xref: /plan9/sys/src/libc/9sys/dirread.c (revision 9a747e4fd48b9f4522c70c07e8f882a15030f964)
13e12c5d1SDavid du Colombier #include <u.h>
23e12c5d1SDavid du Colombier #include <libc.h>
33e12c5d1SDavid du Colombier #include <fcall.h>
43e12c5d1SDavid du Colombier 
5*9a747e4fSDavid du Colombier static
6*9a747e4fSDavid du Colombier long
dirpackage(uchar * buf,long ts,Dir ** d)7*9a747e4fSDavid du Colombier dirpackage(uchar *buf, long ts, Dir **d)
8*9a747e4fSDavid du Colombier {
9*9a747e4fSDavid du Colombier 	char *s;
10*9a747e4fSDavid du Colombier 	long ss, i, n, nn, m;
11*9a747e4fSDavid du Colombier 
12*9a747e4fSDavid du Colombier 	*d = nil;
13*9a747e4fSDavid du Colombier 	if(ts <= 0)
14*9a747e4fSDavid du Colombier 		return 0;
15*9a747e4fSDavid du Colombier 
16*9a747e4fSDavid du Colombier 	/*
17*9a747e4fSDavid du Colombier 	 * first find number of all stats, check they look like stats, & size all associated strings
18*9a747e4fSDavid du Colombier 	 */
19*9a747e4fSDavid du Colombier 	ss = 0;
20*9a747e4fSDavid du Colombier 	n = 0;
21*9a747e4fSDavid du Colombier 	for(i = 0; i < ts; i += m){
22*9a747e4fSDavid du Colombier 		m = BIT16SZ + GBIT16(&buf[i]);
23*9a747e4fSDavid du Colombier 		if(statcheck(&buf[i], m) < 0)
24*9a747e4fSDavid du Colombier 			break;
25*9a747e4fSDavid du Colombier 		ss += m;
26*9a747e4fSDavid du Colombier 		n++;
27*9a747e4fSDavid du Colombier 	}
28*9a747e4fSDavid du Colombier 
29*9a747e4fSDavid du Colombier 	if(i != ts)
30*9a747e4fSDavid du Colombier 		return -1;
31*9a747e4fSDavid du Colombier 
32*9a747e4fSDavid du Colombier 	*d = malloc(n * sizeof(Dir) + ss);
33*9a747e4fSDavid du Colombier 	if(*d == nil)
34*9a747e4fSDavid du Colombier 		return -1;
35*9a747e4fSDavid du Colombier 
36*9a747e4fSDavid du Colombier 	/*
37*9a747e4fSDavid du Colombier 	 * then convert all buffers
38*9a747e4fSDavid du Colombier 	 */
39*9a747e4fSDavid du Colombier 	s = (char*)*d + n * sizeof(Dir);
40*9a747e4fSDavid du Colombier 	nn = 0;
41*9a747e4fSDavid du Colombier 	for(i = 0; i < ts; i += m){
42*9a747e4fSDavid du Colombier 		m = BIT16SZ + GBIT16((uchar*)&buf[i]);
43*9a747e4fSDavid du Colombier 		if(nn >= n || convM2D(&buf[i], m, *d + nn, s) != m){
44*9a747e4fSDavid du Colombier 			free(*d);
45*9a747e4fSDavid du Colombier 			*d = nil;
46*9a747e4fSDavid du Colombier 			return -1;
47*9a747e4fSDavid du Colombier 		}
48*9a747e4fSDavid du Colombier 		nn++;
49*9a747e4fSDavid du Colombier 		s += m;
50*9a747e4fSDavid du Colombier 	}
51*9a747e4fSDavid du Colombier 
52*9a747e4fSDavid du Colombier 	return nn;
53*9a747e4fSDavid du Colombier }
547dd7cddfSDavid du Colombier 
553e12c5d1SDavid du Colombier long
dirread(int fd,Dir ** d)56*9a747e4fSDavid du Colombier dirread(int fd, Dir **d)
573e12c5d1SDavid du Colombier {
58*9a747e4fSDavid du Colombier 	uchar *buf;
59*9a747e4fSDavid du Colombier 	long ts;
603e12c5d1SDavid du Colombier 
61*9a747e4fSDavid du Colombier 	buf = malloc(DIRMAX);
627dd7cddfSDavid du Colombier 	if(buf == nil)
637dd7cddfSDavid du Colombier 		return -1;
64*9a747e4fSDavid du Colombier 	ts = read(fd, buf, DIRMAX);
65*9a747e4fSDavid du Colombier 	if(ts >= 0)
66*9a747e4fSDavid du Colombier 		ts = dirpackage(buf, ts, d);
67*9a747e4fSDavid du Colombier 	free(buf);
68*9a747e4fSDavid du Colombier 	return ts;
69*9a747e4fSDavid du Colombier }
70*9a747e4fSDavid du Colombier 
71*9a747e4fSDavid du Colombier long
dirreadall(int fd,Dir ** d)72*9a747e4fSDavid du Colombier dirreadall(int fd, Dir **d)
73*9a747e4fSDavid du Colombier {
74*9a747e4fSDavid du Colombier 	uchar *buf, *nbuf;
75*9a747e4fSDavid du Colombier 	long n, ts;
76*9a747e4fSDavid du Colombier 
77*9a747e4fSDavid du Colombier 	buf = nil;
78*9a747e4fSDavid du Colombier 	ts = 0;
79*9a747e4fSDavid du Colombier 	for(;;){
80*9a747e4fSDavid du Colombier 		nbuf = realloc(buf, ts+DIRMAX);
81*9a747e4fSDavid du Colombier 		if(nbuf == nil){
827dd7cddfSDavid du Colombier 			free(buf);
833e12c5d1SDavid du Colombier 			return -1;
847dd7cddfSDavid du Colombier 		}
85*9a747e4fSDavid du Colombier 		buf = nbuf;
86*9a747e4fSDavid du Colombier 		n = read(fd, buf+ts, DIRMAX);
87*9a747e4fSDavid du Colombier 		if(n <= 0)
883e12c5d1SDavid du Colombier 			break;
89*9a747e4fSDavid du Colombier 		ts += n;
903e12c5d1SDavid du Colombier 	}
91*9a747e4fSDavid du Colombier 	if(ts >= 0)
92*9a747e4fSDavid du Colombier 		ts = dirpackage(buf, ts, d);
937dd7cddfSDavid du Colombier 	free(buf);
94*9a747e4fSDavid du Colombier 	if(ts == 0 && n < 0)
95*9a747e4fSDavid du Colombier 		return -1;
96*9a747e4fSDavid du Colombier 	return ts;
973e12c5d1SDavid du Colombier }
98