基于Visual C++之Windows核心编程代码分析（2）实现Windows用户管理

转自：http://blog.csdn.net/itcastcpp/article/details/7078745

程序比较简单，主要是学习这几个API

在Windows编程中间，我们经常需要添加用户，删除用户，添加用户组等等功能，那些功能如何实现呢，我们请见实现代码与注释分析

 /* UNICODE */
#ifndef UNICODE
#define UNICODE
#endif
/* 头文件 */
#include <stdio.h>
#include <assert.h>
#include <windows.h>
#include <lm.h>

/*************************************
* AddUser
* 功能 增加用户
* 参数 szServerName，主机名，如果为本机增加用户，设置为NULL
* szUserName，用户名
* szPassword，密码
**************************************/
int AddUser(LPWSTR szServerName,
LPWSTR szUserName,
LPWSTR szPassword)
{
USER_INFO_1 ui;
DWORD dwLevel = 1; // 使用 USER_INFO_1 作为参数
DWORD dwError = 0;
NET_API_STATUS nStatus;
// 填充 USER_INFO_1
ui.usri1_name = szUserName; // 用户名
ui.usri1_password = szPassword; // 密码
ui.usri1_priv = USER_PRIV_USER; // privilege
ui.usri1_home_dir = NULL;
ui.usri1_comment = NULL;
ui.usri1_flags = UF_SCRIPT;
ui.usri1_script_path = NULL;
// 调用 NetUserAdd 增加用户
nStatus = NetUserAdd(szServerName,
dwLevel,
(LPBYTE)&ui,
&dwError);

// 判断结果
if (nStatus == NERR_Success)
{
wprintf(stderr, L"User %s has been successfully added on %s\n",
szUserName, szServerName);
}
else
{
fprintf(stderr, "A system error has occurred: %d\n", nStatus);
}
return 0;
}

/*************************************
* AddUserToGroup
* 功能 为用户组增加用户
* 参数 szServerName，主机名，如果为本机，设置为NULL
* szUserName，用户名
* szGroup，用户组名
**************************************/
int AddUserToGroup(LPWSTR szServerName,
LPWSTR szUserName,
LPWSTR szGroup)
{
NET_API_STATUS nStatus;
// 调用 NetGroupAddUser
nStatus = NetGroupAddUser(
szServerName,
szGroup,
szUserName
);

// 判断结果
if (nStatus == NERR_Success)
fwprintf(stderr, L"User %s has been successfully added on %s\n",
szUserName, szServerName);

else
fprintf(stderr, "NetGroupAddUser A system error has occurred: %d\n", nStatus);
return 0;
}

/*************************************
* DelUser
* 功能 删除用户
* 参数 szServerName，主机名，如果为本机，设置为NULL
* szUserName，用户名
**************************************/
int DelUser(LPWSTR szServerName, LPWSTR szUserName)
{
DWORD dwError = 0;
NET_API_STATUS nStatus;

// 调用 NetUserDel 删除用户
nStatus = NetUserDel(szServerName, szUserName);
// 判断并显示结果
if (nStatus == NERR_Success)
fwprintf(stderr, L"User %s has been successfully deleted on %s\n",
szUserName, szServerName);
else
fprintf(stderr, "A system error has occurred: %d\n", nStatus);

return 0;

}

/*************************************
* int ListUsers(LPWSTR pszServerName)
* 功能 列举用户
* 参数 szServerName，主机名，如果为本机，设置为NULL
**************************************/
int ListUsers(LPWSTR pszServerName)
{
LPUSER_INFO_0 pBuf = NULL;
LPUSER_INFO_0 pTmpBuf;
DWORD dwLevel = 0;
DWORD dwPrefMaxLen = MAX_PREFERRED_LENGTH;
DWORD dwEntriesRead = 0;
DWORD dwTotalEntries = 0;
DWORD dwResumeHandle = 0;
DWORD i;
DWORD dwTotalCount = 0;
NET_API_STATUS nStatus;

// 循环，直到可以成功调用 NetUserEnum
do
{
// 调用NetUserEnum函数
nStatus = NetUserEnum(pszServerName,
dwLevel,// 这里设置为0，使用 LPUSER_INFO_0 返回结果
FILTER_NORMAL_ACCOUNT, // 只列举“正常”类型的用户
(LPBYTE*)&pBuf,// LPUSER_INFO_0 保存返回结果
// MAX_PREFERRED_LENGTH，内存由API分配，需要在之后调用NetApiBufferFree释放
dwPrefMaxLen,
&dwEntriesRead,// 读了的 Entries
&dwTotalEntries,// 一共的 Entries
&dwResumeHandle);
// 判断是否成功
if ((nStatus == NERR_Success) || (nStatus == ERROR_MORE_DATA))
{
if ((pTmpBuf = pBuf) != NULL)
{
// 循环读取用户信息
for (i = 0; (i < dwEntriesRead); i++)
{
assert(pTmpBuf != NULL);

if (pTmpBuf == NULL)
{
fprintf(stderr, "An access violation has occurred\n");
break;
}
// 输出
wprintf(L"\t-- %s\n", pTmpBuf->usri0_name);
// 下一个
pTmpBuf++;
dwTotalCount++;
}
}
}
else
fprintf(stderr, "A system error has occurred: %d\n", nStatus);
// 释放内存
if (pBuf != NULL)
{
NetApiBufferFree(pBuf);
pBuf = NULL;
}
}
while (nStatus == ERROR_MORE_DATA); // end do

// 释放内存
if (pBuf != NULL)
NetApiBufferFree(pBuf);

fprintf(stderr, "Total of %d users\n\n", dwTotalCount);

return 0;
}

/*************************************
* int ListGroup(LPWSTR pszServerName)
* 功能 列举用户组
* 参数 szServerName，主机名，如果为本机，设置为NULL
**************************************/
int ListGroup(LPWSTR pszServerName)
{

DWORD dwLevel = 0;
DWORD dwPrefMaxLen = MAX_PREFERRED_LENGTH;
DWORD dwEntriesRead = 0;
DWORD dwTotalEntries = 0;
DWORD dwResumeHandle = 0;
DWORD i;
DWORD dwTotalCount = 0;
NET_API_STATUS nStatus;

LPLOCALGROUP_INFO_0 pBuf = NULL;
LPLOCALGROUP_INFO_0 pTmpBuf;

do // begin do
{
// 调用NetLocalGroupEnum 参数设置与NetLocalGroup类似
nStatus = NetLocalGroupEnum(
pszServerName,
0,
(LPBYTE*)&pBuf,
dwPrefMaxLen,
&dwEntriesRead,
&dwTotalEntries,
&dwResumeHandle);
// 判断结果
if ((nStatus == NERR_Success) || (nStatus == ERROR_MORE_DATA))
{
if ((pTmpBuf = pBuf) != NULL)
{
// 循环输出
for (i = 0; (i < dwEntriesRead); i++)
{
assert(pTmpBuf != NULL);

if (pTmpBuf == NULL)
{
fprintf(stderr, "An access violation has occurred\n");
break;
}

wprintf(L"\t-- %s\n", pTmpBuf->lgrpi0_name);
pTmpBuf++;
dwTotalCount++;
}
}
}

else
fprintf(stderr, "A system error has occurred: %d\n", nStatus);
// 释放内存
if (pBuf != NULL)
{
NetApiBufferFree(pBuf);
pBuf = NULL;
}
}

while (nStatus == ERROR_MORE_DATA); // end do

if (pBuf != NULL)
NetApiBufferFree(pBuf);

fprintf(stderr, "Total of %d groups\n\n", dwTotalCount);

return 0;
}

/*************************************
* ShowUsersInfo
* 功能 显示指定用户的信息
* 参数 szServerName，主机名，如果为本机，设置为NULL
* pszUserName，用户名
**************************************/
int ShowUsersInfo(LPWSTR pszServerName,LPWSTR pszUserName)
{

DWORD dwLevel = 4;// 使用 LPUSER_INFO_4 返回结果
LPUSER_INFO_4 pBuf = NULL;
NET_API_STATUS nStatus;

nStatus = NetUserGetInfo(pszServerName,
pszUserName,
dwLevel, // pBuf参数类型
(LPBYTE *)&pBuf);

// 判断并输出结果
if (nStatus == NERR_Success)
{
if (pBuf != NULL)
{
wprintf(L"\n\tAccount: %s\n", pBuf->usri4_name);
wprintf(L"\tComment: %s\n", pBuf->usri4_comment);
wprintf(L"\tUser comment: %s\n", pBuf->usri4_usr_comment);
wprintf(L"\tFull name: %s\n", pBuf->usri4_full_name);
wprintf(L"\tpriv: %d\n", pBuf->usri4_priv);
}
}

else
fprintf(stderr, "A system error has occurred: %d\n", nStatus);
// 释放内存
if (pBuf != NULL)
NetApiBufferFree(pBuf);
return 0;

}

/*************************************
* wmain
* 功能 入口函数，根据参数判断需要调用的功能函数
* 参数 参见usage输出
**************************************/
int __cdecl wmain(int ac, wchar_t * av[])
{

if (ac == 4 && lstrcmpW( av[1], L"-a") == 0)
{
AddUser(NULL, av[2], av[3]);
}

else if (ac == 4 && lstrcmpW( av[1], L"-g") == 0)
{
AddUserToGroup(NULL, av[2], av[3]);
}
else if (ac == 3 && lstrcmpW( av[1], L"-i") == 0)
{
ShowUsersInfo(NULL, av[2]);
}
else if (ac == 2 && lstrcmpW( av[1], L"-i") == 0)
{
ListUsers(NULL);
ListGroup(NULL);
}
else if (ac == 3 && lstrcmpW( av[1], L"-d") == 0)
{
DelUser(NULL, av[2]);
}
else
{
printf("usage: \n"
"\t %ws -a <username> <password> to add a user\n"
"\t %ws -g <username> <group> add a user to a group"
"\t %ws -i <username> to show user info\n"
"\t %ws -d
4000
<username> to del a user\n",
av[0], av[0], av[0], av[0]);
}
return 0;
}