/*	$NetBSD: filemon_wrapper.c,v 1.7 2015/06/15 19:45:31 christos Exp $	*/

/*
 * Copyright (c) 2010, Juniper Networks, Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: filemon_wrapper.c,v 1.7 2015/06/15 19:45:31 christos Exp $");

#include <sys/param.h>
#include <sys/types.h>
#include <sys/fcntl.h>
#include <sys/kernel.h>
#include <sys/lwp.h>
#include <sys/proc.h>
#include <sys/syscall.h>
#include <sys/syscallargs.h>

#include "filemon.h"

static int
filemon_wrapper_chdir(struct lwp * l, const struct sys_chdir_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;
	
	if ((error = sys_chdir(l, uap, retval)) != 0)
		return 0;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "C %d %s\n",
	    curproc->p_pid, filemon->fm_fname1);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_execve(struct lwp * l, struct sys_execve_args * uap,
    register_t * retval)
{
	char fname[MAXPATHLEN];
	int error;
	size_t done;
	struct filemon *filemon;
	
	if ((error = sys_execve(l, uap, retval)) != EJUSTRETURN)
		return 0;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), fname, sizeof(fname), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "E %d %s\n", curproc->p_pid, fname);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}


static int
filemon_wrapper_fork(struct lwp * l, const void *v, register_t * retval)
{
	int error;
	struct filemon *filemon;

	if ((error = sys_fork(l, v, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	filemon_printf(filemon, "F %d %ld\n", curproc->p_pid, (long) retval[0]);

	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_vfork(struct lwp * l, const void *v, register_t * retval)
{
	int error;
	struct filemon *filemon;

	if ((error = sys_vfork(l, v, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	filemon_printf(filemon, "F %d %ld\n", curproc->p_pid, (long) retval[0]);

	rw_exit(&filemon->fm_mtx);
	return 0;
}

static void
filemon_flags(struct filemon * filemon, int f)
{
	if (f & O_RDWR) {
		/* we want a separate R record */
		filemon_printf(filemon, "R %d %s\n", curproc->p_pid,
		    filemon->fm_fname1);
	}

	filemon_printf(filemon, "%c %d %s\n", (f & O_ACCMODE) ? 'W' : 'R',
	    curproc->p_pid, filemon->fm_fname1);
}

static int
filemon_wrapper_open(struct lwp * l, struct sys_open_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_open(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	filemon_flags(filemon, SCARG(uap, flags));
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_openat(struct lwp * l, struct sys_openat_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_openat(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	if (filemon->fm_fname1[0] != '/' && SCARG(uap, fd) != AT_FDCWD) {
		/*
		 * Rats we cannot just treat like open.
		 * Output an 'A' record as a clue.
		 */
		filemon_printf(filemon, "A %d %s\n", curproc->p_pid,
		    filemon->fm_fname1);
	}

	filemon_flags(filemon, SCARG(uap, oflags));
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_rename(struct lwp * l, struct sys_rename_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_rename(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, from), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	error = copyinstr(SCARG(uap, to), filemon->fm_fname2,
	    sizeof(filemon->fm_fname2), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "M %d '%s' '%s'\n", curproc->p_pid,
	    filemon->fm_fname1, filemon->fm_fname2);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_link(struct lwp * l, struct sys_link_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_link(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	error = copyinstr(SCARG(uap, link), filemon->fm_fname2,
	    sizeof(filemon->fm_fname2), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "L %d '%s' '%s'\n", curproc->p_pid,
	    filemon->fm_fname1, filemon->fm_fname2);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}

static int
filemon_wrapper_symlink(struct lwp * l, struct sys_symlink_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_symlink(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	error = copyinstr(SCARG(uap, link), filemon->fm_fname2,
	    sizeof(filemon->fm_fname2), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "L %d '%s' '%s'\n", curproc->p_pid,
	    filemon->fm_fname1, filemon->fm_fname2);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}


static void
filemon_wrapper_sys_exit(struct lwp * l, struct sys_exit_args * uap,
    register_t * retval)
{
	struct filemon *filemon;

	filemon = filemon_lookup(curproc);

	if (filemon) {
		filemon_printf(filemon, "X %d %d\n",
		    curproc->p_pid, SCARG(uap, rval));

		/* Check if the monitored process is about to exit. */
		if (filemon->fm_pid == curproc->p_pid)
			filemon_printf(filemon, "# Bye bye\n");

		rw_exit(&filemon->fm_mtx);
	}
	sys_exit(l, uap, retval);
}

static int
filemon_wrapper_unlink(struct lwp * l, struct sys_unlink_args * uap,
    register_t * retval)
{
	int error;
	size_t done;
	struct filemon *filemon;

	if ((error = sys_unlink(l, uap, retval)) != 0)
		return error;

	filemon = filemon_lookup(curproc);
	if (filemon == NULL)
		return 0;

	error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
	    sizeof(filemon->fm_fname1), &done);
	if (error)
		goto out;

	filemon_printf(filemon, "D %d %s\n",
	    curproc->p_pid, filemon->fm_fname1);
out:
	rw_exit(&filemon->fm_mtx);
	return 0;
}


void
filemon_wrapper_install(void)
{
	struct sysent *sv_table = curproc->p_emul->e_sysent;

	sv_table[SYS_chdir].sy_call = (sy_call_t *) filemon_wrapper_chdir;
	sv_table[SYS_execve].sy_call = (sy_call_t *) filemon_wrapper_execve;
	sv_table[SYS_exit].sy_call = (sy_call_t *) filemon_wrapper_sys_exit;
	sv_table[SYS_fork].sy_call = (sy_call_t *) filemon_wrapper_fork;
	sv_table[SYS_link].sy_call = (sy_call_t *) filemon_wrapper_link;
	sv_table[SYS_open].sy_call = (sy_call_t *) filemon_wrapper_open;
	sv_table[SYS_openat].sy_call = (sy_call_t *) filemon_wrapper_openat;
	sv_table[SYS_rename].sy_call = (sy_call_t *) filemon_wrapper_rename;
	sv_table[SYS_symlink].sy_call = (sy_call_t *) filemon_wrapper_symlink;
	sv_table[SYS_unlink].sy_call = (sy_call_t *) filemon_wrapper_unlink;
	sv_table[SYS_vfork].sy_call = (sy_call_t *) filemon_wrapper_vfork;
}

int
filemon_wrapper_deinstall(void)
{
	struct sysent *sv_table = curproc->p_emul->e_sysent;

	if (sv_table[SYS_chdir].sy_call != (sy_call_t *) filemon_wrapper_chdir)
	    return EBUSY;
	sv_table[SYS_chdir].sy_call = (sy_call_t *) sys_chdir;
	sv_table[SYS_execve].sy_call = (sy_call_t *) sys_execve;
	sv_table[SYS_exit].sy_call = (sy_call_t *) sys_exit;
	sv_table[SYS_fork].sy_call = (sy_call_t *) sys_fork;
	sv_table[SYS_link].sy_call = (sy_call_t *) sys_link;
	sv_table[SYS_open].sy_call = (sy_call_t *) sys_open;
	sv_table[SYS_openat].sy_call = (sy_call_t *) sys_openat;
	sv_table[SYS_rename].sy_call = (sy_call_t *) sys_rename;
	sv_table[SYS_symlink].sy_call = (sy_call_t *) sys_symlink;
	sv_table[SYS_unlink].sy_call = (sy_call_t *) sys_unlink;
	sv_table[SYS_vfork].sy_call = (sy_call_t *) sys_vfork;
	return 0;
}