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