twritefile.c - plan9port - [fork] Plan 9 from user space
  HTML git clone git://src.adamsgaard.dk/plan9port
   DIR Log
   DIR Files
   DIR Refs
   DIR README
   DIR LICENSE
       ---
       twritefile.c (1893B)
       ---
            1 #include <u.h>
            2 #include <libc.h>
            3 #include <venti.h>
            4 #include <libsec.h>
            5 #include <thread.h>
            6 
            7 enum
            8 {
            9         Blocksize = 8192
           10 };
           11 
           12 int chatty;
           13 
           14 void
           15 usage(void)
           16 {
           17         fprint(2, "usage: writefile [-v] [-h host] < data\n");
           18         threadexitsall("usage");
           19 }
           20 
           21 void
           22 threadmain(int argc, char *argv[])
           23 {
           24         int n;
           25         uchar score[VtScoreSize];
           26         uchar *buf;
           27         char *host;
           28         vlong off;
           29         VtEntry e;
           30         VtRoot root;
           31         VtCache *c;
           32         VtConn *z;
           33         VtFile *f;
           34 
           35         quotefmtinstall();
           36         fmtinstall('F', vtfcallfmt);
           37         fmtinstall('V', vtscorefmt);
           38 
           39         host = nil;
           40         ARGBEGIN{
           41         case 'V':
           42                 chattyventi++;
           43                 break;
           44         case 'h':
           45                 host = EARGF(usage());
           46                 break;
           47         case 'v':
           48                 chatty++;
           49                 break;
           50         default:
           51                 usage();
           52                 break;
           53         }ARGEND
           54 
           55         if(argc != 0)
           56                 usage();
           57 
           58         buf = vtmallocz(Blocksize);
           59 
           60         z = vtdial(host);
           61         if(z == nil)
           62                 sysfatal("could not connect to server: %r");
           63 
           64         if(vtconnect(z) < 0)
           65                 sysfatal("vtconnect: %r");
           66 
           67         // write file
           68         c = vtcachealloc(z, Blocksize*32);
           69         if(c == nil)
           70                 sysfatal("vtcachealloc: %r");
           71         f = vtfilecreateroot(c, Blocksize, Blocksize, VtDataType);
           72         if(f == nil)
           73                 sysfatal("vtfilecreateroot: %r");
           74         off = 0;
           75         vtfilelock(f, VtOWRITE);
           76         while((n = read(0, buf, Blocksize)) > 0){
           77                 if(vtfilewrite(f, buf, n, off) != n)
           78                         sysfatal("vtfilewrite: %r");
           79                 off += n;
           80                 if(vtfileflushbefore(f, off) < 0)
           81                         sysfatal("vtfileflushbefore: %r");
           82         }
           83         if(vtfileflush(f) < 0)
           84                 sysfatal("vtfileflush: %r");
           85         if(vtfilegetentry(f, &e) < 0)
           86                 sysfatal("vtfilegetentry: %r");
           87         vtfileunlock(f);
           88 
           89         // write directory entry
           90         memset(&root, 0, sizeof root);
           91         vtentrypack(&e, buf, 0);
           92         if(vtwrite(z, root.score, VtDirType, buf, VtEntrySize) < 0)
           93                 sysfatal("vtwrite dir: %r");
           94 
           95         // write root
           96         strcpy(root.name, "data");
           97         strcpy(root.type, "file");
           98         root.blocksize = Blocksize;
           99         vtrootpack(&root, buf);
          100         if(vtwrite(z, score, VtRootType, buf, VtRootSize) < 0)
          101                 sysfatal("vtwrite root: %r");
          102 
          103         print("file:%V\n", score);
          104         threadexitsall(0);
          105 }