programs/oleview/tree.c

   1 /*
   2  * OleView (tree.c)
   3  *
   4  * Copyright 2006 Piotr Caban
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "main.h"
  22
  23 TREE tree;
  24 static const WCHAR wszCLSID[] = { 'C','L','S','I','D','\\','\0' };
  25 static const WCHAR wszInProcServer32[] =
  26     { 'I','n','P','r','o','c','S','e','r','v','e','r','3','2','\0' };
  27 static const WCHAR wszOle32dll[] = { 'o','l','e','3','2','.','d','l','l','\0' };
  28 static const WCHAR wszOleAut32dll[] =
  29     { 'o','l','e','a','u','t','3','2','.','d','l','l','\0' };
  30 static const WCHAR wszImplementedCategories[] =
  31     { 'I','m','p','l','e','m','e','n','t','e','d',' ',
  32         'C','a','t','e','g','o','r','i','e','s','\0' };
  33 static const WCHAR wszAppID[] = { 'A','p','p','I','D','\\','\0' };
  34 static const WCHAR wszTypeLib[] = { 'T','y','p','e','L','i','b','\\','\0' };
  35 static const WCHAR wszInterface[] = { 'I','n','t','e','r','f','a','c','e','\\','\0' };
  36 static const WCHAR wszComponentCategories[] = { 'C','o','m','p','o','n','e','n','t',
  37     ' ','C','a','t','e','g','o','r','i','e','s','\\','\0' };
  38 static const WCHAR wszGetPath[] = { '0','\\','w','i','n','3','2','\0' };
  39
  40 static LPARAM CreateITEM_INFO(INT flag, const WCHAR *info, const WCHAR *clsid, const WCHAR *path)
  41 {
  42     ITEM_INFO *reg;
  43
  44     reg = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ITEM_INFO));
  45
  46     reg->cFlag = flag;
  47     lstrcpyW(reg->info, info);
  48     if(clsid) lstrcpyW(reg->clsid, clsid);
  49     if(path) lstrcpyW(reg->path, path);
  50
  51     return (LPARAM)reg;
  52 }
  53
  54 void CreateInst(HTREEITEM item, WCHAR *wszMachineName)
  55 {
  56     TVITEMW tvi;
  57     HTREEITEM hCur;
  58     TVINSERTSTRUCTW tvis;
  59     WCHAR wszTitle[MAX_LOAD_STRING];
  60     WCHAR wszMessage[MAX_LOAD_STRING];
  61     WCHAR wszFlagName[MAX_LOAD_STRING];
  62     WCHAR wszTreeName[MAX_LOAD_STRING];
  63     WCHAR wszRegPath[MAX_LOAD_STRING];
  64     const WCHAR wszFormat[] = { '\n','%','s',' ','(','$','%','x',')','\n','\0' };
  65     CLSID clsid;
  66     COSERVERINFO remoteInfo;
  67     MULTI_QI qi;
  68     IUnknown *obj, *unk;
  69     HRESULT hRes;
  70
  71     memset(&tvi, 0, sizeof(TVITEMW));
  72     tvi.mask = TVIF_TEXT;
  73     tvi.hItem = item;
  74     tvi.cchTextMax = MAX_LOAD_STRING;
  75     tvi.pszText = wszTreeName;
  76
  77     memset(&tvis, 0, sizeof(TVINSERTSTRUCTW));
  78     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM;
  79     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
  80     tvis.hInsertAfter = TVI_FIRST;
  81     U(tvis).item.pszText = tvi.pszText;
  82     tvis.hParent = item;
  83     tvis.hInsertAfter = TVI_LAST;
  84
  85     if (!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) return;
  86
  87     if(!tvi.lParam || ((ITEM_INFO *)tvi.lParam)->loaded
  88                 || !(((ITEM_INFO *)tvi.lParam)->cFlag&SHOWALL)) return;
  89
  90     if(FAILED(CLSIDFromString(((ITEM_INFO *)tvi.lParam)->clsid, &clsid))) return;
  91
  92     if(wszMachineName)
  93     {
  94         remoteInfo.dwReserved1 = 0;
  95         remoteInfo.dwReserved2 = 0;
  96         remoteInfo.pAuthInfo = NULL;
  97         remoteInfo.pwszName = wszMachineName;
  98
  99         qi.pIID = &IID_IUnknown;
 100
 101         CoCreateInstanceEx(&clsid, NULL, globals.dwClsCtx|CLSCTX_REMOTE_SERVER,
 102                 &remoteInfo, 1, &qi);
 103         hRes = qi.hr;
 104         obj = qi.pItf;
 105     }
 106     else hRes = CoCreateInstance(&clsid, NULL, globals.dwClsCtx,
 107             &IID_IUnknown, (void **)&obj);
 108
 109     if(FAILED(hRes))
 110     {
 111         LoadStringW(globals.hMainInst, IDS_CGCOFAIL, wszMessage,
 112                 sizeof(wszMessage)/sizeof(wszMessage[0]));
 113         LoadStringW(globals.hMainInst, IDS_ABOUT, wszTitle,
 114                 sizeof(wszTitle)/sizeof(wszTitle[0]));
 115
 116 #define CASE_ERR(i) case i: \
 117     MultiByteToWideChar(CP_ACP, 0, #i, -1, wszFlagName, MAX_LOAD_STRING); \
 118     break
 119
 120         switch(hRes)
 121         {
 122             CASE_ERR(REGDB_E_CLASSNOTREG);
 123             CASE_ERR(E_NOINTERFACE);
 124             CASE_ERR(REGDB_E_READREGDB);
 125             CASE_ERR(REGDB_E_KEYMISSING);
 126             CASE_ERR(CO_E_DLLNOTFOUND);
 127             CASE_ERR(CO_E_APPNOTFOUND);
 128             CASE_ERR(E_ACCESSDENIED);
 129             CASE_ERR(CO_E_ERRORINDLL);
 130             CASE_ERR(CO_E_APPDIDNTREG);
 131             CASE_ERR(CLASS_E_CLASSNOTAVAILABLE);
 132             default:
 133                 LoadStringW(globals.hMainInst, IDS_ERROR_UNKN, wszFlagName, sizeof(wszFlagName)/sizeof(wszFlagName[0]));
 134         }
 135
 136         wsprintfW(&wszMessage[lstrlenW(wszMessage)], wszFormat,
 137                 wszFlagName, (unsigned)hRes);
 138         MessageBoxW(globals.hMainWnd, wszMessage, wszTitle, MB_OK|MB_ICONEXCLAMATION);
 139         return;
 140     }
 141
 142     ((ITEM_INFO *)tvi.lParam)->loaded = 1;
 143     ((ITEM_INFO *)tvi.lParam)->pU = obj;
 144
 145     tvi.mask = TVIF_STATE;
 146     tvi.state = TVIS_BOLD;
 147     tvi.stateMask = TVIS_BOLD;
 148     SendMessageW(globals.hTree, TVM_SETITEMW, 0, (LPARAM)&tvi);
 149
 150     tvi.mask = TVIF_TEXT;
 151     hCur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 152             TVGN_CHILD, (LPARAM)tree.hI);
 153
 154     while(hCur)
 155     {
 156         tvi.hItem = hCur;
 157         if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) || !tvi.lParam)
 158         {
 159             hCur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 160                     TVGN_NEXT, (LPARAM)hCur);
 161             continue;
 162         }
 163
 164         CLSIDFromString(((ITEM_INFO *)tvi.lParam)->clsid, &clsid);
 165         hRes = IUnknown_QueryInterface(obj, &clsid, (void *)&unk);
 166
 167         if(SUCCEEDED(hRes))
 168         {
 169             IUnknown_Release(unk);
 170
 171             lstrcpyW(wszRegPath, wszInterface);
 172             lstrcpyW(&wszRegPath[lstrlenW(wszRegPath)], ((ITEM_INFO *)tvi.lParam)->clsid);
 173             U(tvis).item.lParam = CreateITEM_INFO(REGTOP|INTERFACE|REGPATH,
 174                     wszRegPath, ((ITEM_INFO *)tvi.lParam)->clsid, NULL);
 175             SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 176         }
 177         hCur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 178                 TVGN_NEXT, (LPARAM)hCur);
 179     }
 180
 181     RefreshMenu(item);
 182     RefreshDetails(item);
 183 }
 184
 185 void ReleaseInst(HTREEITEM item)
 186 {
 187     TVITEMW tvi;
 188     HTREEITEM cur;
 189     IUnknown *pU;
 190
 191     memset(&tvi, 0, sizeof(TVITEMW));
 192     tvi.hItem = item;
 193     if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) || !tvi.lParam) return;
 194
 195     pU = ((ITEM_INFO *)tvi.lParam)->pU;
 196
 197     if(pU) IUnknown_Release(pU);
 198     ((ITEM_INFO *)tvi.lParam)->loaded = 0;
 199
 200     SendMessageW(globals.hTree, TVM_EXPAND, TVE_COLLAPSE, (LPARAM)item);
 201
 202     cur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 203             TVGN_CHILD, (LPARAM)item);
 204     while(cur)
 205     {
 206         SendMessageW(globals.hTree, TVM_DELETEITEM, 0, (LPARAM)cur);
 207         cur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 208                 TVGN_CHILD, (LPARAM)item);
 209     }
 210
 211     tvi.mask = TVIF_CHILDREN|TVIF_STATE;
 212     tvi.state = 0;
 213     tvi.stateMask = TVIS_BOLD;
 214     tvi.cChildren = 1;
 215     SendMessageW(globals.hTree, TVM_SETITEMW, 0, (LPARAM)&tvi);
 216 }
 217
 218 BOOL CreateRegPath(HTREEITEM item, WCHAR *buffer, int bufSize)
 219 {
 220     TVITEMW tvi;
 221     int bufLen;
 222     BOOL ret = FALSE;
 223
 224     memset(buffer, 0, bufSize * sizeof(WCHAR));
 225     memset(&tvi, 0, sizeof(TVITEMW));
 226     tvi.hItem = item;
 227
 228     if (SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi))
 229         ret = (tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGPATH);
 230
 231     while(TRUE)
 232     {
 233         if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) break;
 234
 235         if(tvi.lParam && (((ITEM_INFO *)tvi.lParam)->cFlag & (REGPATH|REGTOP)))
 236         {
 237             bufLen = lstrlenW(((ITEM_INFO *)tvi.lParam)->info);
 238             memmove(&buffer[bufLen], buffer, (bufSize - bufLen) * sizeof(WCHAR));
 239             memcpy(buffer, ((ITEM_INFO *)tvi.lParam)->info, bufLen * sizeof(WCHAR));
 240         }
 241
 242         if(tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGTOP) break;
 243
 244         if(!tvi.lParam) return FALSE;
 245
 246         tvi.hItem = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 247                 TVGN_PARENT, (LPARAM)tvi.hItem);
 248     }
 249     return ret;
 250 }
 251
 252 static void AddCOMandAll(void)
 253 {
 254     TVINSERTSTRUCTW tvis;
 255     TVITEMW tvi;
 256     HTREEITEM curSearch;
 257     HKEY hKey, hCurKey, hInfo;
 258     WCHAR valName[MAX_LOAD_STRING];
 259     WCHAR buffer[MAX_LOAD_STRING];
 260     WCHAR wszComp[MAX_LOAD_STRING];
 261     LONG lenBuffer;
 262     int i=-1;
 263
 264     memset(&tvi, 0, sizeof(TVITEMW));
 265     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_CHILDREN;
 266     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 267     U(tvis).item.cChildren = 1;
 268     tvis.hInsertAfter = TVI_FIRST;
 269
 270     if(RegOpenKeyW(HKEY_CLASSES_ROOT, wszCLSID, &hKey) != ERROR_SUCCESS) return;
 271
 272     while(TRUE)
 273     {
 274         i++;
 275
 276         if(RegEnumKeyW(hKey, i, valName, sizeof(valName)/sizeof(valName[0])) != ERROR_SUCCESS) break;
 277
 278         if(RegOpenKeyW(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
 279
 280         lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 281         tvis.hParent = tree.hAO;
 282
 283         if(RegOpenKeyW(hCurKey, wszInProcServer32, &hInfo) == ERROR_SUCCESS)
 284         {
 285             if(RegQueryValueW(hInfo, NULL, buffer, &lenBuffer) == ERROR_SUCCESS
 286                     && *buffer)
 287                 if(!memcmp(buffer, wszOle32dll, sizeof(WCHAR[9]))
 288                         ||!memcmp(buffer, wszOleAut32dll, sizeof(WCHAR[12])))
 289                     tvis.hParent = tree.hCLO;
 290
 291             RegCloseKey(hInfo);
 292         }
 293
 294         lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 295
 296         if(RegQueryValueW(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
 297             U(tvis).item.pszText = buffer;
 298         else U(tvis).item.pszText = valName;
 299
 300         U(tvis).item.lParam = CreateITEM_INFO(REGPATH|SHOWALL, valName, valName, NULL);
 301         if(tvis.hParent) SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 302
 303         if(RegOpenKeyW(hCurKey, wszImplementedCategories, &hInfo) == ERROR_SUCCESS)
 304         {
 305             if(RegEnumKeyW(hInfo, 0, wszComp, sizeof(wszComp)/sizeof(wszComp[0])) != ERROR_SUCCESS) break;
 306
 307             RegCloseKey(hInfo);
 308
 309             if(tree.hGBCC) curSearch = (HTREEITEM)SendMessageW(globals.hTree,
 310                     TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)tree.hGBCC);
 311             else curSearch = (HTREEITEM)SendMessageW(globals.hTree,
 312                     TVM_GETNEXTITEM, TVGN_CHILD, (LPARAM)TVI_ROOT);
 313
 314             while(curSearch)
 315             {
 316                 tvi.hItem = curSearch;
 317                 if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) break;
 318
 319                 if(tvi.lParam && !lstrcmpW(((ITEM_INFO *)tvi.lParam)->info, wszComp))
 320                 {
 321                     tvis.hParent = curSearch;
 322
 323                     memmove(&valName[6], valName, sizeof(WCHAR[MAX_LOAD_STRING-6]));
 324                     memmove(valName, wszCLSID, sizeof(WCHAR[6]));
 325                     U(tvis).item.lParam = CreateITEM_INFO(REGTOP|REGPATH|SHOWALL,
 326                             valName, &valName[6], NULL);
 327
 328                     SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 329                     break;
 330                 }
 331                 curSearch = (HTREEITEM)SendMessageW(globals.hTree,
 332                         TVM_GETNEXTITEM, TVGN_NEXT, (LPARAM)curSearch);
 333             }
 334         }
 335         RegCloseKey(hCurKey);
 336     }
 337     RegCloseKey(hKey);
 338
 339     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hCLO);
 340     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hAO);
 341 }
 342
 343 static void AddApplicationID(void)
 344 {
 345     TVINSERTSTRUCTW tvis;
 346     HKEY hKey, hCurKey;
 347     WCHAR valName[MAX_LOAD_STRING];
 348     WCHAR buffer[MAX_LOAD_STRING];
 349     LONG lenBuffer;
 350     int i=-1;
 351
 352     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM;
 353     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 354     tvis.hInsertAfter = TVI_FIRST;
 355     tvis.hParent = tree.hAID;
 356
 357     if(RegOpenKeyW(HKEY_CLASSES_ROOT, wszAppID, &hKey) != ERROR_SUCCESS) return;
 358
 359     while(TRUE)
 360     {
 361         i++;
 362
 363         if(RegEnumKeyW(hKey, i, valName, sizeof(valName)/sizeof(valName[0])) != ERROR_SUCCESS) break;
 364
 365         if(RegOpenKeyW(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
 366
 367         lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 368
 369         if(RegQueryValueW(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
 370             U(tvis).item.pszText = buffer;
 371         else U(tvis).item.pszText = valName;
 372
 373         RegCloseKey(hCurKey);
 374
 375         U(tvis).item.lParam = CreateITEM_INFO(REGPATH, valName, valName, NULL);
 376         SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 377     }
 378     RegCloseKey(hKey);
 379
 380     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hAID);
 381 }
 382
 383 static void AddTypeLib(void)
 384 {
 385     TVINSERTSTRUCTW tvis;
 386     HKEY hKey, hCurKey, hInfoKey, hPath;
 387     WCHAR valName[MAX_LOAD_STRING];
 388     WCHAR valParent[MAX_LOAD_STRING];
 389     WCHAR buffer[MAX_LOAD_STRING];
 390     WCHAR wszVer[MAX_LOAD_STRING];
 391     WCHAR wszPath[MAX_LOAD_STRING];
 392     const WCHAR wszFormat[] = { ' ','(','%','s',' ','%','s',')','\0' };
 393     const WCHAR wszFormat2[] = { '%','s','\\','%','s','\0' };
 394     LONG lenBuffer;
 395     int i=-1, j;
 396
 397     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM;
 398     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 399     tvis.hInsertAfter = TVI_FIRST;
 400     tvis.hParent = tree.hTL;
 401
 402     if(RegOpenKeyW(HKEY_CLASSES_ROOT, wszTypeLib, &hKey) != ERROR_SUCCESS) return;
 403
 404     while(TRUE)
 405     {
 406         i++;
 407
 408         if(RegEnumKeyW(hKey, i, valParent, sizeof(valParent)/sizeof(valParent[0])) != ERROR_SUCCESS) break;
 409
 410         if(RegOpenKeyW(hKey, valParent, &hCurKey) != ERROR_SUCCESS) continue;
 411
 412         j = -1;
 413         while(TRUE)
 414         {
 415             j++;
 416
 417             if(RegEnumKeyW(hCurKey, j, valName, sizeof(valName)/sizeof(valName[0])) != ERROR_SUCCESS) break;
 418
 419             if(RegOpenKeyW(hCurKey, valName, &hInfoKey) != ERROR_SUCCESS) continue;
 420
 421             lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 422
 423             if(RegQueryValueW(hInfoKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS
 424                     && *buffer)
 425             {
 426                 LoadStringW(globals.hMainInst, IDS_TL_VER, wszVer,
 427                         sizeof(wszVer)/sizeof(wszVer[0]));
 428
 429                 wsprintfW(&buffer[lstrlenW(buffer)], wszFormat, wszVer, valName);
 430                 U(tvis).item.pszText = buffer;
 431
 432                 lenBuffer = MAX_LOAD_STRING;
 433                 RegOpenKeyW(hInfoKey, wszGetPath, &hPath);
 434                 RegQueryValueW(hPath, NULL, wszPath, &lenBuffer);
 435                 RegCloseKey(hPath);
 436             }
 437             else U(tvis).item.pszText = valName;
 438
 439             RegCloseKey(hInfoKey);
 440
 441             wsprintfW(wszVer, wszFormat2, valParent, valName);
 442             U(tvis).item.lParam = CreateITEM_INFO(REGPATH, wszVer, valParent, wszPath);
 443
 444             SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 445         }
 446         RegCloseKey(hCurKey);
 447     }
 448
 449     RegCloseKey(hKey);
 450
 451     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hTL);
 452 }
 453
 454 static void AddInterfaces(void)
 455 {
 456     TVINSERTSTRUCTW tvis;
 457     HKEY hKey, hCurKey;
 458     WCHAR valName[MAX_LOAD_STRING];
 459     WCHAR buffer[MAX_LOAD_STRING];
 460     LONG lenBuffer;
 461     int i=-1;
 462
 463     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM;
 464     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 465     tvis.hInsertAfter = TVI_FIRST;
 466     tvis.hParent = tree.hI;
 467
 468     if(RegOpenKeyW(HKEY_CLASSES_ROOT, wszInterface, &hKey) != ERROR_SUCCESS) return;
 469
 470     while(TRUE)
 471     {
 472         i++;
 473
 474         if(RegEnumKeyW(hKey, i, valName, sizeof(valName)/sizeof(valName[0])) != ERROR_SUCCESS) break;
 475
 476         if(RegOpenKeyW(hKey, valName, &hCurKey) != ERROR_SUCCESS) continue;
 477
 478         lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 479
 480         if(RegQueryValueW(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
 481             U(tvis).item.pszText = buffer;
 482         else U(tvis).item.pszText = valName;
 483
 484         RegCloseKey(hCurKey);
 485
 486         U(tvis).item.lParam = CreateITEM_INFO(REGPATH|INTERFACE, valName, valName, NULL);
 487         SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 488     }
 489
 490     RegCloseKey(hKey);
 491
 492     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hI);
 493 }
 494
 495 static void AddComponentCategories(void)
 496 {
 497     TVINSERTSTRUCTW tvis;
 498     HKEY hKey, hCurKey;
 499     WCHAR keyName[MAX_LOAD_STRING];
 500     WCHAR valName[MAX_LOAD_STRING];
 501     WCHAR buffer[MAX_LOAD_STRING];
 502     LONG lenBuffer;
 503     DWORD lenBufferHlp;
 504     DWORD lenValName;
 505     int i=-1;
 506
 507     U(tvis).item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_CHILDREN;
 508     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 509     tvis.hInsertAfter = TVI_FIRST;
 510     if(tree.hGBCC) tvis.hParent = tree.hGBCC;
 511     else tvis.hParent = TVI_ROOT;
 512     U(tvis).item.cChildren = 1;
 513
 514     if(RegOpenKeyW(HKEY_CLASSES_ROOT, wszComponentCategories, &hKey) != ERROR_SUCCESS)
 515         return;
 516
 517     while(TRUE)
 518     {
 519         i++;
 520
 521         if(RegEnumKeyW(hKey, i, keyName, sizeof(keyName)/sizeof(keyName[0])) != ERROR_SUCCESS) break;
 522
 523         if(RegOpenKeyW(hKey, keyName, &hCurKey) != ERROR_SUCCESS) continue;
 524
 525         lenBuffer = sizeof(WCHAR[MAX_LOAD_STRING]);
 526         lenBufferHlp = sizeof(WCHAR[MAX_LOAD_STRING]);
 527         lenValName = sizeof(WCHAR[MAX_LOAD_STRING]);
 528
 529         if(RegQueryValueW(hCurKey, NULL, buffer, &lenBuffer) == ERROR_SUCCESS && *buffer)
 530             U(tvis).item.pszText = buffer;
 531         else if(RegEnumValueW(hCurKey, 0, valName, &lenValName, NULL, NULL,
 532                     (LPBYTE)buffer, &lenBufferHlp) == ERROR_SUCCESS && *buffer)
 533             U(tvis).item.pszText = buffer;
 534         else continue;
 535
 536         RegCloseKey(hCurKey);
 537
 538         U(tvis).item.lParam = CreateITEM_INFO(REGTOP, keyName, keyName, NULL);
 539         SendMessageW(globals.hTree, TVM_INSERTITEMW, 0, (LPARAM)&tvis);
 540     }
 541
 542     RegCloseKey(hKey);
 543
 544     SendMessageW(globals.hTree, TVM_SORTCHILDREN, FALSE, (LPARAM)tree.hGBCC);
 545 }
 546
 547 static void AddBaseEntries(void)
 548 {
 549     TVINSERTSTRUCTW tvis;
 550     WCHAR name[MAX_LOAD_STRING];
 551
 552     U(tvis).item.mask = TVIF_TEXT|TVIF_CHILDREN|TVIF_PARAM;
 553     /* FIXME add TVIF_IMAGE */
 554     U(tvis).item.pszText = name;
 555     U(tvis).item.cchTextMax = MAX_LOAD_STRING;
 556     U(tvis).item.cChildren = 1;
 557     tvis.hInsertAfter = TVI_FIRST;
 558     tvis.hParent = TVI_ROOT;
 559
 560     LoadStringW(globals.hMainInst, IDS_TREE_I, U(tvis).item.pszText,
 561             MAX_LOAD_STRING);
 562     U(tvis).item.lParam = CreateITEM_INFO(REGTOP, wszInterface, NULL, NULL);
 563     tree.hI = TreeView_InsertItemW(globals.hTree, &tvis);
 564
 565     LoadStringW(globals.hMainInst, IDS_TREE_TL, U(tvis).item.pszText,
 566             MAX_LOAD_STRING);
 567     U(tvis).item.lParam = CreateITEM_INFO(REGTOP, wszTypeLib, NULL, NULL);
 568     tree.hTL = TreeView_InsertItemW(globals.hTree, &tvis);
 569
 570     LoadStringW(globals.hMainInst, IDS_TREE_AID, U(tvis).item.pszText,
 571             MAX_LOAD_STRING);
 572     U(tvis).item.lParam = CreateITEM_INFO(REGTOP|REGPATH, wszAppID, NULL, NULL);
 573     tree.hAID = TreeView_InsertItemW(globals.hTree, &tvis);
 574
 575     LoadStringW(globals.hMainInst, IDS_TREE_OC, U(tvis).item.pszText,
 576             MAX_LOAD_STRING);
 577     U(tvis).item.lParam = 0;
 578     tree.hOC = TreeView_InsertItemW(globals.hTree, &tvis);
 579
 580
 581     tvis.hParent = tree.hOC;
 582     LoadStringW(globals.hMainInst, IDS_TREE_AO, U(tvis).item.pszText,
 583             MAX_LOAD_STRING);
 584     U(tvis).item.lParam = CreateITEM_INFO(REGTOP, wszCLSID, NULL, NULL);
 585     tree.hAO = TreeView_InsertItemW(globals.hTree, &tvis);
 586
 587     LoadStringW(globals.hMainInst, IDS_TREE_CLO, U(tvis).item.pszText,
 588             MAX_LOAD_STRING);
 589     tree.hCLO = TreeView_InsertItemW(globals.hTree, &tvis);
 590
 591     LoadStringW(globals.hMainInst, IDS_TREE_O1O, U(tvis).item.pszText,
 592             MAX_LOAD_STRING);
 593     U(tvis).item.lParam = 0;
 594     tree.hO1O = TreeView_InsertItemW(globals.hTree, &tvis);
 595
 596     LoadStringW(globals.hMainInst, IDS_TREE_GBCC, U(tvis).item.pszText,
 597             MAX_LOAD_STRING);
 598     U(tvis).item.lParam = CreateITEM_INFO(REGTOP|REGPATH,
 599             wszComponentCategories, NULL, NULL);
 600     tree.hGBCC = TreeView_InsertItemW(globals.hTree, &tvis);
 601
 602     SendMessageW(globals.hTree, TVM_EXPAND, TVE_EXPAND, (LPARAM)tree.hOC);
 603 }
 604
 605 void EmptyTree(void)
 606 {
 607     HTREEITEM cur, del;
 608     TVITEMW tvi;
 609
 610     tvi.mask = TVIF_PARAM;
 611     cur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 612             TVGN_CHILD, (LPARAM)TVI_ROOT);
 613
 614     while(TRUE)
 615     {
 616         del = cur;
 617         cur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 618                 TVGN_CHILD, (LPARAM)del);
 619
 620         if(!cur) cur = (HTREEITEM)SendMessageW(globals.hTree,
 621                 TVM_GETNEXTITEM, TVGN_NEXT, (LPARAM)del);
 622         if(!cur)
 623         {
 624             cur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
 625                     TVGN_PREVIOUS, (LPARAM)del);
 626             if(!cur) cur = (HTREEITEM)SendMessageW(globals.hTree,
 627                     TVM_GETNEXTITEM, TVGN_PARENT, (LPARAM)del);
 628
 629             tvi.hItem = del;
 630             if(SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) && tvi.lParam)
 631             {
 632                 if(((ITEM_INFO *)tvi.lParam)->loaded) ReleaseInst(del);
 633                 HeapFree(GetProcessHeap(), 0, (ITEM_INFO *)tvi.lParam);
 634
 635                 SendMessageW(globals.hTree, TVM_DELETEITEM, 0, (LPARAM)del);
 636             }
 637
 638             if(!cur) break;
 639         }
 640     }
 641 }
 642
 643 void AddTreeEx(void)
 644 {
 645     AddBaseEntries();
 646     AddComponentCategories();
 647     AddCOMandAll();
 648     AddApplicationID();
 649     AddTypeLib();
 650     AddInterfaces();
 651 }
 652
 653 void AddTree(void)
 654 {
 655     memset(&tree, 0, sizeof(TREE));
 656     AddComponentCategories();
 657     AddCOMandAll();
 658 }
 659
 660 static LRESULT CALLBACK TreeProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
 661 {
 662     switch(uMsg)
 663     {
 664         case WM_CREATE:
 665             globals.hTree = CreateWindowW(WC_TREEVIEWW, NULL,
 666                     WS_CHILD|WS_VISIBLE|TVS_HASLINES|TVS_HASBUTTONS|TVS_LINESATROOT,
 667                     0, 0, 0, 0, hWnd, (HMENU)TREE_WINDOW, globals.hMainInst, NULL);
 668             AddTreeEx();
 669             break;
 670         case WM_NOTIFY:
 671             if((int)wParam != TREE_WINDOW) break;
 672             switch(((LPNMHDR)lParam)->code)
 673             {
 674                 case TVN_ITEMEXPANDINGW:
 675                     CreateInst(((NMTREEVIEWW *)lParam)->itemNew.hItem, NULL);
 676                     break;
 677                 case TVN_SELCHANGEDW:
 678                     RefreshMenu(((NMTREEVIEWW *)lParam)->itemNew.hItem);
 679                     RefreshDetails(((NMTREEVIEWW *)lParam)->itemNew.hItem);
 680                     break;
 681             }
 682             break;
 683         case WM_SIZE:
 684             MoveWindow(globals.hTree, 0, 0, LOWORD(lParam), HIWORD(lParam), TRUE);
 685             break;
 686         default:
 687             return DefWindowProcW(hWnd, uMsg, wParam, lParam);
 688     }
 689     return 0;
 690 }
 691
 692 HWND CreateTreeWindow(HINSTANCE hInst)
 693 {
 694     WNDCLASSW wct;
 695     const WCHAR wszTreeClass[] = { 'T','R','E','E','\0' };
 696
 697     memset(&wct, 0, sizeof(WNDCLASSW));
 698     wct.lpfnWndProc = TreeProc;
 699     wct.lpszClassName = wszTreeClass;
 700     wct.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
 701     wct.hCursor       = LoadCursorW(0, (LPCWSTR)IDC_ARROW);
 702
 703     if(!RegisterClassW(&wct)) return NULL;
 704
 705     return CreateWindowExW(WS_EX_CLIENTEDGE, wszTreeClass, NULL, WS_CHILD|WS_VISIBLE,
 706             0, 0, 0, 0, globals.hPaneWnd, NULL, hInst, NULL);
 707 }