#pragma once
#include <ncxbase.h>
#include <ncatlps.h>
#include <nceh.h>
#include <ncsetup.h>
#include <notifval.h>
#include <winsock2.h>
#include "resource.h"
extern const WCHAR c_szEmpty[];
// AppleTalk Globally visible strings
extern const WCHAR c_chAt;
// Define the possible media types (Used also as array indices)
#define MEDIATYPE_ETHERNET 1
#define MEDIATYPE_TOKENRING 2
#define MEDIATYPE_FDDI 3
#define MEDIATYPE_WAN 4
#define MEDIATYPE_LOCALTALK 5
#define MAX_ZONES 255
#define ZONELISTSIZE 2048
#define MAX_ZONE_NAME_LEN 32
#define MAX_RANGE_ALLOWED 65279
#define MIN_RANGE_ALLOWED 1
#define ZONEBUFFER_LEN 32*255
// Seed Info Validation returns
#define NO_SEED_INFO 0x0
#define VALID_SEED_INFO 0x1
#define INVALID_SEED_INFO 0x2
typedef enum
{
AT_PNP_SWITCH_ROUTING = 0,
AT_PNP_SWITCH_DEFAULT_ADAPTER,
AT_PNP_RECONFIGURE_PARMS
} ATALK_PNP_MSGTYPE;
typedef struct _ATALK_PNP_EVENT
{
ATALK_PNP_MSGTYPE PnpMessage;
} ATALK_PNP_EVENT, *PATALK_PNP_EVENT;
// Class Forwards
class CATlkObj;
class CATLKEnv;
class CAdapterInfo;
// Define a structure for reading/writing all information necessary about an adapter
typedef struct
{
DWORD m_dwAarpRetries;
DWORD m_dwDdpCheckSums;
DWORD m_dwNetworkRangeLowerEnd;
DWORD m_dwNetworkRangeUpperEnd;
DWORD m_dwRouterPramNode;
DWORD m_dwSeedingNetwork;
DWORD m_dwUserPramNode1;
DWORD m_dwUserPramNode2;
DWORD m_dwMediaType;
WCHAR* m_szDefaultZone;
WCHAR* m_szPortName;
} ATLK_ADAPTER;
typedef list<CAdapterInfo *> ATLK_ADAPTER_INFO_LIST;
// Class: CAdapters
//
// Purpose: Contain all information necessary about a single adapter instance
//
class CAdapterInfo
{
friend class CATLKEnv;
public:
CAdapterInfo();
~CAdapterInfo();
// Make a duplicate copy of 'this'
HRESULT HrCopy(CAdapterInfo ** ppAI);
void SetDeletePending(BOOL f) {m_fDeletePending = f;}
BOOL FDeletePending() {return m_fDeletePending;}
void SetDisabled(BOOL f) {m_fDisabled = f;}
BOOL FDisabled() {return m_fDisabled;}
void SetCharacteristics(DWORD dw) {m_dwCharacteristics = dw;}
DWORD GetCharacteristics() {return m_dwCharacteristics;}
BOOL FHidden() {return !!(NCF_HIDDEN & m_dwCharacteristics);}
void SetMediaType(DWORD dw) {m_AdapterInfo.m_dwMediaType = dw;}
DWORD DwMediaType() {return m_AdapterInfo.m_dwMediaType;}
void SetDisplayName(PCWSTR psz) {m_strDisplayName = psz;}
PCWSTR SzDisplayName() {return m_strDisplayName.c_str();}
void SetBindName(PCWSTR psz) {m_strBindName = psz;}
PCWSTR SzBindName() {return m_strBindName.c_str();}
ATLK_ADAPTER* PAdapterInfo() {return &m_AdapterInfo;}
void SetPortName(PCWSTR psz);
PCWSTR SzPortName() {return m_AdapterInfo.m_szPortName ?
m_AdapterInfo.m_szPortName :
c_szEmpty;}
void SetDefaultZone(PCWSTR psz);
PCWSTR SzDefaultZone() {return m_AdapterInfo.m_szDefaultZone ?
m_AdapterInfo.m_szDefaultZone :
c_szEmpty;}
list<tstring*> &LstpstrZoneList() {return m_lstpstrZoneList;}
list<tstring*> &LstpstrDesiredZoneList() {return m_lstpstrDesiredZoneList;}
void SetSeedingNetwork(BOOL fSeeding)
{m_AdapterInfo.m_dwSeedingNetwork = (fSeeding ? 1 : 0);}
DWORD FSeedingNetwork() {return (1 == m_AdapterInfo.m_dwSeedingNetwork);}
void SetRouterOnNetwork(BOOL f) {m_fRouterOnNetwork = f;}
BOOL FRouterOnNetwork() {return m_fRouterOnNetwork;}
void SetExistingNetRange(DWORD dwLower, DWORD dwUpper)
{m_dwNetworkLower = dwLower;
m_dwNetworkUpper = dwUpper;}
DWORD DwQueryNetworkUpper() {return m_dwNetworkUpper;}
DWORD DwQueryNetworkLower() {return m_dwNetworkLower;}
void SetAdapterNetRange(DWORD dwLower, DWORD dwUpper)
{m_AdapterInfo.m_dwNetworkRangeLowerEnd = dwLower;
m_AdapterInfo.m_dwNetworkRangeUpperEnd = dwUpper;}
DWORD DwQueryNetRangeUpper() {return m_AdapterInfo.m_dwNetworkRangeUpperEnd;}
DWORD DwQueryNetRangeLower() {return m_AdapterInfo.m_dwNetworkRangeLowerEnd;}
void SetNetDefaultZone(PCWSTR psz) {m_strNetDefaultZone = psz;}
PCWSTR SzNetDefaultZone() {return m_strNetDefaultZone.c_str();}
HRESULT HrConvertZoneListAndAddToPortInfo(CHAR * szZoneList, ULONG NumZones);
HRESULT HrGetAndSetNetworkInformation(SOCKET, PCWSTR);
VOID SetInstanceGuid(GUID guid) {memcpy(&m_guidInstance, &guid, sizeof(GUID));}
GUID * PInstanceGuid() {return &m_guidInstance;}
BOOL IsDirty() {return m_fDirty;}
VOID SetDirty(BOOL f) {m_fDirty = f;}
BOOL IsRasAdapter() {return m_fRasAdapter;}
VOID SetRasAdapter(BOOL f) {m_fRasAdapter = f;}
VOID ZeroSpecialParams() {m_AdapterInfo.m_dwUserPramNode1 = 0;
m_AdapterInfo.m_dwUserPramNode2 = 0;
m_AdapterInfo.m_dwRouterPramNode = 0; }
// m_guidInstance is the instance guid of the adapter
GUID m_guidInstance;
// m_lstpstrZoneList is the REG_MULTI_SZ value found under
// AppleTalk\Parameters\Adapters\<adapter>\ZoneList
list<tstring*> m_lstpstrZoneList;
// Desired zone will be choosen from this list
list<tstring*> m_lstpstrDesiredZoneList;
// m_AdapterInfo is the collection of values found under
// AppleTalk\Parameters\Adapters\<adapter> with the exception of the
// ZoneList : REG_MULTI_SZ value which is stored in m_lstpstrZoneList above.
ATLK_ADAPTER m_AdapterInfo;
private:
// m_fDisabled is a boolean that, when TRUE, indicates this adapter
// is currently disabled
BOOL m_fDisabled;
// If true, this adapter's info has changed
BOOL m_fDirty;
// m_fDeletePending is a boolean that, when TRUE, indicates this adapter
// is being removed from the adapter list (eventually)
BOOL m_fDeletePending;
// If true, the zone list contents are considered valid
BOOL m_fZoneListValid;
// If true, this adapter exposes ndiswanatlk
BOOL m_fRasAdapter;
// m_dwCharacteristics contains the adapter's characteristic settings
DWORD m_dwCharacteristics;
// m_strBindName is the BindName for the adapter
tstring m_strBindName;
// m_strDisplayName is the display name of the adapter
tstring m_strDisplayName;
// Default zone returned by stack
tstring m_strNetDefaultZone;
// Router network state
BOOL m_fRouterOnNetwork;
DWORD m_dwNetworkUpper; // existing network # returned by stack
DWORD m_dwNetworkLower; // existing network # returned by stack
};
// Define a structure for reading/writing AppleTalk\Parameters values
typedef struct
{
DWORD dwEnableRouter;
WCHAR* szDefaultPort;
WCHAR* szDesiredZone;
} ATLK_PARAMS;
// Class: CATLKEnv
//
// Purpose: Contains the "Known" enviroment state regarding AppleTalk Params/Settings
//
class CATLKEnv
{
private:
CATLKEnv(CATlkObj *pmsc);
public:
// Automatic two-phase constructor
static HRESULT HrCreate(CATLKEnv **, CATlkObj *);
~CATLKEnv();
HRESULT HrCopy(CATLKEnv **ppEnv);
HRESULT HrReadAppleTalkInfo();
HRESULT HrGetAppleTalkInfoFromNetwork(CAdapterInfo * pAI);
HRESULT HrGetAdapterInfo();
HRESULT HrGetOneAdaptersInfo(INetCfgComponent*, CAdapterInfo **);
HRESULT HrAddAdapter(INetCfgComponent * pnccFound);
void SetDefaultPort(PCWSTR psz);
PCWSTR SzDefaultPort() {return (NULL != m_Params.szDefaultPort ?
m_Params.szDefaultPort :
c_szEmpty);}
void SetDesiredZone(PCWSTR psz);
PCWSTR SzDesiredZone() {return (NULL != m_Params.szDesiredZone ?
m_Params.szDesiredZone :
c_szEmpty);}
DWORD DwDefaultAdaptersMediaType() {return m_dwDefaultAdaptersMediaType;}
void SetDefaultMediaType(DWORD dw) {m_dwDefaultAdaptersMediaType = dw;}
void EnableRouting(BOOL f) {m_Params.dwEnableRouter = (f ? 1 : 0);}
BOOL FRoutingEnabled() {return (1 == m_Params.dwEnableRouter);}
ATLK_ADAPTER_INFO_LIST &AdapterInfoList() {return m_lstpAdapters;}
BOOL FIsAppleTalkRunning() {return m_fATrunning;}
void SetATLKRunning(BOOL fRunning) {m_fATrunning = fRunning;}
HRESULT HrUpdateRegistry();
HRESULT HrWriteOneAdapter(CAdapterInfo *pAI);
DWORD DwMediaPriority(DWORD dwMediaType);
void InitDefaultPort();
BOOL FRouterChanged() {return m_fRouterChanged;}
VOID SetRouterChanged(BOOL f) {m_fRouterChanged = f;}
BOOL FDefAdapterChanged() {return m_fDefAdapterChanged;}
VOID SetDefAdapterChanged(BOOL f) {m_fDefAdapterChanged = f;}
CAdapterInfo * PAIFindDefaultPort();
ATLK_ADAPTER_INFO_LIST m_lstpAdapters;
private:
CATlkObj * m_pmsc;
BOOL m_fRouterChanged;
BOOL m_fDefAdapterChanged;
BOOL m_fATrunning;
DWORD m_dwDefaultAdaptersMediaType;
ATLK_PARAMS m_Params;
};
// Class: CATLKGeneralDlg
//
// Purpose: Manage the "General" property page
//
class CATLKGeneralDlg: public CPropSheetPage
{
public:
BEGIN_MSG_MAP(CATLKGeneralDlg)
MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
MESSAGE_HANDLER(WM_CONTEXTMENU, OnContextMenu)
MESSAGE_HANDLER(WM_HELP, OnHelp)
NOTIFY_CODE_HANDLER(PSN_APPLY, OnOk)
COMMAND_ID_HANDLER(CHK_GENERAL_DEFAULT, HandleChkBox)
END_MSG_MAP()
CATLKGeneralDlg(CATlkObj *pmsc, CATLKEnv * pATLKEnv);
~CATLKGeneralDlg();
LRESULT HandleChkBox(WORD wNotifyCode, WORD wID,
HWND hWndCtl, BOOL& bHandled);
LRESULT OnInitDialog(UINT uMsg, WPARAM wParam,
LPARAM lParam, BOOL& bHandled);
LRESULT OnContextMenu(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& Handled);
LRESULT OnHelp(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
LRESULT OnOk(INT idCtrl, LPNMHDR pnmh, BOOL& bHandled);
BOOL FAddZoneListToControl(list<tstring*> * plstpstr);
VOID RefreshZoneCombo();
private:
CATlkObj * m_pmsc;
CATLKEnv * m_pATLKEnv;
};
/////////////////////////////////////////////////////////////////////////////
// CATlkObj
class ATL_NO_VTABLE CATlkObj :
public CComObjectRoot,
public CComCoClass<CATlkObj, &CLSID_CATlkObj>,
public INetCfgComponentControl,
public INetCfgComponentSetup,
public INetCfgComponentPropertyUi,
public INetCfgComponentNotifyBinding
{
public:
CATlkObj();
~CATlkObj();
BEGIN_COM_MAP(CATlkObj)
COM_INTERFACE_ENTRY(INetCfgComponentControl)
COM_INTERFACE_ENTRY(INetCfgComponentSetup)
COM_INTERFACE_ENTRY(INetCfgComponentPropertyUi)
COM_INTERFACE_ENTRY(INetCfgComponentNotifyBinding)
END_COM_MAP()
// DECLARE_NOT_AGGREGATABLE(CATlkObj)
// Remove the comment from the line above if you don't want your object to
// support aggregation. The default is to support it
DECLARE_REGISTRY_RESOURCEID(IDR_REG_ATLKCFG)
// Install Action (Unknown, Install, Remove)
enum INSTALLACTION {eActUnknown, eActInstall, eActRemove};
// INetCfgComponentControl
STDMETHOD (Initialize) (
IN INetCfgComponent* pIComp,
IN INetCfg* pINetCfg,
IN BOOL fInstalling);
STDMETHOD (ApplyRegistryChanges) ();
STDMETHOD (ApplyPnpChanges) (
IN INetCfgPnpReconfigCallback* pICallback) { return S_OK; }
STDMETHOD (CancelChanges) ();
STDMETHOD (Validate) ();
// INetCfgComponentSetup
STDMETHOD (Install) (DWORD dwSetupFlags);
STDMETHOD (Upgrade) (DWORD dwSetupFlags,
DWORD dwUpgradeFromBuildNo) {return S_OK;}
STDMETHOD (ReadAnswerFile) (PCWSTR pszAnswerFile,
PCWSTR pszAnswerSection);
STDMETHOD (Removing) ();
// INetCfgProperties
STDMETHOD (QueryPropertyUi) (
IN IUnknown* pUnk) { return S_OK; }
STDMETHOD (SetContext) (
IN IUnknown* pUnk);
STDMETHOD (MergePropPages) (
IN OUT DWORD* pdwDefPages,
OUT LPBYTE* pahpspPrivate,
OUT UINT* pcPrivate,
IN HWND hwndParent,
OUT PCWSTR* pszStartPage);
STDMETHOD (ValidateProperties) (
HWND hwndSheet);
STDMETHOD (CancelProperties) ();
STDMETHOD (ApplyProperties) ();
// INetCfgNotifyBinding
STDMETHOD (QueryBindingPath) (DWORD dwChangeFlag, INetCfgBindingPath* pncbp);
STDMETHOD (NotifyBindingPath) (DWORD dwChangeFlag, INetCfgBindingPath* pncbp);
INetCfg * PNetCfg() {return m_pNetCfg;}
INetCfgComponent * PNCComponent() {return m_pNCC;}
HRESULT HrCommitInstall();
HRESULT HrCommitRemove();
CATLKEnv * PATLKEnv() {return m_pATLKEnv;}
VOID MarkAdapterListChanged() {m_fAdapterListChanged = TRUE;};
private:
VOID CleanupPropPages();
HRESULT HrProcessAnswerFile(PCWSTR pszAnswerFile,
PCWSTR pszAnswerSection);
HRESULT HrReadAdapterAnswerFileSection(CSetupInfFile * pcsif,
PCWSTR pszSection);
HRESULT HrAtlkReconfig();
private:
INetCfgComponent* m_pNCC;
INetCfg* m_pNetCfg;
INSTALLACTION m_eInstallAction;
CPropSheetPage * m_pspObj;
CATLKEnv * m_pATLKEnv;
CATLKEnv * m_pATLKEnv_PP; // Used by prop pages only
IUnknown * m_pUnkPropContext;
INT m_nIdxAdapterSelected; // Used by Prop Pages only
BOOL m_fAdapterListChanged;
BOOL m_fPropertyChange;
BOOL m_fFirstTimeInstall;
};