|
|
//+---------------------------------------------------------------------------
//
// 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(DWORD dwInput, DWORD dwFlags, DWORD dwNegateCondition){ if (dwFlags & dwNegateCondition) { return ~dwInput; } else { return dwInput; }}
inline BOOL bContains(DWORD dwContainee, DWORD dwContainer, DWORD dwFlags, DWORD dwContaineeNegateCondition, 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(OUT IN 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<DWORD64>(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<DWORD64>(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(int iCmdID, 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(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( HMENU hmenu, BOOL bLan, INT idCmdFirst, 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( const PCONFOLDPIDLVEC& apidlSelected, HMENU hmenu, 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<ConFoldPidl_v2> 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(){ 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<void**>(&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;}
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 Advanced server or data center, delete the bridge menu item
OSVERSIONINFOEXW verInfo = {0}; ULONGLONG ConditionMask = 0;
verInfo.dwOSVersionInfoSize = sizeof(verInfo); verInfo.wSuiteMask = VER_SUITE_ENTERPRISE;
VER_SET_CONDITION(ConditionMask, VER_SUITENAME, VER_AND);
if (VerifyVersionInfo(&verInfo, VER_SUITENAME, ConditionMask)) { nccs = NCCS_NOTSHOWN; return S_OK; } else { BOOL fUserIsAdmin = FALSE; HRESULT hr = S_OK; CComPtr<INetConnectionUiUtilities> pConnectionUi; hr = CoCreateInstance(CLSID_NetConnectionUiUtilities, NULL, CLSCTX_INPROC, IID_INetConnectionUiUtilities, reinterpret_cast<void**>(&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)) { 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)) { 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
}
|
