1 /*
2 * Copyright 2017-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 "../testutil.h"
11 #include "output.h"
12 #include "tu_local.h"
13
14 #include <openssl/crypto.h>
15 #include <openssl/bio.h>
16
17 /* These are available for any test program */
18 BIO *bio_out = NULL;
19 BIO *bio_err = NULL;
20
21 /* These are available for TAP output only (internally) */
22 static BIO *tap_out = NULL;
23 static BIO *tap_err = NULL;
24
test_open_streams(void)25 void test_open_streams(void)
26 {
27 char prefix[] = "# ";
28 tap_out = BIO_new_fp(stdout, BIO_NOCLOSE | BIO_FP_TEXT);
29 tap_err = BIO_new_fp(stderr, BIO_NOCLOSE | BIO_FP_TEXT);
30 #ifdef __VMS
31 tap_out = BIO_push(BIO_new(BIO_f_linebuffer()), tap_out);
32 tap_err = BIO_push(BIO_new(BIO_f_linebuffer()), tap_err);
33 #endif
34 tap_out = BIO_push(BIO_new(BIO_f_prefix()), tap_out);
35 tap_err = BIO_push(BIO_new(BIO_f_prefix()), tap_err);
36
37 bio_out = BIO_push(BIO_new(BIO_f_prefix()), tap_out);
38 bio_err = BIO_push(BIO_new(BIO_f_prefix()), tap_err);
39 BIO_set_prefix(bio_out, prefix);
40 BIO_set_prefix(bio_err, prefix);
41
42 OPENSSL_assert(bio_out != NULL);
43 OPENSSL_assert(bio_err != NULL);
44 }
45
test_adjust_streams_tap_level(int level)46 void test_adjust_streams_tap_level(int level)
47 {
48 BIO_set_indent(tap_out, level);
49 BIO_set_indent(tap_err, level);
50 }
51
test_close_streams(void)52 void test_close_streams(void)
53 {
54 /*
55 * The rest of the chain is freed by the BIO_free_all() calls below, so
56 * we only need to free the last one in the bio_out and bio_err chains.
57 */
58 BIO_free(bio_out);
59 BIO_free(bio_err);
60
61 BIO_free_all(tap_out);
62 BIO_free_all(tap_err);
63 }
64
test_vprintf_stdout(const char * fmt,va_list ap)65 int test_vprintf_stdout(const char *fmt, va_list ap)
66 {
67 return BIO_vprintf(bio_out, fmt, ap);
68 }
69
test_vprintf_stderr(const char * fmt,va_list ap)70 int test_vprintf_stderr(const char *fmt, va_list ap)
71 {
72 return BIO_vprintf(bio_err, fmt, ap);
73 }
74
test_flush_stdout(void)75 int test_flush_stdout(void)
76 {
77 return BIO_flush(bio_out);
78 }
79
test_flush_stderr(void)80 int test_flush_stderr(void)
81 {
82 return BIO_flush(bio_err);
83 }
84
test_vprintf_tapout(const char * fmt,va_list ap)85 int test_vprintf_tapout(const char *fmt, va_list ap)
86 {
87 return BIO_vprintf(tap_out, fmt, ap);
88 }
89
test_vprintf_taperr(const char * fmt,va_list ap)90 int test_vprintf_taperr(const char *fmt, va_list ap)
91 {
92 return BIO_vprintf(tap_err, fmt, ap);
93 }
94
test_flush_tapout(void)95 int test_flush_tapout(void)
96 {
97 return BIO_flush(tap_out);
98 }
99
test_flush_taperr(void)100 int test_flush_taperr(void)
101 {
102 return BIO_flush(tap_err);
103 }
104