1 46f79934 2005-01-04 devnull /* Copyright (C) 2003 Russ Cox, Massachusetts Institute of Technology */
 2 46f79934 2005-01-04 devnull /* See COPYRIGHT */
 3 46f79934 2005-01-04 devnull 
 4 46f79934 2005-01-04 devnull #include <u.h>
 5 46f79934 2005-01-04 devnull #include <libc.h>
 6 46f79934 2005-01-04 devnull #include <fcall.h>
 7 46f79934 2005-01-04 devnull #include <9pclient.h>
 8 46f79934 2005-01-04 devnull #include "fsimpl.h"
 9 46f79934 2005-01-04 devnull 
10 46f79934 2005-01-04 devnull static long
11 46f79934 2005-01-04 devnull _fspwrite(CFid *fid, void *buf, long n, vlong offset)
12 46f79934 2005-01-04 devnull {
13 46f79934 2005-01-04 devnull 	Fcall tx, rx;
14 46f79934 2005-01-04 devnull 	void *freep;
15 46f79934 2005-01-04 devnull 
16 46f79934 2005-01-04 devnull 	tx.type = Twrite;
17 46f79934 2005-01-04 devnull 	tx.fid = fid->fid;
18 46f79934 2005-01-04 devnull 	if(offset == -1){
19 46f79934 2005-01-04 devnull 		qlock(&fid->lk);
20 46f79934 2005-01-04 devnull 		tx.offset = fid->offset;
21 46f79934 2005-01-04 devnull 		qunlock(&fid->lk);
22 46f79934 2005-01-04 devnull 	}else
23 46f79934 2005-01-04 devnull 		tx.offset = offset;
24 46f79934 2005-01-04 devnull 	tx.count = n;
25 46f79934 2005-01-04 devnull 	tx.data = buf;
26 46f79934 2005-01-04 devnull 
27 46f79934 2005-01-04 devnull 	if(_fsrpc(fid->fs, &tx, &rx, &freep) < 0)
28 46f79934 2005-01-04 devnull 		return -1;
29 46f79934 2005-01-04 devnull 	if(rx.type == Rerror){
30 46f79934 2005-01-04 devnull 		werrstr("%s", rx.ename);
31 46f79934 2005-01-04 devnull 		free(freep);
32 46f79934 2005-01-04 devnull 		return -1;
33 46f79934 2005-01-04 devnull 	}
34 46f79934 2005-01-04 devnull 	if(offset == -1 && rx.count){
35 46f79934 2005-01-04 devnull 		qlock(&fid->lk);
36 46f79934 2005-01-04 devnull 		fid->offset += rx.count;
37 46f79934 2005-01-04 devnull 		qunlock(&fid->lk);
38 46f79934 2005-01-04 devnull 	}
39 46f79934 2005-01-04 devnull 	free(freep);
40 46f79934 2005-01-04 devnull 	return rx.count;
41 46f79934 2005-01-04 devnull }
42 46f79934 2005-01-04 devnull 
43 46f79934 2005-01-04 devnull long
44 46f79934 2005-01-04 devnull fspwrite(CFid *fid, void *buf, long n, vlong offset)
45 46f79934 2005-01-04 devnull {
46 46f79934 2005-01-04 devnull 	long tot, want, got, first;
47 46f79934 2005-01-04 devnull 	uint msize;
48 46f79934 2005-01-04 devnull 
49 46f79934 2005-01-04 devnull 	msize = fid->fs->msize - IOHDRSZ;
50 46f79934 2005-01-04 devnull 	tot = 0;
51 46f79934 2005-01-04 devnull 	first = 1;
52 46f79934 2005-01-04 devnull 	while(tot < n || first){
53 46f79934 2005-01-04 devnull 		want = n - tot;
54 46f79934 2005-01-04 devnull 		if(want > msize)
55 46f79934 2005-01-04 devnull 			want = msize;
56 22a88b67 2005-03-15 devnull 		got = _fspwrite(fid, (char*)buf+tot, want, offset);
57 46f79934 2005-01-04 devnull 		first = 0;
58 46f79934 2005-01-04 devnull 		if(got < 0){
59 46f79934 2005-01-04 devnull 			if(tot == 0)
60 46f79934 2005-01-04 devnull 				return got;
61 46f79934 2005-01-04 devnull 			break;
62 46f79934 2005-01-04 devnull 		}
63 46f79934 2005-01-04 devnull 		tot += got;
64 46f79934 2005-01-04 devnull 		if(offset != -1)
65 46f79934 2005-01-04 devnull 			offset += got;
66 46f79934 2005-01-04 devnull 	}
67 46f79934 2005-01-04 devnull 	return tot;
68 46f79934 2005-01-04 devnull }
69 46f79934 2005-01-04 devnull 
70 46f79934 2005-01-04 devnull long
71 46f79934 2005-01-04 devnull fswrite(CFid *fid, void *buf, long n)
72 46f79934 2005-01-04 devnull {
73 46f79934 2005-01-04 devnull 	return fspwrite(fid, buf, n, -1);
74 46f79934 2005-01-04 devnull }