1 a0d146ed 2005-07-12 devnull #ifdef PLAN9PORT	/* SORRY! */
  2 769eb198 2005-07-20 devnull #	include <u.h>
  3 769eb198 2005-07-20 devnull #	include <sys/types.h>
  4 769eb198 2005-07-20 devnull #	ifdef __linux__	/* REALLY SORRY! */
  5 769eb198 2005-07-20 devnull #		define CANBLOCKSIZE 1
  6 769eb198 2005-07-20 devnull #		include <sys/vfs.h>
  7 769eb198 2005-07-20 devnull #	elif defined(__FreeBSD__)
  8 769eb198 2005-07-20 devnull #		define CANBLOCKSIZE 1
  9 769eb198 2005-07-20 devnull #		include <sys/param.h>
 10 ba3c918e 2005-08-11 devnull #		include <sys/stat.h>
 11 769eb198 2005-07-20 devnull #		include <sys/mount.h>
 12 769eb198 2005-07-20 devnull #	endif
 13 a0d146ed 2005-07-12 devnull #endif
 14 a0d146ed 2005-07-12 devnull #include "stdinc.h"
 15 a0d146ed 2005-07-12 devnull #include <ctype.h>
 16 a0d146ed 2005-07-12 devnull #include "dat.h"
 17 a0d146ed 2005-07-12 devnull #include "fns.h"
 18 7e452401 2007-04-27 devnull 
 19 a0d146ed 2005-07-12 devnull u32int	maxblocksize;
 20 a0d146ed 2005-07-12 devnull int	readonly;
 21 a0d146ed 2005-07-12 devnull 
 22 304ab920 2008-07-04 rsc int findsubpart(Part *part, char *name);
 23 304ab920 2008-07-04 rsc 
 24 a0d146ed 2005-07-12 devnull static int
 25 a0d146ed 2005-07-12 devnull strtoullsuf(char *p, char **pp, int rad, u64int *u)
 26 a0d146ed 2005-07-12 devnull {
 27 a0d146ed 2005-07-12 devnull 	u64int v;
 28 a0d146ed 2005-07-12 devnull 
 29 3bd56b04 2005-09-09 devnull 	if(!isdigit((uchar)*p))
 30 a0d146ed 2005-07-12 devnull 		return -1;
 31 a0d146ed 2005-07-12 devnull 	v = strtoull(p, &p, rad);
 32 a0d146ed 2005-07-12 devnull 	switch(*p){
 33 a0d146ed 2005-07-12 devnull 	case 'k':
 34 a0d146ed 2005-07-12 devnull 	case 'K':
 35 a0d146ed 2005-07-12 devnull 		v *= 1024;
 36 a0d146ed 2005-07-12 devnull 		p++;
 37 a0d146ed 2005-07-12 devnull 		break;
 38 a0d146ed 2005-07-12 devnull 	case 'm':
 39 a0d146ed 2005-07-12 devnull 	case 'M':
 40 a0d146ed 2005-07-12 devnull 		v *= 1024*1024;
 41 a0d146ed 2005-07-12 devnull 		p++;
 42 a0d146ed 2005-07-12 devnull 		break;
 43 a0d146ed 2005-07-12 devnull 	case 'g':
 44 a0d146ed 2005-07-12 devnull 	case 'G':
 45 a0d146ed 2005-07-12 devnull 		v *= 1024*1024*1024;
 46 a0d146ed 2005-07-12 devnull 		p++;
 47 a0d146ed 2005-07-12 devnull 		break;
 48 a0d146ed 2005-07-12 devnull 	case 't':
 49 a0d146ed 2005-07-12 devnull 	case 'T':
 50 a0d146ed 2005-07-12 devnull 		v *= 1024*1024;
 51 a0d146ed 2005-07-12 devnull 		v *= 1024*1024;
 52 a0d146ed 2005-07-12 devnull 		p++;
 53 a0d146ed 2005-07-12 devnull 		break;
 54 a0d146ed 2005-07-12 devnull 	}
 55 a0d146ed 2005-07-12 devnull 	*pp = p;
 56 a0d146ed 2005-07-12 devnull 	*u = v;
 57 a0d146ed 2005-07-12 devnull 	return 0;
 58 a0d146ed 2005-07-12 devnull }
 59 fa325e9b 2020-01-10 cross 
 60 a0d146ed 2005-07-12 devnull static int
 61 304ab920 2008-07-04 rsc parsepart(char *name, char **file, char **subpart, u64int *lo, u64int *hi)
 62 a0d146ed 2005-07-12 devnull {
 63 a0d146ed 2005-07-12 devnull 	char *p;
 64 a0d146ed 2005-07-12 devnull 
 65 a0d146ed 2005-07-12 devnull 	*file = estrdup(name);
 66 304ab920 2008-07-04 rsc 	*lo = 0;
 67 304ab920 2008-07-04 rsc 	*hi = 0;
 68 304ab920 2008-07-04 rsc 	*subpart = nil;
 69 304ab920 2008-07-04 rsc 	if((p = strrchr(*file, ':')) == nil)
 70 a0d146ed 2005-07-12 devnull 		return 0;
 71 a0d146ed 2005-07-12 devnull 	*p++ = 0;
 72 304ab920 2008-07-04 rsc 	if(isalpha(*p)){
 73 304ab920 2008-07-04 rsc 		*subpart = p;
 74 304ab920 2008-07-04 rsc 		return 0;
 75 304ab920 2008-07-04 rsc 	}
 76 a0d146ed 2005-07-12 devnull 	if(*p == '-')
 77 a0d146ed 2005-07-12 devnull 		*lo = 0;
 78 a0d146ed 2005-07-12 devnull 	else{
 79 a0d146ed 2005-07-12 devnull 		if(strtoullsuf(p, &p, 0, lo) < 0){
 80 a0d146ed 2005-07-12 devnull 			free(*file);
 81 a0d146ed 2005-07-12 devnull 			return -1;
 82 a0d146ed 2005-07-12 devnull 		}
 83 a0d146ed 2005-07-12 devnull 	}
 84 a0d146ed 2005-07-12 devnull 	if(*p == '-')
 85 a0d146ed 2005-07-12 devnull 		p++;
 86 a0d146ed 2005-07-12 devnull 	if(*p == 0){
 87 a0d146ed 2005-07-12 devnull 		*hi = 0;
 88 a0d146ed 2005-07-12 devnull 		return 0;
 89 a0d146ed 2005-07-12 devnull 	}
 90 a0d146ed 2005-07-12 devnull 	if(strtoullsuf(p, &p, 0, hi) < 0 || *p != 0){
 91 a0d146ed 2005-07-12 devnull 		free(*file);
 92 a0d146ed 2005-07-12 devnull 		return -1;
 93 a0d146ed 2005-07-12 devnull 	}
 94 a0d146ed 2005-07-12 devnull 	return 0;
 95 a0d146ed 2005-07-12 devnull }
 96 a0d146ed 2005-07-12 devnull 
 97 bf930686 2009-08-23 extrudeda #undef min
 98 bf930686 2009-08-23 extrudeda #define min(a, b) ((a) < (b) ? (a) : (b))
 99 a0d146ed 2005-07-12 devnull Part*
100 a0d146ed 2005-07-12 devnull initpart(char *name, int mode)
101 a0d146ed 2005-07-12 devnull {
102 a0d146ed 2005-07-12 devnull 	Part *part;
103 a0d146ed 2005-07-12 devnull 	Dir *dir;
104 304ab920 2008-07-04 rsc 	char *file, *subname;
105 a0d146ed 2005-07-12 devnull 	u64int lo, hi;
106 a0d146ed 2005-07-12 devnull 
107 304ab920 2008-07-04 rsc 	if(parsepart(name, &file, &subname, &lo, &hi) < 0){
108 304ab920 2008-07-04 rsc 		werrstr("cannot parse name %s", name);
109 a0d146ed 2005-07-12 devnull 		return nil;
110 304ab920 2008-07-04 rsc 	}
111 a0d146ed 2005-07-12 devnull 	trace(TraceDisk, "initpart %s file %s lo 0x%llx hi 0x%llx", name, file, lo, hi);
112 a0d146ed 2005-07-12 devnull 	part = MKZ(Part);
113 a0d146ed 2005-07-12 devnull 	part->name = estrdup(name);
114 a0d146ed 2005-07-12 devnull 	part->filename = estrdup(file);
115 a0d146ed 2005-07-12 devnull 	if(readonly){
116 786ce2cf 2007-05-03 devnull 		mode &= ~(OREAD|OWRITE|ORDWR);
117 a0d146ed 2005-07-12 devnull 		mode |= OREAD;
118 a0d146ed 2005-07-12 devnull 	}
119 e46cacb0 2007-04-27 devnull #ifdef __linux__	/* sorry, but linus made O_DIRECT unusable! */
120 e46cacb0 2007-04-27 devnull 	mode &= ~ODIRECT;
121 e46cacb0 2007-04-27 devnull #endif
122 a0d146ed 2005-07-12 devnull 	part->fd = open(file, mode);
123 a0d146ed 2005-07-12 devnull 	if(part->fd < 0){
124 a0d146ed 2005-07-12 devnull 		if((mode&(OREAD|OWRITE|ORDWR)) == ORDWR)
125 a0d146ed 2005-07-12 devnull 			part->fd = open(file, (mode&~ORDWR)|OREAD);
126 a0d146ed 2005-07-12 devnull 		if(part->fd < 0){
127 a0d146ed 2005-07-12 devnull 			freepart(part);
128 a0d146ed 2005-07-12 devnull 			fprint(2, "can't open partition='%s': %r\n", file);
129 a0d146ed 2005-07-12 devnull 			seterr(EOk, "can't open partition='%s': %r", file);
130 a0d146ed 2005-07-12 devnull 			fprint(2, "%r\n");
131 a0d146ed 2005-07-12 devnull 			free(file);
132 a0d146ed 2005-07-12 devnull 			return nil;
133 a0d146ed 2005-07-12 devnull 		}
134 a0d146ed 2005-07-12 devnull 		fprint(2, "warning: %s opened for reading only\n", name);
135 a0d146ed 2005-07-12 devnull 	}
136 a0d146ed 2005-07-12 devnull 	part->offset = lo;
137 c1b1466e 2008-12-06 rsc 	dir = dirfstat(part->fd);
138 a0d146ed 2005-07-12 devnull 	if(dir == nil){
139 a0d146ed 2005-07-12 devnull 		freepart(part);
140 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't stat partition='%s': %r", file);
141 a0d146ed 2005-07-12 devnull 		free(file);
142 a0d146ed 2005-07-12 devnull 		return nil;
143 a0d146ed 2005-07-12 devnull 	}
144 a0d146ed 2005-07-12 devnull 	if(dir->length == 0){
145 a0d146ed 2005-07-12 devnull 		free(dir);
146 afaa1397 2008-12-23 rsc 		dir = dirstat(file);
147 afaa1397 2008-12-23 rsc 		if(dir == nil || dir->length == 0) {
148 afaa1397 2008-12-23 rsc 			freepart(part);
149 afaa1397 2008-12-23 rsc 			seterr(EOk, "can't determine size of partition %s", file);
150 afaa1397 2008-12-23 rsc 			free(file);
151 afaa1397 2008-12-23 rsc 			return nil;
152 afaa1397 2008-12-23 rsc 		}
153 a0d146ed 2005-07-12 devnull 	}
154 a0d146ed 2005-07-12 devnull 	if(dir->length < hi || dir->length < lo){
155 a0d146ed 2005-07-12 devnull 		freepart(part);
156 a0d146ed 2005-07-12 devnull 		seterr(EOk, "partition '%s': bounds out of range (max %lld)", name, dir->length);
157 a0d146ed 2005-07-12 devnull 		free(dir);
158 a0d146ed 2005-07-12 devnull 		free(file);
159 a0d146ed 2005-07-12 devnull 		return nil;
160 a0d146ed 2005-07-12 devnull 	}
161 a0d146ed 2005-07-12 devnull 	if(hi == 0)
162 a0d146ed 2005-07-12 devnull 		hi = dir->length;
163 a0d146ed 2005-07-12 devnull 	part->size = hi - part->offset;
164 0613fe4b 2005-07-19 devnull #ifdef CANBLOCKSIZE
165 a0d146ed 2005-07-12 devnull 	{
166 a0d146ed 2005-07-12 devnull 		struct statfs sfs;
167 afaa1397 2008-12-23 rsc 		if(fstatfs(part->fd, &sfs) >= 0 && sfs.f_bsize > 512)
168 a0d146ed 2005-07-12 devnull 			part->fsblocksize = sfs.f_bsize;
169 a0d146ed 2005-07-12 devnull 	}
170 a0d146ed 2005-07-12 devnull #endif
171 bf930686 2009-08-23 extrudeda 
172 bf930686 2009-08-23 extrudeda 	part->fsblocksize = min(part->fsblocksize, MaxIo);
173 bf930686 2009-08-23 extrudeda 
174 304ab920 2008-07-04 rsc 	if(subname && findsubpart(part, subname) < 0){
175 304ab920 2008-07-04 rsc 		werrstr("cannot find subpartition %s", subname);
176 304ab920 2008-07-04 rsc 		freepart(part);
177 304ab920 2008-07-04 rsc 		return nil;
178 304ab920 2008-07-04 rsc 	}
179 a0d146ed 2005-07-12 devnull 	free(dir);
180 a0d146ed 2005-07-12 devnull 	return part;
181 a0d146ed 2005-07-12 devnull }
182 a0d146ed 2005-07-12 devnull 
183 e46cacb0 2007-04-27 devnull int
184 7e452401 2007-04-27 devnull flushpart(Part *part)
185 7e452401 2007-04-27 devnull {
186 e46cacb0 2007-04-27 devnull 	USED(part);
187 e46cacb0 2007-04-27 devnull #ifdef __linux__	/* grrr! */
188 e46cacb0 2007-04-27 devnull 	if(fsync(part->fd) < 0){
189 e46cacb0 2007-04-27 devnull 		logerr(EAdmin, "flushpart %s: %r", part->name);
190 e46cacb0 2007-04-27 devnull 		return -1;
191 e46cacb0 2007-04-27 devnull 	}
192 e46cacb0 2007-04-27 devnull 	posix_fadvise(part->fd, 0, 0, POSIX_FADV_DONTNEED);
193 e46cacb0 2007-04-27 devnull #endif
194 e46cacb0 2007-04-27 devnull 	return 0;
195 7e452401 2007-04-27 devnull }
196 7e452401 2007-04-27 devnull 
197 7e452401 2007-04-27 devnull void
198 a0d146ed 2005-07-12 devnull freepart(Part *part)
199 a0d146ed 2005-07-12 devnull {
200 a0d146ed 2005-07-12 devnull 	if(part == nil)
201 a0d146ed 2005-07-12 devnull 		return;
202 a0d146ed 2005-07-12 devnull 	if(part->fd >= 0)
203 a0d146ed 2005-07-12 devnull 		close(part->fd);
204 a0d146ed 2005-07-12 devnull 	free(part->name);
205 a0d146ed 2005-07-12 devnull 	free(part);
206 a0d146ed 2005-07-12 devnull }
207 a0d146ed 2005-07-12 devnull 
208 a0d146ed 2005-07-12 devnull void
209 a0d146ed 2005-07-12 devnull partblocksize(Part *part, u32int blocksize)
210 a0d146ed 2005-07-12 devnull {
211 a0d146ed 2005-07-12 devnull 	if(part->blocksize)
212 a0d146ed 2005-07-12 devnull 		sysfatal("resetting partition=%s's block size", part->name);
213 a0d146ed 2005-07-12 devnull 	part->blocksize = blocksize;
214 a0d146ed 2005-07-12 devnull 	if(blocksize > maxblocksize)
215 a0d146ed 2005-07-12 devnull 		maxblocksize = blocksize;
216 a0d146ed 2005-07-12 devnull }
217 a0d146ed 2005-07-12 devnull 
218 a0d146ed 2005-07-12 devnull /*
219 a0d146ed 2005-07-12 devnull  * Read/write some amount of data between a block device or file and a memory buffer.
220 a0d146ed 2005-07-12 devnull  *
221 a0d146ed 2005-07-12 devnull  * Most Unix systems require that when accessing a block device directly,
222 a0d146ed 2005-07-12 devnull  * the buffer, offset, and count are all multiples of the device block size,
223 a0d146ed 2005-07-12 devnull  * making this a lot more complicated than it otherwise would be.
224 fa325e9b 2020-01-10 cross  *
225 a0d146ed 2005-07-12 devnull  * Most of our callers will make things easy on us, but for some callers it's best
226 a0d146ed 2005-07-12 devnull  * if we just do the work here, with only one place to get it right (hopefully).
227 fa325e9b 2020-01-10 cross  *
228 fa325e9b 2020-01-10 cross  * If everything is aligned properly, prwb will try to do big transfers in the main
229 a0d146ed 2005-07-12 devnull  * body of the loop: up to MaxIo bytes at a time.  If everything isn't aligned properly,
230 a0d146ed 2005-07-12 devnull  * we work one block at a time.
231 a0d146ed 2005-07-12 devnull  */
232 a0d146ed 2005-07-12 devnull int
233 a0d146ed 2005-07-12 devnull prwb(char *name, int fd, int isread, u64int offset, void *vbuf, u32int count, u32int blocksize)
234 a0d146ed 2005-07-12 devnull {
235 a0d146ed 2005-07-12 devnull 	char *op;
236 27d28098 2007-04-21 devnull 	u8int *buf, *freetmp, *dst;
237 27d28098 2007-04-21 devnull 	u32int icount, opsize;
238 afaa1397 2008-12-23 rsc 	int r, count1;
239 a0d146ed 2005-07-12 devnull 
240 28b49df3 2006-07-18 devnull 
241 28b49df3 2006-07-18 devnull #ifndef PLAN9PORT
242 27d28098 2007-04-21 devnull 	USED(blocksize);
243 27d28098 2007-04-21 devnull 	icount = count;
244 27d28098 2007-04-21 devnull 	buf = vbuf;
245 28b49df3 2006-07-18 devnull 	op = isread ? "read" : "write";
246 28b49df3 2006-07-18 devnull 	dst = buf;
247 28b49df3 2006-07-18 devnull 	freetmp = nil;
248 28b49df3 2006-07-18 devnull 	while(count > 0){
249 28b49df3 2006-07-18 devnull 		opsize = min(count, 131072 /* blocksize */);
250 28b49df3 2006-07-18 devnull 		if(isread)
251 28b49df3 2006-07-18 devnull 			r = pread(fd, dst, opsize, offset);
252 28b49df3 2006-07-18 devnull 		else
253 28b49df3 2006-07-18 devnull 			r = pwrite(fd, dst, opsize, offset);
254 28b49df3 2006-07-18 devnull 		if(r <= 0)
255 28b49df3 2006-07-18 devnull 			goto Error;
256 28b49df3 2006-07-18 devnull 		offset += r;
257 28b49df3 2006-07-18 devnull 		count -= r;
258 28b49df3 2006-07-18 devnull 		dst += r;
259 28b49df3 2006-07-18 devnull 		if(r != opsize)
260 28b49df3 2006-07-18 devnull 			goto Error;
261 28b49df3 2006-07-18 devnull 	}
262 28b49df3 2006-07-18 devnull 	return icount;
263 27d28098 2007-04-21 devnull #else
264 27d28098 2007-04-21 devnull 	u32int c, delta;
265 27d28098 2007-04-21 devnull 	u8int *tmp;
266 28b49df3 2006-07-18 devnull 
267 27d28098 2007-04-21 devnull 	icount = count;
268 27d28098 2007-04-21 devnull 	buf = vbuf;
269 a0d146ed 2005-07-12 devnull 	tmp = nil;
270 a0d146ed 2005-07-12 devnull 	freetmp = nil;
271 a0d146ed 2005-07-12 devnull 	opsize = blocksize;
272 a0d146ed 2005-07-12 devnull 
273 a0d146ed 2005-07-12 devnull 	if(count == 0){
274 a0d146ed 2005-07-12 devnull 		logerr(EStrange, "pwrb %s called to %s 0 bytes", name, isread ? "read" : "write");
275 a0d146ed 2005-07-12 devnull 		return 0;
276 a0d146ed 2005-07-12 devnull 	}
277 a0d146ed 2005-07-12 devnull 
278 a0d146ed 2005-07-12 devnull 	assert(blocksize > 0);
279 a0d146ed 2005-07-12 devnull 
280 a0d146ed 2005-07-12 devnull 	/* allocate blocksize-aligned temp buffer if needed */
281 a0d146ed 2005-07-12 devnull 	if((ulong)offset%blocksize || (ulong)buf%blocksize || count%blocksize){
282 a0d146ed 2005-07-12 devnull 		if((freetmp = malloc(blocksize*2)) == nil)
283 a0d146ed 2005-07-12 devnull 			return -1;
284 a0d146ed 2005-07-12 devnull 		tmp = freetmp;
285 a0d146ed 2005-07-12 devnull 		tmp += blocksize - (ulong)tmp%blocksize;
286 a0d146ed 2005-07-12 devnull 	}
287 a0d146ed 2005-07-12 devnull 
288 a0d146ed 2005-07-12 devnull 	/* handle beginning fringe */
289 a0d146ed 2005-07-12 devnull 	if((delta = (ulong)offset%blocksize) != 0){
290 a0d146ed 2005-07-12 devnull 		assert(tmp != nil);
291 a0d146ed 2005-07-12 devnull 		if((r=pread(fd, tmp, blocksize, offset-delta)) != blocksize){
292 a0d146ed 2005-07-12 devnull 			dst = tmp;
293 a0d146ed 2005-07-12 devnull 			offset = offset-delta;
294 a0d146ed 2005-07-12 devnull 			op = "read";
295 afaa1397 2008-12-23 rsc 			count1 = blocksize;
296 a0d146ed 2005-07-12 devnull 			goto Error;
297 a0d146ed 2005-07-12 devnull 		}
298 a0d146ed 2005-07-12 devnull 		c = min(count, blocksize-delta);
299 a0d146ed 2005-07-12 devnull 		assert(c > 0 && c < blocksize);
300 a0d146ed 2005-07-12 devnull 		if(isread)
301 a0d146ed 2005-07-12 devnull 			memmove(buf, tmp+delta, c);
302 a0d146ed 2005-07-12 devnull 		else{
303 a0d146ed 2005-07-12 devnull 			memmove(tmp+delta, buf, c);
304 a0d146ed 2005-07-12 devnull 			if((r=pwrite(fd, tmp, blocksize, offset-delta)) != blocksize){
305 a0d146ed 2005-07-12 devnull 				dst = tmp;
306 a0d146ed 2005-07-12 devnull 				offset = offset-delta;
307 a0d146ed 2005-07-12 devnull 				op = "read";
308 afaa1397 2008-12-23 rsc 				count1 = blocksize;
309 a0d146ed 2005-07-12 devnull 				goto Error;
310 a0d146ed 2005-07-12 devnull 			}
311 a0d146ed 2005-07-12 devnull 		}
312 a0d146ed 2005-07-12 devnull 		assert(c > 0);
313 a0d146ed 2005-07-12 devnull 		offset += c;
314 a0d146ed 2005-07-12 devnull 		buf += c;
315 a0d146ed 2005-07-12 devnull 		count -= c;
316 a0d146ed 2005-07-12 devnull 	}
317 a0d146ed 2005-07-12 devnull 
318 a0d146ed 2005-07-12 devnull 	/* handle full blocks */
319 a0d146ed 2005-07-12 devnull 	while(count >= blocksize){
320 a0d146ed 2005-07-12 devnull 		assert((ulong)offset%blocksize == 0);
321 a0d146ed 2005-07-12 devnull 		if((ulong)buf%blocksize){
322 a0d146ed 2005-07-12 devnull 			assert(tmp != nil);
323 a0d146ed 2005-07-12 devnull 			dst = tmp;
324 a0d146ed 2005-07-12 devnull 			opsize = blocksize;
325 a0d146ed 2005-07-12 devnull 		}else{
326 a0d146ed 2005-07-12 devnull 			dst = buf;
327 a0d146ed 2005-07-12 devnull 			opsize = count - count%blocksize;
328 a0d146ed 2005-07-12 devnull 			if(opsize > MaxIo)
329 a0d146ed 2005-07-12 devnull 				opsize = MaxIo;
330 a0d146ed 2005-07-12 devnull 		}
331 a0d146ed 2005-07-12 devnull 		if(isread){
332 a0d146ed 2005-07-12 devnull 			if((r=pread(fd, dst, opsize, offset))<=0 || r%blocksize){
333 a0d146ed 2005-07-12 devnull 				op = "read";
334 afaa1397 2008-12-23 rsc 				count1 = opsize;
335 a0d146ed 2005-07-12 devnull 				goto Error;
336 a0d146ed 2005-07-12 devnull 			}
337 a0d146ed 2005-07-12 devnull 			if(dst == tmp){
338 a0d146ed 2005-07-12 devnull 				assert(r == blocksize);
339 a0d146ed 2005-07-12 devnull 				memmove(buf, tmp, blocksize);
340 a0d146ed 2005-07-12 devnull 			}
341 a0d146ed 2005-07-12 devnull 		}else{
342 a0d146ed 2005-07-12 devnull 			if(dst == tmp){
343 a0d146ed 2005-07-12 devnull 				assert(opsize == blocksize);
344 a0d146ed 2005-07-12 devnull 				memmove(dst, buf, blocksize);
345 a0d146ed 2005-07-12 devnull 			}
346 a0d146ed 2005-07-12 devnull 			if((r=pwrite(fd, dst, opsize, offset))<=0 || r%blocksize){
347 afaa1397 2008-12-23 rsc 				count1 = opsize;
348 a0d146ed 2005-07-12 devnull 				op = "write";
349 a0d146ed 2005-07-12 devnull 				goto Error;
350 a0d146ed 2005-07-12 devnull 			}
351 a0d146ed 2005-07-12 devnull 			if(dst == tmp)
352 a0d146ed 2005-07-12 devnull 				assert(r == blocksize);
353 a0d146ed 2005-07-12 devnull 		}
354 a0d146ed 2005-07-12 devnull 		assert(r > 0);
355 a0d146ed 2005-07-12 devnull 		offset += r;
356 a0d146ed 2005-07-12 devnull 		buf += r;
357 a0d146ed 2005-07-12 devnull 		count -= r;
358 a0d146ed 2005-07-12 devnull 	}
359 a0d146ed 2005-07-12 devnull 
360 a0d146ed 2005-07-12 devnull 	/* handle ending fringe */
361 a0d146ed 2005-07-12 devnull 	if(count > 0){
362 a0d146ed 2005-07-12 devnull 		assert((ulong)offset%blocksize == 0);
363 a0d146ed 2005-07-12 devnull 		assert(tmp != nil);
364 a0d146ed 2005-07-12 devnull 		/*
365 a0d146ed 2005-07-12 devnull 		 * Complicated condition: if we're reading it's okay to get less than
366 a0d146ed 2005-07-12 devnull 		 * a block as long as it's enough to satisfy the read - maybe this is
367 a0d146ed 2005-07-12 devnull 		 * a normal file.  (We never write to normal files, or else things would
368 a0d146ed 2005-07-12 devnull 		 * be even more complicated.)
369 a0d146ed 2005-07-12 devnull 		 */
370 a0d146ed 2005-07-12 devnull 		r = pread(fd, tmp, blocksize, offset);
371 a0d146ed 2005-07-12 devnull 		if((isread && r < count) || (!isread && r != blocksize)){
372 a0d146ed 2005-07-12 devnull print("FAILED isread=%d r=%d count=%d blocksize=%d\n", isread, r, count, blocksize);
373 a0d146ed 2005-07-12 devnull 			dst = tmp;
374 a0d146ed 2005-07-12 devnull 			op = "read";
375 afaa1397 2008-12-23 rsc 			count1 = blocksize;
376 a0d146ed 2005-07-12 devnull 			goto Error;
377 a0d146ed 2005-07-12 devnull 		}
378 a0d146ed 2005-07-12 devnull 		if(isread)
379 a0d146ed 2005-07-12 devnull 			memmove(buf, tmp, count);
380 a0d146ed 2005-07-12 devnull 		else{
381 a0d146ed 2005-07-12 devnull 			memmove(tmp, buf, count);
382 afaa1397 2008-12-23 rsc 			if(pwrite(fd, tmp, blocksize, offset) != blocksize){
383 a0d146ed 2005-07-12 devnull 				dst = tmp;
384 afaa1397 2008-12-23 rsc 				count1 = blocksize;
385 a0d146ed 2005-07-12 devnull 				op = "write";
386 a0d146ed 2005-07-12 devnull 				goto Error;
387 a0d146ed 2005-07-12 devnull 			}
388 a0d146ed 2005-07-12 devnull 		}
389 a0d146ed 2005-07-12 devnull 	}
390 a0d146ed 2005-07-12 devnull 	if(freetmp)
391 a0d146ed 2005-07-12 devnull 		free(freetmp);
392 a0d146ed 2005-07-12 devnull 	return icount;
393 27d28098 2007-04-21 devnull #endif
394 a0d146ed 2005-07-12 devnull 
395 a0d146ed 2005-07-12 devnull Error:
396 a0d146ed 2005-07-12 devnull 	seterr(EAdmin, "%s %s offset 0x%llux count %ud buf %p returned %d: %r",
397 afaa1397 2008-12-23 rsc 		op, name, offset, count1, dst, r);
398 a0d146ed 2005-07-12 devnull 	if(freetmp)
399 a0d146ed 2005-07-12 devnull 		free(freetmp);
400 a0d146ed 2005-07-12 devnull 	return -1;
401 a0d146ed 2005-07-12 devnull }
402 a0d146ed 2005-07-12 devnull 
403 28b49df3 2006-07-18 devnull #ifndef PLAN9PORT
404 28b49df3 2006-07-18 devnull static int sdreset(Part*);
405 28b49df3 2006-07-18 devnull static int reopen(Part*);
406 28b49df3 2006-07-18 devnull static int threadspawnl(int[3], char*, char*, ...);
407 28b49df3 2006-07-18 devnull #endif
408 28b49df3 2006-07-18 devnull 
409 a0d146ed 2005-07-12 devnull int
410 a0d146ed 2005-07-12 devnull rwpart(Part *part, int isread, u64int offset, u8int *buf, u32int count)
411 a0d146ed 2005-07-12 devnull {
412 28b49df3 2006-07-18 devnull 	int n, try;
413 a0d146ed 2005-07-12 devnull 	u32int blocksize;
414 a0d146ed 2005-07-12 devnull 
415 fa325e9b 2020-01-10 cross 	trace(TraceDisk, "%s %s %ud at 0x%llx",
416 a0d146ed 2005-07-12 devnull 		isread ? "read" : "write", part->name, count, offset);
417 a0d146ed 2005-07-12 devnull 	if(offset >= part->size || offset+count > part->size){
418 a0d146ed 2005-07-12 devnull 		seterr(EStrange, "out of bounds %s offset 0x%llux count %ud to partition %s size 0x%llux",
419 a0d146ed 2005-07-12 devnull 			isread ? "read" : "write", offset, count, part->name, part->size);
420 a0d146ed 2005-07-12 devnull 		return -1;
421 a0d146ed 2005-07-12 devnull 	}
422 a0d146ed 2005-07-12 devnull 
423 a0d146ed 2005-07-12 devnull 	blocksize = part->fsblocksize;
424 a0d146ed 2005-07-12 devnull 	if(blocksize == 0)
425 a0d146ed 2005-07-12 devnull 		blocksize = part->blocksize;
426 a0d146ed 2005-07-12 devnull 	if(blocksize == 0)
427 a0d146ed 2005-07-12 devnull 		blocksize = 4096;
428 a0d146ed 2005-07-12 devnull 
429 28b49df3 2006-07-18 devnull 	for(try=0;; try++){
430 28b49df3 2006-07-18 devnull 		n = prwb(part->filename, part->fd, isread, part->offset+offset, buf, count, blocksize);
431 28b49df3 2006-07-18 devnull 		if(n >= 0 || try > 10)
432 28b49df3 2006-07-18 devnull 			break;
433 a0d146ed 2005-07-12 devnull 
434 28b49df3 2006-07-18 devnull #ifndef PLAN9PORT
435 28b49df3 2006-07-18 devnull 	    {
436 28b49df3 2006-07-18 devnull 		char err[ERRMAX];
437 28b49df3 2006-07-18 devnull 		/*
438 28b49df3 2006-07-18 devnull 		 * This happens with the sdmv disks frustratingly often.
439 28b49df3 2006-07-18 devnull 		 * Try to fix things up and continue.
440 28b49df3 2006-07-18 devnull 		 */
441 28b49df3 2006-07-18 devnull 		rerrstr(err, sizeof err);
442 7e452401 2007-04-27 devnull 		if(strstr(err, "i/o timeout") || strstr(err, "i/o error") || strstr(err, "partition has changed")){
443 28b49df3 2006-07-18 devnull 			reopen(part);
444 28b49df3 2006-07-18 devnull 			continue;
445 28b49df3 2006-07-18 devnull 		}
446 28b49df3 2006-07-18 devnull 	    }
447 28b49df3 2006-07-18 devnull #endif
448 28b49df3 2006-07-18 devnull 		break;
449 28b49df3 2006-07-18 devnull 	}
450 e46cacb0 2007-04-27 devnull #ifdef __linux__	/* sigh */
451 e46cacb0 2007-04-27 devnull 	posix_fadvise(part->fd, part->offset+offset, n, POSIX_FADV_DONTNEED);
452 e46cacb0 2007-04-27 devnull #endif
453 28b49df3 2006-07-18 devnull 	return n;
454 28b49df3 2006-07-18 devnull }
455 a0d146ed 2005-07-12 devnull int
456 a0d146ed 2005-07-12 devnull readpart(Part *part, u64int offset, u8int *buf, u32int count)
457 a0d146ed 2005-07-12 devnull {
458 a0d146ed 2005-07-12 devnull 	return rwpart(part, 1, offset, buf, count);
459 a0d146ed 2005-07-12 devnull }
460 a0d146ed 2005-07-12 devnull 
461 a0d146ed 2005-07-12 devnull int
462 a0d146ed 2005-07-12 devnull writepart(Part *part, u64int offset, u8int *buf, u32int count)
463 a0d146ed 2005-07-12 devnull {
464 a0d146ed 2005-07-12 devnull 	return rwpart(part, 0, offset, buf, count);
465 a0d146ed 2005-07-12 devnull }
466 a0d146ed 2005-07-12 devnull 
467 a0d146ed 2005-07-12 devnull ZBlock*
468 a0d146ed 2005-07-12 devnull readfile(char *name)
469 a0d146ed 2005-07-12 devnull {
470 a0d146ed 2005-07-12 devnull 	Part *p;
471 a0d146ed 2005-07-12 devnull 	ZBlock *b;
472 a0d146ed 2005-07-12 devnull 
473 a0d146ed 2005-07-12 devnull 	p = initpart(name, OREAD);
474 a0d146ed 2005-07-12 devnull 	if(p == nil)
475 a0d146ed 2005-07-12 devnull 		return nil;
476 a0d146ed 2005-07-12 devnull 	b = alloczblock(p->size, 0, p->blocksize);
477 a0d146ed 2005-07-12 devnull 	if(b == nil){
478 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't alloc %s: %r", name);
479 a0d146ed 2005-07-12 devnull 		freepart(p);
480 a0d146ed 2005-07-12 devnull 		return nil;
481 a0d146ed 2005-07-12 devnull 	}
482 a0d146ed 2005-07-12 devnull 	if(readpart(p, 0, b->data, p->size) < 0){
483 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't read %s: %r", name);
484 a0d146ed 2005-07-12 devnull 		freepart(p);
485 a0d146ed 2005-07-12 devnull 		freezblock(b);
486 a0d146ed 2005-07-12 devnull 		return nil;
487 a0d146ed 2005-07-12 devnull 	}
488 a0d146ed 2005-07-12 devnull 	freepart(p);
489 a0d146ed 2005-07-12 devnull 	return b;
490 28b49df3 2006-07-18 devnull }
491 304ab920 2008-07-04 rsc 
492 304ab920 2008-07-04 rsc /*
493 304ab920 2008-07-04 rsc  * Search for the Plan 9 partition with the given name.
494 fa325e9b 2020-01-10 cross  * This lets you write things like /dev/ad4:arenas
495 304ab920 2008-07-04 rsc  * if you move a disk from a Plan 9 system to a FreeBSD system.
496 304ab920 2008-07-04 rsc  *
497 304ab920 2008-07-04 rsc  * God I hope I never write this code again.
498 304ab920 2008-07-04 rsc  */
499 304ab920 2008-07-04 rsc #define MAGIC "plan9 partitions"
500 304ab920 2008-07-04 rsc static int
501 304ab920 2008-07-04 rsc tryplan9part(Part *part, char *name)
502 304ab920 2008-07-04 rsc {
503 304ab920 2008-07-04 rsc 	uchar buf[512];
504 304ab920 2008-07-04 rsc 	char *line[40], *f[4];
505 304ab920 2008-07-04 rsc 	int i, n;
506 304ab920 2008-07-04 rsc 	vlong start, end;
507 28b49df3 2006-07-18 devnull 
508 304ab920 2008-07-04 rsc 	/*
509 304ab920 2008-07-04 rsc 	 * Partition table in second sector.
510 304ab920 2008-07-04 rsc 	 * Could also look on 2nd last sector and last sector,
511 304ab920 2008-07-04 rsc 	 * but those disks died out long before venti came along.
512 304ab920 2008-07-04 rsc 	 */
513 304ab920 2008-07-04 rsc 	if(readpart(part, 512, buf, 512) != 512)
514 304ab920 2008-07-04 rsc 		return -1;
515 28b49df3 2006-07-18 devnull 
516 304ab920 2008-07-04 rsc 	/* Plan 9 partition table is just text strings */
517 304ab920 2008-07-04 rsc 	if(strncmp((char*)buf, "part ", 5) != 0)
518 304ab920 2008-07-04 rsc 		return -1;
519 28b49df3 2006-07-18 devnull 
520 304ab920 2008-07-04 rsc 	buf[511] = 0;
521 304ab920 2008-07-04 rsc 	n = getfields((char*)buf, line, 40, 1, "\n");
522 304ab920 2008-07-04 rsc 	for(i=0; i<n; i++){
523 304ab920 2008-07-04 rsc 		if(getfields(line[i], f, 4, 1, " ") != 4)
524 304ab920 2008-07-04 rsc 			break;
525 304ab920 2008-07-04 rsc 		if(strcmp(f[0], "part") != 0)
526 304ab920 2008-07-04 rsc 			break;
527 304ab920 2008-07-04 rsc 		if(strcmp(f[1], name) == 0){
528 304ab920 2008-07-04 rsc 			start = 512*strtoll(f[2], 0, 0);
529 304ab920 2008-07-04 rsc 			end = 512*strtoll(f[3], 0, 0);
530 304ab920 2008-07-04 rsc 			if(start  < end && end <= part->size){
531 304ab920 2008-07-04 rsc 				part->offset += start;
532 304ab920 2008-07-04 rsc 				part->size = end - start;
533 304ab920 2008-07-04 rsc 				return 0;
534 304ab920 2008-07-04 rsc 			}
535 304ab920 2008-07-04 rsc 			return -1;
536 304ab920 2008-07-04 rsc 		}
537 304ab920 2008-07-04 rsc 	}
538 304ab920 2008-07-04 rsc 	return -1;
539 304ab920 2008-07-04 rsc }
540 28b49df3 2006-07-18 devnull 
541 304ab920 2008-07-04 rsc #define	GSHORT(p)	(((p)[1]<<8)|(p)[0])
542 304ab920 2008-07-04 rsc #define	GLONG(p)	((GSHORT(p+2)<<16)|GSHORT(p))
543 28b49df3 2006-07-18 devnull 
544 304ab920 2008-07-04 rsc typedef struct Dospart Dospart;
545 304ab920 2008-07-04 rsc struct Dospart
546 304ab920 2008-07-04 rsc {
547 304ab920 2008-07-04 rsc 	uchar flag;		/* active flag */
548 304ab920 2008-07-04 rsc 	uchar shead;		/* starting head */
549 304ab920 2008-07-04 rsc 	uchar scs[2];		/* starting cylinder/sector */
550 304ab920 2008-07-04 rsc 	uchar type;		/* partition type */
551 304ab920 2008-07-04 rsc 	uchar ehead;		/* ending head */
552 304ab920 2008-07-04 rsc 	uchar ecs[2];		/* ending cylinder/sector */
553 304ab920 2008-07-04 rsc 	uchar offset[4];		/* starting sector */
554 304ab920 2008-07-04 rsc 	uchar size[4];		/* length in sectors */
555 304ab920 2008-07-04 rsc };
556 28b49df3 2006-07-18 devnull 
557 28b49df3 2006-07-18 devnull 
558 304ab920 2008-07-04 rsc int
559 304ab920 2008-07-04 rsc findsubpart(Part *part, char *name)
560 28b49df3 2006-07-18 devnull {
561 304ab920 2008-07-04 rsc 	int i;
562 304ab920 2008-07-04 rsc 	uchar buf[512];
563 304ab920 2008-07-04 rsc 	u64int size;
564 304ab920 2008-07-04 rsc 	Dospart *dp;
565 304ab920 2008-07-04 rsc 
566 304ab920 2008-07-04 rsc 	/* See if this is a Plan 9 partition. */
567 304ab920 2008-07-04 rsc 	if(tryplan9part(part, name) >= 0)
568 304ab920 2008-07-04 rsc 		return 0;
569 fa325e9b 2020-01-10 cross 
570 304ab920 2008-07-04 rsc 	/* Otherwise try for an MBR and then narrow to Plan 9 partition. */
571 304ab920 2008-07-04 rsc 	if(readpart(part, 0, buf, 512) != 512)
572 28b49df3 2006-07-18 devnull 		return -1;
573 304ab920 2008-07-04 rsc 	if(buf[0x1FE] != 0x55 || buf[0x1FF] != 0xAA)
574 304ab920 2008-07-04 rsc 		return -1;
575 304ab920 2008-07-04 rsc 	dp = (Dospart*)(buf+0x1BE);
576 304ab920 2008-07-04 rsc 	size = part->size;
577 304ab920 2008-07-04 rsc 	for(i=0; i<4; i++){
578 304ab920 2008-07-04 rsc 		if(dp[i].type == '9'){
579 304ab920 2008-07-04 rsc 			part->offset = 512LL*GLONG(dp[i].offset);
580 304ab920 2008-07-04 rsc 			part->size = 512LL*GLONG(dp[i].size);
581 304ab920 2008-07-04 rsc 			if(tryplan9part(part, name) >= 0)
582 304ab920 2008-07-04 rsc 				return 0;
583 304ab920 2008-07-04 rsc 			part->offset = 0;
584 304ab920 2008-07-04 rsc 			part->size = size;
585 304ab920 2008-07-04 rsc 		}
586 304ab920 2008-07-04 rsc 		/* Not implementing extended partitions - enough is enough. */
587 28b49df3 2006-07-18 devnull 	}
588 304ab920 2008-07-04 rsc 	return -1;
589 a0d146ed 2005-07-12 devnull }