filipnavara--igb/driver.cpp at ossign · OSSign/filipnavara--igb · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include "precomp.h"

#include "device.h"
#include "trace.h"
#include "adapter.h"
#include "power.h"
#include "interrupt.h"

EXTERN_C __declspec(code_seg("INIT")) DRIVER_INITIALIZE DriverEntry;
EVT_WDF_DRIVER_UNLOAD EvtDriverUnload;
EVT_WDF_DRIVER_DEVICE_ADD EvtDriverDeviceAdd;

_Use_decl_annotations_
__declspec(code_seg("INIT"))
NTSTATUS
DriverEntry(
	_In_ PDRIVER_OBJECT driverObject,
	_In_ PUNICODE_STRING registryPath)
{
	NTSTATUS status = STATUS_SUCCESS;

	DBGPRINT("DriverEntry\n");

	WDF_DRIVER_CONFIG driverConfig;
	WDF_DRIVER_CONFIG_INIT(&driverConfig, EvtDriverDeviceAdd);

	driverConfig.DriverPoolTag = 'I1GB';
	driverConfig.EvtDriverUnload = EvtDriverUnload;

	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		WdfDriverCreate(driverObject, registryPath, WDF_NO_OBJECT_ATTRIBUTES, &driverConfig, NULL));

Exit:

	return status;
}

_Use_decl_annotations_
NTSTATUS
#pragma prefast(suppress: __WARNING_EXCESSIVESTACKUSAGE, "TVS:12813961 call stack depth well-defined")
EvtDriverDeviceAdd(
	_In_ WDFDRIVER driver,
	_Inout_ PWDFDEVICE_INIT deviceInit)
{
	UNREFERENCED_PARAMETER((driver));

	DBGPRINT("EvtDriverDeviceAdd\n");

	NTSTATUS status = STATUS_SUCCESS;
	NETADAPTER_INIT* adapterInit = nullptr;

	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		NetDeviceInitConfig(deviceInit));

	WDF_OBJECT_ATTRIBUTES deviceAttributes;
	WDF_OBJECT_ATTRIBUTES_INIT_CONTEXT_TYPE(&deviceAttributes, IGB_DEVICE);

	{
		WDF_PNPPOWER_EVENT_CALLBACKS pnpPowerCallbacks;
		WDF_PNPPOWER_EVENT_CALLBACKS_INIT(&pnpPowerCallbacks);
		pnpPowerCallbacks.EvtDevicePrepareHardware = EvtDevicePrepareHardware;
		pnpPowerCallbacks.EvtDeviceReleaseHardware = EvtDeviceReleaseHardware;
		pnpPowerCallbacks.EvtDeviceD0Entry = EvtDeviceD0Entry;
		pnpPowerCallbacks.EvtDeviceD0Exit = EvtDeviceD0Exit;

		WdfDeviceInitSetPnpPowerEventCallbacks(deviceInit, &pnpPowerCallbacks);
	}

	// TODO: WdfDeviceInitSetPowerPolicyEventCallbacks

	WDFDEVICE wdfDevice;
	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		WdfDeviceCreate(&deviceInit, &deviceAttributes, &wdfDevice));

	WdfDeviceSetAlignmentRequirement(wdfDevice, FILE_256_BYTE_ALIGNMENT);

	WDF_DEVICE_POWER_POLICY_IDLE_SETTINGS idleSettings;
	WDF_DEVICE_POWER_POLICY_IDLE_SETTINGS_INIT(&idleSettings, IdleCannotWakeFromS0);
	idleSettings.UserControlOfIdleSettings = IdleAllowUserControl;

	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		WdfDeviceAssignS0IdleSettings(wdfDevice, &idleSettings));

	// TODO: WdfDeviceAssignSxWakeSettings

	adapterInit = NetAdapterInitAllocate(wdfDevice);

	GOTO_WITH_INSUFFICIENT_RESOURCES_IF_NULL(Exit, status, adapterInit);

	NET_ADAPTER_DATAPATH_CALLBACKS datapathCallbacks;
	NET_ADAPTER_DATAPATH_CALLBACKS_INIT(
		&datapathCallbacks,
		EvtAdapterCreateTxQueue,
		EvtAdapterCreateRxQueue);

	NetAdapterInitSetDatapathCallbacks(
		adapterInit,
		&datapathCallbacks);

	WDF_OBJECT_ATTRIBUTES adapterAttributes;
	WDF_OBJECT_ATTRIBUTES_INIT_CONTEXT_TYPE(&adapterAttributes, IGB_ADAPTER);

	NETADAPTER netAdapter;
	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		NetAdapterCreate(adapterInit, &adapterAttributes, &netAdapter));

	PIGB_ADAPTER adapter = IgbGetAdapterContext(netAdapter);
	PIGB_DEVICE device = IgbGetDeviceContext(wdfDevice);

	device->Adapter = adapter;

	GOTO_IF_NOT_NT_SUCCESS(Exit, status,
		IgbInitializeAdapterContext(adapter, wdfDevice, netAdapter));

Exit:
	if (adapterInit != nullptr)
	{
		NetAdapterInitFree(adapterInit);
	}

	DBGPRINT("EvtDriverDeviceAdd = %x\n", status);

	return status;
}

_Use_decl_annotations_
VOID
EvtDriverUnload(
	_In_ WDFDRIVER driver)
{
	UNREFERENCED_PARAMETER((driver));
	DBGPRINT("EvtDriverUnload\n");
}