1*e7be843bSPierre Pronchery /*
2*e7be843bSPierre Pronchery * Copyright 2022-2025 The OpenSSL Project Authors. All Rights Reserved.
3*e7be843bSPierre Pronchery *
4*e7be843bSPierre Pronchery * Licensed under the Apache License 2.0 (the "License"). You may not use
5*e7be843bSPierre Pronchery * this file except in compliance with the License. You can obtain a copy
6*e7be843bSPierre Pronchery * in the file LICENSE in the source distribution or at
7*e7be843bSPierre Pronchery * https://www.openssl.org/source/license.html
8*e7be843bSPierre Pronchery */
9*e7be843bSPierre Pronchery #include <stdio.h>
10*e7be843bSPierre Pronchery #include <string.h>
11*e7be843bSPierre Pronchery #include <openssl/evp.h>
12*e7be843bSPierre Pronchery #include <openssl/bio.h>
13*e7be843bSPierre Pronchery #include <openssl/rand.h>
14*e7be843bSPierre Pronchery #include <openssl/comp.h>
15*e7be843bSPierre Pronchery
16*e7be843bSPierre Pronchery #include "testutil.h"
17*e7be843bSPierre Pronchery #include "testutil/output.h"
18*e7be843bSPierre Pronchery #include "testutil/tu_local.h"
19*e7be843bSPierre Pronchery
20*e7be843bSPierre Pronchery #define COMPRESS 1
21*e7be843bSPierre Pronchery #define EXPAND 0
22*e7be843bSPierre Pronchery
23*e7be843bSPierre Pronchery #define BUFFER_SIZE 32 * 1024
24*e7be843bSPierre Pronchery #define NUM_SIZES 4
25*e7be843bSPierre Pronchery static int sizes[NUM_SIZES] = { 64, 512, 2048, 16 * 1024 };
26*e7be843bSPierre Pronchery
27*e7be843bSPierre Pronchery /* using global buffers */
28*e7be843bSPierre Pronchery static unsigned char *original = NULL;
29*e7be843bSPierre Pronchery static unsigned char *result = NULL;
30*e7be843bSPierre Pronchery
31*e7be843bSPierre Pronchery /*
32*e7be843bSPierre Pronchery * For compression:
33*e7be843bSPierre Pronchery * the write operation compresses
34*e7be843bSPierre Pronchery * the read operation decompresses
35*e7be843bSPierre Pronchery */
36*e7be843bSPierre Pronchery
do_bio_comp_test(const BIO_METHOD * meth,size_t size)37*e7be843bSPierre Pronchery static int do_bio_comp_test(const BIO_METHOD *meth, size_t size)
38*e7be843bSPierre Pronchery {
39*e7be843bSPierre Pronchery BIO *bcomp = NULL;
40*e7be843bSPierre Pronchery BIO *bmem = NULL;
41*e7be843bSPierre Pronchery BIO *bexp = NULL;
42*e7be843bSPierre Pronchery int osize;
43*e7be843bSPierre Pronchery int rsize;
44*e7be843bSPierre Pronchery int ret = 0;
45*e7be843bSPierre Pronchery
46*e7be843bSPierre Pronchery /* Compress */
47*e7be843bSPierre Pronchery if (!TEST_ptr(meth))
48*e7be843bSPierre Pronchery goto err;
49*e7be843bSPierre Pronchery if (!TEST_ptr(bcomp = BIO_new(meth)))
50*e7be843bSPierre Pronchery goto err;
51*e7be843bSPierre Pronchery if (!TEST_ptr(bmem = BIO_new(BIO_s_mem())))
52*e7be843bSPierre Pronchery goto err;
53*e7be843bSPierre Pronchery BIO_push(bcomp, bmem);
54*e7be843bSPierre Pronchery osize = BIO_write(bcomp, original, size);
55*e7be843bSPierre Pronchery if (!TEST_int_eq(osize, size)
56*e7be843bSPierre Pronchery || !TEST_true(BIO_flush(bcomp)))
57*e7be843bSPierre Pronchery goto err;
58*e7be843bSPierre Pronchery BIO_free(bcomp);
59*e7be843bSPierre Pronchery bcomp = NULL;
60*e7be843bSPierre Pronchery
61*e7be843bSPierre Pronchery /* decompress */
62*e7be843bSPierre Pronchery if (!TEST_ptr(bexp = BIO_new(meth)))
63*e7be843bSPierre Pronchery goto err;
64*e7be843bSPierre Pronchery BIO_push(bexp, bmem);
65*e7be843bSPierre Pronchery rsize = BIO_read(bexp, result, size);
66*e7be843bSPierre Pronchery
67*e7be843bSPierre Pronchery if (!TEST_int_eq(size, rsize)
68*e7be843bSPierre Pronchery || !TEST_mem_eq(original, osize, result, rsize))
69*e7be843bSPierre Pronchery goto err;
70*e7be843bSPierre Pronchery
71*e7be843bSPierre Pronchery ret = 1;
72*e7be843bSPierre Pronchery err:
73*e7be843bSPierre Pronchery BIO_free(bexp);
74*e7be843bSPierre Pronchery BIO_free(bcomp);
75*e7be843bSPierre Pronchery BIO_free(bmem);
76*e7be843bSPierre Pronchery return ret;
77*e7be843bSPierre Pronchery }
78*e7be843bSPierre Pronchery
do_bio_comp(const BIO_METHOD * meth,int n)79*e7be843bSPierre Pronchery static int do_bio_comp(const BIO_METHOD *meth, int n)
80*e7be843bSPierre Pronchery {
81*e7be843bSPierre Pronchery int i;
82*e7be843bSPierre Pronchery int success = 0;
83*e7be843bSPierre Pronchery int size = sizes[n % 4];
84*e7be843bSPierre Pronchery int type = n / 4;
85*e7be843bSPierre Pronchery
86*e7be843bSPierre Pronchery original = OPENSSL_malloc(BUFFER_SIZE);
87*e7be843bSPierre Pronchery result = OPENSSL_malloc(BUFFER_SIZE);
88*e7be843bSPierre Pronchery
89*e7be843bSPierre Pronchery if (!TEST_ptr(original) || !TEST_ptr(result))
90*e7be843bSPierre Pronchery goto err;
91*e7be843bSPierre Pronchery
92*e7be843bSPierre Pronchery switch (type) {
93*e7be843bSPierre Pronchery case 0:
94*e7be843bSPierre Pronchery TEST_info("zeros of size %d\n", size);
95*e7be843bSPierre Pronchery memset(original, 0, BUFFER_SIZE);
96*e7be843bSPierre Pronchery break;
97*e7be843bSPierre Pronchery case 1:
98*e7be843bSPierre Pronchery TEST_info("ones of size %d\n", size);
99*e7be843bSPierre Pronchery memset(original, 1, BUFFER_SIZE);
100*e7be843bSPierre Pronchery break;
101*e7be843bSPierre Pronchery case 2:
102*e7be843bSPierre Pronchery TEST_info("sequential of size %d\n", size);
103*e7be843bSPierre Pronchery for (i = 0; i < BUFFER_SIZE; i++)
104*e7be843bSPierre Pronchery original[i] = i & 0xFF;
105*e7be843bSPierre Pronchery break;
106*e7be843bSPierre Pronchery case 3:
107*e7be843bSPierre Pronchery TEST_info("random of size %d\n", size);
108*e7be843bSPierre Pronchery if (!TEST_int_gt(RAND_bytes(original, BUFFER_SIZE), 0))
109*e7be843bSPierre Pronchery goto err;
110*e7be843bSPierre Pronchery break;
111*e7be843bSPierre Pronchery default:
112*e7be843bSPierre Pronchery goto err;
113*e7be843bSPierre Pronchery }
114*e7be843bSPierre Pronchery
115*e7be843bSPierre Pronchery if (!TEST_true(do_bio_comp_test(meth, size)))
116*e7be843bSPierre Pronchery goto err;
117*e7be843bSPierre Pronchery success = 1;
118*e7be843bSPierre Pronchery err:
119*e7be843bSPierre Pronchery OPENSSL_free(original);
120*e7be843bSPierre Pronchery OPENSSL_free(result);
121*e7be843bSPierre Pronchery return success;
122*e7be843bSPierre Pronchery }
123*e7be843bSPierre Pronchery
124*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_ZSTD
test_zstd(int n)125*e7be843bSPierre Pronchery static int test_zstd(int n)
126*e7be843bSPierre Pronchery {
127*e7be843bSPierre Pronchery return do_bio_comp(BIO_f_zstd(), n);
128*e7be843bSPierre Pronchery }
129*e7be843bSPierre Pronchery #endif
130*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_BROTLI
test_brotli(int n)131*e7be843bSPierre Pronchery static int test_brotli(int n)
132*e7be843bSPierre Pronchery {
133*e7be843bSPierre Pronchery return do_bio_comp(BIO_f_brotli(), n);
134*e7be843bSPierre Pronchery }
135*e7be843bSPierre Pronchery #endif
136*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_ZLIB
test_zlib(int n)137*e7be843bSPierre Pronchery static int test_zlib(int n)
138*e7be843bSPierre Pronchery {
139*e7be843bSPierre Pronchery return do_bio_comp(BIO_f_zlib(), n);
140*e7be843bSPierre Pronchery }
141*e7be843bSPierre Pronchery #endif
142*e7be843bSPierre Pronchery
setup_tests(void)143*e7be843bSPierre Pronchery int setup_tests(void)
144*e7be843bSPierre Pronchery {
145*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_ZLIB
146*e7be843bSPierre Pronchery ADD_ALL_TESTS(test_zlib, NUM_SIZES * 4);
147*e7be843bSPierre Pronchery #endif
148*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_BROTLI
149*e7be843bSPierre Pronchery ADD_ALL_TESTS(test_brotli, NUM_SIZES * 4);
150*e7be843bSPierre Pronchery #endif
151*e7be843bSPierre Pronchery #ifndef OPENSSL_NO_ZSTD
152*e7be843bSPierre Pronchery ADD_ALL_TESTS(test_zstd, NUM_SIZES * 4);
153*e7be843bSPierre Pronchery #endif
154*e7be843bSPierre Pronchery return 1;
155*e7be843bSPierre Pronchery }
156