/**********************************************************************
* Copyright (c) 2013-2014  Red Hat, Inc.
*
* Developed by Daynix Computing LTD.
*
* Authors:
*     Dmitry Fleytman <dmitry@daynix.com>
*     Pavel Gurvich <pavel@daynix.com>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
**********************************************************************/

#include "stdafx.h"
#include "DriverAccess.h"
#include "Public.h"


void UsbDkDriverAccess::GetDevicesList(PUSB_DK_DEVICE_INFO &DevicesArray, ULONG &DeviceNumber)
{
    DevicesArray = nullptr;
    DWORD   bytesReturned;

    unique_ptr<USB_DK_DEVICE_INFO[]> Result;

    do
    {
        // get number of devices
        Ioctl(IOCTL_USBDK_COUNT_DEVICES, false, nullptr, 0,
              &DeviceNumber, sizeof(DeviceNumber));

        if (DeviceNumber == 0)
        {
            DevicesArray = nullptr;
            return;
        }

        // allocate storage for device list
        Result.reset(new USB_DK_DEVICE_INFO[DeviceNumber]);

    } while (!Ioctl(IOCTL_USBDK_ENUM_DEVICES, true, nullptr, 0,
                    Result.get(), DeviceNumber * sizeof(USB_DK_DEVICE_INFO),
                    &bytesReturned));

    DeviceNumber = bytesReturned / sizeof(USB_DK_DEVICE_INFO);
    DevicesArray = Result.release();
}

void UsbDkDriverAccess::ReleaseConfigurationDescriptor(PUSB_CONFIGURATION_DESCRIPTOR Descriptor)
{
    delete[] Descriptor;
}

void UsbDkDriverAccess::ReleaseDevicesList(PUSB_DK_DEVICE_INFO DevicesArray)
{
    delete[] DevicesArray;
}

PUSB_CONFIGURATION_DESCRIPTOR UsbDkDriverAccess::GetConfigurationDescriptor(USB_DK_CONFIG_DESCRIPTOR_REQUEST &Request, ULONG &Length)
{
    USB_CONFIGURATION_DESCRIPTOR ShortDescriptor;

    Ioctl(IOCTL_USBDK_GET_CONFIG_DESCRIPTOR, true, &Request, sizeof(Request),
          &ShortDescriptor, sizeof(ShortDescriptor));

    Length = ShortDescriptor.wTotalLength;

    auto FullDescriptor = reinterpret_cast<PUSB_CONFIGURATION_DESCRIPTOR>(new BYTE[Length]);

    Ioctl(IOCTL_USBDK_GET_CONFIG_DESCRIPTOR, false, &Request, sizeof(Request),
          FullDescriptor, Length);

    return FullDescriptor;
}

void UsbDkDriverAccess::UpdateRegistryParameters()
{
    Ioctl(IOCTL_USBDK_UPDATE_REG_PARAMETERS);
}

HANDLE UsbDkDriverAccess::AddRedirect(USB_DK_DEVICE_ID &DeviceID)
{
    ULONG64 RedirectorHandle;
    SendIoctlWithDeviceId(IOCTL_USBDK_ADD_REDIRECT, DeviceID, &RedirectorHandle);
    return reinterpret_cast<HANDLE>(RedirectorHandle);
}

void UsbDkHiderAccess::AddHideRule(const USB_DK_HIDE_RULE &Rule)
{
    Ioctl(IOCTL_USBDK_ADD_HIDE_RULE, false, const_cast<PUSB_DK_HIDE_RULE>(&Rule), sizeof(Rule));
}

void UsbDkHiderAccess::ClearHideRules()
{
    Ioctl(IOCTL_USBDK_CLEAR_HIDE_RULES);
}