//+--------------------------------------------------------------------------- // // Microsoft Windows // Copyright (C) Microsoft Corporation, 1997. // // File: C M D T A B L E . C P P // // Contents: Command-table code -- determines which menu options are // available by the selection count, among other criteria // // Notes: // // Author: jeffspr 28 Jan 1998 // //---------------------------------------------------------------------------- #include "pch.h" #pragma hdrstop #include "foldinc.h" // Standard shell\folder includes #include "foldres.h" // Folder resource IDs #include "nsres.h" // Netshell strings #include "cmdtable.h" // Header for this file #include "ncperms.h" // For checking User's rights on actions/menu items #include "cfutils.h" #include "hnetcfg.h" #include "lm.h" //---[ Constants ]------------------------------------------------------------ const DWORD NCCF_ALL = 0xffffffff; // NCCF_ALL - For all Characteristics const DWORD S_REMOVE = 2; #define TRACESTRLEN 65535 // Active must be a subset of Visible (Opening connections folder using a CHK DLL asserts integrity of table). // DO NOT use NCM_ and NCS_ flags in this table!!! You should use NBM_ and NBS_ flags over here. COMMANDENTRY g_cteCommandMatrix[] = { //iCommandId dwDefaultPriority dwFlags HrEnableDisableCB HrCustomMenuStringCB // | | dwValidWhen | | | // | | | | dwMediaTypeVisible | | dwMediaTypeActive // | | | | dwStatusVisible | | dwStatusActive // | | | | dwCharacteristicsVisible | | dwCharacteristicsActive // | | | | (VISIBLE) | | (ACTIVE... (NOT REMOVED)) // v v v v | v v | // v v { CMIDM_HOMENET_WIZARD, 5, NCWHEN_ANYSELECT, NB_REMOVE_TOPLEVEL_ITEM,HrIsHomeNewWizardSupported, NULL, NBM_HNW_WIZARD, NBM_HNW_WIZARD, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_NET_TROUBLESHOOT, 5, NCWHEN_TOPLEVEL, NB_REMOVE_TOPLEVEL_ITEM,HrIsTroubleShootSupported, NULL, NBM_NOMEDIATYPE, NBM_NOMEDIATYPE, // Media Type NBS_NONE, NBS_NONE, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_CONMENU_ADVANCED_CONFIG, 5, NCWHEN_TOPLEVEL, NB_NO_FLAGS, NULL, NULL, NBM_NOMEDIATYPE, NBM_NOMEDIATYPE, // Media Type NBS_NONE, NBS_NONE, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_CONMENU_DIALUP_PREFS, 5, NCWHEN_TOPLEVEL, NB_NO_FLAGS, NULL, NULL, NBM_NOMEDIATYPE, NBM_NOMEDIATYPE, // Media Type NBS_NONE, NBS_NONE, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_NEW_CONNECTION, 5, NCWHEN_ANYSELECT, NB_VERB, HrIsNCWSupported, NULL, NBM_MNC_WIZARD, NBM_MNC_WIZARD, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_CONNECT, 3, NCWHEN_ONESELECT, NB_VERB, NULL, NULL, NBM_SHAREDACCESSHOST_RAS | NBM_ISRASTYPE, NBM_SHAREDACCESSHOST_RAS | NBM_ISRASTYPE, // Media Type NBS_HW_ISSUE | NBS_DISCONNECTED | NBS_CONNECTING, NBS_DISCONNECTED, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_DISCONNECT, 0, NCWHEN_ONESELECT, NB_VERB, NULL, NULL, NBM_SHAREDACCESSHOST_RAS|NBM_ISRASTYPE,NBM_SHAREDACCESSHOST_RAS|NBM_ISRASTYPE, // Media Type NBS_IS_CONNECTED | NBS_DISCONNECTING, NBS_IS_CONNECTED, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_ENABLE, 3, NCWHEN_ONESELECT, NB_VERB, NULL, NULL, NBM_SHAREDACCESSHOST_LAN|NBM_ISLANTYPE,NBM_SHAREDACCESSHOST_LAN|NBM_ISLANTYPE, // Media Type NBS_HW_ISSUE | NBS_DISCONNECTED | NBS_CONNECTING, NBS_DISCONNECTED, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_DISABLE, 0, NCWHEN_ONESELECT, NB_VERB, NULL, NULL, NBM_SHAREDACCESSHOST_LAN|NBM_ISLANTYPE,NBM_SHAREDACCESSHOST_LAN|NBM_ISLANTYPE, // Media Type NBS_DISCONNECTING | NBS_IS_CONNECTED | NBS_MEDIA_DISCONNECTED | NBS_INVALID_ADDRESS, NBS_NOT_DISCONNECT, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_WZCDLG_SHOW, 4, NCWHEN_ONESELECT, NB_VERB, HrIsMediaWireless, NULL, NBM_LAN, NBM_LAN, // Media Type NBS_NOT_DISCONNECT, NBS_NOT_DISCONNECT, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_STATUS, 5, NCWHEN_ONESELECT, NB_VERB, NULL, NULL, NBM_INCOMING | NBM_SHAREDACCESSHOST_LAN|NBM_SHAREDACCESSHOST_RAS|NBM_ISCONNECTIONTYPE, NBM_INCOMING | NBM_SHAREDACCESSHOST_LAN|NBM_SHAREDACCESSHOST_RAS|NBM_ISCONNECTIONTYPE, NBS_ANY, NBS_IS_CONNECTED | NBS_INVALID_ADDRESS, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_FIX, 0, NCWHEN_SOMESELECT, NB_NO_FLAGS, NULL, NULL, NBM_ISLANTYPE, NBM_ISLANTYPE, // Media Type NBS_IS_CONNECTED | NBS_DISCONNECTING| NBS_MEDIA_DISCONNECTED| NBS_INVALID_ADDRESS, NBS_INVALID_ADDRESS | NBS_IS_CONNECTED, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_SEPARATOR, 0,0,0,0,0, 0,0, 0,0, 0,0 }, { CMIDM_SET_DEFAULT, 0, NCWHEN_ONESELECT, NB_NEGATE_CHAR_MATCH, NULL, NULL, NBM_INCOMING | NBM_ISRASTYPE, NBM_INCOMING | NBM_ISRASTYPE, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_INCOMING_ONLY | NCCF_DEFAULT, NCCF_INCOMING_ONLY | NCCF_DEFAULT}, // Characteristics { CMIDM_UNSET_DEFAULT, 0, NCWHEN_ONESELECT, NB_NO_FLAGS, NULL, NULL, NBM_INCOMING | NBM_ISRASTYPE, NBM_INCOMING | NBM_ISRASTYPE, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_DEFAULT, NCCF_DEFAULT}, // Characteristics { CMIDM_SEPARATOR, 0,0,0,0,0, 0,0, 0,0, 0,0 }, { CMIDM_CREATE_BRIDGE, 0, NCWHEN_ANYSELECT, NB_NO_FLAGS, HrIsBridgeSupported, NULL, NBM_LAN, NBM_LAN, // Media Type NBS_IS_CONNECTED | NBS_DISCONNECTING| NBS_MEDIA_DISCONNECTED| NBS_INVALID_ADDRESS, NBS_DISCONNECTING|NBS_IS_CONNECTED|NBS_MEDIA_DISCONNECTED|NBS_INVALID_ADDRESS, // Status NCCF_ALL, NCCF_ALL }, // Characteristics { CMIDM_CONMENU_CREATE_BRIDGE,0, NCWHEN_TOPLEVEL,NB_REMOVE_TOPLEVEL_ITEM,HrIsBridgeSupported, NULL, NBM_NOMEDIATYPE, NBM_NOMEDIATYPE, // Media Type NBS_NONE, NBS_NONE, // Status NCCF_ALL, NCCF_ALL}, // Characteristics { CMIDM_ADD_TO_BRIDGE, 0, NCWHEN_SOMESELECT, NB_NEGATE_CHAR_MATCH, HrIsBridgeSupported, NULL, NBM_LAN, NBM_LAN, // Media Type NBS_IS_CONNECTED | NBS_DISCONNECTING| NBS_MEDIA_DISCONNECTED| NBS_INVALID_ADDRESS, NBS_DISCONNECTING|NBS_IS_CONNECTED|NBS_MEDIA_DISCONNECTED|NBS_INVALID_ADDRESS, // Status NCCF_BRIDGED | NCCF_FIREWALLED | NCCF_SHARED, NCCF_BRIDGED | NCCF_FIREWALLED | NCCF_SHARED }, // Characteristics { CMIDM_REMOVE_FROM_BRIDGE, 0, NCWHEN_SOMESELECT,NB_NO_FLAGS, HrIsBridgeSupported, NULL, NBM_LAN, NBM_LAN, // Media Type NBS_IS_CONNECTED | NBS_DISCONNECTING| NBS_MEDIA_DISCONNECTED| NBS_INVALID_ADDRESS, NBS_DISCONNECTING|NBS_IS_CONNECTED|NBS_MEDIA_DISCONNECTED|NBS_INVALID_ADDRESS, // Status NCCF_BRIDGED, NCCF_BRIDGED }, // Characteristics { CMIDM_SEPARATOR, 0,0,0,0,0, 0,0, 0,0, 0,0 }, { CMIDM_CREATE_COPY, 0, NCWHEN_ONESELECT, NB_NEGATE_VIS_CHAR_MATCH, NULL, NULL, NBM_INCOMING | NBM_SHAREDACCESSHOST_RAS | NBM_ISRASTYPE, NBM_INCOMING | NBM_SHAREDACCESSHOST_RAS | NBM_ISRASTYPE, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_INCOMING_ONLY, NCCF_ALLOW_DUPLICATION}, // Characteristics { CMIDM_SEPARATOR, 0,0,0,0,0, 0,0, 0,0, 0,0 }, { CMIDM_CREATE_SHORTCUT, 0, NCWHEN_ONESELECT, NB_NEGATE_CHAR_MATCH, NULL, NULL, NBM_ANY, NBM_ANY, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_INCOMING_ONLY, NCCF_INCOMING_ONLY}, // Characteristics { CMIDM_DELETE, 0, NCWHEN_SOMESELECT, NB_NO_FLAGS, NULL, NULL, NBM_NOTWIZARD, NBM_NOTWIZARD, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_ALL, NCCF_ALLOW_REMOVAL}, // Characteristics { CMIDM_RENAME, 0, NCWHEN_ONESELECT, NB_NO_FLAGS, HrCanRenameConnection, NULL, NBM_NOTWIZARD, NBM_NOTWIZARD, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_ALL, NCCF_ALLOW_RENAME}, // Characteristics { CMIDM_SEPARATOR, 0,0,0,0,0, 0,0, 0,0, 0,0 }, { CMIDM_PROPERTIES, 2, NCWHEN_ONESELECT, NB_NO_FLAGS, HrCanShowProperties, NULL, NBM_INCOMING|NBM_SHAREDACCESSHOST_LAN|NBM_SHAREDACCESSHOST_RAS|NBM_ISCONNECTIONTYPE, NBM_INCOMING|NBM_SHAREDACCESSHOST_LAN|NBM_SHAREDACCESSHOST_RAS|NBM_ISCONNECTIONTYPE, // Media Type NBS_ANY, NBS_ANY, // Status NCCF_ALL, NCCF_ALL}, // Characteristics }; // What is the difference between NCCF_INCOMING_ONLY, ~NCCF_INCOMING_ONLY and NCCF_INCOMING_ONLY + NB_NEGATE_CHAR_MATCH? // NCCF_INCOMING_ONLY | NCCF_ALLOW_REMOVAL means: NCCF_INCOMING_ONLY or NCCF_ALLOW_REMOVAL or BOTH should be set. // ~NCCF_INCOMING_ONLY means: One or flag (irrespective of NCCF_INCOMING_ONLY flag) should be set. // NB_NEGATE_CHAR_MATCH + NCCF_INCOMING_ONLY means: Check that NCCF_INCOMING_ONLY is not set. const DWORD g_cteCommandMatrixCount = celems(g_cteCommandMatrix); COMMANDPERMISSIONSENTRY g_cteCommandPermissionsMatrix[] = { { CMIDM_NEW_CONNECTION, NBM_ANY, NCCF_ALL, NB_TOPLEVEL_PERM | NB_REMOVE_IF_NOT_MATCH,NBPERM_NewConnectionWizard, APPLY_TO_ALL_USERS }, { CMIDM_CONNECT, NBM_ISRASTYPE, NCCF_ALL, NB_NO_FLAGS, NBPERM_RasConnect, APPLY_TO_ALL_USERS }, { CMIDM_CONNECT, NBM_SHAREDACCESSHOST_RAS, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_ALL_USERS }, { CMIDM_DISCONNECT, NBM_ISRASTYPE, NCCF_INCOMING_ONLY, NB_NEGATE_CHAR_MATCH,NBPERM_RasConnect, APPLY_TO_ALL_USERS }, { CMIDM_DISCONNECT, NBM_ISRASTYPE, NCCF_INCOMING_ONLY, NB_NO_FLAGS, NBPERM_RasConnect, APPLY_TO_ADMIN }, { CMIDM_DISCONNECT, NBM_SHAREDACCESSHOST_RAS, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_ALL_USERS }, { CMIDM_ENABLE, NBM_LAN, NCCF_ALL, NB_NO_FLAGS, NBPERM_LanConnect, APPLY_TO_ALL_USERS }, { CMIDM_ENABLE, NBM_SHAREDACCESSHOST_LAN, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_ALL_USERS }, { CMIDM_ENABLE, NBM_BRIDGE, NCCF_ALL, NB_NO_FLAGS, NBPERM_AllowNetBridge_NLA, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_DISABLE, NBM_LAN, NCCF_ALL, NB_NO_FLAGS, NBPERM_LanConnect, APPLY_TO_ALL_USERS }, { CMIDM_DISABLE, NBM_SHAREDACCESSHOST_LAN, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_ALL_USERS }, { CMIDM_DISABLE, NBM_BRIDGE, NCCF_ALL, NB_NO_FLAGS, NBPERM_AllowNetBridge_NLA, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_STATUS, NBM_ANY, NCCF_INCOMING_ONLY, NB_NEGATE_CHAR_MATCH,NBPERM_Statistics, APPLY_TO_ALL_USERS }, { CMIDM_STATUS, NBM_ANY, NCCF_INCOMING_ONLY, NB_NO_FLAGS, NBPERM_Statistics, APPLY_TO_ADMIN }, { CMIDM_CREATE_BRIDGE, NBM_ANY, NCCF_ALL, NB_NO_FLAGS, NBPERM_AllowNetBridge_NLA, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_ADD_TO_BRIDGE, NBM_ANY, NCCF_ALL, NB_NO_FLAGS, NBPERM_AllowNetBridge_NLA, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_REMOVE_FROM_BRIDGE, NBM_ANY, NCCF_ALL, NB_NO_FLAGS, NBPERM_AllowNetBridge_NLA, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_CREATE_COPY, NBM_ANY, NCCF_ALL_USERS, NB_NO_FLAGS, NBPERM_NewConnectionWizard | NBPERM_RasAllUserProperties, APPLY_TO_ALL_USERS }, { CMIDM_CREATE_COPY, NBM_ANY, NCCF_ALL_USERS, NB_NEGATE_CHAR_MATCH,NBPERM_NewConnectionWizard, APPLY_TO_ALL_USERS }, { CMIDM_FIX, NBM_ANY, NCCF_ALL, NB_NO_FLAGS, NBPERM_Repair, APPLY_TO_POWERUSERSPLUS }, { CMIDM_DELETE, NBM_ANY, NCCF_ALL_USERS, NB_NO_FLAGS, NBPERM_DeleteConnection | NBPERM_DeleteAllUserConnection, APPLY_TO_ALL_USERS }, { CMIDM_DELETE, NBM_ANY, NCCF_ALL_USERS, NB_NEGATE_CHAR_MATCH,NBPERM_DeleteConnection, APPLY_TO_ALL_USERS }, { CMIDM_SET_DEFAULT, NBM_INCOMING | NBM_ISRASTYPE, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_UNSET_DEFAULT, NBM_INCOMING | NBM_ISRASTYPE, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_OPS_OR_ADMIN }, { CMIDM_CONMENU_ADVANCED_CONFIG, NBM_ANY, NCCF_ALL_USERS, NB_TOPLEVEL_PERM, NBPERM_AdvancedSettings, APPLY_TO_ADMIN }, { CMIDM_CONMENU_DIALUP_PREFS, NBM_ANY, NCCF_ALL_USERS, NB_TOPLEVEL_PERM, NBPERM_DialupPrefs, APPLY_TO_ALL_USERS }, { CMIDM_PROPERTIES, NBM_INCOMING, NCCF_ALL, NB_NO_FLAGS, NBPERM_Always, APPLY_TO_ADMIN }, { CMIDM_PROPERTIES, NBM_ISRASTYPE, NCCF_ALL_USERS, NB_NEGATE_CHAR_MATCH,NBPERM_RasMyProperties, APPLY_TO_ALL_USERS }, { CMIDM_PROPERTIES, NBM_ISRASTYPE, NCCF_ALL_USERS, NB_NO_FLAGS, NBPERM_RasAllUserProperties, APPLY_TO_ALL_USERS }, { CMIDM_PROPERTIES, NBM_LAN, NCCF_ALL, NB_NO_FLAGS, NBPERM_LanProperties, APPLY_TO_ALL_USERS } }; const DWORD g_cteCommandPermissionsMatrixCount = celems(g_cteCommandPermissionsMatrix); SFVCOMMANDMAP g_cteSFVCommandMap[] = { { SFVIDM_FILE_DELETE, CMIDM_DELETE}, { SFVIDM_FILE_LINK, CMIDM_CREATE_SHORTCUT}, { SFVIDM_FILE_PROPERTIES, CMIDM_PROPERTIES}, { SFVIDM_FILE_RENAME, CMIDM_RENAME} }; const DWORD g_cteSFVCommandMapCount = celems(g_cteSFVCommandMap); CMDCHKENTRY g_cceFolderChecks[] = { // command id // currently checked // | new check state // | | // v v { CMIDM_CONMENU_OPERATOR_ASSIST, false, false } }; const DWORD g_nFolderCheckCount = celems(g_cceFolderChecks); inline DWORD dwNegateIf(IN DWORD dwInput, IN DWORD dwFlags, IN DWORD dwNegateCondition) { if (dwFlags & dwNegateCondition) { return ~dwInput; } else { return dwInput; } } inline BOOL bContains(IN DWORD dwContainee, IN DWORD dwContainer, IN DWORD dwFlags, IN DWORD dwContaineeNegateCondition, IN DWORD dwContainerNegateCondition) { dwContainer = dwNegateIf(dwContainer, dwFlags, dwContainerNegateCondition); dwContainee = dwNegateIf(dwContainee, dwFlags, dwContaineeNegateCondition); if ( (dwContainee & dwContainer) != dwContainee) { return FALSE; } else { return TRUE; } } //+--------------------------------------------------------------------------- // // Member: HrAssertIntegrityAgainstOldMatrix // // Purpose: Asserts the internal integrity of the Command Matrix // Currently checks for: // 1. No duplicate CMDIDs // 2. Each NCWHEN flag at least specified NCWHEN_ONESELECT // // Arguments: // none // // Returns: // S_OK is succeeded // E_FAIL if not // // Author: deonb 8 Feb 2001 // // Notes: Asserts on failure // HRESULT HrAssertCommandMatrixIntegrity() { TraceFileFunc(ttidMenus); HRESULT hr = S_OK; LPSTR szErr = new CHAR[TRACESTRLEN]; for (DWORD x = 0; x < g_cteCommandMatrixCount; x++) { // Check that there isn't another entry with the same CommandID and Media Type. const COMMANDENTRY& cte = g_cteCommandMatrix[x]; if (CMIDM_SEPARATOR == cte.iCommandId) { continue; } for (DWORD y = x + 1; y < g_cteCommandMatrixCount; y++) { const COMMANDENTRY& ctecmp = g_cteCommandMatrix[y]; if (cte.iCommandId == ctecmp.iCommandId) { sprintf(szErr, "Multiple lines (%d and %d) in the COMMANDENTRY table describe the same CmdID", x+1, y+1); AssertSz(FALSE, szErr); hr = E_FAIL; } } if ( !bContains(cte.dwCharacteristicsActive, cte.dwCharacteristicsVisible, cte.dwFlags, NB_NEGATE_ACT_CHAR_MATCH, NB_NEGATE_VIS_CHAR_MATCH) ) { sprintf(szErr, "Row %d. ACTIVE flags not a subset of VISIBLE flags for Characteristics? ", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } if ( !bContains(cte.dwStatusActive, cte.dwStatusVisible, cte.dwFlags, NB_NEGATE_ACT_NBS_MATCH, NB_NEGATE_VIS_NBS_MATCH) ) { sprintf(szErr, "Row %d. ACTIVE flags not a subset of VISIBLE flags for Status... did you use NCS_ instead of NBM_? ", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } if ( !bContains(cte.dwMediaTypeActive, cte.dwMediaTypeVisible, cte.dwFlags, NB_NEGATE_ACT_NBM_MATCH, NB_NEGATE_VIS_NBM_MATCH) ) { sprintf(szErr, "Row %d. ACTIVE flags not a subset of VISIBLE flags for MediaType... did you use NCM_ instead of NBM_? ", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } } // Assert the permissions table for (x = 0; x < g_cteCommandPermissionsMatrixCount; x++) { const COMMANDPERMISSIONSENTRY cpe = g_cteCommandPermissionsMatrix[x]; // Check that each CMD entry has a corresponding entry in the Command Table BOOL bFound = FALSE; for (DWORD y = 0; y < g_cteCommandMatrixCount; y++) { const COMMANDENTRY& ctecmp = g_cteCommandMatrix[y]; if (cpe.iCommandId == ctecmp.iCommandId) { bFound = TRUE; if ( (cpe.dwMediaType != NBM_ANY) && ((cpe.dwMediaType & ctecmp.dwMediaTypeActive) != cpe.dwMediaType) ) { sprintf(szErr, "A permission has been specified in the Permissions table (row %d) for a MediaType that is not active in the Command Table (row %d)... did you use NCM_ instead of NBM_?", x+1, y+1); AssertSz(FALSE, szErr); hr = E_FAIL; } // if ( (cpe.dwCharacteristicsActive != NCCF_ALL) && // ((cpe.dwCharacteristicsActive & ctecmp.dwCharacteristicsActive) != cpe.dwCharacteristicsActive) ) // { // sprintf(szErr, "A permission has been specified in the Permissions table (row %d) for a Characteristics that is not active in the Command Table (row %d)", x+1, y+1); // AssertSz(FALSE, szErr); // hr = E_FAIL; // } } } if (!bFound) { sprintf(szErr, "An entry has been found in the Permissions table (row %d) without a corresponding CMDID entry in the Command Table", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } // Check that no CmdId/MediaType/Characteristics has been duplicated for (y = x + 1; y < g_cteCommandPermissionsMatrixCount; y++) { const COMMANDPERMISSIONSENTRY& cpecmp = g_cteCommandPermissionsMatrix[y]; if ( (cpe.iCommandId == cpecmp.iCommandId) && (dwNegateIf(cpe.dwMediaType, cpe.dwFlags, NB_NEGATE_NBM_MATCH) & dwNegateIf(cpecmp.dwMediaType, cpecmp.dwFlags, NB_NEGATE_NBM_MATCH)) && (dwNegateIf(cpe.dwCharacteristicsActive, cpe.dwFlags, NB_NEGATE_CHAR_MATCH) & dwNegateIf(cpecmp.dwCharacteristicsActive, cpecmp.dwFlags, NB_NEGATE_CHAR_MATCH)) ) { sprintf(szErr, "Multiple lines (%d and %d) in the COMMANDENTRY table describe the same CmdID/MediaType/Characteristics combo", x+1, y+1); AssertSz(FALSE, szErr); hr = E_FAIL; } } if (! ((APPLY_TO_NETCONFIGOPS & cpe.ncpAppliesTo) || (APPLY_TO_USER & cpe.ncpAppliesTo) || (APPLY_TO_ADMIN & cpe.ncpAppliesTo) || (APPLY_TO_POWERUSERS & cpe.ncpAppliesTo) )) { sprintf(szErr, "Lines (%d) in the Permissionstable - permissions must apply to someone", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } // !!(A & B) != !!(A & C) means: If either B or C is set in A, both B and C must be set (or neither). I hope... // kill me... kill me now. if ((!!(cpe.dwFlags & NB_NEGATE_VIS_NBM_MATCH) != !!(cpe.dwFlags & NB_NEGATE_ACT_NBM_MATCH)) || (!!(cpe.dwFlags & NB_NEGATE_VIS_NBS_MATCH) != !!(cpe.dwFlags & NB_NEGATE_ACT_NBS_MATCH)) || (!!(cpe.dwFlags & NB_NEGATE_VIS_CHAR_MATCH) != !!(cpe.dwFlags & NB_NEGATE_ACT_CHAR_MATCH)) || (!!(cpe.dwFlags & NB_NEGATE_VIS_PERMS_MATCH) != !!(cpe.dwFlags & NB_NEGATE_ACT_PERMS_MATCH)) ) { sprintf(szErr, "Lines (%d) in the Permissionstable should use NB_NEGATE_xxx instead of NB_NEGATE_VIS_xxx or NB_NEGATE_ACT_xxx ", x+1); AssertSz(FALSE, szErr); hr = E_FAIL; } } delete[] szErr; return hr; } //+--------------------------------------------------------------------------- // // Member: HrAssertMenuStructuresValid // // Purpose: Runs various asserts to make sure the menu structures are intact // Called on NetShell startup // // Arguments: // [in] hwndOwner Owner window // // Returns: // S_OK is succeeded // E_FAIL if not // // Author: deonb 8 Feb 2001 // // Notes: Asserts on failure // HRESULT HrAssertMenuStructuresValid(HWND hwndOwner) { #ifdef DBG static fBeenHereDoneThat = FALSE; if (fBeenHereDoneThat) { return S_OK; } else { fBeenHereDoneThat = TRUE; } TraceFileFunc(ttidMenus); HRESULT hr; hr = HrAssertCommandMatrixIntegrity(); return hr; #else return S_OK; #endif } //+--------------------------------------------------------------------------- // // Member: AdjustNCCS // // Purpose: Up-adjusts an NCCS_STATE flag. Will move ENABLED to DISABLED, // and DISABLED to REMOVE but not backwards. // // Arguments: // [in out] nccsCurrent NCCS to be adjusted // [in] nccsNew New state // // Returns: // none // // Author: deonb 8 Feb 2001 // // Notes: // inline void AdjustNCCS(IN OUT NCCS_STATE& nccsCurrent, IN NCCS_STATE nccsNew) { if (nccsNew > nccsCurrent) { nccsCurrent = nccsNew; } } inline BOOL fMatchFlags(IN DWORD dwFlagsMask, IN DWORD dwFlagsTest, IN DWORD dwNegateFlagMask, IN DWORD dwNegateFlagTest) { bool bMatch = FALSE; if ( (0xffffffff == dwFlagsTest) || // Means always succeed. (dwFlagsMask & dwFlagsTest) ) { bMatch = TRUE; } else { bMatch = FALSE; } if ( (dwNegateFlagMask & dwNegateFlagTest) == dwNegateFlagTest) // Do a negative compare { return !bMatch; } else { return bMatch; } } //+--------------------------------------------------------------------------- // // Member: HrGetCommandStateFromCMDTABLEEntry // // Purpose: Get the command state for a given Connection Folder Entry, // given the Command Table Entry entry that should be used. // // Arguments: // [in] cfe Connection Folder Entry // [in] cte Command Table Entry // [in] fMultiSelect Was this part of a multi-selection? // [out] nccs State that the item should be (NCCS_ENABLED/NCCS_DISABLED/NCCS_NOTSHOWN) // // Returns: // none // // Author: deonb 8 Feb 2001 // // Notes: This function uses Cached Permissions. YOU MUST CALL RefreshAllPermission before calling this function. // HRESULT HrGetCommandStateFromCMDTABLEEntry(IN const CConFoldEntry& cfe, IN const COMMANDENTRY& cte, IN BOOL fMultiSelect, OUT NCCS_STATE& nccs, OUT LPDWORD pdwResourceId) { TraceFileFunc(ttidMenus); Assert(pdwResourceId); HRESULT hr = S_OK; nccs = NCCS_ENABLED; DWORD dwNCMbm = (1 << cfe.GetNetConMediaType()); // Convert to bitmask // If we're a wizard, add as Wizard media type. if (cfe.GetWizard() == WIZARD_MNC) { dwNCMbm |= NBM_MNC_WIZARD; dwNCMbm &= ~NBM_INCOMING; // clear the INCOMINGCONNECTIONS flag (old NCM_NONE) if we're a wizard. } else if (cfe.GetWizard() == WIZARD_HNW) { dwNCMbm |= NBM_HNW_WIZARD; dwNCMbm &= ~NBM_INCOMING; // clear the INCOMINGCONNECTIONS flag (old NCM_NONE) if we're a wizard. } DWORD dwNCSbm = (1 << cfe.GetNetConStatus()); // Convert to bitmask DWORD dwNCCF = cfe.GetCharacteristics(); // Already a bitmask // Check if the command can participate in multi-select if ( fMultiSelect && !(cte.dwValidWhen & NCWHEN_MULTISELECT) ) { AdjustNCCS(nccs, NCCS_DISABLED); } // Check if the command should be visible if (!((fMatchFlags(dwNCMbm, cte.dwMediaTypeVisible, cte.dwFlags, NB_NEGATE_VIS_NBM_MATCH)) && (fMatchFlags(dwNCSbm, cte.dwStatusVisible, cte.dwFlags, NB_NEGATE_VIS_NBS_MATCH)) && (fMatchFlags(dwNCCF , cte.dwCharacteristicsVisible,cte.dwFlags, NB_NEGATE_VIS_CHAR_MATCH)) )) { AdjustNCCS(nccs, NCCS_NOTSHOWN); } // Check if the command should be grayed out if (!((fMatchFlags(dwNCMbm, cte.dwMediaTypeActive, cte.dwFlags, NB_NEGATE_ACT_NBM_MATCH)) && (fMatchFlags(dwNCSbm, cte.dwStatusActive, cte.dwFlags, NB_NEGATE_ACT_NBS_MATCH)) && (fMatchFlags(dwNCCF , cte.dwCharacteristicsActive, cte.dwFlags, NB_NEGATE_ACT_CHAR_MATCH)) )) { AdjustNCCS(nccs, NCCS_DISABLED); } // Check if the command should be grayed out based on permissions for (DWORD x = 0; nccs == NCCS_ENABLED, x < g_cteCommandPermissionsMatrixCount; x++)// Permissions won't affect NOT_SHOWN or DISABLED { const COMMANDPERMISSIONSENTRY cpe = g_cteCommandPermissionsMatrix[x]; if ( (cpe.iCommandId == cte.iCommandId) && (fMatchFlags(dwNCMbm, cpe.dwMediaType, cpe.dwFlags, NB_NEGATE_NBM_MATCH)) && (fMatchFlags(dwNCCF, cpe.dwCharacteristicsActive, cpe.dwFlags, NB_NEGATE_CHAR_MATCH)) ) { for (DWORD dwPerm = 0; dwPerm < sizeof(DWORD)*8; dwPerm++) { if (cpe.dwPermissionsActive & (1 << static_cast(dwPerm)) ) { if (!FHasPermissionFromCache(dwPerm)) { if (cpe.dwFlags & NB_REMOVE_IF_NOT_MATCH) { AdjustNCCS(nccs, NCCS_NOTSHOWN); } else { AdjustNCCS(nccs, NCCS_DISABLED); } break; // will break anyway. } } } if (APPLY_TO_USER & cpe.ncpAppliesTo) { break; } if ( (APPLY_TO_POWERUSERS & cpe.ncpAppliesTo) && FIsUserPowerUser() ) { break; } if ( (APPLY_TO_NETCONFIGOPS & cpe.ncpAppliesTo) && FIsUserNetworkConfigOps() ) { break; } if ( (APPLY_TO_ADMIN & cpe.ncpAppliesTo) && FIsUserAdmin()) { break; } // At this point all group access checks failed, so disable the connection. AdjustNCCS(nccs, NCCS_DISABLED); break; } } // Check for callback if ( (nccs != NCCS_NOTSHOWN) && (cte.pfnHrEnableDisableCB) ) { HRESULT hrTmp; NCCS_STATE nccsTemp; hrTmp = (*cte.pfnHrEnableDisableCB)(cfe, fMultiSelect, cte.iCommandId, nccsTemp); if (S_OK == hrTmp) { AdjustNCCS(nccs, nccsTemp); } else { if (FAILED(hrTmp)) { AdjustNCCS(nccs, NCCS_NOTSHOWN); } } // If the function returns S_FALSE it's an indication it didn't change the state. } // Check for Resource String callback: if ( (nccs != NCCS_NOTSHOWN) && // What's the point? (0 == *pdwResourceId) && // Must not already have a resource Id (cte.pfnHrCustomMenuStringCB) ) { HRESULT hrTmp; DWORD dwResourceIdTmp = *pdwResourceId; hrTmp = (*cte.pfnHrCustomMenuStringCB)(cfe, cte.iCommandId, &dwResourceIdTmp); if (S_OK == hr) { *pdwResourceId = dwResourceIdTmp; } } return hr; } //+--------------------------------------------------------------------------- // // Member: HrGetCommandState // // Purpose: Get the command state for a given Connection Folder Entry, // given the Command ID. // // Arguments: // [in] cfpl List (0 or more) of PIDLs that are selected // [in] dwCmdID Command ID // [out] nccs State that the item should be (NCCS_ENABLED/NCCS_DISABLED/NCCS_NOTSHOWN) // // Returns: // none // // Author: deonb 8 Feb 2001 // // Notes: // HRESULT HrGetCommandState(IN const PCONFOLDPIDLVEC& cfpl, IN const DWORD dwCmdID, OUT NCCS_STATE& nccs, OUT LPDWORD pdwResourceId, IN DWORD cteHint, IN DWORD dwOverrideFlag) { TraceFileFunc(ttidMenus); Assert(pdwResourceId); if (!pdwResourceId) { return E_POINTER; } RefreshAllPermission(); HRESULT hr = S_OK; BOOL bFound = FALSE; DWORD dwNumItems = cfpl.size(); *pdwResourceId = 0; if ( dwOverrideFlag & NB_FLAG_ON_TOPMENU ) { for (DWORD x = 0; x < g_cteCommandMatrixCount; x++) { if ( (g_cteCommandMatrix[x].iCommandId == dwCmdID) ) { bFound = TRUE; const COMMANDENTRY& cte = g_cteCommandMatrix[x]; if ((cte.dwValidWhen == NCWHEN_TOPLEVEL) || // Toplevel-ONLY menu (doesn't matter if items). ((cte.dwValidWhen & NCWHEN_TOPLEVEL) && (!dwNumItems || (cte.dwValidWhen & NCWHEN_TOPLEVEL_DISREGARD_ITEM)) ) ) { // Must be marked to allow incompatible selection, nccs = NCCS_ENABLED; // Otherwise, we'll do the item check (below). // Check for permissions for (DWORD x = 0; nccs == NCCS_ENABLED, x < g_cteCommandPermissionsMatrixCount; x++)// Permissions won't affect NOT_SHOWN or DISABLED { const COMMANDPERMISSIONSENTRY cpe = g_cteCommandPermissionsMatrix[x]; if ( (cpe.iCommandId == cte.iCommandId) && (cpe.dwFlags & NB_TOPLEVEL_PERM) ) { for (DWORD dwPerm = 0; dwPerm < sizeof(DWORD)*8; dwPerm++) { if (cpe.dwPermissionsActive & (1 << static_cast(dwPerm)) ) { if (!FHasPermissionFromCache(dwPerm)) { if (cpe.dwFlags & NB_REMOVE_IF_NOT_MATCH) { AdjustNCCS(nccs, NCCS_NOTSHOWN); } else { AdjustNCCS(nccs, NCCS_DISABLED); } break; // will break anyway. } } } if (APPLY_TO_USER & cpe.ncpAppliesTo) { break; } if ( (APPLY_TO_NETCONFIGOPS & cpe.ncpAppliesTo) && FIsUserNetworkConfigOps() ) { break; } if ( (APPLY_TO_ADMIN & cpe.ncpAppliesTo) && FIsUserAdmin()) { break; } // At this point all group access checks failed, so disable the connection. AdjustNCCS(nccs, NCCS_DISABLED); break; } } // Check for callback if (cte.pfnHrEnableDisableCB) { HRESULT hrTmp; NCCS_STATE nccsTemp; CONFOLDENTRY cfe; cfe.clear(); if (dwNumItems > 0) { hrTmp = cfpl[0].ConvertToConFoldEntry(cfe); if (FAILED(hrTmp)) { cfe.clear(); } } hrTmp = (*cte.pfnHrEnableDisableCB)(cfe, dwNumItems > 1, cte.iCommandId, nccsTemp); if (S_OK == hrTmp) { AdjustNCCS(nccs, nccsTemp); } else { if (FAILED(hrTmp)) { AdjustNCCS(nccs, NCCS_NOTSHOWN); } } // If the function returns S_FALSE it's an indication it didn't change the state. } if (!(NB_REMOVE_TOPLEVEL_ITEM & cte.dwFlags)) { if (nccs == NCCS_NOTSHOWN) { nccs = NCCS_DISABLED; } } return S_OK; } break; // we won't find another CMDID } } if (!dwNumItems) { nccs = NCCS_DISABLED; if (bFound) { return S_OK; } else { return E_FILE_NOT_FOUND; } } } AssertSz(dwNumItems, "You don't have any items selected, but you're not a top-level menu... how come?"); bFound = FALSE; nccs = NCCS_ENABLED; // This will effectively loop through all the selected PIDLs and apply the strictest // nccs that applies to everything. for (PCONFOLDPIDLVEC::const_iterator cfp = cfpl.begin(); cfp != cfpl.end(); cfp++) { CONFOLDENTRY cfe; hr = cfp->ConvertToConFoldEntry(cfe); if (FAILED(hr)) { return E_FAIL; } DWORD dwPos = 0xffffffff; // This is a O(n^(2+)) algorithm when called from HrBuildMenu. // We pass a hint to check if we can quickly find the cte. if ( (cteHint != 0xffffffff) && (g_cteCommandMatrix[cteHint].iCommandId == dwCmdID) ) { dwPos = cteHint; } if (dwPos == 0xffffffff) { for (DWORD x = 0; x < g_cteCommandMatrixCount && SUCCEEDED(hr); x++) { if (g_cteCommandMatrix[x].iCommandId == dwCmdID) { dwPos = x; break; } } } if (dwPos == 0xffffffff) { return E_FILE_NOT_FOUND; } else { bFound = TRUE; NCCS_STATE nccsTmp; hr = HrGetCommandStateFromCMDTABLEEntry(cfe, g_cteCommandMatrix[dwPos], dwNumItems != 1, nccsTmp, pdwResourceId); if (FAILED(hr)) { return hr; } AdjustNCCS(nccs, nccsTmp); if ( (dwOverrideFlag & NB_FLAG_ON_TOPMENU) && (!(NB_REMOVE_TOPLEVEL_ITEM & g_cteCommandMatrix[dwPos].dwFlags)) && (nccs == NCCS_NOTSHOWN) ) { nccs = NCCS_DISABLED; } } } if (!bFound) { return E_FILE_NOT_FOUND; } return S_OK; } //+--------------------------------------------------------------------------- // // Member: HrGetCheckState // // Purpose: Get the check state for a given Connection Folder Entry, // given the Command ID. // // Arguments: // [in] cfpl List (0 or more) of PIDLs that are selected // [in] dwCmdID Command ID // [out] nccs State that the item should be (NCCS_CHECKED/NCCS_UNCHECKED) // // Returns: // HRESULT // // Author: deonb 7 Mar 2001 // // Notes: // HRESULT HrGetCheckState(IN const PCONFOLDPIDLVEC& cfpl, IN const DWORD dwCmdID, OUT NCCS_CHECKED_STATE& nccs) { HRESULT hr = S_FALSE; DWORD dwLoop = 0; nccs = NCCS_UNCHECKED; for (; dwLoop < g_nFolderCheckCount; dwLoop++) { if (dwCmdID == g_cceFolderChecks[dwLoop].iCommandId) { switch(g_cceFolderChecks[dwLoop].iCommandId) { case CMIDM_CONMENU_OPERATOR_ASSIST: hr = S_OK; if (g_fOperatorAssistEnabled) { nccs = NCCS_CHECKED; } break; default: break; } } } return S_OK; } //+--------------------------------------------------------------------------- // // Member: NCCMDFromSFV // // Purpose: Return a NetShell CMDID for Shell Internal IDs // // Arguments: // [in] iCmdID CMDID to map // // Returns: // Returns iCmdID if not shell message, or a iCmdID mapping // // Author: deonb 8 Feb 2001 // // Notes: // int NCCMDFromSFV(IN int iCmdID, IN DWORD idCmdFirst) { for (int x = 0; x < g_cteSFVCommandMapCount; x++) { if (g_cteSFVCommandMap[x].iSFVCommandId == iCmdID) { return g_cteSFVCommandMap[x].iCommandId + idCmdFirst; } } return iCmdID; } //+--------------------------------------------------------------------------- // // Function: HrUpdateMenuItemChecks // // Purpose: Walk through the list of checkmark-able commands and check if // applicable. // // Arguments: // None // // Returns: // // Author: deonb 7 Mar 2001 // // Notes: // HRESULT HrUpdateMenuItemChecks(IN PCONFOLDPIDLVEC& cfpl, IN OUT HMENU hMenu, IN DWORD idCmdFirst) { HRESULT hr = S_FALSE; DWORD dwLoop = 0; Assert(hMenu) if (!hMenu) { return E_INVALIDARG; } int cMenuItems = GetMenuItemCount(hMenu); for (int x = 0; x < cMenuItems; x++) { UINT nMenuID = GetMenuItemID(hMenu, x); NCCS_CHECKED_STATE nccs; DWORD dwCustomResourceId = 0; DWORD dwCmdId = NCCMDFromSFV(nMenuID, idCmdFirst) - idCmdFirst; hr = HrGetCheckState(cfpl, dwCmdId, nccs); if (S_OK == hr) // don't need to set if not supported on this item (S_FALSE) { CheckMenuItem( hMenu, x, nccs == NCCS_CHECKED ? MF_CHECKED | MF_BYPOSITION : // checked MF_UNCHECKED | MF_BYPOSITION); // unchecked } } return S_OK; } //+--------------------------------------------------------------------------- // // Member: HrBuildMenu // // Purpose: Build the context menu for for a given Connection Folder Entry.. // // Arguments: // [in out] hMenu Handle to menu which is to be updated // [in] fVerbsOnly Should return Verbs only (shortcuts) // [in] cfpl List (0 or more) of PIDLs that are selected // [in] idCmdFirst Min value the handler can specify for a menu item // // Returns: // none // // Author: deonb 8 Feb 2001 // // Notes: // HRESULT HrBuildMenu(IN OUT HMENU &hMenu, IN BOOL fVerbsOnly, IN PCONFOLDPIDLVEC& cfpl, IN DWORD idCmdFirst) { TraceFileFunc(ttidMenus); HRESULT hr = S_OK; Assert(hMenu) if (!hMenu) { return E_INVALIDARG; } DWORD dwCurrentDefaultPriority = 0; BOOL fShouldAppendSeparator = FALSE; DWORD dwInsertPos = 1; for (DWORD x = 0; x < g_cteCommandMatrixCount && SUCCEEDED(hr); x++) { const COMMANDENTRY& cte = g_cteCommandMatrix[x]; if ( (fVerbsOnly) && !(cte.dwFlags & NB_VERB) ) { continue; } if (CMIDM_SEPARATOR == cte.iCommandId) { fShouldAppendSeparator = TRUE; } else { NCCS_STATE nccs; DWORD dwCustomResourceId = 0; hr = HrGetCommandState(cfpl, cte.iCommandId, nccs, &dwCustomResourceId, x); if (SUCCEEDED(hr)) { if ( nccs != NCCS_NOTSHOWN ) { if (fShouldAppendSeparator) { fShouldAppendSeparator = FALSE; if (!InsertMenu(hMenu, dwInsertPos++, MF_BYPOSITION | MF_SEPARATOR, CMIDM_SEPARATOR, NULL)) { hr = HRESULT_FROM_WIN32(GetLastError()); } } LPCWSTR szMenuString; if (!dwCustomResourceId) { szMenuString = SzLoadIds(IDS_MENU_CMIDM_START + cte.iCommandId - CMIDM_FIRST); } else { szMenuString = SzLoadIds(dwCustomResourceId); } if (!InsertMenu(hMenu, dwInsertPos++, MF_BYPOSITION | MF_STRING | (nccs == NCCS_DISABLED ? MF_GRAYED : MF_ENABLED), idCmdFirst + cte.iCommandId - CMIDM_FIRST, szMenuString)) { AssertSz(FALSE, "Couldn't append menu item"); hr = HRESULT_FROM_WIN32(GetLastError()); } else { if ( (nccs == NCCS_ENABLED) && (cte.dwDefaultPriority > dwCurrentDefaultPriority) ) // Not 0 is implied. { dwCurrentDefaultPriority = cte.dwDefaultPriority; if (!SetMenuDefaultItem(hMenu, idCmdFirst + cte.iCommandId - CMIDM_FIRST, FALSE)) { AssertSz(FALSE, "Couldn't set default menu item"); hr = HRESULT_FROM_WIN32(GetLastError()); } } } } } else { if (E_FILE_NOT_FOUND == hr) { AssertSz(FALSE, "Didn't find the CMDID inside CMDTABLE"); } } } } if (SUCCEEDED(hr)) { hr = HrUpdateMenuItemChecks(cfpl, hMenu, idCmdFirst); if (S_FALSE == hr) { hr = S_OK; } } return hr; } //+--------------------------------------------------------------------------- // // Member: HrUpdateMenu // // Purpose: Update a menu for for a given Connection Folder Entry.. // // Arguments: // [in out] hMenu Handle to menu which is to be updated // [in] cfpl List (0 or more) of PIDLs that are selected // [in] idCmdFirst Min value the handler can specify for a menu item // // Returns: // none // // Author: deonb 8 Feb 2001 // // Notes: // HRESULT HrUpdateMenu(IN OUT HMENU &hMenu, IN PCONFOLDPIDLVEC& cfpl, IN DWORD idCmdFirst) { TraceFileFunc(ttidMenus); HRESULT hr = S_OK; Assert(hMenu) if (!hMenu) { return E_INVALIDARG; } int cMenuItems = GetMenuItemCount(hMenu); for (int x = 0; x < cMenuItems; x++) { UINT nMenuID = GetMenuItemID(hMenu, x); // UINT uiState = GetMenuState(hMenu, nMenuID, MF_BYCOMMAND ); NCCS_STATE nccs; DWORD dwCustomResourceId = 0; DWORD dwCmdId = NCCMDFromSFV(nMenuID, idCmdFirst) - idCmdFirst; hr = HrGetCommandState(cfpl, dwCmdId, nccs, &dwCustomResourceId, 0xffffffff, NB_FLAG_ON_TOPMENU); if (SUCCEEDED(hr)) { if (nccs == NCCS_NOTSHOWN) { #ifdef DBG WCHAR szTemp[MAX_PATH]; GetMenuStringW(hMenu, x, szTemp, MAX_PATH, MF_BYPOSITION ); TraceTag(ttidMenus, "Received request to permanently remove menu item: '%S' for CMDID: %d MenuID: %d", szTemp, dwCmdId, nMenuID); #endif RemoveMenu(hMenu, x, MF_BYPOSITION); } else { EnableMenuItem( hMenu, x, nccs == NCCS_ENABLED ? MF_ENABLED | MF_BYPOSITION: // enable MF_GRAYED | MF_BYPOSITION); } } NCCS_CHECKED_STATE nccCheckedState; hr = HrGetCheckState(cfpl, dwCmdId, nccCheckedState); if (S_OK == hr) // don't need to set if not supported on this item (S_FALSE) { CheckMenuItem( hMenu, x, nccCheckedState == NCCS_CHECKED ? MF_CHECKED | MF_BYPOSITION : // checked MF_UNCHECKED | MF_BYPOSITION); // unchecked } } return hr; } //+--------------------------------------------------------------------------- // // Member: HasPermissionToRenameConnection // // Purpose: Checks if the Thread Local user has access to rename a given // connection // // Arguments: // [in] pcfp PIDL that wants to be renamed // // Returns: // TRUE if has permissions // FALSE if no permissions // // Author: deonb 8 Feb 2001 // // Notes: // ISSUE: Move out of this file // BOOL HasPermissionToRenameConnection(IN const PCONFOLDPIDL& pcfp) { TraceFileFunc(ttidMenus); BOOL fPermission = FALSE; // ISSUE: Due to a clarification in the spec this code is unreasonably complex. // If possible clean it up in the future. if (((!(pcfp->dwCharacteristics & NCCF_ALL_USERS) && FHasPermissionFromCache(NCPERM_RenameMyRasConnection)))) { fPermission = TRUE; } else if (FIsPolicyConfigured(NCPERM_RenameConnection)) { if (((!(pcfp->dwCharacteristics & NCCF_ALL_USERS) && FHasPermissionFromCache(NCPERM_RenameMyRasConnection)))) { fPermission = TRUE; } else if ((pcfp->ncm != NCM_LAN) && (pcfp->dwCharacteristics & NCCF_ALL_USERS) && FHasPermissionFromCache(NCPERM_RenameConnection) || (pcfp->ncm == NCM_LAN) && FHasPermissionFromCache(NCPERM_RenameConnection)) { fPermission = TRUE; } } else if (((pcfp->ncm == NCM_LAN) && FHasPermissionFromCache(NCPERM_RenameLanConnection)) || ((pcfp->ncm != NCM_LAN) && (pcfp->dwCharacteristics & NCCF_ALL_USERS) && FHasPermissionFromCache(NCPERM_RenameAllUserRasConnection))) { fPermission = TRUE; } return fPermission; } //+--------------------------------------------------------------------------- // // Function: SetConnectMenuItem // // Purpose: This function goes in and modifies the first menu item so // that it shows the correct test based on the connection // state (enabled or not) or the connection type (LAN / WAN). // // Arguments: // hmenu [in] Menu to operate on // bLan [in] Whether or not this is a LAN connection // idCmdFirst [in] Are commands are really an offset to this value // bEnable [in] If the connection is enabled or not // // Returns: // // Author: mbend 8 Mar 2000 // // Notes: // VOID SetConnectMenuItem(IN HMENU hmenu, IN BOOL bLan, IN INT idCmdFirst, IN BOOL bEnable) { // Different strings for WAN/LAN INT iEnableString = bLan ? IDS_DISABLE_MENUITEM : IDS_DISCONNECT_MENUITEM; INT iDisableString = bLan ? IDS_ENABLE_MENUITEM : IDS_CONNECT_MENUITEM; INT iMenuString = bEnable ? iEnableString : iDisableString; PCWSTR pszMenuString = SzLoadIds(iMenuString); MENUITEMINFO mii; // Different commands for WAN/LAN INT iConnect = bLan ? CMIDM_ENABLE : CMIDM_CONNECT; INT iDisconnect = bLan ? CMIDM_DISABLE : CMIDM_DISCONNECT; INT iOffset = bEnable ? iDisconnect : iConnect; INT iNewCommand = idCmdFirst + iOffset; Assert(pszMenuString); // Set the menuitem fields // mii.cbSize = sizeof(MENUITEMINFO); mii.fMask = MIIM_TYPE | MIIM_ID; mii.fType = MFT_STRING; mii.dwTypeData = (PWSTR) pszMenuString; mii.wID = iNewCommand; // This is assuming that we want to take out the first menu item. if (!SetMenuItemInfo(hmenu, 0, TRUE, &mii)) { TraceTag(ttidMenus, "SetMenuItemInfo returned: 0x%08x for CMIDM_DISCONNECT", GetLastError()); } } //+--------------------------------------------------------------------------- // // Function: HrSetConnectDisconnectMenuItem // // Purpose: Modify the menu item (if necessary) for connect/disconnect. // We change this back and forth as needed since only one can be // supported at a time, and those in charge don't want both // appearing at any given time. // // Arguments: // apidlSelected [in] List of selected objects // cPidl [in] Count of selected objects // hmenu [in] Our menu handle // // Returns: // // Author: jeffspr 1 May 1998 // // Notes: // HRESULT HrSetConnectDisconnectMenuItem(IN const PCONFOLDPIDLVEC& apidlSelected, IN HMENU hmenu, IN INT idCmdFirst) { HRESULT hr = S_OK; if (apidlSelected.size() == 1) { Assert(!apidlSelected[0].empty() ); const PCONFOLDPIDL& pcfp = apidlSelected[0]; switch(pcfp->ncs) { case NCS_CONNECTED: case NCS_DISCONNECTING: case NCS_MEDIA_DISCONNECTED: case NCS_INVALID_ADDRESS: case NCS_AUTHENTICATING: case NCS_AUTHENTICATION_FAILED: case NCS_AUTHENTICATION_SUCCEEDED: case NCS_CREDENTIALS_REQUIRED: SetConnectMenuItem(hmenu, IsMediaLocalType(pcfp->ncm), idCmdFirst, TRUE); break; case NCS_DISCONNECTED: case NCS_CONNECTING: case NCS_HARDWARE_NOT_PRESENT: case NCS_HARDWARE_DISABLED: case NCS_HARDWARE_MALFUNCTION: SetConnectMenuItem(hmenu, IsMediaLocalType(pcfp->ncm), idCmdFirst, FALSE); break; default: AssertSz(FALSE, "HrSetConnectDisconnectMenuItem: What in the heck state is this?"); break; } } TraceHr(ttidMenus, FAL, hr, FALSE, "HrSetConnectDisconnectMenuItem"); return hr; } HRESULT HrCanRenameConnection( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if (cfe.empty()) { return S_FALSE; } if (cfe.GetCharacteristics() & NCCF_INCOMING_ONLY) { if (cfe.GetNetConMediaType() == NCM_NONE) // Incoming server - don't care { return S_FALSE; } else { nccs = NCCS_NOTSHOWN; return S_OK; } } else { CPConFoldPidl pcfp; cfe.ConvertToPidl(pcfp); if (!HasPermissionToRenameConnection(pcfp)) { nccs = NCCS_DISABLED; } else { return S_FALSE; } return S_OK; } } HRESULT HrCanShowProperties( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if (cfe.empty()) { return S_FALSE; } if (cfe.GetCharacteristics() & NCCF_INCOMING_ONLY) { if (cfe.GetNetConMediaType() == NCM_NONE) // Incoming server - don't care { return S_FALSE; } else { nccs = NCCS_NOTSHOWN; return S_OK; } } return S_FALSE; } BOOL IsBridgeInstalled() throw() { BOOL fBridgePresent = FALSE; // fail to false HRESULT hResult; IHNetCfgMgr* pHomeNetConfigManager; hResult = HrCreateInstance(CLSID_HNetCfgMgr, CLSCTX_INPROC, &pHomeNetConfigManager); if(SUCCEEDED(hResult)) { IHNetBridgeSettings* pNetBridgeSettings; hResult = pHomeNetConfigManager->QueryInterface(IID_IHNetBridgeSettings, reinterpret_cast(&pNetBridgeSettings)); if(SUCCEEDED(hResult)) { IHNetBridge* pNetBridge; IEnumHNetBridges* pNetBridgeEnum; hResult = pNetBridgeSettings->EnumBridges(&pNetBridgeEnum); if(SUCCEEDED(hResult)) { hResult = pNetBridgeEnum->Next(1, &pNetBridge, NULL); if(S_OK == hResult) { fBridgePresent = TRUE; ReleaseObj(pNetBridge); } ReleaseObj(pNetBridgeEnum); } ReleaseObj(pNetBridgeSettings); } ReleaseObj(pHomeNetConfigManager); } return fBridgePresent; } inline BOOL IsNetConnBridgeable (IN const CConFoldEntry& cfe) { return TRUE; // Should check for cfe.GetNetConSubMediaType() != NCSM_BLUETOOTH; } HRESULT HrIsBridgeSupported( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { // if (cfe.empty()) // { // return S_FALSE; // } // #ifdef _WIN64 // Homenet technologies are not available at all on IA64 nccs = NCCS_NOTSHOWN; return S_OK; #else // If the machine is Data Center, Back Office, SBS, or Blade delete the bridge menu item OSVERSIONINFOEXW verInfo = {0}; ULONGLONG ConditionMask = 0; verInfo.dwOSVersionInfoSize = sizeof(verInfo); verInfo.wSuiteMask = VER_SUITE_DATACENTER | VER_SUITE_BACKOFFICE | VER_SUITE_SMALLBUSINESS_RESTRICTED | VER_SUITE_SMALLBUSINESS | VER_SUITE_BLADE; VER_SET_CONDITION(ConditionMask, VER_SUITENAME, VER_OR); if (VerifyVersionInfo(&verInfo, VER_SUITENAME, ConditionMask)) { nccs = NCCS_NOTSHOWN; return S_OK; } else { BOOL fUserIsAdmin = FALSE; HRESULT hr = S_OK; CComPtr pConnectionUi; hr = CoCreateInstance(CLSID_NetConnectionUiUtilities, NULL, CLSCTX_INPROC, IID_INetConnectionUiUtilities, reinterpret_cast(&pConnectionUi)); if (FAILED(hr)) { return hr; } fUserIsAdmin = FIsUserAdmin(); if (IsBridgeInstalled()) { if (CMIDM_CREATE_BRIDGE == iCommandId) { nccs = NCCS_NOTSHOWN; return S_OK; } else if (CMIDM_CONMENU_CREATE_BRIDGE == iCommandId) { nccs = NCCS_DISABLED; return S_OK; } else // CMIDM_ADD_TO_BRIDGE or CMID_REMOVE_FROM_BRIDGE { if (!fUserIsAdmin || !pConnectionUi->UserHasPermission(NCPERM_AllowNetBridge_NLA) || !IsNetConnBridgeable(cfe)) { nccs = NCCS_DISABLED; return S_OK; } else { return S_FALSE; // Leave alone } } } else { if ( (CMIDM_CREATE_BRIDGE == iCommandId) || (CMIDM_CONMENU_CREATE_BRIDGE == iCommandId) ) { if (!fUserIsAdmin || !pConnectionUi->UserHasPermission(NCPERM_AllowNetBridge_NLA) || !IsNetConnBridgeable(cfe)) { nccs = NCCS_DISABLED; return S_OK; } else { return S_FALSE; // Leave alone } } else // CMIDM_ADD_TO_BRIDGE or CMID_REMOVE_FROM_BRIDGE { nccs = NCCS_NOTSHOWN; return S_OK; } } } #endif } HRESULT HrOsIsLikePersonal() { if (IsOS(OS_PERSONAL)) { return S_OK; } if (IsOS(OS_PROFESSIONAL)) { LPWSTR pszDomain; NETSETUP_JOIN_STATUS njs = NetSetupUnknownStatus; if (NERR_Success == NetGetJoinInformation(NULL, &pszDomain, &njs)) { NetApiBufferFree(pszDomain); } if (NetSetupDomainName == njs) { return S_FALSE; // connected to domain } else { return S_OK; // Professional, but not a domain member } } return S_FALSE; // not personal or non-domain professional } HRESULT HrShouldHaveHomeNetWizard() { #ifdef _WIN64 return S_FALSE; #else if ( ( HrOsIsLikePersonal() == S_OK ) && FIsUserAdmin()) { return S_OK; } else { return S_FALSE; } #endif } HRESULT HrIsHomeNewWizardSupported( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if (S_OK == HrShouldHaveHomeNetWizard() ) { nccs = NCCS_ENABLED; return S_OK; } else { nccs = NCCS_NOTSHOWN; return S_OK; } } HRESULT HrIsTroubleShootSupported( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if ( cfe.empty() && (!fMultiSelect) ) { if ( ! IsOS(OS_ANYSERVER) ) { nccs = NCCS_ENABLED; return S_OK; } else { nccs = NCCS_ENABLED; return S_OK; } } else { nccs = NCCS_DISABLED; return S_OK; } } HRESULT HrIsNCWSupported( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if ( (HrOsIsLikePersonal() == S_OK) && !FIsUserAdmin() ) { nccs = NCCS_NOTSHOWN; return S_OK; } else { return S_FALSE; } } HRESULT HrIsMediaWireless( IN const CConFoldEntry& cfe, IN BOOL fMultiSelect, IN int iCommandId, OUT NCCS_STATE& nccs ) { if (cfe.GetNetConSubMediaType() != NCSM_WIRELESS) { nccs = NCCS_NOTSHOWN; return S_OK; } return S_FALSE; // Continue with processing }