Mark socket as bound in list_update_socket_address.
[swifty.git] / src / raw / swift_list.c
1 /*
2  * swift implementation of syscall API
3  *
4  * Simulates the classic socket syscalls (socket, bind, send, recv).
5  * Implementation uses raw sockets (AF_INET, SOCK_RAW).
6  *
7  * Subsequently, implementation is to be ported into kernel space and
8  * the interface is going to be offered by the Linux syscall API.
9  *
10  * Heavily inspired by GLIBC's <sys/socket.h>
11  * (/usr/include/sys/socket.h).
12  *
13  * 2011, Razvan Deaconescu, razvan.deaconescu@cs.pub.ro
14  */
15
16 #include <stdio.h>
17 #include <stdlib.h>
18 #include <string.h>
19 #include <sys/types.h>
20 #include <sys/socket.h>
21 #include <netinet/in.h>
22 #include <unistd.h>
23 #include <errno.h>
24
25 #include "include/swift_types.h"
26 #include "include/swift_list.h"
27 #include "debug.h"
28
29 /*
30  * Add new socket to list. Called by sw_socket "syscall".
31  */
32
33 struct sock_list *list_add_socket(int s)
34 {
35         struct sock_list *ptr = malloc(sizeof(*ptr));
36         if (ptr == NULL)
37                 return NULL;
38
39         ptr->next = &sock_list_head;
40         ptr->prev = sock_list_head.prev;
41         sock_list_head.prev->next = ptr;
42         sock_list_head.prev = ptr;
43         ptr->s = s;
44
45         return ptr;
46 }
47
48 /*
49  * Bind socket to given address. Called by sw_bind "syscall".
50  */
51
52 struct sock_list *list_update_socket_address(int s, __CONST_SOCKADDR_ARG addr)
53 {
54         struct sock_list *ptr;
55
56         for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
57                 if (ptr->s == s) {
58                         memcpy(&ptr->addr, addr, sizeof(ptr->addr));
59                         ptr->bind_state = STATE_BOUND;
60                         return ptr;
61                 }
62
63         return NULL;
64 }
65
66 /*
67  * Get list element containing socket s. Called by sw_send* "syscalls".
68  */
69
70 struct sock_list *list_elem_from_socket(int s)
71 {
72         struct sock_list *ptr;
73
74         for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
75                 if (ptr->s == s)
76                         return ptr;
77
78         return NULL;
79 }
80
81 /*
82  * Get list element containing address addr. Called by sw_bind "syscall".
83  */
84
85 struct sock_list *list_elem_from_address(__CONST_SOCKADDR_ARG addr)
86 {
87         struct sock_list *ptr;
88
89         for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next) {
90                 dprintf("socket address to be checked\n");
91                 if (ptr->bind_state == STATE_NOTBOUND)
92                         continue;
93                 dprintf("bound socket address to be checked\n");
94                 if (memcmp(&ptr->addr, addr, sizeof(addr)) == 0)
95                         return ptr;
96         }
97
98         return NULL;
99 }
100
101 /*
102  * Unlink socket from list. Called by list_remove_socket.
103  */
104 static struct sock_list *list_unlink_socket(int s)
105 {
106         struct sock_list *ptr;
107
108         for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
109                 if (ptr->s == s) {
110                         ptr->next->prev = ptr->prev;
111                         ptr->prev->next = ptr->next;
112                         ptr->next = ptr;
113                         ptr->prev = ptr;
114                         return ptr;
115                 }
116
117         return NULL;
118 }
119
120 /*
121  * Remove socket from list. Called by sw_close "syscall".
122  */
123
124 int list_remove_socket(int s)
125 {
126         struct sock_list *ptr;
127
128         ptr = list_unlink_socket(s);
129         if (ptr == NULL)
130                 return -1;
131
132         free(ptr);
133         return 0;
134 }
135
136 /*
137  * Check if a socket is bound.
138  */
139 int list_socket_is_bound(int s)
140 {
141         struct sock_list *ptr;
142
143         for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
144                 if (ptr->s == s) {
145                         if (ptr->bind_state == STATE_BOUND)
146                                 return 1;
147                         break;
148                 }
149
150         return 0;
151 }