Chinaunix首页 | 论坛 | 博客
  • 博客访问: 19001140
  • 博文数量: 7460
  • 博客积分: 10434
  • 博客等级: 上将
  • 技术积分: 78178
  • 用 户 组: 普通用户
  • 注册时间: 2008-03-02 22:54
文章分类

全部博文(7460)

文章存档

2011年(1)

2009年(669)

2008年(6790)

分类: C/C++

2008-05-28 16:26:43

VC6下编译进Ring0代码的疑惑,操作系统XPSP2,CPU:AMD3000+。现象,VC6总会优化代码,编译出来的代码不是想要的。

代码如下:

// tt.cpp : Defines the entry point for the application.
//

#include "stdafx.h"

#define _X86_

#include <windows.h>
#include <stdio.h>
#include <aclapi.h>
#include <conio.h>
#include <windef.h>
#include <shellapi.h>

typedef long NTSTATUS;
typedef unsigned short USHORT;
#define NT_SUCCESS(Status) ((NTSTATUS)(Status) >= 0)
#define OBJ_INHERIT 0x00000002L
#define OBJ_PERMANENT 0x00000010L
#define OBJ_EXCLUSIVE 0x00000020L
#define OBJ_CASE_INSENSITIVE 0x00000040L
#define OBJ_OPENIF 0x00000080L
#define OBJ_OPENLINK 0x00000100L
#define OBJ_KERNEL_HANDLE 0x00000200L
#define OBJ_VALID_ATTRIBUTES 0x000003F2L

typedef struct _UNICODE_STRING {
	USHORT Length;
	USHORT MaximumLength;
	
#ifdef MIDL_PASS
	[size_is(MaximumLength / 2), length_is((Length) / 2) ] USHORT * Buffer;
#else // MIDL_PASS
	PWSTR Buffer;
#endif // MIDL_PASS
} UNICODE_STRING;

typedef UNICODE_STRING *PUNICODE_STRING;
typedef const UNICODE_STRING *PCUNICODE_STRING;
#define UNICODE_NULL ((WCHAR)0) // winnt

typedef struct _OBJECT_ATTRIBUTES {
	ULONG Length;
	HANDLE RootDirectory;
	PUNICODE_STRING ObjectName;
	ULONG Attributes;
	PVOID SecurityDescriptor; // Points to type SECURITY_DESCRIPTOR
	PVOID SecurityQualityOfService; // Points to type SECURITY_QUALITY_OF_SERVICE
} OBJECT_ATTRIBUTES;

typedef OBJECT_ATTRIBUTES *POBJECT_ATTRIBUTES;

#define InitializeObjectAttributes( p, n, a, r, s ) { \
	(p)->Length = sizeof( OBJECT_ATTRIBUTES );\
	(p)->RootDirectory = r; \
	(p)->Attributes = a;\
	(p)->ObjectName = n;\
	(p)->SecurityDescriptor = s;\
	(p)->SecurityQualityOfService = NULL; \
}

extern "C"
typedef VOID (*pRtlInitUnicodeString)( PUNICODE_STRING DestinationString,PCWSTR SourceString);

extern "C"
typedef NTSTATUS (*pZwOpenSection)(OUT PHANDLE SectionHandle,IN ACCESS_MASK DesiredAccess,IN POBJECT_ATTRIBUTES ObjectAttributes);

extern "C" 
typedef NTSTATUS (*pZwClose)(IN HANDLE Handle);

static const HINSTANCE NTDLLHANDLE=(HINSTANCE)0x7c920000; //ntdll.dll加载的位置可以用GetModuleHandle获取

#define STATUS_SUCCESS ((NTSTATUS)0x00000000L) // ntsubauth
#define STATUS_ACCESS_DENIED ((NTSTATUS)0xC0000022L)
//#pragma comment(lib,"C:\\NTDDK\\libfre\\i386\\ntdll.lib")


#define ENTERRING0 _asm pushad \
	_asm pushf \
_asm cli

#define LEAVERING0 _asm popf \
	_asm popad \
_asm retf

typedef struct gdtr {
	unsigned short Limit;
	unsigned short BaseLow;
	unsigned short BaseHigh;
} Gdtr_t, *PGdtr_t;

typedef struct {
	unsigned short offset_0_15;
	unsigned short selector;
	
	unsigned char param_count : 4;
	unsigned char some_bits : 4;
	
	unsigned char type : 4;
	unsigned char app_system : 1;
	unsigned char dpl : 2;
	unsigned char present : 1;
	
	unsigned short offset_16_31;
} CALLGATE_DESCRIPTOR;


void PrintWin32Error( DWORD ErrorCode )
{
	LPVOID lpMsgBuf;
	
	FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, ErrorCode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &lpMsgBuf, 0, NULL);
	printf("%s\n", lpMsgBuf );
	LocalFree( lpMsgBuf );
}

ULONG MiniMmGetPhysicalAddress(ULONG virtualaddress)
{
	if(virtualaddress<0x80000000||virtualaddress>=0xA0000000)
		return 0;
	return virtualaddress&0x1FFFF000;
}

VOID SetPhyscialMemorySectionCanBeWrited(HANDLE hSection)
{
	PACL pDacl=NULL;
	PACL pNewDacl=NULL;
	PSECURITY_DESCRIPTOR pSD=NULL;
	DWORD dwRes;
	EXPLICIT_ACCESS ea;
	
	if(dwRes=GetSecurityInfo(hSection,SE_KERNEL_OBJECT,DACL_SECURITY_INFORMATION, NULL,NULL,&pDacl,NULL,&pSD) != ERROR_SUCCESS)
	{
		printf( "GetSecurityInfo Error %u\n", dwRes );
		goto CleanUp;
	}
	
	ZeroMemory(&ea, sizeof(EXPLICIT_ACCESS));
	ea.grfAccessPermissions = SECTION_MAP_WRITE;
	ea.grfAccessMode = GRANT_ACCESS;
	ea.grfInheritance= NO_INHERITANCE;
	ea.Trustee.TrusteeForm = TRUSTEE_IS_NAME;
	ea.Trustee.TrusteeType = TRUSTEE_IS_USER;
	ea.Trustee.ptstrName = "CURRENT_USER";
	
	if(dwRes=SetEntriesInAcl(1,&ea,pDacl,&pNewDacl)!=ERROR_SUCCESS)
	{
		printf( "SetEntriesInAcl %u\n", dwRes );
		goto CleanUp;
	}
	
	if(dwRes=SetSecurityInfo(hSection,SE_KERNEL_OBJECT,DACL_SECURITY_INFORMATION,NULL,NULL,pNewDacl,NULL)!=ERROR_SUCCESS)
	{
		printf("SetSecurityInfo %u\n",dwRes);
		goto CleanUp;
	}
	
CleanUp:
	
	if(pSD)
		LocalFree(pSD);
	if(pNewDacl)
		LocalFree(pSD);
}
#define RING0PROC void __declspec (naked)

BOOL ExecRing0Proc(ULONG Entry,ULONG seglen)
{
	Gdtr_t gdt;
	__asm sgdt gdt;
	
	ULONG mapAddr=MiniMmGetPhysicalAddress(gdt.BaseHigh<<16U|gdt.BaseLow);
	if(!mapAddr) return 0;
	
	HANDLE hSection=NULL;
	NTSTATUS status;
	OBJECT_ATTRIBUTES objectAttributes;
	UNICODE_STRING objName;
	CALLGATE_DESCRIPTOR *cg;
	
	status = STATUS_SUCCESS;
	
	pRtlInitUnicodeString RtlInitUnicodeString;
	pZwOpenSection ZwOpenSection;
	pZwClose ZwClose;
	
	RtlInitUnicodeString=(pRtlInitUnicodeString)GetProcAddress(NTDLLHANDLE,"RtlInitUnicodeString");
	ZwOpenSection=(pZwOpenSection)GetProcAddress(NTDLLHANDLE,"ZwOpenSection");
	ZwClose=(pZwClose)GetProcAddress(NTDLLHANDLE,"ZwClose");
	
	RtlInitUnicodeString(&objName,L"\\Device\\PhysicalMemory");
	InitializeObjectAttributes(&objectAttributes, &objName, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, NULL, (PSECURITY_DESCRIPTOR) NULL);
	status = ZwOpenSection(&hSection,SECTION_MAP_READ|SECTION_MAP_WRITE,&objectAttributes);
	
	//if(status == STATUS_ACCESS_DENIED) //这个地方就一直加强改写才行!
	{
		status = ZwOpenSection(&hSection,READ_CONTROL|WRITE_DAC,&objectAttributes);
		SetPhyscialMemorySectionCanBeWrited(hSection);
		ZwClose(hSection);
		status = ZwOpenSection(&hSection,SECTION_MAP_READ|SECTION_MAP_WRITE,&objectAttributes);
	}
	
	if(status != STATUS_SUCCESS)
	{
		printf("Error Open PhysicalMemory Section Object,Status:%08X\n",status);
		return 0;
	}
	
	PVOID BaseAddress;
	BaseAddress=MapViewOfFile(hSection,
		FILE_MAP_READ|FILE_MAP_WRITE,
		0,
		mapAddr, //low part
		(gdt.Limit+1));
	if(!BaseAddress)
	{
		printf("Error MapViewOfFile:");
		PrintWin32Error(GetLastError());
		return 0;
	}
	
	BOOL setcg=FALSE;
	
	for(cg=(CALLGATE_DESCRIPTOR *)((ULONG)BaseAddress+(gdt.Limit&0xFFF8));(ULONG)cg>(ULONG)BaseAddress;cg--)
		if(cg->type == 0){
			cg->offset_0_15 = LOWORD(Entry);
			cg->selector = 8;
			cg->param_count = 0;
			cg->some_bits = 0;
			cg->type = 0xC; // 386 call gate
			cg->app_system = 0; // A system descriptor
			cg->dpl = 3; // Ring 3 code can call
			cg->present = 1;
			cg->offset_16_31 = HIWORD(Entry);
			
			setcg=TRUE;
			break;
		}
		
		if(!setcg){
			ZwClose(hSection);
			return 0;
		}
		char *msg=new char[1000];
		sprintf(msg,"BaseAddress=%x\thSection=%x\tmapAddr=%x",BaseAddress,hSection,mapAddr);
		MessageBox(NULL,msg,NULL,NULL);
		delete [] msg;
		short farcall[3];
		
		farcall[2]=((short)((ULONG)cg-(ULONG)BaseAddress))|3; //Ring 3 callgate;
		if(!VirtualLock((PVOID)Entry,seglen))
		{
			printf("Error VirtualLock:");
			PrintWin32Error(GetLastError());
			return 0;
		}
		
		SetThreadPriority(GetCurrentThread(),THREAD_PRIORITY_TIME_CRITICAL);
		Sleep(0);
		
		_asm call fword ptr [farcall];
		
		MessageBox(NULL,"com",NULL,NULL);
		SetThreadPriority(GetCurrentThread(),THREAD_PRIORITY_NORMAL);
		
		VirtualUnlock((PVOID)Entry,seglen);
		
		//Clear callgate
		*(ULONG *)cg=0;
		*((ULONG *)cg+1)=0;
		ZwClose(hSection);
		MessageBox(NULL,"com2",NULL,NULL);
		return TRUE;
}

struct _RING0DATA
{
	DWORD mcr0,mcr2,mcr3;
	unsigned short BaseMemory;
	unsigned short ExtendedMemory;
}r0Data;

RING0PROC Ring0Proc1()
{
	ENTERRING0;
	_asm {
		mov eax, cr0
			mov r0Data.mcr0, eax;
		mov eax, cr2
			mov r0Data.mcr2, eax;
		mov eax, cr3
			mov r0Data.mcr3, eax;
	}
	LEAVERING0;
}

RING0PROC Ring0Proc2()
{
	ENTERRING0;
	_outp( 0x70, 0x15 );
	
	_asm
	{
		mov ax,0
			in al,71h
			mov r0Data.BaseMemory,ax
	}
	
	_outp( 0x70, 0x16 );
	r0Data.BaseMemory += _inp(0x71) << 8;
	_outp( 0x70, 0x17 );
	r0Data.ExtendedMemory = _inp( 0x71 );
	_outp( 0x70, 0x18 );
	r0Data.ExtendedMemory += _inp(0x71) << 8;
	LEAVERING0;
}



int Freq;
RING0PROC BeepOn()
{
	ENTERRING0;
	
    BYTE b;
	
    if ((Freq >= 20) && (Freq <= 20000))
    {
        Freq = 1193181 / Freq;
        b = _inp(0x61);
        if ((b & 3) == 0)
		{
			_outp(0x61, (BYTE) (b | 3));
			_outp(0x43, 0xb6);
		}
        _outp(0x42, (BYTE) Freq);
        _outp(0x42, (BYTE) (Freq >> 8));
    };
	
	LEAVERING0;
};

RING0PROC BeepOff()
{
	ENTERRING0;
	
    BYTE b;
	
    b= (_inp(0x61) & 0xfc);
    _outp(0x61, b);
	
	LEAVERING0;
};


int APIENTRY WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPSTR     lpCmdLine,
                     int       nCmdShow)
{
	ZeroMemory(&r0Data,sizeof(struct _RING0DATA));
	VirtualLock((PVOID)&r0Data,sizeof(struct _RING0DATA));
	ExecRing0Proc((ULONG)Ring0Proc1,0x100);
	ExecRing0Proc((ULONG)Ring0Proc2,0x100);
	VirtualUnlock((PVOID)&r0Data,sizeof(struct _RING0DATA));
	
	char* msg=new char[100];
	sprintf(msg,"CR0 = %x\tCR2 = %x\tCR3 = %x\t", r0Data.mcr0,r0Data.mcr2,r0Data.mcr3);
	MessageBox(NULL,msg,NULL,NULL);
	
	delete [] msg;
	Freq=5000;
	
	ExecRing0Proc((ULONG)BeepOn,0x100);
	Sleep(1000);
	Freq=3000;
	ExecRing0Proc((ULONG)BeepOn,0x100);
	Sleep(1000);
	ExecRing0Proc((ULONG)BeepOff,0x100);
	MessageBox(NULL,"com3",NULL,NULL);
	return 0;
}
    进Ring0的功能是正确的,问题出现在VC6汇编Sleep,MessageBox这样的函数时,把Sleep,MessageBox调用
地址写在ESI,EDI寄存器内。如主过程:
	ZeroMemory(&r0Data,sizeof(struct _RING0DATA));
	VirtualLock((PVOID)&r0Data,sizeof(struct _RING0DATA));
	ExecRing0Proc((ULONG)Ring0Proc1,0x100);
	ExecRing0Proc((ULONG)Ring0Proc2,0x100);
	VirtualUnlock((PVOID)&r0Data,sizeof(struct _RING0DATA));
	
	char* msg=new char[100];
	sprintf(msg,"CR0 = %x\tCR2 = %x\tCR3 = %x\t", r0Data.mcr0,r0Data.mcr2,r0Data.mcr3);
	MessageBox(NULL,msg,NULL,NULL);
	
	delete [] msg;
	Freq=5000;
	
	ExecRing0Proc((ULONG)BeepOn,0x100);
	Sleep(1000);
	Freq=3000;
	ExecRing0Proc((ULONG)BeepOn,0x100);
	Sleep(1000);
	ExecRing0Proc((ULONG)BeepOff,0x100);
	MessageBox(NULL,"com3",NULL,NULL);
	return 0;


汇编后成为:
004014A0  /___FCKpd___2nbsp; 33C0          XOR EAX,EAX                              ;  tt.00400000
004014A2  |.  56            PUSH ESI
004014A3  |.  A3 287A4000   MOV DWORD PTR DS:[407A28],EAX
004014A8  |.  57            PUSH EDI
004014A9  |.  A3 2C7A4000   MOV DWORD PTR DS:[407A2C],EAX
004014AE  |.  6A 10         PUSH 10
004014B0  |.  A3 307A4000   MOV DWORD PTR DS:[407A30],EAX
004014B5  |.  68 287A4000   PUSH tt.00407A28
004014BA  |.  A3 347A4000   MOV DWORD PTR DS:[407A34],EAX
004014BF  |.  FF15 18604000 CALL DWORD PTR DS:[<&KERNEL32.VirtualLoc>;  kernel32.VirtualLock
004014C5  |.  68 00010000   PUSH 100
004014CA  |.  68 D0134000   PUSH tt.004013D0
004014CF  |.  E8 7CFCFFFF   CALL tt.00401150
004014D4  |.  68 00010000   PUSH 100
004014D9  |.  68 F0134000   PUSH tt.004013F0
004014DE  |.  E8 6DFCFFFF   CALL tt.00401150
004014E3  |.  83C4 10       ADD ESP,10
004014E6  |.  6A 10         PUSH 10                                  ; /Size = 10 (16.)
004014E8  |.  68 287A4000   PUSH tt.00407A28                         ; |Address = tt.00407A28
004014ED  |.  FF15 30604000 CALL DWORD PTR DS:[<&KERNEL32.VirtualUnl>; \VirtualUnlock
004014F3  |.  6A 64         PUSH 64
004014F5  |.  E8 56010000   CALL tt.00401650
004014FA  |.  8B0D 307A4000 MOV ECX,DWORD PTR DS:[407A30]
00401500  |.  8B15 2C7A4000 MOV EDX,DWORD PTR DS:[407A2C]
00401506  |.  8BF0          MOV ESI,EAX
00401508  |.  A1 287A4000   MOV EAX,DWORD PTR DS:[407A28]
0040150D  |.  51            PUSH ECX
0040150E  |.  52            PUSH EDX
0040150F  |.  50            PUSH EAX
00401510  |.  68 88714000   PUSH tt.00407188                         ;  ASCII "CR0 = %x CR2 = %x CR3 = %x "
00401515  |.  56            PUSH ESI
00401516  |.  E8 E3000000   CALL tt.004015FE
0040151B  |.  8B3D D8604000 MOV EDI,DWORD PTR DS:[<&USER32.MessageBo>;  USER32.MessageBoxA
00401521  |.  83C4 18       ADD ESP,18
00401524  |.  6A 00         PUSH 0                                   ; /Style = MB_OK|MB_APPLMODAL
00401526  |.  6A 00         PUSH 0                                   ; |Title = NULL
00401528  |.  56            PUSH ESI                                 ; |Text
00401529  |.  6A 00         PUSH 0                                   ; |hOwner = NULL
0040152B  |.  FFD7          CALL EDI                                 ; \MessageBoxA
0040152D  |.  56            PUSH ESI
0040152E  |.  E8 C0000000   CALL tt.004015F3
00401533  |.  68 00010000   PUSH 100
00401538  |.  68 40144000   PUSH tt.00401440
0040153D  |.  C705 207A4000>MOV DWORD PTR DS:[407A20],1388
00401547  |.  E8 04FCFFFF   CALL tt.00401150
0040154C  |.  8B35 20604000 MOV ESI,DWORD PTR DS:[<&KERNEL32.Sleep>] ;  kernel32.Sleep
00401552  |.  83C4 0C       ADD ESP,0C
00401555  |.  68 E8030000   PUSH 3E8                                 ; /Timeout = 1000. ms
0040155A  |.  FFD6          CALL ESI                                 ; \Sleep
0040155C  |.  68 00010000   PUSH 100
00401561  |.  68 40144000   PUSH tt.00401440
00401566  |.  C705 207A4000>MOV DWORD PTR DS:[407A20],0BB8
00401570  |.  E8 DBFBFFFF   CALL tt.00401150
00401575  |.  83C4 08       ADD ESP,8
00401578  |.  68 E8030000   PUSH 3E8
0040157D  |.  FFD6          CALL ESI ;!!!这是调用Sleep,错误!
0040157F  |.  68 00010000   PUSH 100
00401584  |.  68 90144000   PUSH tt.00401490
00401589  |.  E8 C2FBFFFF   CALL tt.00401150
0040158E  |.  83C4 08       ADD ESP,8
00401591  |.  6A 00         PUSH 0
00401593  |.  6A 00         PUSH 0
00401595  |.  68 80714000   PUSH tt.00407180                         ;  ASCII "com3"
0040159A  |.  6A 00         PUSH 0
0040159C  |.  FFD7          CALL EDI ;!!!这是调用MessageBox,错误!
0040159E  |.  5F            POP EDI
0040159F  |.  33C0          XOR EAX,EAX
004015A1  |.  5E            POP ESI
004015A2  \.  C2 1000       RETN 10

每当Call完 401150,返回后,与只用用户态函数调用不同,寄存器的值都会改变!!!而VC6的编译,无论是优化速度,优化大小,禁止优化,都不能避免类似错误。

有什么办法解决这个问题呢?

我想可以用函数指针,通过指针来调用,不会出错,但是这样也太繁了。
哪位大侠有更好的解决办法啊?

阅读(710) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~