gssproxy/src/gp_debug.c at feb4ad25c43f07e7d94312136b2ffb3a820c6fb5 · simo5/gssproxy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/* Copyright (C) 2011,2018 the GSS-PROXY contributors, see COPYING for license */

#include "config.h"
#include <stdbool.h>
#include <stdlib.h>
#include "gp_proxy.h"
#include "gp_debug.h"
#include "gp_log.h"

/* global debug switch */
int gp_debug = 0;
bool gp_env_trace = false;

/* determine if we can set KRB5_TRACE or the user already set it.
 * we do not override a user's setting */
void gp_debug_check_env_(void)
{
    if (getenv("KRB5_TRACE")) {
        gp_env_trace = true;
    }
}

void (*gp_debug_setup_k5_trace_fn)(int) = NULL;

void gp_debug_set_krb5_tracing_fn(void (*fn)(int))
{
    if (gp_debug_setup_k5_trace_fn != NULL) {
        /* deactivate potential previously set tracing
         * facility */
        gp_debug_setup_k5_trace_fn(0);
    }

    gp_debug_setup_k5_trace_fn = fn;

    /* the tracing setup function may be set after the
     * initial debug level is set via configuration.
     * Make sure to immedaiately call it if debug level is
     * already above 3 */
    if (gp_debug >= 3 && !gp_env_trace) {
        gp_debug_setup_k5_trace_fn(1);
    }
}

void gp_debug_toggle(int level)
{
    if (!gp_env_trace) {
        if (level >= 3) {
            if (gp_debug_setup_k5_trace_fn) {
                gp_debug_setup_k5_trace_fn(1);
            } else {
                setenv("KRB5_TRACE", "/dev/stderr", 1);
            }
        } else if (level < 3) {
            if (gp_debug_setup_k5_trace_fn) {
                gp_debug_setup_k5_trace_fn(0);
            } else {
                unsetenv("KRB5_TRACE");
            }
        }
    }

    gp_debug = level;
    GPDEBUG("Debug Level changed to %d\n", gp_debug);
}

void gp_log_failure(gss_OID mech, uint32_t maj, uint32_t min)
{
    char buf[MAX_LOG_LINE];

    gp_fmt_status(mech, maj, min, buf, MAX_LOG_LINE);

    fprintf(stderr, "Failed with: %s\n", buf);
}

const char *gp_debug_timestamp(void)
{
    static __thread char buffer[24];
    static __thread time_t timestamp = 0;
    struct tm tm_info;
    time_t now;

    time(&now);
    if (now == timestamp) return buffer;

    gmtime_r(&now, &tm_info);
    strftime(buffer, 24, "[%Y/%m/%d %H:%M:%S]: ", &tm_info);
    timestamp = now;
    return buffer;
}

/* thread local connection/client id */
static __thread int cid;

void gp_debug_set_conn_id(int id)
{
    cid = id;
}

static const char*gp_debug_conn_id(void)
{
    static __thread char buffer[18];
    static __thread int last_cid = 0;

    if (cid == 0) {
        buffer[0] = '\0';
        return buffer;
    }

    if (last_cid == cid) return buffer;

    (void)snprintf(buffer, 17, "[CID %d]", cid);
    buffer[17] = '\0';
    last_cid = cid;
    return buffer;
}

void gp_debug_printf(const char *format, ...)
{
    va_list varargs;
    va_start(varargs, format);
    vfprintf(stderr, format, varargs);
    va_end(varargs);
}

void gp_debug_time_printf(const char *format, ...)
{
    va_list varargs;

    fprintf(stderr, "%s%s", gp_debug_conn_id(), gp_debug_timestamp());

    va_start(varargs, format);
    vfprintf(stderr, format, varargs);
    va_end(varargs);
}