Runtime/testcase/tstRTR0DbgKrnlInfoDriver.cpp

68fce2926ac031f46c25460569296f66620e9d52vboxsync/* $Id$ */
68fce2926ac031f46c25460569296f66620e9d52vboxsync/** @file
68fce2926ac031f46c25460569296f66620e9d52vboxsync * IPRT R0 Testcase - Debug kernel information, driver program.
68fce2926ac031f46c25460569296f66620e9d52vboxsync */
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync/*
68fce2926ac031f46c25460569296f66620e9d52vboxsync * Copyright (C) 2012 Oracle Corporation
68fce2926ac031f46c25460569296f66620e9d52vboxsync *
68fce2926ac031f46c25460569296f66620e9d52vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
68fce2926ac031f46c25460569296f66620e9d52vboxsync * available from http://www.virtualbox.org. This file is free software;
68fce2926ac031f46c25460569296f66620e9d52vboxsync * you can redistribute it and/or modify it under the terms of the GNU
68fce2926ac031f46c25460569296f66620e9d52vboxsync * General Public License (GPL) as published by the Free Software
68fce2926ac031f46c25460569296f66620e9d52vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
68fce2926ac031f46c25460569296f66620e9d52vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
68fce2926ac031f46c25460569296f66620e9d52vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
68fce2926ac031f46c25460569296f66620e9d52vboxsync *
68fce2926ac031f46c25460569296f66620e9d52vboxsync * The contents of this file may alternatively be used under the terms
68fce2926ac031f46c25460569296f66620e9d52vboxsync * of the Common Development and Distribution License Version 1.0
68fce2926ac031f46c25460569296f66620e9d52vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
68fce2926ac031f46c25460569296f66620e9d52vboxsync * VirtualBox OSE distribution, in which case the provisions of the
68fce2926ac031f46c25460569296f66620e9d52vboxsync * CDDL are applicable instead of those of the GPL.
68fce2926ac031f46c25460569296f66620e9d52vboxsync *
68fce2926ac031f46c25460569296f66620e9d52vboxsync * You may elect to license modified versions of this file under the
68fce2926ac031f46c25460569296f66620e9d52vboxsync * terms and conditions of either the GPL or the CDDL or both.
68fce2926ac031f46c25460569296f66620e9d52vboxsync */
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync/*******************************************************************************
68fce2926ac031f46c25460569296f66620e9d52vboxsync*   Header Files                                                               *
68fce2926ac031f46c25460569296f66620e9d52vboxsync*******************************************************************************/
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/initterm.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/err.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/path.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/param.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/stream.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/string.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#include <iprt/test.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync#ifdef VBOX
68fce2926ac031f46c25460569296f66620e9d52vboxsync# include <VBox/sup.h>
68fce2926ac031f46c25460569296f66620e9d52vboxsync# include "tstRTR0DbgKrnlInfo.h"
68fce2926ac031f46c25460569296f66620e9d52vboxsync#endif
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync/**
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync *  Entry point.
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync */
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsyncextern "C" DECLEXPORT(int) TrustedMain(int argc, char **argv, char **envp)
68fce2926ac031f46c25460569296f66620e9d52vboxsync{
68fce2926ac031f46c25460569296f66620e9d52vboxsync#ifndef VBOX
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTPrintf("tstSup: SKIPPED\n");
68fce2926ac031f46c25460569296f66620e9d52vboxsync    return 0;
68fce2926ac031f46c25460569296f66620e9d52vboxsync#else
68fce2926ac031f46c25460569296f66620e9d52vboxsync    /*
68fce2926ac031f46c25460569296f66620e9d52vboxsync     * Init.
68fce2926ac031f46c25460569296f66620e9d52vboxsync     */
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTEST hTest;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    int rc = RTTestInitAndCreate("tstRTR0DbgKrnlInfo", &hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (rc)
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return rc;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTestBanner(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    uint8_t *pbPage = (uint8_t *)RTTestGuardedAllocTail(hTest, PAGE_SIZE);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (!pbPage)
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestFailed(hTest, "RTTestGuardedAllocTail failed with rc=%Rrc\n", rc);
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    }
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    PSUPDRVSESSION pSession;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    rc = SUPR3Init(&pSession);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestFailed(hTest, "SUPR3Init failed with rc=%Rrc\n", rc);
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    }
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    char szPath[RTPATH_MAX];
68fce2926ac031f46c25460569296f66620e9d52vboxsync    rc = RTPathExecDir(szPath, sizeof(szPath));
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_SUCCESS(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync        rc = RTPathAppend(szPath, sizeof(szPath), "tstRTR0DbgKrnlInfo.r0");
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestFailed(hTest, "Failed constructing .r0 filename (rc=%Rrc)", rc);
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    }
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    void *pvImageBase;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    rc = SUPR3LoadServiceModule(szPath, "tstRTR0DbgKrnlInfo",
c699250ccce7166a16be96ddc9e9775df6453db2vboxsync                                "TSTR0DbgKrnlInfoSrvReqHandler",
68fce2926ac031f46c25460569296f66620e9d52vboxsync                                &pvImageBase);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestFailed(hTest, "SUPR3LoadServiceModule(%s,,,) failed with rc=%Rrc\n", szPath, rc);
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    }
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    /* test request */
68fce2926ac031f46c25460569296f66620e9d52vboxsync    struct
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        SUPR0SERVICEREQHDR  Hdr;
68fce2926ac031f46c25460569296f66620e9d52vboxsync        char                szMsg[256];
68fce2926ac031f46c25460569296f66620e9d52vboxsync    } Req;
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    /*
68fce2926ac031f46c25460569296f66620e9d52vboxsync     * Sanity checks.
68fce2926ac031f46c25460569296f66620e9d52vboxsync     */
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTestSub(hTest, "Sanity");
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.u32Magic = SUPR0SERVICEREQHDR_MAGIC;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.cbReq = sizeof(Req);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.szMsg[0] = '\0';
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTESTI_CHECK_RC(rc = SUPR3CallR0Service("tstRTR0DbgKrnlInfo", sizeof("tstRTR0DbgKrnlInfo") - 1,
68fce2926ac031f46c25460569296f66620e9d52vboxsync                                             TSTRTR0DBGKRNLINFO_SANITY_OK, 0, &Req.Hdr), VINF_SUCCESS);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTESTI_CHECK_MSG(Req.szMsg[0] == '\0', ("%s", Req.szMsg));
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (Req.szMsg[0] != '\0')
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.u32Magic = SUPR0SERVICEREQHDR_MAGIC;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.cbReq = sizeof(Req);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.szMsg[0] = '\0';
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTESTI_CHECK_RC(rc = SUPR3CallR0Service("tstRTR0DbgKrnlInfo", sizeof("tstRTR0DbgKrnlInfo") - 1,
68fce2926ac031f46c25460569296f66620e9d52vboxsync                                             TSTRTR0DBGKRNLINFO_SANITY_FAILURE, 0, &Req.Hdr), VINF_SUCCESS);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
1939436fa43cbf7f5cdc05a3830ed624d5fe4a6avboxsync    RTTESTI_CHECK_MSG(!strncmp(Req.szMsg, RT_STR_TUPLE("!42failure42")), ("%s", Req.szMsg));
1939436fa43cbf7f5cdc05a3830ed624d5fe4a6avboxsync    if (strncmp(Req.szMsg, RT_STR_TUPLE("!42failure42")))
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    /*
68fce2926ac031f46c25460569296f66620e9d52vboxsync     * Basic tests, bail out on failure.
68fce2926ac031f46c25460569296f66620e9d52vboxsync     */
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTestSub(hTest, "Basics");
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.u32Magic = SUPR0SERVICEREQHDR_MAGIC;
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.Hdr.cbReq = sizeof(Req);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    Req.szMsg[0] = '\0';
68fce2926ac031f46c25460569296f66620e9d52vboxsync    RTTESTI_CHECK_RC(rc = SUPR3CallR0Service("tstRTR0DbgKrnlInfo", sizeof("tstRTR0DbgKrnlInfo") - 1,
68fce2926ac031f46c25460569296f66620e9d52vboxsync                                             TSTRTR0DBGKRNLINFO_BASIC, 0, &Req.Hdr), VINF_SUCCESS);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (RT_FAILURE(rc))
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (Req.szMsg[0] == '!')
68fce2926ac031f46c25460569296f66620e9d52vboxsync    {
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestIFailed("%s", &Req.szMsg[1]);
68fce2926ac031f46c25460569296f66620e9d52vboxsync        return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync    }
68fce2926ac031f46c25460569296f66620e9d52vboxsync    if (Req.szMsg[0])
68fce2926ac031f46c25460569296f66620e9d52vboxsync        RTTestIPrintf(RTTESTLVL_ALWAYS, "%s", Req.szMsg);
68fce2926ac031f46c25460569296f66620e9d52vboxsync
68fce2926ac031f46c25460569296f66620e9d52vboxsync    /*
68fce2926ac031f46c25460569296f66620e9d52vboxsync     * Done.
68fce2926ac031f46c25460569296f66620e9d52vboxsync     */
68fce2926ac031f46c25460569296f66620e9d52vboxsync    return RTTestSummaryAndDestroy(hTest);
68fce2926ac031f46c25460569296f66620e9d52vboxsync#endif
68fce2926ac031f46c25460569296f66620e9d52vboxsync}
68fce2926ac031f46c25460569296f66620e9d52vboxsync
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync#if !defined(VBOX_WITH_HARDENING) || !defined(RT_OS_WINDOWS)
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync/**
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync * Main entry point.
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync */
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsyncint main(int argc, char **argv, char **envp)
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync{
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync    return TrustedMain(argc, argv, envp);
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync}
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync#endif
a0ed9c35186427ae0abbd14b1cfdf43bf2dc8659vboxsync