博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
WINDOWS下调用GetTokenInformation的奇怪之处--两次调用
阅读量:5923 次
发布时间:2019-06-19

本文共 4030 字,大约阅读时间需要 13 分钟。

就是用getLastErr可以得到错误号,同时,会将需要的长度写到参数里,再进行第二次调用,以此来节约内存空间。

神奇的长见识了。

相关说法如下:

======================

The error occurs because the buffer is insufficient. :-) It's not large enough for the content. Once again, I refer you to the documentation. If the function fails, the ReturnLength parameter is set to the size of the buffer needed, so you can allocate sufficient memory and call the function again. The general rule for many API calls that have variable buffer requirements is "Call the function once with a NULL buffer to determine the buffer size needed, allocate the memory, and then call it again to actually retrieve the information"

======================

The TOKEN_USER structure contains pointers (in particular, a pointer to a SID that itself has variable length). Those pointers have to point somewhere. The API function will expect a buffer big enough to hold not only the the TOKEN_USER structure, but also all the things that structure points to. The function tells you how much memory it needs for everything. It will all reside in adjacent memory.

======================

最近又在重新编写一些以前已经实现过的功能, 一来是温习之前学的一些东西, 二是希望多次重写一个功能的过程中提升自己现有的编程技术和水平, 今天这篇文章算是一个小的总结.

 
在写之前请容许我唠叨片刻, 文章中涉及的技术与代码示例可能早遍布在百度与GOOGLE上的各种原创或转载文章中, 所以我的内容多少也会与那些文章中的东西有所相同, 对于那些已经看过那些文章或者早已熟悉这些技术朋友可能会略显平庸, 不过这都算是自己对技术积累的心得吧, 见笑了, 那么言归正传了.
 
Win32 API中提供了GetTokenInformation与SetTokenInformation函数, 这2个函数别分是检索与设置一个访问令牌的信息, 关于访问令牌的具体细节有兴趣的朋友可以去MSDN上找, 因为与文章主题没有直接的关系, 所以就不在具体细说了.
以下是GetTokenInformation的API原型(如果想对这个函数的细节做更多了解, 可以去MSDN上查询它的详细信息)
 
WINADVAPI
BOOL
WINAPI
GetTokenInformation (
          _In_       HANDLE TokenHandle, 
       _In_       TOKEN_INFORMATION_CLASS TokenInformationClass, 
        _Out_opt_  LPVOID TokenInformation, _In_       
DWORD TokenInformationLength, _Out_     
        PDWORD ReturnLength
   );
总得来说, 这个函数算是Win32 API中比较有代表性的一个, 参数的含义也都是一目了然, 这里就不再过多獒述了(注: 关于GetTokenInformation参数含义这里倒是有一个趣事, 如果有兴趣看的朋友可以在文章最后找到我对它的说明), 有兴趣的朋友可以去MSDN上找这个API的具体文档, 下面我用一段代码示例来演示如何使用GetTokenInformation函数检索进程的所有特权信息
 
static PTOKEN_GROUPS GetProcessGroups(HANDLE hProcess)
{
HANDLE hToken;
PTOKEN_GROUPS pGroups = NULL;
DWORD neededSize;
 
// 这里会先获得一个进程访问令牌的句柄, 并且只允许该句柄具有信息查询的权限.
if (!OpenProcessToken(hProcess, TOKEN_QUERY, &hToken))
return NULL;
 
// 这里第一次调用GetTokenInformation是为了获取令牌信息需要的缓冲区的大小,
// 这是在使用Win32 API中的一个常用做法, 很多Win32 API都可以这样使用, 目的
// 是不用让程序员总是创建一个假设一定足够的缓冲区, 这在很多时候会造成空间上的浪费
neededSize = 0;
GetTokenInformation(hToken, TokenGroups, NULL, neededSize, &neededSize);
if (GetLastError() == ERROR_INSUFFICIENT_BUFFER && neededSize > 0)
{
pGroups = HeapAlloc(GetProcessHeap(), 0, neededSize);
if (pGroups != NULL)
{
// 这里是真正去检索特权信息, 并且检查成功或失败.
if (!GetTokenInformation(hToken, TokenGroups, pGroups, neededSize, &neededSize))
{
HeapFree(GetProcessHeap(), 0, pGroups);
pGroups = NULL;
}
}
}
CloseHandle(hToken);
return pGroups;
}
这个函数是用C编写的, 关键性的代码旁边都有注释说明其函数, 到这里文档的主要内容也就结束了, 不过值得多说一句的是, GetTokenInformation不仅可以获取进程的特权信息, 同时还能得到其它更多的信息, 比如进程所属的所有用户组, 进程拥有者以及虚拟化信息等, 而TOKEN_INFORMATION_CLASS会告诉GetTokenInformation你想要哪种信息.
 
关于GetTokenInformation函数参数的趣事
我在使用GetTokenInformation过程中发现了一个有趣的事情, 那就是Microsoft似乎在维护文档与Platform SDK头文件时出现了歧义, 
BOOL
WINAPI
GetTokenInformation (
    __in      HANDLE TokenHandle,
    __in      TOKEN_INFORMATION_CLASS TokenInformationClass,
    __out_bcount_part_opt(TokenInformationLength, *ReturnLength) LPVOID TokenInformation,
    __in      DWORD TokenInformationLength,
    __out_opt PDWORD ReturnLength
    );
此时ReturnLength这个参数带有_out_opt这样的修饰符, 它在WIN32 API中的文档约定是指被修饰的参数可以传一个NULL指针进去而不会任何问题, 最多也就是你不知道实际填充到TokenInformation缓冲区的令牌信息占用了多少字节, 不过有趣的是API的实现似乎是遵照了在MSDN官网文档中定义的形式, 也就是ReturnLength这个值必须指向一个有效的DWORD类型的变量, 所以如果你在调用GetTokenInformation时传递给ReturnLength是一个NULL值, 那么GetTokenInformation会返回失败, 并且调用GetLastError()会返回错误代码87(ERROR_INVALID_PARAMETER), 同样的问题可能不只是出现在GetTokenInformation函数中, 其它一些少数API可能也存在此类问题, 所以使用的时候还需要多加留意才行.
在MSDN的文档中ReturnLength这个参数被定义为"必选"参数, 这里所谓的"必选"一词是指它不是一个NULL指针值, 也就是不允许这样的调用行为GetTokenInformation(...., NULL);, 但如果细心的朋友在Microsoft Visual Stdio 2005中查看这个WinBase.h中这个函数的原型声明时会发现它的声明形式如下:
你可能感兴趣的文章
Java 泛型的使用限制
查看>>
DNS主配置文件的几个选项
查看>>
Liunx的安装详解
查看>>
[AAuto]批量添加按钮及注册命令演示
查看>>
c++中指针的理解demo
查看>>
代理键--surrogate keys
查看>>
Linux一条刚接触的命令该如何去学习它?
查看>>
在系统中复用HTTP Status来映射系统的状态
查看>>
Spring+SpringMVC+Mybatis整合,你可能遇到的问题
查看>>
微信收费的真相
查看>>
Spring的IOC原理
查看>>
JavaScript 正则表达式对象RegExp test方法慎用/g
查看>>
达观数据:文本大数据的机器学习自动分类方法
查看>>
帮助你生成分享和显示社交媒体网络按钮的jQuery插件 - #50C1AL
查看>>
华为:浮动静态路由
查看>>
Cisco ASA 5505配置详解
查看>>
【大数据干货】轻松处理每天2TB的日志数据,支撑运营团队进行大数据分析挖掘,随时洞察用户个性化需求。...
查看>>
专访佰腾科技大数据团队,谈专利大数据领域的挑战与实践
查看>>
聊聊springboot2的ScheduledTasksEndpoint
查看>>
dubbo 异步调用
查看>>