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