NetBSD/sys/dev/filemon/filemon_wrapper.c

386 lines
9.8 KiB
C

/*
* 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.2 2011/03/13 21:26:31 sjg 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 ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_chdir(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
sizeof(filemon->fm_fname1), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"C %d %s\n",
curproc->p_pid, filemon->fm_fname1);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_execve(struct lwp * l, struct sys_execve_args * uap,
register_t * retval)
{
char fname[MAXPATHLEN];
int ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
error = copyinstr(SCARG(uap, path), fname, sizeof(fname), &done);
if ((ret = sys_execve(l, uap, retval)) == EJUSTRETURN && error == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
len = snprintf(filemon->fm_msgbufr, sizeof(filemon->fm_msgbufr),
"E %d %s\n",
curproc->p_pid, fname);
filemon_output(filemon, filemon->fm_msgbufr, len);
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_fork(struct lwp * l, const void *v, register_t * retval)
{
int ret;
size_t len;
struct filemon *filemon;
if ((ret = sys_fork(l, v, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"F %d %ld\n",
curproc->p_pid, (long) retval[0]);
filemon_output(filemon, filemon->fm_msgbufr, len);
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_vfork(struct lwp * l, const void *v, register_t * retval)
{
int ret;
size_t len;
struct filemon *filemon;
if ((ret = sys_vfork(l, v, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"F %d %ld\n",
curproc->p_pid, (long) retval[0]);
filemon_output(filemon, filemon->fm_msgbufr, len);
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_open(struct lwp * l, struct sys_open_args * uap,
register_t * retval)
{
int ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_open(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, path), filemon->fm_fname1,
sizeof(filemon->fm_fname1), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"%c %d %s\n",
(SCARG(uap, flags) & O_ACCMODE) ? 'W' : 'R',
curproc->p_pid, filemon->fm_fname1);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_rename(struct lwp * l, struct sys_rename_args * uap,
register_t * retval)
{
int ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_rename(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, from), filemon->fm_fname1,
sizeof(filemon->fm_fname1), &done);
if (error == 0)
error = copyinstr(SCARG(uap, to),
filemon->fm_fname2,
sizeof(filemon->fm_fname2), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"M %d '%s' '%s'\n",
curproc->p_pid, filemon->fm_fname1,
filemon->fm_fname2);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_link(struct lwp * l, struct sys_link_args * uap,
register_t * retval)
{
int ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_link(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, path),
filemon->fm_fname1, sizeof(filemon->fm_fname1),
&done);
if (error == 0)
error = copyinstr(SCARG(uap, link),
filemon->fm_fname2,
sizeof(filemon->fm_fname2), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr), "L %d '%s' '%s'\n",
curproc->p_pid, filemon->fm_fname1,
filemon->fm_fname2);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static int
filemon_wrapper_symlink(struct lwp * l, struct sys_symlink_args * uap,
register_t * retval)
{
int ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_symlink(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, path),
filemon->fm_fname1,
sizeof(filemon->fm_fname1), &done);
if (error == 0)
error = copyinstr(SCARG(uap, link),
filemon->fm_fname2,
sizeof(filemon->fm_fname2), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"L %d '%s' '%s'\n",
curproc->p_pid, filemon->fm_fname1,
filemon->fm_fname2);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
static void
filemon_wrapper_sys_exit(struct lwp * l, struct sys_exit_args * uap,
register_t * retval)
{
size_t len;
struct filemon *filemon;
filemon = filemon_lookup(curproc);
if (filemon) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr), "X %d %d\n",
curproc->p_pid, SCARG(uap, rval));
filemon_output(filemon, filemon->fm_msgbufr, len);
/* Check if the monitored process is about to exit. */
if (filemon->fm_pid == curproc->p_pid) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr), "# Bye bye\n");
filemon_output(filemon, filemon->fm_msgbufr, len);
}
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 ret;
int error;
size_t done;
size_t len;
struct filemon *filemon;
if ((ret = sys_unlink(l, uap, retval)) == 0) {
filemon = filemon_lookup(curproc);
if (filemon) {
error = copyinstr(SCARG(uap, path),
filemon->fm_fname1,
sizeof(filemon->fm_fname1), &done);
if (error == 0) {
len = snprintf(filemon->fm_msgbufr,
sizeof(filemon->fm_msgbufr),
"D %d %s\n",
curproc->p_pid, filemon->fm_fname1);
filemon_output(filemon, filemon->fm_msgbufr,
len);
}
rw_exit(&filemon->fm_mtx);
}
}
return (ret);
}
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_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;
}
void
filemon_wrapper_deinstall(void)
{
struct sysent *sv_table = curproc->p_emul->e_sysent;
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_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;
}