//中断门描述符 仅做参考
typedef struct _INTER_DESC
{
	short OffsetLow;
	short selector;
	char IST : 3;
	char dumb : 5;
	char Attr0 : 1;
	char Attr1 : 1;
	char Attr2 : 1;
	char Attr3 : 1;
	char Attrzero : 1;
	char AttrDPL : 2;
	char Attrp : 1;
	short OffsetHigh;
}INTER_DESC, *PINTER_DESC;

VOID WriteTrap(ULONG32 FuncAddr)
{
//参数传NewTrap
	_asm
	{
		;写调用门用到的索引
		mov ebx,g_eax
	}
	_asm
	{
		
		sub esp, 20h
		push esi
		; mov esi, ptr IDT
		sidt[esp - 2]
		pop esi

		; 找到第三个中断描述符的位置
		add esi, 18h
		mov ebx, [esi]
		mov g_DescLow, ebx
		mov ebx, [esi + 4]
		;用的时候取高16位
		mov g_DescHigh, ebx
		mov ebx, FuncAddr
		mov[esi + INTERGATE_OFFSETLOW], bx
		shr ebx, 16
		mov[esi + INTERGATE_OFFSETHIGH], bx
		mov[esi + INTERGATE_ATTR], 0xee; ? ?
		mov[esi + INTERGATE_DUMB], 00
		; mov ebx, g_eax
		; shr ebx, 3
		; 1 0 00   Index == 1,RPL == 0
		mov [esi + INTERGATE_SELECTOR], 08h
		
		//保存真实例程地址
		lea eax, g_RealTrap
		mov ebx, g_DescLow
		mov[eax], bx
		mov ebx, g_DescHigh
		shr ebx, 16
		mov[eax + 2], bx

	/*	
		Hook INT 3的时候不好调试，在这先调试下恢复
		mov ebx, g_DescLow
		mov[esi + INTERGATE_OFFSETLOW], bx
		mov ebx, g_DescHigh
		shr ebx, 16
		mov[esi + INTERGATE_OFFSETHIGH], bx
		*/

		add esp,20h
	}
}

//段描述符要用
typedef struct _SEG_DESC
{
	short limit;
	ULONG32 BaseLow : 24;
	
	union OneByte
	{
		struct onebyte
		{
			char type0 : 1;
			char type1 : 1;
			char type2 : 1;
			char type3 : 1;
			char s : 1;
			char DPL : 2;
			char P : 1;
		};
		char dumb0;            //赋值时用
	};
	
	union anotherone
	{
		struct ByteAgain
		{
			char lim : 4;
			char AVL : 1;
			char L : 1;
			char DB : 1;
			char G : 1;
		};
		char dumb1;
	};
	
	char BaseHigh;
}SEG_DESC, *PSEG_DESC;

VOID CoverGDT(ULONG32 FuncAddr)
{
	//修改SegmentBase里的描述符第16~39位 26~63位的base字段
	//本来想直接修改索引号为1的段描述符，这样就不用改IDT了
	//然而一改虚拟机就提示CPU不干....
	ULONG32 Pointer = 0;
	PSEG_DESC pSeg = NULL;
	ULONG32 uTemp = 0;
	_asm
	{
		pushfd
		pushad
		; 找到三号中断入口地址s
		push esi
		; mov esi, ptr IDT
		sidt[esp - 2]
		pop esi

		; 找到第三个中断描述符的位置
		add esi, 18h
		mov ebx, [esi]
		mov g_DescLow, ebx
		mov ebx, [esi + 4]
		mov g_DescHigh, ebx
		;计算原先的中断处理例程地址
		lea eax, g_RealTrap
		mov ebx, g_DescLow
		mov [eax], bx
		mov ebx, g_DescHigh
		shr ebx, 16
		mov[eax + 2], bx
		;算出函数地址偏移 准备赋给Segment Descriptor的base字段
		mov eax, g_RealTrap
		mov ebx, FuncAddr
		sub ebx, eax
		mov uTemp,ebx
		lea ebx,uTemp
		;开始遍历GDT
		push esi
		sgdt[esp - 2]
		; esi: ptr Segment Descriptor
		pop esi
		;eax作计数器
		mov eax, 8
LoopStart :
		lea edx, [esi + eax]
		xor ecx, ecx
		mov cl, [edx + 5]
		test cl, 80h
		; 字面上来看就是 P位为1，其它为0的就过，但是实际上找的是cl == 0的，就是8Bytes全0
		jnz notZero
		; 计数and写段描述符
		mov g_eax, eax
		; 找到空的位置就写我们的描述符
		; 这里只管我们算出来的Base字段，其它的都抄Index == 1的Segment Descriptor了
		mov[edx + SEGDESC_LIMIT], 0xff
		mov[edx + SEGDESC_LIMIT + 1], 0xff
		mov eax,[ebx]
		mov[edx + SEGDESC_BASELOW], ax
		mov al,[ebx + 2]
		mov[edx + SEGDESC_BASELOW + 2],al 
		mov al,[ebx + 3]
		mov[edx + SEGDESC_BASEHIGH], al
		mov[edx + SEGDESC_TYPE], 9bh
		mov[edx + SEGDESC_FLAGS], 0xcf; 为啥有的后面加h就行，有的必须写0x ?
		
                ;开始PatchIDT
		push esi
		; mov esi, ptr IDT
		sidt[esp - 2]
		pop esi
		add esi, 18h
		mov ebx, g_eax
		; 因为描述符大小是8字节 除8得索引
		shr ebx, 3
		mov[esi + INTERGATE_SELECTOR], bx
		
		popad
		popfd
		mov esp, ebp
		pop ebp
		ret

notZero :
		add eax, 8
		test eax, GDT_LIMIT
		jnz LoopStart
		;找不到空的，那就算了
		mov eax, 0
		popad
		popfd
		ret
	}

}

VOID DriverUnload(PDRIVER_OBJECT DriverObject)
{
	DbgPrint("DriverUnload()\r\n");
//	WPOFF();
	_asm
	{
		;Resume IDT
		sub esp, 20h
		push esi
		; mov esi, ptr IDT
		sidt[esp - 2]
		pop esi
		; 找到第三个中断描述符的位置
		add esi, 18h

		mov ebx, g_DescLow
		mov[esi + INTERGATE_OFFSETLOW], bx
		mov ebx, g_DescHigh
		shr ebx, 16
		mov[esi + INTERGATE_OFFSETHIGH], bx

		add esp, 20h
	}
//	WPON();
}

//调用门描述符 参考
typedef struct _CALLGATE_DESC
{
	SHORT	OFFSETL;
	SHORT	SELECTOR;
	CHAR	DCOUNT;
	CHAR	GTYPE;
	SHORT	OFFSETH;
}CALLGATE_DESC;

ULONG32 AddCallGate(ULONG32 FuncAddr)
{
	INT32 i = 0;
	_asm
	{
		; push ebp
		; mov ebp, esp
		sub esp, 20h
		pushfd
		pushad
		push esi
		; esp - 2里写入的是低16位的limit，esp里写入的是32位的Base
		sgdt[esp - 2]
		; 得到ptr GDT
		pop esi
		; 用eax遍历GDT
		mov eax, 8
		mov i, eax
LoopStart :
		lea edx, [esi + eax]
		; edx: ptr CALLGATE
		xor ecx, ecx
		mov cl, [edx + 5]
		;0x80 Attr == 0x80 P == 1,其它标志位为0
		test cl, 80h
		jnz notZero
		mov g_eax, eax
		mov ebx, FuncAddr
		mov[edx], bx
		;selector == 08		1 0 00
		mov[edx + 2], 08h
		mov[edx + 4], 0
		mov[edx + 5], GATE_TYPE
		;将高16位放在结构体的高16位里
		shr ebx, 16
		mov[edx + 6], bx
		popad
		popfd
		mov esp, ebp
		pop ebp
		ret
notZero :
		add eax, 8
		test eax, GDT_LIMIT
		jnz LoopStart
		mov eax, 0
		popad
		popfd
		ret
	}
}