s3:libsmb: add an option to cli_push to let the caller provide the buffers

metze

diff --git a/source3/client/client.c b/source3/client/client.c
index aaa9e35d9689bdc8768db5b99f81aa90cb4cce78..67a2458a94c860a9f146541e0ea55b756bbf52d3 100644 (file)
--- a/source3/client/client.c
+++ b/source3/client/client.c
@@ -220,7 +220,9 @@ struct push_state {
        SMB_OFF_T nread;
 };
 
-static size_t push_source(uint8_t *buf, size_t n, void *priv)
+static size_t push_source(uint8_t *inbuf, size_t n,
+                         const uint8_t **outbuf,
+                         void *priv)
 {
        struct push_state *state = (struct push_state *)priv;
        int result;
@@ -229,7 +231,7 @@ static size_t push_source(uint8_t *buf, size_t n, void *priv)
                return 0;
        }
 
-       result = readfile(buf, n, state->f);
+       result = readfile(inbuf, n, state->f);
        state->nread += result;
        return result;
 }
@@ -1681,8 +1683,8 @@ static int do_put(const char *rname, const char *lname, bool reput)
        state.f = f;
        state.nread = 0;
 
-       status = cli_push(targetcli, fnum, 0, 0, io_bufsize, push_source,
-                         &state);
+       status = cli_push(targetcli, fnum, 0, 0, io_bufsize,
+                         false, push_source, &state);
        if (!NT_STATUS_IS_OK(status)) {
                d_fprintf(stderr, "cli_push returned %s\n", nt_errstr(status));
        }

diff --git a/source3/include/proto.h b/source3/include/proto.h
index a1cafb6837f63da6298818e0f6ca3130156eadf5..794a006a687a6d839c0cde79d1d01ffe44c6b245 100644 (file)
--- a/source3/include/proto.h
+++ b/source3/include/proto.h
@@ -2790,13 +2790,18 @@ struct async_req *cli_push_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
                                struct cli_state *cli,
                                uint16_t fnum, uint16_t mode,
                                off_t start_offset, size_t window_size,
-                               size_t (*source)(uint8_t *buf, size_t n,
+                               bool caller_buffers,
+                               size_t (*source)(uint8_t *inbuf, size_t n,
+                                                const uint8_t **outbuf,
                                                 void *priv),
                                void *priv);
 NTSTATUS cli_push_recv(struct async_req *req);
 NTSTATUS cli_push(struct cli_state *cli, uint16_t fnum, uint16_t mode,
                  off_t start_offset, size_t window_size,
-                 size_t (*source)(uint8_t *buf, size_t n, void *priv),
+                 bool caller_buffers,
+                 size_t (*source)(uint8_t *inbuf, size_t n,
+                                  const uint8_t **outbuf,
+                                  void *priv),
                  void *priv);
 
 /* The following definitions come from libsmb/clisecdesc.c  */

diff --git a/source3/libsmb/clireadwrite.c b/source3/libsmb/clireadwrite.c
index f2f447b4c95661f7d87190473afd6f4b200cc073..7e7cf0d682129a26c173f99061a9cb671ff72704 100644 (file)
--- a/source3/libsmb/clireadwrite.c
+++ b/source3/libsmb/clireadwrite.c
@@ -930,8 +930,11 @@ struct cli_push_state {
        uint16_t mode;
        off_t start_offset;
        size_t window_size;
+       bool caller_buffers;
 
-       size_t (*source)(uint8_t *buf, size_t n, void *priv);
+       size_t (*source)(uint8_t *inbuf, size_t n,
+                        const uint8_t **outbuf,
+                        void *priv);
        void *priv;
 
        bool eof;
@@ -963,13 +966,21 @@ static bool cli_push_write_setup(struct async_req *req,
        substate->req = req;
        substate->idx = idx;
        substate->ofs = state->next_offset;
-       substate->buf = talloc_array(substate, uint8_t, state->chunk_size);
-       if (!substate->buf) {
-               talloc_free(substate);
-               return false;
+       if (state->caller_buffers) {
+               substate->buf = NULL;
+       } else {
+               substate->buf = talloc_array(substate, uint8_t,
+                                            state->chunk_size);
+               if (!substate->buf) {
+                       talloc_free(substate);
+                       return false;
+               }
        }
+
+       /* source function can overwrite substate->buf... */
        substate->size = state->source(substate->buf,
                                       state->chunk_size,
+                                      (const uint8_t **)&substate->buf,
                                       state->priv);
        if (substate->size == 0) {
                state->eof = true;
@@ -1002,7 +1013,9 @@ struct async_req *cli_push_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
                                struct cli_state *cli,
                                uint16_t fnum, uint16_t mode,
                                off_t start_offset, size_t window_size,
-                               size_t (*source)(uint8_t *buf, size_t n,
+                               bool caller_buffers,
+                               size_t (*source)(uint8_t *inbuf, size_t n,
+                                                const uint8_t **outbuf,
                                                 void *priv),
                                void *priv)
 {
@@ -1019,6 +1032,7 @@ struct async_req *cli_push_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
        state->fnum = fnum;
        state->start_offset = start_offset;
        state->mode = mode;
+       state->caller_buffers = caller_buffers;
        state->source = source;
        state->priv = priv;
        state->eof = false;
@@ -1108,7 +1122,10 @@ NTSTATUS cli_push_recv(struct async_req *req)
 
 NTSTATUS cli_push(struct cli_state *cli, uint16_t fnum, uint16_t mode,
                  off_t start_offset, size_t window_size,
-                 size_t (*source)(uint8_t *buf, size_t n, void *priv),
+                 bool caller_buffers,
+                 size_t (*source)(uint8_t *inbuf, size_t n,
+                                  const uint8_t **outbuf,
+                                  void *priv),
                  void *priv)
 {
        TALLOC_CTX *frame = talloc_stackframe();
@@ -1129,7 +1146,7 @@ NTSTATUS cli_push(struct cli_state *cli, uint16_t fnum, uint16_t mode,
        }
 
        req = cli_push_send(frame, ev, cli, fnum, mode, start_offset,
-                           window_size, source, priv);
+                           window_size, caller_buffers, source, priv);
        if (req == NULL) {
                goto nomem;
        }

diff --git a/source3/torture/torture.c b/source3/torture/torture.c
index db89b05603aea293f5c6a079b6b28599eed4ec9f..a563557d5f49bd4ec612c39d6ba8dcd6afa5cba4 100644 (file)
--- a/source3/torture/torture.c
+++ b/source3/torture/torture.c
@@ -5004,7 +5004,9 @@ static bool run_chain1(int dummy)
        return True;
 }
 
-static size_t null_source(uint8_t *buf, size_t n, void *priv)
+static size_t null_source(uint8_t *inbuf, size_t n,
+                         const uint8_t *outbuf,
+                         void *priv)
 {
        size_t *to_pull = (size_t *)priv;
        size_t thistime = *to_pull;
@@ -5014,7 +5016,7 @@ static size_t null_source(uint8_t *buf, size_t n, void *priv)
                return 0;
        }
 
-       memset(buf, 0, thistime);
+       memset(inbuf, 0, thistime);
        *to_pull -= thistime;
        return thistime;
 }
@@ -5057,7 +5059,7 @@ static bool run_windows_write(int dummy)
                }
 
                status = cli_push(cli1, fnum, 0, i * torture_blocksize, torture_blocksize,
-                                 null_source, &to_pull);
+                                 false, null_source, &to_pull);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("cli_push returned: %s\n", nt_errstr(status));
                        goto fail;