/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * Copyright (c) 2013, Joyent, Inc.  All rights reserved.
 */

#include <kvm.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <unistd.h>
#include <limits.h>
#include <fcntl.h>
#include <strings.h>
#include <errno.h>
#include <sys/mem.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <sys/dumphdr.h>
#include <sys/sysmacros.h>

struct _kvmd {
        struct dumphdr  kvm_dump;
        char            *kvm_debug;
        int             kvm_openflag;
        int             kvm_corefd;
        int             kvm_kmemfd;
        int             kvm_memfd;
        size_t          kvm_coremapsize;
        char            *kvm_core;
        dump_map_t      *kvm_map;
        pfn_t           *kvm_pfn;
        struct as       *kvm_kas;
        proc_t          *kvm_practive;
        pid_t           kvm_pid;
        char            kvm_namelist[MAXNAMELEN + 1];
        boolean_t       kvm_namelist_core;
        proc_t          kvm_proc;
};

#define PREAD   (ssize_t (*)(int, void *, size_t, offset_t))pread64
#define PWRITE  (ssize_t (*)(int, void *, size_t, offset_t))pwrite64

static int kvm_nlist_core(kvm_t *kd, struct nlist nl[], const char *err);

static kvm_t *
fail(kvm_t *kd, const char *err, const char *message, ...)
{
        va_list args;

        va_start(args, message);
        if (err || (kd && kd->kvm_debug)) {
                (void) fprintf(stderr, "%s: ", err ? err : "KVM_DEBUG");
                (void) vfprintf(stderr, message, args);
                (void) fprintf(stderr, "\n");
        }
        va_end(args);
        if (kd != NULL)
                (void) kvm_close(kd);
        return (NULL);
}

/*ARGSUSED*/
kvm_t *
kvm_open(const char *namelist, const char *corefile, const char *swapfile,
        int flag, const char *err)
{
        kvm_t *kd;
        struct stat64 memstat, kmemstat, allkmemstat, corestat;
        struct nlist nl[3] = { { "kas" }, { "practive" }, { "" } };

        if ((kd = calloc(1, sizeof (kvm_t))) == NULL)
                return (fail(NULL, err, "cannot allocate space for kvm_t"));

        kd->kvm_corefd = kd->kvm_kmemfd = kd->kvm_memfd = -1;
        kd->kvm_debug = getenv("KVM_DEBUG");

        if ((kd->kvm_openflag = flag) != O_RDONLY && flag != O_RDWR)
                return (fail(kd, err, "illegal flag 0x%x to kvm_open()", flag));

        if (corefile == NULL)
                corefile = "/dev/kmem";

        if (stat64(corefile, &corestat) == -1)
                return (fail(kd, err, "cannot stat %s", corefile));

        if (S_ISCHR(corestat.st_mode)) {
                if (stat64("/dev/mem", &memstat) == -1)
                        return (fail(kd, err, "cannot stat /dev/mem"));

                if (stat64("/dev/kmem", &kmemstat) == -1)
                        return (fail(kd, err, "cannot stat /dev/kmem"));

                if (stat64("/dev/allkmem", &allkmemstat) == -1)
                        return (fail(kd, err, "cannot stat /dev/allkmem"));
                if (corestat.st_rdev == memstat.st_rdev ||
                    corestat.st_rdev == kmemstat.st_rdev ||
                    corestat.st_rdev == allkmemstat.st_rdev) {
                        char *kmem = (corestat.st_rdev == allkmemstat.st_rdev ?
                            "/dev/allkmem" : "/dev/kmem");

                        if ((kd->kvm_kmemfd = open64(kmem, flag)) == -1)
                                return (fail(kd, err, "cannot open %s", kmem));
                        if ((kd->kvm_memfd = open64("/dev/mem", flag)) == -1)
                                return (fail(kd, err, "cannot open /dev/mem"));
                }
        } else {
                if ((kd->kvm_corefd = open64(corefile, flag)) == -1)
                        return (fail(kd, err, "cannot open %s", corefile));
                if (pread64(kd->kvm_corefd, &kd->kvm_dump,
                    sizeof (kd->kvm_dump), 0) != sizeof (kd->kvm_dump))
                        return (fail(kd, err, "cannot read dump header"));
                if (kd->kvm_dump.dump_magic != DUMP_MAGIC)
                        return (fail(kd, err, "%s is not a kernel core file "
                            "(bad magic number %x)", corefile,
                            kd->kvm_dump.dump_magic));
                if (kd->kvm_dump.dump_version != DUMP_VERSION)
                        return (fail(kd, err,
                            "libkvm version (%u) != corefile version (%u)",
                            DUMP_VERSION, kd->kvm_dump.dump_version));
                if (kd->kvm_dump.dump_wordsize != DUMP_WORDSIZE)
                        return (fail(kd, err, "%s is a %d-bit core file - "
                            "cannot examine with %d-bit libkvm", corefile,
                            kd->kvm_dump.dump_wordsize, DUMP_WORDSIZE));
                /*
                 * We try to mmap(2) the entire corefile for performance
                 * (so we can use bcopy(3C) rather than pread(2)).  Failing
                 * that, we insist on at least mmap(2)ing the dump map.
                 */
                kd->kvm_coremapsize = (size_t)corestat.st_size;
                if (corestat.st_size > LONG_MAX ||
                    (kd->kvm_core = mmap64(0, kd->kvm_coremapsize,
                    PROT_READ, MAP_SHARED, kd->kvm_corefd, 0)) == MAP_FAILED) {
                        kd->kvm_coremapsize = kd->kvm_dump.dump_data;
                        if ((kd->kvm_core = mmap64(0, kd->kvm_coremapsize,
                            PROT_READ, MAP_SHARED, kd->kvm_corefd, 0)) ==
                            MAP_FAILED)
                                return (fail(kd, err, "cannot mmap corefile"));
                }
                kd->kvm_map = (void *)(kd->kvm_core + kd->kvm_dump.dump_map);
                kd->kvm_pfn = (void *)(kd->kvm_core + kd->kvm_dump.dump_pfn);
        }

        if (namelist == NULL)
                namelist = "/dev/ksyms";

        (void) strncpy(kd->kvm_namelist, namelist, MAXNAMELEN);

        if (kvm_nlist(kd, nl) == -1) {
                if (kd->kvm_corefd == -1) {
                        return (fail(kd, err, "%s is not a %d-bit "
                            "kernel namelist", namelist, DUMP_WORDSIZE));
                }

                if (kvm_nlist_core(kd, nl, err) == -1)
                        return (NULL);          /* fail() already called */
        }

        kd->kvm_kas = (struct as *)nl[0].n_value;
        kd->kvm_practive = (proc_t *)nl[1].n_value;

        (void) kvm_setproc(kd);
        return (kd);
}

int
kvm_close(kvm_t *kd)
{
        if (kd->kvm_core != NULL && kd->kvm_core != MAP_FAILED)
                (void) munmap(kd->kvm_core, kd->kvm_coremapsize);
        if (kd->kvm_corefd != -1)
                (void) close(kd->kvm_corefd);
        if (kd->kvm_kmemfd != -1)
                (void) close(kd->kvm_kmemfd);
        if (kd->kvm_memfd != -1)
                (void) close(kd->kvm_memfd);
        if (kd->kvm_namelist_core)
                (void) unlink(kd->kvm_namelist);
        free(kd);
        return (0);
}

const char *
kvm_namelist(kvm_t *kd)
{
        return (kd->kvm_namelist);
}

int
kvm_nlist(kvm_t *kd, struct nlist nl[])
{
        return (nlist(kd->kvm_namelist, nl));
}

/*
 * If we don't have a name list, try to dig it out of the kernel crash dump.
 * (The symbols have been present in the dump, uncompressed, for nearly a
 * decade as of this writing -- and it is frankly surprising that the archaic
 * notion of a disjoint symbol table managed to survive that change.)
 */
static int
kvm_nlist_core(kvm_t *kd, struct nlist nl[], const char *err)
{
        dumphdr_t *dump = &kd->kvm_dump;
        char *msg = "couldn't extract symbols from dump";
        char *template = "/tmp/.libkvm.kvm_nlist_core.pid%d.XXXXXX";
        int fd, rval;

        if (dump->dump_ksyms_size != dump->dump_ksyms_csize) {
                (void) fail(kd, err, "%s: kernel symbols are compressed", msg);
                return (-1);
        }

        if (dump->dump_ksyms + dump->dump_ksyms_size > kd->kvm_coremapsize) {
                (void) fail(kd, err, "%s: kernel symbols not mapped", msg);
                return (-1);
        }

        /*
         * Beause this temporary file may be left as a turd if the caller
         * does not properly call kvm_close(), we make sure that it clearly
         * indicates its origins.
         */
        (void) snprintf(kd->kvm_namelist, MAXNAMELEN, template, getpid());

        if ((fd = mkstemp(kd->kvm_namelist)) == -1) {
                (void) fail(kd, err, "%s: couldn't create temporary "
                    "symbols file: %s", msg, strerror(errno));
                return (-1);
        }

        kd->kvm_namelist_core = B_TRUE;

        do {
                rval = write(fd, (caddr_t)((uintptr_t)kd->kvm_core +
                    (uintptr_t)dump->dump_ksyms), dump->dump_ksyms_size);
        } while (rval < dump->dump_ksyms_size && errno == EINTR);

        if (rval < dump->dump_ksyms_size) {
                (void) fail(kd, err, "%s: couldn't write to temporary "
                    "symbols file: %s", msg, strerror(errno));
                (void) close(fd);
                return (-1);
        }

        (void) close(fd);

        if (kvm_nlist(kd, nl) == -1) {
                (void) fail(kd, err, "%s: symbols not valid", msg);
                return (-1);
        }

        return (0);
}

static offset_t
kvm_lookup(kvm_t *kd, struct as *as, uint64_t addr)
{
        uintptr_t pageoff = addr & (kd->kvm_dump.dump_pagesize - 1);
        uint64_t page = addr - pageoff;
        offset_t off = 0;

        if (kd->kvm_debug)
                fprintf(stderr, "kvm_lookup(%p, %llx):", (void *)as, addr);

        if (as == NULL) {               /* physical addressing mode */
                long first = 0;
                long last = kd->kvm_dump.dump_npages - 1;
                pfn_t target = (pfn_t)(page >> kd->kvm_dump.dump_pageshift);
                while (last >= first) {
                        long middle = (first + last) / 2;
                        pfn_t pfn = kd->kvm_pfn[middle];
                        if (kd->kvm_debug)
                                fprintf(stderr, " %ld ->", middle);
                        if (pfn == target) {
                                off = kd->kvm_dump.dump_data + pageoff +
                                    ((uint64_t)middle <<
                                    kd->kvm_dump.dump_pageshift);
                                break;
                        }
                        if (pfn < target)
                                first = middle + 1;
                        else
                                last = middle - 1;
                }
        } else {
                long hash = DUMP_HASH(&kd->kvm_dump, as, page);
                off = kd->kvm_map[hash].dm_first;
                while (off != 0) {
                        dump_map_t *dmp = (void *)(kd->kvm_core + off);
                        if (kd->kvm_debug)
                                fprintf(stderr, " %llx ->", off);
                        if (dmp < kd->kvm_map ||
                            dmp > kd->kvm_map + kd->kvm_dump.dump_hashmask ||
                            (off & (sizeof (offset_t) - 1)) != 0 ||
                            DUMP_HASH(&kd->kvm_dump, dmp->dm_as, dmp->dm_va) !=
                            hash) {
                                if (kd->kvm_debug)
                                        fprintf(stderr, " dump map corrupt\n");
                                return (0);
                        }
                        if (dmp->dm_va == page && dmp->dm_as == as) {
                                off = dmp->dm_data + pageoff;
                                break;
                        }
                        off = dmp->dm_next;
                }
        }
        if (kd->kvm_debug)
                fprintf(stderr, "%s found: %llx\n", off ? "" : " not", off);
        return (off);
}

static ssize_t
kvm_rw(kvm_t *kd, uint64_t addr, void *buf, size_t size,
        struct as *as, ssize_t (*prw)(int, void *, size_t, offset_t))
{
        offset_t off;
        size_t resid = size;

        /*
         * read/write of zero bytes always succeeds
         */
        if (size == 0)
                return (0);

        if (kd->kvm_core == NULL) {
                char procbuf[100];
                int procfd;
                ssize_t rval;

                if (as == kd->kvm_kas)
                        return (prw(kd->kvm_kmemfd, buf, size, addr));
                if (as == NULL)
                        return (prw(kd->kvm_memfd, buf, size, addr));

                (void) sprintf(procbuf, "/proc/%ld/as", kd->kvm_pid);
                if ((procfd = open64(procbuf, kd->kvm_openflag)) == -1)
                        return (-1);
                rval = prw(procfd, buf, size, addr);
                (void) close(procfd);
                return (rval);
        }

        while (resid != 0) {
                uintptr_t pageoff = addr & (kd->kvm_dump.dump_pagesize - 1);
                ssize_t len = MIN(resid, kd->kvm_dump.dump_pagesize - pageoff);

                if ((off = kvm_lookup(kd, as, addr)) == 0)
                        break;

                if (prw == PREAD && off < kd->kvm_coremapsize)
                        bcopy(kd->kvm_core + off, buf, len);
                else if ((len = prw(kd->kvm_corefd, buf, len, off)) <= 0)
                        break;
                resid -= len;
                addr += len;
                buf = (char *)buf + len;
        }
        return (resid < size ? size - resid : -1);
}

ssize_t
kvm_read(kvm_t *kd, uintptr_t addr, void *buf, size_t size)
{
        return (kvm_rw(kd, addr, buf, size, kd->kvm_kas, PREAD));
}

ssize_t
kvm_kread(kvm_t *kd, uintptr_t addr, void *buf, size_t size)
{
        return (kvm_rw(kd, addr, buf, size, kd->kvm_kas, PREAD));
}

ssize_t
kvm_uread(kvm_t *kd, uintptr_t addr, void *buf, size_t size)
{
        return (kvm_rw(kd, addr, buf, size, kd->kvm_proc.p_as, PREAD));
}

ssize_t
kvm_aread(kvm_t *kd, uintptr_t addr, void *buf, size_t size, struct as *as)
{
        return (kvm_rw(kd, addr, buf, size, as, PREAD));
}

ssize_t
kvm_pread(kvm_t *kd, uint64_t addr, void *buf, size_t size)
{
        return (kvm_rw(kd, addr, buf, size, NULL, PREAD));
}

ssize_t
kvm_write(kvm_t *kd, uintptr_t addr, const void *buf, size_t size)
{
        return (kvm_rw(kd, addr, (void *)buf, size, kd->kvm_kas, PWRITE));
}

ssize_t
kvm_kwrite(kvm_t *kd, uintptr_t addr, const void *buf, size_t size)
{
        return (kvm_rw(kd, addr, (void *)buf, size, kd->kvm_kas, PWRITE));
}

ssize_t
kvm_uwrite(kvm_t *kd, uintptr_t addr, const void *buf, size_t size)
{
        return (kvm_rw(kd, addr, (void *)buf, size, kd->kvm_proc.p_as, PWRITE));
}

ssize_t
kvm_awrite(kvm_t *kd, uintptr_t addr, const void *buf, size_t size,
    struct as *as)
{
        return (kvm_rw(kd, addr, (void *)buf, size, as, PWRITE));
}

ssize_t
kvm_pwrite(kvm_t *kd, uint64_t addr, const void *buf, size_t size)
{
        return (kvm_rw(kd, addr, (void *)buf, size, NULL, PWRITE));
}

uint64_t
kvm_physaddr(kvm_t *kd, struct as *as, uintptr_t addr)
{
        mem_vtop_t mem_vtop;
        offset_t off;

        if (kd->kvm_core == NULL) {
                mem_vtop.m_as = as;
                mem_vtop.m_va = (void *)addr;
                if (ioctl(kd->kvm_kmemfd, MEM_VTOP, &mem_vtop) == 0)
                        return ((uint64_t)mem_vtop.m_pfn * getpagesize() +
                            (addr & (getpagesize() - 1)));
        } else {
                if ((off = kvm_lookup(kd, as, addr)) != 0) {
                        long pfn_index =
                            (u_offset_t)(off - kd->kvm_dump.dump_data) >>
                            kd->kvm_dump.dump_pageshift;
                        return (((uint64_t)kd->kvm_pfn[pfn_index] <<
                            kd->kvm_dump.dump_pageshift) +
                            (addr & (kd->kvm_dump.dump_pagesize - 1)));
                }
        }
        return (-1ULL);
}

struct proc *
kvm_getproc(kvm_t *kd, pid_t pid)
{
        (void) kvm_setproc(kd);
        while (kvm_nextproc(kd) != NULL)
                if (kd->kvm_pid == pid)
                        return (&kd->kvm_proc);
        return (NULL);
}

struct proc *
kvm_nextproc(kvm_t *kd)
{
        if (kd->kvm_proc.p_next == NULL ||
            kvm_kread(kd, (uintptr_t)kd->kvm_proc.p_next,
            &kd->kvm_proc, sizeof (proc_t)) != sizeof (proc_t) ||
            kvm_kread(kd, (uintptr_t)&kd->kvm_proc.p_pidp->pid_id,
            &kd->kvm_pid, sizeof (pid_t)) != sizeof (pid_t))
                return (NULL);

        return (&kd->kvm_proc);
}

int
kvm_setproc(kvm_t *kd)
{
        (void) kvm_kread(kd, (uintptr_t)kd->kvm_practive,
            &kd->kvm_proc.p_next, sizeof (proc_t *));
        kd->kvm_pid = -1;
        return (0);
}

/*ARGSUSED*/
struct user *
kvm_getu(kvm_t *kd, struct proc *p)
{
        return (&p->p_user);
}