s3: Add cli_symlink

diff --git a/source3/Makefile.in b/source3/Makefile.in
index 2763277bbcef2d0e91b5a0279c37272ab52b26d3..b37f3158ed7422428962d424b8b2a752cf285541 100644 (file)
--- 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 (file)
index 0000000..e5c2c1a
--- /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 <http://www.gnu.org/licenses/>.
+ */
+
+#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 4d820d6bc676e4e9fc487aa486cdcbdaf0bacee2..a266f7cd41125dad0b8876d5fa433f57978107e5 100644 (file)
--- 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 d071053d5e654f07ad73ce522ac80e623783a450..ee8c4e218484497999415fe833b59cba1a1a19a8 100755 (executable)
--- 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 = '''