1*62c56f98SSadaf Ebrahimi /*
2*62c56f98SSadaf Ebrahimi * Message Processing Stack, Trace module
3*62c56f98SSadaf Ebrahimi *
4*62c56f98SSadaf Ebrahimi * Copyright The Mbed TLS Contributors
5*62c56f98SSadaf Ebrahimi * SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
6*62c56f98SSadaf Ebrahimi */
7*62c56f98SSadaf Ebrahimi
8*62c56f98SSadaf Ebrahimi #include "common.h"
9*62c56f98SSadaf Ebrahimi
10*62c56f98SSadaf Ebrahimi #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
11*62c56f98SSadaf Ebrahimi
12*62c56f98SSadaf Ebrahimi #include "mps_common.h"
13*62c56f98SSadaf Ebrahimi
14*62c56f98SSadaf Ebrahimi #if defined(MBEDTLS_MPS_ENABLE_TRACE)
15*62c56f98SSadaf Ebrahimi
16*62c56f98SSadaf Ebrahimi #include "mps_trace.h"
17*62c56f98SSadaf Ebrahimi #include <stdarg.h>
18*62c56f98SSadaf Ebrahimi
19*62c56f98SSadaf Ebrahimi static int trace_depth = 0;
20*62c56f98SSadaf Ebrahimi
21*62c56f98SSadaf Ebrahimi #define color_default "\x1B[0m"
22*62c56f98SSadaf Ebrahimi #define color_red "\x1B[1;31m"
23*62c56f98SSadaf Ebrahimi #define color_green "\x1B[1;32m"
24*62c56f98SSadaf Ebrahimi #define color_yellow "\x1B[1;33m"
25*62c56f98SSadaf Ebrahimi #define color_blue "\x1B[1;34m"
26*62c56f98SSadaf Ebrahimi #define color_magenta "\x1B[1;35m"
27*62c56f98SSadaf Ebrahimi #define color_cyan "\x1B[1;36m"
28*62c56f98SSadaf Ebrahimi #define color_white "\x1B[1;37m"
29*62c56f98SSadaf Ebrahimi
30*62c56f98SSadaf Ebrahimi static char const *colors[] =
31*62c56f98SSadaf Ebrahimi {
32*62c56f98SSadaf Ebrahimi color_default,
33*62c56f98SSadaf Ebrahimi color_green,
34*62c56f98SSadaf Ebrahimi color_yellow,
35*62c56f98SSadaf Ebrahimi color_magenta,
36*62c56f98SSadaf Ebrahimi color_cyan,
37*62c56f98SSadaf Ebrahimi color_blue,
38*62c56f98SSadaf Ebrahimi color_white
39*62c56f98SSadaf Ebrahimi };
40*62c56f98SSadaf Ebrahimi
41*62c56f98SSadaf Ebrahimi #define MPS_TRACE_BUF_SIZE 100
42*62c56f98SSadaf Ebrahimi
mbedtls_mps_trace_print_msg(int id,int line,const char * format,...)43*62c56f98SSadaf Ebrahimi void mbedtls_mps_trace_print_msg(int id, int line, const char *format, ...)
44*62c56f98SSadaf Ebrahimi {
45*62c56f98SSadaf Ebrahimi int ret;
46*62c56f98SSadaf Ebrahimi char str[MPS_TRACE_BUF_SIZE];
47*62c56f98SSadaf Ebrahimi va_list argp;
48*62c56f98SSadaf Ebrahimi va_start(argp, format);
49*62c56f98SSadaf Ebrahimi ret = mbedtls_vsnprintf(str, MPS_TRACE_BUF_SIZE, format, argp);
50*62c56f98SSadaf Ebrahimi va_end(argp);
51*62c56f98SSadaf Ebrahimi
52*62c56f98SSadaf Ebrahimi if (ret >= 0 && ret < MPS_TRACE_BUF_SIZE) {
53*62c56f98SSadaf Ebrahimi str[ret] = '\0';
54*62c56f98SSadaf Ebrahimi mbedtls_printf("[%d|L%d]: %s\n", id, line, str);
55*62c56f98SSadaf Ebrahimi }
56*62c56f98SSadaf Ebrahimi }
57*62c56f98SSadaf Ebrahimi
mbedtls_mps_trace_get_depth()58*62c56f98SSadaf Ebrahimi int mbedtls_mps_trace_get_depth()
59*62c56f98SSadaf Ebrahimi {
60*62c56f98SSadaf Ebrahimi return trace_depth;
61*62c56f98SSadaf Ebrahimi }
mbedtls_mps_trace_dec_depth()62*62c56f98SSadaf Ebrahimi void mbedtls_mps_trace_dec_depth()
63*62c56f98SSadaf Ebrahimi {
64*62c56f98SSadaf Ebrahimi trace_depth--;
65*62c56f98SSadaf Ebrahimi }
mbedtls_mps_trace_inc_depth()66*62c56f98SSadaf Ebrahimi void mbedtls_mps_trace_inc_depth()
67*62c56f98SSadaf Ebrahimi {
68*62c56f98SSadaf Ebrahimi trace_depth++;
69*62c56f98SSadaf Ebrahimi }
70*62c56f98SSadaf Ebrahimi
mbedtls_mps_trace_color(int id)71*62c56f98SSadaf Ebrahimi void mbedtls_mps_trace_color(int id)
72*62c56f98SSadaf Ebrahimi {
73*62c56f98SSadaf Ebrahimi if (id > (int) (sizeof(colors) / sizeof(*colors))) {
74*62c56f98SSadaf Ebrahimi return;
75*62c56f98SSadaf Ebrahimi }
76*62c56f98SSadaf Ebrahimi printf("%s", colors[id]);
77*62c56f98SSadaf Ebrahimi }
78*62c56f98SSadaf Ebrahimi
mbedtls_mps_trace_indent(int level,mbedtls_mps_trace_type ty)79*62c56f98SSadaf Ebrahimi void mbedtls_mps_trace_indent(int level, mbedtls_mps_trace_type ty)
80*62c56f98SSadaf Ebrahimi {
81*62c56f98SSadaf Ebrahimi if (level > 0) {
82*62c56f98SSadaf Ebrahimi while (--level) {
83*62c56f98SSadaf Ebrahimi printf("| ");
84*62c56f98SSadaf Ebrahimi }
85*62c56f98SSadaf Ebrahimi
86*62c56f98SSadaf Ebrahimi printf("| ");
87*62c56f98SSadaf Ebrahimi }
88*62c56f98SSadaf Ebrahimi
89*62c56f98SSadaf Ebrahimi switch (ty) {
90*62c56f98SSadaf Ebrahimi case MBEDTLS_MPS_TRACE_TYPE_COMMENT:
91*62c56f98SSadaf Ebrahimi mbedtls_printf("@ ");
92*62c56f98SSadaf Ebrahimi break;
93*62c56f98SSadaf Ebrahimi
94*62c56f98SSadaf Ebrahimi case MBEDTLS_MPS_TRACE_TYPE_CALL:
95*62c56f98SSadaf Ebrahimi mbedtls_printf("+--> ");
96*62c56f98SSadaf Ebrahimi break;
97*62c56f98SSadaf Ebrahimi
98*62c56f98SSadaf Ebrahimi case MBEDTLS_MPS_TRACE_TYPE_ERROR:
99*62c56f98SSadaf Ebrahimi mbedtls_printf("E ");
100*62c56f98SSadaf Ebrahimi break;
101*62c56f98SSadaf Ebrahimi
102*62c56f98SSadaf Ebrahimi case MBEDTLS_MPS_TRACE_TYPE_RETURN:
103*62c56f98SSadaf Ebrahimi mbedtls_printf("< ");
104*62c56f98SSadaf Ebrahimi break;
105*62c56f98SSadaf Ebrahimi
106*62c56f98SSadaf Ebrahimi default:
107*62c56f98SSadaf Ebrahimi break;
108*62c56f98SSadaf Ebrahimi }
109*62c56f98SSadaf Ebrahimi }
110*62c56f98SSadaf Ebrahimi
111*62c56f98SSadaf Ebrahimi #endif /* MBEDTLS_MPS_ENABLE_TRACE */
112*62c56f98SSadaf Ebrahimi #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
113