windows-nt/Source/XPSP1/NT/termsrv/license/hserver/prottest/prottest.c
2020-09-26 16:20:57 +08:00

790 lines
19 KiB
C

//+----------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996-1998
//
// File: prottest.c
//
// Contents: Test program for hydra licensing protocol
//
// History: 01-07-98 FredCh Created
//
//-----------------------------------------------------------------------------
#include <windows.h>
#include <stdio.h>
#include <time.h>
#include <limits.h>
#include <stdlib.h>
#include "license.h"
#include "hslice.h"
#include "cryptkey.h"
#include "hccontxt.h"
#include "licecert.h"
#include "lscsp.h"
#include "sysapi.h"
#include "prottest.h"
#define MSG_SIZE 2048
BOOL
LsCsp_UnpackServerCert(
LPBYTE pbCert,
DWORD dwCertLen,
PHydra_Server_Cert pServerCert );
///////////////////////////////////////////////////////////////////////////////
HANDLE g_hServerEvent = NULL, g_hClientEvent = NULL;
BYTE * pbServerMessage = NULL;
PBYTE ClientMessage;//[MSG_SIZE];
DWORD cbServerMessage = 0;
DWORD cbClientMessage;
LPBYTE g_pbPubKey = NULL;
DWORD g_cbPubKey = 0;
Hydra_Server_Cert g_ServerCertificate;
const BYTE g_abServerCertificate[184] = {
0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x01, 0x00, 0x00, 0x00, 0x06, 0x00, 0x5C, 0x00,
0x52, 0x53, 0x41, 0x31, 0x48, 0x00, 0x00, 0x00,
0x00, 0x02, 0x00, 0x00, 0x3F, 0x00, 0x00, 0x00,
0x01, 0x00, 0x01, 0x00, 0x83, 0x76, 0x5B, 0x09,
0x8F, 0xC1, 0x74, 0x12, 0x1B, 0xD3, 0x4E, 0x72,
0x72, 0x4D, 0xBE, 0xCE, 0x55, 0x1D, 0x29, 0x3D,
0x0E, 0xED, 0x28, 0x09, 0x50, 0x66, 0x32, 0xFA,
0x1D, 0xD2, 0xCC, 0x42, 0xDE, 0x5B, 0x4E, 0x3C,
0x35, 0xF6, 0x73, 0x5B, 0x0C, 0x0D, 0xB0, 0xA6,
0x4D, 0x76, 0xBA, 0xC0, 0x88, 0x5F, 0xC4, 0x67,
0x0B, 0xB8, 0xA3, 0x23, 0xA6, 0xC7, 0x79, 0xBD,
0x80, 0xD1, 0xA8, 0x75, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x48, 0x00,
0x19, 0x50, 0x2E, 0x2E, 0x82, 0xB4, 0xEB, 0xB3,
0x87, 0x85, 0xB9, 0x31, 0x4C, 0x29, 0x07, 0x05,
0xD7, 0x37, 0x99, 0x86, 0x15, 0x30, 0x56, 0xE4,
0x47, 0x7A, 0x2C, 0x2F, 0x4C, 0xBD, 0xF0, 0x37,
0xD3, 0x94, 0x01, 0xC8, 0x73, 0xEA, 0x5C, 0x2C,
0x3F, 0x60, 0x27, 0x1E, 0x5D, 0xA9, 0x54, 0x32,
0xDC, 0x49, 0xA4, 0x7E, 0x26, 0xAF, 0xEA, 0x07,
0xCA, 0x4E, 0xE9, 0x95, 0x8E, 0x66, 0xF0, 0x33,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
};
///////////////////////////////////////////////////////////////////////////////
void
_stdcall
HydraServerThread();
void
_stdcall
HydraClientThread();
BOOL
TsIssueLicenseExpirationWarning(
LPDWORD lpdwDays,
PTS_LICENSE_INFO pTsLicenseInfo );
BOOL
FileTimeToUnixTime(
LPFILETIME pft,
time_t * t );
VOID
DisplayLicenseMessage(
DWORD dwDaysLeft );
///////////////////////////////////////////////////////////////////////////////
void _cdecl main(int argc, char *argv[])
{
DWORD ServerThreadID, ClientThreadID;
HANDLE ThreadHandles[2];
DWORD WaitStatus = 0;
Binary_Blob CertBlob;
LICENSE_STATUS Status;
DWORD dwFlag = CERT_DATE_DONT_VALIDATE;
//
// Create the server and client events
//
ThreadHandles[0] = NULL;
ThreadHandles[1] = NULL;
g_hServerEvent = CreateEvent( NULL, FALSE, FALSE, NULL );
g_hClientEvent = CreateEvent( NULL, FALSE, FALSE, NULL );
if( ( NULL == g_hServerEvent ) || ( NULL == g_hClientEvent ) )
{
printf( "Cannot create events\n" );
return;
}
InitializeLicenseLib( TRUE );
memset( &CertBlob, 0, sizeof( CertBlob ) );
//
// get the X509 certificate
//
Status = GetServerCertificate( CERT_TYPE_X509, &CertBlob, KEY_EXCHANGE_ALG_RSA );
if( LICENSE_STATUS_OK == Status )
{
Status = VerifyCertChain(
CertBlob.pBlob,
CertBlob.wBlobLen,
NULL,
&g_cbPubKey,
&dwFlag );
if( LICENSE_STATUS_INSUFFICIENT_BUFFER == Status )
{
g_pbPubKey = LocalAlloc( LPTR, g_cbPubKey );
if( NULL == g_pbPubKey )
{
goto done;
}
Status = VerifyCertChain( CertBlob.pBlob,
CertBlob.wBlobLen,
g_pbPubKey,
&g_cbPubKey,
&dwFlag );
}
if( LICENSE_STATUS_OK != Status )
{
printf( "Cannot verify X509 certificate chain: %x\n", Status );
goto done;
}
}
//
// unpack the hardcoded certificate
//
if( !LsCsp_UnpackServerCert( ( LPBYTE )g_abServerCertificate, sizeof( g_abServerCertificate ),
&g_ServerCertificate ) )
{
printf( "cannot unpack server certificate\n" );
}
//
// Create the server and client threads
//
ThreadHandles[0] = CreateThread( NULL,
0,
( LPTHREAD_START_ROUTINE )HydraServerThread,
NULL,
0,
&ServerThreadID );
ThreadHandles[1] = CreateThread( NULL,
0,
( LPTHREAD_START_ROUTINE )HydraClientThread,
NULL,
0,
&ClientThreadID );
//
// wait for the server and client thread to die
//
WaitStatus = WaitForMultipleObjects( 2, ThreadHandles, TRUE, INFINITE );
//
// close the event handles
//
done:
if( g_hServerEvent )
{
CloseHandle( g_hServerEvent );
}
if( g_hClientEvent )
{
CloseHandle( g_hClientEvent );
}
if( ThreadHandles[0] )
{
CloseHandle( ThreadHandles[0] );
}
if( ThreadHandles[1] )
{
CloseHandle( ThreadHandles[1] );
}
ShutdownLicenseLib();
if( pbServerMessage )
{
LocalFree( pbServerMessage );
}
if( CertBlob.pBlob )
{
LocalFree( CertBlob.pBlob );
}
if( g_pbPubKey )
{
LocalFree( g_pbPubKey );
}
return;
}
#define HYDRA_40_LICENSING_PROTOCOL_FLAG 0
#define HYDRA_40_LICENSING_PROTOCOL_VERSION LICENSE_PROTOCOL_VERSION_1_0 | PREAMBLE_VERSION_2_0
///////////////////////////////////////////////////////////////////////////////
void
_stdcall
HydraServerThread()
{
LICENSE_STATUS Status;
HANDLE hContext = NULL;
LICENSE_CAPABILITIES LicenseCap;
TS_LICENSE_INFO
TsLicenseInfo;
Status = CreateLicenseContext( &hContext, LICENSE_CONTEXT_PER_SEAT );
if( LICENSE_STATUS_OK != Status )
{
printf( "HydraServerThread(): error creating license context\n" );
return;
}
memset( &LicenseCap, 0, sizeof( LicenseCap ) );
LicenseCap.KeyExchangeAlg = KEY_EXCHANGE_ALG_RSA;
#ifdef HYDRA_40_TEST
//
// talking to a Hydra 4.0 client
//
LicenseCap.ProtocolVer = HYDRA_40_LICENSING_PROTOCOL_VERSION;
LicenseCap.fAuthenticateServer = TRUE;
Status = InitializeLicenseContext(
hContext,
0,
&LicenseCap );
#else
#ifdef HYDRA_50_NO_SERVER_AUTHEN_X509
//
// talking to a Hydra 5.0 client and don't send certificate
//
LicenseCap.ProtocolVer = LICENSE_HIGHEST_PROTOCOL_VERSION;
LicenseCap.fAuthenticateServer = FALSE;
LicenseCap.CertType = CERT_TYPE_X509;
Status = InitializeLicenseContext(
hContext,
0,
&LicenseCap );
#else
#ifdef HYDRA_50_NO_SERVER_AUTHEN_PROPRIETORY
//
// talking to a Hydra 5.0 client and don't send certificate
//
LicenseCap.ProtocolVer = LICENSE_HIGHEST_PROTOCOL_VERSION;
LicenseCap.fAuthenticateServer = FALSE;
LicenseCap.CertType = CERT_TYPE_PROPRIETORY;
Status = InitializeLicenseContext(
hContext,
0,
&LicenseCap );
#else
//
// talking to a Hydra 5.0 client and also send certificate
//
LicenseCap.ProtocolVer = LICENSE_HIGHEST_PROTOCOL_VERSION;
LicenseCap.fAuthenticateServer = TRUE;
Status = InitializeLicenseContext(
hContext,
0,
&LicenseCap );
#endif // HYDRA_50_NO_SERVER_AUTHEN_X509
#endif // HYDRA_50_NO_SERVER_AUTHEN_PROPRIETORY
#endif // HYDRA_40_TEST
if( LICENSE_STATUS_OK != Status )
{
printf( "HydraServerThread(): cannot initialize license context: %x\n", Status );
goto done;
}
Status = AcceptLicenseContext( hContext,
0,
NULL,
&cbServerMessage,
&pbServerMessage );
while( LICENSE_STATUS_CONTINUE == Status )
{
SetEvent( g_hClientEvent );
WaitForSingleObject( g_hServerEvent, INFINITE );
if( pbServerMessage )
{
LocalFree( pbServerMessage );
pbServerMessage = NULL;
cbServerMessage = 0;
}
Status = AcceptLicenseContext( hContext,
cbClientMessage,
ClientMessage,
&cbServerMessage,
&pbServerMessage );
}
done:
if( hContext )
{
DWORD
dwDaysLeft = 0;
memset( &TsLicenseInfo, 0, sizeof( TsLicenseInfo ) );
Status = QueryLicenseInfo( hContext, &TsLicenseInfo );
if( LICENSE_STATUS_OK != Status )
{
printf( "HydraSeverThread: cannot query license info: %x\n", Status );
}
if( TsIssueLicenseExpirationWarning( &dwDaysLeft, &TsLicenseInfo ) )
{
DisplayLicenseMessage( dwDaysLeft );
}
FreeLicenseInfo( &TsLicenseInfo );
DeleteLicenseContext( hContext );
}
if( ( LICENSE_STATUS_ISSUED_LICENSE == Status ) ||
( LICENSE_STATUS_OK == Status ) )
{
//
// issueing a license or the license has been successfully validated
//
printf( "HydraServerThread: Protocol completed successfully\n" );
SetEvent( g_hClientEvent );
}
else if( LICENSE_STATUS_SEND_ERROR == Status )
{
printf( "HydraServerThread: sending error to client\n" );
SetEvent( g_hClientEvent );
}
else
{
printf( "HydraServerThread: protocol error: aborted\n" );
}
return;
}
///////////////////////////////////////////////////////////////////////////////
void
_stdcall
HydraClientThread()
{
LICENSE_STATUS Status;
HANDLE hContext;
#ifdef HYDRA_50_NO_SERVER_AUTHEN_X509
//
// no server authentication required. Use the public key in the
// X509 certificate
//
Status = LicenseInitializeContext(
&hContext,
LICENSE_CONTEXT_NO_SERVER_AUTHENTICATION );
if( LICENSE_STATUS_OK != Status )
{
printf( "Cannot initialize client context: %x\n", Status );
return;
}
//
// set the public key
//
Status = LicenseSetPublicKey(
hContext,
g_cbPubKey,
g_pbPubKey );
if( LICENSE_STATUS_OK != Status )
{
printf( "Cannot set public key: %x\n", Status );
goto done;
}
#else
#ifdef HYDRA_50_NO_SERVER_AUTHEN_PROPRIETORY
//
// no server authentication required. Use the proprietory certificate
//
Status = LicenseInitializeContext(
&hContext,
LICENSE_CONTEXT_NO_SERVER_AUTHENTICATION );
if( LICENSE_STATUS_OK != Status )
{
printf( "Cannot initialize client context: %x\n", Status );
return;
}
//
// set the proprietory certificate
//
Status = LicenseSetCertificate(
hContext,
&g_ServerCertificate );
if( LICENSE_STATUS_OK != Status )
{
printf( "cannot set certificate: %x\n", Status );
goto done;
}
#else
//
// hydra 4.0/5.0 licensing protocol.with certificate validation
//
Status = LicenseInitializeContext(
&hContext,
0 );
if( LICENSE_STATUS_OK != Status )
{
printf( "Cannot initialize client context: %x\n", Status );
return;
}
#endif
#endif
if( NULL == hContext )
{
printf( "HydraClientThread(): error creating license context\n" );
return;
}
if( LICENSE_STATUS_OK != Status )
{
printf( "HydraClientThread(): error setting certificate: %x\n", Status );
LicenseDeleteContext( hContext );
return;
}
WaitForSingleObject( g_hClientEvent, INFINITE );
cbClientMessage = 0;//MSG_SIZE;
if( LICENSE_STATUS_CONTINUE != (Status = LicenseAcceptContext( hContext,
0,
pbServerMessage,
cbServerMessage,
NULL,//ClientMessage,
&cbClientMessage )) )
{
printf("Error handling Server request\n");
return;
}
if( NULL == (ClientMessage = (PBYTE)GlobalAlloc(GMEM_FIXED|GMEM_ZEROINIT, cbClientMessage)) )
{
printf("Error allocating memory\n");
return;
}
//memset(ClientMessage, 0x00, cbClientMessage);
Status = LicenseAcceptContext(hContext,
0,
pbServerMessage,
cbServerMessage,
ClientMessage,
&cbClientMessage);
while( LICENSE_STATUS_CONTINUE == Status )
{
SetEvent( g_hServerEvent );
WaitForSingleObject( g_hClientEvent, INFINITE );
cbClientMessage = 0;
if(ClientMessage)
{
GlobalFree((HGLOBAL)ClientMessage);
ClientMessage = NULL;
}
Status = LicenseAcceptContext( hContext,
0,
pbServerMessage,
cbServerMessage,
NULL,//ClientMessage,
&cbClientMessage);
if( ( Status == LICENSE_STATUS_OK ) || ( Status != LICENSE_STATUS_CONTINUE ) )
{
break;
}
if( NULL == (ClientMessage = (PBYTE)GlobalAlloc(GMEM_FIXED|GMEM_ZEROINIT, cbClientMessage)) )
{
printf("Error allocating memory\n");
break;
}
memset(ClientMessage, 0x00, cbClientMessage);
Status = LicenseAcceptContext( hContext,
0,
pbServerMessage,
cbServerMessage,
ClientMessage,
&cbClientMessage );
}
done:
LicenseDeleteContext( hContext );
if( LICENSE_STATUS_OK == Status )
{
printf( "HydraClientThread: license protocol completed successfully\n" );
}
else
{
printf( "HydraClientThread: license protocol failed: 0x%x\n", Status );
}
if( ClientMessage )
{
GlobalFree( ( HGLOBAL )ClientMessage );
}
return;
}
#define SECONDS_IN_A_DAY 86400 // number of seconds in a day
#define ISSUE_LICENSE_WARNING_PERIOD 150 // days to expiration when warning should be issued.
///////////////////////////////////////////////////////////////////////////////
BOOL
TsIssueLicenseExpirationWarning(
LPDWORD lpdwDays,
PTS_LICENSE_INFO pTsLicenseInfo )
{
time_t
Expiration,
CurrentTime;
DWORD
dwDaysLeft;
if( NULL == pTsLicenseInfo )
{
return( FALSE );
}
if( FALSE == pTsLicenseInfo->fTempLicense )
{
return( FALSE );
}
//
// The client license is temporary, figure out how long more
// the license is valid
//
if( FALSE == FileTimeToUnixTime( &pTsLicenseInfo->NotAfter, &Expiration ) )
{
return( FALSE );
}
time( &CurrentTime );
if( CurrentTime >= Expiration )
{
//
// license already expired
//
*lpdwDays = 0xFFFFFFFF;
return( TRUE );
}
dwDaysLeft = ( Expiration - CurrentTime ) / SECONDS_IN_A_DAY;
printf( "Number of days left for temporary license expiration: %d\n", dwDaysLeft );
if( ISSUE_LICENSE_WARNING_PERIOD >= dwDaysLeft )
{
*lpdwDays = dwDaysLeft;
return( TRUE );
}
return( FALSE );
}
///////////////////////////////////////////////////////////////////////////////
BOOL
FileTimeToUnixTime(
LPFILETIME pft,
time_t * t )
{
SYSTEMTIME sysTime;
struct tm gmTime;
if( FileTimeToSystemTime( pft, &sysTime ) == FALSE )
{
return( FALSE );
}
if( sysTime.wYear >= 2038 )
{
*t = INT_MAX;
}
else
{
//
// Unix time support up to 2038/1/18
// restrict any expiration data
//
memset( &gmTime, 0, sizeof( gmTime ) );
gmTime.tm_sec = sysTime.wSecond;
gmTime.tm_min = sysTime.wMinute;
gmTime.tm_hour = sysTime.wHour;
gmTime.tm_year = sysTime.wYear - 1900;
gmTime.tm_mon = sysTime.wMonth - 1;
gmTime.tm_mday = sysTime.wDay;
*t = mktime( &gmTime );
}
return( *t != ( time_t )-1 );
}
///////////////////////////////////////////////////////////////////////////////
VOID
DisplayLicenseMessage(
DWORD dwDaysLeft )
{
TCHAR
szMsgCaption[512],
szMsgTitle[256];
HMODULE
hModule = GetModuleHandle( NULL );
LPDWORD
lpdw;
TCHAR
tszDays[ 10 ];
if( 0xFFFFFFFF == dwDaysLeft )
{
printf( "Temporary License has expired!\n" );
LoadString( hModule, STR_TEMP_LICENSE_EXPIRED_MSG, szMsgCaption, 512 );
}
else
{
printf( "%d days left before temporary license expires\n", dwDaysLeft );
//
// convert the number of days left to UNICODE character
//
_ultow( dwDaysLeft, tszDays, 10 );
lpdw = ( LPDWORD )&tszDays[0];
//
// prepare license about to expire message
//
LoadString( hModule, STR_TEMP_LICENSE_EXPIRATION_MSG, szMsgCaption, 512 );
FormatMessage(
FORMAT_MESSAGE_FROM_STRING | FORMAT_MESSAGE_ARGUMENT_ARRAY,
szMsgCaption,
0,
MAKELANGID( LANG_NEUTRAL, SUBLANG_DEFAULT ),
( LPTSTR )szMsgCaption,
sizeof( szMsgCaption ),
( va_list * )&lpdw );
}
//
// prepare message title
//
LoadString( hModule, STR_TEMP_LICENSE_MSG_TITLE, szMsgTitle, 256 );
MessageBox( NULL, szMsgCaption, szMsgTitle, MB_OK );
return;
}