diff options
Diffstat (limited to 'libwinmain/winmain.c')
-rw-r--r-- | libwinmain/winmain.c | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/libwinmain/winmain.c b/libwinmain/winmain.c new file mode 100644 index 000000000..a4ffc1102 --- /dev/null +++ b/libwinmain/winmain.c @@ -0,0 +1,156 @@ +#include <windows.h> + +#include <stdio.h> +#include <fcntl.h> +#include <io.h> + +#define MAX_CONSOLE_LINES 500 + +int main(int argc, char *argv[], char *penv[]); + +/* Ignore control c and control break signals */ +static BOOL WINAPI IgnoreControlC(DWORD dwCtrlType) +{ + if (dwCtrlType==CTRL_C_EVENT || dwCtrlType==CTRL_BREAK_EVENT) + return TRUE; + return FALSE; +} + +static void CreateConsole(void) +{ + static int ConsoleCreated=0; + if (!ConsoleCreated) + { + int hConHandle; + long lStdHandle; + CONSOLE_SCREEN_BUFFER_INFO coninfo; + + FILE *fp; + ConsoleCreated=1; + if (!AttachConsole(ATTACH_PARENT_PROCESS)) + AllocConsole(); + else + SetConsoleCtrlHandler(IgnoreControlC,TRUE); // Only allow control C interrupt the program when a console was created + + + // set the screen buffer to be big enough to let us scroll text + GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &coninfo); + if (coninfo.dwSize.Y < MAX_CONSOLE_LINES) + { + coninfo.dwSize.Y = MAX_CONSOLE_LINES; + SetConsoleScreenBufferSize(GetStdHandle(STD_OUTPUT_HANDLE), coninfo.dwSize); + } + + // redirect unbuffered STDOUT to the console + lStdHandle = (long)GetStdHandle(STD_OUTPUT_HANDLE); + hConHandle = _open_osfhandle(lStdHandle, _O_TEXT); + if (hConHandle==-1) // error occured, so return + return; + fp = _fdopen( hConHandle, "w" ); + *stdout = *fp; + setvbuf( stdout, NULL, _IONBF, 0 ); + + // redirect unbuffered STDIN to the console + lStdHandle = (long)GetStdHandle(STD_INPUT_HANDLE); + hConHandle = _open_osfhandle(lStdHandle, _O_TEXT); + fp = _fdopen( hConHandle, "r" ); + *stdin = *fp; + setvbuf( stdin, NULL, _IONBF, 0 ); + + // redirect unbuffered STDERR to the console + lStdHandle = (long)GetStdHandle(STD_ERROR_HANDLE); + hConHandle = _open_osfhandle(lStdHandle, _O_TEXT); + fp = _fdopen( hConHandle, "w" ); + *stderr = *fp; + setvbuf( stderr, NULL, _IONBF, 0 ); + } +} + +int __stdcall WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow) +{ + int argc=1; + char ProgramName[255]; + #define MAXNRARGS 100 + char *argv[MAXNRARGS]; + char *pTmp; + char *pProgramName; + char **penv; + int Ret; + LPTCH pEnvStrings; + + hInstance=hInstance; + nCmdShow=nCmdShow; + hPrevInstance=hPrevInstance; + + if (!strncmp(lpCmdLine,"-console",8)) + { + CreateConsole(); + lpCmdLine+=9; + } + else + { + SetConsoleCtrlHandler(IgnoreControlC,TRUE); + } + + GetModuleFileName(NULL,ProgramName,255); + pTmp=strrchr(ProgramName,'\\'); + if (pTmp) + pProgramName=pTmp+1; + else + pProgramName=ProgramName; + pTmp=strrchr(pProgramName,'.'); + if (pTmp) + *pTmp=0; + argv[0]=pProgramName; + + pTmp=lpCmdLine; + while (*pTmp && argc<MAXNRARGS-1) + { + char *pEnd; + if (*pTmp=='"') + { + pTmp++; + pEnd=strchr(pTmp,'"'); + } + else if (*pTmp!=' ') + { + pEnd=strchr(pTmp,' '); + } + else + { + pTmp++; + continue; + } + if (pEnd) + { + *pEnd=0; + argv[argc++]=pTmp; + pTmp=pEnd+1; + } + else + { + argv[argc++]=pTmp; + break; + } + } + + { + LPTCH pTmp=GetEnvironmentStrings(); + int NrEnv=0; + pEnvStrings=pTmp; + penv=malloc(sizeof(*penv)); + while (*pTmp) + { + penv=realloc(penv,(NrEnv+2)*sizeof(*penv)); + penv[NrEnv++]=pTmp; + while (*pTmp) pTmp++; + pTmp++; + } + penv[NrEnv]=NULL; + } + + Ret = main(argc,argv,penv); + free(penv); + FreeEnvironmentStrings(pEnvStrings); + return Ret; +} |