1 files changed, 3 insertions, 3 deletions

diff --git a/receiver.cpp b/receiver.cpp
index 4a0269a..59afdb2 100644
--- a/receiver.cpp
+++ b/receiver.cpp
@@ -107,7 +107,7 @@ public:
         // Dump key
         char hexkey[2*AES_KEY_LENGTH + 1];
         for (int i=0; i<AES_KEY_LENGTH; ++i)
-            sprintf(hexkey + 2 * i, "%02x", _key[i]);
+            snprintf(hexkey + 2 * i, 3, "%02x", _key[i]);
 
         // Glue together serialisation
         char filename[FILENAME_LENGTH + 1];
@@ -297,10 +297,10 @@ int main(int argc, char **argv) {
     mbedtls_ctr_drbg_seed(&ctr_drbg, mbedtls_entropy_func, &entropy, pp, sizeof(pp));
 
     if (privkey_file) {
-        if ((ret = mbedtls_pk_parse_keyfile(&pk, privkey_file, NULL) ) != 0 )
+        if ((ret = mbedtls_pk_parse_keyfile(&pk, privkey_file, NULL, mbedtls_ctr_drbg_random, &ctr_drbg) ) != 0 )
             errx(-1, "Fatal: mbedtls_pk_parse_keyfile returned -0x%04x\n", -ret );
     } else {
-        if ((ret = mbedtls_pk_parse_key(&pk, privkey, sizeof(privkey), NULL, 0) ) != 0 )
+        if ((ret = mbedtls_pk_parse_key(&pk, privkey, sizeof(privkey), NULL, 0, mbedtls_ctr_drbg_random, &ctr_drbg) ) != 0 )
             errx(-1, "Fatal: mbedtls_pk_parse_key returned -0x%04x\n", -ret );
     }