利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)
作者:Tuuzed(土仔)   发表于:2008年3月3日23:12:38 
版权声明:可以任意转载,转载时请务必以超链接形式标明文章原始出处和作者信息及本声明
http://www.cppblog.com/tuuzed/archive/2008/03/03/43631.html


    
    可能很多人都知道NT系统的query user命令,命令返回“使用者名称 工作阶段名称 识别码 状态 闲置时间 登入时间”。如图:

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)

    微软给出了获取终端会话的重要API(见Terminal Services API Functions),与获取当前终端会话功能有关的API有:WTSEnumerateSessionsWTSQuerySessionInformation

        WTSEnumerateSessions:顾名思义就是列出所有的Session,返回一个WTS_SESSION_INFO结构,结构存储了SessionId,WinStationName,State(包括Active、Disconnected等状态)。

        WTSQuerySessionInformation:这个和上面的API有些不同,它只能通过SessionId来查询Session的详细信息,可获取例如用于连接终端客户端工具的ClientName、ClientDirectory等,比WTSEnumerateSessions功能丰富。

    按照MSDN上说的,WTSQuerySessionInformation还可以获取IdleTime、LogonTime、IncomingBytes、OutgoingBytes等信息,可惜,标明是“This value is not used.”,要使用的话必须在Windows Server 2008和Windows Vista SP1下使用,局限性太大了。只好自己上Goolge上搜索一下了,在国外的论坛中,大部分人对于获取Idle Time都是说在WIN2008或VISTA才支持。那么WIN2000、2003里的query命令是怎么获得登入时间的?这里面肯定有什么没有公开的API在里面!果然,我找到了Guy Teverovsky的BLOG,它给出的答案(《Querying TS session idle time with C#》译文:《[翻译]利用C#获取终端服务(Terminal Services)会话的闲置时间》)和我预想的差不错——所要的信息在在Winsta.dll内的一个未公开API函数WinStationQueryInformationW返回的结构WINSTATIONQUERYINFORMATIONW里面。

    要想使用WinStationQueryInformationW必须知道其中两个重要的参数WinStationInformation(枚举类型)值和WINSTATIONINFORMATIONW结构内容。在VS2005对上述两个值有定义(winternl.h):

 

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)typedef enum _WINSTATIONINFOCLASS {
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    WinStationInformation = 8
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)} WINSTATIONINFOCLASS; 

 

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)typedef struct _WINSTATIONINFORMATIONW {
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    BYTE Reserved2[70];
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    ULONG LogonId;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    BYTE Reserved3[1140];
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)} WINSTATIONINFORMATIONW, * PWINSTATIONINFORMATIONW; 

 

    第一个值很清楚了,是8。而后一个结构,保留位达1140位,这里有太多未知的信息了。还好那位牛人给出了C#的定义,我把它转成C++的结构定义:

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)typedef struct _WINSTATIONQUERYINFORMATION
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time){
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    char Reserved1[72];
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    unsigned int SessionId;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    char Reserved2[4];
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    FILETIME ConnectTime;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    FILETIME DisconnectTime;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    FILETIME LastInputTime;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    FILETIME LogonTime;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    char Reserved3[1096];
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    FILETIME CurrentTime;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)} WINSTATIONQUERYINFORMATION, *PWINSTATIONQUERYINFORMATION; 

 

    定义完这个结构,工作已经有眉目了。下面就是载入Winsta.dll内那个未公开的API函数,顺便包装了一下:

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)BOOL WINAPI WinStationQueryInformation(HANDLE hServer, DWORD SessionId, DWORD InfoClass, LPVOID Buffer, DWORD BufferLength, LPDWORD Count)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time){
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    typedef BOOL (WINAPI *PROCPTR)(HANDLE, DWORD, DWORD, LPVOID, DWORD, LPDWORD);
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    static HMODULE hModule = NULL;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    static PROCPTR proc = NULL;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    hModule = LoadLibrary("winsta.dll");
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    if (hModule == NULL)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    {
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)        return FALSE;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    } 
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    if (proc == NULL)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    {
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)        proc = (PROCPTR) GetProcAddress(hModule, "WinStationQueryInformationW");
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    } 
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    if (proc == NULL)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    {
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)        return FALSE;
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    } 
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)    return proc(hServer, SessionId, InfoClass, Buffer, BufferLength, Count);
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)
利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)

 

    这样,只要直接调用自己的WinStationQueryInformation来间接调用DLL里面的WinStationQueryInformationW就可以了。登入时间Logon Time是可以直接获取的,而闲置时间的获取就要参考当前会话的状态了:如果会话是断开(Disconnected)状态,闲置时间=当前时间-断开时间(Idle Time = CurrentTime - DisconnectTime);如果会话是活动的(alive)状态,闲置时间=当前时间-最后输入时间(Idle Time = CurrentTime - LastInputTime)。

    已经做出来了一个DEMO,看截图:

 

利用未公开API获取终端会话闲置时间(Idle Time)和登入时间(Logon Time)

    这是我做的一个ROOKIT工具的截图,其中就包括有对终端会话的管理等。在这次编程后,给我感触最深的就是国外大虾解决问题的方法与国内的我们有很大的不同,他们善于从多角度来解决问题,独自解决问题后再总结,是利用网络资源而不是依赖网络资源。

上一篇:Mysql千万级数据快速分页查询方案


下一篇:linux一些杂项整理