Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
main.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 29k
Category:

Windows Kernel

Development Platform:

Visual C++

main.cpp：Code Content
  1. /*----------------------------------------------------------------------------
  2. / Title;
  3. /   main.cpp
  4. /
  5. / Authors;
  6. /   David De Vorchik (daviddv)
  7. /
  8. / Notes;
  9. /   Query the DS, put the results into a list view.
  10. /   This code is pretty lax with Trace and error reporting.
  11. /----------------------------------------------------------------------------*/
  12. #include "windows.h"
  13. #include "windowsx.h"
  14. #include "commctrl.h"
  15. #include "resource.h"
  16. #include "shlobj.h"
  17. #include "shlobjp.h"
  18. #include "shlwapi.h"
  20. //
  21. // Private headers from the shell/dsui projects
  22. //
  24. #include "comctrlp.h"
  25. #include "shellapi.h"
  26. #include "shsemip.h"
  27. #include "shlwapip.h"
  28. #include "common.h"
  30. #pragma hdrstop
  32. #define INITGUID
  33. #include "initguid.h"
  34. #include "cmnquery.h"
  35. #include "cmnquryp.h"
  36. #include "dsquery.h"
  37. #include "dsshell.h"
  40. /*-----------------------------------------------------------------------------
  41. / Globals and stuff
  42. /----------------------------------------------------------------------------*/
  44. struct
  45. {
  46.     INT idsForm;
  47.     CLSID const * pFormCLSID;
  48. }
  49. forms[] =
  50. {
  51.     IDS_FORMUSERS,            &CLSID_DsFindPeople,
  52.     IDS_FORMCOMPUTERS,        &CLSID_DsFindComputer,
  53.     IDS_FORMPRINT,            &CLSID_DsFindPrinter,
  54.     IDS_FORMFILEFOLDERS,      &CLSID_DsFindVolume,
  55.     IDS_FORMDSFOLDERS,        &CLSID_DsFindContainer,
  56.     IDS_FORMOBJECTS,          &CLSID_DsFindObjects,
  57.     IDS_FORMPWELL,            &CLSID_DsFindAdvanced,
  58.     IDS_FORMDC,               &CLSID_DsFindDomainController,
  59. };
  61. #define INI_FILENAME      TEXT("query.ini")
  62. #define DEFAULT_SCOPE     TEXT("LDAP://ntdev")
  63. #define DEFAULT_SAVELOCN  TEXT("c:\temp")
  65. HINSTANCE g_hInstance = NULL;
  66. UINT g_cfDsObjectNames = 0;
  67. UINT g_cfDsQueryParams = 0;
  69. BOOL CALLBACK Query_DlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
  71. void Query_OnInitDialog(HWND hDlg, LPTSTR pDefaultScope);
  72. void Query_OnDoSearch(HWND hDlg);
  73. void Query_BrowseForScope(HWND hDlg);
  74. void Query_BrowseForSaveLocation(HWND hDlg);
  76. int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow);
  78. struct
  79. {
  80.     UINT ids;
  81.     int  fmt;
  82.     int  cx;
  83. }
  84. columnsResults[] =
  85. {
  86.     IDS_OBJECTNAME,     LVCFMT_LEFT, 192,
  87.     IDS_CLASS,          LVCFMT_LEFT, 64,
  88.     IDS_FLAGS,          LVCFMT_LEFT, 64,
  89.     IDS_PROVIDERFLAGS,  LVCFMT_LEFT, 96,
  90. };
  92. struct
  93. {
  94.     UINT ids;
  95.     int  fmt;
  96.     int  cx;
  97. }
  98. columnsPersistInfo[] =
  99. {
  100.     IDS_SECTION,        LVCFMT_LEFT, 128,
  101.     IDS_TAG,            LVCFMT_LEFT, 128,
  102.     IDS_DATA,           LVCFMT_LEFT, 128,
  103. };
  106. //
  107. // Example persisting iface for poulating the list view
  108. //
  110. class CPersistQuery : public IPersistQuery
  111. {
  112.     private:
  113.         ULONG m_cRefCount;
  114.         HWND m_hwndView;
  116.     public:
  117.         CPersistQuery(HWND hwndView);
  119.         // IUnknown
  120.         STDMETHOD(QueryInterface)(REFIID riid, LPVOID* ppvObject);
  121.         STDMETHOD_(ULONG, AddRef)();
  122.         STDMETHOD_(ULONG, Release)();
  124.         // IPersist
  125.         STDMETHOD(GetClassID)(THIS_ CLSID* pClassID);
  127.         // IPersistQuery
  128.         STDMETHOD(WriteString)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, LPCTSTR pValue);
  129.         STDMETHOD(ReadString)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, LPTSTR pBuffer, INT cchBuffer);
  130.         STDMETHOD(WriteInt)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, INT value);
  131.         STDMETHOD(ReadInt)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, LPINT pValue);
  132.         STDMETHOD(WriteStruct)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, LPVOID pStruct, DWORD cbStruct);
  133.         STDMETHOD(ReadStruct)(THIS_ LPCTSTR pSection, LPCTSTR pValueName, LPVOID pStruct, DWORD cbStruct);
  134.         STDMETHOD(Clear)(THIS);
  135. };
  138. /*-----------------------------------------------------------------------------
  139. / Query_DlgProc
  140. / -------------
  141. /   Handle dialog messages for this window.
  142. /
  143. / In:
  144. /   hDlg -> dialog to be initialized
  145. /
  146. / Out:
  147. /   -
  148. /----------------------------------------------------------------------------*/
  149. BOOL CALLBACK Query_DlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
  150. {
  151.     BOOL fResult = FALSE;
  153.     switch ( uMsg )
  154.     {
  155.         case WM_INITDIALOG:
  156.             Query_OnInitDialog(hDlg, (LPTSTR)lParam);
  157.             break;
  159.         case WM_SHOWWINDOW:
  160.         {
  161. #if 0
  162.             if ( wParam )
  163.                 PostMessage(hDlg, WM_COMMAND, IDC_SEARCH, 0);
  164. #endif
  165.             break;
  166.         }
  168.         case WM_COMMAND:
  169.         {
  170.             switch ( LOWORD(wParam) )
  171.             {
  172.                 case IDOK:
  173.                 case IDCANCEL:
  174.                     EndDialog(hDlg, LOWORD(wParam));
  175.                     break;
  177.                 case IDC_SEARCH:
  178.                     Query_OnDoSearch(hDlg);
  179.                     break;
  181.                 case IDC_SCOPEBROWSE:
  182.                     Query_BrowseForScope(hDlg);
  183.                     break;                                       
  185.                 case IDC_SAVEBROWSE:
  186.                     Query_BrowseForSaveLocation(hDlg);
  187.                     break;                                       
  189.                 case IDC_DEFAULTFORM:
  190.                 {
  191.                     EnableWindow(GetDlgItem(hDlg, IDC_FORMLIST), 
  192.                                     IsDlgButtonChecked(hDlg, IDC_DEFAULTFORM)== BST_CHECKED);
  193.                     break;
  194.                 }
  196.                 case IDC_DEFAULTSCOPE:
  197.                 {
  198.                     EnableWindow(GetDlgItem(hDlg, IDC_SCOPELOCN), 
  199.                                     IsDlgButtonChecked(hDlg, IDC_DEFAULTSCOPE)== BST_CHECKED);
  200.                     EnableWindow(GetDlgItem(hDlg, IDC_SCOPEBROWSE), 
  201.                                     IsDlgButtonChecked(hDlg, IDC_DEFAULTSCOPE)== BST_CHECKED);
  202.                     break;
  203.                 }
  205.                 case IDC_SAVEWHERE:
  206.                 {
  207.                     EnableWindow(GetDlgItem(hDlg, IDC_SAVELOCN), 
  208.                                     IsDlgButtonChecked(hDlg, IDC_SAVEWHERE)== BST_CHECKED);
  209.                     EnableWindow(GetDlgItem(hDlg, IDC_SAVEBROWSE), 
  210.                                     IsDlgButtonChecked(hDlg, IDC_SAVEWHERE)== BST_CHECKED);         
  211.                     break;
  212.                 }
  213.                 
  214.                 case IDC_SAVEONCLOSE:
  215.                 {
  216.                     EnableWindow(GetDlgItem(hDlg, IDC_PERSISTDATA), 
  217.                                     IsDlgButtonChecked(hDlg, IDC_SAVEONCLOSE)== BST_CHECKED);
  218.                     break;
  219.                 }
  221.                 case IDC_GETRESULTS:
  222.                 {
  223.                     EnableWindow(GetDlgItem(hDlg, IDC_RESULTS), 
  224.                                     IsDlgButtonChecked(hDlg, IDC_GETRESULTS)== BST_CHECKED);         
  225.                     break;
  226.                 }
  228.                 default:
  229.                     break;
  231.             }
  232.         }
  233.     }
  235.     return fResult;
  236. }
  239. /*-----------------------------------------------------------------------------
  240. / Query_OnInitDialog
  241. / ------------------
  242. /   Initialize the dialog (the result viewer).
  243. /
  244. / In:
  245. /   hDlg -> dialog to be initialized
  246. /   pDefaultScope -> default scope to use when browsing
  247. /
  248. / Out:
  249. /   -
  250. /----------------------------------------------------------------------------*/
  251. void Query_OnInitDialog(HWND hDlg, LPTSTR pDefaultScope)
  252. {
  253.     TCHAR szBuffer[MAX_PATH];
  254.     LV_COLUMN lvc;
  255.     INT i;
  256.     TCHAR szScopeLocn[MAX_PATH];
  257.     TCHAR szSaveLocn[MAX_PATH];    
  258.     BOOL fSingleSelect = FALSE;
  259.     BOOL fDefaultForm = FALSE;
  260.     BOOL fDefaultScope = FALSE;
  261.     BOOL fNoSave = FALSE;
  262.     BOOL fSaveWhere = FALSE;
  263.     BOOL fShowHidden = FALSE;
  264.     BOOL fRemoveScopes = FALSE;
  265.     BOOL fRemoveForms = FALSE;
  266.     BOOL fIssueOnOpen = FALSE;   
  267.     BOOL fShowOptional = FALSE;
  268.     BOOL fAdminBits = FALSE;
  269.     BOOL fHideSearchPane = FALSE;
  270.     BOOL fPersistOnOK = FALSE;
  271.     BOOL fHideMenus = FALSE;
  272.     BOOL fHideSearchUI = FALSE;
  273.     BOOL fGetResults = TRUE;
  274.     UINT iDefaultForm = ARRAYSIZE(forms);
  275.     
  276.     // Setup the columns within the ListView
  278.     for ( i = 0 ; i < ARRAYSIZE(columnsResults) ; i++ )
  279.     {
  280.         LoadString(g_hInstance, columnsResults[i].ids, szBuffer, ARRAYSIZE(szBuffer));
  282.         lvc.mask = LVCF_FMT|LVCF_TEXT|LVCF_WIDTH;
  283.         lvc.fmt = columnsResults[i].fmt;
  284.         lvc.cx = columnsResults[i].cx;
  285.         lvc.pszText = szBuffer;
  287.         ListView_InsertColumn(GetDlgItem(hDlg, IDC_RESULTS), i, &lvc);
  288.     }
  290.     ListView_SetExtendedListViewStyle(GetDlgItem(hDlg, IDC_RESULTS), LVS_EX_FULLROWSELECT);
  292.     // setup columns in persist window
  294.     for ( i = 0 ; i < ARRAYSIZE(columnsPersistInfo) ; i++ )
  295.     {
  296.         LoadString(g_hInstance, columnsPersistInfo[i].ids, szBuffer, ARRAYSIZE(szBuffer));
  298.         lvc.mask = LVCF_FMT|LVCF_TEXT|LVCF_WIDTH;
  299.         lvc.fmt = columnsPersistInfo[i].fmt;
  300.         lvc.cx = columnsPersistInfo[i].cx;
  301.         lvc.pszText = szBuffer;
  303.         ListView_InsertColumn(GetDlgItem(hDlg, IDC_PERSISTDATA), i, &lvc);
  304.     }
  306.     ListView_SetExtendedListViewStyle(GetDlgItem(hDlg, IDC_PERSISTDATA), LVS_EX_FULLROWSELECT);
  308.     // Setup forms in the form list
  310.     for ( i = 0 ; i < ARRAYSIZE(forms); i++ )
  311.     {
  312.         LoadString(g_hInstance, forms[i].idsForm, szBuffer, ARRAYSIZE(szBuffer));
  313.         ComboBox_AddString(GetDlgItem(hDlg, IDC_FORMLIST), szBuffer);
  314.     }
  316.     // Load settings from the ini file and prep the dialog box with the settings
  318.     GetPrivateProfileStruct(TEXT("Query"), TEXT("SingleSelect"),   &fSingleSelect,   SIZEOF(fSingleSelect),   INI_FILENAME);
  319.     GetPrivateProfileStruct(TEXT("Query"), TEXT("DefaultScope"),   &fDefaultScope,   SIZEOF(fDefaultScope),   INI_FILENAME);
  320.     GetPrivateProfileStruct(TEXT("Query"), TEXT("NoSave"),         &fNoSave,         SIZEOF(fNoSave),         INI_FILENAME);
  321.     GetPrivateProfileStruct(TEXT("Query"), TEXT("SaveWhere"),      &fSaveWhere,      SIZEOF(fSaveWhere),      INI_FILENAME);
  322.     GetPrivateProfileStruct(TEXT("Query"), TEXT("DefaultForm"),    &iDefaultForm,    SIZEOF(iDefaultForm),    INI_FILENAME);
  323.     GetPrivateProfileStruct(TEXT("Query"), TEXT("ShowHidden"),     &fShowHidden,     SIZEOF(fShowHidden),     INI_FILENAME);
  324.     GetPrivateProfileStruct(TEXT("Query"), TEXT("RemoveScope"),    &fRemoveScopes,   SIZEOF(fRemoveScopes),   INI_FILENAME);
  325.     GetPrivateProfileStruct(TEXT("Query"), TEXT("RemoveForms"),    &fRemoveForms,    SIZEOF(fRemoveForms),    INI_FILENAME);
  326.     GetPrivateProfileStruct(TEXT("Query"), TEXT("IssueOnOpen"),    &fIssueOnOpen,    SIZEOF(fIssueOnOpen),    INI_FILENAME);
  327.     GetPrivateProfileStruct(TEXT("Query"), TEXT("ShowOptional"),   &fShowOptional,   SIZEOF(fShowOptional),   INI_FILENAME);
  328.     GetPrivateProfileStruct(TEXT("Query"), TEXT("AdmimnBits"),     &fAdminBits,      SIZEOF(fAdminBits),      INI_FILENAME);
  329.     GetPrivateProfileStruct(TEXT("Query"), TEXT("HideSearchPane"), &fHideSearchPane, SIZEOF(fHideSearchPane), INI_FILENAME);
  330.     GetPrivateProfileStruct(TEXT("Query"), TEXT("PersistOnOK"),    &fPersistOnOK,    SIZEOF(fPersistOnOK),    INI_FILENAME);
  331.     GetPrivateProfileStruct(TEXT("Query"), TEXT("HideMenus"),      &fHideMenus,      SIZEOF(fHideMenus),      INI_FILENAME);
  332.     GetPrivateProfileStruct(TEXT("Query"), TEXT("AsFilterDlg"),    &fHideSearchUI,   SIZEOF(fHideSearchUI),   INI_FILENAME);
  333.     GetPrivateProfileStruct(TEXT("Query"), TEXT("GetResults"),     &fGetResults,     SIZEOF(fGetResults),     INI_FILENAME);
  335.     GetPrivateProfileString(TEXT("Query"), TEXT("ScopeLocn"), DEFAULT_SCOPE, szScopeLocn, ARRAYSIZE(szScopeLocn), INI_FILENAME);
  336.     GetPrivateProfileString(TEXT("Query"), TEXT("SaveLocn"), DEFAULT_SAVELOCN , szSaveLocn, ARRAYSIZE(szSaveLocn), INI_FILENAME);
  338.     if ( iDefaultForm == ARRAYSIZE(forms) )
  339.         iDefaultForm = 0;
  340.     else
  341.         fDefaultForm = TRUE;
  343.     ComboBox_SetCurSel(GetDlgItem(hDlg, IDC_FORMLIST), iDefaultForm);
  345.     if ( pDefaultScope )
  346.     {
  347.         SetDlgItemText(hDlg, IDC_SCOPELOCN, pDefaultScope);
  348.         fDefaultScope = TRUE;
  349.     }
  350.     else
  351.     {
  352.         SetDlgItemText(hDlg, IDC_SCOPELOCN, szScopeLocn);
  353.     }
  355.     SetDlgItemText(hDlg, IDC_SAVELOCN, szSaveLocn);
  357.     CheckDlgButton(hDlg, IDC_SINGLESELECT,   fSingleSelect   ? BST_CHECKED:BST_UNCHECKED);
  358.     CheckDlgButton(hDlg, IDC_DEFAULTFORM,    fDefaultForm    ? BST_CHECKED:BST_UNCHECKED);
  359.     CheckDlgButton(hDlg, IDC_DEFAULTSCOPE,   fDefaultScope   ? BST_CHECKED:BST_UNCHECKED);
  360.     CheckDlgButton(hDlg, IDC_NOSAVE,         fNoSave         ? BST_CHECKED:BST_UNCHECKED);
  361.     CheckDlgButton(hDlg, IDC_SAVEWHERE,      fSaveWhere      ? BST_CHECKED:BST_UNCHECKED);
  362.     CheckDlgButton(hDlg, IDC_SHOWHIDDEN,     fShowHidden     ? BST_CHECKED:BST_UNCHECKED);
  363.     CheckDlgButton(hDlg, IDC_NOSCOPES,       fRemoveScopes   ? BST_CHECKED:BST_UNCHECKED);
  364.     CheckDlgButton(hDlg, IDC_NOFORMS,        fRemoveForms    ? BST_CHECKED:BST_UNCHECKED);
  365.     CheckDlgButton(hDlg, IDC_ISSUEONOPEN,    fIssueOnOpen    ? BST_CHECKED:BST_UNCHECKED);
  366.     CheckDlgButton(hDlg, IDC_SHOWOPTIONAL,   fShowOptional   ? BST_CHECKED:BST_UNCHECKED);
  367.     CheckDlgButton(hDlg, IDC_ADMINBITS,      fAdminBits      ? BST_CHECKED:BST_UNCHECKED);
  368.     CheckDlgButton(hDlg, IDC_HIDESEARCHPANE, fHideSearchPane ? BST_CHECKED:BST_UNCHECKED);
  369.     CheckDlgButton(hDlg, IDC_SAVEONCLOSE,    fPersistOnOK    ? BST_CHECKED:BST_UNCHECKED);
  370.     CheckDlgButton(hDlg, IDC_HIDEMENUS,      fHideMenus      ? BST_CHECKED:BST_UNCHECKED);
  371.     CheckDlgButton(hDlg, IDC_HIDESEARCHUI,   fHideSearchUI   ? BST_CHECKED:BST_UNCHECKED);
  372.     CheckDlgButton(hDlg, IDC_GETRESULTS,     fGetResults     ? BST_CHECKED:BST_UNCHECKED);
  374.     EnableWindow(GetDlgItem(hDlg, IDC_FORMLIST), fDefaultForm);
  375.     EnableWindow(GetDlgItem(hDlg, IDC_SCOPELOCN), fDefaultScope);
  376.     EnableWindow(GetDlgItem(hDlg, IDC_SCOPEBROWSE), fDefaultScope);
  377.     EnableWindow(GetDlgItem(hDlg, IDC_SAVELOCN), fSaveWhere);
  378.     EnableWindow(GetDlgItem(hDlg, IDC_SAVEBROWSE), fSaveWhere);
  379.     EnableWindow(GetDlgItem(hDlg, IDC_PERSISTDATA), fPersistOnOK);
  381.     EnableWindow(GetDlgItem(hDlg, IDC_RESULTS), fGetResults);
  382. }
  385. /*-----------------------------------------------------------------------------
  386. / Query_OnDoSearch
  387. / ----------------
  388. /   Query the DS and populate the specified list view with results we get back.
  389. /
  390. / In:
  391. /   hDlg -> parent of find dialog
  392. /
  393. / Out:
  394. /   -
  395. /----------------------------------------------------------------------------*/
  396. void Query_OnDoSearch(HWND hDlg)
  397. {
  398.     HRESULT hr;
  399.     FORMATETC fmte = {CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
  400.     STGMEDIUM medium = { TYMED_NULL, NULL, NULL };
  401.     DSQUERYINITPARAMS dqip;
  402.     OPENQUERYWINDOW oqw;
  403.     LPTSTR pObjectName, pClass;
  404.     ICommonQuery* pCommonQuery = NULL;
  405.     IDataObject* pDataObject = NULL;
  406.     HWND hwndResults;
  407.     LV_ITEM lvi;
  408.     UINT i;
  409.     TCHAR szBuffer[MAX_PATH];
  410.     TCHAR szScopeLocn[MAX_PATH];
  411.     TCHAR szSaveLocn[MAX_PATH];    
  412.     BOOL fSingleSelect = FALSE;
  413.     BOOL fDefaultForm = FALSE;
  414.     BOOL fDefaultScope = FALSE;
  415.     BOOL fNoSave = FALSE;
  416.     BOOL fSaveWhere = FALSE;
  417.     BOOL fShowHidden = FALSE;
  418.     BOOL fRemoveScopes = FALSE;
  419.     BOOL fRemoveForms = FALSE;
  420.     BOOL fIssueOnOpen = FALSE;   
  421.     BOOL fShowOptional = FALSE;
  422.     BOOL fAdminBits = FALSE;
  423.     BOOL fHideSearchPane = FALSE;
  424.     BOOL fPersistOnOK = FALSE;
  425.     BOOL fHideMenus = FALSE;
  426.     BOOL fHideSearchUI = FALSE;
  427.     BOOL fGetResults = TRUE;
  428.     UINT iDefaultForm = ARRAYSIZE(forms);
  429.     CPersistQuery* pPersistQuery = NULL;
  431.     // Read the state from the dialog and write them out
  433.     fSingleSelect = IsDlgButtonChecked(hDlg, IDC_SINGLESELECT) == BST_CHECKED;
  434.     fDefaultForm = IsDlgButtonChecked(hDlg, IDC_DEFAULTFORM) == BST_CHECKED;
  435.     fDefaultScope = IsDlgButtonChecked(hDlg, IDC_DEFAULTSCOPE) == BST_CHECKED;
  436.     fNoSave = IsDlgButtonChecked(hDlg, IDC_NOSAVE) == BST_CHECKED;
  437.     fSaveWhere = IsDlgButtonChecked(hDlg, IDC_SAVEWHERE) == BST_CHECKED;
  438.     fShowHidden = IsDlgButtonChecked(hDlg, IDC_SHOWHIDDEN) == BST_CHECKED;
  439.     fRemoveScopes = IsDlgButtonChecked(hDlg, IDC_NOSCOPES) == BST_CHECKED;
  440.     fRemoveForms = IsDlgButtonChecked(hDlg, IDC_NOFORMS) == BST_CHECKED;
  441.     fIssueOnOpen = IsDlgButtonChecked(hDlg, IDC_ISSUEONOPEN) == BST_CHECKED;
  442.     fShowOptional = IsDlgButtonChecked(hDlg, IDC_SHOWOPTIONAL) == BST_CHECKED;
  443.     fAdminBits = IsDlgButtonChecked(hDlg, IDC_ADMINBITS) == BST_CHECKED;
  444.     fHideSearchPane = IsDlgButtonChecked(hDlg, IDC_HIDESEARCHPANE) == BST_CHECKED;
  445.     fPersistOnOK = IsDlgButtonChecked(hDlg, IDC_SAVEONCLOSE) == BST_CHECKED;
  446.     fHideMenus = IsDlgButtonChecked(hDlg, IDC_HIDEMENUS) == BST_CHECKED;
  447.     fHideSearchUI = IsDlgButtonChecked(hDlg, IDC_HIDESEARCHUI) == BST_CHECKED;
  448.     fGetResults = IsDlgButtonChecked(hDlg, IDC_GETRESULTS) == BST_CHECKED;
  450.     GetDlgItemText(hDlg, IDC_SCOPELOCN, szScopeLocn, ARRAYSIZE(szScopeLocn));
  451.     GetDlgItemText(hDlg, IDC_SAVELOCN, szSaveLocn, ARRAYSIZE(szSaveLocn));
  453.     if ( fDefaultForm )
  454.         iDefaultForm = ComboBox_GetCurSel(GetDlgItem(hDlg, IDC_FORMLIST));
  456.     WritePrivateProfileStruct(TEXT("Query"), TEXT("SingleSelect"),   &fSingleSelect,   SIZEOF(fSingleSelect),   INI_FILENAME);
  457.     WritePrivateProfileStruct(TEXT("Query"), TEXT("DefaultScope"),   &fDefaultScope,   SIZEOF(fDefaultForm),    INI_FILENAME);
  458.     WritePrivateProfileStruct(TEXT("Query"), TEXT("NoSave"),         &fNoSave,         SIZEOF(fNoSave),         INI_FILENAME);
  459.     WritePrivateProfileStruct(TEXT("Query"), TEXT("SaveWhere"),      &fSaveWhere,      SIZEOF(fSaveWhere),      INI_FILENAME);
  460.     WritePrivateProfileStruct(TEXT("Query"), TEXT("DefaultForm"),    &iDefaultForm,    SIZEOF(iDefaultForm),    INI_FILENAME);
  461.     WritePrivateProfileStruct(TEXT("Query"), TEXT("ShowHidden"),     &fShowHidden,     SIZEOF(fShowHidden),     INI_FILENAME);
  462.     WritePrivateProfileStruct(TEXT("Query"), TEXT("RemoveScope"),    &fRemoveScopes,   SIZEOF(fRemoveScopes),   INI_FILENAME);
  463.     WritePrivateProfileStruct(TEXT("Query"), TEXT("RemoveForms"),    &fRemoveForms,    SIZEOF(fRemoveForms),    INI_FILENAME);
  464.     WritePrivateProfileStruct(TEXT("Query"), TEXT("IssueOnOpen"),    &fIssueOnOpen,    SIZEOF(fIssueOnOpen),    INI_FILENAME);
  465.     WritePrivateProfileStruct(TEXT("Query"), TEXT("ShowOptional"),   &fShowOptional,   SIZEOF(fShowOptional),   INI_FILENAME);
  466.     WritePrivateProfileStruct(TEXT("Query"), TEXT("AdmimnBits"),     &fAdminBits,      SIZEOF(fAdminBits),      INI_FILENAME);
  467.     WritePrivateProfileStruct(TEXT("Query"), TEXT("HideSearchPane"), &fHideSearchPane, SIZEOF(fHideSearchPane), INI_FILENAME);
  468.     WritePrivateProfileStruct(TEXT("Query"), TEXT("PersistOnOK"),    &fPersistOnOK,    SIZEOF(fPersistOnOK),    INI_FILENAME);
  469.     WritePrivateProfileStruct(TEXT("Query"), TEXT("HideMenus"),      &fHideMenus,      SIZEOF(fHideMenus),      INI_FILENAME);
  470.     WritePrivateProfileStruct(TEXT("Query"), TEXT("AsFilterDlg"),    &fHideSearchUI,   SIZEOF(fHideSearchUI),   INI_FILENAME);
  471.     WritePrivateProfileStruct(TEXT("Query"), TEXT("GetResults"),     &fGetResults,     SIZEOF(fGetResults),     INI_FILENAME);
  473.     WritePrivateProfileString(TEXT("Query"), TEXT("ScopeLocn"), szScopeLocn, INI_FILENAME);
  474.     WritePrivateProfileString(TEXT("Query"), TEXT("SaveLocn"),  szSaveLocn,  INI_FILENAME);
  476.     // Fix the structures to reflect these options
  478.     if ( FAILED(CoCreateInstance(CLSID_CommonQuery, NULL, CLSCTX_INPROC_SERVER, IID_ICommonQuery, (LPVOID*)&pCommonQuery)) )
  479.         return;
  481.     dqip.cbStruct = SIZEOF(dqip);
  482.     dqip.dwFlags = 0;
  483.     dqip.pDefaultScope = NULL;
  484.     
  485.     oqw.cbStruct = SIZEOF(oqw);
  486.     oqw.dwFlags = 0;
  487.     oqw.clsidHandler = CLSID_DsQuery;
  488.     oqw.pHandlerParameters = &dqip;
  489.     oqw.clsidDefaultForm = CLSID_DsFindObjects;
  491.     if ( fGetResults )
  492.         oqw.dwFlags |= OQWF_OKCANCEL;
  494.     if ( fSingleSelect )
  495.         oqw.dwFlags |= OQWF_SINGLESELECT;
  497.     if ( fDefaultForm )
  498.     {
  499.         oqw.dwFlags |= OQWF_DEFAULTFORM;
  500.         oqw.clsidDefaultForm = *forms[iDefaultForm].pFormCLSID;
  501.     }
  503.     if ( fDefaultScope )
  504.         dqip.pDefaultScope = szScopeLocn;
  506.     if ( fNoSave )
  507.         dqip.dwFlags |= DSQPF_NOSAVE;
  509.     if ( fSaveWhere )
  510.     {
  511.         dqip.dwFlags |= DSQPF_SAVELOCATION;
  512.         dqip.pDefaultSaveLocation = szSaveLocn;
  513.     }
  515.     if ( fShowHidden )
  516.         dqip.dwFlags |= DSQPF_SHOWHIDDENOBJECTS;
  517.     
  518.     if ( fAdminBits )
  519.         dqip.dwFlags |= DSQPF_ENABLEADMINFEATURES|DSQPF_ENABLEADVANCEDFEATURES;
  521.     if ( fRemoveScopes )
  522.         oqw.dwFlags |= OQWF_REMOVESCOPES;
  524.     if ( fRemoveForms )
  525.         oqw.dwFlags |= OQWF_REMOVEFORMS;
  527.     if ( fIssueOnOpen )
  528.         oqw.dwFlags |= OQWF_ISSUEONOPEN;
  530.     if ( fShowOptional )
  531.         oqw.dwFlags |= OQWF_SHOWOPTIONAL;
  533.     if ( fHideSearchPane )
  534.     oqw.dwFlags |= OQWF_HIDESEARCHPANE;
  535.    
  536.     if ( fPersistOnOK )
  537.     {
  538.         pPersistQuery = new CPersistQuery(GetDlgItem(hDlg, IDC_PERSISTDATA));
  540.         if ( pPersistQuery )
  541.         {
  542.             oqw.dwFlags |= OQWF_SAVEQUERYONOK;
  543.             oqw.pPersistQuery = pPersistQuery;
  544.         }
  545.     }
  547.     if ( fHideMenus )
  548.         oqw.dwFlags |= OQWF_HIDEMENUS;
  550.     if ( fHideSearchUI )
  551.         oqw.dwFlags |= OQWF_HIDESEARCHUI;
  553.     // Now display the dialog, and if we succeeded and get an IDataObject then
  554.     // slurp the results into our list view.
  556.     hwndResults = GetDlgItem(hDlg, IDC_RESULTS);
  557.     ListView_DeleteAllItems(hwndResults);
  559.     SetDlgItemText(hDlg, IDC_CLSIDNAMESPACE, TEXT(""));
  560.     SetDlgItemText(hDlg, IDC_ITEMCOUNT, TEXT(""));
  561.     SetDlgItemText(hDlg, IDC_LDAPFILTER, TEXT(""));
  563.     hr = pCommonQuery->OpenQueryWindow(hDlg, &oqw, &pDataObject);
  565.     if ( SUCCEEDED(hr) && pDataObject )
  566.     {
  567.         // get the DSOBJECTNAMES and fill the list view accordingly
  569.         if ( !g_cfDsObjectNames )
  570.             g_cfDsObjectNames = RegisterClipboardFormat(CFSTR_DSOBJECTNAMES);
  572.         fmte.cfFormat = g_cfDsObjectNames;  
  574.         if ( SUCCEEDED(pDataObject->GetData(&fmte, &medium)) )
  575.         {
  576.             LPDSOBJECTNAMES pDsObjects = (LPDSOBJECTNAMES)medium.hGlobal;
  578.             SHStringFromGUID(pDsObjects->clsidNamespace, szBuffer, ARRAYSIZE(szBuffer));
  579.             SetDlgItemText(hDlg, IDC_CLSIDNAMESPACE, szBuffer);
  581.             wsprintf(szBuffer, TEXT("%d"), pDsObjects->cItems);
  582.             SetDlgItemText(hDlg, IDC_ITEMCOUNT, szBuffer);
  584.             for ( i = 0 ; i != pDsObjects->cItems ; i++ )
  585.             {
  586.                 lvi.mask = LVIF_TEXT;
  587.                 lvi.iItem = i;
  588.                 lvi.iSubItem = 0;
  589.                 lvi.pszText = (LPTSTR)ByteOffset(pDsObjects, pDsObjects->aObjects[i].offsetName);
  590.                 ListView_InsertItem(hwndResults, &lvi);
  592.                 ListView_SetItemText(hwndResults, i, 1, (LPTSTR)ByteOffset(pDsObjects, pDsObjects->aObjects[i].offsetClass));            
  594.                 wsprintf(szBuffer, TEXT("%08x"), pDsObjects->aObjects[i].dwFlags);
  595.                 ListView_SetItemText(hwndResults, i, 2, szBuffer);
  597.                 wsprintf(szBuffer, TEXT("%08x"), pDsObjects->aObjects[i].dwProviderFlags);
  598.                 ListView_SetItemText(hwndResults, i, 3, szBuffer);
  599.             }
  601.             ReleaseStgMedium(&medium);
  602.         }
  604.         // now get the DSQUERYPARAMS and lets get the filter string
  606.         if ( !g_cfDsQueryParams )
  607.             g_cfDsQueryParams = RegisterClipboardFormat(CFSTR_DSQUERYPARAMS);
  609.         fmte.cfFormat = g_cfDsQueryParams;  
  611.         if ( SUCCEEDED(pDataObject->GetData(&fmte, &medium)) )
  612.         {
  613.             LPDSQUERYPARAMS pDsQueryParams = (LPDSQUERYPARAMS)medium.hGlobal;
  614.             LPWSTR pFilter = (LPTSTR)ByteOffset(pDsQueryParams, pDsQueryParams->offsetQuery);
  616.             SetDlgItemText(hDlg, IDC_LDAPFILTER, pFilter);
  617.         }
  619.         pDataObject->Release();
  620.     }
  622.     pCommonQuery->Release();
  624.     if ( pPersistQuery )
  625.         pPersistQuery->Release();
  626. }
  629. /*-----------------------------------------------------------------------------
  630. / Query_BrowseForScope
  631. / --------------------
  632. /   Query the DS and populate the specified list view with results we get back.
  633. /
  634. / In:
  635. /   hDlg -> parent of find dialog
  636. /
  637. / Out:
  638. /   -
  639. /----------------------------------------------------------------------------*/
  640. void Query_BrowseForScope(HWND hDlg)
  641. {
  642.     DSBROWSEINFO dsbi;
  643.     TCHAR szBuffer[MAX_PATH];
  645.     GetDlgItemText(hDlg, IDC_SCOPELOCN, szBuffer, ARRAYSIZE(szBuffer));        
  647.     dsbi.cbStruct = SIZEOF(dsbi);
  648.     dsbi.hwndOwner = hDlg;
  649.     dsbi.pszCaption = TEXT("Browse for default scope");
  650.     dsbi.pszTitle = TEXT("Select a location to start the query from");
  651.     dsbi.pszRoot = NULL;
  652.     dsbi.pszPath = szBuffer;
  653.     dsbi.cchPath = ARRAYSIZE(szBuffer);
  654.     dsbi.dwFlags = DSBI_ENTIREDIRECTORY;
  655.     dsbi.pfnCallback = NULL;
  656.     dsbi.lParam = (LPARAM)0;
  658.     if ( lstrlen(szBuffer) )
  659.         dsbi.dwFlags |= DSBI_EXPANDONOPEN;
  661.     if ( IDOK == DsBrowseForContainer(&dsbi) )
  662.         SetDlgItemText(hDlg, IDC_SCOPELOCN, szBuffer);        
  663. }
  666. /*-----------------------------------------------------------------------------
  667. / Query_BrowseForSaveLocation
  668. / ---------------------------
  669. /   Browse for a new save location setting the IDC_SAVELOCN control
  670. /   to contain the new string.
  671. /
  672. / In:
  673. /   hDlg -> parent of find dialog
  674. /
  675. / Out:
  676. /   -
  677. /----------------------------------------------------------------------------*/
  678. void Query_BrowseForSaveLocation(HWND hDlg)
  679. {
  680.     LPITEMIDLIST pidl;
  681.     BROWSEINFO bi;
  682.     TCHAR szBuffer[MAX_PATH];
  684.     bi.hwndOwner = hDlg;
  685.     bi.pidlRoot = NULL;
  686.     bi.pszDisplayName = NULL;
  687.     bi.lpszTitle = TEXT("Select location to save searches to by default.");
  688.     bi.ulFlags = 0;
  689.     bi.lpfn = NULL;
  690.     bi.lParam = 0;
  691.     bi.iImage = 0;
  693.     pidl = SHBrowseForFolder(&bi);
  695.     if ( pidl )
  696.     {    
  697.         if ( SHGetPathFromIDList(pidl, szBuffer) )
  698.             SetDlgItemText(hDlg, IDC_SAVELOCN, szBuffer);
  700.         ILFree(pidl);
  701.     }
  702. }
  705. /*-----------------------------------------------------------------------------
  706. / Main function for the app - handle initalisation etc.
  707. /----------------------------------------------------------------------------*/
  708. int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
  709. {
  710.     TCHAR szBuffer[MAX_PATH];
  711.     LPTSTR pDefaultScope = NULL;
  713.     g_hInstance = hInstance;
  715.     InitCommonControls();
  716.     CoInitialize(NULL);    
  718.     if ( lpCmdLine && *lpCmdLine )
  719.     {
  720.         MultiByteToWideChar(CP_ACP, 0, lpCmdLine, -1, szBuffer, ARRAYSIZE(szBuffer));
  721.         pDefaultScope = szBuffer;
  722.     }
  724.     DialogBoxParam(g_hInstance, MAKEINTRESOURCE(IDD_SEARCHDS), NULL, Query_DlgProc, (LPARAM)pDefaultScope);
  726.     CoUninitialize();
  728.     return 0;
  729. }
  732. /*-----------------------------------------------------------------------------
  733. / Constructor / IUnknown methods
  734. /----------------------------------------------------------------------------*/
  736. CPersistQuery::CPersistQuery(HWND hwndView)
  737. {
  738.     m_cRefCount = 1;
  739.     m_hwndView = hwndView;
  740. }
  742. //
  743. // IUnknown methods
  744. //
  746. STDMETHODIMP CPersistQuery::QueryInterface(REFIID riid, LPVOID* ppvObject)
  747. {
  748.     if ( IsEqualIID(riid, IID_IPersistQuery) )
  749.     {
  750.         *ppvObject = (LPVOID)(IPersistQuery*)this;
  751.         return S_OK;
  752.     }
  754.     return E_NOTIMPL;
  755. }
  757. STDMETHODIMP_(ULONG) CPersistQuery::AddRef()
  758. {
  759.     return ++m_cRefCount;
  760. }
  762. STDMETHODIMP_(ULONG) CPersistQuery::Release()
  763. {
  764.     if ( --m_cRefCount == 0 )
  765.     {
  766.         delete this;
  767.         return 0;
  768.     }
  770.     return m_cRefCount;
  771. }
  774. /*-----------------------------------------------------------------------------
  775. / IPersist methods
  776. /----------------------------------------------------------------------------*/
  778. STDMETHODIMP CPersistQuery::GetClassID(THIS_ CLSID* pClassID)
  779. {
  780.     return E_NOTIMPL;
  781. }
  784. /*-----------------------------------------------------------------------------
  785. / IPersistQuery methods
  786. /----------------------------------------------------------------------------*/
  788. VOID _AddResultToView(HWND hwndView, LPCTSTR pSection, LPCTSTR pTag, LPCTSTR pValue)
  789. {
  790.     LV_ITEM lvi;
  791.     INT i;
  793.     lvi.mask = LVIF_TEXT;
  794.     lvi.iItem = 0x7fffffff;
  795.     lvi.iSubItem = 0;
  796.     lvi.pszText = (LPTSTR)pSection;
  797.     i = ListView_InsertItem(hwndView, &lvi);
  799.     if ( i >= 0 )
  800.     {
  801.         ListView_SetItemText(hwndView, i, 1, (LPTSTR)pTag);
  802.         ListView_SetItemText(hwndView, i, 2, (LPTSTR)pValue);
  803.     }
  804. }
  806. STDMETHODIMP CPersistQuery::WriteString(THIS_ LPCTSTR pSection, LPCTSTR pKey, LPCTSTR pValue)
  807. {
  808.     _AddResultToView(m_hwndView, pSection, pKey, pValue);
  809.     return S_OK;
  810. }
  812. STDMETHODIMP CPersistQuery::ReadString(THIS_ LPCTSTR pSection, LPCTSTR pKey, LPTSTR pBuffer, INT cchBuffer)
  813. {
  814.     return E_NOTIMPL;
  815. }
  817. STDMETHODIMP CPersistQuery::WriteInt(THIS_ LPCTSTR pSection, LPCTSTR pKey, INT value)
  818. {
  819.     TCHAR szBuffer[64];
  820.     wsprintf(szBuffer, TEXT("%d"), value);
  821.     _AddResultToView(m_hwndView, pSection, pKey, szBuffer);
  822.     return S_OK;
  823. }
  825. STDMETHODIMP CPersistQuery::ReadInt(THIS_ LPCTSTR pSection, LPCTSTR pKey, LPINT pValue)
  826. {
  827.     return E_NOTIMPL;
  828. }
  830. STDMETHODIMP CPersistQuery::WriteStruct(THIS_ LPCTSTR pSection, LPCTSTR pKey, LPVOID pStruct, DWORD cbStruct)
  831. {
  832.     TCHAR szBuffer[128];
  833.     wsprintf(szBuffer, TEXT("Structure (%d byte in size)"), cbStruct);
  834.     _AddResultToView(m_hwndView, pSection, pKey, szBuffer);
  835.     return S_OK;
  836. }
  838. STDMETHODIMP CPersistQuery::ReadStruct(THIS_ LPCTSTR pSection, LPCTSTR pKey, LPVOID pStruct, DWORD cbStruct)
  839. {
  840.     return E_NOTIMPL;
  841. }
  843. STDMETHODIMP CPersistQuery::Clear(THIS)
  844. {
  845.     ListView_DeleteAllItems(m_hwndView);
  846.     return S_OK;
  847. }