sys/src/cmd/tput.c

   1 #include <u.h>
   2 #include <libc.h>
   3
   4 int dopipe;
   5 int buflen = 8192;
   6 uvlong bc, sec;
   7
   8 void
   9 usage(void)
  10 {
  11         fprint(2, "usage: %s [-b buflen] [-p]\n", argv0);
  12         exits("usage");
  13 }
  14
  15 void
  16 main(int argc, char **argv)
  17 {
  18         double speed;
  19         int rc, cpid;
  20         char *buf;
  21
  22         ARGBEGIN {
  23         case 'b':
  24                 buflen = atoi(EARGF(usage()));
  25                 break;
  26         case 'p':
  27                 dopipe = 1;
  28                 break;
  29         default:
  30                 usage();
  31         } ARGEND
  32
  33         if(argc != 0)
  34                 usage();
  35
  36         bc = 0;
  37         sec = 0;
  38         buf = sbrk(buflen);
  39         cpid = rfork(RFPROC | RFMEM);
  40         if(cpid == 0) {
  41                 while(1) {
  42                         sleep(1000);
  43                         speed = bc / ++sec;
  44                         if(speed >= 1073741824) fprint(2, "%.2f GB/s\n", speed / 1073741824);
  45                         else if(speed >= 1048576) fprint(2, "%.2f MB/s\n", speed / 1048576);
  46                         else if(speed >= 1024) fprint(2, "%.2f KB/s\n", speed / 1024);
  47                         else fprint(2, "%.2f B/s\n", speed);
  48                 }
  49         }
  50         while(1) {
  51                 rc = read(0, buf, buflen);
  52                 if(rc <= 0) break;
  53                 if(dopipe) write(1, buf, rc);
  54                 bc += rc;
  55         }
  56         postnote(PNPROC, cpid, "kill");
  57         if(rc < 0) sysfatal("%r");
  58         exits(0);
  59 }