1043 lines
26 KiB
C++
1043 lines
26 KiB
C++
|
//+--------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Copyright (c) 1997-1999 Microsoft Corporation
|
||
|
|
//
|
||
|
|
// File: upg.cpp
|
||
|
|
//
|
||
|
|
// Contents:
|
||
|
|
//
|
||
|
|
// History:
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------------
|
||
|
|
#include "upg.h"
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
//----------------------------------------------------
|
||
|
|
//
|
||
|
|
// Global variables
|
||
|
|
//
|
||
|
|
//
|
||
|
|
#ifndef LICENOC_SMALL_UPG
|
||
|
|
|
||
|
|
JBInstance g_JbInstance;
|
||
|
|
JBSession g_JetSession(g_JbInstance);
|
||
|
|
JBDatabase g_JetDatabase(g_JetSession);
|
||
|
|
|
||
|
|
PBYTE g_pbSetupId=NULL;
|
||
|
|
DWORD g_cbSetupId=0;
|
||
|
|
|
||
|
|
PBYTE g_pbDomainSid=NULL;
|
||
|
|
DWORD g_cbDomainSid=0;
|
||
|
|
|
||
|
|
TCHAR g_szComputerName[MAX_COMPUTERNAME_LENGTH + 1];
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
TCHAR g_szOdbcDsn[128]=NT4LSERVER_DEFAULT_DSN; // ODBC DSN
|
||
|
|
TCHAR g_szOdbcUser[128]=NT4LSERVER_DEFAULT_USER; // ODBC User Name
|
||
|
|
TCHAR g_szOdbcPwd[128]=NT4LSERVER_DEFAULT_PWD; // ODBC Password
|
||
|
|
TCHAR g_szMdbFile[MAX_PATH+1];
|
||
|
|
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#ifndef LICENOC_SMALL_UPG
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpgradeIssuedLicenseTable(
|
||
|
|
CSQLStmt* lpSqlStmt,
|
||
|
|
JBDatabase& jbDatabase
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
|
||
|
|
Note:
|
||
|
|
|
||
|
|
Transaction must be active
|
||
|
|
|
||
|
|
++*/
|
||
|
|
{
|
||
|
|
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
NT4LICENSE nt4License;
|
||
|
|
NT4LICENSERECORD nt4LicenseRecord(&nt4License);
|
||
|
|
|
||
|
|
LICENSEDCLIENT LicensedClient;
|
||
|
|
LicensedTable LicensedTable(jbDatabase);
|
||
|
|
|
||
|
|
TLSLicensedIndexMatchHwid hwidHint;
|
||
|
|
|
||
|
|
BOOL bSuccess;
|
||
|
|
|
||
|
|
|
||
|
|
bSuccess = LicensedTable.OpenTable(
|
||
|
|
TRUE,
|
||
|
|
JET_bitTableUpdatable
|
||
|
|
);
|
||
|
|
|
||
|
|
if(bSuccess == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicensedTable.GetLastJetError());
|
||
|
|
SetLastError(dwStatus);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
|
||
|
|
dwStatus = NT4LicenseEnumBegin(
|
||
|
|
lpSqlStmt,
|
||
|
|
&nt4LicenseRecord
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
while( (dwStatus = NT4RecordEnumNext(lpSqlStmt)) == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
memset(&LicensedClient, 0, sizeof(LicensedClient));
|
||
|
|
|
||
|
|
LicensedClient.ucEntryStatus = 0;
|
||
|
|
LicensedClient.dwLicenseId = nt4License.dwLicenseId;
|
||
|
|
LicensedClient.dwKeyPackId = nt4License.dwKeyPackId;
|
||
|
|
|
||
|
|
LicensedClient.dwKeyPackLicenseId = 0;
|
||
|
|
|
||
|
|
// LicensedClient.ftLastModifyTime;
|
||
|
|
|
||
|
|
SAFESTRCPY(LicensedClient.szMachineName, nt4License.szMachineName);
|
||
|
|
SAFESTRCPY(LicensedClient.szUserName, nt4License.szUserName);
|
||
|
|
|
||
|
|
LicensedClient.ftIssueDate = nt4License.ftIssueDate;
|
||
|
|
LicensedClient.ftExpireDate = nt4License.ftExpireDate;
|
||
|
|
LicensedClient.ucLicenseStatus = nt4License.ucLicenseStatus;
|
||
|
|
LicensedClient.dwNumLicenses = nt4License.dwNumLicenses;
|
||
|
|
|
||
|
|
LicensedClient.dwSystemBiosChkSum = nt4LicenseRecord.checkSum.dwSystemBiosChkSum;
|
||
|
|
LicensedClient.dwVideoBiosChkSum = nt4LicenseRecord.checkSum.dwVideoBiosChkSum;
|
||
|
|
LicensedClient.dwFloppyBiosChkSum = nt4LicenseRecord.checkSum.dwFloppyBiosChkSum;
|
||
|
|
LicensedClient.dwHardDiskSize = nt4LicenseRecord.checkSum.dwHardDiskSize;
|
||
|
|
LicensedClient.dwRamSize = nt4LicenseRecord.checkSum.dwRamSize;
|
||
|
|
|
||
|
|
hwidHint = LicensedClient;
|
||
|
|
|
||
|
|
LicensedClient.dbLowerBound = hwidHint.dbLowerBound;
|
||
|
|
|
||
|
|
//
|
||
|
|
if(LicensedTable.InsertRecord(LicensedClient) == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicensedTable.GetLastJetError());
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
NT4RecordEnumEnd(lpSqlStmt);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_ODBC_NO_DATA_FOUND)
|
||
|
|
{
|
||
|
|
// something wrong in fetch().
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
LicensedTable.CloseTable();
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpgradeLicensePackDescTable(
|
||
|
|
CSQLStmt* lpSqlStmt,
|
||
|
|
JBDatabase& jbDatabase
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
|
||
|
|
Note:
|
||
|
|
|
||
|
|
Transaction must be active
|
||
|
|
|
||
|
|
++*/
|
||
|
|
{
|
||
|
|
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
NT4KEYPACKDESC nt4KpDesc;
|
||
|
|
NT4KEYPACKDESCRECORD nt4KpDescRecord(&nt4KpDesc);
|
||
|
|
|
||
|
|
LICPACKDESC LicPackDesc;
|
||
|
|
LicPackDescTable LicPackDescTable(jbDatabase);
|
||
|
|
|
||
|
|
BOOL bSuccess;
|
||
|
|
|
||
|
|
|
||
|
|
bSuccess = LicPackDescTable.OpenTable(
|
||
|
|
TRUE,
|
||
|
|
JET_bitTableUpdatable
|
||
|
|
);
|
||
|
|
|
||
|
|
if(bSuccess == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicPackDescTable.GetLastJetError());
|
||
|
|
SetLastError(dwStatus);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
|
||
|
|
dwStatus = NT4KeyPackDescEnumBegin(
|
||
|
|
lpSqlStmt,
|
||
|
|
&nt4KpDescRecord
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
while( (dwStatus = NT4RecordEnumNext(lpSqlStmt)) == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
memset(&LicPackDesc, 0, sizeof(LicPackDesc));
|
||
|
|
|
||
|
|
LicPackDesc.ucEntryStatus = 0;
|
||
|
|
LicPackDesc.dwKeyPackId = nt4KpDesc.dwKeyPackId;
|
||
|
|
LicPackDesc.dwLanguageId = nt4KpDesc.dwLanguageId;
|
||
|
|
// LicPackDesc.ftLastModifyTime = 0;
|
||
|
|
|
||
|
|
SAFESTRCPY(LicPackDesc.szCompanyName, nt4KpDesc.szCompanyName);
|
||
|
|
SAFESTRCPY(LicPackDesc.szProductName, nt4KpDesc.szProductName);
|
||
|
|
SAFESTRCPY(LicPackDesc.szProductDesc, nt4KpDesc.szProductDesc);
|
||
|
|
//
|
||
|
|
if(LicPackDescTable.InsertRecord(LicPackDesc) == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicPackDescTable.GetLastJetError());
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
NT4RecordEnumEnd(lpSqlStmt);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_ODBC_NO_DATA_FOUND)
|
||
|
|
{
|
||
|
|
// something wrong in fetch().
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
LicPackDescTable.CloseTable();
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpgradeLicensePackTable(
|
||
|
|
CSQLStmt* lpSqlStmt,
|
||
|
|
JBDatabase& jbDatabase
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
|
||
|
|
Note:
|
||
|
|
|
||
|
|
Transaction must be active
|
||
|
|
|
||
|
|
++*/
|
||
|
|
{
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
NT4KEYPACK nt4KeyPack;
|
||
|
|
NT4KEYPACKRECORD nt4KeyPackRecord(&nt4KeyPack);
|
||
|
|
|
||
|
|
LICENSEPACK LicPack;
|
||
|
|
LicPackTable LicPackTable(jbDatabase);
|
||
|
|
BOOL bSuccess=TRUE;
|
||
|
|
//FILETIME ftCurTime;
|
||
|
|
//SYSTEMTIME sysTime;
|
||
|
|
|
||
|
|
//GetSystemTime(&sysTime);
|
||
|
|
|
||
|
|
//if(SystemTimeToFileTime(&sysTime, &ftCurTime) == FALSE)
|
||
|
|
//{
|
||
|
|
// dwStatus = GetLastError();
|
||
|
|
// goto cleanup;
|
||
|
|
//}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open License KeyPack table
|
||
|
|
//
|
||
|
|
bSuccess = LicPackTable.OpenTable(
|
||
|
|
TRUE,
|
||
|
|
JET_bitTableUpdatable
|
||
|
|
);
|
||
|
|
if(bSuccess == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicPackTable.GetLastJetError());
|
||
|
|
|
||
|
|
SetLastError(dwStatus);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
|
||
|
|
dwStatus = NT4KeyPackEnumBegin(
|
||
|
|
lpSqlStmt,
|
||
|
|
&nt4KeyPackRecord
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
|
||
|
|
while( (dwStatus = NT4RecordEnumNext(lpSqlStmt)) == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
memset(&LicPack, 0, sizeof(LicPack));
|
||
|
|
|
||
|
|
//
|
||
|
|
// Copy over license pack information.
|
||
|
|
//
|
||
|
|
LicPack.ucEntryStatus = 0;
|
||
|
|
LicPack.dwKeyPackId = nt4KeyPack.dwKeyPackId;
|
||
|
|
|
||
|
|
//LicPack.ftLastModifyTime = 0; // don't want this license pack to
|
||
|
|
// be populated to other server
|
||
|
|
LicPack.dwAttribute = 0;
|
||
|
|
|
||
|
|
if( nt4KeyPack.ucKeyPackType > LSKEYPACKTYPE_LAST ||
|
||
|
|
nt4KeyPack.ucKeyPackType < LSKEYPACKTYPE_FIRST)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// This is an invalid key pack, ignore
|
||
|
|
//
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if( nt4KeyPack.ucKeyPackType == LSKEYPACKTYPE_SELECT ||
|
||
|
|
nt4KeyPack.ucKeyPackType == LSKEYPACKTYPE_TEMPORARY ||
|
||
|
|
nt4KeyPack.ucKeyPackType == LSKEYPACKTYPE_FREE)
|
||
|
|
{
|
||
|
|
LicPack.dwNextSerialNumber = nt4KeyPack.dwNumberOfLicenses + 1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
LicPack.dwNextSerialNumber = nt4KeyPack.dwTotalLicenseInKeyPack - nt4KeyPack.dwNumberOfLicenses + 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
LicPack.dwActivateDate = nt4KeyPack.dwActivateDate;
|
||
|
|
LicPack.dwExpirationDate = nt4KeyPack.dwExpirationDate;
|
||
|
|
LicPack.dwNumberOfLicenses = nt4KeyPack.dwNumberOfLicenses;
|
||
|
|
LicPack.ucKeyPackStatus = nt4KeyPack.ucKeyPackStatus;
|
||
|
|
|
||
|
|
LicPack.pbDomainSid = NULL;
|
||
|
|
LicPack.cbDomainSid = 0;
|
||
|
|
|
||
|
|
memcpy(
|
||
|
|
LicPack.szInstallId,
|
||
|
|
g_pbSetupId,
|
||
|
|
min(sizeof(LicPack.szInstallId)/sizeof(LicPack.szInstallId[0]) - 1, g_cbSetupId)
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
// LicPack.szDomainName
|
||
|
|
|
||
|
|
SAFESTRCPY(LicPack.szTlsServerName, g_szComputerName);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Standard KeyPack Property..
|
||
|
|
//
|
||
|
|
|
||
|
|
LicPack.ucAgreementType = nt4KeyPack.ucKeyPackType;
|
||
|
|
SAFESTRCPY(LicPack.szCompanyName, nt4KeyPack.szCompanyName)
|
||
|
|
SAFESTRCPY(LicPack.szProductId, nt4KeyPack.szProductId);
|
||
|
|
LicPack.wMajorVersion = nt4KeyPack.wMajorVersion;
|
||
|
|
LicPack.wMinorVersion = nt4KeyPack.wMinorVersion;
|
||
|
|
|
||
|
|
//
|
||
|
|
// TermSrv specific code
|
||
|
|
//
|
||
|
|
if(_tcsicmp(LicPack.szProductId, NT4HYDRAPRODUCT_FULLVERSION_SKU) == 0)
|
||
|
|
{
|
||
|
|
LicPack.dwPlatformType = 0xFF;
|
||
|
|
SAFESTRCPY(LicPack.szKeyPackId, nt4KeyPack.szProductId);
|
||
|
|
}
|
||
|
|
else if(_tcsicmp(LicPack.szProductId, NT4HYDRAPRODUCT_UPGRADE_SKU) == 0)
|
||
|
|
{
|
||
|
|
LicPack.dwPlatformType = 0x1;
|
||
|
|
SAFESTRCPY(LicPack.szKeyPackId, nt4KeyPack.szProductId);
|
||
|
|
}
|
||
|
|
else if(_tcsicmp(LicPack.szProductId, NT4HYDRAPRODUCT_EXISTING_SKU) == 0)
|
||
|
|
{
|
||
|
|
LicPack.dwPlatformType = 0x2;
|
||
|
|
SAFESTRCPY(LicPack.szKeyPackId, nt4KeyPack.szProductId);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
SAFESTRCPY(LicPack.szKeyPackId, nt4KeyPack.szKeyPackId);
|
||
|
|
LicPack.dwPlatformType = nt4KeyPack.dwPlatformType;
|
||
|
|
}
|
||
|
|
|
||
|
|
LicPack.ucLicenseType = nt4KeyPack.ucLicenseType;
|
||
|
|
LicPack.ucChannelOfPurchase = nt4KeyPack.ucChannelOfPurchase;
|
||
|
|
SAFESTRCPY(LicPack.szBeginSerialNumber, nt4KeyPack.szBeginSerialNumber);
|
||
|
|
LicPack.dwTotalLicenseInKeyPack = nt4KeyPack.dwTotalLicenseInKeyPack;
|
||
|
|
LicPack.dwProductFlags = nt4KeyPack.dwProductFlags;
|
||
|
|
|
||
|
|
//
|
||
|
|
if(LicPackTable.InsertRecord(LicPack) == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = SET_JB_ERROR(LicPackTable.GetLastJetError());
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
NT4RecordEnumEnd(lpSqlStmt);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_ODBC_NO_DATA_FOUND)
|
||
|
|
{
|
||
|
|
// something wrong in fetch().
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
|
||
|
|
lpSqlStmt->Cleanup();
|
||
|
|
LicPackTable.CloseTable();
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
GetNT4DbConfig(
|
||
|
|
LPTSTR pszDsn,
|
||
|
|
LPTSTR pszUserName,
|
||
|
|
LPTSTR pszPwd,
|
||
|
|
LPTSTR pszMdbFile
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
++*/
|
||
|
|
{
|
||
|
|
HKEY hKey = NULL;
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
TCHAR szOdbcDsn[128]=NT4LSERVER_DEFAULT_DSN; // ODBC DSN
|
||
|
|
TCHAR szOdbcUser[128]=NT4LSERVER_DEFAULT_USER; // ODBC User Name
|
||
|
|
TCHAR szOdbcPwd[128]=NT4LSERVER_DEFAULT_PWD; // ODBC Password
|
||
|
|
|
||
|
|
TCHAR szMdbFile[MAX_PATH+1];
|
||
|
|
DWORD dwBuffer=0;
|
||
|
|
|
||
|
|
PBYTE pbData = NULL;
|
||
|
|
DWORD cbData = 0;
|
||
|
|
|
||
|
|
BOOL bSuccess;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open NT4 license server specific registry key
|
||
|
|
//
|
||
|
|
dwStatus = RegOpenKeyEx(
|
||
|
|
HKEY_LOCAL_MACHINE,
|
||
|
|
NT4LSERVER_REGKEY,
|
||
|
|
0,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
&hKey
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
dwStatus = ERROR_INVALID_NT4_SETUP;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Load ODBC DSN and User name from registry,
|
||
|
|
// ignore error return and use default value.
|
||
|
|
//
|
||
|
|
dwBuffer = sizeof(szOdbcDsn);
|
||
|
|
dwStatus = RegQueryValueEx(
|
||
|
|
hKey,
|
||
|
|
NT4LSERVER_PARAMETERS_DSN,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
(LPBYTE)szOdbcDsn,
|
||
|
|
&dwBuffer
|
||
|
|
);
|
||
|
|
if(dwStatus == ERROR_SUCCESS && pszDsn)
|
||
|
|
{
|
||
|
|
lstrcpy(pszDsn, szOdbcDsn);
|
||
|
|
}
|
||
|
|
|
||
|
|
dwBuffer = sizeof(szOdbcUser);
|
||
|
|
dwStatus = RegQueryValueEx(
|
||
|
|
hKey,
|
||
|
|
NT4LSERVER_PARAMETERS_USER,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
(LPBYTE)szOdbcUser,
|
||
|
|
&dwBuffer
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus == ERROR_SUCCESS && pszUserName)
|
||
|
|
{
|
||
|
|
lstrcpy(pszUserName, szOdbcUser);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Load database password from LSA
|
||
|
|
//
|
||
|
|
dwStatus = RetrieveKey(
|
||
|
|
LSERVER_LSA_PASSWORD_KEYNAME,
|
||
|
|
&pbData,
|
||
|
|
&cbData
|
||
|
|
);
|
||
|
|
|
||
|
|
#ifndef PRIVATE_DBG
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Invalid NT4 license server setup or hydra beta2
|
||
|
|
// license server which we don't support.
|
||
|
|
//
|
||
|
|
dwStatus = ERROR_INVALID_NT4_SETUP;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
dwStatus = ERROR_SUCCESS;
|
||
|
|
memset(szOdbcPwd, 0, sizeof(szOdbcPwd));
|
||
|
|
memcpy(
|
||
|
|
(PBYTE)szOdbcPwd,
|
||
|
|
pbData,
|
||
|
|
min(cbData, sizeof(szOdbcPwd) - sizeof(TCHAR))
|
||
|
|
);
|
||
|
|
|
||
|
|
if(pszPwd != NULL)
|
||
|
|
{
|
||
|
|
lstrcpy(pszPwd, szOdbcPwd);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Verify data source is properly installed
|
||
|
|
//
|
||
|
|
bSuccess = IsDataSourceInstalled(
|
||
|
|
szOdbcDsn,
|
||
|
|
ODBC_SYSTEM_DSN,
|
||
|
|
szMdbFile,
|
||
|
|
MAX_PATH
|
||
|
|
);
|
||
|
|
|
||
|
|
if(bSuccess == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = ERROR_INVALID_NT4_SETUP;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(pszMdbFile != NULL)
|
||
|
|
{
|
||
|
|
_tcscpy(pszMdbFile, szMdbFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
|
||
|
|
if(hKey != NULL)
|
||
|
|
{
|
||
|
|
RegCloseKey(hKey);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(pbData != NULL)
|
||
|
|
{
|
||
|
|
LocalFree(pbData);
|
||
|
|
}
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
DWORD
|
||
|
|
DeleteNT4ODBCDataSource()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
BOOL bSuccess;
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get Hydra 4 DB configuration, make sure
|
||
|
|
// data source is properly config.
|
||
|
|
//
|
||
|
|
dwStatus = GetNT4DbConfig(
|
||
|
|
g_szOdbcDsn,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_szMdbFile
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
bSuccess = ConfigDataSource(
|
||
|
|
NULL,
|
||
|
|
FALSE,
|
||
|
|
_TEXT(SZACCESSDRIVERNAME),
|
||
|
|
g_szOdbcDsn,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_szMdbFile
|
||
|
|
);
|
||
|
|
|
||
|
|
if(bSuccess == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = ERROR_DELETE_ODBC_DSN;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#ifndef LICENOC_SMALL_UPG
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpgradeNT4Database(
|
||
|
|
IN DWORD dwServerRole,
|
||
|
|
IN LPCTSTR pszDbFilePath,
|
||
|
|
IN LPCTSTR pszDbFileName,
|
||
|
|
IN BOOL bAlwaysDeleteDataSource
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
This routine is to upgrade Hydra4 Terminal Licensing Server database to
|
||
|
|
NT5. Hydra4 uses ODBC/Jet while NT5 uses JetBlue as database engine,
|
||
|
|
other changes includes table structure and additional table.
|
||
|
|
|
||
|
|
Parameters:
|
||
|
|
|
||
|
|
dwServerRole - Server role in enterprise, 0 - ordinaly server, 1 - enterprise server
|
||
|
|
pszDbFilePath - Directory for NT5 Terminal Licensing Server Database. License
|
||
|
|
Server uses this directory as JetBlue instance's log/temp/system path.
|
||
|
|
pszDbFileName - Database file name, default to TlsLic.edb if NULL.
|
||
|
|
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
|
||
|
|
ERROR_SUCCESS or error code, see upg.h for list of error code and description
|
||
|
|
|
||
|
|
++*/
|
||
|
|
{
|
||
|
|
DWORD dwStatus=ERROR_SUCCESS;
|
||
|
|
TCHAR szFileName[MAX_PATH+1];
|
||
|
|
CSQLStmt sqlStmt;
|
||
|
|
UUID uuid;
|
||
|
|
unsigned short *szUuid=NULL;
|
||
|
|
BOOL bSuccess;
|
||
|
|
DWORD dwNt4DbVersion;
|
||
|
|
DWORD dwComputerName = sizeof(g_szComputerName)/sizeof(g_szComputerName[0]);
|
||
|
|
BOOL bNT4LserverExists = FALSE;
|
||
|
|
|
||
|
|
|
||
|
|
memset(g_szComputerName, 0, sizeof(g_szComputerName));
|
||
|
|
GetComputerName(g_szComputerName, &dwComputerName);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Verify input parameter. pszDbFilePath must not have '\' as last
|
||
|
|
// character.
|
||
|
|
//
|
||
|
|
if(pszDbFilePath == NULL || *pszDbFilePath == _TEXT('0'))
|
||
|
|
{
|
||
|
|
SetLastError(dwStatus = ERROR_INVALID_PARAMETER);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(pszDbFilePath[_tcslen(pszDbFilePath) - 1] == _TEXT('\\'))
|
||
|
|
{
|
||
|
|
SetLastError(dwStatus = ERROR_INVALID_PARAMETER);
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Make sure target directory exist.
|
||
|
|
//
|
||
|
|
if(FileExists(pszDbFilePath, NULL) == FALSE)
|
||
|
|
{
|
||
|
|
dwStatus = ERROR_TARGETFILE_NOT_FOUND;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Database file can't exist, we have no idea what it is.
|
||
|
|
//
|
||
|
|
wsprintf(
|
||
|
|
szFileName,
|
||
|
|
_TEXT("%s\\%s"),
|
||
|
|
pszDbFilePath,
|
||
|
|
(pszDbFileName) ? pszDbFileName : LSERVER_DEFAULT_EDB
|
||
|
|
);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Verify File does not exist, this file might
|
||
|
|
// not be a valid JetBlue database file, setup
|
||
|
|
// should verify and prompt user for different
|
||
|
|
// file name
|
||
|
|
//
|
||
|
|
if(FileExists(szFileName, NULL) == TRUE)
|
||
|
|
{
|
||
|
|
dwStatus = ERROR_DEST_FILE_EXIST;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get Hydra 4 DB configuration, make sure
|
||
|
|
// data source is properly config.
|
||
|
|
//
|
||
|
|
dwStatus = GetNT4DbConfig(
|
||
|
|
g_szOdbcDsn,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_szMdbFile
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
bNT4LserverExists = TRUE;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Generate License Server Setup Id
|
||
|
|
//
|
||
|
|
dwStatus = RetrieveKey(
|
||
|
|
LSERVER_LSA_SETUPID,
|
||
|
|
&g_pbSetupId,
|
||
|
|
&g_cbSetupId
|
||
|
|
);
|
||
|
|
|
||
|
|
#ifndef PRIVATE_DBG
|
||
|
|
if(dwStatus == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// NT4 Hydra does not use this setup ID.
|
||
|
|
//
|
||
|
|
dwStatus = ERROR_INVALID_NT4_SETUP;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
//
|
||
|
|
// Generate License Server Unique Setup ID
|
||
|
|
//
|
||
|
|
UuidCreate(&uuid);
|
||
|
|
UuidToString(&uuid, &szUuid);
|
||
|
|
|
||
|
|
g_pbSetupId = (PBYTE)AllocateMemory((_tcslen(szUuid)+1)*sizeof(TCHAR));
|
||
|
|
if(g_pbSetupId == NULL)
|
||
|
|
{
|
||
|
|
dwStatus = GetLastError();
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(
|
||
|
|
g_pbSetupId,
|
||
|
|
szUuid,
|
||
|
|
(_tcslen(szUuid) + 1)*sizeof(TCHAR)
|
||
|
|
);
|
||
|
|
|
||
|
|
g_cbSetupId = (_tcslen(szUuid) + 1)*sizeof(TCHAR);
|
||
|
|
|
||
|
|
RpcStringFree(&szUuid);
|
||
|
|
szUuid = NULL;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open one SQL handle to Hydra data source
|
||
|
|
//
|
||
|
|
dwStatus = OpenSqlStmtHandle(
|
||
|
|
&sqlStmt,
|
||
|
|
g_szOdbcDsn,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_szMdbFile
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Verify this is latest Hydra 4
|
||
|
|
//
|
||
|
|
dwStatus = GetNt4DatabaseVersion(
|
||
|
|
&sqlStmt,
|
||
|
|
(LONG *)&dwNt4DbVersion
|
||
|
|
);
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(dwNt4DbVersion != HYDRA_DATABASE_NT40_VERSION)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Only support release database version
|
||
|
|
//
|
||
|
|
dwStatus = ERROR_NOTSUPPORT_DB_VERSION;
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Initialize JetBlue and create an empty database
|
||
|
|
//
|
||
|
|
dwStatus = JetBlueInitAndCreateEmptyDatabase(
|
||
|
|
pszDbFilePath,
|
||
|
|
(pszDbFileName) ? pszDbFileName : LSERVER_DEFAULT_EDB,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_JbInstance,
|
||
|
|
g_JetSession,
|
||
|
|
g_JetDatabase
|
||
|
|
);
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Begin JetBlue Transaction,
|
||
|
|
// JetBlue Transaction is session based.
|
||
|
|
//
|
||
|
|
g_JetSession.BeginTransaction();
|
||
|
|
|
||
|
|
//
|
||
|
|
// Copy over licensed key pack table
|
||
|
|
//
|
||
|
|
dwStatus = UpgradeLicensePackTable(
|
||
|
|
&sqlStmt,
|
||
|
|
g_JetDatabase
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Copy over licensed key pack desc. table
|
||
|
|
//
|
||
|
|
dwStatus = UpgradeLicensePackDescTable(
|
||
|
|
&sqlStmt,
|
||
|
|
g_JetDatabase
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Copy over issued license table
|
||
|
|
//
|
||
|
|
dwStatus = UpgradeIssuedLicenseTable(
|
||
|
|
&sqlStmt,
|
||
|
|
g_JetDatabase
|
||
|
|
);
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
goto cleanup;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Write License Server Setup ID to LSA
|
||
|
|
//
|
||
|
|
dwStatus = StoreKey(
|
||
|
|
LSERVER_LSA_SETUPID,
|
||
|
|
g_pbSetupId,
|
||
|
|
g_cbSetupId
|
||
|
|
);
|
||
|
|
|
||
|
|
//
|
||
|
|
// TODO : if we could not write setup ID to LSA
|
||
|
|
//
|
||
|
|
cleanup:
|
||
|
|
|
||
|
|
if(dwStatus == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
g_JetSession.CommitTransaction();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
g_JetSession.RollbackTransaction();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Close ODBC handle
|
||
|
|
//
|
||
|
|
sqlStmt.Close();
|
||
|
|
CSQLStmt::Shutdown();
|
||
|
|
|
||
|
|
g_JetDatabase.CloseDatabase();
|
||
|
|
g_JetSession.EndSession();
|
||
|
|
|
||
|
|
if(dwStatus != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Reset last run status, password is not use on JetBlue,
|
||
|
|
// LSERVER_LSA_SETUPID will not get setup if error.
|
||
|
|
//
|
||
|
|
TLServerLastRunState lastRun;
|
||
|
|
|
||
|
|
memset(&lastRun, 0, sizeof(TLServerLastRunState));
|
||
|
|
lastRun.dwVersion = LSERVER_LSA_STRUCT_VERSION;
|
||
|
|
|
||
|
|
StoreKey(
|
||
|
|
LSERVER_LSA_LASTRUN,
|
||
|
|
(PBYTE)&lastRun,
|
||
|
|
sizeof(TLServerLastRunState)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if( (dwStatus == ERROR_SUCCESS) ||
|
||
|
|
(bAlwaysDeleteDataSource == TRUE && bNT4LserverExists == TRUE) )
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Remove "Hydra License" from data source.
|
||
|
|
// Non-critical error if we can't remove data source.
|
||
|
|
//
|
||
|
|
ConfigDataSource(
|
||
|
|
NULL,
|
||
|
|
FALSE,
|
||
|
|
_TEXT(SZACCESSDRIVERNAME),
|
||
|
|
g_szOdbcDsn,
|
||
|
|
g_szOdbcUser,
|
||
|
|
g_szOdbcPwd,
|
||
|
|
g_szMdbFile
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
FreeMemory(g_pbSetupId);
|
||
|
|
return dwStatus;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
//---------------------------------------------------
|
||
|
|
|
||
|
|
//
|
||
|
|
// License Server secret key info.
|
||
|
|
//
|
||
|
|
#define LSERVER_LSA_PRIVATEKEY_SIGNATURE _TEXT("TermServLiceningSignKey-12d4b7c8-77d5-11d1-8c24-00c04fa3080d")
|
||
|
|
#define LSERVER_LSA_PRIVATEKEY_EXCHANGE _TEXT("TermServLicensingExchKey-12d4b7c8-77d5-11d1-8c24-00c04fa3080d")
|
||
|
|
|
||
|
|
#define LSERVER_LSA_LSERVERID _TEXT("TermServLicensingServerId-12d4b7c8-77d5-11d1-8c24-00c04fa3080d")
|
||
|
|
|
||
|
|
#define LSERVER_SOFTWARE_REGBASE \
|
||
|
|
_TEXT("SOFTWARE\\Microsoft\\") _TEXT(SZSERVICENAME)
|
||
|
|
|
||
|
|
#define LSERVER_CERTIFICATE_STORE _TEXT("Certificates")
|
||
|
|
|
||
|
|
#define LSERVER_SELFSIGN_CERTIFICATE_REGKEY \
|
||
|
|
LSERVER_REGISTRY_BASE _TEXT(SZSERVICENAME) _TEXT("\\") LSERVER_SECRET
|
||
|
|
|
||
|
|
#define LSERVER_SERVER_CERTIFICATE_REGKEY \
|
||
|
|
LSERVER_SOFTWARE_REGBASE _TEXT("\\") LSERVER_CERTIFICATE_STORE
|
||
|
|
|
||
|
|
#define LSERVER_CLIENT_CERTIFICATE_ISSUER _TEXT("Parm0")
|
||
|
|
#define LSERVER_SIGNATURE_CERT_KEY _TEXT("Parm1")
|
||
|
|
#define LSERVER_EXCHANGE_CERT_KEY _TEXT("Parm2")
|
||
|
|
|
||
|
|
|
||
|
|
void
|
||
|
|
CleanLicenseServerSecret()
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DWORD dwStatus = ERROR_SUCCESS;
|
||
|
|
HKEY hKey = NULL;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Wipe out SPK in LSA
|
||
|
|
//
|
||
|
|
dwStatus = StoreKey(
|
||
|
|
LSERVER_LSA_LSERVERID,
|
||
|
|
(PBYTE) NULL,
|
||
|
|
0
|
||
|
|
);
|
||
|
|
|
||
|
|
dwStatus = StoreKey(
|
||
|
|
LSERVER_LSA_LASTRUN,
|
||
|
|
(PBYTE) NULL,
|
||
|
|
0
|
||
|
|
);
|
||
|
|
|
||
|
|
dwStatus = StoreKey(
|
||
|
|
LSERVER_LSA_PRIVATEKEY_EXCHANGE,
|
||
|
|
(PBYTE) NULL,
|
||
|
|
0
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
dwStatus = StoreKey(
|
||
|
|
LSERVER_LSA_PRIVATEKEY_SIGNATURE,
|
||
|
|
(PBYTE) NULL,
|
||
|
|
0
|
||
|
|
);
|
||
|
|
|
||
|
|
dwStatus=RegOpenKeyEx(
|
||
|
|
HKEY_LOCAL_MACHINE,
|
||
|
|
LSERVER_SERVER_CERTIFICATE_REGKEY,
|
||
|
|
0,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
&hKey
|
||
|
|
);
|
||
|
|
if(dwStatus == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Ignore error
|
||
|
|
RegDeleteValue(
|
||
|
|
hKey,
|
||
|
|
LSERVER_SIGNATURE_CERT_KEY
|
||
|
|
);
|
||
|
|
|
||
|
|
RegDeleteValue(
|
||
|
|
hKey,
|
||
|
|
LSERVER_EXCHANGE_CERT_KEY
|
||
|
|
);
|
||
|
|
|
||
|
|
RegDeleteValue(
|
||
|
|
hKey,
|
||
|
|
LSERVER_CLIENT_CERTIFICATE_ISSUER
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(hKey != NULL)
|
||
|
|
{
|
||
|
|
RegCloseKey(hKey);
|
||
|
|
}
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|