mirror of
https://github.com/samba-team/samba.git
synced 2025-03-12 20:58:37 +03:00
r6637: Test for asynchronous function and monitor messages.
rafal (This used to be commit 4ac3c145fccc203323e3bf8772bccd290a82cf63)
This commit is contained in:
parent
c19d6aeca7
commit
e2cec76d85
@ -22,6 +22,7 @@
|
||||
#include "includes.h"
|
||||
#include "librpc/gen_ndr/ndr_samr.h"
|
||||
#include "libnet/composite.h"
|
||||
#include "libcli/composite/monitor.h"
|
||||
|
||||
#define TEST_USERNAME "libnetuserinfotest"
|
||||
|
||||
@ -185,9 +186,9 @@ static BOOL test_userinfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
|
||||
NTSTATUS status;
|
||||
struct rpc_composite_userinfo user;
|
||||
struct dom_sid *user_sid;
|
||||
|
||||
|
||||
user_sid = dom_sid_add_rid(mem_ctx, domain_sid, *rid);
|
||||
|
||||
|
||||
user.in.domain_handle = *domain_handle;
|
||||
user.in.sid = dom_sid_string(mem_ctx, user_sid);
|
||||
user.in.level = 5; /* this should be extended */
|
||||
@ -203,6 +204,57 @@ static BOOL test_userinfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
|
||||
}
|
||||
|
||||
|
||||
void msg_handler(struct monitor_msg *m)
|
||||
{
|
||||
switch (m->type) {
|
||||
case rpc_open_user:
|
||||
printf("monitor_msg: user opened (rid=%d, access_mask=0x%08x)\n",
|
||||
m->data.rpc_open_user.rid, m->data.rpc_open_user.access_mask);
|
||||
break;
|
||||
case rpc_query_user:
|
||||
printf("monitor_msg: user queried (level=%d)\n", m->data.rpc_query_user.level);
|
||||
break;
|
||||
case rpc_close_user:
|
||||
printf("monitor_msg: user closed (rid=%d)\n", m->data.rpc_close_user.rid);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static BOOL test_userinfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
|
||||
struct policy_handle *domain_handle,
|
||||
struct dom_sid2 *domain_sid, const char* user_name,
|
||||
uint32_t *rid)
|
||||
{
|
||||
NTSTATUS status;
|
||||
struct composite_context *c;
|
||||
struct rpc_composite_userinfo user;
|
||||
struct dom_sid *user_sid;
|
||||
|
||||
user_sid = dom_sid_add_rid(mem_ctx, domain_sid, *rid);
|
||||
|
||||
user.in.domain_handle = *domain_handle;
|
||||
user.in.sid = dom_sid_string(mem_ctx, user_sid);
|
||||
user.in.level = 10; /* this should be extended */
|
||||
|
||||
printf("Testing async rpc_composite_userinfo\n");
|
||||
|
||||
c = rpc_composite_userinfo_send(p, &user, msg_handler);
|
||||
if (!c) {
|
||||
printf("Failed to call sync rpc_composite_userinfo_send\n");
|
||||
return False;
|
||||
}
|
||||
|
||||
status = rpc_composite_userinfo_recv(c, mem_ctx, &user);
|
||||
if (!NT_STATUS_IS_OK(status)) {
|
||||
printf("Calling async rpc_composite_userinfo failed - %s\n", nt_errstr(status));
|
||||
return False;
|
||||
}
|
||||
|
||||
return True;
|
||||
}
|
||||
|
||||
|
||||
BOOL torture_userinfo(void)
|
||||
{
|
||||
NTSTATUS status;
|
||||
@ -231,6 +283,9 @@ BOOL torture_userinfo(void)
|
||||
|
||||
name.string = lp_workgroup();
|
||||
|
||||
/*
|
||||
* Testing synchronous version
|
||||
*/
|
||||
if (!test_opendomain(p, mem_ctx, &h, &name, &sid)) {
|
||||
ret = False;
|
||||
goto done;
|
||||
@ -250,6 +305,30 @@ BOOL torture_userinfo(void)
|
||||
ret = False;
|
||||
goto done;
|
||||
}
|
||||
|
||||
/*
|
||||
* Testing asynchronous version and monitor messages
|
||||
*/
|
||||
if (!test_opendomain(p, mem_ctx, &h, &name, &sid)) {
|
||||
ret = False;
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (!test_create(p, mem_ctx, &h, TEST_USERNAME, &rid)) {
|
||||
ret = False;
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (!test_userinfo_async(p, mem_ctx, &h, &sid, TEST_USERNAME, &rid)) {
|
||||
ret = False;
|
||||
goto done;
|
||||
}
|
||||
|
||||
if (!test_cleanup(p, mem_ctx, &h, TEST_USERNAME)) {
|
||||
ret = False;
|
||||
goto done;
|
||||
}
|
||||
|
||||
done:
|
||||
talloc_free(mem_ctx);
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user