一款显示网速、系统利用的软件
選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

TrayS.cpp 152KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801
  1. // TrayS.cpp : 定义应用程序的入口点。
  2. //
  3. #ifdef _WIN64
  4. #pragma comment(linker,"/manifestdependency:\"type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='amd64' publicKeyToken='6595b64144ccf1df' language='*'\"")
  5. #else
  6. #pragma comment(linker,"/manifestdependency:\"type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='x86' publicKeyToken='6595b64144ccf1df' language='*'\"")
  7. #endif
  8. #include "framework.h"
  9. #include "TrayS.h"
  10. int DPI(int pixel)
  11. {
  12. return pixel * iDPI / 96;
  13. }
  14. #if 0
  15. typedef struct _IP_HEADER
  16. {
  17. BYTE bVerAndHLen;
  18. / *版本信息(前4位)和头长度(后4位)* /
  19. BYTE bTypeOfService; //服务类型
  20. USHORT nTotalLength; //数据包长度
  21. USHORT nID; //数据包标识
  22. USHORT nReserved; //保留字段
  23. BYTE bTTL; //生成时间
  24. BYTE bProtocol; //协议类型
  25. USHORT nCheckSum; //校验和
  26. UINT nSourIp; //源IP
  27. UINT nDestIp; //目的IP
  28. }
  29. IP_HEADER, * PIP_HEADER;
  30. typedef struct _TCP_HEADER
  31. {
  32. USHORT nSourPort;
  33. USHORT nDestPort;
  34. UINT nSequNum;
  35. UINT nAcknowledgeNum;
  36. USHORT nHLenAndFlag;
  37. USHORT nWindowSize;
  38. USHORT nCheckSum;
  39. USHORT nrgentPointer;
  40. }
  41. TCP_HEADER, * PTCP_HEADER;
  42. typedef struct _UDP_HEADER
  43. {
  44. USHORT nSourPort;
  45. USHORT nDestPort;
  46. USHORT nLength;
  47. USHORT nCheckSum;
  48. }
  49. UDP_HEADER, * PUDP_HEADER;
  50. typedef struct _PACK_INFO
  51. {
  52. USHORT nLength;
  53. USHORT nProtocol;
  54. UINT nSourIp;
  55. UINT nDestIp;
  56. USHORT nSourPort;
  57. USHORT nDestPort;
  58. }
  59. PACK_INFO, * LPPACK_INFO;
  60. void AnalyseTcp(DWORD dwPort, ULONG64 ul64Flow, bool IsRecv)
  61. {
  62. DWORD dwPid = 0;
  63. MIB_TCPTABLE_OWNER_PID* stcTcpTable = NULL;
  64. DWORD szTcpTableSize = 0;
  65. //获取TCP表大小
  66. GetExtendedTcpTable(stcTcpTable, &szTcpTableSize, FALSE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0);
  67. //分配内存
  68. stcTcpTable = (MIB_TCPTABLE_OWNER_PID*)malloc(szTcpTableSize);
  69. ZeroMemory(stcTcpTable, szTcpTableSize);
  70. //获取TCP表
  71. if (NO_ERROR != GetExtendedTcpTable(stcTcpTable, &szTcpTableSize, FALSE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0))
  72. {
  73. free(stcTcpTable);
  74. return;
  75. }
  76. for (DWORD i = 0;
  77. i < stcTcpTable->dwNumEntries;
  78. i++)
  79. {
  80. if (stcTcpTable->table[i].dwLocalPort == dwPort)
  81. dwPid = stcTcpTable->table[i].dwOwningPid;
  82. }
  83. free(stcTcpTable);
  84. }
  85. void AnalyseUdp(DWORD dwPort, ULONG64 ul64Flow, bool IsRecv)
  86. {
  87. DWORD dwPid = 0;
  88. MIB_UDPTABLE_OWNER_PID* stcUdpTable = NULL;
  89. DWORD szUdpTableSize = 0;
  90. //获取UDP表大小
  91. GetExtendedUdpTable(stcUdpTable, &szUdpTableSize, FALSE, AF_INET, UDP_TABLE_OWNER_PID, 0);
  92. //分配内存
  93. stcUdpTable = (MIB_UDPTABLE_OWNER_PID*)malloc(szUdpTableSize);
  94. ZeroMemory(stcUdpTable, szUdpTableSize);
  95. //获取UDP表
  96. if (NO_ERROR != GetExtendedUdpTable(stcUdpTable, &szUdpTableSize, FALSE, AF_INET, UDP_TABLE_OWNER_PID, 0))
  97. {
  98. //在实际测试中这个地方确实偶尔是会失败的,不过在频繁的更新中一两次失败无关紧要
  99. free(stcUdpTable);
  100. return;
  101. }
  102. for (DWORD i = 0;
  103. i < stcUdpTable->dwNumEntries;
  104. i++)
  105. {
  106. if (stcUdpTable->table[i].dwLocalPort == dwPort)
  107. dwPid = stcUdpTable->table[i].dwOwningPid;
  108. //此处已经找到进程对应的pid了,同时包的大小和是上传还是下载已经通过参数传进来了。
  109. }
  110. free(stcUdpTable);
  111. //对数据的处理
  112. //...
  113. }
  114. void Thread()
  115. {
  116. PACK_INFO PackInfo =
  117. {
  118. 0
  119. };
  120. int nRecvSize = 0;
  121. char szPackBuf[DEF_BUF_SIZE] =
  122. {
  123. 0
  124. };
  125. WSADATA wsaData;
  126. int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
  127. if (iResult != NO_ERROR)
  128. return;
  129. // 获取本地地址信息
  130. sockaddr_in LocalAddr;
  131. char szLocalName[DEF_BUF_SIZE] =
  132. {
  133. 0
  134. };
  135. gethostname(szLocalName, DEF_BUF_SIZE);
  136. hostent* pHost = gethostbyname(szLocalName);
  137. if (pHost != NULL)
  138. {
  139. LocalAddr.sin_family = AF_INET;
  140. LocalAddr.sin_port = htons(0);
  141. memcpy(&(LocalAddr.sin_addr.s_addr), pHost->h_addr_list[0], pHost->h_length);
  142. }
  143. else
  144. return;
  145. // 创建监听套接字
  146. SOCKET MonSock = socket(AF_INET, SOCK_RAW, IPPROTO_IP);
  147. if (MonSock == INVALID_SOCKET)
  148. return;
  149. // 绑定地址信息到套接字
  150. if (bind(MonSock, (sockaddr*)&LocalAddr, sizeof(sockaddr)) == SOCKET_ERROR)
  151. return;
  152. // 设置为混杂模式,收所有IP包
  153. DWORD dwValue = 1;
  154. if (ioctlsocket(MonSock, SIO_RCVALL, &dwValue) != 0)
  155. return;
  156. while (1)
  157. {
  158. // 取得数据包
  159. nRecvSize = recv(MonSock, szPackBuf, DEF_BUF_SIZE, 0);
  160. if (nRecvSize > 0)
  161. {
  162. // 解析IP包头
  163. PIP_HEADER pIpHeader = (PIP_HEADER)szPackBuf;
  164. PackInfo.nLength = nRecvSize;
  165. PackInfo.nProtocol = (USHORT)pIpHeader->bProtocol;
  166. PackInfo.nSourIp = pIpHeader->nSourIp;
  167. PackInfo.nDestIp = pIpHeader->nDestIp;
  168. UINT nIpHeadLength = (pIpHeader->bVerAndHLen & 0x0F) * sizeof(UINT); // IP数据包头长度
  169. // 只检测TCP和UDP包
  170. switch (pIpHeader->bProtocol)
  171. {
  172. case IPPROTO_TCP:
  173. {
  174. // 取得TCP数据包端口号
  175. PTCP_HEADER pTcpHeader = (PTCP_HEADER)&szPackBuf[nIpHeadLength];
  176. PackInfo.nSourPort = pTcpHeader->nSourPort;
  177. PackInfo.nDestPort = pTcpHeader->nDestPort;
  178. //判断上传还是下载
  179. if (PackInfo.nSourIp == LocalAddr.sin_addr.S_un.S_addr)
  180. AnalyseTcp(PackInfo.nSourPort, PackInfo.nLength, FALSE);
  181. else
  182. AnalyseTcp(PackInfo.nDestPort, PackInfo.nLength, TRUE);
  183. }
  184. break;
  185. case IPPROTO_UDP:
  186. {
  187. // 取得UDP数据包端口号
  188. PUDP_HEADER pUdpHeader = (PUDP_HEADER)&szPackBuf[nIpHeadLength];
  189. PackInfo.nSourPort = pUdpHeader->nSourPort;
  190. PackInfo.nDestPort = pUdpHeader->nDestPort;
  191. if (PackInfo.nSourIp == LocalAddr.sin_addr.S_un.S_addr)
  192. AnalyseUdp(PackInfo.nSourPort, PackInfo.nLength, FALSE);
  193. else
  194. AnalyseUdp(PackInfo.nDestPort, PackInfo.nLength, TRUE);
  195. }
  196. break;
  197. }
  198. }
  199. }
  200. }
  201. #endif
  202. BOOL CALLBACK FindWindowFunc(HWND hWnd, LPARAM lpAram)
  203. {
  204. WCHAR szText[16];
  205. GetWindowText(hWnd, szText, 16);
  206. if (lstrcmp(szText, L"_TrayS_") == 0)
  207. {
  208. SendMessage(hWnd, WM_TRAYS, 0, 0);
  209. return FALSE;
  210. }
  211. return TRUE;
  212. }
  213. BOOL CALLBACK IsZoomedFunc(HWND hWnd, LPARAM lpAram)
  214. {
  215. if (::IsWindowVisible(hWnd) && IsZoomed(hWnd))
  216. {
  217. if (MonitorFromWindow(hWnd, MONITOR_DEFAULTTONEAREST) == (HMONITOR)lpAram)
  218. {
  219. BOOL Attribute = FALSE;
  220. if (pDwmGetWindowAttribute)
  221. pDwmGetWindowAttribute(hWnd, 14, &Attribute, sizeof(BOOL));
  222. if (Attribute == FALSE)
  223. {
  224. iWindowMode = 1;
  225. return FALSE;
  226. }
  227. }
  228. }
  229. return TRUE;
  230. }
  231. #if 0
  232. BOOL CreateProcessByExplorer(LPCWSTR process, LPCWSTR szDir, LPCWSTR cmd)
  233. {
  234. BOOL ret = FALSE;
  235. HANDLE hProcess = 0, hToken = 0, hDuplicatedToken = 0;
  236. LPVOID lpEnv = NULL;
  237. do
  238. {
  239. HWND hTrayWnd = ::FindWindow(szShellTray, NULL);
  240. DWORD explorerPid;
  241. GetWindowThreadProcessId(hTrayWnd, &explorerPid); // 获取explorer进程号,自行实现
  242. if (explorerPid == NULL)
  243. break;
  244. hProcess = OpenProcess(PROCESS_QUERY_INFORMATION, TRUE, explorerPid);
  245. if (INVALID_HANDLE_VALUE == hProcess)
  246. break;
  247. if (!OpenProcessToken(hProcess, TOKEN_ALL_ACCESS, &hToken))
  248. break;
  249. DuplicateTokenEx(hToken, MAXIMUM_ALLOWED, NULL, SecurityIdentification, TokenPrimary, &hDuplicatedToken);
  250. CreateEnvironmentBlock(&lpEnv, hDuplicatedToken, FALSE);
  251. / *
  252. WCHAR szDir[MAX_PATH] = L"\"";
  253. wcscpy_s(&szDir[2], MAX_PATH, process);
  254. int iLen = wcslen(szDir);
  255. if (NULL != cmd)
  256. {
  257. wcscpy_s(&szDir[iLen], MAX_PATH, L"\" \"");
  258. iLen = wcslen(szDir);
  259. wcscpy_s(&szDir[iLen], MAX_PATH, cmd);
  260. }
  261. iLen = wcslen(szDir);
  262. wcscpy_s(&szDir[iLen], MAX_PATH, L"\"");
  263. * /
  264. STARTUPINFO si =
  265. {
  266. 0
  267. };
  268. PROCESS_INFORMATION pi =
  269. {
  270. 0
  271. };
  272. si.cb = sizeof(STARTUPINFO);
  273. si.lpDesktop = (LPWSTR)L"winsta0\\default";
  274. si.dwFlags = STARTF_USESHOWWINDOW;
  275. si.wShowWindow = SW_HIDE;
  276. if (!CreateProcessAsUser(hToken, process, NULL / *const_cast<LPWSTR>(szDir) * / , 0, 0, FALSE, CREATE_UNICODE_ENVIRONMENT, lpEnv, NULL, &si, &pi))
  277. break;
  278. ret = TRUE;
  279. } while (0);
  280. if (INVALID_HANDLE_VALUE != hProcess)
  281. CloseHandle(hProcess);
  282. if (INVALID_HANDLE_VALUE != hToken)
  283. CloseHandle(hToken);
  284. if (INVALID_HANDLE_VALUE != hDuplicatedToken)
  285. CloseHandle(hDuplicatedToken);
  286. if (NULL != lpEnv)
  287. DestroyEnvironmentBlock(lpEnv);
  288. return ret;
  289. }
  290. #endif
  291. DWORD iCPU;
  292. /////////////////////////////////////////////////////////////////////////////CPU占用率
  293. FILETIME pre_idleTime;
  294. FILETIME pre_kernelTime;
  295. FILETIME pre_userTime;
  296. __int64 CompareFileTime(FILETIME time1, FILETIME time2)
  297. {
  298. __int64 a = time1.dwHighDateTime;
  299. a = a << 32 | time1.dwLowDateTime;
  300. __int64 b = time2.dwHighDateTime;
  301. b = b << 32 | time2.dwLowDateTime;
  302. return (b - a);
  303. }
  304. typedef struct _PDH_RAW_COUNTER
  305. {
  306. volatile DWORD CStatus;
  307. FILETIME TimeStamp;
  308. LONGLONG FirstValue;
  309. LONGLONG SecondValue;
  310. DWORD MultiCount;
  311. } PDH_RAW_COUNTER, *PPDH_RAW_COUNTER;
  312. PDH_RAW_COUNTER m_last_rawData;
  313. BOOL m_first_get_CPU_utility = TRUE;
  314. int GetCPUUseRate()
  315. {
  316. if (TraySave.bMonitorPDH)
  317. {
  318. #define PDH_FMT_RAW ((DWORD) 0x00000010)
  319. #define PDH_FMT_ANSI ((DWORD) 0x00000020)
  320. #define PDH_FMT_UNICODE ((DWORD) 0x00000040)
  321. #define PDH_FMT_LONG ((DWORD) 0x00000100)
  322. #define PDH_FMT_DOUBLE ((DWORD) 0x00000200)
  323. #define PDH_FMT_LARGE ((DWORD) 0x00000400)
  324. #define PDH_FMT_NOSCALE ((DWORD) 0x00001000)
  325. #define PDH_FMT_1000 ((DWORD) 0x00002000)
  326. #define PDH_FMT_NODATA ((DWORD) 0x00004000)
  327. #define PDH_FMT_NOCAP100 ((DWORD) 0x00008000)
  328. #define PERF_DETAIL_COSTLY ((DWORD) 0x00010000)
  329. #define PERF_DETAIL_STANDARD ((DWORD) 0x0000FFFF)
  330. typedef HANDLE PDH_HCOUNTER;
  331. typedef HANDLE PDH_HQUERY;
  332. typedef HANDLE PDH_HLOG;
  333. typedef PDH_HCOUNTER HCOUNTER;
  334. typedef PDH_HQUERY HQUERY;
  335. typedef struct _PDH_FMT_COUNTERVALUE
  336. {
  337. DWORD CStatus;
  338. union
  339. {
  340. LONG longValue;
  341. double doubleValue;
  342. LONGLONG largeValue;
  343. LPCSTR AnsiStringValue;
  344. LPCWSTR WideStringValue;
  345. };
  346. }
  347. PDH_FMT_COUNTERVALUE, *PPDH_FMT_COUNTERVALUE;
  348. //获取CPU使用率
  349. HQUERY hQuery;
  350. HCOUNTER hCounter;
  351. DWORD counterType;
  352. PDH_RAW_COUNTER rawData;
  353. typedef ULONG(WINAPI *pfnPdhOpenQuery)(_In_opt_ LPCWSTR szDataSource, _In_ DWORD_PTR dwUserData, _Out_ PDH_HQUERY *phQuery);
  354. typedef ULONG(WINAPI *pfnPdhAddCounter)(_In_ PDH_HQUERY hQuery, _In_ LPCWSTR szFullCounterPath, _In_ DWORD_PTR dwUserData, _Out_ PDH_HCOUNTER *phCounter);
  355. typedef ULONG(WINAPI *pfnPdhCollectQueryData)(PDH_HQUERY hQuery);
  356. typedef ULONG(WINAPI *pfnPdhGetRawCounterValue)(PDH_HCOUNTER hCounter, LPDWORD lpdwType, PPDH_RAW_COUNTER pValue);
  357. typedef ULONG(WINAPI *pfnPdhCalculateCounterFromRawValue)(PDH_HCOUNTER hCounter, DWORD dwFormat, PPDH_RAW_COUNTER rawValue1, PPDH_RAW_COUNTER rawValue2, PPDH_FMT_COUNTERVALUE fmtValue);
  358. typedef ULONG(WINAPI *pfnPdhCloseQuery)(PDH_HQUERY hQuery);
  359. if (hPDH == NULL)
  360. hPDH = LoadLibrary(L"pdh.dll");
  361. if (hPDH)
  362. {
  363. pfnPdhOpenQuery PdhOpenQuery = (pfnPdhOpenQuery)GetProcAddress(hPDH, "PdhOpenQueryW");
  364. pfnPdhAddCounter PdhAddCounter = (pfnPdhAddCounter)GetProcAddress(hPDH, "PdhAddCounterW");
  365. pfnPdhCollectQueryData PdhCollectQueryData = (pfnPdhCollectQueryData)GetProcAddress(hPDH, "PdhCollectQueryData");
  366. pfnPdhGetRawCounterValue PdhGetRawCounterValue = (pfnPdhGetRawCounterValue)GetProcAddress(hPDH, "PdhGetRawCounterValue");
  367. pfnPdhCalculateCounterFromRawValue PdhCalculateCounterFromRawValue = (pfnPdhCalculateCounterFromRawValue)GetProcAddress(hPDH, "PdhCalculateCounterFromRawValue");
  368. pfnPdhCloseQuery PdhCloseQuery = (pfnPdhCloseQuery)GetProcAddress(hPDH, "PdhCloseQuery");
  369. if (PdhCloseQuery != NULL && PdhAddCounter != NULL && PdhCollectQueryData != NULL && PdhGetRawCounterValue != NULL && PdhCalculateCounterFromRawValue != NULL && PdhCloseQuery != NULL)
  370. {
  371. PdhOpenQuery(NULL, 0, &hQuery); //开始查询
  372. const wchar_t *query_str{
  373. };
  374. query_str = L"\\Processor Information(_Total)\\% Processor Utility";
  375. PdhAddCounter(hQuery, query_str, NULL, &hCounter);
  376. PdhCollectQueryData(hQuery);
  377. PdhGetRawCounterValue(hCounter, &counterType, &rawData);
  378. PDH_FMT_COUNTERVALUE fmtValue;
  379. PdhCalculateCounterFromRawValue(hCounter, PDH_FMT_DOUBLE, &rawData, &m_last_rawData, &fmtValue); //计算使用率
  380. iCPU = (int)fmtValue.doubleValue; //传出数据
  381. if (iCPU > 100)
  382. iCPU = 100;
  383. m_last_rawData = rawData; //保存上一次数据
  384. PdhCloseQuery(hQuery); //关闭查询
  385. }
  386. }
  387. return iCPU;
  388. }
  389. else
  390. {
  391. if (hPDH)
  392. {
  393. FreeLibrary(hPDH);
  394. hPDH = NULL;
  395. }
  396. int nCPUUseRate = -1;
  397. FILETIME idleTime; //空闲时间
  398. FILETIME kernelTime; //核心态时间
  399. FILETIME userTime; //用户态时间
  400. GetSystemTimes(&idleTime, &kernelTime, &userTime);
  401. __int64 idle = CompareFileTime(pre_idleTime, idleTime);
  402. __int64 kernel = CompareFileTime(pre_kernelTime, kernelTime);
  403. __int64 user = CompareFileTime(pre_userTime, userTime);
  404. if (kernel + user != 0)
  405. nCPUUseRate = (int)((kernel + user - idle) * 100 / (kernel + user));
  406. pre_idleTime = idleTime;
  407. pre_kernelTime = kernelTime;
  408. pre_userTime = userTime;
  409. if (nCPUUseRate < 1)
  410. nCPUUseRate = iCPU;
  411. else if (nCPUUseRate > 100)
  412. nCPUUseRate = 100;
  413. return nCPUUseRate;
  414. }
  415. }
  416. void ReadReg() //读取设置
  417. {
  418. SetToCurrentPath();
  419. HANDLE hFile = CreateFile(szTraySave, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_ARCHIVE, NULL);
  420. if (hFile)
  421. {
  422. DWORD dwBytes;
  423. ReadFile(hFile, &TraySave, sizeof TraySave, &dwBytes, NULL);
  424. CloseHandle(hFile);
  425. }
  426. #if 0
  427. HKEY pKey;
  428. if(IsUserAdmin())
  429. RegOpenKeyEx(HKEY_LOCAL_MACHINE, szSubKey, NULL, KEY_ALL_ACCESS, &pKey);
  430. else
  431. RegOpenKeyEx(HKEY_CURRENT_USER, szSubKey, NULL, KEY_ALL_ACCESS, &pKey);
  432. if (pKey)
  433. {
  434. DWORD dType = REG_BINARY;
  435. DWORD cbData = sizeof(aMode);
  436. RegQueryValueEx(pKey, szMode, NULL, &dType, (BYTE*)aMode, &cbData);
  437. dType = REG_BINARY;
  438. cbData = sizeof(dAlphaColor);
  439. RegQueryValueEx(pKey, szAlphaColor, NULL, &dType, (BYTE*)dAlphaColor, &cbData);
  440. dType = REG_BINARY;
  441. cbData = sizeof(bAlpha);
  442. RegQueryValueEx(pKey, szAlpha, NULL, &dType, (BYTE*)bAlpha, &cbData);
  443. dType = REG_DWORD;
  444. cbData = sizeof(DWORD);
  445. RegQueryValueEx(pKey, szPos, NULL, &dType, (BYTE*)&iPos, &cbData);
  446. dType = REG_DWORD;
  447. cbData = sizeof(DWORD);
  448. RegQueryValueEx(pKey, szUnit, NULL, &dType, (BYTE*)&iUnit, &cbData);
  449. dType = REG_DWORD;
  450. cbData = sizeof(DWORD);
  451. RegQueryValueEx(pKey, szTrayIcon, NULL, &dType, (BYTE*)&bTrayIcon, &cbData);
  452. dType = REG_DWORD;
  453. cbData = sizeof(DWORD);
  454. RegQueryValueEx(pKey, szMonitor, NULL, &dType, (BYTE*)&bMonitor, &cbData);
  455. dType = REG_DWORD;
  456. cbData = sizeof(DWORD);
  457. RegQueryValueEx(pKey, szMonitorLeft, NULL, &dType, (BYTE*)&bMonitorLeft, &cbData);
  458. dType = REG_DWORD;
  459. cbData = sizeof(DWORD);
  460. RegQueryValueEx(pKey, szMonitorFloat, NULL, &dType, (BYTE*)&bMonitorFloat, &cbData);
  461. dType = REG_DWORD;
  462. cbData = sizeof(DWORD);
  463. RegQueryValueEx(pKey, szMonitorTransparent, NULL, &dType, (BYTE*)&bMonitorTransparent, &cbData);
  464. dType = REG_BINARY;
  465. cbData = sizeof(dMonitorPoint);
  466. RegQueryValueEx(pKey, szMonitorPoint, NULL, &dType, (BYTE*)&dMonitorPoint, &cbData);
  467. dType = REG_DWORD;
  468. cbData = sizeof(DWORD);
  469. RegQueryValueEx(pKey, szMonitorTraffic, NULL, &dType, (BYTE*)&bMonitorTraffic, &cbData);
  470. dType = REG_DWORD;
  471. cbData = sizeof(DWORD);
  472. RegQueryValueEx(pKey, szMonitorTemperature, NULL, &dType, (BYTE*)&bMonitorTemperature, &cbData);
  473. dType = REG_DWORD;
  474. cbData = sizeof(DWORD);
  475. RegQueryValueEx(pKey, szMonitorUsage, NULL, &dType, (BYTE*)&bMonitorUsage, &cbData);
  476. dType = REG_DWORD;
  477. cbData = sizeof(DWORD);
  478. RegQueryValueEx(pKey, szSound, NULL, &dType, (BYTE*)&bSound, &cbData);
  479. dType = REG_DWORD;
  480. cbData = sizeof(DWORD);
  481. RegQueryValueEx(pKey, szMonitorPDH, NULL, &dType, (BYTE*)&bMonitorPDH, &cbData);
  482. dType = REG_DWORD;
  483. cbData = sizeof(DWORD);
  484. RegQueryValueEx(pKey, szMonitorSimple, NULL, &dType, (BYTE*)&bMonitorSimple, &cbData);
  485. dType = REG_BINARY;
  486. cbData = sizeof(cMonitorColor);
  487. RegQueryValueEx(pKey, szMonitorColor, NULL, &dType, (BYTE*)cMonitorColor, &cbData);
  488. dType = REG_BINARY;
  489. cbData = sizeof(dNumValues);
  490. RegQueryValueEx(pKey, szNumValues, NULL, &dType, (BYTE*)dNumValues, &cbData);
  491. dType = REG_BINARY;
  492. cbData = 38;
  493. RegQueryValueEx(pKey, szAdapterName, NULL, &dType, (BYTE*)AdpterName, &cbData);
  494. RegCloseKey(pKey);
  495. }
  496. #endif
  497. }
  498. void WriteReg() //写入设置
  499. {
  500. SetToCurrentPath();
  501. HANDLE hFile = CreateFile(szTraySave, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_ARCHIVE, NULL);
  502. if (hFile)
  503. {
  504. DWORD dwBytes;
  505. WriteFile(hFile, &TraySave, sizeof TraySave, &dwBytes, NULL);
  506. CloseHandle(hFile);
  507. }
  508. #if 0
  509. HKEY pKey;
  510. if (IsUserAdmin())
  511. {
  512. RegCreateKey(HKEY_LOCAL_MACHINE, szSubKey, &pKey);
  513. RegCloseKey(pKey);
  514. RegOpenKeyEx(HKEY_LOCAL_MACHINE, szSubKey, NULL, KEY_ALL_ACCESS, &pKey);
  515. }
  516. else
  517. {
  518. RegCreateKey(HKEY_CURRENT_USER, szSubKey, &pKey);
  519. RegCloseKey(pKey);
  520. RegOpenKeyEx(HKEY_CURRENT_USER, szSubKey, NULL, KEY_ALL_ACCESS, &pKey);
  521. }
  522. if (pKey)
  523. {
  524. RegSetValueEx(pKey, szMode, NULL, REG_BINARY, (BYTE*)aMode, sizeof(aMode));
  525. RegSetValueEx(pKey, szAlphaColor, NULL, REG_BINARY, (BYTE*)dAlphaColor, sizeof(dAlphaColor));
  526. RegSetValueEx(pKey, szAlpha, NULL, REG_BINARY, (BYTE*)bAlpha, sizeof(bAlpha));
  527. RegSetValueEx(pKey, szPos, NULL, REG_DWORD, (BYTE*)&iPos, sizeof(iPos));
  528. RegSetValueEx(pKey, szUnit, NULL, REG_DWORD, (BYTE*)&iUnit, sizeof(iUnit));
  529. RegSetValueEx(pKey, szTrayIcon, NULL, REG_DWORD, (BYTE*)&bTrayIcon, sizeof(bTrayIcon));
  530. RegSetValueEx(pKey, szMonitor, NULL, REG_DWORD, (BYTE*)&bMonitor, sizeof(bMonitor));
  531. RegSetValueEx(pKey, szMonitorLeft, NULL, REG_DWORD, (BYTE*)&bMonitorLeft, sizeof(bMonitorLeft));
  532. RegSetValueEx(pKey, szMonitorFloat, NULL, REG_DWORD, (BYTE*)&bMonitorFloat, sizeof(bMonitorFloat));
  533. RegSetValueEx(pKey, szMonitorTransparent, NULL, REG_DWORD, (BYTE*)&bMonitorTransparent, sizeof(bMonitorTransparent));
  534. RegSetValueEx(pKey, szMonitorPoint, NULL, REG_BINARY, (BYTE*)&dMonitorPoint, sizeof(dMonitorPoint));
  535. RegSetValueEx(pKey, szMonitorTraffic, NULL, REG_DWORD, (BYTE*)&bMonitorTraffic, sizeof(bMonitorTraffic));
  536. RegSetValueEx(pKey, szMonitorTemperature, NULL, REG_DWORD, (BYTE*)&bMonitorTemperature, sizeof(bMonitorTemperature));
  537. RegSetValueEx(pKey, szMonitorUsage, NULL, REG_DWORD, (BYTE*)&bMonitorUsage, sizeof(bMonitorUsage));
  538. RegSetValueEx(pKey, szMonitorPDH, NULL, REG_DWORD, (BYTE*)&bMonitorPDH, sizeof(bMonitorPDH));
  539. RegSetValueEx(pKey, szMonitorSimple, NULL, REG_DWORD, (BYTE*)&bMonitorSimple, sizeof(bMonitorSimple));
  540. RegSetValueEx(pKey, szSound, NULL, REG_DWORD, (BYTE*)&bSound, sizeof(bSound));
  541. RegSetValueEx(pKey, szMonitorColor, NULL, REG_BINARY, (BYTE*)cMonitorColor, sizeof(cMonitorColor));
  542. RegSetValueEx(pKey, szNumValues, NULL, REG_BINARY, (BYTE*)dNumValues, sizeof(dNumValues));
  543. RegSetValueEx(pKey, szAdapterName, NULL, REG_BINARY, (BYTE*)AdpterName, 38);
  544. RegCloseKey(pKey);
  545. }
  546. #endif
  547. }
  548. void GetShellAllWnd()
  549. {
  550. while (IsWindow(hTray) == FALSE)
  551. {
  552. hTray = FindWindow(szShellTray, NULL);
  553. if (hTray == NULL)
  554. Sleep(100);
  555. }
  556. while (IsWindow(hReBarWnd) == FALSE)
  557. {
  558. hReBarWnd = FindWindowEx(hTray, 0, L"ReBarWindow32", NULL);
  559. if (hReBarWnd == NULL)
  560. Sleep(100);
  561. }
  562. if (IsWindow(hTaskWnd) == FALSE)
  563. hTaskWnd = FindWindowEx(hReBarWnd, NULL, L"MSTaskSwWClass", NULL);
  564. if (IsWindow(hTaskListWnd) == FALSE)
  565. hTaskListWnd = FindWindowEx(hTaskWnd, NULL, L"MSTaskListWClass", NULL);
  566. }
  567. void CloseTaskBar()
  568. {
  569. if (IsWindow(hTaskBar))
  570. DestroyWindow(hTaskBar);
  571. if (IsWindow(hTaskTips))
  572. DestroyWindow(hTaskTips);
  573. }
  574. void OpenTaskBar()
  575. {
  576. if (IsWindow(hTaskBar) == FALSE)
  577. {
  578. hTaskBar = ::CreateDialog(hInst, MAKEINTRESOURCE(IDD_TASKBAR), NULL, (DLGPROC)TaskBarProc);
  579. if (hTaskBar)
  580. {
  581. GetShellAllWnd();
  582. if (TraySave.bMonitorFloat)
  583. {
  584. LONG exStyle = WS_EX_LAYERED | WS_EX_TOPMOST;
  585. if (TraySave.bMonitorTransparent)
  586. exStyle |= WS_EX_TRANSPARENT;
  587. SetWindowLongPtr(hTaskBar, GWL_EXSTYLE, GetWindowLongPtr(hTaskBar, GWL_EXSTYLE) | exStyle);
  588. SetLayeredWindowAttributes(hTaskBar, RGB(0, 0, 1), 198, LWA_ALPHA | LWA_COLORKEY);
  589. SetParent(hTaskBar, NULL);
  590. }
  591. else
  592. SetParent(hTaskBar, hReBarWnd);
  593. SetWH();
  594. #if 0
  595. if (rovi.dwMajorVersion < 10)
  596. {
  597. SetWindowLongPtr(hTaskBar, GWL_EXSTYLE, GetWindowLongPtr(hTaskBar, GWL_EXSTYLE) | WS_EX_LAYERED);
  598. SetLayeredWindowAttributes(hTaskBar, RGB(128, 128, 129), 0, LWA_COLORKEY);
  599. }
  600. #endif
  601. ShowWindow(hTaskBar, SW_SHOW);
  602. SetTimer(hTaskBar, 3, 1000, NULL);
  603. // SetTimer(hTaskBar, 6, 100, NULL);
  604. }
  605. }
  606. else
  607. {
  608. if (TraySave.bMonitorTransparent)
  609. SetWindowLongPtr(hTaskBar, GWL_EXSTYLE, GetWindowLongPtr(hTaskBar, GWL_EXSTYLE) | WS_EX_TRANSPARENT);
  610. else
  611. SetWindowLongPtr(hTaskBar, GWL_EXSTYLE, GetWindowLongPtr(hTaskBar, GWL_EXSTYLE) & ~WS_EX_TRANSPARENT);
  612. }
  613. }
  614. ////////////////////////////////////////////获取CPU温度
  615. #define MISC_CONTROL_3 (0x3 + ((0x18) << 3))
  616. int GetCpuTemp(DWORD Core)
  617. {
  618. if (bRing0)
  619. {
  620. SetThreadAffinityMask(GetCurrentThread(), Core);
  621. DWORD eax = 0, ebx, ecx, edx;
  622. if (!bIntel)
  623. {
  624. Cpuid(1, &eax, &ebx, &ecx, &edx);
  625. int family = ((eax >> 20) & 0xFF) + ((eax >> 8) & 0xF);
  626. if (family > 0xf)
  627. {
  628. // DWORD pciDevAddr = FindPciDeviceById(0x1022, 0x1203, 0);
  629. DWORD miscReg;
  630. ReadPciConfigDwordEx(MISC_CONTROL_3, 0xa4, &miscReg);
  631. return (miscReg >> 21) >> 3;
  632. }
  633. else
  634. {
  635. // DWORD pciDevAddr = FindPciDeviceById(0x1022, 0x1103, 0);
  636. DWORD miscReg;
  637. ReadPciConfigDwordEx(MISC_CONTROL_3, 0xe4, &miscReg);
  638. return ((miscReg & 0xFF0000) >> 16) - 49;
  639. // return (miscReg >> 16) & 0xFF;
  640. }
  641. }
  642. else
  643. {
  644. DWORD IAcore;
  645. int Tjunction = 100;
  646. Rdmsr(0x1A2, &eax, &edx);
  647. if (eax & 0x20000000)
  648. Tjunction = 85;
  649. Rdmsr(0x19C, &eax, &edx);
  650. IAcore = eax;
  651. IAcore &= 0xFF0000;
  652. IAcore = IAcore >> 16;
  653. return Tjunction - IAcore;
  654. }
  655. }
  656. return 0;
  657. }
  658. //////////////////////////////////////////////////载入温度DLL
  659. void LoadTemperatureDLL()
  660. {
  661. if (!InitOpenLibSys(&m_hOpenLibSys))
  662. bRing0 = FALSE;
  663. else
  664. {
  665. bRing0 = TRUE;
  666. DWORD eax, ebx, ecx, edx;
  667. Cpuid(0, &eax, &ebx, &ecx, &edx);
  668. bIntel = TRUE;
  669. if (ebx == 0x68747541)
  670. {
  671. bIntel = FALSE;
  672. }
  673. }
  674. #ifdef _WIN64
  675. hNVDLL = LoadLibrary(L"nvapi64.dll");
  676. #else
  677. hNVDLL = LoadLibrary(L"nvapi.dll");
  678. #endif
  679. if (hNVDLL)
  680. {
  681. NvAPI_QueryInterface = (NvAPI_QueryInterface_t)GetProcAddress(hNVDLL, "nvapi_QueryInterface");
  682. if (NvAPI_QueryInterface)
  683. {
  684. NvAPI_Initialize_t NvAPI_Initialize = (NvAPI_Initialize_t)NvAPI_QueryInterface(ID_NvAPI_Initialize);
  685. NvAPI_EnumPhysicalGPUs_t NvAPI_EnumPhysicalGPUs = (NvAPI_EnumPhysicalGPUs_t)NvAPI_QueryInterface(ID_NvAPI_EnumPhysicalGPUs);
  686. NvAPI_GPU_GetThermalSettings = (NvAPI_GPU_GetThermalSettings_t)NvAPI_QueryInterface(ID_NvAPI_GPU_GetThermalSettings);
  687. if (NvAPI_Initialize != NULL && NvAPI_EnumPhysicalGPUs != NULL && NvAPI_GPU_GetThermalSettings != NULL)
  688. {
  689. if (NvAPI_Initialize() == 0)
  690. {
  691. for (NvU32 PhysicalGpuIndex = 0; PhysicalGpuIndex < 4; PhysicalGpuIndex++)
  692. {
  693. hPhysicalGpu[PhysicalGpuIndex] = 0;
  694. }
  695. int physicalGpuCount;
  696. NvAPI_EnumPhysicalGPUs(hPhysicalGpu, &physicalGpuCount);
  697. }
  698. else
  699. {
  700. FreeLibrary(hNVDLL);
  701. hNVDLL = NULL;
  702. }
  703. }
  704. else
  705. {
  706. FreeLibrary(hNVDLL);
  707. hNVDLL = NULL;
  708. }
  709. }
  710. else
  711. {
  712. FreeLibrary(hNVDLL);
  713. hNVDLL = NULL;
  714. }
  715. }
  716. #ifdef _WIN64
  717. hATIDLL = LoadLibrary(L"atiadlxx.dll");
  718. #else
  719. hATIDLL = LoadLibrary(L"atiadlxy.dll");
  720. #endif
  721. if (hATIDLL)
  722. {
  723. ADL_Main_Control_Create = (ADL_MAIN_CONTROL_CREATE)GetProcAddress(hATIDLL, "ADL_Main_Control_Create");
  724. ADL_Main_Control_Destroy = (ADL_MAIN_CONTROL_DESTROY)GetProcAddress(hATIDLL, "ADL_Main_Control_Destroy");
  725. ADL_Overdrive5_Temperature_Get = (ADL_OVERDRIVE5_TEMPERATURE_GET)GetProcAddress(hATIDLL, "ADL_Overdrive5_Temperature_Get");
  726. if (NULL != ADL_Main_Control_Create &&
  727. NULL != ADL_Main_Control_Destroy
  728. )
  729. {
  730. if (ADL_OK != ADL_Main_Control_Create(ADL_Main_Memory_Alloc, 1))
  731. {
  732. FreeLibrary(hATIDLL);
  733. hATIDLL = NULL;
  734. }
  735. }
  736. else
  737. {
  738. FreeLibrary(hATIDLL);
  739. hATIDLL = NULL;
  740. }
  741. }
  742. }
  743. ///////////////////////////////////释放温度DLL
  744. void FreeTemperatureDLL()
  745. {
  746. if (hATIDLL)
  747. {
  748. ADL_Main_Control_Destroy();
  749. FreeLibrary(hATIDLL);
  750. hATIDLL = NULL;
  751. }
  752. if (hNVDLL)
  753. {
  754. FreeLibrary(hNVDLL);
  755. hNVDLL = NULL;
  756. }
  757. if (m_hOpenLibSys)
  758. DeinitOpenLibSys(&m_hOpenLibSys);
  759. m_hOpenLibSys = NULL;
  760. }
  761. ///////////////////////////////////////////////打开读取设置
  762. void OpenSetting()
  763. {
  764. if (IsWindow(hSetting))
  765. {
  766. SetForegroundWindow(hSetting);
  767. return;
  768. }
  769. hSetting = ::CreateDialog(hInst, MAKEINTRESOURCE(IDD_SETTING), NULL, (DLGPROC)SettingProc);
  770. if (!hSetting)
  771. {
  772. return;
  773. }
  774. SendMessage(hSetting, WM_SETICON, ICON_BIG, (LPARAM)(HICON)iMain);
  775. SendMessage(hSetting, WM_SETICON, ICON_SMALL, (LPARAM)(HICON)iMain);
  776. CheckRadioButton(hSetting, IDC_RADIO_NORMAL, IDC_RADIO_MAXIMIZE, IDC_RADIO_NORMAL);
  777. iProject = iWindowMode;
  778. if (iProject == 0)
  779. CheckRadioButton(hSetting, IDC_RADIO_NORMAL, IDC_RADIO_MAXIMIZE, IDC_RADIO_NORMAL);
  780. else
  781. CheckRadioButton(hSetting, IDC_RADIO_NORMAL, IDC_RADIO_MAXIMIZE, IDC_RADIO_MAXIMIZE);
  782. if (TraySave.aMode[iProject] == ACCENT_DISABLED)
  783. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_DEFAULT);
  784. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_TRANSPARENTGRADIENT)
  785. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_TRANSPARENT);
  786. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_BLURBEHIND)
  787. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_BLURBEHIND);
  788. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_ACRYLICBLURBEHIND)
  789. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_ACRYLIC);
  790. if (TraySave.iPos == 0)
  791. CheckRadioButton(hSetting, IDC_RADIO_LEFT, IDC_RADIO_RIGHT, IDC_RADIO_LEFT);
  792. else if (TraySave.iPos == 1)
  793. CheckRadioButton(hSetting, IDC_RADIO_LEFT, IDC_RADIO_RIGHT, IDC_RADIO_CENTER);
  794. else if (TraySave.iPos == 2)
  795. CheckRadioButton(hSetting, IDC_RADIO_LEFT, IDC_RADIO_RIGHT, IDC_RADIO_RIGHT);
  796. if (LOWORD(TraySave.iUnit) == 0)
  797. CheckRadioButton(hSetting, IDC_RADIO_AUTO, IDC_RADIO_MB, IDC_RADIO_AUTO);
  798. else if (LOWORD(TraySave.iUnit) == 1)
  799. CheckRadioButton(hSetting, IDC_RADIO_AUTO, IDC_RADIO_MB, IDC_RADIO_KB);
  800. else if (LOWORD(TraySave.iUnit) == 2)
  801. CheckRadioButton(hSetting, IDC_RADIO_AUTO, IDC_RADIO_MB, IDC_RADIO_MB);
  802. if (HIWORD(TraySave.iUnit) == 0)
  803. CheckRadioButton(hSetting, IDC_RADIO_BYTE, IDC_RADIO_BIT, IDC_RADIO_BYTE);
  804. else
  805. CheckRadioButton(hSetting, IDC_RADIO_BYTE, IDC_RADIO_BIT, IDC_RADIO_BIT);
  806. CheckDlgButton(hSetting, IDC_CHECK_TRAYICON, TraySave.bTrayIcon);
  807. CheckDlgButton(hSetting, IDC_CHECK_MONITOR, TraySave.bMonitor);
  808. CheckDlgButton(hSetting, IDC_CHECK_TRAFFIC, TraySave.bMonitorTraffic);
  809. CheckDlgButton(hSetting, IDC_CHECK_TEMPERATURE, TraySave.bMonitorTemperature);
  810. CheckDlgButton(hSetting, IDC_CHECK_USAGE, TraySave.bMonitorUsage);
  811. CheckDlgButton(hSetting, IDC_CHECK_SOUND, TraySave.bSound);
  812. CheckDlgButton(hSetting, IDC_CHECK_MONITOR_PDH, TraySave.bMonitorPDH);
  813. CheckDlgButton(hSetting, IDC_CHECK_MONITOR_SIMPLE, TraySave.iMonitorSimple);
  814. CheckDlgButton(hSetting, IDC_CHECK_MONITOR_LEFT, TraySave.bMonitorLeft);
  815. CheckDlgButton(hSetting, IDC_CHECK_MONITOR_FLOAT, TraySave.bMonitorFloat);
  816. CheckDlgButton(hSetting, IDC_CHECK_TRANSPARENT, TraySave.bMonitorTransparent);
  817. CheckDlgButton(hSetting, IDC_CHECK_TIPS, TraySave.bMonitorTips);
  818. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA, TBM_SETRANGE, 0, MAKELPARAM(0, 255));
  819. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA, TBM_SETPOS, TRUE, TraySave.bAlpha[iProject]);
  820. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA_B, TBM_SETRANGE, 0, MAKELPARAM(0, 255));
  821. BYTE bAlphaB = TraySave.dAlphaColor[iProject] >> 24;
  822. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA_B, TBM_SETPOS, TRUE, bAlphaB);
  823. SendDlgItemMessage(hSetting, IDC_CHECK_AUTORUN, BM_SETCHECK, AutoRun(FALSE, FALSE, szAppName), NULL);
  824. bSettingInit = TRUE;
  825. SetDlgItemInt(hSetting, IDC_EDIT1, TraySave.dNumValues[0] / 1048576, 0);
  826. SetDlgItemInt(hSetting, IDC_EDIT2, TraySave.dNumValues[1] / 1048576, 0);
  827. SetDlgItemInt(hSetting, IDC_EDIT3, TraySave.dNumValues[2], 0);
  828. SetDlgItemInt(hSetting, IDC_EDIT4, TraySave.dNumValues[3], 0);
  829. SetDlgItemInt(hSetting, IDC_EDIT5, TraySave.dNumValues[4], 0);
  830. SetDlgItemInt(hSetting, IDC_EDIT6, TraySave.dNumValues[5], 0);
  831. SetDlgItemInt(hSetting, IDC_EDIT7, TraySave.dNumValues[6], 0);
  832. SetDlgItemInt(hSetting, IDC_EDIT8, TraySave.dNumValues[7], 0);
  833. SetDlgItemInt(hSetting, IDC_EDIT9, TraySave.dNumValues[8] / 1048576, 0);
  834. SetDlgItemInt(hSetting, IDC_EDIT10, TraySave.dNumValues[9], 0);
  835. SetDlgItemInt(hSetting, IDC_EDIT11, TraySave.dNumValues[10], 0);
  836. SetDlgItemInt(hSetting, IDC_EDIT12, TraySave.dNumValues[11], 0);
  837. SetDlgItemInt(hSetting, IDC_EDIT_TIME, TraySave.FlushTime, 0);
  838. SetDlgItemText(hSetting, IDC_EDIT14, TraySave.szTrafficOut);
  839. SetDlgItemText(hSetting, IDC_EDIT15, TraySave.szTrafficIn);
  840. SetDlgItemText(hSetting, IDC_EDIT16, TraySave.szTemperatureCPU);
  841. SetDlgItemText(hSetting, IDC_EDIT17, TraySave.szTemperatureGPU);
  842. SetDlgItemText(hSetting, IDC_EDIT18, TraySave.szTemperatureCPUUnit);
  843. SetDlgItemText(hSetting, IDC_EDIT19, TraySave.szTemperatureGPUUnit);
  844. SetDlgItemText(hSetting, IDC_EDIT20, TraySave.szUsageCPU);
  845. SetDlgItemText(hSetting, IDC_EDIT21, TraySave.szUsageMEM);
  846. SetDlgItemText(hSetting, IDC_EDIT22, TraySave.szUsageCPUUnit);
  847. SetDlgItemText(hSetting, IDC_EDIT23, TraySave.szUsageMEMUnit);
  848. bSettingInit = FALSE;
  849. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  850. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_BACKGROUND), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  851. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_TRAFFIC_LOW), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  852. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_TRAFFIC_MEDIUM), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  853. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_TRAFFIC_HIGH), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  854. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_LOW), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  855. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_MEDUIM), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  856. oldColorButtonPoroc = (WNDPROC)SetWindowLongPtr(GetDlgItem(hSetting, IDC_BUTTON_COLOR_HIGH), GWLP_WNDPROC, (LONG_PTR)ColorButtonProc);
  857. ShowWindow(hSetting, SW_SHOW);
  858. UpdateWindow(hSetting);
  859. SetForegroundWindow(hSetting);
  860. }
  861. #if 0
  862. void SetTaskScheduler(BOOL bAdd)
  863. {
  864. HRESULT hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
  865. if (SUCCEEDED(hr))
  866. {
  867. // Set general COM security levels.
  868. hr = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_PKT_PRIVACY, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, 0, NULL);
  869. if (SUCCEEDED(hr))
  870. {
  871. ITaskService* pService = NULL;
  872. hr = CoCreateInstance(CLSID_TaskScheduler,
  873. NULL,
  874. CLSCTX_INPROC_SERVER,
  875. IID_ITaskService,
  876. (void**)&pService);
  877. if (SUCCEEDED(hr))
  878. {
  879. hr = pService->Connect(_variant_t(), _variant_t(), _variant_t(), _variant_t());
  880. if (SUCCEEDED(hr))
  881. {
  882. ITaskFolder* pRootFolder = NULL;
  883. hr = pService->GetFolder(_bstr_t(L"\\"), &pRootFolder);
  884. if (SUCCEEDED(hr))
  885. {
  886. pRootFolder->DeleteTask(_bstr_t(szAppName), 0);
  887. if (bAdd)
  888. {
  889. ITaskDefinition* pTask = NULL;
  890. hr = pService->NewTask(0, &pTask);
  891. // pService->Release(); // COM clean up. Pointer is no longer used.
  892. if (SUCCEEDED(hr))
  893. {
  894. if (IsUserAdmin())
  895. {
  896. IPrincipal* pPrincipal;
  897. hr = pTask->get_Principal(&pPrincipal);
  898. if (SUCCEEDED(hr))
  899. {
  900. pPrincipal->put_RunLevel(TASK_RUNLEVEL_HIGHEST);
  901. }
  902. }
  903. IRegistrationInfo* pRegInfo = NULL;
  904. hr = pTask->get_RegistrationInfo(&pRegInfo);
  905. if (SUCCEEDED(hr))
  906. {
  907. hr = pRegInfo->put_Author(_bstr_t(L"cgbsmy"));
  908. pRegInfo->Release();
  909. if (SUCCEEDED(hr))
  910. {
  911. ITaskSettings* pSettings = NULL;
  912. hr = pTask->get_Settings(&pSettings);
  913. if (SUCCEEDED(hr))
  914. {
  915. pSettings->put_StopIfGoingOnBatteries(VARIANT_FALSE);
  916. pSettings->put_DisallowStartIfOnBatteries(VARIANT_FALSE);
  917. pSettings->put_AllowHardTerminate(VARIANT_FALSE);
  918. pSettings->put_ExecutionTimeLimit(_bstr_t(L"PT0S"));
  919. pSettings->put_WakeToRun(VARIANT_TRUE);
  920. hr = pSettings->put_StartWhenAvailable(VARIANT_TRUE);
  921. pSettings->Release();
  922. if (SUCCEEDED(hr))
  923. {
  924. ITriggerCollection* pTriggerCollection = NULL;
  925. hr = pTask->get_Triggers(&pTriggerCollection);
  926. if (SUCCEEDED(hr))
  927. {
  928. ITrigger* pTrigger = NULL;
  929. hr = pTriggerCollection->Create(TASK_TRIGGER_LOGON, &pTrigger);
  930. pTriggerCollection->Release();
  931. if (SUCCEEDED(hr))
  932. {
  933. ILogonTrigger* pLogonTrigger = NULL;
  934. hr = pTrigger->QueryInterface(
  935. IID_ILogonTrigger, (void**)&pLogonTrigger);
  936. pTrigger->Release();
  937. if (SUCCEEDED(hr))
  938. {
  939. hr = pLogonTrigger->put_Id(_bstr_t(L"cgbsmy"));
  940. if (SUCCEEDED(hr))
  941. {
  942. // hr = pBootTrigger->put_StartBoundary(_bstr_t(L"2020-06-11T0:00:00"));
  943. // hr = pBootTrigger->put_EndBoundary(_bstr_t(L"2222-06-19T08:00:00"));
  944. // Delay the task to start 30 seconds after system start.
  945. // hr = pBootTrigger->put_Delay(_bstr_t(L"PT1S"));
  946. if (!IsUserAdmin())
  947. {
  948. WCHAR szName[MAX_PATH];
  949. DWORD dwLen = MAX_PATH;
  950. GetUserName(szName, &dwLen);
  951. pLogonTrigger->put_UserId(_bstr_t(szName));
  952. }
  953. pLogonTrigger->Release();
  954. IActionCollection* pActionCollection = NULL;
  955. hr = pTask->get_Actions(&pActionCollection);
  956. if (SUCCEEDED(hr))
  957. {
  958. // Create the action, specifying it as an executable action.
  959. IAction* pAction = NULL;
  960. hr = pActionCollection->Create(TASK_ACTION_EXEC, &pAction);
  961. pActionCollection->Release();
  962. if (SUCCEEDED(hr))
  963. {
  964. IExecAction* pExecAction = NULL;
  965. // QI for the executable task pointer.
  966. hr = pAction->QueryInterface(
  967. IID_IExecAction, (void**)&pExecAction);
  968. pAction->Release();
  969. if (SUCCEEDED(hr))
  970. {
  971. WCHAR szExe[MAX_PATH];
  972. GetModuleFileName(NULL, szExe, MAX_PATH);
  973. size_t sLen = wcslen(szExe);
  974. hr = pExecAction->put_Path(_bstr_t(szExe));
  975. pExecAction->put_Arguments(_bstr_t(L" t"));
  976. pExecAction->Release();
  977. if (SUCCEEDED(hr))
  978. {
  979. IRegisteredTask* pRegisteredTask = NULL;
  980. VARIANT varID;
  981. varID.vt = VT_NULL;
  982. VARIANT varPassword;
  983. varPassword.vt = VT_NULL;
  984. // MessageBox(hSetting, szName, szName, MB_OK);
  985. //hr = pRootFolder->RegisterTaskDefinition(_bstr_t(szAppName), pTask, TASK_CREATE_OR_UPDATE, _variant_t(), _variant_t(), TASK_LOGON_INTERACTIVE_TOKEN, _variant_t(L""), &pRegisteredTask);
  986. //hr = pRootFolder->RegisterTaskDefinition(_bstr_t(szAppName),pTask,TASK_CREATE_OR_UPDATE, _variant_t(),_variant_t(),TASK_LOGON_NONE,_variant_t(L""),&pRegisteredTask);
  987. if(IsUserAdmin())
  988. hr = pRootFolder->RegisterTaskDefinition(_bstr_t(szAppName), pTask, TASK_CREATE_OR_UPDATE, _variant_t(L"Builtin\\Administrators"), _variant_t(), TASK_LOGON_GROUP, _variant_t(L""), &pRegisteredTask);
  989. else
  990. hr = pRootFolder->RegisterTaskDefinition(_bstr_t(szAppName), pTask, TASK_CREATE_OR_UPDATE, _variant_t(), _variant_t(), TASK_LOGON_INTERACTIVE_TOKEN, _variant_t(L""), &pRegisteredTask);
  991. if (SUCCEEDED(hr))
  992. {
  993. IRunningTask* pRunningTask=NULL;
  994. VARIANT param;
  995. param.vt = VT_EMPTY;
  996. hr = pRegisteredTask->Run(param, &pRunningTask);
  997. // hr = pRegisteredTask->RunEx(param, TASK_RUN_IGNORE_CONSTRAINTS, NULL, NULL, &pRunningTask);
  998. if (SUCCEEDED(hr))
  999. {
  1000. }
  1001. pRegisteredTask->Release();
  1002. }
  1003. }
  1004. }
  1005. }
  1006. }
  1007. }
  1008. }
  1009. }
  1010. }
  1011. }
  1012. }
  1013. }
  1014. }
  1015. pTask->Release();
  1016. }
  1017. }
  1018. pRootFolder->Release();
  1019. }
  1020. }
  1021. pService->Release();
  1022. }
  1023. }
  1024. CoUninitialize();
  1025. }
  1026. }
  1027. HRESULT hr = S_OK;
  1028. ITaskScheduler* pITS;
  1029. hr = CoInitialize(NULL);
  1030. if (SUCCEEDED(hr))
  1031. {
  1032. hr = CoInitializeSecurity(
  1033. NULL,
  1034. -1,
  1035. NULL,
  1036. NULL,
  1037. RPC_C_AUTHN_LEVEL_PKT_PRIVACY,
  1038. RPC_C_IMP_LEVEL_IMPERSONATE,
  1039. NULL,
  1040. 0,
  1041. NULL
  1042. );
  1043. hr = CoCreateInstance(CLSID_CTaskScheduler,
  1044. NULL,
  1045. CLSCTX_INPROC_SERVER,
  1046. IID_ITaskScheduler,
  1047. (void**)&pITS);
  1048. if (SUCCEEDED(hr))
  1049. {
  1050. ITask* pITask;
  1051. IPersistFile* pIPersistFile;
  1052. pITS->Delete(szAppName);
  1053. hr = pITS->NewWorkItem(szAppName, // Name of task
  1054. CLSID_CTask, // Class identifier
  1055. IID_ITask, // Interface identifier
  1056. (IUnknown**)&pITask); // Address of task
  1057. pITS->Release(); // Release object
  1058. if (hr == S_OK)
  1059. {
  1060. pITask->
  1061. hr = pITask->QueryInterface(IID_IPersistFile,
  1062. (void**)&pIPersistFile);
  1063. pITask->Release();
  1064. if (hr == S_OK)
  1065. {
  1066. hr = pIPersistFile->Save(NULL, TRUE);
  1067. pIPersistFile->Release();
  1068. }
  1069. }
  1070. }
  1071. CoUninitialize();
  1072. }
  1073. #endif
  1074. #ifndef _DEBUG
  1075. extern "C" void WinMainCRTStartup()
  1076. {
  1077. LPWSTR lpCmdLine = GetCommandLine();
  1078. int iLen = lstrlen(lpCmdLine);
  1079. int flag = 0;
  1080. for (int i = 0; i < iLen; i++)
  1081. {
  1082. if (lpCmdLine[i] == L'\"')
  1083. {
  1084. ++flag;
  1085. }
  1086. else if (flag == 2)
  1087. {
  1088. lpCmdLine = &lpCmdLine[i + 1];
  1089. break;
  1090. }
  1091. else if (lpCmdLine[i] == L' ' && flag == 0)
  1092. {
  1093. lpCmdLine = &lpCmdLine[i + 1];
  1094. break;
  1095. }
  1096. }
  1097. if (lpCmdLine[0] == L'c') ////打开控制面板
  1098. {
  1099. CloseHandle(pShellExecute(NULL, L"open", L"control.exe", &lpCmdLine[1], NULL, SW_SHOW));
  1100. ExitProcess(0);
  1101. }
  1102. else if (lpCmdLine[0] == L'o') //用SHELLEXECUTE打开
  1103. {
  1104. CloseHandle(pShellExecute(NULL, L"open", &lpCmdLine[1], NULL, NULL, SW_SHOW));
  1105. ExitProcess(0);
  1106. }
  1107. else if (lpCmdLine[0] == L's') //打开任务计划
  1108. {
  1109. CloseHandle(pShellExecute(NULL, L"open", L"schtasks", &lpCmdLine[1], NULL, SW_HIDE));
  1110. ExitProcess(0);
  1111. }
  1112. if (IsUserAdmin())
  1113. {
  1114. // lpServiceName = (LPWSTR)szAppName;
  1115. InitService();
  1116. SERVICE_TABLE_ENTRY st[] =
  1117. {
  1118. {(LPWSTR)szAppName, (LPSERVICE_MAIN_FUNCTION)ServiceMain},
  1119. {NULL, NULL}
  1120. };
  1121. if (lstrcmpi(lpCmdLine, L"/install") == 0)
  1122. {
  1123. InstallService();
  1124. ExitProcess(0);
  1125. }
  1126. else if (lstrcmpi(lpCmdLine, L"/uninstall") == 0)
  1127. {
  1128. UninstallService();
  1129. ExitProcess(0);
  1130. }
  1131. else if (lstrcmpi(lpCmdLine, L"/start") == 0)
  1132. {
  1133. ServiceCtrlStart();
  1134. ExitProcess(0);
  1135. }
  1136. else if (lstrcmpi(lpCmdLine, L"/stop") == 0)
  1137. {
  1138. ServiceCtrlStop();
  1139. ExitProcess(0);
  1140. }
  1141. if (ServiceRunState() != SERVICE_RUNNING)
  1142. {
  1143. if (IsServiceInstalled())
  1144. {
  1145. if (ServiceRunState() == SERVICE_STOPPED)
  1146. ServiceCtrlStart();
  1147. StartServiceCtrlDispatcher(st);
  1148. ExitProcess(0);
  1149. }
  1150. }
  1151. ServiceCtrlStop();
  1152. }
  1153. #else
  1154. int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow)
  1155. {
  1156. UNREFERENCED_PARAMETER(hPrevInstance);
  1157. UNREFERENCED_PARAMETER(lpCmdLine);
  1158. if (lpCmdLine[0] == L'c') ////打开控制面板
  1159. {
  1160. CloseHandle(pShellExecute(NULL, L"open", L"control.exe", &lpCmdLine[1], NULL, SW_SHOW));
  1161. return 0;
  1162. }
  1163. else if (lpCmdLine[0] == L'o') //用SHELLEXECUTE打开
  1164. {
  1165. CloseHandle(pShellExecute(NULL, L"open", &lpCmdLine[1], NULL, NULL, SW_SHOW));
  1166. return 0;
  1167. }
  1168. else if (lpCmdLine[0] == L's') //打开任务计划
  1169. {
  1170. CloseHandle(pShellExecute(NULL, L"open", L"schtasks", &lpCmdLine[1], NULL, SW_HIDE));
  1171. return 0;
  1172. }
  1173. if (IsUserAdmin())
  1174. {
  1175. // lpServiceName = (LPWSTR)szAppName;
  1176. InitService();
  1177. SERVICE_TABLE_ENTRY st[] =
  1178. {
  1179. {(LPWSTR)szAppName, (LPSERVICE_MAIN_FUNCTION)ServiceMain},
  1180. {NULL, NULL}
  1181. };
  1182. if (lstrcmpi(lpCmdLine, L"/install") == 0)
  1183. {
  1184. InstallService();
  1185. return 0;
  1186. }
  1187. else if (lstrcmpi(lpCmdLine, L"/uninstall") == 0)
  1188. {
  1189. UninstallService();
  1190. return 0;
  1191. }
  1192. else if (lstrcmpi(lpCmdLine, L"/start") == 0)
  1193. {
  1194. ServiceCtrlStart();
  1195. return 0;
  1196. }
  1197. else if (lstrcmpi(lpCmdLine, L"/stop") == 0)
  1198. {
  1199. ServiceCtrlStop();
  1200. return 0;
  1201. }
  1202. if (ServiceRunState() != SERVICE_RUNNING)
  1203. {
  1204. if (IsServiceInstalled())
  1205. {
  1206. if (ServiceRunState() == SERVICE_STOPPED)
  1207. ServiceCtrlStart();
  1208. StartServiceCtrlDispatcher(st);
  1209. return 0;
  1210. }
  1211. }
  1212. ServiceCtrlStop();
  1213. }
  1214. #endif
  1215. while (hTray == NULL)
  1216. {
  1217. hTray = FindWindow(szShellTray, NULL);
  1218. if (hTray == NULL)
  1219. Sleep(100);
  1220. }
  1221. hInst = GetModuleHandle(NULL); // 将实例句柄存储在全局变量中
  1222. ReadReg();
  1223. /*
  1224. typedef WINUSERAPI DWORD WINAPI RTLGETVERSION(PRTL_OSVERSIONINFOW lpVersionInformation);
  1225. rovi.dwOSVersionInfoSize = sizeof(rovi);
  1226. RTLGETVERSION *RtlGetVersion = (RTLGETVERSION*)GetProcAddress(GetModuleHandleW(L"ntdll.dll"), "RtlGetVersion");
  1227. if (RtlGetVersion)
  1228. RtlGetVersion(&rovi);
  1229. */
  1230. hMutex = CreateMutex(NULL, TRUE, L"_TrayS_");
  1231. if (hMutex != NULL)
  1232. {
  1233. if (ERROR_ALREADY_EXISTS == GetLastError())
  1234. {
  1235. CloseHandle(hMutex);
  1236. if (lpCmdLine[0] != L't')
  1237. {
  1238. if (FindWindow(NULL, szAppName))
  1239. #ifndef _DEBUG
  1240. ExitProcess(0);
  1241. #else
  1242. return 0;
  1243. #endif
  1244. EnumWindows((WNDENUMPROC)FindWindowFunc, 0);
  1245. }
  1246. }
  1247. else
  1248. {
  1249. iMain = LoadIcon(hInst, MAKEINTRESOURCE(IDI_TRAYS));
  1250. hDwmapi = LoadLibrary(L"dwmapi.dll");
  1251. if (hDwmapi)
  1252. {
  1253. pDwmGetWindowAttribute = (pfnDwmGetWindowAttribute)GetProcAddress(hDwmapi, "DwmGetWindowAttribute");
  1254. }
  1255. SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);
  1256. if (TraySave.bMonitorTemperature)
  1257. LoadTemperatureDLL();
  1258. pProcessTime = NULL;
  1259. EnableDebugPrivilege(TRUE);
  1260. SYSTEM_INFO si;
  1261. GetSystemInfo(&si);
  1262. dNumProcessor = si.dwNumberOfProcessors;
  1263. if (dNumProcessor == 0)
  1264. dNumProcessor = 1;
  1265. ppmu[0] = &pmu[0];
  1266. ppmu[1] = &pmu[1];
  1267. ppmu[2] = &pmu[2];
  1268. ppcu[0] = &pcu[0];
  1269. ppcu[1] = &pcu[1];
  1270. ppcu[2] = &pcu[2];
  1271. g_hHeapWindowInfo = HeapCreate(NULL, 0, 0);
  1272. // 执行应用程序初始化:
  1273. if (!InitInstance(hInst, 0))
  1274. {
  1275. #ifndef _DEBUG
  1276. ExitProcess(0);
  1277. #else
  1278. return 0;
  1279. #endif
  1280. }
  1281. MSG msg;
  1282. // 主消息循环:
  1283. while (GetMessage(&msg, nullptr, 0, 0))
  1284. {
  1285. if (!IsDialogMessage(hMain, &msg) && !IsDialogMessage(hSetting, &msg))
  1286. {
  1287. TranslateMessage(&msg);
  1288. DispatchMessage(&msg);
  1289. }
  1290. }
  1291. if (IsWindow(hSetting))
  1292. DestroyWindow(hSetting);
  1293. CloseTaskBar();
  1294. if (IsWindow(hMain))
  1295. DestroyWindow(hMain);
  1296. pShell_NotifyIcon(NIM_DELETE, &nid);
  1297. DestroyIcon(iMain);
  1298. DeleteObject(hFont);
  1299. //free(ipinfo);
  1300. FreeLibrary(hDwmapi);
  1301. if (hIphlpapi)
  1302. FreeLibrary(hIphlpapi);
  1303. if (hOleacc)
  1304. FreeLibrary(hOleacc);
  1305. if (hPDH)
  1306. FreeLibrary(hPDH);
  1307. HeapFree(GetProcessHeap(), 0, mi);
  1308. HeapFree(GetProcessHeap(), 0, piaa);
  1309. HeapFree(GetProcessHeap(), 0, traffic);
  1310. HeapDestroy(g_hHeapWindowInfo);
  1311. if (hMutex)
  1312. CloseHandle(hMutex);
  1313. FreeTemperatureDLL();
  1314. if (bResetRun)
  1315. RunProcess(NULL, NULL);
  1316. }
  1317. }
  1318. // 初始化全局字符串
  1319. // LoadStringW(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
  1320. // LoadStringW(hInstance, IDC_TRAYS, szWindowClass, MAX_LOADSTRING);
  1321. ExitProcess((UINT)0);
  1322. }
  1323. //
  1324. // 函数: InitInstance(HINSTANCE, int)
  1325. //
  1326. // 目标: 保存实例句柄并创建主窗口
  1327. //
  1328. // 注释:
  1329. //
  1330. // 在此函数中,我们在全局变量中保存实例句柄并
  1331. // 创建和显示主程序窗口。
  1332. //
  1333. BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
  1334. {
  1335. hMain = ::CreateDialog(hInst, MAKEINTRESOURCE(IDD_MAIN), NULL, (DLGPROC)MainProc);
  1336. if (!hMain)
  1337. {
  1338. return FALSE;
  1339. }
  1340. ChangeWindowMessageFilter(WM_TRAYS, MSGFLT_ADD);
  1341. ChangeWindowMessageFilter(WM_DROPFILES, MSGFLT_ADD);
  1342. ChangeWindowMessageFilter(0x0049, MSGFLT_ADD);
  1343. //////////////////////////////////////////////////////////////////////////////////设置通知栏图标
  1344. nid.cbSize = sizeof NOTIFYICONDATA;
  1345. nid.uID = WM_IAWENTRAY;
  1346. nid.hWnd = hMain;
  1347. nid.hIcon = iMain;
  1348. nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
  1349. nid.uCallbackMessage = WM_IAWENTRAY;
  1350. // nid.dwInfoFlags = NIIF_INFO;
  1351. LoadString(hInst, IDS_TIPS, nid.szTip, 88);
  1352. if (TraySave.bTrayIcon)
  1353. pShell_NotifyIcon(NIM_ADD, &nid);
  1354. MemoryStatusEx.dwLength = sizeof MEMORYSTATUSEX;
  1355. ////////////////////////////////////////////////////////////当前DPI
  1356. HDC hdc = GetDC(hMain);
  1357. iDPI = GetDeviceCaps(hdc, LOGPIXELSY);
  1358. ::ReleaseDC(hMain, hdc);
  1359. if (TraySave.bMonitor)
  1360. {
  1361. AdjustWindowPos();
  1362. }
  1363. if (TraySave.aMode[0] != ACCENT_DISABLED || TraySave.aMode[1] != ACCENT_DISABLED)
  1364. SetTimer(hMain, 3, TraySave.FlushTime, NULL);
  1365. SetTimer(hMain, 6, 1000, NULL);
  1366. SetTimer(hMain, 11, 6000, NULL);
  1367. // ShowWindow(hMain,SW_SHOW);
  1368. return TRUE;
  1369. }
  1370. BOOL Find(IAccessible *paccParent, int iRole, IAccessible **paccChild) //查找任务图标UI
  1371. {
  1372. HRESULT hr;
  1373. long numChildren;
  1374. unsigned long numFetched;
  1375. VARIANT varChild;
  1376. int indexCount;
  1377. IAccessible *pChild = NULL;
  1378. IEnumVARIANT *pEnum = NULL;
  1379. IDispatch *pDisp = NULL;
  1380. BOOL found = false;
  1381. //Get the IEnumVARIANT interface
  1382. hr = paccParent->QueryInterface(IID_IEnumVARIANT, (PVOID *)&pEnum);
  1383. if (pEnum)
  1384. pEnum->Reset();
  1385. // Get child count
  1386. paccParent->get_accChildCount(&numChildren);
  1387. for (indexCount = 1; indexCount <= numChildren && !found; indexCount++)
  1388. {
  1389. pChild = NULL;
  1390. if (pEnum)
  1391. hr = pEnum->Next(1, &varChild, &numFetched);
  1392. else
  1393. {
  1394. varChild.vt = VT_I4;
  1395. varChild.lVal = indexCount;
  1396. }
  1397. if (varChild.vt == VT_I4)
  1398. {
  1399. pDisp = NULL;
  1400. hr = paccParent->get_accChild(varChild, &pDisp);
  1401. }
  1402. else
  1403. pDisp = varChild.pdispVal;
  1404. if (pDisp)
  1405. {
  1406. hr = pDisp->QueryInterface(IID_IAccessible, (void **)&pChild);
  1407. hr = pDisp->Release();
  1408. }
  1409. if (pChild)
  1410. {
  1411. VariantInit(&varChild);
  1412. varChild.vt = VT_I4;
  1413. varChild.lVal = CHILDID_SELF;
  1414. *paccChild = pChild;
  1415. }
  1416. VARIANT varState;
  1417. pChild->get_accState(varChild, &varState);
  1418. if ((varState.intVal & STATE_SYSTEM_INVISIBLE) == 0)
  1419. {
  1420. VARIANT varRole;
  1421. pChild->get_accRole(varChild, &varRole);
  1422. if (varRole.lVal == iRole)
  1423. {
  1424. paccParent->Release();
  1425. found = true;
  1426. break;
  1427. }
  1428. }
  1429. if (!found && pChild)
  1430. {
  1431. // found = Find(pCAcc, iRole, paccChild);
  1432. // if (*paccChild != pCAcc)
  1433. pChild->Release();
  1434. }
  1435. }
  1436. if (pEnum)
  1437. pEnum->Release();
  1438. return found;
  1439. }
  1440. int oleft = 0, otop = 0;
  1441. void SetTaskBarPos(HWND hTaskListWnd, HWND hTrayWnd, HWND hTaskWnd, HWND hReBarWnd, BOOL bMainTray) //设置任务栏图标位置
  1442. {
  1443. if (hOleacc == NULL)
  1444. {
  1445. hOleacc = LoadLibrary(L"oleacc.dll");
  1446. if (hOleacc)
  1447. {
  1448. AccessibleObjectFromWindowT = (pfnAccessibleObjectFromWindow)GetProcAddress(hOleacc, "AccessibleObjectFromWindow");
  1449. AccessibleChildrenT = (pfnAccessibleChildren)GetProcAddress(hOleacc, "AccessibleChildren");
  1450. }
  1451. }
  1452. if (hOleacc == NULL)
  1453. return;
  1454. IAccessible *pAcc = NULL;
  1455. AccessibleObjectFromWindowT(hTaskListWnd, OBJID_WINDOW, IID_IAccessible, (void **)&pAcc);
  1456. IAccessible *paccChlid = NULL;
  1457. if (pAcc)
  1458. {
  1459. if (Find(pAcc, 22, &paccChlid) == FALSE)
  1460. {
  1461. return;
  1462. }
  1463. }
  1464. else
  1465. return;
  1466. long childCount;
  1467. long returnCount;
  1468. LONG left, top, width, height;
  1469. LONG ol = 0, ot = 0;
  1470. int tWidth = 0;
  1471. int tHeight = 0;
  1472. if (paccChlid)
  1473. {
  1474. if (paccChlid->get_accChildCount(&childCount) == S_OK && childCount != 0)
  1475. {
  1476. VARIANT *pArray = (VARIANT *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof VARIANT *childCount);
  1477. if (AccessibleChildrenT(paccChlid, 0L, childCount, pArray, &returnCount) == S_OK)
  1478. {
  1479. for (int x = 0; x < returnCount; x++)
  1480. {
  1481. VARIANT vtChild = pArray[x];
  1482. VARIANT varState;
  1483. paccChlid->get_accState(vtChild, &varState);
  1484. if ((varState.intVal & STATE_SYSTEM_INVISIBLE) == 0)
  1485. {
  1486. VARIANT varRole;
  1487. paccChlid->get_accRole(vtChild, &varRole);
  1488. if (varRole.intVal == 0x2b || varRole.intVal == 0x39)
  1489. {
  1490. paccChlid->accLocation(&left, &top, &width, &height, vtChild);
  1491. if (ol != left)
  1492. {
  1493. tWidth += width;
  1494. ol = left;
  1495. }
  1496. if (ot != top)
  1497. {
  1498. tHeight += height;
  1499. ot = top;
  1500. }
  1501. }
  1502. }
  1503. }
  1504. }
  1505. HeapFree(GetProcessHeap(), 0, pArray);
  1506. }
  1507. paccChlid->Release();
  1508. }
  1509. else
  1510. return;
  1511. RECT lrc, src, trc;
  1512. GetWindowRect(hTaskListWnd, &lrc);
  1513. GetWindowRect(hTrayWnd, &src);
  1514. GetWindowRect(hTaskWnd, &trc);
  1515. BOOL Vertical = FALSE;
  1516. if (src.right - src.left < src.bottom - src.top)
  1517. Vertical = TRUE;
  1518. SendMessage(hReBarWnd, WM_SETREDRAW, TRUE, 0);
  1519. int lr, tb;
  1520. if (Vertical)
  1521. {
  1522. int t = trc.left - src.left;
  1523. int b = src.bottom - trc.bottom;
  1524. if (bMainTray && TraySave.bMonitor && TraySave.bMonitorFloat == FALSE)
  1525. {
  1526. if (TraySave.bMonitorLeft == FALSE)
  1527. b += mHeight;
  1528. else
  1529. t += mHeight;
  1530. }
  1531. if (t > b)
  1532. tb = t;
  1533. else
  1534. tb = b;
  1535. }
  1536. else
  1537. {
  1538. int l = trc.left - src.left;
  1539. int r = src.right - trc.right;
  1540. if (TraySave.bMonitor && bMainTray && TraySave.bMonitorFloat == FALSE)
  1541. {
  1542. if (TraySave.bMonitorLeft == FALSE)
  1543. r += mWidth;
  1544. else
  1545. l += mWidth;
  1546. }
  1547. if (l > r)
  1548. lr = l;
  1549. else
  1550. lr = r;
  1551. }
  1552. int nleft, ntop;
  1553. if ((TraySave.iPos == 2 || (Vertical == FALSE && tWidth >= trc.right - trc.left - lr) || (Vertical && tHeight >= trc.bottom - trc.top - tb)) && TraySave.iPos != 0)
  1554. {
  1555. if (Vertical)
  1556. {
  1557. ntop = trc.bottom - trc.top - tHeight;
  1558. if (TraySave.bMonitorLeft == FALSE && TraySave.bMonitor && bMainTray && TraySave.bMonitorFloat == FALSE)
  1559. ntop -= mHeight + 2;
  1560. }
  1561. else
  1562. {
  1563. nleft = trc.right - trc.left - tWidth;
  1564. if (TraySave.bMonitorLeft == FALSE && TraySave.bMonitor && bMainTray && TraySave.bMonitorFloat == FALSE)
  1565. nleft -= mWidth + 2;
  1566. }
  1567. }
  1568. else if (TraySave.iPos == 0)
  1569. {
  1570. if (TraySave.bMonitorLeft && TraySave.bMonitor && bMainTray && TraySave.bMonitorFloat == FALSE)
  1571. {
  1572. nleft = mWidth;
  1573. ntop = mHeight;
  1574. }
  1575. else
  1576. {
  1577. nleft = 0;
  1578. ntop = 0;
  1579. if (TraySave.bMonitor == FALSE)
  1580. {
  1581. SetTimer(hMain, 11, 1000, NULL);
  1582. }
  1583. }
  1584. }
  1585. else if (TraySave.iPos == 1)
  1586. {
  1587. if (Vertical)
  1588. ntop = src.top + (src.bottom - src.top) / 2 - trc.top - tHeight / 2;
  1589. else
  1590. nleft = src.left + (src.right - src.left) / 2 - trc.left - tWidth / 2;
  1591. if (bMainTray)
  1592. {
  1593. if (Vertical)
  1594. ntop -= 2;
  1595. else
  1596. nleft -= 2;
  1597. }
  1598. }
  1599. if (Vertical)
  1600. {
  1601. if (bMainTray)
  1602. {
  1603. if (otop == 0)
  1604. lrc.top = ntop;
  1605. else
  1606. lrc.top = otop;
  1607. otop = ntop;
  1608. while (ntop != lrc.top)
  1609. {
  1610. if (ntop > lrc.top)
  1611. ++lrc.top;
  1612. else
  1613. --lrc.top;
  1614. SetWindowPos(hTaskListWnd, 0, 0, lrc.top, lrc.right - lrc.left, lrc.bottom - lrc.top, SWP_NOSIZE | SWP_ASYNCWINDOWPOS | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOSENDCHANGING);
  1615. }
  1616. }
  1617. SetWindowPos(hTaskListWnd, 0, 0, ntop, lrc.right - lrc.left, lrc.bottom - lrc.top, SWP_NOSIZE | SWP_ASYNCWINDOWPOS | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOSENDCHANGING);
  1618. }
  1619. else
  1620. {
  1621. if (bMainTray)
  1622. {
  1623. if (oleft == 0)
  1624. lrc.left = nleft;
  1625. else
  1626. lrc.left = oleft;
  1627. oleft = nleft;
  1628. while (nleft != lrc.left)
  1629. {
  1630. if (nleft > lrc.left)
  1631. ++lrc.left;
  1632. else
  1633. --lrc.left;
  1634. SetWindowPos(hTaskListWnd, 0, lrc.left, 0, lrc.right - lrc.left, lrc.bottom - lrc.top, SWP_NOSIZE | SWP_ASYNCWINDOWPOS | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOSENDCHANGING);
  1635. }
  1636. }
  1637. SetWindowPos(hTaskListWnd, 0, nleft, 0, lrc.right - lrc.left, lrc.bottom - lrc.top, SWP_NOSIZE | SWP_ASYNCWINDOWPOS | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOSENDCHANGING);
  1638. }
  1639. if (TraySave.iPos != 0)
  1640. SendMessage(hReBarWnd, WM_SETREDRAW, FALSE, 0);
  1641. ShowWindow(hTaskWnd, SW_SHOWNOACTIVATE);
  1642. }
  1643. int otleft, ottop;
  1644. void SetWH()
  1645. {
  1646. mWidth = 0;
  1647. mHeight = 0;
  1648. HDC mdc = GetDC(hMain);
  1649. TraySave.TraybarFont.lfHeight = DPI(TraySave.TraybarFontSize);
  1650. DeleteObject(hFont);
  1651. hFont = CreateFontIndirect(&TraySave.TraybarFont); //创建字体
  1652. HFONT oldFont = (HFONT)SelectObject(mdc, hFont);
  1653. SIZE tSize;
  1654. WCHAR sz[16];
  1655. if (TraySave.bMonitorTraffic)
  1656. {
  1657. if (TraySave.iMonitorSimple == 1)
  1658. ::GetTextExtentPoint(mdc, L"↓:8.88M", lstrlen(L"↓:8.88M"), &tSize);
  1659. else if (TraySave.iMonitorSimple == 2)
  1660. ::GetTextExtentPoint(mdc, L"8.88M", lstrlen(L"8.88M"), &tSize);
  1661. else
  1662. {
  1663. wsprintf(sz, L"%s8.88M", TraySave.szTrafficOut);
  1664. ::GetTextExtentPoint(mdc, sz, lstrlen(sz), &tSize);
  1665. }
  1666. wTraffic = tSize.cx + tSize.cy / 4;
  1667. mWidth += wTraffic;
  1668. mHeight += tSize.cy * 2;
  1669. wHeight = tSize.cy + 1;
  1670. }
  1671. if (TraySave.bMonitorTemperature)
  1672. {
  1673. if (TraySave.iMonitorSimple == 1)
  1674. ::GetTextExtentPoint(mdc, L"88℃", lstrlen(L"88℃"), &tSize);
  1675. else if (TraySave.iMonitorSimple == 2)
  1676. ::GetTextExtentPoint(mdc, L"88", lstrlen(L"88"), &tSize);
  1677. else
  1678. {
  1679. wsprintf(sz, L"%s88%s", TraySave.szTemperatureGPU, TraySave.szTemperatureGPUUnit);
  1680. ::GetTextExtentPoint(mdc, sz, lstrlen(sz), &tSize);
  1681. }
  1682. wTemperature = tSize.cx + tSize.cy / 4;
  1683. mWidth += wTemperature;
  1684. wHeight = tSize.cy + 1;
  1685. if (bRing0)
  1686. mHeight += tSize.cy * 2;
  1687. else
  1688. mHeight += tSize.cy;
  1689. }
  1690. if (TraySave.bMonitorUsage)
  1691. {
  1692. if (TraySave.iMonitorSimple == 1)
  1693. ::GetTextExtentPoint(mdc, L"88%", lstrlen(L"88%"), &tSize);
  1694. else if (TraySave.iMonitorSimple == 2)
  1695. ::GetTextExtentPoint(mdc, L"88", lstrlen(L"88"), &tSize);
  1696. else
  1697. {
  1698. wsprintf(sz, L"%s88%s", TraySave.szUsageMEM, TraySave.szUsageMEMUnit);
  1699. ::GetTextExtentPoint(mdc, sz, lstrlen(sz), &tSize);
  1700. }
  1701. wUsage = tSize.cx + tSize.cy / 4;
  1702. mWidth += wUsage;
  1703. wHeight = tSize.cy + 1;
  1704. mHeight += tSize.cy * 2;
  1705. }
  1706. SelectObject(mdc, oldFont);
  1707. ReleaseDC(hMain, mdc);
  1708. ottop = -1;
  1709. otleft = -1;
  1710. AdjustWindowPos();
  1711. }
  1712. void AdjustWindowPos() //设置信息窗口位置大小
  1713. {
  1714. if (IsWindow(hTaskBar) == FALSE)
  1715. OpenTaskBar();
  1716. if (TraySave.bMonitorFloat)
  1717. {
  1718. RECT ScreenRect;
  1719. GetScreenRect(hTaskBar, &ScreenRect, FALSE);
  1720. if (TraySave.dMonitorPoint.x + mWidth > ScreenRect.right)
  1721. TraySave.dMonitorPoint.x = ScreenRect.right - mWidth;
  1722. if (TraySave.dMonitorPoint.y + wHeight * 2 > ScreenRect.bottom)
  1723. TraySave.dMonitorPoint.y = ScreenRect.bottom - wHeight * 2;
  1724. SetWindowPos(hTaskBar, HWND_TOPMOST, TraySave.dMonitorPoint.x, TraySave.dMonitorPoint.y, mWidth, wHeight * 2, SWP_NOACTIVATE);
  1725. return;
  1726. }
  1727. RECT rrc, trc;
  1728. GetWindowRect(hReBarWnd, &rrc);
  1729. GetWindowRect(hTaskWnd, &trc);
  1730. if (rrc.right - rrc.left > rrc.bottom - rrc.top)
  1731. VTray = FALSE;
  1732. else
  1733. VTray = TRUE;
  1734. if (VTray == FALSE)
  1735. {
  1736. int nleft;
  1737. if (TraySave.bMonitorLeft)
  1738. nleft = trc.left - rrc.left;
  1739. else
  1740. nleft = trc.right - trc.left - mWidth + (trc.left - rrc.left);
  1741. int h = wHeight * 2;
  1742. int ntop;
  1743. BOOL sTray = FALSE;
  1744. if (rrc.bottom - rrc.top < h)
  1745. {
  1746. sTray = TRUE;
  1747. h = rrc.bottom - rrc.top - 2;
  1748. ntop = 1;
  1749. }
  1750. else
  1751. ntop = (trc.bottom - trc.top - h) / 2;
  1752. if (nleft != otleft || ottop != ntop)
  1753. {
  1754. HDC hdc = GetDC(hTaskBar);
  1755. RECT crc;
  1756. GetClientRect(hTaskBar, &crc);
  1757. HBRUSH hb = CreateSolidBrush(RGB(0, 0, 0));
  1758. FillRect(hdc, &crc, hb);
  1759. DeleteObject(hb);
  1760. ReleaseDC(hTaskBar, hdc);
  1761. otleft = nleft;
  1762. ottop = ntop;
  1763. MoveWindow(hTaskBar, nleft, ntop, mWidth, h, FALSE);
  1764. ::InvalidateRect(hTaskBar, NULL, TRUE);
  1765. }
  1766. }
  1767. else
  1768. {
  1769. int ntop;
  1770. if (TraySave.bMonitorLeft)
  1771. ntop = trc.top - rrc.top;
  1772. else
  1773. ntop = trc.bottom - trc.top - mHeight + (trc.top - rrc.top);
  1774. int nleft = 2;
  1775. int w = trc.right - trc.left - 4;
  1776. if (ntop != ottop || otleft != w)
  1777. {
  1778. HDC hdc = GetDC(hTaskBar);
  1779. RECT crc;
  1780. GetClientRect(hTaskBar, &crc);
  1781. HBRUSH hb = CreateSolidBrush(RGB(0, 0, 0));
  1782. FillRect(hdc, &crc, hb);
  1783. DeleteObject(hb);
  1784. ReleaseDC(hTaskBar, hdc);
  1785. ottop = ntop;
  1786. otleft = w;
  1787. MoveWindow(hTaskBar, nleft, ntop, w, mHeight, FALSE);
  1788. ::InvalidateRect(hTaskBar, NULL, TRUE);
  1789. }
  1790. }
  1791. }
  1792. DWORD dwIPSize = 0;
  1793. DWORD dwMISize = 0;
  1794. INT_PTR CALLBACK TaskTipsProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) //提示信息窗口过程
  1795. {
  1796. switch (message)
  1797. {
  1798. case WM_INITDIALOG:
  1799. return (INT_PTR)TRUE;
  1800. case WM_MOUSEMOVE:
  1801. {
  1802. POINT pt;
  1803. pt.x = GET_X_LPARAM(lParam);
  1804. pt.y = GET_Y_LPARAM(lParam);
  1805. RECT rc;
  1806. GetClientRect(hDlg, &rc);
  1807. rc.top = nTraffic * wTipsHeight;
  1808. rc.bottom = (nTraffic + 6) * wTipsHeight;
  1809. rc.left = rc.right * 100 / 160;
  1810. rc.right = rc.right * 100 / 148;
  1811. if (PtInRect(&rc, pt))
  1812. {
  1813. inTipsProcessX = TRUE;
  1814. ::InvalidateRect(hDlg, NULL, TRUE);
  1815. }
  1816. else
  1817. {
  1818. inTipsProcessX = FALSE;
  1819. }
  1820. }
  1821. break;
  1822. case WM_LBUTTONDOWN:
  1823. {
  1824. POINT pt;
  1825. pt.x = GET_X_LPARAM(lParam);
  1826. pt.y = GET_Y_LPARAM(lParam);
  1827. if (pt.y < nTraffic * wTipsHeight)
  1828. RunProcess(NULL, szNetCpl);
  1829. else if (pt.y < (nTraffic + 6) * wTipsHeight)
  1830. {
  1831. RECT rc;
  1832. GetClientRect(hDlg, &rc);
  1833. rc.left = rc.right * 100 / 160;
  1834. rc.right = rc.right * 100 / 148;
  1835. if (PtInRect(&rc, pt))
  1836. {
  1837. int x = 0;
  1838. if (wTipsHeight != 0)
  1839. x = (pt.y / wTipsHeight) - nTraffic;
  1840. DWORD pid;
  1841. if (x < 3)
  1842. pid = ppcu[x]->dwProcessID;
  1843. else
  1844. pid = ppmu[x - 3]->dwProcessID;
  1845. HANDLE hProc = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
  1846. if (hProc)
  1847. {
  1848. TerminateProcess(hProc, 0);
  1849. CloseHandle(hProc);
  1850. inTipsProcessX = FALSE;
  1851. GetCursorPos(&pt);
  1852. SetCursorPos(pt.x + 88, pt.y);
  1853. }
  1854. }
  1855. else
  1856. RunProcess(NULL, szTaskmgr);
  1857. }
  1858. else
  1859. RunProcess(NULL, szPowerCpl);
  1860. return TRUE;
  1861. }
  1862. break;
  1863. case WM_MOUSELEAVE:
  1864. {
  1865. POINT pt;
  1866. GetCursorPos(&pt);
  1867. if (WindowFromPoint(pt) != hTaskBar)
  1868. {
  1869. if (pProcessTime != NULL)
  1870. {
  1871. HeapFree(GetProcessHeap(), 0, pProcessTime);
  1872. pProcessTime = NULL;
  1873. }
  1874. DestroyWindow(hTaskTips);
  1875. SetTimer(hMain, 11, 1000, NULL);
  1876. }
  1877. }
  1878. break;
  1879. case WM_ERASEBKGND:
  1880. HDC hdc = (HDC)wParam; //BeginPaint(hDlg, &ps);
  1881. RECT rc, crc;
  1882. GetClientRect(hDlg, &rc);
  1883. crc = rc;
  1884. HDC mdc = CreateCompatibleDC(hdc);
  1885. HBITMAP hMemBmp = CreateCompatibleBitmap(hdc, rc.right - rc.left, rc.bottom - rc.top);
  1886. HBITMAP oldBmp = (HBITMAP)SelectObject(mdc, hMemBmp);
  1887. // if (bErasebkgnd)
  1888. {
  1889. TraySave.TipsFont.lfHeight = TraySave.TipsFontSize;
  1890. HFONT hTipsFont = CreateFontIndirect(&TraySave.TipsFont); //创建字体
  1891. HFONT oldFont = (HFONT)SelectObject(mdc, hTipsFont);
  1892. WCHAR sz[64];
  1893. SetBkMode(mdc, TRANSPARENT);
  1894. COLORREF rgb;
  1895. rgb = RGB(192, 192, 192);
  1896. SetTextColor(mdc, rgb);
  1897. rc.bottom = wTipsHeight;
  1898. HBRUSH hb = CreateSolidBrush(RGB(22, 22, 22));
  1899. for (int i = 0; i < nTraffic / 2 + 4; i++)
  1900. {
  1901. FillRect(mdc, &rc, hb);
  1902. OffsetRect(&rc, 0, wTipsHeight * 2);
  1903. }
  1904. DeleteObject(hb);
  1905. HPEN hp = CreatePen(PS_DOT, 1, RGB(98, 98, 98));
  1906. HPEN oldpen = (HPEN)SelectObject(mdc, hp);
  1907. MoveToEx(mdc, crc.right * 10 / 23, 0, NULL);
  1908. LineTo(mdc, crc.right * 10 / 23, wTipsHeight * nTraffic);
  1909. MoveToEx(mdc, crc.right * 7 / 10, 0, NULL);
  1910. LineTo(mdc, crc.right * 7 / 10, wTipsHeight * nTraffic);
  1911. MoveToEx(mdc, crc.right * 85 / 100, 0, NULL);
  1912. LineTo(mdc, crc.right * 85 / 100, wTipsHeight * nTraffic);
  1913. MoveToEx(mdc, crc.right * 100 / 124, wTipsHeight * nTraffic, NULL);
  1914. LineTo(mdc, crc.right * 100 / 124, wTipsHeight * (nTraffic + 6));
  1915. MoveToEx(mdc, crc.right * 100 / 148, wTipsHeight * nTraffic, NULL);
  1916. LineTo(mdc, crc.right * 100 / 148, wTipsHeight * (nTraffic + 6));
  1917. MoveToEx(mdc, crc.right * 100 / 160, wTipsHeight * nTraffic, NULL);
  1918. LineTo(mdc, crc.right * 100 / 160, wTipsHeight * (nTraffic + 6));
  1919. MoveToEx(mdc, 0, wTipsHeight * nTraffic, NULL);
  1920. LineTo(mdc, crc.right, wTipsHeight * nTraffic);
  1921. MoveToEx(mdc, 0, wTipsHeight * (nTraffic + 3), NULL);
  1922. LineTo(mdc, crc.right, wTipsHeight * (nTraffic + 3));
  1923. MoveToEx(mdc, 0, wTipsHeight * (nTraffic + 6), NULL);
  1924. LineTo(mdc, crc.right, wTipsHeight * (nTraffic + 6));
  1925. SelectObject(mdc, oldpen);
  1926. DeleteObject(hp);
  1927. rc.bottom = wTipsHeight;
  1928. rc.top = 0;
  1929. int cx;
  1930. if (wTipsHeight < 24)
  1931. cx = 16;
  1932. else if (wTipsHeight < 32)
  1933. cx = 24;
  1934. else if (wTipsHeight < 48)
  1935. cx = 32;
  1936. else if (wTipsHeight < 64)
  1937. cx = 48;
  1938. else if (wTipsHeight < 128)
  1939. cx = 64;
  1940. else
  1941. cx = 128;
  1942. // OffsetRect(&rc, 0, DPI(16)*3);
  1943. //PIP_ADAPTER_INFO pai = &ipinfo[0];
  1944. // PIP_ADAPTER_ADDRESSES paa = &piaa[0];
  1945. for (int i = 0; i < nTraffic; i++)
  1946. {
  1947. rc.left = 2; // +wTipsHeight;
  1948. DrawText(mdc, traffic[i].FriendlyName, lstrlen(traffic[i].FriendlyName), &rc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  1949. #if 0
  1950. HICON hIcon = GetIconForCSIDL(CSIDL_CONNECTIONS);
  1951. UINT size = MAKELPARAM(cx, cx);
  1952. WCHAR szGuid[64]; // = L":::";
  1953. MultiByteToWideChar(CP_ACP, 0, traffic[i].AdapterName, 64, szGuid , 64);
  1954. SHFILEINFO shfi;
  1955. SHGetFileInfo(szGuid, 0, &shfi, sizeof(shfi), SHGFI_ICON | SHGFI_USEFILEATTRIBUTES);
  1956. hIcon = shfi.hIcon;
  1957. if (wTipsHeight >= 16)
  1958. DrawIconEx(mdc, 2, rc.top + (rc.bottom - rc.top - cx) / 2, hIcon, cx, cx, 0, NULL, DI_NORMAL);
  1959. else
  1960. DrawIconEx(mdc, 1, rc.top + 1, hIcon, wTipsHeight - 2, wTipsHeight - 2, 0, NULL, DI_NORMAL);
  1961. DestroyIcon(hIcon);
  1962. #endif
  1963. rc.left = crc.right * 10 / 23;
  1964. rc.right = crc.right * 7 / 10;
  1965. DrawText(mdc, traffic[i].IP4, lstrlen(traffic[i].IP4), &rc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  1966. int f_in_byte = traffic[i].in_byte;
  1967. if (traffic[i].in_byte < 1000)
  1968. wsprintf(sz, L"↓:%db", traffic[i].in_byte);
  1969. else if (traffic[i].in_byte < 1000000)
  1970. {
  1971. int k_in_byte = f_in_byte / 1000;
  1972. if (k_in_byte >= 100)
  1973. wsprintf(sz, L"↓:%dk", k_in_byte);
  1974. else if (k_in_byte >= 10)
  1975. wsprintf(sz, L"↓:%d.%dk", k_in_byte, f_in_byte / 100 - k_in_byte * 10);
  1976. else
  1977. wsprintf(sz, L"↓:%d.%dk", k_in_byte, f_in_byte / 10 - k_in_byte * 100);
  1978. }
  1979. else if (traffic[i].in_byte < 1000000000)
  1980. {
  1981. int m_in_byte = f_in_byte / 1000000;
  1982. if (m_in_byte >= 100)
  1983. wsprintf(sz, L"↓:%dm", m_in_byte);
  1984. else if (m_in_byte >= 10)
  1985. wsprintf(sz, L"↓:%d.%dm", m_in_byte, f_in_byte / 100000 - m_in_byte * 10);
  1986. else
  1987. wsprintf(sz, L"↓:%d.%dm", m_in_byte, f_in_byte / 10000 - m_in_byte * 100);
  1988. }
  1989. else
  1990. {
  1991. int g_in_byte = f_in_byte / 1000000000;
  1992. if (g_in_byte >= 100)
  1993. wsprintf(sz, L"↓:%dG", g_in_byte);
  1994. else if (g_in_byte >= 10)
  1995. wsprintf(sz, L"↓:%d.%dG", g_in_byte, f_in_byte / 100000000 - g_in_byte * 10);
  1996. else
  1997. wsprintf(sz, L"↓:%d.%dG", g_in_byte, f_in_byte / 10000000 - g_in_byte * 100);
  1998. }
  1999. rc.left = crc.right * 7 / 10 + 2;
  2000. rc.right += crc.right;
  2001. DrawText(mdc, sz, lstrlen(sz), &rc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2002. int f_out_byte = traffic[i].out_byte;
  2003. if (traffic[i].out_byte < 1000)
  2004. wsprintf(sz, L"↑:%db", traffic[i].out_byte);
  2005. else if (traffic[i].out_byte < 1000000)
  2006. {
  2007. int k_out_byte = f_out_byte / 1000;
  2008. if (k_out_byte >= 100)
  2009. wsprintf(sz, L"↑:%dk", k_out_byte);
  2010. else if (k_out_byte >= 10)
  2011. wsprintf(sz, L"↑:%d.%dk", k_out_byte, f_out_byte / 100 - k_out_byte * 10);
  2012. else
  2013. wsprintf(sz, L"↑:%d.%dk", k_out_byte, f_out_byte / 10 - k_out_byte * 100);
  2014. }
  2015. else if (traffic[i].out_byte < 1000000000)
  2016. {
  2017. int m_out_byte = f_out_byte / 1000000;
  2018. if (m_out_byte >= 100)
  2019. wsprintf(sz, L"↑:%dm", m_out_byte);
  2020. else if (m_out_byte >= 10)
  2021. wsprintf(sz, L"↑:%d.%dm", m_out_byte, f_out_byte / 100000 - m_out_byte * 10);
  2022. else
  2023. wsprintf(sz, L"↑:%d.%dm", m_out_byte, f_out_byte / 10000 - m_out_byte * 100);
  2024. }
  2025. else
  2026. {
  2027. int g_out_byte = f_out_byte / 1000000000;
  2028. if (g_out_byte >= 100)
  2029. wsprintf(sz, L"↑:%dg", g_out_byte);
  2030. else if (g_out_byte >= 10)
  2031. wsprintf(sz, L"↑:%d.%dg", g_out_byte, f_out_byte / 100000000 - g_out_byte * 10);
  2032. else
  2033. wsprintf(sz, L"↑:%d.%dg", g_out_byte, f_out_byte / 10000000 - g_out_byte * 100);
  2034. }
  2035. rc.left = crc.right * 85 / 100 + 2;
  2036. DrawText(mdc, sz, lstrlen(sz), &rc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2037. OffsetRect(&rc, 0, wTipsHeight);
  2038. }
  2039. rc.left = 2; // +wTipsHeight;
  2040. rc.right = crc.right - 2;
  2041. POINT pt;
  2042. GetCursorPos(&pt);
  2043. ScreenToClient(hDlg, &pt);
  2044. for (int i = 0; i < 3; i++)
  2045. {
  2046. SetTextColor(mdc, RGB(192, 192, 0));
  2047. DrawText(mdc, ppcu[i]->szExe, lstrlen(ppcu[i]->szExe), &rc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2048. #if 0
  2049. HICON hIcon = OpenProcessIcon(ppcu[i]->dwProcessID,cx);
  2050. if(wTipsHeight>=16)
  2051. DrawIconEx(mdc, 2, rc.top + (rc.bottom - rc.top - cx) / 2, hIcon, cx, cx, 0, NULL, DI_NORMAL);
  2052. else
  2053. DrawIconEx(mdc, 1, rc.top+1, hIcon, wTipsHeight-2, wTipsHeight-2, 0, NULL, DI_NORMAL);
  2054. DestroyIcon(hIcon);
  2055. #endif
  2056. int iCpuUsage = int(ppcu[i]->fCpuUsage * 100);
  2057. wsprintf(sz, L"%d.%.2d%%", iCpuUsage / 100, iCpuUsage % 100);
  2058. DrawText(mdc, sz, lstrlen(sz), &rc, DT_RIGHT | DT_VCENTER | DT_SINGLELINE);
  2059. RECT cr = rc;
  2060. cr.left = crc.right * 100 / 148;
  2061. cr.right = crc.right * 8 / 10;
  2062. wsprintf(sz, L"%d", ppcu[i]->dwProcessID);
  2063. DrawText(mdc, sz, lstrlen(sz), &cr, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2064. cr.left = crc.right * 100 / 160;
  2065. cr.right = crc.right * 100 / 148;
  2066. if (PtInRect(&cr, pt))
  2067. SetTextColor(mdc, RGB(255, 255, 255));
  2068. DrawText(mdc, L"X", 1, &cr, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2069. OffsetRect(&rc, 0, wTipsHeight);
  2070. }
  2071. for (int i = 0; i < 3; i++)
  2072. {
  2073. SetTextColor(mdc, RGB(0, 192, 192));
  2074. DrawText(mdc, ppmu[i]->szExe, lstrlen(ppmu[i]->szExe), &rc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2075. #if 0
  2076. HICON hIcon = OpenProcessIcon(ppmu[i]->dwProcessID,cx);
  2077. if(wTipsHeight>=16)
  2078. DrawIconEx(mdc, 2, rc.top+(rc.bottom-rc.top-cx)/2, hIcon, cx, cx, 0, NULL, DI_NORMAL);
  2079. else
  2080. DrawIconEx(mdc, 1, rc.top+1, hIcon, wTipsHeight-2, wTipsHeight-2, 0, NULL, DI_NORMAL);
  2081. DestroyIcon(hIcon);
  2082. #endif
  2083. if (ppmu[i]->dwMemUsage >= 1048576000)
  2084. {
  2085. SIZE_T iMemUsage = (ppmu[i]->dwMemUsage * 100 / 1073741824);
  2086. wsprintf(sz, L"%d.%.2dGB", iMemUsage / 100, iMemUsage % 100);
  2087. }
  2088. else
  2089. {
  2090. SIZE_T iMemUsage = (ppmu[i]->dwMemUsage * 100 / 1048576);
  2091. wsprintf(sz, L"%d.%.2dMB", iMemUsage / 100, iMemUsage % 100);
  2092. }
  2093. DrawText(mdc, sz, lstrlen(sz), &rc, DT_RIGHT | DT_VCENTER | DT_SINGLELINE);
  2094. RECT cr = rc;
  2095. cr.left = crc.right * 100 / 148;
  2096. cr.right = crc.right * 8 / 10;
  2097. wsprintf(sz, L"%d", ppmu[i]->dwProcessID);
  2098. DrawText(mdc, sz, lstrlen(sz), &cr, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2099. cr.left = crc.right * 100 / 160;
  2100. cr.right = crc.right * 100 / 148;
  2101. if (PtInRect(&cr, pt))
  2102. SetTextColor(mdc, RGB(255, 255, 255));
  2103. DrawText(mdc, L"X", 1, &cr, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2104. OffsetRect(&rc, 0, wTipsHeight);
  2105. }
  2106. SetTextColor(mdc, RGB(192, 192, 192));
  2107. #if 0
  2108. HBRUSH hb;
  2109. hb=CreateSolidBrush(RGB(0,38,0););
  2110. FillRect(mdc, &rc, hb);
  2111. DeleteObject(hb);
  2112. #endif
  2113. rc.left = 2;
  2114. rc.right -= 2;
  2115. PROCESSOR_POWER_INFORMATION *pi = (PROCESSOR_POWER_INFORMATION *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(PROCESSOR_POWER_INFORMATION) * dNumProcessor);
  2116. if (pCallNtPowerInformation(ProcessorInformation, NULL, 0, &pi[0], sizeof(PROCESSOR_POWER_INFORMATION) *dNumProcessor) == 0)
  2117. {
  2118. int iCGhz = pi[0].CurrentMhz / 10;
  2119. int iMGhz = pi[0].MaxMhz / 10;
  2120. wsprintf(sz, L"%d个逻辑处理器 当前频率%d.%.2dGHz 最大频率%d.%.2dGHz",
  2121. dNumProcessor, iCGhz / 100, iCGhz % 100, iMGhz / 100, iMGhz % 100);
  2122. }
  2123. HeapFree(GetProcessHeap(), 0, pi);
  2124. DrawText(mdc, sz, lstrlen(sz), &rc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2125. OffsetRect(&rc, 0, wTipsHeight);
  2126. DWORDLONG iaPage = MemoryStatusEx.ullAvailPageFile * 100 / 1073741824;
  2127. DWORDLONG itPage = MemoryStatusEx.ullTotalPageFile * 100 / 1073741824;
  2128. DWORDLONG ia = MemoryStatusEx.ullAvailPhys * 100 / 1073741824;
  2129. DWORDLONG it = MemoryStatusEx.ullTotalPhys * 100 / 1073741824;
  2130. wsprintf(sz, L"虚拟内存:%d.%.2d/%d.%.2dGB,可用内存:%d.%.2d/%d.%.2dGB",
  2131. iaPage / 100, iaPage % 100, itPage / 100, itPage % 100, ia / 100, ia % 100, it / 100, it % 100);
  2132. DrawText(mdc, sz, lstrlen(sz), &rc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2133. DeleteObject(hTipsFont);
  2134. SelectObject(mdc, oldFont);
  2135. }
  2136. GetClientRect(hDlg, &rc);
  2137. BitBlt(hdc, 0, 0, rc.right - rc.left, rc.bottom - rc.top, mdc, 0, 0, SRCCOPY);
  2138. SelectObject(mdc, oldBmp);
  2139. DeleteObject(hMemBmp);
  2140. DeleteDC(mdc);
  2141. return TRUE;
  2142. break;
  2143. }
  2144. return (INT_PTR)FALSE;
  2145. }
  2146. void GetProcessCpuUsage() //获取进程CPU占用前三
  2147. {
  2148. if (!inTipsProcessX)
  2149. {
  2150. ppcu[0] = &pcu[0];
  2151. ppcu[1] = &pcu[1];
  2152. ppcu[2] = &pcu[2];
  2153. memset(pcu, 0, sizeof pcu);
  2154. pcu[0].fCpuUsage = 0;
  2155. pcu[1].fCpuUsage = 0;
  2156. pcu[2].fCpuUsage = 0;
  2157. }
  2158. DWORD dCurID = GetCurrentProcessId();
  2159. PROCESSENTRY32 pe;
  2160. pe.dwSize = sizeof(PROCESSENTRY32);
  2161. HANDLE hs = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  2162. if (hs != INVALID_HANDLE_VALUE)
  2163. {
  2164. BOOL ret = Process32First(hs, &pe);
  2165. while (ret)
  2166. {
  2167. if (pe.th32ProcessID != dCurID)
  2168. {
  2169. HANDLE hProc = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pe.th32ProcessID);
  2170. if (hProc)
  2171. {
  2172. int n = -1;
  2173. for (int i = 0; i < nProcess + 31; i++)
  2174. {
  2175. if (pProcessTime[i].dwProcessID == pe.th32ProcessID)
  2176. {
  2177. n = i;
  2178. break;
  2179. }
  2180. else if (n == -1 && pProcessTime[i].dwProcessID == NULL)
  2181. n = i;
  2182. }
  2183. FILETIME CreateTime, ExitTime, KernelTime, UserTime;
  2184. if (GetProcessTimes(hProc, &CreateTime, &ExitTime, &KernelTime, &UserTime))
  2185. {
  2186. float nProcCpuPercent = 0;
  2187. BOOL bRetCode = FALSE;
  2188. FILETIME CreateTime, ExitTime, KernelTime, UserTime;
  2189. LARGE_INTEGER lgKernelTime;
  2190. LARGE_INTEGER lgUserTime;
  2191. LARGE_INTEGER lgCurTime;
  2192. bRetCode = GetProcessTimes(hProc, &CreateTime, &ExitTime, &KernelTime, &UserTime);
  2193. if (bRetCode)
  2194. {
  2195. lgKernelTime.HighPart = KernelTime.dwHighDateTime;
  2196. lgKernelTime.LowPart = KernelTime.dwLowDateTime;
  2197. lgUserTime.HighPart = UserTime.dwHighDateTime;
  2198. lgUserTime.LowPart = UserTime.dwLowDateTime;
  2199. lgCurTime.QuadPart = (lgKernelTime.QuadPart + lgUserTime.QuadPart) / 10000;
  2200. if (pProcessTime[n].g_slgProcessTimeOld.QuadPart == 0)
  2201. nProcCpuPercent = 0;
  2202. else
  2203. nProcCpuPercent = (float)((lgCurTime.QuadPart - pProcessTime[n].g_slgProcessTimeOld.QuadPart) * 100 / 1000);
  2204. pProcessTime[n].g_slgProcessTimeOld = lgCurTime;
  2205. pProcessTime[n].dwProcessID = pe.th32ProcessID;
  2206. nProcCpuPercent = nProcCpuPercent / dNumProcessor;
  2207. }
  2208. else
  2209. {
  2210. nProcCpuPercent = 0;
  2211. }
  2212. if (nProcCpuPercent > 100)
  2213. nProcCpuPercent = 0;
  2214. if (!inTipsProcessX)
  2215. {
  2216. PROCESSCPUUSAGE *ppc;
  2217. if (ppcu[0]->fCpuUsage <= nProcCpuPercent)
  2218. {
  2219. ppc = ppcu[2];
  2220. ppcu[2] = ppcu[1];
  2221. ppcu[1] = ppcu[0];
  2222. ppcu[0] = ppc;
  2223. ppcu[0]->dwProcessID = pe.th32ProcessID;
  2224. ppcu[0]->fCpuUsage = nProcCpuPercent;
  2225. lstrcpyn(ppcu[0]->szExe, pe.szExeFile, 24);
  2226. }
  2227. else if (ppcu[1]->fCpuUsage <= nProcCpuPercent)
  2228. {
  2229. ppc = ppcu[2];
  2230. ppcu[2] = ppcu[1];
  2231. ppcu[1] = ppc;
  2232. ppcu[1]->dwProcessID = pe.th32ProcessID;
  2233. ppcu[1]->fCpuUsage = nProcCpuPercent;
  2234. lstrcpyn(ppcu[1]->szExe, pe.szExeFile, 24);
  2235. }
  2236. else if (ppcu[2]->fCpuUsage <= nProcCpuPercent)
  2237. {
  2238. ppcu[2]->dwProcessID = pe.th32ProcessID;
  2239. ppcu[2]->fCpuUsage = nProcCpuPercent;
  2240. lstrcpyn(ppcu[2]->szExe, pe.szExeFile, 24);
  2241. }
  2242. }
  2243. }
  2244. CloseHandle(hProc);
  2245. }
  2246. }
  2247. ret = Process32Next(hs, &pe);
  2248. }
  2249. CloseHandle(hs);
  2250. }
  2251. }
  2252. int GetProcessMemUsage() //获取进程内存占用前三
  2253. {
  2254. if (!inTipsProcessX)
  2255. {
  2256. ppmu[0] = &pmu[0];
  2257. ppmu[1] = &pmu[1];
  2258. ppmu[2] = &pmu[2];
  2259. memset(pmu, 0, sizeof pmu);
  2260. pmu[0].dwMemUsage = 0;
  2261. pmu[1].dwMemUsage = 0;
  2262. pmu[2].dwMemUsage = 0;
  2263. }
  2264. PROCESSENTRY32 pe;
  2265. pe.dwSize = sizeof(PROCESSENTRY32);
  2266. int n = 0;
  2267. HANDLE hs = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  2268. if (hs != INVALID_HANDLE_VALUE)
  2269. {
  2270. BOOL ret = Process32First(hs, &pe);
  2271. while (ret)
  2272. {
  2273. ++n;
  2274. if (lstrcmp(pe.szExeFile, L"Memory Compression") != 0 && !inTipsProcessX)
  2275. {
  2276. HANDLE hProc = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pe.th32ProcessID);
  2277. if (hProc)
  2278. {
  2279. PROCESS_MEMORY_COUNTERS_EX pmc;
  2280. if (GetProcessMemoryInfo(hProc, (PPROCESS_MEMORY_COUNTERS)&pmc, sizeof(pmc)))
  2281. {
  2282. //QueryWorkingSet()
  2283. PROCESSMEMORYUSAGE *ppm;
  2284. if (ppmu[0]->dwMemUsage <= pmc.WorkingSetSize)
  2285. {
  2286. ppm = ppmu[2];
  2287. ppmu[2] = ppmu[1];
  2288. ppmu[1] = ppmu[0];
  2289. ppmu[0] = ppm;
  2290. ppmu[0]->dwProcessID = pe.th32ProcessID;
  2291. ppmu[0]->dwMemUsage = pmc.WorkingSetSize;
  2292. lstrcpyn(ppmu[0]->szExe, pe.szExeFile, 24);
  2293. }
  2294. else if (ppmu[1]->dwMemUsage <= pmc.WorkingSetSize)
  2295. {
  2296. ppm = ppmu[2];
  2297. ppmu[2] = ppmu[1];
  2298. ppmu[1] = ppm;
  2299. ppmu[1]->dwProcessID = pe.th32ProcessID;
  2300. ppmu[1]->dwMemUsage = pmc.WorkingSetSize;
  2301. lstrcpyn(ppmu[1]->szExe, pe.szExeFile, 24);
  2302. }
  2303. else if (ppmu[2]->dwMemUsage <= pmc.WorkingSetSize)
  2304. {
  2305. ppmu[2]->dwProcessID = pe.th32ProcessID;
  2306. ppmu[2]->dwMemUsage = pmc.WorkingSetSize;
  2307. lstrcpyn(ppmu[2]->szExe, pe.szExeFile, 24);
  2308. }
  2309. }
  2310. CloseHandle(hProc);
  2311. }
  2312. }
  2313. ret = Process32Next(hs, &pe);
  2314. }
  2315. CloseHandle(hs);
  2316. }
  2317. return n;
  2318. }
  2319. void DrawTraffic(HDC mdc, LPRECT lpRect, DWORD dwByte, BOOL bInOut)
  2320. {
  2321. WCHAR szInS[] = L"↓:";
  2322. WCHAR szInS2[] = L"";
  2323. WCHAR szOutS[] = L"↑:";
  2324. WCHAR szOutS2[] = L"";
  2325. WCHAR *szT;
  2326. if (bInOut)
  2327. {
  2328. if (TraySave.iMonitorSimple == 1)
  2329. szT = szInS;
  2330. else if (TraySave.iMonitorSimple == 2)
  2331. szT = szInS2;
  2332. else
  2333. szT = TraySave.szTrafficIn;
  2334. }
  2335. else
  2336. {
  2337. if (TraySave.iMonitorSimple == 1)
  2338. szT = szOutS;
  2339. else if (TraySave.iMonitorSimple == 2)
  2340. szT = szOutS2;
  2341. else
  2342. szT = TraySave.szTrafficOut;
  2343. }
  2344. WCHAR sz[24];
  2345. COLORREF rgb;
  2346. if (dwByte < TraySave.dNumValues[0])
  2347. rgb = TraySave.cMonitorColor[1];
  2348. else if (dwByte < TraySave.dNumValues[1])
  2349. rgb = TraySave.cMonitorColor[2];
  2350. else
  2351. rgb = TraySave.cMonitorColor[3];
  2352. SetTextColor(mdc, rgb);
  2353. if (HIWORD(TraySave.iUnit))
  2354. dwByte *= 8;
  2355. float f_byte = (float)dwByte;
  2356. if (dwByte < 1000 && LOWORD(TraySave.iUnit) == 0)
  2357. wsprintf(sz, L"%s%dB", szT, dwByte);
  2358. else if ((dwByte < 1024000 || (dwByte < 1000000 && HIWORD(TraySave.iUnit))) && LOWORD(TraySave.iUnit) != 2)
  2359. {
  2360. if (HIWORD(TraySave.iUnit))
  2361. f_byte /= 1000;
  2362. else
  2363. f_byte /= 1024;
  2364. int k_byte = int(f_byte * 100);
  2365. if (f_byte >= 100)
  2366. wsprintf(sz, L"%s%dK", szT, k_byte / 100);
  2367. else if (f_byte >= 10)
  2368. wsprintf(sz, L"%s%d.%.1dK", szT, k_byte / 100, (k_byte / 10) % 10);
  2369. else
  2370. wsprintf(sz, L"%s%d.%.2dK", szT, k_byte / 100, k_byte % 100);
  2371. }
  2372. else if (dwByte < 1048576000 || (dwByte < 1000000000 && HIWORD(TraySave.iUnit)))
  2373. {
  2374. if (HIWORD(TraySave.iUnit))
  2375. f_byte /= 1000000;
  2376. else
  2377. f_byte /= 1048576;
  2378. int m_byte = int(f_byte * 100);
  2379. if (f_byte >= 100)
  2380. wsprintf(sz, L"%s%dM", szT, m_byte / 100);
  2381. else if (f_byte >= 10)
  2382. wsprintf(sz, L"%s%d.%.1dM", szT, m_byte / 100, (m_byte / 10) % 10);
  2383. else
  2384. wsprintf(sz, L"%s%d.%.2dM", szT, m_byte / 100, m_byte % 100);
  2385. }
  2386. else
  2387. {
  2388. if (HIWORD(TraySave.iUnit))
  2389. f_byte /= 1000000000;
  2390. else
  2391. f_byte /= 1073741824;
  2392. int g_byte = int(f_byte * 100);
  2393. if (f_byte >= 100)
  2394. wsprintf(sz, L"%s%dG", szT, g_byte / 100);
  2395. else if (f_byte >= 10)
  2396. wsprintf(sz, L"%s%d.%.1dG", szT, g_byte / 100, (g_byte / 10) % 10);
  2397. else
  2398. wsprintf(sz, L"%s%d.%.2dG", szT, g_byte / 100, g_byte % 100);
  2399. }
  2400. if (HIWORD(TraySave.iUnit))
  2401. lstrlwr(sz, 16);
  2402. if (VTray && TraySave.bMonitorFloat == FALSE)
  2403. DrawShadowText(mdc, sz, lstrlen(sz), lpRect, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2404. else
  2405. DrawShadowText(mdc, sz, lstrlen(sz), lpRect, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2406. }
  2407. BOOL bEvent = FALSE; //
  2408. BOOL SetTrackMouseEvent(HWND hWnd, DWORD dwFlags)
  2409. {
  2410. TRACKMOUSEEVENT csTME;
  2411. csTME.cbSize = sizeof(csTME);
  2412. csTME.dwFlags = dwFlags;
  2413. csTME.hwndTrack = hWnd; // 指定要 追踪 的窗口
  2414. csTME.dwHoverTime = 300; // 鼠标在按钮上停留超过 300ms ,才认为状态为 HOVER
  2415. return TrackMouseEvent(&csTME);
  2416. }
  2417. int iGetAddressTime = 10; //10秒一次获取网卡信息
  2418. INT_PTR CALLBACK TaskBarProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) //任务栏信息窗口过程
  2419. {
  2420. switch (message)
  2421. {
  2422. case WM_INITDIALOG:
  2423. return (INT_PTR)TRUE;
  2424. case WM_COMMAND:
  2425. if (LOWORD(wParam) >= IDC_SELECT_ALL && LOWORD(wParam) <= IDC_SELECT_ALL + 99)
  2426. {
  2427. if (LOWORD(wParam) == IDC_SELECT_ALL)
  2428. TraySave.AdpterName[0] = L'\0';
  2429. else
  2430. {
  2431. int x = LOWORD(wParam) - IDC_SELECT_ALL;
  2432. PIP_ADAPTER_ADDRESSES paa;
  2433. paa = &piaa[0];
  2434. int n = 1;
  2435. while (paa)
  2436. {
  2437. if (paa->IfType != IF_TYPE_SOFTWARE_LOOPBACK && paa->IfType != IF_TYPE_TUNNEL)
  2438. {
  2439. if (n == x)
  2440. {
  2441. lstrcpyA(TraySave.AdpterName, paa->AdapterName);
  2442. break;
  2443. }
  2444. n++;
  2445. }
  2446. paa = paa->Next;
  2447. }
  2448. }
  2449. WriteReg();
  2450. m_last_in_bytes = 0;
  2451. m_last_out_bytes = 0;
  2452. s_in_byte = 0;
  2453. s_out_byte = 0;
  2454. }
  2455. break;
  2456. case WM_MOUSEMOVE:
  2457. if (bEvent == FALSE && TraySave.bMonitorTips)
  2458. {
  2459. SetTrackMouseEvent(hTaskBar, TME_LEAVE | TME_HOVER);
  2460. bEvent = TRUE;
  2461. }
  2462. break;
  2463. case WM_MOUSEHOVER:
  2464. {
  2465. if (!IsWindowVisible(hTaskTips))
  2466. {
  2467. if (!IsWindow(hTaskTips))
  2468. {
  2469. hTaskTips = ::CreateDialog(hInst, MAKEINTRESOURCE(IDD_TIPS), NULL, (DLGPROC)TaskTipsProc);
  2470. SetLayeredWindowAttributes(hTaskTips, 0, 255, LWA_ALPHA);
  2471. }
  2472. nProcess = GetProcessMemUsage();
  2473. if (pProcessTime == NULL)
  2474. {
  2475. pProcessTime = (PROCESSTIME *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof PROCESSTIME *(nProcess + 32));
  2476. memset(pProcessTime, 0, sizeof(PROCESSTIME) * (nProcess + 32));
  2477. }
  2478. GetProcessCpuUsage();
  2479. HDC mdc = GetDC(hMain);
  2480. TraySave.TipsFont.lfHeight = DPI(TraySave.TipsFontSize);
  2481. HFONT hTipsFont = CreateFontIndirect(&TraySave.TipsFont); //创建字体
  2482. HFONT oldFont = (HFONT)SelectObject(mdc, hTipsFont);
  2483. SIZE tSize;
  2484. ::GetTextExtentPoint(mdc, L"虚拟内存虚拟内存虚拟内存虚拟内存虚拟内存虚拟内存虚拟内存虚拟内存", 32, &tSize);
  2485. SelectObject(mdc, oldFont);
  2486. DeleteObject(hTipsFont);
  2487. ::ReleaseDC(hMain, mdc);
  2488. int x, y, w, h;
  2489. w = tSize.cx;
  2490. wTipsHeight = tSize.cy;
  2491. h = wTipsHeight * (nTraffic + 8);
  2492. RECT wrc, src;
  2493. GetWindowRect(hDlg, &wrc);
  2494. GetScreenRect(hDlg, &src, TRUE);
  2495. if (wrc.bottom + h > src.bottom)
  2496. y = wrc.top - h;
  2497. else
  2498. y = wrc.bottom;
  2499. if (wrc.right - (wrc.right - wrc.left) / 2 + w / 2 > src.right)
  2500. x = src.right - w;
  2501. else if (wrc.right - (wrc.right - wrc.left) / 2 - w / 2 < src.left)
  2502. x = src.left;
  2503. else
  2504. x = wrc.right - (wrc.right - wrc.left) / 2 - w / 2;
  2505. SetWindowPos(hTaskTips, HWND_TOPMOST, x, y, w, h, SWP_NOACTIVATE | SWP_SHOWWINDOW);
  2506. HRGN hRgn = CreateRoundRectRgn(0, 0, w + 1, h + 1, 3, 3);
  2507. SetWindowRgn(hTaskTips, hRgn, FALSE);
  2508. }
  2509. }
  2510. break;
  2511. case WM_MOUSELEAVE:
  2512. if (TraySave.bMonitorFloat)
  2513. {
  2514. RECT wrc;
  2515. GetWindowRect(hDlg, &wrc);
  2516. TraySave.dMonitorPoint.x = wrc.left;
  2517. TraySave.dMonitorPoint.y = wrc.top;
  2518. WriteReg();
  2519. bTaskBarMoveing = FALSE;
  2520. // SetTimer(hDlg, 6, 100, NULL);
  2521. }
  2522. POINT pt;
  2523. GetCursorPos(&pt);
  2524. if (WindowFromPoint(pt) != hTaskTips)
  2525. {
  2526. if (pProcessTime != NULL)
  2527. {
  2528. HeapFree(GetProcessHeap(), 0, pProcessTime);
  2529. pProcessTime = NULL;
  2530. }
  2531. DestroyWindow(hTaskTips);
  2532. // ShowWindow(hTaskTips, SW_HIDE);
  2533. SetTimer(hMain, 11, 1000, NULL);
  2534. }
  2535. else
  2536. SetTrackMouseEvent(hTaskTips, TME_LEAVE);
  2537. bEvent = FALSE;
  2538. break;
  2539. case WM_RBUTTONDOWN:
  2540. {
  2541. POINT pt;
  2542. GetCursorPos(&pt);
  2543. ScreenToClient(hDlg, &pt);
  2544. if (TraySave.bMonitorTraffic && pt.x < wTraffic && pt.y < wHeight * 2)
  2545. {
  2546. HMENU hMenu = LoadMenu(hInst, MAKEINTRESOURCEW(IDR_MENU));
  2547. HMENU subMenu = GetSubMenu(hMenu, 0);
  2548. PIP_ADAPTER_ADDRESSES paa;
  2549. paa = &piaa[0];
  2550. int n = 1;
  2551. CheckMenuRadioItem(subMenu, IDC_SELECT_ALL, IDC_SELECT_ALL + 99, IDC_SELECT_ALL, MF_BYCOMMAND);
  2552. while (paa)
  2553. {
  2554. if (paa->IfType != IF_TYPE_SOFTWARE_LOOPBACK && paa->IfType != IF_TYPE_TUNNEL)
  2555. {
  2556. AppendMenu(subMenu, MF_BYCOMMAND, IDC_SELECT_ALL + n, paa->FriendlyName);
  2557. if (lstrcmpA(paa->AdapterName, TraySave.AdpterName) == 0)
  2558. CheckMenuRadioItem(subMenu, IDC_SELECT_ALL, IDC_SELECT_ALL + 99, IDC_SELECT_ALL + n, MF_BYCOMMAND);
  2559. n++;
  2560. }
  2561. paa = paa->Next;
  2562. }
  2563. POINT point;
  2564. GetCursorPos(&point);
  2565. SetTimer(hDlg, 5, 1200, NULL);
  2566. TrackPopupMenu(subMenu, TPM_LEFTALIGN, point.x, point.y, NULL, hDlg, NULL);
  2567. DestroyMenu(hMenu);
  2568. }
  2569. else
  2570. OpenSetting();
  2571. return TRUE;
  2572. }
  2573. break;
  2574. case WM_LBUTTONDOWN:
  2575. {
  2576. if (TraySave.bMonitorFloat)
  2577. {
  2578. bTaskBarMoveing = TRUE;
  2579. PostMessage(hDlg, WM_NCLBUTTONDOWN, HTCAPTION, lParam);
  2580. }
  2581. return TRUE;
  2582. /*
  2583. else
  2584. RunProcess(szNetCpl);
  2585. */
  2586. }
  2587. break;
  2588. case WM_LBUTTONUP:
  2589. if (!TraySave.bMonitorFloat)
  2590. {
  2591. ShowWindow(hDlg, SW_HIDE);
  2592. SetTimer(hDlg, 9, 100, NULL);
  2593. return TRUE;
  2594. }
  2595. break;
  2596. case WM_TIMER:
  2597. if (wParam == 9)
  2598. {
  2599. KillTimer(hDlg, wParam);
  2600. POINT pt;
  2601. GetCursorPos(&pt);
  2602. mouse_event(MOUSEEVENTF_LEFTDOWN, pt.x, pt.y, 0, 0);
  2603. mouse_event(MOUSEEVENTF_LEFTUP, pt.x, pt.y, 0, 0);
  2604. ShowWindow(hDlg, SW_SHOWNOACTIVATE);
  2605. }
  2606. else if (wParam == 5) ////////////////////////////////////////////////光标移出弹出式菜单自动隐藏菜单
  2607. {
  2608. HWND hMenu = FindWindow(L"#32768", NULL);
  2609. POINT pt;
  2610. GetCursorPos(&pt);
  2611. if (WindowFromPoint(pt) != hMenu)
  2612. {
  2613. KillTimer(hDlg, wParam);
  2614. PostMessage(hMenu, WM_CLOSE, NULL, NULL);
  2615. }
  2616. }
  2617. else if (wParam == 3)
  2618. {
  2619. if (IsWindowVisible(hTaskTips))
  2620. {
  2621. nProcess = GetProcessMemUsage();
  2622. GetProcessCpuUsage();
  2623. }
  2624. if (TraySave.bMonitorUsage)
  2625. {
  2626. iCPU = GetCPUUseRate();
  2627. GlobalMemoryStatusEx(&MemoryStatusEx);
  2628. }
  2629. if (TraySave.bMonitorTemperature)
  2630. {
  2631. if (bRing0)
  2632. {
  2633. iTemperature1 = GetCpuTemp(1);
  2634. iTemperature2 = GetCpuTemp(dNumProcessor);
  2635. }
  2636. int iATITemperature = 0;
  2637. int iNVTemperature = 0;
  2638. if (hNVDLL)
  2639. {
  2640. NV_GPU_THERMAL_SETTINGS currentTemp; //获取温度的数据结构
  2641. currentTemp.version = NV_GPU_THERMAL_SETTINGS_VER; //一定要设置,不然调用获取温度函数时候会出错
  2642. for (int GpuIndex = 0; GpuIndex < 4; GpuIndex++)
  2643. {
  2644. if (NvAPI_GPU_GetThermalSettings(hPhysicalGpu[GpuIndex], 15, &currentTemp) == 0)
  2645. {
  2646. iNVTemperature = currentTemp.sensor[0].currentTemp;
  2647. break;
  2648. }
  2649. }
  2650. }
  2651. if (hATIDLL)
  2652. {
  2653. adlTemperature.iSize = sizeof(ADLTemperature);
  2654. ADL_Overdrive5_Temperature_Get(0, 0, &adlTemperature);
  2655. iATITemperature = adlTemperature.iTemperature / 1000;
  2656. }
  2657. if (iATITemperature != 0 || iNVTemperature != 0)
  2658. {
  2659. if (iATITemperature > iNVTemperature)
  2660. iTemperature2 = iATITemperature;
  2661. else
  2662. iTemperature2 = iNVTemperature;
  2663. }
  2664. }
  2665. if (TraySave.bMonitorTraffic)
  2666. {
  2667. if (hIphlpapi == NULL)
  2668. {
  2669. hIphlpapi = LoadLibrary(L"iphlpapi.dll");
  2670. if (hIphlpapi)
  2671. {
  2672. GetAdaptersAddressesT = (pfnGetAdaptersAddresses)GetProcAddress(hIphlpapi, "GetAdaptersAddresses");
  2673. GetIfTableT = (pfnGetIfTable)GetProcAddress(hIphlpapi, "GetIfTable");
  2674. }
  2675. }
  2676. if (hIphlpapi)
  2677. {
  2678. PIP_ADAPTER_ADDRESSES paa;
  2679. if (iGetAddressTime == 10)
  2680. {
  2681. // DWORD odwIPSize = dwIPSize;
  2682. dwIPSize = 0;
  2683. if (GetAdaptersAddressesT(AF_INET, 0, 0, piaa, &dwIPSize) == ERROR_BUFFER_OVERFLOW)
  2684. {
  2685. // if (dwIPSize != odwIPSize)
  2686. {
  2687. HeapFree(GetProcessHeap(), 0, piaa);
  2688. int n = 0;
  2689. piaa = (PIP_ADAPTER_ADDRESSES)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwIPSize);
  2690. if (GetAdaptersAddressesT(AF_INET, 0, 0, piaa, &dwIPSize) == ERROR_SUCCESS)
  2691. {
  2692. paa = &piaa[0];
  2693. while (paa)
  2694. {
  2695. if (paa->IfType != IF_TYPE_SOFTWARE_LOOPBACK && paa->IfType != IF_TYPE_TUNNEL)
  2696. {
  2697. ++n;
  2698. }
  2699. paa = paa->Next;
  2700. }
  2701. if (n != nTraffic)
  2702. {
  2703. HeapFree(GetProcessHeap(), 0, traffic);
  2704. nTraffic = n;
  2705. traffic = (TRAFFIC *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, nTraffic * sizeof TRAFFIC);
  2706. }
  2707. }
  2708. }
  2709. }
  2710. iGetAddressTime = 0;
  2711. }
  2712. else
  2713. iGetAddressTime++;
  2714. #if 0
  2715. PIP_ADAPTER_INFO pai;
  2716. if (GetAdaptersInfo(ipinfo, &dwIPSize) == ERROR_BUFFER_OVERFLOW)
  2717. {
  2718. free(ipinfo);
  2719. ipinfo = (PIP_ADAPTER_INFO)malloc(dwIPSize);
  2720. GetAdaptersInfo(ipinfo, &dwIPSize);
  2721. pai = &ipinfo[0];
  2722. nTraffic = 0;
  2723. while (pai)
  2724. {
  2725. ++nTraffic;
  2726. pai = pai->Next;
  2727. }
  2728. free(traffic);
  2729. traffic = (TRAFFIC*)malloc(nTraffic * sizeof TRAFFIC);
  2730. }
  2731. #endif
  2732. if (GetIfTableT(mi, &dwMISize, FALSE) == ERROR_INSUFFICIENT_BUFFER)
  2733. {
  2734. dwMISize += sizeof MIB_IFROW * 2;
  2735. HeapFree(GetProcessHeap(), 0, mi);
  2736. mi = (MIB_IFTABLE *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwMISize);
  2737. GetIfTableT(mi, &dwMISize, FALSE);
  2738. }
  2739. DWORD m_in_bytes = 0;
  2740. DWORD m_out_bytes = 0;
  2741. for (DWORD i = 0; i < mi->dwNumEntries; i++)
  2742. {
  2743. int l = 0;
  2744. paa = &piaa[0];
  2745. while (paa)
  2746. {
  2747. if (paa->IfType != IF_TYPE_SOFTWARE_LOOPBACK && paa->IfType != IF_TYPE_TUNNEL)
  2748. {
  2749. if (paa->IfIndex == mi->table[i].dwIndex)
  2750. {
  2751. traffic[l].in_byte = (mi->table[i].dwInOctets - traffic[l].in_bytes) * 8;
  2752. traffic[l].out_byte = (mi->table[i].dwOutOctets - traffic[l].out_bytes) * 8;
  2753. traffic[l].in_bytes = mi->table[i].dwInOctets;
  2754. traffic[l].out_bytes = mi->table[i].dwOutOctets;
  2755. PIP_ADAPTER_UNICAST_ADDRESS pUnicast = paa->FirstUnicastAddress;
  2756. // char IP[130];
  2757. while (pUnicast)
  2758. {
  2759. if (AF_INET == pUnicast->Address.lpSockaddr->sa_family) // IPV4 地址,使用 IPV4 转换
  2760. {
  2761. void *pAddr = &((sockaddr_in *)pUnicast->Address.lpSockaddr)->sin_addr;
  2762. byte *bp = (byte *)pAddr;
  2763. wsprintf(traffic[l].IP4, L"%d.%d.%d.%d", bp[0], bp[1], bp[2], bp[3]);
  2764. break;
  2765. }
  2766. // else if (AF_INET6 == pUnicast->Address.lpSockaddr->sa_family)// IPV6 地址,使用 IPV6 转换
  2767. // inet_ntop(PF_INET6, &((sockaddr_in6*)pUnicast->Address.lpSockaddr)->sin6_addr, IP, sizeof(IP));
  2768. pUnicast = pUnicast->Next;
  2769. }
  2770. // MultiByteToWideChar(CP_ACP, 0, IP, 15, traffic[l].IP4, 15);
  2771. traffic[l].FriendlyName = paa->FriendlyName;
  2772. traffic[l].AdapterName = paa->AdapterName;
  2773. if (lstrlen(paa->FriendlyName) > 19)
  2774. {
  2775. paa->FriendlyName[16] = L'.';
  2776. paa->FriendlyName[17] = L'.';
  2777. paa->FriendlyName[18] = L'.';
  2778. paa->FriendlyName[19] = L'\0';
  2779. }
  2780. // wcsncpy_s(traffic[l].FriendlyName, 24, paa->FriendlyName,24);
  2781. if (TraySave.AdpterName[0] == L'\0' || lstrcmpA(paa->AdapterName, TraySave.AdpterName) == 0)
  2782. {
  2783. m_in_bytes += mi->table[i].dwInOctets;
  2784. m_out_bytes += mi->table[i].dwOutOctets;
  2785. }
  2786. }
  2787. ++l;
  2788. }
  2789. paa = paa->Next;
  2790. }
  2791. }
  2792. if (m_last_in_bytes != 0)
  2793. {
  2794. s_in_byte = m_in_bytes - m_last_in_bytes;
  2795. s_out_byte = m_out_bytes - m_last_out_bytes;
  2796. }
  2797. m_last_out_bytes = m_out_bytes;
  2798. m_last_in_bytes = m_in_bytes;
  2799. }
  2800. }
  2801. else
  2802. {
  2803. if (hIphlpapi)
  2804. {
  2805. FreeLibrary(hIphlpapi);
  2806. hIphlpapi = NULL;
  2807. }
  2808. }
  2809. if (TraySave.bSound)
  2810. {
  2811. if ((TraySave.dNumValues[8] != 0 && (s_in_byte > TraySave.dNumValues[8] || s_out_byte > TraySave.dNumValues[8]))
  2812. || (TraySave.dNumValues[9] != 0 && ((DWORD)iTemperature1 > TraySave.dNumValues[9] || (DWORD)iTemperature2 > TraySave.dNumValues[9]))
  2813. || (TraySave.dNumValues[10] != 0 && (DWORD)iCPU > TraySave.dNumValues[10])
  2814. || (TraySave.dNumValues[11] != 0 && MemoryStatusEx.dwMemoryLoad > TraySave.dNumValues[11]))
  2815. {
  2816. MessageBeep(MB_ICONHAND);
  2817. //PlaySound((LPCWSTR)SND_ALIAS_SYSTEMQUESTION, NULL, SND_ASYNC | SND_ALIAS_ID);
  2818. }
  2819. }
  2820. ::InvalidateRect(hTaskBar, NULL, TRUE);
  2821. }
  2822. case WM_ERASEBKGND:
  2823. {
  2824. // PAINTSTRUCT ps;
  2825. HDC hdc = (HDC)wParam; //BeginPaint(hDlg, &ps);
  2826. RECT rc;
  2827. GetClientRect(hDlg, &rc);
  2828. HDC mdc = CreateCompatibleDC(hdc);
  2829. HBITMAP hMemBmp = CreateCompatibleBitmap(hdc, rc.right - rc.left, rc.bottom - rc.top);
  2830. HBITMAP oldBmp = (HBITMAP)SelectObject(mdc, hMemBmp);
  2831. if (TraySave.cMonitorColor[0] != 0)
  2832. {
  2833. HBRUSH hb = CreateSolidBrush(TraySave.cMonitorColor[0]);
  2834. FillRect(mdc, &rc, hb);
  2835. DeleteObject(hb);
  2836. }
  2837. // if (bErasebkgnd)
  2838. {
  2839. if (VTray && TraySave.bMonitorFloat == FALSE)
  2840. {
  2841. /*
  2842. int s = (rc.right - rc.left - DPI(54)) / 2;
  2843. rc.left += s;
  2844. rc.right -= s;
  2845. */
  2846. }
  2847. else
  2848. {
  2849. InflateRect(&rc, -2, 0);
  2850. }
  2851. HFONT oldFont = (HFONT)SelectObject(mdc, hFont);
  2852. WCHAR sz[16];
  2853. SetBkMode(mdc, TRANSPARENT);
  2854. COLORREF rgb;
  2855. if (TraySave.bMonitorTraffic)
  2856. {
  2857. RECT crc = rc;
  2858. if (VTray && TraySave.bMonitorFloat == FALSE)
  2859. {
  2860. crc.bottom = wHeight;
  2861. DrawTraffic(mdc, &crc, s_out_byte, FALSE);
  2862. OffsetRect(&crc, 0, wHeight);
  2863. DrawTraffic(mdc, &crc, s_in_byte, TRUE);
  2864. }
  2865. else
  2866. {
  2867. crc.bottom /= 2;
  2868. DrawTraffic(mdc, &crc, s_out_byte, FALSE);
  2869. OffsetRect(&crc, 0, crc.bottom);
  2870. DrawTraffic(mdc, &crc, s_in_byte, TRUE);
  2871. }
  2872. }
  2873. if (TraySave.bMonitorTemperature)
  2874. {
  2875. RECT crc = rc;
  2876. if (VTray && TraySave.bMonitorFloat == FALSE)
  2877. {
  2878. if (TraySave.bMonitorTraffic)
  2879. crc.top = wHeight * 2;
  2880. crc.bottom = crc.top + wHeight;
  2881. }
  2882. else
  2883. {
  2884. if (TraySave.bMonitorTraffic)
  2885. crc.left = wTraffic;
  2886. else
  2887. crc.left = 0;
  2888. crc.bottom /= 2;
  2889. }
  2890. if (bRing0)
  2891. {
  2892. if (iTemperature1 <= TraySave.dNumValues[2])
  2893. rgb = TraySave.cMonitorColor[4];
  2894. else if (iTemperature1 <= TraySave.dNumValues[3])
  2895. rgb = TraySave.cMonitorColor[5];
  2896. else
  2897. rgb = TraySave.cMonitorColor[6];
  2898. SetTextColor(mdc, rgb);
  2899. if (TraySave.iMonitorSimple == 1)
  2900. wsprintf(sz, L"%.2d℃", iTemperature1);
  2901. else if (TraySave.iMonitorSimple == 2)
  2902. wsprintf(sz, L"%.2d", iTemperature1);
  2903. else
  2904. wsprintf(sz, L"%s%.2d%s", TraySave.szTemperatureCPU, iTemperature1, TraySave.szTemperatureCPUUnit);
  2905. if (VTray && TraySave.bMonitorFloat == FALSE)
  2906. DrawShadowText(mdc, sz, lstrlen(sz), &crc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2907. else
  2908. DrawShadowText(mdc, sz, lstrlen(sz), &crc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2909. }
  2910. if (bRing0)
  2911. {
  2912. if (VTray && TraySave.bMonitorFloat == FALSE)
  2913. OffsetRect(&crc, 0, wHeight);
  2914. else
  2915. OffsetRect(&crc, 0, crc.bottom - crc.top);
  2916. }
  2917. else
  2918. {
  2919. if (VTray && TraySave.bMonitorFloat == FALSE)
  2920. {
  2921. // crc.bottom += wHeight;
  2922. }
  2923. else
  2924. crc.bottom += (crc.bottom - crc.top);
  2925. }
  2926. if (iTemperature2 <= TraySave.dNumValues[2])
  2927. rgb = TraySave.cMonitorColor[4];
  2928. else if (iTemperature2 <= TraySave.dNumValues[3])
  2929. rgb = TraySave.cMonitorColor[5];
  2930. else
  2931. rgb = TraySave.cMonitorColor[6];
  2932. SetTextColor(mdc, rgb);
  2933. if (TraySave.iMonitorSimple == 0)
  2934. wsprintf(sz, L"%s%.2d%s", TraySave.szTemperatureGPU, iTemperature2, TraySave.szTemperatureGPUUnit);
  2935. else if (TraySave.iMonitorSimple == 1)
  2936. wsprintf(sz, L"%.2d℃", iTemperature2);
  2937. else
  2938. wsprintf(sz, L"%.2d", iTemperature2);
  2939. if (VTray && TraySave.bMonitorFloat == FALSE)
  2940. DrawShadowText(mdc, sz, lstrlen(sz), &crc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2941. else
  2942. DrawShadowText(mdc, sz, lstrlen(sz), &crc, DT_LEFT | DT_VCENTER | DT_SINGLELINE);
  2943. }
  2944. if (TraySave.bMonitorUsage)
  2945. {
  2946. if (iCPU <= TraySave.dNumValues[4])
  2947. rgb = TraySave.cMonitorColor[4];
  2948. else if (iCPU <= TraySave.dNumValues[5])
  2949. rgb = TraySave.cMonitorColor[5];
  2950. else
  2951. rgb = TraySave.cMonitorColor[6];
  2952. SetTextColor(mdc, rgb);
  2953. /*
  2954. if(bRing0)
  2955. swprintf_s(sz, 16, L"%.2d%%", iCPU);
  2956. else
  2957. */
  2958. if (TraySave.iMonitorSimple == 1)
  2959. wsprintf(sz, L"%.2d%%", iCPU);
  2960. else if (TraySave.iMonitorSimple == 2)
  2961. wsprintf(sz, L"%.2d", iCPU);
  2962. else
  2963. wsprintf(sz, L"%s%.2d%s", TraySave.szUsageCPU, iCPU, TraySave.szUsageCPUUnit);
  2964. int sLen = lstrlen(sz);
  2965. RECT crc = rc;
  2966. if (VTray && TraySave.bMonitorFloat == FALSE)
  2967. {
  2968. if (TraySave.bMonitorTraffic)
  2969. crc.top = wHeight * 2;
  2970. if (TraySave.bMonitorTemperature)
  2971. {
  2972. crc.top += wHeight;
  2973. if (bRing0)
  2974. crc.top += wHeight;
  2975. }
  2976. crc.bottom = crc.top + wHeight;
  2977. DrawShadowText(mdc, sz, sLen, &crc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  2978. }
  2979. else
  2980. {
  2981. crc.bottom /= 2;
  2982. DrawShadowText(mdc, sz, sLen, &crc, DT_RIGHT | DT_VCENTER | DT_SINGLELINE);
  2983. }
  2984. /*
  2985. if(bRing0)
  2986. swprintf_s(sz, 16, L"%.2d%%", MemoryStatusEx.dwMemoryLoad);
  2987. else
  2988. */
  2989. if (TraySave.iMonitorSimple == 1)
  2990. wsprintf(sz, L"%.2d%%", MemoryStatusEx.dwMemoryLoad);
  2991. else if (TraySave.iMonitorSimple == 2)
  2992. wsprintf(sz, L"%.2d", MemoryStatusEx.dwMemoryLoad);
  2993. else
  2994. wsprintf(sz, L"%s%.2d%s", TraySave.szUsageMEM, MemoryStatusEx.dwMemoryLoad, TraySave.szUsageMEMUnit);
  2995. sLen = lstrlen(sz);
  2996. if (MemoryStatusEx.dwMemoryLoad <= TraySave.dNumValues[6])
  2997. rgb = TraySave.cMonitorColor[4];
  2998. else if (MemoryStatusEx.dwMemoryLoad <= TraySave.dNumValues[7])
  2999. rgb = TraySave.cMonitorColor[5];
  3000. else
  3001. rgb = TraySave.cMonitorColor[6];
  3002. SetTextColor(mdc, rgb);
  3003. if (VTray && TraySave.bMonitorFloat == FALSE)
  3004. {
  3005. OffsetRect(&crc, 0, wHeight);
  3006. DrawShadowText(mdc, sz, (int)sLen, &crc, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
  3007. }
  3008. else
  3009. {
  3010. OffsetRect(&crc, 0, crc.bottom);
  3011. DrawShadowText(mdc, sz, (int)sLen, &crc, DT_RIGHT | DT_VCENTER | DT_SINGLELINE);
  3012. }
  3013. }
  3014. SelectObject(mdc, oldFont);
  3015. }
  3016. // GetClientRect(hDlg, &rc);
  3017. InflateRect(&rc, 2, 0);
  3018. /*
  3019. BLENDFUNCTION bf1;
  3020. bf1.BlendOp = AC_SRC_OVER;
  3021. bf1.BlendFlags = 0;
  3022. bf1.SourceConstantAlpha = 88;
  3023. bf1.AlphaFormat = AC_SRC_ALPHA;
  3024. ::AlphaBlend(hdc, 0, 0, rc.right - rc.left, rc.bottom - rc.top, mdc, 0, 0, rc.right - rc.left, rc.bottom - rc.top, bf1);
  3025. */
  3026. if (TraySave.cMonitorColor[0] != RGB(0, 0, 0))
  3027. {
  3028. BYTE *lpvBits = NULL;
  3029. BITMAPINFO binfo;
  3030. memset(&binfo, 0, sizeof(BITMAPINFO));
  3031. binfo.bmiHeader.biBitCount = 32; //每个像素多少位,也可直接写24(RGB)或者32(RGBA)
  3032. binfo.bmiHeader.biCompression = 0;
  3033. binfo.bmiHeader.biHeight = rc.bottom - rc.top;
  3034. binfo.bmiHeader.biPlanes = 1;
  3035. binfo.bmiHeader.biSizeImage = (rc.bottom - rc.top) * (rc.right - rc.left) * 4;
  3036. binfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  3037. binfo.bmiHeader.biWidth = rc.right - rc.left;
  3038. lpvBits = (BYTE *)HeapAlloc(GetProcessHeap(), HEAP_NO_SERIALIZE, binfo.bmiHeader.biSizeImage);
  3039. // GetDIBits(mdc, hMemBmp, 0, rc.bottom - rc.top, lpvBits, &bmpInfo, DIB_RGB_COLORS);
  3040. GetDIBits(mdc, hMemBmp, 0, rc.bottom - rc.top, lpvBits, &binfo, DIB_RGB_COLORS);
  3041. for (DWORD i = 0; i < binfo.bmiHeader.biSizeImage - 4; i += 4)
  3042. {
  3043. if (lpvBits[i] > 3 || lpvBits[i + 1] != 0 || lpvBits[i + 2] != 0)
  3044. lpvBits[i + 3] = 0x80;
  3045. }
  3046. SetDIBits(mdc, hMemBmp, 0, rc.bottom - rc.top, lpvBits, &binfo, DIB_RGB_COLORS);
  3047. HeapFree(GetProcessHeap(), 0, lpvBits);
  3048. }
  3049. BitBlt(hdc, 0, 0, rc.right - rc.left, rc.bottom - rc.top, mdc, 0, 0, SRCCOPY);
  3050. SelectObject(mdc, oldBmp);
  3051. DeleteObject(hMemBmp);
  3052. DeleteDC(mdc);
  3053. // EndPaint(hDlg, &ps);
  3054. return TRUE;
  3055. }
  3056. break;
  3057. }
  3058. return FALSE;
  3059. }
  3060. INT_PTR CALLBACK MainProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) //主窗口过程
  3061. {
  3062. UNREFERENCED_PARAMETER(lParam);
  3063. switch (message)
  3064. {
  3065. case WM_INITDIALOG:
  3066. return (INT_PTR)TRUE;
  3067. /*
  3068. case WM_ENDSESSION:
  3069. if (lParam == ENDSESSION_LOGOFF)
  3070. {
  3071. DestroyWindow(hTray);
  3072. RunProcess(NULL);
  3073. return TRUE;
  3074. }
  3075. break;
  3076. */
  3077. case WM_TRAYS:
  3078. OpenSetting();
  3079. break;
  3080. case WM_DPICHANGED:
  3081. {
  3082. iDPI = LOWORD(wParam);
  3083. SetTimer(hDlg, 8, 1000, NULL);
  3084. }
  3085. break;
  3086. case WM_CLOSE:
  3087. {
  3088. KillTimer(hDlg, 6);
  3089. KillTimer(hDlg, 3);
  3090. SendMessage(hReBarWnd, WM_SETREDRAW, TRUE, 0);
  3091. HWND hSecondaryTray;
  3092. hSecondaryTray = FindWindow(szSecondaryTray, NULL);
  3093. while (hSecondaryTray)
  3094. {
  3095. HWND hSReBarWnd = FindWindowEx(hSecondaryTray, 0, L"WorkerW", NULL);
  3096. SendMessage(hSReBarWnd, WM_SETREDRAW, TRUE, 0);
  3097. ShowWindow(hSReBarWnd, SW_SHOWNOACTIVATE);
  3098. hSecondaryTray = FindWindowEx(NULL, hSecondaryTray, szSecondaryTray, NULL);
  3099. }
  3100. ShowWindow(hTaskListWnd, SW_SHOW);
  3101. PostQuitMessage(0);
  3102. }
  3103. break;
  3104. case WM_TIMER:
  3105. {
  3106. if (wParam == 8)
  3107. {
  3108. KillTimer(hDlg, wParam);
  3109. SetWH();
  3110. }
  3111. else if (wParam == 11) //释放内存
  3112. {
  3113. KillTimer(hDlg, wParam);
  3114. SetTimer(hDlg, wParam, 60000, NULL);
  3115. HANDLE hProcess = GetCurrentProcess();
  3116. SetProcessWorkingSetSize(hProcess, -1, -1);
  3117. EmptyWorkingSet(hProcess);
  3118. }
  3119. else if (wParam == 6) //处理任务栏图标与信息窗口
  3120. {
  3121. if (TraySave.bMonitor)
  3122. {
  3123. // if (!bTaskBarMoveing)
  3124. // AdjustWindowPos();
  3125. if (IsWindowVisible(hTaskTips))
  3126. ::InvalidateRect(hTaskTips, NULL, TRUE);
  3127. }
  3128. if (TraySave.iPos != 0 || TraySave.bMonitor)
  3129. {
  3130. // if (TraySave.bTaskIcon == FALSE)
  3131. {
  3132. GetShellAllWnd();
  3133. SetTaskBarPos(hTaskListWnd, hTray, hTaskWnd, hReBarWnd, TRUE);
  3134. }
  3135. HWND hSecondaryTray;
  3136. hSecondaryTray = FindWindow(szSecondaryTray, NULL);
  3137. while (hSecondaryTray)
  3138. {
  3139. HWND hSReBarWnd = FindWindowEx(hSecondaryTray, 0, L"WorkerW", NULL);
  3140. if (hSReBarWnd)
  3141. {
  3142. HWND hSTaskListWnd = FindWindowEx(hSReBarWnd, NULL, L"MSTaskListWClass", NULL);
  3143. if (hSTaskListWnd)
  3144. {
  3145. SetTaskBarPos(hSTaskListWnd, hSecondaryTray, hSReBarWnd, hSReBarWnd, FALSE);
  3146. }
  3147. }
  3148. hSecondaryTray = FindWindowEx(NULL, hSecondaryTray, szSecondaryTray, NULL);
  3149. }
  3150. }
  3151. }
  3152. else if (wParam == 3) //处理任务栏风格
  3153. {
  3154. if (TraySave.bMonitor)
  3155. {
  3156. if (!bTaskBarMoveing)
  3157. {
  3158. AdjustWindowPos();
  3159. }
  3160. }
  3161. {
  3162. HWND hTray = FindWindow(szShellTray, NULL);
  3163. if (hTray)
  3164. {
  3165. if (iProject == 0)
  3166. iWindowMode = 0;
  3167. else if (iProject == 1)
  3168. iWindowMode = 1;
  3169. else
  3170. {
  3171. iWindowMode = 0;
  3172. EnumWindows(IsZoomedFunc, (LPARAM)MonitorFromWindow(hTray, MONITOR_DEFAULTTONEAREST));
  3173. }
  3174. SetWindowCompositionAttribute(hTray, TraySave.aMode[iWindowMode], TraySave.dAlphaColor[iWindowMode]);
  3175. LONG_PTR exStyle = GetWindowLongPtr(hTray, GWL_EXSTYLE);
  3176. exStyle |= WS_EX_LAYERED;
  3177. SetWindowLongPtr(hTray, GWL_EXSTYLE, exStyle);
  3178. SetLayeredWindowAttributes(hTray, 0, (BYTE)TraySave.bAlpha[iWindowMode], LWA_ALPHA);
  3179. }
  3180. HWND hSecondaryTray = FindWindow(szSecondaryTray, NULL);
  3181. while (hSecondaryTray)
  3182. {
  3183. if (iProject == 0)
  3184. iWindowMode = 0;
  3185. else if (iProject == 1)
  3186. iWindowMode = 1;
  3187. else
  3188. {
  3189. iWindowMode = 0;
  3190. EnumWindows(IsZoomedFunc, (LPARAM)MonitorFromWindow(hSecondaryTray, MONITOR_DEFAULTTONEAREST));
  3191. }
  3192. SetWindowCompositionAttribute(hSecondaryTray, TraySave.aMode[iWindowMode], TraySave.dAlphaColor[iWindowMode]);
  3193. LONG_PTR exStyle = GetWindowLongPtr(hSecondaryTray, GWL_EXSTYLE);
  3194. exStyle |= WS_EX_LAYERED;
  3195. SetWindowLongPtr(hSecondaryTray, GWL_EXSTYLE, exStyle);
  3196. SetLayeredWindowAttributes(hSecondaryTray, 0, (BYTE)TraySave.bAlpha[iWindowMode], LWA_ALPHA);
  3197. hSecondaryTray = FindWindowEx(NULL, hSecondaryTray, szSecondaryTray, NULL);
  3198. }
  3199. }
  3200. if (TraySave.aMode[0] == ACCENT_DISABLED && TraySave.aMode[1] == ACCENT_DISABLED) //默认则关闭定时器
  3201. KillTimer(hDlg, 3);
  3202. }
  3203. }
  3204. break;
  3205. case WM_IAWENTRAY: //////////////////////////////////////////////////////////////////////////////////通知栏左右键处理
  3206. {
  3207. if (wParam == WM_IAWENTRAY)
  3208. {
  3209. if (lParam == WM_LBUTTONDOWN || lParam == WM_RBUTTONDOWN)
  3210. {
  3211. OpenSetting();
  3212. }
  3213. }
  3214. break;
  3215. }
  3216. break;
  3217. }
  3218. return FALSE;
  3219. }
  3220. INT_PTR CALLBACK SettingProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) //设置窗口过程
  3221. {
  3222. UNREFERENCED_PARAMETER(lParam);
  3223. switch (message)
  3224. {
  3225. case WM_INITDIALOG:
  3226. return (INT_PTR)TRUE;
  3227. case WM_NOTIFY:
  3228. switch (((LPNMHDR)lParam)->code)
  3229. {
  3230. case NM_CLICK:
  3231. case NM_RETURN:
  3232. {
  3233. HWND g_hLink = GetDlgItem(hDlg, IDC_SYSLINK);
  3234. PNMLINK pNMLink = (PNMLINK)lParam;
  3235. LITEM item = pNMLink->item;
  3236. if ((((LPNMHDR)lParam)->hwndFrom == g_hLink) && (item.iLink == 0))
  3237. {
  3238. CloseHandle(pShellExecute(NULL, L"open", L"https://gitee.com/cgbsmy/TrayS", NULL, NULL, SW_SHOW));
  3239. //mailto:cgbsmy@live.com?subject=TrayS
  3240. }
  3241. else
  3242. {
  3243. CloseHandle(pShellExecute(NULL, L"open", L"https://www.52pojie.cn/thread-1182669-1-1.html", NULL, NULL, SW_SHOW));
  3244. }
  3245. break;
  3246. }
  3247. }
  3248. break;
  3249. case WM_HSCROLL: //////////////////////////////////////////////////////////////////////////////////透明度处理
  3250. {
  3251. HWND hSlider = GetDlgItem(hDlg, IDC_SLIDER_ALPHA);
  3252. HWND hSliderB = GetDlgItem(hDlg, IDC_SLIDER_ALPHA_B);
  3253. if (hSlider == (HWND)lParam)
  3254. {
  3255. TraySave.bAlpha[iProject] = (int)SendDlgItemMessage(hDlg, IDC_SLIDER_ALPHA, TBM_GETPOS, 0, 0);
  3256. }
  3257. else if (hSliderB == (HWND)lParam)
  3258. {
  3259. DWORD bAlphaB = (int)SendDlgItemMessage(hDlg, IDC_SLIDER_ALPHA_B, TBM_GETPOS, 0, 0);
  3260. bAlphaB = bAlphaB << 24;
  3261. TraySave.dAlphaColor[iProject] = bAlphaB + (TraySave.dAlphaColor[iProject] & 0xffffff);
  3262. }
  3263. SetTimer(hDlg, 3, 500, NULL);
  3264. break;
  3265. }
  3266. case WM_TIMER:
  3267. if (wParam == 3)
  3268. {
  3269. KillTimer(hDlg, wParam);
  3270. WriteReg();
  3271. }
  3272. break;
  3273. case WM_COMMAND:
  3274. if (HIWORD(wParam) == EN_CHANGE && !bSettingInit)
  3275. {
  3276. if (LOWORD(wParam) >= IDC_EDIT1 && LOWORD(wParam) <= IDC_EDIT12)
  3277. {
  3278. int index = LOWORD(wParam) - IDC_EDIT1;
  3279. TraySave.dNumValues[index] = GetDlgItemInt(hDlg, LOWORD(wParam), NULL, 0);
  3280. if (index == 0 || index == 1 || index == 8)
  3281. TraySave.dNumValues[index] *= 1048576;
  3282. SetTimer(hDlg, 3, 500, NULL);
  3283. }
  3284. else if (LOWORD(wParam) == IDC_EDIT_TIME)
  3285. {
  3286. TraySave.FlushTime = GetDlgItemInt(hDlg, LOWORD(wParam), NULL, 0);
  3287. if (TraySave.aMode[0] != ACCENT_DISABLED || TraySave.aMode[1] != ACCENT_DISABLED)
  3288. SetTimer(hMain, 3, TraySave.FlushTime, NULL);
  3289. SetTimer(hDlg, 3, 500, NULL);
  3290. }
  3291. else if (LOWORD(wParam) >= IDC_EDIT14 && LOWORD(wParam) <= IDC_EDIT23)
  3292. {
  3293. GetDlgItemText(hDlg, IDC_EDIT14, TraySave.szTrafficOut, 8);
  3294. GetDlgItemText(hDlg, IDC_EDIT15, TraySave.szTrafficIn, 8);
  3295. GetDlgItemText(hDlg, IDC_EDIT16, TraySave.szTemperatureCPU, 8);
  3296. GetDlgItemText(hDlg, IDC_EDIT17, TraySave.szTemperatureGPU, 8);
  3297. GetDlgItemText(hDlg, IDC_EDIT18, TraySave.szTemperatureCPUUnit, 4);
  3298. GetDlgItemText(hDlg, IDC_EDIT19, TraySave.szTemperatureGPUUnit, 4);
  3299. GetDlgItemText(hDlg, IDC_EDIT20, TraySave.szUsageCPU, 8);
  3300. GetDlgItemText(hDlg, IDC_EDIT21, TraySave.szUsageMEM, 8);
  3301. GetDlgItemText(hDlg, IDC_EDIT22, TraySave.szUsageCPUUnit, 4);
  3302. GetDlgItemText(hDlg, IDC_EDIT23, TraySave.szUsageMEMUnit, 4);
  3303. SetTimer(hDlg, 3, 1500, NULL);
  3304. if (TraySave.iMonitorSimple == 0)
  3305. SetWH();
  3306. }
  3307. }
  3308. else if (LOWORD(wParam) >= IDC_RADIO_DEFAULT && LOWORD(wParam) <= IDC_RADIO_ACRYLIC)
  3309. {
  3310. if (IsDlgButtonChecked(hDlg, IDC_RADIO_DEFAULT))
  3311. TraySave.aMode[iProject] = ACCENT_DISABLED;
  3312. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_TRANSPARENT))
  3313. TraySave.aMode[iProject] = ACCENT_ENABLE_TRANSPARENTGRADIENT;
  3314. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_BLURBEHIND))
  3315. TraySave.aMode[iProject] = ACCENT_ENABLE_BLURBEHIND;
  3316. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_ACRYLIC))
  3317. TraySave.aMode[iProject] = ACCENT_ENABLE_ACRYLICBLURBEHIND;
  3318. WriteReg();
  3319. if (TraySave.aMode[0] != ACCENT_DISABLED || TraySave.aMode[1] != ACCENT_DISABLED)
  3320. SetTimer(hMain, 3, TraySave.FlushTime, NULL);
  3321. else
  3322. KillTimer(hMain, 3);
  3323. }
  3324. else if (LOWORD(wParam) >= IDC_RADIO_LEFT && LOWORD(wParam) <= IDC_RADIO_RIGHT)
  3325. {
  3326. if (IsDlgButtonChecked(hDlg, IDC_RADIO_LEFT))
  3327. {
  3328. TraySave.iPos = 0;
  3329. }
  3330. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_CENTER))
  3331. {
  3332. TraySave.iPos = 1;
  3333. }
  3334. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_RIGHT))
  3335. {
  3336. TraySave.iPos = 2;
  3337. }
  3338. WriteReg();
  3339. }
  3340. else if (LOWORD(wParam) >= IDC_RADIO_BYTE && LOWORD(wParam) <= IDC_RADIO_MB)
  3341. {
  3342. if (IsDlgButtonChecked(hDlg, IDC_RADIO_AUTO))
  3343. TraySave.iUnit = 0;
  3344. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_KB))
  3345. TraySave.iUnit = 1;
  3346. else if (IsDlgButtonChecked(hDlg, IDC_RADIO_MB))
  3347. TraySave.iUnit = 2;
  3348. if (IsDlgButtonChecked(hDlg, IDC_RADIO_BIT))
  3349. TraySave.iUnit |= 0x10000;
  3350. WriteReg();
  3351. }
  3352. if (LOWORD(wParam) == IDC_RADIO_NORMAL || LOWORD(wParam) == IDC_RADIO_MAXIMIZE)
  3353. {
  3354. if (IsDlgButtonChecked(hDlg, IDC_RADIO_NORMAL))
  3355. iProject = 0;
  3356. else
  3357. iProject = 1;
  3358. if (TraySave.aMode[iProject] == ACCENT_DISABLED)
  3359. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_DEFAULT);
  3360. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_TRANSPARENTGRADIENT)
  3361. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_TRANSPARENT);
  3362. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_BLURBEHIND)
  3363. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_BLURBEHIND);
  3364. else if (TraySave.aMode[iProject] == ACCENT_ENABLE_ACRYLICBLURBEHIND)
  3365. CheckRadioButton(hSetting, IDC_RADIO_DEFAULT, IDC_RADIO_ACRYLIC, IDC_RADIO_ACRYLIC);
  3366. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA, TBM_SETPOS, TRUE, TraySave.bAlpha[iProject]);
  3367. BYTE bAlphaB = TraySave.dAlphaColor[iProject] >> 24;
  3368. SendDlgItemMessage(hSetting, IDC_SLIDER_ALPHA_B, TBM_SETPOS, TRUE, bAlphaB);
  3369. ::InvalidateRect(GetDlgItem(hSetting, IDC_BUTTON_COLOR), NULL, FALSE);
  3370. }
  3371. else if (LOWORD(wParam) == IDC_CHECK_SOUND)
  3372. {
  3373. TraySave.bSound = IsDlgButtonChecked(hDlg, IDC_CHECK_SOUND);
  3374. WriteReg();
  3375. }
  3376. else if (LOWORD(wParam) == IDC_CHECK_TIPS)
  3377. {
  3378. TraySave.bMonitorTips = IsDlgButtonChecked(hDlg, IDC_CHECK_TIPS);
  3379. WriteReg();
  3380. }
  3381. else if (LOWORD(wParam) == IDC_CHECK_TRAYICON)
  3382. {
  3383. TraySave.bTrayIcon = IsDlgButtonChecked(hDlg, IDC_CHECK_TRAYICON);
  3384. WriteReg();
  3385. CloseTaskBar();
  3386. if (TraySave.bTrayIcon)
  3387. pShell_NotifyIcon(NIM_ADD, &nid);
  3388. else
  3389. pShell_NotifyIcon(NIM_DELETE, &nid);
  3390. }
  3391. else if (LOWORD(wParam) == IDC_CHECK_MONITOR)
  3392. {
  3393. TraySave.bMonitor = IsDlgButtonChecked(hDlg, IDC_CHECK_MONITOR);
  3394. WriteReg();
  3395. if (!TraySave.bMonitor)
  3396. {
  3397. CloseTaskBar();
  3398. }
  3399. }
  3400. else if (LOWORD(wParam) == IDC_CHECK_TRAFFIC)
  3401. {
  3402. TraySave.bMonitorTraffic = IsDlgButtonChecked(hDlg, IDC_CHECK_TRAFFIC);
  3403. WriteReg();
  3404. SetWH();
  3405. }
  3406. else if (LOWORD(wParam) == IDC_CHECK_TEMPERATURE)
  3407. {
  3408. TraySave.bMonitorTemperature = IsDlgButtonChecked(hDlg, IDC_CHECK_TEMPERATURE);
  3409. if (TraySave.bMonitorTemperature)
  3410. LoadTemperatureDLL();
  3411. else
  3412. FreeTemperatureDLL();
  3413. WriteReg();
  3414. SetWH();
  3415. }
  3416. else if (LOWORD(wParam) == IDC_CHECK_MONITOR_SIMPLE)
  3417. {
  3418. TraySave.iMonitorSimple = IsDlgButtonChecked(hDlg, IDC_CHECK_MONITOR_SIMPLE);
  3419. WriteReg();
  3420. SetWH();
  3421. }
  3422. else if (LOWORD(wParam) == IDC_CHECK_USAGE)
  3423. {
  3424. TraySave.bMonitorUsage = IsDlgButtonChecked(hDlg, IDC_CHECK_USAGE);
  3425. WriteReg();
  3426. SetWH();
  3427. }
  3428. else if (LOWORD(wParam) == IDC_CHECK_MONITOR_PDH)
  3429. {
  3430. TraySave.bMonitorPDH = IsDlgButtonChecked(hDlg, IDC_CHECK_MONITOR_PDH);
  3431. WriteReg();
  3432. }
  3433. else if (LOWORD(wParam) == IDC_CHECK_MONITOR_LEFT)
  3434. {
  3435. TraySave.bMonitorLeft = IsDlgButtonChecked(hDlg, IDC_CHECK_MONITOR_LEFT);
  3436. WriteReg();
  3437. }
  3438. else if (LOWORD(wParam) == IDC_CHECK_MONITOR_FLOAT)
  3439. {
  3440. TraySave.bMonitorFloat = IsDlgButtonChecked(hDlg, IDC_CHECK_MONITOR_FLOAT);
  3441. WriteReg();
  3442. CloseTaskBar();
  3443. }
  3444. else if (LOWORD(wParam) == IDC_CHECK_TRANSPARENT)
  3445. {
  3446. TraySave.bMonitorTransparent = IsDlgButtonChecked(hDlg, IDC_CHECK_TRANSPARENT);
  3447. OpenTaskBar();
  3448. WriteReg();
  3449. }
  3450. else if (LOWORD(wParam) == IDC_CHECK_AUTORUN)
  3451. {
  3452. if (IsDlgButtonChecked(hDlg, IDC_CHECK_AUTORUN))
  3453. AutoRun(TRUE, TRUE, szAppName);
  3454. else
  3455. AutoRun(TRUE, FALSE, szAppName);
  3456. }
  3457. else if (LOWORD(wParam) == IDC_RESTORE_DEFAULT)
  3458. {
  3459. DeleteFile(szTraySave);
  3460. // RegDeleteKey(HKEY_CURRENT_USER, szSubKey);
  3461. SendMessage(hDlg, WM_COMMAND, IDCANCEL, 0);
  3462. }
  3463. else if (LOWORD(wParam) == IDCANCEL)
  3464. {
  3465. /*
  3466. SendMessage(hMain, WM_TIMER, 11, 1000);
  3467. DestroyWindow(hDlg);
  3468. return (INT_PTR)TRUE;
  3469. */
  3470. // SendMessage(hReBarWnd, WM_SETREDRAW, TRUE, 0);
  3471. bResetRun = TRUE;
  3472. SendMessage(hMain, WM_CLOSE, NULL, NULL);
  3473. return (INT_PTR)TRUE;
  3474. }
  3475. else if (LOWORD(wParam) == IDC_CLOSE)
  3476. {
  3477. SendMessage(hMain, WM_CLOSE, NULL, NULL);
  3478. }
  3479. else if (LOWORD(wParam) == IDC_BUTTON_FONT || LOWORD(wParam) == IDC_BUTTON_TIPS_FONT)
  3480. {
  3481. typedef UINT_PTR(CALLBACK * LPCFHOOKPROC)(HWND, UINT, WPARAM, LPARAM);
  3482. typedef struct tagCHOOSEFONTW
  3483. {
  3484. DWORD lStructSize;
  3485. HWND hwndOwner; // caller's window handle
  3486. HDC hDC; // printer DC/IC or NULL
  3487. LPLOGFONTW lpLogFont; // ptr. to a LOGFONT struct
  3488. INT iPointSize; // 10 * size in points of selected font
  3489. DWORD Flags; // enum. type flags
  3490. COLORREF rgbColors; // returned text color
  3491. LPARAM lCustData; // data passed to hook fn.
  3492. LPCFHOOKPROC lpfnHook; // ptr. to hook function
  3493. LPCWSTR lpTemplateName; // custom template name
  3494. HINSTANCE hInstance; // instance handle of.EXE that
  3495. // contains cust. dlg. template
  3496. LPWSTR lpszStyle; // return the style field here
  3497. // must be LF_FACESIZE or bigger
  3498. WORD nFontType; // same value reported to the EnumFonts
  3499. // call back with the extra FONTTYPE_
  3500. // bits added
  3501. WORD ___MISSING_ALIGNMENT__;
  3502. INT nSizeMin; // minimum pt size allowed &
  3503. INT nSizeMax; // max pt size allowed if
  3504. // CF_LIMITSIZE is used
  3505. }
  3506. CHOOSEFONT;
  3507. TraySave.TraybarFont.lfHeight = TraySave.TraybarFontSize;
  3508. TraySave.TipsFont.lfHeight = TraySave.TipsFontSize;
  3509. CHOOSEFONT cf;
  3510. cf.lStructSize = sizeof cf;
  3511. cf.hwndOwner = hDlg;
  3512. cf.hDC = NULL;
  3513. if (LOWORD(wParam) == IDC_BUTTON_FONT)
  3514. cf.lpLogFont = &TraySave.TraybarFont;
  3515. else
  3516. cf.lpLogFont = &TraySave.TipsFont;
  3517. cf.Flags = CF_SCREENFONTS | CF_INITTOLOGFONTSTRUCT | CF_EFFECTS;
  3518. cf.nFontType = SCREEN_FONTTYPE;
  3519. cf.rgbColors = RGB(0, 0, 0);
  3520. typedef BOOL(WINAPI *pfnChooseFont)(CHOOSEFONT *lpcf);
  3521. HMODULE hComdlg32 = LoadLibrary(L"comdlg32.dll");
  3522. if (hComdlg32)
  3523. {
  3524. pfnChooseFont ChooseFont = (pfnChooseFont)GetProcAddress(hComdlg32, "ChooseFontW");
  3525. if (ChooseFont)
  3526. {
  3527. if (ChooseFont(&cf))
  3528. {
  3529. if (LOWORD(wParam) == IDC_BUTTON_FONT)
  3530. {
  3531. TraySave.TraybarFontSize = TraySave.TraybarFont.lfHeight;
  3532. otleft = -1;
  3533. SetWH();
  3534. AdjustWindowPos();
  3535. }
  3536. else
  3537. TraySave.TipsFontSize = TraySave.TipsFont.lfHeight;
  3538. WriteReg();
  3539. }
  3540. }
  3541. FreeLibrary(hComdlg32);
  3542. }
  3543. }
  3544. else if (LOWORD(wParam) == IDC_BUTTON_COLOR || (LOWORD(wParam) >= IDC_BUTTON_COLOR_BACKGROUND && LOWORD(wParam) <= IDC_BUTTON_COLOR_HIGH))
  3545. {
  3546. CHOOSECOLOR stChooseColor;
  3547. stChooseColor.lStructSize = sizeof(CHOOSECOLOR);
  3548. stChooseColor.hwndOwner = hDlg;
  3549. if (LOWORD(wParam) == IDC_BUTTON_COLOR)
  3550. {
  3551. stChooseColor.rgbResult = TraySave.dAlphaColor[iProject];
  3552. stChooseColor.lpCustColors = (LPDWORD)&TraySave.dAlphaColor[iProject];
  3553. }
  3554. else
  3555. {
  3556. stChooseColor.rgbResult = TraySave.cMonitorColor[LOWORD(wParam) - IDC_BUTTON_COLOR_BACKGROUND];
  3557. stChooseColor.lpCustColors = TraySave.cMonitorColor;
  3558. }
  3559. stChooseColor.Flags = CC_RGBINIT | CC_FULLOPEN;
  3560. stChooseColor.lCustData = 0;
  3561. stChooseColor.lpfnHook = NULL;
  3562. stChooseColor.lpTemplateName = NULL;
  3563. typedef BOOL(WINAPI * pfnChooseColor)(LPCHOOSECOLOR lpcc);
  3564. HMODULE hComdlg32 = LoadLibrary(L"comdlg32.dll");
  3565. if (hComdlg32)
  3566. {
  3567. pfnChooseColor ChooseColor = (pfnChooseColor)GetProcAddress(hComdlg32, "ChooseColorW");
  3568. if (ChooseColor)
  3569. {
  3570. if (ChooseColor(&stChooseColor))
  3571. {
  3572. if (LOWORD(wParam) == IDC_BUTTON_COLOR)
  3573. {
  3574. TraySave.dAlphaColor[iProject] = stChooseColor.rgbResult;
  3575. DWORD bAlphaB = (int)SendDlgItemMessage(hDlg, IDC_SLIDER_ALPHA_B, TBM_GETPOS, 0, 0);
  3576. bAlphaB = bAlphaB << 24;
  3577. TraySave.dAlphaColor[iProject] = bAlphaB + (TraySave.dAlphaColor[iProject] & 0xffffff);
  3578. }
  3579. else
  3580. {
  3581. TraySave.cMonitorColor[LOWORD(wParam - IDC_BUTTON_COLOR_BACKGROUND)] = stChooseColor.rgbResult;
  3582. }
  3583. ::InvalidateRect(GetDlgItem(hMain, LOWORD(wParam)), NULL, FALSE);
  3584. }
  3585. }
  3586. FreeLibrary(hComdlg32);
  3587. }
  3588. WriteReg();
  3589. // SendMessage(hMain, WM_TRAYS, NULL, NULL);
  3590. }
  3591. break;
  3592. }
  3593. return (INT_PTR)FALSE;
  3594. }
  3595. INT_PTR CALLBACK ColorButtonProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) //颜色按钮控件过程
  3596. {
  3597. switch (message)
  3598. {
  3599. case WM_PAINT:
  3600. {
  3601. PAINTSTRUCT ps;
  3602. HDC hdc = BeginPaint(hWnd, &ps);
  3603. RECT rc;
  3604. GetClientRect(hWnd, &rc);
  3605. HBRUSH hb;
  3606. int id = GetDlgCtrlID(hWnd);
  3607. if (id >= IDC_BUTTON_COLOR_BACKGROUND && id <= IDC_BUTTON_COLOR_HIGH)
  3608. {
  3609. hb = CreateSolidBrush(TraySave.cMonitorColor[id - IDC_BUTTON_COLOR_BACKGROUND]);
  3610. }
  3611. else
  3612. hb = CreateSolidBrush(TraySave.dAlphaColor[iProject] & 0xffffff);
  3613. FillRect(hdc, &rc, hb);
  3614. DeleteObject(hb);
  3615. EndPaint(hWnd, &ps);
  3616. return TRUE;
  3617. }
  3618. }
  3619. return CallWindowProc(oldColorButtonPoroc, hWnd, message, wParam, lParam);
  3620. }
  3621. int DrawShadowText(HDC hDC, LPCTSTR lpString, int nCount, LPRECT lpRect, UINT uFormat) //绘制阴影文字
  3622. {
  3623. // COLORREF cColor = GetTextColor(hDC);
  3624. // return DrawShadowText(hDC, lpString, nCount, lpRect, uFormat, cColor, RGB(066, 66,66), 1, 1);
  3625. if (TraySave.cMonitorColor[0] == RGB(0, 0, 1))
  3626. {
  3627. COLORREF cColor = GetTextColor(hDC);
  3628. SetTextColor(hDC, RGB(11, 11, 11));
  3629. OffsetRect(lpRect, 1, 0);
  3630. DrawText(hDC, lpString, nCount, lpRect, uFormat);
  3631. OffsetRect(lpRect, 0, 1);
  3632. DrawText(hDC, lpString, nCount, lpRect, uFormat);
  3633. OffsetRect(lpRect, 1, 0);
  3634. DrawText(hDC, lpString, nCount, lpRect, uFormat);
  3635. OffsetRect(lpRect, 0, 1);
  3636. DrawText(hDC, lpString, nCount, lpRect, uFormat);
  3637. OffsetRect(lpRect, -2, -2);
  3638. SetTextColor(hDC, cColor);
  3639. }
  3640. return DrawText(hDC, lpString, nCount, lpRect, uFormat);
  3641. }