pqc_kem_encaps_decaps.c

Exemplo de encapsulamento e desencapsulamento PQC com ML-KEM recebendo um shared secret.

Veja Nota sobre os exemplos.

#include <string.h>
#include <stdlib.h>
#include <stdio.h>
 
#include <dinamo.h> /* header do Dinamo */
 
#define HOST_ADDR       "127.0.0.1"
#define USER_ID         "master"
#define USER_PWD        "12345678"
 
#define KEY_ID          "decaps_key"
 
int main()
{
    int nRet = 0;
    struct AUTH_PWD authPwd;
    HSESSIONCTX hSession = NULL;
    HKEYCTX hEncapsKey = NULL;
 
    BYTE pbEncapsKey[DN_KEM_MAX_PUB_LEN] = {};
    DWORD cbEncapsKey = sizeof(pbEncapsKey);
 
    DN_KEM_INFO stKemInfo = {};
    DN_KEM_ENCAPS_RESULT stEncapsResult = {};
 
    /* Inicializa as bibliotecas do Dinamo */
 
    nRet = DInitialize(0);
    if (nRet){
        printf("Falha na funcao: DInitialize \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Bibliotecas inicializadas.\n");
 
    /* Inicializa a estrutura para conexão com o HSM  */
    
    strncpy(authPwd.szAddr, HOST_ADDR, sizeof(authPwd.szAddr));
    authPwd.nPort = DEFAULT_PORT;
    strncpy(authPwd.szUserId, USER_ID, sizeof(authPwd.szUserId));
    strncpy(authPwd.szPassword, USER_PWD, sizeof(authPwd.szPassword));
 
    nRet = DOpenSession(&hSession, SS_USER_PWD, (BYTE *)&authPwd, sizeof(authPwd), ENCRYPTED_CONN);
    if (nRet){
        printf("Falha na funcao: DOpenSession \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Sessao com o Dinamo estabelecida.\n");
 
    nRet = DGenerateKey(hSession, KEY_ID, ALG_ML_KEM_768, 0, &hEncapsKey);
    if (nRet){
        printf("Falha na funcao: DGenerateKey \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Chave criada com sucesso.\n");
 
    nRet = DExportKey(hEncapsKey, NULL, PUBLICKEY_BLOB, 0, pbEncapsKey, &cbEncapsKey);
    if (nRet){
        printf("Falha na funcao: DExportKey \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Chave encaps exportada com sucesso.\n");
 
    /*
        Faz o encapsulamento da chave utilizando o KEM.
        Faz a geração tradicional retornando o shared secret e o ciphertext.
    */
 
    memset(&stKemInfo, 0, sizeof(stKemInfo));
    stKemInfo.bOp = DN_KEM_OP_SHARED_SECRET;
 
    nRet = DKemEncaps(hSession, DN_KEM_ENCAPS_OPT_ML_KEM, &stKemInfo, pbEncapsKey, 
                      cbEncapsKey, &stEncapsResult, 0);
    if (nRet){
        printf("Falha na funcao: DKemEncaps \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Chave encapsulada com sucesso.\n");
 
    /*
        Desencapsula a chave utilizando o KEM.
        Faz a geração tradicional retornando o shared secret.
        As mesmas opções utilizadas no encapsulamento são utilizadas
        no desencapsulamento.
    */
 
    nRet = DKemDecaps(hSession, DN_KEM_DECAPS_OPT_ML_KEM, &stKemInfo,
                      KEY_ID, stEncapsResult.pbCiphertext, 
                      stEncapsResult.dwCiphertextLen, 
                      stEncapsResult.pbSharedSecret, 0);
    if (nRet){
        printf("Falha na funcao: DKemDecaps \nCodigo de erro: %d\n",nRet);
        goto clean;
    }
 
    printf("Chave desencapsulada com sucesso.\n");
 
    clean:
 
    /*
        Destroi o contexto das chaves usadas no TESTE do HSM e remove as chaves.
        A flag REMOVE_FROM_HSM indica que a chave deve ser removida do HSM.
        Caso não queira que a chave seja removida do HSM, passar 0 no lugar de
        REMOVE_FROM_HSM.
     */
 
    if (hEncapsKey){
        DDestroyKey(&hEncapsKey, REMOVE_FROM_HSM);
        printf("Contexto da chave de desencapsulamento liberado.\n");
    }
 
    if (hSession){
        DCloseSession(&hSession, 0);
        printf("Sessao encerrada.\n");
    }
 
    DFinalize();
 
    printf("Bibliotecas finalizada.\n");
 
    return nRet;
}