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
10 static X509 *cert = NULL;
11 static EVP_PKEY *privkey = NULL;
12
test_encrypt_decrypt(void)13 static 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
setup_tests(void)52 int 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
cleanup_tests(void)89 void cleanup_tests(void)
90 {
91 X509_free(cert);
92 EVP_PKEY_free(privkey);
93 }
94