Skip to content

Commit

Permalink
feat: Add KMDF Sample Driver with pool leak to demonstrate Driver Ver…
Browse files Browse the repository at this point in the history
…ifier (#27)
  • Loading branch information
svasista-ms authored Sep 27, 2024
1 parent 64c426e commit 8b8847f
Show file tree
Hide file tree
Showing 8 changed files with 448 additions and 1 deletion.
11 changes: 11 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 5 additions & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
[workspace]
members = ["general/echo/kmdf/driver/*", "general/echo/kmdf/exe"]
members = [
"general/echo/kmdf/driver/*",
"general/echo/kmdf/exe",
"tools/dv/kmdf/fail_driver_pool_leak",
]
resolver = "2"

[workspace.package]
Expand Down
25 changes: 25 additions & 0 deletions tools/dv/kmdf/fail_driver_pool_leak/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
[package]
name = "fail_driver_pool_leak"
version = "0.1.0"
edition.workspace = true
publish.workspace = true
repository.workspace = true
license.workspace = true

[package.metadata.wdk]

[lib]
crate-type = ["cdylib"]

[dependencies]
wdk.workspace = true
wdk-alloc.workspace = true
wdk-panic.workspace = true
wdk-sys.workspace = true

[build-dependencies]
wdk-build.workspace = true

[features]
default = []
nightly = ["wdk/nightly", "wdk-sys/nightly"]
94 changes: 94 additions & 0 deletions tools/dv/kmdf/fail_driver_pool_leak/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,94 @@
# Fail_Driver_Pool_Leak Sample

This sample KMDF Fail Driver demonstrates the capabilities and features of **Driver Verifier** and the **Device Fundamentals Tests**.

It allocates a pool of memory to a global buffer when a supported device is added by the PnP Manager and intentionally does not free it before the driver is unloaded. This memory leak fault is a system vulnerability that could lead to security and performance issues and bad user experience.

By enabling Driver Verifier on this driver, this pool leak violation can be caught before the driver is unloaded and with an active KDNET session, the bug can be analyzed further.

NOTE: The driver uses WDM's ExAllocatePool2 API directly to allocate memory for its buffer. Ideally, such allocations should be freed by using ExFreePool API. A cleaner way to manage memory in a WDF Driver is to use [wdfmemory](https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/wdfmemory/)


## Steps to reproduce the issue

1. Clone the repository and navigate to the project root.

2. Build the driver project using the following command in a WDK environment (or EWDK prompt) -
```
cargo make
```
3. Prepare a target system (a Hyper-V VM can be used) for testing
Follow the below steps to setup the test system -
1. Disable Secure boot and start the system
2. Run "ipconfig" on the host system and note down the IP (if you are using Default Switch for the VM, note down the IP on the Default Switch)
3. Install and open WinDbg, click on "Attach to Kernel". The key for the connection will be generated in the test system in the next steps.
4. Connect to the test VM and run the following commands -
```
bcdedit /set testsigning on
bcdedit /debug on
bcdedit /dbgsettings net hostip:<PASTE.HOST.IP.HERE> port:<50000-50030>
### Copy the key string output by the above command
```
5. Paste the key in host's WinDbg prompt and connect to the kernel
6. Restart the target/test system
```
shutdown -r -t 0
```
4. Copy the driver package, available under ".\target\debug\fail_driver_pool_leak_package" to the target system.
5. Copy "devgen.exe" from host to the target system. Alternatively you may install WDK on the target system and add the directory that contains "devgen.exe" to PATH variable.
6. Install the driver package and create the device in the target system using the below commands -
```
cd "fail_driver_pool_leak_package"
devgen.exe /add /bus ROOT /hardwareid "fail_driver_pool_leak"
## Copy the Device ID. This will be used later to run the tests
pnputil.exe /add-driver .\fail_driver_pool_leak.inf /install
```
7. Enable Driver Verifier for 'fail_driver_pool_leak.sys' driver package
1. Open run command prompt (Start + R) or cmd as administator and run "verifier"
2. In the verifier manager,
- Create Standard Settings
- Select driver names from list
- Select 'fail_driver_pool_leak.sys'
- Finish
- Restart the system
8. Follow the steps in https://learn.microsoft.com/en-us/windows-hardware/drivers/develop/how-to-test-a-driver-at-runtime-from-a-command-prompt to run tests against the device managed by this driver
9. Install TAEF and WDTF on the test computer and run the following test -
```
cd "C:\Program Files (x86)\Windows Kits\10\Testing\Tests\Additional Tests\x64\DevFund"
TE.exe .\Devfund_PnPDTest_WLK_Certification.dll /P:"DQ=DeviceID='ROOT\DEVGEN\{PASTE-DEVICE-ID-HERE}'" --rebootResumeOption:Manual
```
10. The test will lead to a Bugcheck and a BlueScreen on the target system with the following error -
```
DRIVER_VERIFIER_DETECTED_VIOLATION (c4)
```
The logs will be available in WinDbg
run ```!analyze -v``` for detailed bugcheck report
run ```!verifier 3 fail_driver_pool_leak.sys``` for info on the allocations that were leaked that caused the bugcheck.
11. (Alternatively), the bugcheck can be observed when all the devices managed by this driver are removed, i.e, when the driver is unloaded from the system.
You may use pnputil/devcon to enumerate and remove the devices -
```
# To enumerate the devices
pnputil /enum-devices
# To remove a device
pnputil /remove-device "DEVICE-ID"
```
### References
- [Driver Verifier](https://learn.microsoft.com/en-us/windows-hardware/drivers/devtest/driver-verifier)
- [Device Fundamentals Tests](https://learn.microsoft.com/en-us/windows-hardware/drivers/devtest/device-fundamentals-tests)
- [TAEF](https://learn.microsoft.com/en-us/windows-hardware/drivers/taef/getting-started)
- [WDTF](https://learn.microsoft.com/en-us/windows-hardware/drivers/wdtf/wdtf-runtime-library)
- [Testing a driver at runtime](https://learn.microsoft.com/en-us/windows-hardware/drivers/develop/how-to-test-a-driver-at-runtime-from-a-command-prompt)
- [Using WDF to Develop a Driver](https://learn.microsoft.com/en-us/windows-hardware/drivers/wdf/using-the-framework-to-develop-a-driver)
7 changes: 7 additions & 0 deletions tools/dv/kmdf/fail_driver_pool_leak/build.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
// Copyright (c) Microsoft Corporation
// License: MIT OR Apache-2.0

fn main() -> Result<(), wdk_build::ConfigError> {
wdk_build::Config::from_env_auto()?.configure_binary_build();
Ok(())
}
56 changes: 56 additions & 0 deletions tools/dv/kmdf/fail_driver_pool_leak/fail_driver_pool_leak.inx
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
;===================================================================
; Copyright (c)2023, Microsoft Corporation
;
;Module Name:
; FAIL_DRIVER_POOL_LEAK.INF
;===================================================================

[Version]
Signature = "$WINDOWS NT$"
Class = SoftwareComponent
ClassGuid = {5c4c3332-344d-483c-8739-259e934c9cc8}
Provider = %ProviderString%
PnpLockDown = 1

[DestinationDirs]
DefaultDestDir = 13

[SourceDisksNames]
1 = %DiskId1%,,,""

[SourceDisksFiles]
fail_driver_pool_leak.sys = 1,,

; ================= Install section =================

[Manufacturer]
%StdMfg%=Standard,NT$ARCH$.10.0...16299

[Standard.NT$ARCH$.10.0...16299]
%FAIL_DRIVER_POOL_LEAK.DeviceDesc%=FAIL_DRIVER_POOL_LEAK_DEVICE, fail_driver_pool_leak

[FAIL_DRIVER_POOL_LEAK_DEVICE.NT$ARCH$]
CopyFiles=Drivers_Dir

[Drivers_Dir]
fail_driver_pool_leak.sys

; ================= Service installation =================
[FAIL_DRIVER_POOL_LEAK_Device.NT$ARCH$.Services]
AddService = fail_driver_pool_leak, %SPSVCINST_ASSOCSERVICE%, fail_driver_pool_leak_svc_ins

[fail_driver_pool_leak_svc_ins]
DisplayName = %FAIL_DRIVER_POOL_LEAK.SVCDESC%
ServiceType = 1 ; SERVICE_KERNEL_DRIVER
StartType = 3 ; SERVICE_DEMAND_START
ErrorControl = 1 ; SERVICE_ERROR_NORMAL
ServiceBinary = %13%\fail_driver_pool_leak.sys

; ================= Strings =================
[Strings]
SPSVCINST_ASSOCSERVICE = 0x00000002
ProviderString = "Rust-DV-Fail-Sample"
StdMfg = "(Standard system devices)"
DiskId1 = "WDF FAIL_DRIVER_POOL_LEAK Installation Disk #1"
FAIL_DRIVER_POOL_LEAK.DeviceDesc = "WDF FAIL_DRIVER_POOL_LEAK Device"
FAIL_DRIVER_POOL_LEAK.SVCDESC = "WDF FAIL_DRIVER_POOL_LEAK Service"
Loading

0 comments on commit 8b8847f

Please sign in to comment.