Merge pull request #1402 from microsoft/main

FI: Main -> Develop

1 files changed, 73 insertions, 0 deletions

diff --git a/network/netadaptercx/netvadapter/drivercode/power.cpp b/network/netadaptercx/netvadapter/drivercode/power.cpp
new file mode 100644
index 00000000..68b057ff
--- /dev/null
+++ b/network/netadaptercx/netvadapter/drivercode/power.cpp
@@ -0,0 +1,73 @@
+// Copyright (c) Microsoft Corporation. All rights reserved
+
+#include "pch.hpp"
+#include "netvadapter.h"
+#include "device.h"
+#include "trace.h"
+#include "power.h"
+#include "power.tmh"
+
+void
+NetvDeviceInitializePowerManagement(
+    _In_ NetvDevice * Device
+)
+{
+#ifdef _KERNEL_MODE
+    WDFDEVICE wdfDevice = Device->m_handle;
+    NetvAdapter * adapter = Device->m_adapter;
+
+    if (!adapter->S0Idle)
+    {
+        return;
+    }
+
+    WDF_DEVICE_POWER_POLICY_IDLE_SETTINGS idleSettings;
+    WDF_DEVICE_POWER_POLICY_IDLE_SETTINGS_INIT(
+        &idleSettings,
+        IdleCanWakeFromS0);
+
+    idleSettings.IdleTimeout = 3000;
+    idleSettings.IdleTimeoutType = SystemManagedIdleTimeoutWithHint;
+
+    NTSTATUS ntStatus = WdfDeviceAssignS0IdleSettings(
+        wdfDevice,
+        &idleSettings);
+
+    if (ntStatus != STATUS_SUCCESS)
+    {
+        // MSDN says we should ignore error codes from assign S0 idle settings, do that
+        return;
+    }
+
+    // Now tell NetAdapterCx we support wake on packet filter match
+    NET_ADAPTER_WAKE_PACKET_FILTER_CAPABILITIES wakeOnPacketFilterMatch;
+    NET_ADAPTER_WAKE_PACKET_FILTER_CAPABILITIES_INIT(&wakeOnPacketFilterMatch);
+    wakeOnPacketFilterMatch.PacketFilterMatch = TRUE;
+
+    NetAdapterWakeSetPacketFilterCapabilities(adapter->m_handle, &wakeOnPacketFilterMatch);
+#else
+    UNREFERENCED_PARAMETER(Device);
+#endif
+}
+
+_Use_decl_annotations_
+NTSTATUS
+EvtDeviceArmWakeFromS0(
+    WDFDEVICE Device
+)
+{
+    auto adapter = NetvDeviceGetContext(Device)->m_adapter;
+    adapter->ArmWakeFromS0();
+
+    return STATUS_SUCCESS;
+}
+
+_Use_decl_annotations_
+void
+EvtDeviceDisarmWakeFromS0(
+    WDFDEVICE Device
+)
+{
+    auto adapter = NetvDeviceGetContext(Device)->m_adapter;
+    adapter->DisarmWakeFromS0();
+}