Support/win/SUPR0IdcClient-win.c

	SUPR0IdcClient-win.c revision 8e496a172b2000333186e5c1be75a50f398ed830
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt/* $Id$ */
8e5fce1f9ceba17dd7e3ff0eb287e1e999c14249Mark Andrews/** @file
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * VirtualBox Support Driver - IDC Client Lib, Windows Specific Code.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt/*
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * Copyright (C) 2008 Sun Microsystems, Inc.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * This file is part of VirtualBox Open Source Edition (OSE), as
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * available from http://www.virtualbox.org. This file is free software;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * you can redistribute it and/or modify it under the terms of the GNU
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * General Public License (GPL) as published by the Free Software
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * Foundation, in version 2 as it comes in the "COPYING" file of the
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * The contents of this file may alternatively be used under the terms
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * of the Common Development and Distribution License Version 1.0
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * VirtualBox OSE distribution, in which case the provisions of the
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * CDDL are applicable instead of those of the GPL.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * You may elect to license modified versions of this file under the
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * terms and conditions of either the GPL or the CDDL or both.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * Clara, CA 95054 USA or visit http://www.sun.com if you need
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt * additional information or have any questions.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt/*******************************************************************************
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt*   Header Files                                                               *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt*******************************************************************************/
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt#include "../SUPR0IdcClientInternal.h"
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt#include <VBox/err.h>
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt/*******************************************************************************
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt*   Defined Constants And Macros                                               *
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt*******************************************************************************/
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt/** NT Device name. */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt#define DEVICE_NAME_NT   L"\\Device\\VBoxDrv"
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Huntstatic int supR0IdcNtCallInternal(PDEVICE_OBJECT pDeviceObject, uint32_t iReq, PSUPDRVIDCREQHDR pReq)
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt{
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    int                 rc;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    IO_STATUS_BLOCK     IoStatusBlock;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    KEVENT              Event;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    PIRP                pIrp;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    /*
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt     * Build the request.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt     */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    KeInitializeEvent(&Event, NotificationEvent, FALSE);
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    pIrp = IoBuildDeviceIoControlRequest(iReq,              /* IoControlCode */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         pDeviceObject,
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         pReq,              /* InputBuffer */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         pReq->cb,          /* InputBufferLength */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         pReq,              /* OutputBuffer */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         pReq->cb,          /* OutputBufferLength */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         TRUE,              /* InternalDeviceIoControl (=> IRP_MJ_INTERNAL_DEVICE_CONTROL) */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         &Event,            /* Event */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         &IoStatusBlock);   /* IoStatusBlock */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    if (pIrp)
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    {
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        /*
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt         * Call the driver, wait for an async request to complete (should never happen).
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt         */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        NTSTATUS rcNt = IoCallDriver(pDeviceObject, pIrp);
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        if (rcNt == STATUS_PENDING)
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        {
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt            rcNt = KeWaitForSingleObject(&Event,            /* Object */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         Executive,         /* WaitReason */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         KernelMode,        /* WaitMode */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         FALSE,             /* Altertable */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt                                         NULL);             /* TimeOut */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt            rcNt = IoStatusBlock.Status;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        }
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        if (NT_SUCCESS(rcNt))
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt            rc = pReq->rc;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        else
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt            rc = RTErrConvertFromNtStatus(rcNt);
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    }
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    else
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt        rc = VERR_NO_MEMORY;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    return rc;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt}
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Huntint VBOXCALL supR0IdcNativeOpen(PSUPDRVIDCHANDLE pHandle, PSUPDRVIDCREQCONNECT pReq)
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt{
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    PDEVICE_OBJECT  pDeviceObject = NULL;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    PFILE_OBJECT    pFileObject = NULL;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    UNICODE_STRING  wszDeviceName;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    NTSTATUS        rcNt;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    int             rc;
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    /*
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt     * Get the device object pointer.
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt     */
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    RtlInitUnicodeString(&wszDeviceName, DEVICE_NAME_NT);
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    rcNt = IoGetDeviceObjectPointer(&wszDeviceName, FILE_ALL_ACCESS, &pFileObject, &pDeviceObject);
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    if (NT_SUCCESS(rcNt))
501941f0b6cce74c2ff75b10aff3f230d5d37e4cEvan Hunt    {
        /*
         * Make the connection call.
         */
        rc = supR0IdcNtCallInternal(pDeviceObject, SUPDRV_IDC_REQ_CONNECT, &pReq->Hdr);
        if (RT_SUCCESS(rc))
        {
            pHandle->s.pDeviceObject = pDeviceObject;
            pHandle->s.pFileObject = pFileObject;
            return rc;
        }

        /* only the file object. */
        ObDereferenceObject(pFileObject);
    }
    else
        rc = RTErrConvertFromNtStatus(rcNt);

    pHandle->s.pDeviceObject = NULL;
    pHandle->s.pFileObject = NULL;
    return rc;
}


int VBOXCALL supR0IdcNativeClose(PSUPDRVIDCHANDLE pHandle, PSUPDRVIDCREQHDR pReq)
{
    PDEVICE_OBJECT  pDeviceObject = pHandle->s.pDeviceObject;
    PFILE_OBJECT    pFileObject   = pHandle->s.pFileObject;
    int rc = supR0IdcNtCallInternal(pDeviceObject, SUPDRV_IDC_REQ_DISCONNECT, pReq);
    if (RT_SUCCESS(rc))
    {
        pHandle->s.pDeviceObject = NULL;
        pHandle->s.pFileObject  = NULL;
        ObDereferenceObject(pFileObject);
    }

    return rc;
}


int VBOXCALL supR0IdcNativeCall(PSUPDRVIDCHANDLE pHandle, uint32_t iReq, PSUPDRVIDCREQHDR pReq)
{
    return supR0IdcNtCallInternal(pHandle->s.pDeviceObject, iReq, pReq);
}