2 * WCMD - Wine-compatible command line interface.
4 * Copyright (C) 1999 - 2001 D A Pickles
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 * - Cannot handle parameters in quotes
24 * - Lots of functionality missing from builtins
29 char *inbuilt
[] = {"ATTRIB", "CALL", "CD", "CHDIR", "CLS", "COPY", "CTTY",
30 "DATE", "DEL", "DIR", "ECHO", "ERASE", "FOR", "GOTO",
31 "HELP", "IF", "LABEL", "MD", "MKDIR", "MOVE", "PATH", "PAUSE",
32 "PROMPT", "REM", "REN", "RENAME", "RD", "RMDIR", "SET", "SHIFT",
33 "TIME", "TITLE", "TYPE", "VERIFY", "VER", "VOL", "EXIT" };
37 int echo_mode
= 1, verify_mode
= 0;
38 char nyi
[] = "Not Yet Implemented\n\n";
39 char newline
[] = "\n";
40 char version_string
[] = "WCMD Version 0.17\n\n";
41 char anykey
[] = "Press Return key to continue: ";
42 char quals
[MAX_PATH
], param1
[MAX_PATH
], param2
[MAX_PATH
];
43 BATCH_CONTEXT
*context
= NULL
;
45 /*****************************************************************************
46 * Main entry point. This is a console application so we have a main() not a
50 int main (int argc
, char *argv
[])
56 int opt_c
, opt_k
, opt_q
;
62 if (lstrcmpi(*argv
,"/c")==0) {
66 } else if (lstrcmpi(*argv
,"/q")==0) {
68 } else if (lstrcmpi(*argv
,"/k")==0) {
72 } else if (lstrcmpi(*argv
,"/t")==0 || lstrcmpi(*argv
,"/x")==0 ||
73 lstrcmpi(*argv
,"/y")==0) {
74 /* Ignored for compatibility with Windows */
90 /* Build the command to execute */
92 for (arg
= argv
; *arg
; arg
++)
100 if( !*a
) has_space
=1;
105 if (*a
==' ' || *a
=='\t') {
107 } else if (*a
=='"') {
108 /* doubling of '\' preceeding a '"',
109 * plus escaping of said '"'
117 len
+=(a
-*arg
)+1 /* for the separating space */;
119 len
+=2; /* for the quotes */
122 cmd
= HeapAlloc(GetProcessHeap(), 0, len
);
127 for (arg
= argv
; *arg
; arg
++)
129 int has_space
,has_quote
;
132 /* Check for quotes and spaces in this argument */
133 has_space
=has_quote
=0;
135 if( !*a
) has_space
=1;
137 if (*a
==' ' || *a
=='\t') {
141 } else if (*a
=='"') {
149 /* Now transfer it to the command line */
166 /* Double all the '\\' preceeding this '"', plus one */
167 for (i
=0;i
<=bcount
;i
++)
186 p
--; /* remove last space */
191 /* If we do a "wcmd /c command", we don't want to allocate a new
192 * console since the command returns immediately. Rather, we use
193 * the currently allocated input and output handles. This allows
194 * us to pipe to and read from the command interpreter.
196 if (strchr(cmd
,'|') != NULL
)
199 WCMD_process_command(cmd
);
200 HeapFree(GetProcessHeap(), 0, cmd
);
204 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE
), ENABLE_LINE_INPUT
|
205 ENABLE_ECHO_INPUT
| ENABLE_PROCESSED_INPUT
);
206 SetConsoleTitle("Wine Command Prompt");
209 WCMD_process_command(cmd
);
210 HeapFree(GetProcessHeap(), 0, cmd
);
214 * If there is an AUTOEXEC.BAT file, try to execute it.
217 GetFullPathName ("\\autoexec.bat", sizeof(string
), string
, NULL
);
218 h
= CreateFile (string
, GENERIC_READ
, FILE_SHARE_READ
, NULL
, OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
);
219 if (h
!= INVALID_HANDLE_VALUE
) {
222 WCMD_batch_command (string
);
227 * Loop forever getting commands and executing them.
233 ReadFile (GetStdHandle(STD_INPUT_HANDLE
), string
, sizeof(string
), &count
, NULL
);
235 string
[count
-1] = '\0'; /* ReadFile output is not null-terminated! */
236 if (string
[count
-2] == '\r') string
[count
-2] = '\0'; /* Under Windoze we get CRLF! */
237 if (lstrlen (string
) != 0) {
238 if (strchr(string
,'|') != NULL
) {
242 WCMD_process_command (string
);
250 /*****************************************************************************
251 * Process one command. If the command is EXIT this routine does not return.
252 * We will recurse through here executing batch files.
256 void WCMD_process_command (char *command
)
261 HANDLE old_stdin
= 0, old_stdout
= 0, h
;
266 * Expand up environment variables.
268 len
= ExpandEnvironmentStrings (command
, NULL
, 0);
269 cmd
= HeapAlloc( GetProcessHeap(), 0, len
);
270 status
= ExpandEnvironmentStrings (command
, cmd
, len
);
273 HeapFree( GetProcessHeap(), 0, cmd
);
278 * Changing default drive has to be handled as a special case.
281 if ((cmd
[1] == ':') && IsCharAlpha (cmd
[0]) && (strlen(cmd
) == 2)) {
282 status
= SetCurrentDirectory (cmd
);
283 if (!status
) WCMD_print_error ();
284 HeapFree( GetProcessHeap(), 0, cmd
);
288 /* Dont issue newline WCMD_output (newline); @JED*/
291 * Redirect stdin and/or stdout if required.
294 if ((p
= strchr(cmd
,'<')) != NULL
) {
295 h
= CreateFile (WCMD_parameter (++p
, 0, NULL
), GENERIC_READ
, FILE_SHARE_READ
, NULL
, OPEN_EXISTING
,
296 FILE_ATTRIBUTE_NORMAL
, NULL
);
297 if (h
== INVALID_HANDLE_VALUE
) {
299 HeapFree( GetProcessHeap(), 0, cmd
);
302 old_stdin
= GetStdHandle (STD_INPUT_HANDLE
);
303 SetStdHandle (STD_INPUT_HANDLE
, h
);
305 if ((p
= strchr(cmd
,'>')) != NULL
) {
306 h
= CreateFile (WCMD_parameter (++p
, 0, NULL
), GENERIC_WRITE
, 0, NULL
, CREATE_ALWAYS
,
307 FILE_ATTRIBUTE_NORMAL
, NULL
);
308 if (h
== INVALID_HANDLE_VALUE
) {
310 HeapFree( GetProcessHeap(), 0, cmd
);
313 old_stdout
= GetStdHandle (STD_OUTPUT_HANDLE
);
314 SetStdHandle (STD_OUTPUT_HANDLE
, h
);
317 if ((p
= strchr(cmd
,'<')) != NULL
) *p
= '\0';
320 * Strip leading whitespaces, and a '@' if supplied
322 whichcmd
= WCMD_strtrim_leading_spaces(cmd
);
323 if (whichcmd
[0] == '@') whichcmd
++;
326 * Check if the command entered is internal. If it is, pass the rest of the
327 * line down to the command. If not try to run a program.
331 while (IsCharAlphaNumeric(whichcmd
[count
])) {
334 for (i
=0; i
<=WCMD_EXIT
; i
++) {
335 if (CompareString (LOCALE_USER_DEFAULT
, NORM_IGNORECASE
| SORT_STRINGSORT
,
336 whichcmd
, count
, inbuilt
[i
], -1) == 2) break;
338 p
= WCMD_strtrim_leading_spaces (&whichcmd
[count
]);
339 WCMD_parse (p
, quals
, param1
, param2
);
343 WCMD_setshow_attrib ();
346 WCMD_batch (param1
, p
, 1);
350 WCMD_setshow_default ();
353 WCMD_clear_screen ();
362 WCMD_setshow_date ();
372 /* Use the unstripped version of the following data - step over the space */
373 /* but only if a parameter follows */
374 if (strlen(&whichcmd
[count
]) > 0)
375 WCMD_echo(&whichcmd
[count
+1]);
377 WCMD_echo(&whichcmd
[count
]);
402 WCMD_setshow_path (p
);
408 WCMD_setshow_prompt ();
421 WCMD_setshow_env (p
);
427 WCMD_setshow_time ();
430 if (strlen(&whichcmd
[count
]) > 0)
431 WCMD_title(&whichcmd
[count
+1]);
448 WCMD_run_program (whichcmd
);
450 HeapFree( GetProcessHeap(), 0, cmd
);
452 CloseHandle (GetStdHandle (STD_INPUT_HANDLE
));
453 SetStdHandle (STD_INPUT_HANDLE
, old_stdin
);
456 CloseHandle (GetStdHandle (STD_OUTPUT_HANDLE
));
457 SetStdHandle (STD_OUTPUT_HANDLE
, old_stdout
);
461 /******************************************************************************
464 * Execute a command line as an external program. If no extension given then
465 * precedence is given to .BAT files. Must allow recursion.
467 * FIXME: Case sensitivity in suffixes!
470 void WCMD_run_program (char *command
) {
473 PROCESS_INFORMATION pe
;
479 char filetorun
[MAX_PATH
];
481 WCMD_parse (command
, quals
, param1
, param2
); /* Quick way to get the filename */
482 if (!(*param1
) && !(*param2
))
484 if (strpbrk (param1
, "\\:") == NULL
) { /* No explicit path given */
485 if ((strchr (param1
, '.') == NULL
) || (strstr (param1
, ".bat") != NULL
)) {
486 if (SearchPath (NULL
, param1
, ".bat", sizeof(filetorun
), filetorun
, NULL
)) {
487 WCMD_batch (filetorun
, command
, 0);
491 if ((strchr (param1
, '.') == NULL
) || (strstr (param1
, ".cmd") != NULL
)) {
492 if (SearchPath (NULL
, param1
, ".cmd", sizeof(filetorun
), filetorun
, NULL
)) {
493 WCMD_batch (filetorun
, command
, 0);
498 else { /* Explicit path given */
499 if ((strstr (param1
, ".bat") != NULL
) ||
500 (strstr (param1
, ".cmd") != NULL
)) {
501 WCMD_batch (param1
, command
, 0);
504 if (strchr (param1
, '.') == NULL
) {
505 strcpy (filetorun
, param1
);
506 strcat (filetorun
, ".bat");
507 h
= CreateFile (filetorun
, GENERIC_READ
, FILE_SHARE_READ
,
508 NULL
, OPEN_EXISTING
, FILE_ATTRIBUTE_NORMAL
, NULL
);
509 if (h
!= INVALID_HANDLE_VALUE
) {
511 WCMD_batch (param1
, command
, 0);
517 /* No batch file found, assume executable */
519 hinst
= FindExecutable (param1
, NULL
, filetorun
);
520 if ((int)hinst
< 32) {
524 console
= SHGetFileInfo (filetorun
, 0, &psfi
, sizeof(psfi
), SHGFI_EXETYPE
);
525 ZeroMemory (&st
, sizeof(STARTUPINFO
));
526 st
.cb
= sizeof(STARTUPINFO
);
527 status
= CreateProcess (NULL
, command
, NULL
, NULL
, FALSE
,
528 0, NULL
, NULL
, &st
, &pe
);
533 if (!console
) errorlevel
= 0;
536 if (!HIWORD(console
)) WaitForSingleObject (pe
.hProcess
, INFINITE
);
537 GetExitCodeProcess (pe
.hProcess
, &errorlevel
);
538 if (errorlevel
== STILL_ACTIVE
) errorlevel
= 0;
542 /******************************************************************************
545 * Display the prompt on STDout
549 void WCMD_show_prompt (void) {
552 char out_string
[MAX_PATH
], curdir
[MAX_PATH
], prompt_string
[MAX_PATH
];
555 status
= GetEnvironmentVariable ("PROMPT", prompt_string
, sizeof(prompt_string
));
556 if ((status
== 0) || (status
> sizeof(prompt_string
))) {
557 lstrcpy (prompt_string
, "$P$G");
569 switch (toupper(*p
)) {
577 GetDateFormat (LOCALE_USER_DEFAULT
, DATE_SHORTDATE
, NULL
, NULL
, q
, MAX_PATH
);
590 status
= GetCurrentDirectory (sizeof(curdir
), curdir
);
596 status
= GetCurrentDirectory (sizeof(curdir
), curdir
);
606 GetTimeFormat (LOCALE_USER_DEFAULT
, 0, NULL
, NULL
, q
, MAX_PATH
);
610 lstrcat (q
, version_string
);
621 WCMD_output (out_string
);
624 /****************************************************************************
627 * Print the message for GetLastError
630 void WCMD_print_error (void) {
635 error_code
= GetLastError ();
636 status
= FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER
| FORMAT_MESSAGE_FROM_SYSTEM
,
637 NULL
, error_code
, 0, (LPTSTR
) &lpMsgBuf
, 0, NULL
);
639 WCMD_output ("FIXME: Cannot display message for error %d, status %d\n",
640 error_code
, GetLastError());
643 WCMD_output (lpMsgBuf
);
644 LocalFree ((HLOCAL
)lpMsgBuf
);
645 WCMD_output (newline
);
649 /*******************************************************************
650 * WCMD_parse - parse a command into parameters and qualifiers.
652 * On exit, all qualifiers are concatenated into q, the first string
653 * not beginning with "/" is in p1 and the
654 * second in p2. Any subsequent non-qualifier strings are lost.
655 * Parameters in quotes are handled.
658 void WCMD_parse (char *s
, char *q
, char *p1
, char *p2
) {
662 *q
= *p1
= *p2
= '\0';
667 while ((*s
!= '\0') && (*s
!= ' ') && *s
!= '/') {
668 *q
++ = toupper (*s
++);
677 while ((*s
!= '\0') && (*s
!= '"')) {
678 if (p
== 0) *p1
++ = *s
++;
679 else if (p
== 1) *p2
++ = *s
++;
682 if (p
== 0) *p1
= '\0';
683 if (p
== 1) *p2
= '\0';
690 while ((*s
!= '\0') && (*s
!= ' ')) {
691 if (p
== 0) *p1
++ = *s
++;
692 else if (p
== 1) *p2
++ = *s
++;
695 if (p
== 0) *p1
= '\0';
696 if (p
== 1) *p2
= '\0';
702 /*******************************************************************
703 * WCMD_output - send output to current standard output device.
707 void WCMD_output (char *format
, ...) {
713 vsprintf (string
, format
, ap
);
715 WCMD_output_asis(string
);
719 static int line_count
;
720 static int max_height
;
721 static BOOL paged_mode
;
723 void WCMD_enter_paged_mode(void)
725 CONSOLE_SCREEN_BUFFER_INFO consoleInfo
;
727 if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE
), &consoleInfo
))
728 max_height
= consoleInfo
.dwSize
.Y
;
732 line_count
= 5; /* keep 5 lines from previous output */
735 void WCMD_leave_paged_mode(void)
740 /*******************************************************************
741 * WCMD_output_asis - send output to current standard output device.
742 * without formatting eg. when message contains '%'
745 void WCMD_output_asis (char *message
) {
752 if ((ptr
= strchr(message
, '\n')) != NULL
) ptr
++;
753 WriteFile (GetStdHandle(STD_OUTPUT_HANDLE
), message
,
754 (ptr
) ? ptr
- message
: lstrlen(message
), &count
, NULL
);
756 if (++line_count
>= max_height
- 1) {
758 WCMD_output_asis (anykey
);
759 ReadFile (GetStdHandle(STD_INPUT_HANDLE
), string
, sizeof(string
), &count
, NULL
);
762 } while ((message
= ptr
) != NULL
);
764 WriteFile (GetStdHandle(STD_OUTPUT_HANDLE
), message
, lstrlen(message
), &count
, NULL
);
769 /***************************************************************************
770 * WCMD_strtrim_leading_spaces
772 * Remove leading spaces from a string. Return a pointer to the first
773 * non-space character. Does not modify the input string
776 char *WCMD_strtrim_leading_spaces (char *string
) {
781 while (*ptr
== ' ') ptr
++;
785 /*************************************************************************
786 * WCMD_strtrim_trailing_spaces
788 * Remove trailing spaces from a string. This routine modifies the input
789 * string by placing a null after the last non-space character
792 void WCMD_strtrim_trailing_spaces (char *string
) {
796 ptr
= string
+ lstrlen (string
) - 1;
797 while ((*ptr
== ' ') && (ptr
>= string
)) {
803 /*************************************************************************
806 * Handle pipes within a command - the DOS way using temporary files.
809 void WCMD_pipe (char *command
) {
812 char temp_path
[MAX_PATH
], temp_file
[MAX_PATH
], temp_file2
[MAX_PATH
], temp_cmd
[1024];
814 GetTempPath (sizeof(temp_path
), temp_path
);
815 GetTempFileName (temp_path
, "WCMD", 0, temp_file
);
816 p
= strchr(command
, '|');
818 wsprintf (temp_cmd
, "%s > %s", command
, temp_file
);
819 WCMD_process_command (temp_cmd
);
821 while ((p
= strchr(command
, '|'))) {
823 GetTempFileName (temp_path
, "WCMD", 0, temp_file2
);
824 wsprintf (temp_cmd
, "%s < %s > %s", command
, temp_file
, temp_file2
);
825 WCMD_process_command (temp_cmd
);
826 DeleteFile (temp_file
);
827 lstrcpy (temp_file
, temp_file2
);
830 wsprintf (temp_cmd
, "%s < %s", command
, temp_file
);
831 WCMD_process_command (temp_cmd
);
832 DeleteFile (temp_file
);