基于Visual C++之实现Windows用户管理

来源：互联网发布：宿舍床帘淘宝编辑：程序博客网时间：2024/05/22 05:10

[cpp] view plaincopyprint?

/* UNICODE */
#ifndef UNICODE
#define UNICODE
#endif
/* 头文件 */
#include <stdio.h>
#include <assert.h>
#include <windows.h>
#include <lm.h>
#include "dsgetdc.h"
#pragma comment(lib,"netapi32.lib")
/*************************************
* 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 <username> to del a user\n",
av[0], av[0], av[0], av[0]);
}
return 0;
}