(PureBasic) RSA Encrypt/Decrypt AES Key

Demonstrates how to use RSA to protect a key for AES encryption. It can be used in this scenario: You will provide your RSA public key to any number of counterparts. Your counterpart will generate an AES key, encrypt data (or a file) using it, then encrypt the AES key using your RSA public key. Your counterpart sends you both the encrypted data and the encrypted key. Since you are the only one with access to the RSA private key, only you can decrypt the AES key. You decrypt the key, then decrypt the data using the AES key.

This example will show the entire process. (1) Generate an RSA key and save both private and public parts to PEM files. (2) Encrypt a file using a randomly generated AES encryption key. (3) RSA encrypt the AES key. (4) RSA decrypt the AES key. (5) Use it to AES decrypt the file or data.

Chilkat PureBasic Module Download Chilkat PureBasic Module

IncludeFile "CkPublicKey.pb"
IncludeFile "CkPrivateKey.pb"
IncludeFile "CkRsa.pb"
IncludeFile "CkCrypt2.pb"

Procedure ChilkatExample()

    ; This example assumes the Chilkat API to have been previously unlocked.
    ; See Global Unlock Sample for sample code.

    ; ------------------
    ; Step 1.  Generate an RSA key and save to PEM files.

    rsa.i = CkRsa::ckCreate()
    If rsa.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    ; Generate a 1024-bit key. 
    success.i = CkRsa::ckGenerateKey(rsa,1024)
    If success <> 1
        Debug CkRsa::ckLastErrorText(rsa)
        CkRsa::ckDispose(rsa)
        ProcedureReturn
    EndIf

    ; Keys are exported in XML format:
    pubKeyXml.s = CkRsa::ckExportPublicKey(rsa)

    pubKey.i = CkPublicKey::ckCreate()
    If pubKey.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success = CkPublicKey::ckLoadFromString(pubKey,pubKeyXml)
    ; For brevity, we are not checking the return value...
    success = CkPublicKey::ckSavePemFile(pubKey,0,"qa_temp/pubKey.pem")

    privKeyXml.s = CkRsa::ckExportPrivateKey(rsa)

    privKey.i = CkPrivateKey::ckCreate()
    If privKey.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success = CkPrivateKey::ckLoadXml(privKey,privKeyXml)
    success = CkPrivateKey::ckSavePemFile(privKey,"qa_temp/privKey.pem")

    ; Other methods exist for saving the private key in PKCS8 format,
    ; both encrypted and un-encrypted..

    ; ------------------
    ; Step 2.   This is the code your counterpart will run to 
    ; AES encrypt a file.  It generates a random AES key
    ; to use for the encryption.

    crypt.i = CkCrypt2::ckCreate()
    If crypt.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkCrypt2::setCkCryptAlgorithm(crypt, "aes")
    CkCrypt2::setCkCipherMode(crypt, "cbc")
    CkCrypt2::setCkKeyLength(crypt, 256)

    ; Generate random bytes and return the random bytes
    ; in a hex string:
    CkCrypt2::setCkEncodingMode(crypt, "hex")
    randomKey.s = CkCrypt2::ckGenRandomBytesENC(crypt,32)
    Debug "AES key = " + randomKey

    ; Set the key.
    CkCrypt2::ckSetEncodedKey(crypt,randomKey,"hex")

    ; Set the IV to a known value that will be used on both sides.
    ; (If desired, you could generate a random IV and protect it in the same
    ; way as the key...)
    ; The length of the IV for AES is always 16 bytes, regardless of the key size.
    CkCrypt2::ckSetEncodedIV(crypt,"000102030405060708090A0B0C0D0E0F","hex")

    ; AES Encrypt the file (the file may be any size because it will
    ; stream the file in/out.
    success = CkCrypt2::ckCkEncryptFile(crypt,"qa_data/hamlet.xml","qa_temp/aesEncrypted.dat")
    If success <> 1
        Debug CkCrypt2::ckLastErrorText(crypt)
        CkRsa::ckDispose(rsa)
        CkPublicKey::ckDispose(pubKey)
        CkPrivateKey::ckDispose(privKey)
        CkCrypt2::ckDispose(crypt)
        ProcedureReturn
    EndIf

    ; ------------------
    ; Step 3.
    ; ------------------
    ; RSA encrypt the AES key.
    ; We'll pretend  your counter part has the public-key PEM file and needs
    ; to load it.  Therefore, we'll start with a new RSA object and load
    ; the public key from the PEM..
    rsa2.i = CkRsa::ckCreate()
    If rsa2.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    pubKey2.i = CkPublicKey::ckCreate()
    If pubKey2.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    ; For brevity, don't check the success..  
    success = CkPublicKey::ckLoadFromFile(pubKey2,"qa_temp/pubKey.pem")
    pubKeyXml = CkPublicKey::ckGetXml(pubKey2)
    success = CkRsa::ckImportPublicKey(rsa2,pubKeyXml)

    ; RSA encrypt the AES key and return it as a base64 encoded string.
    CkRsa::setCkEncodingMode(rsa2, "base64")
    bUsePrivateKey.i = 0
    encryptedAesKey.s = CkRsa::ckEncryptStringENC(rsa2,randomKey,bUsePrivateKey)

    ; At this point, your counterpart sends you the encryptedAesKey,
    ; and the encrypted file.

    ; ------------------
    ; Step 4 - RSA decrypt the AES key.
    ; ------------------

    ; Start with a new RSA object and load the private key from the PEM.
    rsa3.i = CkRsa::ckCreate()
    If rsa3.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    privKey2.i = CkPrivateKey::ckCreate()
    If privKey2.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success = CkPrivateKey::ckLoadPemFile(privKey2,"qa_temp/privKey.pem")
    privKeyXml = CkPrivateKey::ckGetXml(privKey2)
    success = CkRsa::ckImportPrivateKey(rsa3,privKeyXml)

    ; The encrypted AES key is encoded using base64, so set
    ; our EncodingMode to "base64".
    CkRsa::setCkEncodingMode(rsa3, "base64")

    bUsePrivateKey = 1
    decryptedAesKey.s = CkRsa::ckDecryptStringENC(rsa3,encryptedAesKey,bUsePrivateKey)
    Debug "decryptedAesKey = " + decryptedAesKey

    ; ------------------
    ; Step 5 - AES decrypt the file.
    ; ------------------

    crypt2.i = CkCrypt2::ckCreate()
    If crypt2.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkCrypt2::setCkCryptAlgorithm(crypt2, "aes")
    CkCrypt2::setCkCipherMode(crypt2, "cbc")
    CkCrypt2::setCkKeyLength(crypt2, 256)

    ; Set the key.
    CkCrypt2::ckSetEncodedKey(crypt2,decryptedAesKey,"hex")

    ; Set the IV 
    CkCrypt2::ckSetEncodedIV(crypt2,"000102030405060708090A0B0C0D0E0F","hex")

    ; AES Decrypt the file (the file may be any size because it will
    ; stream the file in/out.
    success = CkCrypt2::ckCkDecryptFile(crypt2,"qa_temp/aesEncrypted.dat","qa_temp/decrypted.xml")
    If success <> 1
        Debug CkCrypt2::ckLastErrorText(crypt2)
        CkRsa::ckDispose(rsa)
        CkPublicKey::ckDispose(pubKey)
        CkPrivateKey::ckDispose(privKey)
        CkCrypt2::ckDispose(crypt)
        CkRsa::ckDispose(rsa2)
        CkPublicKey::ckDispose(pubKey2)
        CkRsa::ckDispose(rsa3)
        CkPrivateKey::ckDispose(privKey2)
        CkCrypt2::ckDispose(crypt2)
        ProcedureReturn
    EndIf



    CkRsa::ckDispose(rsa)
    CkPublicKey::ckDispose(pubKey)
    CkPrivateKey::ckDispose(privKey)
    CkCrypt2::ckDispose(crypt)
    CkRsa::ckDispose(rsa2)
    CkPublicKey::ckDispose(pubKey2)
    CkRsa::ckDispose(rsa3)
    CkPrivateKey::ckDispose(privKey2)
    CkCrypt2::ckDispose(crypt2)


    ProcedureReturn
EndProcedure