blob: 042e244a40a3ad4b767d485301941d957c2bdebb [file] [log] [blame]
Willy Tarreau67b5a162019-08-11 16:38:56 +02001/*
2 * Event sink management
3 *
4 * Copyright (C) 2000-2019 Willy Tarreau - w@1wt.eu
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation, version 2.1
9 * exclusively.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +020021#include <sys/mman.h>
22#include <errno.h>
23#include <fcntl.h>
24
Willy Tarreau36979d92020-06-05 17:27:29 +020025#include <import/ist.h>
Willy Tarreau4c7e4b72020-05-27 12:58:42 +020026#include <haproxy/api.h>
Christopher Faulet6b0a0fb2022-04-04 11:29:28 +020027#include <haproxy/applet.h>
Willy Tarreau6be78492020-06-05 00:00:29 +020028#include <haproxy/cfgparse.h>
Willy Tarreau83487a82020-06-04 20:19:54 +020029#include <haproxy/cli.h>
Willy Tarreau36979d92020-06-05 17:27:29 +020030#include <haproxy/errors.h>
Willy Tarreau853b2972020-05-27 18:01:47 +020031#include <haproxy/list.h>
Willy Tarreauaeed4a82020-06-04 22:01:04 +020032#include <haproxy/log.h>
Willy Tarreau817538e2021-05-08 20:20:21 +020033#include <haproxy/proxy.h>
Willy Tarreaud2ad57c2020-06-03 19:43:35 +020034#include <haproxy/ring.h>
Willy Tarreau5edca2f2022-05-27 09:25:10 +020035#include <haproxy/sc_strm.h>
Willy Tarreau3727a8a2020-06-04 17:37:26 +020036#include <haproxy/signal.h>
Willy Tarreauba2f73d2020-06-03 20:02:28 +020037#include <haproxy/sink.h>
Willy Tarreaucb086c62022-05-27 09:47:12 +020038#include <haproxy/stconn.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020039#include <haproxy/time.h>
Willy Tarreau4bad5e22021-05-08 13:05:30 +020040#include <haproxy/tools.h>
Willy Tarreau67b5a162019-08-11 16:38:56 +020041
42struct list sink_list = LIST_HEAD_INIT(sink_list);
43
Emeric Brund6e581d2022-09-13 16:16:30 +020044/* sink proxies list */
45struct proxy *sink_proxies_list;
46
Emeric Brun99c453d2020-05-25 15:01:04 +020047struct sink *cfg_sink;
48
Willy Tarreau67b5a162019-08-11 16:38:56 +020049struct sink *sink_find(const char *name)
50{
51 struct sink *sink;
52
53 list_for_each_entry(sink, &sink_list, sink_list)
54 if (strcmp(sink->name, name) == 0)
55 return sink;
56 return NULL;
57}
58
59/* creates a new sink and adds it to the list, it's still generic and not fully
60 * initialized. Returns NULL on allocation failure. If another one already
61 * exists with the same name, it will be returned. The caller can detect it as
62 * a newly created one has type SINK_TYPE_NEW.
63 */
Emeric Brun54648852020-07-06 15:54:06 +020064static struct sink *__sink_new(const char *name, const char *desc, int fmt)
Willy Tarreau67b5a162019-08-11 16:38:56 +020065{
66 struct sink *sink;
67
68 sink = sink_find(name);
69 if (sink)
70 goto end;
71
Emeric Brun494c5052020-05-28 11:13:15 +020072 sink = calloc(1, sizeof(*sink));
Willy Tarreau67b5a162019-08-11 16:38:56 +020073 if (!sink)
74 goto end;
75
Emeric Brun99c453d2020-05-25 15:01:04 +020076 sink->name = strdup(name);
Tim Duesterhusa7ebffe2021-01-03 19:54:11 +010077 if (!sink->name)
78 goto err;
79
Emeric Brun99c453d2020-05-25 15:01:04 +020080 sink->desc = strdup(desc);
Tim Duesterhusa7ebffe2021-01-03 19:54:11 +010081 if (!sink->desc)
82 goto err;
83
Willy Tarreau67b5a162019-08-11 16:38:56 +020084 sink->fmt = fmt;
85 sink->type = SINK_TYPE_NEW;
Christopher Fauleta63a5c22019-11-15 15:10:12 +010086 sink->maxlen = BUFSIZE;
Willy Tarreau67b5a162019-08-11 16:38:56 +020087 /* address will be filled by the caller if needed */
Willy Tarreau973e6622019-08-20 11:57:52 +020088 sink->ctx.fd = -1;
Willy Tarreau67b5a162019-08-11 16:38:56 +020089 sink->ctx.dropped = 0;
90 HA_RWLOCK_INIT(&sink->ctx.lock);
Willy Tarreau2b718102021-04-21 07:32:39 +020091 LIST_APPEND(&sink_list, &sink->sink_list);
Willy Tarreau67b5a162019-08-11 16:38:56 +020092 end:
93 return sink;
Tim Duesterhusa7ebffe2021-01-03 19:54:11 +010094
95 err:
Willy Tarreau61cfdf42021-02-20 10:46:51 +010096 ha_free(&sink->name);
97 ha_free(&sink->desc);
98 ha_free(&sink);
Tim Duesterhusa7ebffe2021-01-03 19:54:11 +010099
100 return NULL;
Willy Tarreau67b5a162019-08-11 16:38:56 +0200101}
102
Willy Tarreau973e6622019-08-20 11:57:52 +0200103/* creates a sink called <name> of type FD associated to fd <fd>, format <fmt>,
104 * and description <desc>. Returns NULL on allocation failure or conflict.
105 * Perfect duplicates are merged (same type, fd, and name).
106 */
Emeric Brun54648852020-07-06 15:54:06 +0200107struct sink *sink_new_fd(const char *name, const char *desc, enum log_fmt fmt, int fd)
Willy Tarreau973e6622019-08-20 11:57:52 +0200108{
109 struct sink *sink;
110
111 sink = __sink_new(name, desc, fmt);
112 if (!sink || (sink->type == SINK_TYPE_FD && sink->ctx.fd == fd))
113 goto end;
114
115 if (sink->type != SINK_TYPE_NEW) {
116 sink = NULL;
117 goto end;
118 }
119
120 sink->type = SINK_TYPE_FD;
121 sink->ctx.fd = fd;
122 end:
123 return sink;
124}
125
Willy Tarreau4ed23ca2019-08-23 15:47:49 +0200126/* creates a sink called <name> of type BUF of size <size>, format <fmt>,
127 * and description <desc>. Returns NULL on allocation failure or conflict.
128 * Perfect duplicates are merged (same type and name). If sizes differ, the
129 * largest one is kept.
130 */
Emeric Brun54648852020-07-06 15:54:06 +0200131struct sink *sink_new_buf(const char *name, const char *desc, enum log_fmt fmt, size_t size)
Willy Tarreau4ed23ca2019-08-23 15:47:49 +0200132{
133 struct sink *sink;
134
135 sink = __sink_new(name, desc, fmt);
136 if (!sink)
137 goto fail;
138
139 if (sink->type == SINK_TYPE_BUFFER) {
140 /* such a buffer already exists, we may have to resize it */
141 if (!ring_resize(sink->ctx.ring, size))
142 goto fail;
143 goto end;
144 }
145
146 if (sink->type != SINK_TYPE_NEW) {
147 /* already exists of another type */
148 goto fail;
149 }
150
151 sink->ctx.ring = ring_new(size);
152 if (!sink->ctx.ring) {
Willy Tarreau2b718102021-04-21 07:32:39 +0200153 LIST_DELETE(&sink->sink_list);
Emeric Brun99c453d2020-05-25 15:01:04 +0200154 free(sink->name);
155 free(sink->desc);
Willy Tarreau4ed23ca2019-08-23 15:47:49 +0200156 free(sink);
157 goto fail;
158 }
159
160 sink->type = SINK_TYPE_BUFFER;
161 end:
162 return sink;
163 fail:
164 return NULL;
165}
166
Willy Tarreau67b5a162019-08-11 16:38:56 +0200167/* tries to send <nmsg> message parts (up to 8, ignored above) from message
Ilya Shipitsin46a030c2020-07-05 16:36:08 +0500168 * array <msg> to sink <sink>. Formatting according to the sink's preference is
Willy Tarreau8f240232019-08-27 16:41:06 +0200169 * done here. Lost messages are NOT accounted for. It is preferable to call
170 * sink_write() instead which will also try to emit the number of dropped
171 * messages when there are any. It returns >0 if it could write anything,
172 * <=0 otherwise.
Willy Tarreau67b5a162019-08-11 16:38:56 +0200173 */
Emeric Brun54648852020-07-06 15:54:06 +0200174 ssize_t __sink_write(struct sink *sink, const struct ist msg[], size_t nmsg,
175 int level, int facility, struct ist *metadata)
176 {
177 struct ist *pfx = NULL;
Willy Tarreaua1426de2019-08-27 14:21:02 +0200178 size_t npfx = 0;
Emeric Brunbd163812020-05-06 14:33:46 +0200179
Emeric Brun54648852020-07-06 15:54:06 +0200180 if (sink->fmt == LOG_FORMAT_RAW)
Emeric Brunbd163812020-05-06 14:33:46 +0200181 goto send;
Willy Tarreau67b5a162019-08-11 16:38:56 +0200182
Emeric Brun54648852020-07-06 15:54:06 +0200183 pfx = build_log_header(sink->fmt, level, facility, metadata, &npfx);
Emeric Brunbd163812020-05-06 14:33:46 +0200184
185send:
Willy Tarreau973e6622019-08-20 11:57:52 +0200186 if (sink->type == SINK_TYPE_FD) {
Willy Tarreau8f240232019-08-27 16:41:06 +0200187 return fd_write_frag_line(sink->ctx.fd, sink->maxlen, pfx, npfx, msg, nmsg, 1);
Willy Tarreau4ed23ca2019-08-23 15:47:49 +0200188 }
189 else if (sink->type == SINK_TYPE_BUFFER) {
Willy Tarreau8f240232019-08-27 16:41:06 +0200190 return ring_write(sink->ctx.ring, sink->maxlen, pfx, npfx, msg, nmsg);
Willy Tarreau973e6622019-08-20 11:57:52 +0200191 }
Willy Tarreau8f240232019-08-27 16:41:06 +0200192 return 0;
193}
Willy Tarreau67b5a162019-08-11 16:38:56 +0200194
Willy Tarreau8f240232019-08-27 16:41:06 +0200195/* Tries to emit a message indicating the number of dropped events. In case of
196 * success, the amount of drops is reduced by as much. It's supposed to be
197 * called under an exclusive lock on the sink to avoid multiple produces doing
198 * the same. On success, >0 is returned, otherwise <=0 on failure.
199 */
Emeric Brun54648852020-07-06 15:54:06 +0200200int sink_announce_dropped(struct sink *sink, int facility)
Willy Tarreau8f240232019-08-27 16:41:06 +0200201{
Emeric Brun54648852020-07-06 15:54:06 +0200202 static THREAD_LOCAL struct ist metadata[LOG_META_FIELDS];
203 static THREAD_LOCAL pid_t curr_pid;
204 static THREAD_LOCAL char pidstr[16];
Willy Tarreau8f240232019-08-27 16:41:06 +0200205 unsigned int dropped;
206 struct buffer msg;
207 struct ist msgvec[1];
208 char logbuf[64];
209
210 while (unlikely((dropped = sink->ctx.dropped) > 0)) {
211 chunk_init(&msg, logbuf, sizeof(logbuf));
212 chunk_printf(&msg, "%u event%s dropped", dropped, dropped > 1 ? "s" : "");
213 msgvec[0] = ist2(msg.area, msg.data);
Emeric Brunbd163812020-05-06 14:33:46 +0200214
Emeric Brun54648852020-07-06 15:54:06 +0200215 if (!metadata[LOG_META_HOST].len) {
216 if (global.log_send_hostname)
Tim Duesterhus77508502022-03-15 13:11:06 +0100217 metadata[LOG_META_HOST] = ist(global.log_send_hostname);
Emeric Brun54648852020-07-06 15:54:06 +0200218 }
219
220 if (!metadata[LOG_META_TAG].len)
221 metadata[LOG_META_TAG] = ist2(global.log_tag.area, global.log_tag.data);
222
223 if (unlikely(curr_pid != getpid()))
224 metadata[LOG_META_PID].len = 0;
225
226 if (!metadata[LOG_META_PID].len) {
227 curr_pid = getpid();
228 ltoa_o(curr_pid, pidstr, sizeof(pidstr));
229 metadata[LOG_META_PID] = ist2(pidstr, strlen(pidstr));
230 }
231
232 if (__sink_write(sink, msgvec, 1, LOG_NOTICE, facility, metadata) <= 0)
Willy Tarreau8f240232019-08-27 16:41:06 +0200233 return 0;
234 /* success! */
235 HA_ATOMIC_SUB(&sink->ctx.dropped, dropped);
236 }
237 return 1;
Willy Tarreau67b5a162019-08-11 16:38:56 +0200238}
239
Willy Tarreau9f830d72019-08-26 18:17:04 +0200240/* parse the "show events" command, returns 1 if a message is returned, otherwise zero */
241static int cli_parse_show_events(char **args, char *payload, struct appctx *appctx, void *private)
242{
243 struct sink *sink;
Willy Tarreaucba88382022-05-05 15:18:57 +0200244 uint ring_flags;
Willy Tarreau1d181e42019-08-30 11:17:01 +0200245 int arg;
Willy Tarreau9f830d72019-08-26 18:17:04 +0200246
247 args++; // make args[1] the 1st arg
248
249 if (!*args[1]) {
250 /* no arg => report the list of supported sink */
Willy Tarreau1d181e42019-08-30 11:17:01 +0200251 chunk_printf(&trash, "Supported events sinks are listed below. Add -w(wait), -n(new). Any key to stop\n");
Willy Tarreau9f830d72019-08-26 18:17:04 +0200252 list_for_each_entry(sink, &sink_list, sink_list) {
253 chunk_appendf(&trash, " %-10s : type=%s, %u dropped, %s\n",
254 sink->name,
255 sink->type == SINK_TYPE_NEW ? "init" :
256 sink->type == SINK_TYPE_FD ? "fd" :
257 sink->type == SINK_TYPE_BUFFER ? "buffer" : "?",
258 sink->ctx.dropped, sink->desc);
259 }
260
261 trash.area[trash.data] = 0;
262 return cli_msg(appctx, LOG_WARNING, trash.area);
263 }
264
265 if (!cli_has_level(appctx, ACCESS_LVL_OPER))
266 return 1;
267
268 sink = sink_find(args[1]);
269 if (!sink)
270 return cli_err(appctx, "No such event sink");
271
272 if (sink->type != SINK_TYPE_BUFFER)
273 return cli_msg(appctx, LOG_NOTICE, "Nothing to report for this sink");
274
Willy Tarreaucba88382022-05-05 15:18:57 +0200275 ring_flags = 0;
Willy Tarreau1d181e42019-08-30 11:17:01 +0200276 for (arg = 2; *args[arg]; arg++) {
277 if (strcmp(args[arg], "-w") == 0)
Willy Tarreaucba88382022-05-05 15:18:57 +0200278 ring_flags |= RING_WF_WAIT_MODE;
Willy Tarreau1d181e42019-08-30 11:17:01 +0200279 else if (strcmp(args[arg], "-n") == 0)
Willy Tarreaucba88382022-05-05 15:18:57 +0200280 ring_flags |= RING_WF_SEEK_NEW;
Willy Tarreau1d181e42019-08-30 11:17:01 +0200281 else if (strcmp(args[arg], "-nw") == 0 || strcmp(args[arg], "-wn") == 0)
Willy Tarreaucba88382022-05-05 15:18:57 +0200282 ring_flags |= RING_WF_WAIT_MODE | RING_WF_SEEK_NEW;
Willy Tarreau1d181e42019-08-30 11:17:01 +0200283 else
284 return cli_err(appctx, "unknown option");
285 }
Willy Tarreaucba88382022-05-05 15:18:57 +0200286 return ring_attach_cli(sink->ctx.ring, appctx, ring_flags);
Willy Tarreau9f830d72019-08-26 18:17:04 +0200287}
288
Ilya Shipitsin47d17182020-06-21 21:42:57 +0500289/* Pre-configures a ring proxy to emit connections */
Emeric Brun494c5052020-05-28 11:13:15 +0200290void sink_setup_proxy(struct proxy *px)
291{
292 px->last_change = now.tv_sec;
Christopher Faulet11a707a2022-10-24 15:10:18 +0200293 px->cap = PR_CAP_BE;
Emeric Brun494c5052020-05-28 11:13:15 +0200294 px->maxconn = 0;
295 px->conn_retries = 1;
296 px->timeout.server = TICK_ETERNITY;
297 px->timeout.client = TICK_ETERNITY;
298 px->timeout.connect = TICK_ETERNITY;
299 px->accept = NULL;
300 px->options2 |= PR_O2_INDEPSTR | PR_O2_SMARTCON | PR_O2_SMARTACC;
Emeric Brund6e581d2022-09-13 16:16:30 +0200301 px->next = sink_proxies_list;
302 sink_proxies_list = px;
Emeric Brun494c5052020-05-28 11:13:15 +0200303}
304
305/*
Willy Tarreau42cc8312022-05-04 20:42:23 +0200306 * IO Handler to handle message push to syslog tcp server.
307 * It takes its context from appctx->svcctx.
Emeric Brun494c5052020-05-28 11:13:15 +0200308 */
309static void sink_forward_io_handler(struct appctx *appctx)
310{
Willy Tarreauc12b3212022-05-27 11:08:15 +0200311 struct stconn *sc = appctx_sc(appctx);
Willy Tarreau0eca5392022-05-27 10:44:25 +0200312 struct stream *s = __sc_strm(sc);
Emeric Brun494c5052020-05-28 11:13:15 +0200313 struct sink *sink = strm_fe(s)->parent;
Willy Tarreau42cc8312022-05-04 20:42:23 +0200314 struct sink_forward_target *sft = appctx->svcctx;
Emeric Brun494c5052020-05-28 11:13:15 +0200315 struct ring *ring = sink->ctx.ring;
316 struct buffer *buf = &ring->buf;
317 uint64_t msg_len;
Willy Tarreau53bfab02022-08-04 17:18:54 +0200318 size_t len, cnt, ofs, last_ofs;
Emeric Brun494c5052020-05-28 11:13:15 +0200319 int ret = 0;
320
Ilya Shipitsin47d17182020-06-21 21:42:57 +0500321 /* if stopping was requested, close immediately */
Emeric Brun494c5052020-05-28 11:13:15 +0200322 if (unlikely(stopping))
323 goto close;
324
Christopher Fauletda89e9b2023-01-04 14:11:10 +0100325 if (unlikely(sc_ic(sc)->flags & CF_SHUTW))
Emeric Brun494c5052020-05-28 11:13:15 +0200326 goto close;
327
328 /* con closed by server side */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200329 if ((sc_oc(sc)->flags & CF_SHUTW))
Emeric Brun494c5052020-05-28 11:13:15 +0200330 goto close;
331
332 /* if the connection is not established, inform the stream that we want
333 * to be notified whenever the connection completes.
334 */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200335 if (sc_opposite(sc)->state < SC_ST_EST) {
Willy Tarreau90e8b452022-05-25 18:21:43 +0200336 applet_need_more_data(appctx);
Willy Tarreaub23edc82022-05-24 16:49:03 +0200337 se_need_remote_conn(appctx->sedesc);
Willy Tarreau4164eb92022-05-25 15:42:03 +0200338 applet_have_more_data(appctx);
Emeric Brun494c5052020-05-28 11:13:15 +0200339 return;
340 }
341
342 HA_SPIN_LOCK(SFT_LOCK, &sft->lock);
343 if (appctx != sft->appctx) {
344 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
345 goto close;
346 }
Emeric Brun494c5052020-05-28 11:13:15 +0200347
348 HA_RWLOCK_WRLOCK(LOGSRV_LOCK, &ring->lock);
349 LIST_DEL_INIT(&appctx->wait_entry);
350 HA_RWLOCK_WRUNLOCK(LOGSRV_LOCK, &ring->lock);
351
352 HA_RWLOCK_RDLOCK(LOGSRV_LOCK, &ring->lock);
353
354 /* explanation for the initialization below: it would be better to do
355 * this in the parsing function but this would occasionally result in
356 * dropped events because we'd take a reference on the oldest message
357 * and keep it while being scheduled. Thus instead let's take it the
358 * first time we enter here so that we have a chance to pass many
359 * existing messages before grabbing a reference to a location. This
360 * value cannot be produced after initialization.
361 */
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100362 if (unlikely(sft->ofs == ~0)) {
363 sft->ofs = b_peek_ofs(buf, 0);
364 HA_ATOMIC_INC(b_orig(buf) + sft->ofs);
Emeric Brun494c5052020-05-28 11:13:15 +0200365 }
366
Emeric Brun494c5052020-05-28 11:13:15 +0200367 /* in this loop, ofs always points to the counter byte that precedes
368 * the message so that we can take our reference there if we have to
369 * stop before the end (ret=0).
370 */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200371 if (sc_opposite(sc)->state == SC_ST_EST) {
Emeric Brunfdabf492020-12-02 17:02:09 +0100372 /* we were already there, adjust the offset to be relative to
373 * the buffer's head and remove us from the counter.
374 */
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100375 ofs = sft->ofs - b_head_ofs(buf);
376 if (sft->ofs < b_head_ofs(buf))
377 ofs += b_size(buf);
Emeric Brunfdabf492020-12-02 17:02:09 +0100378 BUG_ON(ofs >= buf->size);
Willy Tarreau4781b152021-04-06 13:53:36 +0200379 HA_ATOMIC_DEC(b_peek(buf, ofs));
Emeric Brunfdabf492020-12-02 17:02:09 +0100380
Emeric Brun494c5052020-05-28 11:13:15 +0200381 ret = 1;
382 while (ofs + 1 < b_data(buf)) {
383 cnt = 1;
384 len = b_peek_varint(buf, ofs + cnt, &msg_len);
385 if (!len)
386 break;
387 cnt += len;
388 BUG_ON(msg_len + ofs + cnt + 1 > b_data(buf));
389
390 if (unlikely(msg_len + 1 > b_size(&trash))) {
391 /* too large a message to ever fit, let's skip it */
392 ofs += cnt + msg_len;
393 continue;
394 }
395
396 chunk_reset(&trash);
397 len = b_getblk(buf, trash.area, msg_len, ofs + cnt);
398 trash.data += len;
399 trash.area[trash.data++] = '\n';
400
Willy Tarreaud0a06d52022-05-18 15:07:19 +0200401 if (applet_putchk(appctx, &trash) == -1) {
Emeric Brun494c5052020-05-28 11:13:15 +0200402 ret = 0;
403 break;
404 }
405 ofs += cnt + msg_len;
406 }
407
Willy Tarreau4781b152021-04-06 13:53:36 +0200408 HA_ATOMIC_INC(b_peek(buf, ofs));
Willy Tarreaud9c71882023-02-22 14:50:14 +0100409 last_ofs = b_tail_ofs(buf);
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100410 sft->ofs = b_peek_ofs(buf, ofs);
Emeric Brun494c5052020-05-28 11:13:15 +0200411 }
412 HA_RWLOCK_RDUNLOCK(LOGSRV_LOCK, &ring->lock);
413
414 if (ret) {
415 /* let's be woken up once new data arrive */
416 HA_RWLOCK_WRLOCK(LOGSRV_LOCK, &ring->lock);
Willy Tarreau2b718102021-04-21 07:32:39 +0200417 LIST_APPEND(&ring->waiters, &appctx->wait_entry);
Willy Tarreaud9c71882023-02-22 14:50:14 +0100418 ofs = b_tail_ofs(buf);
Emeric Brun494c5052020-05-28 11:13:15 +0200419 HA_RWLOCK_WRUNLOCK(LOGSRV_LOCK, &ring->lock);
Willy Tarreau53bfab02022-08-04 17:18:54 +0200420 if (ofs != last_ofs) {
421 /* more data was added into the ring between the
422 * unlock and the lock, and the writer might not
423 * have seen us. We need to reschedule a read.
424 */
425 applet_have_more_data(appctx);
426 } else
427 applet_have_no_more_data(appctx);
Emeric Brun494c5052020-05-28 11:13:15 +0200428 }
429 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
430
431 /* always drain data from server */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200432 co_skip(sc_oc(sc), sc_oc(sc)->output);
Emeric Brun494c5052020-05-28 11:13:15 +0200433 return;
434
435close:
Willy Tarreau0eca5392022-05-27 10:44:25 +0200436 sc_shutw(sc);
437 sc_shutr(sc);
Emeric Brun494c5052020-05-28 11:13:15 +0200438}
439
Emeric Brun97556472020-05-30 01:42:45 +0200440/*
441 * IO Handler to handle message push to syslog tcp server
442 * using octet counting frames
Willy Tarreau42cc8312022-05-04 20:42:23 +0200443 * It takes its context from appctx->svcctx.
Emeric Brun97556472020-05-30 01:42:45 +0200444 */
445static void sink_forward_oc_io_handler(struct appctx *appctx)
446{
Willy Tarreauc12b3212022-05-27 11:08:15 +0200447 struct stconn *sc = appctx_sc(appctx);
Willy Tarreau0eca5392022-05-27 10:44:25 +0200448 struct stream *s = __sc_strm(sc);
Emeric Brun97556472020-05-30 01:42:45 +0200449 struct sink *sink = strm_fe(s)->parent;
Willy Tarreau42cc8312022-05-04 20:42:23 +0200450 struct sink_forward_target *sft = appctx->svcctx;
Emeric Brun97556472020-05-30 01:42:45 +0200451 struct ring *ring = sink->ctx.ring;
452 struct buffer *buf = &ring->buf;
453 uint64_t msg_len;
454 size_t len, cnt, ofs;
455 int ret = 0;
456 char *p;
457
Ilya Shipitsin47d17182020-06-21 21:42:57 +0500458 /* if stopping was requested, close immediately */
Emeric Brun97556472020-05-30 01:42:45 +0200459 if (unlikely(stopping))
460 goto close;
461
Emeric Brun97556472020-05-30 01:42:45 +0200462 /* an error was detected */
Christopher Fauletda89e9b2023-01-04 14:11:10 +0100463 if (unlikely(sc_ic(sc)->flags & CF_SHUTW))
Emeric Brun97556472020-05-30 01:42:45 +0200464 goto close;
465
466 /* con closed by server side */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200467 if ((sc_oc(sc)->flags & CF_SHUTW))
Emeric Brun97556472020-05-30 01:42:45 +0200468 goto close;
469
470 /* if the connection is not established, inform the stream that we want
471 * to be notified whenever the connection completes.
472 */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200473 if (sc_opposite(sc)->state < SC_ST_EST) {
Willy Tarreau90e8b452022-05-25 18:21:43 +0200474 applet_need_more_data(appctx);
Willy Tarreaub23edc82022-05-24 16:49:03 +0200475 se_need_remote_conn(appctx->sedesc);
Willy Tarreau4164eb92022-05-25 15:42:03 +0200476 applet_have_more_data(appctx);
Emeric Brun97556472020-05-30 01:42:45 +0200477 return;
478 }
479
480 HA_SPIN_LOCK(SFT_LOCK, &sft->lock);
481 if (appctx != sft->appctx) {
482 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
483 goto close;
484 }
Emeric Brun97556472020-05-30 01:42:45 +0200485
486 HA_RWLOCK_WRLOCK(LOGSRV_LOCK, &ring->lock);
487 LIST_DEL_INIT(&appctx->wait_entry);
488 HA_RWLOCK_WRUNLOCK(LOGSRV_LOCK, &ring->lock);
489
490 HA_RWLOCK_RDLOCK(LOGSRV_LOCK, &ring->lock);
491
492 /* explanation for the initialization below: it would be better to do
493 * this in the parsing function but this would occasionally result in
494 * dropped events because we'd take a reference on the oldest message
495 * and keep it while being scheduled. Thus instead let's take it the
496 * first time we enter here so that we have a chance to pass many
497 * existing messages before grabbing a reference to a location. This
498 * value cannot be produced after initialization.
499 */
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100500 if (unlikely(sft->ofs == ~0)) {
501 sft->ofs = b_peek_ofs(buf, 0);
502 HA_ATOMIC_INC(b_orig(buf) + sft->ofs);
Emeric Brun97556472020-05-30 01:42:45 +0200503 }
504
Emeric Brun97556472020-05-30 01:42:45 +0200505 /* in this loop, ofs always points to the counter byte that precedes
506 * the message so that we can take our reference there if we have to
507 * stop before the end (ret=0).
508 */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200509 if (sc_opposite(sc)->state == SC_ST_EST) {
Emeric Brunfdabf492020-12-02 17:02:09 +0100510 /* we were already there, adjust the offset to be relative to
511 * the buffer's head and remove us from the counter.
512 */
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100513 ofs = sft->ofs - b_head_ofs(buf);
514 if (sft->ofs < b_head_ofs(buf))
515 ofs += b_size(buf);
Emeric Brunfdabf492020-12-02 17:02:09 +0100516 BUG_ON(ofs >= buf->size);
Willy Tarreau4781b152021-04-06 13:53:36 +0200517 HA_ATOMIC_DEC(b_peek(buf, ofs));
Emeric Brunfdabf492020-12-02 17:02:09 +0100518
Emeric Brun97556472020-05-30 01:42:45 +0200519 ret = 1;
520 while (ofs + 1 < b_data(buf)) {
521 cnt = 1;
522 len = b_peek_varint(buf, ofs + cnt, &msg_len);
523 if (!len)
524 break;
525 cnt += len;
526 BUG_ON(msg_len + ofs + cnt + 1 > b_data(buf));
527
528 chunk_reset(&trash);
529 p = ulltoa(msg_len, trash.area, b_size(&trash));
530 if (p) {
531 trash.data = (p - trash.area) + 1;
532 *p = ' ';
533 }
534
535 if (!p || (trash.data + msg_len > b_size(&trash))) {
536 /* too large a message to ever fit, let's skip it */
537 ofs += cnt + msg_len;
538 continue;
539 }
540
541 trash.data += b_getblk(buf, p + 1, msg_len, ofs + cnt);
542
Willy Tarreaud0a06d52022-05-18 15:07:19 +0200543 if (applet_putchk(appctx, &trash) == -1) {
Emeric Brun97556472020-05-30 01:42:45 +0200544 ret = 0;
545 break;
546 }
547 ofs += cnt + msg_len;
548 }
549
Willy Tarreau4781b152021-04-06 13:53:36 +0200550 HA_ATOMIC_INC(b_peek(buf, ofs));
Aurelien DARRAGON2c988672023-03-07 16:09:33 +0100551 sft->ofs = b_peek_ofs(buf, ofs);
Emeric Brun97556472020-05-30 01:42:45 +0200552 }
553 HA_RWLOCK_RDUNLOCK(LOGSRV_LOCK, &ring->lock);
554
555 if (ret) {
556 /* let's be woken up once new data arrive */
557 HA_RWLOCK_WRLOCK(LOGSRV_LOCK, &ring->lock);
Willy Tarreau2b718102021-04-21 07:32:39 +0200558 LIST_APPEND(&ring->waiters, &appctx->wait_entry);
Emeric Brun97556472020-05-30 01:42:45 +0200559 HA_RWLOCK_WRUNLOCK(LOGSRV_LOCK, &ring->lock);
Willy Tarreau4164eb92022-05-25 15:42:03 +0200560 applet_have_no_more_data(appctx);
Emeric Brun97556472020-05-30 01:42:45 +0200561 }
562 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
563
564 /* always drain data from server */
Willy Tarreau0eca5392022-05-27 10:44:25 +0200565 co_skip(sc_oc(sc), sc_oc(sc)->output);
Emeric Brun97556472020-05-30 01:42:45 +0200566 return;
567
568close:
Willy Tarreau0eca5392022-05-27 10:44:25 +0200569 sc_shutw(sc);
570 sc_shutr(sc);
Emeric Brun97556472020-05-30 01:42:45 +0200571}
572
Emeric Brun494c5052020-05-28 11:13:15 +0200573void __sink_forward_session_deinit(struct sink_forward_target *sft)
574{
Willy Tarreau0698c802022-05-11 14:09:57 +0200575 struct stream *s = appctx_strm(sft->appctx);
Emeric Brun494c5052020-05-28 11:13:15 +0200576 struct sink *sink;
577
Emeric Brun494c5052020-05-28 11:13:15 +0200578 sink = strm_fe(s)->parent;
579 if (!sink)
580 return;
581
582 HA_RWLOCK_WRLOCK(LOGSRV_LOCK, &sink->ctx.ring->lock);
583 LIST_DEL_INIT(&sft->appctx->wait_entry);
584 HA_RWLOCK_WRUNLOCK(LOGSRV_LOCK, &sink->ctx.ring->lock);
585
586 sft->appctx = NULL;
587 task_wakeup(sink->forward_task, TASK_WOKEN_MSG);
588}
589
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200590static int sink_forward_session_init(struct appctx *appctx)
591{
592 struct sink_forward_target *sft = appctx->svcctx;
593 struct stream *s;
594 struct sockaddr_storage *addr = NULL;
595
596 if (!sockaddr_alloc(&addr, &sft->srv->addr, sizeof(sft->srv->addr)))
597 goto out_error;
598
599 if (appctx_finalize_startup(appctx, sft->sink->forward_px, &BUF_NULL) == -1)
600 goto out_free_addr;
601
602 s = appctx_strm(appctx);
Willy Tarreau7cb9e6c2022-05-17 19:40:40 +0200603 s->scb->dst = addr;
Willy Tarreaucb041662022-05-17 19:44:42 +0200604 s->scb->flags |= SC_FL_NOLINGER;
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200605
606 s->target = &sft->srv->obj_type;
607 s->flags = SF_ASSIGNED;
608
609 s->do_log = NULL;
610 s->uniq_id = 0;
611
612 s->res.flags |= CF_READ_DONTWAIT;
Christopher Faulet2ca4cc12023-02-22 14:22:56 +0100613 applet_expect_no_data(appctx);
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200614 sft->appctx = appctx;
615
616 return 0;
617
618 out_free_addr:
619 sockaddr_free(&addr);
620 out_error:
621 return -1;
622}
Emeric Brun494c5052020-05-28 11:13:15 +0200623
624static void sink_forward_session_release(struct appctx *appctx)
625{
Willy Tarreau42cc8312022-05-04 20:42:23 +0200626 struct sink_forward_target *sft = appctx->svcctx;
Emeric Brun494c5052020-05-28 11:13:15 +0200627
628 if (!sft)
629 return;
630
631 HA_SPIN_LOCK(SFT_LOCK, &sft->lock);
632 if (sft->appctx == appctx)
633 __sink_forward_session_deinit(sft);
634 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
635}
636
637static struct applet sink_forward_applet = {
638 .obj_type = OBJ_TYPE_APPLET,
639 .name = "<SINKFWD>", /* used for logging */
640 .fct = sink_forward_io_handler,
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200641 .init = sink_forward_session_init,
Emeric Brun494c5052020-05-28 11:13:15 +0200642 .release = sink_forward_session_release,
643};
644
Emeric Brun97556472020-05-30 01:42:45 +0200645static struct applet sink_forward_oc_applet = {
646 .obj_type = OBJ_TYPE_APPLET,
647 .name = "<SINKFWDOC>", /* used for logging */
648 .fct = sink_forward_oc_io_handler,
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200649 .init = sink_forward_session_init,
Emeric Brun97556472020-05-30 01:42:45 +0200650 .release = sink_forward_session_release,
651};
652
Emeric Brun494c5052020-05-28 11:13:15 +0200653/*
654 * Create a new peer session in assigned state (connect will start automatically)
Willy Tarreau42cc8312022-05-04 20:42:23 +0200655 * It sets its context into appctx->svcctx.
Emeric Brun494c5052020-05-28 11:13:15 +0200656 */
657static struct appctx *sink_forward_session_create(struct sink *sink, struct sink_forward_target *sft)
658{
Emeric Brun494c5052020-05-28 11:13:15 +0200659 struct appctx *appctx;
Emeric Brun97556472020-05-30 01:42:45 +0200660 struct applet *applet = &sink_forward_applet;
661
662 if (sft->srv->log_proto == SRV_LOG_PROTO_OCTET_COUNTING)
663 applet = &sink_forward_oc_applet;
Emeric Brun494c5052020-05-28 11:13:15 +0200664
Christopher Faulet6095d572022-05-16 17:09:48 +0200665 appctx = appctx_new_here(applet, NULL);
Christopher Faulet2479e5f2022-01-19 14:50:11 +0100666 if (!appctx)
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100667 goto out_close;
Willy Tarreau42cc8312022-05-04 20:42:23 +0200668 appctx->svcctx = (void *)sft;
Emeric Brun494c5052020-05-28 11:13:15 +0200669
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200670 if (appctx_init(appctx) == -1)
Christopher Faulet92202da2022-05-11 12:22:10 +0200671 goto out_free_appctx;
Christopher Faulet13a35e52021-12-20 15:34:16 +0100672
Emeric Brun494c5052020-05-28 11:13:15 +0200673 return appctx;
674
675 /* Error unrolling */
Emeric Brun494c5052020-05-28 11:13:15 +0200676 out_free_appctx:
Christopher Fauletaee57fc2022-05-12 15:34:48 +0200677 appctx_free_on_early_error(appctx);
Emeric Brun494c5052020-05-28 11:13:15 +0200678 out_close:
679 return NULL;
680}
681
682/*
683 * Task to handle connctions to forward servers
684 */
Willy Tarreau144f84a2021-03-02 16:09:26 +0100685static struct task *process_sink_forward(struct task * task, void *context, unsigned int state)
Emeric Brun494c5052020-05-28 11:13:15 +0200686{
687 struct sink *sink = (struct sink *)context;
688 struct sink_forward_target *sft = sink->sft;
689
690 task->expire = TICK_ETERNITY;
691
692 if (!stopping) {
693 while (sft) {
694 HA_SPIN_LOCK(SFT_LOCK, &sft->lock);
695 /* if appctx is NULL, start a new session */
696 if (!sft->appctx)
697 sft->appctx = sink_forward_session_create(sink, sft);
698 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
699 sft = sft->next;
700 }
701 }
702 else {
703 while (sft) {
704 HA_SPIN_LOCK(SFT_LOCK, &sft->lock);
705 /* awake applet to perform a clean close */
706 if (sft->appctx)
707 appctx_wakeup(sft->appctx);
708 HA_SPIN_UNLOCK(SFT_LOCK, &sft->lock);
709 sft = sft->next;
710 }
711 }
712
713 return task;
714}
715/*
716 * Init task to manage connctions to forward servers
717 *
718 * returns 0 in case of error.
719 */
720int sink_init_forward(struct sink *sink)
721{
Willy Tarreaubeeabf52021-10-01 18:23:30 +0200722 sink->forward_task = task_new_anywhere();
Emeric Brun494c5052020-05-28 11:13:15 +0200723 if (!sink->forward_task)
724 return 0;
725
726 sink->forward_task->process = process_sink_forward;
727 sink->forward_task->context = (void *)sink;
728 sink->forward_sighandler = signal_register_task(0, sink->forward_task, 0);
729 task_wakeup(sink->forward_task, TASK_WOKEN_INIT);
730 return 1;
731}
Willy Tarreau32872db2022-08-31 18:52:17 +0200732
733/* This tries to rotate a file-backed ring, but only if it contains contents.
734 * This way empty rings will not cause backups to be overwritten and it's safe
735 * to reload multiple times. That's only best effort, failures are silently
736 * ignored.
737 */
738void sink_rotate_file_backed_ring(const char *name)
739{
740 struct ring ring;
741 char *oldback;
742 int ret;
743 int fd;
744
745 fd = open(name, O_RDONLY);
746 if (fd < 0)
747 return;
748
749 /* check for contents validity */
750 ret = read(fd, &ring, sizeof(ring));
751 close(fd);
752
753 if (ret != sizeof(ring))
754 goto rotate;
755
756 /* contents are present, we want to keep them => rotate. Note that
757 * an empty ring buffer has one byte (the marker).
758 */
759 if (ring.buf.data > 1)
760 goto rotate;
761
762 /* nothing to keep, let's scratch the file and preserve the backup */
763 return;
764
765 rotate:
766 oldback = NULL;
767 memprintf(&oldback, "%s.bak", name);
768 if (oldback) {
769 /* try to rename any possibly existing ring file to
770 * ".bak" and delete remains of older ones. This will
771 * ensure we don't wipe useful debug info upon restart.
772 */
773 unlink(oldback);
774 if (rename(name, oldback) < 0)
775 unlink(oldback);
776 ha_free(&oldback);
777 }
778}
779
Emeric Brun99c453d2020-05-25 15:01:04 +0200780/*
781 * Parse "ring" section and create corresponding sink buffer.
782 *
783 * The function returns 0 in success case, otherwise, it returns error
784 * flags.
785 */
786int cfg_parse_ring(const char *file, int linenum, char **args, int kwm)
787{
788 int err_code = 0;
789 const char *inv;
790 size_t size = BUFSIZE;
Emeric Brun494c5052020-05-28 11:13:15 +0200791 struct proxy *p;
Emeric Brun99c453d2020-05-25 15:01:04 +0200792
Willy Tarreau18d13062022-08-11 16:12:11 +0200793 if (strcmp(args[0], "ring") == 0) { /* new ring section */
Emeric Brun99c453d2020-05-25 15:01:04 +0200794 if (!*args[1]) {
795 ha_alert("parsing [%s:%d] : missing ring name.\n", file, linenum);
796 err_code |= ERR_ALERT | ERR_FATAL;
797 goto err;
798 }
799
800 inv = invalid_char(args[1]);
801 if (inv) {
802 ha_alert("parsing [%s:%d] : invalid ring name '%s' (character '%c' is not permitted).\n", file, linenum, args[1], *inv);
803 err_code |= ERR_ALERT | ERR_FATAL;
804 goto err;
805 }
806
807 if (sink_find(args[1])) {
808 ha_alert("parsing [%s:%d] : sink named '%s' already exists.\n", file, linenum, args[1]);
809 err_code |= ERR_ALERT | ERR_FATAL;
810 goto err;
811 }
812
Emeric Brun54648852020-07-06 15:54:06 +0200813 cfg_sink = sink_new_buf(args[1], args[1], LOG_FORMAT_RAW, size);
Emeric Brun99c453d2020-05-25 15:01:04 +0200814 if (!cfg_sink || cfg_sink->type != SINK_TYPE_BUFFER) {
815 ha_alert("parsing [%s:%d] : unable to create a new sink buffer for ring '%s'.\n", file, linenum, args[1]);
816 err_code |= ERR_ALERT | ERR_FATAL;
817 goto err;
818 }
Emeric Brun494c5052020-05-28 11:13:15 +0200819
820 /* allocate new proxy to handle forwards */
821 p = calloc(1, sizeof *p);
822 if (!p) {
823 ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum);
824 err_code |= ERR_ALERT | ERR_FATAL;
825 goto err;
826 }
827
828 init_new_proxy(p);
829 sink_setup_proxy(p);
830 p->parent = cfg_sink;
831 p->id = strdup(args[1]);
832 p->conf.args.file = p->conf.file = strdup(file);
833 p->conf.args.line = p->conf.line = linenum;
834 cfg_sink->forward_px = p;
Emeric Brun99c453d2020-05-25 15:01:04 +0200835 }
836 else if (strcmp(args[0], "size") == 0) {
Willy Tarreau18d13062022-08-11 16:12:11 +0200837 if (!cfg_sink || (cfg_sink->type != SINK_TYPE_BUFFER)) {
838 ha_alert("parsing [%s:%d] : 'size' directive not usable with this type of sink.\n", file, linenum);
839 err_code |= ERR_ALERT | ERR_FATAL;
840 goto err;
841 }
842
Emeric Brun99c453d2020-05-25 15:01:04 +0200843 size = atol(args[1]);
844 if (!size) {
845 ha_alert("parsing [%s:%d] : invalid size '%s' for new sink buffer.\n", file, linenum, args[1]);
846 err_code |= ERR_ALERT | ERR_FATAL;
847 goto err;
848 }
849
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +0200850 if (cfg_sink->store) {
851 ha_alert("parsing [%s:%d] : cannot resize an already mapped file, please specify 'size' before 'backing-file'.\n", file, linenum);
852 err_code |= ERR_ALERT | ERR_FATAL;
853 goto err;
854 }
855
Willy Tarreau18d13062022-08-11 16:12:11 +0200856 if (size < cfg_sink->ctx.ring->buf.size) {
857 ha_warning("parsing [%s:%d] : ignoring new size '%llu' that is smaller than current size '%llu' for ring '%s'.\n",
858 file, linenum, (ullong)size, (ullong)cfg_sink->ctx.ring->buf.size, cfg_sink->name);
859 err_code |= ERR_ALERT | ERR_FATAL;
860 goto err;
861 }
862
863 if (!ring_resize(cfg_sink->ctx.ring, size)) {
864 ha_alert("parsing [%s:%d] : fail to set sink buffer size '%llu' for ring '%s'.\n", file, linenum,
865 (ullong)cfg_sink->ctx.ring->buf.size, cfg_sink->name);
Emeric Brun99c453d2020-05-25 15:01:04 +0200866 err_code |= ERR_ALERT | ERR_FATAL;
867 goto err;
868 }
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +0200869 }
870 else if (strcmp(args[0], "backing-file") == 0) {
871 /* This tries to mmap file <file> for size <size> and to use it as a backing store
872 * for ring <ring>. Existing data are delete. NULL is returned on error.
873 */
874 const char *backing = args[1];
875 size_t size;
876 void *area;
877 int fd;
878
879 if (!cfg_sink || (cfg_sink->type != SINK_TYPE_BUFFER)) {
880 ha_alert("parsing [%s:%d] : 'backing-file' only usable with existing rings.\n", file, linenum);
881 err_code |= ERR_ALERT | ERR_FATAL;
882 goto err;
883 }
884
885 if (cfg_sink->store) {
886 ha_alert("parsing [%s:%d] : 'backing-file' already specified for ring '%s' (was '%s').\n", file, linenum, cfg_sink->name, cfg_sink->store);
887 err_code |= ERR_ALERT | ERR_FATAL;
888 goto err;
889 }
890
Willy Tarreau32872db2022-08-31 18:52:17 +0200891 /* let's check if the file exists and is not empty. That's the
892 * only condition under which we'll trigger a rotate, so that
893 * config checks, reloads, or restarts that don't emit anything
894 * do not rotate it again.
895 */
896 sink_rotate_file_backed_ring(backing);
Willy Tarreauded77cc2022-08-12 15:38:20 +0200897
Willy Tarreau8e877052022-08-12 15:03:12 +0200898 fd = open(backing, O_RDWR | O_CREAT, 0600);
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +0200899 if (fd < 0) {
900 ha_alert("parsing [%s:%d] : cannot open backing-file '%s' for ring '%s': %s.\n", file, linenum, backing, cfg_sink->name, strerror(errno));
901 err_code |= ERR_ALERT | ERR_FATAL;
902 goto err;
903 }
904
905 size = (cfg_sink->ctx.ring->buf.size + 4095UL) & -4096UL;
906 if (ftruncate(fd, size) != 0) {
907 close(fd);
908 ha_alert("parsing [%s:%d] : could not adjust size of backing-file for ring '%s': %s.\n", file, linenum, cfg_sink->name, strerror(errno));
909 err_code |= ERR_ALERT | ERR_FATAL;
910 goto err;
911 }
912
913 area = mmap(NULL, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
914 if (area == MAP_FAILED) {
915 close(fd);
916 ha_alert("parsing [%s:%d] : failed to use '%s' as a backing file for ring '%s': %s.\n", file, linenum, backing, cfg_sink->name, strerror(errno));
917 err_code |= ERR_ALERT | ERR_FATAL;
918 goto err;
919 }
920
921 /* we don't need the file anymore */
922 close(fd);
923 cfg_sink->store = strdup(backing);
924
925 /* never fails */
926 ring_free(cfg_sink->ctx.ring);
927 cfg_sink->ctx.ring = ring_make_from_area(area, size);
Emeric Brun99c453d2020-05-25 15:01:04 +0200928 }
Emeric Brun494c5052020-05-28 11:13:15 +0200929 else if (strcmp(args[0],"server") == 0) {
Willy Tarreau1b662aa2022-11-16 18:56:34 +0100930 if (!cfg_sink || (cfg_sink->type != SINK_TYPE_BUFFER)) {
931 ha_alert("parsing [%s:%d] : unable to create server '%s'.\n", file, linenum, args[1]);
932 err_code |= ERR_ALERT | ERR_FATAL;
933 goto err;
934 }
935
Amaury Denoyelle30c05372021-03-08 16:36:46 +0100936 err_code |= parse_server(file, linenum, args, cfg_sink->forward_px, NULL,
937 SRV_PARSE_PARSE_ADDR|SRV_PARSE_INITIAL_RESOLVE);
Emeric Brun494c5052020-05-28 11:13:15 +0200938 }
939 else if (strcmp(args[0],"timeout") == 0) {
940 if (!cfg_sink || !cfg_sink->forward_px) {
941 ha_alert("parsing [%s:%d] : unable to set timeout '%s'.\n", file, linenum, args[1]);
942 err_code |= ERR_ALERT | ERR_FATAL;
943 goto err;
944 }
945
946 if (strcmp(args[1], "connect") == 0 ||
947 strcmp(args[1], "server") == 0) {
948 const char *res;
949 unsigned int tout;
950
951 if (!*args[2]) {
952 ha_alert("parsing [%s:%d] : '%s %s' expects <time> as argument.\n",
953 file, linenum, args[0], args[1]);
954 err_code |= ERR_ALERT | ERR_FATAL;
955 goto err;
956 }
957 res = parse_time_err(args[2], &tout, TIME_UNIT_MS);
958 if (res == PARSE_TIME_OVER) {
959 ha_alert("parsing [%s:%d]: timer overflow in argument <%s> to <%s %s>, maximum value is 2147483647 ms (~24.8 days).\n",
960 file, linenum, args[2], args[0], args[1]);
961 err_code |= ERR_ALERT | ERR_FATAL;
962 goto err;
963 }
964 else if (res == PARSE_TIME_UNDER) {
965 ha_alert("parsing [%s:%d]: timer underflow in argument <%s> to <%s %s>, minimum non-null value is 1 ms.\n",
966 file, linenum, args[2], args[0], args[1]);
967 err_code |= ERR_ALERT | ERR_FATAL;
968 goto err;
969 }
970 else if (res) {
971 ha_alert("parsing [%s:%d]: unexpected character '%c' in argument to <%s %s>.\n",
972 file, linenum, *res, args[0], args[1]);
973 err_code |= ERR_ALERT | ERR_FATAL;
974 goto err;
975 }
Christopher Faulet321d1002022-10-19 16:26:21 +0200976 if (args[1][0] == 'c')
Emeric Brun494c5052020-05-28 11:13:15 +0200977 cfg_sink->forward_px->timeout.connect = tout;
978 else
979 cfg_sink->forward_px->timeout.server = tout;
980 }
981 }
Emeric Brun99c453d2020-05-25 15:01:04 +0200982 else if (strcmp(args[0],"format") == 0) {
983 if (!cfg_sink) {
984 ha_alert("parsing [%s:%d] : unable to set format '%s'.\n", file, linenum, args[1]);
985 err_code |= ERR_ALERT | ERR_FATAL;
986 goto err;
987 }
988
Emeric Brun54648852020-07-06 15:54:06 +0200989 cfg_sink->fmt = get_log_format(args[1]);
990 if (cfg_sink->fmt == LOG_FORMAT_UNSPEC) {
Emeric Brun99c453d2020-05-25 15:01:04 +0200991 ha_alert("parsing [%s:%d] : unknown format '%s'.\n", file, linenum, args[1]);
992 err_code |= ERR_ALERT | ERR_FATAL;
993 goto err;
994 }
995 }
996 else if (strcmp(args[0],"maxlen") == 0) {
997 if (!cfg_sink) {
998 ha_alert("parsing [%s:%d] : unable to set event max length '%s'.\n", file, linenum, args[1]);
999 err_code |= ERR_ALERT | ERR_FATAL;
1000 goto err;
1001 }
1002
1003 cfg_sink->maxlen = atol(args[1]);
1004 if (!cfg_sink->maxlen) {
1005 ha_alert("parsing [%s:%d] : invalid size '%s' for new sink buffer.\n", file, linenum, args[1]);
1006 err_code |= ERR_ALERT | ERR_FATAL;
1007 goto err;
1008 }
1009 }
1010 else if (strcmp(args[0],"description") == 0) {
1011 if (!cfg_sink) {
1012 ha_alert("parsing [%s:%d] : unable to set description '%s'.\n", file, linenum, args[1]);
1013 err_code |= ERR_ALERT | ERR_FATAL;
1014 goto err;
1015 }
1016
1017 if (!*args[1]) {
1018 ha_alert("parsing [%s:%d] : missing ring description text.\n", file, linenum);
1019 err_code |= ERR_ALERT | ERR_FATAL;
1020 goto err;
1021 }
1022
1023 free(cfg_sink->desc);
1024
1025 cfg_sink->desc = strdup(args[1]);
1026 if (!cfg_sink->desc) {
1027 ha_alert("parsing [%s:%d] : fail to set description '%s'.\n", file, linenum, args[1]);
1028 err_code |= ERR_ALERT | ERR_FATAL;
1029 goto err;
1030 }
1031 }
Emeric Brun9f2ff3a2020-05-29 15:47:52 +02001032 else {
1033 ha_alert("parsing [%s:%d] : unknown statement '%s'.\n", file, linenum, args[0]);
1034 err_code |= ERR_ALERT | ERR_FATAL;
1035 goto err;
1036 }
Emeric Brun99c453d2020-05-25 15:01:04 +02001037
1038err:
1039 return err_code;
1040}
1041
Emeric Brun94aab062021-04-02 10:41:36 +02001042/* Creates an new sink buffer from a log server.
1043 *
1044 * It uses the logsrvaddress to declare a forward
1045 * server for this buffer. And it initializes the
1046 * forwarding.
1047 *
1048 * The function returns a pointer on the
1049 * allocated struct sink if allocate
1050 * and initialize succeed, else if it fails
1051 * it returns NULL.
1052 *
1053 * Note: the sink is created using the name
Ilya Shipitsinb2be9a12021-04-24 13:25:42 +05001054 * specified into logsrv->ring_name
Emeric Brun94aab062021-04-02 10:41:36 +02001055 */
1056struct sink *sink_new_from_logsrv(struct logsrv *logsrv)
1057{
1058 struct proxy *p = NULL;
1059 struct sink *sink = NULL;
1060 struct server *srv = NULL;
1061 struct sink_forward_target *sft = NULL;
Emeric Brun94aab062021-04-02 10:41:36 +02001062
1063 /* allocate new proxy to handle
1064 * forward to a stream server
1065 */
1066 p = calloc(1, sizeof *p);
1067 if (!p) {
1068 goto error;
1069 }
1070
1071 init_new_proxy(p);
1072 sink_setup_proxy(p);
1073 p->id = strdup(logsrv->ring_name);
1074 p->conf.args.file = p->conf.file = strdup(logsrv->conf.file);
1075 p->conf.args.line = p->conf.line = logsrv->conf.line;
1076
Christopher Fauletd08a25b2022-10-24 15:53:01 +02001077 /* Set default connect and server timeout */
1078 p->timeout.connect = MS_TO_TICKS(1000);
1079 p->timeout.server = MS_TO_TICKS(5000);
1080
Emeric Brun94aab062021-04-02 10:41:36 +02001081 /* allocate a new server to forward messages
1082 * from ring buffer
1083 */
1084 srv = new_server(p);
1085 if (!srv)
1086 goto error;
1087
1088 /* init server */
1089 srv->id = strdup(logsrv->ring_name);
1090 srv->conf.file = strdup(logsrv->conf.file);
1091 srv->conf.line = logsrv->conf.line;
1092 srv->addr = logsrv->addr;
1093 srv->svc_port = get_host_port(&logsrv->addr);
1094 HA_SPIN_INIT(&srv->lock);
1095
1096 /* process per thread init */
Miroslav Zagorac8a8f2702021-06-15 15:33:20 +02001097 if (srv_init_per_thr(srv) == -1)
Emeric Brun94aab062021-04-02 10:41:36 +02001098 goto error;
1099
Emeric Brun94aab062021-04-02 10:41:36 +02001100 /* the servers are linked backwards
1101 * first into proxy
1102 */
1103 p->srv = srv;
1104 srv->next = p->srv;
1105
1106 /* allocate sink_forward_target descriptor */
1107 sft = calloc(1, sizeof(*sft));
1108 if (!sft)
1109 goto error;
1110
1111 /* init sink_forward_target offset */
1112 sft->srv = srv;
1113 sft->appctx = NULL;
1114 sft->ofs = ~0;
1115 HA_SPIN_INIT(&sft->lock);
1116
Ilya Shipitsinb2be9a12021-04-24 13:25:42 +05001117 /* prepare description for the sink */
Emeric Brun94aab062021-04-02 10:41:36 +02001118 chunk_reset(&trash);
1119 chunk_printf(&trash, "created from logserver declared into '%s' at line %d", logsrv->conf.file, logsrv->conf.line);
1120
1121 /* allocate a new sink buffer */
1122 sink = sink_new_buf(logsrv->ring_name, trash.area, logsrv->format, BUFSIZE);
1123 if (!sink || sink->type != SINK_TYPE_BUFFER) {
1124 goto error;
1125 }
1126
1127 /* link sink_forward_target to proxy */
1128 sink->forward_px = p;
1129 p->parent = sink;
1130
1131 /* insert into sink_forward_targets
1132 * list into sink
1133 */
Christopher Faulet2ae25ea2022-05-12 14:50:09 +02001134 sft->sink = sink;
Emeric Brun94aab062021-04-02 10:41:36 +02001135 sft->next = sink->sft;
1136 sink->sft = sft;
1137
1138 /* mark server as an attached reader to the ring */
1139 if (!ring_attach(sink->ctx.ring)) {
1140 /* should never fail since there is
1141 * only one reader
1142 */
1143 goto error;
1144 }
1145
1146 /* initialize sink buffer forwarding */
1147 if (!sink_init_forward(sink))
1148 goto error;
1149
1150 /* reset familyt of logsrv to consider the ring buffer target */
1151 logsrv->addr.ss_family = AF_UNSPEC;
1152
1153 return sink;
1154error:
1155 if (p) {
1156 if (p->id)
1157 free(p->id);
1158 if (p->conf.file)
1159 free(p->conf.file);
1160
1161 free(p);
1162 }
1163
1164 if (srv) {
1165 if (srv->id)
1166 free(srv->id);
1167 if (srv->conf.file)
1168 free((void *)srv->conf.file);
1169 if (srv->per_thr)
1170 free(srv->per_thr);
1171 free(srv);
1172 }
1173
1174 if (sft)
1175 free(sft);
1176
1177 if (sink) {
1178 if (sink->ctx.ring)
1179 ring_free(sink->ctx.ring);
1180
Willy Tarreau2b718102021-04-21 07:32:39 +02001181 LIST_DELETE(&sink->sink_list);
Emeric Brun94aab062021-04-02 10:41:36 +02001182 free(sink->name);
1183 free(sink->desc);
1184 free(sink);
1185 }
1186
1187 return NULL;
1188}
1189
Emeric Brun99c453d2020-05-25 15:01:04 +02001190/*
1191 * Post parsing "ring" section.
1192 *
1193 * The function returns 0 in success case, otherwise, it returns error
1194 * flags.
1195 */
1196int cfg_post_parse_ring()
1197{
1198 int err_code = 0;
Emeric Brun494c5052020-05-28 11:13:15 +02001199 struct server *srv;
Emeric Brun99c453d2020-05-25 15:01:04 +02001200
1201 if (cfg_sink && (cfg_sink->type == SINK_TYPE_BUFFER)) {
1202 if (cfg_sink->maxlen > b_size(&cfg_sink->ctx.ring->buf)) {
1203 ha_warning("ring '%s' event max length '%u' exceeds size, forced to size '%lu'.\n",
Willy Tarreau570a22b2020-06-02 12:00:46 +02001204 cfg_sink->name, cfg_sink->maxlen, (unsigned long)b_size(&cfg_sink->ctx.ring->buf));
Emeric Brun99c453d2020-05-25 15:01:04 +02001205 cfg_sink->maxlen = b_size(&cfg_sink->ctx.ring->buf);
1206 err_code |= ERR_ALERT;
1207 }
Emeric Brun494c5052020-05-28 11:13:15 +02001208
1209 /* prepare forward server descriptors */
1210 if (cfg_sink->forward_px) {
1211 srv = cfg_sink->forward_px->srv;
1212 while (srv) {
1213 struct sink_forward_target *sft;
Emeric Brun99c453d2020-05-25 15:01:04 +02001214
Emeric Brun494c5052020-05-28 11:13:15 +02001215 /* allocate sink_forward_target descriptor */
1216 sft = calloc(1, sizeof(*sft));
1217 if (!sft) {
1218 ha_alert("memory allocation error initializing server '%s' in ring '%s'.\n",srv->id, cfg_sink->name);
1219 err_code |= ERR_ALERT | ERR_FATAL;
1220 break;
1221 }
1222 sft->srv = srv;
1223 sft->appctx = NULL;
1224 sft->ofs = ~0; /* init ring offset */
Christopher Faulet96417f32022-08-04 16:00:13 +02001225 sft->sink = cfg_sink;
Emeric Brun494c5052020-05-28 11:13:15 +02001226 sft->next = cfg_sink->sft;
1227 HA_SPIN_INIT(&sft->lock);
1228
1229 /* mark server attached to the ring */
1230 if (!ring_attach(cfg_sink->ctx.ring)) {
1231 ha_alert("server '%s' sets too many watchers > 255 on ring '%s'.\n", srv->id, cfg_sink->name);
1232 err_code |= ERR_ALERT | ERR_FATAL;
1233 }
1234 cfg_sink->sft = sft;
1235 srv = srv->next;
1236 }
1237 sink_init_forward(cfg_sink);
1238 }
1239 }
Emeric Brun99c453d2020-05-25 15:01:04 +02001240 cfg_sink = NULL;
1241
1242 return err_code;
1243}
1244
1245/* resolve sink names at end of config. Returns 0 on success otherwise error
1246 * flags.
1247*/
1248int post_sink_resolve()
1249{
Christopher Fauletfc633b62020-11-06 15:24:23 +01001250 int err_code = ERR_NONE;
Emeric Brun99c453d2020-05-25 15:01:04 +02001251 struct logsrv *logsrv, *logb;
1252 struct sink *sink;
1253 struct proxy *px;
1254
1255 list_for_each_entry_safe(logsrv, logb, &global.logsrvs, list) {
1256 if (logsrv->type == LOG_TARGET_BUFFER) {
1257 sink = sink_find(logsrv->ring_name);
Emeric Brun94aab062021-04-02 10:41:36 +02001258 if (!sink) {
1259 /* LOG_TARGET_BUFFER but !AF_UNSPEC
1260 * means we must allocate a sink
1261 * buffer to send messages to this logsrv
1262 */
1263 if (logsrv->addr.ss_family != AF_UNSPEC) {
1264 sink = sink_new_from_logsrv(logsrv);
1265 if (!sink) {
1266 ha_alert("global stream log server declared in file '%s' at line %d cannot be initialized'.\n",
1267 logsrv->conf.file, logsrv->conf.line);
1268 err_code |= ERR_ALERT | ERR_FATAL;
1269 }
1270 }
1271 else {
1272 ha_alert("global log server declared in file '%s' at line %d uses unknown ring named '%s'.\n",
1273 logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
1274 err_code |= ERR_ALERT | ERR_FATAL;
1275 }
1276 }
1277 else if (sink->type != SINK_TYPE_BUFFER) {
1278 ha_alert("global log server declared in file '%s' at line %d uses incompatible ring '%s'.\n",
1279 logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
Emeric Brun99c453d2020-05-25 15:01:04 +02001280 err_code |= ERR_ALERT | ERR_FATAL;
1281 }
1282 logsrv->sink = sink;
1283 }
Emeric Brun94aab062021-04-02 10:41:36 +02001284
Emeric Brun99c453d2020-05-25 15:01:04 +02001285 }
1286
1287 for (px = proxies_list; px; px = px->next) {
1288 list_for_each_entry_safe(logsrv, logb, &px->logsrvs, list) {
1289 if (logsrv->type == LOG_TARGET_BUFFER) {
1290 sink = sink_find(logsrv->ring_name);
Emeric Brun94aab062021-04-02 10:41:36 +02001291 if (!sink) {
1292 /* LOG_TARGET_BUFFER but !AF_UNSPEC
1293 * means we must allocate a sink
1294 * buffer to send messages to this logsrv
1295 */
1296 if (logsrv->addr.ss_family != AF_UNSPEC) {
1297 sink = sink_new_from_logsrv(logsrv);
1298 if (!sink) {
1299 ha_alert("log server declared in proxy section '%s' file '%s' at line %d cannot be initialized'.\n",
1300 px->id, logsrv->conf.file, logsrv->conf.line);
1301 err_code |= ERR_ALERT | ERR_FATAL;
1302 }
1303 }
1304 else {
1305 ha_alert("log server declared in proxy section '%s' in file '%s' at line %d uses unknown ring named '%s'.\n",
1306 px->id, logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
1307 err_code |= ERR_ALERT | ERR_FATAL;
1308 }
1309 }
1310 else if (sink->type != SINK_TYPE_BUFFER) {
1311 ha_alert("log server declared in proxy section '%s' in file '%s' at line %d uses incomatible ring named '%s'.\n",
1312 px->id, logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
Emeric Brun99c453d2020-05-25 15:01:04 +02001313 err_code |= ERR_ALERT | ERR_FATAL;
1314 }
1315 logsrv->sink = sink;
1316 }
1317 }
1318 }
Emeric Brun12941c82020-07-07 14:19:42 +02001319
1320 for (px = cfg_log_forward; px; px = px->next) {
1321 list_for_each_entry_safe(logsrv, logb, &px->logsrvs, list) {
1322 if (logsrv->type == LOG_TARGET_BUFFER) {
1323 sink = sink_find(logsrv->ring_name);
Emeric Brun94aab062021-04-02 10:41:36 +02001324 if (!sink) {
1325 /* LOG_TARGET_BUFFER but !AF_UNSPEC
1326 * means we must allocate a sink
1327 * buffer to send messages to this logsrv
1328 */
1329 if (logsrv->addr.ss_family != AF_UNSPEC) {
1330 sink = sink_new_from_logsrv(logsrv);
1331 if (!sink) {
1332 ha_alert("log server declared in log-forward section '%s' file '%s' at line %d cannot be initialized'.\n",
1333 px->id, logsrv->conf.file, logsrv->conf.line);
1334 err_code |= ERR_ALERT | ERR_FATAL;
1335 }
1336 }
1337 else {
1338 ha_alert("log server declared in log-forward section '%s' in file '%s' at line %d uses unknown ring named '%s'.\n",
1339 px->id, logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
1340 err_code |= ERR_ALERT | ERR_FATAL;
1341 }
1342 }
1343 else if (sink->type != SINK_TYPE_BUFFER) {
1344 ha_alert("log server declared in log-forward section '%s' in file '%s' at line %d uses unknown ring named '%s'.\n",
1345 px->id, logsrv->conf.file, logsrv->conf.line, logsrv->ring_name);
Emeric Brun12941c82020-07-07 14:19:42 +02001346 err_code |= ERR_ALERT | ERR_FATAL;
1347 }
1348 logsrv->sink = sink;
1349 }
1350 }
1351 }
Emeric Brun99c453d2020-05-25 15:01:04 +02001352 return err_code;
1353}
1354
1355
Willy Tarreau973e6622019-08-20 11:57:52 +02001356static void sink_init()
1357{
Emeric Brun54648852020-07-06 15:54:06 +02001358 sink_new_fd("stdout", "standard output (fd#1)", LOG_FORMAT_RAW, 1);
1359 sink_new_fd("stderr", "standard output (fd#2)", LOG_FORMAT_RAW, 2);
1360 sink_new_buf("buf0", "in-memory ring buffer", LOG_FORMAT_TIMED, 1048576);
Willy Tarreau4ed23ca2019-08-23 15:47:49 +02001361}
1362
1363static void sink_deinit()
1364{
1365 struct sink *sink, *sb;
1366
1367 list_for_each_entry_safe(sink, sb, &sink_list, sink_list) {
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +02001368 if (sink->type == SINK_TYPE_BUFFER) {
Willy Tarreaufb9a4762023-01-24 12:11:41 +01001369 if (sink->store) {
1370 size_t size = (sink->ctx.ring->buf.size + 4095UL) & -4096UL;
1371 void *area = (sink->ctx.ring->buf.area - sizeof(*sink->ctx.ring));
1372
1373 msync(area, size, MS_SYNC);
1374 munmap(area, size);
Willy Tarreaub9191092023-01-26 15:34:31 +01001375 ha_free(&sink->store);
Willy Tarreaufb9a4762023-01-24 12:11:41 +01001376 }
Willy Tarreau0b8e9ce2022-08-11 16:38:20 +02001377 else
1378 ring_free(sink->ctx.ring);
1379 }
Willy Tarreau2b718102021-04-21 07:32:39 +02001380 LIST_DELETE(&sink->sink_list);
Willy Tarreau09727ee2023-01-26 15:46:08 +01001381 task_destroy(sink->forward_task);
Emeric Brun99c453d2020-05-25 15:01:04 +02001382 free(sink->name);
1383 free(sink->desc);
Willy Tarreau4ed23ca2019-08-23 15:47:49 +02001384 free(sink);
1385 }
Willy Tarreau973e6622019-08-20 11:57:52 +02001386}
1387
1388INITCALL0(STG_REGISTER, sink_init);
Willy Tarreau4ed23ca2019-08-23 15:47:49 +02001389REGISTER_POST_DEINIT(sink_deinit);
Willy Tarreau973e6622019-08-20 11:57:52 +02001390
Willy Tarreau9f830d72019-08-26 18:17:04 +02001391static struct cli_kw_list cli_kws = {{ },{
Willy Tarreaub205bfd2021-05-07 11:38:37 +02001392 { { "show", "events", NULL }, "show events [<sink>] [-w] [-n] : show event sink state", cli_parse_show_events, NULL, NULL },
Willy Tarreau9f830d72019-08-26 18:17:04 +02001393 {{},}
1394}};
1395
1396INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws);
1397
Emeric Brun99c453d2020-05-25 15:01:04 +02001398/* config parsers for this section */
1399REGISTER_CONFIG_SECTION("ring", cfg_parse_ring, cfg_post_parse_ring);
1400REGISTER_POST_CHECK(post_sink_resolve);
1401
Willy Tarreau67b5a162019-08-11 16:38:56 +02001402/*
1403 * Local variables:
1404 * c-indent-level: 8
1405 * c-basic-offset: 8
1406 * End:
1407 */