www.pudn.com > 多线程socket文件传输.rar > SysUtils.cpp
//
// 系统扩展
// 最后更新: 2003.08.18
// 历史:
// 1. create by alone on 2003.08.18
// 2. modified by alone on 2004.06.14: 添加关闭Windows函数
//
#include "stdafx.h"
#include "SysUtils.h"
#pragma comment(lib, "shlwapi.lib")
// 判断文件是否存在
BOOL IsFileExists(char *pszPathName)
{
WIN32_FIND_DATA FindFileData;
HANDLE hFind;
hFind = FindFirstFile(pszPathName, &FindFileData);
if(hFind == INVALID_HANDLE_VALUE)
return FALSE;
else
{
FindClose(hFind);
return TRUE;
}
}
// 提取文件路径
char *ExtractFilePath(char *pszSrc, char *pszDst)
{
int i;
ASSERT(pszSrc != NULL);
ASSERT(pszDst != NULL);
int nLen = strlen(pszSrc);
ASSERT(nLen < MAX_PATH);
for(i = nLen - 1; i >= 0; i-- )
{
if(pszSrc[i] == ':' || pszSrc[i] == '\\')
{
break;
}
}
if(i == -1)
{
pszDst[0] = 0;
}
else
{
strncpy(pszDst, pszSrc, i + 1);
pszDst[i + 1] = 0; // zero terminated;
}
return pszDst;
}
// 创建多层目录
BOOL ForceDirectories(char *pszDir)
{
// 验证字符串的合法性
ASSERT(pszDir != NULL);
int nLen = strlen(pszDir);
if(nLen == 0 || nLen >= MAX_PATH) return FALSE;
char pszCurDir[MAX_PATH]; // 当前需要创建的目录
char pszUpDir[MAX_PATH]; // 更上一层的目录
// 去尾
strcpy(pszCurDir, pszDir);
if(pszCurDir[nLen - 1] == '\\')
{
pszCurDir[nLen -1 ] = 0;
nLen-- ;
}
if (nLen < 3 || PathFileExists(pszCurDir) || strcmp(ExtractFilePath(pszCurDir, pszUpDir), pszCurDir) == 0)
{
return TRUE;
}
return ForceDirectories(pszUpDir) && CreateDirectory(pszCurDir, NULL);
}
// 重新启动Windows 9x
BOOL RebootWin9x()
{
ExitWindowsEx(EWX_REBOOT, 0);
return TRUE;
}
// 重新启动Windows NT/2000/XP
BOOL RebootWinNT()
{
HANDLE hToken = 0, hThread = 0;
DWORD dwRtn = 0;
hThread =(HANDLE)AfxGetApp()->m_hThread;
TOKEN_PRIVILEGES tp;
LUID luid;
tp.PrivilegeCount = 1;
if (!::OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
return FALSE;
}
else
{
if (!LookupPrivilegeValue(NULL, // lookup privilege on local system
SE_SHUTDOWN_NAME, // privilege to lookup
&luid))
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
CloseHandle(hToken);
return FALSE;
}
else
{
// receives LUID of privilege
tp.PrivilegeCount = 1;
tp.Privileges[0].Luid = luid;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
// Enable the privilege or disable all privileges.
AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), &tp, &dwRtn);
if (GetLastError() == ERROR_SUCCESS)
{
ExitWindowsEx(EWX_REBOOT, 0);
CloseHandle(hToken);
return TRUE;
}
else
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
CloseHandle(hToken);
return FALSE;
}
}
}
}
// 重新启动操作系统
BOOL RebootWindows()
{
// 获取操作系统版本
DWORD dwVersion = GetVersion();
if (dwVersion < 0x80000000) // Windows NT/2000/XP
{
return RebootWinNT();
}
else // Windows 98
{
return RebootWin9x();
}
}
// 设置程序是否在操作系统启动后自动运行
void SetAutoRun(BOOL bEnable)
{
HKEY hKey;
// 打开键
if(RegOpenKey(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Windows\\CurrentVersion\\Run", &hKey) !=ERROR_SUCCESS)
return;
if(bEnable) //启动时运行,设置键值
{
char szModule[MAX_PATH];
GetModuleFileName(NULL, szModule, MAX_PATH);
RegSetValueEx(hKey, _T("VLPRModule"), 0, REG_SZ, (CONST BYTE *)szModule, strlen(szModule));
}
else //删除键值
{
RegDeleteValue(hKey, _T("VLPRModule"));
}
RegCloseKey(hKey);
}
// 关闭Windows98
BOOL ShutDownWin98()
{
ExitWindowsEx(EWX_SHUTDOWN, 0);
return TRUE;
}
// 关闭Windows NT/2000/XP
BOOL ShutDownWinNT()
{
HANDLE hToken = 0, hThread = 0;
DWORD dwRtn = 0;
hThread =(HANDLE)AfxGetApp()->m_hThread;
TOKEN_PRIVILEGES tp;
LUID luid;
tp.PrivilegeCount = 1;
if (!::OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
return FALSE;
}
else
{
if (!LookupPrivilegeValue(NULL, // lookup privilege on local system
SE_SHUTDOWN_NAME, // privilege to lookup
&luid))
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
CloseHandle(hToken);
return FALSE;
}
else
{
// receives LUID of privilege
tp.PrivilegeCount = 1;
tp.Privileges[0].Luid = luid;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
// Enable the privilege or disable all privileges.
AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), &tp, &dwRtn);
if (GetLastError() == ERROR_SUCCESS)
{
ExitWindowsEx(EWX_SHUTDOWN, 0);
CloseHandle(hToken);
return TRUE;
}
else
{
// MessageBox("Current proccess can't apply for the reboot priviledge!");
CloseHandle(hToken);
return FALSE;
}
}
}
}
// 删除文件
BOOL DeleteFileEx(char *szPathName, BOOL bAllowUndo)
{
SHFILEOPSTRUCT fo;
char chPathName[MAX_PATH + 1];
memset(chPathName, 0, MAX_PATH + 1);
strcpy(chPathName, szPathName);
memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
fo.wFunc = FO_DELETE;
fo.pFrom = chPathName;
if(bAllowUndo)
fo.fFlags = FOF_ALLOWUNDO;
fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
fo.fAnyOperationsAborted = FALSE;
return (SHFileOperation(&fo) == 0);
}
BOOL RenameFileEx(char *szOldPathName, char *szNewPathName)
{
SHFILEOPSTRUCT fo;
char chOldPathName[MAX_PATH + 1];
char chNewPathName[MAX_PATH + 1];
memset(chOldPathName, 0, MAX_PATH + 1);
strcpy(chOldPathName, szOldPathName);
memset(chNewPathName, 0, MAX_PATH + 1);
strcpy(chNewPathName, szNewPathName);
memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
fo.wFunc = FO_RENAME;
fo.pFrom = chOldPathName;
fo.pTo = chNewPathName;
fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
fo.fAnyOperationsAborted = FALSE;
return (SHFileOperation(&fo) == 0);
}
BOOL MoveFileEx(char *szSrcPathName, char *szDstPathName)
{
SHFILEOPSTRUCT fo;
char chSrcPathName[MAX_PATH + 1];
char chDstPathName[MAX_PATH + 1];
memset(chSrcPathName, 0, MAX_PATH + 1);
strcpy(chSrcPathName, szSrcPathName);
memset(chDstPathName, 0, MAX_PATH + 1);
strcpy(chDstPathName, szDstPathName);
memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
fo.wFunc = FO_MOVE;
fo.pFrom = chSrcPathName;
fo.pTo = chDstPathName;
fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
fo.fAnyOperationsAborted = FALSE;
return (SHFileOperation(&fo) == 0);
}
BOOL CopyFileEx(char *szSrcPathName, char *szDstPathName)
{
SHFILEOPSTRUCT fo;
char chSrcPathName[MAX_PATH + 1];
char chDstPathName[MAX_PATH + 1];
memset(chSrcPathName, 0, MAX_PATH + 1);
strcpy(chSrcPathName, szSrcPathName);
memset(chDstPathName, 0, MAX_PATH + 1);
strcpy(chDstPathName, szDstPathName);
memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
fo.wFunc = FO_COPY;
fo.pFrom = chSrcPathName;
fo.pTo = chDstPathName;
fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
fo.fAnyOperationsAborted = FALSE;
return (SHFileOperation(&fo) == 0);
}
// 判断字符串是否是合法的文件名
// 文件名不能包含的字符有9个字符\/:*?"<>|
BOOL IsLegalFileName(char *szFileName)
{
return (strpbrk(szFileName, "\\/:*?\"<>|") == NULL);
}