tracedump.c

/*
 * Adapted for x86_64 by:
 * Ingvaras Merkys <ingvaras@gmail.com>
 */
/*
 * Copyright (C) 2011-2012 IITiS PAN Gliwice <http://www.iitis.pl/>
 * Author: Paweł Foremski <pjf@iitis.pl>
 * Licensed under GNU GPL v. 3
 */

#include <libpjf/main.h>
#include "tracedump.h"

static bool EXITING = false;

/** Prints usage help screen */
static void help(void)
{
	printf("Usage: tracedump [OPTIONS] <COMMAND...>\n");
	printf("       tracedump [OPTIONS] <PID...>\n");
	printf("\n");
	printf("  A single application IP packet sniffer.\n");
	printf("\n");
	printf("Options:\n");
	printf("  -w <file>              write output to <file> [dump.pcap]\n");
	printf("                         use -w - for stdout output\n");
	printf("  -s <bytes>             number of bytes to capture in each IP packet\n");
	printf("                         use -s 0 for full packet contents\n");
	printf("  --verbose,-V           be verbose (alias for --debug=5)\n");
	printf("  --debug=<num>          set debugging level\n");
	printf("  --help,-h              show this usage help screen\n");
	printf("  --version,-v           show version and copying information\n");
}

/** Prints version and copying information. */
static void version(void)
{
	printf("tracedump %s\n", TRACEDUMP_VERSION);
	printf("Copyright (C) 2011-2012 IITiS PAN\n");
	printf("Licensed under GNU GPL v3\n");
	printf("Author: Paweł Foremski <pjf@iitis.pl>\n");
	printf("Part of the MuTriCs project: <http://mutrics.iitis.pl/>\n");
	printf("Realized under grant nr 2011/01/N/ST6/07202 of the Polish National Science Centre\n");
}

/** Parses arguments and loads modules
 * @retval 0     ok
 * @retval 1     error, main() should exit (eg. wrong arg. given)
 * @retval 2     ok, but main() should exit (eg. on --version or --help) */
static int parse_argv(struct tracedump *td, int argc, char *argv[])
{
	int i, c;

	static char *short_opts = "hvVw:s:";
	static struct option long_opts[] = {
		/* name, has_arg, NULL, short_ch */
		{ "verbose",    0, NULL,  1  },
		{ "debug",      1, NULL,  2  },
		{ "help",       0, NULL,  3  },
		{ "version",    0, NULL,  4  },
		{ 0, 0, 0, 0 }
	};

	/* defaults */
	debug = 1;
	td->opts.outfile = "dump.pcap";

	for (;;) {
		c = getopt_long(argc, argv, short_opts, long_opts, &i);
		if (c == -1) break; /* end of options */

		switch (c) {
			case 'V':
			case  1 : debug = 5; break;
			case  2 : debug = atoi(optarg); break;
			case 'h':
			case  3 : help(); return 2;
			case 'v':
			case  4 : version(); return 2;
			case 'w': td->opts.outfile = mmatic_strdup(td->mm, optarg); break;
			case 's': td->opts.snaplen = atoi(optarg); break;
			default: help(); return 1;
		}
	}

	if (td->opts.snaplen <= 0)
		td->opts.snaplen = UINT16_MAX;

	if (argc - optind > 0) {
		td->opts.src = argv + optind;
		td->opts.srclen = argc - optind;
	} else {
		help();
		return 1;
	}

	return 0;
}

static void sighandler(int signum)
{
	dbg(1, "Caught signal %d, exiting\n", signum);
	signal(SIGINT, SIG_IGN);
	signal(SIGTERM, SIG_IGN);
	EXITING = true;
}

static void handle_socket(struct pid *sp, int fd)
{
	struct tracedump *td = sp->td;
	char buf[128], buf2[128];
	int len, socknum;
	struct sock *ss;
	struct sockaddr_in sa;
	socklen_t optlen;

	/* TODO: speed-up cache? */

	snprintf(buf, sizeof buf, "/proc/%d/fd/%d", sp->pid, fd);
	len = readlink(buf, buf2, sizeof buf2);

	/* is it a socket at all? */
	if (len < 10 || strncmp(buf2, "socket:[", 8) != 0)
		return;

	/* get socknum */
	buf2[len - 1] = 0;
	socknum = atoi(buf2 + 8);

	/* check if in cache */
	if (sp->ss && sp->ss->socknum == socknum)
		return;

	/* get socket info */
	ss = thash_uint_get(td->socks, socknum);
	if (!ss) {
		ss = mmatic_zalloc(td->mm, sizeof *ss);
		ss->td = td;
		ss->socknum = socknum;
		thash_uint_set(td->socks, socknum, ss);

		/* cancel the original socketcall */
		if (sp->in_socketcall)
			inject_escape_socketcall(td, sp);

		/* check if AF_INET, get local address */
		if (inject_getsockname_in(td, sp, fd, &sa) != 0) {
			goto handled;
		}

		/* check if TCP/UDP */
		optlen = sizeof ss->type;
		if (inject_getsockopt(td, sp, fd, SOL_SOCKET, SO_TYPE, &ss->type, &optlen) != 0) {
			goto handled;
		}
		if (optlen != sizeof ss->type || (ss->type != SOCK_STREAM && ss->type != SOCK_DGRAM)) {
			goto handled;
		}

		/* autobind if necessary */
		if (!sa.sin_port) {
			if (inject_autobind(td, sp, fd) != 0) {
				dbg(1, "pid %d fd %d: autobind failed\n", sp->pid, fd);
				goto handled;
			}

			if (inject_getsockname_in(td, sp, fd, &sa) != 0) {
				dbg(1, "pid %d fd %d: getsockname after autobind failed\n", sp->pid, fd);
				goto handled;
			}
		}
		ss->port = ntohs(sa.sin_port);

		port_add(ss, true);
		pcap_update(td);

handled:
		/* finish the original socketcall */
		if (sp->in_socketcall) {
			inject_restore_socketcall(td, sp);
			sp->in_socketcall = false;
		}
	}
}

/** Try connecting to all file descriptors of a just-attached child */
static void handle_attached_pid(struct pid *sp)
{
	DIR *dh;
	char buf[128];
	struct dirent *de;

	snprintf(buf, sizeof buf, "/proc/%d/fd", sp->pid);
	dh = opendir(buf);
	if (!dh)
		return;

	while ((de = readdir(dh))) {
		if (!isdigit(de->d_name[0]))
			continue;

		handle_socket(sp, atoi(de->d_name));
	}

	closedir(dh);
}

int main(int argc, char *argv[])
{
	mmatic *mm;
	struct tracedump *td;
	pid_t pid;
	struct pid *sp;
	unsigned long fd_arg;
	struct user_regs_struct regs;
	int status;
	int stopped_pid;
	int stopsig;
	int i;
	int attached;
	struct sigaction sa;

	/*************/

	/* initialize */
	mm = mmatic_create();
	td = mmatic_zalloc(mm, sizeof *td);
	td->mm = mm;
	pthread_mutex_init(&td->mutex_ports, NULL);

	/* create hashing tables */
	td->pids = thash_create_intkey(mmatic_free, td->mm);
	td->socks = thash_create_intkey(mmatic_free, td->mm);
	td->tcp_ports = thash_create_intkey(mmatic_free, td->mm);
	td->udp_ports = thash_create_intkey(mmatic_free, td->mm);

	/*************/

	/* parse command line options */
	if (parse_argv(td, argc, argv))
		return 1;

	/************ start threads */

	/* start the garbage collector and sniffer threads
	 * note: the GC calls pcap_update(), so the order of initialization is important */
	pcap_init(td);
	port_init(td);

	/************ attach the victim :) */

	attached = 0;

	if (isdigit(td->opts.src[0][0])) {
		/* attach to processes */
		for (i = 0; i < td->opts.srclen; i++) {
			pid = atoi(td->opts.src[i]);
			if (ptrace_attach_pid(pid_get(td, pid), handle_attached_pid) == 0)
				attached++;
		}
	} else {
		/* attach to child */
		pid = fork();

		if (pid == 0) {
			ptrace_traceme();
			execvp(td->opts.src[0], td->opts.src + 0);

			/* if fails: */
			fprintf(stderr, "%s: %s\n", argv[1], strerror(errno));
			exit(127);
		}

		if (ptrace_attach_child(pid_get(td, pid), NULL) == 0)
			attached++;
	}

	if (attached == 0) {
		dbg(0, "No processes to trace, exiting\n");
		return 127;
	} else {
		dbg(2, "Attached, starting\n");
	}

	/************ main thread */

	/* setup signal handling */
	memset(&sa, 0, sizeof sa);
	sa.sa_handler = sighandler;
	sigaction(SIGINT, &sa, NULL);
	sigaction(SIGTERM, &sa, NULL);

	while (!EXITING) {
		/* wait for syscall from any pid */
		stopped_pid = ptrace_wait(NULL, &status);

		if (stopped_pid == -1) {
			break;
		} else if (WIFEXITED(status) || WIFSIGNALED(status)) {
			pid_del(td, stopped_pid);
			continue;
		} else if (!WIFSTOPPED(status)) {
			goto next_syscall;
		}

		/* fetch pid info */
		sp = pid_get(td, stopped_pid);

		/* handle signal passing */
		stopsig = WSTOPSIG(status);
		switch (stopsig) {
			case SIGTRAPS:
				break;
			case SIGTRAP:
			case SIGSTOP:
				goto next_syscall;
			default:
				/* pass the signal to child */
				ptrace_cont_syscall(sp, stopsig, false);
				continue;
		}


		ptrace_getregs(sp, &regs);

		/* filter anything different than bind(), connect() and sendto() */
		if (regs.orig_rax != SYS_bind && regs.orig_rax != SYS_connect && regs.orig_rax != SYS_sendto)
			goto next_syscall;

		sp->in_socketcall = !sp->in_socketcall;

		/* on exit from a successful bind() or enter to connect()/sendto() */
		if ((!sp->in_socketcall && regs.orig_rax == SYS_bind && regs.rax == SYS_restart_syscall) ||
		    (sp->in_socketcall && regs.orig_rax != SYS_bind)) {
			/* get fd number */
			fd_arg = regs.rdi;
			/* handle the socket underlying given fd */
			handle_socket(sp, fd_arg);
		}

next_syscall:
		ptrace_cont_syscall(sp, 0, false);
	}

	/*****************************/

	port_deinit(td);
	pcap_deinit(td);
	pid_detach_all(td);
	mmatic_destroy(mm);

	return 0;
}