1#include <string.h> 2 3#include <openssl/cms.h> 4#include <openssl/bio.h> 5#include <openssl/x509.h> 6#include <openssl/pem.h> 7 8#include "testutil.h" 9 10static X509 *cert = NULL; 11static EVP_PKEY *privkey = NULL; 12 13static int test_encrypt_decrypt(void) 14{ 15 int testresult = 0; 16 STACK_OF(X509) *certstack = sk_X509_new_null(); 17 const char *msg = "Hello world"; 18 BIO *msgbio = BIO_new_mem_buf(msg, strlen(msg)); 19 BIO *outmsgbio = BIO_new(BIO_s_mem()); 20 CMS_ContentInfo* content = NULL; 21 char buf[80]; 22 23 if (!TEST_ptr(certstack) || !TEST_ptr(msgbio) || !TEST_ptr(outmsgbio)) 24 goto end; 25 26 if (!TEST_int_gt(sk_X509_push(certstack, cert), 0)) 27 goto end; 28 29 content = CMS_encrypt(certstack, msgbio, EVP_aes_128_cbc(), CMS_TEXT); 30 if (!TEST_ptr(content)) 31 goto end; 32 33 if (!TEST_true(CMS_decrypt(content, privkey, cert, NULL, outmsgbio, 34 CMS_TEXT))) 35 goto end; 36 37 /* Check we got the message we first started with */ 38 if (!TEST_int_eq(BIO_gets(outmsgbio, buf, sizeof(buf)), strlen(msg)) 39 || !TEST_int_eq(strcmp(buf, msg), 0)) 40 goto end; 41 42 testresult = 1; 43 end: 44 sk_X509_free(certstack); 45 BIO_free(msgbio); 46 BIO_free(outmsgbio); 47 CMS_ContentInfo_free(content); 48 49 return testresult; 50} 51 52int setup_tests(void) 53{ 54 char *certin = NULL, *privkeyin = NULL; 55 BIO *certbio = NULL, *privkeybio = NULL; 56 57 if (!TEST_ptr(certin = test_get_argument(0)) 58 || !TEST_ptr(privkeyin = test_get_argument(1))) 59 return 0; 60 61 certbio = BIO_new_file(certin, "r"); 62 if (!TEST_ptr(certbio)) 63 return 0; 64 if (!TEST_true(PEM_read_bio_X509(certbio, &cert, NULL, NULL))) { 65 BIO_free(certbio); 66 return 0; 67 } 68 BIO_free(certbio); 69 70 privkeybio = BIO_new_file(privkeyin, "r"); 71 if (!TEST_ptr(privkeybio)) { 72 X509_free(cert); 73 cert = NULL; 74 return 0; 75 } 76 if (!TEST_true(PEM_read_bio_PrivateKey(privkeybio, &privkey, NULL, NULL))) { 77 BIO_free(privkeybio); 78 X509_free(cert); 79 cert = NULL; 80 return 0; 81 } 82 BIO_free(privkeybio); 83 84 ADD_TEST(test_encrypt_decrypt); 85 86 return 1; 87} 88 89void cleanup_tests(void) 90{ 91 X509_free(cert); 92 EVP_PKEY_free(privkey); 93} 94