code: purgatorio

ref: 38c0cf0737de03906729054ad7b7b011dd1ca475
dir: /emu/Nt/cmd.c/

#define Unknown win_Unknown
#define UNICODE
#include	<windows.h>
#include <winbase.h>
#include	<winsock.h>
#undef Unknown
#include	"dat.h"
#include	"fns.h"
#include	"error.h"

extern int	nth2fd(HANDLE);
extern wchar_t	*widen(char*);

/*
 * thanks to rcsh for these.
 *
 * windows quoting rules - I think
 * Words are separated 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 *
ntquotedcmd(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);
	if(cmd == nil)
		return nil;
	for(i=0,p=cmd; argv[i]; i++) {
		p = dblquote(p, argv[i]);
		*p++ = ' ';
	}
	if(p != cmd)
		p--;
	*p = 0;

	return cmd;
}

static 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;
}

/* TO DO: check that oserrstr will have the right text on error */

void*
oscmd(char **args, int nice, char *dir, int *fd)
{
	STARTUPINFO si;
	SECURITY_ATTRIBUTES sec;
	HANDLE rh, wh, eh, srh, swh, seh;
	PROCESS_INFORMATION pinfo;
	char *cmd;
	wchar_t *wcmd, *wdir;
	int prio;

	wdir = nil;
	if(dir != nil)
		wdir = widen(dir);

	cmd = ntquotedcmd(args);
	if(cmd == nil)
		error(Enomem);

	wcmd = widen(cmd);
	sec.nLength = sizeof(sec);
	sec.lpSecurityDescriptor = 0;
	sec.bInheritHandle = 0;
	rh = wh = eh = srh = swh = seh = nil;
	if(!CreatePipe(&rh, &swh, &sec, 0))
		goto Error;
	if(!CreatePipe(&srh, &wh, &sec, 0))
		goto Error;
	if(!CreatePipe(&seh, &eh, &sec, 0))
		goto Error;
	rh = exporthandle(rh, 1);
	if(rh == nil)
		goto Error;
	wh = exporthandle(wh, 1);
	if(wh == nil)
		goto Error;
	eh = exporthandle(eh, 1);
	if(eh == nil)
		goto Error;

	memset(&si, 0, sizeof(si));
	si.cb = sizeof(si);
	si.dwFlags = STARTF_USESHOWWINDOW|STARTF_USESTDHANDLES;
	si.wShowWindow = SW_SHOW;
	si.hStdInput = rh;
	si.hStdOutput = wh;
	si.hStdError = eh;

	prio = 0;
	if(nice){
		prio = IDLE_PRIORITY_CLASS;
		if(nice > 1)
			prio |= CREATE_SUSPENDED;
	}

	/* default of nil for wpath seems to be what we want; nil for env exports our current one */
	if(!CreateProcess(nil/*wpath*/, wcmd, 0, 0, 1,
	   CREATE_NEW_PROCESS_GROUP|CREATE_DEFAULT_ERROR_MODE|prio,
	   0 /*env*/, wdir, &si, &pinfo)){
		//print("can't create process '%Q' %d\n", wcmd, GetLastError());
		goto Error;
	}

	fd[0] = nth2fd(swh);
	fd[1] = nth2fd(srh);
	fd[2] = nth2fd(seh);
	if(fd[1] == 1 || fd[2] == 2)
		panic("invalid mapping of handle to fd");
	CloseHandle(si.hStdInput);
	CloseHandle(si.hStdOutput);
	CloseHandle(si.hStdError);

	if(prio & CREATE_SUSPENDED){
		if(nice > 1)
			SetThreadPriority(pinfo.hThread,
				nice>3? THREAD_PRIORITY_IDLE:
				nice>2? THREAD_PRIORITY_LOWEST:
				THREAD_PRIORITY_BELOW_NORMAL);
		ResumeThread(pinfo.hThread);
	}
	CloseHandle(pinfo.hThread);
	/* don't close process handle */
	free(cmd);
	free(wcmd);
	free(wdir);
	return pinfo.hProcess;

Error:
	if(rh)
		CloseHandle(rh);
	if(wh)
		CloseHandle(wh);
	if(eh)
		CloseHandle(eh);
	if(srh)
		CloseHandle(srh);
	if(swh)
		CloseHandle(swh);
	if(seh)
		CloseHandle(seh);
	free(cmd);
	free(wcmd);
	free(wdir);
	return nil;
}

int
oscmdwait(void *v, char *buf, int n)
{
	int status;
	HANDLE proc = (HANDLE)v;

	/* need not worry about being interrupted */
	if(WaitForSingleObject(proc, INFINITE) == WAIT_FAILED)
		return -1;
	if(!GetExitCodeProcess(proc, &status))
		status = 1;
	if(status)
		n = snprint(buf, n, "0 0 0 0 'status %d'", status);
	else
		n = snprint(buf, n, "0 0 0 0 ''");
	return n;

}

int
oscmdkill(void *v)
{
	if(TerminateProcess((HANDLE)v, 666) == FALSE)
		return -1;
	return 0;
}

void
oscmdfree(void *v)
{
	CloseHandle((HANDLE)v);
}