CImageList containing windows system icons

You need also to do some shell programming in order to get the proper icons.

Here's an excerpt from a class the does populate a checkboxed tree with folders and proper icons (starting from 'Desktop' item)

LPITEMIDLIST MakeCopy(LPCITEMIDLIST pidl)
{
      LPITEMIDLIST pidlTemp = (LPITEMIDLIST) pidl;

      // Calculate size of list.
      UINT cb = pidl->mkid.cb + sizeof(pidl->mkid.cb);


      LPITEMIDLIST pidlRet = (LPITEMIDLIST)g_pMalloc->Alloc(cb);
      if (pidlRet)
            CopyMemory(pidlRet, pidl, cb);
      return pidlRet;
}

// CTargetTree message handlers


void CTargetTree::Populate(HTREEITEM hItem /*= NULL*/)
{
      
      LPITEMIDLIST      pidlDesktop;
      HRESULT                  hr;
      STRRET                  strDispName;
      TCHAR                  szDisplayName[100];

      //LPSHELLVIEW            pShellView;
      // // Get a pointer to the Desktop's IShellFolder interface. //
      if(FAILED(SHGetDesktopFolder(&m_pSHDesktopFolder)))
      {
            return;
      }

      hr = SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOP, &pidlDesktop);
      hr = m_pSHDesktopFolder->GetDisplayNameOf(pidlDesktop, SHGDN_INFOLDER, &strDispName);
      hr = StrRetToBuf(&strDispName, pidlDesktop, szDisplayName, sizeof(szDisplayName));

      
      SHFILEINFO      sfi;
      HIMAGELIST      hDesktop = (HIMAGELIST)SHGetFileInfo((LPCTSTR)pidlDesktop, 0, &sfi, sizeof(sfi), SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON);

      if(!hItem)
      {
            CImageList      *pIL = CImageList::FromHandle(hDesktop);

            SetImageList(pIL, TVSIL_NORMAL);
            hItem = InsertItem(szDisplayName, sfi.iIcon, sfi.iIcon);
      }

      CImageList      *pIl2 = GetImageList(TVSIL_STATE);
      ASSERT(pIl2!=0);
      IMAGEINFO      imgInfo;
      ZeroMemory(&imgInfo, sizeof(imgInfo));
      pIl2->GetImageInfo(0, &imgInfo);
      pIl2->Add(LoadIcon(AfxGetResourceHandle(), MAKEINTRESOURCE(IDI_CHECK_GRAY)));

      PopulateSubItem(m_pSHDesktopFolder, hItem);

      //m_pSHDesktopFolder->Release();

      Expand(hItem, TVE_EXPAND);
}


void CTargetTree::PopulateSubItem(LPSHELLFOLDER pSHFolder, HTREEITEM hParent, int iDepth)
{
      LPITEMIDLIST      pIdl;
      LPENUMIDLIST      pEnumFolders;
      //SHFILEINFO            sfi;
      //HIMAGELIST            hDesktop;
      HRESULT                  hr;
      CImageList            *pIL;
      STRRET                  strDispName;
      TCHAR                  szDisplayName[MAX_PATH],szFullName[MAX_PATH];
      HTREEITEM            hItem;
      SFGAOF                  rgf = SFGAO_HASSUBFOLDER;
      _FolderData            *pFolderData;


      hr = pSHFolder->EnumObjects(NULL, SHCONTF_FOLDERS | SHCONTF_INCLUDEHIDDEN, &pEnumFolders);
      if(FAILED(hr))
            return;

      m_iDepth++;

      hr = pEnumFolders->Next(1, &pIdl, NULL);

      while(SUCCEEDED(hr) && (hr != S_FALSE))
      {
            rgf = -1;
            hr = pSHFolder->GetAttributesOf(1, (LPCITEMIDLIST*)&pIdl, &rgf);

            //special case for Control Panel -> skip
            //
            if(rgf  == 0xa0000004)
            {
                  hr = pEnumFolders->Next(1, &pIdl, NULL);
                  continue;
            }

            //if we have a folder
            //
            if((rgf & SFGAO_FOLDER) && !(rgf & SFGAO_LINK))
            {
                  hr = pSHFolder->GetDisplayNameOf(pIdl, SHGDN_INFOLDER, &strDispName);
                  hr = StrRetToBuf(&strDispName, pIdl, szDisplayName, sizeof(szDisplayName));

                  hr = pSHFolder->GetDisplayNameOf(pIdl, SHGDN_NORMAL | SHGDN_FORPARSING, &strDispName);
                  hr = StrRetToBuf(&strDispName, pIdl, szFullName, sizeof(szFullName));
                  
//                  if((rgf & 0xf080014c) != 0xf080014c)
//                        hDesktop = (HIMAGELIST)SHGetFileInfo((LPCTSTR)pIdl, 0, &sfi, sizeof(sfi), SHGFI_PIDL | SHGFI_SYSICONINDEX | SHGFI_SMALLICON | SHGFI_ICON);
//                  else
//                        hDesktop = (HIMAGELIST)SHGetFileInfo(szFullName, 0, &sfi, sizeof(SHFILEINFO), SHGFI_SYSICONINDEX | SHGFI_SMALLICON);

                  LPEXTRACTICON      pIcon;
                  hr = pSHFolder->GetUIObjectOf(NULL, 1, (LPCITEMIDLIST*)&pIdl, IID_IExtractIcon, NULL, (void**)&pIcon);
                  if(SUCCEEDED(hr))
                  {
                        TCHAR      szIconLoc[300];
                        int            iIndex;
                        UINT      uFlags;
                        HICON      hSmall,hLarge;

                        hr = pIcon->GetIconLocation(GIL_FORSHELL, szIconLoc, 300, &iIndex, &uFlags);
                        hr = pIcon->Extract(szIconLoc, iIndex, &hLarge, &hSmall, 16);
                        pIcon->Release();

                        pIL = GetImageList(TVSIL_NORMAL);
                        if(!pIL)
                        {
                              pIL = new CImageList;
                              pIL->Create(16, 16, ILC_COLOR32, 0, 4);
                        }
                        
                        pIL->Add(hLarge);
                        int      iIdx = pIL->GetImageCount()-1;
                        SetImageList(pIL, TVSIL_NORMAL);
                        hItem = InsertItem(szDisplayName, iIdx, iIdx, hParent);
                        
                        //check children
                        //
                        int      iCheck = GetCheck(hParent);
                        if(iCheck == eChecked)
                              SetCheck(hItem, iCheck);

                        //if the item is on excluded, will be unchecked (MUST be done after the above check ;)
                        //if the item is on scan folders, will be checked
                        //
                        CheckItemOnPopulate(hItem, szFullName);
                  }

/*
                  pIL = CImageList::FromHandle(hDesktop);
                  SetImageList(pIL, TVSIL_NORMAL);
                  hItem = InsertItem(szDisplayName, sfi.iIcon, sfi.iIcon, hParent);
                  SetItemData(hItem, NULL);
*/
            
                  //SHGetPathFromIDList(pIdl, szDisplayName);

                  pFolderData = new _FolderData;
                  pFolderData->pIdl = MakeCopy(pIdl);
                  pFolderData->pSHF = pSHFolder;
                  SetItemData(hItem, (DWORD_PTR)pFolderData);

                  //if it has subfolders
                  //
                  if((m_iDepth <= iDepth) && (rgf & SFGAO_HASSUBFOLDER))
                  {
                        LPSHELLFOLDER      pSHSubFolder;
                        hr = pSHFolder->BindToObject(pIdl, NULL, IID_IShellFolder, (void**)&pSHSubFolder);
                        if(SUCCEEDED(hr))
                        {
                              if(m_iDepth < iDepth)
                                    PopulateSubItem(pSHSubFolder, hItem);
                              //pSHSubFolder->Release();

                              //for my computer (and others) do extra stuff
                              //
                              DoExtraStuff(szFullName, hItem);
                        }
                  }
            }
            hr = pEnumFolders->Next(1, &pIdl, NULL);
      }

      m_iDepth--;
}


void CTargetTree::OnTvnItemexpanding(NMHDR *pNMHDR, LRESULT *pResult)
{
      LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
      HTREEITEM      hItem = pNMTreeView->itemNew.hItem;
      
      if(pNMTreeView->action == TVE_EXPAND)
      {
            if(ItemHasChildren(hItem))
            {
                  CWaitCursor      cw;            
                  HTREEITEM      hChild = GetChildItem(hItem);

                  do
                  {
                        if(!ItemHasChildren(hChild))
                        {
                              _FolderData      *pFD = (_FolderData*)GetItemData(hChild);

                              if(pFD)
                                    if(pFD->pSHF)
                                    {
                                          LPSHELLFOLDER      pSHSubFolder;
                                          HRESULT hr = pFD->pSHF->BindToObject(pFD->pIdl, NULL, IID_IShellFolder, (void**)&pSHSubFolder);
                                          if(SUCCEEDED(hr))
                                          {
                                                PopulateSubItem(pSHSubFolder, hChild, 1);
                                          }
                                    }
                                    //pSHFolder->Release();
                        }
                        hChild = GetNextSiblingItem(hChild);
                  }while(hChild);
            }
      }

      *pResult = 0;
}