Reorder source code

Signed-off-by: Simon Rozman <simon@rozman.si>

1 files changed, 210 insertions, 212 deletions

diff --git a/wintun.c b/wintun.c
index 79dbdb0..828f4da 100644
--- a/wintun.c
+++ b/wintun.c
@@ -35,6 +35,18 @@
 #define TUN_EXCH_MIN_BUFFER_SIZE_WRITE  (sizeof(TUN_PACKET))                                // Minimum size of write exchange buffer
 #define TUN_QUEUE_MAX_NBLS              1000
 #define TUN_MEMORY_TAG                  'wtun'
+#define TUN_CSQ_INSERT_HEAD             ((PVOID)TRUE)
+#define TUN_CSQ_INSERT_TAIL             ((PVOID)FALSE)
+
+#if REG_DWORD == REG_DWORD_BIG_ENDIAN
+#define TUN_HTONS(x)    ((USHORT)(x))
+#define TUN_HTONL(x)    ((ULONG)(x))
+#elif REG_DWORD == REG_DWORD_LITTLE_ENDIAN
+#define TUN_HTONS(x)    (((USHORT)(x) & 0x00ff) << 8 | ((USHORT)(x) & 0xff00) >> 8)
+#define TUN_HTONL(x)    (((ULONG)(x) & 0x000000ff) << 24 | ((ULONG)(x) & 0x0000ff00) << 8 | ((ULONG)(x) & 0x00ff0000) >> 8 | ((ULONG)(x) & 0xff000000) >> 24)
+#else
+#error "Unable to determine endianess"
+#endif
 
 typedef struct _TUN_PACKET {
 	ULONG Size;                     // Size of packet data (TUN_EXCH_MAX_IP_PACKET_SIZE max)
@@ -104,19 +116,6 @@ static PVOID TunNotifyInterfaceChangeHandle;
 static NDIS_HANDLE NdisMiniportDriverHandle;
 static volatile LONG64 AdapterCount;
 
-#if REG_DWORD == REG_DWORD_BIG_ENDIAN
-#define TUN_HTONS(x)    ((USHORT)(x))
-#define TUN_HTONL(x)    ((ULONG)(x))
-#elif REG_DWORD == REG_DWORD_LITTLE_ENDIAN
-#define TUN_HTONS(x)    (((USHORT)(x) & 0x00ff) << 8 | ((USHORT)(x) & 0xff00) >> 8)
-#define TUN_HTONL(x)    (((ULONG)(x) & 0x000000ff) << 24 | ((ULONG)(x) & 0x0000ff00) << 8 | ((ULONG)(x) & 0x00ff0000) >> 8 | ((ULONG)(x) & 0xff000000) >> 24)
-#else
-#error "Unable to determine endianess"
-#endif
-
-#define TUN_CSQ_INSERT_HEAD             ((PVOID)TRUE)
-#define TUN_CSQ_INSERT_TAIL             ((PVOID)FALSE)
-
 #define InterlockedGet(val)             (InterlockedAdd((val), 0))
 #define InterlockedGet64(val)           (InterlockedAdd64((val), 0))
 #define InterlockedGetPointer(val)      (InterlockedCompareExchangePointer((val), NULL, NULL))
@@ -259,13 +258,6 @@ static VOID TunCsqCompleteCanceledIrp(IO_CSQ *Csq, IRP *Irp)
 	TunCompleteRequest(ctx, Irp, STATUS_CANCELLED, IO_NO_INCREMENT);
 }
 
-_IRQL_requires_same_
-static void TunSetNBLStatus(_Inout_opt_ NET_BUFFER_LIST *nbl, _In_ NDIS_STATUS status)
-{
-	for (; nbl; nbl = NET_BUFFER_LIST_NEXT_NBL(nbl))
-		NET_BUFFER_LIST_STATUS(nbl) = status;
-}
-
 _IRQL_requires_max_(DISPATCH_LEVEL)
 _Must_inspect_result_
 static NTSTATUS TunGetIrpBuffer(_In_ IRP *Irp, _Out_ UCHAR **buffer, _Out_ ULONG *size)
@@ -588,6 +580,63 @@ static void TunQueueProcess(_Inout_ TUN_CTX *ctx)
 	}
 }
 
+_IRQL_requires_same_
+static void TunSetNBLStatus(_Inout_opt_ NET_BUFFER_LIST *nbl, _In_ NDIS_STATUS status)
+{
+	for (; nbl; nbl = NET_BUFFER_LIST_NEXT_NBL(nbl))
+		NET_BUFFER_LIST_STATUS(nbl) = status;
+}
+
+static MINIPORT_SEND_NET_BUFFER_LISTS TunSendNetBufferLists;
+_Use_decl_annotations_
+static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER_LIST *NetBufferLists, NDIS_PORT_NUMBER PortNumber, ULONG SendFlags)
+{
+	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
+
+	KIRQL irql = ExAcquireSpinLockShared(&ctx->TransitionLock);
+
+	NDIS_STATUS status;
+	if (!NT_SUCCESS(status = TunCheckForPause(ctx))) {
+		TunSetNBLStatus(NetBufferLists, status);
+		NdisMSendNetBufferListsComplete(ctx->MiniportAdapterHandle, NetBufferLists, NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL);
+		goto cleanup_TunCompletePause;
+	}
+
+	TunQueueAppend(ctx, NetBufferLists, TUN_QUEUE_MAX_NBLS);
+
+	TunQueueProcess(ctx);
+
+cleanup_TunCompletePause:
+	TunCompletePause(ctx, TRUE);
+	ExReleaseSpinLockShared(&ctx->TransitionLock, irql);
+}
+
+static MINIPORT_CANCEL_SEND TunCancelSend;
+_Use_decl_annotations_
+static void TunCancelSend(NDIS_HANDLE MiniportAdapterContext, PVOID CancelId)
+{
+	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
+	KLOCK_QUEUE_HANDLE lqh;
+
+	KeAcquireInStackQueuedSpinLock(&ctx->PacketQueue.Lock, &lqh);
+
+	NET_BUFFER_LIST *nbl_last = NULL, **nbl_last_link = &ctx->PacketQueue.FirstNbl;
+	for (NET_BUFFER_LIST *nbl = ctx->PacketQueue.FirstNbl, *nbl_next; nbl; nbl = nbl_next) {
+		nbl_next = NET_BUFFER_LIST_NEXT_NBL(nbl);
+		if (NDIS_GET_NET_BUFFER_LIST_CANCEL_ID(nbl) == CancelId) {
+			NET_BUFFER_LIST_STATUS(nbl) = NDIS_STATUS_SEND_ABORTED;
+			*nbl_last_link = nbl_next;
+			TunNBLRefDec(ctx, nbl, NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL);
+		} else {
+			nbl_last = nbl;
+			nbl_last_link = &NET_BUFFER_LIST_NEXT_NBL(nbl);
+		}
+	}
+	ctx->PacketQueue.LastNbl = nbl_last;
+
+	KeReleaseInStackQueuedSpinLock(&lqh);
+}
+
 #define IRP_REFCOUNT(irp)        ((volatile LONG *)&(irp)->Tail.Overlay.DriverContext[0])
 #define NET_BUFFER_LIST_IRP(nbl) (NET_BUFFER_LIST_MINIPORT_RESERVED(nbl)[0])
 
@@ -716,46 +765,40 @@ cleanup_TunCompletePause:
 	return status;
 }
 
-_IRQL_requires_max_(PASSIVE_LEVEL)
-static void TunForceHandlesClosed(_Inout_ TUN_CTX *ctx)
+static MINIPORT_RETURN_NET_BUFFER_LISTS TunReturnNetBufferLists;
+_Use_decl_annotations_
+static void TunReturnNetBufferLists(NDIS_HANDLE MiniportAdapterContext, PNET_BUFFER_LIST NetBufferLists, ULONG ReturnFlags)
 {
-	NTSTATUS status;
-	PEPROCESS process;
-	KAPC_STATE apc_state;
-	PVOID object;
-	OBJECT_HANDLE_INFORMATION handle_info;
-	SYSTEM_HANDLE_INFORMATION_EX *table = NULL;
+	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
 
-	for (ULONG size = 0, req; (status = ZwQuerySystemInformation(SystemExtendedHandleInformation, table, size, &req)) == STATUS_INFO_LENGTH_MISMATCH; size = req) {
-		if (table)
-			ExFreePoolWithTag(table, TUN_HTONL(TUN_MEMORY_TAG));
-		table = ExAllocatePoolWithTag(PagedPool, req, TUN_HTONL(TUN_MEMORY_TAG));
-		if (!table)
-			return;
-	}
-	if (!NT_SUCCESS(status) || !table)
-		goto out;
+	LONG64 stat_size = 0, stat_p_ok = 0, stat_p_err = 0;
+	for (NET_BUFFER_LIST *nbl = NetBufferLists, *nbl_next; nbl; nbl = nbl_next) {
+		nbl_next = NET_BUFFER_LIST_NEXT_NBL(nbl);
+		NET_BUFFER_LIST_NEXT_NBL(nbl) = NULL;
 
-	for (ULONG_PTR i = 0; i < table->NumberOfHandles; ++i) {
-		FILE_OBJECT *file = table->Handles[i].Object; //XXX: We should probably first look at table->Handles[i].ObjectTypeIndex, but the value changes lots between NT versions.
-		if (!file || file->Type != 5 || file->DeviceObject != ctx->Device.Object)
-			continue;
-		status = PsLookupProcessByProcessId(table->Handles[i].UniqueProcessId, &process);
-		if (!NT_SUCCESS(status))
-			continue;
-		KeStackAttachProcess(process, &apc_state);
-		status = ObReferenceObjectByHandle(table->Handles[i].HandleValue, 0, NULL, UserMode, &object, &handle_info);
-		if (NT_SUCCESS(status)) {
-			if (object == file)
-				ObCloseHandle(table->Handles[i].HandleValue, UserMode);
-			ObfDereferenceObject(object);
+		IRP *irp = NET_BUFFER_LIST_IRP(nbl);
+		MDL *mdl = NET_BUFFER_FIRST_MDL(NET_BUFFER_LIST_FIRST_NB(nbl));
+		if (NT_SUCCESS(NET_BUFFER_LIST_STATUS(nbl))) {
+			ULONG p_size = MmGetMdlByteCount(mdl);
+			stat_size += p_size;
+			stat_p_ok++;
+		} else
+			stat_p_err++;
+
+		NdisFreeMdl(mdl);
+		NdisFreeNetBufferList(nbl);
+
+		ASSERT(InterlockedGet(IRP_REFCOUNT(irp)) > 0);
+		if (InterlockedDecrement(IRP_REFCOUNT(irp)) <= 0) {
+			TunCompleteRequest(ctx, irp, STATUS_SUCCESS, IO_NETWORK_INCREMENT);
+			TunCompletePause(ctx, TRUE);
 		}
-		KeUnstackDetachProcess(&apc_state);
-		ObfDereferenceObject(process);
 	}
-out:
-	if (table)
-		ExFreePoolWithTag(table, TUN_HTONL(TUN_MEMORY_TAG));
+
+	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInOctets, stat_size);
+	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInUcastOctets, stat_size);
+	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInUcastPkts, stat_p_ok);
+	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifInErrors, stat_p_err);
 }
 
 static DRIVER_DISPATCH TunDispatch;
@@ -845,20 +888,6 @@ cleanup_complete_req:
 	return status;
 }
 
-static MINIPORT_PAUSE TunPause;
-_Use_decl_annotations_
-static NDIS_STATUS TunPause(NDIS_HANDLE MiniportAdapterContext, PNDIS_MINIPORT_PAUSE_PARAMETERS MiniportPauseParameters)
-{
-	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
-
-	KIRQL irql = ExAcquireSpinLockExclusive(&ctx->TransitionLock);
-	InterlockedExchange((LONG *)&ctx->State, TUN_STATE_PAUSING);
-	ExReleaseSpinLockExclusive(&ctx->TransitionLock, irql);
-	TunQueueClear(ctx, STATUS_NDIS_PAUSED);
-
-	return TunCompletePause(ctx, FALSE);
-}
-
 static MINIPORT_RESTART TunRestart;
 _Use_decl_annotations_
 static NDIS_STATUS TunRestart(NDIS_HANDLE MiniportAdapterContext, PNDIS_MINIPORT_RESTART_PARAMETERS MiniportRestartParameters)
@@ -872,66 +901,18 @@ static NDIS_STATUS TunRestart(NDIS_HANDLE MiniportAdapterContext, PNDIS_MINIPORT
 	return NDIS_STATUS_SUCCESS;
 }
 
-static MINIPORT_RETURN_NET_BUFFER_LISTS TunReturnNetBufferLists;
-_Use_decl_annotations_
-static void TunReturnNetBufferLists(NDIS_HANDLE MiniportAdapterContext, PNET_BUFFER_LIST NetBufferLists, ULONG ReturnFlags)
-{
-	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
-
-	LONG64 stat_size = 0, stat_p_ok = 0, stat_p_err = 0;
-	for (NET_BUFFER_LIST *nbl = NetBufferLists, *nbl_next; nbl; nbl = nbl_next) {
-		nbl_next = NET_BUFFER_LIST_NEXT_NBL(nbl);
-		NET_BUFFER_LIST_NEXT_NBL(nbl) = NULL;
-
-		IRP *irp = NET_BUFFER_LIST_IRP(nbl);
-		MDL *mdl = NET_BUFFER_FIRST_MDL(NET_BUFFER_LIST_FIRST_NB(nbl));
-		if (NT_SUCCESS(NET_BUFFER_LIST_STATUS(nbl))) {
-			ULONG p_size = MmGetMdlByteCount(mdl);
-			stat_size += p_size;
-			stat_p_ok++;
-		} else
-			stat_p_err++;
-
-		NdisFreeMdl(mdl);
-		NdisFreeNetBufferList(nbl);
-
-		ASSERT(InterlockedGet(IRP_REFCOUNT(irp)) > 0);
-		if (InterlockedDecrement(IRP_REFCOUNT(irp)) <= 0) {
-			TunCompleteRequest(ctx, irp, STATUS_SUCCESS, IO_NETWORK_INCREMENT);
-			TunCompletePause(ctx, TRUE);
-		}
-	}
-
-	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInOctets, stat_size);
-	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInUcastOctets, stat_size);
-	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifHCInUcastPkts, stat_p_ok);
-	InterlockedAdd64((LONG64 *)&ctx->Statistics.ifInErrors, stat_p_err);
-}
-
-static MINIPORT_CANCEL_SEND TunCancelSend;
+static MINIPORT_PAUSE TunPause;
 _Use_decl_annotations_
-static void TunCancelSend(NDIS_HANDLE MiniportAdapterContext, PVOID CancelId)
+static NDIS_STATUS TunPause(NDIS_HANDLE MiniportAdapterContext, PNDIS_MINIPORT_PAUSE_PARAMETERS MiniportPauseParameters)
 {
 	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
-	KLOCK_QUEUE_HANDLE lqh;
-
-	KeAcquireInStackQueuedSpinLock(&ctx->PacketQueue.Lock, &lqh);
 
-	NET_BUFFER_LIST *nbl_last = NULL, **nbl_last_link = &ctx->PacketQueue.FirstNbl;
-	for (NET_BUFFER_LIST *nbl = ctx->PacketQueue.FirstNbl, *nbl_next; nbl; nbl = nbl_next) {
-		nbl_next = NET_BUFFER_LIST_NEXT_NBL(nbl);
-		if (NDIS_GET_NET_BUFFER_LIST_CANCEL_ID(nbl) == CancelId) {
-			NET_BUFFER_LIST_STATUS(nbl) = NDIS_STATUS_SEND_ABORTED;
-			*nbl_last_link = nbl_next;
-			TunNBLRefDec(ctx, nbl, NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL);
-		} else {
-			nbl_last = nbl;
-			nbl_last_link = &NET_BUFFER_LIST_NEXT_NBL(nbl);
-		}
-	}
-	ctx->PacketQueue.LastNbl = nbl_last;
+	KIRQL irql = ExAcquireSpinLockExclusive(&ctx->TransitionLock);
+	InterlockedExchange((LONG *)&ctx->State, TUN_STATE_PAUSING);
+	ExReleaseSpinLockExclusive(&ctx->TransitionLock, irql);
+	TunQueueClear(ctx, STATUS_NDIS_PAUSED);
 
-	KeReleaseInStackQueuedSpinLock(&lqh);
+	return TunCompletePause(ctx, FALSE);
 }
 
 static MINIPORT_DEVICE_PNP_EVENT_NOTIFY TunDevicePnPEventNotify;
@@ -940,30 +921,6 @@ static void TunDevicePnPEventNotify(NDIS_HANDLE MiniportAdapterContext, PNET_DEV
 {
 }
 
-static MINIPORT_SHUTDOWN TunShutdownEx;
-_Use_decl_annotations_
-static void TunShutdownEx(NDIS_HANDLE MiniportAdapterContext, NDIS_SHUTDOWN_ACTION ShutdownAction)
-{
-	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
-
-	if (ShutdownAction == NdisShutdownBugCheck)
-		return;
-
-	InterlockedExchange((LONG *)&ctx->State, TUN_STATE_SHUTDOWN);
-}
-
-static MINIPORT_CANCEL_DIRECT_OID_REQUEST TunCancelDirectOidRequest;
-_Use_decl_annotations_
-static void TunCancelDirectOidRequest(NDIS_HANDLE MiniportAdapterContext, PVOID RequestId)
-{
-}
-
-static MINIPORT_CANCEL_OID_REQUEST TunCancelOidRequest;
-_Use_decl_annotations_
-static void TunCancelOidRequest(NDIS_HANDLE MiniportAdapterContext, PVOID RequestId)
-{
-}
-
 static DRIVER_NOTIFICATION_CALLBACK_ROUTINE TunPnPNotifyDeviceChange;
 _Use_decl_annotations_
 static NTSTATUS TunPnPNotifyDeviceChange(PVOID NotificationStruct, PVOID Context)
@@ -1283,12 +1240,46 @@ cleanup_NdisDeregisterDeviceEx:
 	return status;
 }
 
-static MINIPORT_UNLOAD TunUnload;
-_Use_decl_annotations_
-static VOID TunUnload(PDRIVER_OBJECT DriverObject)
+_IRQL_requires_max_(PASSIVE_LEVEL)
+static void TunForceHandlesClosed(_Inout_ TUN_CTX *ctx)
 {
-	IoUnregisterPlugPlayNotificationEx(TunNotifyInterfaceChangeHandle);
-	NdisMDeregisterMiniportDriver(NdisMiniportDriverHandle);
+	NTSTATUS status;
+	PEPROCESS process;
+	KAPC_STATE apc_state;
+	PVOID object;
+	OBJECT_HANDLE_INFORMATION handle_info;
+	SYSTEM_HANDLE_INFORMATION_EX *table = NULL;
+
+	for (ULONG size = 0, req; (status = ZwQuerySystemInformation(SystemExtendedHandleInformation, table, size, &req)) == STATUS_INFO_LENGTH_MISMATCH; size = req) {
+		if (table)
+			ExFreePoolWithTag(table, TUN_HTONL(TUN_MEMORY_TAG));
+		table = ExAllocatePoolWithTag(PagedPool, req, TUN_HTONL(TUN_MEMORY_TAG));
+		if (!table)
+			return;
+	}
+	if (!NT_SUCCESS(status) || !table)
+		goto out;
+
+	for (ULONG_PTR i = 0; i < table->NumberOfHandles; ++i) {
+		FILE_OBJECT *file = table->Handles[i].Object; //XXX: We should probably first look at table->Handles[i].ObjectTypeIndex, but the value changes lots between NT versions.
+		if (!file || file->Type != 5 || file->DeviceObject != ctx->Device.Object)
+			continue;
+		status = PsLookupProcessByProcessId(table->Handles[i].UniqueProcessId, &process);
+		if (!NT_SUCCESS(status))
+			continue;
+		KeStackAttachProcess(process, &apc_state);
+		status = ObReferenceObjectByHandle(table->Handles[i].HandleValue, 0, NULL, UserMode, &object, &handle_info);
+		if (NT_SUCCESS(status)) {
+			if (object == file)
+				ObCloseHandle(table->Handles[i].HandleValue, UserMode);
+			ObfDereferenceObject(object);
+		}
+		KeUnstackDetachProcess(&apc_state);
+		ObfDereferenceObject(process);
+	}
+out:
+	if (table)
+		ExFreePoolWithTag(table, TUN_HTONL(TUN_MEMORY_TAG));
 }
 
 _IRQL_requires_max_(APC_LEVEL)
@@ -1358,49 +1349,16 @@ static void TunHaltEx(NDIS_HANDLE MiniportAdapterContext, NDIS_HALT_ACTION HaltA
 	NdisDeregisterDeviceEx(ctx->Device.Handle);
 }
 
-_IRQL_requires_max_(PASSIVE_LEVEL)
-static NDIS_STATUS TunOidSet(_Inout_ TUN_CTX *ctx, _Inout_ NDIS_OID_REQUEST *OidRequest)
+static MINIPORT_SHUTDOWN TunShutdownEx;
+_Use_decl_annotations_
+static void TunShutdownEx(NDIS_HANDLE MiniportAdapterContext, NDIS_SHUTDOWN_ACTION ShutdownAction)
 {
-	ASSERT(OidRequest->RequestType == NdisRequestSetInformation);
-
-	OidRequest->DATA.SET_INFORMATION.BytesNeeded =
-	OidRequest->DATA.SET_INFORMATION.BytesRead   = 0;
-
-	switch (OidRequest->DATA.SET_INFORMATION.Oid) {
-	case OID_GEN_CURRENT_PACKET_FILTER:
-	case OID_GEN_CURRENT_LOOKAHEAD:
-		if (OidRequest->DATA.SET_INFORMATION.InformationBufferLength != 4) {
-			OidRequest->DATA.SET_INFORMATION.BytesNeeded = 4;
-			return NDIS_STATUS_INVALID_LENGTH;
-		}
-		OidRequest->DATA.SET_INFORMATION.BytesRead = 4;
-		return NDIS_STATUS_SUCCESS;
-
-	case OID_GEN_LINK_PARAMETERS:
-		OidRequest->DATA.SET_INFORMATION.BytesRead = OidRequest->DATA.SET_INFORMATION.InformationBufferLength;
-		return NDIS_STATUS_SUCCESS;
-
-	case OID_GEN_INTERRUPT_MODERATION:
-		return NDIS_STATUS_INVALID_DATA;
-
-	case OID_PNP_SET_POWER:
-		if (OidRequest->DATA.SET_INFORMATION.InformationBufferLength != sizeof(NDIS_DEVICE_POWER_STATE)) {
-			OidRequest->DATA.SET_INFORMATION.BytesNeeded = sizeof(NDIS_DEVICE_POWER_STATE);
-			return NDIS_STATUS_INVALID_LENGTH;
-		}
-		OidRequest->DATA.SET_INFORMATION.BytesRead = sizeof(NDIS_DEVICE_POWER_STATE);
-
-		NDIS_DEVICE_POWER_STATE state = *(NDIS_DEVICE_POWER_STATE *)OidRequest->DATA.SET_INFORMATION.InformationBuffer;
-		KIRQL irql = ExAcquireSpinLockExclusive(&ctx->TransitionLock);
-		InterlockedExchange((LONG *)&ctx->PowerState, state);
-		ExReleaseSpinLockExclusive(&ctx->TransitionLock, irql);
-		if (state >= NdisDeviceStateD1)
-			TunQueueClear(ctx, STATUS_NDIS_LOW_POWER_STATE);
+	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
 
-		return NDIS_STATUS_SUCCESS;
-	}
+	if (ShutdownAction == NdisShutdownBugCheck)
+		return;
 
-	return NDIS_STATUS_NOT_SUPPORTED;
+	InterlockedExchange((LONG *)&ctx->State, TUN_STATE_SHUTDOWN);
 }
 
 _IRQL_requires_max_(APC_LEVEL)
@@ -1522,6 +1480,50 @@ static NDIS_STATUS TunOidQuery(_Inout_ TUN_CTX *ctx, _Inout_ NDIS_OID_REQUEST *O
 	return NDIS_STATUS_NOT_SUPPORTED;
 }
 
+_IRQL_requires_max_(PASSIVE_LEVEL)
+static NDIS_STATUS TunOidSet(_Inout_ TUN_CTX *ctx, _Inout_ NDIS_OID_REQUEST *OidRequest)
+{
+	ASSERT(OidRequest->RequestType == NdisRequestSetInformation);
+
+	OidRequest->DATA.SET_INFORMATION.BytesNeeded = OidRequest->DATA.SET_INFORMATION.BytesRead = 0;
+
+	switch (OidRequest->DATA.SET_INFORMATION.Oid) {
+	case OID_GEN_CURRENT_PACKET_FILTER:
+	case OID_GEN_CURRENT_LOOKAHEAD:
+		if (OidRequest->DATA.SET_INFORMATION.InformationBufferLength != 4) {
+			OidRequest->DATA.SET_INFORMATION.BytesNeeded = 4;
+			return NDIS_STATUS_INVALID_LENGTH;
+		}
+		OidRequest->DATA.SET_INFORMATION.BytesRead = 4;
+		return NDIS_STATUS_SUCCESS;
+
+	case OID_GEN_LINK_PARAMETERS:
+		OidRequest->DATA.SET_INFORMATION.BytesRead = OidRequest->DATA.SET_INFORMATION.InformationBufferLength;
+		return NDIS_STATUS_SUCCESS;
+
+	case OID_GEN_INTERRUPT_MODERATION:
+		return NDIS_STATUS_INVALID_DATA;
+
+	case OID_PNP_SET_POWER:
+		if (OidRequest->DATA.SET_INFORMATION.InformationBufferLength != sizeof(NDIS_DEVICE_POWER_STATE)) {
+			OidRequest->DATA.SET_INFORMATION.BytesNeeded = sizeof(NDIS_DEVICE_POWER_STATE);
+			return NDIS_STATUS_INVALID_LENGTH;
+		}
+		OidRequest->DATA.SET_INFORMATION.BytesRead = sizeof(NDIS_DEVICE_POWER_STATE);
+
+		NDIS_DEVICE_POWER_STATE state = *(NDIS_DEVICE_POWER_STATE *)OidRequest->DATA.SET_INFORMATION.InformationBuffer;
+		KIRQL irql = ExAcquireSpinLockExclusive(&ctx->TransitionLock);
+		InterlockedExchange((LONG *)&ctx->PowerState, state);
+		ExReleaseSpinLockExclusive(&ctx->TransitionLock, irql);
+		if (state >= NdisDeviceStateD1)
+			TunQueueClear(ctx, STATUS_NDIS_LOW_POWER_STATE);
+
+		return NDIS_STATUS_SUCCESS;
+	}
+
+	return NDIS_STATUS_NOT_SUPPORTED;
+}
+
 static MINIPORT_OID_REQUEST TunOidRequest;
 _Use_decl_annotations_
 static NDIS_STATUS TunOidRequest(NDIS_HANDLE MiniportAdapterContext, PNDIS_OID_REQUEST OidRequest)
@@ -1539,6 +1541,12 @@ static NDIS_STATUS TunOidRequest(NDIS_HANDLE MiniportAdapterContext, PNDIS_OID_R
 	}
 }
 
+static MINIPORT_CANCEL_OID_REQUEST TunCancelOidRequest;
+_Use_decl_annotations_
+static void TunCancelOidRequest(NDIS_HANDLE MiniportAdapterContext, PVOID RequestId)
+{
+}
+
 static MINIPORT_DIRECT_OID_REQUEST TunDirectOidRequest;
 _Use_decl_annotations_
 static NDIS_STATUS TunDirectOidRequest(NDIS_HANDLE MiniportAdapterContext, PNDIS_OID_REQUEST OidRequest)
@@ -1554,28 +1562,18 @@ static NDIS_STATUS TunDirectOidRequest(NDIS_HANDLE MiniportAdapterContext, PNDIS
 	}
 }
 
-static MINIPORT_SEND_NET_BUFFER_LISTS TunSendNetBufferLists;
+static MINIPORT_CANCEL_DIRECT_OID_REQUEST TunCancelDirectOidRequest;
 _Use_decl_annotations_
-static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER_LIST *NetBufferLists, NDIS_PORT_NUMBER PortNumber, ULONG SendFlags)
+static void TunCancelDirectOidRequest(NDIS_HANDLE MiniportAdapterContext, PVOID RequestId)
 {
-	TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
-
-	KIRQL irql = ExAcquireSpinLockShared(&ctx->TransitionLock);
-
-	NDIS_STATUS status;
-	if (!NT_SUCCESS(status = TunCheckForPause(ctx))) {
-		TunSetNBLStatus(NetBufferLists, status);
-		NdisMSendNetBufferListsComplete(ctx->MiniportAdapterHandle, NetBufferLists, NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL);
-		goto cleanup_TunCompletePause;
-	}
-
-	TunQueueAppend(ctx, NetBufferLists, TUN_QUEUE_MAX_NBLS);
-
-	TunQueueProcess(ctx);
+}
 
-cleanup_TunCompletePause:
-	TunCompletePause(ctx, TRUE);
-	ExReleaseSpinLockShared(&ctx->TransitionLock, irql);
+static MINIPORT_UNLOAD TunUnload;
+_Use_decl_annotations_
+static VOID TunUnload(PDRIVER_OBJECT DriverObject)
+{
+	IoUnregisterPlugPlayNotificationEx(TunNotifyInterfaceChangeHandle);
+	NdisMDeregisterMiniportDriver(NdisMiniportDriverHandle);
 }
 
 DRIVER_INITIALIZE DriverEntry;