sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2024-12-25 23:21:54 +03:00
Code

r2185: add a callback function to the dcerpc async API

Browse Source 
also add a demonstration of its use in the netlogon async example
This commit is contained in:
Andrew Tridgell 2004-09-02 12:03:16 +00:00 committed by Gerald (Jerry) Carter
parent 7256945b52
commit f2a0438c66
3 changed files with 45 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
source/librpc/rpc/dcerpc.c
View File

@ -631,6 +631,9 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
req->state = RPC_REQUEST_DONE;
req->status = status;
DLIST_REMOVE(p->pending, req);
if (req->async.callback) {
req->async.callback(req);
}
}
return;
}
@ -655,6 +658,9 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
req->status = status;
req->state = RPC_REQUEST_DONE;
DLIST_REMOVE(p->pending, req);
if (req->async.callback) {
req->async.callback(req);
}
return;
}
@ -664,6 +670,9 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
req->status = NT_STATUS_NET_WRITE_FAULT;
req->state = RPC_REQUEST_DONE;
DLIST_REMOVE(p->pending, req);
if (req->async.callback) {
req->async.callback(req);
}
return;
}
@ -674,6 +683,9 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
req->status = NT_STATUS_NET_WRITE_FAULT;
req->state = RPC_REQUEST_DONE;
DLIST_REMOVE(p->pending, req);
if (req->async.callback) {
req->async.callback(req);
}
return;
}
@ -686,6 +698,9 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
req->status = NT_STATUS_NO_MEMORY;
req->state = RPC_REQUEST_DONE;
DLIST_REMOVE(p->pending, req);
if (req->async.callback) {
req->async.callback(req);
}
return;
}
memcpy(req->payload.data+req->payload.length,
@ -707,6 +722,10 @@ static void dcerpc_request_recv_data(struct dcerpc_pipe *p,
} else {
req->flags &= ~DCERPC_PULL_BIGENDIAN;
}
if (req->async.callback) {
req->async.callback(req);
}
}
@ -737,6 +756,7 @@ struct rpc_request *dcerpc_request_send(struct dcerpc_pipe *p,
req->payload = data_blob(NULL, 0);
req->flags = 0;
req->fault_code = 0;
req->async.callback = NULL;
init_dcerpc_hdr(p, &pkt);

5
source/librpc/rpc/dcerpc.h
View File

@ -166,4 +166,9 @@ struct rpc_request {
size_t struct_size;
TALLOC_CTX *mem_ctx;
} ndr;
struct {
void (*callback)(struct rpc_request *);
void *private;
} async;
};

24
source/torture/rpc/netlogon.c
View File

@ -1625,6 +1625,14 @@ static BOOL test_GetDomainInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
}
static void async_callback(struct rpc_request *req)
{
int *counter = req->async.private;
if (NT_STATUS_IS_OK(req->status)) {
(*counter)++;
}
}
static BOOL test_GetDomainInfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
{
NTSTATUS status;
@ -1636,6 +1644,7 @@ static BOOL test_GetDomainInfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
struct creds_CredentialState creds_async[ASYNC_COUNT];
struct rpc_request *req[ASYNC_COUNT];
int i;
int async_counter = 0;
if (!test_SetupCredentials3(p, mem_ctx, NETLOGON_NEG_AUTH2_ADS_FLAGS, &creds)) {
return False;
@ -1670,6 +1679,13 @@ static BOOL test_GetDomainInfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
creds_async[i] = creds;
req[i] = dcerpc_netr_LogonGetDomainInfo_send(p, mem_ctx, &r);
req[i]->async.callback = async_callback;
req[i]->async.private = &async_counter;
/* even with this flush per request a w2k3 server seems to
clag with multiple outstanding requests. bleergh. */
event_loop_once(dcerpc_event_context(p));
}
for (i=0;i<ASYNC_COUNT;i++) {
@ -1677,18 +1693,18 @@ static BOOL test_GetDomainInfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
if (!NT_STATUS_IS_OK(status) || !NT_STATUS_IS_OK(r.out.result)) {
printf("netr_LogonGetDomainInfo_async(%d) - %s/%s\n",
i, nt_errstr(status), nt_errstr(r.out.result));
return False;
break;
}
if (!creds_client_check(&creds_async[i], &a.cred)) {
printf("Credential chaining failed at async %d\n", i);
return False;
break;
}
}
printf("Testing netr_LogonGetDomainInfo - async count %d OK\n", ASYNC_COUNT);
printf("Testing netr_LogonGetDomainInfo - async count %d OK\n", async_counter);
return True;
return async_counter == ASYNC_COUNT;
}