r19339: Merge my 4.0-unittest branch. This adds an API for more fine-grained
[sfrench/samba-autobuild/.git] / source4 / torture / rpc / eventlog.c
index 1929c4a0681124b838a09d4ea38c8ef801572730..3ecc1a422bede928d19ad28bb483d23bd5802442 100644 (file)
@@ -2,7 +2,7 @@
    Unix SMB/CIFS implementation.
    test suite for eventlog rpc operations
 
-   Copyright (C) Tim Potter 2003
+   Copyright (C) Tim Potter 2003,2005
    Copyright (C) Jelmer Vernooij 2004
    
    This program is free software; you can redistribute it and/or modify
 */
 
 #include "includes.h"
+#include "torture/torture.h"
 #include "librpc/gen_ndr/ndr_eventlog.h"
+#include "librpc/gen_ndr/ndr_eventlog_c.h"
 #include "librpc/gen_ndr/ndr_lsa.h"
+#include "torture/rpc/rpc.h"
 
 static void init_lsa_String(struct lsa_String *name, const char *s)
 {
@@ -31,179 +34,220 @@ static void init_lsa_String(struct lsa_String *name, const char *s)
        name->size = name->length;
 }
 
-static BOOL test_GetNumRecords(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *handle)
+static bool get_policy_handle(struct torture_context *tctx, 
+                                                         struct dcerpc_pipe *p,
+                                                         struct policy_handle *handle)
+{
+       struct eventlog_OpenEventLogW r;
+       struct eventlog_OpenUnknown0 unknown0;
+
+       unknown0.unknown0 = 0x005c;
+       unknown0.unknown1 = 0x0001;
+
+       r.in.unknown0 = &unknown0;
+       init_lsa_String(&r.in.logname, "dns server");
+       init_lsa_String(&r.in.servername, NULL);
+       r.in.unknown2 = 0x00000001;
+       r.in.unknown3 = 0x00000001;
+       r.out.handle = handle;
+
+       torture_assert_ntstatus_ok(tctx, 
+                       dcerpc_eventlog_OpenEventLogW(p, tctx, &r), 
+                       "OpenEventLog failed");
+
+       torture_assert_ntstatus_ok(tctx, r.out.result, "OpenEventLog failed");
+
+       return true;
+}
+
+
+
+static bool test_GetNumRecords(struct torture_context *tctx, struct dcerpc_pipe *p)
 {
-       NTSTATUS status;
        struct eventlog_GetNumRecords r;
+       struct eventlog_CloseEventLog cr;
+       struct policy_handle handle;
 
-       printf("\ntesting GetNumRecords\n");
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
 
-       r.in.handle = handle;
+       r.in.handle = &handle;
 
-       status = dcerpc_eventlog_GetNumRecords(p, mem_ctx, &r);
+       torture_assert_ntstatus_ok(tctx, 
+                       dcerpc_eventlog_GetNumRecords(p, tctx, &r), 
+                       "GetNumRecords failed");
 
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("GetNumRecords failed - %s\n", nt_errstr(status));
-               return False;
-       }
+       torture_comment(tctx, talloc_asprintf(tctx, "%d records\n", r.out.number));
 
-       printf("%d records\n", r.out.number);
+       cr.in.handle = cr.out.handle = &handle;
 
-       return True;
+       torture_assert_ntstatus_ok(tctx, 
+                                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr), 
+                                       "CloseEventLog failed");
+       return true;
 }
 
-static BOOL test_ReadEventLog(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *handle, uint32_t offset)
+static bool test_ReadEventLog(struct torture_context *tctx, 
+                                                         struct dcerpc_pipe *p)
 {
        NTSTATUS status;
        struct eventlog_ReadEventLogW r;
+       struct eventlog_CloseEventLog cr;
+       struct policy_handle handle;
 
-       printf("\ntesting ReadEventLog\n");
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
 
-       r.in.flags = 0x0;
-       r.in.offset = offset;
-       r.in.handle = handle;
-       r.in.number_of_bytes = 0x0;
+       r.in.offset = 0;
+       r.in.handle = &handle;
+       r.in.flags = EVENTLOG_BACKWARDS_READ|EVENTLOG_SEQUENTIAL_READ;
 
-       r.out.data = talloc(mem_ctx, uint8_t);
+       while (1) {
+               DATA_BLOB blob;
+               struct eventlog_Record rec;
+               struct ndr_pull *ndr;
 
-       status = dcerpc_eventlog_ReadEventLogW(p, mem_ctx, &r);
+               /* Read first for number of bytes in record */
 
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("ReadEventLog failed - %s\n", nt_errstr(status));
-               return False;
-       }
+               r.in.number_of_bytes = 0;
+               r.out.data = NULL;
 
-       if (NT_STATUS_IS_OK(r.out.result)) {
-               /* No data */
-               return True;
-       }
+               status = dcerpc_eventlog_ReadEventLogW(p, tctx, &r);
 
-       if (!NT_STATUS_EQUAL(r.out.result, NT_STATUS_BUFFER_TOO_SMALL)) {
-               printf("ReadEventLog failed - %s\n", nt_errstr(r.out.result));
-               return False;
-       }
+               if (NT_STATUS_EQUAL(r.out.result, NT_STATUS_END_OF_FILE)) {
+                       break;
+               }
+
+               torture_assert_ntstatus_ok(tctx, status, "ReadEventLog failed");
+
+               torture_assert_ntstatus_equal(tctx, r.out.result, NT_STATUS_BUFFER_TOO_SMALL,
+                       "ReadEventLog failed");
+               
+               /* Now read the actual record */
+
+               r.in.number_of_bytes = r.out.real_size;
+               r.out.data = talloc_size(tctx, r.in.number_of_bytes);
+
+               status = dcerpc_eventlog_ReadEventLogW(p, tctx, &r);
+
+               torture_assert_ntstatus_ok(tctx, status, "ReadEventLog failed");
+               
+               /* Decode a user-marshalled record */
+
+               blob.length = r.out.sent_size;
+               blob.data = talloc_steal(tctx, r.out.data);
+
+               ndr = ndr_pull_init_blob(&blob, tctx);
+
+               status = ndr_pull_eventlog_Record(
+                       ndr, NDR_SCALARS|NDR_BUFFERS, &rec);
 
-       r.in.number_of_bytes = r.out.real_size;
+               NDR_PRINT_DEBUG(eventlog_Record, &rec);
 
-       status = dcerpc_eventlog_ReadEventLogW(p, mem_ctx, &r);
+               torture_assert_ntstatus_ok(tctx, status, 
+                               "ReadEventLog failed parsing event log record");
 
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("ReadEventLog failed - %s\n", nt_errstr(status));
-               return False;
+               r.in.offset++;
        }
 
+       cr.in.handle = cr.out.handle = &handle;
 
-       return True;
+       torture_assert_ntstatus_ok(tctx, 
+                                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr), 
+                                       "CloseEventLog failed");
+
+       return true;
 }
 
-static BOOL test_CloseEventLog(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
-                       struct policy_handle *handle)
+static bool test_FlushEventLog(struct torture_context *tctx, 
+                                                          struct dcerpc_pipe *p)
 {
-       NTSTATUS status;
-       struct eventlog_CloseEventLog r;
+       struct eventlog_FlushEventLog r;
+       struct eventlog_CloseEventLog cr;
+       struct policy_handle handle;
 
-       r.in.handle = r.out.handle = handle;
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
 
-       printf("Testing CloseEventLog\n");
+       r.in.handle = &handle;
 
-       status = dcerpc_eventlog_CloseEventLog(p, mem_ctx, &r);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("CloseEventLog failed - %s\n", nt_errstr(status));
-               return False;
-       }
+       /* Huh?  Does this RPC always return access denied? */
+       torture_assert_ntstatus_equal(tctx, 
+                       dcerpc_eventlog_FlushEventLog(p, tctx, &r),
+                       NT_STATUS_ACCESS_DENIED, 
+                       "FlushEventLog failed");
+
+       cr.in.handle = cr.out.handle = &handle;
 
-       return True;
+       torture_assert_ntstatus_ok(tctx, 
+                                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr), 
+                                       "CloseEventLog failed");
+
+       return true;
 }
 
-static BOOL test_ClearEventLog(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
-                              struct policy_handle *handle)
+static bool test_ClearEventLog(struct dcerpc_pipe *p, TALLOC_CTX *tctx)
 {
-       NTSTATUS status;
        struct eventlog_ClearEventLogW r;
+       struct eventlog_CloseEventLog cr;
+       struct policy_handle handle;
+
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
 
-       r.in.handle = handle;
+       r.in.handle = &handle;
        r.in.unknown = NULL;
 
-       printf("Testing ClearEventLog\n");
+       torture_assert_ntstatus_ok(tctx, 
+                       dcerpc_eventlog_ClearEventLogW(p, tctx, &r), 
+                       "ClearEventLog failed");
 
-       status = dcerpc_eventlog_ClearEventLogW(p, mem_ctx, &r);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("ClearEventLog failed - %s\n", nt_errstr(status));
-               return False;
-       }
+       cr.in.handle = cr.out.handle = &handle;
+
+       torture_assert_ntstatus_ok(tctx, 
+                                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr), 
+                                       "CloseEventLog failed");
 
-       return True;
+       return true;
 }
 
-static BOOL test_OpenEventLog(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *handle)
+static bool test_OpenEventLog(struct torture_context *tctx, 
+                                                         struct dcerpc_pipe *p)
 {
-       NTSTATUS status;
-       struct eventlog_OpenEventLogW r;
-       struct eventlog_OpenUnknown0 unknown0;
-
-       printf("\ntesting OpenEventLog\n");
-
-       unknown0.unknown0 = 0x005c;
-       unknown0.unknown1 = 0x0001;
+       struct policy_handle handle;
+       struct eventlog_CloseEventLog cr;
 
-       r.in.unknown0 = &unknown0;
-       init_lsa_String(&r.in.logname, "system");
-       init_lsa_String(&r.in.servername, NULL);
-       r.in.unknown2 = 0x00000001;
-       r.in.unknown3 = 0x00000001;
-       r.out.handle = handle;
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
 
-       status = dcerpc_eventlog_OpenEventLogW(p, mem_ctx, &r);
+       cr.in.handle = cr.out.handle = &handle;
 
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("OpenEventLog failed - %s\n", nt_errstr(status));
-               return False;
-       }
+       torture_assert_ntstatus_ok(tctx, 
+                                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr), 
+                                       "CloseEventLog failed");
 
-       if (!NT_STATUS_IS_OK(r.out.result)) {
-               printf("OpenEventLog failed - %s\n", nt_errstr(r.out.result));
-               return False;
-       }
-
-       return True;
+       return true;
 }
 
-BOOL torture_rpc_eventlog(void)
+struct torture_suite *torture_rpc_eventlog(void)
 {
-    NTSTATUS status;
-    struct dcerpc_pipe *p;
-       struct policy_handle handle;
-       TALLOC_CTX *mem_ctx;
-       BOOL ret = True;
-
-       mem_ctx = talloc_init("torture_rpc_atsvc");
-
-       status = torture_rpc_connection(mem_ctx, 
-                                       &p, 
-                                       DCERPC_EVENTLOG_NAME, 
-                                       DCERPC_EVENTLOG_UUID, 
-                                       DCERPC_EVENTLOG_VERSION);
-       if (!NT_STATUS_IS_OK(status)) {
-               talloc_free(mem_ctx);
-               return False;
-       }
+       struct torture_suite *suite;
+       struct torture_tcase *tcase;
 
-       if (!test_OpenEventLog(p, mem_ctx, &handle)) {
-               talloc_free(mem_ctx);
-               return False;
-       }
+       suite = torture_suite_create(talloc_autofree_context(), "EVENTLOG");
+       tcase = torture_suite_add_rpc_iface_tcase(suite, "eventlog", 
+                                                                                         &dcerpc_table_eventlog);
+
+       torture_rpc_tcase_add_test(tcase, "OpenEventLog", test_OpenEventLog);
 
 #if 0
-       test_ClearEventLog(p, mem_ctx, &handle); /* Destructive test */
+       /* Destructive test */
+       torture_rpc_tcase_add_test(tcase, "ClearEventLog", test_ClearEventLog);
 #endif
        
-       test_GetNumRecords(p, mem_ctx, &handle);
-
-       test_ReadEventLog(p, mem_ctx, &handle, 0);
-
-       test_CloseEventLog(p, mem_ctx, &handle);
-
-       talloc_free(mem_ctx);
+       torture_rpc_tcase_add_test(tcase, "GetNumRecords", test_GetNumRecords);
+       torture_rpc_tcase_add_test(tcase, "ReadEventLog", test_ReadEventLog);
+       torture_rpc_tcase_add_test(tcase, "FlushEventLog", test_FlushEventLog);
 
-       return ret;
+       return suite;
 }