Damjan Jovanovic 42fa293d48 start.exe: Search PATH for the executable to launch.
We want to launch all executables through CreateProcess(), but need to
provide the full path, as the searching CreateProcess() does
internally when the application name is NULL is more limited: it only
searches for .EXE files.

Signed-off-by: Damjan Jovanovic <damjan.jov@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
2019-08-19 22:17:20 +02:00

671 lines
21 KiB
C

/*
* Start a program using ShellExecuteEx, optionally wait for it to finish
* Compatible with Microsoft's "c:\windows\command\start.exe"
*
* Copyright 2003 Dan Kegel
* Copyright 2007 Lyutin Anatoly (Etersoft)
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdio.h>
#include <stdlib.h>
#include <windows.h>
#include <shlobj.h>
#include <shellapi.h>
#include <wine/debug.h>
#include "resources.h"
WINE_DEFAULT_DEBUG_CHANNEL(start);
/**
Output given message to stdout without formatting.
*/
static void output(const WCHAR *message)
{
DWORD count;
DWORD res;
int wlen = lstrlenW(message);
if (!wlen) return;
res = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), message, wlen, &count, NULL);
/* If writing to console fails, assume it's file
* i/o so convert to OEM codepage and output
*/
if (!res)
{
DWORD len;
char *mesA;
/* Convert to OEM, then output */
len = WideCharToMultiByte( GetConsoleOutputCP(), 0, message, wlen, NULL, 0, NULL, NULL );
mesA = HeapAlloc(GetProcessHeap(), 0, len*sizeof(char));
if (!mesA) return;
WideCharToMultiByte( GetConsoleOutputCP(), 0, message, wlen, mesA, len, NULL, NULL );
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), mesA, len, &count, FALSE);
HeapFree(GetProcessHeap(), 0, mesA);
}
}
/**
Output given message from string table,
followed by ": ",
followed by description of given GetLastError() value to stdout,
followed by a trailing newline,
then terminate.
*/
static void fatal_error(const WCHAR *msg, DWORD error_code, const WCHAR *filename)
{
DWORD_PTR args[1];
LPVOID lpMsgBuf;
int status;
static const WCHAR colonsW[] = { ':', ' ', 0 };
static const WCHAR newlineW[] = { '\n', 0 };
output(msg);
output(colonsW);
args[0] = (DWORD_PTR)filename;
status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ARGUMENT_ARRAY,
NULL, error_code, 0, (LPWSTR)&lpMsgBuf, 0, (__ms_va_list *)args );
if (!status)
{
WINE_ERR("FormatMessage failed\n");
} else
{
output(lpMsgBuf);
LocalFree((HLOCAL) lpMsgBuf);
output(newlineW);
}
ExitProcess(1);
}
static void fatal_string_error(int which, DWORD error_code, const WCHAR *filename)
{
WCHAR msg[2048];
if (!LoadStringW(GetModuleHandleW(NULL), which, msg, ARRAY_SIZE(msg)))
WINE_ERR("LoadString failed, error %d\n", GetLastError());
fatal_error(msg, error_code, filename);
}
static void fatal_string(int which)
{
WCHAR msg[2048];
if (!LoadStringW(GetModuleHandleW(NULL), which, msg, ARRAY_SIZE(msg)))
WINE_ERR("LoadString failed, error %d\n", GetLastError());
output(msg);
ExitProcess(1);
}
static void usage(void)
{
fatal_string(STRING_USAGE);
}
static WCHAR *build_args( int argc, WCHAR **argvW )
{
int i, wlen = 1;
WCHAR *ret, *p;
static const WCHAR FormatQuotesW[] = { ' ', '\"', '%', 's', '\"', 0 };
static const WCHAR FormatW[] = { ' ', '%', 's', 0 };
for (i = 0; i < argc; i++ )
{
wlen += lstrlenW(argvW[i]) + 1;
if (wcschr(argvW[i], ' '))
wlen += 2;
}
ret = HeapAlloc( GetProcessHeap(), 0, wlen*sizeof(WCHAR) );
ret[0] = 0;
for (i = 0, p = ret; i < argc; i++ )
{
if (wcschr(argvW[i], ' '))
p += swprintf(p, wlen - (p - ret), FormatQuotesW, argvW[i]);
else
p += swprintf(p, wlen - (p - ret), FormatW, argvW[i]);
}
return ret;
}
static WCHAR *get_parent_dir(WCHAR* path)
{
WCHAR *last_slash;
WCHAR *result;
int len;
last_slash = wcsrchr( path, '\\' );
if (last_slash == NULL)
len = 1;
else
len = last_slash - path + 1;
result = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
CopyMemory(result, path, (len-1)*sizeof(WCHAR));
result[len-1] = '\0';
return result;
}
static BOOL is_option(const WCHAR* arg, const WCHAR* opt)
{
return CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE,
arg, -1, opt, -1) == CSTR_EQUAL;
}
static void parse_title(const WCHAR *arg, WCHAR *title, int size)
{
/* See:
* WCMD_start() in programs/cmd/builtins.c
* WCMD_parameter_with_delims() in programs/cmd/batch.c
* The shell has already tokenized the command line for us.
* All we need to do is filter out all the quotes.
*/
int next;
const WCHAR *p = arg;
for (next = 0; next < (size-1) && *p; p++) {
if (*p != '"')
title[next++] = *p;
}
title[next] = '\0';
}
static BOOL search_path(const WCHAR *firstParam, WCHAR **full_path)
{
/* Copied from WCMD_run_program() in programs/cmd/wcmdmain.c */
#define MAXSTRING 8192
static const WCHAR slashW[] = {'\\','\0',};
static const WCHAR envPath[] = {'P','A','T','H','\0'};
static const WCHAR envPathExt[] = {'P','A','T','H','E','X','T','\0'};
static const WCHAR dfltPathExt[] = {'.','b','a','t',';',
'.','c','o','m',';',
'.','c','m','d',';',
'.','e','x','e','\0'};
static const WCHAR delims[] = {'/','\\',':','\0'};
WCHAR temp[MAX_PATH];
WCHAR pathtosearch[MAXSTRING];
WCHAR *pathposn;
WCHAR stemofsearch[MAX_PATH]; /* maximum allowed executable name is
MAX_PATH, including null character */
WCHAR *lastSlash;
WCHAR pathext[MAXSTRING];
BOOL extensionsupplied = FALSE;
DWORD len;
/* Calculate the search path and stem to search for */
if (wcspbrk (firstParam, delims) == NULL) { /* No explicit path given, search path */
static const WCHAR curDir[] = {'.',';','\0'};
lstrcpyW(pathtosearch, curDir);
len = GetEnvironmentVariableW(envPath, &pathtosearch[2], ARRAY_SIZE(pathtosearch)-2);
if ((len == 0) || (len >= ARRAY_SIZE(pathtosearch) - 2)) {
static const WCHAR curDir[] = {'.','\0'};
lstrcpyW (pathtosearch, curDir);
}
if (wcschr(firstParam, '.') != NULL) extensionsupplied = TRUE;
if (lstrlenW(firstParam) >= MAX_PATH) {
return FALSE;
}
lstrcpyW(stemofsearch, firstParam);
} else {
/* Convert eg. ..\fred to include a directory by removing file part */
GetFullPathNameW(firstParam, ARRAY_SIZE(pathtosearch), pathtosearch, NULL);
lastSlash = wcsrchr(pathtosearch, '\\');
if (lastSlash && wcschr(lastSlash, '.') != NULL) extensionsupplied = TRUE;
lstrcpyW(stemofsearch, lastSlash+1);
/* Reduce pathtosearch to a path with trailing '\' to support c:\a.bat and
c:\windows\a.bat syntax */
if (lastSlash) *(lastSlash + 1) = 0x00;
}
/* Now extract PATHEXT */
len = GetEnvironmentVariableW(envPathExt, pathext, ARRAY_SIZE(pathext));
if ((len == 0) || (len >= ARRAY_SIZE(pathext))) {
lstrcpyW (pathext, dfltPathExt);
}
/* Loop through the search path, dir by dir */
pathposn = pathtosearch;
WINE_TRACE("Searching in '%s' for '%s'\n", wine_dbgstr_w(pathtosearch),
wine_dbgstr_w(stemofsearch));
while (pathposn) {
WCHAR thisDir[MAX_PATH] = {'\0'};
int length = 0;
WCHAR *pos = NULL;
BOOL found = FALSE;
BOOL inside_quotes = FALSE;
/* Work on the first directory on the search path */
pos = pathposn;
while ((inside_quotes || *pos != ';') && *pos != 0)
{
if (*pos == '"')
inside_quotes = !inside_quotes;
pos++;
}
if (*pos) { /* Reached semicolon */
memcpy(thisDir, pathposn, (pos-pathposn) * sizeof(WCHAR));
thisDir[(pos-pathposn)] = 0x00;
pathposn = pos+1;
} else { /* Reached string end */
lstrcpyW(thisDir, pathposn);
pathposn = NULL;
}
/* Remove quotes */
length = lstrlenW(thisDir);
if (thisDir[length - 1] == '"')
thisDir[length - 1] = 0;
if (*thisDir != '"')
lstrcpyW(temp, thisDir);
else
lstrcpyW(temp, thisDir + 1);
/* Since you can have eg. ..\.. on the path, need to expand
to full information */
GetFullPathNameW(temp, MAX_PATH, thisDir, NULL);
/* 1. If extension supplied, see if that file exists */
lstrcatW(thisDir, slashW);
lstrcatW(thisDir, stemofsearch);
pos = &thisDir[lstrlenW(thisDir)]; /* Pos = end of name */
/* 1. If extension supplied, see if that file exists */
if (extensionsupplied) {
if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
found = TRUE;
}
}
/* 2. Any .* matches? */
if (!found) {
HANDLE h;
WIN32_FIND_DATAW finddata;
static const WCHAR allFiles[] = {'.','*','\0'};
lstrcatW(thisDir,allFiles);
h = FindFirstFileW(thisDir, &finddata);
FindClose(h);
if (h != INVALID_HANDLE_VALUE) {
WCHAR *thisExt = pathext;
/* 3. Yes - Try each path ext */
while (thisExt) {
WCHAR *nextExt = wcschr(thisExt, ';');
if (nextExt) {
memcpy(pos, thisExt, (nextExt-thisExt) * sizeof(WCHAR));
pos[(nextExt-thisExt)] = 0x00;
thisExt = nextExt+1;
} else {
lstrcpyW(pos, thisExt);
thisExt = NULL;
}
if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
found = TRUE;
thisExt = NULL;
}
}
}
}
if (found) {
int needed_size = lstrlenW(thisDir) + 1;
*full_path = HeapAlloc(GetProcessHeap(), 0, needed_size * sizeof(WCHAR));
if (*full_path)
lstrcpyW(*full_path, thisDir);
return TRUE;
}
}
return FALSE;
}
int wmain (int argc, WCHAR *argv[])
{
SHELLEXECUTEINFOW sei;
DWORD creation_flags;
WCHAR *args = NULL;
int i;
BOOL unix_mode = FALSE;
BOOL progid_open = FALSE;
WCHAR *title = NULL;
const WCHAR *file;
WCHAR *dos_filename = NULL;
WCHAR *fullpath = NULL;
WCHAR *parent_directory = NULL;
DWORD binary_type;
static const WCHAR bW[] = { '/', 'b', 0 };
static const WCHAR minW[] = { '/', 'm', 'i', 'n', 0 };
static const WCHAR maxW[] = { '/', 'm', 'a', 'x', 0 };
static const WCHAR lowW[] = { '/', 'l', 'o', 'w', 0 };
static const WCHAR normalW[] = { '/', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
static const WCHAR highW[] = { '/', 'h', 'i', 'g', 'h', 0 };
static const WCHAR realtimeW[] = { '/', 'r', 'e', 'a', 'l', 't', 'i', 'm', 'e', 0 };
static const WCHAR abovenormalW[] = { '/', 'a', 'b', 'o', 'v', 'e', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
static const WCHAR belownormalW[] = { '/', 'b', 'e', 'l', 'o', 'w', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
static const WCHAR separateW[] = { '/', 's', 'e', 'p', 'a', 'r', 'a', 't', 'e', 0 };
static const WCHAR sharedW[] = { '/', 's', 'h', 'a', 'r', 'e', 'd', 0 };
static const WCHAR nodeW[] = { '/', 'n', 'o', 'd', 'e', 0 };
static const WCHAR affinityW[] = { '/', 'a', 'f', 'f', 'i', 'n', 'i', 't', 'y', 0 };
static const WCHAR wW[] = { '/', 'w', 0 };
static const WCHAR waitW[] = { '/', 'w', 'a', 'i', 't', 0 };
static const WCHAR helpW[] = { '/', '?', 0 };
static const WCHAR unixW[] = { '/', 'u', 'n', 'i', 'x', 0 };
static const WCHAR progIDOpenW[] =
{ '/', 'p', 'r', 'o', 'g', 'I', 'D', 'O', 'p', 'e', 'n', 0};
static const WCHAR openW[] = { 'o', 'p', 'e', 'n', 0 };
static const WCHAR cmdW[] = { 'c', 'm', 'd', '.', 'e', 'x', 'e', 0 };
memset(&sei, 0, sizeof(sei));
sei.cbSize = sizeof(sei);
sei.lpVerb = openW;
sei.nShow = SW_SHOWNORMAL;
/* Dunno what these mean, but it looks like winMe's start uses them */
sei.fMask = SEE_MASK_FLAG_DDEWAIT|
SEE_MASK_FLAG_NO_UI;
sei.lpDirectory = NULL;
creation_flags = CREATE_NEW_CONSOLE;
/* Canonical Microsoft commandline flag processing:
* flags start with / and are case insensitive.
*/
for (i=1; i<argc; i++) {
/* parse first quoted argument as console title */
if (!title && argv[i][0] == '"') {
/* it will remove at least 1 quote */
int maxChars = lstrlenW(argv[1]);
title = HeapAlloc(GetProcessHeap(), 0, maxChars*sizeof(WCHAR));
if (title)
parse_title(argv[i], title, maxChars);
else {
WINE_ERR("out of memory\n");
ExitProcess(1);
}
continue;
}
if (argv[i][0] != '/')
break;
/* Unix paths can start with / so we have to assume anything following /unix is not a flag */
if (unix_mode || progid_open)
break;
if (argv[i][0] == '/' && (argv[i][1] == 'd' || argv[i][1] == 'D')) {
if (argv[i][2])
/* The start directory was concatenated to the option */
sei.lpDirectory = argv[i]+2;
else if (i+1 == argc) {
WINE_ERR("you must specify a directory path for the /d option\n");
usage();
} else
sei.lpDirectory = argv[++i];
}
else if (is_option(argv[i], bW)) {
creation_flags &= ~CREATE_NEW_CONSOLE;
}
else if (argv[i][0] == '/' && (argv[i][1] == 'i' || argv[i][1] == 'I')) {
TRACE("/i is ignored\n"); /* FIXME */
}
else if (is_option(argv[i], minW)) {
sei.nShow = SW_SHOWMINIMIZED;
}
else if (is_option(argv[i], maxW)) {
sei.nShow = SW_SHOWMAXIMIZED;
}
else if (is_option(argv[i], lowW)) {
creation_flags |= IDLE_PRIORITY_CLASS;
}
else if (is_option(argv[i], normalW)) {
creation_flags |= NORMAL_PRIORITY_CLASS;
}
else if (is_option(argv[i], highW)) {
creation_flags |= HIGH_PRIORITY_CLASS;
}
else if (is_option(argv[i], realtimeW)) {
creation_flags |= REALTIME_PRIORITY_CLASS;
}
else if (is_option(argv[i], abovenormalW)) {
creation_flags |= ABOVE_NORMAL_PRIORITY_CLASS;
}
else if (is_option(argv[i], belownormalW)) {
creation_flags |= BELOW_NORMAL_PRIORITY_CLASS;
}
else if (is_option(argv[i], separateW)) {
TRACE("/separate is ignored\n"); /* FIXME */
}
else if (is_option(argv[i], sharedW)) {
TRACE("/shared is ignored\n"); /* FIXME */
}
else if (is_option(argv[i], nodeW)) {
if (i+1 == argc) {
WINE_ERR("you must specify a numa node for the /node option\n");
usage();
} else
{
TRACE("/node is ignored\n"); /* FIXME */
i++;
}
}
else if (is_option(argv[i], affinityW))
{
if (i+1 == argc) {
WINE_ERR("you must specify a numa node for the /node option\n");
usage();
} else
{
TRACE("/affinity is ignored\n"); /* FIXME */
i++;
}
}
else if (is_option(argv[i], wW) || is_option(argv[i], waitW)) {
sei.fMask |= SEE_MASK_NOCLOSEPROCESS;
}
else if (is_option(argv[i], helpW)) {
usage();
}
/* Wine extensions */
else if (is_option(argv[i], unixW)) {
unix_mode = TRUE;
}
else if (is_option(argv[i], progIDOpenW)) {
progid_open = TRUE;
} else
{
WINE_ERR("Unknown option '%s'\n", wine_dbgstr_w(argv[i]));
usage();
}
}
if (progid_open) {
if (i == argc)
usage();
sei.lpClass = argv[i++];
sei.fMask |= SEE_MASK_CLASSNAME;
}
if (i == argc) {
if (progid_open || unix_mode)
usage();
file = cmdW;
}
else
file = argv[i++];
args = build_args( argc - i, &argv[i] );
sei.lpParameters = args;
if (unix_mode || progid_open) {
LPWSTR (*CDECL wine_get_dos_file_name_ptr)(LPCSTR);
char* multibyte_unixpath;
int multibyte_unixpath_len;
wine_get_dos_file_name_ptr = (void*)GetProcAddress(GetModuleHandleA("KERNEL32"), "wine_get_dos_file_name");
if (!wine_get_dos_file_name_ptr)
fatal_string(STRING_UNIXFAIL);
multibyte_unixpath_len = WideCharToMultiByte(CP_UNIXCP, 0, file, -1, NULL, 0, NULL, NULL);
multibyte_unixpath = HeapAlloc(GetProcessHeap(), 0, multibyte_unixpath_len);
WideCharToMultiByte(CP_UNIXCP, 0, file, -1, multibyte_unixpath, multibyte_unixpath_len, NULL, NULL);
dos_filename = wine_get_dos_file_name_ptr(multibyte_unixpath);
HeapFree(GetProcessHeap(), 0, multibyte_unixpath);
if (!dos_filename)
fatal_string(STRING_UNIXFAIL);
sei.lpFile = dos_filename;
if (!sei.lpDirectory)
sei.lpDirectory = parent_directory = get_parent_dir(dos_filename);
sei.fMask &= ~SEE_MASK_FLAG_NO_UI;
} else {
if (search_path(file, &fullpath)) {
if (fullpath != NULL) {
sei.lpFile = fullpath;
} else {
fatal_string_error(STRING_EXECFAIL, ERROR_OUTOFMEMORY, file);
}
} else {
sei.lpFile = file;
}
}
if (GetBinaryTypeW(sei.lpFile, &binary_type)) {
WCHAR *commandline;
STARTUPINFOW startup_info;
PROCESS_INFORMATION process_information;
static const WCHAR commandlineformat[] = {'"','%','s','"','%','s',0};
/* explorer on windows always quotes the filename when running a binary on windows (see bug 5224) so we have to use CreateProcessW in this case */
commandline = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(sei.lpFile)+3+lstrlenW(sei.lpParameters))*sizeof(WCHAR));
swprintf(commandline, lstrlenW(sei.lpFile) + 3 + lstrlenW(sei.lpParameters),
commandlineformat, sei.lpFile, sei.lpParameters);
ZeroMemory(&startup_info, sizeof(startup_info));
startup_info.cb = sizeof(startup_info);
startup_info.wShowWindow = sei.nShow;
startup_info.lpTitle = title;
if (!CreateProcessW(
sei.lpFile, /* lpApplicationName */
commandline, /* lpCommandLine */
NULL, /* lpProcessAttributes */
NULL, /* lpThreadAttributes */
FALSE, /* bInheritHandles */
creation_flags, /* dwCreationFlags */
NULL, /* lpEnvironment */
sei.lpDirectory, /* lpCurrentDirectory */
&startup_info, /* lpStartupInfo */
&process_information /* lpProcessInformation */ ))
{
fatal_string_error(STRING_EXECFAIL, GetLastError(), sei.lpFile);
}
sei.hProcess = process_information.hProcess;
goto done;
}
if (!ShellExecuteExW(&sei))
{
static const WCHAR pathextW[] = {'P','A','T','H','E','X','T',0};
const WCHAR *filename = sei.lpFile;
DWORD size, filename_len;
WCHAR *name, *env;
size = GetEnvironmentVariableW(pathextW, NULL, 0);
if (size)
{
WCHAR *start, *ptr;
env = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
if (!env)
fatal_string_error(STRING_EXECFAIL, ERROR_OUTOFMEMORY, sei.lpFile);
GetEnvironmentVariableW(pathextW, env, size);
filename_len = lstrlenW(filename);
name = HeapAlloc(GetProcessHeap(), 0, (filename_len + size) * sizeof(WCHAR));
if (!name)
fatal_string_error(STRING_EXECFAIL, ERROR_OUTOFMEMORY, sei.lpFile);
sei.lpFile = name;
start = env;
while ((ptr = wcschr(start, ';')))
{
if (start == ptr)
{
start = ptr + 1;
continue;
}
lstrcpyW(name, filename);
memcpy(&name[filename_len], start, (ptr - start) * sizeof(WCHAR));
name[filename_len + (ptr - start)] = 0;
if (ShellExecuteExW(&sei))
{
HeapFree(GetProcessHeap(), 0, name);
HeapFree(GetProcessHeap(), 0, env);
goto done;
}
start = ptr + 1;
}
}
fatal_string_error(STRING_EXECFAIL, GetLastError(), filename);
}
done:
HeapFree( GetProcessHeap(), 0, args );
HeapFree( GetProcessHeap(), 0, dos_filename );
HeapFree( GetProcessHeap(), 0, fullpath );
HeapFree( GetProcessHeap(), 0, parent_directory );
HeapFree( GetProcessHeap(), 0, title );
if (sei.fMask & SEE_MASK_NOCLOSEPROCESS) {
DWORD exitcode;
WaitForSingleObject(sei.hProcess, INFINITE);
GetExitCodeProcess(sei.hProcess, &exitcode);
ExitProcess(exitcode);
}
ExitProcess(0);
}