悲情土仔一生

編程制作管理員登陸報警器

作者：Tuuzed(土仔)    
版權聲明：可以任意轉載，轉載時請務必以超鏈接形式標明文章原始出處和作者信息及本聲明。
http://www.shnenglu.com/tuuzed/archive/2008/03/16/44656.html

        某日，你“偶然”登陸到了一臺服務器上，在確認管理員不在后，開始在服務器上做起一些見不得人的事來。但，這時的你一定不是“心定定”，你或許是在想：萬一這時有用戶登陸進來怎么辦？如果有什么程序能在有用戶登陸進來時提醒一下，那該多好。

        “自己動手，豐衣足食”的時候又來了。搞個思路吧：WIN32系統離不開消息，有人登陸應該會給個消息；得到消息了，判斷是什么人登陸；根據登陸用戶的權限做出反應。

         用戶登陸系統給什么消息？查MSDN，系統會在用戶登陸或斷開時發出WM_WTSSESSION_CHANGE的消息，通過它附帶的參數，可以判斷出是登陸或是斷開。其中WTS_CONSOLE_CONNECT、WTS_REMOTE_CONNECT、WTS_SESSION_LOGON十分有用。

        如何通過消息判斷登陸用戶是誰？還是在WM_WTSSESSION_CHANGE消息中，帶有另一個參數：登陸者的session ID。將session ID傳入API函數WTSQuerySessionInformation中可以查詢到User Name。

        如何判斷用戶權限？獲取的User Name傳入API函數NetUserGetInfo中，利用帶出的USER_INFO_1結構中的usri1_priv得到權限標識。

        下面是利用C++Builder做demo的核心代碼：

        首先在Form建立時加入語句：WTSRegisterSessionNotification(frmMain->Handle, NOTIFY_FOR_ALL_SESSIONS)，告訴系統有Session消息時通知Form。記住要傳入的是主Form的Handle，而不是Application的Handle（BCB用戶應該知道區別）。

        然后在.H中重載void __fastcall WndProc(Messages::TMessage &Message);在.CPP中實現它：

 1void __fastcall TfrmMain::WndProc(Messages::TMessage &Message)
 2{
 3    if (Message.Msg==WM_WTSSESSION_CHANGE)
 4    {
 5        switch (Message.WParam)
 6        {
 7            case WTS_CONSOLE_CONNECT:
 8                //控制臺登陸，報警。
 9                break;
10            case WTS_CONSOLE_DISCONNECT:
11                break;
12            case WTS_REMOTE_CONNECT:
13                //有遠程連接，報警。
14                break;
15            case WTS_REMOTE_DISCONNECT:
16                break;
17            case WTS_SESSION_LOGON:
18                //用戶登陸了，報警。
19                break;
20            case WTS_SESSION_LOGOFF:
21                break;
22            case WTS_SESSION_LOCK:
23               break;
24            case WTS_SESSION_UNLOCK:
25                 break;
26            default:
27                //unknown
28        }
29    }
30} 

         在C++ Builder編制程序的時候，一定要在#include <vcl.h>前先定義#define _WIN32_WINNT 0x0501，要不是的話編譯時會提示找不到WM_WTSSESSION_CHANGE定義（具體原因是在winuser.h中，_WIN32_WINNT必須是大于 0x0501系統才定義了WM_WTSSESSION_CHANGE)。最后，在程序退出前，要告訴系統不要再發消息過來了：WTSUnRegisterSessionNotification。

        程序做出來后，我給它起名：Hacker RUN! 中文名：黑客快跑。