/*
 * 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 "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 *buf, int i)
{
	int val = buf[i];

	if (val == ' ' || val == ',')
		return 1;

	return 0;
}

static void __tprintf_flush(void)
{
	size_t i;
	static ssize_t line_count = 0;
	ssize_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))
				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 ((size_t) 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++;
	}
}
ption value='5'>5</option><option value='6'>6</option><option value='7'>7</option><option value='8'>8</option><option value='9'>9</option><option value='10'>10</option><option value='15'>15</option><option value='20'>20</option><option value='25'>25</option><option value='30'>30</option><option value='35'>35</option><option value='40'>40</option></select></td></tr><tr><td class='label'>space:</td><td class='ctrl'><select name='ignorews' onchange='this.form.submit();'><option value='0' selected='selected'>include</option><option value='1'>ignore</option></select></td></tr><tr><td class='label'>mode:</td><td class='ctrl'><select name='dt' onchange='this.form.submit();'><option value='0' selected='selected'>unified</option><option value='1'>ssdiff</option><option value='2'>stat only</option></select></td></tr><tr><td/><td class='ctrl'><noscript><input type='submit' value='reload'/></noscript></td></tr></table></form></div><table summary='commit info' class='commit-info'>
<tr><th>author</th><td>Dan Carpenter &lt;dan.carpenter@oracle.com&gt;</td><td class='right'>2016-05-26 11:06:53 +0200</td></tr>
<tr><th>committer</th><td>Jean Delvare &lt;jdelvare@suse.de&gt;</td><td class='right'>2016-05-26 11:06:53 +0200</td></tr>
<tr><th>commit</th><td colspan='2' class='oid'><a href='/cgit.cgi/linux/net-next.git/commit/?id=54f0ffc4e249b7f7e7fffb10c83e8de99aca839c'>54f0ffc4e249b7f7e7fffb10c83e8de99aca839c</a> (<a href='/cgit.cgi/linux/net-next.git/patch/?id=54f0ffc4e249b7f7e7fffb10c83e8de99aca839c'>patch</a>)</td></tr>
<tr><th>tree</th><td colspan='2' class='oid'><a href='/cgit.cgi/linux/net-next.git/tree/?id=54f0ffc4e249b7f7e7fffb10c83e8de99aca839c'>355069176980e9a4ed3848d40f847a27483fa3d9</a></td></tr>
<tr><th>parent</th><td colspan='2' class='oid'><a href='/cgit.cgi/linux/net-next.git/commit/?id=2dcd0af568b0cf583645c8a317dd12e344b1c72a'>2dcd0af568b0cf583645c8a317dd12e344b1c72a</a> (<a href='/cgit.cgi/linux/net-next.git/diff/?id=54f0ffc4e249b7f7e7fffb10c83e8de99aca839c&amp;id2=2dcd0af568b0cf583645c8a317dd12e344b1c72a'>diff</a>)</td></tr></table>
<div class='commit-subject'>hwmon: (emc2103) Fix typo in MODULE_PARM_DESC</div><div class='commit-msg'>"apd" was intended here instead of "init".

Signed-off-by: Dan Carpenter &lt;dan.carpenter@oracle.com&gt;
Signed-off-by: Jean Delvare &lt;jdelvare@suse.de&gt;
</div><div class='diffstat-header'><a href='/cgit.cgi/linux/net-next.git/diff/?id=54f0ffc4e249b7f7e7fffb10c83e8de99aca839c'>Diffstat</a></div><table summary='diffstat' class='diffstat'>