14862a39f068c29138cd63cb7f4e758210ec48be
[jelmer/samba4-debian.git] / source / libcli / raw / clisocket.c
1 /* 
2    Unix SMB/CIFS implementation.
3    SMB client socket context management functions
4    Copyright (C) Andrew Tridgell 1994-2003
5    Copyright (C) James Myers 2003 <myersjj@samba.org>
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 /*
25   destroy a socket
26  */
27 static int sock_destructor(void *ptr)
28 {
29         struct smbcli_socket *sock = ptr;
30         if (sock->fd != -1) {
31                 close(sock->fd);
32                 sock->fd = -1;
33         }
34         return 0;
35 }
36
37 /*
38   create a smbcli_socket context
39 */
40 struct smbcli_socket *smbcli_sock_init(TALLOC_CTX *mem_ctx)
41 {
42         struct smbcli_socket *sock;
43
44         sock = talloc_p(mem_ctx, struct smbcli_socket);
45         if (!sock) {
46                 return NULL;
47         }
48
49         ZERO_STRUCTP(sock);
50         sock->fd = -1;
51         sock->port = 0;
52
53         /* 20 second default timeout */
54         sock->timeout = 20000;
55         sock->hostname = NULL;
56
57         talloc_set_destructor(sock, sock_destructor);
58
59         return sock;
60 }
61
62 /*
63   connect a smbcli_socket context to an IP/port pair
64   if port is 0 then choose 445 then 139
65 */
66 BOOL smbcli_sock_connect(struct smbcli_socket *sock, struct in_addr *ip, int port)
67 {
68         if (getenv("LIBSMB_PROG")) {
69                 sock->fd = sock_exec(getenv("LIBSMB_PROG"));
70                 return sock->fd != -1;
71         }
72
73         if (port == 0) {
74                 int i;
75                 const char **ports = lp_smb_ports();
76                 for (i=0;ports[i];i++) {
77                         port = atoi(ports[i]);
78                         if (port != 0 && smbcli_sock_connect(sock, ip, port)) {
79                                 return True;
80                         }
81                 }
82                 return False;
83         }
84
85         sock->dest_ip = *ip;
86         sock->port = port;
87         sock->fd = open_socket_out(SOCK_STREAM,
88                                    &sock->dest_ip,
89                                    sock->port, 
90                                    LONG_CONNECT_TIMEOUT);
91         if (sock->fd == -1) {
92                 return False;
93         }
94
95         set_blocking(sock->fd, False);
96         set_socket_options(sock->fd, lp_socket_options());
97
98         return True;
99 }
100
101
102 /****************************************************************************
103  mark the socket as dead
104 ****************************************************************************/
105 void smbcli_sock_dead(struct smbcli_socket *sock)
106 {
107         if (sock->fd != -1) {
108                 close(sock->fd);
109                 sock->fd = -1;
110         }
111 }
112
113 /****************************************************************************
114  Set socket options on a open connection.
115 ****************************************************************************/
116 void smbcli_sock_set_options(struct smbcli_socket *sock, const char *options)
117 {
118         set_socket_options(sock->fd, options);
119 }
120
121 /****************************************************************************
122  Write to socket. Return amount written.
123 ****************************************************************************/
124 ssize_t smbcli_sock_write(struct smbcli_socket *sock, const char *data, size_t len)
125 {
126         if (sock->fd == -1) {
127                 errno = EIO;
128                 return -1;
129         }
130
131         return write(sock->fd, data, len);
132 }
133
134
135 /****************************************************************************
136  Read from socket. return amount read
137 ****************************************************************************/
138 ssize_t smbcli_sock_read(struct smbcli_socket *sock, char *data, size_t len)
139 {
140         if (sock->fd == -1) {
141                 errno = EIO;
142                 return -1;
143         }
144
145         return read(sock->fd, data, len);
146 }
147
148 /****************************************************************************
149 resolve a hostname and connect 
150 ****************************************************************************/
151 BOOL smbcli_sock_connect_byname(struct smbcli_socket *sock, const char *host, int port)
152 {
153         int name_type = 0x20;
154         struct in_addr ip;
155         char *name, *p;
156         BOOL ret;
157
158         if (getenv("LIBSMB_PROG")) {
159                 sock->fd = sock_exec(getenv("LIBSMB_PROG"));
160                 return sock->fd != -1;
161         }
162
163         name = talloc_strdup(sock, host);
164
165         /* allow hostnames of the form NAME#xx and do a netbios lookup */
166         if ((p = strchr(name, '#'))) {
167                 name_type = strtol(p+1, NULL, 16);
168                 *p = 0;
169         }
170
171         if (!resolve_name(name, name, &ip, name_type)) {
172                 talloc_free(name);
173                 return False;
174         }
175
176         ret = smbcli_sock_connect(sock, &ip, port);
177
178         if (ret) {
179                 sock->hostname = talloc_steal(sock, name);
180         } else {
181                 talloc_free(name);
182         }
183
184         return ret;
185 }