mirror of https://github.com/tongzx/nt5src
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
14 KiB
643 lines
14 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;
|
|
}
|