mirror of
https://github.com/clearlinux/uwsgi.git
synced 2026-08-21 13:07:54 +00:00
Unbit
128 lines
3.1 KiB
C
128 lines
3.1 KiB
C
#include "../../uwsgi.h"
|
|
|
|
extern struct uwsgi_server uwsgi;
|
|
|
|
|
|
struct uwsgi_rsyslog {
|
|
int packet_size;
|
|
int msg_size;
|
|
int split_msg;
|
|
} u_rsyslog;
|
|
|
|
|
|
struct uwsgi_option rsyslog_options[] = {
|
|
{"rsyslog-packet-size", required_argument, 0, "set maximum packet size for syslog messages (default 1024) WARNING! using packets > 1024 breaks RFC 3164 (#4.1)", uwsgi_opt_set_int, &u_rsyslog.packet_size, 0},
|
|
{"rsyslog-split-messages", no_argument, 0, "split big messages into multiple chunks if they are bigger than allowed packet size (default is false)", uwsgi_opt_true, &u_rsyslog.split_msg, 0},
|
|
{0, 0, 0, 0, 0, 0, 0},
|
|
};
|
|
|
|
|
|
ssize_t uwsgi_rsyslog_logger(struct uwsgi_logger *ul, char *message, size_t len) {
|
|
|
|
char ctime_storage[26];
|
|
time_t current_time;
|
|
int portn = 514;
|
|
int rlen;
|
|
|
|
if (!ul->configured) {
|
|
|
|
if (!ul->arg) {
|
|
uwsgi_log_safe("invalid rsyslog syntax\n");
|
|
exit(1);
|
|
}
|
|
|
|
if (ul->arg[0] == '/') {
|
|
ul->fd = socket(AF_UNIX, SOCK_DGRAM, 0);
|
|
}
|
|
else {
|
|
ul->fd = socket(AF_INET, SOCK_DGRAM, 0);
|
|
}
|
|
if (ul->fd < 0) {
|
|
uwsgi_error_safe("socket()");
|
|
exit(1);
|
|
}
|
|
|
|
uwsgi_socket_nb(ul->fd);
|
|
|
|
ul->count = 29;
|
|
|
|
char *comma = strchr(ul->arg, ',');
|
|
if (comma) {
|
|
ul->data = comma+1;
|
|
*comma = 0;
|
|
char *prisev = strchr(ul->data, ',');
|
|
if (prisev) {
|
|
*prisev = 0;
|
|
ul->count = atoi(prisev+1);
|
|
}
|
|
}
|
|
else {
|
|
ul->data = uwsgi_concat2(uwsgi.hostname," uwsgi");
|
|
}
|
|
|
|
|
|
char *port = strchr(ul->arg, ':');
|
|
if (port) {
|
|
portn = atoi(port+1);
|
|
*port = 0;
|
|
}
|
|
|
|
if (ul->arg[0] == '/') {
|
|
ul->addr_len = socket_to_un_addr(ul->arg, &ul->addr.sa_un);
|
|
}
|
|
else {
|
|
ul->addr_len = socket_to_in_addr(ul->arg, NULL, portn, &ul->addr.sa_in);
|
|
}
|
|
|
|
if (port) *port = ':';
|
|
if (comma) *comma = ',';
|
|
|
|
if (!u_rsyslog.packet_size) u_rsyslog.packet_size = 1024;
|
|
if (!u_rsyslog.msg_size) u_rsyslog.msg_size = u_rsyslog.packet_size - 30;
|
|
|
|
ul->buf = uwsgi_malloc(uwsgi.log_master_bufsize);
|
|
|
|
ul->configured = 1;
|
|
}
|
|
|
|
|
|
current_time = uwsgi_now();
|
|
|
|
// drop newline
|
|
if (message[len-1] == '\n') len--;
|
|
#if defined(__sun__) && !defined(__clang__)
|
|
ctime_r(¤t_time, ctime_storage, 26);
|
|
#else
|
|
ctime_r(¤t_time, ctime_storage);
|
|
#endif
|
|
|
|
int pos, msg_len, ret;
|
|
for (pos=0 ; pos < (int) len ;) {
|
|
if (pos > 0 && !u_rsyslog.split_msg) return pos;
|
|
msg_len = ( ((int)len)-pos > u_rsyslog.msg_size ? u_rsyslog.msg_size : ((int)len)-pos);
|
|
rlen = snprintf(ul->buf, u_rsyslog.packet_size, "<%d>%.*s %s: %.*s", ul->count, 15, ctime_storage+4, (char *) ul->data, msg_len, &message[pos]);
|
|
if (rlen > 0 && rlen < u_rsyslog.packet_size) {
|
|
ret = sendto(ul->fd, ul->buf, rlen, 0, (const struct sockaddr *) &ul->addr, ul->addr_len);
|
|
if (ret <= 0) return ret;
|
|
pos += msg_len;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
return pos;
|
|
|
|
}
|
|
|
|
void uwsgi_rsyslog_register() {
|
|
uwsgi_register_logger("rsyslog", uwsgi_rsyslog_logger);
|
|
}
|
|
|
|
struct uwsgi_plugin rsyslog_plugin = {
|
|
|
|
.name = "rsyslog",
|
|
.on_load = uwsgi_rsyslog_register,
|
|
.options = rsyslog_options,
|
|
|
|
};
|
|
|