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