diff options
author | Daniel Borkmann <dborkman@redhat.com> | 2013-03-15 10:41:48 +0100 |
---|---|---|
committer | Daniel Borkmann <dborkman@redhat.com> | 2013-03-15 10:41:48 +0100 |
commit | 1a9fbac03c684f29cff9ac44875bd9504a89f54e (patch) | |
tree | 1b2e40dbe5dc1899ef5b62c4325c9b94c9c450fc /tprintf.c |
all: import netsniff-ng 0.5.8-rc0 source
We decided to get rid of the old Git history and start a new one for
several reasons:
*) Allow / enforce only high-quality commits (which was not the case
for many commits in the history), have a policy that is more close
to the one from the Linux kernel. With high quality commits, we
mean code that is logically split into commits and commit messages
that are signed-off and have a proper subject and message body.
We do not allow automatic Github merges anymore, since they are
total bullshit. However, we will either cherry-pick your patches
or pull them manually.
*) The old archive was about ~27MB for no particular good reason.
This basically derived from the bad decision that also some PDF
files where stored there. From this moment onwards, no binary
objects are allowed to be stored in this repository anymore.
The old archive is not wiped away from the Internet. You will still
be able to find it, e.g. on git.cryptoism.org etc.
Signed-off-by: Daniel Borkmann <dborkman@redhat.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>
Diffstat (limited to 'tprintf.c')
-rw-r--r-- | tprintf.c | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/tprintf.c b/tprintf.c new file mode 100644 index 0000000..46e9574 --- /dev/null +++ b/tprintf.c @@ -0,0 +1,169 @@ +/* + * netsniff-ng - the packet sniffing beast + * Copyright 2009, 2010 Daniel Borkmann. + * Copyright 2013 Tobias Klauser. + * Subject to the GPL, version 2. + */ + +#define _BSD_SOURCE +#include <ctype.h> +#include <stdio.h> +#include <stdarg.h> +#include <sys/ioctl.h> + +#include "xutils.h" +#include "tprintf.h" +#include "die.h" +#include "locking.h" +#include "built_in.h" + +#define term_trailing_size 5 +#define term_starting_size 3 + +#define term_curr_size (get_tty_size() - term_trailing_size) + +static char buffer[1024]; + +static volatile size_t buffer_use = 0; + +static struct spinlock buffer_lock; + +static int get_tty_size(void) +{ +#ifdef TIOCGSIZE + struct ttysize ts = {0}; + + return (ioctl(0, TIOCGSIZE, &ts) == 0 ? ts.ts_cols : DEFAULT_TTY_SIZE); +#elif defined(TIOCGWINSZ) + struct winsize ts; + + return (ioctl(0, TIOCGWINSZ, &ts) == 0 ? ts.ws_col : DEFAULT_TTY_SIZE); +#else + return DEFAULT_TTY_SIZE; +#endif +} + +static inline void __tprintf_flush_newline(void) +{ + int i; + + fputc('\n', stdout); + for (i = 0; i < term_starting_size; ++i) + fputc(' ', stdout); +} + +static inline int __tprintf_flush_skip(char *buffer, int i, size_t max) +{ + int val = buffer[i]; + + if (val == ' ' || val == ',') + return 1; + + return 0; +} + +static void __tprintf_flush(void) +{ + int i; + static ssize_t line_count = 0; + size_t term_len = term_curr_size; + + for (i = 0; i < buffer_use; ++i) { + if (buffer[i] == '\n') { + term_len = term_curr_size; + line_count = -1; + } + + if (line_count == term_len) { + __tprintf_flush_newline(); + line_count = term_starting_size; + + while (i < buffer_use && + __tprintf_flush_skip(buffer, i, buffer_use)) + i++; + } + + fputc(buffer[i], stdout); + line_count++; + } + + fflush(stdout); + buffer_use = 0; +} + +void tprintf_flush(void) +{ + spinlock_lock(&buffer_lock); + __tprintf_flush(); + spinlock_unlock(&buffer_lock); +} + +void tprintf_init(void) +{ + spinlock_init(&buffer_lock); + + setvbuf(stdout, NULL, _IONBF, 0); + setvbuf(stderr, NULL, _IONBF, 0); +} + +void tprintf_cleanup(void) +{ + tprintf_flush(); + spinlock_destroy(&buffer_lock); +} + +void tprintf(char *msg, ...) +{ + ssize_t ret; + ssize_t avail; + va_list vl; + + spinlock_lock(&buffer_lock); + + avail = sizeof(buffer) - buffer_use; + bug_on(avail < 0); + + va_start(vl, msg); + ret = vsnprintf(buffer + buffer_use, avail, msg, vl); + va_end(vl); + + if (ret < 0) + panic("vsnprintf screwed up in tprintf!\n"); + if (ret > sizeof(buffer)) + panic("No mem in tprintf left!\n"); + if (ret >= avail) { + __tprintf_flush(); + + avail = sizeof(buffer) - buffer_use; + bug_on(avail < 0); + + va_start(vl, msg); + ret = vsnprintf(buffer + buffer_use, avail, msg, vl); + va_end(vl); + + if (ret < 0) + panic("vsnprintf screwed up in tprintf!\n"); + } + + buffer_use += ret; + + spinlock_unlock(&buffer_lock); +} + +void tputchar_safe(int c) +{ + unsigned char ch = (unsigned char)(c & 0xff); + + if (isprint(ch)) + tprintf("%c", ch); + else + tprintf("\\0x%02x", ch); +} + +void tputs_safe(const char *str, size_t len) +{ + while (len--) { + tputchar_safe(*str); + str++; + } +} |