Start to tidy-up check_user_ok().
[Samba/vl.git] / source3 / libsmb / clisymlink.c
blobd0a2deb35549c1d707f6038b47c97bc147532121
1 /*
2 * Unix SMB/CIFS implementation.
3 * Client implementation of setting symlinks using reparse points
4 * Copyright (C) Volker Lendecke 2011
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/filesys.h"
22 #include "libsmb/libsmb.h"
23 #include "../lib/util/tevent_ntstatus.h"
24 #include "async_smb.h"
25 #include "libsmb/clirap.h"
26 #include "trans2.h"
27 #include "libcli/security/secdesc.h"
28 #include "libcli/security/security.h"
29 #include "../libcli/smb/smbXcli_base.h"
31 struct cli_symlink_state {
32 struct tevent_context *ev;
33 struct cli_state *cli;
34 const char *oldpath;
35 const char *newpath;
36 uint32_t flags;
38 uint16_t fnum;
40 uint16_t setup[4];
41 NTSTATUS set_reparse_status;
44 static void cli_symlink_create_done(struct tevent_req *subreq);
45 static void cli_symlink_set_reparse_done(struct tevent_req *subreq);
46 static void cli_symlink_delete_on_close_done(struct tevent_req *subreq);
47 static void cli_symlink_close_done(struct tevent_req *subreq);
49 struct tevent_req *cli_symlink_send(TALLOC_CTX *mem_ctx,
50 struct tevent_context *ev,
51 struct cli_state *cli,
52 const char *oldpath,
53 const char *newpath,
54 uint32_t flags)
56 struct tevent_req *req, *subreq;
57 struct cli_symlink_state *state;
59 req = tevent_req_create(mem_ctx, &state, struct cli_symlink_state);
60 if (req == NULL) {
61 return NULL;
63 state->ev = ev;
64 state->cli = cli;
65 state->oldpath = oldpath;
66 state->newpath = newpath;
67 state->flags = flags;
69 subreq = cli_ntcreate_send(
70 state, ev, cli, state->oldpath, 0,
71 SYNCHRONIZE_ACCESS|DELETE_ACCESS|
72 FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES,
73 FILE_ATTRIBUTE_NORMAL, FILE_SHARE_NONE, FILE_CREATE,
74 FILE_OPEN_REPARSE_POINT|FILE_SYNCHRONOUS_IO_NONALERT|
75 FILE_NON_DIRECTORY_FILE, 0);
76 if (tevent_req_nomem(subreq, req)) {
77 return tevent_req_post(req, ev);
79 tevent_req_set_callback(subreq, cli_symlink_create_done, req);
80 return req;
83 static void cli_symlink_create_done(struct tevent_req *subreq)
85 struct tevent_req *req = tevent_req_callback_data(
86 subreq, struct tevent_req);
87 struct cli_symlink_state *state = tevent_req_data(
88 req, struct cli_symlink_state);
89 uint8_t *data;
90 size_t data_len;
91 NTSTATUS status;
93 status = cli_ntcreate_recv(subreq, &state->fnum);
94 TALLOC_FREE(subreq);
95 if (tevent_req_nterror(req, status)) {
96 return;
99 SIVAL(state->setup, 0, FSCTL_SET_REPARSE_POINT);
100 SSVAL(state->setup, 4, state->fnum);
101 SCVAL(state->setup, 6, 1); /* IsFcntl */
102 SCVAL(state->setup, 7, 0); /* IsFlags */
104 if (!symlink_reparse_buffer_marshall(
105 state->newpath, NULL, state->flags, state,
106 &data, &data_len)) {
107 tevent_req_oom(req);
108 return;
111 subreq = cli_trans_send(state, state->ev, state->cli, SMBnttrans,
112 NULL, -1, /* name, fid */
113 NT_TRANSACT_IOCTL, 0,
114 state->setup, 4, 0, /* setup */
115 NULL, 0, 0, /* param */
116 data, data_len, 0); /* data */
117 if (tevent_req_nomem(subreq, req)) {
118 return;
120 tevent_req_set_callback(subreq, cli_symlink_set_reparse_done, req);
123 static void cli_symlink_set_reparse_done(struct tevent_req *subreq)
125 struct tevent_req *req = tevent_req_callback_data(
126 subreq, struct tevent_req);
127 struct cli_symlink_state *state = tevent_req_data(
128 req, struct cli_symlink_state);
130 state->set_reparse_status = cli_trans_recv(
131 subreq, NULL, NULL,
132 NULL, 0, NULL, /* rsetup */
133 NULL, 0, NULL, /* rparam */
134 NULL, 0, NULL); /* rdata */
135 TALLOC_FREE(subreq);
137 if (NT_STATUS_IS_OK(state->set_reparse_status)) {
138 subreq = cli_close_send(state, state->ev, state->cli,
139 state->fnum);
140 if (tevent_req_nomem(subreq, req)) {
141 return;
143 tevent_req_set_callback(subreq, cli_symlink_close_done, req);
144 return;
146 subreq = cli_nt_delete_on_close_send(
147 state, state->ev, state->cli, state->fnum, true);
148 if (tevent_req_nomem(subreq, req)) {
149 return;
151 tevent_req_set_callback(subreq, cli_symlink_delete_on_close_done, req);
154 static void cli_symlink_delete_on_close_done(struct tevent_req *subreq)
156 struct tevent_req *req = tevent_req_callback_data(
157 subreq, struct tevent_req);
158 struct cli_symlink_state *state = tevent_req_data(
159 req, struct cli_symlink_state);
162 * Ignore status, we can't do much anyway in case of failure
165 (void)cli_nt_delete_on_close_recv(subreq);
166 TALLOC_FREE(subreq);
168 subreq = cli_close_send(state, state->ev, state->cli, state->fnum);
169 if (tevent_req_nomem(subreq, req)) {
170 return;
172 tevent_req_set_callback(subreq, cli_symlink_close_done, req);
175 static void cli_symlink_close_done(struct tevent_req *subreq)
177 struct tevent_req *req = tevent_req_callback_data(
178 subreq, struct tevent_req);
179 struct cli_symlink_state *state = tevent_req_data(
180 req, struct cli_symlink_state);
181 NTSTATUS status;
183 status = cli_close_recv(subreq);
184 TALLOC_FREE(subreq);
186 if (tevent_req_nterror(req, status)) {
187 return;
189 if (tevent_req_nterror(req, state->set_reparse_status)) {
190 return;
192 tevent_req_done(req);
195 NTSTATUS cli_symlink_recv(struct tevent_req *req)
197 return tevent_req_simple_recv_ntstatus(req);
200 NTSTATUS cli_symlink(struct cli_state *cli, const char *oldname,
201 const char *newname, uint32_t flags)
203 TALLOC_CTX *frame = talloc_stackframe();
204 struct event_context *ev;
205 struct tevent_req *req;
206 NTSTATUS status = NT_STATUS_NO_MEMORY;
208 if (smbXcli_conn_has_async_calls(cli->conn)) {
209 status = NT_STATUS_INVALID_PARAMETER;
210 goto fail;
212 ev = event_context_init(frame);
213 if (ev == NULL) {
214 goto fail;
216 req = cli_symlink_send(frame, ev, cli, oldname, newname, flags);
217 if (req == NULL) {
218 goto fail;
220 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
221 goto fail;
223 status = cli_symlink_recv(req);
224 fail:
225 TALLOC_FREE(frame);
226 return status;
229 struct cli_readlink_state {
230 struct tevent_context *ev;
231 struct cli_state *cli;
232 uint16_t fnum;
234 uint16_t setup[4];
235 NTSTATUS get_reparse_status;
236 uint8_t *data;
237 uint32_t num_data;
240 static void cli_readlink_opened(struct tevent_req *subreq);
241 static void cli_readlink_got_reparse_data(struct tevent_req *subreq);
242 static void cli_readlink_closed(struct tevent_req *subreq);
244 struct tevent_req *cli_readlink_send(TALLOC_CTX *mem_ctx,
245 struct tevent_context *ev,
246 struct cli_state *cli,
247 const char *fname)
249 struct tevent_req *req, *subreq;
250 struct cli_readlink_state *state;
252 req = tevent_req_create(mem_ctx, &state, struct cli_readlink_state);
253 if (req == NULL) {
254 return NULL;
256 state->ev = ev;
257 state->cli = cli;
259 subreq = cli_ntcreate_send(
260 state, ev, cli, fname, 0, FILE_READ_ATTRIBUTES | FILE_READ_EA,
261 0, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
262 FILE_OPEN, FILE_OPEN_REPARSE_POINT, 0);
263 if (tevent_req_nomem(subreq, req)) {
264 return tevent_req_post(req, ev);
266 tevent_req_set_callback(subreq, cli_readlink_opened, req);
267 return req;
270 static void cli_readlink_opened(struct tevent_req *subreq)
272 struct tevent_req *req = tevent_req_callback_data(
273 subreq, struct tevent_req);
274 struct cli_readlink_state *state = tevent_req_data(
275 req, struct cli_readlink_state);
276 NTSTATUS status;
278 status = cli_ntcreate_recv(subreq, &state->fnum);
279 TALLOC_FREE(subreq);
280 if (tevent_req_nterror(req, status)) {
281 return;
284 SIVAL(state->setup, 0, FSCTL_GET_REPARSE_POINT);
285 SSVAL(state->setup, 4, state->fnum);
286 SCVAL(state->setup, 6, 1); /* IsFcntl */
287 SCVAL(state->setup, 7, 0); /* IsFlags */
289 subreq = cli_trans_send(state, state->ev, state->cli, SMBnttrans,
290 NULL, -1, /* name, fid */
291 NT_TRANSACT_IOCTL, 0,
292 state->setup, 4, 0, /* setup */
293 NULL, 0, 0, /* param */
294 NULL, 0, 16384); /* data */
295 if (tevent_req_nomem(subreq, req)) {
296 return;
298 tevent_req_set_callback(subreq, cli_readlink_got_reparse_data, req);
301 static void cli_readlink_got_reparse_data(struct tevent_req *subreq)
303 struct tevent_req *req = tevent_req_callback_data(
304 subreq, struct tevent_req);
305 struct cli_readlink_state *state = tevent_req_data(
306 req, struct cli_readlink_state);
308 state->get_reparse_status = cli_trans_recv(
309 subreq, state, NULL,
310 NULL, 0, NULL, /* rsetup */
311 NULL, 0, NULL, /* rparam */
312 &state->data, 20, &state->num_data); /* rdata */
313 TALLOC_FREE(subreq);
315 subreq = cli_close_send(state, state->ev, state->cli, state->fnum);
316 if (tevent_req_nomem(subreq, req)) {
317 return;
319 tevent_req_set_callback(subreq, cli_readlink_closed, req);
322 static void cli_readlink_closed(struct tevent_req *subreq)
324 struct tevent_req *req = tevent_req_callback_data(
325 subreq, struct tevent_req);
326 NTSTATUS status;
328 status = cli_close_recv(subreq);
329 TALLOC_FREE(subreq);
330 if (tevent_req_nterror(req, status)) {
331 return;
333 tevent_req_done(req);
336 NTSTATUS cli_readlink_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
337 char **psubstitute_name, char **pprint_name,
338 uint32_t *pflags)
340 struct cli_readlink_state *state = tevent_req_data(
341 req, struct cli_readlink_state);
342 NTSTATUS status;
343 char *substitute_name;
344 char *print_name;
345 uint32_t flags;
347 if (tevent_req_is_nterror(req, &status)) {
348 return status;
351 if (!symlink_reparse_buffer_parse(state->data, state->num_data,
352 talloc_tos(), &substitute_name,
353 &print_name, &flags)) {
354 return NT_STATUS_INVALID_NETWORK_RESPONSE;
357 if (psubstitute_name != NULL) {
358 *psubstitute_name = talloc_move(mem_ctx, &substitute_name);
360 TALLOC_FREE(substitute_name);
362 if (pprint_name != NULL) {
363 *pprint_name = talloc_move(mem_ctx, &print_name);
365 TALLOC_FREE(print_name);
367 if (pflags != NULL) {
368 *pflags = flags;
370 return NT_STATUS_OK;
373 NTSTATUS cli_readlink(struct cli_state *cli, const char *fname,
374 TALLOC_CTX *mem_ctx, char **psubstitute_name,
375 char **pprint_name, uint32_t *pflags)
377 TALLOC_CTX *frame = talloc_stackframe();
378 struct event_context *ev;
379 struct tevent_req *req;
380 NTSTATUS status = NT_STATUS_NO_MEMORY;
382 if (smbXcli_conn_has_async_calls(cli->conn)) {
383 status = NT_STATUS_INVALID_PARAMETER;
384 goto fail;
386 ev = event_context_init(frame);
387 if (ev == NULL) {
388 goto fail;
390 req = cli_readlink_send(frame, ev, cli, fname);
391 if (req == NULL) {
392 goto fail;
394 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
395 goto fail;
397 status = cli_readlink_recv(req, mem_ctx, psubstitute_name,
398 pprint_name, pflags);
399 fail:
400 TALLOC_FREE(frame);
401 return status;