Global.cpp 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368
  1. #pragma once
  2. #include "stdafx.h"
  3. #include "Global.h"
  4. #include <io.h>
  5. #include <direct.h>
  6. #include <tlhelp32.h>
  7. #include <shlwapi.h>
  8. #include <strsafe.h>
  9. #include <WinVer.h>
  10. #pragma comment(lib,"version.lib")
  11. #include "CritSection.h"
  12. // 获取本机ip地址;
  13. #include <Iphlpapi.h>
  14. #pragma comment(lib,"Iphlpapi.lib") //需要添加Iphlpapi.lib库
  15. // 获取网卡地址;
  16. #include <SetupAPI.h>
  17. #include <InitGuid.h>
  18. #include <WinIoCtl.h>
  19. #pragma comment(lib, "SetupAPI.lib")
  20. namespace GLOBAL
  21. {
  22. STSATConfig g_stSATConfig;
  23. //////////////////////////////////////////////////////////////////////////
  24. // 全局变量;
  25. TCHAR g_szIniFile[MAX_PATH] = {0};
  26. TCHAR g_szHostName[MAX_PATH] = {0}; // 主机计算机名;
  27. TCHAR g_szConnectString[MAX_PATH]; // DB连接串;
  28. TCHAR g_szCurModuleFileName[MAX_PATH] = { 0 }; // 软件名称;
  29. TCHAR g_szCurModuleDir[MAX_PATH] = { 0 };
  30. TCHAR g_szCurModulePath[MAX_PATH] = { 0 };
  31. TCHAR g_szFna[_MAX_FNAME] = { 0 };
  32. TCHAR g_szExt[_MAX_EXT] = { 0 };
  33. // 服务器信息;
  34. TCHAR g_szDBSource[MAX_PATH]; // 数据库源(服务所在IP或计算机名);
  35. TCHAR g_szDBSourcePort[MAX_PATH]; // 数据库源端口;
  36. DWORD g_dwDBServerPort; // 数据库源端口;
  37. TCHAR g_szDBAccount[MAX_PATH]; // 数据库登录用户;
  38. TCHAR g_szDBPassWord[MAX_PATH]; // 数据库登录密码;
  39. TCHAR g_szDBName[MAX_PATH]; // 数据库名称;
  40. std::string g_strIPAddress;
  41. //////////////////////////////////////////////////////////////////////////
  42. std::string g_strMacs;
  43. std::vector<MacAddress> g_vtMac;
  44. // usb;
  45. static GUID UsbClassGuid = { 0xA5DCBF10L, 0x6530, 0x11D2, {0x90, 0x1F, 0x00, 0xC0, 0x4F, 0xB9, 0x51, 0xED} };
  46. // mac;
  47. static GUID MacClassGuid = { 0xAD498944, 0x762F, 0x11D0, {0x8D, 0xCB, 0x00, 0xC0, 0x4F, 0xC3, 0x35, 0x8C} };
  48. // hdd;
  49. static GUID HDDClassGuid = { 0x53F56307, 0xB6BF, 0x11D0, {0x94, 0xF2, 0x00, 0xA0, 0xC9, 0x1E, 0xFB, 0x8B} };
  50. //////////////////////////////////////////////////////////////////////////
  51. ThreadSection g_critSection;
  52. ThreadSection g_csTextLog;
  53. //////////////////////////////////////////////////////////////////////////
  54. static const int BEGIN_YEAR = 1901;
  55. static const int NUMBER_YEAR = 199;
  56. static const unsigned int LUNAR_YEARS[199] =
  57. {
  58. 0x04AE53, 0x0A5748, 0x5526BD, 0x0D2650, 0x0D9544, 0x46AAB9, 0x056A4D, 0x09AD42, 0x24AEB6, 0x04AE4A, //1901-1910
  59. 0x6A4DBE, 0x0A4D52, 0x0D2546, 0x5D52BA, 0x0B544E, 0x0D6A43, 0x296D37, 0x095B4B, 0x749BC1, 0x049754, //1911-1920
  60. 0x0A4B48, 0x5B25BC, 0x06A550, 0x06D445, 0x4ADAB8, 0x02B64D, 0x095742, 0x2497B7, 0x04974A, 0x664B3E, //1921-1930
  61. 0x0D4A51, 0x0EA546, 0x56D4BA, 0x05AD4E, 0x02B644, 0x393738, 0x092E4B, 0x7C96BF, 0x0C9553, 0x0D4A48, //1931-1940
  62. 0x6DA53B, 0x0B554F, 0x056A45, 0x4AADB9, 0x025D4D, 0x092D42, 0x2C95B6, 0x0A954A, 0x7B4ABD, 0x06CA51, //1941-1950
  63. 0x0B5546, 0x555ABB, 0x04DA4E, 0x0A5B43, 0x352BB8, 0x052B4C, 0x8A953F, 0x0E9552, 0x06AA48, 0x7AD53C, //1951-1960
  64. 0x0AB54F, 0x04B645, 0x4A5739, 0x0A574D, 0x052642, 0x3E9335, 0x0D9549, 0x75AABE, 0x056A51, 0x096D46, //1961-1970
  65. 0x54AEBB, 0x04AD4F, 0x0A4D43, 0x4D26B7, 0x0D254B, 0x8D52BF, 0x0B5452, 0x0B6A47, 0x696D3C, 0x095B50, //1971-1980
  66. 0x049B45, 0x4A4BB9, 0x0A4B4D, 0xAB25C2, 0x06A554, 0x06D449, 0x6ADA3D, 0x0AB651, 0x093746, 0x5497BB, //1981-1990
  67. 0x04974F, 0x064B44, 0x36A537, 0x0EA54A, 0x86B2BF, 0x05AC53, 0x0AB647, 0x5936BC, 0x092E50, 0x0C9645, //1991-2000
  68. 0x4D4AB8, 0x0D4A4C, 0x0DA541, 0x25AAB6, 0x056A49, 0x7AADBD, 0x025D52, 0x092D47, 0x5C95BA, 0x0A954E, //2001-2010
  69. 0x0B4A43, 0x4B5537, 0x0AD54A, 0x955ABF, 0x04BA53, 0x0A5B48, 0x652BBC, 0x052B50, 0x0A9345, 0x474AB9, //2011-2020
  70. 0x06AA4C, 0x0AD541, 0x24DAB6, 0x04B64A, 0x69573D, 0x0A4E51, 0x0D2646, 0x5E933A, 0x0D534D, 0x05AA43, //2021-2030
  71. 0x36B537, 0x096D4B, 0xB4AEBF, 0x04AD53, 0x0A4D48, 0x6D25BC, 0x0D254F, 0x0D5244, 0x5DAA38, 0x0B5A4C, //2031-2040
  72. 0x056D41, 0x24ADB6, 0x049B4A, 0x7A4BBE, 0x0A4B51, 0x0AA546, 0x5B52BA, 0x06D24E, 0x0ADA42, 0x355B37, //2041-2050
  73. 0x09374B, 0x8497C1, 0x049753, 0x064B48, 0x66A53C, 0x0EA54F, 0x06B244, 0x4AB638, 0x0AAE4C, 0x092E42, //2051-2060
  74. 0x3C9735, 0x0C9649, 0x7D4ABD, 0x0D4A51, 0x0DA545, 0x55AABA, 0x056A4E, 0x0A6D43, 0x452EB7, 0x052D4B, //2061-2070
  75. 0x8A95BF, 0x0A9553, 0x0B4A47, 0x6B553B, 0x0AD54F, 0x055A45, 0x4A5D38, 0x0A5B4C, 0x052B42, 0x3A93B6, //2071-2080
  76. 0x069349, 0x7729BD, 0x06AA51, 0x0AD546, 0x54DABA, 0x04B64E, 0x0A5743, 0x452738, 0x0D264A, 0x8E933E, //2081-2090
  77. 0x0D5252, 0x0DAA47, 0x66B53B, 0x056D4F, 0x04AE45, 0x4A4EB9, 0x0A4D4C, 0x0D1541, 0x2D92B5 //2091-2099
  78. };
  79. //////////////////////////////////////////////////////////////////////////
  80. BOOL IsValidDate(IN LPCTSTR lpDate) // 判断公历;
  81. {
  82. if (lpDate == NULL || _tcscmp(lpDate, _T("")) == 0)
  83. return FALSE;
  84. CString strDate = lpDate;
  85. if (strDate.IsEmpty()) return FALSE;
  86. // 判断日期合法性;
  87. static int nYear = 0;
  88. static int nMonth = 0;
  89. static int nDay = 0;
  90. nYear = nMonth = nDay = 0;
  91. nYear = _ttoi(strDate.Left(4));
  92. nMonth = _ttoi(strDate.Mid(5, 2));
  93. nDay = _ttoi(strDate.Mid(8, 2));
  94. CONST static INT nNormalMonth[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  95. CONST static INT nLeapMonth[12] = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  96. if ((nDay <= 0) || (nDay > 31) || (nMonth <= 0) || (nMonth > 12)) return FALSE;
  97. if ((nYear % 400 == 0) || (nYear % 4 == 0 && nYear % 100 != 0))
  98. {
  99. if (nDay > nLeapMonth[nMonth - 1]) return FALSE;
  100. }
  101. else
  102. {
  103. if (nDay > nNormalMonth[nMonth - 1]) return FALSE;
  104. }
  105. return TRUE;
  106. }
  107. BOOL IsValidLunanrDate(IN LPCTSTR lpDate) // 判断农历;
  108. {
  109. if (lpDate == NULL || _tcscmp(lpDate, _T("")) == 0)
  110. return FALSE;
  111. CString strDate = lpDate;
  112. if (strDate.IsEmpty()) return FALSE;
  113. static int lunar_year, lunar_month, lunar_day;
  114. lunar_year = lunar_month = lunar_day = 0;
  115. lunar_year = _ttoi(strDate.Left(4));
  116. lunar_month = _ttoi(strDate.Mid(5, 2));
  117. lunar_day = _ttoi(strDate.Mid(8, 2));
  118. //越界检查,如果越界,返回无效日期;
  119. if (lunar_year <= BEGIN_YEAR || lunar_year > BEGIN_YEAR + NUMBER_YEAR - 1)
  120. return FALSE;
  121. if (lunar_month < 1 || lunar_month > 12)
  122. return FALSE;
  123. // 1.农历年是否有闰月;
  124. int nYearIndex = lunar_year - BEGIN_YEAR;
  125. int leap_month = (LUNAR_YEARS[nYearIndex] >> 20) & 0xF;
  126. // 2.月份大小月天数;
  127. int month_day = 0;
  128. if ((LUNAR_YEARS[nYearIndex] >> (7 + 13 - lunar_month)) & 0x1)
  129. month_day = 30;
  130. else
  131. month_day = 29;
  132. if (lunar_day < 0 || lunar_day > month_day)
  133. return FALSE;
  134. return TRUE;
  135. }
  136. int GetIniInfo(IN const TCHAR *szPath , const TCHAR *szIniName )
  137. {
  138. TCHAR szDrive[_MAX_DRIVE] = { 0 };
  139. TCHAR szDir[_MAX_DIR] = { 0 };
  140. TCHAR szFna[_MAX_FNAME] = { 0 };
  141. TCHAR szExt[_MAX_EXT] = { 0 };
  142. DWORD dwRet = ::GetModuleFileName(NULL, g_szCurModuleFileName, sizeof(g_szCurModuleFileName) / sizeof(TCHAR));
  143. _tsplitpath_s(g_szCurModuleFileName, szDrive, szDir, g_szFna, g_szExt);
  144. StringCchCat(g_szCurModuleDir, MAX_PATH, szDrive);
  145. StringCchCat(g_szCurModuleDir, MAX_PATH, szDir);
  146. // -----------------------------------------------------//
  147. TCHAR szIniPath[MAX_PATH] = {0};
  148. if (szPath != NULL && szIniName != NULL)
  149. StringCchPrintf(szIniPath, MAX_PATH, _T("%s%s"), szPath, szIniName);
  150. else
  151. StringCchPrintf(szIniPath, MAX_PATH, _T("%sSATService.ini"), g_szCurModuleDir);
  152. HANDLE hFile = CreateFile(szIniPath, 0/*GENERIC_READ*/, 0, NULL, OPEN_EXISTING, 0, NULL);
  153. if (ERROR_FILE_NOT_FOUND == GetLastError())
  154. {
  155. return -1;
  156. }
  157. // 获取服务器端信息;
  158. GetPrivateProfileString(_T("SATService"), _T("ScriptDir"), _T(""), g_stSATConfig.szScriptDir, MAX_PATH, szIniPath);
  159. if ( g_stSATConfig.szScriptDir[_tcslen(g_stSATConfig.szScriptDir)-1] != '\\')
  160. {
  161. _tcscat_s(g_stSATConfig.szScriptDir, MAX_PATH, "\\");
  162. WritePrivateProfileString(_T("SATService"), _T("ScriptDir"), g_stSATConfig.szScriptDir, szIniPath);
  163. }
  164. g_stSATConfig.dwAdbTimeout = GetPrivateProfileInt(_T("SATService"), _T("AdbTimeout"), 120000, szIniPath);
  165. g_stSATConfig.dwTCPSvrPort = GetPrivateProfileInt(_T("SATService"), _T("TCPSvrPort"), 5588, szIniPath);
  166. g_stSATConfig.dwScriptTimeout = GetPrivateProfileInt(_T("SATService"), _T("ScriptTimeout"), 300000, szIniPath);
  167. GetPrivateProfileString(_T("SATService"), _T("ExecuteServer"), _T(""), g_stSATConfig.szExecuteServer, MAX_PATH, szIniPath);
  168. GetPrivateProfileString(_T("SATService"), _T("ExecuteSyscode"), _T(""), g_stSATConfig.szExecuteSysCode, MAX_PATH, szIniPath);
  169. GetPrivateProfileString(_T("SATService"), _T("ApkServer"), _T(""), g_stSATConfig.szApkServer, MAX_PATH, szIniPath);
  170. GetPrivateProfileString(_T("SATService"), _T("ApkSyscode"), _T(""), g_stSATConfig.szApkSysCode, MAX_PATH, szIniPath);
  171. GetPrivateProfileString(_T("SATService"), _T("CaseServer"), _T(""), g_stSATConfig.szCaseServer, MAX_PATH, szIniPath);
  172. GetPrivateProfileString(_T("SATService"), _T("CaseSyscode"), _T(""), g_stSATConfig.szCaseSysCode, MAX_PATH, szIniPath);
  173. GetPrivateProfileString(_T("SATService"), _T("TaskServer"), _T(""), g_stSATConfig.szTaskServer, MAX_PATH, szIniPath);
  174. GetPrivateProfileString(_T("SATService"), _T("TaskSyscode"), _T(""), g_stSATConfig.szTaskSysCode, MAX_PATH, szIniPath);
  175. GetPrivateProfileString(_T("SATService"), _T("ResultServer"), _T(""), g_stSATConfig.szResultServer, MAX_PATH, szIniPath);
  176. GetPrivateProfileString(_T("SATService"), _T("ResultExcelDir"), _T(""), g_stSATConfig.szResultExcelDir, MAX_PATH, szIniPath);
  177. GetPrivateProfileString(_T("LinuxDevice"), _T("device"), _T(""), g_stSATConfig.stLinuxDevice.szDevice, MAX_PATH, szIniPath);
  178. GetPrivateProfileString(_T("LinuxDevice"), _T("manu"), _T(""), g_stSATConfig.stLinuxDevice.szManu, MAX_PATH, szIniPath);
  179. GetPrivateProfileString(_T("LinuxDevice"), _T("hardwareversion"), _T(""), g_stSATConfig.stLinuxDevice.szHardwareVersion, MAX_PATH, szIniPath);
  180. GetPrivateProfileString(_T("LinuxDevice"), _T("model"), _T(""), g_stSATConfig.stLinuxDevice.szModel, MAX_PATH, szIniPath);
  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. return 0;
  390. do
  391. {
  392. if (_tcscmp(lpExeName, pe32.szExeFile) == 0)
  393. dwCount++;
  394. } while (Process32Next(hProcessSnap, &pe32));
  395. CloseHandle(hProcessSnap);
  396. }
  397. return dwCount;
  398. }
  399. // 根据路径名查找进程,查找成功并结束;
  400. int FindAndCloseProcess(IN CString strProName)
  401. {
  402. int nIndex = strProName.ReverseFind('\\');
  403. if (nIndex != -1) // 如果传的是全路径;
  404. strProName = strProName.Right(strProName.GetLength() - nIndex - 1);
  405. DWORD dwProcessID = 0;
  406. PROCESSENTRY32 pe32 = { 0 };
  407. HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  408. if (hProcessSnap == NULL)
  409. {
  410. return 0;
  411. }
  412. pe32.dwSize = sizeof(PROCESSENTRY32);
  413. if (Process32First(hProcessSnap, &pe32))
  414. {
  415. do
  416. {
  417. if (_tcscmp(strProName.GetBuffer(), pe32.szExeFile) == 0)
  418. {
  419. dwProcessID = pe32.th32ProcessID;
  420. break;
  421. }
  422. } while (Process32Next(hProcessSnap, &pe32));
  423. }
  424. CloseHandle(hProcessSnap);
  425. if (dwProcessID == 0)
  426. return 0;
  427. HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS, TRUE, dwProcessID);
  428. if (hProcess == NULL)
  429. return 0;
  430. DWORD dwError;
  431. if (!TerminateProcess(hProcess, 0))
  432. {
  433. dwError = GetLastError();
  434. CloseHandle(hProcess);
  435. return -1;
  436. }
  437. // 等待进程结束响应;
  438. if (WAIT_OBJECT_0 != WaitForSingleObject(hProcess, INFINITE))
  439. {
  440. CloseHandle(hProcess);
  441. AfxMessageBox(_T("结束进程失败"));
  442. return -1;
  443. }
  444. CloseHandle(hProcess);
  445. return 1;
  446. }
  447. /************************************************************************/
  448. /*
  449. 函数:GetDiskSpaceInfo
  450. 描述:获取指定目录或盘符的容量信息;
  451. 参数:
  452. IN: lpRootPathName 根目录名称;
  453. INOUT: dTotalNum 总容量;
  454. INOUT: dFreeNum 可用容量;
  455. 返回:
  456. 注意:
  457. GetDiskFreeSpace函数不能满足当前超过2T的硬盘容量,因为里面的数据类型是UINT,最大支持32位(4G);
  458. GetDiskFreeSpaceEx支持64位(16384P=1024*16384T)
  459. 能获取本机硬盘信息,但路径必须包含":"冒号, 同时还可以获取网络路径的共享硬盘大小;
  460. 调用约定:
  461. */
  462. /************************************************************************/
  463. BOOL GetDiskSpaceInfo(IN LPCTSTR lpRootPathName, IN OUT double &dTotalNum, IN OUT double &dFreeNum)
  464. {
  465. #if 1
  466. // 校验参数有效性;
  467. if (lpRootPathName == NULL || _tcscmp(lpRootPathName, _T("")) == 0)
  468. {
  469. //_tprintf_s(_T("根目录为空!"));
  470. return FALSE;
  471. }
  472. // 如果长度==1;
  473. int nlen = _tcslen(lpRootPathName);
  474. if ((1 == nlen) || (2 == nlen && lpRootPathName[nlen - 1] != _T(':')))
  475. {
  476. //_tprintf_s(_T("根目录无效!\n"));
  477. return FALSE;
  478. }
  479. if (!PathFileExists(lpRootPathName))
  480. {
  481. //_tprintf_s(_T("根目录不存在!\n"));
  482. return FALSE;
  483. }
  484. #else
  485. // 使用IsValipath判断网络路径时较慢;
  486. if (!IsValidpath(lpRootPathName) || !PathFileExists(lpRootPathName)) return FALSE;
  487. #endif
  488. BOOL fResult = FALSE;
  489. DWORD dwSectPerClust; // 每簇的扇区数;
  490. DWORD dwBytesPerSect; // 每个扇区的字节数;
  491. DWORD dwFreeClusters; // 可用簇数;
  492. DWORD dwTotalClusters; // 总簇数;
  493. typedef BOOL(WINAPI *P_GDFSE)(LPCTSTR, PULARGE_INTEGER, PULARGE_INTEGER, PULARGE_INTEGER);
  494. P_GDFSE pGetDiskFreeSpaceEx = NULL;
  495. /*unsigned*/ __int64 i64FreeBytesToCaller;
  496. /*unsigned*/ __int64 i64TotalBytes;
  497. /*unsigned*/ __int64 i64FreeBytes;
  498. // 调用GetDiskFreeSpaceEx失败则调用GetDiskSpace;
  499. #ifdef UNICODE
  500. pGetDiskFreeSpaceEx = (P_GDFSE)GetProcAddress(GetModuleHandle(_T("kernel32.dll")), "GetDiskFreeSpaceExW");
  501. #else
  502. pGetDiskFreeSpaceEx = (P_GDFSE)GetProcAddress(GetModuleHandle("kernel32.dll"), "GetDiskFreeSpaceExA");
  503. #endif
  504. if (pGetDiskFreeSpaceEx)
  505. {
  506. fResult = pGetDiskFreeSpaceEx(lpRootPathName,
  507. (PULARGE_INTEGER)&i64FreeBytesToCaller,
  508. (PULARGE_INTEGER)&i64TotalBytes,
  509. (PULARGE_INTEGER)&i64FreeBytes);
  510. if (fResult)
  511. {
  512. // 将单位由byte转为G byte;
  513. dTotalNum = (double)i64TotalBytes / 1024 / 1024 / 1024;
  514. dFreeNum = (double)i64FreeBytes / 1024 / 1024 / 1024;
  515. }
  516. }
  517. else
  518. {
  519. fResult = GetDiskFreeSpace(lpRootPathName,
  520. &dwSectPerClust,
  521. &dwBytesPerSect,
  522. &dwFreeClusters,
  523. &dwTotalClusters);
  524. if (fResult)
  525. {
  526. // 成功获得容量信息;
  527. i64TotalBytes = (__int64)dwTotalClusters * dwSectPerClust * dwBytesPerSect;
  528. i64FreeBytes = (__int64)dwFreeClusters * dwSectPerClust * dwBytesPerSect;
  529. // 将单位由byte转为G byte;
  530. dTotalNum = (double)i64TotalBytes / 1024 / 1024 / 1024;
  531. dFreeNum = (double)i64FreeBytes / 1024 / 1024 / 1024;
  532. }
  533. }
  534. return fResult;
  535. }
  536. /************************************************************************/
  537. /*
  538. 函数:GetFileVersion
  539. 描述:获取可执行文件的文件版号;
  540. 参数:
  541. hModule[IN] 可执行文件模块句柄;
  542. dwArray[OUT] 返回的文件版本号;
  543. 返回:
  544. 成功返回TRUE,失败返回FALSE;
  545. 注意:
  546. 当hModule为空时,表示要获取的可执行文件为本程序的文件版本号;
  547. */
  548. /************************************************************************/
  549. BOOL GetFileVersion( IN HMODULE hModule, OUT DWORD (&dwArray)[4])
  550. {
  551. TCHAR fname[MAX_PATH];
  552. VS_FIXEDFILEINFO *pVi;
  553. DWORD dwHandle;
  554. if ( GetModuleFileName(hModule, fname, MAX_PATH))
  555. {
  556. INT nSize = GetFileVersionInfoSize(fname, &dwHandle);
  557. if (nSize > 0)
  558. {
  559. BYTE *pBuffer = new BYTE[nSize];
  560. memset(pBuffer, 0, nSize);
  561. if (GetFileVersionInfo(fname, dwHandle, nSize, pBuffer))
  562. {
  563. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  564. {
  565. dwArray[0] = HIWORD(pVi->dwFileVersionMS);
  566. dwArray[1] = LOWORD(pVi->dwFileVersionMS);
  567. dwArray[2] = HIWORD(pVi->dwFileVersionLS);
  568. dwArray[3] = LOWORD(pVi->dwFileVersionLS);
  569. delete[]pBuffer;
  570. return TRUE;
  571. }
  572. }
  573. if ( pBuffer )
  574. delete[]pBuffer;
  575. }
  576. }
  577. return FALSE;
  578. }
  579. /************************************************************************/
  580. /*
  581. 函数:GetFileVersion
  582. 描述:获取可执行文件的文件版号;
  583. 参数:
  584. lpFileName[IN] 可执行文件名全路径;
  585. dwArray[OUT] 返回的文件版本号;
  586. 返回:
  587. 成功返回TRUE,失败返回FALSE;
  588. 注意:
  589. */
  590. /************************************************************************/
  591. BOOL GetFileVersionEx( IN LPCTSTR lpFileName, IN DWORD (&dwArray)[4] )
  592. {
  593. if ( lpFileName == NULL || !PathFileExists(lpFileName) )
  594. {
  595. OutputDebugString(_T("文件名错误或文件不存在\n"));
  596. return FALSE;
  597. }
  598. DWORD dwHandle = 0;
  599. VS_FIXEDFILEINFO *pVi = NULL;
  600. INT nSize = GetFileVersionInfoSize(lpFileName, &dwHandle);
  601. if ( nSize > 0 )
  602. {
  603. BYTE *pBuffer = new BYTE[nSize];
  604. memset(pBuffer, 0, nSize);
  605. if ( GetFileVersionInfo(lpFileName, dwHandle, nSize, pBuffer) )
  606. {
  607. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  608. {
  609. dwArray[0] = HIWORD(pVi->dwFileVersionMS);
  610. dwArray[1] = LOWORD(pVi->dwFileVersionMS);
  611. dwArray[2] = HIWORD(pVi->dwFileVersionLS);
  612. dwArray[3] = LOWORD(pVi->dwFileVersionLS);
  613. if (pBuffer)
  614. delete[]pBuffer;
  615. return TRUE;
  616. }
  617. }
  618. if (pBuffer)
  619. delete[]pBuffer;
  620. }
  621. return FALSE;
  622. }
  623. /************************************************************************/
  624. /*
  625. 函数:GetProductVersion
  626. 描述:获取可执行文件的产品版号;
  627. 参数:
  628. hModule[IN] 可执行文件模块句柄;
  629. dwArray[OUT] 返回的产品版本号;
  630. 返回:
  631. 成功返回TRUE,失败返回FALSE;
  632. 注意:
  633. 当hModule为空时,表示要获取的可执行文件为本程序的产品版本号;
  634. */
  635. /************************************************************************/
  636. BOOL GetProductVersion(IN HMODULE hModule, IN DWORD (&dwArray)[4] )
  637. {
  638. TCHAR fname[MAX_PATH];
  639. VS_FIXEDFILEINFO *pVi;
  640. DWORD dwHandle;
  641. if (::GetModuleFileName(hModule, fname, MAX_PATH))
  642. {
  643. INT nSize = GetFileVersionInfoSize(fname, &dwHandle);
  644. if (nSize > 0)
  645. {
  646. BYTE *pBuffer = new BYTE[nSize];
  647. memset(pBuffer, 0, nSize);
  648. if (GetFileVersionInfo(fname, dwHandle, nSize, pBuffer))
  649. {
  650. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  651. {
  652. dwArray[0] = HIWORD(pVi->dwProductVersionMS);
  653. dwArray[1] = LOWORD(pVi->dwProductVersionMS);
  654. dwArray[2] = HIWORD(pVi->dwProductVersionLS);
  655. dwArray[3] = LOWORD(pVi->dwProductVersionLS);
  656. if(pBuffer)
  657. delete[]pBuffer;
  658. return TRUE;
  659. }
  660. }
  661. if(pBuffer)
  662. delete[]pBuffer;
  663. }
  664. }
  665. return FALSE;
  666. }
  667. /************************************************************************/
  668. /*
  669. 函数:GetProductVersion
  670. 描述:获取可执行文件的产品版号;
  671. 参数:
  672. lpFileName[IN] 可执行文件名全路径;
  673. dwArray[OUT] 返回的产品版本号;
  674. 返回:
  675. 成功返回TRUE,失败返回FALSE;
  676. 注意:
  677. */
  678. /************************************************************************/
  679. BOOL GetProductVersionEx( IN LPCTSTR lpFileName, IN DWORD (&dwArray)[4] )
  680. {
  681. if ( lpFileName == NULL || !PathFileExists(lpFileName) )
  682. {
  683. OutputDebugString(_T("文件名错误或文件不存在\n"));
  684. return FALSE;
  685. }
  686. DWORD dwHandle = 0;
  687. VS_FIXEDFILEINFO *pVi = NULL;
  688. INT nSize = GetFileVersionInfoSize(lpFileName, &dwHandle);
  689. if ( nSize > 0 )
  690. {
  691. BYTE *pBuffer = new BYTE[nSize];
  692. memset(pBuffer, 0, nSize);
  693. if ( GetFileVersionInfo(lpFileName, dwHandle, nSize, pBuffer) )
  694. {
  695. if (VerQueryValue(pBuffer, _T("\\"), (LPVOID *)&pVi, (PUINT)&nSize))
  696. {
  697. dwArray[0] = HIWORD(pVi->dwProductVersionMS);
  698. dwArray[1] = LOWORD(pVi->dwProductVersionMS);
  699. dwArray[2] = HIWORD(pVi->dwProductVersionLS);
  700. dwArray[3] = LOWORD(pVi->dwProductVersionLS);
  701. if (pBuffer)
  702. delete[]pBuffer;
  703. return TRUE;
  704. }
  705. }
  706. if (pBuffer)
  707. delete[]pBuffer;
  708. }
  709. return FALSE;
  710. }
  711. /************************************************************************/
  712. /*
  713. 函数:
  714. 描述:
  715. 参数:
  716. 返回:
  717. 注意:
  718. */
  719. /************************************************************************/
  720. BOOL GetDirLastWriteTime(IN LPCTSTR lpDirName, OUT LPTSTR lpszString, IN CONST DWORD& dwSize)
  721. {
  722. if ( lpDirName == NULL || !PathFileExists(lpDirName))
  723. {
  724. OutputDebugString(_T("文件夹不存在\n"));
  725. return FALSE;
  726. }
  727. // 打开文件夹;
  728. HANDLE hDir = CreateFile(
  729. lpDirName,
  730. GENERIC_READ,
  731. FILE_SHARE_READ|FILE_SHARE_DELETE,
  732. NULL, OPEN_EXISTING,
  733. FILE_FLAG_BACKUP_SEMANTICS, NULL);
  734. if ( hDir == INVALID_HANDLE_VALUE )
  735. {
  736. OutputDebugString(_T("打开文件夹失败\n"));
  737. return FALSE;
  738. }
  739. if ( !GetLastWriteTime(hDir, lpszString, dwSize) )
  740. {
  741. OutputDebugString(_T("获取文件修改时间失败\n"));
  742. return FALSE;
  743. }
  744. // 关闭文件句柄;
  745. CloseHandle(hDir);
  746. return TRUE;
  747. }
  748. BOOL GetLastWriteTime(IN HANDLE hFile, OUT LPTSTR lpszString, IN CONST DWORD& dwSize)
  749. {
  750. // 创建时间、访问时间、修改时间;
  751. FILETIME ftCreate, ftAccess, ftWrite;
  752. // UTC时间、系统时间;
  753. SYSTEMTIME stUTC, stLocal;
  754. // 返回的结果;
  755. DWORD dwRet = 0;
  756. // Retrieve the file times for the file.
  757. if (!GetFileTime(hFile, &ftCreate, &ftAccess, &ftWrite))
  758. return FALSE;
  759. // Convert the last-write time to local time.
  760. FileTimeToSystemTime(&ftWrite, &stUTC);
  761. SystemTimeToTzSpecificLocalTime(NULL, &stUTC, &stLocal);
  762. // Build a string showing the date and time.
  763. dwRet = StringCchPrintf(lpszString, dwSize,
  764. TEXT("%02d/%02d/%d %02d:%02d"),
  765. stLocal.wMonth, stLocal.wDay, stLocal.wYear,
  766. stLocal.wHour, stLocal.wMinute);
  767. return (S_OK == dwRet ? TRUE : FALSE);
  768. }
  769. BOOL SetFileLastWirteTime(IN LPCTSTR lpDirName, IN SYSTEMTIME &st)
  770. {
  771. if ( lpDirName == NULL || !PathFileExists(lpDirName))
  772. {
  773. OutputDebugString(_T("文件夹不存在\n"));
  774. return FALSE;
  775. }
  776. // 打开文件夹;
  777. HANDLE hDir = CreateFile(
  778. lpDirName,
  779. GENERIC_WRITE,
  780. FILE_SHARE_READ|FILE_SHARE_DELETE,
  781. NULL, OPEN_EXISTING,
  782. FILE_FLAG_BACKUP_SEMANTICS, NULL);
  783. if ( hDir == INVALID_HANDLE_VALUE )
  784. {
  785. OutputDebugString(_T("打开文件夹失败\n"));
  786. return FALSE;
  787. }
  788. FILETIME ft;
  789. BOOL f;
  790. SystemTimeToFileTime(&st, &ft); // converts to file time format
  791. // sets last-write time for file
  792. f = SetFileTime(hDir, (LPFILETIME) NULL, (LPFILETIME) NULL, &ft);
  793. CloseHandle(hDir);
  794. return f;
  795. }
  796. void DeleteDirectory(CString strDir)
  797. {
  798. if ( strDir.IsEmpty() || !PathFileExists(strDir) )
  799. {
  800. OutputDebugString(_T("DeleteDirectory:路径无效!"));
  801. return;
  802. }
  803. try
  804. {
  805. // 首先删除文件及子文件夹;
  806. CFileFind ff;
  807. BOOL bFound = ff.FindFile(strDir + _T("\\*"), 0);
  808. while (bFound)
  809. {
  810. bFound = ff.FindNextFile();
  811. if (ff.GetFileName() == _T(".") || ff.GetFileName() == _T(".."))
  812. continue;
  813. // 去掉文件(夹)只读等属性;
  814. if ( SetFileAttributes(ff.GetFilePath(), FILE_ATTRIBUTE_NORMAL) )
  815. {
  816. if (ff.IsDirectory())
  817. { // 递归删除子文件夹;
  818. DeleteDirectory(ff.GetFilePath());
  819. RemoveDirectory(ff.GetFilePath());
  820. }
  821. else
  822. {
  823. DeleteFile(ff.GetFilePath()); // Jeff:删除某文件夹下的所有子文件;
  824. OutputDebugString(ff.GetFilePath());
  825. }
  826. }
  827. else
  828. {
  829. DWORD dwError = GetLastError();
  830. CString strError = _T("");
  831. strError.Format(_T("设置文件[%s]属性失败:%d"), ff.GetFilePath(), dwError);
  832. }
  833. }
  834. ff.Close();
  835. // 然后删除该文件夹;
  836. RemoveDirectory(strDir);
  837. }
  838. catch (...)
  839. {
  840. OutputDebugString(_T("出错在函数deldir中"));
  841. }
  842. }
  843. void GetHostName()
  844. {
  845. WSADATA WSAData = { 0 };
  846. if(0 != WSAStartup(MAKEWORD(2, 2), &WSAData))
  847. {
  848. AfxMessageBox(_T("网络环境初始化失败!"));
  849. return;
  850. }
  851. CHAR szHostName[MAX_PATH] = {0};
  852. gethostname(szHostName,MAX_PATH);
  853. #ifdef UNICODE
  854. INT wSize = MultiByteToWideChar(CP_ACP, 0, szHostName, -1, NULL, 0);
  855. if (wSize == 0) return ;
  856. WCHAR *pHostName = new WCHAR[wSize];
  857. int convresult = MultiByteToWideChar(CP_ACP, 0, szHostName, -1, pHostName, wSize);
  858. if (convresult != wSize)
  859. {
  860. if (pHostName) delete []pHostName;
  861. return;
  862. }
  863. #else
  864. TCHAR *pHostName = szHostName;
  865. #endif
  866. StringCchPrintf(g_szHostName, MAX_PATH, _T("%s"), pHostName);
  867. #ifdef UNICODE
  868. delete []pHostName;
  869. #endif
  870. }
  871. bool MKDIR(LPCTSTR dir)
  872. {
  873. //////////////////////////////////////////////////////////////////////////
  874. // 创建目录;
  875. int nleft = 0;
  876. int nIndex = -1;
  877. string strdir = dir;
  878. strdir = strdir.substr(0, strdir.find_last_of(_T("\\")));
  879. if (strdir.at(strdir.size() - 1) != _T('\\'))
  880. strdir.append(_T("\\"));
  881. // 共享路径和硬盘盘符;
  882. if (_tcscmp(strdir.substr(0, 2).c_str(), _T("\\\\")) == 0)
  883. nleft = strdir.find_first_of(_T("\\"), 2) + 1; // 去除共享主机名;
  884. else if (strdir.at(2) == _T('\\'))
  885. nleft = 3;
  886. do
  887. {
  888. nIndex = strdir.substr(nleft, -1).find_first_of(_T("\\"));
  889. if (nIndex != string::npos)
  890. {
  891. if (_mkdir(strdir.substr(0, nIndex + nleft).c_str()) == -1 && (errno != EEXIST))
  892. {
  893. //WriteTextLog(_T("创建目录失败:%s,错误码:%d"), strdir.substr(0, nIndex + nleft).c_str(), errno);
  894. return false;
  895. }
  896. nleft += nIndex + 1;
  897. }
  898. } while (nIndex != -1);
  899. return true;
  900. };
  901. /************************************************************************/
  902. /* 函数:WriteTextLog[7/28/2009 Jeff];
  903. /* 描述:写文本日志;
  904. /* 参数:;
  905. /* [IN] :;
  906. /* 返回:void;
  907. /* 注意:;
  908. /* 示例:;
  909. /*
  910. /* 修改:;
  911. /* 日期:;
  912. /* 内容:;
  913. /************************************************************************/
  914. void WriteTextLog(const TCHAR* format, ...)
  915. {
  916. AutoThreadSection aSection(&g_csTextLog);
  917. // 解析出日志路径;
  918. TCHAR szlogpath[MAX_PATH] = { 0 };
  919. _stprintf_s(szlogpath, _T("%s%s.txt"), g_szCurModuleDir, g_szFna);
  920. // 打开或创建文件;
  921. FILE* fp = NULL;
  922. if (_taccess(szlogpath, 0) != -1)
  923. {// 存在;
  924. if (0 == _tfopen_s(&fp, szlogpath, _T("a+")))
  925. // 移动到末尾;
  926. fseek(fp, 0, SEEK_END);
  927. }
  928. else
  929. {// 不存在;
  930. _tfopen_s(&fp, szlogpath, _T("w+"));
  931. }
  932. if (fp == NULL)
  933. return;
  934. // 格式化前设置语言区域;
  935. TCHAR* old_locale = _tcsdup(_tsetlocale(LC_CTYPE, NULL));
  936. _tsetlocale(LC_CTYPE, _T("chs"));//设定中文;
  937. // 格式化日志内容;
  938. va_list args = NULL;
  939. int len = 0;
  940. TCHAR* buffer = NULL;
  941. va_start(args, format);
  942. // _vscprintf doesn't count. terminating '\0'
  943. len = _vsctprintf(format, args) + 1;
  944. buffer = (TCHAR*)malloc(len * sizeof(TCHAR));
  945. _vstprintf_s(buffer, len, format, args);
  946. // 将日志内容输入到文件中;
  947. // 获取今年年份;
  948. __time64_t gmt = time(NULL);// 获取当前日历时间(1900-01-01开始的Unix时间戳);
  949. struct tm gmtm = { 0 };
  950. localtime_s(&gmtm, &gmt); // 时间戳转成本地时间;
  951. _ftprintf(fp, _T("%04d-%02d-%02d %02d:%02d:%02d %s\n"), gmtm.tm_year + 1990, gmtm.tm_mon + 1, gmtm.tm_mday, gmtm.tm_hour, gmtm.tm_min, gmtm.tm_sec, buffer);
  952. // 关闭文件,释放资源并设置回原语言区域;
  953. free(buffer);
  954. fclose(fp);
  955. _tsetlocale(LC_CTYPE, old_locale);
  956. free(old_locale);//还原区域设定;
  957. }
  958. void WritePythonLog(LPCTSTR file, const TCHAR* msg)
  959. {
  960. AutoThreadSection aSection(&g_critSection);
  961. // 将日志内容输入到文件中;
  962. // 获取今年年份;
  963. //__time64_t gmt = time(NULL); // 获取当前日历时间(1900-01-01开始的Unix时间戳);
  964. //struct tm gmtm = { 0 };
  965. //localtime_s(&gmtm, &gmt); // 时间戳转成本地时间;
  966. int nErr = -1;
  967. // 打开或创建文件;
  968. FILE* fp = NULL;
  969. #ifndef UNICODE
  970. if (_access(file, 0) != -1)
  971. #else
  972. if (_taccess(file, 0) != -1)
  973. #endif
  974. { // 存在;
  975. if (0 == _tfopen_s(&fp, file, _T("a+")))
  976. {
  977. // 移动到末尾;
  978. fseek(fp, 0, SEEK_END);
  979. }
  980. else
  981. {
  982. //WriteTextLog(_T("打开文件失败:%s,%d"), file, GetLastError());
  983. return;
  984. }
  985. }
  986. else
  987. { // 不存在;
  988. MKDIR(file);
  989. if (0 != _tfopen_s(&fp, file, _T("w+")) )
  990. {
  991. //WriteTextLog(_T("创建文件失败:%s,%d"), file, GetLastError());
  992. return;
  993. }
  994. }
  995. if (fp == NULL)
  996. {
  997. //WriteTextLog(_T("文件句柄空:%s,%d"), file, GetLastError());
  998. return;
  999. }
  1000. // 格式化前设置语言区域;
  1001. TCHAR* old_locale = _tcsdup(_tsetlocale(LC_CTYPE, NULL));
  1002. _tsetlocale(LC_CTYPE, _T("chs")); //设定中文;
  1003. //_ftprintf(fp, _T("%04d-%02d-%02d %02d:%02d:%02d %s\n"), gmtm.tm_year + 1990, gmtm.tm_mon + 1, gmtm.tm_mday, gmtm.tm_hour, gmtm.tm_min, gmtm.tm_sec, msg);
  1004. _ftprintf(fp, _T("%s"), msg);
  1005. // 关闭文件,释放资源并设置回原语言区域;
  1006. fclose(fp);
  1007. _tsetlocale(LC_CTYPE, old_locale);
  1008. free(old_locale); //还原区域设定;
  1009. }
  1010. std::string GetLocalAddress(int nType/*= MIB_IF_TYPE_ETHERNET*/)
  1011. {
  1012. int nCardNo = 1;
  1013. // PIP_ADAPTER_INFO结构体指针存储本机网卡信息
  1014. PIP_ADAPTER_INFO pIpAdapterInfo = new IP_ADAPTER_INFO();
  1015. // 得到结构体大小,用于GetAdaptersInfo参数
  1016. unsigned long stSize = sizeof(IP_ADAPTER_INFO);
  1017. // 调用GetAdaptersInfo函数,填充pIpAdapterInfo指针变量;其中stSize参数既是一个输入量也是一个输出量
  1018. int nRel = GetAdaptersInfo(pIpAdapterInfo,&stSize);
  1019. // 记录网卡数量
  1020. int netCardNum = 0;
  1021. // 记录每张网卡上的IP地址数量
  1022. int IPnumPerNetCard = 0;
  1023. if(ERROR_BUFFER_OVERFLOW == nRel)
  1024. {
  1025. // 如果函数返回的是ERROR_BUFFER_OVERFLOW
  1026. // 则说明GetAdaptersInfo参数传递的内存空间不够,同时其传出stSize,表示需要的空间大小
  1027. // 这也是说明为什么stSize既是一个输入量也是一个输出量
  1028. // 释放原来的内存空间
  1029. delete pIpAdapterInfo;
  1030. // 重新申请内存空间用来存储所有网卡信息
  1031. pIpAdapterInfo = (PIP_ADAPTER_INFO)new BYTE[stSize];
  1032. // 再次调用GetAdaptersInfo函数,填充pIpAdapterInfo指针变量
  1033. nRel = GetAdaptersInfo(pIpAdapterInfo,&stSize);
  1034. }
  1035. if(ERROR_SUCCESS == nRel)
  1036. {
  1037. // 输出网卡信息
  1038. // 可能有多网卡,因此通过循环去判断
  1039. while (pIpAdapterInfo)
  1040. {
  1041. // 可能网卡有多IP,因此通过循环去判断
  1042. IP_ADDR_STRING*pIpAddrString = &(pIpAdapterInfo->IpAddressList);
  1043. if ( pIpAdapterInfo->Type == nType )
  1044. {
  1045. g_strIPAddress = pIpAddrString->IpAddress.String;
  1046. break;
  1047. }
  1048. #if 0
  1049. switch(pIpAdapterInfo->Type)
  1050. {
  1051. case MIB_IF_TYPE_OTHER:
  1052. case MIB_IF_TYPE_ETHERNET:
  1053. case MIB_IF_TYPE_TOKENRING:
  1054. case MIB_IF_TYPE_FDDI:
  1055. case MIB_IF_TYPE_PPP:
  1056. case MIB_IF_TYPE_LOOPBACK:
  1057. case MIB_IF_TYPE_SLIP:
  1058. {
  1059. strAddress = pIpAddrString->IpAddress.String;
  1060. // 需要注意的是有时可能获取的IP地址是0.0.0.0,这时需要过滤掉
  1061. if(std::string("0.0.0.0") == strAddress)
  1062. break;
  1063. std::cout<<_T("第")<<nCardNo<<_T("张网卡的IP地址是")<<strAddress<<std::endl;
  1064. nCardNo++;
  1065. break;
  1066. }
  1067. default:
  1068. // 未知类型网卡就跳出
  1069. break;
  1070. }
  1071. #endif
  1072. pIpAdapterInfo = pIpAdapterInfo->Next;
  1073. }
  1074. }
  1075. // 释放内存空间
  1076. if (pIpAdapterInfo)
  1077. {
  1078. delete pIpAdapterInfo;
  1079. }
  1080. return g_strIPAddress;
  1081. }
  1082. INT GetMacAddress()
  1083. {
  1084. HDEVINFO hDevInfo;
  1085. DWORD MemberIndex, RequiredSize;
  1086. SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;
  1087. PSP_DEVICE_INTERFACE_DETAIL_DATA DeviceInterfaceDetailData;
  1088. INT nTotal = 0;
  1089. INT nNICKind = 0;
  1090. // 获取设备信息集;
  1091. hDevInfo = SetupDiGetClassDevs(&MacClassGuid, NULL, NULL, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE);
  1092. if (hDevInfo == INVALID_HANDLE_VALUE)
  1093. {
  1094. return -1;
  1095. }
  1096. g_vtMac.clear();
  1097. // 枚举设备信息集中所有设备;
  1098. DeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA);
  1099. for (MemberIndex = 0; SetupDiEnumDeviceInterfaces(hDevInfo, NULL, &MacClassGuid, MemberIndex, &DeviceInterfaceData); MemberIndex++)
  1100. {
  1101. // 获取接收缓冲区大小,函数返回值为FALSE,GetLastError()=ERROR_INSUFFICIENT_BUFFER;
  1102. SetupDiGetDeviceInterfaceDetail(hDevInfo, &DeviceInterfaceData, NULL, 0, &RequiredSize, NULL);
  1103. // 申请接收缓冲区;
  1104. DeviceInterfaceDetailData = (PSP_DEVICE_INTERFACE_DETAIL_DATA)malloc(RequiredSize);
  1105. if ( DeviceInterfaceDetailData == NULL)
  1106. continue;
  1107. DeviceInterfaceDetailData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
  1108. // 获取设备细节信息;
  1109. if (SetupDiGetDeviceInterfaceDetail(hDevInfo, &DeviceInterfaceData, DeviceInterfaceDetailData, RequiredSize, NULL, NULL))
  1110. {
  1111. HANDLE hDeviceFile;
  1112. BOOL isOK = FALSE;
  1113. if (_tcsnicmp(DeviceInterfaceDetailData->DevicePath + 4, TEXT("pci"), 3) != 0)
  1114. {
  1115. free(DeviceInterfaceDetailData);
  1116. DeviceInterfaceDetailData = NULL;
  1117. continue;
  1118. }
  1119. MacAddress tagMacAddress;
  1120. tagMacAddress.nNICKind = NIC_PCI;
  1121. _stprintf_s(tagMacAddress.szNICKind, _T("%s"), _T("NIC_PCI"));
  1122. // 获取设备句柄;
  1123. hDeviceFile = CreateFile(DeviceInterfaceDetailData->DevicePath,
  1124. 0,
  1125. FILE_SHARE_READ | FILE_SHARE_WRITE,
  1126. NULL,
  1127. OPEN_EXISTING,
  1128. 0,
  1129. NULL);
  1130. if (hDeviceFile != INVALID_HANDLE_VALUE)
  1131. {
  1132. ULONG dwID;
  1133. BYTE ucData[8];
  1134. DWORD dwByteRet;
  1135. // 获取原生MAC地址;
  1136. dwID = OID_802_3_PERMANENT_ADDRESS;
  1137. isOK = DeviceIoControl(hDeviceFile, IOCTL_NDIS_QUERY_GLOBAL_STATS, &dwID, sizeof(dwID), ucData, sizeof(ucData), &dwByteRet, NULL);
  1138. if (isOK)
  1139. {
  1140. ++nTotal;
  1141. _stprintf_s(tagMacAddress.szDevicePath, _T("%s"), DeviceInterfaceDetailData->DevicePath);
  1142. memset(tagMacAddress.szMacAddress, 0, sizeof(TCHAR) * MAX_PATH);
  1143. // 将字节数组转换成16进制字符串;
  1144. for (DWORD i = 0; i < dwByteRet; i++)
  1145. {
  1146. _stprintf_s(&tagMacAddress.szMacAddress[i * 3], MAX_PATH - (i * 3), (i != dwByteRet - 1) ? _T("%02X-") : _T("%02X"), ucData[i]);
  1147. }
  1148. g_vtMac.push_back(tagMacAddress);
  1149. }
  1150. CloseHandle(hDeviceFile);
  1151. }
  1152. }
  1153. free(DeviceInterfaceDetailData);
  1154. DeviceInterfaceDetailData = NULL;
  1155. }
  1156. SetupDiDestroyDeviceInfoList(hDevInfo);
  1157. #if 1
  1158. g_strMacs.clear();
  1159. std::vector<MacAddress>::iterator it = g_vtMac.begin();
  1160. for (; it != g_vtMac.end(); it++)
  1161. {
  1162. g_strMacs.append(it->szMacAddress);
  1163. break;
  1164. }
  1165. #endif
  1166. return nTotal;
  1167. }
  1168. std::string Replace(std::string strSrc, std::string strNew, std::string strOld)
  1169. {
  1170. while(true) {
  1171. std::string::size_type pos(0);
  1172. if(( pos = strSrc.find(strOld)) != std::string::npos )
  1173. strSrc.replace(pos,strOld.length(),strNew);
  1174. else
  1175. break;
  1176. }
  1177. return strSrc;
  1178. }
  1179. };