code: purgatorio

ref: 254b07675e1a5348f2a8723f55e6b5b1b18a4721
dir: /utils/ntsrv/ntsrv.c/

View raw version
#ifndef ForNT4
/* only for XP, 2000 and above - JobObject only available on these*/
#undef _WIN32_WINNT
#define _WIN32_WINNT 0x0500
#endif

#include "lib9.h"
#include <windows.h>
#include <winsvc.h>
#include <stdarg.h>

#ifdef JOB_OBJECT_TERMINATE
#define	Jobs
#endif

#ifdef ForNT4
#undef Jobs
#endif

/*
 * ntsrv add Name Inferno-root Cmds
 * ntsrv del Name
 * ntsrv run Name Inferno-root Cmds
 *
 * 'add' registers service: Name with args "run Name Inferno-root Cmds"
 * 'del' unregisters service Name
 * 'run' - only given by NT service manager when starting the service (see 'add')
 *
 * Cmds are cat'd (with space separator) and requoted for CreateProcess()
 *
 * There must be an ntsrv.exe in Inferno-root/Nt/386/bin
 */


SERVICE_STATUS	status = {
	SERVICE_WIN32_OWN_PROCESS,	/* dwServiceType */
	0,							/* dwCurrentState */
	SERVICE_ACCEPT_STOP			/* dwControlsAccepted */
};

typedef struct Emu Emu;
struct Emu {
	HANDLE	proc;		/* NULL if error */
	HANDLE	job;			/* job for all children */
	HANDLE	stdin;		/* stdio pipes */
	HANDLE	stdout;
	DWORD	notepg;		/* process group ID (in case we lack Jobs) */
};

typedef struct Copyargs Copyargs;
struct Copyargs {
	HANDLE in;
	HANDLE out;
};

#ifdef Jobs
static char *myname = "ntsrv.exe";
#else
static char *myname = "ntsrv4.exe";
#endif
#define LOGFILE "grid\\slave\\svclog"
static char *name;
static char *root;
static char *cmds;
static SERVICE_STATUS_HANDLE	statush;
static HANDLE	emujob;		/* win32 job object for emu session */
static DWORD	emugroup;		/* process group ID for emu session */
static HANDLE emuin;		/* stdin pipe of emu */
static HANDLE logfile;

HANDLE openlog(char*);
void logmsg(char*, ...);
void WINAPI infmain(ulong, LPTSTR[]);
void WINAPI infctl(ulong);
Emu runemu(char*);
HANDLE exporthandle(HANDLE, int);
DWORD WINAPI copy(LPVOID);
int shuttingdown = 0;
int nice = 0;

static void
usage()
{
	fprint(2, "usage: ntsrv [-n] add name root cmds | del name\n");
}

/* (from rcsh)
 * windows quoting rules - I think
 * Words are seperated by space or tab
 * Words containing a space or tab can be quoted using "
 * 2N backslashes + " ==> N backslashes and end quote
 * 2N+1 backslashes + " ==> N backslashes + literal "
 * N backslashes not followed by " ==> N backslashes
 */
static char *
dblquote(char *cmd, char *s)
{
	int nb;
	char *p;

	for(p=s; *p; p++)
		if(*p == ' ' || *p == '\t' || *p == '\n' || *p == '\r' || *p == '"')
			break;

	if(*p == 0){				/* easy case */
		strcpy(cmd, s);
		return cmd+(p-s);
	}

	*cmd++ = '"';
	for(;;) {
		for(nb=0; *s=='\\'; nb++)
			*cmd++ = *s++;

		if(*s == 0) {			/* trailing backslashes -> 2N */
			while(nb-- > 0)
				*cmd++ = '\\';
			break;
		}

		if(*s == '"') {			/* literal quote -> 2N+1 backslashes */
			while(nb-- > 0)
				*cmd++ = '\\';
			*cmd++ = '\\';		/* escape the quote */
		}
		*cmd++ = *s++;
	}

	*cmd++ = '"';
	*cmd = 0;

	return cmd;
}

static char *
proccmd(char **argv)
{
	int i, n;
	char *cmd, *p;

	/* conservatively calculate length of command;
	 * backslash expansion can cause growth in dblquote().
	 */
	for(i=0,n=0; argv[i]; i++) {
		n += 2*strlen(argv[i]);
	}
	n++;
	
	cmd = malloc(n);
	for(i=0,p=cmd; argv[i]; i++) {
		p = dblquote(p, argv[i]);
		*p++ = ' ';
	}
	if(p != cmd)
		p--;
	*p = 0;

	return cmd;
}

int
installnewemu()
{
	LPCTSTR currpath, newpath;
	currpath = smprint("%s\\Nt\\386\\bin\\emu.exe", root);
	newpath = smprint("%s\\Nt\\386\\bin\\newemu.exe", root);
	if(GetFileAttributes(newpath) == 0xffffffff)	// INVALID_FILE_ATTRIBUTES is not defined
		return 0;
	DeleteFile(currpath);			// ignore error message - it might not be there.
	if(MoveFile(newpath, currpath) == 0){
		logmsg("cannot rename %s to %s: %r", newpath, currpath);
		return -1;
	}
	return 0;
}	

void WINAPI
infmain(ulong argc, char *argv[])
{
	HANDLE cpt;
	Emu emu;
	Copyargs cp;
	DWORD tid;
	char *cmd;

	argc--;
	argv++;
	cmd = smprint("%s\\%s", root, LOGFILE);
	logfile = openlog(cmd);
	free(cmd);
	statush = RegisterServiceCtrlHandler(name, infctl);
	if (statush == 0)
		return;

	status.dwCurrentState = SERVICE_START_PENDING;
	SetServiceStatus(statush, &status);

	while(installnewemu() != -1){
		/* start the service */
		cmd = smprint("%s\\Nt\\386\\bin\\emu.exe -r%s %s", root, root, cmds);
		logmsg("starting %s", cmd);
		emu = runemu(cmd);
		free(cmd);
		if (emu.proc == NULL) {
			logmsg("runemu failed: %r");
			status.dwCurrentState = SERVICE_STOPPED;
			SetServiceStatus(statush, &status);
			return;
		}
	
		cp.in = emu.stdout;
		cp.out = logfile;
		cpt = CreateThread(NULL, 0, copy, (void*)&cp, 0, &tid);
		if (cpt == NULL) {
			logmsg("failed to create copy thread: %r");
			CloseHandle(emu.stdout);
		}
	
		logmsg("infmain blocking on emu proc");
		emujob = emu.job;
		emugroup = emu.notepg;
		status.dwCurrentState = SERVICE_RUNNING;
		SetServiceStatus(statush, &status);
		WaitForSingleObject(emu.proc, INFINITE);
		logmsg("infmain emu proc terminated");
		emujob = NULL;
		emugroup = 0;
#ifdef Jobs
		logmsg("terminating job");
		TerminateJobObject(emu.job, 0);
#else
		logmsg("notepg (%d)", emu.notepg);
		if(emu.notepg)
			GenerateConsoleCtrlEvent(CTRL_C_EVENT, emu.notepg);
#endif
		if (cpt) {
			/* copy() sees eof on emu.stdout and exits */
			WaitForSingleObject(cpt, INFINITE);
			CloseHandle(cpt);
			CloseHandle(emu.stdout);
		}
		CloseHandle(emu.proc);
		if(emu.job != NULL)
			CloseHandle(emu.job);
		CloseHandle(emu.stdin);
		// XXX should check to see that we're not starting up again too quickly, as
		// it's quite possible to get into an infinite loop here.
		// but what are good criteria? 5 times? 100 times?
		// 10 times within a minute?
		// for the moment, just sleep for a while before restarting...
		if(shuttingdown)
			break;
		SleepEx(10000, FALSE);
	}
	logmsg("infmain done");
	if (logfile)
		CloseHandle(logfile);
	status.dwCurrentState = SERVICE_STOPPED;
	SetServiceStatus(statush, &status);
	return;
}

void WINAPI
infctl(ulong op)
{
	if (op != SERVICE_CONTROL_STOP)
		return;

	/* stop the service (status set by infmain()
	 *
	 * NOTE: there is a race for emujob - may have been closed
	 * after test, but before TerminateJobObject()
	 * MSDN is unclear as to whether TerminatJobObject() handles
	 * NULL job ptr - should probably use a mutex
	 */
	shuttingdown = 1;
#ifdef Jobs
	logmsg("svc stop: stopping job");
	if (emujob)
		TerminateJobObject(emujob, 0);
#else
	logmsg("svc stop: interrupting emu");
	if (emugroup)
		GenerateConsoleCtrlEvent(CTRL_C_EVENT, emugroup);
#endif
}

void
printerror(char *s)
{
	char *msg;
	FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
		FORMAT_MESSAGE_FROM_SYSTEM|
		FORMAT_MESSAGE_IGNORE_INSERTS,
		NULL,
		GetLastError(),
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		(LPTSTR)&msg,
		0,
		NULL);
	fprint(2, "%s: %s\n", s, msg);
	LocalFree(msg);
}

int
add(char *name, char *root, char *cmds)
{
	char *path;
	int r;
	SC_HANDLE scm, scs;
	char *nopt;

	nopt = nice ? " -n" : "";
	path = smprint("%s\\Nt\\386\\bin\\%s%s run %s %s %s", root, myname, nopt, name, root, cmds);
	r = 0;
	scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
	if (scm == NULL) {
		printerror("cannot open service control manager");
		return -1;
	}
	scs = CreateService(scm,
		name,
		name,
		SERVICE_START|SERVICE_STOP,
		SERVICE_WIN32_OWN_PROCESS,
		SERVICE_AUTO_START,
		SERVICE_ERROR_IGNORE,
		path,
		NULL,
		NULL,
		NULL,
		NULL,
		NULL
	);
	if (scs == NULL) {
		printerror("cannot create service");
		r = -1;
	} else {
		CloseServiceHandle(scs);
	}
	CloseServiceHandle(scm);
	return r;
}

int
del(char *name)
{
	SC_HANDLE scm, scs;

	scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
	if (scm == NULL) {
		printerror("cannot open service control manager");
		return -1;
	}

	scs = OpenService(scm, name, DELETE);
	if (scs == NULL) {
		printerror("cannot open service");
		CloseServiceHandle(scm);
		return -1;
	}
	if (!DeleteService(scs)) {
		printerror("cannot delete Iservice");
		CloseServiceHandle(scs);
		CloseServiceHandle(scm);
		return -1;
	}
	CloseServiceHandle(scs);
	CloseServiceHandle(scm);
	return 0;
}

HANDLE
openlog(char *p)
{
	HANDLE h;
	h = CreateFile(p, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	if (h == INVALID_HANDLE_VALUE)
		return NULL;
	SetFilePointer(h, 0, NULL, FILE_END);
	return h;
}

void
logmsg(char *fmt, ...)
{
	int n;
	char *p;
	va_list args;
	if(logfile == 0)
		return;
	va_start(args, fmt);
	p = vsmprint(fmt, args);
	va_end(args);
	n = strlen(p);
	if (n)
		WriteFile(logfile, p, n, &n, NULL);
	WriteFile(logfile, "\n", 1, &n, NULL);
}

Emu
runemu(char *cmd)
{
	Emu r = {NULL, NULL, NULL};
	STARTUPINFO si;
	PROCESS_INFORMATION pinfo;
	HANDLE job, emu, emut, stdin, stdout, stderr, emui, emuo;
	SECURITY_ATTRIBUTES sec;
	DWORD flags;

	job = emu = emut = stdin = stdout = stderr = emui = emuo = NULL;
#ifdef Jobs
	job = CreateJobObject(NULL, NULL);
	if (job == NULL) {
		logmsg("cannot create job object: %r");
		goto error;
	}
#endif

	/* set up pipes */
	sec.nLength = sizeof(sec);
	sec.lpSecurityDescriptor = 0;
	sec.bInheritHandle = 0;
	if (!CreatePipe(&stdin, &emui, &sec, 0)) {
		logmsg("cannot create stdin pipe: %r");
		goto error;
	}
	if (!CreatePipe(&emuo, &stdout, &sec, 0)) {
		logmsg("cannot create stdout pipe: %r");
		goto error;
	}
	stdin = exporthandle(stdin, 1);
	stdout = exporthandle(stdout, 1);
	stderr = exporthandle(stdout, 0);

	/* create emu process (suspended) */
	memset(&si, 0, sizeof(si));
	si.cb = sizeof(si);
	si.dwFlags = STARTF_USESTDHANDLES;
	si.hStdInput = stdin;
	si.hStdOutput = stdout;
	si.hStdError = stderr;

	flags = CREATE_NEW_PROCESS_GROUP|CREATE_DEFAULT_ERROR_MODE|CREATE_SUSPENDED;
	if(nice)
		flags |= IDLE_PRIORITY_CLASS;
	if(!CreateProcess(0, cmd, 0, 0, 1, flags, 0, 0, &si, &pinfo)) {
		logmsg("cannot create process: %r");
		goto error;
	}
	emu = pinfo.hProcess;
	emut = pinfo.hThread;
	CloseHandle(stdin);
	stdin = NULL;
	CloseHandle(stdout);
	stdout = NULL;
	CloseHandle(stderr);
	stderr = NULL;

#ifdef Jobs
	if(!AssignProcessToJobObject(job, emu)) {
		logmsg("failed to assign emu to job: %r");
		goto error;
	}
#endif
	ResumeThread(emut);
	CloseHandle(emut);

	r.proc = emu;
	r.notepg = pinfo.dwProcessId;
	r.job = job;	/* will be NULL if not implemented (NT4) */
	r.stdin = emui;
	r.stdout = emuo;
	return r;

error:
	if (stdin)
		CloseHandle(stdin);
	if (stdout)
		CloseHandle(stdout);
	if (stderr)
		CloseHandle(stderr);
	if (emui)
		CloseHandle(emuin);
	if (emuo)
		CloseHandle(emuo);
	if (emut)
		CloseHandle(emut);
	if (emu) {
		TerminateProcess(emu, 0);
		CloseHandle(emu);
	}
	if (job)
		CloseHandle(job);
	return r;
}

HANDLE
exporthandle(HANDLE h, int close)
{
	HANDLE cp, dh;
	DWORD flags = DUPLICATE_SAME_ACCESS;
	if (close)
		flags |= DUPLICATE_CLOSE_SOURCE;
	cp = GetCurrentProcess();
	if (!DuplicateHandle(cp, h, cp, &dh, DUPLICATE_SAME_ACCESS, 1, flags))
		return nil;
	return dh;
}

DWORD WINAPI
copy(void *arg)
{
	Copyargs *cp = (Copyargs*)arg;
	char buf[1024];
	DWORD n;

	while (ReadFile(cp->in, buf, sizeof(buf), &n, NULL)) {
		if (n && cp->out)
			WriteFile(cp->out, buf, n, &n, NULL);
	}
	return 0;
}

void
main(int argc, char *argv[])
{
	char *verb;
	SERVICE_TABLE_ENTRY services[2];

	memset(services, 0, sizeof(services));

	ARGBEGIN{
	case 'n':
		nice = 1;
		break;
	default:
		usage();
	}ARGEND

	if (argc < 2) {
		usage();
		return;
	}

	verb = argv[0];
	name = argv[1];
	if (argc > 2)
		root = argv[2];
	if (argc > 3)
		cmds = proccmd(argv+3);

	if (strcmp(verb, "del") == 0)
		exit(del(name));
	if (strcmp(verb, "add") == 0) {
		if (root == NULL || cmds == NULL) {
			usage();
			return;
		}
		exit(add(name, root, cmds));
	}
	if (strcmp(verb, "run") == 0) {
		if (root == NULL || cmds == NULL || *cmds == '\0') {
			usage();
			return;
		}
		services[0].lpServiceName = name;
		services[0].lpServiceProc = infmain;
		StartServiceCtrlDispatcher(services);
		exit(0);
	}
	usage();
}