author | Mikael Hallendal <micke@imendio.com> |
Sun, 10 Jun 2007 18:14:31 +0200 | |
changeset 268 | 7eb1cedb3241 |
parent 161 | 05ddf1d0440f |
child 196 | 209bfa124066 |
permissions | -rw-r--r-- |
147 | 1 |
/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */ |
2 |
/* |
|
3 |
* Copyright (C) 2006 Imendio AB |
|
4 |
* |
|
5 |
* This program is free software; you can redistribute it and/or |
|
6 |
* modify it under the terms of the GNU Lesser General Public License as |
|
7 |
* published by the Free Software Foundation; either version 2 of the |
|
8 |
* License, or (at your option) any later version. |
|
9 |
* |
|
10 |
* This program is distributed in the hope that it will be useful, |
|
11 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
12 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
13 |
* Lesser General Public License for more details. |
|
14 |
* |
|
15 |
* You should have received a copy of the GNU Lesser General Public |
|
16 |
* License along with this program; if not, write to the |
|
17 |
* Free Software Foundation, Inc., 59 Temple Place - Suite 330, |
|
18 |
* Boston, MA 02111-1307, USA. |
|
19 |
*/ |
|
20 |
||
21 |
#include "lm-socket.h" |
|
22 |
||
149 | 23 |
#ifndef G_OS_WIN32 |
24 |
typedef int LmSock; |
|
25 |
#else /* G_OS_WIN32 */ |
|
26 |
typedef SOCKET LmSock; |
|
27 |
#endif /* G_OS_WIN32 */ |
|
28 |
||
29 |
/* FIXME: Integrate with the SSL stuff */ |
|
30 |
||
154 | 31 |
/* FIXME: IO Buffering both in/out here? */ |
32 |
||
149 | 33 |
struct _LmSocket { |
151 | 34 |
LmSock sock; |
154 | 35 |
GIOChannel *io_channel; |
36 |
guint io_watch; |
|
147 | 37 |
/* FIXME: Add the rest */ |
151 | 38 |
|
39 |
LmSSL *ssl; |
|
147 | 40 |
|
151 | 41 |
LmSocketFuncs funcs; |
42 |
||
43 |
LmSocketState state; |
|
149 | 44 |
|
151 | 45 |
gboolean is_blocking; |
147 | 46 |
|
151 | 47 |
gint ref; |
147 | 48 |
}; |
49 |
||
154 | 50 |
typedef void (* SocketDNSCallback) (LmSocket *socket, |
51 |
SocketDNSData *data, |
|
52 |
gboolean success); |
|
53 |
||
54 |
typedef struct { |
|
55 |
/* Used when socket tries to connect */ |
|
56 |
struct addrinfo *resolved_addrs; |
|
57 |
struct addrinfo *current_addr; |
|
58 |
||
59 |
/* -- Internal during DNS lookup -- */ |
|
60 |
LmSocket *socket; |
|
61 |
gchar *host; |
|
62 |
SocketDNSCallback callback; |
|
63 |
} SocketDNSData; |
|
64 |
||
65 |
static LmSocket * socket_create (void); |
|
66 |
static void socket_free (LmSocket *socket); |
|
67 |
static gboolean socket_channel_event (GIOChannel *source, |
|
68 |
GIOCondition condition, |
|
69 |
LmSocket *socket); |
|
70 |
static void socket_dns_lookup (LmSocket *socket, |
|
71 |
const gchar *host, |
|
72 |
SocketDNSCallback callback); |
|
73 |
static SocketDNSData * socket_dns_data_new (LmSocket *socket, |
|
74 |
const gchar *host, |
|
75 |
SocketDNSCallback callbac); |
|
76 |
static void socket_dns_data_free (SocketDNSData *data); |
|
77 |
static void socket_start_connect (LmSocket *socket, |
|
78 |
SocketDNSData *data, |
|
79 |
gboolean success); |
|
149 | 80 |
|
81 |
static LmSocket * |
|
82 |
socket_create (void) |
|
83 |
{ |
|
84 |
LmSocket *socket; |
|
85 |
||
86 |
socket = g_new0 (LmSocket, 1); |
|
87 |
socket->ref_count = 1; |
|
88 |
socket->is_blocking = FALSE; |
|
151 | 89 |
socket->state = LM_SOCKET_STATE_CLOSED; |
90 |
socket->ssl = NULL; |
|
149 | 91 |
|
92 |
return socket; |
|
93 |
} |
|
147 | 94 |
|
95 |
static void |
|
149 | 96 |
socket_free (LmSocket *socket) |
147 | 97 |
{ |
98 |
/* FIXME: Free up the rest of the memory */ |
|
151 | 99 |
|
154 | 100 |
if (socket->io_channel) { |
101 |
if (socket->io_watch != 0) { |
|
102 |
g_source_destroy (g_main_context_find_source_by_id (socket->context), |
|
103 |
socket->io_watch); |
|
104 |
socket->io_watch = 0; |
|
105 |
} |
|
106 |
||
107 |
g_io_channel_unref (socket->io_channel); |
|
108 |
socket->io_channel = NULL; |
|
109 |
||
110 |
socket->fd = -1; |
|
111 |
} |
|
112 |
||
151 | 113 |
if (socket->ssl) { |
114 |
_lm_ssl_unref (socket->ssl); |
|
115 |
} |
|
116 |
||
149 | 117 |
g_free (socket->host); |
154 | 118 |
g_free (socket); |
119 |
} |
|
149 | 120 |
|
154 | 121 |
static gboolean |
122 |
socket_channel_event (GIOChannel *source, |
|
123 |
GIOCondition condition, |
|
124 |
LmSocket *socket) |
|
125 |
{ |
|
161
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
126 |
if (condition & G_IO_IN) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
127 |
socket_signal_read_available (); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
128 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
129 |
if (condition & G_IO_OUT) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
130 |
socket_attempt_write (); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
131 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
132 |
if (condition & G_IO_ERR || |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
133 |
condition & G_IO_HUP) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
134 |
socket_disconnected (); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
135 |
} |
154 | 136 |
} |
137 |
||
138 |
static void |
|
161
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
139 |
socket_dns_lookup (LmSocket *socket, |
154 | 140 |
const gchar *host, |
141 |
SocketDNSCallback callback); |
|
142 |
{ |
|
143 |
SocketDNSData *data; |
|
144 |
struct addrinfo req; |
|
145 |
struct addrinfo ans; |
|
146 |
int err; |
|
147 |
||
148 |
/* FIXME: This should not be synchronous */ |
|
149 |
data = socket_dns_data_new (socket, host, callback); |
|
150 |
||
151 |
memset (&req, 0, sizeof (req)); |
|
152 |
req.ai_family = AF_UNSPEC; |
|
153 |
req.ai_socktype = SOCK_STREAM; |
|
154 |
req.ai_protocol = IPPROTO_TCP; |
|
155 |
||
156 |
err = getaddrinfo (data->host, NULL, &req, &ans); |
|
157 |
if (err != 0) { |
|
158 |
/* FIXME: Handle error */ |
|
159 |
data->callback (data->socket, data, FALSE); |
|
160 |
} |
|
161 |
||
162 |
data->resolved_addrs = ans; |
|
163 |
data->current_addr = ans; |
|
164 |
||
165 |
data->callback (data->socket, data, TRUE); |
|
166 |
} |
|
167 |
||
168 |
static SocketDNSData * |
|
169 |
socket_dns_data_new (LmSocket *socket, |
|
170 |
const gchar *host, |
|
171 |
SocketDNSCallback callbac) |
|
172 |
{ |
|
173 |
SocketDNSData *data; |
|
174 |
||
175 |
data = g_new0 (SocketDNSData, 1); |
|
176 |
||
177 |
data->socket = lm_socket_ref (socket); |
|
178 |
data->host = g_strdup (host); |
|
179 |
data->callback = callback; |
|
180 |
||
181 |
data->resolved_addrs = data->current_addr = NULL; |
|
182 |
||
183 |
return data; |
|
184 |
} |
|
185 |
||
186 |
static void |
|
187 |
socket_dns_data_free (SocketDNSData *data) |
|
188 |
{ |
|
189 |
lm_socket_unref (data->socket); |
|
190 |
g_free (data->host); |
|
191 |
||
192 |
if (data->resolved_addrs) { |
|
193 |
freeaddrinfo (data->resolved_addrs);; |
|
194 |
deta->resolved_addrs = NULL; |
|
195 |
} |
|
196 |
||
197 |
g_free (data); |
|
198 |
} |
|
199 |
||
200 |
static void |
|
201 |
socket_start_connect (LmSocket *socket, |
|
202 |
SocketDNSData *data, |
|
203 |
gboolean success) |
|
204 |
{ |
|
161
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
205 |
struct addrinfo *addr; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
206 |
int port; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
207 |
char name[NI_MAXHOST]; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
208 |
char portname[NI_MAXSERV]; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
209 |
gint fd; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
210 |
|
154 | 211 |
if (!success) { |
212 |
socket_dns_data_free (data); |
|
213 |
/* FIXME: Report error */ |
|
161
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
214 |
return; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
215 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
216 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
217 |
addr = data->current_addr; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
218 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
219 |
if (socket->proxy) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
220 |
port = htons (lm_proxy_get_port (socket->proxy)); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
221 |
} else { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
222 |
port = htons (socket->port); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
223 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
224 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
225 |
((struct sockaddr_in *) addr->ai_addr)->sin_port = port; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
226 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
227 |
res = getnameinfo (addr->ai_addr, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
228 |
(socklen_t) addr->ai_addrlen, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
229 |
name, sizeof (name), |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
230 |
portname, sizeof (portname), |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
231 |
NI_NUMERICHOST | NI_NUMERICSERV); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
232 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
233 |
if (res < 0) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
234 |
/* FIXME: Report failure */ |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
235 |
return; |
154 | 236 |
} |
237 |
||
161
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
238 |
fd = _lm_sock_makesocket (addr->ai_family, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
239 |
addr->ai_socktype, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
240 |
addr->ai_protocol); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
241 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
242 |
if (!_LM_SOCK_VALID (fd)) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
243 |
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
244 |
"Failed making socket, error:%d...\n", |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
245 |
_lm_sock_get_last_error ()); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
246 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
247 |
/* FIXME: Report failure */ |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
248 |
return; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
249 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
250 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
251 |
/* Even though it says _unix_new(), it is supported by glib on |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
252 |
* win32 because glib does some cool stuff to find out if it |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
253 |
* can treat it as a FD or a windows SOCKET. |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
254 |
*/ |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
255 |
socket->fd = fd; |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
256 |
socket->io_channel = g_io_channel_unix_new (fd); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
257 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
258 |
g_io_channel_set_encoding (connect_data->io_channel, NULL, NULL); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
259 |
g_io_channel_set_buffered (connect_data->io_channel, FALSE); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
260 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
261 |
_lm_sock_set_blocking (socket->fd, FALSE); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
262 |
|
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
263 |
if (socket->proxy) { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
264 |
socket->io_watch_connect = |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
265 |
socket_add_watch (socket, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
266 |
socket->io_channel, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
267 |
G_IO_IN|G_IO_OUT|G_IO_ERR|G_IO_HUP, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
268 |
(GIOFunc) _lm_proxy_connect_cb, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
269 |
socket); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
270 |
} else { |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
271 |
socket->io_watch_connect = |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
272 |
socket_add_watch (socket, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
273 |
socket->io_channel, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
274 |
G_IO_IN|G_IO_OUT|G_IO_ERR|G_IO_HUP, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
275 |
(GIOFunc) socket_connect_cb, |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
276 |
socket); |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
277 |
} |
05ddf1d0440f
2006-08-01 Mikael Hallendal <micke@imendio.com>
hallski <hallski>
parents:
154
diff
changeset
|
278 |
|
154 | 279 |
/* FIXME: Continue and connect */ |
147 | 280 |
} |
281 |
||
149 | 282 |
LmSocket * |
283 |
lm_socket_new (LmSocketFuncs funcs, const gchar *host, guint port) |
|
284 |
{ |
|
285 |
LmSocket *socket; |
|
147 | 286 |
|
149 | 287 |
socket = socket_create (); |
288 |
||
289 |
socket->funcs = funcs; |
|
290 |
socket->host = g_strdup (host); |
|
291 |
socket->port = port; |
|
151 | 292 |
socket->is_blocking = FALSE; |
149 | 293 |
|
294 |
return socket; |
|
295 |
} |
|
147 | 296 |
|
149 | 297 |
void |
298 |
lm_socket_open (LmSocket *socket) |
|
299 |
{ |
|
151 | 300 |
g_return_if_fail (socket != NULL); |
301 |
||
154 | 302 |
socket_dns_lookup (socket, socket->host, socket_start_connect); |
149 | 303 |
} |
304 |
||
305 |
int |
|
306 |
lm_socket_get_fd (LmSocket *socket) |
|
147 | 307 |
{ |
149 | 308 |
g_return_val_if_fail (socket != NULL, -1); |
309 |
||
310 |
return socket->fd; |
|
311 |
} |
|
312 |
||
313 |
gboolean |
|
314 |
lm_socket_get_is_blocking (LmSocket *socket) |
|
315 |
{ |
|
316 |
return socket->is_blocking; |
|
147 | 317 |
} |
318 |
||
319 |
void |
|
149 | 320 |
lm_socket_set_is_blocking (LmSocket *socket, gboolean is_block) |
147 | 321 |
{ |
149 | 322 |
int res; |
323 |
||
324 |
g_return_if_fail (socket != NULL); |
|
325 |
||
326 |
/* FIXME: Don't unset all flags */ |
|
327 |
||
328 |
#ifndef G_OS_WIN32 |
|
329 |
res = fcntl (socket->sock, F_SETFL, is_block ? 0 : O_NONBLOCK); |
|
330 |
#else /* G_OS_WIN32 */ |
|
331 |
u_long mode = (is_block ? 0 : 1); |
|
332 |
res = ioctlsocket (socket->sock, FIONBIO, &mode); |
|
333 |
#endif /* G_OS_WIN32 */ |
|
334 |
||
335 |
if (res != 0) { |
|
336 |
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, |
|
337 |
"Could not set socket to be %s\n", |
|
338 |
block ? "blocking" : "non-blocking"); |
|
339 |
} |
|
340 |
||
341 |
socket->is_blocking = is_block; |
|
342 |
} |
|
343 |
||
344 |
int |
|
345 |
lm_socket_write (LmSocket *socket, |
|
346 |
gsize size, |
|
347 |
gchar *buf, |
|
348 |
GError **error) |
|
349 |
{ |
|
151 | 350 |
gint b_written; |
351 |
||
352 |
g_return_val_if_fail (socket != NULL, -1); |
|
353 |
||
354 |
if (socket->ssl) { |
|
355 |
b_written = _lm_ssl_send (socket->ssl, buf, len); |
|
356 |
} else { |
|
357 |
GIOStatus io_status = G_IO_STATUS_AGAIN; |
|
358 |
gsize bytes_written = 0; |
|
359 |
||
360 |
while (io_status == G_IO_STATUS_AGAIN) { |
|
154 | 361 |
io_status = g_io_channel_write_chars (socket->io_channel, |
151 | 362 |
buf, size, |
363 |
&bytes_written, |
|
364 |
NULL); |
|
365 |
} |
|
366 |
||
367 |
b_written = bytes_written; |
|
368 |
||
369 |
if (io_status != G_IO_STATUS_NORMAL) { |
|
370 |
b_written = -1; |
|
371 |
} |
|
372 |
} |
|
373 |
||
374 |
return b_written; |
|
149 | 375 |
} |
376 |
||
377 |
int |
|
378 |
lm_socket_read (LmSocket *socket, |
|
379 |
gsize size, |
|
380 |
gchar *buf, |
|
381 |
GError **error) |
|
382 |
{ |
|
151 | 383 |
gsize bytes_read = 0; |
384 |
GIOStatus status = G_IO_STATUS_AGAIN; |
|
385 |
||
386 |
g_return_val_if_fail (socket != NULL, -1); |
|
387 |
||
388 |
while (status == G_IO_STATUS_AGAIN) { |
|
389 |
if (socket->ssl) { |
|
390 |
status = _lm_ssl_read (socket->ssl, |
|
391 |
buf, size, &bytes_read); |
|
392 |
} else { |
|
154 | 393 |
status = g_io_channel_read_chars (socket->io_channel, |
151 | 394 |
buf, size, |
395 |
&bytes_read, |
|
396 |
NULL); |
|
397 |
} |
|
398 |
} |
|
399 |
||
400 |
if (status != G_IO_STATUS_NORMAL || bytes_read < 0) { |
|
401 |
/* FIXME: Set error */ |
|
402 |
||
403 |
return -1; |
|
404 |
} |
|
405 |
||
406 |
return bytes_read; |
|
149 | 407 |
} |
408 |
||
409 |
gboolean |
|
410 |
lm_socket_close (LmSocket *socket, GError **error) |
|
411 |
{ |
|
150 | 412 |
#ifndef G_OS_WIN32 |
413 |
close (socket->sock); |
|
414 |
#else /* G_OS_WIN32 */ |
|
415 |
closesocket (socket->sock); |
|
416 |
#endif /* G_OS_WIN32 */ |
|
149 | 417 |
} |
418 |
||
419 |
LmSocket * |
|
420 |
lm_socket_ref (LmSocket *socket) |
|
421 |
{ |
|
422 |
g_return_val_if_fail (socket != NULL, NULL); |
|
151 | 423 |
|
149 | 424 |
socket->ref++; |
147 | 425 |
|
149 | 426 |
return socket; |
427 |
} |
|
428 |
||
429 |
void |
|
430 |
lm_socket_unref (LmSocket *socket) |
|
431 |
{ |
|
432 |
g_return_if_fail (socket != NULL); |
|
147 | 433 |
|
149 | 434 |
socket->ref--; |
435 |
||
436 |
if (socket->ref <= 0) { |
|
437 |
socket_free (socket); |
|
147 | 438 |
} |
439 |
} |
|
149 | 440 |