980 lines
18 KiB
C++
980 lines
18 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (C) Microsoft Corporation, 1995 - 1999
|
||
|
|
All rights reserved.
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
folder.hxx
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Holds TServerNotify definitions.
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Albert Ting (AlbertT) 30-Oct-1995
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#ifndef _FOLDER_HXX
|
||
|
|
#define _FOLDER_HXX
|
||
|
|
|
||
|
|
class TFolder;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Class summary:
|
||
|
|
|
||
|
|
TFolder: This objects represents the print folder (local or
|
||
|
|
remote) that a user opens in the UI. It uses TConnectionNotify
|
||
|
|
to watch for new changes, and then creates TServers to hold
|
||
|
|
the server/printer data.
|
||
|
|
|
||
|
|
Destruction refcounted.
|
||
|
|
|
||
|
|
TConnectionNotify: if the Folder is local, then TFolder creates
|
||
|
|
and owns this. This object is responsible for telling
|
||
|
|
TFolder (via ConnectionNotifyClient) that the registry has
|
||
|
|
changed and TFolder should walk through its TServers, either
|
||
|
|
adding new ones or deleting old ones. (Compare EnumPrinters
|
||
|
|
with INFO_4 vs TFolder's linked list.)
|
||
|
|
|
||
|
|
Destruction sync'd with TFolder.
|
||
|
|
|
||
|
|
VDataSource: watch set of printers. This class is responsible for
|
||
|
|
keeping and updating data about a server/printer. It also
|
||
|
|
sends out notifications to the shell when something changes.
|
||
|
|
It uses TPrinter to get the data. TDSServer and VDSConnection
|
||
|
|
derive from this class.
|
||
|
|
|
||
|
|
Destruction refcounted.
|
||
|
|
|
||
|
|
TDSServer: watch a server with >=0 printers.
|
||
|
|
The single data source will return many printers. It
|
||
|
|
will internally enumerate masq printers, but it won't
|
||
|
|
return them via Get/EnumPrinters.
|
||
|
|
|
||
|
|
Inherits from VDataSource.
|
||
|
|
|
||
|
|
VDSConnection: watch a connection with 1 printer. This is
|
||
|
|
needed is true connects don't appear on the server, and
|
||
|
|
masq printers appear but are inaccurate.
|
||
|
|
|
||
|
|
Inherits from VDataSource.
|
||
|
|
|
||
|
|
TDSCTrue: watch a VDSConnection (true connect)
|
||
|
|
|
||
|
|
Inherits from VDSConnection since we gather information
|
||
|
|
from a separate handle.
|
||
|
|
|
||
|
|
TDSCMasq: watch a VDSConnection (masq connect).
|
||
|
|
|
||
|
|
Inherits from VDSConnection since we gather information
|
||
|
|
from a separate handle.
|
||
|
|
|
||
|
|
TNotifyHandler:
|
||
|
|
|
||
|
|
Utility class. Holder of IFolderNotify* handlers allowing to be
|
||
|
|
chained in a linked list.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
TConnectionNotify
|
||
|
|
|
||
|
|
Watch the registry looking for new printer addition. When
|
||
|
|
a new connection has been created, add the the TServer (actually
|
||
|
|
a printer) to pFolder.
|
||
|
|
|
||
|
|
Client constructs with a registry path off of HKEY_CURRENT_USER
|
||
|
|
and passes in a callback. When the registry node changes, the
|
||
|
|
callback is triggered.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TConnectionNotify : public MNotifyWork
|
||
|
|
{
|
||
|
|
SIGNATURE( 'regn' )
|
||
|
|
SAFE_NEW
|
||
|
|
|
||
|
|
public:
|
||
|
|
TConnectionNotify(
|
||
|
|
TFolder* pFolder
|
||
|
|
);
|
||
|
|
~TConnectionNotify(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
private:
|
||
|
|
TFolder *_pFolder;
|
||
|
|
CAutoHandleHKEY _shKeyConnections;
|
||
|
|
CAutoHandleNT _shEvent;
|
||
|
|
BOOL _bRegistered;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bSetupNotify(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Virtual definition MNotifyWork.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
HANDLE
|
||
|
|
hEvent(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vProcessNotifyWork(
|
||
|
|
TNotify* pNotify
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
inline
|
||
|
|
BOOL
|
||
|
|
TConnectionNotify::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return _shEvent != NULL && MNotifyWork::bValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
VDataSource
|
||
|
|
|
||
|
|
VDataSource uses the name constructor class factory idiom.
|
||
|
|
Instead of using the normal ctr, client should use pNew(),
|
||
|
|
which instantiates a derieved, concrete class of VDataSource.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class VDataSource : public MPrinterClient {
|
||
|
|
|
||
|
|
SIGNATURE( 'daso' )
|
||
|
|
SAFE_NEW
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
enum CONNECT_TYPE {
|
||
|
|
kServer,
|
||
|
|
kTrue,
|
||
|
|
kMasq
|
||
|
|
};
|
||
|
|
|
||
|
|
VAR( TString, strDataSource );
|
||
|
|
VAR( CONNECT_TYPE, ConnectType );
|
||
|
|
DLINK( VDataSource, DataSource );
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Separate constructor/destructor since we always build
|
||
|
|
a derived class of VDataSource (based on ConnectType).
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
static
|
||
|
|
VDataSource*
|
||
|
|
pNew(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszServer,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vDelete(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
HANDLE
|
||
|
|
hItemFindByName(
|
||
|
|
LPCTSTR pszPrinter
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bSkipItem(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Overrides that derived classes can define.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
virtual
|
||
|
|
BOOL
|
||
|
|
bRefresh(
|
||
|
|
VOID
|
||
|
|
) = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
BOOL
|
||
|
|
bReopen(
|
||
|
|
VOID
|
||
|
|
) = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
BOOL
|
||
|
|
bGetPrinter(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
PFOLDER_PRINTER_DATA pData,
|
||
|
|
DWORD cbData,
|
||
|
|
PDWORD pcbNeeded
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
BOOL
|
||
|
|
bAdministrator(
|
||
|
|
VOID
|
||
|
|
) const = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
COUNTB
|
||
|
|
cbSinglePrinterData(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
VOID
|
||
|
|
vPackSinglePrinterData(
|
||
|
|
HANDLE hItem,
|
||
|
|
PBYTE& pBegin,
|
||
|
|
PBYTE& pEnd
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
COUNTB
|
||
|
|
cbAllPrinterData(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
COUNT
|
||
|
|
cPackAllPrinterData(
|
||
|
|
PBYTE& pBegin,
|
||
|
|
PBYTE& pEnd
|
||
|
|
) const;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
|
||
|
|
TFolder* _pFolder;
|
||
|
|
COUNT _cIgnoreNotifications;
|
||
|
|
LONG _lItems;
|
||
|
|
|
||
|
|
VDataSource(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszDataSource,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
~VDataSource(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
virtual
|
||
|
|
VOID
|
||
|
|
vContainerChanged(
|
||
|
|
CONTAINER_CHANGE ContainerChange,
|
||
|
|
INFO Info
|
||
|
|
);
|
||
|
|
|
||
|
|
virtual
|
||
|
|
BOOL
|
||
|
|
bGetPrintLib(
|
||
|
|
TRefLock<TPrintLib> &refLock
|
||
|
|
) const;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Data retrieval.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
virtual
|
||
|
|
LPCTSTR
|
||
|
|
pszGetPrinterName(
|
||
|
|
HANDLE hItem
|
||
|
|
) const = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
LPCTSTR
|
||
|
|
pszGetCommentString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
LPCTSTR
|
||
|
|
pszGetLocationString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
LPCTSTR
|
||
|
|
pszGetModelString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
LPCTSTR
|
||
|
|
VDataSource::
|
||
|
|
pszGetPortString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Change callbacks and support.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
virtual
|
||
|
|
VOID
|
||
|
|
vReloadItems(
|
||
|
|
VOID
|
||
|
|
) = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
VOID
|
||
|
|
vRefreshComplete(
|
||
|
|
VOID
|
||
|
|
) const = 0;
|
||
|
|
|
||
|
|
virtual
|
||
|
|
FOLDER_NOTIFY_TYPE
|
||
|
|
uItemCreate(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
BOOL bNotify
|
||
|
|
) = 0;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
MPrinterClient / MDataClient virtual definitions.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vItemChanged(
|
||
|
|
ITEM_CHANGE ItemChange,
|
||
|
|
HITEM hItem,
|
||
|
|
INFO Info,
|
||
|
|
INFO InfoNew
|
||
|
|
);
|
||
|
|
|
||
|
|
VDataNotify*
|
||
|
|
pNewNotify(
|
||
|
|
MDataClient* pDataClient
|
||
|
|
) const;
|
||
|
|
|
||
|
|
VDataRefresh*
|
||
|
|
pNewRefresh(
|
||
|
|
MDataClient* pDataClient
|
||
|
|
) const;
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vRefZeroed(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
inline
|
||
|
|
BOOL
|
||
|
|
VDataSource::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return _pPrinter != NULL && MPrinterClient::bValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
TDSServer
|
||
|
|
|
||
|
|
Watch a server for printers. We can also watch a particular
|
||
|
|
printer (\\server\share) and it will act like a server with
|
||
|
|
just one printer.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TDSServer : public VDataSource {
|
||
|
|
friend
|
||
|
|
VDataSource*
|
||
|
|
VDataSource::pNew(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszServer,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
SIGNATURE( 'dssv' )
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
BOOL _bDiscardRefresh;
|
||
|
|
|
||
|
|
TDSServer(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszDataSource
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bRefresh(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bReopen(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Data retrieval.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
LPCTSTR
|
||
|
|
pszGetPrinterName(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bAdministrator(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bGetPrinter(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
PFOLDER_PRINTER_DATA pData,
|
||
|
|
DWORD cbData,
|
||
|
|
PDWORD pcbNeeded
|
||
|
|
) const;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Change callbacks and support.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vContainerChanged(
|
||
|
|
CONTAINER_CHANGE ContainerChange,
|
||
|
|
INFO Info
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vReloadItems(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vRefreshComplete(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
FOLDER_NOTIFY_TYPE
|
||
|
|
uItemCreate(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
BOOL bNotify
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
VDSConnection
|
||
|
|
|
||
|
|
Watch a connection.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class VDSConnection : public VDataSource {
|
||
|
|
|
||
|
|
SIGNATURE( 'dscn' )
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
static
|
||
|
|
BOOL
|
||
|
|
bStaticInitShutdown(
|
||
|
|
BOOL bShutdown = FALSE
|
||
|
|
);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
|
||
|
|
VDSConnection(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszDataSource,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
friend
|
||
|
|
VDataSource*
|
||
|
|
VDataSource::pNew(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszServer,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
CONNECT_STATUS _ConnectStatus;
|
||
|
|
|
||
|
|
static TString *gpstrConnectStatusOpen;
|
||
|
|
static TString *gpstrConnectStatusOpenError;
|
||
|
|
static TString *gpstrConnectStatusAccessDenied;
|
||
|
|
static TString *gpstrConnectStatusInvalidPrinterName;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bRefresh(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bReopen(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Data retrieval.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
LPCTSTR
|
||
|
|
pszGetPrinterName(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
LPCTSTR
|
||
|
|
pszGetCommentString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
LPCTSTR
|
||
|
|
pszGetStatusString(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bAdministrator(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bGetPrinter(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
PFOLDER_PRINTER_DATA pData,
|
||
|
|
DWORD cbData,
|
||
|
|
PDWORD pcbNeeded
|
||
|
|
) const;
|
||
|
|
|
||
|
|
COUNTB
|
||
|
|
cbSinglePrinterData(
|
||
|
|
HANDLE hItem
|
||
|
|
) const;
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vPackSinglePrinterData(
|
||
|
|
HANDLE hItem,
|
||
|
|
PBYTE& pBegin,
|
||
|
|
PBYTE& pEnd
|
||
|
|
) const;
|
||
|
|
|
||
|
|
COUNTB
|
||
|
|
cbAllPrinterData(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
COUNT
|
||
|
|
cPackAllPrinterData(
|
||
|
|
PBYTE& pBegin,
|
||
|
|
PBYTE& pEnd
|
||
|
|
) const;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bUseFakePrinterData(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Change callbacks and support.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vContainerChanged(
|
||
|
|
CONTAINER_CHANGE ContainerChange,
|
||
|
|
INFO Info
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vReloadItems(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vRefreshComplete(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
FOLDER_NOTIFY_TYPE
|
||
|
|
uItemCreate(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
BOOL bNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vUpdateConnectStatus(
|
||
|
|
CONNECT_STATUS ConnectStatusNew
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
TDSCTrue: watch a true connection.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TDSCTrue : public VDSConnection {
|
||
|
|
friend
|
||
|
|
VDataSource*
|
||
|
|
VDataSource::pNew(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszServer,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
TDSCTrue(
|
||
|
|
TFolder *pFolder,
|
||
|
|
LPCTSTR pszDataSource
|
||
|
|
);
|
||
|
|
|
||
|
|
~TDSCTrue(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
};
|
||
|
|
|
||
|
|
inline
|
||
|
|
BOOL
|
||
|
|
TDSCTrue::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return VDSConnection::bValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
TDSCMasq: watch a masq or 'false' connection.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TDSCMasq : public VDSConnection {
|
||
|
|
friend
|
||
|
|
VDataSource*
|
||
|
|
VDataSource::pNew(
|
||
|
|
TFolder* pFolder,
|
||
|
|
LPCTSTR pszServer,
|
||
|
|
CONNECT_TYPE ConnectType
|
||
|
|
);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
TDSCMasq(
|
||
|
|
TFolder *pFolder,
|
||
|
|
LPCTSTR pszDataSource
|
||
|
|
);
|
||
|
|
|
||
|
|
~TDSCMasq(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
inline
|
||
|
|
BOOL
|
||
|
|
TDSCMasq::
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const
|
||
|
|
{
|
||
|
|
return VDSConnection::bValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
TFolder
|
||
|
|
|
||
|
|
Watch a server. If the server is local (NULL) then we have
|
||
|
|
to watch printer connections also. Too bad the spooler wasn't
|
||
|
|
architected to do this for us... maybe the next one will.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TFolder : public MRefCom
|
||
|
|
{
|
||
|
|
SIGNATURE( 'fold' )
|
||
|
|
SAFE_NEW
|
||
|
|
|
||
|
|
class TNotifyHandler {
|
||
|
|
|
||
|
|
SIGNATURE( 'tnhd' )
|
||
|
|
SAFE_NEW
|
||
|
|
|
||
|
|
//
|
||
|
|
// To prevent accidental copying of this object
|
||
|
|
//
|
||
|
|
TNotifyHandler( const TNotifyHandler& );
|
||
|
|
TNotifyHandler& operator =( const TNotifyHandler& );
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
TNotifyHandler(
|
||
|
|
IFolderNotify *pClientNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
~TNotifyHandler(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
VAR( BOOL, bValid );
|
||
|
|
VAR( IFolderNotify*, pClientNotify );
|
||
|
|
DLINK( TNotifyHandler, Link );
|
||
|
|
};
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
VAR( BOOL, bValid );
|
||
|
|
VAR( TConnectionNotify*, pConnectionNotify );
|
||
|
|
DLINK_BASE( VDataSource, DataSource, DataSource );
|
||
|
|
VAR( CCSLock, CritSec );
|
||
|
|
VAR( TRefLock<TPrintLib>, pPrintLib );
|
||
|
|
|
||
|
|
DLINK_BASE( TNotifyHandler, Handlers, Link );
|
||
|
|
VAR( TString, strLocalDataSource );
|
||
|
|
DLINK( TFolder, Link );
|
||
|
|
|
||
|
|
TFolder(
|
||
|
|
IN LPCTSTR pszDataSource
|
||
|
|
);
|
||
|
|
|
||
|
|
~TFolder(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
pLookupNotifyHandler(
|
||
|
|
IN IFolderNotify *pClientNotify,
|
||
|
|
OUT TNotifyHandler **ppHandler
|
||
|
|
);
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
RegisterNotifyHandler(
|
||
|
|
IN IFolderNotify *pClientNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
UnregisterNotifyHandler(
|
||
|
|
IN IFolderNotify *pClientNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bNotifyAllClients(
|
||
|
|
IN FOLDER_NOTIFY_TYPE NotifyType,
|
||
|
|
IN LPCWSTR pszName,
|
||
|
|
IN LPCWSTR pszNewName
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bLocal(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Called when the system needs to refresh a window because
|
||
|
|
// of a large change.
|
||
|
|
//
|
||
|
|
VOID
|
||
|
|
vRefreshUI(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vCleanup(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vAddDataSource(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
VDataSource::CONNECT_TYPE ConnectType,
|
||
|
|
BOOL bNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vAddMasqDataSource(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
BOOL bNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vDeleteMasqDataSource(
|
||
|
|
LPCTSTR pszPrinter
|
||
|
|
);
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vRevalidateMasqPrinters(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
static
|
||
|
|
VOID
|
||
|
|
vDefaultPrinterChanged(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
|
||
|
|
static
|
||
|
|
VOID
|
||
|
|
vCheckDeleteDefault(
|
||
|
|
LPCTSTR pszDeleted
|
||
|
|
);
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
Connection notify override.
|
||
|
|
|
||
|
|
We receive change notification callbacks from this virtual
|
||
|
|
function definition.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vConnectionNotifyChange(
|
||
|
|
BOOL bNotify
|
||
|
|
);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
VDataSource*
|
||
|
|
pFindDataSource(
|
||
|
|
LPCTSTR pszPrinter,
|
||
|
|
VDataSource::CONNECT_TYPE ConnectType
|
||
|
|
) const;
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
MRefCom override.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
vRefZeroed(
|
||
|
|
VOID
|
||
|
|
);
|
||
|
|
};
|
||
|
|
|
||
|
|
/********************************************************************
|
||
|
|
|
||
|
|
class TFolderList
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
class TFolderList {
|
||
|
|
|
||
|
|
SIGNATURE( 'fldl' )
|
||
|
|
SAFE_NEW
|
||
|
|
|
||
|
|
//
|
||
|
|
// To prevent accidental copying of this object
|
||
|
|
//
|
||
|
|
TFolderList( const TFolderList& );
|
||
|
|
TFolderList& operator =( const TFolderList& );
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
TFolderList( );
|
||
|
|
~TFolderList( );
|
||
|
|
|
||
|
|
static HRESULT
|
||
|
|
RegisterDataSource(
|
||
|
|
IN LPCTSTR pszDataSource,
|
||
|
|
IN IFolderNotify *pClientNotify,
|
||
|
|
OUT LPHANDLE phFolder,
|
||
|
|
OUT PBOOL pbAdministrator OPTIONAL
|
||
|
|
);
|
||
|
|
|
||
|
|
static HRESULT
|
||
|
|
UnregisterDataSource(
|
||
|
|
IN LPCTSTR pszDataSource,
|
||
|
|
IN IFolderNotify *pClientNotify,
|
||
|
|
OUT LPHANDLE phFolder
|
||
|
|
);
|
||
|
|
|
||
|
|
static BOOL
|
||
|
|
bValidFolderObject(
|
||
|
|
IN const TFolder *pFolder
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
bValid(
|
||
|
|
VOID
|
||
|
|
) const;
|
||
|
|
|
||
|
|
//
|
||
|
|
// This global lock serializes the calls to the external PrintUI cache
|
||
|
|
// private APIs.
|
||
|
|
//
|
||
|
|
static CCSLock *gpFolderLock;
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
static HRESULT
|
||
|
|
pLookupFolder(
|
||
|
|
IN LPCTSTR pszDataSource,
|
||
|
|
OUT TFolder **ppFolder
|
||
|
|
);
|
||
|
|
|
||
|
|
DLINK_BASE( TFolder, Folders, Link ); // List of the registered folders
|
||
|
|
static TFolderList *gpFolders; // Global TFolder cashe for the library
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif // ndef _FOLDER_HXX
|