tCcli.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
       ---
       tCcli.c (1560B)
       ---
            1 #include "stdinc.h"
            2 
            3 #include "9.h"
            4 
            5 typedef struct {
            6         char*        argv0;
            7         int        (*cmd)(int, char*[]);
            8 } Cmd;
            9 
           10 static struct {
           11         QLock        lock;
           12         Cmd*        cmd;
           13         int        ncmd;
           14         int        hi;
           15 } cbox;
           16 
           17 enum {
           18         NCmdIncr        = 20,
           19 };
           20 
           21 int
           22 cliError(char* fmt, ...)
           23 {
           24         char *p;
           25         va_list arg;
           26 
           27         va_start(arg, fmt);
           28         p = vsmprint(fmt, arg);
           29         werrstr("%s", p);
           30         free(p);
           31         va_end(arg);
           32 
           33         return 0;
           34 }
           35 
           36 int
           37 cliExec(char* buf)
           38 {
           39         int argc, i, r;
           40         char *argv[20], *p;
           41 
           42         p = vtstrdup(buf);
           43         if((argc = tokenize(p, argv, nelem(argv)-1)) == 0){
           44                 vtfree(p);
           45                 return 1;
           46         }
           47         argv[argc] = 0;
           48 
           49         if(argv[0][0] == '#'){
           50                 vtfree(p);
           51                 return 1;
           52         }
           53 
           54         qlock(&cbox.lock);
           55         for(i = 0; i < cbox.hi; i++){
           56                 if(strcmp(cbox.cmd[i].argv0, argv[0]) == 0){
           57                         qunlock(&cbox.lock);
           58                         if(!(r = cbox.cmd[i].cmd(argc, argv)))
           59                                 consPrint("%r\n");
           60                         vtfree(p);
           61                         return r;
           62                 }
           63         }
           64         qunlock(&cbox.lock);
           65 
           66         consPrint("%s: - eh?\n", argv[0]);
           67         vtfree(p);
           68 
           69         return 0;
           70 }
           71 
           72 int
           73 cliAddCmd(char* argv0, int (*cmd)(int, char*[]))
           74 {
           75         int i;
           76         Cmd *opt;
           77 
           78         qlock(&cbox.lock);
           79         for(i = 0; i < cbox.hi; i++){
           80                 if(strcmp(argv0, cbox.cmd[i].argv0) == 0){
           81                         qunlock(&cbox.lock);
           82                         return 0;
           83                 }
           84         }
           85         if(i >= cbox.hi){
           86                 if(cbox.hi >= cbox.ncmd){
           87                         cbox.cmd = vtrealloc(cbox.cmd,
           88                                         (cbox.ncmd+NCmdIncr)*sizeof(Cmd));
           89                         memset(&cbox.cmd[cbox.ncmd], 0, NCmdIncr*sizeof(Cmd));
           90                         cbox.ncmd += NCmdIncr;
           91                 }
           92         }
           93 
           94         opt = &cbox.cmd[cbox.hi];
           95         opt->argv0 = argv0;
           96         opt->cmd = cmd;
           97         cbox.hi++;
           98         qunlock(&cbox.lock);
           99 
          100         return 1;
          101 }
          102 
          103 int
          104 cliInit(void)
          105 {
          106         cbox.cmd = vtmallocz(NCmdIncr*sizeof(Cmd));
          107         cbox.ncmd = NCmdIncr;
          108         cbox.hi = 0;
          109 
          110         return 1;
          111 }