test/cmsapitest.c

   1 /*
   2  * Copyright 2018-2020 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 #include <string.h>
  11
  12 #include <openssl/cms.h>
  13 #include <openssl/bio.h>
  14 #include <openssl/x509.h>
  15 #include <openssl/pem.h>
  16
  17 #include "testutil.h"
  18
  19 DEFINE_STACK_OF(X509)
  20
  21 static X509 *cert = NULL;
  22 static EVP_PKEY *privkey = NULL;
  23
  24 static int test_encrypt_decrypt(void)
  25 {
  26     int testresult = 0;
  27     STACK_OF(X509) *certstack = sk_X509_new_null();
  28     const char *msg = "Hello world";
  29     BIO *msgbio = BIO_new_mem_buf(msg, strlen(msg));
  30     BIO *outmsgbio = BIO_new(BIO_s_mem());
  31     CMS_ContentInfo* content = NULL;
  32     char buf[80];
  33
  34     if (!TEST_ptr(certstack) || !TEST_ptr(msgbio) || !TEST_ptr(outmsgbio))
  35         goto end;
  36
  37     if (!TEST_int_gt(sk_X509_push(certstack, cert), 0))
  38         goto end;
  39
  40     content = CMS_encrypt(certstack, msgbio, EVP_aes_128_cbc(), CMS_TEXT);
  41     if (!TEST_ptr(content))
  42         goto end;
  43
  44     if (!TEST_true(CMS_decrypt(content, privkey, cert, NULL, outmsgbio,
  45                                CMS_TEXT)))
  46         goto end;
  47
  48     /* Check we got the message we first started with */
  49     if (!TEST_int_eq(BIO_gets(outmsgbio, buf, sizeof(buf)), strlen(msg))
  50             || !TEST_int_eq(strcmp(buf, msg), 0))
  51         goto end;
  52
  53     testresult = 1;
  54  end:
  55     sk_X509_free(certstack);
  56     BIO_free(msgbio);
  57     BIO_free(outmsgbio);
  58     CMS_ContentInfo_free(content);
  59
  60     return testresult;
  61 }
  62
  63 OPT_TEST_DECLARE_USAGE("certfile privkeyfile\n")
  64
  65 int setup_tests(void)
  66 {
  67     char *certin = NULL, *privkeyin = NULL;
  68     BIO *certbio = NULL, *privkeybio = NULL;
  69
  70     if (!test_skip_common_options()) {
  71         TEST_error("Error parsing test options\n");
  72         return 0;
  73     }
  74
  75     if (!TEST_ptr(certin = test_get_argument(0))
  76             || !TEST_ptr(privkeyin = test_get_argument(1)))
  77         return 0;
  78
  79     certbio = BIO_new_file(certin, "r");
  80     if (!TEST_ptr(certbio))
  81         return 0;
  82     if (!TEST_true(PEM_read_bio_X509(certbio, &cert, NULL, NULL))) {
  83         BIO_free(certbio);
  84         return 0;
  85     }
  86     BIO_free(certbio);
  87
  88     privkeybio = BIO_new_file(privkeyin, "r");
  89     if (!TEST_ptr(privkeybio)) {
  90         X509_free(cert);
  91         cert = NULL;
  92         return 0;
  93     }
  94     if (!TEST_true(PEM_read_bio_PrivateKey(privkeybio, &privkey, NULL, NULL))) {
  95         BIO_free(privkeybio);
  96         X509_free(cert);
  97         cert = NULL;
  98         return 0;
  99     }
 100     BIO_free(privkeybio);
 101
 102     ADD_TEST(test_encrypt_decrypt);
 103
 104     return 1;
 105 }
 106
 107 void cleanup_tests(void)
 108 {
 109     X509_free(cert);
 110     EVP_PKEY_free(privkey);
 111 }