Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

676 lines
17 KiB

//****************************************************************************
//
// Module: ULS.DLL
// File: localprt.cpp
// Content: This file contains the LocalProtocol object.
// History:
// Wed 17-Apr-1996 11:13:54 -by- Viroon Touranachun [viroont]
//
// Copyright (c) Microsoft Corporation 1996-1997
//
//****************************************************************************
#include "ulsp.h"
#include "localprt.h"
#include "attribs.h"
#include "callback.h"
#include "culs.h"
//****************************************************************************
// Event Notifiers
//****************************************************************************
//
//****************************************************************************
// Class Implementation
//****************************************************************************
//
//****************************************************************************
// CLocalProt::CLocalProt (void)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
CLocalProt::CLocalProt (void)
:m_cRef (0),
m_fReadonly (FALSE),
m_hProt (NULL),
m_szName (NULL),
m_uPort (0),
m_szMimeType (NULL),
m_pAttrs (NULL),
m_pConnPt (NULL),
m_pIlsServer (NULL),
m_pszUser (NULL),
m_pszApp (NULL)
{
}
//****************************************************************************
// CLocalProt::~CLocalProt (void)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
CLocalProt::~CLocalProt (void)
{
// Release the connection point
//
if (m_pConnPt != NULL)
{
m_pConnPt->ContainerReleased();
((IConnectionPoint*)m_pConnPt)->Release();
};
// Release the attributes object
//
if (m_pAttrs != NULL)
{
m_pAttrs->Release();
};
// Release the buffer resources
//
::MemFree (m_szName);
::MemFree (m_szMimeType);
::MemFree (m_pszUser);
::MemFree (m_pszApp);
if (m_pIlsServer != NULL)
m_pIlsServer->Release ();
if (m_hProt != NULL)
::UlsLdap_VirtualUnRegisterProtocol(m_hProt);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::Init (BSTR bstrName, ULONG uPort, BSTR bstrMimeType)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::Init (BSTR bstrName, ULONG uPort, BSTR bstrMimeType)
{
HRESULT hr;
// Set the port number
//
this->m_uPort = uPort;
hr = BSTR_to_LPTSTR(&m_szName, bstrName);
if (SUCCEEDED(hr))
{
hr = BSTR_to_LPTSTR(&m_szMimeType, bstrMimeType);
if (SUCCEEDED(hr))
{
#ifdef LATER
// Initialize the attributes list
//
m_pAttrs = new CAttributes;
if (m_pAttrs != NULL)
m_pAttrs->SetAccessType (ILS_ATTRTYPE_NAME_VALUE);
#endif //LATER
// Make the connection point
//
m_pConnPt = new CConnectionPoint (&IID_IIlsProtocolNotify,
(IConnectionPointContainer *)this);
if (m_pConnPt != NULL)
{
((IConnectionPoint*)m_pConnPt)->AddRef();
hr = NOERROR;
}
else
{
hr = ILS_E_MEMORY;
};
};
};
// Make this as read/write access
//
ASSERT (! m_fReadonly);
return hr;
}
//****************************************************************************
// STDMETHODIMP
// CIlsProt::Init (LPTSTR szServerName, LPTSTR szUserName,
// LPTSTR szAppName, PLDAP_PROTINFO ppi)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::Init (CIlsServer *pIlsServer, LPTSTR szUserName,
LPTSTR szAppName, PLDAP_PROTINFO ppi)
{
HRESULT hr;
// Validate parameter
//
if (ppi == NULL)
return ILS_E_POINTER;
if (ppi->uSize != sizeof(*ppi))
return ILS_E_PARAMETER;
// Make this a readonly guy
//
m_fReadonly = TRUE;
// Remember port name
//
m_uPort = ppi->uPortNumber;
// Remember the server name
//
m_pIlsServer = pIlsServer;
pIlsServer->AddRef ();
hr = SetLPTSTR(&m_pszUser, szUserName);
if (SUCCEEDED(hr))
{
hr = SetLPTSTR(&m_pszApp, szAppName);
if (SUCCEEDED(hr))
{
hr = SetLPTSTR(&m_szName,
(LPCTSTR)(((PBYTE)ppi)+ppi->uOffsetName));
if (SUCCEEDED(hr))
{
hr = SetLPTSTR(&m_szMimeType,
(LPCTSTR)(((PBYTE)ppi)+ppi->uOffsetMimeType));
};
};
};
// cleanup is done in destructor
return hr;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::QueryInterface (REFIID riid, void **ppv)
//
// History:
// Wed 17-Apr-1996 11:14:08 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::QueryInterface (REFIID riid, void **ppv)
{
*ppv = NULL;
if (riid == IID_IIlsProtocol || riid == IID_IUnknown)
{
*ppv = (IIlsMain *) this;
}
else
{
if (riid == IID_IConnectionPointContainer)
{
*ppv = (IConnectionPointContainer *) this;
};
};
if (*ppv != NULL)
{
((LPUNKNOWN)*ppv)->AddRef();
return S_OK;
}
else
{
return ILS_E_NO_INTERFACE;
};
}
//****************************************************************************
// STDMETHODIMP_(ULONG)
// CLocalProt::AddRef (void)
//
// History:
// Wed 17-Apr-1996 11:14:17 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP_(ULONG)
CLocalProt::AddRef (void)
{
DllLock();
MyDebugMsg ((DM_REFCOUNT, "CLocalProt::AddRef: ref=%ld\r\n", m_cRef));
::InterlockedIncrement (&m_cRef);
return (ULONG) m_cRef;
}
//****************************************************************************
// STDMETHODIMP_(ULONG)
// CLocalProt::Release (void)
//
// History:
// Wed 17-Apr-1996 11:14:26 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP_(ULONG)
CLocalProt::Release (void)
{
DllRelease();
ASSERT (m_cRef > 0);
MyDebugMsg ((DM_REFCOUNT, "CLocalProt::Release: ref=%ld\r\n", m_cRef));
if (::InterlockedDecrement (&m_cRef) == 0)
{
delete this;
return 0;
}
return (ULONG) m_cRef;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::IsWritable (BOOL *pfWriteable)
//
//****************************************************************************
STDMETHODIMP
CLocalProt::IsWritable(BOOL *pfWriteable)
{
HRESULT hr;
if (pfWriteable)
{
*pfWriteable = !m_fReadonly;
hr = S_OK;
}
else
{
hr = ILS_E_POINTER;
}
return (hr);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::GetStandardAttribute (ILS_STD_ATTR_NAME stdAttr, BSTR *pbstrStdAttr)
//
// History:
// 1-16-97 Shishir Pardikar
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::GetStandardAttribute(
ILS_STD_ATTR_NAME stdAttr,
BSTR *pbstrStdAttr
)
{
LPTSTR lpszAttr = NULL;
BOOL fValid = TRUE;
HRESULT hr;
TCHAR sz[16];
if (pbstrStdAttr == NULL) {
return ILS_E_POINTER;
}
switch(stdAttr) {
case ILS_STDATTR_PROTOCOL_PORT:
lpszAttr = &sz[0];
wsprintf (&sz[0], TEXT ("%lu"), m_uPort);
break;
case ILS_STDATTR_PROTOCOL_NAME:
lpszAttr = m_szName;
break;
case ILS_STDATTR_PROTOCOL_MIME_TYPE:
lpszAttr = m_szMimeType;
break;
default:
fValid = FALSE;
break;
}
if (fValid) {
if (lpszAttr){
hr = LPTSTR_to_BSTR(pbstrStdAttr, lpszAttr);
}
else {
*pbstrStdAttr = NULL;
hr = NOERROR;
}
}
else {
hr = ILS_E_PARAMETER;
}
return (hr);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::SetStandardAttribute (ILS_STD_ATTR_NAME stdAttr, BSTR bstrStdAttr)
//
// History:
// 1-16-97 Shishir Pardikar
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::SetStandardAttribute(
ILS_STD_ATTR_NAME stdAttr,
BSTR bstrStdAttr
)
{
return (ILS_E_FAIL);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::Update(BSTR bstrServerName, ULONG *pulReqId)
//
// History:
// 1-16-97 Shishir Pardikar
// Created.
//****************************************************************************
STDMETHODIMP CLocalProt::
Update ( ULONG *pulReqID )
{
return (ILS_E_FAIL);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::IsSameAs (CLocalProt *pProtocol)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::IsSameAs (CLocalProt *pProtocol)
{
return (!lstrcmp(pProtocol->m_szName, this->m_szName) ?
NOERROR : S_FALSE);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::GetProtocolInfo (PLDAP_PROTINFO *ppProtInfo)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::GetProtocolInfo (PLDAP_PROTINFO *ppProtInfo)
{
PLDAP_PROTINFO ppi;
ULONG cName, cMime;
HRESULT hr;
// Assume failure
//
*ppProtInfo = NULL;
// Calculate the buffer size
//
cName = lstrlen(m_szName)+1;
cMime = lstrlen(m_szMimeType)+1;
// Allocate the buffer
//
ULONG cbTotalSize = sizeof (LDAP_PROTINFO) + (cName + cMime) * sizeof (TCHAR);
ppi = (PLDAP_PROTINFO) ::MemAlloc (cbTotalSize);
if (ppi == NULL)
{
hr = ILS_E_MEMORY;
}
else
{
// Fill the structure content
//
ppi->uSize = cbTotalSize;
ppi->uOffsetName = sizeof(*ppi);
ppi->uPortNumber = m_uPort;
ppi->uOffsetMimeType = ppi->uOffsetName + (cName*sizeof(TCHAR));
// Copy the user information
//
lstrcpy((LPTSTR)(((PBYTE)ppi)+ppi->uOffsetName), m_szName);
lstrcpy((LPTSTR)(((PBYTE)ppi)+ppi->uOffsetMimeType), m_szMimeType);
// Return the structure
//
*ppProtInfo = ppi;
};
return NOERROR;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::NotifySink (void *pv, CONN_NOTIFYPROC pfn)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::NotifySink (void *pv, CONN_NOTIFYPROC pfn)
{
HRESULT hr = S_OK;
if (m_pConnPt != NULL)
{
hr = m_pConnPt->Notify(pv, pfn);
};
return hr;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::GetID (BSTR *pbstrID)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::GetPortNumber (ULONG *pulPort)
{
// Validate parameter
//
if (pulPort == NULL)
{
return ILS_E_POINTER;
};
*pulPort = m_uPort;
return (NOERROR);
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::SetAttributes (IIlsAttributes *pAttributes, ULONG *puReqID)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP CLocalProt::
SetExtendedAttribute ( BSTR bstrName, BSTR bstrValue )
{
return ILS_E_NOT_IMPL;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::RemoveAttributes (IIlsAttributes *pAttributes, ULONG *puReqID)
//
// History:
// Wed 17-Apr-1996 11:14:03 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP CLocalProt::
RemoveExtendedAttribute ( BSTR bstrName )
{
return ILS_E_NOT_IMPL;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::GetAttributes (IIlsAttributes **pAttributes)
//
//****************************************************************************
STDMETHODIMP CLocalProt::
GetExtendedAttribute ( BSTR bstrName, BSTR *pbstrValue )
{
return ILS_E_NOT_IMPL;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::GetAllExtendedAttributes (IIlsAttributes **pAttributes)
//
//****************************************************************************
STDMETHODIMP CLocalProt::
GetAllExtendedAttributes ( IIlsAttributes **ppAttributes )
{
return ILS_E_NOT_IMPL;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::EnumConnectionPoints(IEnumConnectionPoints **ppEnum)
//
// History:
// Wed 17-Apr-1996 11:15:02 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::EnumConnectionPoints(IEnumConnectionPoints **ppEnum)
{
CEnumConnectionPoints *pecp;
HRESULT hr;
// Validate parameters
//
if (ppEnum == NULL)
{
return ILS_E_POINTER;
};
// Assume failure
//
*ppEnum = NULL;
// Create an enumerator
//
pecp = new CEnumConnectionPoints;
if (pecp == NULL)
return ILS_E_MEMORY;
// Initialize the enumerator
//
hr = pecp->Init((IConnectionPoint *)m_pConnPt);
if (FAILED(hr))
{
delete pecp;
return hr;
};
// Give it back to the caller
//
pecp->AddRef();
*ppEnum = pecp;
return S_OK;
}
//****************************************************************************
// STDMETHODIMP
// CLocalProt::FindConnectionPoint(REFIID riid, IConnectionPoint **ppcp)
//
// History:
// Wed 17-Apr-1996 11:15:09 -by- Viroon Touranachun [viroont]
// Created.
//****************************************************************************
STDMETHODIMP
CLocalProt::FindConnectionPoint(REFIID riid, IConnectionPoint **ppcp)
{
IID siid;
HRESULT hr;
// Validate parameters
//
if (ppcp == NULL)
{
return ILS_E_POINTER;
};
// Assume failure
//
*ppcp = NULL;
if (m_pConnPt != NULL)
{
hr = m_pConnPt->GetConnectionInterface(&siid);
if (SUCCEEDED(hr))
{
if (riid == siid)
{
*ppcp = (IConnectionPoint *)m_pConnPt;
(*ppcp)->AddRef();
hr = S_OK;
}
else
{
hr = ILS_E_NO_INTERFACE;
};
};
}
else
{
hr = ILS_E_NO_INTERFACE;
};
return hr;
}