2 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3 * Copyright (C) 1999-2009 Hiroyuki Yamamoto and the Claws Mail team
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 3 of the License, or
8 * (at your option) any later version.
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
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #if (defined (_XOPEN_SOURCE) && !defined (_BSD_SOURCE))
29 #include <glib/gi18n.h>
32 #include <sys/types.h>
34 # include <winsock2.h>
36 # define EINPROGRESS WSAEINPROGRESS
41 # include <sys/wait.h>
43 # include <sys/socket.h>
44 # include <sys/stat.h>
46 # include <netinet/in.h>
47 # include <arpa/inet.h>
50 #endif /* G_OS_WIN32 */
60 # include <sys/select.h>
71 #error USE_GIO is currently not supported
81 typedef gint (*SockAddrFunc) (GList *addr_list,
84 typedef struct _SockConnectData SockConnectData;
85 typedef struct _SockLookupData SockLookupData;
86 typedef struct _SockAddrData SockAddrData;
87 typedef struct _SockSource SockSource;
89 struct _SockConnectData {
95 SockLookupData *lookup_data;
100 gchar *canonical_name;
103 struct _SockLookupData {
112 gchar *canonical_name;
115 struct _SockAddrData {
120 struct sockaddr *addr;
128 static guint io_timeout = 60;
130 static GList *sock_connect_data_list = NULL;
132 static gboolean sock_prepare (GSource *source,
134 static gboolean sock_check (GSource *source);
135 static gboolean sock_dispatch (GSource *source,
136 GSourceFunc callback,
139 GSourceFuncs sock_watch_funcs = {
146 static gint sock_connect_with_timeout (gint sock,
147 const struct sockaddr *serv_addr,
152 static gint sock_connect_by_hostname (gint sock,
153 const gchar *hostname,
156 static gint sock_connect_by_getaddrinfo (const gchar *hostname,
160 static SockInfo *sockinfo_from_fd(const gchar *hostname,
163 static void sock_address_list_free (GList *addr_list);
165 static gboolean sock_connect_async_cb (GIOChannel *source,
166 GIOCondition condition,
168 static gint sock_connect_async_get_address_info_cb
172 static gint sock_connect_address_list_async (SockConnectData *conn_data);
174 static gboolean sock_get_address_info_async_cb (GIOChannel *source,
175 GIOCondition condition,
177 static SockLookupData *sock_get_address_info_async
178 (const gchar *hostname,
182 static gint sock_get_address_info_async_cancel (SockLookupData *lookup_data);
191 result = WSAStartup(MAKEWORD(2, 2), &wsadata);
192 if (result != NO_ERROR) {
193 g_warning("WSAStartup() failed\n");
200 gint sock_cleanup(void)
208 gint sock_set_io_timeout(guint sec)
214 void refresh_resolvers(void)
217 static time_t resolv_conf_changed = (time_t)NULL;
220 /* This makes the glibc re-read resolv.conf, if it changed
221 * since our startup. Maybe that should be #ifdef'ed, I don't
222 * know if it'd work on BSDs.
223 * Why doesn't the glibc do it by itself?
225 if (stat("/etc/resolv.conf", &s) == 0) {
226 if (s.st_mtime > resolv_conf_changed) {
227 resolv_conf_changed = s.st_mtime;
231 we'll have bigger problems. */
236 #define SOCKET_IS_VALID(s) ((s) != INVALID_SOCKET)
238 #define SOCKET_IS_VALID(s) TRUE
241 /* Due to the fact that socket under Windows are not represented by
242 standard file descriptors, we sometimes need to check whether a
243 given file descriptor is actually a socket. This is done by
244 testing for an error. Returns true under W32 if FD is a socket. */
245 static int fd_is_w32_socket(gint fd)
249 gint retval = sizeof(optval);
251 return !getsockopt(fd, SOL_SOCKET, SO_TYPE, (char*)&optval, &retval);
257 gint fd_connect_inet(gushort port)
260 struct sockaddr_in addr;
262 sock = socket(AF_INET, SOCK_STREAM, 0);
263 if (!SOCKET_IS_VALID(sock)) {
265 debug_print("fd_connect_inet(): socket() failed: %d\n",
268 perror("fd_connect_inet(): socket");
273 memset(&addr, 0, sizeof(addr));
274 addr.sin_family = AF_INET;
275 addr.sin_port = htons(port);
276 addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
278 if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
285 gint fd_open_inet(gushort port)
288 struct sockaddr_in addr;
291 sock = socket(AF_INET, SOCK_STREAM, 0);
292 if (!SOCKET_IS_VALID(sock)) {
294 g_warning("fd_open_inet(): socket() failed: %d\n",
297 perror("fd_open_inet(): socket");
303 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char *)&val,
305 perror("setsockopt");
310 memset(&addr, 0, sizeof(addr));
311 addr.sin_family = AF_INET;
312 addr.sin_port = htons(port);
313 addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
315 if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
321 if (listen(sock, 1) < 0) {
330 gint fd_connect_unix(const gchar *path)
334 struct sockaddr_un addr;
336 sock = socket(PF_UNIX, SOCK_STREAM, 0);
338 perror("sock_connect_unix(): socket");
342 memset(&addr, 0, sizeof(addr));
343 addr.sun_family = AF_UNIX;
344 strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
346 if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
357 gint fd_open_unix(const gchar *path)
361 struct sockaddr_un addr;
363 sock = socket(PF_UNIX, SOCK_STREAM, 0);
366 perror("sock_open_unix(): socket");
370 memset(&addr, 0, sizeof(addr));
371 addr.sun_family = AF_UNIX;
372 strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
374 if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
375 gchar *buf = g_strdup_printf("can't bind to %s", path);
382 if (listen(sock, 1) < 0) {
383 gchar *buf = g_strdup_printf("can't listen on %s", path);
396 gint fd_accept(gint sock)
398 struct sockaddr_in caddr;
401 caddr_len = sizeof(caddr);
402 return accept(sock, (struct sockaddr *)&caddr, &caddr_len);
406 static gint set_nonblocking_mode(gint fd, gboolean nonblock)
411 flags = fcntl(fd, F_GETFL, 0);
420 flags &= ~O_NONBLOCK;
422 return fcntl(fd, F_SETFL, flags);
428 gint sock_set_nonblocking_mode(SockInfo *sock, gboolean nonblock)
430 cm_return_val_if_fail(sock != NULL, -1);
432 return set_nonblocking_mode(sock->sock, nonblock);
435 static gboolean is_nonblocking_mode(gint fd)
440 flags = fcntl(fd, F_GETFL, 0);
446 return ((flags & O_NONBLOCK) != 0);
452 gboolean sock_is_nonblocking_mode(SockInfo *sock)
454 cm_return_val_if_fail(sock != NULL, FALSE);
456 return is_nonblocking_mode(sock->sock);
460 static gboolean sock_prepare(GSource *source, gint *timeout)
466 static gboolean sock_check(GSource *source)
468 SockInfo *sock = ((SockSource *)source)->sock;
469 struct timeval timeout = {0, 0};
471 GIOCondition condition = 0;
473 if (!sock || !sock->sock)
476 condition = sock->condition;
479 FD_SET(sock->sock, &fds);
481 select(sock->sock + 1,
482 (condition & G_IO_IN) ? &fds : NULL,
483 (condition & G_IO_OUT) ? &fds : NULL,
486 return FD_ISSET(sock->sock, &fds) != 0;
489 static gboolean sock_dispatch(GSource *source, GSourceFunc callback,
492 SockInfo *sock = ((SockSource *)source)->sock;
494 if (!sock || !sock->callback || !sock->data)
497 return sock->callback(sock, sock->condition, sock->data);
500 static gboolean sock_watch_cb(GIOChannel *source, GIOCondition condition,
503 SockInfo *sock = (SockInfo *)data;
505 if ((condition & sock->condition) == 0)
508 return sock->callback(sock, sock->condition, sock->data);
511 guint sock_add_watch(SockInfo *sock, GIOCondition condition, SockFunc func,
517 sock->callback = func;
518 sock->condition = condition;
524 GSource *source = g_source_new(&sock_watch_funcs,
526 ((SockSource *) source)->sock = sock;
527 g_source_set_priority(source, G_PRIORITY_DEFAULT);
528 g_source_set_can_recurse(source, FALSE);
529 sock->g_source = g_source_attach(source, NULL);
530 g_source_unref (source); /* Refcount back down to 1 */
531 return sock->g_source;
535 return g_io_add_watch(sock->sock_ch, condition, sock_watch_cb, sock);
538 static gint fd_check_io(gint fd, GIOCondition cond)
540 struct timeval timeout;
543 if (is_nonblocking_mode(fd))
546 timeout.tv_sec = io_timeout;
552 if (cond == G_IO_IN) {
553 select(fd + 1, &fds, NULL, NULL,
554 io_timeout > 0 ? &timeout : NULL);
556 select(fd + 1, NULL, &fds, NULL,
557 io_timeout > 0 ? &timeout : NULL);
560 if (FD_ISSET(fd, &fds)) {
563 g_warning("Socket IO timeout\n");
569 static sigjmp_buf jmpenv;
571 static void timeout_handler(gint sig)
573 siglongjmp(jmpenv, 1);
577 static gint sock_connect_with_timeout(gint sock,
578 const struct sockaddr *serv_addr,
584 void (*prev_handler)(gint);
587 prev_handler = signal(SIGALRM, timeout_handler);
588 if (sigsetjmp(jmpenv, 1)) {
590 signal(SIGALRM, prev_handler);
597 ret = connect(sock, serv_addr, addrlen);
601 signal(SIGALRM, prev_handler);
607 struct hostent *my_gethostbyname(const gchar *hostname)
611 void (*prev_handler)(gint);
614 prev_handler = signal(SIGALRM, timeout_handler);
615 if (sigsetjmp(jmpenv, 1)) {
617 signal(SIGALRM, prev_handler);
618 g_printerr("%s: host lookup timed out.\n", hostname);
625 if ((hp = gethostbyname(hostname)) == NULL) {
628 signal(SIGALRM, prev_handler);
630 g_printerr("%s: unknown host.\n", hostname);
637 signal(SIGALRM, prev_handler);
644 static gint my_inet_aton(const gchar *hostname, struct in_addr *inp)
647 return inet_aton(hostname, inp);
652 inaddr = inet_addr(hostname);
654 memcpy(inp, &inaddr, sizeof(inaddr));
661 #endif /* HAVE_INET_ATON */
664 static gint sock_connect_by_hostname(gint sock, const gchar *hostname,
668 struct sockaddr_in ad;
670 memset(&ad, 0, sizeof(ad));
671 ad.sin_family = AF_INET;
672 ad.sin_port = htons(port);
676 if (!my_inet_aton(hostname, &ad.sin_addr)) {
677 if ((hp = my_gethostbyname(hostname)) == NULL) {
678 g_printerr("%s: unknown host.\n", hostname);
683 if (hp->h_length != 4 && hp->h_length != 8) {
684 g_printerr("illegal address length received for host %s\n", hostname);
689 memcpy(&ad.sin_addr, hp->h_addr, hp->h_length);
692 return sock_connect_with_timeout(sock, (struct sockaddr *)&ad,
693 sizeof(ad), io_timeout);
697 static gint sock_connect_by_getaddrinfo(const gchar *hostname, gushort port)
699 gint sock = -1, gai_error;
700 struct addrinfo hints, *res, *ai;
705 memset(&hints, 0, sizeof(hints));
706 /* hints.ai_flags = AI_CANONNAME; */
707 hints.ai_family = AF_UNSPEC;
708 hints.ai_socktype = SOCK_STREAM;
709 hints.ai_protocol = IPPROTO_TCP;
711 /* convert port from integer to string. */
712 g_snprintf(port_str, sizeof(port_str), "%d", port);
714 if ((gai_error = getaddrinfo(hostname, port_str, &hints, &res)) != 0) {
715 g_printerr("getaddrinfo for %s:%s failed: %s\n",
716 hostname, port_str, gai_strerror(gai_error));
720 for (ai = res; ai != NULL; ai = ai->ai_next) {
721 sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
725 if (sock_connect_with_timeout
726 (sock, ai->ai_addr, ai->ai_addrlen, io_timeout) == 0)
742 SockInfo *sock_connect(const gchar *hostname, gushort port)
751 if ((sock = sock_connect_by_getaddrinfo(hostname, port)) < 0)
755 if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET) {
756 g_warning("socket() failed: %d\n", WSAGetLastError());
758 if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
760 #endif /* G_OS_WIN32 */
764 if (sock_connect_by_hostname(sock, hostname, port) < 0) {
765 if (errno != 0) perror("connect");
771 return sockinfo_from_fd(hostname, port, sock);
775 static void sock_address_list_free(GList *addr_list)
779 for (cur = addr_list; cur != NULL; cur = cur->next) {
780 SockAddrData *addr_data = (SockAddrData *)cur->data;
781 g_free(addr_data->addr);
785 g_list_free(addr_list);
788 /* asynchronous TCP connection */
790 static gboolean sock_connect_async_cb(GIOChannel *source,
791 GIOCondition condition, gpointer data)
793 SockConnectData *conn_data = (SockConnectData *)data;
799 if (conn_data->io_tag == 0 && conn_data->channel == NULL)
802 fd = g_io_channel_unix_get_fd(source);
804 conn_data->io_tag = 0;
805 conn_data->channel = NULL;
806 g_io_channel_unref(source);
809 if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &val, &len) < 0) {
810 perror("getsockopt");
812 sock_connect_address_list_async(conn_data);
818 sock_connect_address_list_async(conn_data);
822 sockinfo = g_new0(SockInfo, 1);
825 sockinfo->sock_ch = g_io_channel_unix_new(fd);
827 sockinfo->sock_ch = g_io_channel_win32_new_socket(fd);
829 sockinfo->hostname = g_strdup(conn_data->hostname);
830 sockinfo->port = conn_data->port;
831 sockinfo->state = CONN_ESTABLISHED;
832 sockinfo->canonical_name = g_strdup(conn_data->canonical_name);
834 conn_data->func(sockinfo, conn_data->data);
836 sock_connect_async_cancel(conn_data->id);
841 static gint sock_connect_async_get_address_info_cb(GList *addr_list,
844 SockConnectData *conn_data = (SockConnectData *)data;
846 conn_data->addr_list = addr_list;
847 conn_data->cur_addr = addr_list;
848 if (conn_data->lookup_data) {
849 conn_data->canonical_name = conn_data->lookup_data->canonical_name;
850 conn_data->lookup_data->canonical_name = NULL;
851 conn_data->lookup_data = NULL;
853 return sock_connect_address_list_async(conn_data);
856 gint sock_connect_async(const gchar *hostname, gushort port,
857 SockConnectFunc func, gpointer data)
860 SockConnectData *conn_data;
862 conn_data = g_new0(SockConnectData, 1);
863 conn_data->id = id++;
864 conn_data->hostname = g_strdup(hostname);
865 conn_data->port = port;
866 conn_data->addr_list = NULL;
867 conn_data->cur_addr = NULL;
868 conn_data->io_tag = 0;
869 conn_data->func = func;
870 conn_data->data = data;
872 conn_data->lookup_data = sock_get_address_info_async
873 (hostname, port, sock_connect_async_get_address_info_cb,
876 if (conn_data->lookup_data == NULL) {
877 g_free(conn_data->hostname);
882 sock_connect_data_list = g_list_append(sock_connect_data_list,
885 return conn_data->id;
888 gint sock_connect_async_cancel(gint id)
890 SockConnectData *conn_data = NULL;
893 for (cur = sock_connect_data_list; cur != NULL; cur = cur->next) {
894 if (((SockConnectData *)cur->data)->id == id) {
895 conn_data = (SockConnectData *)cur->data;
901 sock_connect_data_list = g_list_remove(sock_connect_data_list,
904 if (conn_data->lookup_data)
905 sock_get_address_info_async_cancel
906 (conn_data->lookup_data);
908 if (conn_data->io_tag > 0)
909 g_source_remove(conn_data->io_tag);
910 if (conn_data->channel) {
912 g_io_channel_shutdown(conn_data->channel, TRUE, &err);
915 g_io_channel_unref(conn_data->channel);
918 sock_address_list_free(conn_data->addr_list);
919 g_free(conn_data->canonical_name);
920 g_free(conn_data->hostname);
923 g_warning("sock_connect_async_cancel: id %d not found.\n", id);
930 static gint sock_connect_address_list_async(SockConnectData *conn_data)
932 SockAddrData *addr_data;
935 for (; conn_data->cur_addr != NULL;
936 conn_data->cur_addr = conn_data->cur_addr->next) {
937 addr_data = (SockAddrData *)conn_data->cur_addr->data;
939 if ((sock = socket(addr_data->family, addr_data->socktype,
940 addr_data->protocol)) < 0) {
945 set_nonblocking_mode(sock, TRUE);
947 if (connect(sock, addr_data->addr, addr_data->addr_len) < 0) {
948 if (EINPROGRESS == errno) {
958 if (conn_data->cur_addr == NULL) {
959 g_warning("sock_connect_address_list_async: "
960 "connection to %s:%d failed\n",
961 conn_data->hostname, conn_data->port);
962 conn_data->func(NULL, conn_data->data);
963 sock_connect_async_cancel(conn_data->id);
967 conn_data->cur_addr = conn_data->cur_addr->next;
970 conn_data->channel = g_io_channel_unix_new(sock);
972 conn_data->channel = g_io_channel_win32_new_socket(sock);
974 conn_data->io_tag = g_io_add_watch(conn_data->channel, G_IO_IN|G_IO_OUT,
975 sock_connect_async_cb, conn_data);
980 /* asynchronous DNS lookup */
982 static gboolean sock_get_address_info_async_cb(GIOChannel *source,
983 GIOCondition condition,
986 SockLookupData *lookup_data = (SockLookupData *)data;
987 GList *addr_list = NULL;
988 SockAddrData *addr_data;
991 struct sockaddr *addr;
992 gchar *canonical_name = NULL;
996 g_io_channel_set_encoding(source, NULL, &err);
998 g_warning("can unset encoding: %s\n", err->message);
1002 g_io_channel_set_buffered(source, FALSE);
1003 if (g_io_channel_read_chars(source, &len, sizeof(len),
1004 &bytes_read, &err) == G_IO_STATUS_NORMAL) {
1006 g_warning("g_io_channel_read_chars: %s\n", err->message);
1010 if (bytes_read == sizeof(len) && len > 0) {
1013 canonical_name = g_malloc0(len + 1);
1014 cur = canonical_name;
1016 if (g_io_channel_read_chars(source, cur, todo,
1017 &bytes_read, &err) != G_IO_STATUS_NORMAL) {
1019 g_warning("canonical name not read %s\n", err->message);
1020 g_free(canonical_name);
1021 canonical_name = NULL;
1030 if (bytes_read == 0) {
1031 g_warning("canonical name not read\n");
1032 g_free(canonical_name);
1033 canonical_name = NULL;
1040 if (g_io_channel_read_chars(source, (gchar *)ai_member,
1041 sizeof(ai_member), &bytes_read, &err)
1042 != G_IO_STATUS_NORMAL) {
1044 g_warning("g_io_channel_read_chars: addr len %s\n", err->message);
1051 if (bytes_read == 0 || bytes_read != sizeof(ai_member))
1054 if (ai_member[0] == AF_UNSPEC) {
1055 g_warning("DNS lookup failed\n");
1059 addr = g_malloc(ai_member[3]);
1060 if (g_io_channel_read_chars(source, (gchar *)addr, ai_member[3],
1062 != G_IO_STATUS_NORMAL) {
1064 g_warning("g_io_channel_read_chars: addr data read %s\n", err->message);
1072 if (bytes_read != ai_member[3]) {
1073 g_warning("sock_get_address_info_async_cb: "
1074 "incomplete address data\n");
1079 addr_data = g_new0(SockAddrData, 1);
1080 addr_data->family = ai_member[0];
1081 addr_data->socktype = ai_member[1];
1082 addr_data->protocol = ai_member[2];
1083 addr_data->addr_len = ai_member[3];
1084 addr_data->addr = addr;
1086 addr_list = g_list_append(addr_list, addr_data);
1089 g_io_channel_shutdown(source, TRUE, &err);
1092 g_io_channel_unref(source);
1095 /* FIXME: We would need to cancel the thread. */
1097 kill(lookup_data->child_pid, SIGKILL);
1098 waitpid(lookup_data->child_pid, NULL, 0);
1100 lookup_data->canonical_name = canonical_name;
1102 lookup_data->func(addr_list, lookup_data->data);
1104 g_free(lookup_data->canonical_name);
1105 g_free(lookup_data->hostname);
1106 g_free(lookup_data);
1112 /* For better readability we use a separate function to implement the
1113 child code of sock_get_address_info_async. Note, that under W32
1114 this is actually not a child but a thread and this is the reason
1115 why we pass only a void pointer. */
1116 static void address_info_async_child(void *opaque)
1118 SockLookupData *parm = opaque;
1121 struct addrinfo hints, *res, *ai;
1125 gchar **addr_list_p;
1126 struct sockaddr_in ad;
1128 gint ai_member[4] = {AF_UNSPEC, 0, 0, 0};
1131 close(parm->pipe_fds[0]);
1132 parm->pipe_fds[0] = -1;
1136 memset(&hints, 0, sizeof(hints));
1137 hints.ai_flags = AI_CANONNAME;
1138 hints.ai_family = AF_UNSPEC;
1139 hints.ai_socktype = SOCK_STREAM;
1140 hints.ai_protocol = IPPROTO_TCP;
1142 g_snprintf(port_str, sizeof(port_str), "%d", parm->port);
1144 gai_err = getaddrinfo(parm->hostname, port_str, &hints, &res);
1147 g_warning("getaddrinfo for %s:%s failed: %s\n",
1148 parm->hostname, port_str, gai_strerror(gai_err));
1149 fd_write_all(parm->pipe_fds[1], &len,
1151 fd_write_all(parm->pipe_fds[1], (gchar *)ai_member,
1153 close(parm->pipe_fds[1]);
1154 parm->pipe_fds[1] = -1;
1163 if (res->ai_canonname && strlen(res->ai_canonname) < 255) {
1164 gchar len = strlen(res->ai_canonname);
1165 fd_write_all(parm->pipe_fds[1], &len,
1167 fd_write_all(parm->pipe_fds[1], res->ai_canonname,
1171 fd_write_all(parm->pipe_fds[1], &len,
1176 fd_write_all(parm->pipe_fds[1], &len,
1180 for (ai = res; ai != NULL; ai = ai->ai_next) {
1181 ai_member[0] = ai->ai_family;
1182 ai_member[1] = ai->ai_socktype;
1183 ai_member[2] = ai->ai_protocol;
1184 ai_member[3] = ai->ai_addrlen;
1186 fd_write_all(parm->pipe_fds[1], (gchar *)ai_member,
1188 fd_write_all(parm->pipe_fds[1], (gchar *)ai->ai_addr,
1195 hp = my_gethostbyname(parm->hostname);
1196 if (hp == NULL || hp->h_addrtype != AF_INET) {
1198 fd_write_all(parm->pipe_fds[1], &len,
1200 fd_write_all(parm->pipe_fds[1], (gchar *)ai_member,
1202 close(parm->pipe_fds[1]);
1203 parm->pipe_fds[1] = -1;
1211 ai_member[0] = AF_INET;
1212 ai_member[1] = SOCK_STREAM;
1213 ai_member[2] = IPPROTO_TCP;
1214 ai_member[3] = sizeof(ad);
1216 memset(&ad, 0, sizeof(ad));
1217 ad.sin_family = AF_INET;
1218 ad.sin_port = htons(parm->port);
1220 if (hp->h_name && strlen(hp->h_name) < 255) {
1221 gchar len = strlen(hp->h_name);
1222 fd_write_all(parm->pipe_fds[1], &len,
1224 fd_write_all(parm->pipe_fds[1], hp->h_name,
1228 fd_write_all(parm->pipe_fds[1], &len,
1231 for (addr_list_p = hp->h_addr_list; *addr_list_p != NULL;
1233 memcpy(&ad.sin_addr, *addr_list_p, hp->h_length);
1234 fd_write_all(parm->pipe_fds[1], (gchar *)ai_member,
1236 fd_write_all(parm->pipe_fds[1], (gchar *)&ad, sizeof(ad));
1240 close(parm->pipe_fds[1]);
1241 parm->pipe_fds[1] = -1;
1250 static SockLookupData *sock_get_address_info_async(const gchar *hostname,
1255 SockLookupData *lookup_data = NULL;
1257 refresh_resolvers();
1259 lookup_data = g_new0(SockLookupData, 1);
1260 lookup_data->hostname = g_strdup(hostname);
1261 lookup_data->func = func;
1262 lookup_data->data = data;
1263 lookup_data->port = port;
1264 lookup_data->child_pid = (pid_t)(-1);
1265 lookup_data->pipe_fds[0] = -1;
1266 lookup_data->pipe_fds[1] = -1;
1268 if (pipe(lookup_data->pipe_fds) < 0) {
1271 g_free (lookup_data->hostname);
1272 g_free (lookup_data);
1277 if ((lookup_data->child_pid = fork()) < 0) {
1280 g_free (lookup_data->hostname);
1281 g_free (lookup_data);
1285 if (lookup_data->child_pid == 0) {
1286 /* Child process. */
1287 address_info_async_child (lookup_data);
1288 g_assert_not_reached ();
1290 /* Parent process. */
1291 close(lookup_data->pipe_fds[1]);
1292 lookup_data->pipe_fds[1] = -1;
1293 #endif /*!G_OS_WIN32 */
1296 lookup_data->channel = g_io_channel_unix_new(lookup_data->pipe_fds[0]);
1298 lookup_data->channel = g_io_channel_win32_new_fd(lookup_data->pipe_fds[0]);
1300 lookup_data->io_tag = g_io_add_watch(lookup_data->channel, G_IO_IN,
1301 sock_get_address_info_async_cb,
1304 lookup_data->child_pid = _beginthread(
1305 address_info_async_child, 0, lookup_data);
1311 static gint sock_get_address_info_async_cancel(SockLookupData *lookup_data)
1313 if (lookup_data->io_tag > 0)
1314 g_source_remove(lookup_data->io_tag);
1315 if (lookup_data->channel) {
1317 g_io_channel_shutdown(lookup_data->channel, TRUE, &err);
1321 g_io_channel_unref(lookup_data->channel);
1324 if (lookup_data->child_pid > 0) {
1326 /* FIXME: Need a way to cancel the thread. */
1328 kill(lookup_data->child_pid, SIGKILL);
1329 waitpid(lookup_data->child_pid, NULL, 0);
1333 g_free(lookup_data->canonical_name);
1334 g_free(lookup_data->hostname);
1335 g_free(lookup_data);
1341 static SockInfo *sockinfo_from_fd(const gchar *hostname,
1347 sockinfo = g_new0(SockInfo, 1);
1348 sockinfo->sock = sock;
1350 sockinfo->sock_ch = g_io_channel_unix_new(sock);
1352 sockinfo->sock_ch = g_io_channel_win32_new_socket(sock);
1354 sockinfo->hostname = g_strdup(hostname);
1355 sockinfo->port = port;
1356 sockinfo->state = CONN_ESTABLISHED;
1361 static gint fd_read(gint fd, gchar *buf, gint len)
1363 if (fd_check_io(fd, G_IO_IN) < 0)
1366 if (fd_is_w32_socket(fd))
1367 return recv(fd, buf, len, 0);
1368 return read(fd, buf, len);
1372 static gint ssl_read(gnutls_session ssl, gchar *buf, gint len)
1376 if (gnutls_record_check_pending(ssl) == 0) {
1377 if (fd_check_io(GPOINTER_TO_INT(gnutls_transport_get_ptr(ssl)), G_IO_IN) < 0)
1382 r = gnutls_record_recv(ssl, buf, len);
1387 case 0: /* closed connection */
1390 case GNUTLS_E_REHANDSHAKE:
1392 r = gnutls_handshake(ssl);
1393 } while (r == GNUTLS_E_AGAIN || r == GNUTLS_E_INTERRUPTED);
1394 break; /* re-receive */
1395 case GNUTLS_E_AGAIN:
1396 case GNUTLS_E_INTERRUPTED:
1408 gint sock_read(SockInfo *sock, gchar *buf, gint len)
1412 cm_return_val_if_fail(sock != NULL, -1);
1416 ret = ssl_read(sock->ssl, buf, len);
1419 ret = fd_read(sock->sock, buf, len);
1422 sock->state = CONN_DISCONNECTED;
1426 gint fd_write(gint fd, const gchar *buf, gint len)
1428 if (fd_check_io(fd, G_IO_OUT) < 0)
1431 if (fd_is_w32_socket (fd))
1432 return send(fd, buf, len, 0);
1433 return write(fd, buf, len);
1437 static gint ssl_write(gnutls_session ssl, const gchar *buf, gint len)
1441 if (fd_check_io(GPOINTER_TO_INT(gnutls_transport_get_ptr(ssl)), G_IO_OUT) < 0)
1444 ret = gnutls_record_send(ssl, buf, len);
1449 case GNUTLS_E_AGAIN:
1450 case GNUTLS_E_INTERRUPTED:
1460 gint sock_write(SockInfo *sock, const gchar *buf, gint len)
1464 cm_return_val_if_fail(sock != NULL, -1);
1468 ret = ssl_write(sock->ssl, buf, len);
1471 ret = fd_write(sock->sock, buf, len);
1474 sock->state = CONN_DISCONNECTED;
1478 gint fd_write_all(gint fd, const gchar *buf, gint len)
1483 if (fd_check_io(fd, G_IO_OUT) < 0)
1486 signal(SIGPIPE, SIG_IGN);
1488 if (fd_is_w32_socket(fd))
1489 n = send(fd, buf, len, 0);
1491 n = write(fd, buf, len);
1494 log_error(LOG_PROTOCOL, _("write on fd%d: %s\n"), fd, strerror(errno));
1506 static gint ssl_write_all(gnutls_session ssl, const gchar *buf, gint len)
1511 n = ssl_write(ssl, buf, len);
1523 gint sock_write_all(SockInfo *sock, const gchar *buf, gint len)
1527 cm_return_val_if_fail(sock != NULL, -1);
1531 ret = ssl_write_all(sock->ssl, buf, len);
1534 ret = fd_write_all(sock->sock, buf, len);
1537 sock->state = CONN_DISCONNECTED;
1541 static gint fd_recv(gint fd, gchar *buf, gint len, gint flags)
1543 if (fd_check_io(fd, G_IO_IN) < 0)
1546 return recv(fd, buf, len, flags);
1549 gint fd_gets(gint fd, gchar *buf, gint len)
1551 gchar *newline, *bp = buf;
1561 MSKB Article ID: Q147714
1562 Windows Sockets 2 Service Provider Interface Limitations
1563 Polling with recv(MSG_PEEK) to determine when a complete message
1565 Reason and Workaround not available.
1567 Single-byte send() and recv().
1568 Reason: Couple one-byte sends with Nagle disabled.
1569 Workaround: Send modest amounts and receive as much as possible.
1572 if (recv(fd, bp, 1, 0) <= 0)
1579 #else /*!G_OS_WIN32*/
1581 if ((n = fd_recv(fd, bp, len, MSG_PEEK)) <= 0)
1583 if ((newline = memchr(bp, '\n', n)) != NULL)
1584 n = newline - bp + 1;
1585 if ((n = fd_read(fd, bp, n)) < 0)
1589 } while (!newline && len);
1590 #endif /*!G_OS_WIN32*/
1596 gint sock_close(SockInfo *sock)
1604 g_io_channel_unref(sock->sock_ch);
1608 ssl_done_socket(sock);
1609 if (sock->g_source != 0)
1610 g_source_remove(sock->g_source);
1614 shutdown(sock->sock, 1); /* complete transfer before close */
1615 ret = closesocket(sock->sock);
1617 ret = fd_close(sock->sock);
1620 g_free(sock->canonical_name);
1621 g_free(sock->hostname);
1627 gint fd_close(gint fd)