[CMD]: Small refactoring:
[reactos.git] / reactos / base / shell / cmd / start.c
1 /*
2 * START.C - start internal command.
3 *
4 *
5 * History:
6 *
7 * 24-Jul-1999 (Eric Kohl)
8 * Started.
9 *
10 * 30-Apr-2005 (Magnus Olsen <magnus@greatlord.com>)
11 * Remove all hardcoded strings in En.rc
12 */
13
14 #include "precomp.h"
15
16 #ifdef INCLUDE_CMD_START
17
18 /* Find the end of an option, and turn it into a nul-terminated string
19 * in place. (It's moved back one character, to make room for the nul) */
20 static TCHAR *GetParameter(TCHAR **pPointer)
21 {
22 BOOL bInQuote = FALSE;
23 TCHAR *start = *pPointer;
24 TCHAR *p;
25 for (p = start; *p; p++)
26 {
27 if (!bInQuote && (*p == _T('/') || _istspace(*p)))
28 break;
29 bInQuote ^= (*p == _T('"'));
30 p[-1] = *p;
31 }
32 p[-1] = _T('\0');
33 *pPointer = p;
34 return start - 1;
35 }
36
37 INT cmd_start (LPTSTR Rest)
38 {
39 TCHAR szFullName[CMDLINE_LENGTH];
40 TCHAR szUnquotedName[CMDLINE_LENGTH];
41 TCHAR *param = NULL;
42 TCHAR *dot;
43 INT size;
44 LPTSTR comspec;
45 BOOL bWait = FALSE;
46 BOOL bBat = FALSE;
47 BOOL bCreate = FALSE;
48 TCHAR szFullCmdLine[CMDLINE_LENGTH];
49 PROCESS_INFORMATION prci;
50 STARTUPINFO stui;
51 #ifdef UNICODE
52 DWORD dwCreationFlags = CREATE_NEW_CONSOLE | CREATE_UNICODE_ENVIRONMENT;
53 #else
54 DWORD dwCreationFlags = CREATE_NEW_CONSOLE;
55 #endif
56 DWORD dwAffinityMask = 0;
57 LPTSTR lpTitle = NULL;
58 LPTSTR lpDirectory = NULL;
59 LPTSTR lpEnvironment = NULL;
60 WORD wShowWindow = SW_SHOWNORMAL;
61
62 while (1)
63 {
64 if (_istspace(*Rest))
65 {
66 Rest++;
67 }
68 else if (*Rest == _T('"') && !lpTitle)
69 {
70 lpTitle = GetParameter(&Rest);
71 StripQuotes(lpTitle);
72 }
73 else if (*Rest == L'/')
74 {
75 LPTSTR option;
76 Rest++;
77 option = GetParameter(&Rest);
78 if (*option == _T('?'))
79 {
80 ConOutResPaging(TRUE,STRING_START_HELP1);
81 return 0;
82 }
83 else if (_totupper(*option) == _T('D'))
84 {
85 lpDirectory = option + 1;
86 if (!*lpDirectory)
87 {
88 while (_istspace(*Rest))
89 Rest++;
90 lpDirectory = GetParameter(&Rest);
91 }
92 StripQuotes(lpDirectory);
93 }
94 else if (_totupper(*option) == _T('I'))
95 {
96 /* rest of the option is apparently ignored */
97 lpEnvironment = lpOriginalEnvironment;
98 }
99 else if (!_tcsicmp(option, _T("MIN")))
100 {
101 wShowWindow = SW_MINIMIZE;
102 }
103 else if (!_tcsicmp(option, _T("MAX")))
104 {
105 wShowWindow = SW_MAXIMIZE;
106 }
107 else if (!_tcsicmp(option, _T("AFFINITY")))
108 {
109 TCHAR *end;
110 while (_istspace(*Rest))
111 Rest++;
112 option = GetParameter(&Rest);
113 /* Affinity mask is given in hexadecimal */
114 dwAffinityMask = _tcstoul(option, &end, 16);
115 if (*end != _T('\0') || dwAffinityMask == 0 ||
116 dwAffinityMask == (DWORD)-1)
117 {
118 ConErrResPrintf(STRING_ERROR_INVALID_PARAM_FORMAT, option);
119 return 1;
120 }
121 dwCreationFlags |= CREATE_SUSPENDED;
122 }
123 else if (!_tcsicmp(option, _T("B")))
124 {
125 dwCreationFlags &= ~CREATE_NEW_CONSOLE;
126 dwCreationFlags |= CREATE_NEW_PROCESS_GROUP;
127 }
128 else if (!_tcsicmp(option, _T("LOW")))
129 {
130 dwCreationFlags |= IDLE_PRIORITY_CLASS;
131 }
132 else if (!_tcsicmp(option, _T("NORMAL")))
133 {
134 dwCreationFlags |= NORMAL_PRIORITY_CLASS;
135 }
136 else if (!_tcsicmp(option, _T("HIGH")))
137 {
138 dwCreationFlags |= HIGH_PRIORITY_CLASS;
139 }
140 else if (!_tcsicmp(option, _T("REALTIME")))
141 {
142 dwCreationFlags |= REALTIME_PRIORITY_CLASS;
143 }
144 else if (!_tcsicmp(option, _T("ABOVENORMAL")))
145 {
146 dwCreationFlags |= ABOVE_NORMAL_PRIORITY_CLASS;
147 }
148 else if (!_tcsicmp(option, _T("BELOWNORMAL")))
149 {
150 dwCreationFlags |= BELOW_NORMAL_PRIORITY_CLASS;
151 }
152 else if (!_tcsicmp(option, _T("SEPARATE")))
153 {
154 dwCreationFlags |= CREATE_SEPARATE_WOW_VDM;
155 }
156 else if (!_tcsicmp(option, _T("SHARED")))
157 {
158 dwCreationFlags |= CREATE_SHARED_WOW_VDM;
159 }
160 else if (!_tcsicmp(option, _T("W")) ||
161 !_tcsicmp(option, _T("WAIT")))
162 {
163 bWait = TRUE;
164 }
165 else
166 {
167 ConErrResPrintf(STRING_TYPE_ERROR1, option);
168 return 0;
169 }
170 }
171 else
172 {
173 /* It's not an option - must be the beginning of
174 * the actual command. Leave the loop. */
175 break;
176 }
177 }
178
179 /* get comspec */
180 comspec = cmd_alloc ( MAX_PATH * sizeof(TCHAR));
181 if (comspec == NULL)
182 {
183 error_out_of_memory();
184 return 1;
185 }
186 SetLastError(0);
187 size = GetEnvironmentVariable (_T("COMSPEC"), comspec, MAX_PATH);
188 if (GetLastError() == ERROR_ENVVAR_NOT_FOUND)
189 {
190 _tcscpy(comspec, _T("cmd"));
191 }
192 else
193 {
194 if (size > MAX_PATH)
195 {
196 LPTSTR Oldcomspec = comspec;
197 comspec = cmd_realloc(comspec,size * sizeof(TCHAR) );
198 if (comspec==NULL)
199 {
200 cmd_free(Oldcomspec);
201 return 1;
202 }
203 size = GetEnvironmentVariable (_T("COMSPEC"), comspec, size);
204 }
205 }
206
207 nErrorLevel = 0;
208
209 if (!*Rest)
210 {
211 Rest = _T("cmd.exe");
212 }
213 else
214 /* Parsing the command that gets called by start, and it's parameters */
215 {
216 BOOL bInside = FALSE;
217 INT i;
218 /* find the end of the command and put the arguments in param */
219 for (i = 0; Rest[i]; i++)
220 {
221 if (Rest[i] == _T('\"'))
222 bInside = !bInside;
223 if (_istspace(Rest[i]) && !bInside)
224 {
225 param = &Rest[i+1];
226 Rest[i] = _T('\0');
227 break;
228 }
229 }
230 }
231
232 _tcscpy(szUnquotedName, Rest);
233 StripQuotes(szUnquotedName);
234
235 /* get the PATH environment variable and parse it */
236 /* search the PATH environment variable for the binary */
237 if (SearchForExecutable(szUnquotedName, szFullName))
238 {
239 /* check if this is a .BAT or .CMD file */
240 dot = _tcsrchr(szFullName, _T('.'));
241 if (dot && (!_tcsicmp(dot, _T(".bat")) || !_tcsicmp(dot, _T(".cmd"))))
242 {
243 bBat = TRUE;
244 _stprintf(szFullCmdLine, _T("\"%s\" /K %s"), comspec, Rest);
245 TRACE ("[BATCH: %s %s]\n", debugstr_aw(szFullName), debugstr_aw(Rest));
246 }
247 else
248 {
249 TRACE ("[EXEC: %s %s]\n", debugstr_aw(szFullName), debugstr_aw(Rest));
250 _tcscpy(szFullCmdLine, szFullName);
251 }
252
253 /* build command line for CreateProcess() */
254 if (param != NULL)
255 {
256 _tcsncat(szFullCmdLine, _T(" "), CMDLINE_LENGTH - _tcslen(szFullCmdLine));
257 _tcsncat(szFullCmdLine, param, CMDLINE_LENGTH - _tcslen(szFullCmdLine));
258 }
259
260 /* fill startup info */
261 memset (&stui, 0, sizeof (STARTUPINFO));
262 stui.cb = sizeof (STARTUPINFO);
263 stui.dwFlags = STARTF_USESHOWWINDOW;
264 stui.lpTitle = lpTitle;
265 stui.wShowWindow = wShowWindow;
266
267 bCreate = CreateProcess(bBat ? comspec : szFullName,
268 szFullCmdLine, NULL, NULL, TRUE, dwCreationFlags,
269 lpEnvironment, lpDirectory, &stui, &prci);
270 if (bCreate)
271 {
272 if (dwAffinityMask)
273 {
274 SetProcessAffinityMask(prci.hProcess, dwAffinityMask);
275 ResumeThread(prci.hThread);
276 }
277 CloseHandle(prci.hThread);
278 }
279 }
280 else
281 {
282 /* The file name did not seem to be valid, but maybe it's actually a
283 * directory or URL, so we still want to pass it to ShellExecute. */
284 _tcscpy(szFullName, szUnquotedName);
285 }
286
287 if (!bCreate)
288 {
289 /* CreateProcess didn't work; try ShellExecute */
290 DWORD flags = SEE_MASK_NOCLOSEPROCESS;
291 if (!(dwCreationFlags & CREATE_NEW_CONSOLE))
292 flags |= SEE_MASK_NO_CONSOLE;
293 prci.hProcess = RunFile(flags, szFullName, param, lpDirectory, wShowWindow);
294 }
295
296 if (prci.hProcess != NULL)
297 {
298 if (bWait)
299 {
300 DWORD dwExitCode;
301 WaitForSingleObject (prci.hProcess, INFINITE);
302 GetExitCodeProcess (prci.hProcess, &dwExitCode);
303 nErrorLevel = (INT)dwExitCode;
304 }
305 CloseHandle (prci.hProcess);
306
307 /* Get New code page if it has change */
308 InputCodePage= GetConsoleCP();
309 OutputCodePage = GetConsoleOutputCP();
310 }
311 else
312 {
313 ErrorMessage(GetLastError (),
314 _T("Error executing CreateProcess()!!\n"));
315 }
316
317 cmd_free(comspec);
318 return 0;
319 }
320
321 #endif
322
323 /* EOF */