test/cmp_asn_test.c

   1 /*
   2  * Copyright 2007-2021 The OpenSSL Project Authors. All Rights Reserved.
   3  * Copyright Nokia 2007-2019
   4  * Copyright Siemens AG 2015-2019
   5  *
   6  * Licensed under the Apache License 2.0 (the "License").  You may not use
   7  * this file except in compliance with the License.  You can obtain a copy
   8  * in the file LICENSE in the source distribution or at
   9  * https://www.openssl.org/source/license.html
  10  */
  11
  12 #include "helpers/cmp_testlib.h"
  13
  14 static unsigned char rand_data[OSSL_CMP_TRANSACTIONID_LENGTH];
  15
  16 typedef struct test_fixture {
  17     const char *test_case_name;
  18     int expected;
  19     ASN1_OCTET_STRING *src_string;
  20     ASN1_OCTET_STRING *tgt_string;
  21
  22 } CMP_ASN_TEST_FIXTURE;
  23
  24 static CMP_ASN_TEST_FIXTURE *set_up(const char *const test_case_name)
  25 {
  26     CMP_ASN_TEST_FIXTURE *fixture;
  27
  28     if (!TEST_ptr(fixture = OPENSSL_zalloc(sizeof(*fixture))))
  29         return NULL;
  30     fixture->test_case_name = test_case_name;
  31     return fixture;
  32 }
  33
  34 static void tear_down(CMP_ASN_TEST_FIXTURE *fixture)
  35 {
  36     ASN1_OCTET_STRING_free(fixture->src_string);
  37     if (fixture->tgt_string != fixture->src_string)
  38         ASN1_OCTET_STRING_free(fixture->tgt_string);
  39
  40     OPENSSL_free(fixture);
  41 }
  42
  43 static int execute_cmp_asn1_get_int_test(CMP_ASN_TEST_FIXTURE *fixture)
  44 {
  45     int res;
  46     ASN1_INTEGER *asn1integer = ASN1_INTEGER_new();
  47
  48     if (!TEST_ptr(asn1integer))
  49         return 0;
  50     ASN1_INTEGER_set(asn1integer, 77);
  51     res = TEST_int_eq(77, ossl_cmp_asn1_get_int(asn1integer));
  52     ASN1_INTEGER_free(asn1integer);
  53     return res;
  54 }
  55
  56 static int test_cmp_asn1_get_int(void)
  57 {
  58     SETUP_TEST_FIXTURE(CMP_ASN_TEST_FIXTURE, set_up);
  59     fixture->expected = 1;
  60     EXECUTE_TEST(execute_cmp_asn1_get_int_test, tear_down);
  61     return result;
  62 }
  63
  64 static int execute_CMP_ASN1_OCTET_STRING_set1_test(CMP_ASN_TEST_FIXTURE *
  65                                                    fixture)
  66 {
  67     if (!TEST_int_eq(fixture->expected,
  68                      ossl_cmp_asn1_octet_string_set1(&fixture->tgt_string,
  69                                                      fixture->src_string)))
  70         return 0;
  71     if (fixture->expected != 0)
  72         return TEST_int_eq(0, ASN1_OCTET_STRING_cmp(fixture->tgt_string,
  73                                                     fixture->src_string));
  74     return 1;
  75 }
  76
  77 static int test_ASN1_OCTET_STRING_set(void)
  78 {
  79     SETUP_TEST_FIXTURE(CMP_ASN_TEST_FIXTURE, set_up);
  80     fixture->expected = 1;
  81     if (!TEST_ptr(fixture->tgt_string = ASN1_OCTET_STRING_new())
  82             || !TEST_ptr(fixture->src_string = ASN1_OCTET_STRING_new())
  83             || !TEST_true(ASN1_OCTET_STRING_set(fixture->src_string, rand_data,
  84                                                 sizeof(rand_data)))) {
  85         tear_down(fixture);
  86         fixture = NULL;
  87     }
  88     EXECUTE_TEST(execute_CMP_ASN1_OCTET_STRING_set1_test, tear_down);
  89     return result;
  90 }
  91
  92 static int test_ASN1_OCTET_STRING_set_tgt_is_src(void)
  93 {
  94     SETUP_TEST_FIXTURE(CMP_ASN_TEST_FIXTURE, set_up);
  95     fixture->expected = 1;
  96     if (!TEST_ptr(fixture->src_string = ASN1_OCTET_STRING_new())
  97             || !(fixture->tgt_string = fixture->src_string)
  98             || !TEST_true(ASN1_OCTET_STRING_set(fixture->src_string, rand_data,
  99                                                 sizeof(rand_data)))) {
 100         tear_down(fixture);
 101         fixture = NULL;
 102     }
 103     EXECUTE_TEST(execute_CMP_ASN1_OCTET_STRING_set1_test, tear_down);
 104     return result;
 105 }
 106
 107
 108 void cleanup_tests(void)
 109 {
 110     return;
 111 }
 112
 113 int setup_tests(void)
 114 {
 115     RAND_bytes(rand_data, OSSL_CMP_TRANSACTIONID_LENGTH);
 116     /* ASN.1 related tests */
 117     ADD_TEST(test_cmp_asn1_get_int);
 118     ADD_TEST(test_ASN1_OCTET_STRING_set);
 119     ADD_TEST(test_ASN1_OCTET_STRING_set_tgt_is_src);
 120     return 1;
 121 }