shithub: libtags

ref: 19f590de69c3f287c128ecd6e58e4ebf9a24a24f
dir: /wav.c/

#include "tagspriv.h"

#define le16u(d) (u16int)((d)[0] | (d)[1]<<8)

static struct {
	char *s;
	int type;
}t[] = {
	{"IART", Tartist},
	{"ICRD", Tdate},
	{"IGNR", Tgenre},
	{"INAM", Ttitle},
	{"IPRD", Talbum},
	{"ITRK", Ttrack},
};

int
tagwav(Tagctx *ctx)
{
	uchar *d;
	int i, n, info;
	u32int csz, x;
	uvlong sz;

	d = (uchar*)ctx->buf;

	sz = 1;
	info = 0;
	for(i = 0; i < 8 && sz > 0; i++){
		if(ctx->read(ctx, d, 4+4+(i?0:4)) != 4+4+(i?0:4))
			return -1;
		if(i == 0){
			if(memcmp(d, "RIFF", 4) != 0 || memcmp(d+8, "WAVE", 4) != 0)
				return -1;
			sz = leuint(d+4);
			if(sz < 4)
				return -1;
			sz -= 4;
			continue;
		}else if(memcmp(d, "INFO", 4) == 0){
			info = 1;
			ctx->seek(ctx, -4, 1);
			continue;
		}

		if(sz <= 8)
			break;
		sz -= 4+4;
		csz = leuint(d+4);
		if(sz < csz)
			break;
		sz -= csz;

		if(i == 1){
			if(memcmp(d, "fmt ", 4) != 0 || csz < 16)
				return -1;
			if(ctx->read(ctx, d, 16) != 16)
				return -1;
			csz -= 16;
			ctx->channels = le16u(d+2);
			ctx->samplerate = leuint(d+4);
			x = leuint(d+8);
			if(ctx->channels < 1 || ctx->samplerate < 1 || x < 1)
				return -1;
			ctx->duration = sz*1000 / x;
		}else if(memcmp(d, "LIST", 4) == 0){
			sz = csz - 4;
			continue;
		}else if(memcmp(d, "data", 4) == 0){
			break;
		}else if(info){
			csz++;
			for(n = 0; n < nelem(t); n++){
				if(memcmp(d, t[n].s, 4) == 0){
					if(ctx->read(ctx, d, csz) != (int)csz)
						return -1;
					d[csz-1] = 0;
					txtcb(ctx, t[n].type, "", d);
					csz = 0;
					break;
				}
			}
		}

		if(ctx->seek(ctx, csz, 1) < 0)
			return -1;
	}

	return i > 0 ? 0 : -1;
}