mirror of
https://github.com/frida/tinycc
synced 2024-12-23 05:26:49 +03:00
97 lines
2.6 KiB
C
97 lines
2.6 KiB
C
//+---------------------------------------------------------------------------
|
|
|
|
// _UNICODE for tchar.h, UNICODE for API
|
|
#include <tchar.h>
|
|
|
|
#include <windows.h>
|
|
#include <stdlib.h>
|
|
|
|
#define __UNKNOWN_APP 0
|
|
#define __CONSOLE_APP 1
|
|
#define __GUI_APP 2
|
|
void __set_app_type(int);
|
|
void _controlfp(unsigned a, unsigned b);
|
|
|
|
#ifdef _UNICODE
|
|
#define __tgetmainargs __wgetmainargs
|
|
#define _twinstart _wwinstart
|
|
#define _runtwinmain _runwwinmain
|
|
int APIENTRY wWinMain(HINSTANCE, HINSTANCE, LPWSTR, int);
|
|
#else
|
|
#define __tgetmainargs __getmainargs
|
|
#define _twinstart _winstart
|
|
#define _runtwinmain _runwinmain
|
|
#endif
|
|
|
|
extern void (*__init_array_start[]) (void);
|
|
extern void (*__init_array_end[]) (void);
|
|
extern void (*__fini_array_start[]) (void);
|
|
extern void (*__fini_array_end[]) (void);
|
|
|
|
typedef struct { int newmode; } _startupinfo;
|
|
int __cdecl __tgetmainargs(int *pargc, _TCHAR ***pargv, _TCHAR ***penv, int globb, _startupinfo*);
|
|
|
|
static int go_winmain(TCHAR *arg1)
|
|
{
|
|
STARTUPINFO si;
|
|
_TCHAR *szCmd, *p;
|
|
int fShow;
|
|
int retval;
|
|
int i;
|
|
|
|
GetStartupInfo(&si);
|
|
if (si.dwFlags & STARTF_USESHOWWINDOW)
|
|
fShow = si.wShowWindow;
|
|
else
|
|
fShow = SW_SHOWDEFAULT;
|
|
|
|
szCmd = NULL, p = GetCommandLine();
|
|
if (arg1)
|
|
szCmd = _tcsstr(p, arg1);
|
|
if (NULL == szCmd)
|
|
szCmd = _tcsdup(__T(""));
|
|
else if (szCmd > p && szCmd[-1] == __T('"'))
|
|
--szCmd;
|
|
#if defined __i386__ || defined __x86_64__
|
|
_controlfp(0x10000, 0x30000);
|
|
#endif
|
|
i = 0;
|
|
while (&__init_array_start[i] != __init_array_end) {
|
|
(*__init_array_start[i++])();
|
|
}
|
|
retval = _tWinMain(GetModuleHandle(NULL), NULL, szCmd, fShow);
|
|
i = 0;
|
|
while (&__fini_array_end[i] != __fini_array_start) {
|
|
(*__fini_array_end[--i])();
|
|
}
|
|
return retval;
|
|
}
|
|
|
|
static LONG WINAPI catch_sig(EXCEPTION_POINTERS *ex)
|
|
{
|
|
return _XcptFilter(ex->ExceptionRecord->ExceptionCode, ex);
|
|
}
|
|
|
|
int _twinstart(void)
|
|
{
|
|
_startupinfo start_info_con = {0};
|
|
SetUnhandledExceptionFilter(catch_sig);
|
|
__set_app_type(__GUI_APP);
|
|
__tgetmainargs(&__argc, &__targv, &_tenviron, 0, &start_info_con);
|
|
exit(go_winmain(__argc > 1 ? __targv[1] : NULL));
|
|
}
|
|
|
|
int _runtwinmain(int argc, /* as tcc passed in */ char **argv)
|
|
{
|
|
#ifdef UNICODE
|
|
_startupinfo start_info = {0};
|
|
__tgetmainargs(&__argc, &__targv, &_tenviron, 0, &start_info);
|
|
/* may be wrong when tcc has received wildcards (*.c) */
|
|
if (argc < __argc)
|
|
__targv += __argc - argc, __argc = argc;
|
|
#else
|
|
__argc = argc, __targv = argv;
|
|
#endif
|
|
return go_winmain(__argc > 1 ? __targv[1] : NULL);
|
|
}
|