内核知识第六讲,内核编写规范,以及获取GDT表
一丶内核驱动编写规范
我们都知道,在ring3下,如果我们的程序出错了.那么就崩溃了.但是在ring0下,只要我们的程序崩溃了.那么直接就蓝屏了.
那么有那些规范?
1.最基本的检查要有. 比如检查指针是否为NULL,基本的校验错误必须有
2.对内存进行操作的时候,要进行内存判断.下面提供内存判断的API
BOOLEAN
MmIsAddressValid(
IN PVOID VirtualAddress
);
给定一个缓冲区,然后检查其内存是否有效
VOID
ProbeForWrite(
IN CONST VOID *Address,
IN SIZE_T Length,
IN ULONG Alignment
);
给定一个缓冲区.然后检查是否可以写. 但是注意,检测的是ring3下的内存地址.
3.尽量不要使用太大变量,因为在内核中.内存地址都是共享的.如果你使用了很大.那么相应的内核空间就会变少.
比如不能这样写 char szBuf[0x1000];这样是使用栈内存了.但是编译是不通过了.操作系统不让你这样使用.
但是你可以写成 char szBuf1[0x500], char szBuf2[0x500]; 但是没有任何意义.
4.ring0下不能直接访问ring3的内存
这个是有原因的.因为ring0知道一块物理内存.但是不知道是那个进程的.所以不能直接访问.
二丶分页内存与非分页内存管理
我们在内核下申请空间的API:
PVOID
ExAllocatePool(
IN POOL_TYPE PoolType, //指定内存的类型
IN SIZE_T NumberOfBytes //申请的字节数
);
但是第一个参数是我们制定内存的内容是一个结构体
typedef enum _POOL_TYPE {
NonPagedPool, //非分页管理
PagedPool, //分页管理
NonPagedPoolMustSucceed,
DontUseThisType,
NonPagedPoolCacheAligned,
PagedPoolCacheAligned,
NonPagedPoolCacheAlignedMustS
} POOL_TYPE;
然后类型中有个非分页管理以及分页管理,什么意思?
意思就是在内核中申请一块内存,这块内存可否与磁盘交互.
非分页内存: 不可以进行交互.申请了这块内存就不能动了.
分页内存: 可以进行交互.当内存资源紧张的是否,而这块内存很少是否.可以暂时放到磁盘.
操作系统的代码都是放到非分页内存中的.
三丶设置分页内存
了解了分页内存和非分页内存.那么此时就产生了一个新的问题.
我们的代码.有的只调用一次,不该占着这个空间不放.那么我们就要设置为分页内存中.
关键字:
总共两种方式:
.#pragma code_seg("PAGE") //每个函数都加上,设置到PAGE节中
例如:
但是这种方式,很麻烦,且每写一个函数都要设置.那么此时就有第二种方式
.#pragma Alloc_text("PAGE",函数名,.....) //类似于一个数组一样,可以设置多个.
但是此时就要注意了.我们就需要一个头文件了.把我们的函数声明都放到头文件中.
然后下方设置.
Page节,和INIT节. 内核中,我们编写的sys驱动程序,其实也是一个PE文件. 类似于我们的ring3下的DLL, ring3的DLL是给应用程序使用的.而sys则是给操作系统使用的. 只不过一个是3环一个是0环
Page节: 这个节则是我们说的分页内存节. 我们给了函数指针.那么就代表这块内存很*了.操作系统可以在资源紧张的时候使用.
INIT节: 这个节则是初始化的节.例如我们的入口函数,只用一次初始化而已,不用占着内存.所以给INIT节中.那么操作系统会初始化之后释放这块内存.这样可以留给其它驱动使用.
四丶设置内核代码运行的CPU在那个核心上跑.并获取出来每个核心的GDT表.
代码:
int shift = ;
char szGDT[];
nCount = KeQueryActiveProcessors();//获取CUP核心个数,按位来算.
while(shift != 0x80000000)
{
if (nCount & shift)
{
KeSetSystemAffinityThread(nCount & shift);//设置在那个线程跑
__asm
{
sgdt szGDT
}
KdPrint(("limit:%p GDT:%p\n", *(short*)szGDT, *(int*)(szGDT + )));
} shift <<= ;
}