windows-nt/Source/XPSP1/NT/termsrv/license/tlserver/tlsapi/tlsapip.cpp

864 lines
20 KiB
C++
Raw Permalink Normal View History

2020-09-26 03:20:57 -05:00
//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996-1998
//
// File: tlsapip.cpp
//
// Contents: Private API
//
// History: 09-09-97 HueiWang Created
//
//---------------------------------------------------------------------------
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <tchar.h>
#include <rpc.h>
#include "lscommon.h"
#include <wincrypt.h>
#include "tlsrpc.h"
#include "tlsapi.h"
#include "tlsapip.h"
//----------------------------------------------------------------------------
BOOL
TLSIsLicenseEnforceEnable()
/*++
--*/
{
#if ENFORCE_LICENSING
return TRUE;
#else
return FALSE;
#endif
}
//----------------------------------------------------------------------------
BOOL
TLSIsBetaNTServer()
/*++
Abstract:
Detemine if base NT is a beta or RTM version.
Parameter:
None.
Return:
TRUE/FALSE
--*/
{
BOOL bBetaNt = FALSE;
DWORD dwStatus;
DWORD cbData;
DWORD cbType;
HKEY hKey = NULL;
__try {
LARGE_INTEGER Time = USER_SHARED_DATA->SystemExpirationDate;
if(Time.QuadPart)
{
bBetaNt = TRUE;
// check our special registry key - force
// issuing a RTM license.
dwStatus = RegOpenKeyEx(
HKEY_LOCAL_MACHINE,
L"SOFTWARE\\Microsoft\\TermServLicensing",
0,
KEY_ALL_ACCESS,
&hKey
);
if(dwStatus == ERROR_SUCCESS)
{
dwStatus = RegQueryValueEx(
hKey,
_TEXT("RunAsRTM"),
NULL,
&cbType,
NULL,
&cbData
);
// for testing, force it to run as RTM version.
// key must exist and must be DWORD type
if(dwStatus == ERROR_SUCCESS && cbType == REG_DWORD)
{
bBetaNt = FALSE;
}
}
}
}
__except(EXCEPTION_EXECUTE_HANDLER) {
ASSERT(FALSE);
}
if(hKey != NULL)
{
RegCloseKey(hKey);
}
return bBetaNt;
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSAllocateInternetLicenseEx(
IN TLS_HANDLE hHandle,
IN const CHALLENGE_CONTEXT ChallengeContext,
IN const LICENSEREQUEST* pRequest,
IN LPTSTR pszMachineName,
IN LPTSTR pszUserName,
IN const DWORD cbChallengeResponse,
IN const PBYTE pbChallengeResponse,
OUT PTLSInternetLicense pInternetLicense,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
TLSLICENSEREQUEST rpcRequest;
RequestToTlsRequest( pRequest, &rpcRequest );
return TLSRpcAllocateInternetLicenseEx(
hHandle,
ChallengeContext,
&rpcRequest,
pszMachineName,
pszUserName,
cbChallengeResponse,
pbChallengeResponse,
pInternetLicense,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSReturnInternetLicenseEx(
IN TLS_HANDLE hHandle,
IN const LICENSEREQUEST* pRequest,
IN const ULARGE_INTEGER* pulSerialNumber,
IN const DWORD dwQuantity,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
TLSLICENSEREQUEST rpcRequest;
RequestToTlsRequest( pRequest, &rpcRequest );
return TLSRpcReturnInternetLicenseEx(
hHandle,
&rpcRequest,
pulSerialNumber,
dwQuantity,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSRegisterLicenseKeyPack(
TLS_HANDLE hHandle,
LPBYTE pbCHCertBlob,
DWORD cbCHCertBlobSize,
LPBYTE pbRootCertBlob,
DWORD cbRootCertBlob,
LPBYTE lpKeyPackBlob,
DWORD dwKeyPackBlobLen,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcRegisterLicenseKeyPack(
hHandle,
pbCHCertBlob,
cbCHCertBlobSize,
pbRootCertBlob,
cbRootCertBlob,
lpKeyPackBlob,
dwKeyPackBlobLen,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSTelephoneRegisterLKP(
IN TLS_HANDLE hHandle,
IN DWORD cbData,
IN PBYTE pbData,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
return TLSRpcTelephoneRegisterLKP(
hHandle,
cbData,
pbData,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
RequestToTlsRequest(
const LICENSEREQUEST* lpRequest,
TLSLICENSEREQUEST* lpRpcRequest
)
/*++
++*/
{
if(lpRequest == NULL || lpRpcRequest == NULL || lpRequest->pProductInfo == NULL)
return ERROR_INVALID_PARAMETER;
//
// NOTE : No memory allocation, DO NOT FREE ...
//
lpRpcRequest->cbEncryptedHwid = lpRequest->cbEncryptedHwid;
lpRpcRequest->pbEncryptedHwid = lpRequest->pbEncryptedHwid;
lpRpcRequest->dwLanguageID = lpRequest->dwLanguageID;
lpRpcRequest->dwPlatformID = lpRequest->dwPlatformID;
lpRpcRequest->ProductInfo.dwVersion = lpRequest->pProductInfo->dwVersion;
lpRpcRequest->ProductInfo.cbCompanyName = lpRequest->pProductInfo->cbCompanyName;
lpRpcRequest->ProductInfo.pbCompanyName = lpRequest->pProductInfo->pbCompanyName;
lpRpcRequest->ProductInfo.cbProductID = lpRequest->pProductInfo->cbProductID;
lpRpcRequest->ProductInfo.pbProductID = lpRequest->pProductInfo->pbProductID;
return ERROR_SUCCESS;
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSReturnLicense(
TLS_HANDLE hHandle,
DWORD dwKeyPackId,
DWORD dwLicenseId,
DWORD dwRetrunReason,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcReturnLicense(
hHandle,
dwKeyPackId,
dwLicenseId,
dwRetrunReason,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSGetLSPKCS10CertRequest(
TLS_HANDLE hHandle,
DWORD dwCertType,
PDWORD pcbData,
PBYTE* ppbData,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcGetLSPKCS10CertRequest(
hHandle,
dwCertType,
pcbData,
ppbData,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSRequestTermServCert(
TLS_HANDLE hHandle,
LPLSHydraCertRequest pRequest,
PDWORD cbChallengeData,
PBYTE* pbChallengeData,
PDWORD pdwErrCode
)
/*++
++*/
{
TLSHYDRACERTREQUEST CertRequest;
CertRequest.dwHydraVersion = pRequest->dwHydraVersion;
CertRequest.pbEncryptedHwid = pRequest->pbEncryptedHwid;
CertRequest.cbEncryptedHwid = pRequest->cbEncryptedHwid;
CertRequest.szSubjectRdn = pRequest->szSubjectRdn;
CertRequest.pSubjectPublicKeyInfo = (TLSCERT_PUBLIC_KEY_INFO *)pRequest->SubjectPublicKeyInfo;
CertRequest.dwNumCertExtension = pRequest->dwNumCertExtension;
CertRequest.pCertExtensions = (TLSCERT_EXTENSION *)pRequest->pCertExtensions;
return TLSRpcRequestTermServCert(
hHandle,
&CertRequest,
cbChallengeData,
pbChallengeData,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSRetrieveTermServCert(
TLS_HANDLE hHandle,
DWORD cbResponseData,
PBYTE pbResponseData,
PDWORD pcbCert,
PBYTE* ppbCert,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcRetrieveTermServCert(
hHandle,
cbResponseData,
pbResponseData,
pcbCert,
ppbCert,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSInstallCertificate(
TLS_HANDLE hHandle,
DWORD dwCertType,
DWORD dwCertLevel,
DWORD cbSingnatureCert,
PBYTE pbSingnatureCert,
DWORD cbExchangeCert,
PBYTE pbExchangeCert,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcInstallCertificate(
hHandle,
dwCertType,
dwCertLevel,
cbSingnatureCert,
pbSingnatureCert,
cbExchangeCert,
pbExchangeCert,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSGetServerCertificate(
TLS_HANDLE hHandle,
BOOL bSignCert,
LPBYTE *ppbCertBlob,
LPDWORD lpdwCertBlobLen,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcGetServerCertificate(
hHandle,
bSignCert,
ppbCertBlob,
lpdwCertBlobLen,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSKeyPackAdd(
TLS_HANDLE hHandle,
LPLSKeyPack lpKeypack,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcKeyPackAdd(
hHandle,
lpKeypack,
pdwErrCode
);
}
//----------------------------------------------------------------------------
DWORD WINAPI
TLSKeyPackSetStatus(
TLS_HANDLE hHandle,
DWORD dwSetParm,
LPLSKeyPack lpKeyPack,
PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcKeyPackSetStatus(
hHandle,
dwSetParm,
lpKeyPack,
pdwErrCode
);
}
//-----------------------------------------------------------------
DWORD WINAPI
TLSAnnounceServer(
IN TLS_HANDLE hHandle,
IN DWORD dwType,
IN FILETIME* pftTime,
IN LPTSTR pszSetupId,
IN LPTSTR pszDomainName,
IN LPTSTR pszMachineName,
OUT PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcAnnounceServer(
hHandle,
dwType,
pftTime,
pszSetupId,
pszDomainName,
pszMachineName,
pdwErrCode
);
}
//-----------------------------------------------------------------
DWORD WINAPI
TLSLookupServer(
IN TLS_HANDLE hHandle,
IN LPTSTR pszLookupSetupId,
OUT LPTSTR pszLsSetupId,
IN OUT PDWORD pcbSetupId,
OUT LPTSTR pszDomainName,
IN OUT PDWORD pcbDomainName,
IN LPTSTR pszLsName,
IN OUT PDWORD pcbMachineName,
OUT PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcLookupServer(
hHandle,
pszLookupSetupId,
pszLsSetupId,
pcbSetupId,
pszDomainName,
pcbDomainName,
pszLsName,
pcbMachineName,
pdwErrCode
);
}
//-------------------------------------------------------
DWORD WINAPI
TLSAnnounceLicensePack(
IN TLS_HANDLE hHandle,
IN PTLSReplRecord pReplRecord,
OUT PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcAnnounceLicensePack(
hHandle,
pReplRecord,
pdwErrCode
);
}
//-------------------------------------------------------
DWORD WINAPI
TLSReturnLicensedProduct(
IN TLS_HANDLE hHandle,
IN PTLSLicenseToBeReturn pClientLicense,
OUT PDWORD pdwErrCode
)
/*++
++*/
{
return TLSRpcReturnLicensedProduct(
hHandle,
pClientLicense,
pdwErrCode
);
}
//-------------------------------------------------------
DWORD WINAPI
TLSChallengeServer(
IN TLS_HANDLE hHandle,
IN DWORD dwClientType,
IN PTLSCHALLENGEDATA pClientChallenge,
OUT PTLSCHALLENGERESPONSEDATA* ppServerResponse,
OUT PTLSCHALLENGEDATA* ppServerChallenge,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
return TLSRpcChallengeServer(
hHandle,
dwClientType,
pClientChallenge,
ppServerResponse,
ppServerChallenge,
pdwErrCode
);
}
//-------------------------------------------------------
DWORD WINAPI
TLSResponseServerChallenge(
IN TLS_HANDLE hHandle,
IN PTLSCHALLENGERESPONSEDATA pClientResponse,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
return TLSRpcResponseServerChallenge(
hHandle,
pClientResponse,
pdwErrCode
);
}
//------------------------------------------------------
DWORD WINAPI
TLSGetTlsPrivateData(
IN TLS_HANDLE hHandle,
IN DWORD dwGetDataType,
IN PTLSPrivateDataUnion pGetParm,
OUT PDWORD pdwRetDataType,
OUT PTLSPrivateDataUnion* ppRetData,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
return TLSRpcGetTlsPrivateData(
hHandle,
dwGetDataType,
pGetParm,
pdwRetDataType,
ppRetData,
pdwErrCode
);
}
//------------------------------------------------------
DWORD WINAPI
TLSTriggerReGenKey(
IN TLS_HANDLE hHandle,
IN BOOL bKeepSPK,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
return TLSRpcTriggerReGenKey(
hHandle,
bKeepSPK,
pdwErrCode
);
}
//------------------------------------------------------
DWORD
GetPrivateBinaryDataFromServer(
TLS_HANDLE hHandle,
DWORD dwType,
PDWORD pcbData,
PBYTE* ppbData,
PDWORD pdwErrCode
)
/*++
--*/
{
TLSPrivateDataUnion SearchParm;
PTLSPrivateDataUnion pPrivateData = NULL;
DWORD dwRetType;
DWORD dwStatus;
memset(
&SearchParm,
0,
sizeof(TLSPrivateDataUnion)
);
dwStatus = TLSRpcGetTlsPrivateData(
hHandle,
dwType,
&SearchParm,
&dwRetType,
&pPrivateData,
pdwErrCode
);
if(dwStatus != RPC_S_OK || *pdwErrCode != ERROR_SUCCESS)
{
goto cleanup;
}
if(dwRetType != dwType)
{
//
// License Server error
//
*pdwErrCode = LSERVER_E_INVALID_RETURN;
goto cleanup;
}
//
// Copy over unique ID
//
*ppbData = (PBYTE)MIDL_user_allocate(pPrivateData->BinaryData.cbData);
if(*ppbData != NULL)
{
memset(
*ppbData,
0,
pPrivateData->BinaryData.cbData
);
*pcbData = pPrivateData->BinaryData.cbData;
memcpy(
*ppbData,
pPrivateData->BinaryData.pbData,
pPrivateData->BinaryData.cbData
);
}
else
{
*pdwErrCode = LSERVER_E_OUTOFMEMORY;
}
cleanup:
if(pPrivateData != NULL)
{
midl_user_free(pPrivateData);
}
return dwStatus;
}
//------------------------------------------------------
DWORD WINAPI
TLSGetServerPID(
TLS_HANDLE hHandle,
PDWORD pcbData,
PBYTE* ppbData,
PDWORD pdwErrCode
)
/*++
--*/
{
return GetPrivateBinaryDataFromServer(
hHandle,
TLS_PRIVATEDATA_PID,
pcbData,
ppbData,
pdwErrCode
);
}
//------------------------------------------------------
DWORD WINAPI
TLSGetServerSPK(
TLS_HANDLE hHandle,
PDWORD pcbData,
PBYTE* ppbData,
PDWORD pdwErrCode
)
/*++
--*/
{
TLSPrivateDataUnion SearchParm;
PTLSPrivateDataUnion pPrivateData = NULL;
DWORD dwRetType;
DWORD dwStatus;
memset(
&SearchParm,
0,
sizeof(TLSPrivateDataUnion)
);
dwStatus = TLSRpcGetTlsPrivateData(
hHandle,
TLS_PRIVATEDATA_SPK,
&SearchParm,
&dwRetType,
&pPrivateData,
pdwErrCode
);
if(dwStatus != RPC_S_OK || *pdwErrCode != ERROR_SUCCESS)
{
goto cleanup;
}
if(dwRetType != TLS_PRIVATEDATA_SPK)
{
//
// License Server error
//
*pdwErrCode = LSERVER_E_INVALID_RETURN;
goto cleanup;
}
//
// Copy over Server's SPK.
// Server never return CH's cert extension.
//
*ppbData = (PBYTE)MIDL_user_allocate(pPrivateData->SPK.cbSPK);
if(*ppbData != NULL)
{
memset(
*ppbData,
0,
pPrivateData->SPK.cbSPK
);
*pcbData = pPrivateData->SPK.cbSPK;
memcpy(
*ppbData,
pPrivateData->SPK.pbSPK,
pPrivateData->SPK.cbSPK
);
}
else
{
*pdwErrCode = LSERVER_E_OUTOFMEMORY;
}
cleanup:
if(pPrivateData != NULL)
{
midl_user_free(pPrivateData);
}
return dwStatus;
}
//-----------------------------------------------------------
DWORD WINAPI
TLSDepositeServerSPK(
IN TLS_HANDLE hHandle,
IN DWORD cbSPK,
IN PBYTE pbSPK,
IN PCERT_EXTENSIONS pCertExtensions,
OUT PDWORD pdwErrCode
)
/*++
--*/
{
TLSPrivateDataUnion SetData;
DWORD dwStatus;
SetData.SPK.cbSPK = cbSPK;
SetData.SPK.pbSPK = pbSPK;
SetData.SPK.pCertExtensions = pCertExtensions;
return TLSRpcSetTlsPrivateData(
hHandle,
TLS_PRIVATEDATA_SPK,
&SetData,
pdwErrCode
);
}