/* Copyright (C) 1996-1997 Id Software, Inc. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ // sys_win.h #include "quakedef.h" #include "winquake.h" #include "resource.h" #include "errno.h" #include "fcntl.h" #include #include #include #include #include "pr_common.h" #include "fs.h" #ifdef _MSC_VER #define MSVC_SEH #endif //#define RESTARTTEST #ifdef MULTITHREAD #include #endif //exports that 3rd-party drivers can see in order to use descrete graphics cards over integrated ones, FOR MORE POWAH! __declspec(dllexport) DWORD NvOptimusEnablement = 0x00000001; __declspec(dllexport) int AmdPowerXpressRequestHighPerformance = 1; //13.35+ #ifdef WINRT //you're going to need a different sys_ port. qboolean isDedicated = false; qboolean ActiveApp; void VARGS Sys_Error (const char *error, ...){} //eep void VARGS Sys_Printf (char *fmt, ...){} //safe, but not ideal (esp for debugging) void Sys_SendKeyEvents (void){} //safe, but not ideal void Sys_ServerActivity(void){} //empty is safe void Sys_RecentServer(char *command, char *target, char *title, char *desc){} //empty is safe qboolean Sys_InitTerminal(void){return false;} //failure will break 'setrenderer sv' char *Sys_ConsoleInput (void){return NULL;} //safe to stub void Sys_CloseTerminal (void){} //called when switching from dedicated->non-dedicated dllhandle_t *Sys_LoadLibrary(const char *name, dllfunction_t *funcs){return NULL;} //can just about get away with it void *Sys_GetAddressForName(dllhandle_t *module, const char *exportname){return NULL;} void Sys_CloseLibrary(dllhandle_t *lib){} //safe, ish void Sys_Init (void){} //safe, stub is fine. used to register system-specific cvars/commands. void Sys_Shutdown(void){} //safe qboolean Sys_RandomBytes(qbyte *string, int len){return false;} qboolean Sys_GetDesktopParameters(int *width, int *height, int *bpp, int *refreshrate){return false;} void INS_Move(float *movements, int pnum){} //safe void INS_Commands(void){} //safe void INS_Init(void){} //safe. should be xinput2 I guess. nothing else is actually supported. touchscreens don't really count. void INS_ReInit(void){} //safe void INS_Shutdown(void){} //safe void INS_UpdateGrabs(int fullscreen, int activeapp){} //safe void *RT_GetCoreWindow(int *width, int *height){return NULL;} //I already wrote the d3d code, but it needs a window to attach to. you can override width+height by writing to them void D3D11_DoResize(int newwidth, int newheight); //already written, call if resized since getcorewindow static char *clippy; char *Sys_GetClipboard(void) { return Z_StrDup(clippy); } void Sys_CloseClipboard(char *bf) { Z_Free(bf); } void Sys_SaveClipboard(char *text) { Z_Free(clippy); clippy = Z_StrDup(text); } static LARGE_INTEGER timestart, timefreq; static qboolean timeinited = false; unsigned int Sys_Milliseconds(void) { LARGE_INTEGER cur, diff; if (!timeinited) { timeinited = true; QueryPerformanceFrequency(&timefreq); QueryPerformanceCounter(×tart); } QueryPerformanceCounter(&cur); diff.QuadPart = cur.QuadPart - timestart.QuadPart; diff.QuadPart *= 1000; diff.QuadPart /= timefreq.QuadPart; return diff.QuadPart; } double Sys_DoubleTime (void) { LARGE_INTEGER cur, diff; if (!timeinited) { timeinited = true; QueryPerformanceFrequency(&timefreq); QueryPerformanceCounter(×tart); } QueryPerformanceCounter(&cur); diff.QuadPart = cur.QuadPart - timestart.QuadPart; diff.QuadPart *= 1000; return (double)diff.QuadPart / (double)timefreq.QuadPart; //I hope the timefreq doesn't get rounded and cause milliseconds and doubletime to start to drift apart. } void Sys_Quit (void) { Host_Shutdown (); exit(0); } void Sys_mkdir (char *path) { wchar_t wide[MAX_OSPATH]; widen(wide, sizeof(wide), path); CreateDirectoryW(wide, NULL); } qboolean Sys_remove (char *path) { wchar_t wide[MAX_OSPATH]; widen(wide, sizeof(wide), path); if (DeleteFileW(wide)) return true; //success if (GetLastError() == ERROR_FILE_NOT_FOUND) return true; //succeed when the file already didn't exist return false; //other errors? panic } qboolean Sys_Rename (char *oldfname, char *newfname) { wchar_t oldwide[MAX_OSPATH]; wchar_t newwide[MAX_OSPATH]; widen(oldwide, sizeof(oldwide), oldfname); widen(newwide, sizeof(newwide), newfname); return MoveFileExW(oldwide, newwide, MOVEFILE_REPLACE_EXISTING|MOVEFILE_COPY_ALLOWED); } //enumeratefiles is recursive for */* to work static int Sys_EnumerateFiles2 (const char *match, int matchstart, int neststart, int (QDECL *func)(const char *fname, qofs_t fsize, time_t mtime, void *parm, searchpathfuncs_t *spath), void *parm, searchpathfuncs_t *spath) { qboolean go; HANDLE r; WIN32_FIND_DATAW fd; int nest = neststart; //neststart refers to just after a / qboolean wild = false; while(match[nest] && match[nest] != '/') { if (match[nest] == '?' || match[nest] == '*') wild = true; nest++; } if (match[nest] == '/') { char submatch[MAX_OSPATH]; char tmproot[MAX_OSPATH]; if (!wild) return Sys_EnumerateFiles2(match, matchstart, nest+1, func, parm, spath); if (nest-neststart+1> MAX_OSPATH) return 1; memcpy(submatch, match+neststart, nest - neststart); submatch[nest - neststart] = 0; nest++; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); { wchar_t wroot[MAX_OSPATH]; r = FindFirstFileExW(widen(wroot, sizeof(wroot), tmproot), FindExInfoStandard, &fd, FindExSearchNameMatch, NULL, 0); } strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { char utf8[MAX_OSPATH]; char file[MAX_OSPATH]; narrowen(utf8, sizeof(utf8), fd.cFileName); if (*utf8 == '.'); //don't ever find files with a name starting with '.' else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, utf8)) { int newnest; if (strlen(tmproot) + strlen(utf8) + strlen(match+nest) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot, utf8); newnest = strlen(file); strcpy(file+newnest, match+nest); go = Sys_EnumerateFiles2(file, matchstart, newnest, func, parm, spath); } } } } while(FindNextFileW(r, &fd) && go); FindClose(r); } else { const char *submatch = match + neststart; char tmproot[MAX_OSPATH]; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); { wchar_t wroot[MAX_OSPATH]; r = FindFirstFileExW(widen(wroot, sizeof(wroot), tmproot), FindExInfoStandard, &fd, FindExSearchNameMatch, NULL, 0); } strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { char utf8[MAX_OSPATH]; char file[MAX_OSPATH]; narrowen(utf8, sizeof(utf8), fd.cFileName); if (*utf8 == '.') ; //don't ever find files with a name starting with '.' (includes .. and . directories, and unix hidden files) else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, utf8)) { if (strlen(tmproot+matchstart) + strlen(utf8) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot+matchstart, utf8); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), 0, parm, spath); } } } else { if (wildcmp(submatch, utf8)) { if (strlen(tmproot+matchstart) + strlen(utf8) + 1 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s", tmproot+matchstart, utf8); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), 0, parm, spath); } } } } while(FindNextFileW(r, &fd) && go); FindClose(r); } return go; } int Sys_EnumerateFiles (const char *gpath, const char *match, int (QDECL *func)(const char *fname, qofs_t fsize, void *parm, searchpathfuncs_t *spath), void *parm, searchpathfuncs_t *spath) { char fullmatch[MAX_OSPATH]; int start; if (strlen(gpath) + strlen(match) + 2 > MAX_OSPATH) return 1; strcpy(fullmatch, gpath); start = strlen(fullmatch); if (start && fullmatch[start-1] != '/') fullmatch[start++] = '/'; fullmatch[start] = 0; strcat(fullmatch, match); return Sys_EnumerateFiles2(fullmatch, start, start, func, parm, spath); } #else #if defined(GLQUAKE) #define PRINTGLARRAYS #endif #if defined(_DEBUG) || defined(DEBUG) #define CATCHCRASH #endif #if !defined(CLIENTONLY) && !defined(SERVERONLY) qboolean isDedicated = false; #endif extern qboolean isPlugin; qboolean debugout; float gammapending; //to cope with ATI. When it times out, v_gamma is reforced in order to correct/update gamma now the drivers think that they have won. HWND sys_parentwindow; unsigned int sys_parentleft; //valid if sys_parentwindow is set unsigned int sys_parenttop; unsigned int sys_parentwidth; //valid if sys_parentwindow is set unsigned int sys_parentheight; static struct { int width; int height; int rate; int bpp; } desktopsettings; static void Sys_QueryDesktopParameters(void); //used to do special things with awkward windows versions. int qwinvermaj; int qwinvermin; char *sys_argv[MAX_NUM_ARGVS]; #ifdef RESTARTTEST jmp_buf restart_jmpbuf; #endif /* ================ Sys_RandomBytes ================ */ #include qboolean Sys_RandomBytes(qbyte *string, int len) { HCRYPTPROV prov; if(!CryptAcquireContext( &prov, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) { return false; } if(!CryptGenRandom(prov, len, (BYTE *)string)) { CryptReleaseContext( prov, 0); return false; } CryptReleaseContext(prov, 0); return true; } /* ================= Library loading ================= */ void Sys_CloseLibrary(dllhandle_t *lib) { FreeLibrary((HMODULE)lib); } HMODULE LoadLibraryU(const char *name) { if (WinNT) { wchar_t wide[MAX_OSPATH]; widen(wide, sizeof(wide), name); return LoadLibraryW(wide); } else { wchar_t wide[MAX_OSPATH]; char ansi[MAX_OSPATH]; widen(wide, sizeof(wide), name); ansi[WideCharToMultiByte(CP_ACP, 0, wide, wcslen(wide), ansi, sizeof(ansi)-1, NULL, NULL)] = 0; return LoadLibraryA(ansi); } } dllhandle_t *Sys_LoadLibrary(const char *name, dllfunction_t *funcs) { int i; HMODULE lib; lib = LoadLibraryU(name); if (!lib) { if (!strstr(COM_SkipPath(name), ".dll")) { //.dll implies that it is a system dll, or something that is otherwise windows-specific already. char libname[MAX_OSPATH]; #ifdef _WIN64 Q_snprintfz(libname, sizeof(libname), "%s_64", name); #elif defined(_WIN32) Q_snprintfz(libname, sizeof(libname), "%s_32", name); #else #error wut? not win32? #endif lib = LoadLibraryU(libname); } if (!lib) return NULL; } if (funcs) { for (i = 0; funcs[i].name; i++) { *funcs[i].funcptr = GetProcAddress(lib, funcs[i].name); if (!*funcs[i].funcptr) break; } if (funcs[i].name) { Con_DPrintf("Missing export \"%s\" in \"%s\"\n", funcs[i].name, name); Sys_CloseLibrary((dllhandle_t*)lib); lib = NULL; } } return (dllhandle_t*)lib; } void *Sys_GetAddressForName(dllhandle_t *module, const char *exportname) { if (!module) return NULL; return GetProcAddress((HINSTANCE)module, exportname); } #ifdef HLSERVER char *Sys_GetNameForAddress(dllhandle_t *module, void *address) { //windows doesn't provide a function to do this, so we have to do it ourselves. //this isn't the fastest way... //halflife needs this function. char *base = (char *)module; IMAGE_DATA_DIRECTORY *datadir; IMAGE_EXPORT_DIRECTORY *block; IMAGE_NT_HEADERS *ntheader; IMAGE_DOS_HEADER *dosheader = (void*)base; int i, j; DWORD *funclist; DWORD *namelist; SHORT *ordilist; if (!dosheader || dosheader->e_magic != IMAGE_DOS_SIGNATURE) return NULL; //yeah, that wasn't an exe ntheader = (void*)(base + dosheader->e_lfanew); if (!dosheader->e_lfanew || ntheader->Signature != IMAGE_NT_SIGNATURE) return NULL; //urm, wait, a 16bit dos exe? datadir = &ntheader->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT]; block = (IMAGE_EXPORT_DIRECTORY *)(base + datadir->VirtualAddress); funclist = (DWORD*)(base+block->AddressOfFunctions); namelist = (DWORD*)(base+block->AddressOfNames); ordilist = (SHORT*)(base+block->AddressOfNameOrdinals); for (i = 0; i < block->NumberOfFunctions; i++) { if (base+funclist[i] == address) { for (j = 0; j < block->NumberOfNames; j++) { if (ordilist[j] == i) { return base+namelist[i]; } } //it has no name. huh? return NULL; } } return NULL; } #endif int starttime; qboolean ActiveApp, Minimized; qboolean WinNT; //NT has a) proper unicode support that does not unconditionally result in errors. b) a few different registry paths. static HANDLE hinput, houtput; HANDLE qwclsemaphore; static HANDLE tevent; void Sys_InitFloatTime (void); int VARGS Sys_DebugLog(char *file, char *fmt, ...) { FILE *fd; va_list argptr; static char data[1024]; va_start(argptr, fmt); vsnprintf(data, sizeof(data)-1, fmt, argptr); va_end(argptr); #if defined(CRAZYDEBUGGING) && CRAZYDEBUGGING > 1 { static int sock; if (!sock) { struct sockaddr_in sa; netadr_t na; int _true = true; int listip; listip = COM_CheckParm("-debugip"); NET_StringToAdr(listip?com_argv[listip+1]:"127.0.0.1", &na); NetadrToSockadr(&na, (struct sockaddr_qstorage*)&sa); sa.sin_port = htons(10000); sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (-1==connect(sock, (struct sockaddr*)&sa, sizeof(sa))) Sys_Error("Couldn't send debug log lines\n"); setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (char*)&_true, sizeof(_true)); } send(sock, data, strlen(data), 0); } #endif fd = fopen(file, "ab"); if (fd) { fprintf(fd, "%s", data); fclose(fd); return 0; } return 1; }; #ifdef CATCHCRASH #include "dbghelp.h" typedef BOOL (WINAPI *MINIDUMPWRITEDUMP) ( HANDLE hProcess, DWORD ProcessId, HANDLE hFile, MINIDUMP_TYPE DumpType, PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam, PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam, PMINIDUMP_CALLBACK_INFORMATION CallbackParam ); void DumpGLState(void); void *watchdogthread; DWORD CrashExceptionHandler (qboolean iswatchdog, DWORD exceptionCode, LPEXCEPTION_POINTERS exceptionInfo) { char dumpPath[1024]; char msg[1024]; HANDLE hProc = GetCurrentProcess(); DWORD procid = GetCurrentProcessId(); HANDLE dumpfile; HMODULE hDbgHelp; MINIDUMPWRITEDUMP fnMiniDumpWriteDump; HMODULE hKernel; BOOL (WINAPI *pIsDebuggerPresent)(void); DWORD (WINAPI *pSymSetOptions)(DWORD SymOptions); BOOL (WINAPI *pSymInitialize)(HANDLE hProcess, PSTR UserSearchPath, BOOL fInvadeProcess); BOOL (WINAPI *pSymFromAddr)(HANDLE hProcess, DWORD64 Address, PDWORD64 Displacement, PSYMBOL_INFO Symbol); #ifdef _WIN64 #define DBGHELP_POSTFIX "64" BOOL (WINAPI *pStackWalkX)(DWORD MachineType, HANDLE hProcess, HANDLE hThread, LPSTACKFRAME64 StackFrame, PVOID ContextRecord, PREAD_PROCESS_MEMORY_ROUTINE64 ReadMemoryRoutine, PFUNCTION_TABLE_ACCESS_ROUTINE64 FunctionTableAccessRoutine, PGET_MODULE_BASE_ROUTINE64 GetModuleBaseRoutine, PTRANSLATE_ADDRESS_ROUTINE64 TranslateAddress); PVOID (WINAPI *pSymFunctionTableAccessX)(HANDLE hProcess, DWORD64 AddrBase); DWORD64 (WINAPI *pSymGetModuleBaseX)(HANDLE hProcess, DWORD64 qwAddr); BOOL (WINAPI *pSymGetLineFromAddrX)(HANDLE hProcess, DWORD64 qwAddr, PDWORD pdwDisplacement, PIMAGEHLP_LINE64 Line64); BOOL (WINAPI *pSymGetModuleInfoX)(HANDLE hProcess, DWORD64 qwAddr, PIMAGEHLP_MODULE64 ModuleInfo); #define STACKFRAMEX STACKFRAME64 #define IMAGEHLP_LINEX IMAGEHLP_LINE64 #define IMAGEHLP_MODULEX IMAGEHLP_MODULE64 #else #define DBGHELP_POSTFIX "" BOOL (WINAPI *pStackWalkX)(DWORD MachineType, HANDLE hProcess, HANDLE hThread, LPSTACKFRAME StackFrame, PVOID ContextRecord, PREAD_PROCESS_MEMORY_ROUTINE ReadMemoryRoutine, PFUNCTION_TABLE_ACCESS_ROUTINE FunctionTableAccessRoutine, PGET_MODULE_BASE_ROUTINE GetModuleBaseRoutine, PTRANSLATE_ADDRESS_ROUTINE TranslateAddress); PVOID (WINAPI *pSymFunctionTableAccessX)(HANDLE hProcess, DWORD AddrBase); DWORD (WINAPI *pSymGetModuleBaseX)(HANDLE hProcess, DWORD dwAddr); BOOL (WINAPI *pSymGetLineFromAddrX)(HANDLE hProcess, DWORD dwAddr, PDWORD pdwDisplacement, PIMAGEHLP_LINE Line); BOOL (WINAPI *pSymGetModuleInfoX)(HANDLE hProcess, DWORD dwAddr, PIMAGEHLP_MODULE ModuleInfo); #define STACKFRAMEX STACKFRAME #define IMAGEHLP_LINEX IMAGEHLP_LINE #define IMAGEHLP_MODULEX IMAGEHLP_MODULE #endif dllfunction_t debughelpfuncs[] = { {(void*)&pSymFromAddr, "SymFromAddr"}, {(void*)&pSymSetOptions, "SymSetOptions"}, {(void*)&pSymInitialize, "SymInitialize"}, {(void*)&pStackWalkX, "StackWalk"DBGHELP_POSTFIX}, {(void*)&pSymFunctionTableAccessX, "SymFunctionTableAccess"DBGHELP_POSTFIX}, {(void*)&pSymGetModuleBaseX, "SymGetModuleBase"DBGHELP_POSTFIX}, {(void*)&pSymGetLineFromAddrX, "SymGetLineFromAddr"DBGHELP_POSTFIX}, {(void*)&pSymGetModuleInfoX, "SymGetModuleInfo"DBGHELP_POSTFIX}, {NULL, NULL} }; switch(exceptionCode) { case EXCEPTION_ACCESS_VIOLATION: case EXCEPTION_DATATYPE_MISALIGNMENT: case EXCEPTION_SINGLE_STEP: case EXCEPTION_ARRAY_BOUNDS_EXCEEDED: case EXCEPTION_FLT_DENORMAL_OPERAND: case EXCEPTION_FLT_DIVIDE_BY_ZERO: case EXCEPTION_FLT_INEXACT_RESULT: case EXCEPTION_FLT_INVALID_OPERATION: case EXCEPTION_FLT_OVERFLOW: case EXCEPTION_FLT_STACK_CHECK: case EXCEPTION_FLT_UNDERFLOW: case EXCEPTION_INT_DIVIDE_BY_ZERO: case EXCEPTION_INT_OVERFLOW: case EXCEPTION_PRIV_INSTRUCTION: case EXCEPTION_IN_PAGE_ERROR: case EXCEPTION_ILLEGAL_INSTRUCTION: case EXCEPTION_NONCONTINUABLE_EXCEPTION: case EXCEPTION_STACK_OVERFLOW: case EXCEPTION_INVALID_DISPOSITION: case EXCEPTION_GUARD_PAGE: case EXCEPTION_INVALID_HANDLE: // case EXCEPTION_POSSIBLE_DEADLOCK: break; case EXCEPTION_BREAKPOINT: hKernel = LoadLibrary ("kernel32"); pIsDebuggerPresent = (void*)GetProcAddress(hKernel, "IsDebuggerPresent"); if (pIsDebuggerPresent && pIsDebuggerPresent()) return EXCEPTION_CONTINUE_SEARCH; break; return EXCEPTION_CONTINUE_EXECUTION; default: //because windows is a steaming pile of shite, we have to ignore any software-generated exceptions, because most of them are not in fact fatal, *EVEN IF THEY CLAIM TO BE NON-CONTINUABLE* return EXCEPTION_CONTINUE_SEARCH; } #ifdef PRINTGLARRAYS if (!iswatchdog && qrenderer == QR_OPENGL && Sys_IsMainThread()) DumpGLState(); #endif hKernel = LoadLibrary ("kernel32"); pIsDebuggerPresent = (void*)GetProcAddress(hKernel, "IsDebuggerPresent"); #ifdef GLQUAKE //restores gamma GLVID_Crashed(); #endif if (!iswatchdog && pIsDebuggerPresent && pIsDebuggerPresent ()) { /*if we have a current window, minimize it to bring us out of fullscreen*/ extern qboolean vid_initializing; qboolean oldval = vid_initializing; vid_initializing = true; // ShowWindow(mainwindow, SW_MINIMIZE); vid_initializing = oldval; return EXCEPTION_CONTINUE_SEARCH; } /*if we have a current window, kill it, so it can't steal input of handle window messages or anything risky like that*/ if (iswatchdog) { } else { DestroyWindow(mainwindow); if (Sys_LoadLibrary("DBGHELP", debughelpfuncs)) { STACKFRAMEX stack; CONTEXT *pcontext = exceptionInfo->ContextRecord; IMAGEHLP_LINEX line; IMAGEHLP_MODULEX module; struct { SYMBOL_INFO sym; char name[1024]; } sym; int frameno; char stacklog[8192]; int logpos, logstart; char *logline; stacklog[logpos=0] = 0; pSymInitialize(hProc, NULL, TRUE); pSymSetOptions(SYMOPT_LOAD_LINES); memset(&stack, 0, sizeof(stack)); #ifdef _WIN64 #define IMAGE_FILE_MACHINE_THIS IMAGE_FILE_MACHINE_AMD64 stack.AddrPC.Mode = AddrModeFlat; stack.AddrPC.Offset = pcontext->Rip; stack.AddrFrame.Mode = AddrModeFlat; stack.AddrFrame.Offset = pcontext->Rbp; stack.AddrStack.Mode = AddrModeFlat; stack.AddrStack.Offset = pcontext->Rsp; #else #define IMAGE_FILE_MACHINE_THIS IMAGE_FILE_MACHINE_I386 stack.AddrPC.Mode = AddrModeFlat; stack.AddrPC.Offset = pcontext->Eip; stack.AddrFrame.Mode = AddrModeFlat; stack.AddrFrame.Offset = pcontext->Ebp; stack.AddrStack.Mode = AddrModeFlat; stack.AddrStack.Offset = pcontext->Esp; #endif Q_strncpyz(stacklog+logpos, FULLENGINENAME " or dependancy has crashed. The following stack dump been copied to your windows clipboard.\n" #ifdef _MSC_VER "Would you like to generate a core dump too?\n" #endif "\n", sizeof(stacklog)-logpos); logstart = logpos += strlen(stacklog+logpos); //so I know which one it is #if defined(DEBUG) || defined(_DEBUG) #define BUILDDEBUGREL "Debug" #else #define BUILDDEBUGREL "Optimised" #endif #ifdef MINIMAL #define BUILDMINIMAL "Min" #else #define BUILDMINIMAL "" #endif #if defined(GLQUAKE) && !defined(D3DQUAKE) #define BUILDTYPE "GL" #elif !defined(GLQUAKE) && defined(D3DQUAKE) #define BUILDTYPE "D3D" #else #define BUILDTYPE "Merged" #endif Q_snprintfz(stacklog+logpos, sizeof(stacklog)-logpos, "Build: %s %s %s: %s\r\n", BUILDDEBUGREL, PLATFORM, BUILDMINIMAL BUILDTYPE, version_string()); logpos += strlen(stacklog+logpos); for(frameno = 0; ; frameno++) { DWORD64 symdisp; DWORD linedisp; DWORD_PTR symaddr; if (!pStackWalkX(IMAGE_FILE_MACHINE_THIS, hProc, GetCurrentThread(), &stack, pcontext, NULL, pSymFunctionTableAccessX, pSymGetModuleBaseX, NULL)) break; memset(&module, 0, sizeof(module)); module.SizeOfStruct = sizeof(module); pSymGetModuleInfoX(hProc, stack.AddrPC.Offset, &module); memset(&line, 0, sizeof(line)); line.SizeOfStruct = sizeof(line); symdisp = 0; memset(&sym, 0, sizeof(sym)); sym.sym.MaxNameLen = sizeof(sym.name); symaddr = stack.AddrPC.Offset; sym.sym.SizeOfStruct = sizeof(sym.sym); if (pSymFromAddr(hProc, symaddr, &symdisp, &sym.sym)) { if (pSymGetLineFromAddrX(hProc, stack.AddrPC.Offset, &linedisp, &line)) logline = va("%-20s - %s:%i (%s)\r\n", sym.sym.Name, line.FileName, (int)line.LineNumber, module.LoadedImageName); else logline = va("%-20s+%#x (%s)\r\n", sym.sym.Name, (unsigned int)symdisp, module.LoadedImageName); } else logline = va("0x%p (%s)\r\n", (void*)(DWORD_PTR)stack.AddrPC.Offset, module.LoadedImageName); Q_strncpyz(stacklog+logpos, logline, sizeof(stacklog)-logpos); logpos += strlen(stacklog+logpos); if (logpos+1 >= sizeof(stacklog)) break; } Sys_SaveClipboard(stacklog+logstart); #ifdef _MSC_VER if (MessageBoxA(0, stacklog, "KABOOM!", MB_ICONSTOP|MB_YESNO) != IDYES) { if (pIsDebuggerPresent && pIsDebuggerPresent ()) { //its possible someone attached a debugger while we were showing that message return EXCEPTION_CONTINUE_SEARCH; } return EXCEPTION_EXECUTE_HANDLER; } #else MessageBox(0, stacklog, "KABOOM!", MB_ICONSTOP); return EXCEPTION_EXECUTE_HANDLER; #endif } else { MessageBoxA(NULL, "We crashed.\nUnable to load dbghelp library. Stack info is not available", DISTRIBUTION " Sucks", MB_ICONSTOP); return EXCEPTION_EXECUTE_HANDLER; } } //generate a minidump, but only if we were compiled by something that used usable debugging info. its a bit pointless otherwise. #ifdef _MSC_VER hDbgHelp = LoadLibraryA ("DBGHELP"); if (hDbgHelp) fnMiniDumpWriteDump = (MINIDUMPWRITEDUMP)GetProcAddress (hDbgHelp, "MiniDumpWriteDump"); else fnMiniDumpWriteDump = NULL; if (fnMiniDumpWriteDump) { if (iswatchdog) { switch (MessageBoxA(NULL, "Fizzle... We hit an infinite loop! Or something is just really slow.\nBlame the monkey in the corner.\nI hope you saved your work.\nWould you like to take a dump now?\n(click cancel to wait a bit longer)", DISTRIBUTION " Sucks", MB_ICONSTOP|MB_YESNOCANCEL|MB_DEFBUTTON3)) { case IDYES: break; //take a dump. case IDNO: exit(0); default: //cancel = run the exception handler, which means we reset the watchdog. return EXCEPTION_EXECUTE_HANDLER; } } /*take a dump*/ if (*com_homepath) Q_strncpyz(dumpPath, com_homepath, sizeof(dumpPath)); else if (*com_gamepath) Q_strncpyz(dumpPath, com_gamepath, sizeof(dumpPath)); else GetTempPathA (sizeof(dumpPath)-16, dumpPath); Q_strncatz(dumpPath, DISTRIBUTION"CrashDump.dmp", sizeof(dumpPath)); dumpfile = CreateFileA (dumpPath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (dumpfile) { MINIDUMP_EXCEPTION_INFORMATION crashinfo; crashinfo.ClientPointers = TRUE; crashinfo.ExceptionPointers = exceptionInfo; crashinfo.ThreadId = GetCurrentThreadId (); if (fnMiniDumpWriteDump(hProc, procid, dumpfile, MiniDumpWithIndirectlyReferencedMemory|MiniDumpWithDataSegs, &crashinfo, NULL, NULL)) { CloseHandle(dumpfile); Q_snprintfz(msg, sizeof(msg), "You can find the crashdump at:\n%s\nPlease send this file to someone.\n\nWarning: sensitive information (like your current user name) might be present in the dump.\nYou will probably want to compress it.", dumpPath); MessageBoxA(NULL, msg, DISTRIBUTION " Sucks", 0); } else MessageBoxA(NULL, "MiniDumpWriteDump failed", "oh noes", 0); } else { Q_snprintfz(msg, sizeof(msg), "unable to open %s\nno dump created.", dumpPath); MessageBoxA(NULL, msg, "oh noes", 0); } } else MessageBoxA(NULL, "Kaboom! Sorry. No MiniDumpWriteDump function.", DISTRIBUTION " Sucks", 0); #endif return EXCEPTION_EXECUTE_HANDLER; } //most compilers do not support __try. perhaps we should avoid its use entirely? LONG CALLBACK nonmsvc_CrashExceptionHandler(PEXCEPTION_POINTERS ExceptionInfo) { DWORD foo = EXCEPTION_CONTINUE_SEARCH; foo = CrashExceptionHandler(false, ExceptionInfo->ExceptionRecord->ExceptionCode, ExceptionInfo); //we have no handler. thus we handle it by exiting. if (foo == EXCEPTION_EXECUTE_HANDLER) exit(1); return foo; } volatile int watchdogframe; //incremented each frame. void *watchdogthread; int watchdogthreadfunction(void *arg) { #ifdef _MSC_VER int oldframe = watchdogframe; int newframe; int secs = 0; while(1) { newframe = watchdogframe; if (oldframe != newframe) { oldframe = newframe; secs = 0; } else { secs++; if (secs > 10) { secs = 0; __try { *(int*)arg = -3; } __except (CrashExceptionHandler(true, GetExceptionCode(), GetExceptionInformation())) { } } } Sleep(1000); } #endif return 0; } #endif int *debug; #ifndef SERVERONLY #if (_WIN32_WINNT < 0x0400) #define LLKHF_ALTDOWN 0x00000020 #define LLKHF_UP 0x00000080 #define WH_KEYBOARD_LL 13 typedef struct { DWORD vkCode; DWORD scanCode; DWORD flags; DWORD time; DWORD dwExtraInfo; } KBDLLHOOKSTRUCT; #elif defined(MINGW) #ifndef LLKHF_UP #define LLKHF_UP 0x00000080 #endif #endif HHOOK llkeyboardhook; cvar_t sys_disableWinKeys = SCVAR("sys_disableWinKeys", "0"); cvar_t sys_disableTaskSwitch = SCVARF("sys_disableTaskSwitch", "0", CVAR_NOTFROMSERVER); // please don't encourage people to use this... LRESULT CALLBACK LowLevelKeyboardProc (INT nCode, WPARAM wParam, LPARAM lParam) { KBDLLHOOKSTRUCT *pkbhs = (KBDLLHOOKSTRUCT *) lParam; if (ActiveApp) switch (nCode) { case HC_ACTION: { //Trap the Left Windowskey if (pkbhs->vkCode == VK_SNAPSHOT) { IN_KeyEvent (0, !(pkbhs->flags & LLKHF_UP), K_PRINTSCREEN, 0); return 1; } if (sys_disableWinKeys.ival) { if (pkbhs->vkCode == VK_LWIN) { IN_KeyEvent (0, !(pkbhs->flags & LLKHF_UP), K_LWIN, 0); return 1; } //Trap the Right Windowskey if (pkbhs->vkCode == VK_RWIN) { IN_KeyEvent(0, !(pkbhs->flags & LLKHF_UP), K_RWIN, 0); return 1; } //Trap the Application Key (what a pointless key) if (pkbhs->vkCode == VK_APPS) { IN_KeyEvent (0, !(pkbhs->flags & LLKHF_UP), K_APP, 0); return 1; } } // Disable CTRL+ESC //this works, but we've got to give some way to tab out... if (sys_disableTaskSwitch.ival) { if (pkbhs->vkCode == VK_ESCAPE && GetAsyncKeyState (VK_CONTROL) >> ((sizeof(SHORT) * 8) - 1)) return 1; // Disable ATL+TAB if (pkbhs->vkCode == VK_TAB && pkbhs->flags & LLKHF_ALTDOWN) return 1; // Disable ALT+ESC if (pkbhs->vkCode == VK_ESCAPE && pkbhs->flags & LLKHF_ALTDOWN) return 1; } break; } default: break; } return CallNextHookEx (llkeyboardhook, nCode, wParam, lParam); } void SetHookState(qboolean state) { if (!state == !llkeyboardhook) //not so types are comparable return; if (llkeyboardhook) { UnhookWindowsHookEx(llkeyboardhook); llkeyboardhook = NULL; } if (state) llkeyboardhook = SetWindowsHookEx(WH_KEYBOARD_LL, LowLevelKeyboardProc, GetModuleHandle(NULL), 0); } #endif /* =============================================================================== FILE IO =============================================================================== */ void Sys_mkdir (char *path) { if (WinNT) { wchar_t wide[MAX_OSPATH]; widen(wide, sizeof(wide), path); CreateDirectoryW(wide, NULL); } else _mkdir (path); } qboolean Sys_remove (char *path) { if (WinNT) { wchar_t wide[MAX_OSPATH]; widen(wide, sizeof(wide), path); if (DeleteFileW(wide)) return true; //success if (GetLastError() == ERROR_FILE_NOT_FOUND) return true; //succeed when the file already didn't exist return false; //other errors? panic } else { if (remove (path) != 0) { int e = errno; if (e == ENOENT) return true; //return success if it doesn't already exist. return false; } return true; } } qboolean Sys_Rename (char *oldfname, char *newfname) { if (WinNT) { wchar_t oldwide[MAX_OSPATH]; wchar_t newwide[MAX_OSPATH]; widen(oldwide, sizeof(oldwide), oldfname); widen(newwide, sizeof(newwide), newfname); return MoveFileW(oldwide, newwide); } else return !rename(oldfname, newfname); } static time_t Sys_FileTimeToTime(FILETIME ft) { ULARGE_INTEGER ull; ull.LowPart = ft.dwLowDateTime; ull.HighPart = ft.dwHighDateTime; return ull.QuadPart / 10000000ULL - 11644473600ULL; } static int Sys_EnumerateFiles2 (const char *match, int matchstart, int neststart, int (QDECL *func)(const char *fname, qofs_t fsize, time_t mtime, void *parm, searchpathfuncs_t *spath), void *parm, searchpathfuncs_t *spath) { qboolean go; if (!WinNT) { HANDLE r; WIN32_FIND_DATAA fd; int nest = neststart; //neststart refers to just after a / qboolean wild = false; while(match[nest] && match[nest] != '/') { if (match[nest] == '?' || match[nest] == '*') wild = true; nest++; } if (match[nest] == '/') { char submatch[MAX_OSPATH]; char tmproot[MAX_OSPATH]; char file[MAX_OSPATH]; if (!wild) return Sys_EnumerateFiles2(match, matchstart, nest+1, func, parm, spath); if (nest-neststart+1> MAX_OSPATH) return 1; memcpy(submatch, match+neststart, nest - neststart); submatch[nest - neststart] = 0; nest++; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); r = FindFirstFileA(tmproot, &fd); strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { if (*fd.cFileName == '.'); //don't ever find files with a name starting with '.' else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, fd.cFileName)) { int newnest; if (strlen(tmproot) + strlen(fd.cFileName) + strlen(match+nest) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot, fd.cFileName); newnest = strlen(file); strcpy(file+newnest, match+nest); go = Sys_EnumerateFiles2(file, matchstart, newnest, func, parm, spath); } } } } while(FindNextFileA(r, &fd) && go); FindClose(r); } else { const char *submatch = match + neststart; char tmproot[MAX_OSPATH]; char file[MAX_OSPATH]; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); r = FindFirstFileA(tmproot, &fd); strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { if (*fd.cFileName == '.') ; //don't ever find files with a name starting with '.' (includes .. and . directories, and unix hidden files) else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, fd.cFileName)) { if (strlen(tmproot+matchstart) + strlen(fd.cFileName) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot+matchstart, fd.cFileName); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), Sys_FileTimeToTime(fd.ftLastWriteTime), parm, spath); } } } else { if (wildcmp(submatch, fd.cFileName)) { if (strlen(tmproot+matchstart) + strlen(fd.cFileName) + 1 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s", tmproot+matchstart, fd.cFileName); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), Sys_FileTimeToTime(fd.ftLastWriteTime), parm, spath); } } } } while(FindNextFileA(r, &fd) && go); FindClose(r); } } else { HANDLE r; WIN32_FIND_DATAW fd; int nest = neststart; //neststart refers to just after a / qboolean wild = false; while(match[nest] && match[nest] != '/') { if (match[nest] == '?' || match[nest] == '*') wild = true; nest++; } if (match[nest] == '/') { char submatch[MAX_OSPATH]; char tmproot[MAX_OSPATH]; if (!wild) return Sys_EnumerateFiles2(match, matchstart, nest+1, func, parm, spath); if (nest-neststart+1> MAX_OSPATH) return 1; memcpy(submatch, match+neststart, nest - neststart); submatch[nest - neststart] = 0; nest++; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); { wchar_t wroot[MAX_OSPATH]; r = FindFirstFileW(widen(wroot, sizeof(wroot), tmproot), &fd); } strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { char utf8[MAX_OSPATH]; char file[MAX_OSPATH]; narrowen(utf8, sizeof(utf8), fd.cFileName); if (*utf8 == '.'); //don't ever find files with a name starting with '.' else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, utf8)) { int newnest; if (strlen(tmproot) + strlen(utf8) + strlen(match+nest) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot, utf8); newnest = strlen(file); strcpy(file+newnest, match+nest); go = Sys_EnumerateFiles2(file, matchstart, newnest, func, parm, spath); } } } } while(FindNextFileW(r, &fd) && go); FindClose(r); } else { const char *submatch = match + neststart; char tmproot[MAX_OSPATH]; if (neststart+4 > MAX_OSPATH) return 1; memcpy(tmproot, match, neststart); strcpy(tmproot+neststart, "*.*"); { wchar_t wroot[MAX_OSPATH]; r = FindFirstFileW(widen(wroot, sizeof(wroot), tmproot), &fd); } strcpy(tmproot+neststart, ""); if (r==(HANDLE)-1) return 1; go = true; do { char utf8[MAX_OSPATH]; char file[MAX_OSPATH]; narrowen(utf8, sizeof(utf8), fd.cFileName); if (*utf8 == '.') ; //don't ever find files with a name starting with '.' (includes .. and . directories, and unix hidden files) else if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) //is a directory { if (wildcmp(submatch, utf8)) { if (strlen(tmproot+matchstart) + strlen(utf8) + 2 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s/", tmproot+matchstart, utf8); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), Sys_FileTimeToTime(fd.ftLastWriteTime), parm, spath); } } } else { if (wildcmp(submatch, utf8)) { if (strlen(tmproot+matchstart) + strlen(utf8) + 1 < MAX_OSPATH) { Q_snprintfz(file, sizeof(file), "%s%s", tmproot+matchstart, utf8); go = func(file, qofs_Make(fd.nFileSizeLow, fd.nFileSizeHigh), Sys_FileTimeToTime(fd.ftLastWriteTime), parm, spath); } } } } while(FindNextFileW(r, &fd) && go); FindClose(r); } } return go; } int Sys_EnumerateFiles (const char *gpath, const char *match, int (QDECL *func)(const char *fname, qofs_t fsize, time_t mtime, void *parm, searchpathfuncs_t *spath), void *parm, searchpathfuncs_t *spath) { char fullmatch[MAX_OSPATH]; int start; if (strlen(gpath) + strlen(match) + 2 > MAX_OSPATH) return 1; strcpy(fullmatch, gpath); start = strlen(fullmatch); if (start && fullmatch[start-1] != '/') fullmatch[start++] = '/'; fullmatch[start] = 0; strcat(fullmatch, match); return Sys_EnumerateFiles2(fullmatch, start, start, func, parm, spath); } //wide only. we let the windows api sort out the mess of file urls. system-wide consistancy. qboolean Sys_ResolveFileURL(const char *inurl, int inlen, char *out, int outlen) { char *cp; wchar_t wurl[MAX_PATH]; wchar_t local[MAX_PATH]; DWORD grr; static HRESULT (WINAPI *pPathCreateFromUrlW)(PCWSTR pszUrl, PWSTR pszPath, DWORD *pcchPath, DWORD dwFlags); if (!pPathCreateFromUrlW) pPathCreateFromUrlW = Sys_GetAddressForName(Sys_LoadLibrary("Shlwapi.dll", NULL), "PathCreateFromUrlW"); if (!pPathCreateFromUrlW) return false; //need to make a copy, because we can't terminate the inurl easily. cp = malloc(inlen+1); memcpy(cp, inurl, inlen); cp[inlen] = 0; widen(wurl, sizeof(wurl), cp); free(cp); grr = sizeof(local)/sizeof(wchar_t); if (FAILED(pPathCreateFromUrlW(wurl, local, &grr, 0))) return false; narrowen(out, outlen, local); return true; } /* =============================================================================== SYSTEM IO =============================================================================== */ /* ================ Sys_MakeCodeWriteable ================ */ #if 0 void Sys_MakeCodeWriteable (void *startaddr, unsigned long length) { DWORD flOldProtect; //@@@ copy on write or just read-write? if (!VirtualProtect(startaddr, length, PAGE_EXECUTE_READWRITE, &flOldProtect)) { char str[1024]; FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), str, sizeof(str), NULL); Sys_Error("Protection change failed!\nError %d: %s\n", (int)GetLastError(), str); } } #endif void Sys_DoFileAssociations(qboolean elevated); void Sys_Register_File_Associations_f(void) { Sys_DoFileAssociations(0); } /* ================ Sys_Init ================ */ void Sys_Init (void) { // LARGE_INTEGER PerformanceFreq; // unsigned int lowpart, highpart; OSVERSIONINFO vinfo; Sys_QueryDesktopParameters(); #ifndef SERVERONLY Cvar_Register(&sys_disableWinKeys, "System vars"); Cvar_Register(&sys_disableTaskSwitch, "System vars"); Cmd_AddCommandD("sys_register_file_associations", Sys_Register_File_Associations_f, "Register FTE as the system handler for .bsp .mvd .qwd .dem files. Also register the qw:// URL protocol. This command will probably trigger a UAC prompt in Windows Vista and up. Deny it for current-user-only asociations (will also prevent listing in windows' 'default programs' ui due to microsoft bugs/limitations)."); #ifdef QUAKESPYAPI #ifndef CLIENTONLY if (!isDedicated && !COM_CheckParm("-nomutex")) #else if (!COM_CheckParm("-nomutex")) //we need to create a mutex to allow gamespy to realise that we're running, but it might not be desired as it prevents other clients from running too. #endif { // allocate a named semaphore on the client so the // front end can tell if it is alive // mutex will fail if semephore already exists qwclsemaphore = CreateMutex( NULL, // Security attributes 0, // owner "qwcl"); // Semaphore name // if (!qwclsemaphore) // Sys_Error ("QWCL is already running on this system"); CloseHandle (qwclsemaphore); qwclsemaphore = CreateSemaphore( NULL, // Security attributes 0, // Initial count 1, // Maximum count "qwcl"); // Semaphore name } #endif #endif #if 0 if (!QueryPerformanceFrequency (&PerformanceFreq)) Sys_Error ("No hardware timer available"); // get 32 out of the 64 time bits such that we have around // 1 microsecond resolution lowpart = (unsigned int)PerformanceFreq.LowPart; highpart = (unsigned int)PerformanceFreq.HighPart; lowshift = 0; while (highpart || (lowpart > 2000000.0)) { lowshift++; lowpart >>= 1; lowpart |= (highpart & 1) << 31; highpart >>= 1; } pfreq = 1.0 / (double)lowpart; Sys_InitFloatTime (); #endif // make sure the timer is high precision, otherwise // NT gets 18ms resolution timeBeginPeriod( 1 ); vinfo.dwOSVersionInfoSize = sizeof(vinfo); if (!GetVersionEx (&vinfo)) Sys_Error ("Couldn't get OS info"); if ((vinfo.dwMajorVersion < 4) || (vinfo.dwPlatformId == VER_PLATFORM_WIN32s)) { Sys_Error (FULLENGINENAME " requires at least Win95 or NT 4.0"); } if (vinfo.dwPlatformId == VER_PLATFORM_WIN32_NT) WinNT = true; else WinNT = false; qwinvermaj = vinfo.dwMajorVersion; qwinvermin = vinfo.dwMinorVersion; } void Sys_Shutdown(void) { int i; if (tevent) CloseHandle (tevent); tevent = NULL; if (qwclsemaphore) CloseHandle (qwclsemaphore); qwclsemaphore = NULL; for (i = 0; i < MAX_NUM_ARGVS; i++) { if (!sys_argv[i]) break; free(sys_argv[i]); sys_argv[i] = NULL; } } void VARGS Sys_Error (const char *error, ...) { va_list argptr; char text[1024]; // DWORD dummy; va_start (argptr, error); vsnprintf (text, sizeof(text), error, argptr); va_end (argptr); COM_WorkerAbort(text); #ifndef SERVERONLY SetHookState(false); Host_Shutdown (); #else SV_Shutdown(); #endif { wchar_t wtext[1024]; widen(wtext, sizeof(wtext), text); MessageBoxW(NULL, wtext, L"Error", 0); } #ifndef SERVERONLY CloseHandle (qwclsemaphore); SetHookState(false); #endif TL_Shutdown(); #ifdef USE_MSVCRT_DEBUG if (_CrtDumpMemoryLeaks()) OutputDebugStringA("Leaks detected\n"); #endif exit (1); } void VARGS Sys_Printf (char *fmt, ...) { va_list argptr; char text[1024]; DWORD dummy; conchar_t msg[1024], *end, *in; wchar_t wide[1024], *out; int wlen; if (!houtput && !debugout && !SSV_IsSubServer()) return; va_start (argptr,fmt); vsnprintf (text, sizeof(text), fmt, argptr); va_end (argptr); #ifdef SUBSERVERS if (SSV_IsSubServer()) { SSV_PrintToMaster(text); return; } #endif end = COM_ParseFunString(CON_WHITEMASK, text, msg, sizeof(msg), false); out = wide; in = msg; wlen = 0; for (in = msg; in < end && wlen+3 < countof(wide); ) { unsigned int flags, cp; in = Font_Decode(in, &flags, &cp); if (!(flags & CON_HIDDEN)) { cp = COM_DeQuake(cp); if (cp > 0xffff) { cp -= 0x10000; *out++ = 0xD800 | (cp>>10); *out++ = 0xDC00 | (cp&0x3ff); wlen += 2; } else { *out++ = cp; wlen++; } } } *out = 0; if (debugout) OutputDebugStringW(wide); if (houtput) { if (WinNT) WriteConsoleW(houtput, wide, wlen, &dummy, NULL); else { //win95 doesn't support wide chars *sigh*. blank consoles suck. this conversion might loose stuff if the multibytes are too long. WriteConsole(houtput, text, WideCharToMultiByte(CP_ACP, 0, wide, wlen, text, sizeof(text), NULL, NULL), &dummy, NULL); } } } void Sys_Quit (void) { #ifndef SERVERONLY SetHookState(false); Host_Shutdown (); SetHookState(false); #else SV_Shutdown(); #endif TL_Shutdown(); #ifdef RESTARTTEST longjmp(restart_jmpbuf, 1); #endif #ifdef USE_MSVCRT_DEBUG if (_CrtDumpMemoryLeaks()) OutputDebugStringA("Leaks detected\n"); #endif exit(1); } #if 0 /* ================ Sys_DoubleTime ================ */ double Sys_DoubleTime (void) { static int first = 1; static LARGE_INTEGER qpcfreq; LARGE_INTEGER PerformanceCount; static LONGLONG oldcall; static LONGLONG firsttime; LONGLONG diff; QueryPerformanceCounter (&PerformanceCount); if (first) { first = 0; QueryPerformanceFrequency(&qpcfreq); firsttime = PerformanceCount.QuadPart; diff = 0; } else diff = PerformanceCount.QuadPart - oldcall; if (diff >= 0) oldcall = PerformanceCount.QuadPart; return (oldcall - firsttime) / (double)qpcfreq.QuadPart; } unsigned int Sys_Milliseconds (void) { return Sys_DoubleTime()*1000; } #else unsigned int Sys_Milliseconds (void) { static DWORD starttime; static qboolean first = true; DWORD now; // double t; now = timeGetTime(); if (first) { first = false; starttime = now; return 0.0; } /* if (now < starttime) // wrapped? { double r; r = (now) + (LONG_MAX - starttime); starttime = now; return r; } if (now - starttime == 0) return 0.0; */ return (now - starttime); } double Sys_DoubleTime (void) { return Sys_Milliseconds()/1000.f; } #endif ///////////////////////////////////////////////////////////// //clipboard HANDLE clipboardhandle; char *cliputf8; char *Sys_GetClipboard(void) { if (OpenClipboard(NULL)) { //windows programs interpret CF_TEXT as ansi (aka: gibberish) //so grab utf-16 text and convert it to utf-8 if our console parsing is set to accept that. clipboardhandle = GetClipboardData(CF_UNICODETEXT); if (clipboardhandle) { unsigned short *clipWText = GlobalLock(clipboardhandle); if (clipWText) { unsigned int l, c; char *utf8; for (l = 0; clipWText[l]; l++) ; l = l*4 + 1; utf8 = cliputf8 = malloc(l); while(*clipWText) { unsigned int cp = *clipWText++; if (cp == '\r' && *clipWText == '\n') //bloomin microsoft. cp = *clipWText++; if (cp >= 0xD800u && cp <= 0xDBFFu) { //handle utf-16 surrogates if (*clipWText >= 0xDC00u && *clipWText <= 0xDFFFu) { cp = (cp&0x3ff)<<10; cp |= *clipWText++ & 0x3ff; } else cp = 0xFFFDu; } c = utf8_encode(utf8, cp, l); if (!c) break; l -= c; utf8 += c; } *utf8 = 0; return cliputf8; } //failed at the last hurdle GlobalUnlock(clipboardhandle); } clipboardhandle = GetClipboardData(CF_TEXT); if (clipboardhandle) { char *clipText = GlobalLock(clipboardhandle); if (clipText) { unsigned int l, c; char *utf8; for (l = 0; clipText[l]; l++) ; l = l*4 + 1; utf8 = cliputf8 = malloc(l); while(*clipText) { if (clipText[0] == '\r' && clipText[1] == '\n') //bloomin microsoft. clipText++; c = utf8_encode(utf8, *clipText++, l); if (!c) break; l -= c; utf8 += c; } *utf8 = 0; return cliputf8; } //failed at the last hurdle GlobalUnlock(clipboardhandle); } CloseClipboard(); } clipboardhandle = NULL; return NULL; } void Sys_CloseClipboard(char *bf) { if (clipboardhandle) { free(cliputf8); cliputf8 = NULL; GlobalUnlock(clipboardhandle); CloseClipboard(); clipboardhandle = NULL; } } void Sys_SaveClipboard(char *text) { HANDLE glob; char *temp; unsigned short *tempw; unsigned int codepoint; if (!OpenClipboard(NULL)) return; EmptyClipboard(); if (WinNT) { glob = GlobalAlloc(GMEM_MOVEABLE, (strlen(text) + 1)*4); if (glob) { tempw = GlobalLock(glob); if (tempw != NULL) { int error; while(*text) { codepoint = unicode_decode(&error, text, &text, false); if (codepoint == '\n') { //windows is stupid and annoying. *tempw++ = '\r'; *tempw++ = '\n'; } else if (codepoint > 0xffff) { //and badly designed, too. codepoint -= 0x10000; *tempw++ = 0xD800 | ((codepoint>>10)&0x3ff); *tempw++ = 0xDC00 | (codepoint&0x3ff); } else *tempw++ = codepoint; } *tempw = 0; GlobalUnlock(glob); SetClipboardData(CF_UNICODETEXT, glob); } else GlobalFree(glob); } } else { glob = GlobalAlloc(GMEM_MOVEABLE, strlen(text)*2 + 1); if (glob) { //yes, quake chars will get mangled horribly. temp = GlobalLock(glob); if (temp != NULL) { int error; while (*text) { codepoint = unicode_decode(&error, text, &text, false); if (codepoint == '\n') *temp++ = '\r'; *temp++ = codepoint; } *temp = 0; strcpy(temp, text); GlobalUnlock(glob); SetClipboardData(CF_TEXT, glob); } else GlobalFree(glob); } } CloseClipboard(); } //end of clipboard ///////////////////////////////////////////////////////////// ///////////////////////////////////////// //the system console stuff char *Sys_ConsoleInput (void) { static char text[256]; static int len; INPUT_RECORD recs[1024]; // int count; int ch; DWORD numevents, numread, dummy=0; HANDLE th; char *clipText, *textCopied; #ifdef SUBSERVERS if (SSV_IsSubServer()) { DWORD avail; static char text[1024], *nl; static int textpos = 0; HANDLE input = GetStdHandle(STD_INPUT_HANDLE); for (;;) { if (!PeekNamedPipe(input, NULL, 0, NULL, &avail, NULL)) { SV_FinalMessage("Cluster shut down\n"); Cmd_ExecuteString("quit force", RESTRICT_LOCAL); } else if (avail) { if (avail > sizeof(text)-1-textpos) avail = sizeof(text)-1-textpos; if (ReadFile(input, text+textpos, avail, &avail, NULL)) { textpos += avail; while(textpos >= 2) { unsigned short len = text[0] | (text[1]<<8); if (textpos >= len && len >= 2) { memcpy(net_message.data, text+2, len-2); net_message.cursize = len-2; MSG_BeginReading (msg_nullnetprim); SSV_ReadFromControlServer(); memmove(text, text+len, textpos - len); textpos -= len; } else break; } } } else break; } return NULL; } #endif if (!hinput) return NULL; for ( ;; ) { if (!GetNumberOfConsoleInputEvents (hinput, &numevents)) Sys_Error ("Error getting # of console events"); if (numevents <= 0) break; if (WinNT) { if (!ReadConsoleInputW(hinput, recs, 1, &numread)) Sys_Error ("Error reading console input"); } else { if (!ReadConsoleInputA(hinput, recs, 1, &numread)) Sys_Error ("Error reading console input"); } if (numread != 1) Sys_Error ("Couldn't read console input"); if (recs[0].EventType == KEY_EVENT) { if (recs[0].Event.KeyEvent.bKeyDown) { ch = recs[0].Event.KeyEvent.uChar.UnicodeChar; switch (ch) { case '\r': WriteFile(houtput, "\r\n", 2, &dummy, NULL); if (len) { text[len] = 0; len = 0; return text; } break; case '\b': if (len) { len--; WriteFile(houtput, "\b \b", 3, &dummy, NULL); } break; default: if (((ch=='V' || ch=='v') && (recs[0].Event.KeyEvent.dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))) || ((recs[0].Event.KeyEvent.dwControlKeyState & SHIFT_PRESSED) && (recs[0].Event.KeyEvent.wVirtualKeyCode ==VK_INSERT))) { if (OpenClipboard(NULL)) { th = GetClipboardData(CF_TEXT); if (th) { clipText = GlobalLock(th); if (clipText) { int i; textCopied = BZ_Malloc(GlobalSize(th)+1); strcpy(textCopied, clipText); /* Substitutes a NULL for every token */strtok(textCopied, "\n\r\b"); i = strlen(textCopied); if (i+len>=256) i=256-len; if (i>0) { textCopied[i]=0; text[len]=0; strcat(text, textCopied); len+=dummy; WriteFile(houtput, textCopied, i, &dummy, NULL); } BZ_Free(textCopied); } GlobalUnlock(th); } CloseClipboard(); } } else if (ch >= ' ') { wchar_t wch = ch; if (WinNT) WriteConsoleW(houtput, &wch, 1, &dummy, NULL); else { char mb[8]; //hopefully ucs-2-only will be sufficient... WriteConsoleA(houtput, mb, WideCharToMultiByte(CP_ACP, 0, &wch, 1, mb, sizeof(mb), NULL, NULL), &dummy, NULL); } len += utf8_encode(text+len, ch, sizeof(text)-1-len); } break; } } } } return NULL; } BOOL WINAPI HandlerRoutine (DWORD dwCtrlType) { switch (dwCtrlType) { case CTRL_C_EVENT: case CTRL_BREAK_EVENT: case CTRL_CLOSE_EVENT: case CTRL_LOGOFF_EVENT: case CTRL_SHUTDOWN_EVENT: Cbuf_AddText ("quit\n", RESTRICT_LOCAL); return true; } return false; } #ifndef CP_UTF8 #define CP_UTF8 65001 #endif qboolean Sys_InitTerminal (void) { DWORD m; if (SSV_IsSubServer()) return true; //just pretend we did if (!AllocConsole()) return false; #ifndef SERVERONLY if (qwclsemaphore) { CloseHandle(qwclsemaphore); qwclsemaphore = NULL; } #endif SetConsoleCtrlHandler (HandlerRoutine, TRUE); SetConsoleCP(CP_UTF8); SetConsoleOutputCP(CP_UTF8); SetConsoleTitleW (_L(FULLENGINENAME) L" dedicated server"); if (isPlugin) { hinput = CreateFileA("CONIN$",GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ,0,OPEN_EXISTING,0,0); houtput = CreateFileA("CONOUT$",GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ,0,OPEN_EXISTING,0,0); } else { hinput = GetStdHandle (STD_INPUT_HANDLE); houtput = GetStdHandle (STD_OUTPUT_HANDLE); } GetConsoleMode(hinput, &m); SetConsoleMode(hinput, m | 0x40 | 0x80); return true; } void Sys_CloseTerminal (void) { FreeConsole(); hinput = NULL; houtput = NULL; } // //////////////////////////// qboolean QCExternalDebuggerCommand(char *text); void Sys_SendKeyEvents (void) { MSG msg; #ifdef _MSC_VER #define strtoull _strtoui64 #endif if (isPlugin) { DWORD avail; static char text[256], *nl; static int textpos = 0; HANDLE input = GetStdHandle(STD_INPUT_HANDLE); if (!PeekNamedPipe(input, NULL, 0, NULL, &avail, NULL)) { wantquit = true; Cmd_ExecuteString("quit force", RESTRICT_LOCAL); } else if (avail) { if (avail > sizeof(text)-1-textpos) avail = sizeof(text)-1-textpos; if (ReadFile(input, text+textpos, avail, &avail, NULL)) { textpos += avail; if (textpos > sizeof(text)-1) Sys_Error("No."); } } while (textpos) { text[textpos] = 0; nl = strchr(text, '\n'); if (nl) { *nl++ = 0; if (qrenderer <= QR_NONE && !strncmp(text, "vid_recenter ", 13)) { Cmd_TokenizeString(text, false, false); sys_parentleft = strtoul(Cmd_Argv(1), NULL, 0); sys_parenttop = strtoul(Cmd_Argv(2), NULL, 0); sys_parentwidth = strtoul(Cmd_Argv(3), NULL, 0); sys_parentheight = strtoul(Cmd_Argv(4), NULL, 0); sys_parentwindow = (HWND)(intptr_t)strtoull(Cmd_Argv(5), NULL, 16); } #if !defined(CLIENTONLY) || defined(CSQC_DAT) || defined(MENU_DAT) else if (QCExternalDebuggerCommand(text)) /*handled elsewhere*/; #endif else { Cbuf_AddText(text, RESTRICT_LOCAL); Cbuf_AddText("\n", RESTRICT_LOCAL); } memmove(text, nl, textpos - (nl - text)); textpos -= (nl - text); } else break; } } if (isDedicated) { #ifndef CLIENTONLY SV_GetConsoleCommands (); #endif return; } if (gammapending) { gammapending -= host_frametime; if (gammapending < host_frametime) { gammapending = 0; Cvar_ForceCallback(&v_gamma); } } while (PeekMessage (&msg, NULL, 0, 0, PM_REMOVE)) { // we always update if there are any event, even if we're paused //if (!GetMessage (&msg, NULL, 0, 0)) // break; // Sys_Quit (); // if (TranslateMessage (&msg)) // continue; DispatchMessage (&msg); } } void Sys_ServerActivity(void) { #ifndef SERVERONLY if (GetActiveWindow() != mainwindow) FlashWindow(mainwindow, true); #endif } /* ============================================================================== WINDOWS CRAP ============================================================================== */ /* ================== WinMain ================== */ void SleepUntilInput (int time) { MsgWaitForMultipleObjects(1, &tevent, FALSE, time, QS_ALLINPUT); } qboolean Sys_Startup_CheckMem(quakeparms_t *parms) { return true; } /* ================== WinMain ================== */ HINSTANCE global_hInstance; int global_nCmdShow; HWND hwnd_dialog; #include typedef struct qSHARDAPPIDINFOLINK { IShellLinkW *psl; PCWSTR pszAppID; } qSHARDAPPIDINFOLINK; #define qSHARD_APPIDINFOLINK 0x00000007 typedef struct { GUID fmtid; DWORD pid; } qPROPERTYKEY; typedef struct qIPropertyStore qIPropertyStore; struct qIPropertyStore { CONST_VTBL struct { /*IUnknown*/ HRESULT ( STDMETHODCALLTYPE *QueryInterface )( qIPropertyStore * This, REFIID riid, void **ppvObject); ULONG ( STDMETHODCALLTYPE *AddRef )( qIPropertyStore * This); ULONG ( STDMETHODCALLTYPE *Release )( qIPropertyStore * This); /*property store stuff*/ HRESULT ( STDMETHODCALLTYPE *GetCount)( qIPropertyStore * This, ULONG *count); HRESULT ( STDMETHODCALLTYPE *GetAt)( qIPropertyStore * This, DWORD prop, qPROPERTYKEY * key); HRESULT ( STDMETHODCALLTYPE *GetValue)( qIPropertyStore * This, qPROPERTYKEY * key, PROPVARIANT * val); HRESULT ( STDMETHODCALLTYPE *SetValue)( qIPropertyStore * This, qPROPERTYKEY * key, PROPVARIANT * val); HRESULT ( STDMETHODCALLTYPE *Commit)( qIPropertyStore * This); } *lpVtbl; }; static const IID qIID_IPropertyStore = {0x886d8eeb, 0x8cf2, 0x4446, {0x8d, 0x02, 0xcd, 0xba, 0x1d, 0xbd, 0xcf, 0x99}}; #define qIObjectArray IUnknown static const IID qIID_IObjectArray = {0x92ca9dcd, 0x5622, 0x4bba, {0xa8,0x05,0x5e,0x9f,0x54,0x1b,0xd8,0xc9}}; typedef struct qIObjectCollection { struct qIObjectCollectionVtbl { HRESULT ( __stdcall *QueryInterface )( /* [in] IShellLink*/ void *This, /* [in] */ const GUID * const riid, /* [out] */ void **ppvObject); ULONG ( __stdcall *AddRef )( /* [in] IShellLink*/ void *This); ULONG ( __stdcall *Release )( /* [in] IShellLink*/ void *This); HRESULT ( __stdcall *GetCount )( /* [in] IShellLink*/ void *This, /* [out] */ UINT *pcObjects); HRESULT ( __stdcall *GetAt )( /* [in] IShellLink*/ void *This, /* [in] */ UINT uiIndex, /* [in] */ const GUID * const riid, /* [iid_is][out] */ void **ppv); HRESULT ( __stdcall *AddObject )( /* [in] IShellLink*/ void *This, /* [in] */ void *punk); HRESULT ( __stdcall *AddFromArray )( /* [in] IShellLink*/ void *This, /* [in] */ qIObjectArray *poaSource); HRESULT ( __stdcall *RemoveObjectAt )( /* [in] IShellLink*/ void *This, /* [in] */ UINT uiIndex); HRESULT ( __stdcall *Clear )( /* [in] IShellLink*/ void *This); } *lpVtbl; } qIObjectCollection; static const IID qIID_IObjectCollection = {0x5632b1a4, 0xe38a, 0x400a, {0x92,0x8a,0xd4,0xcd,0x63,0x23,0x02,0x95}}; static const CLSID qCLSID_EnumerableObjectCollection = {0x2d3468c1, 0x36a7, 0x43b6, {0xac,0x24,0xd3,0xf0,0x2f,0xd9,0x60,0x7a}}; typedef struct qICustomDestinationList { struct qICustomDestinationListVtbl { HRESULT ( __stdcall *QueryInterface ) ( /* [in] ICustomDestinationList*/ void *This, /* [in] */ const GUID * const riid, /* [out] */ void **ppvObject); ULONG ( __stdcall *AddRef )( /* [in] ICustomDestinationList*/ void *This); ULONG ( __stdcall *Release )( /* [in] ICustomDestinationList*/ void *This); HRESULT ( __stdcall *SetAppID )( /* [in] ICustomDestinationList*/ void *This, /* [string][in] */ LPCWSTR pszAppID); HRESULT ( __stdcall *BeginList )( /* [in] ICustomDestinationList*/ void *This, /* [out] */ UINT *pcMinSlots, /* [in] */ const GUID * const riid, /* [out] */ void **ppv); HRESULT ( __stdcall *AppendCategory )( /* [in] ICustomDestinationList*/ void *This, /* [string][in] */ LPCWSTR pszCategory, /* [in] IObjectArray*/ void *poa); HRESULT ( __stdcall *AppendKnownCategory )( /* [in] ICustomDestinationList*/ void *This, /* [in] KNOWNDESTCATEGORY*/ int category); HRESULT ( __stdcall *AddUserTasks )( /* [in] ICustomDestinationList*/ void *This, /* [in] IObjectArray*/ void *poa); HRESULT ( __stdcall *CommitList )( /* [in] ICustomDestinationList*/ void *This); HRESULT ( __stdcall *GetRemovedDestinations )( /* [in] ICustomDestinationList*/ void *This, /* [in] */ const IID * const riid, /* [out] */ void **ppv); HRESULT ( __stdcall *DeleteList )( /* [in] ICustomDestinationList*/ void *This, /* [string][unique][in] */ LPCWSTR pszAppID); HRESULT ( __stdcall *AbortList )( /* [in] ICustomDestinationList*/ void *This); } *lpVtbl; } qICustomDestinationList; static const IID qIID_ICustomDestinationList = {0x6332debf, 0x87b5, 0x4670, {0x90,0xc0,0x5e,0x57,0xb4,0x08,0xa4,0x9e}}; static const CLSID qCLSID_DestinationList = {0x77f10cf0, 0x3db5, 0x4966, {0xb5,0x20,0xb7,0xc5,0x4f,0xd3,0x5e,0xd6}}; static const IID qIID_IShellLinkW = {0x000214F9L, 0, 0, {0xc0,0,0,0,0,0,0,0x46}}; static const IID qIID_IPersistFile = {0x0000010BL, 0, 0, {0xc0,0,0,0,0,0,0,0x46}}; #define WIN7_APPNAME L"FTEQuake" static IShellLinkW *CreateShellLink(char *command, char *target, char *title, char *desc) { HRESULT hr; IShellLinkW *link; qIPropertyStore *prop_store; WCHAR buf[1024]; char tmp[1024], *s; // Get a pointer to the IShellLink interface. hr = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &qIID_IShellLinkW, (void**)&link); if (FAILED(hr)) return NULL; GetModuleFileNameW(NULL, buf, countof(buf)-1); IShellLinkW_SetIconLocation(link, buf, 0); /*grab the first icon from our exe*/ IShellLinkW_SetPath(link, buf); /*program to run*/ Q_strncpyz(tmp, com_gamepath, sizeof(tmp)); /*normalize the gamedir, so we don't end up with the same thing multiple times*/ for(s = tmp; *s; s++) { if (*s == '\\') *s = '/'; else *s = tolower(*s); } IShellLinkW_SetArguments(link, widen(buf, sizeof(buf), va("%s \"%s\" -basedir \"%s\"", command, target, tmp))); /*args*/ IShellLinkW_SetDescription(link, widen(buf, sizeof(buf), desc)); /*tooltip*/ hr = IShellLinkW_QueryInterface(link, &qIID_IPropertyStore, (void**)&prop_store); if(SUCCEEDED(hr)) { PROPVARIANT pv; qPROPERTYKEY PKEY_Title; pv.vt=VT_LPSTR; pv.pszVal=title; /*item text*/ CLSIDFromString(L"{F29F85E0-4FF9-1068-AB91-08002B27B3D9}", &(PKEY_Title.fmtid)); PKEY_Title.pid=2; hr = prop_store->lpVtbl->SetValue(prop_store, &PKEY_Title, &pv); hr = prop_store->lpVtbl->Commit(prop_store); prop_store->lpVtbl->Release(prop_store); } return link; } void Sys_RecentServer(char *command, char *target, char *title, char *desc) { qSHARDAPPIDINFOLINK appinfo; IShellLinkW *link; link = CreateShellLink(command, target, title, desc); if (!link) return; appinfo.pszAppID=WIN7_APPNAME; appinfo.psl=link; SHAddToRecentDocs(qSHARD_APPIDINFOLINK, &appinfo); IShellLinkW_Release(link); } typedef struct { LPCWSTR pcszFile; LPCWSTR pcszClass; int oaifInFlags; } qOPENASINFO; HRESULT (WINAPI *pSHOpenWithDialog)(HWND hwndParent, const qOPENASINFO *poainfo); LPITEMIDLIST (STDAPICALLTYPE *pSHBrowseForFolderW)(LPBROWSEINFOW lpbi); BOOL (STDAPICALLTYPE *pSHGetPathFromIDListW)(LPCITEMIDLIST pidl, LPWSTR pszPath); BOOL (STDAPICALLTYPE *pSHGetSpecialFolderPathW)(HWND hwnd, LPWSTR pszPath, int csidl, BOOL fCreate); BOOL (STDAPICALLTYPE *pShell_NotifyIconW)(DWORD dwMessage, PNOTIFYICONDATAW lpData); void Win7_Init(void) { HANDLE h; HRESULT (WINAPI *pSetCurrentProcessExplicitAppUserModelID)(PCWSTR AppID); h = LoadLibraryU("shell32.dll"); if (h) { pSHBrowseForFolderW = (void*)GetProcAddress(h, "SHBrowseForFolderW"); pSHGetPathFromIDListW = (void*)GetProcAddress(h, "SHGetPathFromIDListW"); pSHGetSpecialFolderPathW = (void*)GetProcAddress(h, "SHGetSpecialFolderPathW"); pShell_NotifyIconW = (void*)GetProcAddress(h, "Shell_NotifyIconW"); pSHOpenWithDialog = (void*)GetProcAddress(h, "SHOpenWithDialog"); pSetCurrentProcessExplicitAppUserModelID = (void*)GetProcAddress(h, "SetCurrentProcessExplicitAppUserModelID"); if (pSetCurrentProcessExplicitAppUserModelID) pSetCurrentProcessExplicitAppUserModelID(WIN7_APPNAME); } } void Win7_TaskListInit(void) { qICustomDestinationList *cdl; qIObjectCollection *col; qIObjectArray *arr; IShellLinkW *link; CoInitialize(NULL); if (SUCCEEDED(CoCreateInstance(&qCLSID_DestinationList, NULL, CLSCTX_INPROC_SERVER, &qIID_ICustomDestinationList, (void**)&cdl))) { UINT minslots; IUnknown *removed; cdl->lpVtbl->BeginList(cdl, &minslots, &qIID_IObjectArray, (void**)&removed); if (SUCCEEDED(CoCreateInstance(&qCLSID_EnumerableObjectCollection, NULL, CLSCTX_INPROC_SERVER, &qIID_IObjectCollection, (void**)&col))) { switch(M_GameType()) { case MGT_QUAKE1: link = CreateShellLink("+menu_servers", "", "Server List", "Pick a multiplayer server to join"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } link = CreateShellLink("+map start", "", "Start New Game (Quake)", "Begin a new single-player game"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } break; case MGT_QUAKE2: link = CreateShellLink("+menu_servers", "", "Quake2 Server List", "Pick a multiplayer server to join"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } link = CreateShellLink("+map unit1", "", "Start New Game (Quake2)", "Begin a new game"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } break; #ifdef HEXEN2 case MGT_HEXEN2: link = CreateShellLink("+menu_servers", "", "Hexen2 Server List", "Pick a multiplayer server to join"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } link = CreateShellLink("+map demo1", "", "Start New Game (Hexen2)", "Begin a new game"); if (link) { col->lpVtbl->AddObject(col, (IUnknown*)link); link->lpVtbl->Release(link); } break; #endif } if (SUCCEEDED(col->lpVtbl->QueryInterface(col, &qIID_IObjectArray, (void**)&arr))) { cdl->lpVtbl->AddUserTasks(cdl, arr); arr->lpVtbl->Release(arr); } col->lpVtbl->Release(col); } cdl->lpVtbl->AppendKnownCategory(cdl, 1); cdl->lpVtbl->CommitList(cdl); cdl->lpVtbl->Release(cdl); } } BOOL CopyFileU(const char *src, const char *dst, BOOL bFailIfExists) { wchar_t wide1[2048]; wchar_t wide2[2048]; return CopyFileW(widen(wide1, sizeof(wide1), src), widen(wide2, sizeof(wide2), dst), bFailIfExists); } //#define SVNREVISION 1 #if defined(SVNREVISION) && !defined(MINIMAL) && !defined(NOLEGACY) #define SVNREVISIONSTR STRINGIFY(SVNREVISION) #if defined(OFFICIAL_RELEASE) #define UPD_BUILDTYPE "rel" #else #define UPD_BUILDTYPE "test" //WARNING: Security comes from the fact that the triptohell.info certificate is hardcoded in the tls code. //this will correctly detect insecure tls proxies also. #define UPDATE_URL_ROOT "https://triptohell.info/moodles/" #define UPDATE_URL_TESTED UPDATE_URL_ROOT "autoup/" #define UPDATE_URL_NIGHTLY UPDATE_URL_ROOT #define UPDATE_URL_VERSION "%sversion.txt" #ifdef NOLEGACY #ifdef _WIN64 #define UPDATE_URL_BUILD "%snocompat64/fte" EXETYPE "64.exe" #else #define UPDATE_URL_BUILD "%snocompat/fte" EXETYPE ".exe" #endif #else #ifdef _WIN64 #define UPDATE_URL_BUILD "%swin64/fte" EXETYPE "64.exe" #else #define UPDATE_URL_BUILD "%swin32/fte" EXETYPE ".exe" #endif #endif #endif #endif #if defined(SERVERONLY) #define EXETYPE "qwsv" //not gonna happen, but whatever. #elif defined(GLQUAKE) && defined(D3DQUAKE) #define EXETYPE "qw" #elif defined(GLQUAKE) #ifdef MINIMAL #define EXETYPE "minglqw" #else #define EXETYPE "glqw" #endif #elif defined(D3DQUAKE) #define EXETYPE "d3dqw" #elif defined(SWQUAKE) #define EXETYPE "swqw" #else //erm... #define EXETYPE "qw" #endif #ifdef UPDATE_URL_ROOT int sys_autoupdatesetting; qboolean Update_GetHomeDirectory(char *homedir, int homedirsize) { HMODULE shfolder = LoadLibraryA("shfolder.dll"); if (shfolder) { HRESULT (WINAPI *dSHGetFolderPathW) (HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath); dSHGetFolderPathW = (void *)GetProcAddress(shfolder, "SHGetFolderPathW"); if (dSHGetFolderPathW) { wchar_t folderw[MAX_PATH]; // 0x5 == CSIDL_PERSONAL if (dSHGetFolderPathW(NULL, 0x5, NULL, 0, folderw) == S_OK) { narrowen(homedir, homedirsize, folderw); Q_strncatz(homedir, "/My Games/"FULLENGINENAME"/", homedirsize); return true; } } // FreeLibrary(shfolder); } return false; } static void Update_CreatePath (char *path) { char *ofs; for (ofs = path+1 ; *ofs ; ofs++) { if (*ofs == '/') { // create the directory *ofs = 0; Sys_mkdir (path); *ofs = '/'; } } } //ctx is a pointer to the original frontend process void Update_PromptedDownloaded(void *ctx, int foo) { if (foo == 0 && ctx) { PROCESS_INFORMATION childinfo; STARTUPINFOW startinfo = {sizeof(startinfo)}; wchar_t widearg[2048]; wchar_t wideexe[2048]; char cmdline[2048]; #ifndef SERVERONLY SetHookState(false); Host_Shutdown (); CloseHandle (qwclsemaphore); SetHookState(false); #else SV_Shutdown(); #endif TL_Shutdown(); narrowen(cmdline, sizeof(cmdline), GetCommandLineW()); widen(wideexe, sizeof(wideexe), ctx); widen(widearg, sizeof(widearg), va("\"%s\" %s", (char*)ctx, COM_Parse(cmdline))); CreateProcessW(wideexe, widearg, NULL, NULL, TRUE, 0, NULL, NULL, &startinfo, &childinfo); Z_Free(ctx); exit(1); } else Z_Free(ctx); } void Update_Version_Updated(struct dl_download *dl) { //happens in a thread, avoid va if (dl->file) { if (dl->status == DL_FINISHED) { char buf[8192]; unsigned int size = 0, chunk; char pendingname[MAX_OSPATH]; vfsfile_t *pending; Update_GetHomeDirectory(pendingname, sizeof(pendingname)); Q_strncatz(pendingname, DISTRIBUTION UPD_BUILDTYPE EXETYPE".tmp", sizeof(pendingname)); Update_CreatePath(pendingname); pending = VFSOS_Open(pendingname, "wb"); if (!pending) Con_Printf("Unable to write to \"%s\"\n", pendingname); else { while(1) { chunk = VFS_READ(dl->file, buf, sizeof(buf)); if (!chunk) break; size += VFS_WRITE(pending, buf, chunk); } VFS_CLOSE(pending); if (VFS_GETLEN(dl->file) != size) Con_Printf("Download was the wrong size / corrupt\n"); else { //figure out the original binary that was executed, so we can start from scratch. //this is to attempt to avoid the new process appearing as 'foo.tmp'. which needlessly confuses firewall rules etc. int ffe = COM_CheckParm("--fromfrontend"); wchar_t wbinarypath[MAX_PATH]; char ubinarypath[MAX_PATH]; char *ffp; GetModuleFileNameW(NULL, wbinarypath, countof(wbinarypath)-1); narrowen(ubinarypath, sizeof(ubinarypath), wbinarypath); ffp = Z_StrDup(ffe?com_argv[ffe+2]:ubinarypath); //make it pending MyRegSetValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "pending" UPD_BUILDTYPE EXETYPE, REG_SZ, pendingname, strlen(pendingname)+1); Key_Dest_Remove(kdm_console); M_Menu_Prompt(Update_PromptedDownloaded, ffp, "An update was downloaded", "Restart to activate.", "", ffp?"Restart":NULL, "", "Okay"); } } } else Con_Printf("Update download failed\n"); } } void Update_PromptedForUpdate(void *ctx, int foo) { if (foo == 0) { struct dl_download *dl; Con_Printf("Downloading update\n"); dl = HTTP_CL_Get(va(UPDATE_URL_BUILD, (char*)ctx), NULL, Update_Version_Updated); dl->file = FS_OpenTemp(); #ifdef MULTITHREAD DL_CreateThread(dl, NULL, NULL); #endif } else Con_Printf("Not downloading update\n"); } void Update_Versioninfo_Available(struct dl_download *dl) { if (dl->file) { if (dl->status == DL_FINISHED) { char linebuf[1024]; while(VFS_GETS(dl->file, linebuf, sizeof(linebuf))) { if (!strnicmp(linebuf, "Revision: ", 10)) { if (atoi(linebuf+10) > atoi(SVNREVISIONSTR)) { char *revision = va("Revision %i", atoi(linebuf+10)); char *current = va("Current %i", atoi(SVNREVISIONSTR)); Con_Printf("An update is available, revision %i\n", atoi(linebuf+10)); if (COM_CheckParm("-autoupdate") || COM_CheckParm("--autoupdate")) Update_PromptedForUpdate(dl->user_ctx, 0); else { Key_Dest_Remove(kdm_console); M_Menu_Prompt(Update_PromptedForUpdate, dl->user_ctx, "An update is available.", revision, current, "Download", "", "Ignore"); } } else Con_Printf("autoupdate: already at latest version\n"); return; } } } } } void Update_Check(void) { static qboolean doneupdatecheck; //once per run struct dl_download *dl; if (sys_autoupdatesetting < 2) //not if disabled (do it once it does get enabled) return; if (!doneupdatecheck) { char *updateroot = (sys_autoupdatesetting>=3)?UPDATE_URL_NIGHTLY:UPDATE_URL_TESTED; doneupdatecheck = true; dl = HTTP_CL_Get(va(UPDATE_URL_VERSION, updateroot), NULL, Update_Versioninfo_Available); dl->file = FS_OpenTemp(); dl->user_ctx = updateroot; dl->isquery = true; #ifdef MULTITHREAD DL_CreateThread(dl, NULL, NULL); #endif } } int Sys_GetAutoUpdateSetting(void) { return sys_autoupdatesetting; } void Sys_SetAutoUpdateSetting(int newval) { static qboolean doneupdatecheck; if (sys_autoupdatesetting == newval) return; sys_autoupdatesetting = newval; MyRegSetValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "AutoUpdateEnabled", REG_DWORD, &sys_autoupdatesetting, sizeof(sys_autoupdatesetting)); Update_Check(); } BOOL DeleteFileU(const char *path) { wchar_t wide[2048]; return DeleteFileW(widen(wide, sizeof(wide), path)); } BOOL MoveFileU(const char *src, const char *dst) { wchar_t wide1[2048]; wchar_t wide2[2048]; return MoveFileW(widen(wide1, sizeof(wide1), src), widen(wide2, sizeof(wide2), dst)); } qboolean Sys_CheckUpdated(char *bindir, size_t bindirsize) { wchar_t wide1[2048]; int ffe = COM_CheckParm("--fromfrontend"); PROCESS_INFORMATION childinfo; STARTUPINFOW startinfo = {sizeof(startinfo)}; char *e; strtoul(SVNREVISIONSTR, &e, 10); if (!*SVNREVISIONSTR || *e) //svn revision didn't parse as an exact number. this implies it has an 'M' in it to mark it as modified, or a - to mean unknown. either way, its bad and autoupdates when we don't know what we're updating from is a bad idea. sys_autoupdatesetting = -1; else if (COM_CheckParm("-noupdate") || COM_CheckParm("--noupdate") || COM_CheckParm("-noautoupdate") || COM_CheckParm("--noautoupdate")) sys_autoupdatesetting = 0; else if (COM_CheckParm("-autoupdate") || COM_CheckParm("--autoupdate")) sys_autoupdatesetting = 3; else { //favour 'tested' sys_autoupdatesetting = MyRegGetIntValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "AutoUpdateEnabled", 2); } if (!strcmp(SVNREVISIONSTR, "-")) return false; //no revision info in this build, meaning its custom built and thus cannot check against the available updated versions. else if (sys_autoupdatesetting == 0) return false; else if (isPlugin == 1) { //download, but don't invoke. the caller is expected to start us up properly (once installed). } else if (!ffe) { //if we're not from the frontend (ie: we ARE the frontend), we should run the updated build instead char pendingpath[MAX_OSPATH]; char updatedpath[MAX_OSPATH]; //FIXME: store versions instead of names MyRegGetStringValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "pending" UPD_BUILDTYPE EXETYPE, pendingpath, sizeof(pendingpath)); if (*pendingpath) { qboolean okay; MyRegDeleteKeyValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "pending" UPD_BUILDTYPE EXETYPE); Update_GetHomeDirectory(updatedpath, sizeof(updatedpath)); Update_CreatePath(updatedpath); Q_strncatz(updatedpath, "cur" UPD_BUILDTYPE EXETYPE".exe", sizeof(updatedpath)); DeleteFileU(updatedpath); okay = MoveFileU(pendingpath, updatedpath); if (!okay) { //if we just downloaded an update, we may need to wait for the existing process to close. //sadly I'm too lazy to provide any sync mechanism (and wouldn't trust any auto-released handles or whatever), so lets just retry after a delay. Sleep(2000); okay = MoveFileU(pendingpath, updatedpath); } if (okay) MyRegSetValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, UPD_BUILDTYPE EXETYPE, REG_SZ, updatedpath, strlen(updatedpath)+1); else { MessageBox(NULL, va("Unable to rename %s to %s", pendingpath, updatedpath), FULLENGINENAME" autoupdate", 0); DeleteFileU(pendingpath); } } MyRegGetStringValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, UPD_BUILDTYPE EXETYPE, updatedpath, sizeof(updatedpath)); if (*updatedpath) { wchar_t widefe[MAX_OSPATH], wargs[2048]; GetModuleFileNameW(NULL, widefe, countof(widefe)-1); _snwprintf(wargs, countof(wargs), L"%s --fromfrontend \"%s\" \"%s\"", GetCommandLineW(), widen(wide1, sizeof(wide1), SVNREVISIONSTR), widefe); if (CreateProcessW(widen(wide1, sizeof(wide1), updatedpath), wargs, NULL, NULL, TRUE, 0, NULL, NULL, &startinfo, &childinfo)) return true; } } else { // char frontendpath[MAX_OSPATH]; //com_argv[ffe+1] is frontend revision //com_argv[ffe+2] is frontend location if (atoi(com_argv[ffe+1]) > atoi(SVNREVISIONSTR)) { //ping-pong it back, to make sure we're running the most recent version. // GetModuleFileName(NULL, frontendpath, sizeof(frontendpath)-1); if (CreateProcessW(widen(wide1, sizeof(wide1), com_argv[ffe+2]), GetCommandLineW(), NULL, NULL, TRUE, 0, NULL, NULL, &startinfo, &childinfo)) return true; } if (com_argv[ffe+2]) { com_argv[0] = com_argv[ffe+2]; Q_strncpyz(bindir, com_argv[0], bindirsize); *COM_SkipPath(bindir) = 0; } } return false; } #else int Sys_GetAutoUpdateSetting(void) { return -1; } void Sys_SetAutoUpdateSetting(int newval) { } qboolean Sys_CheckUpdated(char *bindir, size_t bindirsize) { return false; } void Update_Check(void) { } #endif #include "shellapi.h" const GUID qIID_IApplicationAssociationRegistrationUI = {0x1f76a169,0xf994,0x40ac, {0x8f,0xc8,0x09,0x59,0xe8,0x87,0x47,0x10}}; const GUID qCLSID_ApplicationAssociationRegistrationUI = {0x1968106d,0xf3b5,0x44cf,{0x89,0x0e,0x11,0x6f,0xcb,0x9e,0xce,0xf1}}; struct qIApplicationAssociationRegistrationUI; typedef struct qIApplicationAssociationRegistrationUI { struct qIApplicationAssociationRegistrationUI_vtab { HRESULT (WINAPI *QueryInterface) (struct qIApplicationAssociationRegistrationUI *, const GUID *riid, void **ppvObject); HRESULT (WINAPI *AddRef) (struct qIApplicationAssociationRegistrationUI *); HRESULT (WINAPI *Release) (struct qIApplicationAssociationRegistrationUI *); HRESULT (WINAPI *LaunchAdvancedAssociationUI) (struct qIApplicationAssociationRegistrationUI *, LPCWSTR app); } *lpVtbl; } qIApplicationAssociationRegistrationUI; void Sys_DoFileAssociations(qboolean elevated) { char command[1024]; qboolean ok = true; HKEY root; //I'd do everything in current_user if I could, but windows sucks too much for that. //'registered applications' simply does not work in hkcu, we MUST use hklm for that. //if there's a registered application and we are not, we are unable to grab that association, ever. //thus we HAVE to do things to the local machine or we might as well not bother doing anything. //still, with a manifest not giving false success, if the user clicks 'no' to the UAC prompt, we'll write everything to the current user anyway, so if microsoft do ever fix things, then yay. //also, I hate the idea of creating a 'registered application' in globally without the file types it uses being local. //on xp, we use ONLY current user. no 'registered applications' means no 'registered applications bug', which means no need to use hklm at all. //in vista/7, we have to create stuff in local_machine. in which case we might as well put ALL associations in there. the ui stuff will allow user-specific settings, so this is not an issue other than the fact that it triggers uac. //in 8, we cannot programatically force ownership of our associations, so we might as well just use the ui method even for vista+7 instead of the ruder version. if (qwinvermaj < 6) elevated = 2; root = elevated == 2?HKEY_CURRENT_USER:HKEY_LOCAL_MACHINE; #define ASSOC_VERSION 2 #define ASSOCV "1" //register the basic demo class Q_snprintfz(command, sizeof(command), "Quake or QuakeWorld Demo"); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_DemoFile."ASSOCV, "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\",0", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_DemoFile."ASSOCV"\\DefaultIcon", "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\" \"%%1\"", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_DemoFile."ASSOCV"\\shell\\open\\command", "", REG_SZ, command, strlen(command)); //register the basic map class. yeah, the command is the same as for demos. but the description is different! Q_snprintfz(command, sizeof(command), "Quake Map"); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_BSPFile."ASSOCV, "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\",0", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_BSPFile."ASSOCV"\\DefaultIcon", "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\" \"%%1\"", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_BSPFile."ASSOCV"\\shell\\open\\command", "", REG_SZ, command, strlen(command)); //register the basic protocol class Q_snprintfz(command, sizeof(command), "QuakeWorld Server"); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_Server."ASSOCV"", "", REG_SZ, command, strlen(command)); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_Server."ASSOCV"", "URL Protocol", REG_SZ, "", strlen("")); Q_snprintfz(command, sizeof(command), "\"%s\",0", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_Server."ASSOCV"\\DefaultIcon", "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\" \"%%1\"", com_argv[0]); ok = ok & MyRegSetValue(root, "Software\\Classes\\"DISTRIBUTION"_Server."ASSOCV"\\shell\\open\\command", "", REG_SZ, command, strlen(command)); //try to get ourselves listed in windows' 'default programs' ui. Q_snprintfz(command, sizeof(command), "%s", FULLENGINENAME); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities", "ApplicationName", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "%s", FULLENGINENAME" is an awesome hybrid game engine able to run multiple Quake-compatible/derived games."); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities", "ApplicationDescription", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), DISTRIBUTION"_DemoFile.1"); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".qtv", REG_SZ, command, strlen(command)); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".mvd", REG_SZ, command, strlen(command)); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".qwd", REG_SZ, command, strlen(command)); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".dem", REG_SZ, command, strlen(command)); // ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".dm2", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), DISTRIBUTION"_BSPFile.1"); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".bsp", REG_SZ, command, strlen(command)); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".map", REG_SZ, command, strlen(command)); // ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\FileAssociations", ".fmf", REG_SZ, DISTRIBUTION"_ManifestFile", strlen(DISTRIBUTION"_ManifestFile")); // ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\MIMEAssociations", "application/x-ftemanifest", REG_SZ, DISTRIBUTION"_ManifestFile", strlen(DISTRIBUTION"_ManifestFile")); Q_snprintfz(command, sizeof(command), DISTRIBUTION"_Server.1"); ok = ok & MyRegSetValue(root, "Software\\"FULLENGINENAME"\\Capabilities\\UrlAssociations", "qw", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "Software\\"FULLENGINENAME"\\Capabilities"); ok = ok & MyRegSetValue(root, "Software\\RegisteredApplications", FULLENGINENAME, REG_SZ, command, strlen(command)); SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL); if (!ok && elevated < 2) { HINSTANCE ch = ShellExecute(mainwindow, "runas", com_argv[0], va("-register_types %i", elevated+1), NULL, SW_SHOWNORMAL); if ((intptr_t)ch <= 32) Sys_DoFileAssociations(2); return; } if (ok) { // char buf[1]; //attempt to display the vista+ prompt (only way possible in win8, apparently) qIApplicationAssociationRegistrationUI *aarui = NULL; //needs to be done anyway to ensure that its listed, and so that we get the association if nothing else has it. //however, the popup for when you start new programs is very annoying, so lets try to avoid that. our file associations are somewhat explicit anyway. //note that you'll probably still get the clumsy prompt if you try to run fte as a different user. really depends if you gave it local machine write access. // if (!aarui || elevated==2 || !MyRegGetStringValue(root, "Software\\Classes\\.qtv", "", buf, sizeof(buf))) MyRegSetValue(root, "Software\\Classes\\.qtv", "", REG_SZ, DISTRIBUTION"_DemoFile."ASSOCV, strlen(DISTRIBUTION"_DemoFile.1")); // if (!aarui || elevated==2 || !MyRegGetStringValue(root, "Software\\Classes\\.mvd", "", buf, sizeof(buf))) MyRegSetValue(root, "Software\\Classes\\.mvd", "", REG_SZ, DISTRIBUTION"_DemoFile."ASSOCV, strlen(DISTRIBUTION"_DemoFile.1")); // if (!aarui || elevated==2 || !MyRegGetStringValue(root, "Software\\Classes\\.qwd", "", buf, sizeof(buf))) MyRegSetValue(root, "Software\\Classes\\.qwd", "", REG_SZ, DISTRIBUTION"_DemoFile."ASSOCV, strlen(DISTRIBUTION"_DemoFile.1")); // if (!aarui || elevated==2 || !MyRegGetStringValue(root, "Software\\Classes\\.dem", "", buf, sizeof(buf))) MyRegSetValue(root, "Software\\Classes\\.dem", "", REG_SZ, DISTRIBUTION"_DemoFile."ASSOCV, strlen(DISTRIBUTION"_DemoFile.1")); // if (!aarui || elevated==2 || !MyRegGetStringValue(root, "Software\\Classes\\.bsp", "", buf, sizeof(buf))) MyRegSetValue(root, "Software\\Classes\\.bsp", "", REG_SZ, DISTRIBUTION"_BSPFile."ASSOCV, strlen(DISTRIBUTION"_BSPFile.1")); //legacy url associations are a bit more explicit // if (!aarui || elevated==2 || !MyRegGetStringValue(HKEY_CURRENT_USER, "Software\\Classes\\qw", "", buf, sizeof(buf))) { Q_snprintfz(command, sizeof(command), "QuakeWorld Server"); MyRegSetValue(root, "Software\\Classes\\qw", "", REG_SZ, command, strlen(command)); MyRegSetValue(root, "Software\\Classes\\qw", "URL Protocol", REG_SZ, "", strlen("")); Q_snprintfz(command, sizeof(command), "\"%s\",0", com_argv[0]); MyRegSetValue(root, "Software\\Classes\\qw\\DefaultIcon", "", REG_SZ, command, strlen(command)); Q_snprintfz(command, sizeof(command), "\"%s\" \"%%1\"", com_argv[0]); MyRegSetValue(root, "Software\\Classes\\qw\\shell\\open\\command", "", REG_SZ, command, strlen(command)); } CoInitialize(NULL); if (FAILED(CoCreateInstance(&qCLSID_ApplicationAssociationRegistrationUI, 0, CLSCTX_INPROC_SERVER, &qIID_IApplicationAssociationRegistrationUI, (LPVOID*)&aarui))) aarui = NULL; if (aarui) { #define wideify2(a) L##a #define wideify(a) wideify2(a) aarui->lpVtbl->LaunchAdvancedAssociationUI(aarui, wideify(FULLENGINENAME)); aarui->lpVtbl->Release(aarui); } else { /* #define wideify2(a) L##a #define wideify(a) wideify2(a) qOPENASINFO open_as_info = {0}; open_as_info.pcszFile = L".mvd"; open_as_info.pcszClass = wideify(DISTRIBUTION)L"_DemoFile.1"; open_as_info.oaifInFlags = 8 | 2;//OAIF_FORCE_REGISTRATION | OAIF_REGISTER_EXT; if (pSHOpenWithDialog) pSHOpenWithDialog(NULL, &open_as_info); SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL); */ } } } /* #ifdef _MSC_VER #include void VARGS Signal_Error_Handler(int i) { int *basepointer; __asm {mov basepointer,ebp}; Sys_Error("Received signal, offset was 0x%8x", basepointer[73]); } #endif */ extern char sys_language[64]; static int Sys_ProcessCommandline(char **argv, int maxargc, char *argv0) { int argc = 0, i; wchar_t *wc = GetCommandLineW(); unsigned char utf8cmdline[4096], *cl = utf8cmdline; narrowen(utf8cmdline, sizeof(utf8cmdline), wc); // argv[argc] = argv0; // argc++; while (*cl && (argc < maxargc)) { while (*cl && *cl <= 32) cl++; if (*cl) { if (*cl == '\"') { cl++; argv[argc] = cl; argc++; while (*cl && *cl != '\"') cl++; } else { argv[argc] = cl; argc++; while (*cl && *cl > 32) cl++; } if (*cl) { *cl = 0; cl++; } } } if (argc < 1) { argv[0] = argv0; argc = 1; } for (i = 0; i < argc; i++) argv[i] = strdup(argv[i]); return i; } int MessageBoxU(HWND hWnd, char *lpText, char *lpCaption, UINT uType) { wchar_t widecaption[256]; wchar_t widetext[2048]; widen(widetext, sizeof(widetext), lpText); widen(widecaption, sizeof(widecaption), lpCaption); return MessageBoxW(hWnd, widetext, widecaption, uType); } #ifdef WEBCLIENT //using this like posix' access function, but with much more code, microsoftisms, and no errno codes/info //no, I don't really have a clue why it needs to be so long. //#include #ifndef ACCESS_READ #define ACCESS_READ 0x1 #define ACCESS_WRITE 0x2 #endif static BOOL microsoft_accessW(LPWSTR pszFolder, DWORD dwAccessDesired) { HANDLE hToken; PRIVILEGE_SET PrivilegeSet; DWORD dwPrivSetSize; DWORD dwAccessGranted; BOOL fAccessGranted = FALSE; GENERIC_MAPPING GenericMapping; SECURITY_INFORMATION si = (SECURITY_INFORMATION)( OWNER_SECURITY_INFORMATION|GROUP_SECURITY_INFORMATION|DACL_SECURITY_INFORMATION); PSECURITY_DESCRIPTOR psdSD = NULL; DWORD dwNeeded; GetFileSecurityW(pszFolder,si,NULL,0,&dwNeeded); psdSD = malloc(dwNeeded); GetFileSecurityW(pszFolder,si,psdSD,dwNeeded,&dwNeeded); ImpersonateSelf(SecurityImpersonation); OpenThreadToken(GetCurrentThread(), TOKEN_ALL_ACCESS, TRUE, &hToken); memset(&GenericMapping, 0xff, sizeof(GENERIC_MAPPING)); GenericMapping.GenericRead = ACCESS_READ; GenericMapping.GenericWrite = ACCESS_WRITE; GenericMapping.GenericExecute = 0; GenericMapping.GenericAll = ACCESS_READ | ACCESS_WRITE; MapGenericMask(&dwAccessDesired, &GenericMapping); dwPrivSetSize = sizeof(PRIVILEGE_SET); AccessCheck(psdSD, hToken, dwAccessDesired, &GenericMapping, &PrivilegeSet, &dwPrivSetSize, &dwAccessGranted, &fAccessGranted); free(psdSD); return fAccessGranted; } static BOOL microsoft_accessU(LPCSTR pszFolder, DWORD dwAccessDesired) { wchar_t wpath[MAX_OSPATH]; return microsoft_accessW(widen(wpath, sizeof(wpath), pszFolder), dwAccessDesired); } static WNDPROC omgwtfwhyohwhy; static LRESULT CALLBACK stoopidstoopidstoopid(HWND w, UINT m, WPARAM wp, LPARAM lp) { switch (m) { case WM_NOTIFY: switch (((LPNMHDR)lp)->code) { case TVN_ENDLABELEDITW: { LRESULT r; NMTVDISPINFOW *fu = (NMTVDISPINFOW*)lp; NMTREEVIEWW gah; gah.action = TVC_UNKNOWN; gah.itemOld = fu->item; gah.itemNew = fu->item; gah.ptDrag.x = gah.ptDrag.y = 0; gah.hdr = fu->hdr; gah.hdr.code = TVN_SELCHANGEDW; r = CallWindowProcW(omgwtfwhyohwhy,w,m,wp,lp); CallWindowProcW(omgwtfwhyohwhy,w,WM_NOTIFY,wp,(LPARAM)&gah); return r; } break; case TVN_ENDLABELEDITA: { LRESULT r; NMTVDISPINFOA *fu = (NMTVDISPINFOA*)lp; NMTREEVIEWA gah; gah.action = TVC_UNKNOWN; gah.itemOld = fu->item; gah.itemNew = fu->item; gah.ptDrag.x = gah.ptDrag.y = 0; gah.hdr = fu->hdr; gah.hdr.code = TVN_SELCHANGEDA; r = CallWindowProcW(omgwtfwhyohwhy,w,m,wp,lp); CallWindowProcW(omgwtfwhyohwhy,w,WM_NOTIFY,wp,(LPARAM)&gah); return r; } break; case TVN_SELCHANGEDA: case TVN_SELCHANGEDW: break; } break; } return CallWindowProcW(omgwtfwhyohwhy,w,m,wp,lp); } struct egadsthisisretarded { wchar_t title[MAX_OSPATH]; char subdir[MAX_OSPATH]; char parentdir[MAX_OSPATH]; char statustext[MAX_OSPATH]; }; void FS_Directorize(char *fname, size_t fnamesize) { size_t l = strlen(fname); if (!l) //technically already a directory return; if (fname[l-1] == '\\' || fname[l-1] == '/') return; //already a directory Q_strncatz(fname, "/", fnamesize); } static INT CALLBACK StupidBrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lp, LPARAM pDatafoo) { //'stolen' from microsoft's knowledge base. //required to work around microsoft being annoying. struct egadsthisisretarded *pData = (struct egadsthisisretarded*)pDatafoo; // char *foo; HWND edit = FindWindowEx(hwnd, NULL, "EDIT", NULL); HWND list; extern qboolean com_homepathenabled; // OutputDebugString(va("got %u (%u)\n", uMsg, lp)); switch(uMsg) { case BFFM_INITIALIZED: //combat windows putting new windows behind everything else if it takes a while for UAC prompts to go away SetWindowPos(hwnd, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE); //combat windows bug where renaming something doesn't update the dialog's path list = FindWindowExW(hwnd, NULL, L"SHBROWSEFORFOLDER SHELLNAMESPACE CONTROL", NULL); if (list) omgwtfwhyohwhy = (WNDPROC)SetWindowLongPtr(list, GWLP_WNDPROC, (LONG_PTR)stoopidstoopidstoopid); { wchar_t szDir[MAX_PATH]; #ifndef _DEBUG //the standard location iiuc if (com_homepathenabled && pSHGetSpecialFolderPathW(NULL, szDir, CSIDL_PROGRAM_FILES, TRUE) && microsoft_accessW(szDir, ACCESS_READ | ACCESS_WRITE)) ; else if (microsoft_accessU("C:\\Games\\", ACCESS_READ | ACCESS_WRITE)) widen(szDir, sizeof(szDir), "C:\\Games\\"); else if (microsoft_accessU("C:\\", ACCESS_READ | ACCESS_WRITE)) widen(szDir, sizeof(szDir), "C:\\"); //if we're not an admin, install it somewhere else. else if (pSHGetSpecialFolderPathW(NULL, szDir, CSIDL_LOCAL_APPDATA, TRUE) && microsoft_accessW(szDir, ACCESS_READ | ACCESS_WRITE)) ; else #endif if (GetCurrentDirectoryW(countof(szDir), szDir))// && microsoft_access(szDir, ACCESS_READ | ACCESS_WRITE)) ; SendMessageW(hwnd, BFFM_SETSELECTIONW, TRUE, (LPARAM)szDir); SendMessageW(hwnd, BFFM_SETEXPANDED, TRUE, (LPARAM)szDir); SendMessageW(hwnd, BFFM_SETOKTEXT, TRUE, (LPARAM)widen(szDir, sizeof(szDir), "Install")); } break; case BFFM_VALIDATEFAILEDA: case BFFM_VALIDATEFAILEDW: if (!microsoft_accessU(pData->parentdir, ACCESS_READ | ACCESS_WRITE)) { MessageBoxU(hwnd, va("%s is not writable.", pData->parentdir), fs_gamename.string, 0); return 1; } if (edit) { wchar_t wide[256]; GetWindowTextW(edit, wide, countof(wide)); narrowen(pData->subdir, sizeof(pData->subdir), wide); } if (microsoft_accessU(va("%s%s", pData->parentdir, pData->subdir), ACCESS_READ)) return MessageBoxU(hwnd, va("%s%s already exists!\nThis installer will (generally) not overwrite existing data files.\nIf you want to re-install, you must manually uninstall it first.\n\nContinue?", pData->parentdir, pData->subdir), fs_gamename.string, MB_ICONWARNING|MB_OKCANCEL|MB_TOPMOST) == IDCANCEL; else return MessageBoxU(hwnd, va("Install to %s%s ?", pData->parentdir, pData->subdir), fs_gamename.string, MB_OKCANCEL) == IDCANCEL; case BFFM_SELCHANGED: { wchar_t wide[MAX_PATH*2+2]; char *foo; if (pSHGetPathFromIDListW((LPITEMIDLIST)lp, wide)) { narrowen(pData->parentdir, sizeof(pData->parentdir), wide); FS_Directorize(pData->parentdir, sizeof(pData->parentdir)); //this'll make microsoft happy. while(foo = strchr(pData->parentdir, '/')) *foo = '\\'; if (edit) { wchar_t wide[128]; SetWindowTextW(edit, widen(wide, sizeof(wide), fs_gamename.string)); SendMessageW(hwnd, BFFM_SETSTATUSTEXT, 0, (LPARAM)widen(wide, sizeof(wide), va("%s", pData->parentdir))); } else SendMessageW(hwnd, BFFM_SETSTATUSTEXT, 0, (LPARAM)widen(wide, sizeof(wide), va("%s%s", pData->parentdir, fs_gamename.string))); } } break; // case BFFM_IUNKNOWN: // break; } return 0; } LRESULT CALLBACK NoCloseWindowProc(HWND w, UINT m, WPARAM wp, LPARAM lp) { if (m == WM_CLOSE) return 0; return DefWindowProc(w, m, wp, lp); } void FS_CreateBasedir(const char *path); qboolean Sys_DoInstall(void) { extern ftemanifest_t *fs_manifest; char exepath[MAX_OSPATH]; char newexepath[MAX_OSPATH]; wchar_t wide[MAX_PATH]; wchar_t wide2[MAX_PATH]; char resultpath[MAX_OSPATH]; BROWSEINFOW bi; LPITEMIDLIST il; struct egadsthisisretarded diediedie; if (!pSHGetSpecialFolderPathW) { MessageBoxU(NULL, "SHGetSpecialFolderPathW is not supported\n", fs_gamename.string, 0); return TRUE; } if (fs_manifest && fs_manifest->eula) { if (MessageBoxU(NULL, fs_manifest->eula, fs_gamename.string, MB_OKCANCEL|MB_TOPMOST|MB_DEFBUTTON2) != IDOK) return TRUE; } Q_strncpyz(diediedie.subdir, fs_gamename.string, sizeof(diediedie.subdir)); _snwprintf(diediedie.title, countof(diediedie.title), L"Where would you like to install %s to?", widen(wide, sizeof(wide), fs_gamename.string)); GetCurrentDirectoryW(countof(wide)-1, wide); memset(&bi, 0, sizeof(bi)); bi.hwndOwner = mainwindow; //note that this is usually still null bi.pidlRoot = NULL; bi.pszDisplayName = wide; bi.lpszTitle = diediedie.title; bi.ulFlags = BIF_RETURNONLYFSDIRS|BIF_STATUSTEXT | BIF_EDITBOX|BIF_NEWDIALOGSTYLE|BIF_VALIDATE; bi.lpfn = StupidBrowseCallbackProc; bi.lParam = (LPARAM)&diediedie; bi.iImage = 0; il = pSHBrowseForFolderW?pSHBrowseForFolderW(&bi):NULL; if (il) { pSHGetPathFromIDListW(il, wide); CoTaskMemFree(il); } else return true; narrowen(resultpath, sizeof(resultpath), wide); FS_Directorize(resultpath, sizeof(resultpath)); if (*diediedie.subdir) { Q_strncatz(resultpath, diediedie.subdir, sizeof(resultpath)); FS_Directorize(resultpath, sizeof(resultpath)); } FS_CreateBasedir(resultpath); GetModuleFileNameW(NULL, wide, countof(wide)); narrowen(exepath, sizeof(exepath), wide); FS_NativePath(va("%s.exe", fs_gamename.string), FS_ROOT, newexepath, sizeof(newexepath)); CopyFileU(exepath, newexepath, FALSE); /*the game can now be run (using regular autoupdate stuff), but most installers are expected to install the data instead of just more downloaders, so lets do that with a 'nice' progress box*/ { HINSTANCE hInstance = NULL; HWND progress, label, wnd; WNDCLASS wc; RECT ca; int sh; int pct = -100; char fname[MAX_OSPATH]; memset(&wc, 0, sizeof(wc)); wc.style = 0; wc.lpfnWndProc = NoCloseWindowProc;//Progress_Wnd; wc.hInstance = hInstance; wc.hCursor = LoadCursor (NULL,IDC_ARROW); wc.hbrBackground = (void *)COLOR_WINDOW; wc.lpszClassName = "FTEPROG"; RegisterClass(&wc); ca.right = GetSystemMetrics(SM_CXSCREEN); ca.bottom = GetSystemMetrics(SM_CYSCREEN); mainwindow = wnd = CreateWindowEx(0, wc.lpszClassName, va("%s Installer", fs_gamename.string), 0, (ca.right-320)/2, (ca.bottom-100)/2, 320, 100, NULL, NULL, hInstance, NULL); GetClientRect(wnd, &ca); sh = GetSystemMetrics(SM_CYVSCROLL); Sys_LoadLibrary("comctl32.dll", NULL); // InitCommonControls(); label = CreateWindow("STATIC","", WS_CHILD | WS_VISIBLE | SS_PATHELLIPSIS, sh, ((ca.bottom-ca.top-sh)/3), ca.right-ca.left-2*sh, sh, wnd, NULL, hInstance, NULL); progress = CreateWindowEx(0, PROGRESS_CLASS, NULL, WS_CHILD | WS_VISIBLE | PBS_SMOOTH, sh, ((ca.bottom-ca.top-sh)/3)*2, ca.right-ca.left-2*sh, sh, wnd, NULL, hInstance, NULL); ShowWindow(wnd, SW_NORMAL); SetWindowPos(wnd, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE); SendMessage(progress, PBM_SETRANGE32, 0, 10000); *fname = 0; HTTP_CL_Think(); while(FS_DownloadingPackage()) { MSG msg; char *cur = cls.download?COM_SkipPath(cls.download->localname):"Please Wait"; int newpct = cls.download?cls.download->percent*100:0; if (cls.download && cls.download->sizeunknown) { //marquee needs manifest bollocks in order to work. so lets just not bother. float time = Sys_DoubleTime(); newpct = 10000 * (time - (int)time); if ((int)time & 1) newpct = 10000 - newpct; } if (Q_strcmp(fname, cur)) { Q_strncpyz(fname, cur, sizeof(fname)); SetWindowText(label, fname); } if (pct != newpct) { SendMessage(progress, PBM_SETPOS, pct, 0); pct = newpct; } while (PeekMessage (&msg, NULL, 0, 0, PM_REMOVE)) DispatchMessage (&msg); Sleep(10); HTTP_CL_Think(); } DestroyWindow(progress); DestroyWindow(wnd); UnregisterClass("FTEPROG", hInstance); mainwindow = NULL; } /*create startmenu icon*/ if (MessageBoxU(NULL, va("Create Startmenu icon for %s?", fs_gamename.string), fs_gamename.string, MB_YESNO|MB_ICONQUESTION|MB_TOPMOST) == IDYES) { HRESULT hres; IShellLinkW *psl; hres = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &qIID_IShellLinkW, (LPVOID*)&psl); if (SUCCEEDED(hres)) { char startmenu[MAX_OSPATH]; WCHAR wsz[MAX_PATH]; IPersistFile *ppf; widen(wsz, sizeof(wsz), newexepath); psl->lpVtbl->SetPath(psl, wsz); widen(wsz, sizeof(wsz), resultpath); psl->lpVtbl->SetWorkingDirectory(psl, wsz); hres = psl->lpVtbl->QueryInterface(psl, &qIID_IPersistFile, (LPVOID*)&ppf); if (SUCCEEDED(hres) && pSHGetSpecialFolderPathW(NULL, wsz, CSIDL_COMMON_PROGRAMS, TRUE)) { WCHAR wsz[MAX_PATH]; narrowen(startmenu, sizeof(startmenu), wsz); widen(wsz, sizeof(wsz), va("%s/%s.lnk", startmenu, fs_gamename.string)); hres = ppf->lpVtbl->Save(ppf, wsz, TRUE); if (hres == E_ACCESSDENIED && pSHGetSpecialFolderPathW(NULL, wsz, CSIDL_PROGRAMS, TRUE)) { narrowen(startmenu, sizeof(startmenu), wsz); widen(wsz, sizeof(wsz), va("%s/%s.lnk", startmenu, fs_gamename.string)); hres = ppf->lpVtbl->Save(ppf, wsz, TRUE); } ppf->lpVtbl->Release(ppf); } psl->lpVtbl->Release(psl); } } //now start it up properly. ShellExecuteW(mainwindow, L"open", widen(wide, sizeof(wide), newexepath), Q_strcasecmp(fs_manifest->installation, "quake")?L"":L"+sys_register_file_associations", widen(wide2, sizeof(wide2), resultpath), SW_SHOWNORMAL); return true; } qboolean Sys_RunInstaller(void) { HINSTANCE ch; char exepath[MAX_OSPATH]; if (COM_CheckParm("-doinstall")) return Sys_DoInstall(); if (!com_installer) return false; if (MessageBoxU(NULL, va("%s is not installed. Install now?", fs_gamename.string), fs_gamename.string, MB_OKCANCEL|MB_ICONQUESTION|MB_TOPMOST) == IDOK) { GetModuleFileName(NULL, exepath, sizeof(exepath)); ch = ShellExecute(mainwindow, "runas", com_argv[0], va("%s -doinstall", COM_Parse(GetCommandLine())), NULL, SW_SHOWNORMAL); if ((intptr_t)ch > 32) return true; //succeeded. should quit out. return Sys_DoInstall(); //if it failed, try doing it with the current privileges } return true; } #define RESLANG MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_UK) static const char *Sys_FindManifest(void) { HRSRC hdl = FindResource(NULL, MAKEINTRESOURCE(1), RT_RCDATA); HGLOBAL hgl = LoadResource(NULL, hdl); return LockResource(hgl); } //size info that microsoft recommends static const struct { int width; int height; int bpp; } icosizes[] = { // {96, 96, 32}, {48, 48, 32}, {32, 32, 32}, {16, 16, 32}, // {16, 16, 4}, // {48, 48, 4}, // {32, 32, 4}, // {16, 16, 1}, // {48, 48, 1}, // {32, 32, 1}, {256, 256, 32} //vista! }; //dates back to 16bit windows. bah. #pragma pack(push) #pragma pack(2) typedef struct { WORD idReserved; WORD idType; WORD idCount; struct { BYTE bWidth; BYTE bHeight; BYTE bColorCount; BYTE bReserved; WORD wPlanes; WORD wBitCount; DWORD dwBytesInRes; WORD nId; } idEntries[sizeof(icosizes)/sizeof(icosizes[0])]; } icon_group_t; #pragma pack(pop) static void Sys_MakeInstaller(const char *name) { vfsfile_t *filehandle; qbyte *filedata; unsigned int filelen; char *error = NULL; char *warn = NULL; HANDLE bin; char ourname[MAX_OSPATH]; char newname[MAX_OSPATH]; char tmpname[MAX_OSPATH]; Q_snprintfz(newname, sizeof(newname), "%s.exe", name); Q_snprintfz(tmpname, sizeof(tmpname), "tmp.exe"); GetModuleFileName(NULL, ourname, sizeof(ourname)); if (!CopyFile(ourname, tmpname, FALSE)) error = va("\"%s\" already exists or cannot be written", tmpname); if (!(bin = BeginUpdateResource(tmpname, FALSE))) error = "BeginUpdateResource failed"; else { filehandle = VFSOS_Open(va("%s.png", name), "rb"); if (filehandle) { icon_group_t icondata; qbyte *rgbadata; int imgwidth, imgheight; int iconid = 1; qboolean hasalpha; memset(&icondata, 0, sizeof(icondata)); icondata.idType = 1; filelen = VFS_GETLEN(filehandle); filedata = BZ_Malloc(filelen); VFS_READ(filehandle, filedata, filelen); VFS_CLOSE(filehandle); //nuke existing icons. UpdateResource(bin, RT_GROUP_ICON, MAKEINTRESOURCE(1), RESLANG, NULL, 0); UpdateResource(bin, RT_GROUP_ICON, MAKEINTRESOURCE(2), RESLANG, NULL, 0); // UpdateResource(bin, RT_GROUP_ICON, MAKEINTRESOURCE(3), RESLANG, NULL, 0); rgbadata = Read32BitImageFile(filedata, filelen, &imgwidth, &imgheight, &hasalpha, va("%s.png", name)); if (!rgbadata) error = "unable to read icon image"; else { void *data = NULL; unsigned int datalen = 0; unsigned int i; extern cvar_t gl_lerpimages; gl_lerpimages.ival = 1; for (i = 0; i < sizeof(icosizes)/sizeof(icosizes[0]); i++) { unsigned int x,y; unsigned int pixels; if (icosizes[i].width > imgwidth || icosizes[i].height > imgheight) continue; //ignore icons if they're bigger than the original icon. if (icosizes[i].bpp == 32 && icosizes[i].width >= 128 && icosizes[i].height >= 128 && icosizes[i].width == imgwidth && icosizes[i].height == imgheight) { //png compression. oh look. we originally loaded a png! data = filedata; datalen = filelen; } else { //generate the bitmap info BITMAPV4HEADER *bi; qbyte *out, *outmask; qbyte *in, *inrow; unsigned int outidx; pixels = icosizes[i].width * icosizes[i].height; bi = data = Z_Malloc(sizeof(*bi) + icosizes[i].width * icosizes[i].height * 5 + icosizes[i].height*4); memset(bi,0, sizeof(BITMAPINFOHEADER)); bi->bV4Size = sizeof(BITMAPINFOHEADER); bi->bV4Width = icosizes[i].width; bi->bV4Height = icosizes[i].height * 2; bi->bV4Planes = 1; bi->bV4BitCount = icosizes[i].bpp; bi->bV4V4Compression = BI_RGB; bi->bV4ClrUsed = (icosizes[i].bpp>=32?0:(1u<bV4Size; out = (qbyte*)data + datalen; datalen += ((icosizes[i].width*icosizes[i].bpp/8+3)&~3) * icosizes[i].height; outmask = (qbyte*)data + datalen; datalen += ((icosizes[i].width+31)&~31)/8 * icosizes[i].height; in = malloc(pixels*4); Image_ResampleTexture((unsigned int*)rgbadata, imgwidth, imgheight, (unsigned int*)in, icosizes[i].width, icosizes[i].height); inrow = in; outidx = 0; if (icosizes[i].bpp == 32) { for (y = 0; y < icosizes[i].height; y++) { inrow = in + 4*icosizes[i].width*(icosizes[i].height-1-y); for (x = 0; x < icosizes[i].width; x++) { if (inrow[3] == 0) //transparent outmask[outidx>>3] |= 1u<<(outidx&7); else { out[0] = inrow[2]; out[1] = inrow[1]; out[2] = inrow[0]; } out += 4; outidx++; inrow += 4; } if (x & 3) out += 4 - (x&3); outidx = (outidx + 31)&~31; } } } if (!error && !UpdateResource(bin, RT_ICON, MAKEINTRESOURCE(iconid), 0, data, datalen)) error = "UpdateResource failed (icon data)"; //and make a copy of it in the icon list icondata.idEntries[icondata.idCount].bWidth = (icosizes[i].width<256)?icosizes[i].width:0; icondata.idEntries[icondata.idCount].bHeight = (icosizes[i].height<256)?icosizes[i].height:0; icondata.idEntries[icondata.idCount].wBitCount = icosizes[i].bpp; icondata.idEntries[icondata.idCount].wPlanes = 1; icondata.idEntries[icondata.idCount].bColorCount = (icosizes[i].bpp>=8)?0:(1u< 1200 Win7_Init(); // #endif // #endif #ifdef _MSC_VER #if _M_IX86_FP >= 1 { int idedx; char cpuname[13]; /*I'm not going to check to make sure cpuid works.*/ __asm { xor eax, eax cpuid mov dword ptr [cpuname+0],ebx mov dword ptr [cpuname+4],edx mov dword ptr [cpuname+8],ecx } cpuname[12] = 0; __asm { mov eax, 0x1 cpuid mov idedx, edx } #if _M_IX86_FP >= 2 if (!(idedx&(1<<26))) MessageBox(NULL, "This is an SSE2 optimised build, and your cpu doesn't seem to support it", DISTRIBUTION, 0); else #endif if (!(idedx&(1<<25))) MessageBox(NULL, "This is an SSE optimised build, and your cpu doesn't seem to support it", DISTRIBUTION, 0); } #endif #endif #ifdef CATCHCRASH LoadLibraryU ("DBGHELP"); //heap corruption can prevent loadlibrary from working properly, so do this in advance. #ifdef MSVC_SEH __try #else { PVOID (WINAPI *pAddVectoredExceptionHandler)(ULONG FirstHandler, PVECTORED_EXCEPTION_HANDLER VectoredHandler); dllfunction_t dbgfuncs[] = {{(void*)&pAddVectoredExceptionHandler, "AddVectoredExceptionHandler"}, {NULL,NULL}}; if (Sys_LoadLibrary("kernel32.dll", dbgfuncs) && pAddVectoredExceptionHandler) pAddVectoredExceptionHandler(0, nonmsvc_CrashExceptionHandler); } #endif #endif { /* #ifndef _DEBUG #ifdef _MSC_VER signal (SIGFPE, Signal_Error_Handler); signal (SIGILL, Signal_Error_Handler); signal (SIGSEGV, Signal_Error_Handler); #endif #endif */ global_hInstance = hInstance; global_nCmdShow = nCmdShow; #ifdef RESTARTTEST setjmp (restart_jmpbuf); #endif if (WinNT) { wchar_t widebindir[1024]; GetModuleFileNameW(NULL, widebindir, sizeof(widebindir)/sizeof(widebindir[0])-1); narrowen(bindir, sizeof(bindir)-1, widebindir); } else GetModuleFileNameA(NULL, bindir, sizeof(bindir)-1); parms.argc = Sys_ProcessCommandline(sys_argv, MAX_NUM_ARGVS, bindir); *COM_SkipPath(bindir) = 0; parms.argv = (const char **)sys_argv; parms.binarydir = bindir; COM_InitArgv (parms.argc, parms.argv); #ifdef PLUGINS c = COM_CheckParm("--plugwrapper"); if (c) { int (QDECL *thefunc) (void); void *dummy; dllhandle_t *QVM_LoadDLL(const char *name, qboolean binroot, void **vmMain, sys_calldll_t syscall); dllhandle_t *lib = QVM_LoadDLL(com_argv[c+1], true, &dummy, NULL); thefunc = Sys_GetAddressForName(lib, com_argv[c+2]); if (thefunc) return thefunc(); return 0; } #endif c = COM_CheckParm("-qcdebug"); if (c) isPlugin = 3; else { c = COM_CheckParm("-plugin"); if (c) { if (c < com_argc && !strcmp(com_argv[c+1], "qcdebug")) isPlugin = 2; else isPlugin = 1; } else isPlugin = 0; } if (Sys_CheckUpdated(bindir, sizeof(bindir))) return true; if (COM_CheckParm("-register_types")) { Sys_DoFileAssociations(1); return true; } /* else if (!isPlugin) { if (MyRegGetIntValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "filetypes", -1) != ASSOC_VERSION) { DWORD dw = ASSOC_VERSION; if (IDYES == MessageBox(NULL, "Register file associations?", "FTE First Start", MB_YESNO)) Sys_DoFileAssociations(0); MyRegSetValue(HKEY_CURRENT_USER, "Software\\"FULLENGINENAME, "filetypes", REG_DWORD, &dw, sizeof(dw)); } } */ #if defined(CATCHCRASH) && defined(MULTITHREAD) #ifdef _MSC_VER if (COM_CheckParm("-watchdog")) watchdogthread = Sys_CreateThread("watchdog", watchdogthreadfunction, NULL, 0, 0); #endif #endif if (isPlugin==1) { printf("status Starting up!\n"); fflush(stdout); } if (COM_CheckParm("--version") || COM_CheckParm("-v")) { printf("version: %s\n", version_string()); return true; } if (COM_CheckParm("-outputdebugstring")) debugout = true; if (WinNT) { wchar_t wcwd[MAX_OSPATH]; if (!GetCurrentDirectoryW (sizeof(wcwd)/sizeof(wchar_t), wcwd)) Sys_Error ("Couldn't determine current directory"); narrowen(cwd, sizeof(cwd), wcwd); } else { if (!GetCurrentDirectoryA (sizeof(cwd), cwd)) Sys_Error ("Couldn't determine current directory"); } #ifdef WEBCLIENT c = COM_CheckParm("-makeinstaller"); if (c) { Sys_MakeInstaller(parms.argv[c+1]); return true; } parms.manifest = Sys_FindManifest(); #endif if (parms.argc >= 2) { if (*parms.argv[1] != '-' && *parms.argv[1] != '+') { char *e; if (parms.argc == 2 && !strchr(parms.argv[1], '\"') && !strchr(parms.argv[1], ';') && !strchr(parms.argv[1], '\n') && !strchr(parms.argv[1], '\r')) { HWND old; qtvfile = parms.argv[1]; old = FindWindowW(L"FTEGLQuake", NULL); if (!old) old = FindWindowW(L"FTED3D11QUAKE", NULL); if (!old) old = FindWindowW(L"FTED3D9QUAKE", NULL); if (old) { COPYDATASTRUCT cds; cds.dwData = 0xdeadf11eu; cds.cbData = strlen(qtvfile); cds.lpData = (void*)qtvfile; if (SendMessage(old, WM_COPYDATA, (WPARAM)GetDesktopWindow(), (LPARAM)&cds)) { Sleep(10*1000); //sleep for 10 secs so the real engine has a chance to open it, if the program that gave it is watching to see if we quit. return 0; //message sent. } } } else { MessageBox(NULL, va("Invalid commandline:\n%s", lpCmdLine), FULLENGINENAME, 0); return 0; } GetModuleFileName(NULL, cwd, sizeof(cwd)-1); for (e = cwd+strlen(cwd)-1; e >= cwd; e--) { if (*e == '/' || *e == '\\') { *e = 0; break; } } } } //98+/nt4+ if (GetLocaleInfo(LOCALE_USER_DEFAULT, LOCALE_SISO639LANGNAME, lang, sizeof(lang)) > 0) { if (GetLocaleInfoA(LOCALE_USER_DEFAULT, LOCALE_SISO3166CTRYNAME, ctry, sizeof(ctry)) > 0) Q_snprintfz(sys_language, sizeof(sys_language), "%s_%s", lang, ctry); else Q_snprintfz(sys_language, sizeof(sys_language), "%s", lang); } TL_InitLanguages(parms.binarydir); //tprints are now allowed if (*cwd && cwd[strlen(cwd)-1] != '\\' && cwd[strlen(cwd)-1] != '/') Q_strncatz(cwd, "/", sizeof(cwd)); parms.basedir = cwd; parms.argc = com_argc; parms.argv = com_argv; #if !defined(CLIENTONLY) && !defined(SERVERONLY) if (COM_CheckParm ("-dedicated")) isDedicated = true; #ifdef SUBSERVERS if (COM_CheckParm("-clusterslave")) isDedicated = isClusterSlave = true; #endif #endif if (isDedicated) { #if !defined(CLIENTONLY) if (!Sys_InitTerminal()) Sys_Error ("Couldn't allocate dedicated server console"); #endif } if (!Sys_Startup_CheckMem(&parms)) Sys_Error ("Not enough memory free; check disk space\n"); #ifndef CLIENTONLY if (isDedicated) //compleate denial to switch to anything else - many of the client structures are not initialized. { float delay; SV_Init (&parms); delay = SV_Frame(); while (1) { if (!isDedicated) Sys_Error("Dedicated was cleared"); NET_Sleep(delay, false); delay = SV_Frame(); } return TRUE; } #endif tevent = CreateEvent(NULL, FALSE, FALSE, NULL); if (!tevent) Sys_Error ("Couldn't create event"); #ifdef SERVERONLY Sys_Printf ("SV_Init\n"); SV_Init(&parms); #else Sys_Printf ("Host_Init\n"); Host_Init (&parms); #endif oldtime = Sys_DoubleTime (); if (qtvfile) { if (!Host_RunFile(qtvfile, strlen(qtvfile), NULL)) { SetHookState(false); Host_Shutdown (); return TRUE; } } //client console should now be initialized. #ifndef MINGW #if _MSC_VER > 1200 Win7_TaskListInit(); #endif #endif if (isPlugin==1) { printf("status Running!\n"); fflush(stdout); } Update_Check(); /* main window message loop */ while (1) { #ifdef CATCHCRASH watchdogframe++; #endif if (isDedicated) { #ifndef CLIENTONLY NET_Sleep(delay, false); // find time passed since last cycle newtime = Sys_DoubleTime (); time = newtime - oldtime; oldtime = newtime; delay = SV_Frame (); #else Sys_Error("wut?"); #endif } else { #ifndef SERVERONLY double sleeptime; newtime = Sys_DoubleTime (); time = newtime - oldtime; sleeptime = Host_Frame (time); oldtime = newtime; SetHookState(ActiveApp); /*sleep if its not yet time for a frame*/ if (sleeptime) Sys_Sleep(sleeptime); #else Sys_Error("wut?"); #endif } } } #ifdef CATCHCRASH #ifdef MSVC_SEH __except (CrashExceptionHandler(false, GetExceptionCode(), GetExceptionInformation())) { return 1; } #endif #endif /* return success of application */ return TRUE; } #ifdef _MSC_VER #pragma optimize( "", on) //revert back to default optimisations again. #endif int __cdecl main(void) { char *cmdline; FreeConsole(); cmdline = GetCommandLine(); while (*cmdline && *cmdline == ' ') cmdline++; if (*cmdline == '\"') { cmdline++; while (*cmdline && *cmdline != '\"') cmdline++; if (*cmdline == '\"') cmdline++; } else { while (*cmdline && *cmdline != ' ') cmdline++; } return WinMain(GetModuleHandle(NULL), NULL, cmdline, SW_NORMAL); } //now queries at startup and then caches, to avoid mode changes from giving weird results. qboolean Sys_GetDesktopParameters(int *width, int *height, int *bpp, int *refreshrate) { *width = desktopsettings.width; *height = desktopsettings.height; *bpp = desktopsettings.bpp; *refreshrate = desktopsettings.rate; return true; } static void Sys_QueryDesktopParameters(void) { HDC hdc; hdc = GetDC(NULL); desktopsettings.width = GetDeviceCaps(hdc, HORZRES); desktopsettings.height = GetDeviceCaps(hdc, VERTRES); desktopsettings.bpp = GetDeviceCaps(hdc, BITSPIXEL); desktopsettings.rate = GetDeviceCaps(hdc, VREFRESH); if (desktopsettings.rate == 1) desktopsettings.rate = 0; ReleaseDC(NULL, hdc); } void Sys_Sleep (double seconds) { Sleep(seconds * 1000); } HCURSOR hArrowCursor, hCustomCursor; void *WIN_CreateCursor(char *filename, float hotx, float hoty, float scale) { int width, height; BITMAPV5HEADER bi; DWORD x,y; HCURSOR hAlphaCursor = NULL; ICONINFO ii; HDC maindc; qbyte *rgbadata, *rgbadata_start, *bgradata, *bgradata_start; qboolean hasalpha; void *filedata; int filelen; if (!filename || !*filename) return NULL; filelen = FS_LoadFile(filename, &filedata); if (!filedata) return NULL; rgbadata_start = Read32BitImageFile(filedata, filelen, &width, &height, &hasalpha, "cursor"); FS_FreeFile(filedata); if (!rgbadata_start) return NULL; if (scale != 1) { int nw,nh; qbyte *nd; nw = width * scale; nh = height * scale; if (nw <= 0 || nh <= 0 || nw > 128 || nh > 128) //don't go crazy. return NULL; nd = BZ_Malloc(nw*nh*4); Image_ResampleTexture((unsigned int*)rgbadata_start, width, height, (unsigned int*)nd, nw, nh); width = nw; height = nh; BZ_Free(rgbadata_start); rgbadata_start = nd; } memset(&bi,0, sizeof(BITMAPV5HEADER)); bi.bV5Size = sizeof(BITMAPV5HEADER); bi.bV5Width = width; bi.bV5Height = height; bi.bV5Planes = 1; bi.bV5BitCount = 32; bi.bV5Compression = BI_BITFIELDS; // The following mask specification specifies a supported 32 BPP // alpha format for Windows XP. //FIXME: can we not just specify it as RGBA? meh. bi.bV5RedMask = 0x00FF0000; bi.bV5GreenMask = 0x0000FF00; bi.bV5BlueMask = 0x000000FF; bi.bV5AlphaMask = 0xFF000000; // Create the DIB section with an alpha channel. maindc = GetDC(mainwindow); ii.hbmColor = CreateDIBSection(maindc, (BITMAPINFO *)&bi, DIB_RGB_COLORS, (void **)&bgradata_start, NULL, 0); ReleaseDC(mainwindow, maindc); if (!ii.hbmColor) { BZ_Free(rgbadata_start); return NULL; } for (rgbadata=rgbadata_start,y=0;y