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
135
136
137
|
/* I/O for printing warnings, errors and debug messages
* Copyright (C) 2005 Red Hat Inc.
*
* This file is part of systemtap, and is free software. You can
* redistribute it and/or modify it under the terms of the GNU General
* Public License (GPL); either version 2, or (at your option) any
* later version.
*/
#ifndef _IO_C_
#define _IO_C_
#include "transport/transport.c"
void _stp_print_flush (void);
void _stp_string_cat_cstr (String str1, const char *str2);
/** @file io.c
* @brief I/O for printing warnings, errors and debug messages.
*/
/** @addtogroup io I/O
* @{
*/
/** private buffer for _stp_log() */
#define STP_LOG_BUF_LEN 2047
#define WARN_STRING "\033[33mWARNING: \033[0m"
#define ERR_STRING "\033[31mERROR: \033[0m"
static char _stp_lbuf[NR_CPUS][STP_LOG_BUF_LEN + 1];
enum code { INFO=0, WARN, ERROR, DBUG };
static void _stp_vlog (enum code type, char *func, int line, const char *fmt, va_list args)
{
int num;
char *buf = &_stp_lbuf[get_cpu()][0];
int start = 0;
if (type == DBUG) {
start = scnprintf(buf, STP_LOG_BUF_LEN, "\033[36m%s:%d:\033[0m ", func, line);
} else if (type == WARN) {
strcpy (buf, WARN_STRING);
start = sizeof(WARN_STRING) - 1;
} else if (type == ERROR) {
strcpy (buf, ERR_STRING);
start = sizeof(ERR_STRING) - 1;
}
num = vscnprintf (buf + start, STP_LOG_BUF_LEN - start, fmt, args);
if (num + start) {
if (buf[num + start - 1] != '\n') {
buf[num + start] = '\n';
num++;
}
buf[num + start] = '\0';
_stp_ctrl_send(STP_REALTIME_DATA, buf, start + num + 1, _stp_pid);
}
put_cpu();
}
/** Logs Data.
* This function sends the message immediately to stpd. It
* will also be sent over the bulk transport (relayfs) if it is
* being used. If the last character is not a newline, then one
* is added. This function is not as efficient as _stp_printf()
* and should only be used for urgent messages. You probably want
* dbug(), or _stp_warn().
* @param fmt A variable number of args.
* @todo Evaluate if this function is necessary.
*/
void _stp_log (const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
_stp_vlog (INFO, NULL, 0, fmt, args);
va_end(args);
}
/** Prints warning.
* This function sends a warning message immediately to stpd. It
* will also be sent over the bulk transport (relayfs) if it is
* being used. If the last character is not a newline, then one
* is added.
* @param fmt A variable number of args.
*/
void _stp_warn (const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
_stp_vlog (WARN, NULL, 0, fmt, args);
va_end(args);
}
/** Exits and unloads the module.
* This function sends a signal to stpd to tell it to
* unload the module and exit. The module will not be
* unloaded until after the current probe returns.
* @note Be careful to not treat this like the Linux exit()
* call. You should probably call return immediately after
* calling _stp_exit().
*/
void _stp_exit (void)
{
schedule_work (&stp_exit);
}
/** Prints error message and exits.
* This function sends an error message immediately to stpd. It
* will also be sent over the bulk transport (relayfs) if it is
* being used. If the last character is not a newline, then one
* is added.
*
* After the error message is displayed, the module will be unloaded.
* @param fmt A variable number of args.
* @sa _stp_exit().
*/
void _stp_error (const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
_stp_vlog (ERROR, NULL, 0, fmt, args);
va_end(args);
_stp_exit();
}
static void _stp_dbug (char *func, int line, const char *fmt, ...)
{
va_list args;
va_start(args, fmt);
_stp_vlog (DBUG, func, line, fmt, args);
va_end(args);
}
/** @} */
#endif /* _IO_C_ */
|
