summaryrefslogtreecommitdiffstats
path: root/stdiobufhack/stdiobufhack.c
blob: 604c54604f12447f0ed172ee193b915d9a723017 (plain)
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
/* Simple LD_PRELOAD library to force stdout/stderr into line-buffered mode.
 * This file is released to the Public Domain. */

#define _GNU_SOURCE

#include <stdio.h>
#include <stdlib.h>
#include <dlfcn.h>

#define TARGET_MODE	_IOLBF


static int (*libc_setvbuf)(FILE *stream, char *buf, int mode, size_t size);
static int in_stdiobufhack;

int setvbuf(FILE *stream, char *buf, int mode, size_t size)
{
	if (in_stdiobufhack)
		return 0;
	in_stdiobufhack = 1;

	if (!libc_setvbuf)
		libc_setvbuf = dlsym(RTLD_NEXT, "setvbuf");
	if (!libc_setvbuf) {
		fprintf(stderr, "stdiobufhack: Did not find libc setvbuf()\n");
		exit(1);
	}

	if (stream == stdout || stream == stderr)
		mode = TARGET_MODE;

	in_stdiobufhack = 0;

	return libc_setvbuf(stream, buf, mode, size);
}

static void __attribute__((__constructor__)) stdiobufhack_ctor(void)
{
	setvbuf(stdout, NULL, TARGET_MODE, 1024);
	setvbuf(stderr, NULL, TARGET_MODE, 1024);
}
bues.ch cgit interface