2 Unix SMB/CIFS implementation.
4 SMB client socket context management functions
6 Copyright (C) Andrew Tridgell 1994-2005
7 Copyright (C) James Myers 2003 <myersjj@samba.org>
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
26 #include "libcli/raw/libcliraw.h"
27 #include "libcli/composite/composite.h"
30 this private structure is used during async connection handling
32 struct clisocket_connect {
34 struct smbcli_socket *sock;
35 const char *dest_host;
39 create a smbcli_socket context
41 struct smbcli_socket *smbcli_sock_init(TALLOC_CTX *mem_ctx)
43 struct smbcli_socket *sock;
45 sock = talloc_zero(mem_ctx, struct smbcli_socket);
50 sock->event.ctx = event_context_init(sock);
51 if (sock->event.ctx == NULL) {
59 static NTSTATUS smbcli_sock_connect_one(struct smbcli_socket *sock,
60 const char *hostaddr, int port);
63 handle socket write events during an async connect. These happen when the OS
64 has either completed the connect() or has returned an error
66 static void smbcli_sock_connect_handler(struct event_context *ev, struct fd_event *fde,
67 struct timeval t, uint16_t flags)
69 struct smbcli_composite *c = talloc_get_type(fde->private, struct smbcli_composite);
70 struct clisocket_connect *conn = talloc_get_type(c->private, struct clisocket_connect);
73 c->status = socket_connect_complete(conn->sock->sock, 0);
74 if (NT_STATUS_IS_OK(c->status)) {
75 socket_set_option(conn->sock->sock, lp_socket_options(), NULL);
76 conn->sock->hostname = talloc_strdup(conn->sock, conn->dest_host);
77 c->state = SMBCLI_REQUEST_DONE;
84 /* that port failed - try the next port */
85 for (i=c->stage+1;conn->iports[i];i++) {
87 c->status = smbcli_sock_connect_one(conn->sock,
90 if (NT_STATUS_IS_OK(c->status) ||
91 NT_STATUS_EQUAL(c->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
92 conn->sock->event.fde->private = c;
97 c->state = SMBCLI_REQUEST_ERROR;
105 try to connect to the given address/port
107 static NTSTATUS smbcli_sock_connect_one(struct smbcli_socket *sock,
108 const char *hostaddr, int port)
114 talloc_free(sock->sock);
118 if (sock->event.fde) {
119 event_remove_fd(sock->event.ctx, sock->event.fde);
120 sock->event.fde = NULL;
123 status = socket_create("ip", SOCKET_TYPE_STREAM, &sock->sock, 0);
124 if (!NT_STATUS_IS_OK(status)) {
128 talloc_steal(sock, sock->sock);
130 /* we initially look for write - see the man page on
131 non-blocking connect */
132 fde.fd = socket_get_fd(sock->sock);
133 fde.flags = EVENT_FD_WRITE;
134 fde.handler = smbcli_sock_connect_handler;
137 sock->event.fde = event_add_fd(sock->event.ctx, &fde);
139 set_blocking(fde.fd, False);
141 return socket_connect(sock->sock, NULL, 0, hostaddr, port, 0);
146 connect a smbcli_socket context to an IP/port pair
147 if port is 0 then choose 445 then 139
149 this is the async send side of the interface
151 struct smbcli_composite *smbcli_sock_connect_send(struct smbcli_socket *sock,
152 const char *host_addr, int port)
154 struct smbcli_composite *c;
155 struct clisocket_connect *conn;
158 c = talloc_zero(sock, struct smbcli_composite);
159 if (c == NULL) return NULL;
161 c->event_ctx = sock->event.ctx;
163 conn = talloc(c, struct clisocket_connect);
164 if (conn == NULL) goto failed;
168 /* work out what ports we will try */
170 const char **ports = lp_smb_ports();
171 for (i=0;ports[i];i++) /* noop */ ;
172 conn->iports = talloc_array(c, int, i+1);
173 if (conn->iports == NULL) goto failed;
174 for (i=0;ports[i];i++) {
175 conn->iports[i] = atoi(ports[i]);
179 conn->iports = talloc_array(c, int, 2);
180 if (conn->iports == NULL) goto failed;
181 conn->iports[0] = port;
185 conn->dest_host = talloc_strdup(c, host_addr);
186 if (conn->dest_host == NULL) goto failed;
189 c->state = SMBCLI_REQUEST_SEND;
191 /* startup the connect process for each port in turn until one
192 succeeds or tells us that it is pending */
193 for (i=0;conn->iports[i];i++) {
195 conn->sock->port = conn->iports[i];
196 c->status = smbcli_sock_connect_one(sock,
199 if (NT_STATUS_IS_OK(c->status) ||
200 NT_STATUS_EQUAL(c->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
201 sock->event.fde->private = c;
206 c->state = SMBCLI_REQUEST_ERROR;
215 finish a smbcli_sock_connect_send() operation
217 NTSTATUS smbcli_sock_connect_recv(struct smbcli_composite *c)
220 status = smb_composite_wait(c);
226 connect a smbcli_socket context to an IP/port pair
227 if port is 0 then choose the ports listed in smb.conf (normally 445 then 139)
229 sync version of the function
231 NTSTATUS smbcli_sock_connect(struct smbcli_socket *sock, const char *host_addr, int port)
233 struct smbcli_composite *c;
235 c = smbcli_sock_connect_send(sock, host_addr, port);
237 return NT_STATUS_NO_MEMORY;
240 return smbcli_sock_connect_recv(c);
244 /****************************************************************************
245 mark the socket as dead
246 ****************************************************************************/
247 void smbcli_sock_dead(struct smbcli_socket *sock)
249 if (sock->sock != NULL) {
250 talloc_free(sock->sock);
255 /****************************************************************************
256 Set socket options on a open connection.
257 ****************************************************************************/
258 void smbcli_sock_set_options(struct smbcli_socket *sock, const char *options)
260 socket_set_option(sock->sock, options, NULL);
263 /****************************************************************************
264 Write to socket. Return amount written.
265 ****************************************************************************/
266 ssize_t smbcli_sock_write(struct smbcli_socket *sock, const uint8_t *data, size_t len)
272 if (sock->sock == NULL) {
277 blob.data = discard_const(data);
280 status = socket_send(sock->sock, &blob, &nsent, 0);
281 if (NT_STATUS_IS_ERR(status)) {
289 /****************************************************************************
290 Read from socket. return amount read
291 ****************************************************************************/
292 ssize_t smbcli_sock_read(struct smbcli_socket *sock, uint8_t *data, size_t len)
297 if (sock->sock == NULL) {
302 status = socket_recv(sock->sock, data, len, &nread, 0);
303 if (NT_STATUS_IS_ERR(status)) {
311 /****************************************************************************
312 resolve a hostname and connect
313 ****************************************************************************/
314 BOOL smbcli_sock_connect_byname(struct smbcli_socket *sock, const char *host, int port)
316 int name_type = 0x20;
321 name = talloc_strdup(sock, host);
323 /* allow hostnames of the form NAME#xx and do a netbios lookup */
324 if ((p = strchr(name, '#'))) {
325 name_type = strtol(p+1, NULL, 16);
329 if (!resolve_name(name, name, &ip, name_type)) {
333 sock->hostname = name;
335 status = smbcli_sock_connect(sock, sys_inet_ntoa(ip), port);
337 return NT_STATUS_IS_OK(status);