libcli/smb: make use of samba_tevent_context_init()
[Samba/gebeck_regimport.git] / libcli / smb / smb2cli_echo.c
blob29cbf26115fba95f221bcfcc423812225a8b7b1a
1 /*
2 Unix SMB/CIFS implementation.
3 smb2 lib
4 Copyright (C) Stefan Metzmacher 2012
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
20 #include "includes.h"
21 #include "system/network.h"
22 #include "lib/util/tevent_ntstatus.h"
23 #include "smb_common.h"
24 #include "smbXcli_base.h"
26 struct smb2cli_echo_state {
27 uint8_t fixed[0x4];
30 static void smb2cli_echo_done(struct tevent_req *subreq);
32 struct tevent_req *smb2cli_echo_send(TALLOC_CTX *mem_ctx,
33 struct tevent_context *ev,
34 struct smbXcli_conn *conn,
35 uint32_t timeout_msec)
37 struct tevent_req *req, *subreq;
38 struct smb2cli_echo_state *state;
39 uint8_t *fixed;
41 req = tevent_req_create(mem_ctx, &state,
42 struct smb2cli_echo_state);
43 if (req == NULL) {
44 return NULL;
46 fixed = state->fixed;
47 SSVAL(fixed, 0, 4);
48 SSVAL(fixed, 2, 0);
50 subreq = smb2cli_req_send(state, ev, conn, SMB2_OP_KEEPALIVE,
51 0, 0, /* flags */
52 timeout_msec,
53 NULL, /* tcon */
54 NULL, /* session */
55 state->fixed, sizeof(state->fixed),
56 NULL, 0);
57 if (tevent_req_nomem(subreq, req)) {
58 return tevent_req_post(req, ev);
60 tevent_req_set_callback(subreq, smb2cli_echo_done, req);
61 return req;
64 static void smb2cli_echo_done(struct tevent_req *subreq)
66 struct tevent_req *req =
67 tevent_req_callback_data(subreq,
68 struct tevent_req);
69 NTSTATUS status;
70 static const struct smb2cli_req_expected_response expected[] = {
72 .status = NT_STATUS_OK,
73 .body_size = 0x04
77 status = smb2cli_req_recv(subreq, NULL, NULL,
78 expected, ARRAY_SIZE(expected));
79 TALLOC_FREE(subreq);
80 if (tevent_req_nterror(req, status)) {
81 return;
83 tevent_req_done(req);
86 NTSTATUS smb2cli_echo_recv(struct tevent_req *req)
88 return tevent_req_simple_recv_ntstatus(req);
91 NTSTATUS smb2cli_echo(struct smbXcli_conn *conn,
92 uint32_t timeout_msec)
94 TALLOC_CTX *frame = talloc_stackframe();
95 struct tevent_context *ev;
96 struct tevent_req *req;
97 NTSTATUS status = NT_STATUS_NO_MEMORY;
99 if (smbXcli_conn_has_async_calls(conn)) {
101 * Can't use sync call while an async call is in flight
103 status = NT_STATUS_INVALID_PARAMETER;
104 goto fail;
106 ev = samba_tevent_context_init(frame);
107 if (ev == NULL) {
108 goto fail;
110 req = smb2cli_echo_send(frame, ev, conn, timeout_msec);
111 if (req == NULL) {
112 goto fail;
114 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
115 goto fail;
117 status = smb2cli_echo_recv(req);
118 fail:
119 TALLOC_FREE(frame);
120 return status;