这宏，我写的，怎么看怎么顺眼，能用宏，为什么要写函数呢，费事不讨好，而且还有可能影响效率

不过，实话实说，看着是有点low。。。

 

如果Windows的编译器能支持gcc的那种 ({}) ，的话，这个宏我能写得更完美，

就不用像最后，还得单独一行写宏，然后再起一行if __ret 了，

那样的话，代码会更好看。

 

其实，这样写还可能出现另一个问题，就是类型不兼容的警告，

这需要 typeof 关键字来解决，但是。我上哪找去。

 

 1 //    判断 __src 的长度 __srcLen 是否小于 __dstLen ，如果是的话，
 2 //    那么就用 __memType 和 __tag 重新申请一块内存，否则不动，
 3 //    不动、重新申请成功，__ret 返回 1
 4 //    申请失败，__ret 返回 0
 5 //    原始地址指针PP，原始内存长度PD，目标长度DW，内存页属性DW，TAG，返回值PD
 6 #define ComperMemoryAlloc(__src, __srcLen, __dstLen, __memType, __tag, __ret)            \
 7     do {                                                                                \
 8         LPVOID pv = NULL;                                                                \
 9         *(__ret) = 1;                                                                    \
10         if (*(__srcLen) < (__dstLen))                                                    \
11         {                                                                                \
12             pv = ExAllocatePoolWithTag((__memType), (__dstLen), (__tag));                \
13             if (pv)                                                                        \
14             {                                                                            \
15                 ExFreePoolWithTag(*(__src), (__tag));                                    \
16                 *(__srcLen) = (__dstLen);                                                \
17                 *(__src) = pv;                                                            \
18                 *(__ret) = 1;                                                            \
19             }                                                                            \
20             else                                                                        \
21             {                                                                            \
22                 *(__ret) = 0;                                                            \
23             }                                                                            \
24         }                                                                                \
25     } while (0)

 

不过，至于使用方法。。。

有点自己给自己找罪受的感觉

 1     DWORD dwRet = 0;
 2     ComperMemoryAlloc(&p, &dwLen, 200, NULL, 0, &dwRet);
 3     if (dwRet)
 4     {
 5 
 6     }
 7     else
 8     {
 9 
10     }