util.c - platform/external/tcpdump - Git at Google

 /*
  * Copyright (c) 1990, 1991, 1993, 1994, 1995, 1996, 1997
  *	The Regents of the University of California.  All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that: (1) source code distributions
  * retain the above copyright notice and this paragraph in its entirety, (2)
  * distributions including binary code include the above copyright notice and
  * this paragraph in its entirety in the documentation or other materials
  * provided with the distribution, and (3) all advertising materials mentioning
  * features or use of this software display the following acknowledgement:
  * ``This product includes software developed by the University of California,
  * Lawrence Berkeley Laboratory and its contributors.'' Neither the name of
  * the University nor the names of its contributors may be used to endorse
  * or promote products derived from this software without specific prior
  * written permission.
  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
  * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  */

 #ifndef lint
 static const char rcsid[] =
     "@(#) $Header: /tcpdump/master/tcpdump/util.c,v 1.73 2002-06-11 17:09:01 itojun Exp $ (LBL)";
 #endif

 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif

 #include <sys/types.h>
 #include <sys/time.h>
 #include <sys/file.h>
 #include <sys/stat.h>

 #include <ctype.h>
 #include <errno.h>
 #ifdef HAVE_FCNTL_H
 #include <fcntl.h>
 #endif
 #include <pcap.h>
 #include <stdio.h>
 #include <stdarg.h>
 #include <stdlib.h>
 #include <string.h>
 #ifdef TIME_WITH_SYS_TIME
 #include <time.h>
 #endif
 #include <unistd.h>

 #include "interface.h"

 /*
  * Print out a filename (or other ascii string).
  * If ep is NULL, assume no truncation check is needed.
  * Return true if truncated.
  */
 int
 fn_print(register const u_char *s, register const u_char *ep)
 {
 	register int ret;
 	register u_char c;

 	ret = 1;			/* assume truncated */
 	while (ep == NULL || s < ep) {
 		c = *s++;
 		if (c == '\0') {
 			ret = 0;
 			break;
 		}
 		if (!isascii(c)) {
 			c = toascii(c);
 			putchar('M');
 			putchar('-');
 		}
 		if (!isprint(c)) {
 			c ^= 0x40;	/* DEL to ?, others to alpha */
 			putchar('^');
 		}
 		putchar(c);
 	}
 	return(ret);
 }

 /*
  * Print out a counted filename (or other ascii string).
  * If ep is NULL, assume no truncation check is needed.
  * Return true if truncated.
  */
 int
 fn_printn(register const u_char *s, register u_int n,
 	  register const u_char *ep)
 {
 	register int ret;
 	register u_char c;

 	ret = 1;			/* assume truncated */
 	while (ep == NULL || s < ep) {
 		if (n-- <= 0) {
 			ret = 0;
 			break;
 		}
 		c = *s++;
 		if (!isascii(c)) {
 			c = toascii(c);
 			putchar('M');
 			putchar('-');
 		}
 		if (!isprint(c)) {
 			c ^= 0x40;	/* DEL to ?, others to alpha */
 			putchar('^');
 		}
 		putchar(c);
 	}
 	return(ret);
 }

 /*
  * Print the timestamp
  */
 void
 ts_print(register const struct timeval *tvp)
 {
 	register int s;
 	struct tm *tm;
 	time_t Time;
 	static unsigned b_sec;
 	static unsigned b_usec;

 	switch(tflag) {
 	case 1: /* Default */
 		s = (tvp->tv_sec + thiszone) % 86400;
 		(void)printf("%02d:%02d:%02d.%06u ",
 			     s / 3600, (s % 3600) / 60, s % 60,
 			     (unsigned)tvp->tv_usec);
 		break;
 	case -1: /* Unix timeval style */
 		(void)printf("%u.%06u ",
 			     (unsigned)tvp->tv_sec,
 			     (unsigned)tvp->tv_usec);
 		break;
 	case -2:
 		if (b_sec == 0) {
 			printf("000000 ");
 		} else {
 			int d_usec = tvp->tv_usec - b_usec;
 			int d_sec = tvp->tv_sec - b_sec;

 			while (d_usec < 0) {
 				d_usec += 1000000;
 				d_sec--;
 			}
 			if (d_sec)
 				printf("%d. ", d_sec);
 			printf("%06d ", d_usec);
 		}
 		b_sec = tvp->tv_sec;
 		b_usec = tvp->tv_usec;
 		break;
 	case -3: /* Default + Date*/
 		s = (tvp->tv_sec + thiszone) % 86400;
 		Time = (tvp->tv_sec + thiszone) - s;
 		tm  = gmtime (&Time);
 		(void)printf("%02d/%02d/%04d %02d:%02d:%02d.%06u ",
 			     tm->tm_mon+1, tm->tm_mday,
 			     tm->tm_year+1900,
 			     s / 3600, (s % 3600) / 60,
 			     s % 60, (unsigned)tvp->tv_usec);
 		break;
 	}
 }

 /*
  * Print a relative number of seconds (e.g. hold time, prune timer)
  * in the form 5m1s.  This does no truncation, so 32230861 seconds
  * is represented as 1y1w1d1h1m1s.
  */
 void
 relts_print(int secs)
 {
 	static const char *lengths[] = {"y", "w", "d", "h", "m", "s"};
 	static const int seconds[] = {31536000, 604800, 86400, 3600, 60, 1};
 	const char **l = lengths;
 	const int *s = seconds;

 	if (secs == 0) {
 		(void)printf("0s");
 		return;
 	}
 	if (secs < 0) {
 		(void)printf("-");
 		secs = -secs;
 	}
 	while (secs > 0) {
 		if (secs >= *s) {
 			(void)printf("%d%s", secs / *s, *l);
 			secs -= (secs / *s) * *s;
 		}
 		s++;
 		l++;
 	}
 }

 /*
  * Convert a token value to a string; use "fmt" if not found.
  */
 const char *
 tok2str(register const struct tok *lp, register const char *fmt,
 	register int v)
 {
 	static char buf[128];

 	while (lp->s != NULL) {
 		if (lp->v == v)
 			return (lp->s);
 		++lp;
 	}
 	if (fmt == NULL)
 		fmt = "#%d";
 	(void)snprintf(buf, sizeof(buf), fmt, v);
 	return (buf);
 }

 /*
  * Convert a value to a string using an array; the macro
  * tok2strary() in <interface.h> is the public interface to
  * this function and ensures that the second argument is
  * correct for bounds-checking.
  */
 const char *
 tok2strary_internal(register const char **lp, int n, register const char *fmt,
 	register int v)
 {
 	static char buf[128];

 	if (v >= 0 && v < n && lp[v] != NULL)
 		return lp[v];
 	if (fmt == NULL)
 		fmt = "#%d";
 	(void)snprintf(buf, sizeof(buf), fmt, v);
 	return (buf);
 }

 /* VARARGS */
 void
 error(const char *fmt, ...)
 {
 	va_list ap;

 	(void)fprintf(stderr, "%s: ", program_name);
 	va_start(ap, fmt);
 	(void)vfprintf(stderr, fmt, ap);
 	va_end(ap);
 	if (*fmt) {
 		fmt += strlen(fmt);
 		if (fmt[-1] != '\n')
 			(void)fputc('\n', stderr);
 	}
 	exit(1);
 	/* NOTREACHED */
 }

 /* VARARGS */
 void
 warning(const char *fmt, ...)
 {
 	va_list ap;

 	(void)fprintf(stderr, "%s: WARNING: ", program_name);
 	va_start(ap, fmt);
 	(void)vfprintf(stderr, fmt, ap);
 	va_end(ap);
 	if (*fmt) {
 		fmt += strlen(fmt);
 		if (fmt[-1] != '\n')
 			(void)fputc('\n', stderr);
 	}
 }

 /*
  * Copy arg vector into a new buffer, concatenating arguments with spaces.
  */
 char *
 copy_argv(register char **argv)
 {
 	register char **p;
 	register u_int len = 0;
 	char *buf;
 	char *src, *dst;

 	p = argv;
 	if (*p == 0)
 		return 0;

 	while (*p)
 		len += strlen(*p++) + 1;

 	buf = (char *)malloc(len);
 	if (buf == NULL)
 		error("copy_argv: malloc");

 	p = argv;
 	dst = buf;
 	while ((src = *p++) != NULL) {
 		while ((*dst++ = *src++) != '\0')
 			;
 		dst[-1] = ' ';
 	}
 	dst[-1] = '\0';

 	return buf;
 }

 char *
 read_infile(char *fname)
 {
 	register int fd, cc;
 	register char *cp;
 	struct stat buf;

 	fd = open(fname, O_RDONLY);
 	if (fd < 0)
 		error("can't open %s: %s", fname, pcap_strerror(errno));

 	if (fstat(fd, &buf) < 0)
 		error("can't stat %s: %s", fname, pcap_strerror(errno));

 	cp = malloc((u_int)buf.st_size + 1);
 	if (cp == NULL)
 		error("malloc(%d) for %s: %s", (u_int)buf.st_size + 1,
 			fname, pcap_strerror(errno));
 	cc = read(fd, cp, (u_int)buf.st_size);
 	if (cc < 0)
 		error("read %s: %s", fname, pcap_strerror(errno));
 	if (cc != buf.st_size)
 		error("short read %s (%d != %d)", fname, cc, (int)buf.st_size);
 	cp[(int)buf.st_size] = '\0';

 	return (cp);
 }

 void
 safeputs(const char *s)
 {
 	while (*s) {
 		safeputchar(*s);
 		s++;
 	}
 }

 void
 safeputchar(int c)
 {
 	unsigned char ch;

 	ch = (unsigned char)(c & 0xff);
 	if (c < 0x80 && isprint(c))
 		printf("%c", c & 0xff);
 	else
 		printf("\\%03o", c & 0xff);
 }
	/*
	* Copyright (c) 1990, 1991, 1993, 1994, 1995, 1996, 1997
	* The Regents of the University of California. All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that: (1) source code distributions
	* retain the above copyright notice and this paragraph in its entirety, (2)
	* distributions including binary code include the above copyright notice and
	* this paragraph in its entirety in the documentation or other materials
	* provided with the distribution, and (3) all advertising materials mentioning
	* features or use of this software display the following acknowledgement:
	* ``This product includes software developed by the University of California,
	* Lawrence Berkeley Laboratory and its contributors.'' Neither the name of
	* the University nor the names of its contributors may be used to endorse
	* or promote products derived from this software without specific prior
	* written permission.
	* THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
	* WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
	* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
	*/

	#ifndef lint
	static const char rcsid[] =
	"@(#) $Header: /tcpdump/master/tcpdump/util.c,v 1.73 2002-06-11 17:09:01 itojun Exp $ (LBL)";
	#endif

	#ifdef HAVE_CONFIG_H
	#include "config.h"
	#endif

	#include <sys/types.h>
	#include <sys/time.h>
	#include <sys/file.h>
	#include <sys/stat.h>

	#include <ctype.h>
	#include <errno.h>
	#ifdef HAVE_FCNTL_H
	#include <fcntl.h>
	#endif
	#include <pcap.h>
	#include <stdio.h>
	#include <stdarg.h>
	#include <stdlib.h>
	#include <string.h>
	#ifdef TIME_WITH_SYS_TIME
	#include <time.h>
	#endif
	#include <unistd.h>

	#include "interface.h"

	/*
	* Print out a filename (or other ascii string).
	* If ep is NULL, assume no truncation check is needed.
	* Return true if truncated.
	*/
	int
	fn_print(register const u_char s, register const u_char ep)
	{
	register int ret;
	register u_char c;

	ret = 1; /* assume truncated */
	while (ep == NULL \|\| s < ep) {
	c = *s++;
	if (c == '\0') {
	ret = 0;
	break;
	}
	if (!isascii(c)) {
	c = toascii(c);
	putchar('M');
	putchar('-');
	}
	if (!isprint(c)) {
	c ^= 0x40; /* DEL to ?, others to alpha */
	putchar('^');
	}
	putchar(c);
	}
	return(ret);
	}

	/*
	* Print out a counted filename (or other ascii string).
	* If ep is NULL, assume no truncation check is needed.
	* Return true if truncated.
	*/
	int
	fn_printn(register const u_char *s, register u_int n,
	register const u_char *ep)
	{
	register int ret;
	register u_char c;

	ret = 1; /* assume truncated */
	while (ep == NULL \|\| s < ep) {
	if (n-- <= 0) {
	ret = 0;
	break;
	}
	c = *s++;
	if (!isascii(c)) {
	c = toascii(c);
	putchar('M');
	putchar('-');
	}
	if (!isprint(c)) {
	c ^= 0x40; /* DEL to ?, others to alpha */
	putchar('^');
	}
	putchar(c);
	}
	return(ret);
	}

	/*
	* Print the timestamp
	*/
	void
	ts_print(register const struct timeval *tvp)
	{
	register int s;
	struct tm *tm;
	time_t Time;
	static unsigned b_sec;
	static unsigned b_usec;

	switch(tflag) {
	case 1: /* Default */
	s = (tvp->tv_sec + thiszone) % 86400;
	(void)printf("%02d:%02d:%02d.%06u ",
	s / 3600, (s % 3600) / 60, s % 60,
	(unsigned)tvp->tv_usec);
	break;
	case -1: /* Unix timeval style */
	(void)printf("%u.%06u ",
	(unsigned)tvp->tv_sec,
	(unsigned)tvp->tv_usec);
	break;
	case -2:
	if (b_sec == 0) {
	printf("000000 ");
	} else {
	int d_usec = tvp->tv_usec - b_usec;
	int d_sec = tvp->tv_sec - b_sec;

	while (d_usec < 0) {
	d_usec += 1000000;
	d_sec--;
	}
	if (d_sec)
	printf("%d. ", d_sec);
	printf("%06d ", d_usec);
	}
	b_sec = tvp->tv_sec;
	b_usec = tvp->tv_usec;
	break;
	case -3: /* Default + Date*/
	s = (tvp->tv_sec + thiszone) % 86400;
	Time = (tvp->tv_sec + thiszone) - s;
	tm = gmtime (&Time);
	(void)printf("%02d/%02d/%04d %02d:%02d:%02d.%06u ",
	tm->tm_mon+1, tm->tm_mday,
	tm->tm_year+1900,
	s / 3600, (s % 3600) / 60,
	s % 60, (unsigned)tvp->tv_usec);
	break;
	}
	}

	/*
	* Print a relative number of seconds (e.g. hold time, prune timer)
	* in the form 5m1s. This does no truncation, so 32230861 seconds
	* is represented as 1y1w1d1h1m1s.
	*/
	void
	relts_print(int secs)
	{
	static const char *lengths[] = {"y", "w", "d", "h", "m", "s"};
	static const int seconds[] = {31536000, 604800, 86400, 3600, 60, 1};
	const char **l = lengths;
	const int *s = seconds;

	if (secs == 0) {
	(void)printf("0s");
	return;
	}
	if (secs < 0) {
	(void)printf("-");
	secs = -secs;
	}
	while (secs > 0) {
	if (secs >= *s) {
	(void)printf("%d%s", secs / s, l);
	secs -= (secs / s) *s;
	}
	s++;
	l++;
	}
	}

	/*
	* Convert a token value to a string; use "fmt" if not found.
	*/
	const char *
	tok2str(register const struct tok lp, register const char fmt,
	register int v)
	{
	static char buf[128];

	while (lp->s != NULL) {
	if (lp->v == v)
	return (lp->s);
	++lp;
	}
	if (fmt == NULL)
	fmt = "#%d";
	(void)snprintf(buf, sizeof(buf), fmt, v);
	return (buf);
	}

	/*
	* Convert a value to a string using an array; the macro
	* tok2strary() in <interface.h> is the public interface to
	* this function and ensures that the second argument is
	* correct for bounds-checking.
	*/
	const char *
	tok2strary_internal(register const char *lp, int n, register const char fmt,
	register int v)
	{
	static char buf[128];

	if (v >= 0 && v < n && lp[v] != NULL)
	return lp[v];
	if (fmt == NULL)
	fmt = "#%d";
	(void)snprintf(buf, sizeof(buf), fmt, v);
	return (buf);
	}

	/* VARARGS */
	void
	error(const char *fmt, ...)
	{
	va_list ap;

	(void)fprintf(stderr, "%s: ", program_name);
	va_start(ap, fmt);
	(void)vfprintf(stderr, fmt, ap);
	va_end(ap);
	if (*fmt) {
	fmt += strlen(fmt);
	if (fmt[-1] != '\n')
	(void)fputc('\n', stderr);
	}
	exit(1);
	/* NOTREACHED */
	}

	/* VARARGS */
	void
	warning(const char *fmt, ...)
	{
	va_list ap;

	(void)fprintf(stderr, "%s: WARNING: ", program_name);
	va_start(ap, fmt);
	(void)vfprintf(stderr, fmt, ap);
	va_end(ap);
	if (*fmt) {
	fmt += strlen(fmt);
	if (fmt[-1] != '\n')
	(void)fputc('\n', stderr);
	}
	}

	/*
	* Copy arg vector into a new buffer, concatenating arguments with spaces.
	*/
	char *
	copy_argv(register char **argv)
	{
	register char **p;
	register u_int len = 0;
	char *buf;
	char src, dst;

	p = argv;
	if (*p == 0)
	return 0;

	while (*p)
	len += strlen(*p++) + 1;

	buf = (char *)malloc(len);
	if (buf == NULL)
	error("copy_argv: malloc");

	p = argv;
	dst = buf;
	while ((src = *p++) != NULL) {
	while ((dst++ = src++) != '\0')
	;
	dst[-1] = ' ';
	}
	dst[-1] = '\0';

	return buf;
	}

	char *
	read_infile(char *fname)
	{
	register int fd, cc;
	register char *cp;
	struct stat buf;

	fd = open(fname, O_RDONLY);
	if (fd < 0)
	error("can't open %s: %s", fname, pcap_strerror(errno));

	if (fstat(fd, &buf) < 0)
	error("can't stat %s: %s", fname, pcap_strerror(errno));

	cp = malloc((u_int)buf.st_size + 1);
	if (cp == NULL)
	error("malloc(%d) for %s: %s", (u_int)buf.st_size + 1,
	fname, pcap_strerror(errno));
	cc = read(fd, cp, (u_int)buf.st_size);
	if (cc < 0)
	error("read %s: %s", fname, pcap_strerror(errno));
	if (cc != buf.st_size)
	error("short read %s (%d != %d)", fname, cc, (int)buf.st_size);
	cp[(int)buf.st_size] = '\0';

	return (cp);
	}

	void
	safeputs(const char *s)
	{
	while (*s) {
	safeputchar(*s);
	s++;
	}
	}

	void
	safeputchar(int c)
	{
	unsigned char ch;

	ch = (unsigned char)(c & 0xff);
	if (c < 0x80 && isprint(c))
	printf("%c", c & 0xff);
	else
	printf("\\%03o", c & 0xff);
	}