BOOL
CNdasDeviceRegistrar::Bootstrap()
{
BOOL fSuccess = FALSE;
BOOL fMigrated = FALSE;
//
// Set bootstrapping flag to prevent multiple events
// for DeviceSetChange Events
//
m_fBootstrapping = TRUE;
TCHAR szSubcontainer[30];
for (DWORD i = 0; i < MAX_SLOT_NUMBER; ++i) {
HRESULT hr = ::StringCchPrintf(szSubcontainer, 30, TEXT("%s\\%04d"), CFG_CONTAINER, i);
_ASSERTE(SUCCEEDED(hr) && "CFG_CONTAINER is too large???");
BOOL fAutoRegistered = FALSE;
fSuccess = _NdasSystemCfg.GetValueEx(
szSubcontainer,
_T("AutoRegistered"),
&fAutoRegistered);
if (fSuccess && fAutoRegistered)
{
DBGPRT_WARN_EX(_FT("Deleting %s: "), szSubcontainer);
// Auto registered devices are not persistent
// it is an error to show up here.
// We just ignore those entries
fSuccess = _NdasSystemCfg.DeleteContainer(szSubcontainer, TRUE);
if (!fSuccess)
{
DBGPRT_WARN_EX(_FT("Deleting %s failed: "), szSubcontainer);
}
continue;
}
NDAS_DEVICE_ID deviceId;
fSuccess = _NdasSystemCfg.GetSecureValueEx(
szSubcontainer,
_T("DeviceID"),
&deviceId,
sizeof(deviceId));
// ignore read fault - tampered or not exists
if (!fSuccess) {
continue;
}
PCNdasDevice pDevice = Register(deviceId, i);
// _ASSERTE(NULL != pDevice && "Failure of registration should not happed during bootstrap!");
// This may happen due to auto-register feature!
if (NULL == pDevice) {
continue;
}
ACCESS_MASK grantedAccess = GENERIC_READ;
const DWORD cbBuffer = sizeof(ACCESS_MASK) + sizeof(NDAS_DEVICE_ID);
BYTE pbBuffer[cbBuffer];
fSuccess = _NdasSystemCfg.GetSecureValueEx(
szSubcontainer,
_T("GrantedAccess"),
pbBuffer,
cbBuffer);
if (fSuccess) {
grantedAccess = *((ACCESS_MASK*)(pbBuffer));
}
grantedAccess |= GENERIC_READ; // to prevent invalid access mask configuration
// _ASSERTE(grantedAccess & GENERIC_READ); // invalid configuration?
pDevice->SetGrantedAccess(grantedAccess);
TCHAR szDeviceName[MAX_NDAS_DEVICE_NAME_LEN + 1];
fSuccess = _NdasSystemCfg.GetValueEx(
szSubcontainer,
_T("DeviceName"),
szDeviceName,
sizeof(TCHAR)*(MAX_NDAS_DEVICE_NAME_LEN + 1));
if (fSuccess)
{
pDevice->SetName(szDeviceName);
}
BOOL fEnabled = FALSE;
fSuccess = _NdasSystemCfg.GetValueEx(
szSubcontainer,
_T("Enabled"),
&fEnabled);
if (fSuccess && fEnabled)
{
pDevice->Enable(fEnabled);
}
}
//
// Migration will be done only once
// if there is no registered devices in the current configurations
// and if the migration flag (Install\Migrate = 1) is set
//
if (m_deviceSlotMap.size() == 0) {
fSuccess = _NdasSystemCfg.GetValueEx(_T("Install"), _T("Migrated"), &fMigrated);
if (!fSuccess || !fMigrated) {
fMigrated = TRUE;
ImportLegacySettings();
_NdasSystemCfg.SetValueEx(_T("Install"), _T("Migrated"), fMigrated);
}
}
//
// Clear bootstrapping state
//
m_fBootstrapping = FALSE;
return TRUE;
}
BOOL
CNdasDeviceRegistrar::ImportLegacyEntry(DWORD dwSlotNo, HKEY hEntryKey)
{
static CONST size_t CB_ADDR = sizeof(TCHAR) * 18;
HRESULT hr = E_FAIL;
TCHAR szAddrVal[CB_ADDR + 1];
DWORD cbAddrVal = sizeof(szAddrVal);
DWORD dwValueType;
LONG lResult = ::RegQueryValueEx(
hEntryKey,
_T("Address"),
0,
&dwValueType,
(LPBYTE)szAddrVal,
&cbAddrVal);
if (ERROR_SUCCESS != lResult) {
// Ignore invalid values
return FALSE;
}
if (cbAddrVal != CB_ADDR) {
DBGPRT_ERR(_FT("Invalid Entry(A): %s, ignored\n"), szAddrVal);
return FALSE;
}
//
// 00:0B:D0:00:D4:2F to NDAS_DEVICE_ID
//
NDAS_DEVICE_ID deviceId = {0};
BOOL fSuccess = pConvertStringToDeviceId(szAddrVal, &deviceId);
if (!fSuccess) {
DBGPRT_ERR(_FT("Invalid Entry(D): %s, ignored\n"), szAddrVal);
return FALSE;
}
DBGPRT_INFO(_FT("Importing an entry: %s\n"), CNdasDeviceId(deviceId).ToString());
TCHAR szNameVal[MAX_NDAS_DEVICE_NAME_LEN + 1] = {0};
DWORD cbNameVal = sizeof(szNameVal);
lResult = ::RegQueryValueEx(
hEntryKey,
_T("Name"),
0,
&dwValueType,
(LPBYTE)szNameVal,
&cbNameVal);
if (ERROR_SUCCESS != lResult || _T('\0') == szNameVal[0]) {
TCHAR szDefaultName[MAX_NDAS_DEVICE_NAME_LEN + 1] = {0};
fSuccess = _NdasSystemCfg.GetValueEx(
_T("Devices"),
_T("DefaultPrefix"),
szDefaultName,
sizeof(szDefaultName));
if (!fSuccess) {
hr = ::StringCchCopy(
szDefaultName,
MAX_NDAS_DEVICE_NAME_LEN + 1,
_T("NDAS Device "));
_ASSERTE(SUCCEEDED(hr));
}
hr = ::StringCchPrintf(
szNameVal,
MAX_NDAS_DEVICE_NAME_LEN,
_T("%s %d"),
szDefaultName,
dwSlotNo);
}
BYTE pbSerialKeyVal[9];
DWORD cbSerialKeyVal = sizeof(pbSerialKeyVal);
lResult = ::RegQueryValueEx(
hEntryKey,
_T("SerialKey"),
0,
&dwValueType,
(LPBYTE)pbSerialKeyVal,
&cbSerialKeyVal);
if (ERROR_SUCCESS != lResult) {
return FALSE;
}
if (cbSerialKeyVal != sizeof(pbSerialKeyVal)) {
return FALSE;
}
ACCESS_MASK fAccessMode = GENERIC_READ;
if (0xFF == pbSerialKeyVal[8]) {
// Registered as RW
fAccessMode |= GENERIC_WRITE;
} else if (0x00 == pbSerialKeyVal[8]) {
// Registered as RO
} else {
// Invalid value
return FALSE;
}
PCNdasDevice pDevice = Register(deviceId, dwSlotNo);
if (NULL == pDevice) {
DBGPRT_ERR_EX(_FT("Failed to register %s at %d during import: "),
CNdasDeviceId(deviceId).ToString(),
dwSlotNo);
return FALSE;
}
// Always enable this!
pDevice->Enable(TRUE);
pDevice->SetName(szNameVal);
pDevice->SetGrantedAccess(fAccessMode);
return TRUE;
}
