一款显示网速、系统利用的软件
Du kannst nicht mehr als 25 Themen auswählen Themen müssen mit entweder einem Buchstaben oder einer Ziffer beginnen. Sie können Bindestriche („-“) enthalten und bis zu 35 Zeichen lang sein.

TrayS.cpp 113KB

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