266 lines
6.2 KiB
C
266 lines
6.2 KiB
C
|
#ifndef __CONNECTION_POINT_STUFF_H
|
||
|
|
#define __CONNECTION_POINT_STUFF_H
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
/* File: connect.h
|
||
|
|
|
||
|
|
Description: Provides declarations required for the quota controller to
|
||
|
|
support OLE connection points.
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
Date Description Programmer
|
||
|
|
-------- --------------------------------------------------- ----------
|
||
|
|
06/19/96 Initial creation. BrianAu
|
||
|
|
*/
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
#ifndef _INC_DSKQUOTA_H
|
||
|
|
# include "dskquota.h"
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_USER_H
|
||
|
|
# include "user.h" // MAX_USERNAME
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_OADISP_H
|
||
|
|
# include "oadisp.h" // OleAutoDispatch class.
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_DISPATCH_H
|
||
|
|
# include "dispatch.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
class ConnectionPoint : public IConnectionPoint, public IDispatch
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
LONG m_cRef; // Class object ref count.
|
||
|
|
DWORD m_cConnections; // Number of connections.
|
||
|
|
DWORD m_dwCookieNext; // Next cookie value to hand out.
|
||
|
|
LPUNKNOWN m_pUnkContainer; // IUnknown of ConnectionPointEnumerator
|
||
|
|
REFIID m_riid; // Reference to IID supported by conn pt.
|
||
|
|
HANDLE m_hMutex;
|
||
|
|
OleAutoDispatch m_Dispatch;
|
||
|
|
CArray<CONNECTDATA> m_ConnectionList;
|
||
|
|
|
||
|
|
void Lock(void)
|
||
|
|
{ WaitForSingleObject(m_hMutex, INFINITE); }
|
||
|
|
void ReleaseLock(void)
|
||
|
|
{ ReleaseMutex(m_hMutex); }
|
||
|
|
|
||
|
|
//
|
||
|
|
// Prevent copying.
|
||
|
|
//
|
||
|
|
ConnectionPoint(const ConnectionPoint&);
|
||
|
|
void operator = (const ConnectionPoint&);
|
||
|
|
|
||
|
|
|
||
|
|
public:
|
||
|
|
ConnectionPoint(LPUNKNOWN pUnkContainer, REFIID riid);
|
||
|
|
~ConnectionPoint(void);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
QueryInterface(
|
||
|
|
REFIID,
|
||
|
|
LPVOID *);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
AddRef(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
Release(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IConnectionPoint methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
GetConnectionInterface(
|
||
|
|
LPIID pIID);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetConnectionPointContainer(
|
||
|
|
PCONNECTIONPOINTCONTAINER *ppCPC);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Advise(
|
||
|
|
LPUNKNOWN pUnkSink,
|
||
|
|
LPDWORD pdwCookie);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Unadvise(
|
||
|
|
DWORD dwCookie);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
EnumConnections(
|
||
|
|
PENUMCONNECTIONS *ppEnum);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IDispatch methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
GetIDsOfNames(
|
||
|
|
REFIID riid,
|
||
|
|
OLECHAR ** rgszNames,
|
||
|
|
UINT cNames,
|
||
|
|
LCID lcid,
|
||
|
|
DISPID *rgDispId);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetTypeInfo(
|
||
|
|
UINT iTInfo,
|
||
|
|
LCID lcid,
|
||
|
|
ITypeInfo **ppTInfo);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetTypeInfoCount(
|
||
|
|
UINT *pctinfo);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Invoke(
|
||
|
|
DISPID dispIdMember,
|
||
|
|
REFIID riid,
|
||
|
|
LCID lcid,
|
||
|
|
WORD wFlags,
|
||
|
|
DISPPARAMS *pDispParams,
|
||
|
|
VARIANT *pVarResult,
|
||
|
|
EXCEPINFO *pExcepInfo,
|
||
|
|
UINT *puArgErr);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class ConnectionEnum : public IEnumConnections
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
LONG m_cRef; // Object ref count.
|
||
|
|
UINT m_iCurrent; // "Current" enum index.
|
||
|
|
UINT m_cConnections; // Connection count.
|
||
|
|
PCONNECTDATA m_rgConnections; // Array of connection info.
|
||
|
|
LPUNKNOWN m_pUnkContainer; // Connection pt container.
|
||
|
|
|
||
|
|
//
|
||
|
|
// Prevent assignment.
|
||
|
|
//
|
||
|
|
void operator = (const ConnectionEnum&);
|
||
|
|
|
||
|
|
public:
|
||
|
|
ConnectionEnum(
|
||
|
|
LPUNKNOWN pUnkContainer,
|
||
|
|
UINT cConnection,
|
||
|
|
PCONNECTDATA rgConnections);
|
||
|
|
|
||
|
|
ConnectionEnum(
|
||
|
|
const ConnectionEnum& refEnum);
|
||
|
|
|
||
|
|
~ConnectionEnum(void);
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
Initialize(
|
||
|
|
UINT cConnection,
|
||
|
|
PCONNECTDATA rgConnections);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
QueryInterface(
|
||
|
|
REFIID,
|
||
|
|
LPVOID *);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
AddRef(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
Release(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IEnumConnections methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
Next(
|
||
|
|
DWORD,
|
||
|
|
PCONNECTDATA,
|
||
|
|
LPDWORD);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Skip(
|
||
|
|
DWORD);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Reset(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Clone(
|
||
|
|
PENUMCONNECTIONS *);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class ConnectionPointEnum : public IEnumConnectionPoints
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
LONG m_cRef; // Object ref count.
|
||
|
|
UINT m_iCurrent; // "Current" enum index.
|
||
|
|
UINT m_cConnPts; // Connection point count.
|
||
|
|
PCONNECTIONPOINT *m_rgConnPts; // Array of connection info.
|
||
|
|
LPUNKNOWN m_pUnkContainer; // IUnknown of DiskQuotaController.
|
||
|
|
|
||
|
|
//
|
||
|
|
// Prevent assignment.
|
||
|
|
//
|
||
|
|
void operator = (const ConnectionPointEnum&);
|
||
|
|
|
||
|
|
public:
|
||
|
|
ConnectionPointEnum(
|
||
|
|
LPUNKNOWN pUnkContainer,
|
||
|
|
UINT cConnPts,
|
||
|
|
PCONNECTIONPOINT *rgConnPts);
|
||
|
|
|
||
|
|
ConnectionPointEnum(
|
||
|
|
const ConnectionPointEnum& refEnum);
|
||
|
|
|
||
|
|
~ConnectionPointEnum(void);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
QueryInterface(
|
||
|
|
REFIID,
|
||
|
|
LPVOID *);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
AddRef(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
Release(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IEnumConnections methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
Next(
|
||
|
|
DWORD,
|
||
|
|
PCONNECTIONPOINT *,
|
||
|
|
LPDWORD);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Skip(
|
||
|
|
DWORD);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Reset(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Clone(
|
||
|
|
PENUMCONNECTIONPOINTS *);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif // CONNECTION_POINT_STUFF_H
|