312 lines
10 KiB
C
312 lines
10 KiB
C
|
/*
|
||
|
|
LUFA Library
|
||
|
|
Copyright (C) Dean Camera, 2019.
|
||
|
|
|
||
|
|
dean [at] fourwalledcubicle [dot] com
|
||
|
|
www.lufa-lib.org
|
||
|
|
*/
|
||
|
|
|
||
|
|
/*
|
||
|
|
Copyright 2019 Dean Camera (dean [at] fourwalledcubicle [dot] com)
|
||
|
|
|
||
|
|
Permission to use, copy, modify, distribute, and sell this
|
||
|
|
software and its documentation for any purpose is hereby granted
|
||
|
|
without fee, provided that the above copyright notice appear in
|
||
|
|
all copies and that both that the copyright notice and this
|
||
|
|
permission notice and warranty disclaimer appear in supporting
|
||
|
|
documentation, and that the name of the author not be used in
|
||
|
|
advertising or publicity pertaining to distribution of the
|
||
|
|
software without specific, written prior permission.
|
||
|
|
|
||
|
|
The author disclaims all warranties with regard to this
|
||
|
|
software, including all implied warranties of merchantability
|
||
|
|
and fitness. In no event shall the author be liable for any
|
||
|
|
special, indirect or consequential damages or any damages
|
||
|
|
whatsoever resulting from loss of use, data or profits, whether
|
||
|
|
in an action of contract, negligence or other tortious action,
|
||
|
|
arising out of or in connection with the use or performance of
|
||
|
|
this software.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/** \file
|
||
|
|
*
|
||
|
|
* RNDIS Device commands, to issue RNDIS commands to the device for
|
||
|
|
* the control and data transfer between the host and RNDIS device.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "RNDISCommands.h"
|
||
|
|
|
||
|
|
/** Current RNDIS Request ID, for associating sent commands with received data */
|
||
|
|
uint32_t RequestID = 0;
|
||
|
|
|
||
|
|
|
||
|
|
/** Function to send the given encapsulated RNDIS command to the device.
|
||
|
|
*
|
||
|
|
* \param[in] Buffer Source command data buffer to send to the device
|
||
|
|
* \param[in] Length Number of bytes to send
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_SendEncapsulatedCommand(void* const Buffer,
|
||
|
|
const uint16_t Length)
|
||
|
|
{
|
||
|
|
USB_ControlRequest = (USB_Request_Header_t)
|
||
|
|
{
|
||
|
|
.bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE),
|
||
|
|
.bRequest = RNDIS_REQ_SendEncapsulatedCommand,
|
||
|
|
.wValue = 0,
|
||
|
|
.wIndex = 0,
|
||
|
|
.wLength = Length,
|
||
|
|
};
|
||
|
|
|
||
|
|
/* Select the control pipe for the request transfer */
|
||
|
|
Pipe_SelectPipe(PIPE_CONTROLPIPE);
|
||
|
|
|
||
|
|
return USB_Host_SendControlRequest(Buffer);
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Function to receive the given encapsulated RNDIS response from the device.
|
||
|
|
*
|
||
|
|
* \param[out] Buffer Destination command data buffer to write read data from the device to
|
||
|
|
* \param[in] Length Number of bytes to read
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_GetEncapsulatedResponse(void* const Buffer,
|
||
|
|
const uint16_t Length)
|
||
|
|
{
|
||
|
|
USB_ControlRequest = (USB_Request_Header_t)
|
||
|
|
{
|
||
|
|
.bmRequestType = (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE),
|
||
|
|
.bRequest = RNDIS_REQ_GetEncapsulatedResponse,
|
||
|
|
.wValue = 0,
|
||
|
|
.wIndex = 0,
|
||
|
|
.wLength = Length,
|
||
|
|
};
|
||
|
|
|
||
|
|
/* Select the control pipe for the request transfer */
|
||
|
|
Pipe_SelectPipe(PIPE_CONTROLPIPE);
|
||
|
|
|
||
|
|
return USB_Host_SendControlRequest(Buffer);
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Sends a RNDIS KEEPALIVE command to the device, to ensure that it does not enter standby mode after periods
|
||
|
|
* of long inactivity.
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum or RNDIS_COMMAND_FAILED if the device returned a
|
||
|
|
* logical command failure
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_SendKeepAlive(void)
|
||
|
|
{
|
||
|
|
uint8_t ErrorCode;
|
||
|
|
|
||
|
|
RNDIS_KeepAlive_Message_t KeepAliveMessage;
|
||
|
|
RNDIS_KeepAlive_Complete_t KeepAliveMessageResponse;
|
||
|
|
|
||
|
|
KeepAliveMessage.MessageType = REMOTE_NDIS_KEEPALIVE_MSG;
|
||
|
|
KeepAliveMessage.MessageLength = sizeof(RNDIS_KeepAlive_Message_t);
|
||
|
|
KeepAliveMessage.RequestId = RequestID++;
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_SendEncapsulatedCommand(&KeepAliveMessage,
|
||
|
|
sizeof(RNDIS_KeepAlive_Message_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_GetEncapsulatedResponse(&KeepAliveMessageResponse,
|
||
|
|
sizeof(RNDIS_KeepAlive_Complete_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
return HOST_SENDCONTROL_Successful;
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Initializes the attached RNDIS device's RNDIS interface.
|
||
|
|
*
|
||
|
|
* \param[in] HostMaxPacketSize Size of the packet buffer on the host
|
||
|
|
* \param[out] DeviceMaxPacketSize Pointer to where the packet buffer size of the device is to be stored
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum or RNDIS_COMMAND_FAILED if the device returned a
|
||
|
|
* logical command failure
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_InitializeDevice(const uint16_t HostMaxPacketSize,
|
||
|
|
uint16_t* const DeviceMaxPacketSize)
|
||
|
|
{
|
||
|
|
uint8_t ErrorCode;
|
||
|
|
|
||
|
|
RNDIS_Initialize_Message_t InitMessage;
|
||
|
|
RNDIS_Initialize_Complete_t InitMessageResponse;
|
||
|
|
|
||
|
|
InitMessage.MessageType = REMOTE_NDIS_INITIALIZE_MSG;
|
||
|
|
InitMessage.MessageLength = sizeof(RNDIS_Initialize_Message_t);
|
||
|
|
InitMessage.RequestId = RequestID++;
|
||
|
|
|
||
|
|
InitMessage.MajorVersion = REMOTE_NDIS_VERSION_MAJOR;
|
||
|
|
InitMessage.MinorVersion = REMOTE_NDIS_VERSION_MINOR;
|
||
|
|
InitMessage.MaxTransferSize = HostMaxPacketSize;
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_SendEncapsulatedCommand(&InitMessage,
|
||
|
|
sizeof(RNDIS_Initialize_Message_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_GetEncapsulatedResponse(&InitMessageResponse,
|
||
|
|
sizeof(RNDIS_Initialize_Complete_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (InitMessageResponse.Status != REMOTE_NDIS_STATUS_SUCCESS)
|
||
|
|
return RNDIS_COMMAND_FAILED;
|
||
|
|
|
||
|
|
*DeviceMaxPacketSize = InitMessageResponse.MaxTransferSize;
|
||
|
|
|
||
|
|
return HOST_SENDCONTROL_Successful;
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Sets a given RNDIS property of an attached RNDIS device.
|
||
|
|
*
|
||
|
|
* \param[in] Oid OID number of the parameter to set
|
||
|
|
* \param[in] Buffer Pointer to where the property data is to be sourced from
|
||
|
|
* \param[in] Length Length in bytes of the property data to sent to the device
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum or RNDIS_COMMAND_FAILED if the device returned a
|
||
|
|
* logical command failure
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_SetRNDISProperty(const uint32_t Oid,
|
||
|
|
void* Buffer,
|
||
|
|
const uint16_t Length)
|
||
|
|
{
|
||
|
|
uint8_t ErrorCode;
|
||
|
|
|
||
|
|
struct
|
||
|
|
{
|
||
|
|
RNDIS_Set_Message_t SetMessage;
|
||
|
|
uint8_t ContiguousBuffer[Length];
|
||
|
|
} SetMessageData;
|
||
|
|
|
||
|
|
RNDIS_Set_Complete_t SetMessageResponse;
|
||
|
|
|
||
|
|
SetMessageData.SetMessage.MessageType = REMOTE_NDIS_SET_MSG;
|
||
|
|
SetMessageData.SetMessage.MessageLength = sizeof(RNDIS_Set_Message_t) + Length;
|
||
|
|
SetMessageData.SetMessage.RequestId = RequestID++;
|
||
|
|
|
||
|
|
SetMessageData.SetMessage.Oid = Oid;
|
||
|
|
SetMessageData.SetMessage.InformationBufferLength = Length;
|
||
|
|
SetMessageData.SetMessage.InformationBufferOffset = (sizeof(RNDIS_Set_Message_t) - sizeof(RNDIS_Message_Header_t));
|
||
|
|
SetMessageData.SetMessage.DeviceVcHandle = 0;
|
||
|
|
|
||
|
|
memcpy(SetMessageData.ContiguousBuffer, Buffer, Length);
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_SendEncapsulatedCommand(&SetMessageData,
|
||
|
|
SetMessageData.SetMessage.MessageLength)) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_GetEncapsulatedResponse(&SetMessageResponse,
|
||
|
|
sizeof(RNDIS_Set_Complete_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (SetMessageResponse.Status != REMOTE_NDIS_STATUS_SUCCESS)
|
||
|
|
return RNDIS_COMMAND_FAILED;
|
||
|
|
|
||
|
|
return HOST_SENDCONTROL_Successful;
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Gets a given RNDIS property of an attached RNDIS device.
|
||
|
|
*
|
||
|
|
* \param[in] Oid OID number of the parameter to get
|
||
|
|
* \param[in] Buffer Pointer to where the property data is to be written to
|
||
|
|
* \param[in] MaxLength Length in bytes of the destination buffer size
|
||
|
|
*
|
||
|
|
* \return A value from the USB_Host_SendControlErrorCodes_t enum or RNDIS_COMMAND_FAILED if the device returned a
|
||
|
|
* logical command failure
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_QueryRNDISProperty(const uint32_t Oid,
|
||
|
|
void* Buffer,
|
||
|
|
const uint16_t MaxLength)
|
||
|
|
{
|
||
|
|
uint8_t ErrorCode;
|
||
|
|
|
||
|
|
RNDIS_Query_Message_t QueryMessage;
|
||
|
|
|
||
|
|
struct
|
||
|
|
{
|
||
|
|
RNDIS_Query_Complete_t QueryMessageResponse;
|
||
|
|
uint8_t ContiguousBuffer[MaxLength];
|
||
|
|
} QueryMessageResponseData;
|
||
|
|
|
||
|
|
QueryMessage.MessageType = REMOTE_NDIS_QUERY_MSG;
|
||
|
|
QueryMessage.MessageLength = sizeof(RNDIS_Query_Message_t);
|
||
|
|
QueryMessage.RequestId = RequestID++;
|
||
|
|
|
||
|
|
QueryMessage.Oid = Oid;
|
||
|
|
QueryMessage.InformationBufferLength = 0;
|
||
|
|
QueryMessage.InformationBufferOffset = 0;
|
||
|
|
QueryMessage.DeviceVcHandle = 0;
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_SendEncapsulatedCommand(&QueryMessage,
|
||
|
|
sizeof(RNDIS_Query_Message_t))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((ErrorCode = RNDIS_GetEncapsulatedResponse(&QueryMessageResponseData,
|
||
|
|
sizeof(QueryMessageResponseData))) != HOST_SENDCONTROL_Successful)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (QueryMessageResponseData.QueryMessageResponse.Status != REMOTE_NDIS_STATUS_SUCCESS)
|
||
|
|
return RNDIS_COMMAND_FAILED;
|
||
|
|
|
||
|
|
memcpy(Buffer, &QueryMessageResponseData.ContiguousBuffer, MaxLength);
|
||
|
|
|
||
|
|
return HOST_SENDCONTROL_Successful;
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Retrieves the size of a received packet, discarding the remainder of the RNDIS packet header to leave only the
|
||
|
|
* packet contents for processing by the host.
|
||
|
|
*
|
||
|
|
* \param[out] PacketLength Size of the packet currently in the pipe
|
||
|
|
*
|
||
|
|
* \return A value from the Pipe_Stream_RW_ErrorCodes_t enum
|
||
|
|
*/
|
||
|
|
uint8_t RNDIS_GetPacketLength(uint16_t* const PacketLength)
|
||
|
|
{
|
||
|
|
uint8_t ErrorCode;
|
||
|
|
|
||
|
|
Pipe_SelectPipe(RNDIS_DATA_IN_PIPE);
|
||
|
|
Pipe_SetPipeToken(PIPE_TOKEN_IN);
|
||
|
|
Pipe_Unfreeze();
|
||
|
|
|
||
|
|
if (!(Pipe_IsReadWriteAllowed()))
|
||
|
|
{
|
||
|
|
*PacketLength = 0;
|
||
|
|
Pipe_Freeze();
|
||
|
|
return PIPE_RWSTREAM_NoError;
|
||
|
|
}
|
||
|
|
|
||
|
|
RNDIS_Packet_Message_t DeviceMessage;
|
||
|
|
|
||
|
|
if ((ErrorCode = Pipe_Read_Stream_LE(&DeviceMessage, sizeof(RNDIS_Packet_Message_t), NULL)) != PIPE_RWSTREAM_NoError)
|
||
|
|
{
|
||
|
|
return ErrorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
*PacketLength = (uint16_t)DeviceMessage.DataLength;
|
||
|
|
|
||
|
|
Pipe_Discard_Stream(DeviceMessage.DataOffset - (sizeof(RNDIS_Packet_Message_t) - sizeof(RNDIS_Message_Header_t)),
|
||
|
|
NULL);
|
||
|
|
|
||
|
|
Pipe_Freeze();
|
||
|
|
|
||
|
|
return PIPE_RWSTREAM_NoError;
|
||
|
|
}
|
||
|
|
|