sys/src/cmd/plumb/plumb.c

   1 #include <u.h>
   2 #include <libc.h>
   3 #include <plumb.h>
   4
   5 char *plumbfile = nil;
   6 Plumbmsg m;
   7
   8 void
   9 usage(void)
  10 {
  11         fprint(2, "usage:  plumb [-p plumbfile] [-a 'attr=value ...'] [-s src] [-d dst] [-t type] [-w wdir] -i | data1\n");
  12         exits("usage");
  13 }
  14
  15 void
  16 gather(void)
  17 {
  18         char buf[8192];
  19         int n;
  20
  21         m.ndata = 0;
  22         m.data = nil;
  23         while((n = read(0, buf, sizeof buf)) > 0){
  24                 m.data = realloc(m.data, m.ndata+n);
  25                 if(m.data == nil){
  26                         fprint(2, "plumb: alloc failed: %r\n");
  27                         exits("alloc");
  28                 }
  29                 memmove(m.data+m.ndata, buf, n);
  30                 m.ndata += n;
  31         }
  32         if(n < 0){
  33                 fprint(2, "plumb: i/o error on input: %r\n");
  34                 exits("read");
  35         }
  36 }
  37
  38 void
  39 main(int argc, char *argv[])
  40 {
  41         char buf[1024], *p;
  42         int fd, i, input;
  43
  44         input = 0;
  45         m.src = "plumb";
  46         m.dst = nil;
  47         m.wdir = getwd(buf, sizeof buf);
  48         m.type = "text";
  49         m.attr = nil;
  50         ARGBEGIN{
  51         case 'a':
  52                 p = ARGF();
  53                 if(p == nil)
  54                         usage();
  55                 m.attr = plumbaddattr(m.attr, plumbunpackattr(p));
  56                 break;
  57         case 'd':
  58                 m.dst = ARGF();
  59                 if(m.dst == nil)
  60                         usage();
  61                 break;
  62         case 'i':
  63                 input++;
  64                 break;
  65         case 't':
  66         case 'k':       /* for backwards compatibility */
  67                 m.type = ARGF();
  68                 if(m.type == nil)
  69                         usage();
  70                 break;
  71         case 'p':
  72                 plumbfile = ARGF();
  73                 if(plumbfile == nil)
  74                         usage();
  75                 break;
  76         case 's':
  77                 m.src = ARGF();
  78                 if(m.src == nil)
  79                         usage();
  80                 break;
  81         case 'w':
  82                 m.wdir = ARGF();
  83                 if(m.wdir == nil)
  84                         usage();
  85                 break;
  86         }ARGEND
  87
  88         if((input && argc>0) || (!input && argc<1))
  89                 usage();
  90         if(plumbfile != nil)
  91                 fd = open(plumbfile, OWRITE);
  92         else
  93                 fd = plumbopen("send", OWRITE);
  94         if(fd < 0){
  95                 fprint(2, "plumb: can't open plumb file: %r\n");
  96                 exits("open");
  97         }
  98         if(input){
  99                 gather();
 100                 if(plumblookup(m.attr, "action") == nil)
 101                         m.attr = plumbaddattr(m.attr, plumbunpackattr("action=showdata"));
 102                 if(plumbsend(fd, &m) < 0){
 103                         fprint(2, "plumb: can't send message: %r\n");
 104                         exits("error");
 105                 }
 106                 exits(nil);
 107         }
 108         for(i=0; i<argc; i++){
 109                 if(input == 0){
 110                         m.data = argv[i];
 111                         m.ndata = -1;
 112                 }
 113                 if(plumbsend(fd, &m) < 0){
 114                         fprint(2, "plumb: can't send message: %r\n");
 115                         exits("error");
 116                 }
 117         }
 118         exits(nil);
 119 }