301 lines · 4.5 KB · C
RawDownload
#include "stdinc.h"#include "dat.h"#include "fns.h" char magic[] = "s3venti-1";int magiclen = sizeof(magic); /* * disk structure conversion routines */#define	U8GET(p)	((p)[0])#define	U16GET(p)	(((p)[0]<<8)|(p)[1])#define	U32GET(p)	((u32int)(((p)[0]<<24)|((p)[1]<<16)|((p)[2]<<8)|(p)[3]))#define	U64GET(p)	(((u64int)U32GET(p)<<32)|(u64int)U32GET((p)+4)) #define	U8PUT(p,v)	(p)[0]=(v)&0xFF#define	U16PUT(p,v)	(p)[0]=((v)>>8)&0xFF;(p)[1]=(v)&0xFF#define	U32PUT(p,v)	(p)[0]=((v)>>24)&0xFF;(p)[1]=((v)>>16)&0xFF;(p)[2]=((v)>>8)&0xFF;(p)[3]=(v)&0xFF#define	U64PUT(p,v,t32)	t32=(v)>>32;U32PUT(p,t32);t32=(v);U32PUT((p)+4,t32) u32inthashbits(u8int *sc, int bits){	u32int v; 	v = (sc[0] << 24) | (sc[1] << 16) | (sc[2] << 8) | sc[3];	if(bits < 32)		 v >>= (32 - bits);	return v;} intnamecmp(char *s, char *t){	return strncmp(s, t, ANameSize);} voidnamecp(char *dst, char *src){	strncpy(dst, src, ANameSize - 1);	dst[ANameSize - 1] = '\0';} intnameok(char *name){	char *t;	int c; 	if(name == nil)		return -1;	for(t = name; c = *t; t++)		if(t - name >= ANameSize		|| c < ' ' || c >= 0x7f)			return -1;	return 0;} intstru32int(char *s, u32int *r){	char *t;	u32int n, nn, m;	int c; 	m = TWID32 / 10;	n = 0;	for(t = s; ; t++){		c = *t;		if(c < '0' || c > '9')			break;		if(n > m)			return -1;		nn = n * 10 + c - '0';		if(nn < n)			return -1;		n = nn;	}	*r = n;	return s != t && *t == '\0';} intstru64int(char *s, u64int *r){	char *t;	u64int n, nn, m;	int c; 	m = TWID64 / 10;	n = 0;	for(t = s; ; t++){		c = *t;		if(c < '0' || c > '9')			break;		if(n > m)			return -1;		nn = n * 10 + c - '0';		if(nn < n)			return -1;		n = nn;	}	*r = n;	return s != t && *t == '\0';} intvttypevalid(int type){	return type < VtMaxType;} static char*logit(int severity, char *fmt, va_list args){	char *s; 	s = vsmprint(fmt, args);	if(s == nil)		return nil;	if(severity != EOk){		if(argv0 == nil)			fprint(2, "%T %s: err %d: %s\n", argv0, severity, s);		else			fprint(2, "%T err %d: %s\n", severity, s);	}	return s;} voidseterr(int severity, char *fmt, ...){	char *s;	va_list args; 	va_start(args, fmt);	s = logit(severity, fmt, args);	va_end(args);	if(s == nil)		werrstr("error setting error");	else{		werrstr("%s", s);		free(s);	}} voidlogerr(int severity, char *fmt, ...){	char *s;	va_list args; 	va_start(args, fmt);	s = logit(severity, fmt, args);	va_end(args);	free(s);} u32intnow(void){	return time(nil);} int abortonmem = 1; void *emalloc(ulong n){	void *p; 	p = malloc(n);	if(p == nil){		if(abortonmem)			abort();		sysfatal("out of memory allocating %lud", n);	}	memset(p, 0xa5, n);	setmalloctag(p, getcallerpc(&n));if(0)print("emalloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&n));	return p;} void *ezmalloc(ulong n){	void *p; 	p = malloc(n);	if(p == nil){		if(abortonmem)			abort();		sysfatal("out of memory allocating %lud", n);	}	memset(p, 0, n);	setmalloctag(p, getcallerpc(&n));if(0)print("ezmalloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&n));	return p;} void *erealloc(void *p, ulong n){	p = realloc(p, n);	if(p == nil){		if(abortonmem)			abort();		sysfatal("out of memory allocating %lud", n);	}	setrealloctag(p, getcallerpc(&p));if(0)print("erealloc %p-%p by %lux\n", p, (char*)p+n, getcallerpc(&p));	return p;} char *estrdup(char *s){	char *t;	int n; 	n = strlen(s) + 1;	t = emalloc(n);	memmove(t, s, n);	setmalloctag(t, getcallerpc(&s));if(0)print("estrdup %p-%p by %lux\n", t, (char*)t+n, getcallerpc(&s));	return t;} /* * return floor(log2(v)) */intu64log2(u64int v){	int i; 	for(i = 0; i < 64; i++)		if((v >> i) <= 1)			break;	return i;} intvtproc(void (*fn)(void*), void *arg){	proccreate(fn, arg, 256*1024);	return 0;} voidventifmtinstall(void){	fmtinstall('F', vtfcallfmt);	fmtinstall('H', encodefmt);	fmtinstall('T', vttimefmt);	fmtinstall('V', vtscorefmt);} uintmsec(void){	return nsec()/1000000;} uintcountbits(uint n){	n = (n&0x55555555)+((n>>1)&0x55555555);	n = (n&0x33333333)+((n>>2)&0x33333333);	n = (n&0x0F0F0F0F)+((n>>4)&0x0F0F0F0F);	n = (n&0x00FF00FF)+((n>>8)&0x00FF00FF);	n = (n&0x0000FFFF)+((n>>16)&0x0000FFFF);	return n;} voidvhdrpack(S3Vhdr *h, uchar *p){	U8PUT(p, h->blocktype);	p += U8Size;	U8PUT(p, h->codec);	p += U8Size;	U16PUT(p, h->size);	p += U16Size;	U16PUT(p, h->csize);	p += U16Size;} voidvhdrunpack(S3Vhdr *h, uchar *p){	h->blocktype = U8GET(p);	p += U8Size;	h->codec = U8GET(p);	p += U8Size;	h->size = U16GET(p);	p += U16Size;	h->csize = U16GET(p);	p += U16Size;}