From bbe8e0bcf47022584f4d908adf52dec522f3d736 Mon Sep 17 00:00:00 2001 From: Volker Lendecke Date: Thu, 23 Jun 2011 16:49:17 +0200 Subject: s3: Add cli_symlink --- source3/Makefile.in | 1 + source3/libsmb/clisymlink.c | 228 ++++++++++++++++++++++++++++++++++++++++++++ source3/libsmb/proto.h | 12 +++ source3/wscript_build | 1 + 4 files changed, 242 insertions(+) create mode 100644 source3/libsmb/clisymlink.c diff --git a/source3/Makefile.in b/source3/Makefile.in index 2763277bbc..b37f3158ed 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -591,6 +591,7 @@ LIBSMB_OBJ = libsmb/clientgen.o libsmb/cliconnect.o libsmb/clifile.o \ ../libcli/auth/spnego_parse.o \ ../lib/util/asn1.o \ libsmb/reparse_symlink.o \ + libsmb/clisymlink.o \ libsmb/clirap.o libsmb/clierror.o libsmb/climessage.o \ libsmb/clireadwrite.o libsmb/clilist.o libsmb/cliprint.o \ libsmb/clitrans.o libsmb/clisecdesc.o libsmb/clidgram.o \ diff --git a/source3/libsmb/clisymlink.c b/source3/libsmb/clisymlink.c new file mode 100644 index 0000000000..e5c2c1a89d --- /dev/null +++ b/source3/libsmb/clisymlink.c @@ -0,0 +1,228 @@ +/* + * Unix SMB/CIFS implementation. + * Client implementation of setting symlinks using reparse points + * Copyright (C) Volker Lendecke 2011 + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include "includes.h" +#include "system/filesys.h" +#include "libsmb/libsmb.h" +#include "../lib/util/tevent_ntstatus.h" +#include "async_smb.h" +#include "libsmb/clirap.h" +#include "trans2.h" +#include "ntioctl.h" +#include "libcli/security/secdesc.h" +#include "libcli/security/security.h" + +struct cli_symlink_state { + struct tevent_context *ev; + struct cli_state *cli; + const char *oldpath; + const char *newpath; + uint32_t flags; + + uint16_t fnum; + + uint16_t setup[4]; + NTSTATUS set_reparse_status; +}; + +static void cli_symlink_create_done(struct tevent_req *subreq); +static void cli_symlink_set_reparse_done(struct tevent_req *subreq); +static void cli_symlink_delete_on_close_done(struct tevent_req *subreq); +static void cli_symlink_close_done(struct tevent_req *subreq); + +struct tevent_req *cli_symlink_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct cli_state *cli, + const char *oldpath, + const char *newpath, + uint32_t flags) +{ + struct tevent_req *req, *subreq; + struct cli_symlink_state *state; + + req = tevent_req_create(mem_ctx, &state, struct cli_symlink_state); + if (req == NULL) { + return NULL; + } + state->ev = ev; + state->cli = cli; + state->oldpath = oldpath; + state->newpath = newpath; + state->flags = flags; + + subreq = cli_ntcreate_send( + state, ev, cli, state->oldpath, 0, + SYNCHRONIZE_ACCESS|DELETE_ACCESS| + FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES, + FILE_ATTRIBUTE_NORMAL, FILE_SHARE_NONE, FILE_CREATE, + FILE_OPEN_REPARSE_POINT|FILE_SYNCHRONOUS_IO_NONALERT| + FILE_NON_DIRECTORY_FILE, 0); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, cli_symlink_create_done, req); + return req; +} + +static void cli_symlink_create_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct cli_symlink_state *state = tevent_req_data( + req, struct cli_symlink_state); + uint8_t *data; + size_t data_len; + NTSTATUS status; + + status = cli_ntcreate_recv(subreq, &state->fnum); + TALLOC_FREE(subreq); + if (tevent_req_nterror(req, status)) { + return; + } + + SIVAL(state->setup, 0, FSCTL_SET_REPARSE_POINT); + SSVAL(state->setup, 4, state->fnum); + SCVAL(state->setup, 6, 1); /* IsFcntl */ + SCVAL(state->setup, 7, 0); /* IsFlags */ + + if (!symlink_reparse_buffer_marshall( + state->newpath, NULL, state->flags, state, + &data, &data_len)) { + tevent_req_oom(req); + return; + } + + subreq = cli_trans_send(state, state->ev, state->cli, SMBnttrans, + NULL, -1, /* name, fid */ + NT_TRANSACT_IOCTL, 0, + state->setup, 4, 0, /* setup */ + NULL, 0, 0, /* param */ + data, data_len, 0); /* data */ + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, cli_symlink_set_reparse_done, req); +} + +static void cli_symlink_set_reparse_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct cli_symlink_state *state = tevent_req_data( + req, struct cli_symlink_state); + + state->set_reparse_status = cli_trans_recv( + subreq, NULL, NULL, + NULL, 0, NULL, /* rsetup */ + NULL, 0, NULL, /* rparam */ + NULL, 0, NULL); /* rdata */ + TALLOC_FREE(subreq); + + if (NT_STATUS_IS_OK(state->set_reparse_status)) { + subreq = cli_close_send(state, state->ev, state->cli, + state->fnum); + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, cli_symlink_close_done, req); + return; + } + subreq = cli_nt_delete_on_close_send( + state, state->ev, state->cli, state->fnum, true); + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, cli_symlink_delete_on_close_done, req); +} + +static void cli_symlink_delete_on_close_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct cli_symlink_state *state = tevent_req_data( + req, struct cli_symlink_state); + NTSTATUS status; + + status = cli_nt_delete_on_close_recv(subreq); + TALLOC_FREE(subreq); + + /* + * Ignore status, we can't do much anyway in case of failure + */ + + subreq = cli_close_send(state, state->ev, state->cli, state->fnum); + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, cli_symlink_close_done, req); +} + +static void cli_symlink_close_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct cli_symlink_state *state = tevent_req_data( + req, struct cli_symlink_state); + NTSTATUS status; + + status = cli_close_recv(subreq); + TALLOC_FREE(subreq); + + if (tevent_req_nterror(req, status)) { + return; + } + if (tevent_req_nterror(req, state->set_reparse_status)) { + return; + } + tevent_req_done(req); +} + +NTSTATUS cli_symlink_recv(struct tevent_req *req) +{ + return tevent_req_simple_recv_ntstatus(req); +} + +NTSTATUS cli_symlink(struct cli_state *cli, const char *oldname, + const char *newname, uint32_t flags) +{ + TALLOC_CTX *frame = talloc_stackframe(); + struct event_context *ev; + struct tevent_req *req; + NTSTATUS status = NT_STATUS_NO_MEMORY; + + if (cli_has_async_calls(cli)) { + status = NT_STATUS_INVALID_PARAMETER; + goto fail; + } + ev = event_context_init(frame); + if (ev == NULL) { + goto fail; + } + req = cli_symlink_send(frame, ev, cli, oldname, newname, flags); + if (req == NULL) { + goto fail; + } + if (!tevent_req_poll_ntstatus(req, ev, &status)) { + goto fail; + } + status = cli_symlink_recv(req); + fail: + TALLOC_FREE(frame); + return status; +} diff --git a/source3/libsmb/proto.h b/source3/libsmb/proto.h index 4d820d6bc6..a266f7cd41 100644 --- a/source3/libsmb/proto.h +++ b/source3/libsmb/proto.h @@ -860,4 +860,16 @@ bool symlink_reparse_buffer_parse( const uint8_t *src, size_t srclen, TALLOC_CTX *mem_ctx, char **psubstitute_name, char **pprint_name, uint32_t *pflags); +/* The following definitions come from libsmb/clisymlink.c */ + +struct tevent_req *cli_symlink_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct cli_state *cli, + const char *oldpath, + const char *newpath, + uint32_t flags); +NTSTATUS cli_symlink_recv(struct tevent_req *req); +NTSTATUS cli_symlink(struct cli_state *cli, const char *oldname, + const char *newname, uint32_t flags); + #endif /* _LIBSMB_PROTO_H_ */ diff --git a/source3/wscript_build b/source3/wscript_build index d071053d5e..ee8c4e2184 100755 --- a/source3/wscript_build +++ b/source3/wscript_build @@ -138,6 +138,7 @@ LIBSMB_SRC = '''libsmb/clientgen.c libsmb/cliconnect.c libsmb/clifile.c libsmb/smb_seal.c libsmb/async_smb.c libsmb/read_smb.c libsmb/cli_np_tstream.c libsmb/reparse_symlink.c + libsmb/clisymlink.c libsmb/smbsock_connect.c''' LIBMSRPC_SRC = ''' -- cgit