summaryrefslogtreecommitdiff
path: root/source4/libcli/smb_composite
diff options
context:
space:
mode:
authorVolker Lendecke <vlendec@samba.org>2005-10-02 10:02:35 +0000
committerGerald (Jerry) Carter <jerry@samba.org>2007-10-10 13:39:17 -0500
commit68c70ef396f84077dd08f97bf700f0c2963c8676 (patch)
treef54fb72fa4adbdc8bcf469f863b292de88d1a374 /source4/libcli/smb_composite
parent33834e2dda513681dc952e5a983cecb5871e3417 (diff)
downloadsamba-68c70ef396f84077dd08f97bf700f0c2963c8676.tar.gz
samba-68c70ef396f84077dd08f97bf700f0c2963c8676.tar.bz2
samba-68c70ef396f84077dd08f97bf700f0c2963c8676.zip
r10677: Add smb_composite_connectmulti: Send out multiple SYN packets at once, use the
first one that replies correctly. Add a talloc context to smb_composite_connect() Volker (This used to be commit 6b88de182e40cb00a833c085f801fd47c92bbe94)
Diffstat (limited to 'source4/libcli/smb_composite')
-rw-r--r--source4/libcli/smb_composite/connect.c44
-rw-r--r--source4/libcli/smb_composite/connect_multi.c188
-rw-r--r--source4/libcli/smb_composite/fetchfile.c2
-rw-r--r--source4/libcli/smb_composite/fsinfo.c3
-rw-r--r--source4/libcli/smb_composite/smb_composite.h18
5 files changed, 245 insertions, 10 deletions
diff --git a/source4/libcli/smb_composite/connect.c b/source4/libcli/smb_composite/connect.c
index 56d9d988c3..53cc8a9ac0 100644
--- a/source4/libcli/smb_composite/connect.c
+++ b/source4/libcli/smb_composite/connect.c
@@ -25,6 +25,7 @@
#include "libcli/raw/libcliraw.h"
#include "libcli/composite/composite.h"
#include "libcli/smb_composite/smb_composite.h"
+#include "lib/events/events.h"
/* the stages of this call */
enum connect_stage {CONNECT_RESOLVE,
@@ -39,6 +40,7 @@ struct connect_state {
struct smbcli_socket *sock;
struct smbcli_transport *transport;
struct smbcli_session *session;
+ struct smb_composite_connectmulti *conn;
struct smb_composite_connect *io;
union smb_tcon *io_tcon;
struct smb_composite_sesssetup *io_setup;
@@ -213,9 +215,11 @@ static NTSTATUS connect_socket(struct composite_context *c,
NTSTATUS status;
struct nbt_name calling, called;
- status = smbcli_sock_connect_recv(state->creq);
+ status = smb_composite_connectmulti_recv(state->creq, state);
NT_STATUS_NOT_OK_RETURN(status);
+ state->sock = state->conn->out.socket;
+
/* the socket is up - we can initialise the smbcli transport layer */
state->transport = smbcli_transport_init(state->sock, state, True);
NT_STATUS_HAVE_NO_MEMORY(state->transport);
@@ -254,11 +258,33 @@ static NTSTATUS connect_resolve(struct composite_context *c,
struct connect_state *state = talloc_get_type(c->private_data, struct connect_state);
NTSTATUS status;
const char *address;
+ struct smb_composite_connectmulti *conn;
status = resolve_name_recv(state->creq, state, &address);
NT_STATUS_NOT_OK_RETURN(status);
- state->creq = smbcli_sock_connect_send(state->sock, address, state->io->in.port, io->in.dest_host);
+ conn = talloc(state, struct smb_composite_connectmulti);
+ NT_STATUS_HAVE_NO_MEMORY(conn);
+ state->conn = conn;
+ conn->in.num_dests = 1;
+
+ conn->in.addresses = talloc_array(state->conn, const char *, 1);
+ NT_STATUS_HAVE_NO_MEMORY(conn->in.addresses);
+ conn->in.addresses[0] = address;
+
+ conn->in.hostnames = talloc_array(state->conn, const char *, 1);
+ NT_STATUS_HAVE_NO_MEMORY(conn->in.hostnames);
+ conn->in.hostnames[0] = state->io->in.dest_host;
+
+ conn->in.ports = NULL;
+ if (state->io->in.port != 0) {
+ conn->in.ports = talloc_array(state->conn, int, 1);
+ NT_STATUS_HAVE_NO_MEMORY(conn->in.ports);
+ conn->in.ports[0] = state->io->in.port;
+ }
+
+ state->creq = smb_composite_connectmulti_send(conn, state,
+ c->event_ctx);
NT_STATUS_HAVE_NO_MEMORY(state->creq);
state->stage = CONNECT_SOCKET;
@@ -332,25 +358,27 @@ static void composite_handler(struct composite_context *creq)
a function to establish a smbcli_tree from scratch
*/
struct composite_context *smb_composite_connect_send(struct smb_composite_connect *io,
- struct event_context *event_ctx)
+ TALLOC_CTX *mem_ctx,
+ struct event_context *event_ctx)
{
struct composite_context *c;
struct connect_state *state;
struct nbt_name name;
- c = talloc_zero(NULL, struct composite_context);
+ c = talloc_zero(mem_ctx, struct composite_context);
if (c == NULL) goto failed;
state = talloc(c, struct connect_state);
if (state == NULL) goto failed;
- state->sock = smbcli_sock_init(state, event_ctx);
- if (state->sock == NULL) goto failed;
+ if (event_ctx == NULL) {
+ event_ctx = event_context_init(mem_ctx);
+ }
state->io = io;
c->state = COMPOSITE_STATE_IN_PROGRESS;
- c->event_ctx = talloc_reference(c, state->sock->event.ctx);
+ c->event_ctx = talloc_reference(c, event_ctx);
c->private_data = state;
state->stage = CONNECT_RESOLVE;
@@ -391,6 +419,6 @@ NTSTATUS smb_composite_connect_recv(struct composite_context *c, TALLOC_CTX *mem
NTSTATUS smb_composite_connect(struct smb_composite_connect *io, TALLOC_CTX *mem_ctx,
struct event_context *ev)
{
- struct composite_context *c = smb_composite_connect_send(io, ev);
+ struct composite_context *c = smb_composite_connect_send(io, mem_ctx, ev);
return smb_composite_connect_recv(c, mem_ctx);
}
diff --git a/source4/libcli/smb_composite/connect_multi.c b/source4/libcli/smb_composite/connect_multi.c
new file mode 100644
index 0000000000..a539ba4bc6
--- /dev/null
+++ b/source4/libcli/smb_composite/connect_multi.c
@@ -0,0 +1,188 @@
+/*
+ Unix SMB/CIFS implementation.
+
+ Copyright (C) Volker Lendecke
+
+ 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 2 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, write to the Free Software
+ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+/*
+ a composite API to fire connect calls to multiple targets, picking the first
+ one.
+*/
+
+#include "includes.h"
+#include "libcli/raw/libcliraw.h"
+#include "libcli/composite/composite.h"
+#include "libcli/smb_composite/smb_composite.h"
+
+struct connectmulti_state {
+ struct smb_composite_connectmulti *io;
+ struct composite_context *creq;
+ struct smbcli_socket *result;
+ int num_socks, socks_left;
+ struct smbcli_socket **socks;
+ struct composite_context **creqs;
+};
+
+static void connect_receive(struct composite_context *c)
+
+{
+ struct connectmulti_state *state =
+ talloc_get_type(c->async.private_data,
+ struct connectmulti_state);
+ int i;
+
+ for (i=0; i<state->num_socks; i++) {
+ if (state->creqs[i] == c) {
+ break;
+ }
+ }
+
+ if (i == state->num_socks) {
+ c->status = NT_STATUS_INTERNAL_ERROR;
+ c->state = COMPOSITE_STATE_ERROR;
+ if (state->creq->async.fn != NULL) {
+ state->creq->async.fn(state->creq);
+ }
+ return;
+ }
+
+ state->creq->status = smbcli_sock_connect_recv(c);
+ if (!NT_STATUS_IS_OK(state->creq->status)) {
+ talloc_free(state->socks[i]);
+ state->socks[i] = NULL;
+ state->creqs[i] = NULL;
+ state->socks_left -= 1;
+ if (state->socks_left == 0) {
+ state->creq->state = COMPOSITE_STATE_ERROR;
+ if (state->creq->async.fn != NULL) {
+ state->creq->async.fn(state->creq);
+ }
+ }
+ return;
+ }
+
+ state->result = talloc_steal(state, state->socks[i]);
+ talloc_free(state->socks);
+
+ state->creq->state = COMPOSITE_STATE_DONE;
+ if (state->creq->async.fn != NULL) {
+ state->creq->async.fn(state->creq);
+ }
+}
+
+struct composite_context *smb_composite_connectmulti_send(struct smb_composite_connectmulti *io,
+ TALLOC_CTX *mem_ctx,
+ struct event_context *event_ctx)
+{
+ struct composite_context *c;
+ struct connectmulti_state *state;
+ int num_socks = io->in.num_dests;
+ const char **hostnames = io->in.hostnames;
+ const char **addresses = io->in.addresses;
+ int *ports = io->in.ports;
+ int i;
+
+ c = talloc_zero(mem_ctx, struct composite_context);
+ if (c == NULL) goto failed;
+
+ state = talloc(c, struct connectmulti_state);
+ if (state == NULL) goto failed;
+
+ c->state = COMPOSITE_STATE_IN_PROGRESS;
+ c->event_ctx = talloc_reference(c, event_ctx);
+ c->private_data = state;
+
+ if (ports == NULL) {
+ int j, nports;
+ const char **smb_ports = lp_smb_ports();
+
+ for (nports=0; smb_ports[nports]; nports++) /* noop */;
+
+ num_socks *= nports;
+ hostnames = talloc_array(state, const char *, num_socks);
+ if (hostnames == NULL) goto failed;
+ addresses = talloc_array(state, const char *, num_socks);
+ if (addresses == NULL) goto failed;
+ ports = talloc_array(state, int, num_socks);
+ if (ports == NULL) goto failed;
+
+ for (i=0; i<io->in.num_dests; i++) {
+ for (j=0; j<nports; j++) {
+ hostnames[i*nports+j] = io->in.hostnames[i];
+ addresses[i*nports+j] = io->in.addresses[i];
+ ports[i*nports+j] = atoi(smb_ports[j]);
+ }
+ }
+ }
+
+ state->io = io;
+ state->creq = c;
+ state->num_socks = num_socks;
+ state->socks_left = num_socks;
+ state->socks = talloc_array(state, struct smbcli_socket *, num_socks);
+ state->creqs = talloc_array(state, struct composite_context *,
+ num_socks);
+ if ((state->socks == NULL) || (state->creqs == NULL)) goto failed;
+
+ for (i=0; i<num_socks; i++) {
+ state->socks[i] = smbcli_sock_init(state->socks, event_ctx);
+ if (state->socks[i] == NULL) goto failed;
+
+ /* If the event_ctx we got given is NULL, the first socket
+ * creates one and all others need to refer to it. */
+ event_ctx = state->socks[i]->event.ctx;
+
+ state->creqs[i] = smbcli_sock_connect_send(state->socks[i],
+ addresses[i],
+ ports[i],
+ hostnames[i]);
+ if (state->creqs[i] == NULL) goto failed;
+ state->creqs[i]->async.fn = connect_receive;
+ state->creqs[i]->async.private_data = state;
+ }
+
+ return c;
+
+ failed:
+ talloc_free(c);
+ return NULL;
+}
+
+NTSTATUS smb_composite_connectmulti_recv(struct composite_context *c,
+ TALLOC_CTX *mem_ctx)
+{
+ NTSTATUS status;
+ status = composite_wait(c);
+
+ if (NT_STATUS_IS_OK(status)) {
+ struct connectmulti_state *state =
+ talloc_get_type(c->private_data,
+ struct connectmulti_state);
+ state->io->out.socket = talloc_steal(mem_ctx, state->result);
+ }
+
+ talloc_free(c);
+ return status;
+}
+
+NTSTATUS smb_composite_connectmulti(struct smb_composite_connectmulti *io,
+ TALLOC_CTX *mem_ctx,
+ struct event_context *ev)
+{
+ struct composite_context *c =
+ smb_composite_connectmulti_send(io, mem_ctx, ev);
+ return smb_composite_connectmulti_recv(c, mem_ctx);
+}
diff --git a/source4/libcli/smb_composite/fetchfile.c b/source4/libcli/smb_composite/fetchfile.c
index 9e88fb669d..8aa91bf3a9 100644
--- a/source4/libcli/smb_composite/fetchfile.c
+++ b/source4/libcli/smb_composite/fetchfile.c
@@ -146,7 +146,7 @@ struct composite_context *smb_composite_fetchfile_send(struct smb_composite_fetc
state->connect->in.credentials = io->in.credentials;
state->connect->in.workgroup = io->in.workgroup;
- state->creq = smb_composite_connect_send(state->connect, event_ctx);
+ state->creq = smb_composite_connect_send(state->connect, state, event_ctx);
if (state->creq == NULL) goto failed;
state->creq->async.private_data = c;
diff --git a/source4/libcli/smb_composite/fsinfo.c b/source4/libcli/smb_composite/fsinfo.c
index e870225906..fa9f18d132 100644
--- a/source4/libcli/smb_composite/fsinfo.c
+++ b/source4/libcli/smb_composite/fsinfo.c
@@ -156,7 +156,8 @@ struct composite_context *smb_composite_fsinfo_send(struct smbcli_tree *tree,
c->event_ctx = talloc_reference(c, tree->session->transport->socket->event.ctx);
c->private_data = state;
- state->creq = smb_composite_connect_send(state->connect, c->event_ctx);
+ state->creq = smb_composite_connect_send(state->connect, state,
+ c->event_ctx);
if (state->creq == NULL) goto failed;
diff --git a/source4/libcli/smb_composite/smb_composite.h b/source4/libcli/smb_composite/smb_composite.h
index 08031c2f4b..ec3a7af22d 100644
--- a/source4/libcli/smb_composite/smb_composite.h
+++ b/source4/libcli/smb_composite/smb_composite.h
@@ -150,3 +150,21 @@ struct smb_composite_appendacl {
struct security_descriptor *sd;
} out;
};
+
+/*
+ a composite API to fire connect() calls to multiple targets, picking the
+ first one.
+*/
+
+struct smb_composite_connectmulti {
+ struct {
+ int num_dests;
+ const char **hostnames;
+ const char **addresses;
+ int *ports; /* Either NULL for lp_smb_ports() per
+ * destination or a list of explicit ports */
+ } in;
+ struct {
+ struct smbcli_socket *socket;
+ } out;
+};