setupapi: Refactor freeing the INF file into free_inf_file. Use this new function...
[wine/multimedia.git] / programs / cmd / wcmdmain.c
blobe94cb3bb706ea87bb991135cfdab3d211b4ab45f
1 /*
2 * CMD - Wine-compatible command line interface.
4 * Copyright (C) 1999 - 2001 D A Pickles
5 * Copyright (C) 2007 J Edmeades
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 * FIXME:
24 * - Cannot handle parameters in quotes
25 * - Lots of functionality missing from builtins
28 #include "config.h"
29 #include "wcmd.h"
30 #include "wine/debug.h"
32 WINE_DEFAULT_DEBUG_CHANNEL(cmd);
34 const WCHAR inbuilt[][10] = {
35 {'A','T','T','R','I','B','\0'},
36 {'C','A','L','L','\0'},
37 {'C','D','\0'},
38 {'C','H','D','I','R','\0'},
39 {'C','L','S','\0'},
40 {'C','O','P','Y','\0'},
41 {'C','T','T','Y','\0'},
42 {'D','A','T','E','\0'},
43 {'D','E','L','\0'},
44 {'D','I','R','\0'},
45 {'E','C','H','O','\0'},
46 {'E','R','A','S','E','\0'},
47 {'F','O','R','\0'},
48 {'G','O','T','O','\0'},
49 {'H','E','L','P','\0'},
50 {'I','F','\0'},
51 {'L','A','B','E','L','\0'},
52 {'M','D','\0'},
53 {'M','K','D','I','R','\0'},
54 {'M','O','V','E','\0'},
55 {'P','A','T','H','\0'},
56 {'P','A','U','S','E','\0'},
57 {'P','R','O','M','P','T','\0'},
58 {'R','E','M','\0'},
59 {'R','E','N','\0'},
60 {'R','E','N','A','M','E','\0'},
61 {'R','D','\0'},
62 {'R','M','D','I','R','\0'},
63 {'S','E','T','\0'},
64 {'S','H','I','F','T','\0'},
65 {'T','I','M','E','\0'},
66 {'T','I','T','L','E','\0'},
67 {'T','Y','P','E','\0'},
68 {'V','E','R','I','F','Y','\0'},
69 {'V','E','R','\0'},
70 {'V','O','L','\0'},
71 {'E','N','D','L','O','C','A','L','\0'},
72 {'S','E','T','L','O','C','A','L','\0'},
73 {'P','U','S','H','D','\0'},
74 {'P','O','P','D','\0'},
75 {'A','S','S','O','C','\0'},
76 {'C','O','L','O','R','\0'},
77 {'F','T','Y','P','E','\0'},
78 {'M','O','R','E','\0'},
79 {'E','X','I','T','\0'}
82 HINSTANCE hinst;
83 DWORD errorlevel;
84 int echo_mode = 1, verify_mode = 0, defaultColor = 7;
85 static int opt_c, opt_k, opt_s;
86 const WCHAR newline[] = {'\r','\n','\0'};
87 static const WCHAR equalsW[] = {'=','\0'};
88 static const WCHAR closeBW[] = {')','\0'};
89 WCHAR anykey[100];
90 WCHAR version_string[100];
91 WCHAR quals[MAX_PATH], param1[MAXSTRING], param2[MAXSTRING];
92 BATCH_CONTEXT *context = NULL;
93 extern struct env_stack *pushd_directories;
94 static const WCHAR *pagedMessage = NULL;
95 static char *output_bufA = NULL;
96 #define MAX_WRITECONSOLE_SIZE 65535
97 BOOL unicodePipes = FALSE;
100 /*******************************************************************
101 * WCMD_output_asis_len - send output to current standard output
103 * Output a formatted unicode string. Ideally this will go to the console
104 * and hence required WriteConsoleW to output it, however if file i/o is
105 * redirected, it needs to be WriteFile'd using OEM (not ANSI) format
107 static void WCMD_output_asis_len(const WCHAR *message, int len, HANDLE device) {
109 DWORD nOut= 0;
110 DWORD res = 0;
112 /* If nothing to write, return (MORE does this sometimes) */
113 if (!len) return;
115 /* Try to write as unicode assuming it is to a console */
116 res = WriteConsoleW(device, message, len, &nOut, NULL);
118 /* If writing to console fails, assume its file
119 i/o so convert to OEM codepage and output */
120 if (!res) {
121 BOOL usedDefaultChar = FALSE;
122 DWORD convertedChars;
124 if (!unicodePipes) {
126 * Allocate buffer to use when writing to file. (Not freed, as one off)
128 if (!output_bufA) output_bufA = HeapAlloc(GetProcessHeap(), 0,
129 MAX_WRITECONSOLE_SIZE);
130 if (!output_bufA) {
131 WINE_FIXME("Out of memory - could not allocate ansi 64K buffer\n");
132 return;
135 /* Convert to OEM, then output */
136 convertedChars = WideCharToMultiByte(GetConsoleOutputCP(), 0, message,
137 len, output_bufA, MAX_WRITECONSOLE_SIZE,
138 "?", &usedDefaultChar);
139 WriteFile(device, output_bufA, convertedChars,
140 &nOut, FALSE);
141 } else {
142 WriteFile(device, message, len*sizeof(WCHAR),
143 &nOut, FALSE);
146 return;
149 /*******************************************************************
150 * WCMD_output - send output to current standard output device.
154 void WCMD_output (const WCHAR *format, ...) {
156 va_list ap;
157 WCHAR string[1024];
158 int ret;
160 va_start(ap,format);
161 ret = vsnprintfW(string, sizeof(string)/sizeof(WCHAR), format, ap);
162 if( ret >= (sizeof(string)/sizeof(WCHAR))) {
163 WINE_ERR("Output truncated in WCMD_output\n" );
164 ret = (sizeof(string)/sizeof(WCHAR)) - 1;
165 string[ret] = '\0';
167 va_end(ap);
168 WCMD_output_asis_len(string, ret, GetStdHandle(STD_OUTPUT_HANDLE));
172 static int line_count;
173 static int max_height;
174 static int max_width;
175 static BOOL paged_mode;
176 static int numChars;
178 void WCMD_enter_paged_mode(const WCHAR *msg)
180 CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
182 if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo)) {
183 max_height = consoleInfo.dwSize.Y;
184 max_width = consoleInfo.dwSize.X;
185 } else {
186 max_height = 25;
187 max_width = 80;
189 paged_mode = TRUE;
190 line_count = 0;
191 numChars = 0;
192 pagedMessage = (msg==NULL)? anykey : msg;
195 void WCMD_leave_paged_mode(void)
197 paged_mode = FALSE;
198 pagedMessage = NULL;
201 /***************************************************************************
202 * WCMD_Readfile
204 * Read characters in from a console/file, returning result in Unicode
205 * with signature identical to ReadFile
207 BOOL WCMD_ReadFile(const HANDLE hIn, WCHAR *intoBuf, const DWORD maxChars,
208 LPDWORD charsRead, const LPOVERLAPPED unused) {
210 BOOL res;
212 /* Try to read from console as Unicode */
213 res = ReadConsoleW(hIn, intoBuf, maxChars, charsRead, NULL);
215 /* If reading from console has failed we assume its file
216 i/o so read in and convert from OEM codepage */
217 if (!res) {
219 DWORD numRead;
221 * Allocate buffer to use when reading from file. Not freed
223 if (!output_bufA) output_bufA = HeapAlloc(GetProcessHeap(), 0,
224 MAX_WRITECONSOLE_SIZE);
225 if (!output_bufA) {
226 WINE_FIXME("Out of memory - could not allocate ansi 64K buffer\n");
227 return 0;
230 /* Read from file (assume OEM codepage) */
231 res = ReadFile(hIn, output_bufA, maxChars, &numRead, unused);
233 /* Convert from OEM */
234 *charsRead = MultiByteToWideChar(GetConsoleCP(), 0, output_bufA, numRead,
235 intoBuf, maxChars);
238 return res;
241 /*******************************************************************
242 * WCMD_output_asis - send output to current standard output device.
243 * without formatting eg. when message contains '%'
245 void WCMD_output_asis (const WCHAR *message) {
246 DWORD count;
247 const WCHAR* ptr;
248 WCHAR string[1024];
250 if (paged_mode) {
251 do {
252 ptr = message;
253 while (*ptr && *ptr!='\n' && (numChars < max_width)) {
254 numChars++;
255 ptr++;
257 if (*ptr == '\n') ptr++;
258 WCMD_output_asis_len(message, (ptr) ? ptr - message : strlenW(message),
259 GetStdHandle(STD_OUTPUT_HANDLE));
260 if (ptr) {
261 numChars = 0;
262 if (++line_count >= max_height - 1) {
263 line_count = 0;
264 WCMD_output_asis_len(pagedMessage, strlenW(pagedMessage),
265 GetStdHandle(STD_OUTPUT_HANDLE));
266 WCMD_ReadFile (GetStdHandle(STD_INPUT_HANDLE), string,
267 sizeof(string)/sizeof(WCHAR), &count, NULL);
270 } while (((message = ptr) != NULL) && (*ptr));
271 } else {
272 WCMD_output_asis_len(message, lstrlenW(message),
273 GetStdHandle(STD_OUTPUT_HANDLE));
277 /****************************************************************************
278 * WCMD_print_error
280 * Print the message for GetLastError
283 void WCMD_print_error (void) {
284 LPVOID lpMsgBuf;
285 DWORD error_code;
286 int status;
288 error_code = GetLastError ();
289 status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
290 NULL, error_code, 0, (LPWSTR) &lpMsgBuf, 0, NULL);
291 if (!status) {
292 WINE_FIXME ("Cannot display message for error %d, status %d\n",
293 error_code, GetLastError());
294 return;
297 WCMD_output_asis_len(lpMsgBuf, lstrlenW(lpMsgBuf),
298 GetStdHandle(STD_ERROR_HANDLE));
299 LocalFree (lpMsgBuf);
300 WCMD_output_asis_len (newline, lstrlenW(newline),
301 GetStdHandle(STD_ERROR_HANDLE));
302 return;
305 /******************************************************************************
306 * WCMD_show_prompt
308 * Display the prompt on STDout
312 static void WCMD_show_prompt (void) {
314 int status;
315 WCHAR out_string[MAX_PATH], curdir[MAX_PATH], prompt_string[MAX_PATH];
316 WCHAR *p, *q;
317 DWORD len;
318 static const WCHAR envPrompt[] = {'P','R','O','M','P','T','\0'};
320 len = GetEnvironmentVariableW(envPrompt, prompt_string,
321 sizeof(prompt_string)/sizeof(WCHAR));
322 if ((len == 0) || (len >= (sizeof(prompt_string)/sizeof(WCHAR)))) {
323 const WCHAR dfltPrompt[] = {'$','P','$','G','\0'};
324 strcpyW (prompt_string, dfltPrompt);
326 p = prompt_string;
327 q = out_string;
328 *q++ = '\r';
329 *q++ = '\n';
330 *q = '\0';
331 while (*p != '\0') {
332 if (*p != '$') {
333 *q++ = *p++;
334 *q = '\0';
336 else {
337 p++;
338 switch (toupper(*p)) {
339 case '$':
340 *q++ = '$';
341 break;
342 case 'A':
343 *q++ = '&';
344 break;
345 case 'B':
346 *q++ = '|';
347 break;
348 case 'C':
349 *q++ = '(';
350 break;
351 case 'D':
352 GetDateFormatW(LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL, NULL, q, MAX_PATH);
353 while (*q) q++;
354 break;
355 case 'E':
356 *q++ = '\E';
357 break;
358 case 'F':
359 *q++ = ')';
360 break;
361 case 'G':
362 *q++ = '>';
363 break;
364 case 'H':
365 *q++ = '\b';
366 break;
367 case 'L':
368 *q++ = '<';
369 break;
370 case 'N':
371 status = GetCurrentDirectoryW(sizeof(curdir)/sizeof(WCHAR), curdir);
372 if (status) {
373 *q++ = curdir[0];
375 break;
376 case 'P':
377 status = GetCurrentDirectoryW(sizeof(curdir)/sizeof(WCHAR), curdir);
378 if (status) {
379 strcatW (q, curdir);
380 while (*q) q++;
382 break;
383 case 'Q':
384 *q++ = '=';
385 break;
386 case 'S':
387 *q++ = ' ';
388 break;
389 case 'T':
390 GetTimeFormatW(LOCALE_USER_DEFAULT, 0, NULL, NULL, q, MAX_PATH);
391 while (*q) q++;
392 break;
393 case 'V':
394 strcatW (q, version_string);
395 while (*q) q++;
396 break;
397 case '_':
398 *q++ = '\n';
399 break;
400 case '+':
401 if (pushd_directories) {
402 memset(q, '+', pushd_directories->u.stackdepth);
403 q = q + pushd_directories->u.stackdepth;
405 break;
407 p++;
408 *q = '\0';
411 WCMD_output_asis (out_string);
415 /*************************************************************************
416 * WCMD_strdupW
417 * A wide version of strdup as its missing from unicode.h
419 WCHAR *WCMD_strdupW(WCHAR *input) {
420 int len=strlenW(input)+1;
421 WCHAR *result = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
422 memcpy(result, input, len * sizeof(WCHAR));
423 return result;
426 /*************************************************************************
427 * WCMD_strsubstW
428 * Replaces a portion of a Unicode string with the specified string.
429 * It's up to the caller to ensure there is enough space in the
430 * destination buffer.
432 void WCMD_strsubstW(WCHAR* start, WCHAR* next, WCHAR* insert, int len) {
434 if (len < 0)
435 len=insert ? lstrlenW(insert) : 0;
436 if (start+len != next)
437 memmove(start+len, next, (strlenW(next) + 1) * sizeof(*next));
438 if (insert)
439 memcpy(start, insert, len * sizeof(*insert));
442 /***************************************************************************
443 * WCMD_strtrim_leading_spaces
445 * Remove leading spaces from a string. Return a pointer to the first
446 * non-space character. Does not modify the input string
448 WCHAR *WCMD_strtrim_leading_spaces (WCHAR *string) {
450 WCHAR *ptr;
452 ptr = string;
453 while (*ptr == ' ') ptr++;
454 return ptr;
457 /*************************************************************************
458 * WCMD_opt_s_strip_quotes
460 * Remove first and last quote WCHARacters, preserving all other text
462 void WCMD_opt_s_strip_quotes(WCHAR *cmd) {
463 WCHAR *src = cmd + 1, *dest = cmd, *lastq = NULL;
464 while((*dest=*src) != '\0') {
465 if (*src=='\"')
466 lastq=dest;
467 dest++, src++;
469 if (lastq) {
470 dest=lastq++;
471 while ((*dest++=*lastq++) != 0)
477 /*************************************************************************
478 * WCMD_expand_envvar
480 * Expands environment variables, allowing for WCHARacter substitution
482 static WCHAR *WCMD_expand_envvar(WCHAR *start, WCHAR *forVar, WCHAR *forVal) {
483 WCHAR *endOfVar = NULL, *s;
484 WCHAR *colonpos = NULL;
485 WCHAR thisVar[MAXSTRING];
486 WCHAR thisVarContents[MAXSTRING];
487 WCHAR savedchar = 0x00;
488 int len;
490 static const WCHAR ErrorLvl[] = {'E','R','R','O','R','L','E','V','E','L','\0'};
491 static const WCHAR ErrorLvlP[] = {'%','E','R','R','O','R','L','E','V','E','L','%','\0'};
492 static const WCHAR Date[] = {'D','A','T','E','\0'};
493 static const WCHAR DateP[] = {'%','D','A','T','E','%','\0'};
494 static const WCHAR Time[] = {'T','I','M','E','\0'};
495 static const WCHAR TimeP[] = {'%','T','I','M','E','%','\0'};
496 static const WCHAR Cd[] = {'C','D','\0'};
497 static const WCHAR CdP[] = {'%','C','D','%','\0'};
498 static const WCHAR Random[] = {'R','A','N','D','O','M','\0'};
499 static const WCHAR RandomP[] = {'%','R','A','N','D','O','M','%','\0'};
500 static const WCHAR Delims[] = {'%',' ',':','\0'};
502 WINE_TRACE("Expanding: %s (%s,%s)\n", wine_dbgstr_w(start),
503 wine_dbgstr_w(forVal), wine_dbgstr_w(forVar));
505 /* Find the end of the environment variable, and extract name */
506 endOfVar = strpbrkW(start+1, Delims);
508 if (endOfVar == NULL || *endOfVar==' ') {
510 /* In batch program, missing terminator for % and no following
511 ':' just removes the '%' */
512 if (context) {
513 WCMD_strsubstW(start, start + 1, NULL, 0);
514 return start;
515 } else {
517 /* In command processing, just ignore it - allows command line
518 syntax like: for %i in (a.a) do echo %i */
519 return start+1;
523 /* If ':' found, process remaining up until '%' (or stop at ':' if
524 a missing '%' */
525 if (*endOfVar==':') {
526 WCHAR *endOfVar2 = strchrW(endOfVar+1, '%');
527 if (endOfVar2 != NULL) endOfVar = endOfVar2;
530 memcpy(thisVar, start, ((endOfVar - start) + 1) * sizeof(WCHAR));
531 thisVar[(endOfVar - start)+1] = 0x00;
532 colonpos = strchrW(thisVar+1, ':');
534 /* If there's complex substitution, just need %var% for now
535 to get the expanded data to play with */
536 if (colonpos) {
537 *colonpos = '%';
538 savedchar = *(colonpos+1);
539 *(colonpos+1) = 0x00;
542 WINE_TRACE("Retrieving contents of %s\n", wine_dbgstr_w(thisVar));
544 /* Expand to contents, if unchanged, return */
545 /* Handle DATE, TIME, ERRORLEVEL and CD replacements allowing */
546 /* override if existing env var called that name */
547 if ((CompareStringW(LOCALE_USER_DEFAULT,
548 NORM_IGNORECASE | SORT_STRINGSORT,
549 thisVar, 12, ErrorLvlP, -1) == 2) &&
550 (GetEnvironmentVariableW(ErrorLvl, thisVarContents, 1) == 0) &&
551 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
552 static const WCHAR fmt[] = {'%','d','\0'};
553 wsprintfW(thisVarContents, fmt, errorlevel);
554 len = strlenW(thisVarContents);
556 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
557 NORM_IGNORECASE | SORT_STRINGSORT,
558 thisVar, 6, DateP, -1) == 2) &&
559 (GetEnvironmentVariableW(Date, thisVarContents, 1) == 0) &&
560 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
562 GetDateFormatW(LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL,
563 NULL, thisVarContents, MAXSTRING);
564 len = strlenW(thisVarContents);
566 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
567 NORM_IGNORECASE | SORT_STRINGSORT,
568 thisVar, 6, TimeP, -1) == 2) &&
569 (GetEnvironmentVariableW(Time, thisVarContents, 1) == 0) &&
570 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
571 GetTimeFormatW(LOCALE_USER_DEFAULT, TIME_NOSECONDS, NULL,
572 NULL, thisVarContents, MAXSTRING);
573 len = strlenW(thisVarContents);
575 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
576 NORM_IGNORECASE | SORT_STRINGSORT,
577 thisVar, 4, CdP, -1) == 2) &&
578 (GetEnvironmentVariableW(Cd, thisVarContents, 1) == 0) &&
579 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
580 GetCurrentDirectoryW(MAXSTRING, thisVarContents);
581 len = strlenW(thisVarContents);
583 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
584 NORM_IGNORECASE | SORT_STRINGSORT,
585 thisVar, 8, RandomP, -1) == 2) &&
586 (GetEnvironmentVariableW(Random, thisVarContents, 1) == 0) &&
587 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
588 static const WCHAR fmt[] = {'%','d','\0'};
589 wsprintfW(thisVarContents, fmt, rand() % 32768);
590 len = strlenW(thisVarContents);
592 /* Look for a matching 'for' variable */
593 } else if (forVar &&
594 (CompareStringW(LOCALE_USER_DEFAULT,
595 SORT_STRINGSORT,
596 thisVar,
597 (colonpos - thisVar) - 1,
598 forVar, -1) == 2)) {
599 strcpyW(thisVarContents, forVal);
600 len = strlenW(thisVarContents);
602 } else {
604 len = ExpandEnvironmentStringsW(thisVar, thisVarContents,
605 sizeof(thisVarContents)/sizeof(WCHAR));
608 if (len == 0)
609 return endOfVar+1;
611 /* In a batch program, unknown env vars are replaced with nothing,
612 note syntax %garbage:1,3% results in anything after the ':'
613 except the %
614 From the command line, you just get back what you entered */
615 if (lstrcmpiW(thisVar, thisVarContents) == 0) {
617 /* Restore the complex part after the compare */
618 if (colonpos) {
619 *colonpos = ':';
620 *(colonpos+1) = savedchar;
623 /* Command line - just ignore this */
624 if (context == NULL) return endOfVar+1;
627 /* Batch - replace unknown env var with nothing */
628 if (colonpos == NULL) {
629 WCMD_strsubstW(start, endOfVar + 1, NULL, 0);
630 } else {
631 len = strlenW(thisVar);
632 thisVar[len-1] = 0x00;
633 /* If %:...% supplied, : is retained */
634 if (colonpos == thisVar+1) {
635 WCMD_strsubstW(start, endOfVar + 1, colonpos, -1);
636 } else {
637 WCMD_strsubstW(start, endOfVar + 1, colonpos + 1, -1);
640 return start;
644 /* See if we need to do complex substitution (any ':'s), if not
645 then our work here is done */
646 if (colonpos == NULL) {
647 WCMD_strsubstW(start, endOfVar + 1, thisVarContents, -1);
648 return start;
651 /* Restore complex bit */
652 *colonpos = ':';
653 *(colonpos+1) = savedchar;
656 Handle complex substitutions:
657 xxx=yyy (replace xxx with yyy)
658 *xxx=yyy (replace up to and including xxx with yyy)
659 ~x (from x WCHARs in)
660 ~-x (from x WCHARs from the end)
661 ~x,y (from x WCHARs in for y WCHARacters)
662 ~x,-y (from x WCHARs in until y WCHARacters from the end)
665 /* ~ is substring manipulation */
666 if (savedchar == '~') {
668 int substrposition, substrlength = 0;
669 WCHAR *commapos = strchrW(colonpos+2, ',');
670 WCHAR *startCopy;
672 substrposition = atolW(colonpos+2);
673 if (commapos) substrlength = atolW(commapos+1);
675 /* Check bounds */
676 if (substrposition >= 0) {
677 startCopy = &thisVarContents[min(substrposition, len)];
678 } else {
679 startCopy = &thisVarContents[max(0, len+substrposition-1)];
682 if (commapos == NULL) {
683 /* Copy the lot */
684 WCMD_strsubstW(start, endOfVar + 1, startCopy, -1);
685 } else if (substrlength < 0) {
687 int copybytes = (len+substrlength-1)-(startCopy-thisVarContents);
688 if (copybytes > len) copybytes = len;
689 else if (copybytes < 0) copybytes = 0;
690 WCMD_strsubstW(start, endOfVar + 1, startCopy, copybytes);
691 } else {
692 WCMD_strsubstW(start, endOfVar + 1, startCopy, substrlength);
695 return start;
697 /* search and replace manipulation */
698 } else {
699 WCHAR *equalspos = strstrW(colonpos, equalsW);
700 WCHAR *replacewith = equalspos+1;
701 WCHAR *found = NULL;
702 WCHAR *searchIn;
703 WCHAR *searchFor;
705 if (equalspos == NULL) return start+1;
706 s = WCMD_strdupW(endOfVar + 1);
708 /* Null terminate both strings */
709 thisVar[strlenW(thisVar)-1] = 0x00;
710 *equalspos = 0x00;
712 /* Since we need to be case insensitive, copy the 2 buffers */
713 searchIn = WCMD_strdupW(thisVarContents);
714 CharUpperBuffW(searchIn, strlenW(thisVarContents));
715 searchFor = WCMD_strdupW(colonpos+1);
716 CharUpperBuffW(searchFor, strlenW(colonpos+1));
718 /* Handle wildcard case */
719 if (*(colonpos+1) == '*') {
720 /* Search for string to replace */
721 found = strstrW(searchIn, searchFor+1);
723 if (found) {
724 /* Do replacement */
725 strcpyW(start, replacewith);
726 strcatW(start, thisVarContents + (found-searchIn) + strlenW(searchFor+1));
727 strcatW(start, s);
728 } else {
729 /* Copy as is */
730 strcpyW(start, thisVarContents);
731 strcatW(start, s);
734 } else {
735 /* Loop replacing all instances */
736 WCHAR *lastFound = searchIn;
737 WCHAR *outputposn = start;
739 *start = 0x00;
740 while ((found = strstrW(lastFound, searchFor))) {
741 lstrcpynW(outputposn,
742 thisVarContents + (lastFound-searchIn),
743 (found - lastFound)+1);
744 outputposn = outputposn + (found - lastFound);
745 strcatW(outputposn, replacewith);
746 outputposn = outputposn + strlenW(replacewith);
747 lastFound = found + strlenW(searchFor);
749 strcatW(outputposn,
750 thisVarContents + (lastFound-searchIn));
751 strcatW(outputposn, s);
753 HeapFree(GetProcessHeap(), 0, s);
754 HeapFree(GetProcessHeap(), 0, searchIn);
755 HeapFree(GetProcessHeap(), 0, searchFor);
756 return start;
758 return start+1;
761 /*****************************************************************************
762 * Expand the command. Native expands lines from batch programs as they are
763 * read in and not again, except for 'for' variable substitution.
764 * eg. As evidence, "echo %1 && shift && echo %1" or "echo %%path%%"
766 static void handleExpansion(WCHAR *cmd, BOOL justFors, WCHAR *forVariable, WCHAR *forValue) {
768 /* For commands in a context (batch program): */
769 /* Expand environment variables in a batch file %{0-9} first */
770 /* including support for any ~ modifiers */
771 /* Additionally: */
772 /* Expand the DATE, TIME, CD, RANDOM and ERRORLEVEL special */
773 /* names allowing environment variable overrides */
774 /* NOTE: To support the %PATH:xxx% syntax, also perform */
775 /* manual expansion of environment variables here */
777 WCHAR *p = cmd;
778 WCHAR *t;
779 int i;
781 while ((p = strchrW(p, '%'))) {
783 WINE_TRACE("Translate command:%s %d (at: %s)\n",
784 wine_dbgstr_w(cmd), justFors, wine_dbgstr_w(p));
785 i = *(p+1) - '0';
787 /* Don't touch %% unless its in Batch */
788 if (!justFors && *(p+1) == '%') {
789 if (context) {
790 WCMD_strsubstW(p, p+1, NULL, 0);
792 p+=1;
794 /* Replace %~ modifications if in batch program */
795 } else if (*(p+1) == '~') {
796 WCMD_HandleTildaModifiers(&p, forVariable, forValue, justFors);
797 p++;
799 /* Replace use of %0...%9 if in batch program*/
800 } else if (!justFors && context && (i >= 0) && (i <= 9)) {
801 t = WCMD_parameter (context -> command, i + context -> shift_count[i], NULL);
802 WCMD_strsubstW(p, p+2, t, -1);
804 /* Replace use of %* if in batch program*/
805 } else if (!justFors && context && *(p+1)=='*') {
806 WCHAR *startOfParms = NULL;
807 t = WCMD_parameter (context -> command, 1, &startOfParms);
808 if (startOfParms != NULL)
809 WCMD_strsubstW(p, p+2, startOfParms, -1);
810 else
811 WCMD_strsubstW(p, p+2, NULL, 0);
813 } else if (forVariable &&
814 (CompareStringW(LOCALE_USER_DEFAULT,
815 SORT_STRINGSORT,
817 strlenW(forVariable),
818 forVariable, -1) == 2)) {
819 WCMD_strsubstW(p, p + strlenW(forVariable), forValue, -1);
821 } else if (!justFors) {
822 p = WCMD_expand_envvar(p, forVariable, forValue);
824 /* In a FOR loop, see if this is the variable to replace */
825 } else { /* Ignore %'s on second pass of batch program */
826 p++;
830 return;
834 /*******************************************************************
835 * WCMD_parse - parse a command into parameters and qualifiers.
837 * On exit, all qualifiers are concatenated into q, the first string
838 * not beginning with "/" is in p1 and the
839 * second in p2. Any subsequent non-qualifier strings are lost.
840 * Parameters in quotes are handled.
842 static void WCMD_parse (WCHAR *s, WCHAR *q, WCHAR *p1, WCHAR *p2)
844 int p = 0;
846 *q = *p1 = *p2 = '\0';
847 while (TRUE) {
848 switch (*s) {
849 case '/':
850 *q++ = *s++;
851 while ((*s != '\0') && (*s != ' ') && *s != '/') {
852 *q++ = toupperW (*s++);
854 *q = '\0';
855 break;
856 case ' ':
857 case '\t':
858 s++;
859 break;
860 case '"':
861 s++;
862 while ((*s != '\0') && (*s != '"')) {
863 if (p == 0) *p1++ = *s++;
864 else if (p == 1) *p2++ = *s++;
865 else s++;
867 if (p == 0) *p1 = '\0';
868 if (p == 1) *p2 = '\0';
869 p++;
870 if (*s == '"') s++;
871 break;
872 case '\0':
873 return;
874 default:
875 while ((*s != '\0') && (*s != ' ') && (*s != '\t')
876 && (*s != '=') && (*s != ',') ) {
877 if (p == 0) *p1++ = *s++;
878 else if (p == 1) *p2++ = *s++;
879 else s++;
881 /* Skip concurrent parms */
882 while ((*s == ' ') || (*s == '\t') || (*s == '=') || (*s == ',') ) s++;
884 if (p == 0) *p1 = '\0';
885 if (p == 1) *p2 = '\0';
886 p++;
891 static void init_msvcrt_io_block(STARTUPINFOW* st)
893 STARTUPINFOW st_p;
894 /* fetch the parent MSVCRT info block if any, so that the child can use the
895 * same handles as its grand-father
897 st_p.cb = sizeof(STARTUPINFOW);
898 GetStartupInfoW(&st_p);
899 st->cbReserved2 = st_p.cbReserved2;
900 st->lpReserved2 = st_p.lpReserved2;
901 if (st_p.cbReserved2 && st_p.lpReserved2)
903 /* Override the entries for fd 0,1,2 if we happened
904 * to change those std handles (this depends on the way wcmd sets
905 * it's new input & output handles)
907 size_t sz = max(sizeof(unsigned) + (sizeof(char) + sizeof(HANDLE)) * 3, st_p.cbReserved2);
908 BYTE* ptr = HeapAlloc(GetProcessHeap(), 0, sz);
909 if (ptr)
911 unsigned num = *(unsigned*)st_p.lpReserved2;
912 char* flags = (char*)(ptr + sizeof(unsigned));
913 HANDLE* handles = (HANDLE*)(flags + num * sizeof(char));
915 memcpy(ptr, st_p.lpReserved2, st_p.cbReserved2);
916 st->cbReserved2 = sz;
917 st->lpReserved2 = ptr;
919 #define WX_OPEN 0x01 /* see dlls/msvcrt/file.c */
920 if (num <= 0 || (flags[0] & WX_OPEN))
922 handles[0] = GetStdHandle(STD_INPUT_HANDLE);
923 flags[0] |= WX_OPEN;
925 if (num <= 1 || (flags[1] & WX_OPEN))
927 handles[1] = GetStdHandle(STD_OUTPUT_HANDLE);
928 flags[1] |= WX_OPEN;
930 if (num <= 2 || (flags[2] & WX_OPEN))
932 handles[2] = GetStdHandle(STD_ERROR_HANDLE);
933 flags[2] |= WX_OPEN;
935 #undef WX_OPEN
940 /******************************************************************************
941 * WCMD_run_program
943 * Execute a command line as an external program. Must allow recursion.
945 * Precedence:
946 * Manual testing under windows shows PATHEXT plays a key part in this,
947 * and the search algorithm and precedence appears to be as follows.
949 * Search locations:
950 * If directory supplied on command, just use that directory
951 * If extension supplied on command, look for that explicit name first
952 * Otherwise, search in each directory on the path
953 * Precedence:
954 * If extension supplied on command, look for that explicit name first
955 * Then look for supplied name .* (even if extension supplied, so
956 * 'garbage.exe' will match 'garbage.exe.cmd')
957 * If any found, cycle through PATHEXT looking for name.exe one by one
958 * Launching
959 * Once a match has been found, it is launched - Code currently uses
960 * findexecutable to achieve this which is left untouched.
963 void WCMD_run_program (WCHAR *command, int called) {
965 WCHAR temp[MAX_PATH];
966 WCHAR pathtosearch[MAXSTRING];
967 WCHAR *pathposn;
968 WCHAR stemofsearch[MAX_PATH]; /* maximum allowed executable name is
969 MAX_PATH, including null character */
970 WCHAR *lastSlash;
971 WCHAR pathext[MAXSTRING];
972 BOOL extensionsupplied = FALSE;
973 BOOL launched = FALSE;
974 BOOL status;
975 BOOL assumeInternal = FALSE;
976 DWORD len;
977 static const WCHAR envPath[] = {'P','A','T','H','\0'};
978 static const WCHAR envPathExt[] = {'P','A','T','H','E','X','T','\0'};
979 static const WCHAR delims[] = {'/','\\',':','\0'};
981 WCMD_parse (command, quals, param1, param2); /* Quick way to get the filename */
982 if (!(*param1) && !(*param2))
983 return;
985 /* Calculate the search path and stem to search for */
986 if (strpbrkW (param1, delims) == NULL) { /* No explicit path given, search path */
987 static const WCHAR curDir[] = {'.',';','\0'};
988 strcpyW(pathtosearch, curDir);
989 len = GetEnvironmentVariableW(envPath, &pathtosearch[2], (sizeof(pathtosearch)/sizeof(WCHAR))-2);
990 if ((len == 0) || (len >= (sizeof(pathtosearch)/sizeof(WCHAR)) - 2)) {
991 static const WCHAR curDir[] = {'.','\0'};
992 strcpyW (pathtosearch, curDir);
994 if (strchrW(param1, '.') != NULL) extensionsupplied = TRUE;
995 if (strlenW(param1) >= MAX_PATH)
997 WCMD_output_asis(WCMD_LoadMessage(WCMD_LINETOOLONG));
998 return;
1001 strcpyW(stemofsearch, param1);
1003 } else {
1005 /* Convert eg. ..\fred to include a directory by removing file part */
1006 GetFullPathNameW(param1, sizeof(pathtosearch)/sizeof(WCHAR), pathtosearch, NULL);
1007 lastSlash = strrchrW(pathtosearch, '\\');
1008 if (lastSlash && strchrW(lastSlash, '.') != NULL) extensionsupplied = TRUE;
1009 strcpyW(stemofsearch, lastSlash+1);
1011 /* Reduce pathtosearch to a path with trailing '\' to support c:\a.bat and
1012 c:\windows\a.bat syntax */
1013 if (lastSlash) *(lastSlash + 1) = 0x00;
1016 /* Now extract PATHEXT */
1017 len = GetEnvironmentVariableW(envPathExt, pathext, sizeof(pathext)/sizeof(WCHAR));
1018 if ((len == 0) || (len >= (sizeof(pathext)/sizeof(WCHAR)))) {
1019 static const WCHAR dfltPathExt[] = {'.','b','a','t',';',
1020 '.','c','o','m',';',
1021 '.','c','m','d',';',
1022 '.','e','x','e','\0'};
1023 strcpyW (pathext, dfltPathExt);
1026 /* Loop through the search path, dir by dir */
1027 pathposn = pathtosearch;
1028 WINE_TRACE("Searching in '%s' for '%s'\n", wine_dbgstr_w(pathtosearch),
1029 wine_dbgstr_w(stemofsearch));
1030 while (!launched && pathposn) {
1032 WCHAR thisDir[MAX_PATH] = {'\0'};
1033 WCHAR *pos = NULL;
1034 BOOL found = FALSE;
1035 const WCHAR slashW[] = {'\\','\0'};
1037 /* Work on the first directory on the search path */
1038 pos = strchrW(pathposn, ';');
1039 if (pos) {
1040 memcpy(thisDir, pathposn, (pos-pathposn) * sizeof(WCHAR));
1041 thisDir[(pos-pathposn)] = 0x00;
1042 pathposn = pos+1;
1044 } else {
1045 strcpyW(thisDir, pathposn);
1046 pathposn = NULL;
1049 /* Since you can have eg. ..\.. on the path, need to expand
1050 to full information */
1051 strcpyW(temp, thisDir);
1052 GetFullPathNameW(temp, MAX_PATH, thisDir, NULL);
1054 /* 1. If extension supplied, see if that file exists */
1055 strcatW(thisDir, slashW);
1056 strcatW(thisDir, stemofsearch);
1057 pos = &thisDir[strlenW(thisDir)]; /* Pos = end of name */
1059 /* 1. If extension supplied, see if that file exists */
1060 if (extensionsupplied) {
1061 if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
1062 found = TRUE;
1066 /* 2. Any .* matches? */
1067 if (!found) {
1068 HANDLE h;
1069 WIN32_FIND_DATAW finddata;
1070 static const WCHAR allFiles[] = {'.','*','\0'};
1072 strcatW(thisDir,allFiles);
1073 h = FindFirstFileW(thisDir, &finddata);
1074 FindClose(h);
1075 if (h != INVALID_HANDLE_VALUE) {
1077 WCHAR *thisExt = pathext;
1079 /* 3. Yes - Try each path ext */
1080 while (thisExt) {
1081 WCHAR *nextExt = strchrW(thisExt, ';');
1083 if (nextExt) {
1084 memcpy(pos, thisExt, (nextExt-thisExt) * sizeof(WCHAR));
1085 pos[(nextExt-thisExt)] = 0x00;
1086 thisExt = nextExt+1;
1087 } else {
1088 strcpyW(pos, thisExt);
1089 thisExt = NULL;
1092 if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
1093 found = TRUE;
1094 thisExt = NULL;
1100 /* Internal programs won't be picked up by this search, so even
1101 though not found, try one last createprocess and wait for it
1102 to complete.
1103 Note: Ideally we could tell between a console app (wait) and a
1104 windows app, but the API's for it fail in this case */
1105 if (!found && pathposn == NULL) {
1106 WINE_TRACE("ASSUMING INTERNAL\n");
1107 assumeInternal = TRUE;
1108 } else {
1109 WINE_TRACE("Found as %s\n", wine_dbgstr_w(thisDir));
1112 /* Once found, launch it */
1113 if (found || assumeInternal) {
1114 STARTUPINFOW st;
1115 PROCESS_INFORMATION pe;
1116 SHFILEINFOW psfi;
1117 DWORD console;
1118 HINSTANCE hinst;
1119 WCHAR *ext = strrchrW( thisDir, '.' );
1120 static const WCHAR batExt[] = {'.','b','a','t','\0'};
1121 static const WCHAR cmdExt[] = {'.','c','m','d','\0'};
1123 launched = TRUE;
1125 /* Special case BAT and CMD */
1126 if (ext && !strcmpiW(ext, batExt)) {
1127 WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE);
1128 return;
1129 } else if (ext && !strcmpiW(ext, cmdExt)) {
1130 WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE);
1131 return;
1132 } else {
1134 /* thisDir contains the file to be launched, but with what?
1135 eg. a.exe will require a.exe to be launched, a.html may be iexplore */
1136 hinst = FindExecutableW (thisDir, NULL, temp);
1137 if ((INT_PTR)hinst < 32)
1138 console = 0;
1139 else
1140 console = SHGetFileInfoW(temp, 0, &psfi, sizeof(psfi), SHGFI_EXETYPE);
1142 ZeroMemory (&st, sizeof(STARTUPINFOW));
1143 st.cb = sizeof(STARTUPINFOW);
1144 init_msvcrt_io_block(&st);
1146 /* Launch the process and if a CUI wait on it to complete
1147 Note: Launching internal wine processes cannot specify a full path to exe */
1148 status = CreateProcessW(assumeInternal?NULL : thisDir,
1149 command, NULL, NULL, TRUE, 0, NULL, NULL, &st, &pe);
1150 if ((opt_c || opt_k) && !opt_s && !status
1151 && GetLastError()==ERROR_FILE_NOT_FOUND && command[0]=='\"') {
1152 /* strip first and last quote WCHARacters and try again */
1153 WCMD_opt_s_strip_quotes(command);
1154 opt_s=1;
1155 WCMD_run_program(command, called);
1156 return;
1159 if (!status)
1160 break;
1162 if (!assumeInternal && !console) errorlevel = 0;
1163 else
1165 /* Always wait when called in a batch program context */
1166 if (assumeInternal || context || !HIWORD(console)) WaitForSingleObject (pe.hProcess, INFINITE);
1167 GetExitCodeProcess (pe.hProcess, &errorlevel);
1168 if (errorlevel == STILL_ACTIVE) errorlevel = 0;
1170 CloseHandle(pe.hProcess);
1171 CloseHandle(pe.hThread);
1172 return;
1177 /* Not found anywhere - give up */
1178 SetLastError(ERROR_FILE_NOT_FOUND);
1179 WCMD_print_error ();
1181 /* If a command fails to launch, it sets errorlevel 9009 - which
1182 does not seem to have any associated constant definition */
1183 errorlevel = 9009;
1184 return;
1188 /*****************************************************************************
1189 * Process one command. If the command is EXIT this routine does not return.
1190 * We will recurse through here executing batch files.
1192 void WCMD_execute (WCHAR *command, WCHAR *redirects,
1193 WCHAR *forVariable, WCHAR *forValue,
1194 CMD_LIST **cmdList)
1196 WCHAR *cmd, *p, *redir;
1197 int status, i;
1198 DWORD count, creationDisposition;
1199 HANDLE h;
1200 WCHAR *whichcmd;
1201 SECURITY_ATTRIBUTES sa;
1202 WCHAR *new_cmd = NULL;
1203 WCHAR *new_redir = NULL;
1204 HANDLE old_stdhandles[3] = {GetStdHandle (STD_INPUT_HANDLE),
1205 GetStdHandle (STD_OUTPUT_HANDLE),
1206 GetStdHandle (STD_ERROR_HANDLE)};
1207 DWORD idx_stdhandles[3] = {STD_INPUT_HANDLE,
1208 STD_OUTPUT_HANDLE,
1209 STD_ERROR_HANDLE};
1210 BOOL piped = FALSE;
1212 WINE_TRACE("command on entry:%s (%p), with '%s'='%s'\n",
1213 wine_dbgstr_w(command), cmdList,
1214 wine_dbgstr_w(forVariable), wine_dbgstr_w(forValue));
1216 /* If the next command is a pipe then we implement pipes by redirecting
1217 the output from this command to a temp file and input into the
1218 next command from that temp file.
1219 FIXME: Use of named pipes would make more sense here as currently this
1220 process has to finish before the next one can start but this requires
1221 a change to not wait for the first app to finish but rather the pipe */
1222 if (cmdList && (*cmdList)->nextcommand &&
1223 (*cmdList)->nextcommand->prevDelim == CMD_PIPE) {
1225 WCHAR temp_path[MAX_PATH];
1226 static const WCHAR cmdW[] = {'C','M','D','\0'};
1228 /* Remember piping is in action */
1229 WINE_TRACE("Output needs to be piped\n");
1230 piped = TRUE;
1232 /* Generate a unique temporary filename */
1233 GetTempPathW(sizeof(temp_path)/sizeof(WCHAR), temp_path);
1234 GetTempFileNameW(temp_path, cmdW, 0, (*cmdList)->nextcommand->pipeFile);
1235 WINE_TRACE("Using temporary file of %s\n",
1236 wine_dbgstr_w((*cmdList)->nextcommand->pipeFile));
1239 /* Move copy of the command onto the heap so it can be expanded */
1240 new_cmd = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
1241 if (!new_cmd)
1243 WINE_ERR("Could not allocate memory for new_cmd\n");
1244 return;
1246 strcpyW(new_cmd, command);
1248 /* Move copy of the redirects onto the heap so it can be expanded */
1249 new_redir = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
1250 if (!new_redir)
1252 WINE_ERR("Could not allocate memory for new_redir\n");
1253 HeapFree( GetProcessHeap(), 0, new_cmd );
1254 return;
1257 /* If piped output, send stdout to the pipe by appending >filename to redirects */
1258 if (piped) {
1259 static const WCHAR redirOut[] = {'%','s',' ','>',' ','%','s','\0'};
1260 wsprintfW (new_redir, redirOut, redirects, (*cmdList)->nextcommand->pipeFile);
1261 WINE_TRACE("Redirects now %s\n", wine_dbgstr_w(new_redir));
1262 } else {
1263 strcpyW(new_redir, redirects);
1266 /* Expand variables in command line mode only (batch mode will
1267 be expanded as the line is read in, except for 'for' loops) */
1268 handleExpansion(new_cmd, (context != NULL), forVariable, forValue);
1269 handleExpansion(new_redir, (context != NULL), forVariable, forValue);
1270 cmd = new_cmd;
1273 * Changing default drive has to be handled as a special case.
1276 if ((cmd[1] == ':') && IsCharAlphaW(cmd[0]) && (strlenW(cmd) == 2)) {
1277 WCHAR envvar[5];
1278 WCHAR dir[MAX_PATH];
1280 /* According to MSDN CreateProcess docs, special env vars record
1281 the current directory on each drive, in the form =C:
1282 so see if one specified, and if so go back to it */
1283 strcpyW(envvar, equalsW);
1284 strcatW(envvar, cmd);
1285 if (GetEnvironmentVariableW(envvar, dir, MAX_PATH) == 0) {
1286 static const WCHAR fmt[] = {'%','s','\\','\0'};
1287 wsprintfW(cmd, fmt, cmd);
1288 WINE_TRACE("No special directory settings, using dir of %s\n", wine_dbgstr_w(cmd));
1290 WINE_TRACE("Got directory %s as %s\n", wine_dbgstr_w(envvar), wine_dbgstr_w(cmd));
1291 status = SetCurrentDirectoryW(cmd);
1292 if (!status) WCMD_print_error ();
1293 HeapFree( GetProcessHeap(), 0, cmd );
1294 HeapFree( GetProcessHeap(), 0, new_redir );
1295 return;
1298 sa.nLength = sizeof(sa);
1299 sa.lpSecurityDescriptor = NULL;
1300 sa.bInheritHandle = TRUE;
1303 * Redirect stdin, stdout and/or stderr if required.
1306 /* STDIN could come from a preceding pipe, so delete on close if it does */
1307 if (cmdList && (*cmdList)->pipeFile[0] != 0x00) {
1308 WINE_TRACE("Input coming from %s\n", wine_dbgstr_w((*cmdList)->pipeFile));
1309 h = CreateFileW((*cmdList)->pipeFile, GENERIC_READ,
1310 FILE_SHARE_READ, &sa, OPEN_EXISTING,
1311 FILE_ATTRIBUTE_NORMAL | FILE_FLAG_DELETE_ON_CLOSE, NULL);
1312 if (h == INVALID_HANDLE_VALUE) {
1313 WCMD_print_error ();
1314 HeapFree( GetProcessHeap(), 0, cmd );
1315 HeapFree( GetProcessHeap(), 0, new_redir );
1316 return;
1318 SetStdHandle (STD_INPUT_HANDLE, h);
1320 /* No need to remember the temporary name any longer once opened */
1321 (*cmdList)->pipeFile[0] = 0x00;
1323 /* Otherwise STDIN could come from a '<' redirect */
1324 } else if ((p = strchrW(new_redir,'<')) != NULL) {
1325 h = CreateFileW(WCMD_parameter (++p, 0, NULL), GENERIC_READ, FILE_SHARE_READ, &sa, OPEN_EXISTING,
1326 FILE_ATTRIBUTE_NORMAL, NULL);
1327 if (h == INVALID_HANDLE_VALUE) {
1328 WCMD_print_error ();
1329 HeapFree( GetProcessHeap(), 0, cmd );
1330 HeapFree( GetProcessHeap(), 0, new_redir );
1331 return;
1333 SetStdHandle (STD_INPUT_HANDLE, h);
1336 /* Scan the whole command looking for > and 2> */
1337 redir = new_redir;
1338 while (redir != NULL && ((p = strchrW(redir,'>')) != NULL)) {
1339 int handle = 0;
1341 if (*(p-1)!='2') {
1342 handle = 1;
1343 } else {
1344 handle = 2;
1347 p++;
1348 if ('>' == *p) {
1349 creationDisposition = OPEN_ALWAYS;
1350 p++;
1352 else {
1353 creationDisposition = CREATE_ALWAYS;
1356 /* Add support for 2>&1 */
1357 redir = p;
1358 if (*p == '&') {
1359 int idx = *(p+1) - '0';
1361 if (DuplicateHandle(GetCurrentProcess(),
1362 GetStdHandle(idx_stdhandles[idx]),
1363 GetCurrentProcess(),
1365 0, TRUE, DUPLICATE_SAME_ACCESS) == 0) {
1366 WINE_FIXME("Duplicating handle failed with gle %d\n", GetLastError());
1368 WINE_TRACE("Redirect %d (%p) to %d (%p)\n", handle, GetStdHandle(idx_stdhandles[idx]), idx, h);
1370 } else {
1371 WCHAR *param = WCMD_parameter (p, 0, NULL);
1372 h = CreateFileW(param, GENERIC_WRITE, 0, &sa, creationDisposition,
1373 FILE_ATTRIBUTE_NORMAL, NULL);
1374 if (h == INVALID_HANDLE_VALUE) {
1375 WCMD_print_error ();
1376 HeapFree( GetProcessHeap(), 0, cmd );
1377 HeapFree( GetProcessHeap(), 0, new_redir );
1378 return;
1380 if (SetFilePointer (h, 0, NULL, FILE_END) ==
1381 INVALID_SET_FILE_POINTER) {
1382 WCMD_print_error ();
1384 WINE_TRACE("Redirect %d to '%s' (%p)\n", handle, wine_dbgstr_w(param), h);
1387 SetStdHandle (idx_stdhandles[handle], h);
1391 * Strip leading whitespaces, and a '@' if supplied
1393 whichcmd = WCMD_strtrim_leading_spaces(cmd);
1394 WINE_TRACE("Command: '%s'\n", wine_dbgstr_w(cmd));
1395 if (whichcmd[0] == '@') whichcmd++;
1398 * Check if the command entered is internal. If it is, pass the rest of the
1399 * line down to the command. If not try to run a program.
1402 count = 0;
1403 while (IsCharAlphaNumericW(whichcmd[count])) {
1404 count++;
1406 for (i=0; i<=WCMD_EXIT; i++) {
1407 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1408 whichcmd, count, inbuilt[i], -1) == 2) break;
1410 p = WCMD_strtrim_leading_spaces (&whichcmd[count]);
1411 WCMD_parse (p, quals, param1, param2);
1412 WINE_TRACE("param1: %s, param2: %s\n", wine_dbgstr_w(param1), wine_dbgstr_w(param2));
1414 switch (i) {
1416 case WCMD_ATTRIB:
1417 WCMD_setshow_attrib ();
1418 break;
1419 case WCMD_CALL:
1420 WCMD_call (p);
1421 break;
1422 case WCMD_CD:
1423 case WCMD_CHDIR:
1424 WCMD_setshow_default (p);
1425 break;
1426 case WCMD_CLS:
1427 WCMD_clear_screen ();
1428 break;
1429 case WCMD_COPY:
1430 WCMD_copy ();
1431 break;
1432 case WCMD_CTTY:
1433 WCMD_change_tty ();
1434 break;
1435 case WCMD_DATE:
1436 WCMD_setshow_date ();
1437 break;
1438 case WCMD_DEL:
1439 case WCMD_ERASE:
1440 WCMD_delete (p, TRUE);
1441 break;
1442 case WCMD_DIR:
1443 WCMD_directory (p);
1444 break;
1445 case WCMD_ECHO:
1446 WCMD_echo(&whichcmd[count]);
1447 break;
1448 case WCMD_FOR:
1449 WCMD_for (p, cmdList);
1450 break;
1451 case WCMD_GOTO:
1452 WCMD_goto (cmdList);
1453 break;
1454 case WCMD_HELP:
1455 WCMD_give_help (p);
1456 break;
1457 case WCMD_IF:
1458 WCMD_if (p, cmdList);
1459 break;
1460 case WCMD_LABEL:
1461 WCMD_volume (1, p);
1462 break;
1463 case WCMD_MD:
1464 case WCMD_MKDIR:
1465 WCMD_create_dir ();
1466 break;
1467 case WCMD_MOVE:
1468 WCMD_move ();
1469 break;
1470 case WCMD_PATH:
1471 WCMD_setshow_path (p);
1472 break;
1473 case WCMD_PAUSE:
1474 WCMD_pause ();
1475 break;
1476 case WCMD_PROMPT:
1477 WCMD_setshow_prompt ();
1478 break;
1479 case WCMD_REM:
1480 break;
1481 case WCMD_REN:
1482 case WCMD_RENAME:
1483 WCMD_rename ();
1484 break;
1485 case WCMD_RD:
1486 case WCMD_RMDIR:
1487 WCMD_remove_dir (p);
1488 break;
1489 case WCMD_SETLOCAL:
1490 WCMD_setlocal(p);
1491 break;
1492 case WCMD_ENDLOCAL:
1493 WCMD_endlocal();
1494 break;
1495 case WCMD_SET:
1496 WCMD_setshow_env (p);
1497 break;
1498 case WCMD_SHIFT:
1499 WCMD_shift (p);
1500 break;
1501 case WCMD_TIME:
1502 WCMD_setshow_time ();
1503 break;
1504 case WCMD_TITLE:
1505 if (strlenW(&whichcmd[count]) > 0)
1506 WCMD_title(&whichcmd[count+1]);
1507 break;
1508 case WCMD_TYPE:
1509 WCMD_type (p);
1510 break;
1511 case WCMD_VER:
1512 WCMD_version ();
1513 break;
1514 case WCMD_VERIFY:
1515 WCMD_verify (p);
1516 break;
1517 case WCMD_VOL:
1518 WCMD_volume (0, p);
1519 break;
1520 case WCMD_PUSHD:
1521 WCMD_pushd(p);
1522 break;
1523 case WCMD_POPD:
1524 WCMD_popd();
1525 break;
1526 case WCMD_ASSOC:
1527 WCMD_assoc(p, TRUE);
1528 break;
1529 case WCMD_COLOR:
1530 WCMD_color();
1531 break;
1532 case WCMD_FTYPE:
1533 WCMD_assoc(p, FALSE);
1534 break;
1535 case WCMD_MORE:
1536 WCMD_more(p);
1537 break;
1538 case WCMD_EXIT:
1539 WCMD_exit (cmdList);
1540 break;
1541 default:
1542 WCMD_run_program (whichcmd, 0);
1544 HeapFree( GetProcessHeap(), 0, cmd );
1545 HeapFree( GetProcessHeap(), 0, new_redir );
1547 /* Restore old handles */
1548 for (i=0; i<3; i++) {
1549 if (old_stdhandles[i] != GetStdHandle(idx_stdhandles[i])) {
1550 CloseHandle (GetStdHandle (idx_stdhandles[i]));
1551 SetStdHandle (idx_stdhandles[i], old_stdhandles[i]);
1555 /*************************************************************************
1556 * WCMD_LoadMessage
1557 * Load a string from the resource file, handling any error
1558 * Returns string retrieved from resource file
1560 WCHAR *WCMD_LoadMessage(UINT id) {
1561 static WCHAR msg[2048];
1562 static const WCHAR failedMsg[] = {'F','a','i','l','e','d','!','\0'};
1564 if (!LoadStringW(GetModuleHandleW(NULL), id, msg, sizeof(msg)/sizeof(WCHAR))) {
1565 WINE_FIXME("LoadString failed with %d\n", GetLastError());
1566 strcpyW(msg, failedMsg);
1568 return msg;
1571 /***************************************************************************
1572 * WCMD_DumpCommands
1574 * Dumps out the parsed command line to ensure syntax is correct
1576 static void WCMD_DumpCommands(CMD_LIST *commands) {
1577 CMD_LIST *thisCmd = commands;
1579 WINE_TRACE("Parsed line:\n");
1580 while (thisCmd != NULL) {
1581 WINE_TRACE("%p %d %2.2d %p %s Redir:%s\n",
1582 thisCmd,
1583 thisCmd->prevDelim,
1584 thisCmd->bracketDepth,
1585 thisCmd->nextcommand,
1586 wine_dbgstr_w(thisCmd->command),
1587 wine_dbgstr_w(thisCmd->redirects));
1588 thisCmd = thisCmd->nextcommand;
1592 /***************************************************************************
1593 * WCMD_addCommand
1595 * Adds a command to the current command list
1597 static void WCMD_addCommand(WCHAR *command, int *commandLen,
1598 WCHAR *redirs, int *redirLen,
1599 WCHAR **copyTo, int **copyToLen,
1600 CMD_DELIMITERS prevDelim, int curDepth,
1601 CMD_LIST **lastEntry, CMD_LIST **output) {
1603 CMD_LIST *thisEntry = NULL;
1605 /* Allocate storage for command */
1606 thisEntry = HeapAlloc(GetProcessHeap(), 0, sizeof(CMD_LIST));
1608 /* Copy in the command */
1609 if (command) {
1610 thisEntry->command = HeapAlloc(GetProcessHeap(), 0,
1611 (*commandLen+1) * sizeof(WCHAR));
1612 memcpy(thisEntry->command, command, *commandLen * sizeof(WCHAR));
1613 thisEntry->command[*commandLen] = 0x00;
1615 /* Copy in the redirects */
1616 thisEntry->redirects = HeapAlloc(GetProcessHeap(), 0,
1617 (*redirLen+1) * sizeof(WCHAR));
1618 memcpy(thisEntry->redirects, redirs, *redirLen * sizeof(WCHAR));
1619 thisEntry->redirects[*redirLen] = 0x00;
1620 thisEntry->pipeFile[0] = 0x00;
1622 /* Reset the lengths */
1623 *commandLen = 0;
1624 *redirLen = 0;
1625 *copyToLen = commandLen;
1626 *copyTo = command;
1628 } else {
1629 thisEntry->command = NULL;
1632 /* Fill in other fields */
1633 thisEntry->nextcommand = NULL;
1634 thisEntry->prevDelim = prevDelim;
1635 thisEntry->bracketDepth = curDepth;
1636 if (*lastEntry) {
1637 (*lastEntry)->nextcommand = thisEntry;
1638 } else {
1639 *output = thisEntry;
1641 *lastEntry = thisEntry;
1645 /***************************************************************************
1646 * WCMD_IsEndQuote
1648 * Checks if the quote pointet to is the end-quote.
1650 * Quotes end if:
1652 * 1) The current parameter ends at EOL or at the beginning
1653 * of a redirection or pipe and not in a quote section.
1655 * 2) If the next character is a space and not in a quote section.
1657 * Returns TRUE if this is an end quote, and FALSE if it is not.
1660 static BOOL WCMD_IsEndQuote(WCHAR *quote, int quoteIndex)
1662 int quoteCount = quoteIndex;
1663 int i;
1665 /* If we are not in a quoted section, then we are not an end-quote */
1666 if(quoteIndex == 0)
1668 return FALSE;
1671 /* Check how many quotes are left for this parameter */
1672 for(i=0;quote[i];i++)
1674 if(quote[i] == '"')
1676 quoteCount++;
1679 /* Quote counting ends at EOL, redirection, space or pipe if current quote is complete */
1680 else if(((quoteCount % 2) == 0)
1681 && ((quote[i] == '<') || (quote[i] == '>') || (quote[i] == '|') || (quote[i] == ' ')))
1683 break;
1687 /* If the quote is part of the last part of a series of quotes-on-quotes, then it must
1688 be an end-quote */
1689 if(quoteIndex >= (quoteCount / 2))
1691 return TRUE;
1694 /* No cigar */
1695 return FALSE;
1698 /***************************************************************************
1699 * WCMD_ReadAndParseLine
1701 * Either uses supplied input or
1702 * Reads a file from the handle, and then...
1703 * Parse the text buffer, spliting into separate commands
1704 * - unquoted && strings split 2 commands but the 2nd is flagged as
1705 * following an &&
1706 * - ( as the first character just ups the bracket depth
1707 * - unquoted ) when bracket depth > 0 terminates a bracket and
1708 * adds a CMD_LIST structure with null command
1709 * - Anything else gets put into the command string (including
1710 * redirects)
1712 WCHAR *WCMD_ReadAndParseLine(WCHAR *optionalcmd, CMD_LIST **output, HANDLE readFrom) {
1714 WCHAR *curPos;
1715 int inQuotes = 0;
1716 WCHAR curString[MAXSTRING];
1717 int curStringLen = 0;
1718 WCHAR curRedirs[MAXSTRING];
1719 int curRedirsLen = 0;
1720 WCHAR *curCopyTo;
1721 int *curLen;
1722 int curDepth = 0;
1723 CMD_LIST *lastEntry = NULL;
1724 CMD_DELIMITERS prevDelim = CMD_NONE;
1725 static WCHAR *extraSpace = NULL; /* Deliberately never freed */
1726 const WCHAR remCmd[] = {'r','e','m',' ','\0'};
1727 const WCHAR forCmd[] = {'f','o','r',' ','\0'};
1728 const WCHAR ifCmd[] = {'i','f',' ','\0'};
1729 const WCHAR ifElse[] = {'e','l','s','e',' ','\0'};
1730 BOOL inRem = FALSE;
1731 BOOL inFor = FALSE;
1732 BOOL inIn = FALSE;
1733 BOOL inIf = FALSE;
1734 BOOL inElse= FALSE;
1735 BOOL onlyWhiteSpace = FALSE;
1736 BOOL lastWasWhiteSpace = FALSE;
1737 BOOL lastWasDo = FALSE;
1738 BOOL lastWasIn = FALSE;
1739 BOOL lastWasElse = FALSE;
1740 BOOL lastWasRedirect = TRUE;
1742 /* Allocate working space for a command read from keyboard, file etc */
1743 if (!extraSpace)
1744 extraSpace = HeapAlloc(GetProcessHeap(), 0, (MAXSTRING+1) * sizeof(WCHAR));
1745 if (!extraSpace)
1747 WINE_ERR("Could not allocate memory for extraSpace\n");
1748 return NULL;
1751 /* If initial command read in, use that, otherwise get input from handle */
1752 if (optionalcmd != NULL) {
1753 strcpyW(extraSpace, optionalcmd);
1754 } else if (readFrom == INVALID_HANDLE_VALUE) {
1755 WINE_FIXME("No command nor handle supplied\n");
1756 } else {
1757 if (WCMD_fgets(extraSpace, MAXSTRING, readFrom) == NULL) return NULL;
1759 curPos = extraSpace;
1761 /* Handle truncated input - issue warning */
1762 if (strlenW(extraSpace) == MAXSTRING -1) {
1763 WCMD_output_asis(WCMD_LoadMessage(WCMD_TRUNCATEDLINE));
1764 WCMD_output_asis(extraSpace);
1765 WCMD_output_asis(newline);
1768 /* Replace env vars if in a batch context */
1769 if (context) handleExpansion(extraSpace, FALSE, NULL, NULL);
1770 /* Show prompt before batch line IF echo is on and in batch program */
1771 if (context && echo_mode && extraSpace[0] && (extraSpace[0] != '@')) {
1772 WCMD_show_prompt();
1773 WCMD_output_asis(extraSpace);
1774 WCMD_output_asis(newline);
1777 /* Start with an empty string, copying to the command string */
1778 curStringLen = 0;
1779 curRedirsLen = 0;
1780 curCopyTo = curString;
1781 curLen = &curStringLen;
1782 lastWasRedirect = FALSE; /* Required for eg spaces between > and filename */
1784 /* Parse every character on the line being processed */
1785 while (*curPos != 0x00) {
1787 WCHAR thisChar;
1789 /* Debugging AID:
1790 WINE_TRACE("Looking at '%c' (len:%d, lws:%d, ows:%d)\n", *curPos, *curLen,
1791 lastWasWhiteSpace, onlyWhiteSpace);
1794 /* Certain commands need special handling */
1795 if (curStringLen == 0 && curCopyTo == curString) {
1796 const WCHAR forDO[] = {'d','o',' ','\0'};
1798 /* If command starts with 'rem', ignore any &&, ( etc */
1799 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1800 curPos, 4, remCmd, -1) == 2) {
1801 inRem = TRUE;
1803 /* If command starts with 'for', handle ('s mid line after IN or DO */
1804 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1805 curPos, 4, forCmd, -1) == 2) {
1806 inFor = TRUE;
1808 /* If command starts with 'if' or 'else', handle ('s mid line. We should ensure this
1809 is only true in the command portion of the IF statement, but this
1810 should suffice for now
1811 FIXME: Silly syntax like "if 1(==1( (
1812 echo they equal
1813 )" will be parsed wrong */
1814 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1815 curPos, 3, ifCmd, -1) == 2) {
1816 inIf = TRUE;
1818 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1819 curPos, 5, ifElse, -1) == 2) {
1820 inElse = TRUE;
1821 lastWasElse = TRUE;
1822 onlyWhiteSpace = TRUE;
1823 memcpy(&curCopyTo[*curLen], curPos, 5*sizeof(WCHAR));
1824 (*curLen)+=5;
1825 curPos+=5;
1826 continue;
1828 /* In a for loop, the DO command will follow a close bracket followed by
1829 whitespace, followed by DO, ie closeBracket inserts a NULL entry, curLen
1830 is then 0, and all whitespace is skipped */
1831 } else if (inFor &&
1832 (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1833 curPos, 3, forDO, -1) == 2)) {
1834 WINE_TRACE("Found DO\n");
1835 lastWasDo = TRUE;
1836 onlyWhiteSpace = TRUE;
1837 memcpy(&curCopyTo[*curLen], curPos, 3*sizeof(WCHAR));
1838 (*curLen)+=3;
1839 curPos+=3;
1840 continue;
1842 } else if (curCopyTo == curString) {
1844 /* Special handling for the 'FOR' command */
1845 if (inFor && lastWasWhiteSpace) {
1846 const WCHAR forIN[] = {'i','n',' ','\0'};
1848 WINE_TRACE("Found 'FOR', comparing next parm: '%s'\n", wine_dbgstr_w(curPos));
1850 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1851 curPos, 3, forIN, -1) == 2) {
1852 WINE_TRACE("Found IN\n");
1853 lastWasIn = TRUE;
1854 onlyWhiteSpace = TRUE;
1855 memcpy(&curCopyTo[*curLen], curPos, 3*sizeof(WCHAR));
1856 (*curLen)+=3;
1857 curPos+=3;
1858 continue;
1863 /* Nothing 'ends' a REM statement and &&, quotes etc are ineffective,
1864 so just use the default processing ie skip character specific
1865 matching below */
1866 if (!inRem) thisChar = *curPos;
1867 else thisChar = 'X'; /* Character with no special processing */
1869 lastWasWhiteSpace = FALSE; /* Will be reset below */
1871 switch (thisChar) {
1873 case '=': /* drop through - ignore token delimiters at the start of a command */
1874 case ',': /* drop through - ignore token delimiters at the start of a command */
1875 case '\t':/* drop through - ignore token delimiters at the start of a command */
1876 case ' ':
1877 /* If a redirect in place, it ends here */
1878 if (!inQuotes && !lastWasRedirect) {
1880 /* If finishing off a redirect, add a whitespace delimiter */
1881 if (curCopyTo == curRedirs) {
1882 curCopyTo[(*curLen)++] = ' ';
1884 curCopyTo = curString;
1885 curLen = &curStringLen;
1887 if (*curLen > 0) {
1888 curCopyTo[(*curLen)++] = *curPos;
1891 /* Remember just processed whitespace */
1892 lastWasWhiteSpace = TRUE;
1894 break;
1896 case '>': /* drop through - handle redirect chars the same */
1897 case '<':
1898 /* Make a redirect start here */
1899 if (!inQuotes) {
1900 curCopyTo = curRedirs;
1901 curLen = &curRedirsLen;
1902 lastWasRedirect = TRUE;
1905 /* See if 1>, 2> etc, in which case we have some patching up
1906 to do */
1907 if (curPos != extraSpace &&
1908 *(curPos-1)>='1' && *(curPos-1)<='9') {
1910 curStringLen--;
1911 curString[curStringLen] = 0x00;
1912 curCopyTo[(*curLen)++] = *(curPos-1);
1915 curCopyTo[(*curLen)++] = *curPos;
1917 /* If a redirect is immediately followed by '&' (ie. 2>&1) then
1918 do not process that ampersand as an AND operator */
1919 if (thisChar == '>' && *(curPos+1) == '&') {
1920 curCopyTo[(*curLen)++] = *(curPos+1);
1921 curPos++;
1923 break;
1925 case '|': /* Pipe character only if not || */
1926 if (!inQuotes) {
1927 lastWasRedirect = FALSE;
1929 /* Add an entry to the command list */
1930 if (curStringLen > 0) {
1932 /* Add the current command */
1933 WCMD_addCommand(curString, &curStringLen,
1934 curRedirs, &curRedirsLen,
1935 &curCopyTo, &curLen,
1936 prevDelim, curDepth,
1937 &lastEntry, output);
1941 if (*(curPos+1) == '|') {
1942 curPos++; /* Skip other | */
1943 prevDelim = CMD_ONFAILURE;
1944 } else {
1945 prevDelim = CMD_PIPE;
1947 } else {
1948 curCopyTo[(*curLen)++] = *curPos;
1950 break;
1952 case '"': if (WCMD_IsEndQuote(curPos, inQuotes)) {
1953 inQuotes--;
1954 } else {
1955 inQuotes++; /* Quotes within quotes are fun! */
1957 curCopyTo[(*curLen)++] = *curPos;
1958 lastWasRedirect = FALSE;
1959 break;
1961 case '(': /* If a '(' is the first non whitespace in a command portion
1962 ie start of line or just after &&, then we read until an
1963 unquoted ) is found */
1964 WINE_TRACE("Found '(' conditions: curLen(%d), inQ(%d), onlyWS(%d)"
1965 ", for(%d, In:%d, Do:%d)"
1966 ", if(%d, else:%d, lwe:%d)\n",
1967 *curLen, inQuotes,
1968 onlyWhiteSpace,
1969 inFor, lastWasIn, lastWasDo,
1970 inIf, inElse, lastWasElse);
1971 lastWasRedirect = FALSE;
1973 /* Ignore open brackets inside the for set */
1974 if (*curLen == 0 && !inIn) {
1975 curDepth++;
1977 /* If in quotes, ignore brackets */
1978 } else if (inQuotes) {
1979 curCopyTo[(*curLen)++] = *curPos;
1981 /* In a FOR loop, an unquoted '(' may occur straight after
1982 IN or DO
1983 In an IF statement just handle it regardless as we don't
1984 parse the operands
1985 In an ELSE statement, only allow it straight away after
1986 the ELSE and whitespace
1988 } else if (inIf ||
1989 (inElse && lastWasElse && onlyWhiteSpace) ||
1990 (inFor && (lastWasIn || lastWasDo) && onlyWhiteSpace)) {
1992 /* If entering into an 'IN', set inIn */
1993 if (inFor && lastWasIn && onlyWhiteSpace) {
1994 WINE_TRACE("Inside an IN\n");
1995 inIn = TRUE;
1998 /* Add the current command */
1999 WCMD_addCommand(curString, &curStringLen,
2000 curRedirs, &curRedirsLen,
2001 &curCopyTo, &curLen,
2002 prevDelim, curDepth,
2003 &lastEntry, output);
2005 curDepth++;
2006 } else {
2007 curCopyTo[(*curLen)++] = *curPos;
2009 break;
2011 case '&': if (!inQuotes) {
2012 lastWasRedirect = FALSE;
2014 /* Add an entry to the command list */
2015 if (curStringLen > 0) {
2017 /* Add the current command */
2018 WCMD_addCommand(curString, &curStringLen,
2019 curRedirs, &curRedirsLen,
2020 &curCopyTo, &curLen,
2021 prevDelim, curDepth,
2022 &lastEntry, output);
2026 if (*(curPos+1) == '&') {
2027 curPos++; /* Skip other & */
2028 prevDelim = CMD_ONSUCCESS;
2029 } else {
2030 prevDelim = CMD_NONE;
2032 } else {
2033 curCopyTo[(*curLen)++] = *curPos;
2035 break;
2037 case ')': if (!inQuotes && curDepth > 0) {
2038 lastWasRedirect = FALSE;
2040 /* Add the current command if there is one */
2041 if (curStringLen) {
2043 /* Add the current command */
2044 WCMD_addCommand(curString, &curStringLen,
2045 curRedirs, &curRedirsLen,
2046 &curCopyTo, &curLen,
2047 prevDelim, curDepth,
2048 &lastEntry, output);
2051 /* Add an empty entry to the command list */
2052 prevDelim = CMD_NONE;
2053 WCMD_addCommand(NULL, &curStringLen,
2054 curRedirs, &curRedirsLen,
2055 &curCopyTo, &curLen,
2056 prevDelim, curDepth,
2057 &lastEntry, output);
2058 curDepth--;
2060 /* Leave inIn if necessary */
2061 if (inIn) inIn = FALSE;
2062 } else {
2063 curCopyTo[(*curLen)++] = *curPos;
2065 break;
2066 default:
2067 lastWasRedirect = FALSE;
2068 curCopyTo[(*curLen)++] = *curPos;
2071 curPos++;
2073 /* At various times we need to know if we have only skipped whitespace,
2074 so reset this variable and then it will remain true until a non
2075 whitespace is found */
2076 if ((thisChar != ' ') && (thisChar != '\n')) onlyWhiteSpace = FALSE;
2078 /* Flag end of interest in FOR DO and IN parms once something has been processed */
2079 if (!lastWasWhiteSpace) {
2080 lastWasIn = lastWasDo = FALSE;
2083 /* If we have reached the end, add this command into the list */
2084 if (*curPos == 0x00 && *curLen > 0) {
2086 /* Add an entry to the command list */
2087 WCMD_addCommand(curString, &curStringLen,
2088 curRedirs, &curRedirsLen,
2089 &curCopyTo, &curLen,
2090 prevDelim, curDepth,
2091 &lastEntry, output);
2094 /* If we have reached the end of the string, see if bracketing outstanding */
2095 if (*curPos == 0x00 && curDepth > 0 && readFrom != INVALID_HANDLE_VALUE) {
2096 inRem = FALSE;
2097 prevDelim = CMD_NONE;
2098 inQuotes = 0;
2099 memset(extraSpace, 0x00, (MAXSTRING+1) * sizeof(WCHAR));
2101 /* Read more, skipping any blank lines */
2102 while (*extraSpace == 0x00) {
2103 if (!context) WCMD_output_asis( WCMD_LoadMessage(WCMD_MOREPROMPT));
2104 if (WCMD_fgets(extraSpace, MAXSTRING, readFrom) == NULL) break;
2106 curPos = extraSpace;
2107 if (context) handleExpansion(extraSpace, FALSE, NULL, NULL);
2108 /* Continue to echo commands IF echo is on and in batch program */
2109 if (context && echo_mode && extraSpace[0] && (extraSpace[0] != '@')) {
2110 WCMD_output_asis(extraSpace);
2111 WCMD_output_asis(newline);
2116 /* Dump out the parsed output */
2117 WCMD_DumpCommands(*output);
2119 return extraSpace;
2122 /***************************************************************************
2123 * WCMD_process_commands
2125 * Process all the commands read in so far
2127 CMD_LIST *WCMD_process_commands(CMD_LIST *thisCmd, BOOL oneBracket,
2128 WCHAR *var, WCHAR *val) {
2130 int bdepth = -1;
2132 if (thisCmd && oneBracket) bdepth = thisCmd->bracketDepth;
2134 /* Loop through the commands, processing them one by one */
2135 while (thisCmd) {
2137 CMD_LIST *origCmd = thisCmd;
2139 /* If processing one bracket only, and we find the end bracket
2140 entry (or less), return */
2141 if (oneBracket && !thisCmd->command &&
2142 bdepth <= thisCmd->bracketDepth) {
2143 WINE_TRACE("Finished bracket @ %p, next command is %p\n",
2144 thisCmd, thisCmd->nextcommand);
2145 return thisCmd->nextcommand;
2148 /* Ignore the NULL entries a ')' inserts (Only 'if' cares
2149 about them and it will be handled in there)
2150 Also, skip over any batch labels (eg. :fred) */
2151 if (thisCmd->command && thisCmd->command[0] != ':') {
2152 WINE_TRACE("Executing command: '%s'\n", wine_dbgstr_w(thisCmd->command));
2153 WCMD_execute (thisCmd->command, thisCmd->redirects, var, val, &thisCmd);
2156 /* Step on unless the command itself already stepped on */
2157 if (thisCmd == origCmd) thisCmd = thisCmd->nextcommand;
2159 return NULL;
2162 /***************************************************************************
2163 * WCMD_free_commands
2165 * Frees the storage held for a parsed command line
2166 * - This is not done in the process_commands, as eventually the current
2167 * pointer will be modified within the commands, and hence a single free
2168 * routine is simpler
2170 void WCMD_free_commands(CMD_LIST *cmds) {
2172 /* Loop through the commands, freeing them one by one */
2173 while (cmds) {
2174 CMD_LIST *thisCmd = cmds;
2175 cmds = cmds->nextcommand;
2176 HeapFree(GetProcessHeap(), 0, thisCmd->command);
2177 HeapFree(GetProcessHeap(), 0, thisCmd);
2182 /*****************************************************************************
2183 * Main entry point. This is a console application so we have a main() not a
2184 * winmain().
2187 int wmain (int argc, WCHAR *argvW[])
2189 int args;
2190 WCHAR *cmd = NULL;
2191 WCHAR string[1024];
2192 WCHAR envvar[4];
2193 HANDLE h;
2194 int opt_q;
2195 int opt_t = 0;
2196 static const WCHAR autoexec[] = {'\\','a','u','t','o','e','x','e','c','.',
2197 'b','a','t','\0'};
2198 static const WCHAR promptW[] = {'P','R','O','M','P','T','\0'};
2199 static const WCHAR defaultpromptW[] = {'$','P','$','G','\0'};
2200 char ansiVersion[100];
2201 CMD_LIST *toExecute = NULL; /* Commands left to be executed */
2203 srand(time(NULL));
2205 /* Pre initialize some messages */
2206 strcpy(ansiVersion, PACKAGE_VERSION);
2207 MultiByteToWideChar(CP_ACP, 0, ansiVersion, -1, string, 1024);
2208 wsprintfW(version_string, WCMD_LoadMessage(WCMD_VERSION), string);
2209 strcpyW(anykey, WCMD_LoadMessage(WCMD_ANYKEY));
2211 args = argc;
2212 opt_c=opt_k=opt_q=opt_s=0;
2213 while (args > 0)
2215 WCHAR c;
2216 WINE_TRACE("Command line parm: '%s'\n", wine_dbgstr_w(*argvW));
2217 if ((*argvW)[0]!='/' || (*argvW)[1]=='\0') {
2218 argvW++;
2219 args--;
2220 continue;
2223 c=(*argvW)[1];
2224 if (tolowerW(c)=='c') {
2225 opt_c=1;
2226 } else if (tolowerW(c)=='q') {
2227 opt_q=1;
2228 } else if (tolowerW(c)=='k') {
2229 opt_k=1;
2230 } else if (tolowerW(c)=='s') {
2231 opt_s=1;
2232 } else if (tolowerW(c)=='a') {
2233 unicodePipes=FALSE;
2234 } else if (tolowerW(c)=='u') {
2235 unicodePipes=TRUE;
2236 } else if (tolowerW(c)=='t' && (*argvW)[2]==':') {
2237 opt_t=strtoulW(&(*argvW)[3], NULL, 16);
2238 } else if (tolowerW(c)=='x' || tolowerW(c)=='y') {
2239 /* Ignored for compatibility with Windows */
2242 if ((*argvW)[2]==0) {
2243 argvW++;
2244 args--;
2246 else /* handle `cmd /cnotepad.exe` and `cmd /x/c ...` */
2248 *argvW+=2;
2251 if (opt_c || opt_k) /* break out of parsing immediately after c or k */
2252 break;
2255 if (opt_q) {
2256 const WCHAR eoff[] = {'O','F','F','\0'};
2257 WCMD_echo(eoff);
2260 if (opt_c || opt_k) {
2261 int len,qcount;
2262 WCHAR** arg;
2263 int argsLeft;
2264 WCHAR* p;
2266 /* opt_s left unflagged if the command starts with and contains exactly
2267 * one quoted string (exactly two quote characters). The quoted string
2268 * must be an executable name that has whitespace and must not have the
2269 * following characters: &<>()@^| */
2271 /* Build the command to execute */
2272 len = 0;
2273 qcount = 0;
2274 argsLeft = args;
2275 for (arg = argvW; argsLeft>0; arg++,argsLeft--)
2277 int has_space,bcount;
2278 WCHAR* a;
2280 has_space=0;
2281 bcount=0;
2282 a=*arg;
2283 if( !*a ) has_space=1;
2284 while (*a!='\0') {
2285 if (*a=='\\') {
2286 bcount++;
2287 } else {
2288 if (*a==' ' || *a=='\t') {
2289 has_space=1;
2290 } else if (*a=='"') {
2291 /* doubling of '\' preceding a '"',
2292 * plus escaping of said '"'
2294 len+=2*bcount+1;
2295 qcount++;
2297 bcount=0;
2299 a++;
2301 len+=(a-*arg) + 1; /* for the separating space */
2302 if (has_space)
2304 len+=2; /* for the quotes */
2305 qcount+=2;
2309 if (qcount!=2)
2310 opt_s=1;
2312 /* check argvW[0] for a space and invalid characters */
2313 if (!opt_s) {
2314 opt_s=1;
2315 p=*argvW;
2316 while (*p!='\0') {
2317 if (*p=='&' || *p=='<' || *p=='>' || *p=='(' || *p==')'
2318 || *p=='@' || *p=='^' || *p=='|') {
2319 opt_s=1;
2320 break;
2322 if (*p==' ')
2323 opt_s=0;
2324 p++;
2328 cmd = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
2329 if (!cmd)
2330 exit(1);
2332 p = cmd;
2333 argsLeft = args;
2334 for (arg = argvW; argsLeft>0; arg++,argsLeft--)
2336 int has_space,has_quote;
2337 WCHAR* a;
2339 /* Check for quotes and spaces in this argument */
2340 has_space=has_quote=0;
2341 a=*arg;
2342 if( !*a ) has_space=1;
2343 while (*a!='\0') {
2344 if (*a==' ' || *a=='\t') {
2345 has_space=1;
2346 if (has_quote)
2347 break;
2348 } else if (*a=='"') {
2349 has_quote=1;
2350 if (has_space)
2351 break;
2353 a++;
2356 /* Now transfer it to the command line */
2357 if (has_space)
2358 *p++='"';
2359 if (has_quote) {
2360 int bcount;
2361 WCHAR* a;
2363 bcount=0;
2364 a=*arg;
2365 while (*a!='\0') {
2366 if (*a=='\\') {
2367 *p++=*a;
2368 bcount++;
2369 } else {
2370 if (*a=='"') {
2371 int i;
2373 /* Double all the '\\' preceding this '"', plus one */
2374 for (i=0;i<=bcount;i++)
2375 *p++='\\';
2376 *p++='"';
2377 } else {
2378 *p++=*a;
2380 bcount=0;
2382 a++;
2384 } else {
2385 strcpyW(p,*arg);
2386 p+=strlenW(*arg);
2388 if (has_space)
2389 *p++='"';
2390 *p++=' ';
2392 if (p > cmd)
2393 p--; /* remove last space */
2394 *p = '\0';
2396 WINE_TRACE("/c command line: '%s'\n", wine_dbgstr_w(cmd));
2398 /* strip first and last quote characters if opt_s; check for invalid
2399 * executable is done later */
2400 if (opt_s && *cmd=='\"')
2401 WCMD_opt_s_strip_quotes(cmd);
2404 if (opt_c) {
2405 /* If we do a "wcmd /c command", we don't want to allocate a new
2406 * console since the command returns immediately. Rather, we use
2407 * the currently allocated input and output handles. This allows
2408 * us to pipe to and read from the command interpreter.
2411 /* Parse the command string, without reading any more input */
2412 WCMD_ReadAndParseLine(cmd, &toExecute, INVALID_HANDLE_VALUE);
2413 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2414 WCMD_free_commands(toExecute);
2415 toExecute = NULL;
2417 HeapFree(GetProcessHeap(), 0, cmd);
2418 return errorlevel;
2421 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_LINE_INPUT |
2422 ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT);
2423 SetConsoleTitleW(WCMD_LoadMessage(WCMD_CONSTITLE));
2425 /* Note: cmd.exe /c dir does not get a new color, /k dir does */
2426 if (opt_t) {
2427 if (!(((opt_t & 0xF0) >> 4) == (opt_t & 0x0F))) {
2428 defaultColor = opt_t & 0xFF;
2429 param1[0] = 0x00;
2430 WCMD_color();
2432 } else {
2433 /* Check HKCU\Software\Microsoft\Command Processor
2434 Then HKLM\Software\Microsoft\Command Processor
2435 for defaultcolour value
2436 Note Can be supplied as DWORD or REG_SZ
2437 Note2 When supplied as REG_SZ it's in decimal!!! */
2438 HKEY key;
2439 DWORD type;
2440 DWORD value=0, size=4;
2441 static const WCHAR regKeyW[] = {'S','o','f','t','w','a','r','e','\\',
2442 'M','i','c','r','o','s','o','f','t','\\',
2443 'C','o','m','m','a','n','d',' ','P','r','o','c','e','s','s','o','r','\0'};
2444 static const WCHAR dfltColorW[] = {'D','e','f','a','u','l','t','C','o','l','o','r','\0'};
2446 if (RegOpenKeyExW(HKEY_CURRENT_USER, regKeyW,
2447 0, KEY_READ, &key) == ERROR_SUCCESS) {
2448 WCHAR strvalue[4];
2450 /* See if DWORD or REG_SZ */
2451 if (RegQueryValueExW(key, dfltColorW, NULL, &type,
2452 NULL, NULL) == ERROR_SUCCESS) {
2453 if (type == REG_DWORD) {
2454 size = sizeof(DWORD);
2455 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2456 (LPBYTE)&value, &size);
2457 } else if (type == REG_SZ) {
2458 size = sizeof(strvalue)/sizeof(WCHAR);
2459 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2460 (LPBYTE)strvalue, &size);
2461 value = strtoulW(strvalue, NULL, 10);
2464 RegCloseKey(key);
2467 if (value == 0 && RegOpenKeyExW(HKEY_LOCAL_MACHINE, regKeyW,
2468 0, KEY_READ, &key) == ERROR_SUCCESS) {
2469 WCHAR strvalue[4];
2471 /* See if DWORD or REG_SZ */
2472 if (RegQueryValueExW(key, dfltColorW, NULL, &type,
2473 NULL, NULL) == ERROR_SUCCESS) {
2474 if (type == REG_DWORD) {
2475 size = sizeof(DWORD);
2476 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2477 (LPBYTE)&value, &size);
2478 } else if (type == REG_SZ) {
2479 size = sizeof(strvalue)/sizeof(WCHAR);
2480 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2481 (LPBYTE)strvalue, &size);
2482 value = strtoulW(strvalue, NULL, 10);
2485 RegCloseKey(key);
2488 /* If one found, set the screen to that colour */
2489 if (!(((value & 0xF0) >> 4) == (value & 0x0F))) {
2490 defaultColor = value & 0xFF;
2491 param1[0] = 0x00;
2492 WCMD_color();
2497 /* Save cwd into appropriate env var */
2498 GetCurrentDirectoryW(1024, string);
2499 if (IsCharAlphaW(string[0]) && string[1] == ':') {
2500 static const WCHAR fmt[] = {'=','%','c',':','\0'};
2501 wsprintfW(envvar, fmt, string[0]);
2502 SetEnvironmentVariableW(envvar, string);
2503 WINE_TRACE("Set %s to %s\n", wine_dbgstr_w(envvar), wine_dbgstr_w(string));
2506 if (opt_k) {
2507 /* Parse the command string, without reading any more input */
2508 WCMD_ReadAndParseLine(cmd, &toExecute, INVALID_HANDLE_VALUE);
2509 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2510 WCMD_free_commands(toExecute);
2511 toExecute = NULL;
2512 HeapFree(GetProcessHeap(), 0, cmd);
2516 * If there is an AUTOEXEC.BAT file, try to execute it.
2519 GetFullPathNameW (autoexec, sizeof(string)/sizeof(WCHAR), string, NULL);
2520 h = CreateFileW(string, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
2521 if (h != INVALID_HANDLE_VALUE) {
2522 CloseHandle (h);
2523 #if 0
2524 WCMD_batch (autoexec, autoexec, 0, NULL, INVALID_HANDLE_VALUE);
2525 #endif
2529 * Loop forever getting commands and executing them.
2532 SetEnvironmentVariableW(promptW, defaultpromptW);
2533 WCMD_version ();
2534 while (TRUE) {
2536 /* Read until EOF (which for std input is never, but if redirect
2537 in place, may occur */
2538 WCMD_show_prompt ();
2539 if (WCMD_ReadAndParseLine(NULL, &toExecute,
2540 GetStdHandle(STD_INPUT_HANDLE)) == NULL)
2541 break;
2542 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2543 WCMD_free_commands(toExecute);
2544 toExecute = NULL;
2546 return 0;