Merge branch 'master' of ssh://git.samba.org/data/git/samba
[amitay/samba.git] / source4 / torture / rpc / eventlog.c
index 5da205bcdeba9ae34945fd5da7d12968f0824f45..2fd9d923f1e54671e749799c0de25f73b41569d6 100644 (file)
@@ -4,6 +4,7 @@
 
    Copyright (C) Tim Potter 2003,2005
    Copyright (C) Jelmer Vernooij 2004
+   Copyright (C) Guenther Deschner 2009
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
@@ -27,6 +28,8 @@
 #include "torture/rpc/rpc.h"
 #include "param/param.h"
 
+#define TEST_BACKUP_NAME "samrtorturetest"
+
 static void init_lsa_String(struct lsa_String *name, const char *s)
 {
        name->string = s;
@@ -122,9 +125,10 @@ static bool test_ReadEventLog(struct torture_context *tctx,
 
        while (1) {
                DATA_BLOB blob;
-               struct eventlog_Record rec;
-               struct ndr_pull *ndr;
+               struct EVENTLOGRECORD rec;
                enum ndr_err_code ndr_err;
+               uint32_t size = 0;
+               uint32_t pos = 0;
 
                /* Read first for number of bytes in record */
 
@@ -137,6 +141,7 @@ static bool test_ReadEventLog(struct torture_context *tctx,
                status = dcerpc_eventlog_ReadEventLogW(p, tctx, &r);
 
                if (NT_STATUS_EQUAL(r.out.result, NT_STATUS_END_OF_FILE)) {
+                       /* FIXME: still need to decode then */
                        break;
                }
 
@@ -153,17 +158,32 @@ static bool test_ReadEventLog(struct torture_context *tctx,
                torture_assert_ntstatus_ok(tctx, status, "ReadEventLog failed");
 
                /* Decode a user-marshalled record */
+               size = IVAL(r.out.data, pos);
+
+               while (size > 0) {
+
+                       blob = data_blob_const(r.out.data + pos, size);
+                       dump_data(0, blob.data, blob.length);
 
-               blob.length = *r.out.sent_size;
-               blob.data = talloc_steal(tctx, r.out.data);
+                       ndr_err = ndr_pull_struct_blob_all(&blob, tctx,
+                               lp_iconv_convenience(tctx->lp_ctx), &rec,
+                               (ndr_pull_flags_fn_t)ndr_pull_EVENTLOGRECORD);
+                       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                               status = ndr_map_error2ntstatus(ndr_err);
+                               torture_assert_ntstatus_ok(tctx, status,
+                                       "ReadEventLog failed parsing event log record");
+                       }
 
-               ndr = ndr_pull_init_blob(&blob, tctx, lp_iconv_convenience(tctx->lp_ctx));
+                       NDR_PRINT_DEBUG(EVENTLOGRECORD, &rec);
 
-               ndr_err = ndr_pull_eventlog_Record(
-                       ndr, NDR_SCALARS|NDR_BUFFERS, &rec);
-               status = ndr_map_error2ntstatus(ndr_err);
+                       pos += size;
 
-               NDR_PRINT_DEBUG(eventlog_Record, &rec);
+                       if (pos + 4 > *r.out.sent_size) {
+                               break;
+                       }
+
+                       size = IVAL(r.out.data, pos);
+               }
 
                torture_assert_ntstatus_ok(tctx, status,
                                "ReadEventLog failed parsing event log record");
@@ -282,6 +302,53 @@ static bool test_ClearEventLog(struct torture_context *tctx,
        return true;
 }
 
+static bool test_GetLogInformation(struct torture_context *tctx,
+                                  struct dcerpc_pipe *p)
+{
+       NTSTATUS status;
+       struct eventlog_GetLogIntormation r;
+       struct eventlog_CloseEventLog cr;
+       struct policy_handle handle;
+       uint32_t bytes_needed = 0;
+
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
+
+       r.in.handle = &handle;
+       r.in.level = 1;
+       r.in.buf_size = 0;
+       r.out.buffer = NULL;
+       r.out.bytes_needed = &bytes_needed;
+
+       status = dcerpc_eventlog_GetLogIntormation(p, tctx, &r);
+
+       torture_assert_ntstatus_equal(tctx, status, NT_STATUS_INVALID_LEVEL,
+                                     "GetLogInformation failed");
+
+       r.in.level = 0;
+
+       status = dcerpc_eventlog_GetLogIntormation(p, tctx, &r);
+
+       torture_assert_ntstatus_equal(tctx, status, NT_STATUS_BUFFER_TOO_SMALL,
+                                     "GetLogInformation failed");
+
+       r.in.buf_size = bytes_needed;
+       r.out.buffer = talloc_array(tctx, uint8_t, bytes_needed);
+
+       status = dcerpc_eventlog_GetLogIntormation(p, tctx, &r);
+
+       torture_assert_ntstatus_ok(tctx, status, "GetLogInformation failed");
+
+       cr.in.handle = cr.out.handle = &handle;
+
+       torture_assert_ntstatus_ok(tctx,
+                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr),
+                       "CloseEventLog failed");
+
+       return true;
+}
+
+
 static bool test_OpenEventLog(struct torture_context *tctx,
                              struct dcerpc_pipe *p)
 {
@@ -300,6 +367,72 @@ static bool test_OpenEventLog(struct torture_context *tctx,
        return true;
 }
 
+static bool test_BackupLog(struct torture_context *tctx,
+                          struct dcerpc_pipe *p)
+{
+       NTSTATUS status;
+       struct policy_handle handle, backup_handle;
+       struct eventlog_BackupEventLogW r;
+       struct eventlog_OpenBackupEventLogW b;
+       struct eventlog_CloseEventLog cr;
+       const char *tmp;
+       struct lsa_String backup_filename;
+       struct eventlog_OpenUnknown0 unknown0;
+
+       if (!get_policy_handle(tctx, p, &handle))
+               return false;
+
+       tmp = talloc_asprintf(tctx, "C:\\%s", TEST_BACKUP_NAME);
+       init_lsa_String(&backup_filename, tmp);
+
+       r.in.handle = &handle;
+       r.in.backup_filename = &backup_filename;
+
+       status = dcerpc_eventlog_BackupEventLogW(p, tctx, &r);
+       torture_assert_ntstatus_equal(tctx, status,
+               NT_STATUS_OBJECT_PATH_SYNTAX_BAD, "BackupEventLogW failed");
+
+       tmp = talloc_asprintf(tctx, "\\??\\C:\\%s", TEST_BACKUP_NAME);
+       init_lsa_String(&backup_filename, tmp);
+
+       r.in.handle = &handle;
+       r.in.backup_filename = &backup_filename;
+
+       status = dcerpc_eventlog_BackupEventLogW(p, tctx, &r);
+       torture_assert_ntstatus_ok(tctx, status, "BackupEventLogW failed");
+
+       status = dcerpc_eventlog_BackupEventLogW(p, tctx, &r);
+       torture_assert_ntstatus_equal(tctx, status,
+               NT_STATUS_OBJECT_NAME_COLLISION, "BackupEventLogW failed");
+
+       cr.in.handle = cr.out.handle = &handle;
+
+       torture_assert_ntstatus_ok(tctx,
+                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr),
+                       "BackupLog failed");
+
+       unknown0.unknown0 = 0x005c;
+       unknown0.unknown1 = 0x0001;
+
+       b.in.unknown0 = &unknown0;
+       b.in.backup_logname = &backup_filename;
+       b.in.major_version = 1;
+       b.in.minor_version = 1;
+       b.out.handle = &backup_handle;
+
+       status = dcerpc_eventlog_OpenBackupEventLogW(p, tctx, &b);
+
+       torture_assert_ntstatus_ok(tctx, status, "OpenBackupEventLogW failed");
+
+       cr.in.handle = cr.out.handle = &backup_handle;
+
+       torture_assert_ntstatus_ok(tctx,
+                       dcerpc_eventlog_CloseEventLog(p, tctx, &cr),
+                       "CloseEventLog failed");
+
+       return true;
+}
+
 struct torture_suite *torture_rpc_eventlog(TALLOC_CTX *mem_ctx)
 {
        struct torture_suite *suite;
@@ -318,6 +451,8 @@ struct torture_suite *torture_rpc_eventlog(TALLOC_CTX *mem_ctx)
        torture_rpc_tcase_add_test(tcase, "ReadEventLog", test_ReadEventLog);
        torture_rpc_tcase_add_test(tcase, "ReportEventLog", test_ReportEventLog);
        torture_rpc_tcase_add_test(tcase, "FlushEventLog", test_FlushEventLog);
+       torture_rpc_tcase_add_test(tcase, "GetLogIntormation", test_GetLogInformation);
+       torture_rpc_tcase_add_test(tcase, "BackupLog", test_BackupLog);
 
        return suite;
 }