test/stack_test.c - third_party/openssl - Git at Google

 /*
  * Copyright 2017 The OpenSSL Project Authors. All Rights Reserved.
  *
  * Licensed under the OpenSSL license (the "License").  You may not use
  * this file except in compliance with the License.  You can obtain a copy
  * in the file LICENSE in the source distribution or at
  * https://www.openssl.org/source/license.html
  */

 /* ====================================================================
  * Copyright (c) 2017 Oracle and/or its affiliates.  All rights reserved.
  */

 #include <stdio.h>
 #include <string.h>

 #include <openssl/opensslconf.h>
 #include <openssl/safestack.h>
 #include <openssl/err.h>
 #include <openssl/crypto.h>

 #include "e_os.h"
 #include "test_main.h"
 #include "testutil.h"

 /* The macros below generate unused functions which error out one of the clang
  * builds.  We disable this check here.
  */
 #ifdef __clang__
 #pragma clang diagnostic ignored "-Wunused-function"
 #endif

 typedef struct {
     int n;
     char c;
 } SS;

 typedef union {
     int n;
     char c;
 } SU;

 DEFINE_SPECIAL_STACK_OF(sint, int)
 DEFINE_SPECIAL_STACK_OF_CONST(uchar, unsigned char)
 DEFINE_STACK_OF(SS)
 DEFINE_STACK_OF_CONST(SU)

 static int int_compare(const int *const *a, const int *const *b)
 {
     if (**a < **b)
         return -1;
     if (**a > **b)
         return 1;
     return 0;
 }

 static int test_int_stack(void)
 {
     static int v[] = { 1, 2, -4, 16, 999, 1, -173, 1, 9 };
     const int n = OSSL_NELEM(v);
     static struct {
         int value;
         int unsorted;
         int sorted;
         int ex;
     } finds[] = {
         { 2,    1,  5,  5   },
         { 9,    7,  6,  6   },
         { -173, 5,  0,  0   },
         { 999,  3,  8,  8   },
         { 0,   -1, -1,  1   }
     };
     const int n_finds = OSSL_NELEM(finds);
     static struct {
         int value;
         int ex;
     } exfinds[] = {
         { 3,    5   },
         { 1000, 8   },
         { 20,   8   },
         { -999, 0   },
         { -5,   0   },
         { 8,    5   }
     };
     const int n_exfinds = OSSL_NELEM(exfinds);
     STACK_OF(sint) *s = sk_sint_new_null();
     int i;
     int testresult = 0;

     /* Check push and num */
     for (i = 0; i < n; i++) {
         if (sk_sint_num(s) != i) {
             fprintf(stderr, "test int stack size %d\n", i);
             goto end;
         }
         sk_sint_push(s, v + i);
     }
     if (sk_sint_num(s) != n) {
         fprintf(stderr, "test int stack size %d\n", n);
         goto end;
     }

     /* check the values */
     for (i = 0; i < n; i++)
         if (sk_sint_value(s, i) != v + i) {
             fprintf(stderr, "test int value %d\n", i);
             goto end;
         }

     /* find unsorted -- the pointers are compared */
     for (i = 0; i < n_finds; i++)
         if (sk_sint_find(s, v + finds[i].unsorted) != finds[i].unsorted) {
             fprintf(stderr, "test int unsorted find %d\n", i);
             goto end;
         }

     /* find_ex unsorted */
     for (i = 0; i < n_finds; i++)
         if (sk_sint_find_ex(s, v + finds[i].unsorted) != finds[i].unsorted) {
             fprintf(stderr, "test int unsorted find_ex %d\n", i);
             goto end;
         }

     /* sorting */
     if (sk_sint_is_sorted(s)) {
         fprintf(stderr, "test int unsorted\n");
         goto end;
     }
     sk_sint_set_cmp_func(s, &int_compare);
     sk_sint_sort(s);
     if (!sk_sint_is_sorted(s)) {
         fprintf(stderr, "test int sorted\n");
         goto end;
     }

     /* find sorted -- the value is matched so we don't need to locate it */
     for (i = 0; i < n_finds; i++)
         if (sk_sint_find(s, &finds[i].value) != finds[i].sorted) {
             fprintf(stderr, "test int sorted find %d\n", i);
             goto end;
         }

     /* find_ex sorted */
     for (i = 0; i < n_finds; i++)
         if (sk_sint_find_ex(s, &finds[i].value) != finds[i].ex) {
             fprintf(stderr, "test int sorted find_ex present %d\n", i);
             goto end;
         }
     for (i = 0; i < n_exfinds; i++)
         if (sk_sint_find_ex(s, &exfinds[i].value) != exfinds[i].ex) {
             fprintf(stderr, "test int sorted find_ex absent %d\n", i);
             goto end;
         }

     /* shift */
     if (sk_sint_shift(s) != v + 6) {
         fprintf(stderr, "test int shift\n");
         goto end;
     }

     testresult = 1;
 end:
     sk_sint_free(s);
     return testresult;
 }

 static int uchar_compare(const unsigned char *const *a,
                          const unsigned char *const *b)
 {
     return **a - (signed int)**b;
 }

 static int test_uchar_stack(void)
 {
     static const unsigned char v[] = { 1, 3, 7, 5, 255, 0 };
     const int n = OSSL_NELEM(v);
     STACK_OF(uchar) *s = sk_uchar_new(&uchar_compare), *r = NULL;
     int i;
     int testresult = 0;

     /* unshift and num */
     for (i = 0; i < n; i++) {
         if (sk_uchar_num(s) != i) {
             fprintf(stderr, "test uchar stack size %d\n", i);
             goto end;
         }
         sk_uchar_unshift(s, v + i);
     }
     if (sk_uchar_num(s) != n) {
         fprintf(stderr, "test uchar stack size %d\n", n);
         goto end;
     }

     /* dup */
     r = sk_uchar_dup(s);
     if (sk_uchar_num(r) != n) {
         fprintf(stderr, "test uchar dup size %d\n", n);
         goto end;
     }
     sk_uchar_sort(r);

     /* pop */
     for (i = 0; i < n; i++)
         if (sk_uchar_pop(s) != v + i) {
             fprintf(stderr, "test uchar pop %d\n", i);
             goto end;
         }

     /* free -- we rely on the debug malloc to detect leakage here */
     sk_uchar_free(s);
     s = NULL;

     /* dup again */
     if (sk_uchar_num(r) != n) {
         fprintf(stderr, "test uchar dup size %d\n", n);
         goto end;
     }

     /* zero */
     sk_uchar_zero(r);
     if (sk_uchar_num(r) != 0) {
         fprintf(stderr, "test uchar zero %d\n", n);
         goto end;
     }

     /* insert */
     sk_uchar_insert(r, v, 0);
     sk_uchar_insert(r, v + 2, -1);
     sk_uchar_insert(r, v + 1, 1);
     for (i = 0; i < 3; i++)
         if (sk_uchar_value(r, i) != v + i) {
             fprintf(stderr, "test uchar insert %d\n", i);
             goto end;
         }

     /* delete */
     if (sk_uchar_delete(r, 12) != NULL) {
         fprintf(stderr, "test uchar delete missing %d\n", n);
         goto end;
     }
     if (sk_uchar_delete(r, 1) != v + 1) {
         fprintf(stderr, "test uchar delete middle %d\n", n);
         goto end;
     }

     /* set */
     sk_uchar_set(r, 1, v + 1);
     for (i = 0; i < 2; i++)
         if (sk_uchar_value(r, i) != v + i) {
             fprintf(stderr, "test uchar set %d\n", i);
             goto end;
         }

     testresult = 1;
 end:
     sk_uchar_free(r);
     sk_uchar_free(s);
     return testresult;
 }

 static SS *SS_copy(const SS *p)
 {
     SS *q = OPENSSL_malloc(sizeof(*q));

     if (q != NULL)
         memcpy(q, p, sizeof(*q));
     return q;
 }

 static void SS_free(SS *p) {
     OPENSSL_free(p);
 }

 static int test_SS_stack(void)
 {
     STACK_OF(SS) *s = sk_SS_new_null();
     STACK_OF(SS) *r = NULL;
     SS *v[10], *p;
     const int n = OSSL_NELEM(v);
     int i;
     int testresult = 0;

     /* allocate and push */
     for (i = 0; i < n; i++) {
         v[i] = OPENSSL_malloc(sizeof(*v[i]));

         if (v[i] == NULL) {
             fprintf(stderr, "test SS memory allocation failure\n");
             goto end;
         }
         v[i]->n = i;
         v[i]->c = 'A' + i;
         if (sk_SS_num(s) != i) {
             fprintf(stderr, "test SS stack size %d\n", i);
             goto end;
         }
         sk_SS_push(s, v[i]);
     }
     if (sk_SS_num(s) != n) {
         fprintf(stderr, "test SS size %d\n", n);
         goto end;
     }

     /* deepcopy */
     r = sk_SS_deep_copy(s, &SS_copy, &SS_free);
     if (r == NULL) {
         fprintf(stderr, "test SS deepcopy failure\n");
         goto end;
     }
     for (i = 0; i < n; i++) {
         p = sk_SS_value(r, i);
         if (p == v[i]) {
             fprintf(stderr, "test SS deepcopy non-copy %d\n", i);
             goto end;
         }
         if (p->n != v[i]->n || p->c != v[i]->c) {
             fprintf(stderr, "test SS deepcopy values %d\n", i);
             goto end;
         }
     }

     /* pop_free - we rely on the malloc debug to catch the leak */
     sk_SS_pop_free(r, &SS_free);
     r = NULL;

     /* delete_ptr */
     if ((p = sk_SS_delete_ptr(s, v[3])) == NULL) {
         fprintf(stderr, "test SS delete ptr not found\n");
         goto end;
     }
     SS_free(p);
     if (sk_SS_num(s) != n-1) {
         fprintf(stderr, "test SS delete ptr size\n");
         goto end;
     }
     for (i = 0; i < n-1; i++)
         if (sk_SS_value(s, i) != v[i<3 ? i : 1+i]) {
             fprintf(stderr, "test SS delete ptr item %d\n", i);
             goto end;
         }

     testresult = 1;
 end:
     sk_SS_pop_free(r, &SS_free);
     sk_SS_pop_free(s, &SS_free);
     return testresult;
 }

 static int test_SU_stack(void)
 {
     STACK_OF(SU) *s = sk_SU_new_null();
     SU v[10];
     const int n = OSSL_NELEM(v);
     int i;
     int testresult = 0;

     /* allocate and push */
     for (i = 0; i < n; i++) {
         if ((i & 1) == 0)
             v[i].n = i;
         else
             v[i].c = 'A' + i;
         if (sk_SU_num(s) != i) {
             fprintf(stderr, "test SU stack size %d\n", i);
             goto end;
         }
         sk_SU_push(s, v + i);
     }
     if (sk_SU_num(s) != n) {
         fprintf(stderr, "test SU size %d\n", n);
         goto end;
     }

     /* check the pointers are correct */
     for (i = 0; i < n; i++)
         if (sk_SU_value(s, i) != v + i) {
             fprintf(stderr, "test SU pointer check %d\n", i);
             goto end;
         }

     testresult = 1;
 end:
     sk_SU_free(s);
     return testresult;
 }

 void register_tests(void)
 {
     ADD_TEST(test_int_stack);
     ADD_TEST(test_uchar_stack);
     ADD_TEST(test_SS_stack);
     ADD_TEST(test_SU_stack);
 }
	/*
	* Copyright 2017 The OpenSSL Project Authors. All Rights Reserved.
	*
	* Licensed under the OpenSSL license (the "License"). You may not use
	* this file except in compliance with the License. You can obtain a copy
	* in the file LICENSE in the source distribution or at
	* https://www.openssl.org/source/license.html
	*/

	/* ====================================================================
	* Copyright (c) 2017 Oracle and/or its affiliates. All rights reserved.
	*/

	#include <stdio.h>
	#include <string.h>

	#include <openssl/opensslconf.h>
	#include <openssl/safestack.h>
	#include <openssl/err.h>
	#include <openssl/crypto.h>

	#include "e_os.h"
	#include "test_main.h"
	#include "testutil.h"

	/* The macros below generate unused functions which error out one of the clang
	* builds. We disable this check here.
	*/
	#ifdef __clang__
	#pragma clang diagnostic ignored "-Wunused-function"
	#endif

	typedef struct {
	int n;
	char c;
	} SS;

	typedef union {
	int n;
	char c;
	} SU;

	DEFINE_SPECIAL_STACK_OF(sint, int)
	DEFINE_SPECIAL_STACK_OF_CONST(uchar, unsigned char)
	DEFINE_STACK_OF(SS)
	DEFINE_STACK_OF_CONST(SU)

	static int int_compare(const int const a, const int const b)
	{
	if (a < b)
	return -1;
	if (a > b)
	return 1;
	return 0;
	}

	static int test_int_stack(void)
	{
	static int v[] = { 1, 2, -4, 16, 999, 1, -173, 1, 9 };
	const int n = OSSL_NELEM(v);
	static struct {
	int value;
	int unsorted;
	int sorted;
	int ex;
	} finds[] = {
	{ 2, 1, 5, 5 },
	{ 9, 7, 6, 6 },
	{ -173, 5, 0, 0 },
	{ 999, 3, 8, 8 },
	{ 0, -1, -1, 1 }
	};
	const int n_finds = OSSL_NELEM(finds);
	static struct {
	int value;
	int ex;
	} exfinds[] = {
	{ 3, 5 },
	{ 1000, 8 },
	{ 20, 8 },
	{ -999, 0 },
	{ -5, 0 },
	{ 8, 5 }
	};
	const int n_exfinds = OSSL_NELEM(exfinds);
	STACK_OF(sint) *s = sk_sint_new_null();
	int i;
	int testresult = 0;

	/* Check push and num */
	for (i = 0; i < n; i++) {
	if (sk_sint_num(s) != i) {
	fprintf(stderr, "test int stack size %d\n", i);
	goto end;
	}
	sk_sint_push(s, v + i);
	}
	if (sk_sint_num(s) != n) {
	fprintf(stderr, "test int stack size %d\n", n);
	goto end;
	}

	/* check the values */
	for (i = 0; i < n; i++)
	if (sk_sint_value(s, i) != v + i) {
	fprintf(stderr, "test int value %d\n", i);
	goto end;
	}

	/* find unsorted -- the pointers are compared */
	for (i = 0; i < n_finds; i++)
	if (sk_sint_find(s, v + finds[i].unsorted) != finds[i].unsorted) {
	fprintf(stderr, "test int unsorted find %d\n", i);
	goto end;
	}

	/* find_ex unsorted */
	for (i = 0; i < n_finds; i++)
	if (sk_sint_find_ex(s, v + finds[i].unsorted) != finds[i].unsorted) {
	fprintf(stderr, "test int unsorted find_ex %d\n", i);
	goto end;
	}

	/* sorting */
	if (sk_sint_is_sorted(s)) {
	fprintf(stderr, "test int unsorted\n");
	goto end;
	}
	sk_sint_set_cmp_func(s, &int_compare);
	sk_sint_sort(s);
	if (!sk_sint_is_sorted(s)) {
	fprintf(stderr, "test int sorted\n");
	goto end;
	}

	/* find sorted -- the value is matched so we don't need to locate it */
	for (i = 0; i < n_finds; i++)
	if (sk_sint_find(s, &finds[i].value) != finds[i].sorted) {
	fprintf(stderr, "test int sorted find %d\n", i);
	goto end;
	}

	/* find_ex sorted */
	for (i = 0; i < n_finds; i++)
	if (sk_sint_find_ex(s, &finds[i].value) != finds[i].ex) {
	fprintf(stderr, "test int sorted find_ex present %d\n", i);
	goto end;
	}
	for (i = 0; i < n_exfinds; i++)
	if (sk_sint_find_ex(s, &exfinds[i].value) != exfinds[i].ex) {
	fprintf(stderr, "test int sorted find_ex absent %d\n", i);
	goto end;
	}

	/* shift */
	if (sk_sint_shift(s) != v + 6) {
	fprintf(stderr, "test int shift\n");
	goto end;
	}

	testresult = 1;
	end:
	sk_sint_free(s);
	return testresult;
	}

	static int uchar_compare(const unsigned char const a,
	const unsigned char const b)
	{
	return a - (signed int)b;
	}

	static int test_uchar_stack(void)
	{
	static const unsigned char v[] = { 1, 3, 7, 5, 255, 0 };
	const int n = OSSL_NELEM(v);
	STACK_OF(uchar) s = sk_uchar_new(&uchar_compare), r = NULL;
	int i;
	int testresult = 0;

	/* unshift and num */
	for (i = 0; i < n; i++) {
	if (sk_uchar_num(s) != i) {
	fprintf(stderr, "test uchar stack size %d\n", i);
	goto end;
	}
	sk_uchar_unshift(s, v + i);
	}
	if (sk_uchar_num(s) != n) {
	fprintf(stderr, "test uchar stack size %d\n", n);
	goto end;
	}

	/* dup */
	r = sk_uchar_dup(s);
	if (sk_uchar_num(r) != n) {
	fprintf(stderr, "test uchar dup size %d\n", n);
	goto end;
	}
	sk_uchar_sort(r);

	/* pop */
	for (i = 0; i < n; i++)
	if (sk_uchar_pop(s) != v + i) {
	fprintf(stderr, "test uchar pop %d\n", i);
	goto end;
	}

	/* free -- we rely on the debug malloc to detect leakage here */
	sk_uchar_free(s);
	s = NULL;

	/* dup again */
	if (sk_uchar_num(r) != n) {
	fprintf(stderr, "test uchar dup size %d\n", n);
	goto end;
	}

	/* zero */
	sk_uchar_zero(r);
	if (sk_uchar_num(r) != 0) {
	fprintf(stderr, "test uchar zero %d\n", n);
	goto end;
	}

	/* insert */
	sk_uchar_insert(r, v, 0);
	sk_uchar_insert(r, v + 2, -1);
	sk_uchar_insert(r, v + 1, 1);
	for (i = 0; i < 3; i++)
	if (sk_uchar_value(r, i) != v + i) {
	fprintf(stderr, "test uchar insert %d\n", i);
	goto end;
	}

	/* delete */
	if (sk_uchar_delete(r, 12) != NULL) {
	fprintf(stderr, "test uchar delete missing %d\n", n);
	goto end;
	}
	if (sk_uchar_delete(r, 1) != v + 1) {
	fprintf(stderr, "test uchar delete middle %d\n", n);
	goto end;
	}

	/* set */
	sk_uchar_set(r, 1, v + 1);
	for (i = 0; i < 2; i++)
	if (sk_uchar_value(r, i) != v + i) {
	fprintf(stderr, "test uchar set %d\n", i);
	goto end;
	}

	testresult = 1;
	end:
	sk_uchar_free(r);
	sk_uchar_free(s);
	return testresult;
	}

	static SS SS_copy(const SS p)
	{
	SS q = OPENSSL_malloc(sizeof(q));

	if (q != NULL)
	memcpy(q, p, sizeof(*q));
	return q;
	}

	static void SS_free(SS *p) {
	OPENSSL_free(p);
	}

	static int test_SS_stack(void)
	{
	STACK_OF(SS) *s = sk_SS_new_null();
	STACK_OF(SS) *r = NULL;
	SS v[10], p;
	const int n = OSSL_NELEM(v);
	int i;
	int testresult = 0;

	/* allocate and push */
	for (i = 0; i < n; i++) {
	v[i] = OPENSSL_malloc(sizeof(*v[i]));

	if (v[i] == NULL) {
	fprintf(stderr, "test SS memory allocation failure\n");
	goto end;
	}
	v[i]->n = i;
	v[i]->c = 'A' + i;
	if (sk_SS_num(s) != i) {
	fprintf(stderr, "test SS stack size %d\n", i);
	goto end;
	}
	sk_SS_push(s, v[i]);
	}
	if (sk_SS_num(s) != n) {
	fprintf(stderr, "test SS size %d\n", n);
	goto end;
	}

	/* deepcopy */
	r = sk_SS_deep_copy(s, &SS_copy, &SS_free);
	if (r == NULL) {
	fprintf(stderr, "test SS deepcopy failure\n");
	goto end;
	}
	for (i = 0; i < n; i++) {
	p = sk_SS_value(r, i);
	if (p == v[i]) {
	fprintf(stderr, "test SS deepcopy non-copy %d\n", i);
	goto end;
	}
	if (p->n != v[i]->n \|\| p->c != v[i]->c) {
	fprintf(stderr, "test SS deepcopy values %d\n", i);
	goto end;
	}
	}

	/* pop_free - we rely on the malloc debug to catch the leak */
	sk_SS_pop_free(r, &SS_free);
	r = NULL;

	/* delete_ptr */
	if ((p = sk_SS_delete_ptr(s, v[3])) == NULL) {
	fprintf(stderr, "test SS delete ptr not found\n");
	goto end;
	}
	SS_free(p);
	if (sk_SS_num(s) != n-1) {
	fprintf(stderr, "test SS delete ptr size\n");
	goto end;
	}
	for (i = 0; i < n-1; i++)
	if (sk_SS_value(s, i) != v[i<3 ? i : 1+i]) {
	fprintf(stderr, "test SS delete ptr item %d\n", i);
	goto end;
	}

	testresult = 1;
	end:
	sk_SS_pop_free(r, &SS_free);
	sk_SS_pop_free(s, &SS_free);
	return testresult;
	}

	static int test_SU_stack(void)
	{
	STACK_OF(SU) *s = sk_SU_new_null();
	SU v[10];
	const int n = OSSL_NELEM(v);
	int i;
	int testresult = 0;

	/* allocate and push */
	for (i = 0; i < n; i++) {
	if ((i & 1) == 0)
	v[i].n = i;
	else
	v[i].c = 'A' + i;
	if (sk_SU_num(s) != i) {
	fprintf(stderr, "test SU stack size %d\n", i);
	goto end;
	}
	sk_SU_push(s, v + i);
	}
	if (sk_SU_num(s) != n) {
	fprintf(stderr, "test SU size %d\n", n);
	goto end;
	}

	/* check the pointers are correct */
	for (i = 0; i < n; i++)
	if (sk_SU_value(s, i) != v + i) {
	fprintf(stderr, "test SU pointer check %d\n", i);
	goto end;
	}

	testresult = 1;
	end:
	sk_SU_free(s);
	return testresult;
	}

	void register_tests(void)
	{
	ADD_TEST(test_int_stack);
	ADD_TEST(test_uchar_stack);
	ADD_TEST(test_SS_stack);
	ADD_TEST(test_SU_stack);
	}