src/main/win32/SDL_win32_main.c
author Sam Lantinga
Sat, 29 Dec 2007 18:58:09 +0000
branchSDL-1.2
changeset 4118 3c068a8c016f
parent 4084 70e77cbba87c
child 4246 8b8314cc34a6
permissions -rw-r--r--
Tony White fixed bug #503

Support escaped quotes on Win32 command lines.
slouken@754
     1
/*
slouken@754
     2
    SDL_main.c, placed in the public domain by Sam Lantinga  4/13/98
slouken@754
     3
slouken@754
     4
    The WinMain function -- calls your program's main() function
slouken@754
     5
*/
slouken@754
     6
slouken@754
     7
#include <stdio.h>
slouken@754
     8
#include <stdlib.h>
slouken@754
     9
slouken@1433
    10
#define WIN32_LEAN_AND_MEAN
slouken@1433
    11
#include <windows.h>
slouken@1336
    12
slouken@754
    13
#ifdef _WIN32_WCE
slouken@754
    14
# define DIR_SEPERATOR TEXT("\\")
slouken@766
    15
# undef _getcwd
slouken@766
    16
# define _getcwd(str,len)	wcscpy(str,TEXT(""))
slouken@766
    17
# define setbuf(f,b)
slouken@766
    18
# define setvbuf(w,x,y,z)
slouken@754
    19
# define fopen		_wfopen
slouken@754
    20
# define freopen	_wfreopen
slouken@754
    21
# define remove(x)	DeleteFile(x)
slouken@754
    22
#else
slouken@754
    23
# define DIR_SEPERATOR TEXT("/")
slouken@754
    24
# include <direct.h>
slouken@754
    25
#endif
slouken@754
    26
slouken@754
    27
/* Include the SDL main definition header */
slouken@754
    28
#include "SDL.h"
slouken@754
    29
#include "SDL_main.h"
slouken@754
    30
slouken@754
    31
#ifdef main
slouken@754
    32
# ifndef _WIN32_WCE_EMULATION
slouken@754
    33
#  undef main
slouken@754
    34
# endif /* _WIN32_WCE_EMULATION */
slouken@754
    35
#endif /* main */
slouken@754
    36
slouken@754
    37
/* The standard output files */
slouken@754
    38
#define STDOUT_FILE	TEXT("stdout.txt")
slouken@754
    39
#define STDERR_FILE	TEXT("stderr.txt")
slouken@754
    40
slouken@754
    41
#ifndef NO_STDIO_REDIRECT
slouken@754
    42
# ifdef _WIN32_WCE
slouken@754
    43
  static wchar_t stdoutPath[MAX_PATH];
slouken@754
    44
  static wchar_t stderrPath[MAX_PATH];
slouken@754
    45
# else
slouken@754
    46
  static char stdoutPath[MAX_PATH];
slouken@754
    47
  static char stderrPath[MAX_PATH];
slouken@754
    48
# endif
slouken@754
    49
#endif
slouken@754
    50
slouken@754
    51
#if defined(_WIN32_WCE) && _WIN32_WCE < 300
slouken@754
    52
/* seems to be undefined in Win CE although in online help */
slouken@754
    53
#define isspace(a) (((CHAR)a == ' ') || ((CHAR)a == '\t'))
slouken@754
    54
#endif /* _WIN32_WCE < 300 */
slouken@754
    55
slouken@4118
    56
static void UnEscapeQuotes( char *arg )
slouken@4118
    57
{
slouken@4118
    58
	char *last = NULL;
slouken@4118
    59
slouken@4118
    60
	while( *arg ) {
slouken@4118
    61
		if( *arg == '"' && *last == '\\' ) {
slouken@4118
    62
			char *c_curr = arg;
slouken@4118
    63
			char *c_last = last;
slouken@4118
    64
slouken@4118
    65
			while( *c_curr ) {
slouken@4118
    66
				*c_last = *c_curr;
slouken@4118
    67
				c_last = c_curr;
slouken@4118
    68
				c_curr++;
slouken@4118
    69
			}
slouken@4118
    70
			*c_last = '\0';
slouken@4118
    71
		}
slouken@4118
    72
		last = arg;
slouken@4118
    73
		arg++;
slouken@4118
    74
	}
slouken@4118
    75
}
slouken@4118
    76
slouken@754
    77
/* Parse a command line buffer into arguments */
slouken@754
    78
static int ParseCommandLine(char *cmdline, char **argv)
slouken@754
    79
{
slouken@754
    80
	char *bufp;
slouken@4118
    81
	char *lastp = NULL;
slouken@4118
    82
	int argc, last_argc;
slouken@754
    83
slouken@4118
    84
	argc = last_argc = 0;
slouken@754
    85
	for ( bufp = cmdline; *bufp; ) {
slouken@754
    86
		/* Skip leading whitespace */
slouken@754
    87
		while ( isspace(*bufp) ) {
slouken@754
    88
			++bufp;
slouken@754
    89
		}
slouken@754
    90
		/* Skip over argument */
slouken@754
    91
		if ( *bufp == '"' ) {
slouken@754
    92
			++bufp;
slouken@754
    93
			if ( *bufp ) {
slouken@754
    94
				if ( argv ) {
slouken@754
    95
					argv[argc] = bufp;
slouken@754
    96
				}
slouken@754
    97
				++argc;
slouken@754
    98
			}
slouken@754
    99
			/* Skip over word */
slouken@4118
   100
			while ( *bufp && ( *bufp != '"' || *lastp == '\\' ) ) {
slouken@4118
   101
				lastp = bufp;
slouken@754
   102
				++bufp;
slouken@754
   103
			}
slouken@754
   104
		} else {
slouken@754
   105
			if ( *bufp ) {
slouken@754
   106
				if ( argv ) {
slouken@754
   107
					argv[argc] = bufp;
slouken@754
   108
				}
slouken@754
   109
				++argc;
slouken@754
   110
			}
slouken@754
   111
			/* Skip over word */
slouken@754
   112
			while ( *bufp && ! isspace(*bufp) ) {
slouken@754
   113
				++bufp;
slouken@754
   114
			}
slouken@754
   115
		}
slouken@754
   116
		if ( *bufp ) {
slouken@754
   117
			if ( argv ) {
slouken@754
   118
				*bufp = '\0';
slouken@754
   119
			}
slouken@754
   120
			++bufp;
slouken@754
   121
		}
slouken@4118
   122
slouken@4118
   123
		/* Strip out \ from \" sequences */
slouken@4118
   124
		if( argv && last_argc != argc ) {
slouken@4118
   125
			UnEscapeQuotes( argv[last_argc] );	
slouken@4118
   126
		}
slouken@4118
   127
		last_argc = argc;	
slouken@754
   128
	}
slouken@754
   129
	if ( argv ) {
slouken@754
   130
		argv[argc] = NULL;
slouken@754
   131
	}
slouken@754
   132
	return(argc);
slouken@754
   133
}
slouken@754
   134
slouken@754
   135
/* Show an error message */
slouken@754
   136
static void ShowError(const char *title, const char *message)
slouken@754
   137
{
slouken@754
   138
/* If USE_MESSAGEBOX is defined, you need to link with user32.lib */
slouken@754
   139
#ifdef USE_MESSAGEBOX
slouken@754
   140
	MessageBox(NULL, message, title, MB_ICONEXCLAMATION|MB_OK);
slouken@754
   141
#else
slouken@754
   142
	fprintf(stderr, "%s: %s\n", title, message);
slouken@754
   143
#endif
slouken@754
   144
}
slouken@754
   145
slouken@754
   146
/* Pop up an out of memory message, returns to Windows */
slouken@754
   147
static BOOL OutOfMemory(void)
slouken@754
   148
{
slouken@754
   149
	ShowError("Fatal Error", "Out of memory - aborting");
slouken@754
   150
	return FALSE;
slouken@754
   151
}
slouken@754
   152
slouken@1769
   153
/* SDL_Quit() shouldn't be used with atexit() directly because
slouken@1769
   154
   calling conventions may differ... */
slouken@1769
   155
static void cleanup(void)
slouken@1769
   156
{
slouken@1769
   157
	SDL_Quit();
slouken@1769
   158
}
slouken@1769
   159
slouken@754
   160
/* Remove the output files if there was no output written */
slouken@1769
   161
static void cleanup_output(void)
slouken@754
   162
{
slouken@754
   163
#ifndef NO_STDIO_REDIRECT
slouken@754
   164
	FILE *file;
slouken@754
   165
	int empty;
slouken@754
   166
#endif
slouken@754
   167
slouken@754
   168
	/* Flush the output in case anything is queued */
slouken@754
   169
	fclose(stdout);
slouken@754
   170
	fclose(stderr);
slouken@754
   171
slouken@754
   172
#ifndef NO_STDIO_REDIRECT
slouken@754
   173
	/* See if the files have any output in them */
slouken@754
   174
	if ( stdoutPath[0] ) {
slouken@754
   175
		file = fopen(stdoutPath, TEXT("rb"));
slouken@754
   176
		if ( file ) {
slouken@754
   177
			empty = (fgetc(file) == EOF) ? 1 : 0;
slouken@754
   178
			fclose(file);
slouken@754
   179
			if ( empty ) {
slouken@754
   180
				remove(stdoutPath);
slouken@754
   181
			}
slouken@754
   182
		}
slouken@754
   183
	}
slouken@754
   184
	if ( stderrPath[0] ) {
slouken@754
   185
		file = fopen(stderrPath, TEXT("rb"));
slouken@754
   186
		if ( file ) {
slouken@754
   187
			empty = (fgetc(file) == EOF) ? 1 : 0;
slouken@754
   188
			fclose(file);
slouken@754
   189
			if ( empty ) {
slouken@754
   190
				remove(stderrPath);
slouken@754
   191
			}
slouken@754
   192
		}
slouken@754
   193
	}
slouken@754
   194
#endif
slouken@754
   195
}
slouken@754
   196
slouken@754
   197
#if defined(_MSC_VER) && !defined(_WIN32_WCE)
slouken@754
   198
/* The VC++ compiler needs main defined */
slouken@754
   199
#define console_main main
slouken@754
   200
#endif
slouken@754
   201
slouken@754
   202
/* This is where execution begins [console apps] */
slouken@754
   203
int console_main(int argc, char *argv[])
slouken@754
   204
{
slouken@1472
   205
	size_t n;
slouken@754
   206
	char *bufp, *appname;
slouken@1423
   207
	int status;
slouken@754
   208
slouken@754
   209
	/* Get the class name from argv[0] */
slouken@754
   210
	appname = argv[0];
slouken@1336
   211
	if ( (bufp=SDL_strrchr(argv[0], '\\')) != NULL ) {
slouken@754
   212
		appname = bufp+1;
slouken@754
   213
	} else
slouken@1336
   214
	if ( (bufp=SDL_strrchr(argv[0], '/')) != NULL ) {
slouken@754
   215
		appname = bufp+1;
slouken@754
   216
	}
slouken@754
   217
slouken@1336
   218
	if ( (bufp=SDL_strrchr(appname, '.')) == NULL )
slouken@1336
   219
		n = SDL_strlen(appname);
slouken@754
   220
	else
slouken@754
   221
		n = (bufp-appname);
slouken@754
   222
slouken@1423
   223
	bufp = SDL_stack_alloc(char, n+1);
slouken@754
   224
	if ( bufp == NULL ) {
slouken@754
   225
		return OutOfMemory();
slouken@754
   226
	}
slouken@1769
   227
	SDL_strlcpy(bufp, appname, n+1);
slouken@754
   228
	appname = bufp;
slouken@754
   229
slouken@754
   230
	/* Load SDL dynamic link library */
slouken@754
   231
	if ( SDL_Init(SDL_INIT_NOPARACHUTE) < 0 ) {
slouken@754
   232
		ShowError("WinMain() error", SDL_GetError());
slouken@754
   233
		return(FALSE);
slouken@754
   234
	}
slouken@754
   235
	atexit(cleanup_output);
slouken@1769
   236
	atexit(cleanup);
slouken@754
   237
slouken@754
   238
	/* Sam:
slouken@754
   239
	   We still need to pass in the application handle so that
slouken@754
   240
	   DirectInput will initialize properly when SDL_RegisterApp()
slouken@754
   241
	   is called later in the video initialization.
slouken@754
   242
	 */
slouken@754
   243
	SDL_SetModuleHandle(GetModuleHandle(NULL));
slouken@754
   244
slouken@754
   245
	/* Run the application main() code */
slouken@1423
   246
	status = SDL_main(argc, argv);
slouken@754
   247
slouken@754
   248
	/* Exit cleanly, calling atexit() functions */
slouken@1423
   249
	exit(status);
slouken@754
   250
slouken@754
   251
	/* Hush little compiler, don't you cry... */
slouken@1423
   252
	return 0;
slouken@754
   253
}
slouken@754
   254
slouken@754
   255
/* This is where execution begins [windowed apps] */
slouken@754
   256
#ifdef _WIN32_WCE
slouken@754
   257
int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPWSTR szCmdLine, int sw)
slouken@754
   258
#else
slouken@754
   259
int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR szCmdLine, int sw)
slouken@754
   260
#endif
slouken@754
   261
{
slouken@754
   262
	HINSTANCE handle;
slouken@754
   263
	char **argv;
slouken@754
   264
	int argc;
slouken@754
   265
	char *cmdline;
slouken@754
   266
#ifdef _WIN32_WCE
slouken@754
   267
	wchar_t *bufp;
slouken@754
   268
	int nLen;
slouken@754
   269
#else
slouken@754
   270
	char *bufp;
slouken@1379
   271
	size_t nLen;
slouken@754
   272
#endif
slouken@754
   273
#ifndef NO_STDIO_REDIRECT
slouken@4084
   274
	DWORD pathlen;
slouken@4084
   275
#ifdef _WIN32_WCE
slouken@4084
   276
	wchar_t path[MAX_PATH];
slouken@4084
   277
#else
slouken@4084
   278
	char path[MAX_PATH];
slouken@4084
   279
#endif
slouken@754
   280
	FILE *newfp;
slouken@754
   281
#endif
slouken@754
   282
slouken@754
   283
	/* Start up DDHELP.EXE before opening any files, so DDHELP doesn't
slouken@754
   284
	   keep them open.  This is a hack.. hopefully it will be fixed 
slouken@754
   285
	   someday.  DDHELP.EXE starts up the first time DDRAW.DLL is loaded.
slouken@754
   286
	 */
slouken@754
   287
	handle = LoadLibrary(TEXT("DDRAW.DLL"));
slouken@754
   288
	if ( handle != NULL ) {
slouken@754
   289
		FreeLibrary(handle);
slouken@754
   290
	}
slouken@754
   291
slouken@754
   292
#ifndef NO_STDIO_REDIRECT
slouken@1379
   293
	pathlen = GetModuleFileName(NULL, path, SDL_arraysize(path));
slouken@1286
   294
	while ( pathlen > 0 && path[pathlen] != '\\' ) {
slouken@1286
   295
		--pathlen;
slouken@1286
   296
	}
slouken@1286
   297
	path[pathlen] = '\0';
slouken@1286
   298
slouken@1465
   299
#ifdef _WIN32_WCE
slouken@1465
   300
	wcsncpy( stdoutPath, path, SDL_arraysize(stdoutPath) );
slouken@1465
   301
	wcsncat( stdoutPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );
slouken@1465
   302
#else
slouken@1379
   303
	SDL_strlcpy( stdoutPath, path, SDL_arraysize(stdoutPath) );
slouken@1379
   304
	SDL_strlcat( stdoutPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );
slouken@1465
   305
#endif
slouken@754
   306
    
slouken@754
   307
	/* Redirect standard input and standard output */
slouken@754
   308
	newfp = freopen(stdoutPath, TEXT("w"), stdout);
slouken@754
   309
slouken@754
   310
#ifndef _WIN32_WCE
slouken@754
   311
	if ( newfp == NULL ) {	/* This happens on NT */
slouken@754
   312
#if !defined(stdout)
slouken@754
   313
		stdout = fopen(stdoutPath, TEXT("w"));
slouken@754
   314
#else
slouken@754
   315
		newfp = fopen(stdoutPath, TEXT("w"));
slouken@754
   316
		if ( newfp ) {
slouken@754
   317
			*stdout = *newfp;
slouken@754
   318
		}
slouken@754
   319
#endif
slouken@754
   320
	}
slouken@754
   321
#endif /* _WIN32_WCE */
slouken@754
   322
slouken@1465
   323
#ifdef _WIN32_WCE
slouken@1465
   324
	wcsncpy( stderrPath, path, SDL_arraysize(stdoutPath) );
slouken@1465
   325
	wcsncat( stderrPath, DIR_SEPERATOR STDOUT_FILE, SDL_arraysize(stdoutPath) );
slouken@1465
   326
#else
slouken@1379
   327
	SDL_strlcpy( stderrPath, path, SDL_arraysize(stderrPath) );
slouken@1379
   328
	SDL_strlcat( stderrPath, DIR_SEPERATOR STDERR_FILE, SDL_arraysize(stderrPath) );
slouken@1465
   329
#endif
slouken@754
   330
slouken@754
   331
	newfp = freopen(stderrPath, TEXT("w"), stderr);
slouken@754
   332
#ifndef _WIN32_WCE
slouken@754
   333
	if ( newfp == NULL ) {	/* This happens on NT */
slouken@754
   334
#if !defined(stderr)
slouken@754
   335
		stderr = fopen(stderrPath, TEXT("w"));
slouken@754
   336
#else
slouken@754
   337
		newfp = fopen(stderrPath, TEXT("w"));
slouken@754
   338
		if ( newfp ) {
slouken@754
   339
			*stderr = *newfp;
slouken@754
   340
		}
slouken@754
   341
#endif
slouken@754
   342
	}
slouken@754
   343
#endif /* _WIN32_WCE */
slouken@754
   344
slouken@754
   345
	setvbuf(stdout, NULL, _IOLBF, BUFSIZ);	/* Line buffered */
slouken@754
   346
	setbuf(stderr, NULL);			/* No buffering */
slouken@754
   347
#endif /* !NO_STDIO_REDIRECT */
slouken@754
   348
slouken@754
   349
#ifdef _WIN32_WCE
slouken@754
   350
	nLen = wcslen(szCmdLine)+128+1;
slouken@1423
   351
	bufp = SDL_stack_alloc(wchar_t, nLen*2);
slouken@754
   352
	wcscpy (bufp, TEXT("\""));
slouken@754
   353
	GetModuleFileName(NULL, bufp+1, 128-3);
slouken@754
   354
	wcscpy (bufp+wcslen(bufp), TEXT("\" "));
slouken@754
   355
	wcsncpy(bufp+wcslen(bufp), szCmdLine,nLen-wcslen(bufp));
slouken@754
   356
	nLen = wcslen(bufp)+1;
slouken@1465
   357
	cmdline = SDL_stack_alloc(char, nLen);
slouken@754
   358
	if ( cmdline == NULL ) {
slouken@754
   359
		return OutOfMemory();
slouken@754
   360
	}
slouken@754
   361
	WideCharToMultiByte(CP_ACP, 0, bufp, -1, cmdline, nLen, NULL, NULL);
slouken@754
   362
#else
slouken@1423
   363
	/* Grab the command line */
slouken@754
   364
	bufp = GetCommandLine();
slouken@1379
   365
	nLen = SDL_strlen(bufp)+1;
slouken@1379
   366
	cmdline = SDL_stack_alloc(char, nLen);
slouken@754
   367
	if ( cmdline == NULL ) {
slouken@754
   368
		return OutOfMemory();
slouken@754
   369
	}
slouken@1379
   370
	SDL_strlcpy(cmdline, bufp, nLen);
slouken@754
   371
#endif
slouken@754
   372
slouken@754
   373
	/* Parse it into argv and argc */
slouken@754
   374
	argc = ParseCommandLine(cmdline, NULL);
slouken@1379
   375
	argv = SDL_stack_alloc(char*, argc+1);
slouken@754
   376
	if ( argv == NULL ) {
slouken@754
   377
		return OutOfMemory();
slouken@754
   378
	}
slouken@754
   379
	ParseCommandLine(cmdline, argv);
slouken@754
   380
slouken@754
   381
	/* Run the main program (after a little SDL initialization) */
slouken@1423
   382
	console_main(argc, argv);
slouken@1379
   383
slouken@1423
   384
	/* Hush little compiler, don't you cry... */
slouken@1423
   385
	return 0;
slouken@754
   386
}