ConsoleFSM.cpp 122 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589
  1. #include "stdafx2.h"
  2. #include "ConsoleFSM.h"
  3. #include "windows.h"
  4. #include "..\EventCode.h"
  5. #include "MaintainWatcher_server_g.h"
  6. #include "tlhelp32.h"
  7. #define OP_START_ENTITY 1
  8. #define OP_STOP_ENTITY 0
  9. #define USB_NONE 0
  10. #define USB_INSERT 1
  11. #define USB_PULLOUT 2
  12. #pragma comment(lib, "setupapi.lib")
  13. #include "Dbt.h"
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include <setupapi.h>
  17. HANDLE hDetectMutex = NULL;//only one functioning detector in a single OS
  18. HWND hDetectWnd = NULL;
  19. HANDLE hDetectThread = NULL;
  20. HDEVNOTIFY hDevNotify = NULL;
  21. DWORD g_USBStatus = USB_NONE; //USB状态值,0:无状态;1:USB插入;2:USB拔出
  22. BOOL g_FindUsbHdset = FALSE;
  23. BOOL g_FindUsbCamera = FALSE;
  24. BOOL g_FindEwsCamera = FALSE;
  25. char g_UsbHdsetName[MAX_PATH] = {0};
  26. char g_UsbCameraName[MAX_PATH] = {0};
  27. char g_UkeyName[MAX_PATH] = {0};
  28. BOOL g_FindKey = FALSE;
  29. BOOL g_FindUkey = FALSE;
  30. BOOL g_bErrStatus = FALSE; //是否处于错误状态 0:不在;1:在
  31. BOOL g_bWorkingStatus = FALSE; //上一个状态是否处于业务状态,0:不处于;1:处于
  32. //string g_UsbCameraName;
  33. BOOL g_HasVerifyTask = FALSE;
  34. USBKEY_INFO g_USBKeyInfo; //Ukey信息
  35. int g_nSurveillanceRecordAllowed;//是否使用广角摄像头,0:不使用;1:使用
  36. CSimpleStringA g_strEwsCameraID;//广角摄像头设备示例ID
  37. using namespace MaintainWatcher;
  38. class ConsoleFSM;
  39. #define MAX_MEDIA_DEVICE_NUM 10 //最大的音视频设备个数
  40. #define USB_IMAGE_DEVICE L"{6BDD1FC6-810F-11D0-BEC7-08002BE2092F}" //Audio and video devices 类GUID
  41. #define USB_MEDIA_DEVICE L"{4D36E96C-E325-11CE-BFC1-08002BE10318}" //Cameras and scanners 类GUID
  42. #define NEWUSB_IMAGE_DEVICE L"{CA3E7AB9-B4C3-4AE6-8251-579EF933890F}" //新增一类摄像头GUID,add zl 20190428
  43. #define DETECT_TEMPLATE_MW21 "USB\\VID_04B4&PID_1234" //MW21 SmartCardReader
  44. #define DETECT_TEMPLATE_GD31 "USB\\VID_096E&PID_0005" //GD31 SmartCardReader
  45. #define DETECT_TEMPLATE_GD33 "USB\\VID_096E&PID_A010" //GD33 HIDClass
  46. #define DETECT_TEMPLATE_MW23 "USB\\VID_1483&PID_C007" //MW23 HIDClass
  47. #define DETECT_TEMPLATE_GD35 "USB\\VID_05E3&PID_0723" //GD35
  48. #define DETECT_TEMPLATE_TDR61_1 "USB\\VID_1780&PID_0490" //TDR61
  49. #define DETECT_TEMPLATE_TDR61_2 "USB\\VID_1780&PID_0491" //TDR61
  50. #define DETECT_TEMPLATE_TDR61_3 "USB\\VID_1780&PID_0492" //TDR61
  51. #define DETECT_TEMPLATE_TDR61_4 "USB\\VID_1780&PID_0493" //TDR61
  52. #define DETECT_TEMPLATE_FT62 "USB\\VID_096E&PID_0708" //FT62
  53. #define DETECT_TEMPLATE_WD63 "USB\\VID_163C&PID_0704" //WD63
  54. // 增加三种新的OTP KEY类型 add by wr 20130731
  55. #define DETECT_TEMPLATE_TDR81 "USB\\VID_1780&PID_049A" //TDR81
  56. #define DETECT_TEMPLATE_FT82 "USB\\VID_096E&PID_0310" //FT82
  57. #define DETECT_TEMPLATE_WD83 "USB\\VID_163C&PID_0708" //WD83
  58. // 增加三种新的NFC KEY类型 add by wr 20150618
  59. #define DETECT_TEMPLATE_TDR91 "USB\\VID_1780&PID_049B" //TDR91
  60. #define DETECT_TEMPLATE_WDC92 "USB\\VID_1EA8&PID_C005" //WDC92
  61. #define DETECT_TEMPLATE_WD93 "USB\\VID_163C&PID_0709" //WD93
  62. // 增加三种新的一代KEY类型 add by wr 20160322
  63. #define DETECT_TEMPLATE_FT24 "USB\\VID_096E&PID_031B" //FT24
  64. // add by wr 20160328
  65. #define DETECT_TEMPLATE_WD25 "USB\\VID_163C&PID_070A" //WD25
  66. #define UKEY_NUMBER 19
  67. string g_UkeyIDList[UKEY_NUMBER]=
  68. {DETECT_TEMPLATE_MW21,
  69. DETECT_TEMPLATE_GD31,
  70. DETECT_TEMPLATE_GD33,
  71. DETECT_TEMPLATE_MW23,
  72. DETECT_TEMPLATE_GD35,
  73. DETECT_TEMPLATE_TDR61_1,
  74. DETECT_TEMPLATE_TDR61_2,
  75. DETECT_TEMPLATE_TDR61_3,
  76. DETECT_TEMPLATE_TDR61_4,
  77. DETECT_TEMPLATE_FT62,
  78. DETECT_TEMPLATE_WD63,
  79. DETECT_TEMPLATE_TDR81,
  80. DETECT_TEMPLATE_FT82,
  81. DETECT_TEMPLATE_WD83,
  82. DETECT_TEMPLATE_TDR91,
  83. DETECT_TEMPLATE_WDC92,
  84. DETECT_TEMPLATE_WD93,
  85. DETECT_TEMPLATE_FT24,
  86. DETECT_TEMPLATE_WD25};
  87. typedef struct
  88. {
  89. char cDeviceName[MAX_PATH];
  90. }DEVICE_INFO, *PDEVICE_INFO;
  91. DEVICE_INFO g_FixedAudioDeviceName[MAX_MEDIA_DEVICE_NUM] = {0};
  92. DEVICE_INFO g_FixedImageDeviceName[MAX_MEDIA_DEVICE_NUM] = {0};
  93. unsigned int g_FixedAudioDeviceNum = 0;
  94. unsigned int g_FixedImageDeviceNum = 0;
  95. GUID* StrToGUID(const WCHAR *guid_string)
  96. {
  97. char buffer[64];
  98. GUID* guid = (GUID*) buffer;
  99. WCHAR upperGuid[40+1]={0};
  100. for (int i =0 ; i < 40+1; i++)
  101. {
  102. upperGuid[i] = guid_string[i];
  103. if (upperGuid[i]>=L'a' && upperGuid[i]<=L'z')
  104. {
  105. upperGuid[i] += L'A' - L'a';
  106. }
  107. }
  108. swscanf_s(upperGuid, L"{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
  109. &guid->Data1,
  110. &guid->Data2,
  111. &guid->Data3,
  112. &guid->Data4[0],
  113. &guid->Data4[1],
  114. &guid->Data4[2],
  115. &guid->Data4[3],
  116. &guid->Data4[4],
  117. &guid->Data4[5],
  118. &guid->Data4[6],
  119. &guid->Data4[7]);
  120. return guid;
  121. }
  122. DWORD FindSubStr(const char* source, const char* target)
  123. {
  124. unsigned int i,j;
  125. if (strlen(source) < strlen(target))
  126. {
  127. return -1;
  128. }
  129. if (1 == strlen(source)
  130. && 1== strlen(target)
  131. && source[0] == target[0])
  132. {
  133. return 0;
  134. }
  135. for (i = 0;i<=strlen(source)-strlen(target);i++)
  136. {
  137. if (source[i]==target[0])
  138. {
  139. for (j = 1;j<=strlen(target);j++)
  140. {
  141. if (source[i+j]!=target[j])
  142. break;
  143. }
  144. if ((j == strlen(target))
  145. || (j == strlen(target)+1))
  146. {
  147. return i;
  148. }
  149. j = 0;
  150. }
  151. }
  152. return -1;
  153. }
  154. //判断当前设备是否是内置音频设备,说明:实体启动时检测到得音频设备都认为是内置音频设备
  155. BOOL IsFixedMediaDevice(const char* pDeviceName)
  156. {
  157. int nCount = g_FixedAudioDeviceNum;
  158. for (int i=0; i<nCount; i++)
  159. {
  160. if (0 == strcmp(g_FixedAudioDeviceName[i].cDeviceName, pDeviceName))
  161. {
  162. return TRUE;
  163. }
  164. }
  165. return FALSE;
  166. }
  167. BOOL IsFixedImageDevice(const char* pDeviceName)
  168. {
  169. int nCount = g_FixedImageDeviceNum;
  170. for (int i=0; i<nCount; i++)
  171. {
  172. if (0 == strcmp(g_FixedImageDeviceName[i].cDeviceName, pDeviceName))
  173. {
  174. return TRUE;
  175. }
  176. }
  177. return FALSE;
  178. }
  179. //判断是否是高拍仪
  180. bool IsPortableScannerDetected(LPCTSTR strDeviceID, BOOL& bUpstairOne)
  181. {
  182. //TODO: 根据良田和南天提供的设备标识名,判断插入的是否为高拍仪所有的摄像头
  183. CSimpleStringA strDevice(strDeviceID);
  184. bUpstairOne = FALSE;
  185. LPCTSTR split = strrchr(strDeviceID, '#');
  186. if(split != NULL) {
  187. if(!strnicmp(split+1, "upcamera", strlen("upcamera"))) {
  188. bUpstairOne = TRUE;
  189. return true;
  190. }
  191. else if(!strnicmp(split+1, "downcamera", strlen("downcamera"))) {
  192. return true;
  193. }
  194. }
  195. return false;
  196. }
  197. //获取耳机个数
  198. unsigned int GetMediaDeviceNum()
  199. {
  200. DWORD dwErrCode = ERROR_SUCCESS;
  201. unsigned int nDeviceNum = 0;
  202. GUID mediaGuid = *StrToGUID(L"{4D36E96C-E325-11CE-BFC1-08002BE10318}");//Audio and video devices
  203. HDEVINFO hDev = SetupDiGetClassDevs(&mediaGuid, NULL, NULL, DIGCF_PRESENT);
  204. if(INVALID_HANDLE_VALUE == hDev)
  205. {
  206. dwErrCode = GetLastError();
  207. return 0;
  208. }
  209. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  210. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  211. SetupDiGetClassImageList(&_spImageData);
  212. SP_DEVINFO_DATA DevInfo;
  213. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  214. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  215. DWORD DeviceIndex = 0;
  216. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  217. {
  218. char szBuf[MAX_PATH] = {0};
  219. int wImageIdx = 0;
  220. short wItem = 0;
  221. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  222. {
  223. DeviceIndex++;
  224. continue;
  225. }
  226. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  227. {
  228. char szName[MAX_PATH] = {0};
  229. //先获取获取设备名
  230. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  231. {
  232. nDeviceNum++;
  233. DeviceIndex++;
  234. Dbg("第[%d]个音视频设备,设备名[%s]",nDeviceNum, szName);
  235. continue;
  236. }
  237. else
  238. {
  239. //如果无法获取设备名,则获取设备描述名
  240. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  241. {
  242. nDeviceNum++;
  243. DeviceIndex++;
  244. Dbg("第[%d]个音视频设备,设备描述名[%s]",nDeviceNum, szName);
  245. continue;
  246. }
  247. }
  248. }
  249. DeviceIndex++;
  250. }
  251. SetupDiDestroyClassImageList(&_spImageData);
  252. if(hDev)
  253. {
  254. SetupDiDestroyDeviceInfoList(hDev);
  255. }
  256. return nDeviceNum;
  257. }
  258. //获取摄像头个数
  259. //unsigned int GetImageDeviceNum()
  260. //{
  261. // DWORD dwErrCode = ERROR_SUCCESS;
  262. // unsigned int nDeviceNum = 0;
  263. // GUID devcieGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  264. //
  265. // HDEVINFO hDev = SetupDiGetClassDevs(&devcieGuid, NULL, NULL, DIGCF_PRESENT);
  266. // if(INVALID_HANDLE_VALUE == hDev)
  267. // {
  268. // dwErrCode = GetLastError();
  269. // return 0;
  270. // }
  271. //
  272. // SP_CLASSIMAGELIST_DATA _spImageData = {0};
  273. // _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  274. // SetupDiGetClassImageList(&_spImageData);
  275. //
  276. // SP_DEVINFO_DATA DevInfo;
  277. // ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  278. // DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  279. // DWORD DeviceIndex = 0;
  280. //
  281. // while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  282. // {
  283. // char szBuf[MAX_PATH] = {0};
  284. // int wImageIdx = 0;
  285. // short wItem = 0;
  286. // if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  287. // {
  288. // DeviceIndex++;
  289. // continue;
  290. // }
  291. //
  292. // if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  293. // {
  294. // char szName[MAX_PATH] = {0};
  295. //
  296. // //先获取获取设备名
  297. // //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_HARDWAREID, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  298. // if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  299. // {
  300. // nDeviceNum++;
  301. // DeviceIndex++;
  302. // Dbg("第[%d]个摄像头,设备名[%s]",nDeviceNum, szName);
  303. // continue;
  304. // }
  305. // else
  306. // {
  307. // //如果无法获取设备名,则获取设备描述名
  308. // if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  309. // {
  310. // nDeviceNum++;
  311. // DeviceIndex++;
  312. // Dbg("第[%d]个摄像头,设备描述名[%s]",nDeviceNum, szName);
  313. // continue;
  314. // }
  315. // }
  316. // }
  317. //
  318. // DeviceIndex++;
  319. // }
  320. //
  321. // SetupDiDestroyClassImageList(&_spImageData);
  322. //
  323. // if(hDev)
  324. // {
  325. // SetupDiDestroyDeviceInfoList(hDev);
  326. // }
  327. //
  328. // return nDeviceNum;
  329. //}
  330. unsigned int GetImageDeviceNum()
  331. {
  332. DWORD dwErrCode = ERROR_SUCCESS;
  333. unsigned int nDeviceNum = 0;
  334. GUID devcieGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  335. HDEVINFO hDev = SetupDiGetClassDevs(&devcieGuid, NULL, NULL, DIGCF_PRESENT);
  336. if(INVALID_HANDLE_VALUE == hDev)
  337. {
  338. dwErrCode = GetLastError();
  339. return 0;
  340. }
  341. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  342. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  343. SetupDiGetClassImageList(&_spImageData);
  344. SP_DEVINFO_DATA DevInfo;
  345. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  346. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  347. DWORD DeviceIndex = 0;
  348. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  349. {
  350. char szBuf[MAX_PATH] = {0};
  351. int wImageIdx = 0;
  352. short wItem = 0;
  353. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  354. {
  355. DeviceIndex++;
  356. continue;
  357. }
  358. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  359. {
  360. char szName[MAX_PATH] = {0};
  361. //先获取获取设备名
  362. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_HARDWAREID, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  363. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  364. {
  365. nDeviceNum++;
  366. DeviceIndex++;
  367. Dbg("第[%d]个摄像头,设备名[%s]",nDeviceNum, szName);
  368. continue;
  369. }
  370. else
  371. {
  372. //如果无法获取设备名,则获取设备描述名
  373. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  374. {
  375. nDeviceNum++;
  376. DeviceIndex++;
  377. Dbg("第[%d]个摄像头,设备描述名[%s]",nDeviceNum, szName);
  378. continue;
  379. }
  380. }
  381. }
  382. DeviceIndex++;
  383. }
  384. SetupDiDestroyClassImageList(&_spImageData);
  385. if(hDev)
  386. {
  387. SetupDiDestroyDeviceInfoList(hDev);
  388. }
  389. //add by zl 20190510,增加新一类外界摄像头支持
  390. devcieGuid = *StrToGUID(NEWUSB_IMAGE_DEVICE);//Audio and video devices
  391. hDev = SetupDiGetClassDevs(&devcieGuid, NULL, NULL, DIGCF_PRESENT);
  392. if(INVALID_HANDLE_VALUE == hDev)
  393. {
  394. dwErrCode = GetLastError();
  395. return 0;
  396. }
  397. ZeroMemory(&_spImageData,sizeof(SP_CLASSIMAGELIST_DATA));
  398. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  399. SetupDiGetClassImageList(&_spImageData);
  400. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  401. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  402. DeviceIndex = 0;
  403. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  404. {
  405. char szBuf[MAX_PATH] = {0};
  406. int wImageIdx = 0;
  407. short wItem = 0;
  408. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  409. {
  410. DeviceIndex++;
  411. continue;
  412. }
  413. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  414. {
  415. char szName[MAX_PATH] = {0};
  416. //先获取获取设备名
  417. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_HARDWAREID, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  418. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  419. {
  420. nDeviceNum++;
  421. DeviceIndex++;
  422. Dbg("第[%d]个摄像头,设备名[%s]",nDeviceNum, szName);
  423. continue;
  424. }
  425. else
  426. {
  427. //如果无法获取设备名,则获取设备描述名
  428. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  429. {
  430. nDeviceNum++;
  431. DeviceIndex++;
  432. Dbg("第[%d]个摄像头,设备描述名[%s]",nDeviceNum, szName);
  433. continue;
  434. }
  435. }
  436. }
  437. DeviceIndex++;
  438. }
  439. SetupDiDestroyClassImageList(&_spImageData);
  440. if(hDev)
  441. {
  442. SetupDiDestroyDeviceInfoList(hDev);
  443. }
  444. return nDeviceNum;
  445. }
  446. //获取高拍仪个数
  447. unsigned int GetPortableScannerNum()
  448. {
  449. DWORD dwErrCode = ERROR_SUCCESS;
  450. unsigned int nDeviceNum = 0;
  451. GUID devcieGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  452. HDEVINFO hDev = SetupDiGetClassDevs(&devcieGuid, NULL, NULL, DIGCF_PRESENT);
  453. if(INVALID_HANDLE_VALUE == hDev)
  454. {
  455. dwErrCode = GetLastError();
  456. return 0;
  457. }
  458. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  459. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  460. SetupDiGetClassImageList(&_spImageData);
  461. SP_DEVINFO_DATA DevInfo;
  462. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  463. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  464. DWORD DeviceIndex = 0;
  465. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  466. {
  467. char szBuf[MAX_PATH] = {0};
  468. int wImageIdx = 0;
  469. short wItem = 0;
  470. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  471. {
  472. DeviceIndex++;
  473. continue;
  474. }
  475. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  476. {
  477. char szName[MAX_PATH] = {0};
  478. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  479. {
  480. BOOL bUpstair = FALSE;
  481. if (IsPortableScannerDetected(szName, bUpstair))
  482. {
  483. nDeviceNum++;
  484. DeviceIndex++;
  485. Dbg("第[%d]个高拍仪,FRIENDLYNAME[%s]",nDeviceNum, szName);
  486. continue;
  487. }
  488. }
  489. }
  490. DeviceIndex++;
  491. }
  492. SetupDiDestroyClassImageList(&_spImageData);
  493. if(hDev)
  494. {
  495. SetupDiDestroyDeviceInfoList(hDev);
  496. }
  497. return nDeviceNum;
  498. }
  499. //查找USB耳机
  500. BOOL FindNewMediaDevice(char** pDeviceName)
  501. {
  502. if (NULL == pDeviceName)
  503. {
  504. return FALSE;
  505. }
  506. DWORD dwErrCode = ERROR_SUCCESS;
  507. BOOL bFind = FALSE;
  508. GUID mediaGuid = *StrToGUID(L"{4D36E96C-E325-11CE-BFC1-08002BE10318}");//Audio and video devices
  509. HDEVINFO hDev = SetupDiGetClassDevs(&mediaGuid, NULL, NULL, DIGCF_PRESENT);
  510. if(INVALID_HANDLE_VALUE == hDev)
  511. {
  512. dwErrCode = GetLastError();
  513. return FALSE;
  514. }
  515. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  516. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  517. SetupDiGetClassImageList(&_spImageData);
  518. SP_DEVINFO_DATA DevInfo;
  519. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  520. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  521. DWORD DeviceIndex = 0;
  522. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  523. {
  524. char szBuf[MAX_PATH] = {0};
  525. int wImageIdx = 0;
  526. short wItem = 0;
  527. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  528. {
  529. DeviceIndex++;
  530. continue;
  531. }
  532. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  533. {
  534. char szName[MAX_PATH] = {0};
  535. //先获取获取设备名
  536. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  537. {
  538. if (IsFixedMediaDevice(szName))
  539. {
  540. //如果是内置音频设备则继续找下一个
  541. DeviceIndex++;
  542. continue;
  543. }
  544. memcpy((void*)g_UsbHdsetName, szName, MAX_PATH);
  545. memcpy(*pDeviceName, szName, MAX_PATH);
  546. bFind = TRUE;
  547. break;
  548. }
  549. else
  550. {
  551. //如果无法获取设备名,则获取设备描述名
  552. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  553. {
  554. if (IsFixedMediaDevice(szName))
  555. {
  556. //如果是内置音频设备则继续找下一个
  557. DeviceIndex++;
  558. continue;
  559. }
  560. memcpy((void*)g_UsbHdsetName, szName, MAX_PATH);
  561. memcpy(*pDeviceName, szName, MAX_PATH);
  562. bFind = TRUE;
  563. break;
  564. }
  565. }
  566. }
  567. DeviceIndex++;
  568. }
  569. SetupDiDestroyClassImageList(&_spImageData);
  570. if(hDev)
  571. {
  572. SetupDiDestroyDeviceInfoList(hDev);
  573. }
  574. return bFind;
  575. }
  576. //查找摄像头
  577. //BOOL FindNewImageDevice(char** pDeviceName)
  578. //{
  579. // if (NULL == pDeviceName)
  580. // {
  581. // return FALSE;
  582. // }
  583. //
  584. // DWORD dwErrCode = ERROR_SUCCESS;
  585. // BOOL bFind = FALSE;
  586. // GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  587. //
  588. // HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  589. // if(INVALID_HANDLE_VALUE == hDev)
  590. // {
  591. // dwErrCode = GetLastError();
  592. // return FALSE;
  593. // }
  594. //
  595. // SP_CLASSIMAGELIST_DATA _spImageData = {0};
  596. // _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  597. // SetupDiGetClassImageList(&_spImageData);
  598. //
  599. // SP_DEVINFO_DATA DevInfo;
  600. // ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  601. // DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  602. // DWORD DeviceIndex = 0;
  603. //
  604. // while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  605. // {
  606. // char szBuf[MAX_PATH] = {0};
  607. // int wImageIdx = 0;
  608. // short wItem = 0;
  609. // if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  610. // {
  611. // DeviceIndex++;
  612. // continue;
  613. // }
  614. //
  615. // if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  616. // {
  617. // char szName[MAX_PATH] = {0};
  618. // char szFriendlyName[MAX_PATH] = {0};
  619. //
  620. // //add by zl 20171109 如果是高拍仪,直接返回不处理
  621. // if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szFriendlyName, MAX_PATH - 1, 0))
  622. // {
  623. // BOOL bUpstair = FALSE;
  624. // if (IsPortableScannerDetected(szFriendlyName, bUpstair))
  625. // {
  626. // DeviceIndex++;
  627. // continue;
  628. // }
  629. // }
  630. //
  631. // //先获取获取设备名
  632. // if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  633. // //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  634. // {
  635. // if (IsFixedImageDevice(szName))
  636. // {
  637. // //如果是内置音频设备则继续找下一个
  638. // DeviceIndex++;
  639. // continue;
  640. // }
  641. //
  642. // memset((void*)g_UsbCameraName, 0, MAX_PATH);
  643. // memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  644. // memcpy(*pDeviceName, szName, MAX_PATH);
  645. // bFind = TRUE;
  646. // break;
  647. // }
  648. // else
  649. // {
  650. // //如果无法获取设备名,则获取设备描述名
  651. // if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  652. // {
  653. // if (IsFixedImageDevice(szName))
  654. // {
  655. // //如果是内置音频设备则继续找下一个
  656. // DeviceIndex++;
  657. // continue;
  658. // }
  659. // memset((void*)g_UsbCameraName, 0, MAX_PATH);
  660. // memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  661. // memcpy(*pDeviceName, szName, MAX_PATH);
  662. // bFind = TRUE;
  663. // break;
  664. // }
  665. // }
  666. // }
  667. //
  668. // DeviceIndex++;
  669. // }
  670. //
  671. // SetupDiDestroyClassImageList(&_spImageData);
  672. //
  673. // if(hDev)
  674. // {
  675. // SetupDiDestroyDeviceInfoList(hDev);
  676. // }
  677. //
  678. // return bFind;
  679. //}
  680. //查找新的广角摄像头
  681. BOOL FindNewImageDevice(char** pDeviceName)
  682. {
  683. if (NULL == pDeviceName)
  684. {
  685. return FALSE;
  686. }
  687. DWORD dwErrCode = ERROR_SUCCESS;
  688. BOOL bFind = FALSE;
  689. GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  690. HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  691. if(INVALID_HANDLE_VALUE == hDev)
  692. {
  693. dwErrCode = GetLastError();
  694. return FALSE;
  695. }
  696. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  697. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  698. SetupDiGetClassImageList(&_spImageData);
  699. SP_DEVINFO_DATA DevInfo;
  700. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  701. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  702. DWORD DeviceIndex = 0;
  703. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  704. {
  705. char szBuf[MAX_PATH] = {0};
  706. int wImageIdx = 0;
  707. short wItem = 0;
  708. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  709. {
  710. DeviceIndex++;
  711. continue;
  712. }
  713. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  714. {
  715. char szName[MAX_PATH] = {0};
  716. char szFriendlyName[MAX_PATH] = {0};
  717. //add by zl 20171109 如果是高拍仪,直接返回不处理
  718. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szFriendlyName, MAX_PATH - 1, 0))
  719. {
  720. BOOL bUpstair = FALSE;
  721. if (IsPortableScannerDetected(szFriendlyName, bUpstair))
  722. {
  723. DeviceIndex++;
  724. continue;
  725. }
  726. }
  727. //先获取获取设备名
  728. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  729. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  730. {
  731. if (IsFixedImageDevice(szName))
  732. {
  733. //如果是内置音频设备则继续找下一个
  734. DeviceIndex++;
  735. continue;
  736. }
  737. memset((void*)g_UsbCameraName, 0, MAX_PATH);
  738. memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  739. memcpy(*pDeviceName, szName, MAX_PATH);
  740. bFind = TRUE;
  741. break;
  742. }
  743. else
  744. {
  745. //如果无法获取设备名,则获取设备描述名
  746. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  747. {
  748. if (IsFixedImageDevice(szName))
  749. {
  750. //如果是内置音频设备则继续找下一个
  751. DeviceIndex++;
  752. continue;
  753. }
  754. memset((void*)g_UsbCameraName, 0, MAX_PATH);
  755. memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  756. memcpy(*pDeviceName, szName, MAX_PATH);
  757. bFind = TRUE;
  758. break;
  759. }
  760. }
  761. }
  762. DeviceIndex++;
  763. }
  764. SetupDiDestroyClassImageList(&_spImageData);
  765. if(hDev)
  766. {
  767. SetupDiDestroyDeviceInfoList(hDev);
  768. }
  769. if (bFind)
  770. {
  771. return TRUE;
  772. }
  773. //add by zl 20190510,增加新一类外界摄像头支持
  774. deviceGuid = *StrToGUID(NEWUSB_IMAGE_DEVICE);//Audio and video devices
  775. hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  776. if(INVALID_HANDLE_VALUE == hDev)
  777. {
  778. dwErrCode = GetLastError();
  779. return FALSE;
  780. }
  781. ZeroMemory(&_spImageData, sizeof(SP_CLASSIMAGELIST_DATA));
  782. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  783. SetupDiGetClassImageList(&_spImageData);
  784. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  785. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  786. DeviceIndex = 0;
  787. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  788. {
  789. char szBuf[MAX_PATH] = {0};
  790. int wImageIdx = 0;
  791. short wItem = 0;
  792. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  793. {
  794. DeviceIndex++;
  795. continue;
  796. }
  797. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  798. {
  799. char szName[MAX_PATH] = {0};
  800. char szFriendlyName[MAX_PATH] = {0};
  801. //add by zl 20171109 如果是高拍仪,直接返回不处理
  802. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szFriendlyName, MAX_PATH - 1, 0))
  803. {
  804. BOOL bUpstair = FALSE;
  805. if (IsPortableScannerDetected(szFriendlyName, bUpstair))
  806. {
  807. DeviceIndex++;
  808. continue;
  809. }
  810. }
  811. //先获取获取设备名
  812. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  813. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  814. {
  815. if (IsFixedImageDevice(szName))
  816. {
  817. //如果是内置音频设备则继续找下一个
  818. DeviceIndex++;
  819. continue;
  820. }
  821. memset((void*)g_UsbCameraName, 0, MAX_PATH);
  822. memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  823. memcpy(*pDeviceName, szName, MAX_PATH);
  824. bFind = TRUE;
  825. break;
  826. }
  827. else
  828. {
  829. //如果无法获取设备名,则获取设备描述名
  830. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  831. {
  832. if (IsFixedImageDevice(szName))
  833. {
  834. //如果是内置音频设备则继续找下一个
  835. DeviceIndex++;
  836. continue;
  837. }
  838. memset((void*)g_UsbCameraName, 0, MAX_PATH);
  839. memcpy((void*)g_UsbCameraName, szName, MAX_PATH);
  840. memcpy(*pDeviceName, szName, MAX_PATH);
  841. bFind = TRUE;
  842. break;
  843. }
  844. }
  845. }
  846. DeviceIndex++;
  847. }
  848. SetupDiDestroyClassImageList(&_spImageData);
  849. if(hDev)
  850. {
  851. SetupDiDestroyDeviceInfoList(hDev);
  852. }
  853. return bFind;
  854. }
  855. //查找新的广角摄像头
  856. BOOL FindNewEwsCamera()
  857. {
  858. DWORD dwErrCode = ERROR_SUCCESS;
  859. BOOL bFind = FALSE;
  860. GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  861. HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  862. if(INVALID_HANDLE_VALUE == hDev)
  863. {
  864. dwErrCode = GetLastError();
  865. return FALSE;
  866. }
  867. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  868. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  869. SetupDiGetClassImageList(&_spImageData);
  870. SP_DEVINFO_DATA DevInfo;
  871. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  872. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  873. DWORD DeviceIndex = 0;
  874. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  875. {
  876. char szBuf[MAX_PATH] = {0};
  877. int wImageIdx = 0;
  878. short wItem = 0;
  879. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  880. {
  881. DeviceIndex++;
  882. continue;
  883. }
  884. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  885. {
  886. char szName[MAX_PATH] = {0};
  887. //先获取获取设备名
  888. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  889. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  890. {
  891. if (IsFixedImageDevice(szName))
  892. {
  893. //如果是内置音频设备则继续找下一个
  894. DeviceIndex++;
  895. continue;
  896. }
  897. if (0 == _stricmp(szName, g_strEwsCameraID.GetData()))
  898. {
  899. bFind = TRUE;
  900. break;
  901. }
  902. }
  903. else
  904. {
  905. //如果无法获取设备名,则获取设备描述名
  906. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  907. {
  908. if (IsFixedImageDevice(szName))
  909. {
  910. //如果是内置音频设备则继续找下一个
  911. DeviceIndex++;
  912. continue;
  913. }
  914. if (0 == _stricmp(szName, g_strEwsCameraID.GetData()))
  915. {
  916. bFind = TRUE;
  917. break;
  918. }
  919. }
  920. }
  921. }
  922. DeviceIndex++;
  923. }
  924. SetupDiDestroyClassImageList(&_spImageData);
  925. if(hDev)
  926. {
  927. SetupDiDestroyDeviceInfoList(hDev);
  928. }
  929. return bFind;
  930. }
  931. //查找指定的USB耳机
  932. BOOL FindMyMediaDevice(const char* pDeviceName)
  933. {
  934. if (NULL == pDeviceName)
  935. {
  936. return FALSE;
  937. }
  938. DWORD dwErrCode = ERROR_SUCCESS;
  939. BOOL bFind = FALSE;
  940. GUID mediaGuid = *StrToGUID(L"{4D36E96C-E325-11CE-BFC1-08002BE10318}");//Audio and video devices
  941. HDEVINFO hDev = SetupDiGetClassDevs(&mediaGuid, NULL, NULL, DIGCF_PRESENT);
  942. if(INVALID_HANDLE_VALUE == hDev)
  943. {
  944. dwErrCode = GetLastError();
  945. return FALSE;
  946. }
  947. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  948. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  949. SetupDiGetClassImageList(&_spImageData);
  950. SP_DEVINFO_DATA DevInfo;
  951. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  952. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  953. DWORD DeviceIndex = 0;
  954. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  955. {
  956. char szBuf[MAX_PATH] = {0};
  957. int wImageIdx = 0;
  958. short wItem = 0;
  959. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  960. {
  961. DeviceIndex++;
  962. continue;
  963. }
  964. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  965. {
  966. char szName[MAX_PATH] = {0};
  967. //先获取获取设备名
  968. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  969. {
  970. if (0 == strcmp(pDeviceName, (const char*)szName))
  971. {
  972. bFind = TRUE;
  973. }
  974. break;
  975. }
  976. else
  977. {
  978. //如果无法获取设备名,则获取设备描述名
  979. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  980. {
  981. if (0 == strcmp(pDeviceName, (const char*)szName))
  982. {
  983. bFind = TRUE;
  984. }
  985. break;
  986. }
  987. }
  988. }
  989. DeviceIndex++;
  990. }
  991. SetupDiDestroyClassImageList(&_spImageData);
  992. if(hDev)
  993. {
  994. SetupDiDestroyDeviceInfoList(hDev);
  995. }
  996. return bFind;
  997. }
  998. //查找指定的摄像头
  999. BOOL FindMyImageDevice(const char* pDeviceName)
  1000. {
  1001. if (NULL == pDeviceName)
  1002. {
  1003. Dbg("pDeviceName is null");
  1004. return FALSE;
  1005. }
  1006. DWORD dwErrCode = ERROR_SUCCESS;
  1007. BOOL bFind = FALSE;
  1008. GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  1009. HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  1010. if(INVALID_HANDLE_VALUE == hDev)
  1011. {
  1012. Dbg("SetupDiGetClassDevs failed");
  1013. dwErrCode = GetLastError();
  1014. return FALSE;
  1015. }
  1016. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  1017. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1018. SetupDiGetClassImageList(&_spImageData);
  1019. SP_DEVINFO_DATA DevInfo;
  1020. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1021. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1022. DWORD DeviceIndex = 0;
  1023. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1024. {
  1025. char szBuf[MAX_PATH] = {0};
  1026. int wImageIdx = 0;
  1027. short wItem = 0;
  1028. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1029. {
  1030. Dbg("SetupDiGetDeviceRegistryPropertyA failed");
  1031. DeviceIndex++;
  1032. continue;
  1033. }
  1034. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1035. {
  1036. char szName[MAX_PATH] = {0};
  1037. //先获取获取设备名
  1038. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1039. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1040. {
  1041. Dbg("DeviceInstanceId[%s]",szName);
  1042. if (0 == strcmp(pDeviceName, (const char*)szName))
  1043. {
  1044. bFind = TRUE;
  1045. break;
  1046. }
  1047. }
  1048. else
  1049. {
  1050. //如果无法获取设备名,则获取设备描述名
  1051. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1052. {
  1053. Dbg("SPDRP_DEVICEDESC[%s]",szName);
  1054. if (0 == strcmp(pDeviceName, (const char*)szName))
  1055. {
  1056. bFind = TRUE;
  1057. break;
  1058. }
  1059. }
  1060. else
  1061. {
  1062. Dbg("找不到此设备示例ID和设备描述名");
  1063. }
  1064. }
  1065. }
  1066. DeviceIndex++;
  1067. }
  1068. SetupDiDestroyClassImageList(&_spImageData);
  1069. if(hDev)
  1070. {
  1071. SetupDiDestroyDeviceInfoList(hDev);
  1072. }
  1073. if (bFind)
  1074. {
  1075. return TRUE;
  1076. }
  1077. //add by zl 20190510,增加新一类外界摄像头支持
  1078. deviceGuid = *StrToGUID(NEWUSB_IMAGE_DEVICE);//Audio and video devices
  1079. hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  1080. if(INVALID_HANDLE_VALUE == hDev)
  1081. {
  1082. Dbg("SetupDiGetClassDevs failed");
  1083. dwErrCode = GetLastError();
  1084. return FALSE;
  1085. }
  1086. ZeroMemory(&_spImageData, sizeof(SP_CLASSIMAGELIST_DATA));
  1087. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1088. SetupDiGetClassImageList(&_spImageData);
  1089. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1090. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1091. DeviceIndex = 0;
  1092. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1093. {
  1094. char szBuf[MAX_PATH] = {0};
  1095. int wImageIdx = 0;
  1096. short wItem = 0;
  1097. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1098. {
  1099. Dbg("SetupDiGetDeviceRegistryPropertyA failed");
  1100. DeviceIndex++;
  1101. continue;
  1102. }
  1103. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1104. {
  1105. char szName[MAX_PATH] = {0};
  1106. //先获取获取设备名
  1107. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1108. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1109. {
  1110. Dbg("DeviceInstanceId[%s]",szName);
  1111. if (0 == strcmp(pDeviceName, (const char*)szName))
  1112. {
  1113. bFind = TRUE;
  1114. break;
  1115. }
  1116. }
  1117. else
  1118. {
  1119. //如果无法获取设备名,则获取设备描述名
  1120. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1121. {
  1122. Dbg("SPDRP_DEVICEDESC[%s]",szName);
  1123. if (0 == strcmp(pDeviceName, (const char*)szName))
  1124. {
  1125. bFind = TRUE;
  1126. break;
  1127. }
  1128. }
  1129. else
  1130. {
  1131. Dbg("找不到此设备示例ID和设备描述名");
  1132. }
  1133. }
  1134. }
  1135. DeviceIndex++;
  1136. }
  1137. SetupDiDestroyClassImageList(&_spImageData);
  1138. if(hDev)
  1139. {
  1140. SetupDiDestroyDeviceInfoList(hDev);
  1141. }
  1142. return bFind;
  1143. }
  1144. //查找内置音频设备
  1145. BOOL FindFixedMediaDevice()
  1146. {
  1147. DWORD dwErrCode = ERROR_SUCCESS;
  1148. BOOL bFind = FALSE;
  1149. GUID mediaGuid = *StrToGUID(L"{4D36E96C-E325-11CE-BFC1-08002BE10318}");//Audio and video devices
  1150. HDEVINFO hDev = SetupDiGetClassDevs(&mediaGuid, NULL, NULL, DIGCF_PRESENT);
  1151. if(INVALID_HANDLE_VALUE == hDev)
  1152. {
  1153. dwErrCode = GetLastError();
  1154. return FALSE;
  1155. }
  1156. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  1157. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1158. SetupDiGetClassImageList(&_spImageData);
  1159. SP_DEVINFO_DATA DevInfo;
  1160. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1161. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1162. DWORD DeviceIndex = 0;
  1163. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1164. {
  1165. char szBuf[MAX_PATH] = {0};
  1166. int wImageIdx = 0;
  1167. short wItem = 0;
  1168. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1169. {
  1170. DeviceIndex++;
  1171. continue;
  1172. }
  1173. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1174. {
  1175. char szName[MAX_PATH] = {0};
  1176. //先获取获取设备名
  1177. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1178. {
  1179. g_FixedAudioDeviceNum++;
  1180. memcpy((void*)g_FixedAudioDeviceName[g_FixedAudioDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1181. bFind = TRUE;
  1182. DeviceIndex++;
  1183. continue;
  1184. }
  1185. else
  1186. {
  1187. //如果无法获取设备名,则获取设备描述名
  1188. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1189. {
  1190. g_FixedAudioDeviceNum++;
  1191. memcpy((void*)g_FixedAudioDeviceName[g_FixedAudioDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1192. bFind = TRUE;
  1193. DeviceIndex++;
  1194. continue;
  1195. }
  1196. }
  1197. }
  1198. DeviceIndex++;
  1199. }
  1200. SetupDiDestroyClassImageList(&_spImageData);
  1201. if(hDev)
  1202. {
  1203. SetupDiDestroyDeviceInfoList(hDev);
  1204. }
  1205. return bFind;
  1206. }
  1207. //查找内置摄像头
  1208. BOOL FindFixedImageDevice()
  1209. {
  1210. DWORD dwErrCode = ERROR_SUCCESS;
  1211. BOOL bFind = FALSE;
  1212. GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  1213. HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  1214. if(INVALID_HANDLE_VALUE == hDev)
  1215. {
  1216. dwErrCode = GetLastError();
  1217. return FALSE;
  1218. }
  1219. SP_CLASSIMAGELIST_DATA _spImageData = {0};
  1220. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1221. SetupDiGetClassImageList(&_spImageData);
  1222. SP_DEVINFO_DATA DevInfo;
  1223. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1224. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1225. DWORD DeviceIndex = 0;
  1226. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1227. {
  1228. char szBuf[MAX_PATH] = {0};
  1229. int wImageIdx = 0;
  1230. short wItem = 0;
  1231. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1232. {
  1233. DeviceIndex++;
  1234. continue;
  1235. }
  1236. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1237. {
  1238. char szName[MAX_PATH] = {0};
  1239. //先获取获取设备名
  1240. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1241. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1242. {
  1243. g_FixedImageDeviceNum++;
  1244. memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1245. bFind = TRUE;
  1246. DeviceIndex++;
  1247. continue;
  1248. }
  1249. else
  1250. {
  1251. //如果无法获取设备名,则获取设备描述名
  1252. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1253. {
  1254. g_FixedImageDeviceNum++;
  1255. memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1256. bFind = TRUE;
  1257. DeviceIndex++;
  1258. continue;
  1259. }
  1260. }
  1261. }
  1262. DeviceIndex++;
  1263. }
  1264. SetupDiDestroyClassImageList(&_spImageData);
  1265. if(hDev)
  1266. {
  1267. SetupDiDestroyDeviceInfoList(hDev);
  1268. }
  1269. //add by zl 20190510,增加新一类外界摄像头支持
  1270. deviceGuid = *StrToGUID(NEWUSB_IMAGE_DEVICE);//Audio and video devices
  1271. hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  1272. if(INVALID_HANDLE_VALUE == hDev)
  1273. {
  1274. dwErrCode = GetLastError();
  1275. return FALSE;
  1276. }
  1277. ZeroMemory(&_spImageData, sizeof(SP_CLASSIMAGELIST_DATA));
  1278. _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1279. SetupDiGetClassImageList(&_spImageData);
  1280. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1281. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1282. DeviceIndex = 0;
  1283. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1284. {
  1285. char szBuf[MAX_PATH] = {0};
  1286. int wImageIdx = 0;
  1287. short wItem = 0;
  1288. if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1289. {
  1290. DeviceIndex++;
  1291. continue;
  1292. }
  1293. if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1294. {
  1295. char szName[MAX_PATH] = {0};
  1296. //先获取获取设备名
  1297. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1298. //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1299. {
  1300. g_FixedImageDeviceNum++;
  1301. memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1302. bFind = TRUE;
  1303. DeviceIndex++;
  1304. continue;
  1305. }
  1306. else
  1307. {
  1308. //如果无法获取设备名,则获取设备描述名
  1309. if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1310. {
  1311. g_FixedImageDeviceNum++;
  1312. memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1313. bFind = TRUE;
  1314. DeviceIndex++;
  1315. continue;
  1316. }
  1317. }
  1318. }
  1319. DeviceIndex++;
  1320. }
  1321. SetupDiDestroyClassImageList(&_spImageData);
  1322. if(hDev)
  1323. {
  1324. SetupDiDestroyDeviceInfoList(hDev);
  1325. }
  1326. return bFind;
  1327. }
  1328. //BOOL FindFixedImageDevice()
  1329. //{
  1330. // DWORD dwErrCode = ERROR_SUCCESS;
  1331. // BOOL bFind = FALSE;
  1332. // GUID deviceGuid = *StrToGUID(USB_IMAGE_DEVICE);//Audio and video devices
  1333. //
  1334. // HDEVINFO hDev = SetupDiGetClassDevs(&deviceGuid, NULL, NULL, DIGCF_PRESENT);
  1335. // if(INVALID_HANDLE_VALUE == hDev)
  1336. // {
  1337. // dwErrCode = GetLastError();
  1338. // return FALSE;
  1339. // }
  1340. //
  1341. // SP_CLASSIMAGELIST_DATA _spImageData = {0};
  1342. // _spImageData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
  1343. // SetupDiGetClassImageList(&_spImageData);
  1344. //
  1345. // SP_DEVINFO_DATA DevInfo;
  1346. // ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1347. // DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1348. // DWORD DeviceIndex = 0;
  1349. //
  1350. // while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1351. // {
  1352. // char szBuf[MAX_PATH] = {0};
  1353. // int wImageIdx = 0;
  1354. // short wItem = 0;
  1355. // if (!SetupDiGetDeviceRegistryPropertyA(hDev, &DevInfo, SPDRP_CLASS, NULL, (PBYTE)szBuf, MAX_PATH, 0))
  1356. // {
  1357. // DeviceIndex++;
  1358. // continue;
  1359. // }
  1360. //
  1361. // if (SetupDiGetClassImageIndex(&_spImageData, &DevInfo.ClassGuid, &wImageIdx))
  1362. // {
  1363. // char szName[MAX_PATH] = {0};
  1364. //
  1365. // //先获取获取设备名
  1366. // if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1367. // //if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_FRIENDLYNAME, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1368. // {
  1369. // g_FixedImageDeviceNum++;
  1370. // memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1371. // bFind = TRUE;
  1372. // DeviceIndex++;
  1373. // continue;
  1374. // }
  1375. // else
  1376. // {
  1377. // //如果无法获取设备名,则获取设备描述名
  1378. // if (SetupDiGetDeviceRegistryProperty(hDev,&DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)szName, MAX_PATH - 1, 0))
  1379. // {
  1380. // g_FixedImageDeviceNum++;
  1381. // memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum-1].cDeviceName, szName, MAX_PATH);
  1382. // bFind = TRUE;
  1383. // DeviceIndex++;
  1384. // continue;
  1385. // }
  1386. // }
  1387. // }
  1388. //
  1389. // DeviceIndex++;
  1390. // }
  1391. //
  1392. // SetupDiDestroyClassImageList(&_spImageData);
  1393. //
  1394. // if(hDev)
  1395. // {
  1396. // SetupDiDestroyDeviceInfoList(hDev);
  1397. // }
  1398. //
  1399. // return bFind;
  1400. //}
  1401. //查找被拔出的摄像头
  1402. BOOL FindOutputImageDevice(char* pDeviceName)
  1403. {
  1404. if (NULL == pDeviceName)
  1405. {
  1406. return FALSE;
  1407. }
  1408. for (unsigned int i=0; i<g_FixedImageDeviceNum; i++)
  1409. {
  1410. if (FindMyImageDevice(g_FixedImageDeviceName[i].cDeviceName))
  1411. {
  1412. continue;
  1413. }
  1414. else
  1415. {
  1416. memcpy(pDeviceName,g_FixedImageDeviceName[i].cDeviceName,MAX_PATH);
  1417. return TRUE;
  1418. }
  1419. }
  1420. return FALSE;
  1421. }
  1422. //检测是否插入专业版UKEY
  1423. BOOL FindUkey()
  1424. {
  1425. DWORD dwErrCode = ERROR_SUCCESS;
  1426. BOOL bFind = FALSE;
  1427. HDEVINFO hDev = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_PRESENT|DIGCF_ALLCLASSES);
  1428. if(INVALID_HANDLE_VALUE == hDev)
  1429. {
  1430. Dbg("SetupDiGetClassDevs failed");
  1431. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_UKEY,
  1432. GetOutPutStr("%s%d","SetupDiGetClassDevs",hDev).c_str());
  1433. dwErrCode = GetLastError();
  1434. return FALSE;
  1435. }
  1436. SP_DEVINFO_DATA DevInfo;
  1437. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1438. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1439. DWORD DeviceIndex = 0;
  1440. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1441. {
  1442. char szBuf[MAX_PATH] = {0};
  1443. int wImageIdx = 0;
  1444. short wItem = 0;
  1445. char szName[MAX_PATH] = {0};
  1446. //获取获取设备示例ID
  1447. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1448. {
  1449. //Dbg("DeviceInstanceId[%s]",szName);
  1450. for (int i=0; i<UKEY_NUMBER; i++)
  1451. {
  1452. if (-1 != FindSubStr(szName, g_UkeyIDList[i].c_str()))
  1453. {
  1454. Dbg("find UKey[%s]",g_UkeyIDList[i].c_str());
  1455. string DebugInfoStr = "find u key,id is" + g_UkeyIDList[i] + ".";
  1456. memset((void*)g_UkeyName, 0, MAX_PATH);
  1457. memcpy((void*)g_UkeyName, szName, MAX_PATH);
  1458. if(hDev)
  1459. {
  1460. SetupDiDestroyDeviceInfoList(hDev);
  1461. }
  1462. return TRUE;
  1463. }
  1464. }
  1465. }
  1466. else
  1467. {
  1468. Dbg("找不到此USB设备的实例ID");
  1469. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_UKEY_GET_INSTANCEID,
  1470. GetOutPutStr("%s%s", "SetupDiGetDeviceInstanceId", "False").c_str());
  1471. }
  1472. DeviceIndex++;
  1473. }
  1474. if(hDev)
  1475. {
  1476. SetupDiDestroyDeviceInfoList(hDev);
  1477. }
  1478. return bFind;
  1479. }
  1480. //查找指定的专业版UKEY
  1481. BOOL FindTheUkey(const char* pUkeyID)
  1482. {
  1483. if (NULL == pUkeyID)
  1484. {
  1485. Dbg("pUkeyID is null");
  1486. LogWarn(Severity_Low,Error_Unexpect,ERR_MAINTAINWATCHER_THE_UKEY,
  1487. GetOutPutStr("%s%s", "pUkeyID", "NULL").c_str());
  1488. return FALSE;
  1489. }
  1490. DWORD dwErrCode = ERROR_SUCCESS;
  1491. BOOL bFind = FALSE;
  1492. HDEVINFO hDev = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_PRESENT|DIGCF_ALLCLASSES);
  1493. if(INVALID_HANDLE_VALUE == hDev)
  1494. {
  1495. Dbg("SetupDiGetClassDevs failed");
  1496. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_THE_UKEY,
  1497. GetOutPutStr("%s%d", "SetupDiGetClassDevs", hDev).c_str());
  1498. dwErrCode = GetLastError();
  1499. return FALSE;
  1500. }
  1501. SP_DEVINFO_DATA DevInfo;
  1502. ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
  1503. DevInfo.cbSize = sizeof(SP_DEVINFO_DATA);
  1504. DWORD DeviceIndex = 0;
  1505. while(SetupDiEnumDeviceInfo(hDev, DeviceIndex, &DevInfo))
  1506. {
  1507. char szBuf[MAX_PATH] = {0};
  1508. int wImageIdx = 0;
  1509. short wItem = 0;
  1510. char szName[MAX_PATH] = {0};
  1511. //获取获取设备示例ID
  1512. if(SetupDiGetDeviceInstanceId(hDev,&DevInfo,szName,MAX_PATH - 1,NULL))
  1513. {
  1514. //Dbg("DeviceInstanceId[%s]",szName);
  1515. if (-1 != FindSubStr(szName, pUkeyID))
  1516. {
  1517. Dbg("find the UKey[%s]",pUkeyID);
  1518. bFind = TRUE;
  1519. break;
  1520. }
  1521. }
  1522. else
  1523. {
  1524. Dbg("找不到此USB设备的实例ID");
  1525. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_THE_UKEY_GET_INSTANCEID,
  1526. GetOutPutStr("%s%s", "SetupDiGetDeviceInstanceId", "False").c_str());
  1527. }
  1528. DeviceIndex++;
  1529. }
  1530. if(hDev)
  1531. {
  1532. SetupDiDestroyDeviceInfoList(hDev);
  1533. }
  1534. return bFind;
  1535. }
  1536. LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
  1537. {
  1538. int i = 0;
  1539. unsigned int uiDeviceNum = 0;
  1540. unsigned int uiCameraNum = 0;
  1541. char szMessage[128] = {0};
  1542. char* pName = new char[MAX_PATH];
  1543. char* pAudioName = new char[MAX_PATH];
  1544. memset(pName, 0, MAX_PATH);
  1545. memset(pAudioName, 0, MAX_PATH);
  1546. sprintf(szMessage, "message=%d\n", message);
  1547. OutputDebugString(szMessage);
  1548. switch (message)
  1549. {
  1550. case WM_DESTROY:
  1551. PostQuitMessage(0);
  1552. break;
  1553. case WM_DEVICECHANGE:
  1554. {
  1555. switch(wParam)
  1556. {
  1557. case DBT_DEVICEARRIVAL:
  1558. {
  1559. Sleep(1000);
  1560. g_USBStatus = USB_INSERT;
  1561. LogEvent(Severity_Middle, EVENT_USB_INSERT, "USB insert");
  1562. Dbg("检测到USB设备插入.");
  1563. //把广角摄像头当做一个内置摄像头
  1564. if (1 == g_nSurveillanceRecordAllowed)
  1565. {
  1566. //
  1567. if (g_strEwsCameraID.GetLength() < 1)
  1568. {
  1569. Dbg("Root.ini没有配置广角摄像头实例ID!");
  1570. }
  1571. else if (!g_FindEwsCamera)
  1572. {
  1573. if (FindNewEwsCamera())
  1574. {
  1575. g_FindEwsCamera = TRUE;
  1576. LogEvent(Severity_Middle, EVENT_USB_CAMERA_INPUT, g_strEwsCameraID.GetData());
  1577. Dbg("广角摄像头:[%s]插入", g_strEwsCameraID);
  1578. memcpy((void*)g_FixedImageDeviceName[g_FixedImageDeviceNum].cDeviceName, g_strEwsCameraID.GetData(), g_strEwsCameraID.GetLength());
  1579. g_FixedImageDeviceNum++;
  1580. break;
  1581. }
  1582. }
  1583. }
  1584. //TODO:当前摄像头个数小于2时(只是两个内置摄像头),处于异常,要求重启框架
  1585. uiCameraNum = GetImageDeviceNum();
  1586. if (uiCameraNum > g_FixedImageDeviceNum + 1)
  1587. {
  1588. //add by zl 20171109 获取高拍仪个数,允许同时插入普通外接摄像头和高拍仪,且支持同时插入多个高拍仪
  1589. int nPortableScannerNum = GetPortableScannerNum();
  1590. int nNewScannerNum = uiCameraNum - g_FixedImageDeviceNum - nPortableScannerNum;
  1591. if (nNewScannerNum > 1)
  1592. {
  1593. //如果当前找到的摄像头个数>实体启动时找到的摄像头个数+高拍仪个数+1
  1594. //说明系统识别到了非USB接口的摄像头,不支持处理这种异常情况
  1595. //打印当前找到的摄像头名称和个数
  1596. Dbg("系统识别[%d]个外接摄像头,系统只支持1个,请全部拔出,并重新插入唯一设备!",uiCameraNum-g_FixedImageDeviceNum-nPortableScannerNum);
  1597. LogEvent(Severity_Middle, EVENT_USB_CAMERA_NOT_ONLY, "Not only");
  1598. //MessageBox(0,"系统识别多个外接摄像头,系统只支持1个,请全部拔出,并重新插入唯一设备!",NULL,0);
  1599. break;
  1600. }
  1601. else if (0 == nNewScannerNum)
  1602. {
  1603. //如果只插入高拍仪,不处理直接返回
  1604. break;
  1605. }
  1606. }
  1607. else if (uiCameraNum < g_FixedImageDeviceNum)
  1608. {
  1609. //实体启动时插入了非USB外接摄像头,在此刻插入USB摄像头之前已经拔出
  1610. //此情况此处不做特殊处理,在拔出设备时,会刷新内置摄像头列表
  1611. Dbg("系统识别[%d]个摄像头,少于内置摄像头个数[%d]!",uiDeviceNum, g_FixedAudioDeviceNum);
  1612. }
  1613. if (!g_FindUsbCamera)
  1614. {
  1615. if (FindNewImageDevice(&pName))
  1616. {
  1617. g_FindUsbCamera = TRUE;
  1618. LogEvent(Severity_Middle, EVENT_USB_CAMERA_INPUT, pName);
  1619. Dbg("摄像头:[%s]插入",pName);
  1620. //找到摄像头后,不再查找音频设备
  1621. break;
  1622. }
  1623. }
  1624. }
  1625. break;
  1626. case DBT_DEVICEREMOVECOMPLETE:
  1627. {
  1628. Sleep(100);
  1629. g_USBStatus = USB_PULLOUT;
  1630. LogEvent(Severity_Middle, EVENT_USB_OUTPUT, "USB output");
  1631. Dbg("检测到USB设备拔出.");
  1632. //广角摄像头当做内置摄像头处理
  1633. if (g_FindEwsCamera)
  1634. {
  1635. if (!FindMyImageDevice(g_strEwsCameraID.GetData()))
  1636. {
  1637. g_FindEwsCamera = FALSE;
  1638. LogEvent(Severity_Middle, EVENT_USB_CAMERA_OUTPUT, g_strEwsCameraID.GetData());
  1639. Dbg("广角摄像头:[%s]拔出\n",g_strEwsCameraID);
  1640. //从内置摄像头列表中去掉广角摄像头
  1641. DEVICE_INFO newFixedImageDeviceName[MAX_MEDIA_DEVICE_NUM] = {0};
  1642. int newCount = 0;
  1643. for (unsigned int i=0;i<g_FixedImageDeviceNum;i++)
  1644. {
  1645. if (-1 == FindSubStr(g_FixedImageDeviceName[i].cDeviceName,g_strEwsCameraID.GetData()))
  1646. {
  1647. memcpy((void*)newFixedImageDeviceName[newCount].cDeviceName, g_FixedImageDeviceName[i].cDeviceName, MAX_PATH);
  1648. newCount++;
  1649. }
  1650. memset(g_FixedImageDeviceName[i].cDeviceName, 0, MAX_PATH);
  1651. }
  1652. Dbg("拔出广角摄像头,刷新后的内置摄像头列表:");
  1653. for (unsigned int i=0;i<newCount;i++)
  1654. {
  1655. memcpy((void*)g_FixedImageDeviceName[i].cDeviceName, newFixedImageDeviceName[i].cDeviceName, MAX_PATH);
  1656. Dbg("内置摄像头:[%s]",g_FixedImageDeviceName[i].cDeviceName);
  1657. }
  1658. g_FixedImageDeviceNum--;
  1659. break;
  1660. }
  1661. }
  1662. uiCameraNum = GetImageDeviceNum();
  1663. if (uiCameraNum < g_FixedImageDeviceNum)
  1664. {
  1665. //如果当前的音频设备个数小于实体启动时发现的摄像头个数
  1666. //则认为实体启动时已经插入外接的摄像头,此时已经拔出,需刷新内置摄像头列表
  1667. Dbg("当前识别到的摄像头个数[%d]小于实体启动时识别到的摄像头个数[%d]",uiCameraNum, g_FixedImageDeviceNum);
  1668. Dbg("说明实体启动时已经插入了外接的摄像头,此时拔出该外接摄像头。准备刷新内置摄像头列表");
  1669. //找到被拔出的这个摄像头
  1670. char* poutName = new char[MAX_PATH];
  1671. memset(poutName, 0, MAX_PATH);
  1672. if(FindOutputImageDevice(poutName))
  1673. {
  1674. LogEvent(Severity_Middle, EVENT_USB_CAMERA_OUTPUT, poutName);
  1675. Dbg("摄像头:[%s]拔出\n",poutName);
  1676. }
  1677. if (NULL != poutName)
  1678. {
  1679. delete poutName;
  1680. poutName = NULL;
  1681. }
  1682. //清空
  1683. g_FixedImageDeviceNum = 0;
  1684. for (i=0; i<MAX_MEDIA_DEVICE_NUM; i++)
  1685. {
  1686. memset((void*)g_FixedImageDeviceName[i].cDeviceName, 0, MAX_PATH);
  1687. }
  1688. //刷新
  1689. if (!FindFixedImageDevice())
  1690. {
  1691. break;
  1692. }
  1693. else
  1694. {
  1695. Dbg("刷新后的内置摄像头列表:");
  1696. for (unsigned int i=0;i<g_FixedImageDeviceNum;i++)
  1697. {
  1698. Dbg("内置摄像头:[%s]",g_FixedImageDeviceName[i].cDeviceName);
  1699. }
  1700. }
  1701. }
  1702. else if (uiCameraNum > g_FixedImageDeviceNum)
  1703. {
  1704. //add by zl 20171109 获取高拍仪个数,允许同时插入普通外接摄像头和高拍仪,且支持同时插入多个高拍仪
  1705. int nPortableScannerNum = GetPortableScannerNum();
  1706. int nScannerNum = uiCameraNum - g_FixedImageDeviceNum - nPortableScannerNum;
  1707. if (nScannerNum > 1)
  1708. {
  1709. //如果当前的摄像头个数大于实体启动时发现的摄像头个数
  1710. //则认为实体启动后,系统自动识别到新的非USB摄像头
  1711. //此情况此处不做特殊处理,在下一次插入USB设备时,会弹出错误提示
  1712. //int nInNum = uiCameraNum - g_FixedImageDeviceNum;
  1713. Dbg("系统仍识别[%d]个外接摄像头!",nScannerNum);
  1714. break;//add by zl 20160705,电子钥匙插拔时,会进入此处
  1715. }
  1716. }
  1717. else
  1718. {
  1719. Dbg("g_FixedImageDeviceNum=uiCameraNum");
  1720. }
  1721. if (g_FindUsbCamera)
  1722. {
  1723. if (!FindMyImageDevice(g_UsbCameraName))
  1724. {
  1725. g_FindUsbCamera = FALSE;
  1726. LogEvent(Severity_Middle, EVENT_USB_CAMERA_OUTPUT, g_UsbCameraName);
  1727. Dbg("摄像头:[%s]拔出\n",g_UsbCameraName);
  1728. }
  1729. }
  1730. }
  1731. break;
  1732. }
  1733. }
  1734. break;
  1735. default:
  1736. return DefWindowProc(hWnd, message, wParam, lParam);
  1737. }
  1738. if (NULL != pName)
  1739. {
  1740. delete pName;
  1741. }
  1742. return 0;
  1743. }
  1744. DWORD WINAPI DetectThreadFunc(LPVOID ready)
  1745. {
  1746. char szClassName[256] = {'c','g','b',0};
  1747. SYSTEMTIME st;
  1748. GetLocalTime(&st);
  1749. srand(GetTickCount());
  1750. sprintf_s((char *)&(szClassName[3]),250,"%d%d%d%d%d%d",rand(),rand(),st.wMinute,st.wSecond,st.wMilliseconds,GetTickCount());
  1751. MSG msg;
  1752. WNDCLASSEX wcex = {0};
  1753. wcex.cbSize = sizeof(WNDCLASSEX);
  1754. wcex.style = CS_HREDRAW | CS_VREDRAW;
  1755. wcex.lpfnWndProc = (WNDPROC)WndProc;
  1756. wcex.lpszClassName = szClassName;/*"GDB_DETECT_WINDOW";*/
  1757. ATOM a = RegisterClassEx(&wcex);
  1758. if (a == 0)
  1759. {
  1760. //WRITE_INFORMATION_LOG("Error::DetectThreadFun!!!RegisterClassEx!!!!!!!!!!!!!!Start--%d--",GetLastError());
  1761. }
  1762. else
  1763. {
  1764. //WRITE_INFORMATION_LOG("DetectThreadFunc!!!RegisterClassEx!!!!!!!!!!!!!Start--%d--",GetLastError());
  1765. }
  1766. hDetectWnd = CreateWindow(szClassName, "", WS_OVERLAPPEDWINDOW,
  1767. CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, NULL, NULL);
  1768. HideCaret(hDetectWnd);
  1769. SetEvent((HANDLE)ready);
  1770. if(hDetectWnd)
  1771. {
  1772. while (GetMessage(&msg, NULL, 0, 0))
  1773. {
  1774. if (!TranslateAccelerator(msg.hwnd, NULL, &msg))
  1775. {
  1776. TranslateMessage(&msg);
  1777. DispatchMessage(&msg);
  1778. }
  1779. }
  1780. }
  1781. return 0;
  1782. }
  1783. bool RegisterDetector()
  1784. {
  1785. hDetectMutex = CreateMutex(0, TRUE, "CMBDETECTOR");
  1786. if (hDetectMutex == NULL)
  1787. {
  1788. //WRITE_INFORMATION_LOG("CreateMutex fail %08x.\n", GetLastError());
  1789. return false;
  1790. }
  1791. if (GetLastError() == ERROR_ALREADY_EXISTS)
  1792. {
  1793. CloseHandle(hDetectMutex);
  1794. hDetectMutex = NULL;
  1795. ////WRITE_INFORMATION_LOG("Detector exists.\n");
  1796. return false;
  1797. }
  1798. DWORD ThreadId;
  1799. HANDLE ready = CreateEvent(0,0,0,0);
  1800. hDetectThread = CreateThread(NULL,0,DetectThreadFunc,ready,0,&ThreadId);
  1801. WaitForSingleObject(ready,-1);
  1802. if (hDetectWnd == NULL)
  1803. {
  1804. //WRITE_INFORMATION_LOG("Create window fail %08x.\n", GetLastError());
  1805. CloseHandle(hDetectMutex);
  1806. hDetectMutex = NULL;
  1807. return false;
  1808. }
  1809. DEV_BROADCAST_DEVICEINTERFACE devBroadcastDevIF = {0};
  1810. devBroadcastDevIF.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
  1811. devBroadcastDevIF.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
  1812. devBroadcastDevIF.dbcc_reserved = 0;
  1813. devBroadcastDevIF.dbcc_classguid = *StrToGUID(L"{A5DCBF10-6530-11D2-901F-00C04FB951ED}");
  1814. //devBroadcastDevIF.dbcc_classguid = *StrToGUID(L"{4D36E96C-E325-11CE-BFC1-08002BE10318}");
  1815. DWORD flag = DEVICE_NOTIFY_WINDOW_HANDLE;
  1816. //if (!g_IsWin2K)
  1817. //{
  1818. // flag |= DEVICE_NOTIFY_ALL_INTERFACE_CLASSES;
  1819. //}
  1820. hDevNotify = RegisterDeviceNotification(
  1821. hDetectWnd,
  1822. &devBroadcastDevIF,
  1823. flag
  1824. );
  1825. if (hDevNotify == NULL)
  1826. {
  1827. SendMessage(hDetectWnd,WM_CLOSE,0,0);
  1828. if (hDetectThread)
  1829. {
  1830. WaitForSingleObject(hDetectThread,-1);
  1831. }
  1832. hDetectThread = NULL;
  1833. hDetectWnd = NULL;
  1834. //WRITE_INFORMATION_LOG("RegisterDeviceNotification fail %08x.\n", GetLastError());
  1835. CloseHandle(hDetectMutex);
  1836. hDetectMutex = NULL;
  1837. return false;
  1838. }
  1839. //WRITE_INFORMATION_LOG("RegisterDeviceNotification succeed.\n");
  1840. return true;
  1841. }
  1842. void InitializeDetector(PVOID pWnd)
  1843. {
  1844. RegisterDetector();
  1845. }
  1846. void UninitializeDetector(PVOID pWnd)
  1847. {
  1848. if (hDevNotify)
  1849. {
  1850. UnregisterDeviceNotification(hDevNotify);
  1851. hDevNotify = NULL;
  1852. }
  1853. if (hDetectWnd)
  1854. {
  1855. SendMessage(hDetectWnd,WM_CLOSE,0,0);
  1856. if (hDetectThread)
  1857. {
  1858. WaitForSingleObject(hDetectThread,-1);
  1859. }
  1860. hDetectThread = NULL;
  1861. hDetectWnd = NULL;
  1862. }
  1863. if (hDetectMutex)
  1864. {
  1865. CloseHandle(hDetectMutex);
  1866. hDetectMutex = NULL;
  1867. }
  1868. }
  1869. struct callback_entry : public IReleasable
  1870. {
  1871. virtual ~callback_entry() {}
  1872. CSimpleStringA EntityName;
  1873. union {
  1874. void *pRawData;
  1875. int state;
  1876. };
  1877. int op;
  1878. ErrorCodeEnum ErrorResult;
  1879. };
  1880. struct GuidUSBTask : public ITaskSp
  1881. {
  1882. ConsoleFSM* fsm;
  1883. GuidUSBTask(ConsoleFSM* f) : fsm(f) {}
  1884. void Process()
  1885. {
  1886. USBKEY_INFO info;
  1887. unsigned count = 0;
  1888. DWORD dwRet = DONGLE_SUCCESS;
  1889. BOOL bRet = FALSE;
  1890. while (1)
  1891. {
  1892. if (USB_INSERT == g_USBStatus)
  1893. {
  1894. memset(&g_USBKeyInfo, 0, sizeof(DONGLE_INFO));
  1895. dwRet = FindUsbKey(&g_USBKeyInfo, &count);
  1896. if (dwRet == DONGLE_SUCCESS)
  1897. {
  1898. if (!g_FindKey)
  1899. {
  1900. g_FindKey = TRUE;
  1901. fsm->PostEventFIFO(new FSMEvent(USER_EVT_JMP_UKEYINSERT));
  1902. }
  1903. }
  1904. //如果是大机,检测是否专业版Ukey插入
  1905. if (!g_FindUkey)
  1906. {
  1907. if (FindUkey())
  1908. {
  1909. g_FindUkey = TRUE;
  1910. LogEvent(Severity_Middle, EVENT_UKEY_INSERT, "Ukey insert");
  1911. TheUkeyInsert e ={};
  1912. SpSendBroadcast(fsm->GetFunction(), eMsg_TheUkeyInsert, eMsgSig_TheUkeyInsert, e);
  1913. Dbg("UKey[%s]插入",g_UkeyName);
  1914. }
  1915. }
  1916. g_USBStatus = 0;
  1917. }
  1918. else if (USB_PULLOUT == g_USBStatus)
  1919. {
  1920. memset(&info, 0, sizeof(DONGLE_INFO));
  1921. dwRet = FindUsbKey(&info, &count);
  1922. if (dwRet != DONGLE_SUCCESS)
  1923. {
  1924. if (g_FindKey)
  1925. {
  1926. g_FindKey = FALSE;
  1927. fsm->PostEventFIFO(new FSMEvent(USER_EVT_JMP_UKEYPULLOUT));
  1928. }
  1929. }
  1930. //检测是否专业版Ukey拔出
  1931. if (g_FindUkey)
  1932. {
  1933. if (!FindTheUkey(g_UkeyName))
  1934. {
  1935. g_FindUkey = FALSE;
  1936. LogEvent(Severity_Middle, EVENT_UKEY_OUTPUT, "Ukey output");
  1937. TheUkeyPullOut e ={};
  1938. SpSendBroadcast(fsm->GetFunction(), eMsg_TheUkeyPullOut, eMsgSig_TheUkeyPullOut, e);
  1939. Dbg("UKey[%s]拔出",g_UkeyName);
  1940. //如果是大机,拔出Ukey,关闭Gpio
  1941. if (0 != strcmp(fsm->m_strMachineType.GetData(), MACHINETYPE_PAD))
  1942. {
  1943. bRet = fsm->OpenGpio(FALSE);
  1944. }
  1945. }
  1946. }
  1947. g_USBStatus = 0;
  1948. }
  1949. Sleep(50);
  1950. }
  1951. }
  1952. };
  1953. struct GuidVerifyTask : public ITaskSp
  1954. {
  1955. ConsoleFSM* fsm;
  1956. GuidVerifyTask(ConsoleFSM* f) : fsm(f) {}
  1957. void Process()
  1958. {
  1959. while(1)
  1960. {
  1961. if (fsm->m_bVerifyPwd)
  1962. {
  1963. fsm->m_bVerifyPwd = FALSE;
  1964. //fsm->PostEventLIFO(new FSMEvent(USER_EVT_JMP_RELEASE));
  1965. }
  1966. Sleep(100);
  1967. }
  1968. }
  1969. };
  1970. ConsoleFSM::ConsoleFSM() : m_pConnection(NULL)
  1971. {
  1972. m_Handle = 0;
  1973. g_FindKey = FALSE;
  1974. g_FindUkey = FALSE;
  1975. g_bWorkingStatus = FALSE;
  1976. m_bVerifySuccess = FALSE;
  1977. m_nUkeyStatus = 0;
  1978. m_nCheckTerminalNo = 0;
  1979. m_nCheckMaintainTask = 0;
  1980. m_nVerifyBlackList = 1;
  1981. g_nSurveillanceRecordAllowed = 0;
  1982. g_strEwsCameraID = "";
  1983. m_strUIstate = "";
  1984. m_nSRState = 0;
  1985. m_strPwd.Clear();
  1986. m_errcode = 0;
  1987. m_bMenu = FALSE;
  1988. g_bErrStatus = FALSE;
  1989. m_bAns = FALSE;
  1990. m_bVerifyPwd = FALSE;
  1991. m_bInputpwdDlg = FALSE;
  1992. g_HasVerifyTask = FALSE;
  1993. memset(m_TaskStatus,'D',1);//初始状态设置为D
  1994. memset(m_USBKeyInitTime,0,20);
  1995. m_RecordNum = 0;
  1996. memset(&g_USBKeyInfo, 0, sizeof(DONGLE_INFO));
  1997. m_pCardIssuerService = NULL;
  1998. m_pDeviceControlService = NULL;
  1999. for (int i=0; i<128; i++)
  2000. {
  2001. m_TaskRecords[i] = new TaskRecord();
  2002. memset(m_TaskRecords[i]->startTime, 0, 20);
  2003. memset(m_TaskRecords[i]->endTime, 0, 20);
  2004. m_TaskRecords[i]->strWarningId = "";
  2005. }
  2006. InitializeCriticalSection(&cs);
  2007. }
  2008. ConsoleFSM::~ConsoleFSM()
  2009. {
  2010. for (int i=0; i<128; i++)
  2011. {
  2012. delete m_TaskRecords[i];
  2013. }
  2014. DeleteCriticalSection(&cs);
  2015. }
  2016. void ConsoleFSM::OnStateTrans( int iSrcState, int iDstState )
  2017. {
  2018. Dbg("trans from %s to %s", GetStateName(iSrcState), GetStateName(iDstState));
  2019. }
  2020. void ConsoleFSM::OnSysVarEvent( const char *pszKey, const char *pszValue,const char *pszOldValue,const char *pszEntityName )
  2021. {
  2022. }
  2023. ErrorCodeEnum ConsoleFSM::OnInit()
  2024. {
  2025. ErrorCodeEnum Error;
  2026. Error = LoadServerConfig();
  2027. if (Error != Error_Succeed)
  2028. {
  2029. LOG_TRACE("load MaintainWatcher.ini failed!");
  2030. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_READ_CONFIG,
  2031. GetOutPutStr("%s%08X%s%s", "LoadServerConfig", Error,"File","MaintainWatcher.ini").c_str());
  2032. }
  2033. Error = LoadRootConfig();
  2034. if (Error != Error_Succeed)
  2035. {
  2036. LOG_TRACE("load root.ini failed!");
  2037. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_READ_CONFIG,
  2038. GetOutPutStr("%s%08X%s%s", "LoadRootConfig", Error, "File", "root.ini").c_str());
  2039. return Error;
  2040. }
  2041. Error = LoadCenterConfig();
  2042. if (Error != Error_Succeed)
  2043. {
  2044. //没有读到CenterSetting.ini中的CheckTerMinalNo项,不返回错误,默认为不校验终端ID
  2045. LOG_TRACE("load CenterSetting.ini failed!");
  2046. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_READ_CONFIG,
  2047. GetOutPutStr("%s%08X%s%s", "LoadCenterConfig", Error, "File", "CenterSetting.ini").c_str());
  2048. //return Error;
  2049. }
  2050. Error = SetRunConfig();
  2051. if (Error_Succeed != Error)
  2052. {
  2053. LOG_TRACE("set runcfg MaintainWatcher.ini failed!");
  2054. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_READ_CONFIG,
  2055. GetOutPutStr("%s%08X%s%s", "SetRunConfig", Error, "File", "MaintainWatcher.ini").c_str());
  2056. }
  2057. if (Error == Error_Succeed)
  2058. {
  2059. AddStateHooker(this);
  2060. }
  2061. //获取终端号
  2062. CSystemStaticInfo si;
  2063. m_pEntity->GetFunction()->GetSystemStaticInfo(si);
  2064. m_strTerminalId = si.strTerminalID;
  2065. Dbg("Get TerminalId=%s from root.ini",m_strTerminalId.c_str());
  2066. //创建证书目录
  2067. DWORD dwAttr=GetFileAttributes(CERT_DIR);
  2068. if(dwAttr==0xFFFFFFFF) //目录不存在则创建
  2069. {
  2070. if(!CreateDirectory(CERT_DIR,NULL))
  2071. {
  2072. Dbg("Create cert dir failed!");
  2073. LogWarn(Severity_Low,Error_Unexpect,ERR_MAINTAINWATCHER_CREATE_DIR,
  2074. GetOutPutStr("%s%s%s%s", "CreateDirectory", "False", "Dir", CERT_DIR).c_str());
  2075. }
  2076. }
  2077. //创建文件目录
  2078. char path[MAX_PATH];
  2079. GetSystemDirectory(path,MAX_PATH);
  2080. string strRvcPath = path;
  2081. strRvcPath += ":\\RVC";
  2082. dwAttr=GetFileAttributes("C:\\RVC");
  2083. if(dwAttr==0xFFFFFFFF) //目录不存在则创建
  2084. {
  2085. if(!CreateDirectory("C:\\RVC",NULL))
  2086. {
  2087. Dbg("Create C:\\RVC failed!");
  2088. LogWarn(Severity_Low,Error_Unexpect,ERR_MAINTAINWATCHER_CREATE_DIR,
  2089. GetOutPutStr("%s%s%s%s", "CreateDirectory", "False", "Dir", "C:\\RVC").c_str());
  2090. }
  2091. }
  2092. string strCertPath = path;
  2093. strCertPath += ":\\RVC\\UKey";
  2094. dwAttr=GetFileAttributes("C:\\RVC\\Ukey");
  2095. if(dwAttr==0xFFFFFFFF) //目录不存在则创建
  2096. {
  2097. if(!CreateDirectory("C:\\RVC\\Ukey",NULL))
  2098. {
  2099. Dbg("Create C:\\RVC\\Ukey failed!");
  2100. LogWarn(Severity_Low,Error_Unexpect, ERR_MAINTAINWATCHER_CREATE_DIR,
  2101. GetOutPutStr("%s%s%s%s", "CreateDirectory", "False", "Dir", "C:\\RVC\\Ukey").c_str());
  2102. }
  2103. }
  2104. GuidUSBTask* task2 = new GuidUSBTask(this);
  2105. GetEntityBase()->GetFunction()->PostThreadPoolTask(task2);
  2106. if (1)
  2107. {
  2108. FindFixedImageDevice();
  2109. Dbg("找到[%d]个摄像头 at OnInit",g_FixedImageDeviceNum);
  2110. for (unsigned int i=0;i<g_FixedImageDeviceNum;i++)
  2111. {
  2112. Dbg("摄像头:[%s]",g_FixedImageDeviceName[i].cDeviceName);
  2113. }
  2114. if (FindNewEwsCamera())
  2115. {
  2116. g_FindEwsCamera = TRUE;
  2117. LogEvent(Severity_Middle, EVENT_USB_CAMERA_INPUT, g_strEwsCameraID.GetData());
  2118. Dbg("实体启动时,广角摄像头:[%s]插入", g_strEwsCameraID);
  2119. }
  2120. //启动USB监控
  2121. InitializeDetector(NULL);
  2122. }
  2123. //下载根证书和CA证书
  2124. Dbg("开始下载基本证书");
  2125. DownloadCerts();
  2126. Dbg("开始下载黑名单");
  2127. DownloadCertBlackList();
  2128. //......
  2129. return Error;
  2130. }
  2131. ErrorCodeEnum ConsoleFSM::OnExit()
  2132. {
  2133. return Error_Succeed;
  2134. }
  2135. void ConsoleFSM::DownloadCerts()
  2136. {
  2137. //下载根证书和CA证书
  2138. if (!m_pConnection)
  2139. {
  2140. m_pConnection = new CAConnection(m_pEntity, this);
  2141. if (!m_pConnection->ConnectFromCentralSetting())
  2142. {
  2143. Dbg("connect server fail,config or Servers Error!");
  2144. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_DOWNLOAD_CERT,
  2145. GetOutPutStr("%s%s", "ConnectFromCentralSetting", "False").c_str());
  2146. }
  2147. }
  2148. if (m_pConnection->IsConnectionOK())
  2149. {
  2150. EnterCriticalSection(&cs);
  2151. m_pConnection->SendVersionReq();
  2152. Dbg("send cert req success");
  2153. LeaveCriticalSection(&cs);
  2154. }
  2155. else
  2156. {
  2157. Dbg("connect to CA server failed, can not download CA cert!");
  2158. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_DOWNLOAD_CERT,
  2159. GetOutPutStr("%s%s", "IsConnectionOK", "False").c_str());
  2160. }
  2161. }
  2162. void ConsoleFSM::DownloadCertBlackList()
  2163. {
  2164. if (!m_pConnection)
  2165. {
  2166. m_pConnection = new CAConnection(m_pEntity, this);
  2167. if (!m_pConnection->ConnectFromCentralSetting())
  2168. {
  2169. Dbg("connect server fail,config or Servers Error!");
  2170. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_DOWBLOAD_CERT_BLACKLIST,
  2171. GetOutPutStr("%s%s", "ConnectFromCentralSetting", "False").c_str());
  2172. }
  2173. }
  2174. if (m_pConnection->IsConnectionOK())
  2175. {
  2176. EnterCriticalSection(&cs);
  2177. m_pConnection->SendCertBlackListReq();
  2178. Dbg("Ask cert blackList req success");
  2179. LeaveCriticalSection(&cs);
  2180. }
  2181. else
  2182. {
  2183. Dbg("connect to CA server failed, can not download cert blackList!");
  2184. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_DOWBLOAD_CERT_BLACKLIST,
  2185. GetOutPutStr("%s%s", "IsConnectionOK", "False").c_str());
  2186. }
  2187. }
  2188. BOOL ConsoleFSM::GetMenuStatus()
  2189. {
  2190. //判断终端是否在业务界面首页,不在首页不弹出客户经理登录框
  2191. ErrorCodeEnum Error = GetEntityBase()->GetFunction()->GetSysVar("UIState",m_strUIstate);
  2192. if (Error_NotExist == Error)
  2193. {
  2194. Dbg("UIState is not exist, errcode=%d!", Error);
  2195. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHERGET_UISTATE,
  2196. GetOutPutStr("%s%08X%s%s", "GetSysVar", Error,"UIState", m_strUIstate).c_str());
  2197. m_bMenu = TRUE;
  2198. return m_bMenu;
  2199. }
  2200. else if (Error_Succeed != Error)
  2201. {
  2202. Dbg("Get UIState failed, errcode=%d!", Error);
  2203. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHERGET_UISTATE,
  2204. GetOutPutStr("%s%08X%s%s", "GetSysVar", Error, "UIState", m_strUIstate).c_str());
  2205. m_bMenu = TRUE;
  2206. return m_bMenu;
  2207. }
  2208. else
  2209. {
  2210. //Dbg("Get UIState success, m_strUIstate=%s!", m_strUIstate);
  2211. if ((-1 != FindSubStr(m_strUIstate, "M"))
  2212. || (-1 != FindSubStr(m_strUIstate, "A"))
  2213. || (-1 != FindSubStr(m_strUIstate, "T"))
  2214. || (-1 != FindSubStr(m_strUIstate, "X")))
  2215. {
  2216. //界面状态UIState。定义当前交互层所处的交互模式。有8种状态:目录浏览M、随机广告A、目标广告T、业务填单F、交易确认C、业务提交S、请求回报R、暂停响应Z。初始状态为M。
  2217. //M、A、T三种状态都认为在首页
  2218. Dbg("首页状态!");
  2219. m_bMenu = TRUE;
  2220. }
  2221. else
  2222. {
  2223. //不在首页
  2224. Dbg("非首页状态!");
  2225. m_bMenu = FALSE;
  2226. }
  2227. return m_bMenu;
  2228. }
  2229. }
  2230. DWORD ConsoleFSM::VerifyUser(const char* pcPIN)
  2231. {
  2232. if (NULL == pcPIN)
  2233. {
  2234. return INPUT_NULL_ERR;
  2235. }
  2236. USBKEY_INFO info;
  2237. unsigned count = 0;
  2238. DWORD dwRet = 0;
  2239. DONGLE_HANDLE hHandle;
  2240. CSimpleStringA errNum;
  2241. //找锁
  2242. dwRet = FindUsbKey(&info, &count);
  2243. if (DONGLE_SUCCESS != dwRet)
  2244. {
  2245. errNum.Format("0x%08X", dwRet);
  2246. Dbg("没有检测到电子钥匙, errcode=%s", errNum);
  2247. LogWarn(Severity_Middle,Error_Unexpect,ERR_MAINTAINWATCHER_EKEY_VERIFY,
  2248. GetOutPutStr("%s%08X", "FindUsbKey", dwRet).c_str());
  2249. //MessageBoxA(NULL,(LPCSTR)("没有检测到电子钥匙,请插入电子钥匙!"),NULL,0);
  2250. return dwRet;
  2251. }
  2252. Dbg("检测到[%d]把电子钥匙!",count);
  2253. //打开锁
  2254. dwRet = OpenUsbKey(&hHandle,0);
  2255. if (DONGLE_SUCCESS != dwRet)
  2256. {
  2257. errNum.Format("0x%08X", dwRet);
  2258. Dbg("无法打开电子钥匙, errcode=%s", errNum);
  2259. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_EKEY_VERIFY,
  2260. GetOutPutStr("%s%08X", "OpenUsbKey", dwRet).c_str());
  2261. return dwRet;
  2262. }
  2263. m_Handle = hHandle;
  2264. Dbg("打开电子钥匙成功!");
  2265. //验证开发商PIN码
  2266. char PIN[9] = {0};
  2267. memcpy(PIN, pcPIN, 8);
  2268. char* upPIN = _tcsupr(PIN);
  2269. dwRet = VerifyAdminPIN(hHandle, upPIN);
  2270. if (DONGLE_SUCCESS != dwRet)
  2271. {
  2272. errNum.Format("0x%08X", dwRet);
  2273. Dbg("电子钥匙密码错误, errcode=%s", errNum);
  2274. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_EKEY_VERIFY,
  2275. GetOutPutStr("%s%08X", "VerifyAdminPIN", dwRet).c_str());
  2276. return dwRet;
  2277. }
  2278. else
  2279. {
  2280. Dbg("验证电子钥匙密码成功!");
  2281. }
  2282. //读取根证书
  2283. int rootCertLen = 0;
  2284. dwRet = GetRootCaCertificateLength(hHandle, &rootCertLen);
  2285. if (DONGLE_SUCCESS == dwRet)
  2286. {
  2287. Dbg("获取根证书长度成功!");
  2288. BYTE* rootCertContent = new BYTE[rootCertLen];
  2289. if (NULL == rootCertContent)
  2290. {
  2291. return NEW_MEM_ERR;
  2292. }
  2293. memset(rootCertContent,0,rootCertLen);
  2294. dwRet = ReadRootCaCertificate(hHandle, rootCertContent, rootCertLen);
  2295. if (DONGLE_SUCCESS == dwRet)
  2296. {
  2297. Dbg("读根根证书成功!");
  2298. FILE* fp = fopen(ROOT_CERT_PATH, "w");
  2299. fwrite(rootCertContent, 1, rootCertLen, fp);
  2300. fclose(fp);
  2301. delete rootCertContent;
  2302. }
  2303. else
  2304. {
  2305. errNum.Format("0x%08X", dwRet);
  2306. Dbg("读根证书失败, errcode=%s", errNum);
  2307. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_ROOT_CERT,
  2308. GetOutPutStr("%s%08X", "ReadRootCaCertificate", dwRet).c_str());
  2309. delete rootCertContent;
  2310. return dwRet;
  2311. }
  2312. }
  2313. else
  2314. {
  2315. errNum.Format("0x%08X", dwRet);
  2316. Dbg("获取根证书长度失败, errcode=%s", errNum);
  2317. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_ROOT_CERT,
  2318. GetOutPutStr("%s%08X", "GetRootCaCertificateLength", dwRet).c_str());
  2319. return dwRet;
  2320. }
  2321. //读取分行CA证书
  2322. int subCaCertLen = 0;
  2323. dwRet = GetSubCaCertificateLength(hHandle, &subCaCertLen);
  2324. if (DONGLE_SUCCESS == dwRet)
  2325. {
  2326. Dbg("获取分行CA证书长度成功!");
  2327. BYTE* subCaCertContent = new BYTE[subCaCertLen];
  2328. if (NULL == subCaCertContent)
  2329. {
  2330. return NEW_MEM_ERR;
  2331. }
  2332. memset(subCaCertContent,0,subCaCertLen);
  2333. dwRet = ReadSubCaCertificate(hHandle, subCaCertContent, subCaCertLen);
  2334. if (DONGLE_SUCCESS == dwRet)
  2335. {
  2336. Dbg("读根分行CA证书成功!");
  2337. FILE* fp = fopen(CA_CERT_PATH, "w");
  2338. fwrite(subCaCertContent, 1, subCaCertLen, fp);
  2339. fclose(fp);
  2340. delete subCaCertContent;
  2341. }
  2342. else
  2343. {
  2344. errNum.Format("0x%08X", dwRet);
  2345. Dbg("读分行CA证书失败, errcode=%s", errNum);
  2346. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_BRANCH_CERT,
  2347. GetOutPutStr("%s%08X", "ReadSubCaCertificate", dwRet).c_str());
  2348. delete subCaCertContent;
  2349. return dwRet;
  2350. }
  2351. }
  2352. else
  2353. {
  2354. errNum.Format("0x%08X", dwRet);
  2355. Dbg("获取分行CA证书长度失败, errcode=%s", errNum);
  2356. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_BRANCH_CERT,
  2357. GetOutPutStr("%s%08X", "GetSubCaCertificateLength", dwRet).c_str());
  2358. return dwRet;
  2359. }
  2360. //读取Ukey中的用户证书
  2361. int userCertLen = 0;
  2362. dwRet = GetUserCertificateLength(hHandle, &userCertLen);
  2363. if (DONGLE_SUCCESS == dwRet)
  2364. {
  2365. Dbg("获取用户证书长度成功!");
  2366. BYTE* userCertContent = new BYTE[userCertLen];
  2367. if (NULL == userCertContent)
  2368. {
  2369. return NEW_MEM_ERR;
  2370. }
  2371. memset(userCertContent,0,userCertLen);
  2372. dwRet = ReadUserCertificate(hHandle, userCertContent, userCertLen);
  2373. if (DONGLE_SUCCESS == dwRet)
  2374. {
  2375. Dbg("读取用户证书成功!");
  2376. FILE* fp = fopen(USER_CERT_PATH, "w");
  2377. fwrite(userCertContent, 1, userCertLen, fp);
  2378. fclose(fp);
  2379. delete userCertContent;
  2380. }
  2381. else
  2382. {
  2383. errNum.Format("0x%08X", dwRet);
  2384. Dbg("读用户证书失败, errcode=%s", errNum);
  2385. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_USER_CERT,
  2386. GetOutPutStr("%s%08X", "ReadUserCertificate", dwRet).c_str());
  2387. delete userCertContent;
  2388. return dwRet;
  2389. }
  2390. }
  2391. else
  2392. {
  2393. errNum.Format("0x%08X", dwRet);
  2394. Dbg("获取用户证书长度失败, errcode=%s", errNum);
  2395. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_USER_CERT,
  2396. GetOutPutStr("%s%08X", "GetUserCertificateLength", dwRet).c_str());
  2397. return dwRet;
  2398. }
  2399. //获取证书认证类型
  2400. char cVerifyType[32] = {0};
  2401. bool bRet = GetVerifyTypeFromCert(USER_CERT_PATH, FORMAT_PEM, cVerifyType);
  2402. if (!bRet)
  2403. {
  2404. Dbg("获取证书类型失败!");
  2405. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_USER_CERT,
  2406. GetOutPutStr("%s%s", "GetVerifyTypeFromCert", "False").c_str());
  2407. return GET_CERT_CERTTYPE_ERR;
  2408. }
  2409. //判断认证类型
  2410. Dbg("电子钥匙类型:%s(Framework:框架认证,System:系统认证, admin:超级认证)", cVerifyType);
  2411. //删除此处判断,分行维护电子钥匙也能进入框架维护,add by zl80170228
  2412. /*if ((0 != strncmp(cVerifyType, "Framework", 9))
  2413. && (0 != strncmp(cVerifyType, "admin", 5)))
  2414. {
  2415. Dbg("电子钥匙认证类型不具备框架认证权限!");
  2416. return GET_NO_FRAMEWORK_PERMIT_ERR;
  2417. }*/
  2418. //验证用户证书
  2419. dwRet = VerifyUserCert(ROOT_CERT_PATH, CA_CERT_PATH, USER_CERT_PATH, NULL, FORMAT_PEM);
  2420. if (DONGLE_SUCCESS != dwRet)
  2421. {
  2422. errNum.Format("0x%08X", dwRet);
  2423. Dbg("验证用户证书失败, errcode=%s", errNum);
  2424. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_USER_CERT_VERIFY,
  2425. GetOutPutStr("%s%08X", "VerifyUserCert", dwRet).c_str());
  2426. return dwRet;
  2427. }
  2428. Dbg("验证用户证书成功!");
  2429. dwRet = VerifyTerminalInfo();
  2430. if (DONGLE_SUCCESS != dwRet)
  2431. {
  2432. errNum.Format("0x%08X", dwRet);
  2433. Dbg("验证任务信息失败, errcode=%s", errNum);
  2434. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_TERMINAL_VERIFY,
  2435. GetOutPutStr("%s%08X", "VerifyTerminalInfo", dwRet).c_str());
  2436. return dwRet;
  2437. }
  2438. //校验证书时间,给出续期提醒,过期前15天开始提醒
  2439. dwRet = CheckCertVaildity();
  2440. if (SUCCESS != dwRet && OVERTTIMER_WAR != dwRet)
  2441. {
  2442. Dbg("CheckCertVaildity failed!");
  2443. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CERT_VERIFY,
  2444. GetOutPutStr("%s%08X", "CheckCertVaildity", dwRet).c_str());
  2445. }
  2446. return dwRet;
  2447. }
  2448. DWORD ConsoleFSM::VerifyOther()
  2449. {
  2450. Dbg("enter VerifyOther");
  2451. m_bVerifyPwd = TRUE;
  2452. ((0 == m_errcode) || (OVERTTIMER_WAR == m_errcode))? PostEventLIFO(new FSMEvent(USER_EVT_JMP_PWDSUCCESS)):PostEventLIFO(new FSMEvent(USER_EVT_JMP_PWDERR));
  2453. return 0;
  2454. }
  2455. BOOL CALLBACK spEnumWindowsProc(HWND hwnd,LPARAM lParam)
  2456. {
  2457. char title[300] = {0};
  2458. SendMessage(hwnd,WM_GETTEXT,256,(LPARAM)title);
  2459. if (_stricmp(title, "密码") == 0)
  2460. {
  2461. SendMessage(hwnd, WM_CLOSE, 0,0);
  2462. return FALSE;
  2463. }
  2464. return TRUE;
  2465. }
  2466. time_t Str2Timet(const string& ATime, const string& AFormat="%d-%d-%d")
  2467. {
  2468. struct tm tm_Temp;
  2469. time_t time_Ret;
  2470. try
  2471. {
  2472. int i = sscanf(ATime.c_str(), AFormat.c_str(),// "%d/%d/%d %d:%d:%d" ,
  2473. &(tm_Temp.tm_year),
  2474. &(tm_Temp.tm_mon),
  2475. &(tm_Temp.tm_mday),
  2476. &(tm_Temp.tm_hour),
  2477. &(tm_Temp.tm_min),
  2478. &(tm_Temp.tm_sec),
  2479. &(tm_Temp.tm_wday),
  2480. &(tm_Temp.tm_yday));
  2481. tm_Temp.tm_year -= 1900;
  2482. tm_Temp.tm_mon --;
  2483. tm_Temp.tm_hour=0;
  2484. tm_Temp.tm_min=0;
  2485. tm_Temp.tm_sec=0;
  2486. tm_Temp.tm_isdst = 0;
  2487. time_Ret = mktime(&tm_Temp);
  2488. return time_Ret;
  2489. } catch(...) {
  2490. return 0;
  2491. }
  2492. }
  2493. string Timet2Str(const time_t &ATime_t, const string& AFormat="%d-%02d-%02d")
  2494. {
  2495. char chRet[15];
  2496. string sRet;
  2497. struct tm *p;
  2498. try
  2499. {
  2500. p = localtime(&ATime_t);
  2501. p->tm_year = p->tm_year + 1900;
  2502. p->tm_mon = p->tm_mon + 1;
  2503. sprintf(chRet, AFormat.c_str(),
  2504. p->tm_year, p->tm_mon, p->tm_mday);
  2505. sRet = chRet;
  2506. return sRet;
  2507. } catch(...) {
  2508. return "";
  2509. }
  2510. }
  2511. time_t NowTime()
  2512. {
  2513. time_t t_Now = time(0);
  2514. struct tm* tm_Now = localtime(&t_Now);
  2515. tm_Now->tm_hour =0;
  2516. tm_Now->tm_min = 0;
  2517. tm_Now->tm_sec = 0;
  2518. return mktime(tm_Now);
  2519. }
  2520. BOOL ConsoleFSM::GetTaskStatus(const char* pTaskID)
  2521. {
  2522. //获取任务状态
  2523. if (!m_pConnection)
  2524. {
  2525. m_pConnection = new CAConnection(m_pEntity, this);
  2526. if (NULL == m_pConnection)
  2527. {
  2528. Dbg("new CAConnection failed!");
  2529. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2530. GetOutPutStr("%s%s", "CAConnection", "NULL").c_str());
  2531. return FALSE;
  2532. }
  2533. if (!m_pConnection->ConnectFromCentralSetting())
  2534. {
  2535. Dbg("connect server fail,config or Servers Error!");
  2536. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2537. GetOutPutStr("%s%s", "ConnectFromCentralSetting", "False").c_str());
  2538. return FALSE;
  2539. }
  2540. }
  2541. if (m_pConnection->IsConnectionOK())
  2542. {
  2543. EnterCriticalSection(&cs);
  2544. m_pConnection->SendTaskReq(pTaskID);
  2545. LeaveCriticalSection(&cs);
  2546. }
  2547. else
  2548. {
  2549. Dbg("connect to CA server failed, can not get <TaskID=%s> status!", pTaskID);
  2550. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2551. GetOutPutStr("%s%s%s%s", "IsConnectionOK", "False","GetTaskStatus",pTaskID).c_str());
  2552. return FALSE;
  2553. }
  2554. return TRUE;
  2555. }
  2556. //校验电子钥匙生命周期,超过3年提示更新硬件
  2557. BOOL ConsoleFSM::CheckUSBKeyLife(const char* pUSBKeyID)
  2558. {
  2559. //获取任务状态
  2560. if (!m_pConnection)
  2561. {
  2562. m_pConnection = new CAConnection(m_pEntity, this);
  2563. if (NULL == m_pConnection)
  2564. {
  2565. Dbg("new CAConnection failed!");
  2566. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2567. GetOutPutStr("%s%s%s%s", "CAConnection", "NULL","CheckUSBKeyLife", pUSBKeyID).c_str());
  2568. return FALSE;
  2569. }
  2570. if (!m_pConnection->ConnectFromCentralSetting())
  2571. {
  2572. Dbg("connect server fail,config or Servers Error!");
  2573. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2574. GetOutPutStr("%s%s%s%s", "ConnectFromCentralSetting", "False", "CheckUSBKeyLife", pUSBKeyID).c_str());
  2575. return FALSE;
  2576. }
  2577. }
  2578. if (m_pConnection->IsConnectionOK())
  2579. {
  2580. EnterCriticalSection(&cs);
  2581. m_pConnection->SendUSBKeyInfoReq(pUSBKeyID);
  2582. LeaveCriticalSection(&cs);
  2583. }
  2584. else
  2585. {
  2586. Dbg("connect to CA server failed, can not get <USBKeyID=%s> info!", pUSBKeyID);
  2587. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CA_CONNECT,
  2588. GetOutPutStr("%s%s%s%s", "IsConnectionOK", "False", "CheckUSBKeyLife", pUSBKeyID).c_str());
  2589. return FALSE;
  2590. }
  2591. time_t tNow = NowTime();
  2592. Dbg("电子钥匙第一次初始化时间[%s]", m_USBKeyInitTime);
  2593. time_t tUSBKeyInitTime = Str2Timet(m_USBKeyInitTime, "%4d%2d%2d%2d%2d%2d");
  2594. if (tUSBKeyInitTime >= tNow)
  2595. {
  2596. string strNow = Timet2Str(tNow);
  2597. Dbg("电子钥匙第一次初始化时间[%s]大于机器当前时间[%s], 请确认当前VTM的系统时间是否正确!", m_USBKeyInitTime, strNow.c_str());
  2598. CString msg;
  2599. msg.Format(_T("电子钥匙第一次初始化时间[%s]大于机器当前时间[%s], 请确认当前VTM的系统时间是否正确!"), m_USBKeyInitTime, strNow.c_str());
  2600. ::MessageBox(NULL,msg, "提示",0);
  2601. return FALSE;
  2602. }
  2603. else
  2604. {
  2605. UINT nRestSecond = tNow - tUSBKeyInitTime;
  2606. UINT nRestDay = nRestSecond/(60*60*24);
  2607. if (nRestDay >= 365*3)
  2608. {
  2609. CString msg;
  2610. msg.Format(_T("电子钥匙硬件已超过使用期限(最长3年),剩余有效天数[%d],请及时更换新电子钥匙!"), nRestDay);
  2611. ::MessageBox(NULL,msg, "提示",0);
  2612. Dbg("电子钥匙硬件已超过使用期限(最长3年),剩余有效天数[%d],请及时更换新电子钥匙!", nRestDay);
  2613. return TRUE;
  2614. }
  2615. }
  2616. return TRUE;
  2617. }
  2618. //在线校验任务状态
  2619. BOOL ConsoleFSM::VerifyTaskStatus(const char* pTaskID)
  2620. {
  2621. //查询任务状态
  2622. BOOL bRet = GetTaskStatus(pTaskID);
  2623. //此处不判断返回值,如果获取不成功,则认为终端处于离线状态,返回任务状态校验成功
  2624. /*if (!bRet)
  2625. {
  2626. return FALSE;
  2627. }*/
  2628. Dbg("<TaskID=%s> status is <%c>!", pTaskID, m_TaskStatus[0]);
  2629. if ('C' == m_TaskStatus[0])
  2630. {
  2631. Dbg("<TaskID=%s> is been closed!", pTaskID);
  2632. m_TaskStatus[0] = 'D';
  2633. return FALSE;
  2634. }
  2635. return TRUE;
  2636. }
  2637. DWORD ConsoleFSM::VerifyTerminalInfo()
  2638. {
  2639. //获取用户信息,用于写入日志
  2640. BOOL bRet = GetUserInfo();
  2641. if (!bRet)
  2642. {
  2643. Dbg("GetUserInfo failed");
  2644. return GET_USERINFO_ERR;
  2645. }
  2646. string strUserType = "";
  2647. int nUserType = atoi(m_strUserType.c_str());
  2648. if (1 == m_nVerifyBlackList)
  2649. {
  2650. char ukeyID[100];
  2651. memset(ukeyID, 0, sizeof(ukeyID));
  2652. m_strKeyHID.copy(ukeyID, m_strKeyHID.length(), 0);
  2653. if (IsCertInBlack(ukeyID))
  2654. {
  2655. Dbg("CertUkeyID is in black List");
  2656. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_EKEY_VERIFY,
  2657. GetOutPutStr("%s%s", "IsCertInBlack", "True").c_str());
  2658. //MessageBox(0,"当前电子钥匙已被禁用,请更换其他电子钥匙!",NULL,0);
  2659. return IN_BLACKLIST_ERR;
  2660. }
  2661. else
  2662. {
  2663. Dbg("CertUkeyID is not in black List");
  2664. }
  2665. }
  2666. //只有维护和物料类型的电子钥匙需要验证终端任务信息
  2667. if (((USER_TYPE_MATERIAL_MANAGER == nUserType)&&(1 == m_nCheckTerminalNo))//物料管理类型电子钥匙是否校验任务,通过集中配置CheckTerminalNo开关控制
  2668. ||((USER_TYPE_MAINTAIN == nUserType)&&(1 == m_nCheckMaintainTask)))//维护类型电子钥匙是否校验任务,通过集中配置CheckMaintainTask开关控制
  2669. {
  2670. bRet = FALSE;
  2671. //兼容支持清机任务的卡机号认证
  2672. if((USER_TYPE_MATERIAL_MANAGER == nUserType)
  2673. &&(0 == strcmp(m_strMachineType.GetData(), MACHINETYPE_PAD)))
  2674. {
  2675. //获取卡箱号
  2676. bRet = GetIssuerNum();
  2677. if (!bRet)
  2678. {
  2679. Dbg("GetIssuerNum failed");
  2680. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_GET_CARDISSUER_SN,
  2681. GetOutPutStr("%s%s", "GetIssuerNum", "False").c_str());
  2682. return GET_ISSUERNUM_ERR;
  2683. }
  2684. else
  2685. {
  2686. Dbg("GetIssuerNum succeed");
  2687. }
  2688. //如果是Pad做物料维护,校验卡箱号
  2689. DWORD dwRet = VerifyCardIssuerSN();
  2690. if (SUCCESS != dwRet)
  2691. {
  2692. Dbg("VerifyCardIssuerSN failed");
  2693. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_CARDISSUER_SN_VERIFY,
  2694. GetOutPutStr("%s%d", "VerifyCardIssuerSN", dwRet).c_str());
  2695. return dwRet;
  2696. }
  2697. else
  2698. {
  2699. Dbg("VerifyCardIssuerSN succeed");
  2700. }
  2701. }
  2702. else
  2703. {
  2704. //校验操作权限(终端ID)
  2705. DWORD dwRet = VerifyTerminalNo();
  2706. if (SUCCESS != dwRet)
  2707. {
  2708. Dbg("VerifyTerminalNo failed");
  2709. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_TERMINAL_TASK_VERIFY,
  2710. GetOutPutStr("%s%d", "VerifyCardIssuerSN", dwRet).c_str());
  2711. return dwRet;
  2712. }
  2713. }
  2714. //获取用户类型
  2715. char cUserType[3] = {0};
  2716. bRet = GetUserTypeFromCert(USER_CERT_PATH, FORMAT_PEM, cUserType);
  2717. if (!bRet)
  2718. {
  2719. Dbg("GetUserTypeFromCert err!");
  2720. LogWarn(Severity_Middle,Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_TYPE_FROM_CERT,
  2721. GetOutPutStr("%s%s", "GetUserTypeFromCert", "False").c_str());
  2722. return GET_USERINFO_ERR;
  2723. }
  2724. //发送验证通过事件
  2725. UkeyVerify e = {};
  2726. UkeyVerifyEx ex = {};
  2727. char cUserInfo[2048] = {0};
  2728. char cUserID[17] = {0};
  2729. char cOfficeID[9] = {0};
  2730. int nUserType = atoi(cUserType);
  2731. //改成广播方式发送事件
  2732. bRet = GetUserIDFromCert(USER_CERT_PATH, FORMAT_PEM, cUserID);
  2733. if (bRet)
  2734. {
  2735. e.AgentId = cUserID;
  2736. }
  2737. else
  2738. {
  2739. Dbg("GetUserIDFromCert failed!");
  2740. e.AgentId = "";
  2741. }
  2742. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerify, eMsgSig_UkeyVerify, e);
  2743. Dbg("AgentId=%s", e.AgentId);
  2744. //遵循向前兼容原则,新增一个广播消息 modify by zl 20140918
  2745. bRet = GetUserInfoFromCert(USER_CERT_PATH, FORMAT_PEM, cUserInfo);
  2746. if (bRet)
  2747. {
  2748. string strUserInfo = cUserInfo;
  2749. strUserInfo += "UserType=";
  2750. strUserInfo += cUserType;
  2751. strUserInfo += ";";
  2752. memcpy(cUserInfo, strUserInfo.c_str(), strUserInfo.length());
  2753. ex.UserInfo = cUserInfo;
  2754. }
  2755. else
  2756. {
  2757. Dbg("GetUserInfoFromCert failed!");
  2758. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_INFO_FROM_CERT,
  2759. GetOutPutStr("%s%s", "GetUserInfoFromCert", "False").c_str());
  2760. ex.UserInfo = "";
  2761. }
  2762. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerifyEx, eMsgSig_UkeyVerifyEx, ex);
  2763. Dbg("UserInfo=%s", ex.UserInfo);
  2764. Dbg("%s has right to operator %s", m_strUserID.c_str(), m_strTerminalId.c_str());
  2765. }
  2766. switch (nUserType)
  2767. {
  2768. case 0: //维护
  2769. strUserType = "Maintian";
  2770. break;
  2771. case 1: //物料管理
  2772. strUserType = "MaterialManager";
  2773. break;
  2774. case 2: //业务
  2775. strUserType = "Business";
  2776. break;
  2777. case 3: //Agent3
  2778. strUserType = "Agent3";
  2779. break;
  2780. default:
  2781. strUserType = "unknown";
  2782. Dbg("userType=%d is wrong",nUserType);
  2783. break;
  2784. }
  2785. //只有维护类型电子钥匙写日志;频繁操作会导致日志写满,报错
  2786. //物料管理和维护类电子钥匙日志上传方式待业务流程讨论清楚后,再写日志
  2787. if (0)
  2788. {
  2789. if (0 == nUserType)
  2790. {
  2791. //写操作日志
  2792. //操作起始时间
  2793. time_t rawtime;
  2794. struct tm * timeinfo;
  2795. time(&rawtime);
  2796. timeinfo = localtime(&rawtime);
  2797. sprintf(m_StartTime, "%4d-%02d-%02d %02d:%02d:%02d",1900+timeinfo->tm_year, 1+timeinfo->tm_mon,timeinfo->tm_mday,timeinfo->tm_hour,timeinfo->tm_min,timeinfo->tm_sec);
  2798. //拼装日志记录
  2799. string logRecord = "";
  2800. logRecord += m_strTerminalId;
  2801. logRecord += "@";
  2802. logRecord += m_strUserID;
  2803. logRecord += "@";
  2804. logRecord += strUserType;
  2805. logRecord += "@";
  2806. logRecord += m_StartTime;
  2807. logRecord += "@";
  2808. logRecord += m_StartTime;
  2809. logRecord += ";";
  2810. Dbg("WriteLogFile ,logRecord=%s",logRecord.c_str());
  2811. DWORD dwRet = WriteLogFile(m_Handle, (BYTE*)logRecord.c_str(), logRecord.length());
  2812. if (DONGLE_SUCCESS == dwRet)
  2813. {
  2814. Dbg("WriteLogFile success");
  2815. }
  2816. else
  2817. {
  2818. Dbg("WriteLogFile failed,errcode=0x%08X",dwRet);
  2819. //MessageBox(0,"操作日志记录已满,请先登录作业平台上传操作日志!",NULL,0);
  2820. return LOG_FULL_ERR;
  2821. }
  2822. }
  2823. }
  2824. return DONGLE_SUCCESS;
  2825. }
  2826. DWORD ConsoleFSM::VerifyTerminalNo()
  2827. {
  2828. //校验终端操作权限
  2829. Dbg("start to verify terminal nubmer");
  2830. //获取电子钥匙中的授权终端列表
  2831. //终端列表格式:任务号1,终端号1,有效开始时间,有效结束时间@任务号2,终端号2,有效开始时间,有效结束时间@任务号3,终端号3,有效开始时间,有效结束时间@
  2832. char readDeviceList[4096] = {0};
  2833. DWORD dwRet = GetAuthorizedDevices(m_Handle, readDeviceList);
  2834. if (DONGLE_SUCCESS != dwRet)
  2835. {
  2836. Dbg("GetAuthorizedDevices failed,Return: 0x%08X",dwRet);
  2837. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_AUTHER_DEVICES_LIST,
  2838. GetOutPutStr("%s%08X", "GetAuthorizedDevices", dwRet).c_str());
  2839. if (0xF000000F == dwRet)
  2840. {
  2841. Dbg("电子钥匙未下载设备的操作权限,请先下载!");
  2842. }
  2843. return TERMINAL_NO_TASK;
  2844. }
  2845. string strDeviceList = readDeviceList;
  2846. int nLastPos = strDeviceList.find_last_of("@");
  2847. if (-1 == nLastPos)
  2848. {
  2849. Dbg("find '@' in device list failed, DeviceList=%s", readDeviceList);
  2850. return TERMINAL_AUTHORIZED_ERR;
  2851. }
  2852. string strNewDeviceList = strDeviceList.substr(0, nLastPos+1);
  2853. Dbg("GetAuthorizedDevices success, DeviceList=%s", strNewDeviceList.c_str());
  2854. //校验终端号
  2855. BOOL bOverTime = FALSE;
  2856. BOOL bNoBeginValid = FALSE;
  2857. unsigned int nDeviceListLen = strNewDeviceList.length();
  2858. while (nDeviceListLen >= 0)
  2859. {
  2860. DWORD dwPos = FindSubStr(strNewDeviceList.c_str(), m_strTerminalId.c_str());
  2861. if (-1 == dwPos)
  2862. {
  2863. Dbg("current TerminalNo=%s , can't find it in authorized device list=%s", m_strTerminalId.c_str(), strNewDeviceList.c_str());
  2864. if (bOverTime)
  2865. {
  2866. //如果一个终端ID存在多个授权记录,且同时存在过期和未生效的记录,则报过期错误
  2867. //MessageBoxA(NULL, "设备授权已过期,请确认授权时间!",NULL,0);
  2868. Dbg("设备授权已过期,请确认授权时间!");
  2869. return TERMINAL_AUTHORIZED_OVERTIME_ERR;
  2870. }
  2871. else if (bNoBeginValid)
  2872. {
  2873. //MessageBoxA(NULL, "设备授权未开始生效,请确认授权时间!",NULL,0);
  2874. Dbg("设备授权未开始生效,请确认授权时间!");
  2875. return TERMINAL_AUTHORIZED_NOTBEGIN_ERR;
  2876. }
  2877. else
  2878. {
  2879. //MessageBoxA(NULL, "电子钥匙不具备此设备操作授权,请确认是否已授权!",NULL,0);
  2880. Dbg("电子钥匙不具备此设备操作授权,请确认是否已授权!");
  2881. return TERMINAL_AUTHORIZED_ERR;
  2882. }
  2883. }
  2884. else
  2885. {
  2886. //兼容带任务ID和不带任务ID的操作任务
  2887. int nSubCnt = 0;//分割符(",")的个数
  2888. int nFirstRecordEndPos = strNewDeviceList.find_first_of("@");
  2889. if(-1 != nFirstRecordEndPos)
  2890. {
  2891. string firstRecord = strNewDeviceList.substr(0, nFirstRecordEndPos+1);
  2892. int nfirstSubEndPos = firstRecord.find_first_of(",");
  2893. while (-1 != nfirstSubEndPos)
  2894. {
  2895. nSubCnt++;
  2896. firstRecord = firstRecord.substr(nfirstSubEndPos+1, firstRecord.length()-nfirstSubEndPos-1);
  2897. nfirstSubEndPos = firstRecord.find_first_of(",");
  2898. }
  2899. Dbg("找到[%d]个分隔符", nSubCnt);
  2900. if (3==nSubCnt)
  2901. {
  2902. Dbg("任务包含TaskID");
  2903. //获取任务ID长度
  2904. string rest = strNewDeviceList.substr(0, dwPos);
  2905. int restEndPos = rest.find_last_of("@");
  2906. int nTaskIDLen = 0;
  2907. if (-1 != restEndPos)
  2908. {
  2909. nTaskIDLen = dwPos - restEndPos - 1 -1;
  2910. }
  2911. else
  2912. {
  2913. nTaskIDLen = dwPos - 1;
  2914. }
  2915. //校验任务ID、终端号和准入时间
  2916. string strRest = strNewDeviceList.substr(0, dwPos-nTaskIDLen-1);
  2917. string strNew = strNewDeviceList.substr(dwPos-nTaskIDLen-1, strNewDeviceList.length()-strRest.length());
  2918. unsigned int nEndPos = strNew.find_first_of("@");
  2919. string strRecord = strNew.substr(0, nEndPos+1);
  2920. //获取任务ID
  2921. unsigned int nRecordLen = strRecord.length();
  2922. int nPos = strRecord.find_first_of(",");
  2923. string strTaskId = strRecord.substr(0, nPos);
  2924. string strTemp = strRecord.substr(nPos+1, nRecordLen-nPos-1);
  2925. //获取任务状态
  2926. if (!VerifyTaskStatus(strTaskId.c_str()))
  2927. {
  2928. strNewDeviceList = strNewDeviceList.substr(dwPos-nTaskIDLen-1+nRecordLen, strNewDeviceList.length()-dwPos+nTaskIDLen-1-nRecordLen);
  2929. nDeviceListLen -= (dwPos-nTaskIDLen-1 + nRecordLen);
  2930. continue;
  2931. }
  2932. //获取终端号
  2933. nRecordLen = strTemp.length();
  2934. nPos = strTemp.find_first_of(",");
  2935. string strDeviceId = strTemp.substr(0, nPos);
  2936. strTemp = strTemp.substr(nPos+1, nRecordLen-nPos-1);
  2937. //获取有效时间
  2938. unsigned int nStartTimeEndPos = strTemp.find_first_of(",");
  2939. string strStartTime = strTemp.substr(0, nStartTimeEndPos);
  2940. string strEndTime = strTemp.substr(nStartTimeEndPos+1,strTemp.length()-strStartTime.length()-2);
  2941. if ((strStartTime.length() <=1) || (strEndTime.length() <= 1))
  2942. {
  2943. Dbg("此任务中设备授权有效期格式错误");
  2944. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  2945. nDeviceListLen -= (dwPos + nRecordLen);
  2946. continue;
  2947. }
  2948. time_t tNow = NowTime();
  2949. string strNowTime = Timet2Str(tNow);
  2950. time_t tStartTime = Str2Timet(strStartTime);
  2951. time_t tEndTime = Str2Timet(strEndTime);
  2952. Dbg("current time=%s , TaskRecord=%s", strNowTime.c_str(), strRecord.c_str());
  2953. if (tEndTime < tNow)
  2954. {
  2955. Dbg("此任务中的设备授权已过期");
  2956. bOverTime = TRUE;
  2957. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  2958. nDeviceListLen -= (dwPos + nRecordLen);
  2959. continue;
  2960. }
  2961. else if (tStartTime > tNow)
  2962. {
  2963. Dbg("此任务中的设备授权未开始生效");
  2964. bNoBeginValid = TRUE;
  2965. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  2966. nDeviceListLen -= (dwPos + nRecordLen);
  2967. continue;
  2968. }
  2969. else
  2970. {
  2971. Dbg("找到当前设备有效授权任务");
  2972. return SUCCESS;
  2973. }
  2974. }
  2975. else if (2==nSubCnt)
  2976. {
  2977. Dbg("任务不包含TaskID");
  2978. //只校验终端号和准入时间
  2979. string strRest = strNewDeviceList.substr(0, dwPos);
  2980. string strNew = strNewDeviceList.substr(dwPos, strNewDeviceList.length()-strRest.length());
  2981. unsigned int nEndPos = strNew.find_first_of("@");
  2982. string strRecord = strNew.substr(0, nEndPos+1);
  2983. //获取终端号
  2984. unsigned int nRecordLen = strRecord.length();
  2985. int nPos = strRecord.find_first_of(",");
  2986. string strDeviceId = strRecord.substr(0, nPos);
  2987. string strTemp = strRecord.substr(nPos+1, nRecordLen-nPos-1);
  2988. //获取有效时间
  2989. unsigned int nStartTimeEndPos = strTemp.find_first_of(",");
  2990. string strStartTime = strTemp.substr(0, nStartTimeEndPos);
  2991. string strEndTime = strTemp.substr(nStartTimeEndPos+1,strTemp.length()-strStartTime.length()-2);
  2992. if ((strStartTime.length() <=1) || (strEndTime.length() <= 1))
  2993. {
  2994. Dbg("此任务中设备授权有效期格式错误");
  2995. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  2996. nDeviceListLen -= (dwPos + nRecordLen);
  2997. continue;
  2998. }
  2999. time_t tNow = NowTime();
  3000. string strNowTime = Timet2Str(tNow);
  3001. time_t tStartTime = Str2Timet(strStartTime);
  3002. time_t tEndTime = Str2Timet(strEndTime);
  3003. Dbg("current time=%s , TaskRecord=%s", strNowTime.c_str(), strRecord.c_str());
  3004. if (tEndTime < tNow)
  3005. {
  3006. Dbg("此任务中的设备授权已过期");
  3007. bOverTime = TRUE;
  3008. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  3009. nDeviceListLen -= (dwPos + nRecordLen);
  3010. continue;
  3011. }
  3012. else if (tStartTime > tNow)
  3013. {
  3014. Dbg("此任务中的设备授权未开始生效");
  3015. bNoBeginValid = TRUE;
  3016. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  3017. nDeviceListLen -= (dwPos + nRecordLen);
  3018. continue;
  3019. }
  3020. else
  3021. {
  3022. Dbg("找到当前设备有效授权任务");
  3023. return SUCCESS;
  3024. }
  3025. }
  3026. else
  3027. {
  3028. //异常
  3029. //MessageBoxA(NULL, "设备授权格式错误,请联系技术人员!",NULL,0);
  3030. Dbg("设备授权格式错误,请联系技术人员!");
  3031. }
  3032. }
  3033. }
  3034. }
  3035. return SUCCESS;
  3036. }
  3037. DWORD ConsoleFSM::VerifyCardIssuerSN()
  3038. {
  3039. //校验终端操作权限
  3040. Dbg("start to verify card issuer sn");
  3041. //获取电子钥匙中的授权终端列表
  3042. //终端列表格式:任务号1,终端号1,有效开始时间,有效结束时间@任务号2,终端号2,有效开始时间,有效结束时间@任务号3,终端号3,有效开始时间,有效结束时间@
  3043. char readDeviceList[4096] = {0};
  3044. DWORD dwRet = GetAuthorizedDevices(m_Handle, readDeviceList);
  3045. if (DONGLE_SUCCESS != dwRet)
  3046. {
  3047. Dbg("GetAuthorizedDevices failed,Return: 0x%08X",dwRet);
  3048. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_AUTHER_DEVICES_LIST,
  3049. GetOutPutStr("%s%08X", "GetAuthorizedDevices", dwRet).c_str());
  3050. if (0xF000000F == dwRet)
  3051. {
  3052. //MessageBoxA(NULL, "电子钥匙未下载设备的操作权限,请先下载!",NULL,0);
  3053. Dbg("电子钥匙未下载设备的操作权限,请先下载!");
  3054. }
  3055. return TERMINAL_NO_TASK;
  3056. }
  3057. string strDeviceList = readDeviceList;
  3058. int nLastPos = strDeviceList.find_last_of("@");
  3059. if (-1 == nLastPos)
  3060. {
  3061. Dbg("find '@' in device list failed, DeviceList=%s", readDeviceList);
  3062. return TERMINAL_AUTHORIZED_ERR;
  3063. }
  3064. string strNewDeviceList = strDeviceList.substr(0, nLastPos+1);
  3065. Dbg("GetAuthorizedDevices success, DeviceList=%s", strNewDeviceList.c_str());
  3066. //校验终端号
  3067. BOOL bOverTime = FALSE;
  3068. BOOL bNoBeginValid = FALSE;
  3069. unsigned int nDeviceListLen = strNewDeviceList.length();
  3070. while (nDeviceListLen >= 0)
  3071. {
  3072. DWORD dwPos = FindSubStr(strNewDeviceList.c_str(), m_CardIssuerSN.GetData());
  3073. if (-1 == dwPos)
  3074. {
  3075. Dbg("current CardIssuerSN=%s , can't find it in authorized device list=%s", m_CardIssuerSN.GetData(), strNewDeviceList.c_str());
  3076. if (bOverTime)
  3077. {
  3078. //如果一个终端ID存在多个授权记录,且同时存在过期和未生效的记录,则报过期错误
  3079. Dbg("设备授权已过期,请确认授权时间!");
  3080. return TERMINAL_AUTHORIZED_OVERTIME_ERR;
  3081. }
  3082. else if (bNoBeginValid)
  3083. {
  3084. Dbg("设备授权未开始生效,请确认授权时间!");
  3085. return TERMINAL_AUTHORIZED_NOTBEGIN_ERR;
  3086. }
  3087. else
  3088. {
  3089. Dbg("电子钥匙不具备此设备操作授权,请确认是否已授权!");
  3090. return TERMINAL_AUTHORIZED_ERR;
  3091. }
  3092. }
  3093. else
  3094. {
  3095. //校验任务号、卡机号和准入时间
  3096. string strRest = strNewDeviceList.substr(0, dwPos-17);
  3097. string strNew = strNewDeviceList.substr(dwPos-17, strNewDeviceList.length()-strRest.length());
  3098. unsigned int nEndPos = strNew.find_first_of("@");
  3099. string strRecord = strNew.substr(0, nEndPos+1);
  3100. //获取任务ID
  3101. unsigned int nRecordLen = strRecord.length();
  3102. int nPos = strRecord.find_first_of(",");
  3103. string strTaskId = strRecord.substr(0, nPos);
  3104. string strTemp = strRecord.substr(nPos+1, nRecordLen-nPos-1);
  3105. //获取任务状态
  3106. if(16 != strTaskId.length())
  3107. {
  3108. Dbg("<TaskId=%s>任务ID的长度不等于16,格式错误!",strTaskId.c_str());
  3109. strNewDeviceList = strNewDeviceList.substr(dwPos-17+nRecordLen, strNewDeviceList.length()-dwPos+17-nRecordLen);
  3110. nDeviceListLen -= (dwPos-17 + nRecordLen);
  3111. continue;
  3112. }
  3113. if (!VerifyTaskStatus(strTaskId.c_str()))
  3114. {
  3115. strNewDeviceList = strNewDeviceList.substr(dwPos-17+nRecordLen, strNewDeviceList.length()-dwPos+17-nRecordLen);
  3116. nDeviceListLen -= (dwPos-17 + nRecordLen);
  3117. continue;
  3118. }
  3119. //获取卡机号
  3120. nRecordLen = strTemp.length();
  3121. nPos = strTemp.find_first_of(",");
  3122. string strDeviceId = strTemp.substr(0, nPos);
  3123. strTemp = strTemp.substr(nPos+1, nRecordLen-nPos-1);
  3124. //获取有效时间
  3125. unsigned int nStartTimeEndPos = strTemp.find_first_of(",");
  3126. string strStartTime = strTemp.substr(0, nStartTimeEndPos);
  3127. string strEndTime = strTemp.substr(nStartTimeEndPos+1,strTemp.length()-strStartTime.length()-2);
  3128. if ((strStartTime.length() <=1) || (strEndTime.length() <= 1))
  3129. {
  3130. Dbg("此任务中设备授权有效期格式错误");
  3131. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  3132. nDeviceListLen -= (dwPos + nRecordLen);
  3133. continue;
  3134. }
  3135. time_t tNow = NowTime();
  3136. string strNowTime = Timet2Str(tNow);
  3137. time_t tStartTime = Str2Timet(strStartTime);
  3138. time_t tEndTime = Str2Timet(strEndTime);
  3139. Dbg("current time=%s , TaskRecord=%s", strNowTime.c_str(), strRecord.c_str());
  3140. if (tEndTime < tNow)
  3141. {
  3142. Dbg("此任务中的设备授权已过期");
  3143. bOverTime = TRUE;
  3144. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  3145. nDeviceListLen -= (dwPos + nRecordLen);
  3146. continue;
  3147. }
  3148. else if (tStartTime > tNow)
  3149. {
  3150. Dbg("此任务中的设备授权未开始生效");
  3151. bNoBeginValid = TRUE;
  3152. strNewDeviceList = strNewDeviceList.substr(dwPos+nRecordLen, strNewDeviceList.length()-dwPos-nRecordLen);
  3153. nDeviceListLen -= (dwPos + nRecordLen);
  3154. continue;
  3155. }
  3156. else
  3157. {
  3158. Dbg("找到当前设备有效授权任务");
  3159. return SUCCESS;
  3160. }
  3161. }
  3162. }
  3163. return SUCCESS;
  3164. }
  3165. DWORD ConsoleFSM::CheckCertVaildity()
  3166. {
  3167. time_t tNow = NowTime();
  3168. char strRootStartTime[32] = {0};
  3169. char strRootEndTime[32] = {0};
  3170. BOOL bRet = GetValidityFromCert(ROOT_CERT_PATH, FORMAT_PEM, strRootStartTime, strRootEndTime);
  3171. if (!bRet)
  3172. {
  3173. Dbg("获取根证书有效期失败");
  3174. return GET_CERT_VALIDITY_ERR;
  3175. }
  3176. Dbg("根证书有效期[%s,%s]", strRootStartTime, strRootEndTime);
  3177. string strNewRootStartTime = "20";
  3178. string strNewRootEndTime = "20";
  3179. strNewRootStartTime += strRootStartTime;
  3180. strNewRootEndTime += strRootEndTime;
  3181. time_t tRootStartTime = Str2Timet(strNewRootStartTime.c_str(), "%4d%2d%2d%2d%2d%2d");
  3182. time_t tRootEndTime = Str2Timet(strNewRootEndTime, "%4d%2d%2d%2d%2d%2d");
  3183. UINT nRestSecond = tRootEndTime - tNow;
  3184. UINT nRestDay = nRestSecond/(60*60*24);
  3185. if (nRestDay <= 15)
  3186. {
  3187. /*CString msg;
  3188. msg.Format(_T("电子钥匙即将过期,剩余有效天数[%d],请及时续期!"), nRestDay);
  3189. ::MessageBox(NULL,msg, "提示",0);*/
  3190. Dbg("根证书即将过期, 有效剩余天数:%d", nRestDay);
  3191. return OVERTTIMER_WAR;
  3192. }
  3193. char strCaStartTime[32] = {0};
  3194. char strCaEndTime[32] = {0};
  3195. bRet = GetValidityFromCert(CA_CERT_PATH, FORMAT_PEM, strCaStartTime, strCaEndTime);
  3196. if (!bRet)
  3197. {
  3198. Dbg("获取分行证书有效期失败");
  3199. return GET_CERT_VALIDITY_ERR;
  3200. }
  3201. Dbg("分行证书有效期[%s,%s]", strCaStartTime, strCaEndTime);
  3202. string strNewCaStartTime = "20";
  3203. string strNewCaEndTime = "20";
  3204. strNewCaStartTime += strCaStartTime;
  3205. strNewCaEndTime += strCaEndTime;
  3206. time_t tCaStartTime = Str2Timet(strNewCaStartTime.c_str(), "%4d%2d%2d%2d%2d%2d");
  3207. time_t tCaEndTime = Str2Timet(strNewCaEndTime, "%4d%2d%2d%2d%2d%2d");
  3208. nRestSecond = tCaEndTime - tNow;
  3209. nRestDay = nRestSecond/(60*60*24);
  3210. if (nRestDay <= 15)
  3211. {
  3212. /*CString msg;
  3213. msg.Format(_T("电子钥匙即将过期,剩余有效天数[%d],请及时续期!"), nRestDay);
  3214. ::MessageBox(NULL,msg, "提示",0);*/
  3215. Dbg("分行证书即将过期, 有效剩余天数:%d", nRestDay);
  3216. return OVERTTIMER_WAR;
  3217. }
  3218. char strUserStartTime[32] = {0};
  3219. char strUserEndTime[32] = {0};
  3220. bRet = GetValidityFromCert(USER_CERT_PATH, FORMAT_PEM, strUserStartTime, strUserEndTime);
  3221. if (!bRet)
  3222. {
  3223. Dbg("获取用户证书有效期失败");
  3224. return GET_CERT_VALIDITY_ERR;
  3225. }
  3226. Dbg("用户证书有效期[%s,%s]", strUserStartTime, strUserEndTime);
  3227. string strNewUserStartTime = "20";
  3228. string strNewUserEndTime = "20";
  3229. strNewUserStartTime += strUserStartTime;
  3230. strNewUserEndTime += strUserEndTime;
  3231. time_t tUserStartTime = Str2Timet(strNewUserStartTime.c_str(), "%4d%2d%2d%2d%2d%2d");
  3232. time_t tUserEndTime = Str2Timet(strNewUserEndTime, "%4d%2d%2d%2d%2d%2d");
  3233. nRestSecond = tUserEndTime - tNow;
  3234. nRestDay = nRestSecond/(60*60*24);
  3235. if (nRestDay <= 15)
  3236. {
  3237. /*CString msg;
  3238. msg.Format(_T("电子钥匙即将过期,剩余有效天数[%d],请及时续期!"), nRestDay);
  3239. ::MessageBox(NULL,msg, "提示",0);*/
  3240. Dbg("用户证书即将过期, 有效剩余天数:%d", nRestDay);
  3241. return OVERTTIMER_WAR;
  3242. }
  3243. return SUCCESS;
  3244. }
  3245. BOOL ConsoleFSM::GetUserInfo()
  3246. {
  3247. //电子钥匙HID
  3248. char strUkeyHID[32] = {0};
  3249. bool bRet = GetHIDFromCert(USER_CERT_PATH, FORMAT_PEM, strUkeyHID);
  3250. if (!bRet)
  3251. {
  3252. Dbg("GetHIDFromCert failed");
  3253. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_EKEY_HID,
  3254. GetOutPutStr("%s%s", "GetHIDFromCert", "False").c_str());
  3255. return FALSE;
  3256. }
  3257. m_strKeyHID = strUkeyHID;
  3258. Dbg("HID=%s", m_strKeyHID.c_str());
  3259. //用户ID
  3260. char strUserID[32] = {0};
  3261. bRet = GetUserIDFromCert(USER_CERT_PATH, FORMAT_PEM, strUserID);
  3262. if (!bRet)
  3263. {
  3264. Dbg("GetUserIDFromCert failed");
  3265. return FALSE;
  3266. }
  3267. m_strUserID = strUserID;
  3268. Dbg("userId=%s", m_strUserID.c_str());
  3269. //用户类型
  3270. char cUserType[3] = {0};
  3271. bRet = GetUserTypeFromCert(USER_CERT_PATH, FORMAT_PEM, cUserType);
  3272. if (!bRet)
  3273. {
  3274. Dbg("GetUserTypeFromCert failed");
  3275. return FALSE;
  3276. }
  3277. m_strUserType = cUserType;
  3278. Dbg("userType=%s", m_strUserType.c_str());
  3279. return TRUE;
  3280. }
  3281. //获取当前连接的卡箱号
  3282. BOOL ConsoleFSM::GetIssuerNum()
  3283. {
  3284. ErrorCodeEnum eErr = Error_Succeed;
  3285. if (m_pCardIssuerService == NULL)
  3286. {
  3287. m_pCardIssuerService = new CardIssuerService_ClientBase(this->GetEntityBase());
  3288. eErr = m_pCardIssuerService->Connect();
  3289. if (Error_Succeed != eErr)
  3290. {
  3291. Dbg("CardIssuer connected failed.");
  3292. m_pCardIssuerService = NULL;
  3293. return FALSE;
  3294. }
  3295. else
  3296. {
  3297. Dbg("CardIssuer connected ok.");
  3298. }
  3299. }
  3300. Dbg("CardIssuer connected.");
  3301. CardIssuerService_GetSCIInfo_Req req;
  3302. CardIssuerService_GetSCIInfo_Ans ans;
  3303. eErr = m_pCardIssuerService->GetSCIInfo(req, ans,5000);
  3304. if (Error_Succeed != eErr)
  3305. {
  3306. Dbg("GetSCIInfo failed.");
  3307. m_pCardIssuerService = NULL;
  3308. return FALSE;
  3309. }
  3310. else
  3311. {
  3312. m_CardIssuerSN = ans.DevSN;
  3313. Dbg("GetSCIInfo succeed, DevSN=[%s]", m_CardIssuerSN);
  3314. }
  3315. return TRUE;
  3316. }
  3317. //GPIO控制,true:打开;false:关闭
  3318. ErrorCodeEnum ConsoleFSM::OpenGpio(BOOL bOpen)
  3319. {
  3320. ErrorCodeEnum eErr = Error_Succeed;
  3321. if (m_pDeviceControlService == NULL)
  3322. {
  3323. m_pDeviceControlService = new DeviceControlService_ClientBase(this->GetEntityBase());
  3324. eErr = m_pDeviceControlService->Connect();
  3325. if (Error_Succeed != eErr)
  3326. {
  3327. Dbg("DeviceControl connected failed.");
  3328. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_DEVICECONTROL_CONNECT,
  3329. GetOutPutStr("%s%08X", "Connect", eErr).c_str());
  3330. m_pDeviceControlService = NULL;
  3331. return eErr;
  3332. }
  3333. else
  3334. {
  3335. Dbg("DeviceControl connected ok.");
  3336. }
  3337. }
  3338. Dbg("DeviceControl connected.");
  3339. DeviceControlService_USB_Req req;
  3340. DeviceControlService_USB_Ans ans;
  3341. req.open = bOpen;
  3342. eErr = m_pDeviceControlService->USB(req, ans,10000);
  3343. if (Error_Succeed != eErr)
  3344. {
  3345. Dbg("m_pDeviceControlService->USB failed.");
  3346. m_pDeviceControlService = NULL;
  3347. return eErr;
  3348. }
  3349. else
  3350. {
  3351. Dbg("m_pDeviceControlService->USB succeed, bOpen[%d]", bOpen);
  3352. }
  3353. return eErr;
  3354. }
  3355. CSmartPointer<IEntityFunction> ConsoleFSM::GetFunction()
  3356. {
  3357. return m_pEntity->GetFunction();
  3358. }
  3359. ErrorCodeEnum ConsoleFSM::SetRunConfig()
  3360. {
  3361. char* pSap = "00000001";
  3362. char* pTime = "493751309";
  3363. CSmartPointer<IEntityFunction> spFunction = m_pEntity->GetFunction();
  3364. CSmartPointer<IConfigInfo> spConfig;
  3365. ErrorCodeEnum Error = spFunction->OpenConfig(Config_Run, spConfig);
  3366. if (Error_Succeed == Error)
  3367. {
  3368. CSimpleStringA strSAP;
  3369. Error = spConfig->ReadConfigValue("UserInfo", "SAP", strSAP);
  3370. if ((Error_Succeed == Error) && (strSAP.IsNullOrEmpty()))
  3371. {
  3372. Error = spConfig->WriteConfigValue("UserInfo", "SAP", pSap);
  3373. Error = spConfig->WriteConfigValue("UserInfo", "Time", pTime);
  3374. }
  3375. }
  3376. return Error;
  3377. }
  3378. ErrorCodeEnum ConsoleFSM::LoadRunConfig()
  3379. {
  3380. CSmartPointer<IEntityFunction> spFunction = m_pEntity->GetFunction();
  3381. CSmartPointer<IConfigInfo> spConfig;
  3382. ErrorCodeEnum Error = spFunction->OpenConfig(Config_Run, spConfig);
  3383. if (Error == Error_Succeed)
  3384. {
  3385. Error = spConfig->ReadConfigValueInt("Ukey", "UkeyStatus", m_nUkeyStatus);
  3386. }
  3387. return Error;
  3388. }
  3389. ErrorCodeEnum ConsoleFSM::LoadCenterConfig()
  3390. {
  3391. CSmartPointer<IEntityFunction> spFunction = m_pEntity->GetFunction();
  3392. CSmartPointer<IConfigInfo> spConfig;
  3393. ErrorCodeEnum Error = spFunction->OpenConfig(Config_CenterSetting, spConfig);
  3394. if (Error_Succeed == Error)
  3395. {
  3396. Error = spConfig->ReadConfigValueInt("MaintainWatcher", "CheckTerMinalNo", m_nCheckTerminalNo);
  3397. if (Error_Succeed == Error)
  3398. {
  3399. Dbg("get CheckTerMinalNo=%d from CenterSetting.ini", m_nCheckTerminalNo);
  3400. }
  3401. else
  3402. {
  3403. Dbg("get CheckTerMinalNo from CenterSetting.ini failed");
  3404. }
  3405. Error = spConfig->ReadConfigValueInt("MaintainWatcher", "CheckMaintainTask", m_nCheckMaintainTask);
  3406. if (Error_Succeed == Error)
  3407. {
  3408. Dbg("get CheckMaintainTask=%d from CenterSetting.ini", m_nCheckMaintainTask);
  3409. }
  3410. else
  3411. {
  3412. Dbg("get CheckMaintainTask from CenterSetting.ini failed");
  3413. }
  3414. Error = spConfig->ReadConfigValueInt("MaintainWatcher", "Verify_BlackList", m_nVerifyBlackList);
  3415. if (Error_Succeed == Error)
  3416. {
  3417. Dbg("get Verify_BlackList=%d from CenterSetting.ini", m_nVerifyBlackList);
  3418. }
  3419. else
  3420. {
  3421. Dbg("get Verify_BlackList from CenterSetting.ini failed");
  3422. }
  3423. Error = spConfig->ReadConfigValueInt("InteractiveControl", "SurveillanceRecordAllowed", g_nSurveillanceRecordAllowed);
  3424. if (Error_Succeed == Error)
  3425. {
  3426. Dbg("get SurveillanceRecordAllowed=%d from CenterSetting.ini", g_nSurveillanceRecordAllowed);
  3427. }
  3428. else
  3429. {
  3430. Dbg("get SurveillanceRecordAllowed from CenterSetting.ini failed");
  3431. }
  3432. }
  3433. return Error;
  3434. }
  3435. ErrorCodeEnum ConsoleFSM::LoadServerConfig()
  3436. {
  3437. CSmartPointer<IEntityFunction> spFunction = m_pEntity->GetFunction();
  3438. CSmartPointer<IConfigInfo> spConfig;
  3439. ErrorCodeEnum Error = spFunction->OpenConfig(Config_Software, spConfig);
  3440. if (Error == Error_Succeed)
  3441. {
  3442. Error = spConfig->ReadConfigValue("VerifyKey", "Key", m_StringVerifyKey);
  3443. if(Error == Error_Succeed)
  3444. Error = spConfig->ReadConfigValue("GuiConsole", "Name", m_StringConsoleName);
  3445. }
  3446. return Error;
  3447. }
  3448. ErrorCodeEnum ConsoleFSM::LoadRootConfig()
  3449. {
  3450. CSmartPointer<IEntityFunction> spFunction = m_pEntity->GetFunction();
  3451. CSmartPointer<IConfigInfo> spConfig;
  3452. ErrorCodeEnum Error = spFunction->OpenConfig(Config_Root, spConfig);
  3453. if (Error == Error_Succeed)
  3454. {
  3455. Error = spConfig->ReadConfigValue("Terminal", "MachineType", m_strMachineType);
  3456. if(Error == Error_Succeed)
  3457. {
  3458. Error = spConfig->ReadConfigValue("Terminal", "Site", m_strSite);
  3459. if (Error != Error_Succeed)
  3460. {
  3461. Dbg("ReadConfigValue, get Site value failed");
  3462. }
  3463. }
  3464. else
  3465. {
  3466. Dbg("ReadConfigValue, get MachineType value failed");
  3467. }
  3468. Error = spConfig->ReadConfigValue("Video", "EwsCamera", g_strEwsCameraID);
  3469. if (Error_Succeed == Error)
  3470. {
  3471. Dbg("get EwsCamera=%s from root.ini", g_strEwsCameraID);
  3472. }
  3473. else
  3474. {
  3475. Dbg("get EwsCamera from root.ini failed");
  3476. }
  3477. }
  3478. return Error;
  3479. }
  3480. void ConsoleFSM::CloseConnection()
  3481. {
  3482. if (m_bAns && m_pConnection != NULL)
  3483. {
  3484. m_pConnection->Close();
  3485. m_pConnection->DecRefCount();
  3486. m_pConnection = NULL;
  3487. m_bAns = FALSE;
  3488. Dbg("Close m_pConnection");
  3489. }
  3490. }
  3491. void ConsoleFSM::s0_on_entry()
  3492. {
  3493. //MessageBoxA(0,0,0,0);
  3494. Dbg("s0_on_entry, starting");
  3495. unsigned int count = 0;
  3496. DWORD dwRet = 0;
  3497. //状态机启动时检测电子钥匙
  3498. dwRet = FindUsbKey(&g_USBKeyInfo, &count);
  3499. if (dwRet == DONGLE_SUCCESS)
  3500. {
  3501. Dbg("when s0_on_entry,FindUsbKey success!");
  3502. if (!g_FindKey)
  3503. {
  3504. g_FindKey = TRUE;
  3505. PostEventLIFO(new FSMEvent(USER_EVT_JMP_UKEYINSERT));
  3506. }
  3507. }
  3508. else
  3509. {
  3510. Dbg("when s0_on_entry,FindUsbKey Fail!");
  3511. LogWarn(Severity_Low,Error_Unexpect, ERR_MAINTAINWATCHER_CANNOT_FIND_EKEY,
  3512. GetOutPutStr("%s%d", "FindUsbKey", dwRet).c_str());
  3513. if (g_FindKey)
  3514. {
  3515. g_FindKey = FALSE;
  3516. PostEventLIFO(new FSMEvent(USER_EVT_JMP_UKEYPULLOUT));
  3517. return;
  3518. }
  3519. ScheduleTimer(1, 10);
  3520. }
  3521. }
  3522. void ConsoleFSM::s0_on_exit()
  3523. {
  3524. CancelTimer(1);
  3525. }
  3526. unsigned int ConsoleFSM::s0_on_event( FSMEvent* event )
  3527. {
  3528. //收到回复,释放链接
  3529. CloseConnection();
  3530. if (EVT_TIMER == event->iEvt)
  3531. {
  3532. ScheduleTimer(1, 100);
  3533. }
  3534. else if (USER_EVT_JMP_UKEYINSERT == event->iEvt)
  3535. {
  3536. PostEventLIFO(new FSMEvent(USER_EVT_JMP_UKEYINSERT));
  3537. }
  3538. else if (USER_EVT_JMP_UKEYPULLOUT == event->iEvt)
  3539. {
  3540. PostEventLIFO(new FSMEvent(USER_EVT_JMP_UKEYPULLOUT));
  3541. }
  3542. return 0;
  3543. }
  3544. void ConsoleFSM::s1_on_entry()
  3545. {
  3546. Dbg("s1_on_entry");
  3547. Dbg("检测到电子钥匙插入");
  3548. //收到回复,释放链接
  3549. CloseConnection();
  3550. LogEvent(Severity_Middle, EVENT_UKEY_INSERTED, "Ukey insert!");
  3551. UkeyInsert e = {};
  3552. for (int i=0; i<8; i++)
  3553. {
  3554. string id;
  3555. sprintf((char*)id.c_str(),"%02X", g_USBKeyInfo.m_HID[i]);
  3556. e.UkeyID += id.c_str();
  3557. }
  3558. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyInsert, eMsgSig_UkeyInsert, e);
  3559. Dbg("UkeyID=%s", e.UkeyID);
  3560. //根据是否在首页状态,决定下一个状态机状态
  3561. //获取首页状态
  3562. GetMenuStatus();
  3563. if (m_bMenu)
  3564. {
  3565. //首页状态,进入验证流程
  3566. PostEventLIFO(new FSMEvent(USER_EVT_JMP_VERIFYPWD));
  3567. }
  3568. else
  3569. {
  3570. //非首页状态
  3571. PostEventLIFO(new FSMEvent(USER_EVT_JMP_RELEASE));
  3572. }
  3573. }
  3574. void ConsoleFSM::s1_on_exit()
  3575. {
  3576. CancelTimer(1);
  3577. }
  3578. unsigned int ConsoleFSM::s1_on_event(FSMEvent* event)
  3579. {
  3580. //收到回复,释放链接
  3581. CloseConnection();
  3582. if (event->iEvt == EVT_TIMER)
  3583. {
  3584. //Dbg("Verify cert err");
  3585. }
  3586. if (event->iEvt == USER_EVT_JMP_UKEYPULLOUT)
  3587. {
  3588. Dbg("s1_on_event receive USER_EVT_JMP_UKEYPULLOUT");
  3589. }
  3590. return 0;
  3591. }
  3592. void ConsoleFSM::s2_on_entry()
  3593. {
  3594. Dbg("s2_on_entry");
  3595. Dbg("检测到电子钥匙拔出");
  3596. //收到回复,释放链接
  3597. CloseConnection();
  3598. //处理拔出状态
  3599. g_bWorkingStatus = FALSE;
  3600. g_bErrStatus = FALSE;
  3601. m_bInputpwdDlg = FALSE;
  3602. LogEvent(Severity_Middle, EVENT_UKEY_PULLOUT, "Ukey pullout!");
  3603. UkeyPullOut e = {};
  3604. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyPullOut, eMsgSig_UkeyPullOut, e);
  3605. PostEventLIFO(new FSMEvent(USER_EVT_JMP_RELEASE));
  3606. }
  3607. void ConsoleFSM::s2_on_exit()
  3608. {
  3609. CancelTimer(2);
  3610. }
  3611. unsigned int ConsoleFSM::s2_on_event(FSMEvent* event)
  3612. {
  3613. if (event->iEvt == EVT_TIMER)
  3614. {
  3615. }
  3616. return 0;
  3617. }
  3618. void ConsoleFSM::s3_on_entry()
  3619. {
  3620. Dbg("s3_on_entry");
  3621. Dbg("开始启动验证线程");
  3622. //收到回复,释放链接
  3623. CloseConnection();
  3624. //启动验证线程
  3625. if (!g_HasVerifyTask)
  3626. {
  3627. Dbg("创建等待验密线程");
  3628. GuidVerifyTask* pVerifyTask = new GuidVerifyTask(this);
  3629. GetEntityBase()->GetFunction()->PostThreadPoolTask(pVerifyTask);
  3630. g_HasVerifyTask = TRUE;
  3631. }
  3632. if (!m_bInputpwdDlg && g_FindKey)
  3633. {
  3634. //发送弹出密码输入界面
  3635. InputPwd e = {};
  3636. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_InputPwd, eMsgSig_InputPwd, e);
  3637. Dbg("弹出密码输入界面");
  3638. m_bInputpwdDlg = TRUE;
  3639. }
  3640. }
  3641. void ConsoleFSM::s3_on_exit()
  3642. {
  3643. }
  3644. unsigned int ConsoleFSM::s3_on_event(FSMEvent* event)
  3645. {
  3646. if (USER_EVT_JMP_PWDSUCCESS == event->iEvt)
  3647. {
  3648. DWORD dwRet = 0;
  3649. Dbg("revceive USER_EVT_JMP_PWDSUCCESS, m_strPwd[%s],m_errcode[%08X]",m_strPwd, m_errcode);
  3650. if (!m_strPwd.IsNullOrEmpty() && ((0 == m_errcode) || (OVERTTIMER_WAR == m_errcode)))
  3651. {
  3652. //清空密码
  3653. m_strPwd.Clear();
  3654. if (0)
  3655. {
  3656. BOOL bRet = CheckUSBKeyLife(m_strKeyHID.c_str());
  3657. if (!bRet)
  3658. {
  3659. Dbg("CheckUSBKeyLife failed!");
  3660. LogWarn(Severity_Low, Error_Unexpect, ERR_MAINTAINWATCHER_CHECK_EKEY_LIFE,
  3661. GetOutPutStr("%s%s", "CheckUSBKeyLife", "False").c_str());
  3662. }
  3663. }
  3664. //获取用户类型,根据不同的用户类型发送不同的事件
  3665. char cUserType[3] = {0};
  3666. BOOL bRet = GetUserTypeFromCert(USER_CERT_PATH, FORMAT_PEM, cUserType);
  3667. if (!bRet)
  3668. {
  3669. Dbg("GetUserTypeFromCert err!");
  3670. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_TYPE_FROM_CERT,
  3671. GetOutPutStr("%s%s%s%s", "GetUserTypeFromCert", "False","cUserType", cUserType).c_str());
  3672. PostEventLIFO(new FSMEvent(USER_EVT_JMP_RELEASE));
  3673. return 1;
  3674. }
  3675. UkeyVerify e = {};
  3676. UkeyVerifyEx ex = {};
  3677. char cUserInfo[2048] = {0};
  3678. char cUserID[17] = {0};
  3679. char cOfficeID[9] = {0};
  3680. int nUserType = atoi(cUserType);
  3681. switch(nUserType)
  3682. {
  3683. case USER_TYPE_MAINTAIN:
  3684. //LogEvent(Severity_Middle, EVENT_UKEY_MAINTTIAN, "维护权限!");
  3685. LogEvent(Severity_Middle, EVENT_UKEY_LOW_LEVEL, "Start local maintance!");
  3686. if (0 == m_nCheckMaintainTask)
  3687. {
  3688. bRet = GetUserInfoFromCert(USER_CERT_PATH, FORMAT_PEM, cUserInfo);
  3689. if (bRet)
  3690. {
  3691. string strUserInfo = cUserInfo;
  3692. strUserInfo += "UserType=";
  3693. strUserInfo += cUserType;
  3694. strUserInfo += ";";
  3695. memcpy(cUserInfo, strUserInfo.c_str(), strUserInfo.length());
  3696. ex.UserInfo = cUserInfo;
  3697. }
  3698. else
  3699. {
  3700. Dbg("GetUserInfoFromCert failed!");
  3701. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_INFO_FROM_CERT,
  3702. GetOutPutStr("%s%s%s%s", "GetUserInfoFromCert", "False", "cUserInfo", cUserInfo).c_str());
  3703. ex.UserInfo = "";
  3704. }
  3705. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyNormalMaintain, eMsgSig_UkeyNormalMaintain, ex);
  3706. Dbg("UserInfo=%s", ex.UserInfo);
  3707. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerifyEx, eMsgSig_UkeyVerifyEx, ex);
  3708. Dbg("eMsg_UkeyVerifyEx:UserInfo=%s", ex.UserInfo);
  3709. }
  3710. break;
  3711. case USER_TYPE_MATERIAL_MANAGER:
  3712. LogEvent(Severity_Middle, EVENT_UKEY_MATERIAL_MANAGER, "物料管理权限!");
  3713. if (0 == m_nCheckTerminalNo)
  3714. {
  3715. //改成广播方式发送事件 modify by zl 20140619
  3716. bRet = GetUserIDFromCert(USER_CERT_PATH, FORMAT_PEM, cUserID);
  3717. if (bRet)
  3718. {
  3719. e.AgentId = cUserID;
  3720. }
  3721. else
  3722. {
  3723. Dbg("GetUserIDFromCert failed!");
  3724. e.AgentId = "";
  3725. }
  3726. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerify, eMsgSig_UkeyVerify, e);
  3727. Dbg("AgentId=%s", e.AgentId);
  3728. //遵循向前兼容原则,新增一个广播消息 modify by zl 20140918
  3729. bRet = GetUserInfoFromCert(USER_CERT_PATH, FORMAT_PEM, cUserInfo);
  3730. if (bRet)
  3731. {
  3732. string strUserInfo = cUserInfo;
  3733. strUserInfo += "UserType=";
  3734. strUserInfo += cUserType;
  3735. strUserInfo += ";";
  3736. memcpy(cUserInfo, strUserInfo.c_str(), strUserInfo.length());
  3737. ex.UserInfo = cUserInfo;
  3738. }
  3739. else
  3740. {
  3741. Dbg("GetUserInfoFromCert failed!");
  3742. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_INFO_FROM_CERT,
  3743. GetOutPutStr("%s%s%s%s", "GetUserInfoFromCert", "False", "cUserInfo", cUserInfo).c_str());
  3744. ex.UserInfo = "";
  3745. }
  3746. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerifyEx, eMsgSig_UkeyVerifyEx, ex);
  3747. Dbg("UserInfo=%s", ex.UserInfo);
  3748. }
  3749. break;
  3750. case USER_TYPE_CUSTOMER_MANAGER:
  3751. LogEvent(Severity_Middle, EVENT_UKEY_CUSTOMER_MANAGER, "客户经理权限!");
  3752. bRet = GetUserInfoFromCert(USER_CERT_PATH, FORMAT_PEM, cUserInfo);
  3753. if (bRet)
  3754. {
  3755. string strUserInfo = cUserInfo;
  3756. strUserInfo += "UserType=";
  3757. strUserInfo += cUserType;
  3758. strUserInfo += ";";
  3759. memcpy(cUserInfo, strUserInfo.c_str(), strUserInfo.length());
  3760. ex.UserInfo = cUserInfo;
  3761. }
  3762. else
  3763. {
  3764. Dbg("GetUserInfoFromCert failed!");
  3765. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_GET_USER_INFO_FROM_CERT,
  3766. GetOutPutStr("%s%s%s%s", "GetUserInfoFromCert", "False", "cUserInfo", cUserInfo).c_str());
  3767. ex.UserInfo = "";
  3768. }
  3769. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_UkeyVerifyEx, eMsgSig_UkeyVerifyEx, ex);
  3770. Dbg("UserInfo=%s", ex.UserInfo);
  3771. Dbg("m_strMachineType=%s", m_strMachineType.GetData());
  3772. break;
  3773. case USER_TYPE_AGENT3:
  3774. LogEvent(Severity_Middle, EVENT_UKEY_AGENT3, "Agent3用户权限!");
  3775. break;
  3776. case USER_TYPE_FINANCING_MANAGER:
  3777. LogEvent(Severity_Middle, EVENT_UKEY_FINANCING_MANAGER, "理财经理权限!");
  3778. break;
  3779. default:
  3780. Dbg("未知用户类型!");
  3781. break;
  3782. }
  3783. }
  3784. else
  3785. {
  3786. Dbg("VerifyUser failed, errcode=0x%08X ", m_errcode);
  3787. char tmp[100] = { 0 };
  3788. sprintf(tmp,"VerifyUser failed, errcode=0x%08X",m_errcode);
  3789. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_EKEY_VERIFY,
  3790. GetOutPutStr("%s%08X", "VerifyUser", m_errcode).c_str());
  3791. //清空密码
  3792. m_strPwd.Clear();
  3793. m_errcode = 0;
  3794. g_bErrStatus = TRUE;
  3795. }
  3796. PostEventLIFO(new FSMEvent(USER_EVT_JMP_RELEASE));
  3797. }
  3798. else if (USER_EVT_JMP_PWDERR == event->iEvt)
  3799. {
  3800. //当验证错误且电子钥匙没有拔出时,状态机停留在s3验证状态,等待用户下一步操作
  3801. Dbg("when verifyUser return failed, keep s3 state!");
  3802. }
  3803. return 0;
  3804. }
  3805. void ConsoleFSM::s4_on_entry()
  3806. {
  3807. Dbg("s4_on_entry, release");
  3808. //收到回复,释放链接
  3809. CloseConnection();
  3810. //close电子钥匙,关闭界面
  3811. DWORD dwRet = CloseUsbKey(m_Handle);
  3812. if (DONGLE_SUCCESS != dwRet)
  3813. {
  3814. Dbg("CloseUsbKey failed,errcode= 0x%08X", dwRet);
  3815. char tmp[100];
  3816. sprintf(tmp, "CloseUsbKey failed,errcode= 0x%08X", dwRet);
  3817. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_CLOSE_EKEY,
  3818. GetOutPutStr("%s%d", "CloseUsbKey", dwRet).c_str());
  3819. }
  3820. if (AsyncStopEntity(m_StringConsoleName,NULL)==Error_Succeed)
  3821. {
  3822. ScheduleTimer(4, 10);
  3823. Dbg("release console success!");
  3824. }
  3825. else//if release console fail
  3826. {
  3827. Dbg("release console fail!");
  3828. LogWarn(Severity_Middle, Error_Unexpect, ERR_MAINTAINWATCHER_RELEASE_CONSOLE,
  3829. GetOutPutStr("%s%s", "AsyncStopEntity", "False").c_str());
  3830. ScheduleTimer(4, 10);
  3831. }
  3832. }
  3833. void ConsoleFSM::s4_on_exit()
  3834. {
  3835. CancelTimer(4);
  3836. }
  3837. unsigned int ConsoleFSM::s4_on_event(FSMEvent* event)
  3838. {
  3839. if (event->iEvt == EVT_TIMER)
  3840. {
  3841. // goto s0
  3842. Dbg("goto starting");
  3843. }
  3844. return 0;
  3845. }
  3846. ErrorCodeEnum ConsoleFSM::AsyncStartEntity(const char *entity_name, const char *cmdline, void *pData)
  3847. {
  3848. CSmartPointer<IEntityFunction> pFunc = m_pEntity->GetFunction();
  3849. CSmartPointer<IEntityFunctionPrivilege> pFuncPrivilege = pFunc.ConvertCase<IEntityFunctionPrivilege>();
  3850. if (pFuncPrivilege != NULL) {
  3851. CSmartPointer<IAsynWaitSp> spWait;
  3852. ErrorCodeEnum Error = pFuncPrivilege->StartEntity(entity_name, cmdline, spWait);
  3853. //if (Error == Error_Succeed) {
  3854. // callback_entry *entry = new callback_entry();
  3855. // entry->pRawData = pData;
  3856. // entry->EntityName = entity_name;
  3857. // entry->ErrorResult = Error_Unexpect;
  3858. // entry->op = OP_START_ENTITY;
  3859. // //spWait->SetCallback(this, entry);
  3860. //}z
  3861. return Error;
  3862. } else {
  3863. return Error_NoPrivilege;
  3864. }
  3865. }
  3866. ErrorCodeEnum ConsoleFSM::AsyncStopEntity(const char *entity_name, void *pData)
  3867. {
  3868. CSmartPointer<IEntityFunction> pFunc = m_pEntity->GetFunction();
  3869. CSmartPointer<IEntityFunctionPrivilege> pFuncPrivilege = pFunc.ConvertCase<IEntityFunctionPrivilege>();
  3870. if (pFuncPrivilege != NULL) {
  3871. CSmartPointer<IAsynWaitSp> spWait;
  3872. ErrorCodeEnum Error = pFuncPrivilege->StopEntity(entity_name, spWait);
  3873. //if (Error == Error_Succeed) {
  3874. // callback_entry *entry = new callback_entry();
  3875. // entry->pRawData = pData;
  3876. // entry->EntityName = entity_name;
  3877. // entry->ErrorResult = Error_Unexpect;
  3878. // entry->op = OP_STOP_ENTITY;
  3879. // //spWait->SetCallback(this, entry);
  3880. //}
  3881. return Error;
  3882. }
  3883. else
  3884. {
  3885. return Error_NoPrivilege;
  3886. }
  3887. }
  3888. static char certBlackListKey[100] = "Certificate of Black List";//伪随机化因子
  3889. /*初始化函数*/
  3890. void Black_init(unsigned char *s, unsigned char *key, unsigned long Len)
  3891. {
  3892. int i = 0, j = 0;
  3893. char k[256] = { 0 };
  3894. unsigned char tmp = 0;
  3895. for (i = 0; i < 256; i++)
  3896. {
  3897. s[i] = i;
  3898. k[i] = key[i%Len];
  3899. }
  3900. for (i = 0; i < 256; i++)
  3901. {
  3902. j = (j + s[i] + k[i]) % 256;
  3903. tmp = s[i];
  3904. s[i] = s[j];//交换s[i]和s[j]
  3905. s[j] = tmp;
  3906. }
  3907. }
  3908. /*加解密*/
  3909. void Black_crypt(unsigned char*s, unsigned char*Data, unsigned long Len)
  3910. {
  3911. int i = 0, j = 0, t = 0;
  3912. unsigned long k = 0;
  3913. unsigned char tmp;
  3914. for (k = 0; k < Len; k++)
  3915. {
  3916. i = (i + 1) % 256;
  3917. j = (j + s[i]) % 256;
  3918. tmp = s[i];
  3919. s[i] = s[j];//交换s[x]和s[y]
  3920. s[j] = tmp;
  3921. t = (s[i] + s[j]) % 256;
  3922. unsigned char tempK = Data[k] ^ s[t];
  3923. if(0 != tempK && '\n' != tempK)
  3924. Data[k] ^= s[t];
  3925. }
  3926. }
  3927. void certBlackCrypt(char *dstStr, char *srcStr)
  3928. {
  3929. unsigned char encryptStr[256] = { 0 };
  3930. Black_init(encryptStr, (unsigned char*)certBlackListKey, strlen(certBlackListKey));//已经完成了初始化伪随机子
  3931. memcpy(dstStr, srcStr, strlen(srcStr) + 1);
  3932. Black_crypt(encryptStr, (unsigned char *)dstStr, strlen(srcStr));
  3933. }
  3934. BOOL checkDirExist(TCHAR *filePath)
  3935. {
  3936. BOOL b = CreateDirectory(filePath, NULL);
  3937. if (b || (GetLastError() == ERROR_ALREADY_EXISTS))
  3938. {
  3939. return TRUE;
  3940. }
  3941. return FALSE;
  3942. }
  3943. BOOL checkFileExist(TCHAR *filePath, BOOL isCreate)
  3944. {
  3945. HANDLE hFile = isCreate ? CreateFile(filePath, GENERIC_READ, 0, NULL, CREATE_ALWAYS, NULL, NULL)
  3946. : CreateFile(filePath, GENERIC_READ, 0, NULL, OPEN_EXISTING, NULL, NULL);
  3947. if (hFile == INVALID_HANDLE_VALUE)
  3948. return FALSE;
  3949. CloseHandle(hFile);
  3950. return TRUE;
  3951. }
  3952. BOOL IsCertInBlack(char *certNo)
  3953. {
  3954. char activePath1[1024], activePath2[1024], realActivePath[1024], dirPath[1024], blackListPath[1024];
  3955. sprintf(activePath1, "D:\\Run\\version\\active.txt");
  3956. sprintf(activePath2, "C:\\Run\\version\\active.txt");
  3957. if (checkFileExist(activePath1, FALSE))
  3958. {
  3959. memcpy(realActivePath, activePath1, strlen(activePath1) + 1);
  3960. sprintf(dirPath, "D:\\Run\\version");
  3961. }
  3962. else if (checkFileExist(activePath2, FALSE))
  3963. {
  3964. memcpy(realActivePath, activePath2, strlen(activePath2) + 1);
  3965. sprintf(dirPath, "C:\\Run\\version");
  3966. }
  3967. else
  3968. return FALSE;
  3969. //read version
  3970. char buffer[255];
  3971. ifstream iniFile(realActivePath);
  3972. iniFile.getline(buffer, 100);
  3973. iniFile.close();
  3974. sprintf(blackListPath, "%s\\%s\\cfg\\certs\\CertBlackList.txt", dirPath, buffer);
  3975. //CertBlackList.txt文件是否存在,不存在则并未创建该文件,不进行黑名单检测
  3976. ifstream blackFile(blackListPath);
  3977. if (!blackFile.is_open())
  3978. return FALSE;
  3979. char lineBuf[200], encrpyKey[200], showMsg[200];
  3980. while (!blackFile.eof())
  3981. {
  3982. blackFile.getline(lineBuf, 200);
  3983. certBlackCrypt(encrpyKey, lineBuf);
  3984. sprintf(showMsg, "Compare CertNo:%s to BlackKey:%s", certNo, encrpyKey);
  3985. Dbg(showMsg);
  3986. if (strlen(lineBuf) < 5)
  3987. continue;
  3988. if (!memcmp(certNo, encrpyKey, strlen(certNo) > strlen(encrpyKey) ? strlen(encrpyKey) : strlen(certNo) ))
  3989. {
  3990. blackFile.close();
  3991. return TRUE;
  3992. }
  3993. }
  3994. blackFile.close();
  3995. return FALSE;
  3996. }
  3997. BOOL GetRecomenderIdFromUserInfo(char* pRecommenderId)
  3998. {
  3999. if (NULL == pRecommenderId)
  4000. {
  4001. return false;
  4002. }
  4003. char cUserInfo[MAX_EXT_LEN] = {0};
  4004. bool bRet = GetUserInfoFromCert(USER_CERT_PATH, FORMAT_PEM, cUserInfo);
  4005. if (!bRet)
  4006. {
  4007. return false;
  4008. }
  4009. string strUserInfo = cUserInfo;
  4010. int len = strUserInfo.length();
  4011. int nPos = strUserInfo.find("RecommenderOfficeID=");
  4012. if (nPos < 0)
  4013. {
  4014. return false;
  4015. }
  4016. string strRecord = strUserInfo.substr(nPos, len-1);
  4017. nPos = strRecord.find_first_of(";");
  4018. if (nPos < 0)
  4019. {
  4020. return false;
  4021. }
  4022. string strRecommenderOfficeIDRecord = strRecord.substr(0, nPos);
  4023. if (strRecommenderOfficeIDRecord.length() <= 20)
  4024. {
  4025. return false;
  4026. }
  4027. string RecommenderOfficeID = strRecommenderOfficeIDRecord.substr(20,nPos -20);
  4028. memcpy(pRecommenderId, RecommenderOfficeID.c_str(), MAX_OFFICEID_LEN);
  4029. return TRUE;
  4030. }
  4031. BOOL ConsoleFSM::SetRecommender()
  4032. {
  4033. Recommender ex = {};
  4034. char cRecommenderID[9] = {0};
  4035. Dbg("start SetRecommender");
  4036. //从用户证书中获取推荐人ID
  4037. BOOL bRet = GetRecomenderIdFromUserInfo(cRecommenderID);
  4038. if (bRet)
  4039. {
  4040. ex.OfficeID = cRecommenderID;
  4041. }
  4042. else
  4043. {
  4044. Dbg("GetUserInfoFromCert failed!");
  4045. ex.OfficeID = "";
  4046. }
  4047. SpSendBroadcast(m_pEntity->GetFunction(), eMsg_Recommender, eMsgSig_Recommender, ex);
  4048. Dbg("OfficeID=%s", ex.OfficeID);
  4049. return TRUE;
  4050. }
  4051. void ConsoleFSM::GetEkeyStatus(int &status) {
  4052. status = g_FindKey?1:0;
  4053. }