3 silcwin32socketstream.c
5 Author: Pekka Riikonen <priikone@silcnet.org>
7 Copyright (C) 2007 - 2008 Pekka Riikonen
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; version 2 of the License.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
20 #include "silcruntime.h"
22 /************************ Static utility functions **************************/
24 /* The IO process callback that calls the notifier callback to upper layer. */
26 SILC_TASK_CALLBACK(silc_socket_stream_io)
28 SilcSocketStream stream = context;
30 if (!stream->notifier)
35 stream->notifier(stream, SILC_STREAM_CAN_READ, stream->notifier_context);
39 stream->notifier(stream, SILC_STREAM_CAN_WRITE, stream->notifier_context);
47 /**************************** Stream Operations *****************************/
49 /* Stream read operation */
51 int silc_socket_stream_read(SilcStream stream, unsigned char *buf,
54 SilcSocketStream sock = stream;
55 SOCKET fd = sock->sock;
58 SILC_LOG_DEBUG(("Reading data from socket %d", fd));
60 /* Check whether there is data available, without calling recv(). */
61 ioctlsocket(fd, FIONREAD, (unsigned long *)&argp);
63 /* Is this kludge or what? Without this thing this contraption
64 does not work at all!?. */
66 SILC_LOG_DEBUG(("Could not read immediately, will do it later"));
67 silc_schedule_set_listen_fd(sock->schedule, sock->sock,
68 silc_schedule_get_fd_events(sock->schedule,
70 SILC_TASK_READ, FALSE);
71 silc_set_errno(SILC_ERR_WOULD_BLOCK);
75 /* Read the data from the socket. */
76 len = recv(fd, buf, buf_len, 0);
77 if (len == SOCKET_ERROR) {
78 len = WSAGetLastError();
79 silc_set_errno_posix(ret);
80 if (len == WSAEWOULDBLOCK || len == WSAEINTR) {
81 SILC_LOG_DEBUG(("Could not read immediately, will do it later"));
82 silc_schedule_set_listen_fd(sock->schedule, sock->sock,
83 silc_schedule_get_fd_events(sock->schedule,
85 SILC_TASK_READ, FALSE);
88 SILC_LOG_DEBUG(("Cannot read from socket: %d", sock->sock));
89 silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
93 SILC_LOG_DEBUG(("Read %d bytes", len));
96 silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
101 /* Stream write operation */
103 int silc_socket_stream_write(SilcStream stream, const unsigned char *data,
106 SilcSocketStream sock = stream;
107 SOCKET fd = sock->sock;
110 SILC_LOG_DEBUG(("Writing data to socket %d", fd));
112 ret = send(fd, data, data_len, 0);
113 if (ret == SOCKET_ERROR) {
114 ret = WSAGetLastError();
115 silc_set_errno_posix(ret);
116 if (ret == WSAEWOULDBLOCK) {
117 SILC_LOG_DEBUG(("Could not write immediately, will do it later"));
118 silc_schedule_set_listen_fd(sock->schedule, sock->sock,
119 SILC_TASK_READ | SILC_TASK_WRITE, FALSE);
122 SILC_LOG_DEBUG(("Cannot write to socket"));
123 silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
127 SILC_LOG_DEBUG(("Wrote data %d bytes", ret));
128 if (silc_schedule_get_fd_events(sock->schedule, sock->sock) &
130 silc_schedule_set_listen_fd(sock->schedule, sock->sock,
131 SILC_TASK_READ, FALSE);
136 /* Receive UDP packet. QoS is not supported. */
138 int silc_socket_udp_stream_read(SilcStream stream, unsigned char *buf,
141 return silc_net_udp_receive(stream, NULL, 0, NULL, buf, buf_len);
144 /* Send UDP packet. This always succeeds. */
146 int silc_socket_udp_stream_write(SilcStream stream, const unsigned char *data,
149 SilcSocketStream sock = stream;
151 /* In connectionless state check if remote IP and port is provided */
152 if (!sock->connected && sock->ip && sock->port)
153 return silc_net_udp_send(stream, sock->ip, sock->port, data, data_len);
155 /* In connected state use normal writing to socket. */
156 return silc_socket_stream_write(stream, data, data_len);
161 SilcBool silc_socket_stream_close(SilcStream stream)
163 SilcSocketStream socket_stream = stream;
165 if (socket_stream->schedule) {
166 silc_schedule_unset_listen_fd(socket_stream->schedule,
167 socket_stream->sock);
168 silc_schedule_task_del_by_fd(socket_stream->schedule,
169 socket_stream->sock);
171 silc_net_close_connection(socket_stream->sock);
176 /* Destroys the stream */
178 void silc_socket_stream_destroy(SilcStream stream)
180 SilcSocketStream socket_stream = stream;
182 silc_socket_stream_close(socket_stream);
183 silc_free(socket_stream->ip);
184 silc_free(socket_stream->hostname);
185 if (socket_stream->schedule)
186 silc_schedule_task_del_by_fd(socket_stream->schedule, socket_stream->sock);
188 if (socket_stream->schedule)
189 silc_schedule_wakeup(socket_stream->schedule);
191 silc_free(socket_stream);
194 /* Sets stream notification callback for the stream */
196 SilcBool silc_socket_stream_notifier(SilcStream stream,
197 SilcSchedule schedule,
198 SilcStreamNotifier callback,
201 SilcSocketStream socket_stream = stream;
203 SILC_LOG_DEBUG(("Setting stream notifier callback"));
205 socket_stream->notifier = callback;
206 socket_stream->notifier_context = context;
207 socket_stream->schedule = schedule;
209 if (socket_stream->notifier && socket_stream->schedule) {
210 /* Add the socket to scheduler. Safe to call if already added. */
211 if (!silc_schedule_task_add_fd(socket_stream->schedule,
213 silc_socket_stream_io, socket_stream))
216 /* Initially set socket for reading */
217 if (!silc_schedule_set_listen_fd(socket_stream->schedule,
219 SILC_TASK_READ, FALSE))
221 } else if (socket_stream->schedule) {
222 /* Unschedule the socket */
223 silc_schedule_unset_listen_fd(socket_stream->schedule,
224 socket_stream->sock);
225 silc_schedule_task_del_by_fd(socket_stream->schedule,
226 socket_stream->sock);
229 if (socket_stream->schedule)
230 silc_schedule_wakeup(socket_stream->schedule);