tstRTR0Thread.cpp revision 01bfc3834fc0d724e808adf928b2bfe541f9f211
/* $Id$ */
/** @file
* IPRT R0 Testcase - Kernel thread.
*/
/*
* Copyright (C) 2015 Oracle Corporation
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* you can redistribute it and/or modify it under the terms of the GNU
* General Public License (GPL) as published by the Free Software
* Foundation, in version 2 as it comes in the "COPYING" file of the
* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
*
* The contents of this file may alternatively be used under the terms
* of the Common Development and Distribution License Version 1.0
* (CDDL) only, as it comes in the "COPYING.CDDL" file of the
* VirtualBox OSE distribution, in which case the provisions of the
* CDDL are applicable instead of those of the GPL.
*
* You may elect to license modified versions of this file under the
* terms and conditions of either the GPL or the CDDL or both.
*/
/*******************************************************************************
* Header Files *
*******************************************************************************/
#include <iprt/thread.h>
#include <iprt/asm-amd64-x86.h>
#include <iprt/err.h>
#include <VBox/sup.h>
#include "tstRTR0Thread.h"
#include "tstRTR0Common.h"
#define TSTRTR0THREADDATA_MAGIC 0xcececece
/**
* State structure for tstRTR0ThreadCallback().
*/
typedef struct TSTRTR0THREADDATA
{
/** The magic value. */
uint32_t uMagic;
/** Sample counter. */
uint32_t volatile cCounter;
/** The thread handle. */
RTTHREAD hThread;
} TSTRTR0THREADDATA;
/** Pointer to state structure for tstRTR0ThreadCallback(). */
typedef TSTRTR0THREADDATA *PTSTRTR0THREADDATA;
static DECLCALLBACK(int) tstRTR0ThreadCallback(RTTHREAD hThread, void *pvUser)
{
PTSTRTR0THREADDATA pData = (PTSTRTR0THREADDATA)pvUser;
if (RT_LIKELY(pData))
{
if (pData->uMagic == TSTRTR0THREADDATA_MAGIC)
pData->uMagic = ~pData->uMagic;
if (pData->cCounter == 127)
pData->cCounter = 196;
}
RTThreadUserSignal(RTThreadSelf());
return VINF_SUCCESS;
}
/**
* Service request callback function.
*
* @returns VBox status code.
* @param pSession The caller's session.
* @param u64Arg 64-bit integer argument.
* @param pReqHdr The request header. Input / Output. Optional.
*/
DECLEXPORT(int) TSTRTR0ThreadSrvReqHandler(PSUPDRVSESSION pSession, uint32_t uOperation,
uint64_t u64Arg, PSUPR0SERVICEREQHDR pReqHdr)
{
RTR0TESTR0_SRV_REQ_PROLOG_RET(pReqHdr);
/*
* The big switch.
*/
switch (uOperation)
{
RTR0TESTR0_IMPLEMENT_SANITY_CASES();
RTR0TESTR0_IMPLEMENT_DEFAULT_CASE(uOperation);
case TSTRTR0THREAD_BASIC:
{
TSTRTR0THREADDATA Data;
RT_ZERO(Data);
Data.uMagic = TSTRTR0THREADDATA_MAGIC;
Data.cCounter = 127;
/* Create the kernel thread. */
RTR0TESTR0_CHECK_RC_BREAK(RTThreadCreate(&Data.hThread, tstRTR0ThreadCallback, &Data, 0 /* cbStack */,
RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE, "tstRTR0Thr"), VINF_SUCCESS);
/* Wait for thread to signal. */
RTR0TESTR0_CHECK_RC(RTThreadUserWait(Data.hThread, 500 /* ms */), VINF_SUCCESS);
/* Reset the semaphore. */
RTR0TESTR0_CHECK_RC(RTThreadUserReset(Data.hThread), VINF_SUCCESS);
/* Check if the thread has modified data as expected. */
RTR0TESTR0_CHECK_MSG_BREAK(Data.cCounter = 196 && Data.uMagic == ~TSTRTR0THREADDATA_MAGIC,
("Thread didn't modify data as expected.\n"));
break;
}
}
RTR0TESTR0_SRV_REQ_EPILOG(pReqHdr);
/* The error indicator is the '!' in the message buffer. */
return VINF_SUCCESS;
}