开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 进程与线程 > 查看源码
SysUtils.cpp
资源名称：基于TCP-IP多线程文件传输.rar　[点击查看]
上传用户：fengliqin8
上传日期：2007-04-13
资源大小：28k
文件大小：7k
源码类别：

进程与线程

开发平台：

Visual C++

SysUtils.cpp：源码内容
  1. //
  2. // 系统扩展
  3. // 最后更新: 2003.08.18
  4. // 历史:
  5. // 1. create by alone on 2003.08.18
  6. // 2. modified by alone on 2004.06.14: 添加关闭Windows函数
  7. //
  8. #include "stdafx.h"
  9. #include "SysUtils.h"
  11. #pragma comment(lib, "shlwapi.lib")
  13. // 判断文件是否存在
  14. BOOL IsFileExists(char *pszPathName)
  15. {
  16. WIN32_FIND_DATA FindFileData;
  17. HANDLE hFind;
  19. hFind = FindFirstFile(pszPathName, &FindFileData);
  20. if(hFind == INVALID_HANDLE_VALUE)
  21. return FALSE;
  22. else
  23. {
  24. FindClose(hFind);
  25. return TRUE;
  26. }
  27. }
  29. // 提取文件路径
  30. char *ExtractFilePath(char *pszSrc, char *pszDst)
  31. {
  32. int i;
  34. ASSERT(pszSrc != NULL);
  35. ASSERT(pszDst != NULL);
  36. int nLen = strlen(pszSrc);
  37. ASSERT(nLen < MAX_PATH);
  40. for(i = nLen - 1; i >= 0; i-- )
  41. {
  42. if(pszSrc[i] == ':' || pszSrc[i] == '\')
  43. {
  44. break;
  45. }
  46. }
  48. if(i == -1)
  49. {
  50. pszDst[0] = 0;
  51. }
  52. else
  53. {
  54. strncpy(pszDst, pszSrc, i + 1);
  55. pszDst[i + 1] = 0; // zero terminated;
  56. }
  58. return pszDst;
  59. }
  61. // 创建多层目录
  62. BOOL ForceDirectories(char *pszDir)
  63. {
  64. // 验证字符串的合法性
  65. ASSERT(pszDir != NULL);
  66. int nLen = strlen(pszDir);
  67. if(nLen == 0 || nLen >= MAX_PATH) return FALSE;
  69. char pszCurDir[MAX_PATH]; // 当前需要创建的目录
  70. char pszUpDir[MAX_PATH]; // 更上一层的目录
  72. // 去尾
  73. strcpy(pszCurDir, pszDir);
  74. if(pszCurDir[nLen - 1] == '\')
  75. {
  76. pszCurDir[nLen -1 ] = 0;
  77. nLen-- ;
  78. }
  79.   
  80. if (nLen < 3 || PathFileExists(pszCurDir) || strcmp(ExtractFilePath(pszCurDir, pszUpDir), pszCurDir) == 0)
  81. {
  82. return TRUE;
  83. }
  85.   return  ForceDirectories(pszUpDir) && CreateDirectory(pszCurDir, NULL);
  86. }
  88. // 重新启动Windows 9x
  89. BOOL RebootWin9x()
  90. {
  91. ExitWindowsEx(EWX_REBOOT, 0);
  92. return TRUE;
  93. }
  95. // 重新启动Windows NT/2000/XP
  96. BOOL RebootWinNT()
  97. {
  98. HANDLE hToken = 0, hThread = 0;
  99. DWORD dwRtn = 0;
  100. hThread =(HANDLE)AfxGetApp()->m_hThread;
  102. TOKEN_PRIVILEGES tp;
  103. LUID luid;
  104. tp.PrivilegeCount = 1;
  106. if (!::OpenProcessToken(GetCurrentProcess(),
  107. TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
  108. {
  109. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  110. return FALSE;
  111. }
  112. else
  113. {
  114. if (!LookupPrivilegeValue(NULL, // lookup privilege on local system
  115. SE_SHUTDOWN_NAME, // privilege to lookup 
  116. &luid))
  117. {
  118. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  119. CloseHandle(hToken);
  120. return FALSE;
  121. }
  122. else
  123. {
  124. // receives LUID of privilege
  125. tp.PrivilegeCount = 1;
  126. tp.Privileges[0].Luid = luid;
  127. tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
  129. // Enable the privilege or disable all privileges.
  130. AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), &tp, &dwRtn);
  131. if (GetLastError() == ERROR_SUCCESS)
  132. {
  133. ExitWindowsEx(EWX_REBOOT, 0);
  134. CloseHandle(hToken);
  135. return TRUE;
  136. }
  137. else
  138. {
  139. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  140. CloseHandle(hToken);
  141. return FALSE;
  142. }
  143. }
  144. }
  145. }
  147. // 重新启动操作系统
  148. BOOL RebootWindows()
  149. {
  150. // 获取操作系统版本
  151. DWORD dwVersion = GetVersion();
  152. if (dwVersion < 0x80000000) // Windows NT/2000/XP
  153. {
  154. return RebootWinNT();
  155. }
  156. else // Windows 98
  157. {
  158. return RebootWin9x();
  159. }
  160. }
  162. // 设置程序是否在操作系统启动后自动运行
  163. void SetAutoRun(BOOL bEnable)
  164. {
  165. HKEY hKey;
  167. // 打开键
  168. if(RegOpenKey(HKEY_LOCAL_MACHINE, "Software\Microsoft\Windows\CurrentVersion\Run", &hKey) !=ERROR_SUCCESS)
  169. return;
  171. if(bEnable) //启动时运行,设置键值
  172. {
  173. char szModule[MAX_PATH];
  174. GetModuleFileName(NULL, szModule, MAX_PATH);
  175. RegSetValueEx(hKey, _T("VLPRModule"), 0, REG_SZ, (CONST BYTE *)szModule, strlen(szModule));
  176. }
  177. else //删除键值
  178. {
  179. RegDeleteValue(hKey, _T("VLPRModule"));
  180. }
  181. RegCloseKey(hKey);
  182. }
  185. // 关闭Windows98
  186. BOOL ShutDownWin98()
  187. {
  188. ExitWindowsEx(EWX_SHUTDOWN, 0);
  189. return TRUE;
  190. }
  192. // 关闭Windows NT/2000/XP
  193. BOOL ShutDownWinNT()
  194. {
  195. HANDLE hToken = 0, hThread = 0;
  196. DWORD dwRtn = 0;
  197. hThread =(HANDLE)AfxGetApp()->m_hThread;
  199. TOKEN_PRIVILEGES tp;
  200. LUID luid;
  201. tp.PrivilegeCount = 1;
  203. if (!::OpenProcessToken(GetCurrentProcess(),
  204. TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
  205. {
  206. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  207. return FALSE;
  208. }
  209. else
  210. {
  211. if (!LookupPrivilegeValue(NULL, // lookup privilege on local system
  212. SE_SHUTDOWN_NAME, // privilege to lookup 
  213. &luid))
  214. {
  215. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  216. CloseHandle(hToken);
  217. return FALSE;
  218. }
  219. else
  220. {
  221. // receives LUID of privilege
  222. tp.PrivilegeCount = 1;
  223. tp.Privileges[0].Luid = luid;
  224. tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
  226. // Enable the privilege or disable all privileges.
  227. AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), &tp, &dwRtn);
  228. if (GetLastError() == ERROR_SUCCESS)
  229. {
  230. ExitWindowsEx(EWX_SHUTDOWN, 0);
  231. CloseHandle(hToken);
  232. return TRUE;
  233. }
  234. else
  235. {
  236. // MessageBox("Current proccess can't apply for the reboot priviledge!");
  237. CloseHandle(hToken);
  238. return FALSE;
  239. }
  240. }
  241. }
  242. }
  244. // 删除文件
  245. BOOL DeleteFileEx(char *szPathName, BOOL bAllowUndo)
  246. {
  247. SHFILEOPSTRUCT fo;
  248. char chPathName[MAX_PATH + 1];
  250. memset(chPathName, 0, MAX_PATH + 1);
  251. strcpy(chPathName, szPathName);
  253. memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
  254. fo.wFunc = FO_DELETE;
  255. fo.pFrom = chPathName;
  256. if(bAllowUndo)
  257. fo.fFlags = FOF_ALLOWUNDO;
  258. fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
  259. fo.fAnyOperationsAborted = FALSE;
  261. return (SHFileOperation(&fo) == 0);
  262. }
  264. BOOL RenameFileEx(char *szOldPathName, char *szNewPathName)
  265. {
  266. SHFILEOPSTRUCT fo;
  267. char chOldPathName[MAX_PATH + 1];
  268. char chNewPathName[MAX_PATH + 1];
  270. memset(chOldPathName, 0, MAX_PATH + 1);
  271. strcpy(chOldPathName, szOldPathName);
  273. memset(chNewPathName, 0, MAX_PATH + 1);
  274. strcpy(chNewPathName, szNewPathName);
  276. memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
  277. fo.wFunc = FO_RENAME;
  278. fo.pFrom = chOldPathName;
  279. fo.pTo = chNewPathName;
  280. fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
  281. fo.fAnyOperationsAborted = FALSE;
  283. return (SHFileOperation(&fo) == 0);
  284. }
  286. BOOL MoveFileEx(char *szSrcPathName, char *szDstPathName)
  287. {
  288. SHFILEOPSTRUCT fo;
  289. char chSrcPathName[MAX_PATH + 1];
  290. char chDstPathName[MAX_PATH + 1];
  292. memset(chSrcPathName, 0, MAX_PATH + 1);
  293. strcpy(chSrcPathName, szSrcPathName);
  295. memset(chDstPathName, 0, MAX_PATH + 1);
  296. strcpy(chDstPathName, szDstPathName);
  298. memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
  299. fo.wFunc = FO_MOVE;
  300. fo.pFrom = chSrcPathName;
  301. fo.pTo = chDstPathName;
  302. fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
  303. fo.fAnyOperationsAborted = FALSE;
  305. return (SHFileOperation(&fo) == 0);
  306. }
  308. BOOL CopyFileEx(char *szSrcPathName, char *szDstPathName)
  309. {
  310. SHFILEOPSTRUCT fo;
  311. char chSrcPathName[MAX_PATH + 1];
  312. char chDstPathName[MAX_PATH + 1];
  314. memset(chSrcPathName, 0, MAX_PATH + 1);
  315. strcpy(chSrcPathName, szSrcPathName);
  317. memset(chDstPathName, 0, MAX_PATH + 1);
  318. strcpy(chDstPathName, szDstPathName);
  320. memset(&fo, 0, sizeof(SHFILEOPSTRUCT));
  321. fo.wFunc = FO_COPY;
  322. fo.pFrom = chSrcPathName;
  323. fo.pTo = chDstPathName;
  324. fo.fFlags = FOF_NOCONFIRMATION | FOF_SILENT;
  325. fo.fAnyOperationsAborted = FALSE;
  327. return (SHFileOperation(&fo) == 0);
  328. }
  330. // 判断字符串是否是合法的文件名
  331. // 文件名不能包含的字符有9个字符/:*?"<>|
  332. BOOL IsLegalFileName(char *szFileName)
  333. {
  334. return (strpbrk(szFileName, "\/:*?"<>|") == NULL);
  335. }