rostests/kmtests/ntos_ps/PsNotify.c

   1 /*
   2  * PROJECT:         ReactOS kernel-mode tests
   3  * LICENSE:         GPLv2+ - See COPYING in the top level directory
   4  * PURPOSE:         Kernel-Mode Test Suite Process Notification Routines test
   5  * PROGRAMMER:      Thomas Faber <thomas.faber@reactos.org>
   6  */
   7
   8 #include <kmt_test.h>
   9
  10 //#define NDEBUG
  11 #include <debug.h>
  12
  13 static
  14 VOID
  15 NTAPI
  16 CreateProcessNotifyRoutine(
  17     IN HANDLE ParentId,
  18     IN HANDLE ProcessId,
  19     IN BOOLEAN Create)
  20 {
  21     ok_irql(PASSIVE_LEVEL);
  22     if (!Create)
  23         ok_eq_pointer(ProcessId, PsGetCurrentProcessId());
  24     else
  25         ok(ProcessId != PsGetCurrentProcessId(),
  26            "ProcessId %p equals current\n", ProcessId);
  27     DPRINT("%s(%p, %p, %d)\n", __FUNCTION__, ParentId, ProcessId, Create);
  28 }
  29
  30 static
  31 VOID
  32 TestCreateProcessNotify(VOID)
  33 {
  34     NTSTATUS Status;
  35
  36     Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
  37     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
  38
  39     Status = PsSetCreateProcessNotifyRoutine(NULL, FALSE);
  40     ok_eq_hex(Status, STATUS_SUCCESS);
  41
  42     Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
  43     ok_eq_hex(Status, STATUS_SUCCESS);
  44
  45     Status = PsSetCreateProcessNotifyRoutine(NULL, TRUE);
  46     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
  47
  48     Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
  49     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
  50
  51     Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, FALSE);
  52     ok_eq_hex(Status, STATUS_SUCCESS);
  53
  54     /* TODO: test whether the callback is notified on process creation */
  55
  56     Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
  57     ok_eq_hex(Status, STATUS_SUCCESS);
  58
  59     Status = PsSetCreateProcessNotifyRoutine(CreateProcessNotifyRoutine, TRUE);
  60     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
  61
  62     /* TODO: register the same routine twice */
  63     /* TODO: register more than the maximum number of notifications */
  64 }
  65
  66 static
  67 VOID
  68 NTAPI
  69 CreateThreadNotifyRoutine(
  70     IN HANDLE ProcessId,
  71     IN HANDLE ThreadId,
  72     IN BOOLEAN Create)
  73 {
  74     ok_irql(PASSIVE_LEVEL);
  75     if (!Create)
  76     {
  77         ok_eq_pointer(ProcessId, PsGetCurrentProcessId());
  78         ok_eq_pointer(ThreadId, PsGetCurrentThreadId());
  79     }
  80     else
  81     {
  82         ok(ThreadId != PsGetCurrentThreadId(),
  83            "ThreadId %p equals current\n", ThreadId);
  84     }
  85     DPRINT("%s(%p, %p, %d)\n", __FUNCTION__, ProcessId, ThreadId, Create);
  86 }
  87
  88 static
  89 VOID
  90 TestCreateThreadNotify(VOID)
  91 {
  92     NTSTATUS Status;
  93
  94     Status = PsRemoveCreateThreadNotifyRoutine(NULL);
  95     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
  96
  97     Status = PsSetCreateThreadNotifyRoutine(NULL);
  98     ok_eq_hex(Status, STATUS_SUCCESS);
  99
 100     Status = PsRemoveCreateThreadNotifyRoutine(NULL);
 101     ok_eq_hex(Status, STATUS_SUCCESS);
 102
 103     Status = PsRemoveCreateThreadNotifyRoutine(NULL);
 104     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 105
 106     Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
 107     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 108
 109     Status = PsSetCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
 110     ok_eq_hex(Status, STATUS_SUCCESS);
 111
 112     /* TODO: test whether the callback is notified on thread creation */
 113
 114     Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
 115     ok_eq_hex(Status, STATUS_SUCCESS);
 116
 117     Status = PsRemoveCreateThreadNotifyRoutine(CreateThreadNotifyRoutine);
 118     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 119
 120     /* TODO: register the same routine twice */
 121     /* TODO: register more than the maximum number of notifications */
 122 }
 123
 124 static
 125 VOID
 126 NTAPI
 127 LoadImageNotifyRoutine(
 128     IN PUNICODE_STRING FullImageName OPTIONAL,
 129     IN HANDLE ProcessId,
 130     IN PIMAGE_INFO ImageInfo)
 131 {
 132     ok_irql(PASSIVE_LEVEL);
 133     DPRINT("%s('%wZ', %p, %p)\n", __FUNCTION__, FullImageName, ProcessId, ImageInfo);
 134 }
 135
 136 static
 137 VOID
 138 TestLoadImageNotify(VOID)
 139 {
 140     NTSTATUS Status;
 141
 142     Status = PsRemoveLoadImageNotifyRoutine(NULL);
 143     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 144
 145     Status = PsSetLoadImageNotifyRoutine(NULL);
 146     ok_eq_hex(Status, STATUS_SUCCESS);
 147
 148     Status = PsRemoveLoadImageNotifyRoutine(NULL);
 149     ok_eq_hex(Status, STATUS_SUCCESS);
 150
 151     Status = PsRemoveLoadImageNotifyRoutine(NULL);
 152     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 153
 154     Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
 155     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 156
 157     Status = PsSetLoadImageNotifyRoutine(LoadImageNotifyRoutine);
 158     ok_eq_hex(Status, STATUS_SUCCESS);
 159
 160     /* TODO: test whether the callback is notified on image load */
 161
 162     Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
 163     ok_eq_hex(Status, STATUS_SUCCESS);
 164
 165     Status = PsRemoveLoadImageNotifyRoutine(LoadImageNotifyRoutine);
 166     ok_eq_hex(Status, STATUS_PROCEDURE_NOT_FOUND);
 167
 168     /* TODO: register the same routine twice */
 169     /* TODO: register more than the maximum number of notifications */
 170 }
 171
 172 START_TEST(PsNotify)
 173 {
 174     TestCreateProcessNotify();
 175     TestCreateThreadNotify();
 176     TestLoadImageNotify();
 177 }