sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-03-11 16:58:40 +03:00
Code

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

Browse Source 
metze
This commit is contained in:
Stefan Metzmacher 2009-03-10 12:34:20 +01:00
parent c2993f74af
commit 9579a6f193
4 changed files with 43 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
source3/client/client.c
View File

@ -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));
}

9
source3/include/proto.h
View File

@ -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 */

33
source3/libsmb/clireadwrite.c
View File

@ -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;
}

8
source3/torture/torture.c
View File

@ -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;