Global.cpp 44 KB

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