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