root/src/lib9/run_windows.c

DEFINITIONS

1. runcmd

// Copyright 2013 The Go Authors.  All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

#include <u.h>
#include <windows.h>
#define NOPLAN9DEFINES
#include <libc.h>
#include "win.h"

int
runcmd(char **argv)
{
        // Mostly copied from ../cmd/dist/windows.c.
        // If there's a bug here, fix the logic there too.
        int i, j, nslash;
        Fmt fmt;
        char *q;
        WinRune *r;
        STARTUPINFOW si;
        PROCESS_INFORMATION pi;
        DWORD code;

        fmtstrinit(&fmt);
        for(i=0; argv[i]; i++) {
                if(i > 0)
                        fmtprint(&fmt, " ");
                q = argv[i];
                if(strstr(q, " ") || strstr(q, "\t") || strstr(q, "\"") || strstr(q, "\\\\") || (strlen(q) > 0 && q[strlen(q)-1] == '\\')) {
                        fmtprint(&fmt, "\"");
                        nslash = 0;
                        for(; *q; q++) {
                                if(*q == '\\') {
                                        nslash++;
                                        continue;
                                }
                                if(*q == '"') {
                                        for(j=0; j<2*nslash+1; j++)
                                                fmtprint(&fmt, "\\");
                                        nslash = 0;
                                }
                                for(j=0; j<nslash; j++)
                                        fmtprint(&fmt, "\\");
                                nslash = 0;
                                fmtprint(&fmt, "\"");
                        }
                        for(j=0; j<2*nslash; j++)
                                fmtprint(&fmt, "\\");
                        fmtprint(&fmt, "\"");
                } else {
                        fmtprint(&fmt, "%s", q);
                }
        }
        
        q = fmtstrflush(&fmt);
        r = torune(q);
        free(q);

        memset(&si, 0, sizeof si);
        si.cb = sizeof si;
        si.dwFlags = STARTF_USESTDHANDLES;
        si.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
        si.hStdError = GetStdHandle(STD_ERROR_HANDLE);

        if(!CreateProcessW(nil, r, nil, nil, TRUE, 0, nil, nil, &si, &pi)) {
                free(r);
                return -1;
        }

        free(r);
        if(WaitForMultipleObjects(1, &pi.hProcess, FALSE, INFINITE) != 0)
                return -1;
        i = GetExitCodeProcess(pi.hProcess, &code);
        CloseHandle(pi.hProcess);
        CloseHandle(pi.hThread);
        if(!i)
                return -1;
        if(code != 0) {
                werrstr("unsuccessful exit status: %d", (int)code);
                return -1;
        }
        return 0;
}