Drop main() prototype. Syncs with NetBSD-8

[minix.git] / minix / usr.bin / trace / call.c

#include "inc.h"

#include <minix/com.h>
#include <minix/callnr.h>
#include <minix/endpoint.h>

static const struct calls *call_table[] = {
        &pm_calls,
        &vfs_calls,
        &rs_calls,
        &mib_calls,
        &vm_calls,
        &ipc_calls,
};

/*
 * Find a call handler for the given endpoint, call number pair.  Return NULL
 * if no call handler for this call exists.
 */
static const struct call_handler *
find_handler(endpoint_t endpt, int call_nr)
{
        unsigned int i, index;

        for (i = 0; i < COUNT(call_table); i++) {
                if (call_table[i]->endpt != ANY &&
                    call_table[i]->endpt != endpt)
                        continue;

                if ((unsigned int)call_nr < call_table[i]->base)
                        continue;

                index = (unsigned int)call_nr - call_table[i]->base;

                if (index >= call_table[i]->count)
                        continue;

                if (call_table[i]->map[index].outfunc == NULL)
                        continue;

                return &call_table[i]->map[index];
        }

        return NULL;
}

/*
 * Print an endpoint.
 */
void
put_endpoint(struct trace_proc * proc, const char * name, endpoint_t endpt)
{
        const char *text = NULL;

        if (!valuesonly) {
                switch (endpt) {
                TEXT(ASYNCM);
                TEXT(IDLE);
                TEXT(CLOCK);
                TEXT(SYSTEM);
                TEXT(KERNEL);
                TEXT(PM_PROC_NR);
                TEXT(VFS_PROC_NR);
                TEXT(RS_PROC_NR);
                TEXT(MEM_PROC_NR);
                TEXT(SCHED_PROC_NR);
                TEXT(TTY_PROC_NR);
                TEXT(DS_PROC_NR);
                TEXT(MIB_PROC_NR);
                TEXT(VM_PROC_NR);
                TEXT(PFS_PROC_NR);
                TEXT(ANY);
                TEXT(NONE);
                TEXT(SELF);
                }
        }

        if (text != NULL)
                put_field(proc, name, text);
        else
                put_value(proc, name, "%d", endpt);
}

/*
 * Print a message structure.  The source field will be printed only if the
 * PF_ALT flag is given.
 */
static void
put_message(struct trace_proc * proc, const char * name, int flags,
        vir_bytes addr)
{
        message m;

        if (!put_open_struct(proc, name, flags, addr, &m, sizeof(m)))
                return;

        if (flags & PF_ALT)
                put_endpoint(proc, "m_source", m.m_source);

        put_value(proc, "m_type", "0x%x", m.m_type);

        put_close_struct(proc, FALSE /*all*/);
}

/*
 * Print the call's equals sign, which also implies that the parameters part of
 * the call has been fully printed and the corresponding closing parenthesis
 * may have to be printed, if it has not been printed already.
 */
void
put_equals(struct trace_proc * proc)
{

        /*
         * Do not allow multiple equals signs on a single line.  This check is
         * protection against badly written handlers.  It does not work for the
         * no-return type, but such calls are rare and less error prone anyway.
         */
        assert((proc->call_flags & (CF_DONE | CF_NORETURN)) != CF_DONE);

        /*
         * We allow (and in fact force) handlers to call put_equals in order to
         * indicate that the call's parameters block has ended, so we must end
         * the block here, if we hadn't done so before.
         */
        if (!(proc->call_flags & CF_DONE)) {
                put_close(proc, ") ");

                proc->call_flags |= CF_DONE;
        }

        put_align(proc);
        put_text(proc, "= ");

        format_set_sep(proc, NULL);
}

/*
 * Print the primary result of a call, after the equals sign.  It is always
 * possible that this is an IPC-level or other low-level error, in which case
 * this takes precedence, which is why this function must be called to print
 * the result if the call failed in any way at all; it may or may not be used
 * if the call succeeded.  For regular call results, default MINIX3/POSIX
 * semantics are used: if the return value is negative, the actual call failed
 * with -1 and the negative return value is the call's error code.  The caller
 * may consider other cases a failure (e.g., waitpid() returning 0), but
 * negative return values *not* signifying an error are currently not supported
 * since they are not present in MINIX3.
 */
void
put_result(struct trace_proc * proc)
{
        const char *errname;
        int value;

        /* This call should always be preceded by a put_equals call. */
        assert(proc->call_flags & CF_DONE);

        /*
         * If we failed to copy in the result register or message, print a
         * basic error and nothing else.
         */
        if (proc->call_flags & (CF_REG_ERR | CF_MSG_ERR)) {
                put_text(proc, "<fault>");

                return;
        }

        /*
         * If we are printing a system call rather than an IPC call, and an
         * error occurred at the IPC level, prefix the output with "<ipc>" to
         * indicate the IPC failure.  If we are printing an IPC call, an IPC-
         * level result is implied, so we do not print this.
         */
        if (proc->call_handler != NULL && (proc->call_flags & CF_IPC_ERR))
                put_text(proc, "<ipc> ");

        value = proc->call_result;

        if (value >= 0)
                put_fmt(proc, "%d", value);
        else if (!valuesonly && (errname = get_error_name(-value)) != NULL)
                put_fmt(proc, "-1 [%s]", errname);
        else
                put_fmt(proc, "-1 [%d]", -value);

        format_set_sep(proc, " ");
}

/*
 * The default enter-call (out) printer, which prints no parameters and is thus
 * immediately done with printing parameters.
 */
int
default_out(struct trace_proc * __unused proc, const message * __unused m_out)
{

        return CT_DONE;
}

/*
 * The default leave-call (in) printer, which simply prints the call result,
 * possibly preceded by an equals sign if none was printed yet.  For obvious
 * reasons, if the handler's out printer returned CT_NOTDONE, this default
 * printer must not be used.
 */
void
default_in(struct trace_proc * proc, const message * __unused m_out,
        const message * __unused m_in, int __unused failed)
{

        if ((proc->call_flags & (CF_DONE | CF_NORETURN)) != CF_DONE)
                put_equals(proc);
        put_result(proc);
}

/*
 * Prepare a sendrec call, by copying in the request message, determining
 * whether it is one of the calls that the tracing engine should know about,
 * searching for a handler for the call, and returning a name for the call.
 */
static const char *
sendrec_prepare(struct trace_proc * proc, endpoint_t endpt, vir_bytes addr,
        int * trace_class)
{
        const char *name;
        int r;

        r = mem_get_data(proc->pid, addr, &proc->m_out, sizeof(proc->m_out));

        if (r == 0) {
                if (endpt == PM_PROC_NR) {
                        if (proc->m_out.m_type == PM_EXEC)
                                *trace_class = TC_EXEC;
                        else if (proc->m_out.m_type == PM_SIGRETURN)
                                *trace_class = TC_SIGRET;
                }

                proc->call_handler = find_handler(endpt, proc->m_out.m_type);
        } else
                proc->call_handler = NULL;

        if (proc->call_handler != NULL) {
                if (proc->call_handler->namefunc != NULL)
                        name = proc->call_handler->namefunc(&proc->m_out);
                else
                        name = proc->call_handler->name;

                assert(name != NULL);
        } else
                name = "ipc_sendrec";

        return name;
}

/*
 * Print the outgoing (request) part of a sendrec call.  If we found a call
 * handler for the call, let the handler generate output.  Otherwise, print the
 * sendrec call at the kernel IPC level.  Return the resulting call flags.
 */
static unsigned int
sendrec_out(struct trace_proc * proc, endpoint_t endpt, vir_bytes addr)
{

        if (proc->call_handler != NULL) {
                return proc->call_handler->outfunc(proc, &proc->m_out);
        } else {
                put_endpoint(proc, "src_dest", endpt);
                /*
                 * We have already copied in the message, but if we used m_out
                 * and PF_LOCADDR here, a copy failure would cause "&.." to be
                 * printed rather than the actual message address.
                 */
                put_message(proc, "m_ptr", 0, addr);

                return CT_DONE;
        }
}

/*
 * Print the incoming (reply) part of a sendrec call.  Copy in the reply
 * message, determine whether the call is considered to have failed, and let
 * the call handler do the rest.  If no call handler was found, print an
 * IPC-level result.
 */
static void
sendrec_in(struct trace_proc * proc, int failed)
{
        message m_in;

        if (failed) {
                /* The call failed at the IPC level. */
                memset(&m_in, 0, sizeof(m_in)); /* not supposed to be used */
                assert(proc->call_flags & CF_IPC_ERR);
        } else if (mem_get_data(proc->pid, proc->m_addr, &m_in,
            sizeof(m_in)) != 0) {
                /* The reply message is somehow unavailable to us. */
                memset(&m_in, 0, sizeof(m_in)); /* not supposed to be used */
                proc->call_result = EGENERIC; /* not supposed to be used */
                proc->call_flags |= CF_MSG_ERR;
                failed = PF_FAILED;
        } else {
                /* The result is for the actual call. */
                proc->call_result = m_in.m_type;
                failed = (proc->call_result < 0) ? PF_FAILED : 0;
        }

        if (proc->call_handler != NULL)
                proc->call_handler->infunc(proc, &proc->m_out, &m_in, failed);
        else
                put_result(proc);
}

/*
 * Perform preparations for printing a system call.  Return two things: the
 * name to use for the call, and the trace class of the call.
 * special treatment).
 */
static const char *
call_prepare(struct trace_proc * proc, reg_t reg[3], int * trace_class)
{

        switch (proc->call_type) {
        case SENDREC:
                return sendrec_prepare(proc, (endpoint_t)reg[1],
                    (vir_bytes)reg[2], trace_class);

        case SEND:
                return "ipc_send";

        case SENDNB:
                return "ipc_sendnb";

        case RECEIVE:
                return "ipc_receive";

        case NOTIFY:
                return "ipc_notify";

        case SENDA:
                return "ipc_senda";

        case MINIX_KERNINFO:
                return "minix_kerninfo";

        default:
                /*
                 * It would be nice to include the call number here, but we
                 * must return a string that will last until the entire call is
                 * finished.  Adding another buffer to the trace_proc structure
                 * is an option, but it seems overkill..
                 */
                return "ipc_unknown";
        }
}

/*
 * Print the outgoing (request) part of a system call.  Return the resulting
 * call flags.
 */
static unsigned int
call_out(struct trace_proc * proc, reg_t reg[3])
{

        switch (proc->call_type) {
        case SENDREC:
                proc->m_addr = (vir_bytes)reg[2];

                return sendrec_out(proc, (endpoint_t)reg[1],
                    (vir_bytes)reg[2]);

        case SEND:
        case SENDNB:
                put_endpoint(proc, "dest", (endpoint_t)reg[1]);
                put_message(proc, "m_ptr", 0, (vir_bytes)reg[2]);

                return CT_DONE;

        case RECEIVE:
                proc->m_addr = (vir_bytes)reg[2];

                put_endpoint(proc, "src", (endpoint_t)reg[1]);

                return CT_NOTDONE;

        case NOTIFY:
                put_endpoint(proc, "dest", (endpoint_t)reg[1]);

                return CT_DONE;

        case SENDA:
                put_ptr(proc, "table", (vir_bytes)reg[2]);
                put_value(proc, "count", "%zu", (size_t)reg[1]);

                return CT_DONE;

        case MINIX_KERNINFO:
        default:
                return CT_DONE;
        }
}

/*
 * Print the incoming (reply) part of a call.
 */
static void
call_in(struct trace_proc * proc, int failed)
{

        switch (proc->call_type) {
        case SENDREC:
                sendrec_in(proc, failed);

                break;

        case RECEIVE:
                /* Print the source as well. */
                put_message(proc, "m_ptr", failed | PF_ALT, proc->m_addr);
                put_equals(proc);
                put_result(proc);

                break;

        case MINIX_KERNINFO:
                /*
                 * We do not have a platform-independent means to access the
                 * secondary IPC return value, so we cannot print the receive
                 * status or minix_kerninfo address.
                 */
                /* FALLTHROUGH */
        default:
                put_result(proc);

                break;
        }
}

/*
 * Determine whether to skip printing the given call, based on its name.
 */
static int
call_hide(const char * __unused name)
{

        /*
         * TODO: add support for such filtering, with an strace-like -e command
         * line option.  For now, we filter nothing, although calls may still
         * be hidden as the result of a register retrieval error.
         */
        return FALSE;
}

/*
 * The given process entered a system call.  Return the trace class of the
 * call: TC_EXEC for an execve() call, TC_SIGRET for a sigreturn() call, or
 * TC_NORMAL for a call that requires no exceptions in the trace engine.
 */
int
call_enter(struct trace_proc * proc, int show_stack)
{
        const char *name;
        reg_t reg[3];
        int trace_class, type;

        /* Get the IPC-level type and parameters of the system call. */
        if (kernel_get_syscall(proc->pid, reg) < 0) {
                /*
                 * If obtaining the details of the system call failed, even
                 * though we know the process is stopped on a system call, we
                 * are going to assume that the process got killed somehow.
                 * Thus, the best we can do is ignore the system call entirely,
                 * and hope that the next thing we hear about this process is
                 * its termination.  At worst, we ignore a serious error..
                 */
                proc->call_flags = CF_HIDE;

                return FALSE;
        }

        /*
         * Obtain the call name that is to be used for this call, and decide
         * whether we want to print this call at all.
         */
        proc->call_type = (int)reg[0];
        trace_class = TC_NORMAL;

        name = call_prepare(proc, reg, &trace_class);

        proc->call_name = name;

        if (call_hide(name)) {
                proc->call_flags = CF_HIDE;

                return trace_class;
        }

        /* Only print a stack trace if we are printing the call itself. */
        if (show_stack)
                kernel_put_stacktrace(proc);

        /*
         * Start a new line, start recording, and print the call name and
         * opening parenthesis.
         */
        put_newline();

        format_reset(proc);

        record_start(proc);

        put_text(proc, name);
        put_open(proc, NULL, PF_NONAME, "(", ", ");

        /*
         * Print the outgoing part of the call, that is, some or all of its
         * parameters.  This call returns flags indicating how far printing
         * got, and may be one of the following combinations:
         * - CT_NOTDONE (0) if printing parameters is not yet complete; after
         *   the call split, the in handler must print the rest itself;
         * - CT_DONE (CF_DONE) if printing parameters is complete, and we
         *   should now print the closing parenthesis and equals sign;
         * - CT_NORETURN (CF_DONE|CF_NORETURN) if printing parameters is
         *   complete, but we should not print the equals sign, because the
         *   call is expected not to return (the no-return call type).
         */
        type = call_out(proc, reg);
        assert(type == CT_NOTDONE || type == CT_DONE || type == CT_NORETURN);

        /*
         * Print whatever the handler told us to print for now.
         */
        if (type & CF_DONE) {
                if (type & CF_NORETURN) {
                        put_close(proc, ")");

                        put_space(proc);

                        proc->call_flags |= type;
                } else {
                        /*
                         * The equals sign is printed implicitly for the
                         * CT_DONE type only.  For CT_NORETURN and CT_NOTDONE,
                         * the "in" handler has to do it explicitly.
                         */
                        put_equals(proc);
                }
        } else {
                /*
                 * If at least one parameter was printed, print the separator
                 * now.  We know that another parameter will follow (otherwise
                 * the caller would have returned CT_DONE), and this way the
                 * output looks better.
                 */
                format_push_sep(proc);
        }

        /*
         * We are now at the call split; further printing will be done once the
         * call returns, through call_leave.  Stop recording; if the call gets
         * suspended and later resumed, we should replay everything up to here.
         */
#if DEBUG
        put_text(proc, "|"); /* warning, this may push a space */
#endif

        record_stop(proc);

        output_flush();

        return trace_class;
}

/*
 * The given process left a system call, or if skip is set, the leave phase of
 * the current system call should be ended.
 */
void
call_leave(struct trace_proc * proc, int skip)
{
        reg_t retreg;
        int hide, failed;

        /* If the call is skipped, it must be a no-return type call. */
        assert(!skip || (proc->call_flags & (CF_NORETURN | CF_HIDE)));

        /*
         * Start by replaying the current call, if necessary.  If the call was
         * suspended and we are about to print the "in" part, this is obviously
         * needed.  If the call is hidden, replaying will be a no-op, since
         * nothing was recorded for this call.  The special case is a skipped
         * call (which, as established above, must be a no-return call, e.g.
         * exec), for which replaying has the effect that if the call was
         * previously suspended, it will now be replayed, without suspension:
         *
         *       2| execve("./test", ["./test"], [..(12)]) <..>
         *       3| sigsuspend([]) = <..>
         * [A]   2| execve("./test", ["./test"], [..(12)])
         *       2| ---
         *       2| Tracing test (pid 2)
         *
         * The [A] line is the result of replaying the skipped call.
         */
        call_replay(proc);

        hide = (proc->call_flags & CF_HIDE);

        if (!hide && !skip) {
                /* Get the IPC-level result of the call. */
                if (kernel_get_retreg(proc->pid, &retreg) < 0) {
                        /* This should never happen.  Deal with it anyway. */
                        proc->call_flags |= CF_REG_ERR;
                        failed = PF_FAILED;
                } else if ((proc->call_result = (int)retreg) < 0) {
                        proc->call_flags |= CF_IPC_ERR;
                        failed = PF_FAILED;
                } else
                        failed = 0;

                /*
                 * Print the incoming part of the call, that is, possibly some
                 * or all of its parameters and the call's closing parenthesis
                 * (if CT_NOTDONE), and the equals sign (if not CT_DONE), then
                 * the call result.
                 */
                call_in(proc, failed);
        }

        if (!hide) {
                /*
                 * The call is complete now, so clear the recording.  This also
                 * implies that no suspension marker will be printed anymore.
                 */
                record_clear(proc);

                put_newline();
        }

        /*
         * For calls not of the no-return type, an equals sign must have been
         * printed by now.  This is protection against badly written handlers.
         */
        assert(proc->call_flags & CF_DONE);

        proc->call_name = NULL;
        proc->call_flags = 0;
}

/*
 * Replay the recorded text, if any, for the enter phase of the given process.
 * If there is no recorded text, start a new line anyway.
 */
void
call_replay(struct trace_proc * proc)
{

        /*
         * We get TRUE if the recorded call should be replayed, but the
         * recorded text for the call did not fit in the recording buffer.
         * In that case, we have to come up with a replacement text for the
         * call up to the call split.
         */
        if (record_replay(proc) == TRUE) {
                /*
                 * We basically place a "<..>" suspension marker in the
                 * parameters part of the call, and use its call name and flags
                 * for the rest.  There is a trailing space in all cases.
                 */
                put_fmt(proc, "%s(<..>%s", proc->call_name,
                    !(proc->call_flags & CF_DONE) ? "," :
                    ((proc->call_flags & CF_NORETURN) ? ")" : ") ="));
                put_space(proc);
        }
}

/*
 * Return the human-readable name of the call currently being made by the given
 * process.  The process is guaranteed to be in a call, although the call may
 * be hidden.  Under no circumstances may this function return a NULL pointer.
 */
const char *
call_name(struct trace_proc * proc)
{

        assert(proc->call_name != NULL);

        return proc->call_name;
}

/*
 * Return whether the current call failed due to an error at the system call
 * level, and if so, return the error code as well.  May be called during the
 * leave phase of a call only.
 */
int
call_errno(struct trace_proc * proc, int * err)
{

        if (proc->call_flags & (CF_REG_ERR | CF_MSG_ERR | CF_IPC_ERR))
                return FALSE;

        if (proc->call_result >= 0)
                return FALSE;

        *err = -proc->call_result;
        return TRUE;
}