Global.cpp 28 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024
  1. #pragma once
  2. #include "stdafx.h"
  3. #include "Global.h"
  4. #include <tlhelp32.h>
  5. #include <shlwapi.h>
  6. #include <strsafe.h>
  7. #include <WinVer.h>
  8. #pragma comment(lib,"version.lib")
  9. namespace Global
  10. {
  11. //////////////////////////////////////////////////////////////////////////
  12. // 全局变量;
  13. TCHAR g_szIniFile[MAX_PATH] = {0};
  14. TCHAR g_szHostName[MAX_PATH] = {0}; // 主机计算机名;
  15. TCHAR g_szModulePath[MAX_PATH]; // 软件目录;
  16. TCHAR g_szModuleFileName[MAX_PATH]; // 软件名称;
  17. TCHAR g_szConnectString[MAX_PATH]; // DB连接串;
  18. TCHAR g_szBranchId[MAX_PATH]; // 本店加密锁域名;
  19. // 服务器信息;
  20. TCHAR g_szDBSource[MAX_PATH]; // 数据库源(服务所在IP或计算机名);
  21. TCHAR g_szDBSourcePort[MAX_PATH]; // 数据库源端口;
  22. DWORD g_dwDBServerPort; // 数据库源端口;
  23. TCHAR g_szDBAccount[MAX_PATH]; // 数据库登录用户;
  24. TCHAR g_szDBPassWord[MAX_PATH]; // 数据库登录密码;
  25. TCHAR g_szDBName[MAX_PATH]; // 数据库名称;
  26. DWORD g_dwDBPoolMaxCount = 1; // 数据库连接池最大对象数;
  27. DWORD g_dwDBPoolDef = 1; // 数据库连接池默认对象数;
  28. //////////////////////////////////////////////////////////////////////////
  29. static const int BEGIN_YEAR = 1901;
  30. static const int NUMBER_YEAR = 199;
  31. static const unsigned int LUNAR_YEARS[199] =
  32. {
  33. 0x04AE53, 0x0A5748, 0x5526BD, 0x0D2650, 0x0D9544, 0x46AAB9, 0x056A4D, 0x09AD42, 0x24AEB6, 0x04AE4A, //1901-1910
  34. 0x6A4DBE, 0x0A4D52, 0x0D2546, 0x5D52BA, 0x0B544E, 0x0D6A43, 0x296D37, 0x095B4B, 0x749BC1, 0x049754, //1911-1920
  35. 0x0A4B48, 0x5B25BC, 0x06A550, 0x06D445, 0x4ADAB8, 0x02B64D, 0x095742, 0x2497B7, 0x04974A, 0x664B3E, //1921-1930
  36. 0x0D4A51, 0x0EA546, 0x56D4BA, 0x05AD4E, 0x02B644, 0x393738, 0x092E4B, 0x7C96BF, 0x0C9553, 0x0D4A48, //1931-1940
  37. 0x6DA53B, 0x0B554F, 0x056A45, 0x4AADB9, 0x025D4D, 0x092D42, 0x2C95B6, 0x0A954A, 0x7B4ABD, 0x06CA51, //1941-1950
  38. 0x0B5546, 0x555ABB, 0x04DA4E, 0x0A5B43, 0x352BB8, 0x052B4C, 0x8A953F, 0x0E9552, 0x06AA48, 0x7AD53C, //1951-1960
  39. 0x0AB54F, 0x04B645, 0x4A5739, 0x0A574D, 0x052642, 0x3E9335, 0x0D9549, 0x75AABE, 0x056A51, 0x096D46, //1961-1970
  40. 0x54AEBB, 0x04AD4F, 0x0A4D43, 0x4D26B7, 0x0D254B, 0x8D52BF, 0x0B5452, 0x0B6A47, 0x696D3C, 0x095B50, //1971-1980
  41. 0x049B45, 0x4A4BB9, 0x0A4B4D, 0xAB25C2, 0x06A554, 0x06D449, 0x6ADA3D, 0x0AB651, 0x093746, 0x5497BB, //1981-1990
  42. 0x04974F, 0x064B44, 0x36A537, 0x0EA54A, 0x86B2BF, 0x05AC53, 0x0AB647, 0x5936BC, 0x092E50, 0x0C9645, //1991-2000
  43. 0x4D4AB8, 0x0D4A4C, 0x0DA541, 0x25AAB6, 0x056A49, 0x7AADBD, 0x025D52, 0x092D47, 0x5C95BA, 0x0A954E, //2001-2010
  44. 0x0B4A43, 0x4B5537, 0x0AD54A, 0x955ABF, 0x04BA53, 0x0A5B48, 0x652BBC, 0x052B50, 0x0A9345, 0x474AB9, //2011-2020
  45. 0x06AA4C, 0x0AD541, 0x24DAB6, 0x04B64A, 0x69573D, 0x0A4E51, 0x0D2646, 0x5E933A, 0x0D534D, 0x05AA43, //2021-2030
  46. 0x36B537, 0x096D4B, 0xB4AEBF, 0x04AD53, 0x0A4D48, 0x6D25BC, 0x0D254F, 0x0D5244, 0x5DAA38, 0x0B5A4C, //2031-2040
  47. 0x056D41, 0x24ADB6, 0x049B4A, 0x7A4BBE, 0x0A4B51, 0x0AA546, 0x5B52BA, 0x06D24E, 0x0ADA42, 0x355B37, //2041-2050
  48. 0x09374B, 0x8497C1, 0x049753, 0x064B48, 0x66A53C, 0x0EA54F, 0x06B244, 0x4AB638, 0x0AAE4C, 0x092E42, //2051-2060
  49. 0x3C9735, 0x0C9649, 0x7D4ABD, 0x0D4A51, 0x0DA545, 0x55AABA, 0x056A4E, 0x0A6D43, 0x452EB7, 0x052D4B, //2061-2070
  50. 0x8A95BF, 0x0A9553, 0x0B4A47, 0x6B553B, 0x0AD54F, 0x055A45, 0x4A5D38, 0x0A5B4C, 0x052B42, 0x3A93B6, //2071-2080
  51. 0x069349, 0x7729BD, 0x06AA51, 0x0AD546, 0x54DABA, 0x04B64E, 0x0A5743, 0x452738, 0x0D264A, 0x8E933E, //2081-2090
  52. 0x0D5252, 0x0DAA47, 0x66B53B, 0x056D4F, 0x04AE45, 0x4A4EB9, 0x0A4D4C, 0x0D1541, 0x2D92B5 //2091-2099
  53. };
  54. //////////////////////////////////////////////////////////////////////////
  55. BOOL IsValidDate(IN LPCTSTR lpDate) // 判断公历;
  56. {
  57. if (lpDate == NULL || _tcscmp(lpDate, _T("")) == 0)
  58. return FALSE;
  59. CString strDate = lpDate;
  60. if (strDate.IsEmpty()) return FALSE;
  61. // 判断日期合法性;
  62. static int nYear = 0;
  63. static int nMonth = 0;
  64. static int nDay = 0;
  65. nYear = nMonth = nDay = 0;
  66. nYear = _ttoi(strDate.Left(4));
  67. nMonth = _ttoi(strDate.Mid(5, 2));
  68. nDay = _ttoi(strDate.Mid(8, 2));
  69. CONST static INT nNormalMonth[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  70. CONST static INT nLeapMonth[12] = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  71. if ((nDay <= 0) || (nDay > 31) || (nMonth <= 0) || (nMonth > 12)) return FALSE;
  72. if ((nYear % 400 == 0) || (nYear % 4 == 0 && nYear % 100 != 0))
  73. {
  74. if (nDay > nLeapMonth[nMonth - 1]) return FALSE;
  75. }
  76. else
  77. {
  78. if (nDay > nNormalMonth[nMonth - 1]) return FALSE;
  79. }
  80. return TRUE;
  81. }
  82. BOOL IsValidLunanrDate(IN LPCTSTR lpDate) // 判断农历;
  83. {
  84. if (lpDate == NULL || _tcscmp(lpDate, _T("")) == 0)
  85. return FALSE;
  86. CString strDate = lpDate;
  87. if (strDate.IsEmpty()) return FALSE;
  88. static int lunar_year, lunar_month, lunar_day;
  89. lunar_year = lunar_month = lunar_day = 0;
  90. lunar_year = _ttoi(strDate.Left(4));
  91. lunar_month = _ttoi(strDate.Mid(5, 2));
  92. lunar_day = _ttoi(strDate.Mid(8, 2));
  93. //越界检查,如果越界,返回无效日期;
  94. if (lunar_year <= BEGIN_YEAR || lunar_year > BEGIN_YEAR + NUMBER_YEAR - 1)
  95. return FALSE;
  96. if (lunar_month < 1 || lunar_month > 12)
  97. return FALSE;
  98. // 1.农历年是否有闰月;
  99. int nYearIndex = lunar_year - BEGIN_YEAR;
  100. int leap_month = (LUNAR_YEARS[nYearIndex] >> 20) & 0xF;
  101. // 2.月份大小月天数;
  102. int month_day = 0;
  103. if ((LUNAR_YEARS[nYearIndex] >> (7 + 13 - lunar_month)) & 0x1)
  104. month_day = 30;
  105. else
  106. month_day = 29;
  107. if (lunar_day < 0 || lunar_day > month_day)
  108. return FALSE;
  109. return TRUE;
  110. }
  111. int GetIniInfo(IN const TCHAR *szPath , const TCHAR *szIniName )
  112. {
  113. TCHAR szDrive[_MAX_DRIVE] = { 0 };
  114. TCHAR szDir[_MAX_DIR] = { 0 };
  115. TCHAR szFna[_MAX_DIR] = { 0 };
  116. TCHAR szExt[_MAX_DIR] = { 0 };
  117. DWORD dwRet = ::GetModuleFileName(NULL, g_szModuleFileName, sizeof(g_szModuleFileName) / sizeof(TCHAR));
  118. _tsplitpath_s(g_szModuleFileName, szDrive, szDir, szFna, szExt);
  119. StringCchCat(g_szModulePath, MAX_PATH, szDrive);
  120. StringCchCat(g_szModulePath, MAX_PATH, szDir);
  121. // -----------------------------------------------------//
  122. TCHAR szIniPath[MAX_PATH] = {0};
  123. if (szPath != NULL && szIniName != NULL)
  124. StringCchPrintf(szIniPath, MAX_PATH, _T("%s%s"), szPath, szIniName);
  125. else
  126. StringCchPrintf(szIniPath, MAX_PATH, _T("%sServiceInfo.ini"), g_szModulePath);
  127. HANDLE hFile = CreateFile(szIniPath, 0/*GENERIC_READ*/, 0, NULL, OPEN_EXISTING, 0, NULL);
  128. if (ERROR_FILE_NOT_FOUND == GetLastError())
  129. {
  130. return -1;
  131. }
  132. // 获取服务器端信息;
  133. //GetPrivateProfileString(_T("DatabaseInfo"), _T("dbSource"), _T(""), g_szDBSource, MAX_PATH, szIniPath);
  134. //g_dwDBServerPort = GetPrivateProfileInt(_T("DatabaseInfo"), _T("dbServerPort"), 0, szIniPath);
  135. //GetPrivateProfileString(_T("DatabaseInfo"), _T("dbAccount"), _T(""), g_szDBAccount, MAX_PATH, szIniPath);
  136. //GetPrivateProfileString(_T("DatabaseInfo"), _T("dbPassWord"), _T(""), g_szDBPassWord, MAX_PATH, szIniPath);
  137. //GetPrivateProfileString(_T("DatabaseInfo"), _T("dbName"), _T(""), g_szDBName, MAX_PATH, szIniPath);
  138. ////g_dwCSPort = GetPrivateProfileInt(_T("NetWorkInfo"), _T("TCPChatPort"), 5678, szIniPath);
  139. //if (g_dwDBServerPort != 0)
  140. // StringCchPrintf(g_szConnectString, MAX_PATH, _T("driver={SQL Server};Server=%s,%d;database=%s;uid=%s;pwd=%s"),
  141. // g_szDBSource, g_dwDBServerPort, g_szDBName, g_szDBAccount, g_szDBPassWord);
  142. //else
  143. // StringCchPrintf(g_szConnectString, MAX_PATH, _T("driver={SQL Server};Server=%s;database=%s;uid=%s;pwd=%s"),
  144. // g_szDBSource, g_szDBName, g_szDBAccount, g_szDBPassWord);
  145. // 获取服务器端信息;
  146. GetPrivateProfileString(_T("DatabaseInfo"), _T("dbSource"), _T(""), g_szDBSource, MAX_PATH, szIniPath);
  147. g_dwDBServerPort = GetPrivateProfileInt(_T("DatabaseInfo"), _T("dbServerPort"), 0, szIniPath);
  148. GetPrivateProfileString(_T("DatabaseInfo"), _T("dbAccount"), _T(""), g_szDBAccount, MAX_PATH, szIniPath);
  149. GetPrivateProfileString(_T("DatabaseInfo"), _T("dbPassWord"), _T(""), g_szDBPassWord, MAX_PATH, szIniPath);
  150. GetPrivateProfileString(_T("DatabaseInfo"), _T("dbName"), _T(""), g_szDBName, MAX_PATH, szIniPath);
  151. g_dwDBPoolMaxCount = GetPrivateProfileInt(_T("DatabaseInfo"), _T("dbpoolMaxObj"), 1, szIniPath);
  152. g_dwDBPoolDef = GetPrivateProfileInt(_T("DatabaseInfo"), _T("dbpoolDef"), 1, szIniPath);
  153. //g_dwSvrPort = GetPrivateProfileInt(_T("NetWorkInfo"), _T("TCPChatPort"), 0, szIniPath);
  154. if (g_dwDBServerPort != 0)
  155. {
  156. if (_tcscmp(g_szDBAccount, _T("")) == 0)
  157. _stprintf_s(g_szConnectString, DB_SW_CONN_WITH_PORT, g_szDBSource, g_dwDBServerPort, g_szDBName);
  158. else
  159. _stprintf_s(g_szConnectString, DB_SS_CONN_WITH_PORT, g_szDBSource, g_dwDBServerPort, g_szDBName, g_szDBAccount, g_szDBPassWord);
  160. _stprintf_s(g_szConnectString,
  161. _T("Provider=sqloledb;Data Source=%s,%ld;Initial Catalog=%s;User Id=%s;Password=%s; "),
  162. g_szDBSource, // IP地址或服务名称;
  163. g_dwDBServerPort,
  164. g_szDBName,
  165. g_szDBAccount,
  166. g_szDBPassWord);
  167. }
  168. else
  169. {
  170. if (_tcscmp(g_szDBAccount, _T("")) == 0)
  171. _stprintf_s(g_szConnectString, DB_SW_CONN_WITHOUT_PORT, g_szDBSource, g_szDBName);
  172. else
  173. _stprintf_s(g_szConnectString, DB_SS_CONN_WITHOUT_PORT, g_szDBSource, g_szDBName, g_szDBAccount, g_szDBPassWord);
  174. _stprintf_s(g_szConnectString,
  175. _T("Provider=sqloledb;Data Source=%s;Initial Catalog=%s;User Id=%s;Password=%s; "),
  176. g_szDBSource, // IP地址或服务名称;
  177. g_szDBName,
  178. g_szDBAccount,
  179. g_szDBPassWord);
  180. }
  181. return 0;
  182. }
  183. // 判断指定字符串是否在数组里(区分大小写);
  184. BOOL IsStringExist(IN CString& str, IN CStringArray &tagAry)
  185. {
  186. int nSize = tagAry.GetSize();
  187. if (nSize == 0) return FALSE;
  188. BOOL bExist = FALSE;
  189. for (int i = 0; i < nSize; i++)
  190. {
  191. if (str == tagAry.ElementAt(i))
  192. {
  193. bExist = TRUE;
  194. break;
  195. }
  196. }
  197. return bExist;
  198. }
  199. // 判断指定字符串是否在数组里(区分大小写);
  200. BOOL IsStringExist(IN TString& str, IN vector<TString> &tagVt)
  201. {
  202. int nSize = tagVt.size();
  203. if (nSize == 0) return FALSE;
  204. BOOL bExist = FALSE;
  205. for (vector<TString>::iterator it = tagVt.begin(); it != tagVt.end(); it++)
  206. {
  207. if (str.compare(it->c_str()) == 0)
  208. {
  209. bExist = TRUE;
  210. break;
  211. }
  212. }
  213. return bExist;
  214. }
  215. // 判断指定字符串是否在数组里(不区分大小写);
  216. BOOL IsStringExistNoCase(IN CString& str, IN CStringArray &tagAry)
  217. {
  218. int nSize = tagAry.GetSize();
  219. if (nSize == 0) return FALSE;
  220. BOOL bExist = FALSE;
  221. for (int i = 0; i < nSize; i++)
  222. {
  223. if (str.CompareNoCase(tagAry.ElementAt(i)) == 0)
  224. {
  225. bExist = TRUE;
  226. break;
  227. }
  228. }
  229. return bExist;
  230. }
  231. BOOL IsStringExistNoCase(IN CONST TString& str, IN CStringArray &tagAry)
  232. {
  233. int nSize = tagAry.GetSize();
  234. if (nSize == 0) return FALSE;
  235. BOOL bExist = FALSE;
  236. for (int i = 0; i < nSize; i++)
  237. {
  238. if ((tagAry.ElementAt(i).CompareNoCase(str.c_str())) == 0)
  239. {
  240. bExist = TRUE;
  241. break;
  242. }
  243. }
  244. return bExist;
  245. }
  246. // 判断指定字符串是否在数组里(不区分大小写);
  247. BOOL IsStringExistNoCase(IN CONST TString& str, IN vector<TString> &tagVt)
  248. {
  249. int nSize = tagVt.size();
  250. if (nSize == 0) return FALSE;
  251. BOOL bExist = FALSE;
  252. TString stmp1 = uppercase(str);
  253. TString stmp2;
  254. for (vector<TString>::iterator it = tagVt.begin(); it != tagVt.end(); it++)
  255. {
  256. stmp2 = uppercase(*it);
  257. if (stmp1.compare(stmp2) == 0)
  258. {
  259. bExist = TRUE;
  260. break;
  261. }
  262. }
  263. return bExist;
  264. }
  265. TString uppercase(IN const TString &Str)
  266. {
  267. const TCHAR aazz = _T('z') - _T('Z');
  268. TString sResult;
  269. for (TString::const_iterator it = Str.begin(); Str.end() != it; it++)
  270. {
  271. if (_T('a') <= *it && *it <= _T('z'))
  272. sResult.append(1, *it - aazz);
  273. else
  274. sResult.append(1, *it);
  275. }
  276. return sResult;
  277. }
  278. int match(IN CONST TString &sFileExt, IN CONST TString &sFile)
  279. {
  280. TString fext = uppercase(sFileExt);
  281. TString file = uppercase(sFile);
  282. int pos = file.find_last_of(_T('.'));
  283. if (TString::npos != pos)
  284. file = file.substr(pos);
  285. return (TString::npos != fext.find(file));
  286. }
  287. TString getfilename(IN CONST TString &file)
  288. {
  289. TString name;
  290. TString strfile = file;
  291. int nIndex = strfile.find_last_of(_T('\\')); // 如果file不包含 '\\' ,得不到文件名;
  292. if (nIndex == TString::npos)
  293. {
  294. nIndex = strfile.find_last_of(_T('.'));
  295. if ( nIndex == TString::npos )
  296. return _T("");
  297. return strfile.substr(0, nIndex);
  298. }
  299. name = strfile.substr(nIndex+1);
  300. nIndex = name.find_last_of(_T('.'));
  301. if (nIndex == TString::npos)
  302. return _T("");
  303. return name.substr(0, nIndex);
  304. }
  305. // 获取文件名;
  306. CString getfilename(IN CONST CString &file)
  307. {
  308. CString name;
  309. CString strfile = file;
  310. int nIndex = strfile.ReverseFind(_T('\\')); // 如果file不包含 '\\' ,得不到文件名;
  311. if (nIndex == -1)
  312. {
  313. nIndex = strfile.ReverseFind(_T('.'));
  314. if (nIndex == -1)
  315. return _T("");
  316. return strfile.Left(nIndex);
  317. }
  318. name = strfile.Mid(nIndex+1);
  319. nIndex = name.Find(_T('.'));
  320. if (nIndex == -1)
  321. return _T("");
  322. return name.Left(nIndex);
  323. }
  324. // 通过路径名查找进程,返回进程ID;
  325. BOOL FindAppProcessID(IN CString path, OUT DWORD &dwProcessID)
  326. {
  327. int nIndex = path.ReverseFind('\\');
  328. path = path.Right(path.GetLength() - nIndex - 1);
  329. dwProcessID = FindProcess(path);
  330. return (dwProcessID != 0 ? TRUE : FALSE);
  331. BOOL bFound = FALSE;
  332. HANDLE handle = ::CreateToolhelp32Snapshot(TH32CS_SNAPALL, 0);
  333. if ( handle == NULL )
  334. return FALSE;
  335. PROCESSENTRY32 Info;
  336. Info.dwSize = sizeof(PROCESSENTRY32);
  337. path.MakeLower();
  338. if (::Process32First(handle, &Info))
  339. {
  340. do{
  341. CString ss = Info.szExeFile;
  342. ss.MakeLower();
  343. if (ss == path)
  344. {
  345. dwProcessID = Info.th32ProcessID;
  346. bFound = TRUE;
  347. break;
  348. }
  349. } while (::Process32Next(handle, &Info));
  350. }
  351. CloseHandle(handle);
  352. return (bFound ? TRUE : FALSE);
  353. }
  354. // 根据路径名查找进程,返回进程ID;
  355. DWORD FindProcess(IN CString strProName)
  356. {
  357. DWORD dwProcessID = 0;
  358. PROCESSENTRY32 pe32 = { 0 };
  359. HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  360. if (hProcessSnap == NULL)
  361. {
  362. return 0;
  363. }
  364. pe32.dwSize = sizeof(PROCESSENTRY32);
  365. if (Process32First(hProcessSnap, &pe32))
  366. {
  367. do
  368. {
  369. if (_tcscmp(strProName.GetString(), pe32.szExeFile) == 0)
  370. {
  371. dwProcessID = pe32.th32ProcessID;
  372. break;
  373. }
  374. } while (Process32Next(hProcessSnap, &pe32));
  375. }
  376. CloseHandle(hProcessSnap);
  377. return dwProcessID;
  378. }
  379. DWORD FindProcessCount(IN LPCTSTR lpExeName)
  380. {
  381. DWORD dwCount = 0;
  382. if ( lpExeName && lpExeName[0] != _T('\0'))
  383. {
  384. PROCESSENTRY32 pe32 = { 0 };
  385. HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  386. if (hProcessSnap == NULL) return 0;
  387. pe32.dwSize = sizeof(PROCESSENTRY32);
  388. if ( !Process32First(hProcessSnap, &pe32) )
  389. {
  390. CloseHandle(hProcessSnap);
  391. return 0;
  392. }
  393. do
  394. {
  395. if (_tcscmp(lpExeName, pe32.szExeFile) == 0)
  396. dwCount++;
  397. } while (Process32Next(hProcessSnap, &pe32));
  398. CloseHandle(hProcessSnap);
  399. }
  400. return dwCount;
  401. }
  402. // 根据路径名查找进程,查找成功并结束;
  403. int FindAndCloseProcess(IN CString strProName)
  404. {
  405. int nIndex = strProName.ReverseFind('\\');
  406. if (nIndex != -1) // 如果传的是全路径;
  407. strProName = strProName.Right(strProName.GetLength() - nIndex - 1);
  408. DWORD dwProcessID = 0;
  409. PROCESSENTRY32 pe32 = { 0 };
  410. HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  411. if (hProcessSnap == NULL)
  412. {
  413. return 0;
  414. }
  415. pe32.dwSize = sizeof(PROCESSENTRY32);
  416. if (Process32First(hProcessSnap, &pe32))
  417. {
  418. do
  419. {
  420. if (_tcscmp(strProName.GetBuffer(), pe32.szExeFile) == 0)
  421. {
  422. dwProcessID = pe32.th32ProcessID;
  423. break;
  424. }
  425. } while (Process32Next(hProcessSnap, &pe32));
  426. }
  427. CloseHandle(hProcessSnap);
  428. if (dwProcessID == 0)
  429. return 0;
  430. HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS, TRUE, dwProcessID);
  431. if (hProcess == NULL)
  432. return 0;
  433. DWORD dwError;
  434. if (!TerminateProcess(hProcess, 0))
  435. {
  436. dwError = GetLastError();
  437. CloseHandle(hProcess);
  438. return -1;
  439. }
  440. // 等待进程结束响应;
  441. if (WAIT_OBJECT_0 != WaitForSingleObject(hProcess, INFINITE))
  442. {
  443. CloseHandle(hProcess);
  444. AfxMessageBox(_T("结束进程失败"));
  445. return -1;
  446. }
  447. CloseHandle(hProcess);
  448. return 1;
  449. }
  450. /************************************************************************/
  451. /*
  452. 函数:GetDiskSpaceInfo
  453. 描述:获取指定目录或盘符的容量信息;
  454. 参数:
  455. IN: lpRootPathName 根目录名称;
  456. INOUT: dTotalNum 总容量;
  457. INOUT: dFreeNum 可用容量;
  458. 返回:
  459. 注意:
  460. GetDiskFreeSpace函数不能满足当前超过2T的硬盘容量,因为里面的数据类型是UINT,最大支持32位(4G);
  461. GetDiskFreeSpaceEx支持64位(16384P=1024*16384T)
  462. 能获取本机硬盘信息,但路径必须包含":"冒号, 同时还可以获取网络路径的共享硬盘大小;
  463. 调用约定:
  464. */
  465. /************************************************************************/
  466. BOOL GetDiskSpaceInfo(IN LPCTSTR lpRootPathName, IN OUT double &dTotalNum, IN OUT double &dFreeNum)
  467. {
  468. #if 1
  469. // 校验参数有效性;
  470. if (lpRootPathName == NULL || _tcscmp(lpRootPathName, _T("")) == 0)
  471. {
  472. //_tprintf_s(_T("根目录为空!"));
  473. return FALSE;
  474. }
  475. // 如果长度==1;
  476. int nlen = _tcslen(lpRootPathName);
  477. if ((1 == nlen) || (2 == nlen && lpRootPathName[nlen - 1] != _T(':')))
  478. {
  479. //_tprintf_s(_T("根目录无效!\n"));
  480. return FALSE;
  481. }
  482. if (!PathFileExists(lpRootPathName))
  483. {
  484. //_tprintf_s(_T("根目录不存在!\n"));
  485. return FALSE;
  486. }
  487. #else
  488. // 使用IsValipath判断网络路径时较慢;
  489. if (!IsValidpath(lpRootPathName) || !PathFileExists(lpRootPathName)) return FALSE;
  490. #endif
  491. BOOL fResult = FALSE;
  492. DWORD dwSectPerClust; // 每簇的扇区数;
  493. DWORD dwBytesPerSect; // 每个扇区的字节数;
  494. DWORD dwFreeClusters; // 可用簇数;
  495. DWORD dwTotalClusters; // 总簇数;
  496. typedef BOOL(WINAPI *P_GDFSE)(LPCTSTR, PULARGE_INTEGER, PULARGE_INTEGER, PULARGE_INTEGER);
  497. P_GDFSE pGetDiskFreeSpaceEx = NULL;
  498. /*unsigned*/ __int64 i64FreeBytesToCaller;
  499. /*unsigned*/ __int64 i64TotalBytes;
  500. /*unsigned*/ __int64 i64FreeBytes;
  501. // 调用GetDiskFreeSpaceEx失败则调用GetDiskSpace;
  502. #ifdef UNICODE
  503. pGetDiskFreeSpaceEx = (P_GDFSE)GetProcAddress(GetModuleHandle(_T("kernel32.dll")), "GetDiskFreeSpaceExW");
  504. #else
  505. pGetDiskFreeSpaceEx = (P_GDFSE)GetProcAddress(GetModuleHandle("kernel32.dll"), "GetDiskFreeSpaceExA");
  506. #endif
  507. if (pGetDiskFreeSpaceEx)
  508. {
  509. fResult = pGetDiskFreeSpaceEx(lpRootPathName,
  510. (PULARGE_INTEGER)&i64FreeBytesToCaller,
  511. (PULARGE_INTEGER)&i64TotalBytes,
  512. (PULARGE_INTEGER)&i64FreeBytes);
  513. if (fResult)
  514. {
  515. // 将单位由byte转为G byte;
  516. dTotalNum = (double)i64TotalBytes / 1024 / 1024 / 1024;
  517. dFreeNum = (double)i64FreeBytes / 1024 / 1024 / 1024;
  518. }
  519. }
  520. else
  521. {
  522. fResult = GetDiskFreeSpace(lpRootPathName,
  523. &dwSectPerClust,
  524. &dwBytesPerSect,
  525. &dwFreeClusters,
  526. &dwTotalClusters);
  527. if (fResult)
  528. {
  529. // 成功获得容量信息;
  530. i64TotalBytes = (__int64)dwTotalClusters * dwSectPerClust * dwBytesPerSect;
  531. i64FreeBytes = (__int64)dwFreeClusters * dwSectPerClust * dwBytesPerSect;
  532. // 将单位由byte转为G byte;
  533. dTotalNum = (double)i64TotalBytes / 1024 / 1024 / 1024;
  534. dFreeNum = (double)i64FreeBytes / 1024 / 1024 / 1024;
  535. }
  536. }
  537. return fResult;
  538. }
  539. /************************************************************************/
  540. /*
  541. 函数:GetFileVersion
  542. 描述:获取可执行文件的文件版号;
  543. 参数:
  544. hModule[IN] 可执行文件模块句柄;
  545. dwArray[OUT] 返回的文件版本号;
  546. 返回:
  547. 成功返回TRUE,失败返回FALSE;
  548. 注意:
  549. 当hModule为空时,表示要获取的可执行文件为本程序的文件版本号;
  550. */
  551. /************************************************************************/
  552. BOOL GetFileVersion( IN HMODULE hModule, OUT DWORD (&dwArray)[4])
  553. {
  554. TCHAR fname[MAX_PATH];
  555. VS_FIXEDFILEINFO *pVi;
  556. DWORD dwHandle;
  557. if ( GetModuleFileName(hModule, fname, MAX_PATH))
  558. {
  559. INT nSize = GetFileVersionInfoSize(fname, &dwHandle);
  560. if (nSize > 0)
  561. {
  562. BYTE *pBuffer = new BYTE[nSize];
  563. memset(pBuffer, 0, nSize);
  564. if (GetFileVersionInfo(fname, dwHandle, nSize, pBuffer))
  565. {
  566. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  567. {
  568. dwArray[0] = HIWORD(pVi->dwFileVersionMS);
  569. dwArray[1] = LOWORD(pVi->dwFileVersionMS);
  570. dwArray[2] = HIWORD(pVi->dwFileVersionLS);
  571. dwArray[3] = LOWORD(pVi->dwFileVersionLS);
  572. delete[]pBuffer;
  573. return TRUE;
  574. }
  575. }
  576. if ( pBuffer )
  577. delete[]pBuffer;
  578. }
  579. }
  580. return FALSE;
  581. }
  582. /************************************************************************/
  583. /*
  584. 函数:GetFileVersion
  585. 描述:获取可执行文件的文件版号;
  586. 参数:
  587. lpFileName[IN] 可执行文件名全路径;
  588. dwArray[OUT] 返回的文件版本号;
  589. 返回:
  590. 成功返回TRUE,失败返回FALSE;
  591. 注意:
  592. */
  593. /************************************************************************/
  594. BOOL GetFileVersionEx( IN LPCTSTR lpFileName, IN DWORD (&dwArray)[4] )
  595. {
  596. if ( lpFileName == NULL || !PathFileExists(lpFileName) )
  597. {
  598. OutputDebugString(_T("文件名错误或文件不存在\n"));
  599. return FALSE;
  600. }
  601. DWORD dwHandle = 0;
  602. VS_FIXEDFILEINFO *pVi = NULL;
  603. INT nSize = GetFileVersionInfoSize(lpFileName, &dwHandle);
  604. if ( nSize > 0 )
  605. {
  606. BYTE *pBuffer = new BYTE[nSize];
  607. memset(pBuffer, 0, nSize);
  608. if ( GetFileVersionInfo(lpFileName, dwHandle, nSize, pBuffer) )
  609. {
  610. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  611. {
  612. dwArray[0] = HIWORD(pVi->dwFileVersionMS);
  613. dwArray[1] = LOWORD(pVi->dwFileVersionMS);
  614. dwArray[2] = HIWORD(pVi->dwFileVersionLS);
  615. dwArray[3] = LOWORD(pVi->dwFileVersionLS);
  616. if (pBuffer)
  617. delete[]pBuffer;
  618. return TRUE;
  619. }
  620. }
  621. if (pBuffer)
  622. delete[]pBuffer;
  623. }
  624. return FALSE;
  625. }
  626. /************************************************************************/
  627. /*
  628. 函数:GetProductVersion
  629. 描述:获取可执行文件的产品版号;
  630. 参数:
  631. hModule[IN] 可执行文件模块句柄;
  632. dwArray[OUT] 返回的产品版本号;
  633. 返回:
  634. 成功返回TRUE,失败返回FALSE;
  635. 注意:
  636. 当hModule为空时,表示要获取的可执行文件为本程序的产品版本号;
  637. */
  638. /************************************************************************/
  639. BOOL GetProductVersion(IN HMODULE hModule, IN DWORD (&dwArray)[4] )
  640. {
  641. TCHAR fname[MAX_PATH];
  642. VS_FIXEDFILEINFO *pVi;
  643. DWORD dwHandle;
  644. if (::GetModuleFileName(hModule, fname, MAX_PATH))
  645. {
  646. INT nSize = GetFileVersionInfoSize(fname, &dwHandle);
  647. if (nSize > 0)
  648. {
  649. BYTE *pBuffer = new BYTE[nSize];
  650. memset(pBuffer, 0, nSize);
  651. if (GetFileVersionInfo(fname, dwHandle, nSize, pBuffer))
  652. {
  653. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  654. {
  655. dwArray[0] = HIWORD(pVi->dwProductVersionMS);
  656. dwArray[1] = LOWORD(pVi->dwProductVersionMS);
  657. dwArray[2] = HIWORD(pVi->dwProductVersionLS);
  658. dwArray[3] = LOWORD(pVi->dwProductVersionLS);
  659. if(pBuffer)
  660. delete[]pBuffer;
  661. return TRUE;
  662. }
  663. }
  664. if(pBuffer)
  665. delete[]pBuffer;
  666. }
  667. }
  668. return FALSE;
  669. }
  670. /************************************************************************/
  671. /*
  672. 函数:GetProductVersion
  673. 描述:获取可执行文件的产品版号;
  674. 参数:
  675. lpFileName[IN] 可执行文件名全路径;
  676. dwArray[OUT] 返回的产品版本号;
  677. 返回:
  678. 成功返回TRUE,失败返回FALSE;
  679. 注意:
  680. */
  681. /************************************************************************/
  682. BOOL GetProductVersionEx( IN LPCTSTR lpFileName, IN DWORD (&dwArray)[4] )
  683. {
  684. if ( lpFileName == NULL || !PathFileExists(lpFileName) )
  685. {
  686. OutputDebugString(_T("文件名错误或文件不存在\n"));
  687. return FALSE;
  688. }
  689. DWORD dwHandle = 0;
  690. VS_FIXEDFILEINFO *pVi = NULL;
  691. INT nSize = GetFileVersionInfoSize(lpFileName, &dwHandle);
  692. if ( nSize > 0 )
  693. {
  694. BYTE *pBuffer = new BYTE[nSize];
  695. memset(pBuffer, 0, nSize);
  696. if ( GetFileVersionInfo(lpFileName, dwHandle, nSize, pBuffer) )
  697. {
  698. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  699. {
  700. dwArray[0] = HIWORD(pVi->dwProductVersionMS);
  701. dwArray[1] = LOWORD(pVi->dwProductVersionMS);
  702. dwArray[2] = HIWORD(pVi->dwProductVersionLS);
  703. dwArray[3] = LOWORD(pVi->dwProductVersionLS);
  704. if (pBuffer)
  705. delete[]pBuffer;
  706. return TRUE;
  707. }
  708. }
  709. if (pBuffer)
  710. delete[]pBuffer;
  711. }
  712. return FALSE;
  713. }
  714. /************************************************************************/
  715. /*
  716. 函数:
  717. 描述:
  718. 参数:
  719. 返回:
  720. 注意:
  721. */
  722. /************************************************************************/
  723. BOOL GetDirLastWriteTime(IN LPCTSTR lpDirName, OUT LPTSTR lpszString, IN CONST DWORD& dwSize)
  724. {
  725. if ( lpDirName == NULL || !PathFileExists(lpDirName))
  726. {
  727. OutputDebugString(_T("文件夹不存在\n"));
  728. return FALSE;
  729. }
  730. // 打开文件夹;
  731. HANDLE hDir = CreateFile(
  732. lpDirName,
  733. GENERIC_READ,
  734. FILE_SHARE_READ|FILE_SHARE_DELETE,
  735. NULL, OPEN_EXISTING,
  736. FILE_FLAG_BACKUP_SEMANTICS, NULL);
  737. if ( hDir == INVALID_HANDLE_VALUE )
  738. {
  739. OutputDebugString(_T("打开文件夹失败\n"));
  740. return FALSE;
  741. }
  742. if ( !GetLastWriteTime(hDir, lpszString, dwSize) )
  743. {
  744. OutputDebugString(_T("获取文件修改时间失败\n"));
  745. return FALSE;
  746. }
  747. // 关闭文件句柄;
  748. CloseHandle(hDir);
  749. return TRUE;
  750. }
  751. BOOL GetLastWriteTime(IN HANDLE hFile, OUT LPTSTR lpszString, IN CONST DWORD& dwSize)
  752. {
  753. // 创建时间、访问时间、修改时间;
  754. FILETIME ftCreate, ftAccess, ftWrite;
  755. // UTC时间、系统时间;
  756. SYSTEMTIME stUTC, stLocal;
  757. // 返回的结果;
  758. DWORD dwRet = 0;
  759. // Retrieve the file times for the file.
  760. if (!GetFileTime(hFile, &ftCreate, &ftAccess, &ftWrite))
  761. return FALSE;
  762. // Convert the last-write time to local time.
  763. FileTimeToSystemTime(&ftWrite, &stUTC);
  764. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  765. // Build a string showing the date and time.
  766. dwRet = StringCchPrintf(lpszString, dwSize,
  767. TEXT("%02d/%02d/%d %02d:%02d"),
  768. stLocal.wMonth, stLocal.wDay, stLocal.wYear,
  769. stLocal.wHour, stLocal.wMinute);
  770. return (S_OK == dwRet ? TRUE : FALSE);
  771. }
  772. BOOL SetFileLastWirteTime(IN LPCTSTR lpDirName, IN SYSTEMTIME &st)
  773. {
  774. if ( lpDirName == NULL || !PathFileExists(lpDirName))
  775. {
  776. OutputDebugString(_T("文件夹不存在\n"));
  777. return FALSE;
  778. }
  779. // 打开文件夹;
  780. HANDLE hDir = CreateFile(
  781. lpDirName,
  782. GENERIC_WRITE,
  783. FILE_SHARE_READ|FILE_SHARE_DELETE,
  784. NULL, OPEN_EXISTING,
  785. FILE_FLAG_BACKUP_SEMANTICS, NULL);
  786. if ( hDir == INVALID_HANDLE_VALUE )
  787. {
  788. OutputDebugString(_T("打开文件夹失败\n"));
  789. return FALSE;
  790. }
  791. FILETIME ft;
  792. BOOL f;
  793. SystemTimeToFileTime(&st, &ft); // converts to file time format
  794. // sets last-write time for file
  795. f = SetFileTime(hDir, (LPFILETIME) NULL, (LPFILETIME) NULL, &ft);
  796. CloseHandle(hDir);
  797. return f;
  798. }
  799. void DeleteDirectory(CString strDir)
  800. {
  801. if ( strDir.IsEmpty() || !PathFileExists(strDir) )
  802. {
  803. OutputDebugString(_T("DeleteDirectory:路径无效!"));
  804. return;
  805. }
  806. try
  807. {
  808. // 首先删除文件及子文件夹;
  809. CFileFind ff;
  810. BOOL bFound = ff.FindFile(strDir + _T("\\*"), 0);
  811. while (bFound)
  812. {
  813. bFound = ff.FindNextFile();
  814. if (ff.GetFileName() == _T(".") || ff.GetFileName() == _T(".."))
  815. continue;
  816. // 去掉文件(夹)只读等属性;
  817. if ( SetFileAttributes(ff.GetFilePath(), FILE_ATTRIBUTE_NORMAL) )
  818. {
  819. if (ff.IsDirectory())
  820. { // 递归删除子文件夹;
  821. DeleteDirectory(ff.GetFilePath());
  822. RemoveDirectory(ff.GetFilePath());
  823. }
  824. else
  825. {
  826. DeleteFile(ff.GetFilePath()); // Jeff:删除某文件夹下的所有子文件;
  827. OutputDebugString(ff.GetFilePath());
  828. }
  829. }
  830. else
  831. {
  832. DWORD dwError = GetLastError();
  833. CString strError = _T("");
  834. strError.Format(_T("设置文件[%s]属性失败:%d"), ff.GetFilePath(), dwError);
  835. }
  836. }
  837. ff.Close();
  838. // 然后删除该文件夹;
  839. RemoveDirectory(strDir);
  840. }
  841. catch (...)
  842. {
  843. OutputDebugString(_T("出错在函数deldir中"));
  844. }
  845. }
  846. void GetHostName()
  847. {
  848. WSADATA WSAData = { 0 };
  849. if(0 != WSAStartup(MAKEWORD(2, 2), &WSAData))
  850. {
  851. AfxMessageBox(_T("网络环境初始化失败!"));
  852. return;
  853. }
  854. CHAR szHostName[MAX_PATH] = {0};
  855. gethostname(szHostName,MAX_PATH);
  856. #ifdef UNICODE
  857. INT wSize = MultiByteToWideChar(CP_ACP, 0, szHostName, -1, NULL, 0);
  858. if (wSize == 0) return ;
  859. WCHAR *pHostName = new WCHAR[wSize];
  860. int convresult = MultiByteToWideChar(CP_ACP, 0, szHostName, -1, pHostName, wSize);
  861. if (convresult != wSize)
  862. {
  863. if (pHostName) delete []pHostName;
  864. return;
  865. }
  866. #else
  867. TCHAR *pHostName = szHostName;
  868. #endif
  869. StringCchPrintf(g_szHostName, MAX_PATH, _T("%s"), pHostName);
  870. #ifdef UNICODE
  871. delete []pHostName;
  872. #endif
  873. }
  874. };