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
|
/*++
Module Name:
shared.h
Abstract:
This contains declarations shared by the Ob/Ps callback test driver and
the user mode test app.
// Notice:
//
// Use this sample code at your own risk; there is no support from Microsoft for the sample code.
// In addition, this sample code is licensed to you under the terms of the Microsoft Public License
// (http://www.microsoft.com/opensource/licenses.mspx)
--*/
#pragma once
#pragma warning(disable:4214) // bit field types other than int
#pragma warning(disable:4201) // nameless struct/union
//
// TD_ASSERT
//
// This macro is identical to NT_ASSERT but works in fre builds as well.
//
// It is used for error checking in the driver in cases where
// we can't easily report the error to the user mode app, or the
// error is so severe that we should break in immediately to
// investigate.
//
// It's better than DbgBreakPoint because it provides additional info
// that can be dumped with .exr -1, and individual asserts can be disabled
// from kd using 'ahi' command.
//
#define TD_ASSERT(_exp) \
((!(_exp)) ? \
(__annotation(L"Debug", L"AssertFail", L#_exp), \
DbgRaiseAssertionFailure(), FALSE) : \
TRUE)
//
// Driver and device names
// It is important to change the names of the binaries
// in the sample code to be unique for your own use.
//
#define TD_DRIVER_NAME L"ObCallbackTest"
#define TD_DRIVER_NAME_WITH_EXT L"ObCallbackTest.sys"
#define TD_NT_DEVICE_NAME L"\\Device\\ObCallbackTest"
#define TD_DOS_DEVICES_LINK_NAME L"\\DosDevices\\ObCallbackTest"
#define TD_WIN32_DEVICE_NAME L"\\\\.\\ObCallbackTest"
#define NAME_SIZE 200
#define TD_INVALID_CALLBACK_ID ((ULONG)-1)
//
// IOCTLs exposed by the driver.
//
// #define TD_IOCTL_REGISTER_CALLBACK CTL_CODE (FILE_DEVICE_UNKNOWN, (0x800 + 0), METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
// #define TD_IOCTL_UNREGISTER_CALLBACK CTL_CODE (FILE_DEVICE_UNKNOWN, (0x800 + 1), METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
#define TD_IOCTL_PROTECT_NAME_CALLBACK CTL_CODE (FILE_DEVICE_UNKNOWN, (0x800 + 2), METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
#define TD_IOCTL_UNPROTECT_CALLBACK CTL_CODE (FILE_DEVICE_UNKNOWN, (0x800 + 3), METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
#define TDProtectName_Protect 0 // name of programs to proect and filter out the desiredAccess on Process Open
#define TDProtectName_Reject 1 // name of programs to reject during ProcessCreate
//
// Structures used by TD_IOCTL_PROTECTNAME
//
typedef struct _TD_PROTECTNAME_INPUT {
ULONG Operation;
WCHAR Name[NAME_SIZE+1]; // what is the filename to protect - extra wchar for forced NULL
}
TD_PROTECTNAME_INPUT, *PTD_PROTECTNAME_INPUT;
//
// Structures used by TD_IOCTL_UNPROTECT_CALLBACK
//
typedef struct _TD_UNPROTECT_CALLBACK_INPUT {
ULONG UnusedParameter;
}
TD_UNPROTECT_CALLBACK_INPUT, *PTD_UNPROTECT_CALLBACK_INPUT;
|