[httpdisk] Fix deprecation warnings
authorShao Miller <Shao.Miller@yrdsb.edu.on.ca>
Fri, 7 Jan 2011 20:46:09 +0000 (15:46 -0500)
committerShao Miller <Shao.Miller@yrdsb.edu.on.ca>
Fri, 7 Jan 2011 20:46:09 +0000 (15:46 -0500)
Use HttpDisk[M|P]alloc() instead of ExAllocatePool().

If Bo's ksocket and/or ktdi ever come out of HTTPDisk,
their references to allocation functions will have to
be changed.

src/httpdisk/httpdisk.c
src/httpdisk/ksocket.c
src/httpdisk/ktdi.c

index 9d9c2d8..e16b6b9 100644 (file)
@@ -225,6 +225,20 @@ __int64 __cdecl _atoi64(const char *);
 int __cdecl _snprintf(char *, size_t, const char *, ...);
 int __cdecl swprintf(wchar_t *, const wchar_t *, ...);
 
+/** Memory allocation functions. */
+PVOID HttpDiskMalloc(SIZE_T size) {
+    return ExAllocatePoolWithTag(NonPagedPool, size, 'DHvW');
+  }
+
+PVOID HttpDiskPalloc(SIZE_T size) {
+    /*
+     * The call-points for this function merely place-hold for where
+     * Bo's original work allocated from paged pool.  Since this
+     * version is intended for booting, we don't use paged pool.
+     */
+    return ExAllocatePoolWithTag(NonPagedPool, size, 'DHvW');
+  }
+
 #pragma code_seg("INIT")
 
 NTSTATUS
@@ -246,7 +260,7 @@ DriverEntry (
 
     parameter_path.MaximumLength = RegistryPath->Length + sizeof(PARAMETER_KEY);
 
-    parameter_path.Buffer = (PWSTR) ExAllocatePool(PagedPool, parameter_path.MaximumLength);
+    parameter_path.Buffer = HttpDiskPalloc(parameter_path.MaximumLength);
 
     if (parameter_path.Buffer == NULL)
     {
@@ -1028,7 +1042,9 @@ HttpDiskConnect (
 
     device_extension->port = http_disk_information->Port;
 
-    device_extension->host_name = ExAllocatePool(NonPagedPool, http_disk_information->HostNameLength + 1);
+    device_extension->host_name = HttpDiskMalloc(
+        http_disk_information->HostNameLength + 1
+      );
 
     if (device_extension->host_name == NULL)
     {
@@ -1044,7 +1060,9 @@ HttpDiskConnect (
 
     device_extension->host_name[http_disk_information->HostNameLength] = '\0';
 
-    device_extension->file_name = ExAllocatePool(NonPagedPool, http_disk_information->FileNameLength + 1);
+    device_extension->file_name = HttpDiskMalloc(
+        http_disk_information->FileNameLength + 1
+      );
 
     if (device_extension->file_name == NULL)
     {
@@ -1170,7 +1188,7 @@ HttpDiskGetHeader (
     ASSERT(IoStatus != NULL);
     ASSERT(HttpHeader != NULL);
 
-    buffer = ExAllocatePool(PagedPool, BUFFER_SIZE);
+    buffer = HttpDiskPalloc(BUFFER_SIZE);
 
     if (buffer == NULL)
     {
@@ -1312,7 +1330,7 @@ HttpDiskGetBlock (
 
     IoStatus->Information = 0;
 
-    buffer = ExAllocatePool(PagedPool, BUFFER_SIZE + 1);
+    buffer = HttpDiskPalloc(BUFFER_SIZE + 1);
 
     if (buffer == NULL)
     {
index d52d3a8..27a81f1 100644 (file)
 #include "ktdi.h"
 #include "ksocket.h"
 
+/* From httpdisk.c */
+extern PVOID HttpDiskMalloc(SIZE_T);
+extern PVOID HttpDiskPalloc(SIZE_T);
+
 typedef struct _STREAM_SOCKET {
     HANDLE              connectionHandle;
     PFILE_OBJECT        connectionFileObject;
@@ -184,7 +188,7 @@ int __cdecl connect(int socket, const struct sockaddr *addr, int addrlen)
 
         if (!s->streamSocket)
         {
-            s->streamSocket = (PSTREAM_SOCKET) ExAllocatePool(NonPagedPool, sizeof(STREAM_SOCKET));
+            s->streamSocket = HttpDiskMalloc(sizeof *s->streamSocket);
 
             if (!s->streamSocket)
             {
@@ -513,7 +517,7 @@ int __cdecl socket(int af, int type, int protocol)
         return STATUS_INVALID_PARAMETER;
     }
 
-    s = (PSOCKET) ExAllocatePool(NonPagedPool, sizeof(SOCKET));
+    s = HttpDiskMalloc(sizeof *s);
 
     if (!s)
     {
index 606bee8..283c7ba 100644 (file)
 #include <tdikrnl.h>
 #include "ktdi.h"
 
+/* From httpdisk.c */
+extern PVOID HttpDiskMalloc(SIZE_T);
+extern PVOID HttpDiskPalloc(SIZE_T);
+
 NTSTATUS tdi_open_transport_address(PUNICODE_STRING devName, ULONG addr, USHORT port, BOOLEAN shared, PHANDLE addressHandle, PFILE_OBJECT *addressFileObject)
 {
     OBJECT_ATTRIBUTES           attr;
@@ -38,7 +42,7 @@ NTSTATUS tdi_open_transport_address(PUNICODE_STRING devName, ULONG addr, USHORT
              1                                                 +
              sizeof(TA_IP_ADDRESS);
 
-    eaBuffer = (PFILE_FULL_EA_INFORMATION) ExAllocatePool(PagedPool, eaSize);
+    eaBuffer = HttpDiskPalloc(eaSize);
 
     if (eaBuffer == NULL)
     {
@@ -114,7 +118,7 @@ NTSTATUS tdi_open_connection_endpoint(PUNICODE_STRING devName, PVOID connectionC
              1                                                 +
              sizeof(int);
 
-    eaBuffer = (PFILE_FULL_EA_INFORMATION) ExAllocatePool(PagedPool, eaSize);
+    eaBuffer = HttpDiskPalloc(eaSize);
 
     if (eaBuffer == NULL)
     {
@@ -281,7 +285,10 @@ NTSTATUS tdi_connect(PFILE_OBJECT connectionFileObject, ULONG addr, USHORT port)
 
     KeInitializeEvent(&event, NotificationEvent, FALSE);
 
-    remoteInfo = (PTDI_CONNECTION_INFORMATION) ExAllocatePool(NonPagedPool, 2 * sizeof(TDI_CONNECTION_INFORMATION) + 2 * sizeof(TA_IP_ADDRESS));
+    remoteInfo = HttpDiskMalloc(
+        2 * sizeof (TDI_CONNECTION_INFORMATION) +
+        2 * sizeof (TA_IP_ADDRESS)
+      );
 
     if (remoteInfo == NULL)
     {
@@ -382,7 +389,10 @@ NTSTATUS tdi_send_dgram(PFILE_OBJECT addressFileObject, ULONG addr, USHORT port,
 
     KeInitializeEvent(&event, NotificationEvent, FALSE);
 
-    remoteInfo = (PTDI_CONNECTION_INFORMATION) ExAllocatePool(NonPagedPool, sizeof(TDI_CONNECTION_INFORMATION) + sizeof(TA_IP_ADDRESS));
+    remoteInfo = HttpDiskMalloc(
+        sizeof (TDI_CONNECTION_INFORMATION) +
+        sizeof (TA_IP_ADDRESS)
+      );
 
     if (remoteInfo == NULL)
     {
@@ -471,7 +481,10 @@ NTSTATUS tdi_recv_dgram(PFILE_OBJECT addressFileObject, PULONG addr, PUSHORT por
 
     KeInitializeEvent(&event, NotificationEvent, FALSE);
 
-    remoteInfo = (PTDI_CONNECTION_INFORMATION) ExAllocatePool(NonPagedPool, 2 * sizeof(TDI_CONNECTION_INFORMATION) + sizeof(TA_IP_ADDRESS));
+    remoteInfo = HttpDiskMalloc(
+        2 * sizeof (TDI_CONNECTION_INFORMATION) +
+        sizeof (TA_IP_ADDRESS)
+      );
 
     if (remoteInfo == NULL)
     {
@@ -694,7 +707,7 @@ NTSTATUS tdi_query_address(PFILE_OBJECT addressFileObject, PULONG addr, PUSHORT
 
     KeInitializeEvent(&event, NotificationEvent, FALSE);
 
-    localInfo = (PTRANSPORT_ADDRESS) ExAllocatePool(NonPagedPool, sizeof(TDI_ADDRESS_INFO)*10);
+    localInfo = HttpDiskMalloc(sizeof (TDI_ADDRESS_INFO) * 10);
 
     if (localInfo == NULL)
     {