1 /*
2 Unix SMB/CIFS implementation.
3 SMB client oplock functions
4 Copyright (C) Andrew Tridgell 2001
5
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.
10
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.
15
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/>.
18 */
19
20 #include "includes.h"
21 #include "../lib/util/tevent_ntstatus.h"
22 #include "async_smb.h"
23 #include "libsmb/libsmb.h"
24 #include "../libcli/smb/smbXcli_base.h"
25
26 struct cli_smb_oplock_break_waiter_state {
27 uint16_t fnum;
28 uint8_t level;
29 };
30
31 static void cli_smb_oplock_break_waiter_done(struct tevent_req *subreq);
32
cli_smb_oplock_break_waiter_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli)33 struct tevent_req *cli_smb_oplock_break_waiter_send(TALLOC_CTX *mem_ctx,
34 struct tevent_context *ev,
35 struct cli_state *cli)
36 {
37 struct tevent_req *req, *subreq;
38 struct cli_smb_oplock_break_waiter_state *state;
39
40 req = tevent_req_create(mem_ctx, &state,
41 struct cli_smb_oplock_break_waiter_state);
42 if (req == NULL) {
43 return NULL;
44 }
45
46 /*
47 * Create a fake SMB request that we will never send out. This is only
48 * used to be set into the pending queue with the right mid.
49 */
50 subreq = smb1cli_req_create(mem_ctx, ev, cli->conn, 0, 0, 0, 0, 0, 0,
51 0, NULL, NULL, 0, NULL, 0, NULL);
52 if (tevent_req_nomem(subreq, req)) {
53 return tevent_req_post(req, ev);
54 }
55 smb1cli_req_set_mid(subreq, 0xffff);
56
57 if (!smbXcli_req_set_pending(subreq)) {
58 tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
59 return tevent_req_post(req, ev);
60 }
61 tevent_req_set_callback(subreq, cli_smb_oplock_break_waiter_done, req);
62 return req;
63 }
64
cli_smb_oplock_break_waiter_done(struct tevent_req * subreq)65 static void cli_smb_oplock_break_waiter_done(struct tevent_req *subreq)
66 {
67 struct tevent_req *req = tevent_req_callback_data(
68 subreq, struct tevent_req);
69 struct cli_smb_oplock_break_waiter_state *state = tevent_req_data(
70 req, struct cli_smb_oplock_break_waiter_state);
71 struct iovec *iov;
72 uint8_t wct;
73 uint16_t *vwv;
74 NTSTATUS status;
75
76 status = smb1cli_req_recv(subreq, state,
77 &iov, /* piov */
78 NULL, /* phdr */
79 &wct,
80 &vwv,
81 NULL, /* pvwv_offset */
82 NULL, /* pnum_bytes */
83 NULL, /* pbytes */
84 NULL, /* pbytes_offset */
85 NULL, /* pinbuf */
86 NULL, 0); /* expected */
87 TALLOC_FREE(subreq);
88 if (!NT_STATUS_IS_OK(status)) {
89 tevent_req_nterror(req, status);
90 return;
91 }
92 if (wct < 8) {
93 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
94 return;
95 }
96 state->fnum = SVAL(vwv+2, 0);
97 state->level = CVAL(vwv+3, 1);
98 tevent_req_done(req);
99 }
100
cli_smb_oplock_break_waiter_recv(struct tevent_req * req,uint16_t * pfnum,uint8_t * plevel)101 NTSTATUS cli_smb_oplock_break_waiter_recv(struct tevent_req *req,
102 uint16_t *pfnum,
103 uint8_t *plevel)
104 {
105 struct cli_smb_oplock_break_waiter_state *state = tevent_req_data(
106 req, struct cli_smb_oplock_break_waiter_state);
107 NTSTATUS status;
108
109 if (tevent_req_is_nterror(req, &status)) {
110 return status;
111 }
112 *pfnum = state->fnum;
113 *plevel = state->level;
114 return NT_STATUS_OK;
115 }
116
117 /****************************************************************************
118 send an ack for an oplock break request
119 ****************************************************************************/
120
121 struct cli_oplock_ack_state {
122 uint8_t dummy;
123 };
124
125 static void cli_oplock_ack_done(struct tevent_req *subreq);
126
cli_oplock_ack_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint8_t level)127 struct tevent_req *cli_oplock_ack_send(TALLOC_CTX *mem_ctx,
128 struct tevent_context *ev,
129 struct cli_state *cli,
130 uint16_t fnum, uint8_t level)
131 {
132 struct tevent_req *req, *subreq;
133 struct cli_oplock_ack_state *state;
134
135 req = tevent_req_create(mem_ctx, &state, struct cli_oplock_ack_state);
136 if (req == NULL) {
137 return NULL;
138 }
139
140 subreq = cli_lockingx_send(
141 state, /* mem_ctx */
142 ev, /* tevent_context */
143 cli, /* cli */
144 fnum, /* fnum */
145 LOCKING_ANDX_OPLOCK_RELEASE, /* typeoflock */
146 level, /* newoplocklevel */
147 0, /* timeout */
148 0, /* num_unlocks */
149 NULL, /* unlocks */
150 0, /* num_locks */
151 NULL); /* locks */
152
153 if (tevent_req_nomem(subreq, req)) {
154 return tevent_req_post(req, ev);
155 }
156 tevent_req_set_callback(subreq, cli_oplock_ack_done, req);
157 return req;
158 }
159
cli_oplock_ack_done(struct tevent_req * subreq)160 static void cli_oplock_ack_done(struct tevent_req *subreq)
161 {
162 NTSTATUS status = cli_lockingx_recv(subreq);
163 tevent_req_simple_finish_ntstatus(subreq, status);
164 }
165
cli_oplock_ack_recv(struct tevent_req * req)166 NTSTATUS cli_oplock_ack_recv(struct tevent_req *req)
167 {
168 return tevent_req_simple_recv_ntstatus(req);
169 }
170
171