170 lines
6.3 KiB
C
Executable File
170 lines
6.3 KiB
C
Executable File
//Copyright (C) 2007 Anders Kjersem. Licensed under the zlib/libpng license, see License.txt for details.
|
|
#pragma once
|
|
/** /#define BUILD_DBGRELEASE // Include simple debug output in release version */
|
|
/** /#define BUILD_DBGSELECTELVMODE //Test MyRunAs*/
|
|
|
|
/** /#define UNICODE // Unicode build */
|
|
/**/#define FEAT_CUSTOMRUNASDLG // Include custom runas dialog */
|
|
/**/#define FEAT_CUSTOMRUNASDLG_TRANSLATE //*/
|
|
/**/#define FEAT_MSRUNASDLGMODHACK // Default to other user radio button */
|
|
|
|
|
|
#if !defined(APSTUDIO_INVOKED) && !defined(RC_INVOKED)
|
|
|
|
#if (defined(_MSC_VER) && !defined(_DEBUG))
|
|
#pragma comment(linker,"/opt:nowin98")
|
|
#pragma comment(linker,"/ignore:4078")
|
|
#pragma comment(linker,"/merge:.rdata=.text")
|
|
|
|
//#pragma intrinsic(memset) //http://www.codeguru.com/forum/showthread.php?t=371491&page=2&pp=15 | http://www.ddj.com/windows/184416623
|
|
#endif
|
|
|
|
#if defined(UNICODE) && !defined(_UNICODE)
|
|
#define _UNICODE
|
|
#endif
|
|
#ifdef _UNICODE
|
|
#define TFUNCSUFFIX W
|
|
#else
|
|
#define TFUNCSUFFIX A
|
|
#endif
|
|
#define _PCJOIN(a,b) a##b
|
|
#define PCJOIN(a,b) _PCJOIN(a,b)
|
|
|
|
|
|
#define _WIN32_WINNT 0x0501
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <ShellAPI.h>
|
|
#include <TChar.h>
|
|
#include "NSISUtil.h"
|
|
|
|
#ifndef SEE_MASK_NOZONECHECKS
|
|
#define SEE_MASK_NOZONECHECKS 0x00800000
|
|
#endif
|
|
|
|
#define COUNTOF(___c) ( sizeof(___c)/sizeof(___c[0]) )
|
|
#ifndef ARRAYSIZE
|
|
#define ARRAYSIZE COUNTOF
|
|
#endif
|
|
#define FORCEINLINE __forceinline
|
|
|
|
#if _MSC_VER >= 1400
|
|
extern void* __cdecl memset(void*mem,int c,size_t len);
|
|
#endif
|
|
|
|
FORCEINLINE LRESULT MySndDlgItemMsg(HWND hDlg,int id,UINT Msg,WPARAM wp=0,LPARAM lp=0) {return SendMessage(GetDlgItem(hDlg,id),Msg,wp,lp);}
|
|
#ifndef UAC_NOCUSTOMIMPLEMENTATIONS
|
|
FORCEINLINE HANDLE WINAPI GetCurrentProcess(){return ((HANDLE)(-1));}
|
|
FORCEINLINE HANDLE WINAPI GetCurrentThread(){return ((HANDLE)(-2));}
|
|
|
|
#define MyTStrLen lstrlen
|
|
|
|
#undef lstrcpy
|
|
#define lstrcpy MyTStrCpy
|
|
FORCEINLINE LPTSTR MyTStrCpy(LPTSTR s1,LPCTSTR s2) {return PCJOIN(lstr,PCJOIN(cpyn,TFUNCSUFFIX))(s1,s2,0x7FFFFFFF);}
|
|
|
|
#undef lstrcat
|
|
#define lstrcat MyTStrCat
|
|
LPTSTR MyTStrCat(LPTSTR s1,LPCTSTR s2)
|
|
#ifdef UAC_INITIMPORTS
|
|
{return s1?MyTStrCpy(&s1[MyTStrLen(s1)],s2):NULL;}
|
|
#else
|
|
;
|
|
#endif //UAC_INITIMPORTS
|
|
|
|
#endif //UAC_NOCUSTOMIMPLEMENTATIONS
|
|
|
|
|
|
//DelayLoaded functions:
|
|
typedef BOOL (WINAPI*ALLOWSETFOREGROUNDWINDOW)(DWORD dwProcessId);
|
|
typedef BOOL (WINAPI*OPENPROCESSTOKEN)(HANDLE ProcessHandle,DWORD DesiredAccess,PHANDLE TokenHandle);
|
|
typedef BOOL (WINAPI*OPENTHREADTOKEN)(HANDLE ThreadHandle,DWORD DesiredAccess,BOOL OpenAsSelf,PHANDLE TokenHandle);
|
|
typedef BOOL (WINAPI*GETTOKENINFORMATION)(HANDLE hToken,TOKEN_INFORMATION_CLASS TokInfoClass,LPVOID TokInfo,DWORD TokInfoLenh,PDWORD RetLen);
|
|
typedef BOOL (WINAPI*ALLOCATEANDINITIALIZESID)(PSID_IDENTIFIER_AUTHORITY pIdentifierAuthority,BYTE nSubAuthorityCount,DWORD sa0,DWORD sa1,DWORD sa2,DWORD sa3,DWORD sa4,DWORD sa5,DWORD sa6,DWORD sa7,PSID*pSid);
|
|
typedef PVOID (WINAPI*FREESID)(PSID pSid);
|
|
typedef BOOL (WINAPI*EQUALSID)(PSID pSid1,PSID pSid2);
|
|
typedef BOOL (WINAPI*CHECKTOKENMEMBERSHIP)(HANDLE TokenHandle,PSID SidToCheck,PBOOL IsMember);
|
|
#ifdef FEAT_CUSTOMRUNASDLG
|
|
typedef BOOL (WINAPI*GETUSERNAME)(LPTSTR lpBuffer,LPDWORD nSize);
|
|
typedef BOOL (WINAPI*CREATEPROCESSWITHLOGONW)(LPCWSTR lpUsername,LPCWSTR lpDomain,LPCWSTR lpPassword,DWORD dwLogonFlags,LPCWSTR lpApplicationName,LPWSTR lpCommandLine,DWORD dwCreationFlags,LPVOID pEnv,LPCWSTR WorkDir,LPSTARTUPINFOW pSI,LPPROCESS_INFORMATION pPI);
|
|
#define SECURITY_WIN32
|
|
#include <Security.h>//NameSamCompatible
|
|
typedef BOOLEAN (WINAPI*GETUSERNAMEEX)(EXTENDED_NAME_FORMAT NameFormat,LPTSTR lpNameBuffer,PULONG nSize);
|
|
#endif
|
|
#ifdef UAC_INITIMPORTS
|
|
ALLOWSETFOREGROUNDWINDOW _AllowSetForegroundWindow=0;
|
|
OPENPROCESSTOKEN _OpenProcessToken=0;
|
|
OPENTHREADTOKEN _OpenThreadToken=0;
|
|
GETTOKENINFORMATION _GetTokenInformation=0;
|
|
ALLOCATEANDINITIALIZESID _AllocateAndInitializeSid=0;
|
|
FREESID _FreeSid=0;
|
|
EQUALSID _EqualSid=0;
|
|
CHECKTOKENMEMBERSHIP _CheckTokenMembership=0;
|
|
#ifdef FEAT_CUSTOMRUNASDLG
|
|
GETUSERNAME _GetUserName=0;
|
|
GETUSERNAMEEX _GetUserNameEx=0;
|
|
CREATEPROCESSWITHLOGONW _CreateProcessWithLogonW=0;
|
|
#endif
|
|
#else
|
|
#ifdef FEAT_CUSTOMRUNASDLG
|
|
extern GETUSERNAME _GetUserName;
|
|
extern GETUSERNAMEEX _GetUserNameEx;
|
|
extern CREATEPROCESSWITHLOGONW _CreateProcessWithLogonW;
|
|
#endif
|
|
#endif /* UAC_INITIMPORTS */
|
|
|
|
|
|
extern DWORD DelayLoadDlls();
|
|
#ifdef FEAT_CUSTOMRUNASDLG
|
|
extern DWORD MyRunAs(HINSTANCE hInstDll,SHELLEXECUTEINFO&sei);
|
|
#endif
|
|
|
|
#if !defined(NTDDI_VISTA) || defined(BUILD_OLDSDK)
|
|
//#if !defined(NTDDI_VERSION) || (NTDDI_VERSION < 0x06000000) || !defined(NTDDI_VISTA)
|
|
//#if !defined(TOKEN_ELEVATION_TYPE) || defined(BUILD_OLDSDK)
|
|
enum TOKEN_ELEVATION_TYPE {
|
|
TokenElevationTypeDefault = 1,
|
|
TokenElevationTypeFull,
|
|
TokenElevationTypeLimited
|
|
};
|
|
enum _TOKEN_INFORMATION_CLASS___VISTA {
|
|
TokenElevationType = (TokenOrigin+1),
|
|
TokenLinkedToken,
|
|
TokenElevation,
|
|
TokenHasRestrictions,
|
|
TokenAccessInformation,
|
|
TokenVirtualizationAllowed,
|
|
TokenVirtualizationEnabled,
|
|
TokenIntegrityLevel,
|
|
TokenUIAccess,
|
|
TokenMandatoryPolicy,
|
|
TokenLogonSid,
|
|
};
|
|
#endif
|
|
|
|
|
|
#if defined(_DEBUG) || defined(BUILD_DBGRELEASE)
|
|
//Simple debug helpers:
|
|
#define BUILD_DBG
|
|
/** /#define BUILD_XPTEST //Pretend UAC exists and "elevate" with NT runas */
|
|
#define DBG_RESETDBGVIEW() do{HWND hDbgView=FindWindowA("dbgviewClass",0);PostMessage(hDbgView,WM_COMMAND,40020,0);if(0)SetForegroundWindow(hDbgView);}while(0)
|
|
#define _pp_MakeStr_(x) #x
|
|
#define pp_MakeStr(x) _pp_MakeStr_(x)
|
|
#define TRACE OutputDebugStringA
|
|
#define DBGONLY(_x) _x
|
|
#ifndef ASSERT
|
|
#define ASSERT(_x) do{if(!(_x)){MessageBoxA(GetActiveWindow(),#_x##"\n\n"##__FILE__##":"## pp_MakeStr(__LINE__),"SimpleAssert",0);/*TRACE(#_x##"\n"##__FILE__##":" pp_MakeStr(__LINE__)"\n");*/}}while(0)
|
|
#endif
|
|
#define VERIFY(_x) ASSERT(_x)
|
|
static void TRACEF(const char*fmt,...) {va_list a;va_start(a,fmt);static TCHAR b[1024*4];if (sizeof(TCHAR)!=sizeof(char)){static TCHAR fmtBuf[COUNTOF(b)];VERIFY(wsprintf(fmtBuf,_T("%hs"),fmt)<COUNTOF(fmtBuf));fmt=(LPCSTR)fmtBuf;}wvsprintf(b,(TCHAR*)fmt,a);OutputDebugString(b);}
|
|
#else
|
|
#define TRACE /*(void)0*/
|
|
#define DBGONLY(_x)
|
|
#define ASSERT(_x)
|
|
#define VERIFY(_x) ((void)(_x))
|
|
#define TRACEF TRACE
|
|
#endif /* DEBUG */
|
|
|
|
#endif /* APSTUDIO_INVOKED */
|
|
|