2 * swift implementation of syscall API>
4 * Simulates the classic socket syscalls (socket, bind, send, recv).
5 * Implementation uses raw sockets (AF_INET, SOCK_RAW).
7 * Subsequently, implementation is to be ported into kernel space and
8 * the interface is going to be offered by the Linux syscall API.
10 * Heavily inspired by GLIBC's <sys/socket.h>
11 * (/usr/include/sys/socket.h).
13 * 2011, Razvan Deaconescu, razvan.deaconescu@cs.pub.ro
19 #include <sys/types.h>
20 #include <sys/socket.h>
21 #include <netinet/in.h>
25 #include "swift_types.h"
26 #include "swift_raw.h"
35 /* socket management structure */
38 struct sockaddr_sw addr;
39 enum sock_rw_state rw_state;
40 struct sock_list *next;
41 struct sock_list *prev;
44 static struct sock_list sock_list_head = {
45 .next = &sock_list_head,
46 .prev = &sock_list_head
50 * Add new socket to list. Called by sw_socket "syscall".
53 static struct sock_list *list_add_socket(int s)
55 struct sock_list *ptr = malloc(sizeof(*ptr));
59 ptr->next = &sock_list_head;
60 ptr->prev = sock_list_head.prev;
61 sock_list_head.prev->next = ptr;
62 sock_list_head.prev = ptr;
69 * Bind socket to given address. Called by sw_bind "syscall".
72 static struct sock_list *list_update_socket_address(int s, __CONST_SOCKADDR_ARG addr)
74 struct sock_list *ptr;
76 for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
78 memcpy(&ptr->addr, addr, sizeof(ptr->addr));
86 * Get list element containing socket s. Called by sw_send* "syscalls".
89 static struct sock_list *list_elem_from_socket(int s)
91 struct sock_list *ptr;
93 for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
102 * Get list element containing address addr. Called by sw_bind "syscall".
105 static struct sock_list *list_elem_from_address(__CONST_SOCKADDR_ARG addr)
107 struct sock_list *ptr;
109 for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
110 if (memcmp(&ptr->addr, addr, sizeof(addr)) == 0)
117 * Remove socket from list. Called by sw_close "syscall".
120 static struct sock_list *list_unlink_socket(int s)
122 struct sock_list *ptr;
124 for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
126 ptr->next->prev = ptr->prev;
127 ptr->prev->next = ptr->next;
137 * Create a new socket of type TYPE in domain DOMAIN, using
138 * protocol PROTOCOL. If PROTOCOL is zero, one is chosen automatically.
139 * Returns a file descriptor for the new socket, or -1 for errors.
141 * swift: PROTOCOL is IPPROTO_SWIFT. Ignore TYPE.
143 int sw_socket(int __domain, int __type, int __protocol)
146 struct sock_list *list;
148 s = socket(__domain, SOCK_RAW, IPPROTO_SWIFT);
152 list = list_add_socket(s);
156 /* Socket is fully open. */
157 list->rw_state = STATE_NO_SHUT;
159 if (__domain != AF_INET || __type != SOCK_RAW || __protocol != IPPROTO_SWIFT) {
164 s = socket(AF_INET, SOCK_RAW, IPPROTO_SWIFT);
175 * Give the socket FD the local address ADDR (which is LEN bytes long).
177 * swift: ADDR is of type struct sockaddr_sw.
179 int sw_bind(int __fd, __CONST_SOCKADDR_ARG __addr, socklen_t __len)
181 struct sock_list *list;
183 /* Check whether address is already in use. */
184 list = list_elem_from_address(__addr);
189 /* Update __fd entry in socket management list. */
190 list = list_update_socket_address(__fd, __addr);
193 goto list_update_err;
203 /* Put the local address of FD into *ADDR and its length in *LEN. */
204 int sw_getsockname(int __fd, __SOCKADDR_ARG __addr,
205 socklen_t *__restrict __len)
213 * Send N bytes of BUF on socket FD to peer at address ADDR (which is
214 * ADDR_LEN bytes long). Returns the number sent, or -1 for errors.
216 * This function is a cancellation point and therefore not marked with
219 ssize_t sw_sendto(int __fd, __const void *__buf, size_t __n,
220 int __flags, __CONST_SOCKADDR_ARG __addr,
221 socklen_t __addr_len)
229 * Read N bytes into BUF through socket FD.
230 * If ADDR is not NULL, fill in *ADDR_LEN bytes of it with tha address of
231 * the sender, and store the actual size of the address in *ADDR_LEN.
232 * Returns the number of bytes read or -1 for errors.
234 * This function is a cancellation point and therefore not marked with
237 ssize_t sw_recvfrom(int __fd, void *__restrict __buf, size_t __n,
238 int __flags, __SOCKADDR_ARG __addr,
239 socklen_t *__restrict __addr_len)
249 * Send a message described MESSAGE on socket FD.
250 * Returns the number of bytes sent, or -1 for errors.
252 * This function is a cancellation point and therefore not marked with
255 ssize_t sw_sendmsg(int __fd, __const struct msghdr *__message,
260 return sendmsg(__fd, __message, __flags);
264 * Receive a message as described by MESSAGE from socket FD.
265 * Returns the number of bytes read or -1 for errors.
267 * This function is a cancellation point and therefore not marked with
270 ssize_t sw_recvmsg(int __fd, struct msghdr *__message, int __flags)
280 * Put the current value for socket FD's option OPTNAME at protocol level
281 * LEVEL into OPTVAL (which is *OPTLEN bytes long), and set *OPTLEN to the
282 * value's actual length. Returns 0 on success, -1 for errors.
284 int sw_getsockopt(int __fd, int __level, int __optname,
285 void *__restrict __optval,
286 socklen_t *__restrict __optlen)
288 /* Call classical interface of getsockopt(2). */
289 return getsockopt(__fd, __level, __optname, __optval, __optlen);
293 * Set socket FD's option OPTNAME at protocol level LEVEL
294 * to *OPTVAL (which is OPTLEN bytes long).
295 * Returns 0 on success, -1 for errors.
298 int sw_setsockopt(int __fd, int __level, int __optname,
299 __const void *__optval, socklen_t __optlen)
301 /* Call classical interface of setsockopt(2). */
302 return setsockopt(__fd, __level, __optname, __optval, __optlen);
306 * Shut down all or part of the connection open on socket FD.
307 * HOW determines what to shut down:
308 * SHUT_RD = No more receptions;
309 * SHUT_WR = No more transmissions;
310 * SHUT_RDWR = No more receptions or transmissions.
311 * Returns 0 on success, -1 for errors.
313 int sw_shutdown(int __fd, int __how)
315 struct sock_list *list;
317 /* Find socket in management structure. */
318 list = list_elem_from_socket(__fd);
324 /* Check and update socket state. */
325 if (__how == STATE_SHUT_RDWR)
326 list->rw_state = STATE_SHUT_RDWR;
327 else if (__how == STATE_SHUT_WR) {
328 if (list->rw_state == STATE_SHUT_RD)
329 list->rw_state = STATE_SHUT_RDWR;
330 else if (list->rw_state == STATE_SHUT_WR) {
335 else if (__how == STATE_SHUT_RD) {
336 if (list->rw_state == STATE_SHUT_WR)
337 list->rw_state = STATE_SHUT_RDWR;
338 else if (list->rw_state == STATE_SHUT_RD) {
344 /* Remove socket from socket management structure. */
345 if (list->rw_state == STATE_SHUT_RDWR) {
346 list = list_unlink_socket(__fd);
349 goto list_unlink_err;
353 /* Call classical interface of shutdown(2). */
354 return shutdown(__fd, __how);
363 * Close file descriptor for socket FD.
364 * Returns 0 on success, -1 for errors.
366 int sw_close(int __fd)
368 struct sock_list *list;
370 /* Remove socket from socket management structure. */
371 list = list_unlink_socket(__fd);
374 goto list_unlink_err;
377 /* Call classical interface of close(2). */