sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2024-12-23 17:34:34 +03:00
Code

s4:libcli/smb_composite: convert smb2_composite_setpathinfo_send/rev to tevent_req

Browse Source 
metze
This commit is contained in:
Stefan Metzmacher 2011-05-01 18:10:20 +02:00
parent 1dbe9706bc
commit 8d12649b31
2 changed files with 130 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

186
source4/libcli/smb_composite/smb2.c
View File

@ -25,6 +25,8 @@
#include "includes.h"
#include <tevent.h>
#include "lib/util/tevent_ntstatus.h"
#include "libcli/raw/libcliraw.h"
#include "libcli/raw/raw_proto.h"
#include "libcli/composite/composite.h"
@ -264,107 +266,145 @@ NTSTATUS smb2_composite_rmdir(struct smb2_tree *tree, struct smb_rmdir *io)
return composite_wait_free(c);
}
struct smb2_composite_setpathinfo_state {
struct smb2_tree *tree;
union smb_setfileinfo io;
NTSTATUS set_status;
struct smb2_create cr;
struct smb2_close cl;
};
static void smb2_composite_setpathinfo_create_done(struct smb2_request *smb2req);
/*
composite SMB2 setpathinfo call
*/
struct composite_context *smb2_composite_setpathinfo_send(struct smb2_tree *tree,
union smb_setfileinfo *io)
struct tevent_req *smb2_composite_setpathinfo_send(TALLOC_CTX *mem_ctx,
struct tevent_context *ev,
struct smb2_tree *tree,
const union smb_setfileinfo *io)
{
struct composite_context *ctx;
struct smb2_create create_parm;
struct tevent_req *req;
struct smb2_composite_setpathinfo_state *state;
struct smb2_request *smb2req;
union smb_setfileinfo *io2;
ctx = composite_create(tree, tree->session->transport->socket->event.ctx);
if (ctx == NULL) return NULL;
req = tevent_req_create(mem_ctx, &state,
struct smb2_composite_setpathinfo_state);
if (req == NULL) {
return NULL;
}
ZERO_STRUCT(create_parm);
create_parm.in.desired_access = SEC_FLAG_MAXIMUM_ALLOWED;
create_parm.in.create_disposition = NTCREATEX_DISP_OPEN;
create_parm.in.share_access =
state->tree = tree;
state->io = *io;
state->cr.in.desired_access = SEC_FLAG_MAXIMUM_ALLOWED;
state->cr.in.create_disposition = NTCREATEX_DISP_OPEN;
state->cr.in.share_access =
NTCREATEX_SHARE_ACCESS_DELETE|
NTCREATEX_SHARE_ACCESS_READ|
NTCREATEX_SHARE_ACCESS_WRITE;
create_parm.in.create_options = 0;
create_parm.in.fname = io->generic.in.file.path;
if (create_parm.in.fname[0] == '\\') {
create_parm.in.fname++;
state->cr.in.create_options = 0;
state->cr.in.fname = state->io.generic.in.file.path;
if (state->cr.in.fname[0] == '\\') {
state->cr.in.fname++;
}
smb2req = smb2_create_send(tree, &create_parm);
io2 = talloc(ctx, union smb_setfileinfo);
if (composite_nomem(io2, ctx)) {
return ctx;
smb2req = smb2_create_send(tree, &state->cr);
if (tevent_req_nomem(smb2req, req)) {
return tevent_req_post(req, ev);
}
*io2 = *io;
smb2req->async.fn = smb2_composite_setpathinfo_create_done;
smb2req->async.private_data = req;
ctx->private_data = io2;
composite_continue_smb2(ctx, smb2req, smb2_composite_setpathinfo_create_done, ctx);
return ctx;
return req;
}
static void smb2_composite_setpathinfo_setinfo_done(struct smb2_request *smb2req);
static void smb2_composite_setpathinfo_create_done(struct smb2_request *smb2req)
{
struct composite_context *ctx = talloc_get_type(smb2req->async.private_data,
struct composite_context);
struct smb2_tree *tree = smb2req->tree;
struct smb2_create create_parm;
struct tevent_req *req =
talloc_get_type_abort(smb2req->async.private_data,
struct tevent_req);
struct smb2_composite_setpathinfo_state *state =
tevent_req_data(req,
struct smb2_composite_setpathinfo_state);
NTSTATUS status;
union smb_setfileinfo *io2 = talloc_get_type(ctx->private_data,
union smb_setfileinfo);
status = smb2_create_recv(smb2req, ctx, &create_parm);
if (!NT_STATUS_IS_OK(status)) {
composite_error(ctx, status);
status = smb2_create_recv(smb2req, state, &state->cr);
if (tevent_req_nterror(req, status)) {
return;
}
io2->generic.in.file.handle = create_parm.out.file.handle;
state->io.generic.in.file.handle = state->cr.out.file.handle;
smb2req = smb2_setinfo_file_send(tree, io2);
composite_continue_smb2(ctx, smb2req, smb2_composite_setpathinfo_setinfo_done, ctx);
smb2req = smb2_setinfo_file_send(state->tree, &state->io);
if (tevent_req_nomem(smb2req, req)) {
return;
}
smb2req->async.fn = smb2_composite_setpathinfo_setinfo_done;
smb2req->async.private_data = req;
}
static void smb2_composite_setpathinfo_close_done(struct smb2_request *smb2req);
static void smb2_composite_setpathinfo_setinfo_done(struct smb2_request *smb2req)
{
struct composite_context *ctx = talloc_get_type(smb2req->async.private_data,
struct composite_context);
struct smb2_tree *tree = smb2req->tree;
struct smb2_close close_parm;
struct tevent_req *req =
talloc_get_type_abort(smb2req->async.private_data,
struct tevent_req);
struct smb2_composite_setpathinfo_state *state =
tevent_req_data(req,
struct smb2_composite_setpathinfo_state);
NTSTATUS status;
union smb_setfileinfo *io2 = talloc_get_type(ctx->private_data,
union smb_setfileinfo);
status = smb2_setinfo_recv(smb2req);
if (!NT_STATUS_IS_OK(status)) {
composite_error(ctx, status);
state->set_status = status;
state->cl.in.file.handle = state->io.generic.in.file.handle;
smb2req = smb2_close_send(state->tree, &state->cl);
if (tevent_req_nomem(smb2req, req)) {
return;
}
ZERO_STRUCT(close_parm);
close_parm.in.file.handle = io2->generic.in.file.handle;
smb2req = smb2_close_send(tree, &close_parm);
composite_continue_smb2(ctx, smb2req, smb2_composite_setpathinfo_close_done, ctx);
smb2req->async.fn = smb2_composite_setpathinfo_close_done;
smb2req->async.private_data = req;
}
static void smb2_composite_setpathinfo_close_done(struct smb2_request *smb2req)
{
struct composite_context *ctx = talloc_get_type(smb2req->async.private_data,
struct composite_context);
struct tevent_req *req =
talloc_get_type_abort(smb2req->async.private_data,
struct tevent_req);
struct smb2_composite_setpathinfo_state *state =
tevent_req_data(req,
struct smb2_composite_setpathinfo_state);
NTSTATUS status;
struct smb2_close close_parm;
status = smb2_close_recv(smb2req, &close_parm);
composite_error(ctx, status);
status = smb2_close_recv(smb2req, &state->cl);
if (tevent_req_nterror(req, state->set_status)) {
return;
}
if (tevent_req_nterror(req, status)) {
return;
}
tevent_req_done(req);
}
NTSTATUS smb2_composite_setpathinfo_recv(struct tevent_req *req)
{
NTSTATUS status;
if (tevent_req_is_nterror(req, &status)) {
tevent_req_received(req);
return status;
}
tevent_req_received(req);
return NT_STATUS_OK;
}
/*
@ -372,6 +412,36 @@ static void smb2_composite_setpathinfo_close_done(struct smb2_request *smb2req)
*/
NTSTATUS smb2_composite_setpathinfo(struct smb2_tree *tree, union smb_setfileinfo *io)
{
struct composite_context *c = smb2_composite_setpathinfo_send(tree, io);
return composite_wait_free(c);
struct tevent_req *subreq;
NTSTATUS status;
bool ok;
TALLOC_CTX *frame = talloc_stackframe();
struct tevent_context *ev = tree->session->transport->socket->event.ctx;
if (frame == NULL) {
return NT_STATUS_NO_MEMORY;
}
subreq = smb2_composite_setpathinfo_send(frame, ev, tree, io);
if (subreq == NULL) {
TALLOC_FREE(frame);
return NT_STATUS_NO_MEMORY;
}
ok = tevent_req_poll(subreq, ev);
if (!ok) {
status = map_nt_error_from_unix(errno);
TALLOC_FREE(frame);
return status;
}
status = smb2_composite_setpathinfo_recv(subreq);
TALLOC_FREE(subreq);
if (!NT_STATUS_IS_OK(status)) {
TALLOC_FREE(frame);
return status;
}
TALLOC_FREE(frame);
return NT_STATUS_OK;
}

4
source4/libcli/wscript_build
View File

@ -36,8 +36,8 @@ bld.SAMBA_SUBSYSTEM('LIBCLI_COMPOSITE',
bld.SAMBA_SUBSYSTEM('LIBCLI_SMB_COMPOSITE',
source='smb_composite/loadfile.c smb_composite/savefile.c smb_composite/connect.c smb_composite/sesssetup.c smb_composite/fetchfile.c smb_composite/appendacl.c smb_composite/fsinfo.c smb_composite/smb2.c',
autoproto='smb_composite/proto.h',
deps='LIBCLI_SMB2',
public_deps='LIBCLI_COMPOSITE credentials gensec LIBCLI_RESOLVE'
deps='LIBCLI_SMB2 UTIL_TEVENT',
public_deps='LIBCLI_COMPOSITE credentials gensec LIBCLI_RESOLVE tevent'
)