Fixed range of random produced in BN_is_prime_fasttest_ex() to be 1 < rand < w-1...
[openssl.git] / test / cmsapitest.c
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
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
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
89 void cleanup_tests(void)
90 {
91     X509_free(cert);
92     EVP_PKEY_free(privkey);
93 }