Global.cpp 41 KB

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