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.
 
 
 
 
 
 

643 lines
15 KiB

#include <windows.h>
#include "netconn.h"
#include "nconnwrap.h"
#pragma warning(disable:4100) // unreferenced formal parameter
#define ARRAYSIZE(a) (sizeof(a) / sizeof((a)[0]))
#define ASSERT(a)
BOOL WINAPI IsProtocolInstalledA(LPCSTR pszProtocolDeviceID, BOOL bExhaustive);
HRESULT WINAPI InstallProtocolA(LPCSTR pszProtocol, HWND hwndParent, PROGRESS_CALLBACK pfnCallback, LPVOID pvCallbackParam);
HRESULT WINAPI RemoveProtocolA(LPCSTR pszProtocol);
BOOL WINAPI FindConflictingServiceA(LPCSTR pszWantService, NETSERVICE* pConflict);
//
// String helper class. Takes in a WCHAR string and converts it to multibyte.
// Allocates and frees space if the string is long.
//
class CStrIn
{
public:
CStrIn(LPCWSTR pwsz);
~CStrIn();
operator char*() {return _psz;};
private:
LPSTR _psz;
char _sz[128];
};
CStrIn::CStrIn(LPCWSTR pwsz)
{
if (pwsz)
{
//
// If a string was passed in always return a string - even if it's
// the empty string.
//
_psz = _sz;
int cch = WideCharToMultiByte(CP_ACP, 0, pwsz, -1, _sz, ARRAYSIZE(_sz),
NULL, NULL);
//
// If the conversion failed try to allocate a buffer to hold the
// multibyte version of the string.
//
if (0 == cch)
{
*_sz = '\0';
cch = WideCharToMultiByte(CP_ACP, 0, pwsz, -1, NULL, 0, NULL, NULL);
if (cch)
{
_psz = new char[cch];
if (_psz)
{
cch = WideCharToMultiByte(CP_ACP, 0, pwsz, -1, _psz, cch,
NULL, NULL);
if (0 == cch)
{
delete [] _psz;
_psz = _sz;
}
}
else
{
_psz = _sz;
}
}
}
}
else
{
_psz = NULL;
}
}
CStrIn::~CStrIn()
{
if (_psz != _sz && _psz)
{
delete [] _psz;
}
}
//
//
//
class CStrOut
{
public:
CStrOut(LPWSTR psz, int cch);
operator LPSTR() { return _sz;}
operator int() { return ARRAYSIZE(_sz);}
void Convert();
private:
LPWSTR _psz;
int _cch;
char _sz[MAX_PATH];
};
CStrOut::CStrOut(LPWSTR psz, int cch)
{
ASSERT(cch <= ARRAYSIZE(_sz));
_psz = psz;
_cch = cch;
}
void CStrOut::Convert()
{
if (_psz)
{
MultiByteToWideChar(CP_ACP, 0, _sz, -1, _psz, _cch);
}
}
//
//
//
class CStrsOut
{
public:
CStrsOut(LPWSTR** pppsz) {_pStructW = pppsz;};
operator LPSTR**() {return _pStructW ? &_pStructA : NULL;};
int Convert(int cStrs);
private:
int ConvertStruct(LPSTR* pStructA, int nStrs, LPWSTR* pStructW, LPWSTR pszLast);
DWORD SizeOfWideCharStruct(LPSTR* pStructA, int cStrs);
DWORD SizeOfPointerArea(int cStrs);
DWORD SizeOfWideCharStringArea(LPSTR* pStruct, int cStrs);
LPSTR EndOfLastString(LPSTR* pStruct, int cStrs);
LPSTR StartOfFirstString(LPSTR* pStruct);
private:
LPWSTR** _pStructW;
LPSTR* _pStructA;
};
int CStrsOut::ConvertStruct(LPSTR* pStructA, int nStrs, LPWSTR* pStructW, LPWSTR pszLast)
{
ASSERT(pStructA);
ASSERT(nStrs);
ASSERT(pStructW);
int nRet;
LPWSTR pszCurrent = (LPWSTR)&pStructW[nStrs];
for (nRet = 0; nRet < nStrs; nRet++)
{
pStructW[nRet] = pszCurrent;
pszCurrent += MultiByteToWideChar(CP_ACP, 0, pStructA[nRet], -1,
pszCurrent, pszLast - pszCurrent);
}
return nRet;
}
int CStrsOut::Convert(int cStrs)
{
int nRet;
if (_pStructW)
{
nRet = 0;
if (_pStructA)
{
DWORD cbStruct = SizeOfWideCharStruct(_pStructA, cStrs);
*_pStructW = (LPWSTR*)NetConnAlloc(cbStruct);
if (*_pStructW)
{
nRet = ConvertStruct(_pStructA, cStrs, *_pStructW,
(LPWSTR)((BYTE*)*_pStructW + cbStruct));
}
NetConnFree(_pStructA);
}
else
{
*_pStructW = NULL;
}
}
else
{
nRet = cStrs;
}
return nRet;
}
inline LPSTR CStrsOut::EndOfLastString(LPSTR* pStructA, int nStrs)
{
return pStructA[nStrs - 1] + lstrlenA(pStructA[nStrs - 1]);
}
inline LPSTR CStrsOut::StartOfFirstString(LPSTR* pStructA)
{
return *pStructA;
}
inline DWORD CStrsOut::SizeOfWideCharStringArea(LPSTR* pStructA, int nStrs)
{
return ((EndOfLastString(pStructA, nStrs) + 1) - StartOfFirstString(pStructA))
* sizeof(WCHAR);
}
inline DWORD CStrsOut::SizeOfPointerArea(int nStrs)
{
return nStrs * sizeof(LPWSTR);
}
DWORD CStrsOut::SizeOfWideCharStruct(LPSTR* pStructA, int nStrs)
{
return SizeOfPointerArea(nStrs) + SizeOfWideCharStringArea(pStructA, nStrs);
}
//
// Netservice helper class. Passes in an ansi NETSERVICE structure when
// cast to NETSERVICEA. Copies the ansi structure to a unicode structure
// coverting the strings.
//
#undef NETSERVICE
class CNetServiceOut
{
public:
CNetServiceOut(NETSERVICE* pNS) {_pNS = pNS;};
operator NETSERVICEA*() {return _pNS ? &_NSA : NULL;};
void Convert();
private:
NETSERVICEA _NSA;
NETSERVICE* _pNS;
};
void CNetServiceOut::Convert()
{
if (_pNS)
{
MultiByteToWideChar(CP_ACP, 0, _NSA.szClassKey, -1, _pNS->szClassKey,
ARRAYSIZE(_pNS->szClassKey));
MultiByteToWideChar(CP_ACP, 0, _NSA.szDeviceID, -1, _pNS->szDeviceID,
ARRAYSIZE(_pNS->szDeviceID));
MultiByteToWideChar(CP_ACP, 0, _NSA.szDisplayName, -1, _pNS->szDisplayName,
ARRAYSIZE(_pNS->szDisplayName));
}
}
//
//
//
#undef NETADAPTER
class CNetAdaptersOut
{
public:
CNetAdaptersOut(NETADAPTER** ppNA) {_ppNA = ppNA;};
operator NETADAPTERA**(){return _ppNA ? &_pNAA : NULL;};
int Convert(int cNAA);
private:
void ConvertNA(NETADAPTERA* pNAA, NETADAPTER* pNA);
private:
NETADAPTERA* _pNAA;
NETADAPTER** _ppNA;
};
int CNetAdaptersOut::Convert(int cNAA)
{
int nRet = 0;
*_ppNA = NULL;
if (cNAA > 0)
{
if (_pNAA)
{
*_ppNA = (NETADAPTER*)NetConnAlloc(sizeof(NETADAPTER) * cNAA);
if (*_ppNA)
{
for (nRet = 0; nRet < cNAA; nRet++)
{
ConvertNA(&_pNAA[nRet], &(*_ppNA)[nRet]);
}
NetConnFree(_pNAA);
}
}
}
return nRet;
}
void CNetAdaptersOut::ConvertNA(NETADAPTERA* pNAA, NETADAPTER* pNA)
{
ASSERT(pNAA);
ASSERT(pNA);
MultiByteToWideChar(CP_ACP, 0, pNAA->szDisplayName, -1, pNA->szDisplayName,
ARRAYSIZE(pNA->szDisplayName));
MultiByteToWideChar(CP_ACP, 0, pNAA->szDeviceID, -1, pNA->szDeviceID,
ARRAYSIZE(pNA->szDeviceID));
MultiByteToWideChar(CP_ACP, 0, pNAA->szEnumKey, -1, pNA->szEnumKey,
ARRAYSIZE(pNA->szEnumKey));
MultiByteToWideChar(CP_ACP, 0, pNAA->szClassKey, -1, pNA->szClassKey,
ARRAYSIZE(pNA->szClassKey));
MultiByteToWideChar(CP_ACP, 0, pNAA->szManufacturer, -1, pNA->szManufacturer,
ARRAYSIZE(pNA->szManufacturer));
MultiByteToWideChar(CP_ACP, 0, pNAA->szInfFileName, -1, pNA->szInfFileName,
ARRAYSIZE(pNA->szInfFileName));
pNA->bNicType = pNAA->bNicType;
pNA->bNetType = pNAA->bNetType;
pNA->bNetSubType = pNAA->bNetSubType;
pNA->bIcsStatus = pNAA->bIcsStatus;
pNA->bError = pNAA->bError;
pNA->bWarning = pNAA->bWarning;
pNA->devnode = pNAA->devnode;
}
//
//
//
class CNetAdapterIn
{
public:
CNetAdapterIn(const NETADAPTER* pNA);
operator NETADAPTERA*() {return _pNA ? &_NAA : NULL;};
private:
const NETADAPTER* _pNA;
NETADAPTERA _NAA;
};
CNetAdapterIn::CNetAdapterIn(const NETADAPTER* pNA)
{
_pNA = pNA;
if (pNA)
{
WideCharToMultiByte(CP_ACP, 0, pNA->szDisplayName, -1, _NAA.szDisplayName,
ARRAYSIZE(_NAA.szDisplayName), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pNA->szDeviceID, -1, _NAA.szDeviceID,
ARRAYSIZE(_NAA.szDeviceID), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pNA->szEnumKey, -1, _NAA.szEnumKey,
ARRAYSIZE(_NAA.szEnumKey), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pNA->szClassKey, -1, _NAA.szClassKey,
ARRAYSIZE(_NAA.szClassKey), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pNA->szManufacturer, -1, _NAA.szManufacturer,
ARRAYSIZE(_NAA.szManufacturer), NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pNA->szInfFileName, -1, _NAA.szInfFileName,
ARRAYSIZE(_NAA.szInfFileName), NULL, NULL);
_NAA.bNicType = pNA->bNicType;
_NAA.bNetType = pNA->bNetType;
_NAA.bNetSubType = pNA->bNetSubType;
_NAA.bIcsStatus = pNA->bIcsStatus;
_NAA.bError = pNA->bError;
_NAA.bWarning = pNA->bWarning;
}
}
//
//
//
#undef IsProtocolInstalled
BOOL WINAPI IsProtocolInstalled(LPCWSTR pszProtocolDeviceID, BOOL bExhaustive)
{
CStrIn CStrProtocolDeviceID(pszProtocolDeviceID);
return IsProtocolInstalledA(CStrProtocolDeviceID, bExhaustive);
}
#undef InstallProtocol
HRESULT WINAPI InstallProtocol(LPCWSTR pszProtocol, HWND hwndParent, PROGRESS_CALLBACK pfnCallback, LPVOID pvCallbackParam)
{
CStrIn CStrProtocol(pszProtocol);
return InstallProtocolA(CStrProtocol, hwndParent, pfnCallback, pvCallbackParam);
}
#undef RemoveProtocol
HRESULT WINAPI RemoveProtocol(LPCWSTR pszProtocol)
{
CStrIn CStrProtocol(pszProtocol);
return RemoveProtocolA(CStrProtocol);
}
#undef FindConflictingService
BOOL WINAPI FindConflictingService(LPCWSTR pszWantService, NETSERVICE* pConflict)
{
BOOL fRet;
CStrIn CStrWantService(pszWantService);
CNetServiceOut CNSConflict(pConflict);
fRet = FindConflictingServiceA(CStrWantService, CNSConflict);
if (fRet)
{
CNSConflict.Convert();
}
return fRet;
}
#undef EnumNetAdapters
int WINAPI EnumNetAdapters(NETADAPTER FAR** pprgNetAdapters)
{
int nRet;
CNetAdaptersOut CNANetAdapters(pprgNetAdapters);
nRet = EnumNetAdaptersA(CNANetAdapters);
nRet = CNANetAdapters.Convert(nRet);
return nRet;
}
/*
#undef InstallNetAdapters
HRESULT WINAPI InstallNetAdapter(LPCWSTR pszDeviceID, LPCWSTR pszInfPath, HWND hwndParent, PROGRESS_CALLBACK pfnProgress, LPVOID pvCallbackParam)
{
CStrIn CStrDeviceID(pszDeviceID);
CStrIn CStrInfPath(pszInfPath);
return InstallNetAdapterA(CStrDeviceID, CStrInfPath, hwndParent, pfnProgress, pvCallbackParam);
}
*/
#undef IsProtocolBoundToAdapter
BOOL WINAPI IsProtocolBoundToAdapter(LPCWSTR pszProtocolID, const NETADAPTER* pAdapter)
{
CStrIn CStrProtocolID(pszProtocolID);
CNetAdapterIn CNAAdapter(pAdapter);
return IsProtocolBoundToAdapterA(CStrProtocolID, CNAAdapter);
}
/*
#undef ENableNetAdapter
HRESULT WINAPI EnableNetAdapter(const NETADAPTER* pAdapter)
{
CNetAdapterIn CNAAdapter(pAdapter);
return EnableNetAdapterA(CNAAdapter);
}
*/
#undef IsClientInstalled
BOOL WINAPI IsClientInstalled(LPCWSTR pszClient, BOOL bExhaustive)
{
CStrIn CStrClient(pszClient);
return IsClientInstalledA(CStrClient, bExhaustive);
}
/*
#undef RemoveClient
HRESULT WINAPI RemoveClient(LPCWSTR pszClient)
{
CStrIn CStrClient(pszClient);
return RemoveClientA(pszClient);
}
*/
/*
#undef RemoveGhostedAdapters
HRESULT WINAPI RemoveGhostedAdapters(LPCWSTR pszDeviceID)
{
CStrIn CStrDeviceID(pszDeviceID);
return RemoveGhostedAdaptersA(CStrDeviceID);
}
*/
/*
#undef RemoveUnknownAdapters
HRESULT WINAPI RemoveUnknownAdapters(LPCWSTR pszDeviceID)
{
CStrIn CStrDeviceID(pszDeviceID);
return RemoveUnknownAdaptersA(CStrDeviceID);
}
*/
/*
#undef DoesAdapterMatchDeviceID
BOOL WINAPI DoesAdapterMatchDeviceID(const NETADAPTER* pAdapter, LPCWSTR pszDeviceID)
{
CNetAdapterIn CNAAdapter(pAdapter);
CStrIn CStrDeviceID(pszDeviceID);
return DoesAdapterMatchDeviceIDA(CNAAdapter, CStrDeviceID);
}
*/
#undef IsAdapterBroadband
BOOL WINAPI IsAdapterBroadband(const NETADAPTER* pAdapter)
{
CNetAdapterIn CNAAdapter(pAdapter);
return IsAdapterBroadbandA(CNAAdapter);
}
#undef SaveBroadbandSettings
void WINAPI SaveBroadbandSettings(LPCWSTR pszBroadbandAdapterNumber)
{
CStrIn CStrBroadbandAdapterNumber(pszBroadbandAdapterNumber);
SaveBroadbandSettingsA(CStrBroadbandAdapterNumber);
}
/*
#undef UpdateBroadbandSettings
BOOL WINAPI UpdateBroadbandSettings(LPWSTR pszEnumKeyBuf, int cchEnumKeyBuf)
{
return FALSE;
}
*/
#undef DetectHardware
HRESULT WINAPI DetectHardware(LPCWSTR pszDeviceID)
{
CStrIn CStrDeviceID(pszDeviceID);
return DetectHardwareA(CStrDeviceID);
}
#undef EnableAutodial
void WINAPI EnableAutodial(BOOL bAutodial, LPCWSTR pszConnection)
{
CStrIn CStrConnection(pszConnection);
EnableAutodialA(bAutodial, CStrConnection);
}
#undef SetDefaultDialupConnection
void WINAPI SetDefaultDialupConnection(LPCWSTR pszConnectionName)
{
CStrIn CStrConnectionName(pszConnectionName);
SetDefaultDialupConnectionA(CStrConnectionName);
}
#undef GetDefaultDialupConnection
void WINAPI GetDefaultDialupConnection(LPWSTR pszConnectionName, int cchMax)
{
CStrOut CStrConnectionName(pszConnectionName, cchMax);
GetDefaultDialupConnectionA(CStrConnectionName, CStrConnectionName);
CStrConnectionName.Convert();
}
#undef EnumMatchingNetBindings
int WINAPI EnumMatchingNetBindings(LPCWSTR pszParentBinding, LPCWSTR pszDeviceID, LPWSTR** pprgBindings)
{
int nRet;
CStrIn CStrParentBinding(pszParentBinding);
CStrIn CStrDeviceID(pszDeviceID);
CStrsOut CStrsBindings(pprgBindings);
nRet = EnumMatchingNetBindingsA(CStrParentBinding, CStrDeviceID, CStrsBindings);
nRet = CStrsBindings.Convert(nRet);
return nRet;
}