plan9port

fork of plan9port with libvec, libstr and libsdb
Log | Files | Refs | README | LICENSE

Ccli.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 }