/src/openssl/crypto/trace.c

Source
/*
 * Copyright 2019-2025 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (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
 */

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

#include "internal/thread_once.h"
#include <openssl/bio.h>
#include <openssl/crypto.h>
#include <openssl/trace.h>
#include "internal/bio.h"
#include "internal/nelem.h"
#include "internal/refcount.h"
#include "crypto/cryptlib.h"
#include "crypto/ctype.h"

#ifndef OPENSSL_NO_TRACE

static CRYPTO_RWLOCK *trace_lock = NULL;

static const BIO *current_channel = NULL;

/*-
 * INTERNAL TRACE CHANNEL IMPLEMENTATION
 *
 * For our own flexibility, all trace categories are associated with a
 * BIO sink object, also called the trace channel. Instead of a BIO object,
 * the application can also provide a callback function, in which case an
 * internal trace channel is attached, which simply calls the registered
 * callback function.
 */
static int trace_write(BIO *b, const char *buf,
    size_t num, size_t *written);
static int trace_puts(BIO *b, const char *str);
static long trace_ctrl(BIO *channel, int cmd, long argl, void *argp);
static int trace_free(BIO *b);

static const BIO_METHOD trace_method = {
    BIO_TYPE_SOURCE_SINK,
    "trace",
    trace_write,
    NULL, /* old write */
    NULL, /* read_ex */
    NULL, /* read */
    trace_puts,
    NULL, /* gets */
    trace_ctrl, /* ctrl */
    NULL, /* create */
    trace_free, /* free */
    NULL, /* callback_ctrl */
};

struct trace_data_st {
    OSSL_trace_cb callback;
    int category;
    void *data;
};

static int trace_write(BIO *channel,
    const char *buf, size_t num, size_t *written)
{
    struct trace_data_st *ctx = BIO_get_data(channel);
    size_t cnt = ctx->callback(buf, num, ctx->category, OSSL_TRACE_CTRL_WRITE,
        ctx->data);

    *written = cnt;
    return cnt != 0;
}

static int trace_puts(BIO *channel, const char *str)
{
    size_t written;

    if (trace_write(channel, str, strlen(str), &written))
        return (int)written;

    return EOF;
}

static long trace_ctrl(BIO *channel, int cmd, long argl, void *argp)
{
    struct trace_data_st *ctx = BIO_get_data(channel);

    switch (cmd) {
    case OSSL_TRACE_CTRL_BEGIN:
    case OSSL_TRACE_CTRL_END:
        /* We know that the callback is likely to return 0 here */
        ctx->callback("", 0, ctx->category, cmd, ctx->data);
        return 1;
    default:
        break;
    }
    return -2; /* Unsupported */
}

static int trace_free(BIO *channel)
{
    if (channel == NULL)
        return 0;
    OPENSSL_free(BIO_get_data(channel));
    return 1;
}
#endif

/*-
 * TRACE
 */

/* Helper struct and macro to get name string to number mapping */
struct trace_category_st {
    const char *const name;
    const int num;
};
#define TRACE_CATEGORY_(name) { #name, OSSL_TRACE_CATEGORY_##name }

static const struct trace_category_st
    trace_categories[OSSL_TRACE_CATEGORY_NUM]
    = {
          TRACE_CATEGORY_(ALL),
          TRACE_CATEGORY_(TRACE),
          TRACE_CATEGORY_(INIT),
          TRACE_CATEGORY_(TLS),
          TRACE_CATEGORY_(TLS_CIPHER),
          TRACE_CATEGORY_(CONF),
          TRACE_CATEGORY_(PKCS5V2),
          TRACE_CATEGORY_(PKCS12_KEYGEN),
          TRACE_CATEGORY_(PKCS12_DECRYPT),
          TRACE_CATEGORY_(X509V3_POLICY),
          TRACE_CATEGORY_(BN_CTX),
          TRACE_CATEGORY_(CMP),
          TRACE_CATEGORY_(STORE),
          TRACE_CATEGORY_(DECODER),
          TRACE_CATEGORY_(ENCODER),
          TRACE_CATEGORY_(REF_COUNT),
          TRACE_CATEGORY_(HTTP),
          TRACE_CATEGORY_(PROVIDER),
          TRACE_CATEGORY_(QUERY),
      }; /* KEEP THIS LIST IN SYNC with #define OSSL_TRACE_CATEGORY_... in trace.h */

const char *OSSL_trace_get_category_name(int num)
{
    if (num < 0 || (size_t)num >= OSSL_NELEM(trace_categories))
        return NULL;
    /*
     * Partial check that OSSL_TRACE_CATEGORY_... macros
     * are synced with trace_categories array
     */
    if (!ossl_assert(trace_categories[num].name != NULL)
        || !ossl_assert(trace_categories[num].num == num))
        return NULL;
    return trace_categories[num].name;
}

int OSSL_trace_get_category_num(const char *name)
{
    size_t i;

    if (name == NULL)
        return -1;

    for (i = 0; i < OSSL_NELEM(trace_categories); i++)
        if (OPENSSL_strcasecmp(name, trace_categories[i].name) == 0)
            return trace_categories[i].num;

    return -1; /* not found */
}

#ifndef OPENSSL_NO_TRACE

/* We use one trace channel for each trace category */
static struct {
    enum { SIMPLE_CHANNEL,
        CALLBACK_CHANNEL } type;
    BIO *bio;
    char *prefix;
    char *suffix;
} trace_channels[OSSL_TRACE_CATEGORY_NUM] = {
    { 0, NULL, NULL, NULL },
};

#endif

#ifndef OPENSSL_NO_TRACE

enum {
    CHANNEL,
    PREFIX,
    SUFFIX
};

static int trace_attach_cb(int category, int type, const void *data)
{
    switch (type) {
    case CHANNEL:
        OSSL_TRACE2(TRACE, "Attach channel %p to category '%s'\n",
            data, trace_categories[category].name);
        break;
    case PREFIX:
        OSSL_TRACE2(TRACE, "Attach prefix \"%s\" to category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    case SUFFIX:
        OSSL_TRACE2(TRACE, "Attach suffix \"%s\" to category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    default: /* No clue */
        break;
    }
    return 1;
}

static int trace_detach_cb(int category, int type, const void *data)
{
    switch (type) {
    case CHANNEL:
        OSSL_TRACE2(TRACE, "Detach channel %p from category '%s'\n",
            data, trace_categories[category].name);
        break;
    case PREFIX:
        OSSL_TRACE2(TRACE, "Detach prefix \"%s\" from category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    case SUFFIX:
        OSSL_TRACE2(TRACE, "Detach suffix \"%s\" from category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    default: /* No clue */
        break;
    }
    return 1;
}

static int do_ossl_trace_init(void);
static CRYPTO_ONCE trace_inited = CRYPTO_ONCE_STATIC_INIT;
DEFINE_RUN_ONCE_STATIC(ossl_trace_init)
{
    return do_ossl_trace_init();
}

static int set_trace_data(int category, int type, BIO **channel,
    const char **prefix, const char **suffix,
    int (*attach_cb)(int, int, const void *),
    int (*detach_cb)(int, int, const void *))
{
    BIO *curr_channel = NULL;
    char *curr_prefix = NULL;
    char *curr_suffix = NULL;

    /* Ensure do_ossl_trace_init() is called once */
    if (!RUN_ONCE(&trace_inited, ossl_trace_init))
        return 0;

    curr_channel = trace_channels[category].bio;
    curr_prefix = trace_channels[category].prefix;
    curr_suffix = trace_channels[category].suffix;

    /* Make sure to run the detach callback first on all data */
    if (prefix != NULL && curr_prefix != NULL) {
        detach_cb(category, PREFIX, curr_prefix);
    }

    if (suffix != NULL && curr_suffix != NULL) {
        detach_cb(category, SUFFIX, curr_suffix);
    }

    if (channel != NULL && curr_channel != NULL) {
        detach_cb(category, CHANNEL, curr_channel);
    }

    /* After detach callbacks are done, clear data where appropriate */
    if (prefix != NULL && curr_prefix != NULL) {
        OPENSSL_free(curr_prefix);
        trace_channels[category].prefix = NULL;
    }

    if (suffix != NULL && curr_suffix != NULL) {
        OPENSSL_free(curr_suffix);
        trace_channels[category].suffix = NULL;
    }

    if (channel != NULL && curr_channel != NULL) {
        BIO_free(curr_channel);
        trace_channels[category].type = 0;
        trace_channels[category].bio = NULL;
    }

    /* Before running callbacks are done, set new data where appropriate */
    if (prefix != NULL && *prefix != NULL) {
        if ((curr_prefix = OPENSSL_strdup(*prefix)) == NULL)
            return 0;
        trace_channels[category].prefix = curr_prefix;
    }

    if (suffix != NULL && *suffix != NULL) {
        if ((curr_suffix = OPENSSL_strdup(*suffix)) == NULL)
            return 0;
        trace_channels[category].suffix = curr_suffix;
    }

    if (channel != NULL && *channel != NULL) {
        trace_channels[category].type = type;
        trace_channels[category].bio = *channel;
        /*
         * This must not be done before setting prefix/suffix,
         * as those may fail, and then the caller is mislead to free *channel.
         */
    }

    /* Finally, run the attach callback on the new data */
    if (channel != NULL && *channel != NULL) {
        attach_cb(category, CHANNEL, *channel);
    }

    if (prefix != NULL && *prefix != NULL) {
        attach_cb(category, PREFIX, *prefix);
    }

    if (suffix != NULL && *suffix != NULL) {
        attach_cb(category, SUFFIX, *suffix);
    }

    return 1;
}

static int do_ossl_trace_init(void)
{
    trace_lock = CRYPTO_THREAD_lock_new();
    return trace_lock != NULL;
}

#endif

void ossl_trace_cleanup(void)
{
#ifndef OPENSSL_NO_TRACE
    int category;
    BIO *channel = NULL;
    const char *prefix = NULL;
    const char *suffix = NULL;

    for (category = 0; category < OSSL_TRACE_CATEGORY_NUM; category++) {
        /* We force the TRACE category to be treated last */
        if (category == OSSL_TRACE_CATEGORY_TRACE)
            continue;
        set_trace_data(category, 0, &channel, &prefix, &suffix,
            trace_attach_cb, trace_detach_cb);
    }
    set_trace_data(OSSL_TRACE_CATEGORY_TRACE, 0, &channel,
        &prefix, &suffix,
        trace_attach_cb, trace_detach_cb);
    CRYPTO_THREAD_lock_free(trace_lock);
#endif
}

int OSSL_trace_set_channel(int category, BIO *channel)
{
#ifndef OPENSSL_NO_TRACE
    if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
        return set_trace_data(category, SIMPLE_CHANNEL, &channel, NULL, NULL,
            trace_attach_cb, trace_detach_cb);
#endif
    return 0;
}

#ifndef OPENSSL_NO_TRACE
static int trace_attach_w_callback_cb(int category, int type, const void *data)
{
    switch (type) {
    case CHANNEL:
        OSSL_TRACE2(TRACE,
            "Attach channel %p to category '%s' (with callback)\n",
            data, trace_categories[category].name);
        break;
    case PREFIX:
        OSSL_TRACE2(TRACE, "Attach prefix \"%s\" to category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    case SUFFIX:
        OSSL_TRACE2(TRACE, "Attach suffix \"%s\" to category '%s'\n",
            (const char *)data, trace_categories[category].name);
        break;
    default: /* No clue */
        break;
    }
    return 1;
}
#endif

int OSSL_trace_set_callback(int category, OSSL_trace_cb callback, void *data)
{
#ifndef OPENSSL_NO_TRACE
    BIO *channel = NULL;
    struct trace_data_st *trace_data = NULL;

    if (category < 0 || category >= OSSL_TRACE_CATEGORY_NUM)
        return 0;

    if (callback != NULL) {
        if ((channel = BIO_new(&trace_method)) == NULL
            || (trace_data = OPENSSL_zalloc(sizeof(struct trace_data_st))) == NULL)
            goto err;

        trace_data->callback = callback;
        trace_data->category = category;
        trace_data->data = data;

        BIO_set_data(channel, trace_data);
    }

    if (!set_trace_data(category, CALLBACK_CHANNEL, &channel, NULL, NULL,
            trace_attach_w_callback_cb, trace_detach_cb))
        goto err;

    return 1;

err:
    BIO_free(channel);
    OPENSSL_free(trace_data);
#endif

    return 0;
}

int OSSL_trace_set_prefix(int category, const char *prefix)
{
#ifndef OPENSSL_NO_TRACE
    if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
        return set_trace_data(category, 0, NULL, &prefix, NULL,
            trace_attach_cb, trace_detach_cb);
#endif
    return 0;
}

int OSSL_trace_set_suffix(int category, const char *suffix)
{
#ifndef OPENSSL_NO_TRACE
    if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
        return set_trace_data(category, 0, NULL, NULL, &suffix,
            trace_attach_cb, trace_detach_cb);
#endif
    return 0;
}

#ifndef OPENSSL_NO_TRACE
static int ossl_trace_get_category(int category)
{
    if (category < 0 || category >= OSSL_TRACE_CATEGORY_NUM)
        return -1;
    if (trace_channels[category].bio != NULL)
        return category;
    return OSSL_TRACE_CATEGORY_ALL;
}
#endif

int OSSL_trace_enabled(int category)
{
    int ret = 0;
#ifndef OPENSSL_NO_TRACE
    category = ossl_trace_get_category(category);
    if (category >= 0)
        ret = trace_channels[category].bio != NULL;
#endif
    return ret;
}

BIO *OSSL_trace_begin(int category)
{
    BIO *channel = NULL;
#ifndef OPENSSL_NO_TRACE
    char *prefix = NULL;

    category = ossl_trace_get_category(category);
    if (category < 0 || !OSSL_trace_enabled(category))
        return NULL;

    channel = trace_channels[category].bio;
    prefix = trace_channels[category].prefix;

    if (channel != NULL) {
        if (!CRYPTO_THREAD_write_lock(trace_lock))
            return NULL;
        current_channel = channel;
        switch (trace_channels[category].type) {
        case SIMPLE_CHANNEL:
            if (prefix != NULL) {
                (void)BIO_puts(channel, prefix);
                (void)BIO_puts(channel, "\n");
            }
            break;
        case CALLBACK_CHANNEL:
            (void)BIO_ctrl(channel, OSSL_TRACE_CTRL_BEGIN,
                prefix == NULL ? 0 : (int)strlen(prefix), prefix);
            break;
        }
    }
#endif
    return channel;
}

void OSSL_trace_end(int category, BIO *channel)
{
#ifndef OPENSSL_NO_TRACE
    char *suffix = NULL;

    category = ossl_trace_get_category(category);
    if (category < 0)
        return;
    suffix = trace_channels[category].suffix;
    if (channel != NULL
        && ossl_assert(channel == current_channel)) {
        (void)BIO_flush(channel);
        switch (trace_channels[category].type) {
        case SIMPLE_CHANNEL:
            if (suffix != NULL) {
                (void)BIO_puts(channel, suffix);
                (void)BIO_puts(channel, "\n");
            }
            break;
        case CALLBACK_CHANNEL:
            (void)BIO_ctrl(channel, OSSL_TRACE_CTRL_END,
                suffix == NULL ? 0 : (int)strlen(suffix), suffix);
            break;
        }
        current_channel = NULL;
        CRYPTO_THREAD_unlock(trace_lock);
    }
#endif
}

int OSSL_trace_string(BIO *out, int text, int full,
    const unsigned char *data, size_t size)
{
    unsigned char buf[OSSL_TRACE_STRING_MAX + 1];
    int len, i;

    if (!full && size > OSSL_TRACE_STRING_MAX) {
        BIO_printf(out, "[len %zu limited to %d]: ",
            size, OSSL_TRACE_STRING_MAX);
        len = OSSL_TRACE_STRING_MAX;
    } else {
        len = (int)size;
    }
    if (!text) { /* mask control characters while preserving newlines */
        for (i = 0; i < len; i++, data++)
            buf[i] = (char)*data != '\n' && ossl_iscntrl((int)*data)
                ? ' '
                : *data;
        if (len == 0 || data[-1] != '\n')
            buf[len++] = '\n';
        data = buf;
    }
    return BIO_printf(out, "%.*s", len, data);
}

Coverage Report

Created: 2025-12-10 06:24

Line	Count	Source
1		/*
2		* Copyright 2019-2025 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 <stdio.h>
11		#include <string.h>
12
13		#include "internal/thread_once.h"
14		#include <openssl/bio.h>
15		#include <openssl/crypto.h>
16		#include <openssl/trace.h>
17		#include "internal/bio.h"
18		#include "internal/nelem.h"
19		#include "internal/refcount.h"
20		#include "crypto/cryptlib.h"
21		#include "crypto/ctype.h"
22
23		#ifndef OPENSSL_NO_TRACE
24
25		static CRYPTO_RWLOCK *trace_lock = NULL;
26
27		static const BIO *current_channel = NULL;
28
29		/*-
30		* INTERNAL TRACE CHANNEL IMPLEMENTATION
31		*
32		* For our own flexibility, all trace categories are associated with a
33		* BIO sink object, also called the trace channel. Instead of a BIO object,
34		* the application can also provide a callback function, in which case an
35		* internal trace channel is attached, which simply calls the registered
36		* callback function.
37		*/
38		static int trace_write(BIO b, const char buf,
39		size_t num, size_t *written);
40		static int trace_puts(BIO b, const char str);
41		static long trace_ctrl(BIO channel, int cmd, long argl, void argp);
42		static int trace_free(BIO *b);
43
44		static const BIO_METHOD trace_method = {
45		BIO_TYPE_SOURCE_SINK,
46		"trace",
47		trace_write,
48		NULL, /* old write */
49		NULL, /* read_ex */
50		NULL, /* read */
51		trace_puts,
52		NULL, /* gets */
53		trace_ctrl, /* ctrl */
54		NULL, /* create */
55		trace_free, /* free */
56		NULL, /* callback_ctrl */
57		};
58
59		struct trace_data_st {
60		OSSL_trace_cb callback;
61		int category;
62		void *data;
63		};
64
65		static int trace_write(BIO *channel,
66		const char buf, size_t num, size_t written)
67		{
68		struct trace_data_st *ctx = BIO_get_data(channel);
69		size_t cnt = ctx->callback(buf, num, ctx->category, OSSL_TRACE_CTRL_WRITE,
70		ctx->data);
71
72		*written = cnt;
73		return cnt != 0;
74		}
75
76		static int trace_puts(BIO channel, const char str)
77		{
78		size_t written;
79
80		if (trace_write(channel, str, strlen(str), &written))
81		return (int)written;
82
83		return EOF;
84		}
85
86		static long trace_ctrl(BIO channel, int cmd, long argl, void argp)
87		{
88		struct trace_data_st *ctx = BIO_get_data(channel);
89
90		switch (cmd) {
91		case OSSL_TRACE_CTRL_BEGIN:
92		case OSSL_TRACE_CTRL_END:
93		/* We know that the callback is likely to return 0 here */
94		ctx->callback("", 0, ctx->category, cmd, ctx->data);
95		return 1;
96		default:
97		break;
98		}
99		return -2; /* Unsupported */
100		}
101
102		static int trace_free(BIO *channel)
103		{
104		if (channel == NULL)
105		return 0;
106		OPENSSL_free(BIO_get_data(channel));
107		return 1;
108		}
109		#endif
110
111		/*-
112		* TRACE
113		*/
114
115		/* Helper struct and macro to get name string to number mapping */
116		struct trace_category_st {
117		const char *const name;
118		const int num;
119		};
120		#define TRACE_CATEGORY_(name) { #name, OSSL_TRACE_CATEGORY_##name }
121
122		static const struct trace_category_st
123		trace_categories[OSSL_TRACE_CATEGORY_NUM]
124		= {
125		TRACE_CATEGORY_(ALL),
126		TRACE_CATEGORY_(TRACE),
127		TRACE_CATEGORY_(INIT),
128		TRACE_CATEGORY_(TLS),
129		TRACE_CATEGORY_(TLS_CIPHER),
130		TRACE_CATEGORY_(CONF),
131		TRACE_CATEGORY_(PKCS5V2),
132		TRACE_CATEGORY_(PKCS12_KEYGEN),
133		TRACE_CATEGORY_(PKCS12_DECRYPT),
134		TRACE_CATEGORY_(X509V3_POLICY),
135		TRACE_CATEGORY_(BN_CTX),
136		TRACE_CATEGORY_(CMP),
137		TRACE_CATEGORY_(STORE),
138		TRACE_CATEGORY_(DECODER),
139		TRACE_CATEGORY_(ENCODER),
140		TRACE_CATEGORY_(REF_COUNT),
141		TRACE_CATEGORY_(HTTP),
142		TRACE_CATEGORY_(PROVIDER),
143		TRACE_CATEGORY_(QUERY),
144		}; /* KEEP THIS LIST IN SYNC with #define OSSL_TRACE_CATEGORY_... in trace.h */
145
146		const char *OSSL_trace_get_category_name(int num)
147	0	{
148	0	if (num < 0 \|\| (size_t)num >= OSSL_NELEM(trace_categories))
149	0	return NULL;
150		/*
151		* Partial check that OSSL_TRACE_CATEGORY_... macros
152		* are synced with trace_categories array
153		*/
154	0	if (!ossl_assert(trace_categories[num].name != NULL)
155	0	\|\| !ossl_assert(trace_categories[num].num == num))
156	0	return NULL;
157	0	return trace_categories[num].name;
158	0	}
159
160		int OSSL_trace_get_category_num(const char *name)
161	0	{
162	0	size_t i;
163
164	0	if (name == NULL)
165	0	return -1;
166
167	0	for (i = 0; i < OSSL_NELEM(trace_categories); i++)
168	0	if (OPENSSL_strcasecmp(name, trace_categories[i].name) == 0)
169	0	return trace_categories[i].num;
170
171	0	return -1; /* not found */
172	0	}
173
174		#ifndef OPENSSL_NO_TRACE
175
176		/* We use one trace channel for each trace category */
177		static struct {
178		enum { SIMPLE_CHANNEL,
179		CALLBACK_CHANNEL } type;
180		BIO *bio;
181		char *prefix;
182		char *suffix;
183		} trace_channels[OSSL_TRACE_CATEGORY_NUM] = {
184		{ 0, NULL, NULL, NULL },
185		};
186
187		#endif
188
189		#ifndef OPENSSL_NO_TRACE
190
191		enum {
192		CHANNEL,
193		PREFIX,
194		SUFFIX
195		};
196
197		static int trace_attach_cb(int category, int type, const void *data)
198		{
199		switch (type) {
200		case CHANNEL:
201		OSSL_TRACE2(TRACE, "Attach channel %p to category '%s'\n",
202		data, trace_categories[category].name);
203		break;
204		case PREFIX:
205		OSSL_TRACE2(TRACE, "Attach prefix \"%s\" to category '%s'\n",
206		(const char *)data, trace_categories[category].name);
207		break;
208		case SUFFIX:
209		OSSL_TRACE2(TRACE, "Attach suffix \"%s\" to category '%s'\n",
210		(const char *)data, trace_categories[category].name);
211		break;
212		default: /* No clue */
213		break;
214		}
215		return 1;
216		}
217
218		static int trace_detach_cb(int category, int type, const void *data)
219		{
220		switch (type) {
221		case CHANNEL:
222		OSSL_TRACE2(TRACE, "Detach channel %p from category '%s'\n",
223		data, trace_categories[category].name);
224		break;
225		case PREFIX:
226		OSSL_TRACE2(TRACE, "Detach prefix \"%s\" from category '%s'\n",
227		(const char *)data, trace_categories[category].name);
228		break;
229		case SUFFIX:
230		OSSL_TRACE2(TRACE, "Detach suffix \"%s\" from category '%s'\n",
231		(const char *)data, trace_categories[category].name);
232		break;
233		default: /* No clue */
234		break;
235		}
236		return 1;
237		}
238
239		static int do_ossl_trace_init(void);
240		static CRYPTO_ONCE trace_inited = CRYPTO_ONCE_STATIC_INIT;
241		DEFINE_RUN_ONCE_STATIC(ossl_trace_init)
242		{
243		return do_ossl_trace_init();
244		}
245
246		static int set_trace_data(int category, int type, BIO **channel,
247		const char prefix, const char suffix,
248		int (attach_cb)(int, int, const void ),
249		int (detach_cb)(int, int, const void ))
250		{
251		BIO *curr_channel = NULL;
252		char *curr_prefix = NULL;
253		char *curr_suffix = NULL;
254
255		/* Ensure do_ossl_trace_init() is called once */
256		if (!RUN_ONCE(&trace_inited, ossl_trace_init))
257		return 0;
258
259		curr_channel = trace_channels[category].bio;
260		curr_prefix = trace_channels[category].prefix;
261		curr_suffix = trace_channels[category].suffix;
262
263		/* Make sure to run the detach callback first on all data */
264		if (prefix != NULL && curr_prefix != NULL) {
265		detach_cb(category, PREFIX, curr_prefix);
266		}
267
268		if (suffix != NULL && curr_suffix != NULL) {
269		detach_cb(category, SUFFIX, curr_suffix);
270		}
271
272		if (channel != NULL && curr_channel != NULL) {
273		detach_cb(category, CHANNEL, curr_channel);
274		}
275
276		/* After detach callbacks are done, clear data where appropriate */
277		if (prefix != NULL && curr_prefix != NULL) {
278		OPENSSL_free(curr_prefix);
279		trace_channels[category].prefix = NULL;
280		}
281
282		if (suffix != NULL && curr_suffix != NULL) {
283		OPENSSL_free(curr_suffix);
284		trace_channels[category].suffix = NULL;
285		}
286
287		if (channel != NULL && curr_channel != NULL) {
288		BIO_free(curr_channel);
289		trace_channels[category].type = 0;
290		trace_channels[category].bio = NULL;
291		}
292
293		/* Before running callbacks are done, set new data where appropriate */
294		if (prefix != NULL && *prefix != NULL) {
295		if ((curr_prefix = OPENSSL_strdup(*prefix)) == NULL)
296		return 0;
297		trace_channels[category].prefix = curr_prefix;
298		}
299
300		if (suffix != NULL && *suffix != NULL) {
301		if ((curr_suffix = OPENSSL_strdup(*suffix)) == NULL)
302		return 0;
303		trace_channels[category].suffix = curr_suffix;
304		}
305
306		if (channel != NULL && *channel != NULL) {
307		trace_channels[category].type = type;
308		trace_channels[category].bio = *channel;
309		/*
310		* This must not be done before setting prefix/suffix,
311		* as those may fail, and then the caller is mislead to free *channel.
312		*/
313		}
314
315		/* Finally, run the attach callback on the new data */
316		if (channel != NULL && *channel != NULL) {
317		attach_cb(category, CHANNEL, *channel);
318		}
319
320		if (prefix != NULL && *prefix != NULL) {
321		attach_cb(category, PREFIX, *prefix);
322		}
323
324		if (suffix != NULL && *suffix != NULL) {
325		attach_cb(category, SUFFIX, *suffix);
326		}
327
328		return 1;
329		}
330
331		static int do_ossl_trace_init(void)
332		{
333		trace_lock = CRYPTO_THREAD_lock_new();
334		return trace_lock != NULL;
335		}
336
337		#endif
338
339		void ossl_trace_cleanup(void)
340	3	{
341		#ifndef OPENSSL_NO_TRACE
342		int category;
343		BIO *channel = NULL;
344		const char *prefix = NULL;
345		const char *suffix = NULL;
346
347		for (category = 0; category < OSSL_TRACE_CATEGORY_NUM; category++) {
348		/* We force the TRACE category to be treated last */
349		if (category == OSSL_TRACE_CATEGORY_TRACE)
350		continue;
351		set_trace_data(category, 0, &channel, &prefix, &suffix,
352		trace_attach_cb, trace_detach_cb);
353		}
354		set_trace_data(OSSL_TRACE_CATEGORY_TRACE, 0, &channel,
355		&prefix, &suffix,
356		trace_attach_cb, trace_detach_cb);
357		CRYPTO_THREAD_lock_free(trace_lock);
358		#endif
359	3	}
360
361		int OSSL_trace_set_channel(int category, BIO *channel)
362	3	{
363		#ifndef OPENSSL_NO_TRACE
364		if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
365		return set_trace_data(category, SIMPLE_CHANNEL, &channel, NULL, NULL,
366		trace_attach_cb, trace_detach_cb);
367		#endif
368	3	return 0;
369	3	}
370
371		#ifndef OPENSSL_NO_TRACE
372		static int trace_attach_w_callback_cb(int category, int type, const void *data)
373		{
374		switch (type) {
375		case CHANNEL:
376		OSSL_TRACE2(TRACE,
377		"Attach channel %p to category '%s' (with callback)\n",
378		data, trace_categories[category].name);
379		break;
380		case PREFIX:
381		OSSL_TRACE2(TRACE, "Attach prefix \"%s\" to category '%s'\n",
382		(const char *)data, trace_categories[category].name);
383		break;
384		case SUFFIX:
385		OSSL_TRACE2(TRACE, "Attach suffix \"%s\" to category '%s'\n",
386		(const char *)data, trace_categories[category].name);
387		break;
388		default: /* No clue */
389		break;
390		}
391		return 1;
392		}
393		#endif
394
395		int OSSL_trace_set_callback(int category, OSSL_trace_cb callback, void *data)
396	0	{
397		#ifndef OPENSSL_NO_TRACE
398		BIO *channel = NULL;
399		struct trace_data_st *trace_data = NULL;
400
401		if (category < 0 \|\| category >= OSSL_TRACE_CATEGORY_NUM)
402		return 0;
403
404		if (callback != NULL) {
405		if ((channel = BIO_new(&trace_method)) == NULL
406		\|\| (trace_data = OPENSSL_zalloc(sizeof(struct trace_data_st))) == NULL)
407		goto err;
408
409		trace_data->callback = callback;
410		trace_data->category = category;
411		trace_data->data = data;
412
413		BIO_set_data(channel, trace_data);
414		}
415
416		if (!set_trace_data(category, CALLBACK_CHANNEL, &channel, NULL, NULL,
417		trace_attach_w_callback_cb, trace_detach_cb))
418		goto err;
419
420		return 1;
421
422		err:
423		BIO_free(channel);
424		OPENSSL_free(trace_data);
425		#endif
426
427	0	return 0;
428	0	}
429
430		int OSSL_trace_set_prefix(int category, const char *prefix)
431	0	{
432		#ifndef OPENSSL_NO_TRACE
433		if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
434		return set_trace_data(category, 0, NULL, &prefix, NULL,
435		trace_attach_cb, trace_detach_cb);
436		#endif
437	0	return 0;
438	0	}
439
440		int OSSL_trace_set_suffix(int category, const char *suffix)
441	0	{
442		#ifndef OPENSSL_NO_TRACE
443		if (category >= 0 && category < OSSL_TRACE_CATEGORY_NUM)
444		return set_trace_data(category, 0, NULL, NULL, &suffix,
445		trace_attach_cb, trace_detach_cb);
446		#endif
447	0	return 0;
448	0	}
449
450		#ifndef OPENSSL_NO_TRACE
451		static int ossl_trace_get_category(int category)
452		{
453		if (category < 0 \|\| category >= OSSL_TRACE_CATEGORY_NUM)
454		return -1;
455		if (trace_channels[category].bio != NULL)
456		return category;
457		return OSSL_TRACE_CATEGORY_ALL;
458		}
459		#endif
460
461		int OSSL_trace_enabled(int category)
462	0	{
463	0	int ret = 0;
464		#ifndef OPENSSL_NO_TRACE
465		category = ossl_trace_get_category(category);
466		if (category >= 0)
467		ret = trace_channels[category].bio != NULL;
468		#endif
469	0	return ret;
470	0	}
471
472		BIO *OSSL_trace_begin(int category)
473	0	{
474	0	BIO *channel = NULL;
475		#ifndef OPENSSL_NO_TRACE
476		char *prefix = NULL;
477
478		category = ossl_trace_get_category(category);
479		if (category < 0 \|\| !OSSL_trace_enabled(category))
480		return NULL;
481
482		channel = trace_channels[category].bio;
483		prefix = trace_channels[category].prefix;
484
485		if (channel != NULL) {
486		if (!CRYPTO_THREAD_write_lock(trace_lock))
487		return NULL;
488		current_channel = channel;
489		switch (trace_channels[category].type) {
490		case SIMPLE_CHANNEL:
491		if (prefix != NULL) {
492		(void)BIO_puts(channel, prefix);
493		(void)BIO_puts(channel, "\n");
494		}
495		break;
496		case CALLBACK_CHANNEL:
497		(void)BIO_ctrl(channel, OSSL_TRACE_CTRL_BEGIN,
498		prefix == NULL ? 0 : (int)strlen(prefix), prefix);
499		break;
500		}
501		}
502		#endif
503	0	return channel;
504	0	}
505
506		void OSSL_trace_end(int category, BIO *channel)
507	0	{
508		#ifndef OPENSSL_NO_TRACE
509		char *suffix = NULL;
510
511		category = ossl_trace_get_category(category);
512		if (category < 0)
513		return;
514		suffix = trace_channels[category].suffix;
515		if (channel != NULL
516		&& ossl_assert(channel == current_channel)) {
517		(void)BIO_flush(channel);
518		switch (trace_channels[category].type) {
519		case SIMPLE_CHANNEL:
520		if (suffix != NULL) {
521		(void)BIO_puts(channel, suffix);
522		(void)BIO_puts(channel, "\n");
523		}
524		break;
525		case CALLBACK_CHANNEL:
526		(void)BIO_ctrl(channel, OSSL_TRACE_CTRL_END,
527		suffix == NULL ? 0 : (int)strlen(suffix), suffix);
528		break;
529		}
530		current_channel = NULL;
531		CRYPTO_THREAD_unlock(trace_lock);
532		}
533		#endif
534	0	}
535
536		int OSSL_trace_string(BIO *out, int text, int full,
537		const unsigned char *data, size_t size)
538	0	{
539	0	unsigned char buf[OSSL_TRACE_STRING_MAX + 1];
540	0	int len, i;
541
542	0	if (!full && size > OSSL_TRACE_STRING_MAX) {
543	0	BIO_printf(out, "[len %zu limited to %d]: ",
544	0	size, OSSL_TRACE_STRING_MAX);
545	0	len = OSSL_TRACE_STRING_MAX;
546	0	} else {
547	0	len = (int)size;
548	0	}
549	0	if (!text) { /* mask control characters while preserving newlines */
550	0	for (i = 0; i < len; i++, data++)
551	0	buf[i] = (char)data != '\n' && ossl_iscntrl((int)data)
552	0	? ' '
553	0	: *data;
554	0	if (len == 0 \|\| data[-1] != '\n')
555	0	buf[len++] = '\n';
556	0	data = buf;
557	0	}
558	0	return BIO_printf(out, "%.*s", len, data);
559	0	}