[KS]
[reactos.git] / reactos / drivers / ksfilter / ks / image.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Kernel Streaming
4 * FILE: drivers/ksfilter/ks/allocators.c
5 * PURPOSE: KS Allocator functions
6 * PROGRAMMER: Johannes Anderwald
7 */
8
9 #include "priv.h"
10
11 /*
12 @implemented
13 */
14 KSDDKAPI
15 NTSTATUS
16 NTAPI
17 KsLoadResource(
18 IN PVOID ImageBase,
19 IN POOL_TYPE PoolType,
20 IN ULONG_PTR ResourceName,
21 IN ULONG ResourceType,
22 OUT PVOID* Resource,
23 OUT PULONG ResourceSize)
24 {
25 NTSTATUS Status;
26 LDR_RESOURCE_INFO ResourceInfo;
27 PIMAGE_RESOURCE_DATA_ENTRY ResourceDataEntry;
28 PVOID Data;
29 ULONG Size;
30 PVOID _SEH2_VOLATILE Result = NULL;
31
32 /* set up resource info */
33 ResourceInfo.Type = ResourceType;
34 ResourceInfo.Name = ResourceName;
35 ResourceInfo.Language = 0;
36
37 _SEH2_TRY
38 {
39 /* find the resource */
40 Status = LdrFindResource_U(ImageBase, &ResourceInfo, RESOURCE_DATA_LEVEL, &ResourceDataEntry);
41 if (NT_SUCCESS(Status))
42 {
43 /* try accessing it */
44 Status = LdrAccessResource(ImageBase, ResourceDataEntry, &Data, &Size);
45 if (NT_SUCCESS(Status))
46 {
47 /* allocate resource buffer */
48 Result = AllocateItem(PoolType, Size);
49 if (Result)
50 {
51 /* copy resource */
52 RtlMoveMemory(Result, Data, Size);
53 /* store result */
54 *Resource = Result;
55
56 if (ResourceSize)
57 {
58 /* resource size is optional */
59 *ResourceSize = Size;
60 }
61 }
62 else
63 {
64 /* not enough memory */
65 Status = STATUS_INSUFFICIENT_RESOURCES;
66 }
67 }
68 }
69 }
70 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
71 {
72 /* Exception, get the error code */
73 Status = _SEH2_GetExceptionCode();
74 }
75 _SEH2_END;
76
77 if (!NT_SUCCESS(Status))
78 {
79 /* failed */
80 if (Result)
81 {
82 /* free resource buffer in case of a failure */
83 FreeItem(Result);
84 }
85 }
86 /* done */
87 return Status;
88 }
89
90
91 NTSTATUS
92 KspQueryRegValue(
93 IN HANDLE KeyHandle,
94 IN LPWSTR KeyName,
95 IN PVOID Buffer,
96 IN OUT PULONG BufferLength,
97 OUT PULONG Type)
98 {
99 UNIMPLEMENTED
100 return STATUS_NOT_IMPLEMENTED;
101 }
102
103 /*
104 @implemented
105 */
106 KSDDKAPI
107 NTSTATUS
108 NTAPI
109 KsGetImageNameAndResourceId(
110 IN HANDLE RegKey,
111 OUT PUNICODE_STRING ImageName,
112 OUT PULONG_PTR ResourceId,
113 OUT PULONG ValueType)
114 {
115 NTSTATUS Status;
116 ULONG ImageLength;
117 WCHAR ImagePath[] = {L"\\SystemRoot\\system32\\drivers\\"};
118
119 /* first clear the provided ImageName */
120 ImageName->Buffer = NULL;
121 ImageName->Length = ImageName->MaximumLength = 0;
122
123 ImageLength = 0;
124 /* retrieve length of image name */
125 Status = KspQueryRegValue(RegKey, L"Image", NULL, &ImageLength, NULL);
126
127 if (Status != STATUS_BUFFER_OVERFLOW)
128 {
129 /* key value doesnt exist */
130 return Status;
131 }
132
133 /* allocate image name buffer */
134 ImageName->MaximumLength = (USHORT)(sizeof(ImagePath) + ImageLength);
135 ImageName->Buffer = AllocateItem(PagedPool, ImageName->MaximumLength);
136
137 /* check for success */
138 if (!ImageName->Buffer)
139 {
140 /* insufficient memory */
141 return STATUS_INSUFFICIENT_RESOURCES;
142 }
143
144 /* copy image name */
145 RtlCopyMemory(ImageName->Buffer, ImagePath, sizeof(ImagePath));
146
147 /* retrieve image name */
148 Status = KspQueryRegValue(RegKey,
149 L"Image",
150 &ImageName->Buffer[sizeof(ImagePath) / sizeof(WCHAR)],
151 &ImageLength,
152 NULL);
153
154 if (!NT_SUCCESS(Status))
155 {
156 /* unexpected error */
157 FreeItem(ImageName->Buffer);
158 return Status;
159 }
160
161 /* now query for resource id length*/
162 ImageLength = 0;
163 Status = KspQueryRegValue(RegKey, L"ResourceId", NULL, &ImageLength, ValueType);
164
165 /* allocate resource id buffer*/
166 *ResourceId = (ULONG_PTR)AllocateItem(PagedPool, ImageLength);
167
168 /* check for success */
169 if (!*ResourceId)
170 {
171 /* insufficient memory */
172 FreeItem(ImageName->Buffer);
173 return STATUS_INSUFFICIENT_RESOURCES;
174 }
175 /* now query for resource id */
176 Status = KspQueryRegValue(RegKey, L"ResourceId", (PVOID)*ResourceId, &ImageLength, ValueType);
177
178 if (!NT_SUCCESS(Status))
179 {
180 /* unexpected error */
181 FreeItem(ImageName->Buffer);
182 FreeItem((PVOID)*ResourceId);
183 }
184
185 /* return result */
186 return Status;
187 }
188
189 /*
190 @implemented
191 */
192 KSDDKAPI
193 NTSTATUS
194 NTAPI
195 KsMapModuleName(
196 IN PDEVICE_OBJECT PhysicalDeviceObject,
197 IN PUNICODE_STRING ModuleName,
198 OUT PUNICODE_STRING ImageName,
199 OUT PULONG_PTR ResourceId,
200 OUT PULONG ValueType)
201 {
202 NTSTATUS Status;
203 UNICODE_STRING SubKeyName;
204 UNICODE_STRING Modules = RTL_CONSTANT_STRING(L"Modules\\");
205 HANDLE hKey, hSubKey;
206 OBJECT_ATTRIBUTES ObjectAttributes;
207
208 /* first open device key */
209 Status = IoOpenDeviceRegistryKey(PhysicalDeviceObject, PLUGPLAY_REGKEY_DEVICE, GENERIC_READ, &hKey);
210
211 /* check for success */
212 if (!NT_SUCCESS(Status))
213 {
214 /* invalid parameter */
215 return STATUS_INVALID_PARAMETER;
216 }
217
218 /* initialize subkey buffer */
219 SubKeyName.Length = 0;
220 SubKeyName.MaximumLength = Modules.MaximumLength + ModuleName->MaximumLength;
221 SubKeyName.Buffer = AllocateItem(PagedPool, SubKeyName.MaximumLength);
222
223 /* check for success */
224 if (!SubKeyName.Buffer)
225 {
226 /* not enough memory */
227 ZwClose(hKey);
228 return STATUS_NO_MEMORY;
229 }
230
231 /* build subkey string */
232 RtlAppendUnicodeStringToString(&SubKeyName, &Modules);
233 RtlAppendUnicodeStringToString(&SubKeyName, ModuleName);
234
235 /* initialize subkey attributes */
236 InitializeObjectAttributes(&ObjectAttributes, &SubKeyName, OBJ_CASE_INSENSITIVE, hKey, NULL);
237
238 /* now open the subkey */
239 Status = ZwOpenKey(&hSubKey, GENERIC_READ, &ObjectAttributes);
240
241 /* check for success */
242 if (NT_SUCCESS(Status))
243 {
244 /* defer work */
245 Status = KsGetImageNameAndResourceId(hSubKey, ImageName, ResourceId, ValueType);
246
247 /* close subkey */
248 ZwClose(hSubKey);
249 }
250
251 /* free subkey string */
252 FreeItem(SubKeyName.Buffer);
253
254 /* close device key */
255 ZwClose(hKey);
256
257 /* return status */
258 return Status;
259 }
260
261
262