NTSTATUS
TLInspectLoadConfig(
_In_ const WDFKEY key
)
{
NTSTATUS status;
DECLARE_CONST_UNICODE_STRING(valueName, L"RemoteAddressToInspect");
DECLARE_UNICODE_STRING_SIZE(value, INET6_ADDRSTRLEN);
status = WdfRegistryQueryUnicodeString(key, &valueName, NULL, &value);
if (NT_SUCCESS(status))
{
PWSTR terminator;
// Defensively null-terminate the string
value.Length = min(value.Length, value.MaximumLength - sizeof(WCHAR));
value.Buffer[value.Length/sizeof(WCHAR)] = UNICODE_NULL;
status = RtlIpv4StringToAddressW(
value.Buffer,
TRUE,
&terminator,
&remoteAddrStorageV4
);
if (NT_SUCCESS(status))
{
remoteAddrStorageV4.S_un.S_addr =
RtlUlongByteSwap(remoteAddrStorageV4.S_un.S_addr);
configInspectRemoteAddrV4 = &remoteAddrStorageV4.S_un.S_un_b.s_b1;
}
else
{
status = RtlIpv6StringToAddressW(
value.Buffer,
&terminator,
&remoteAddrStorageV6
);
if (NT_SUCCESS(status))
{
configInspectRemoteAddrV6 = (UINT8*)(&remoteAddrStorageV6.u.Byte[0]);
}
}
}
return status;
}
NTSTATUS
DeviceConfigure(
_In_ PDEVICE_CONTEXT DeviceContext
)
/*++
Routine Description:
This method is called after the device callback object has been initialized
and returned to the driver. It would setup the device's queues and their
corresponding callback objects.
Arguments:
FxDevice - the framework device object for which we're handling events.
Return Value:
status
--*/
{
NTSTATUS status;
WDFDEVICE device = DeviceContext->Device;
WDFKEY key;
LPGUID guid;
errno_t errorNo;
DECLARE_CONST_UNICODE_STRING(portName, REG_VALUENAME_PORTNAME);
DECLARE_UNICODE_STRING_SIZE (comPort, 10);
DECLARE_UNICODE_STRING_SIZE (symbolicLinkName, SYMBOLIC_LINK_NAME_LENGTH);
#ifdef _FAKE_MODEM
//
// Compiled as fake modem
//
guid = (LPGUID) &GUID_DEVINTERFACE_MODEM;
#else
//
// Compiled as virtual serial port
//
guid = (LPGUID) &GUID_DEVINTERFACE_COMPORT;
#endif
//
// Create device interface
//
status = WdfDeviceCreateDeviceInterface(
device,
guid,
NULL);
if (!NT_SUCCESS(status)) {
Trace(TRACE_LEVEL_ERROR,
"Error: Cannot create device interface");
goto Exit;
}
//
// Read the COM port number from the registry, which has been automatically
// created by "MsPorts!PortsClassInstaller" if INF file says "Class=Ports"
//
status = WdfDeviceOpenRegistryKey(
device,
PLUGPLAY_REGKEY_DEVICE,
KEY_QUERY_VALUE,
WDF_NO_OBJECT_ATTRIBUTES,
&key);
if (!NT_SUCCESS(status)) {
Trace(TRACE_LEVEL_ERROR,
"Error: Failed to retrieve device hardware key root");
goto Exit;
}
status = WdfRegistryQueryUnicodeString(
key,
&portName,
NULL,
&comPort);
if (!NT_SUCCESS(status)) {
Trace(TRACE_LEVEL_ERROR,
"Error: Failed to read PortName");
goto Exit;
}
//
// Manually create the symbolic link name. Length is the length in
// bytes not including the NULL terminator.
//
// 6054 and 26035 are code analysis warnings that comPort.Buffer might
// not be NULL terminated, while we know that they are.
//
#pragma warning(suppress: 6054 26035)
symbolicLinkName.Length = (USHORT)((wcslen(comPort.Buffer) * sizeof(wchar_t))
+ sizeof(SYMBOLIC_LINK_NAME_PREFIX) - sizeof(UNICODE_NULL));
if (symbolicLinkName.Length >= symbolicLinkName.MaximumLength) {
Trace(TRACE_LEVEL_ERROR, "Error: Buffer overflow when creating COM port name. Size"
" is %d, buffer length is %d", symbolicLinkName.Length, symbolicLinkName.MaximumLength);
status = STATUS_BUFFER_OVERFLOW;
goto Exit;
}
errorNo = wcscpy_s(symbolicLinkName.Buffer,
SYMBOLIC_LINK_NAME_LENGTH,
SYMBOLIC_LINK_NAME_PREFIX);
if (errorNo != 0) {
Trace(TRACE_LEVEL_ERROR,
"Failed to copy %ws to buffer with error %d",
SYMBOLIC_LINK_NAME_PREFIX, errorNo);
status = STATUS_INVALID_PARAMETER;
goto Exit;
}
errorNo = wcscat_s(symbolicLinkName.Buffer,
SYMBOLIC_LINK_NAME_LENGTH,
comPort.Buffer);
if (errorNo != 0) {
Trace(TRACE_LEVEL_ERROR,
"Failed to copy %ws to buffer with error %d",
comPort.Buffer, errorNo);
status = STATUS_INVALID_PARAMETER;
goto Exit;
}
//
// Create symbolic link
//
status = WdfDeviceCreateSymbolicLink(
device,
&symbolicLinkName);
if (!NT_SUCCESS(status)) {
Trace(TRACE_LEVEL_ERROR,
"Error: Cannot create symbolic link %ws", symbolicLinkName.Buffer);
goto Exit;
}
status = DeviceGetPdoName(DeviceContext);
if (!NT_SUCCESS(status)) {
goto Exit;
}
status = DeviceWriteLegacyHardwareKey(
DeviceContext->PdoName,
comPort.Buffer,
DeviceContext->Device);
if (NT_SUCCESS(status)) {
DeviceContext->CreatedLegacyHardwareKey = TRUE;
}
status = QueueCreate(DeviceContext);
if (!NT_SUCCESS(status)) {
goto Exit;
}
Exit:
return status;
}
NTSTATUS
StreamEditLoadConfig(
const WDFKEY key
)
{
NTSTATUS status = STATUS_SUCCESS;
DECLARE_CONST_UNICODE_STRING(stringToFindKey, L"StringToFind");
DECLARE_CONST_UNICODE_STRING(stringToReplaceKey, L"StringToReplace");
DECLARE_CONST_UNICODE_STRING(inspectionPortKey, L"InspectionPort");
DECLARE_CONST_UNICODE_STRING(editInlineKey, L"EditInline");
DECLARE_CONST_UNICODE_STRING(inspectOutboundKey, L"InspectOutbound");
UNICODE_STRING stringValue;
WCHAR buffer[128];
USHORT requiredSize;
ULONG valueSize;
ULONG ulongValue;
stringValue.Buffer = buffer;
stringValue.Length = 0;
stringValue.MaximumLength = sizeof(buffer) - sizeof(buffer[0]);
if (NT_SUCCESS(
WdfRegistryQueryUnicodeString(
key,
&stringToFindKey,
&requiredSize,
&stringValue
)))
{
stringValue.Buffer[stringValue.Length/sizeof(stringValue.Buffer[0])] =
UNICODE_NULL;
status = RtlUnicodeToMultiByteN(
configStringToFind,
sizeof(configStringToFind) - 1,
&valueSize,
stringValue.Buffer,
(ULONG)requiredSize
);
if (!NT_SUCCESS(status))
{
goto Exit;
}
configStringToFind[valueSize] = '\0';
}
if (NT_SUCCESS(
WdfRegistryQueryUnicodeString(
key,
&stringToReplaceKey,
&requiredSize,
&stringValue
)))
{
status = RtlUnicodeToMultiByteN(
configStringToReplace,
sizeof(configStringToReplace) - 1,
&valueSize,
stringValue.Buffer,
(ULONG)requiredSize
);
if (!NT_SUCCESS(status))
{
goto Exit;
}
configStringToReplace[valueSize] = '\0';
}
if (NT_SUCCESS(
WdfRegistryQueryULong(
key,
&inspectionPortKey,
&ulongValue
)))
{
configInspectionPort = (USHORT) ulongValue;
}
if (NT_SUCCESS(
WdfRegistryQueryULong(
key,
&editInlineKey,
&ulongValue
)))
{
configEditInline = (ulongValue != 0);
}
if (NT_SUCCESS(
WdfRegistryQueryULong(
key,
&inspectOutboundKey,
&ulongValue
)))
{
configInspectionOutbound = (ulongValue != 0);
}
Exit:
return status;
}
