ring0-内存可读、可写、有效性、指针是否为空、深度校验字符串（随手代码）

http://blog.csdn.net/hgy413/article/details/7907057

1.如在ring3下，则要判断是否可读可写：

[cpp] view plaincopy

1. KPROCESSOR_MODE PreviousMode;
2. ULONG PID;
3. PreviousMode = ExGetPreviousMode();
4. // 如果非内核模式,就要开始检查IN的这些参数都否可读
5. if (PreviousMode != KernelMode)
6. {
7. try
8. {
9. ProbeForRead(ClientId, sizeof(CLIENT_ID), sizeof(ULONG));
10. ProbeForWrite(ClientId, sizeof(CLIENT_ID), sizeof(ULONG));
11. }
12. __except(EXCEPTION_EXECUTE_HANDLER)
13. {
14. return GetExceptionCode();
15. }
16. }

2. 校验指针是否为空，内存有效性：

[cpp] view plaincopy

1. // 效验ClientId是否为NULL
2. if (ARGUMENT_PRESENT(ClientId)
3. && MmIsAddressValid(ClientId))
4. {
5. //更安全的访问。
6. PID = (ULONG)ClientId->UniqueProcess;
7. KdPrint(("OpenProcess %d\r\n",PID));
8. }

3.深度校验字符串函数：

[cpp] view plaincopy

1. //深度的效验
2. BOOLEAN ValidateUnicodeString(PUNICODE_STRING usStr)
3. {
4. ULONG i;
5. __try
6. {
7. if (!MmIsAddressValid(usStr))
8. {
9. return FALSE;
10. }
11. if (usStr->Buffer == NULL || usStr->Length == 0)
12. {
13. return FALSE;
14. }
15. for (i = 0; i < usStr->Length; i++)
16. {
17. if (!MmIsAddressValid((PUCHAR)usStr->Buffer + i))
18. {
19. return FALSE;
20. }
21. }
22. }__except(EXCEPTION_EXECUTE_HANDLER){
23. }
24. return TRUE;
25. }

[cpp] view plaincopy

1. //验证一个WCHAR内容的指针是否可以访问
2. BOOLEAN ValidateWCHARString(WCHAR *pwzStr,ULONG_PTR Length)
3. {
4. ULONG i;
5. __try
6. {
7. //第一步判断指针和大小是否为NULL，是的话就没必要验证了
8. if  (pwzStr == NULL || Length == 0)
9. {
10. return FALSE;
11. }
12. //以length长度循环检查指针pwzStr里面的值
13. for (i = 0; i < Length; i++)
14. {
15. //检查是否可以访问。
16. if  (!MmIsAddressValid((PUCHAR)pwzStr + i))
17. {
18. //只要有一个字节是不可读取
19. return FALSE;
20. }
21. }
22. }
23. __except(EXCEPTION_EXECUTE_HANDLER)
24. { //触发了异常
25. return FALSE;
26. }
27. return TRUE;
28. }