wmi-1.3.16 from opsview.com
This commit is contained in:
Are Casilla
@@ -0,0 +1,253 @@
|
||||
/*
|
||||
Unix SMB/CIFS implementation.
|
||||
test suite for eventlog rpc operations
|
||||
|
||||
Copyright (C) Tim Potter 2003,2005
|
||||
Copyright (C) Jelmer Vernooij 2004
|
||||
|
||||
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
|
||||
the Free Software Foundation; either version 2 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*/
|
||||
|
||||
#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)
|
||||
{
|
||||
name->string = s;
|
||||
name->length = 2*strlen_m(s);
|
||||
name->size = name->length;
|
||||
}
|
||||
|
||||
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)
|
||||
{
|
||||
struct eventlog_GetNumRecords r;
|
||||
struct eventlog_CloseEventLog cr;
|
||||
struct policy_handle handle;
|
||||
|
||||
if (!get_policy_handle(tctx, p, &handle))
|
||||
return false;
|
||||
|
||||
r.in.handle = &handle;
|
||||
|
||||
torture_assert_ntstatus_ok(tctx,
|
||||
dcerpc_eventlog_GetNumRecords(p, tctx, &r),
|
||||
"GetNumRecords failed");
|
||||
|
||||
torture_comment(tctx, talloc_asprintf(tctx, "%d records\n", *r.out.number));
|
||||
|
||||
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_ReadEventLog(struct torture_context *tctx,
|
||||
struct dcerpc_pipe *p)
|
||||
{
|
||||
NTSTATUS status;
|
||||
struct eventlog_ReadEventLogW r;
|
||||
struct eventlog_CloseEventLog cr;
|
||||
struct policy_handle handle;
|
||||
|
||||
if (!get_policy_handle(tctx, p, &handle))
|
||||
return false;
|
||||
|
||||
r.in.offset = 0;
|
||||
r.in.handle = &handle;
|
||||
r.in.flags = EVENTLOG_BACKWARDS_READ|EVENTLOG_SEQUENTIAL_READ;
|
||||
|
||||
while (1) {
|
||||
DATA_BLOB blob;
|
||||
struct eventlog_Record rec;
|
||||
struct ndr_pull *ndr;
|
||||
|
||||
/* Read first for number of bytes in record */
|
||||
|
||||
r.in.number_of_bytes = 0;
|
||||
r.out.data = NULL;
|
||||
|
||||
status = dcerpc_eventlog_ReadEventLogW(p, tctx, &r);
|
||||
|
||||
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);
|
||||
|
||||
NDR_PRINT_DEBUG(eventlog_Record, &rec);
|
||||
|
||||
torture_assert_ntstatus_ok(tctx, status,
|
||||
"ReadEventLog failed parsing event log record");
|
||||
|
||||
r.in.offset++;
|
||||
}
|
||||
|
||||
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_FlushEventLog(struct torture_context *tctx,
|
||||
struct dcerpc_pipe *p)
|
||||
{
|
||||
struct eventlog_FlushEventLog r;
|
||||
struct eventlog_CloseEventLog cr;
|
||||
struct policy_handle handle;
|
||||
|
||||
if (!get_policy_handle(tctx, p, &handle))
|
||||
return false;
|
||||
|
||||
r.in.handle = &handle;
|
||||
|
||||
/* 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;
|
||||
|
||||
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 *tctx)
|
||||
{
|
||||
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.unknown = NULL;
|
||||
|
||||
torture_assert_ntstatus_ok(tctx,
|
||||
dcerpc_eventlog_ClearEventLogW(p, tctx, &r),
|
||||
"ClearEventLog 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)
|
||||
{
|
||||
struct policy_handle handle;
|
||||
struct eventlog_CloseEventLog cr;
|
||||
|
||||
if (!get_policy_handle(tctx, p, &handle))
|
||||
return false;
|
||||
|
||||
cr.in.handle = cr.out.handle = &handle;
|
||||
|
||||
torture_assert_ntstatus_ok(tctx,
|
||||
dcerpc_eventlog_CloseEventLog(p, tctx, &cr),
|
||||
"CloseEventLog failed");
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
struct torture_suite *torture_rpc_eventlog(void)
|
||||
{
|
||||
struct torture_suite *suite;
|
||||
struct torture_tcase *tcase;
|
||||
|
||||
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
|
||||
/* Destructive test */
|
||||
torture_rpc_tcase_add_test(tcase, "ClearEventLog", test_ClearEventLog);
|
||||
#endif
|
||||
|
||||
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 suite;
|
||||
}
|
||||
Reference in New Issue
Block a user