#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include "p6sax2contenthandler.h"
#include "cconsolestream.h"
static p6ISafeString *m_pStr = NULL;
static p6IRandom *m_pRandom = NULL;
static p6ICryptoKey *m_pSignKey = NULL;
static p6ISymmetricCrypto *m_pCrypto = NULL;
static P6WCHAR* m_pHostName = NULL;
static p6IKeystore *m_pKeystore = NULL;
static p6IKeystoreInit *m_pStoreInit = NULL;
static P6ERR getSafeString( p6ISafeString **ppStr )
{
printf("getSafeString [ OK ]\n");
else printf("getSafeString [ FAILED ] %x\n",err);
return err;
}
static P6ERR getRNG( p6IRandom **ppRandom )
{
p6IEntropySource *pSource = NULL;
p6IRandomInit *pInit = NULL;
{
if (
P6SUCCEEDED( err = pSource->lpVtbl->initialize( pSource, P6ENTROPY_HIGH )))
{
{
if (
P6SUCCEEDED( err = pInit->lpVtbl->initialize( pInit, P6RAND_NOFLAGS, pSource ))) {
}
}
}
}
if (NULL != pSource) pSource->lpVtbl->release( pSource );
if (NULL != pInit ) pInit->lpVtbl->release( pInit );
return err;
}
P6ERR getIGenKeys( p6IGenKeys **ppGenKeys )
{
p6IGenKeys *pTempKeys = NULL;
*ppGenKeys = NULL;
{
if (
P6FAILED( err = pTempKeys->lpVtbl->initialize( pTempKeys, P6GENKEY_NOFLAGS, m_pRandom ))) {
pTempKeys->lpVtbl->release( pTempKeys );
}
else {
*ppGenKeys = pTempKeys;
}
}
return err;
}
P6ERR createPKCS8Key( P6BSTR keyBuffer,
P6SIZE keySize, p6ICryptoKey **ppNewKey )
{
p6ICryptoKeyInit *pKeyInit = NULL;
{
if (
P6SUCCEEDED( err = pKeyInit->lpVtbl->initialize( pKeyInit, P6CKF_NONE, m_pRandom )))
{
if (
P6SUCCEEDED( err = pKeyInit->lpVtbl->loadPKCS8Key( pKeyInit, keyBuffer.pString, (
P6UINT32)keyBuffer.length, (
P6UINT32)keySize ))) {
err = pKeyInit->lpVtbl->queryInterface( pKeyInit, &IID_p6ICryptoKey, (
P6VOID**)ppNewKey );
}
}
}
return err;
}
P6ERR keystoreAddRootCertFromFile( p6IKeystore *pKeystore,
const P6WCHAR *pszCertificateFile )
{
p6IKeystoreSSL *pSSLHelp = NULL;
if (NULL == pKeystore || NULL == pszCertificateFile) return eInvalidArg;
if (
P6FAILED( err = m_pStr->lpVtbl->wstrlcat( m_pStr, certPath,
P6CNTOF(certPath), pszCertificateFile, NULL )))
return err;
if (
P6FAILED( err = pKeystore->lpVtbl->queryInterface( pKeystore, &IID_p6IKeystoreSSL, (
P6VOID**)&pSSLHelp )))
return err;
err = pSSLHelp->lpVtbl->importTrustedRootCertFromPEMFile( pSSLHelp, certPath, NULL );
pSSLHelp->lpVtbl->release( pSSLHelp );
return err;
}
P6ERR keystoreAddClientCertFromFile( p6IKeystore* pKeystore,
const P6WCHAR* pszHostname,
const P6WCHAR* pszPrivateKeyFile,
const P6WCHAR* pszCertificateFile )
{
p6IKeystoreSSL *pSSLHelp = NULL;
if (NULL == pKeystore || NULL == pszHostname || NULL == pszPrivateKeyFile || NULL == pszCertificateFile ) return eInvalidArg;
if (
P6FAILED( err = m_pStr->lpVtbl->wstrlcat( m_pStr, keyPath,
P6CNTOF(keyPath), pszPrivateKeyFile, NULL )))
return err;
if (
P6FAILED( err = m_pStr->lpVtbl->wstrlcat( m_pStr, certPath,
P6CNTOF(certPath), pszCertificateFile, NULL )))
return err;
if (
P6FAILED( err = pKeystore->lpVtbl->queryInterface( pKeystore, &IID_p6IKeystoreSSL, (
P6VOID**)&pSSLHelp )))
return err;
err = pSSLHelp->lpVtbl->importCredentialsPEM( pSSLHelp,
P6TRUE, pszHostname, keyPath, certPath, NULL, NULL );
pSSLHelp->lpVtbl->release( pSSLHelp );
return err;
}
P6ERR createKeystore(
const P6WCHAR* pKeystoreName,
const P6WCHAR* rootPEM,
const P6WCHAR* certPEM,
const P6WCHAR* privPEM, p6IKeystoreInit** ppInit , p6IKeystore** ppKeystore )
{
*ppInit = NULL;
*ppKeystore = NULL;
err = (*ppInit)->lpVtbl->queryInterface( (*ppInit), &IID_p6IKeystore, (
P6VOID**)ppKeystore );
if (
P6FAILED( err = (*ppInit)->lpVtbl->initialize( (*ppInit), P6KEYSTORE_NOFLAGS, m_pCrypto,
SH_SHA256, m_pSignKey )))
{
if (NULL != (*ppKeystore)) (*ppKeystore)->lpVtbl->release( (*ppKeystore) );
(*ppKeystore) = NULL;
(*ppInit)->lpVtbl->release( (*ppInit) );
(*ppInit) = NULL;
return err;
}
if (
P6FAILED( err = (*ppInit)->lpVtbl->openSigned( (*ppInit), NULL, pKeystoreName )))
{
if (NULL != (*ppKeystore)) (*ppKeystore)->lpVtbl->release( (*ppKeystore) );
(*ppKeystore) = NULL;
(*ppInit)->lpVtbl->release( (*ppInit) );
(*ppInit) = NULL;
return err;
}
if (
P6FAILED( err = keystoreAddRootCertFromFile( (*ppKeystore), rootPEM )))
return err;
if (
P6FAILED( err = keystoreAddClientCertFromFile( (*ppKeystore), m_pHostName, privPEM, certPEM )))
return err;
return eOk;
}
{
if (NULL != m_pStr ) m_pStr->lpVtbl->release( m_pStr );
if (NULL != m_pRandom ) m_pRandom->lpVtbl->release( m_pRandom );
if (NULL != m_pSignKey ) m_pSignKey->lpVtbl->release( m_pSignKey );
if (NULL != m_pCrypto ) m_pCrypto->lpVtbl->release( m_pCrypto );
if (NULL != m_pStoreInit) m_pStoreInit->lpVtbl->release( m_pStoreInit );
if (NULL != m_pKeystore ) m_pKeystore->lpVtbl->release( m_pKeystore );
}
{
p6IGenKeys* pGenKey = NULL;
p6ICryptoKey* pKey = NULL;
p6IDir* pDir = NULL;
if (
P6FAILED( err = getSafeString( &m_pStr )))
return err;
if (
P6FAILED( err = getRNG( &m_pRandom )))
return err;
m_port = 5696;
m_compatMask = 0;
if (
P6FAILED( err = m_pStr->lpVtbl->wstrdup( m_pStr, (
const P6WCHAR*)
P6TEXT(
"fqdn.com"), &m_pHostName )))
return err;
if (
P6FAILED( err = getIGenKeys( &pGenKey )))
return err;
if (
P6SUCCEEDED( err = pGenKey->lpVtbl->genSymmetricKey( pGenKey, &m_pSignKey, 256,
P6FALSE ))) {
err = pGenKey->lpVtbl->genSymmetricKey( pGenKey, &pKey, 256,
P6FALSE );
}
err = pGenKey->lpVtbl->
release( pGenKey );
if (
P6SUCCEEDED( err )) err = m_pCrypto->lpVtbl->setKey( m_pCrypto, pKey );
if (NULL != pKey) pKey->lpVtbl->release( pKey );
if (
P6FAILED( err = pDir->lpVtbl->initialize( pDir ))) {
pDir->lpVtbl->release( pDir );
return err;
}
dbPath[0] = '\0';
pDir->lpVtbl->release( pDir );
return err;
}
pDir->lpVtbl->release( pDir );
return err;
}
pDir->lpVtbl->unlink( pDir, dbPath );
dbPath[0] = '\0';
pDir->lpVtbl->release( pDir );
return err;
}
if (
P6FAILED( err = m_pStr->lpVtbl->wstrlcat( m_pStr, dbPath,
P6CNTOF(dbPath), (
const P6WCHAR*)
P6TEXT(
"/db/KMIP12_keystore.sig"), NULL ))) {
pDir->lpVtbl->release( pDir );
return err;
}
pDir->lpVtbl->unlink( pDir, dbPath );
pDir->lpVtbl->release( pDir );
return err;
}
{
m_pStr->lpVtbl->setMem( m_pStr, pPrefs, 0, sizeof( P6KMIP_PREF ));
pPrefs->pVersion = "1.2"; pPrefs->pSubdirectory = pLogDir;
pPrefs->asynchronous = asynchronous; pPrefs->maxBufferSize = maxBufferSize;
pPrefs->connectTimeout = connectTimeout; pPrefs->sendTimeout = sendTimeout;
pPrefs->receiveTimeout = receiveTimeout; pPrefs->initialBufCount = initialBufCount;
pPrefs->growBufsBy = growBufsBy; pPrefs->privateKeyEncoding = privateKeyEncoding;
pPrefs->publicKeyEncoding = publicKeyEncoding; pPrefs->symmetricKeyEncoding = symmetricKeyEncoding;
pPrefs->compatibility1 = m_compatMask;
return eOk;
}
P6ERR createSession( p6IKMIPClient *pClient,
P6BOOL bWithCredentials )
{
P6KMIP_CREDENTIAL credential;
if (
P6FAILED( err = pClient->lpVtbl->setSSLOptions( pClient, NULL, (P6SSF_METHOD_TLS1 | P6SSF_SECURE_CLIENT | P6SSF_SECURE_CLIENT_AUTH | P6SSF_LOG_X509SUBJECTLOOKUPS | P6SSF_VRFY_DISABLEHOSTMATCH))))
return err;
m_pStr->lpVtbl->setMem( m_pStr, &credential, 0, sizeof( P6KMIP_CREDENTIAL ));
credential.value.password.userName.pString = "Fred";
credential.value.password.userName.length = 4;
credential.value.password.password.pString = "password1";
credential.value.password.password.length = 9;
return pClient->lpVtbl->open( pClient, m_pHostName, m_port, (bWithCredentials ? &credential : NULL));
}
P6ERR extractUniqueId( p6IKMIPStr *pEnum, P6NCSTR *pUniqueId )
{
P6NCSTR buffer = { NULL, 0 };
pUniqueId->length = 0;
err = pEnum->lpVtbl->reset( pEnum );
err = pEnum->lpVtbl->next( pEnum, &buffer );
if ( 0 < buffer.length )
{
if (NULL == (pGUID = (
P6CHAR*)malloc(
sizeof(
P6CHAR) * (buffer.length + 2) )))
return eNoMemory;
pGUID[0] = 0;
buffer.pString = pGUID;
buffer.length += 2;
if (
P6FAILED( err = pEnum->lpVtbl->next( pEnum, &buffer )))
return err;
pUniqueId->pString = buffer.pString;
pUniqueId->length = buffer.length;
}
else err = eFail;
return err;
}
P6VOID verifyStateAttribute( p6IKMIPClient *pClient, P6NCSTR keyId,
P6UINT32 expectedState )
{
P6KMIP_RESULT resultCodes;
P6KMIP_ATTRIBRESULT getResult;
P6KMIP_GETATTRIBPARAMS attribParams;
P6KMIP_OBJECT_ATTRIBUTE objAttribute;
P6NCSTR attribNames[2];
attribNames[0].pString = "State";
attribNames[0].length = 5;
m_pStr->lpVtbl->setMem( m_pStr, &resultCodes, 0, sizeof( P6KMIP_RESULT ));
m_pStr->lpVtbl->setMem( m_pStr, &getResult, 0, sizeof( P6KMIP_ATTRIBRESULT ));
m_pStr->lpVtbl->setMem( m_pStr, &attribParams, 0, sizeof( P6KMIP_GETATTRIBPARAMS ));
attribParams.uniqueId = keyId;
attribParams.attribCount = 1;
attribParams.pAttributeNames = attribNames;
if (
P6SUCCEEDED( err = pClient->lpVtbl->getAttributes( pClient, attribParams, &getResult, &resultCodes )))
{
if (NULL != getResult.pAttribute)
{
if (
P6SUCCEEDED( err = getResult.pAttribute->lpVtbl->next( getResult.pAttribute, &attribType )))
{
m_pStr->lpVtbl->setMem( m_pStr, &objAttribute, 0, sizeof( P6KMIP_OBJECT_ATTRIBUTE ));
if (KMIP_ATTRIB_STATE == attribType )
{
err = getResult.pAttribute->lpVtbl->getValue( getResult.pAttribute, &objAttribute );
if (expectedState != objAttribute.value.state) {
printf("\nExample 15, expected state is %d but the key has %d\n", expectedState, objAttribute.value.state );
}
}
}
getResult.pAttribute->lpVtbl->release( getResult.pAttribute );
}
}
}
P6ERR run( p6IDataStream *pStreamDebug )
{
p6IKMIPClient* pClient = NULL;
p6IKMIPStr* pUniqueId = NULL;
P6KMIP_PREF preferences;
P6KMIP_RESULT resultCodes;
P6KMIP_KEYPARAMS keyParams;
P6KMIP_TEMPLATEATTRIBUTE attributes;
P6KMIP_ATTRIBUTE attributeList[4];
P6KMIP_NEWOBJECT newKey;
P6KMIP_REVOCATION reason;
P6NCSTR keyId = {NULL, 0};
setPreferences( &preferences,
P6CTEXT(
"TC_311_12"), 0, 0, 0, 0, 5000, 2000, 2000, 120000, 2, 2 );
if (
P6FAILED( err = pClient->lpVtbl->initialize( pClient, (P6KMIPFLG_TRACE_MSGS | P6KMIPFLG_TRACE_FORMATKMIPXML), m_pKeystore, preferences ))) {
pClient->lpVtbl->release( pClient );
return err;
}
pClient->lpVtbl->release( pClient );
return err;
}
m_pStr->lpVtbl->setMem( m_pStr, &resultCodes, 0, sizeof( P6KMIP_RESULT ));
m_pStr->lpVtbl->setMem( m_pStr, &newKey, 0, sizeof( P6KMIP_NEWOBJECT ));
m_pStr->lpVtbl->setMem( m_pStr, &keyParams, 0, sizeof( P6KMIP_KEYPARAMS ));
m_pStr->lpVtbl->setMem( m_pStr, &attributes, 0, sizeof( P6KMIP_TEMPLATEATTRIBUTE ));
attributeList[0].type = KMIP_ATTRIB_CRYPTOALGORITHM;
attributeList[0].index = 0;
attributeList[0].value.cryptoAlgorithm = KMIP_AES;
attributeList[1].type = KMIP_ATTRIB_CRYPTOLENGTH;
attributeList[1].index = 0;
attributeList[1].value.cryptoLength = 128;
attributeList[2].type = KMIP_ATTRIB_CRYPTOUSAGEMASK;
attributeList[2].index = 0;
attributeList[2].value.cryptoUsageMask = KMIP_USE_ENCRYPT | KMIP_USE_DECRYPT;
attributeList[3].type = KMIP_ATTRIB_EXTENSION;
attributeList[3].index = 0;
attributeList[3].value.extension.xName.pString = "x-ID";
attributeList[3].value.extension.xName.length = 4;
attributeList[3].value.extension.xValue.vText.pString = "TC-311-12";
attributeList[3].value.extension.xValue.vText.length = 9;
attributeList[3].value.extension.xType = KMIP_TYPE_TEXTSTRING;
attributes.attribCount = 4;
attributes.pAttributeList = attributeList;
keyParams.pAttributes = &attributes;
if (
P6FAILED(err = pClient->lpVtbl->createKeyObject( pClient, keyParams, &newKey, &resultCodes ))) {
printf("\ncall to createKeyObject has failed %x\n", err );
}
else if (KMIP_RESULT_SUCCESS != resultCodes.resultStatus) {
printf("\nKMIP server returned an error\n");
}
if (NULL != newKey.pUniqueId)
{
err = extractUniqueId( newKey.pUniqueId, &keyId );
newKey.pUniqueId->lpVtbl->release( newKey.pUniqueId );
newKey.pUniqueId = NULL;
}
verifyStateAttribute( pClient, keyId, KMIP_STATE_PREACTIVE );
m_pStr->lpVtbl->setMem( m_pStr, &resultCodes, 0, sizeof( P6KMIP_RESULT ));
err = pClient->lpVtbl->activateObject( pClient, keyId, NULL, &pUniqueId, &resultCodes );
if (NULL != pUniqueId) pUniqueId->lpVtbl->release( pUniqueId);
verifyStateAttribute( pClient, keyId, KMIP_STATE_ACTIVE );
if (NULL != keyId.pString)
{
m_pStr->lpVtbl->setMem( m_pStr, &resultCodes, 0, sizeof( P6KMIP_RESULT ));
m_pStr->lpVtbl->setMem( m_pStr, &reason, 0, sizeof( P6KMIP_REVOCATION ));
pUniqueId = NULL;
reason.reasonCode = KMIP_REVOC_UNSPECIFIED;
if (
P6FAILED( err = pClient->lpVtbl->revokeObject( pClient, keyId, reason, NULL, NULL, &pUniqueId, &resultCodes ))) {
printf("\ncall to revokeObject has failed %x\n", err );
}
else if (KMIP_RESULT_SUCCESS != resultCodes.resultStatus) {
printf("\nKMIP server returned an error on revoke key\n");
}
if (NULL != pUniqueId) pUniqueId->lpVtbl->release( pUniqueId );
m_pStr->lpVtbl->setMem( m_pStr, &resultCodes, 0, sizeof( P6KMIP_RESULT ));
err = pClient->lpVtbl->destroyObject( pClient, keyId, NULL, &newKey.pUniqueId, &resultCodes );
printf("\ndestroying key call has failed\n");
}
else if (KMIP_RESULT_SUCCESS != resultCodes.resultStatus) {
printf("\nKMIP server returned an error\n");
}
free( (
P6VOID*)keyId.pString );
}
err = pClient->lpVtbl->close( pClient );
pClient->lpVtbl->release( pClient );
return err;
}
P6VOID KMIP_TC_311_12( p6IDataStream *pDataStream )
{
P6ERR err = eNotInitialized;
err = run( pDataStream );
}
freeGlobals();
}
int main(int argc,char *argv[])
{
p6IDataStream *pDataStream = NULL;
if (
P6FAILED( err = createStream( &pDataStream ))) {
printf("ERROR: Failed to create a datastream [ %x ]\n", err );
return 1;
}
{
KMIP_TC_311_12( pDataStream );
}
else printf("ERROR: Failed to initialize the loader [ %x ]\n", err );
return 0;
}