9p: fix writen (sqweek)

1 files changed, 13 insertions, 8 deletions

diff --git a/src/cmd/9p.c b/src/cmd/9p.c
index 689e00ee..135d9c50 100644
--- a/src/cmd/9p.c
+++ b/src/cmd/9p.c
@@ -28,16 +28,21 @@ usage(void)
 int
 writen(int fd, void *buf, int n)
 {
-	long m, tot;
+	int m, tot;
 
-	for(tot=0; tot<n; tot+=m){
-		m = n - tot;
-		if(m > 8192)
-			m = 8192;
-		if(write(fd, (uchar*)buf+tot, m) != m)
-			break;
+	if(n < 0){
+		werrstr("bad count");
+		return -1;
 	}
-	return tot;
+	if(n == 0)
+		return 0;
+	
+	tot = 0;
+	while((m = write(fd, (char*)buf+tot, n-tot)) > 0)
+		tot += m;
+	if(tot < n)
+		return -1;
+	return n;
 }
 
 CFsys *(*nsmnt)(char*, char*) = nsamount;