plan9port

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

fossil.c (2714B)

      1 #include "stdinc.h"
      2 #include <ctype.h>
      3 
      4 #include "9.h"
      5 
      6 int Dflag;
      7 int mempcnt;			/* for 9fsys.c */
      8 char* none = "none";
      9 char* foptname = "/none/such";
     10 
     11 static void
     12 usage(void)
     13 {
     14 	fprint(2, "usage: %s [-Dt] [-c cmd] [-f partition] [-m %%]\n", argv0);
     15 	threadexitsall("usage");
     16 }
     17 
     18 static void
     19 readCmdPart(char *file, char ***pcmd, int *pncmd)
     20 {
     21 	char buf[1024+1], *f[1024];
     22 	char tbuf[1024];
     23 	int nf;
     24 	int i, fd, n;
     25 	char **cmd, *p;
     26 	int ncmd;
     27 
     28 	cmd = *pcmd;
     29 	ncmd = *pncmd;
     30 
     31 	if((fd = open(file, OREAD)) < 0)
     32 		sysfatal("open %s: %r", file);
     33 	if(seek(fd, 127*1024, 0) != 127*1024)
     34 		sysfatal("seek %s 127kB: %r", file);
     35 	n = readn(fd, buf, sizeof buf-1);
     36 	if(n == 0)
     37 		sysfatal("short read of %s at 127kB", file);
     38 	if(n < 0)
     39 		sysfatal("read %s: %r", file);
     40 	buf[n] = 0;
     41 	if(memcmp(buf, "fossil config\n", 6+1+6+1) != 0)
     42 		sysfatal("bad config magic in %s", file);
     43 	nf = getfields(buf+6+1+6+1, f, nelem(f), 1, "\n");
     44 	for(i=0; i<nf; i++){
     45 		if(f[i][0] == '#')
     46 			continue;
     47 		cmd = vtrealloc(cmd, (ncmd+1)*sizeof(char*));
     48 		/* expand argument '*' to mean current file */
     49 		if((p = strchr(f[i], '*')) && (p==f[i]||isspace(p[-1])) && (p[1]==0||isspace(p[1]))){
     50 			memmove(tbuf, f[i], p-f[i]);
     51 			strecpy(tbuf+(p-f[i]), tbuf+sizeof tbuf, file);
     52 			strecpy(tbuf+strlen(tbuf), tbuf+sizeof tbuf, p+1);
     53 			f[i] = tbuf;
     54 		}
     55 		cmd[ncmd++] = vtstrdup(f[i]);
     56 	}
     57 	close(fd);
     58 	*pcmd = cmd;
     59 	*pncmd = ncmd;
     60 }
     61 
     62 int
     63 threadmaybackground(void)
     64 {
     65 	return 1;
     66 }
     67 
     68 void
     69 threadmain(int argc, char* argv[])
     70 {
     71 	char **cmd, *p;
     72 	int i, ncmd, tflag;
     73 
     74 	fmtinstall('D', dirfmt);
     75 	fmtinstall('F', fcallfmt);
     76 	fmtinstall('M', dirmodefmt);
     77 	quotefmtinstall();
     78 
     79 	/*
     80 	 * Insulate from the invoker's environment.
     81 	 */
     82 #ifdef PLAN9PORT
     83 	if(rfork(RFNAMEG) < 0)
     84 #else
     85 	if(rfork(RFREND|RFNOTEG|RFNAMEG) < 0)
     86 #endif
     87 		sysfatal("rfork: %r");
     88 
     89 	close(0);
     90 	open("/dev/null", OREAD);
     91 	close(1);
     92 	open("/dev/null", OWRITE);
     93 
     94 	cmd = nil;
     95 	ncmd = tflag = 0;
     96 
     97 	ARGBEGIN{
     98 	case '?':
     99 	default:
    100 		usage();
    101 		break;
    102 	case 'c':
    103 		p = EARGF(usage());
    104 		currfsysname = p;
    105 		cmd = vtrealloc(cmd, (ncmd+1)*sizeof(char*));
    106 		cmd[ncmd++] = p;
    107 		break;
    108 	case 'D':
    109 		Dflag ^= 1;
    110 		break;
    111 	case 'f':
    112 		p = EARGF(usage());
    113 		currfsysname = foptname = p;
    114 		readCmdPart(p, &cmd, &ncmd);
    115 		break;
    116 	case 'm':
    117 		mempcnt = atoi(EARGF(usage()));
    118 		if(mempcnt <= 0 || mempcnt >= 100)
    119 			usage();
    120 		break;
    121 	case 't':
    122 		tflag = 1;
    123 		break;
    124 	}ARGEND
    125 	if(argc != 0)
    126 		usage();
    127 
    128 	consInit();
    129 	cliInit();
    130 	msgInit();
    131 	conInit();
    132 	cmdInit();
    133 	fsysInit();
    134 	exclInit();
    135 	fidInit();
    136 
    137 	srvInit();
    138 	lstnInit();
    139 	usersInit();
    140 
    141 	for(i = 0; i < ncmd; i++)
    142 		if(cliExec(cmd[i]) == 0)
    143 			fprint(2, "%s: %r\n", cmd[i]);
    144 	vtfree(cmd);
    145 
    146 	if(tflag && consTTY() == 0)
    147 		consPrint("%r\n");
    148 
    149 	threadexits(0);
    150 }