Make the socket parser more general
[olsrd.git] / src / socket_parser.c
1 /*
2  * The olsr.org Optimized Link-State Routing daemon(olsrd)
3  * Copyright (c) 2004, Andreas T√łnnesen(andreto@olsr.org)
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without 
7  * modification, are permitted provided that the following conditions 
8  * are met:
9  *
10  * * Redistributions of source code must retain the above copyright 
11  *   notice, this list of conditions and the following disclaimer.
12  * * Redistributions in binary form must reproduce the above copyright 
13  *   notice, this list of conditions and the following disclaimer in 
14  *   the documentation and/or other materials provided with the 
15  *   distribution.
16  * * Neither the name of olsr.org, olsrd nor the names of its 
17  *   contributors may be used to endorse or promote products derived 
18  *   from this software without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 
23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 
24  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 
25  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 
26  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
27  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 
28  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 
30  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
31  * POSSIBILITY OF SUCH DAMAGE.
32  *
33  * Visit http://www.olsr.org for more information.
34  *
35  * If you find this software useful feel free to make a donation
36  * to the project. For more information see the website or contact
37  * the copyright holders.
38  *
39  */
40
41 #include "socket_parser.h"
42 #include "scheduler.h"
43 #include "olsr.h"
44 #include "defs.h"
45 #include "log.h"
46 #include "net_os.h"
47
48 #include <errno.h>
49 #include <stdlib.h>
50
51
52 #ifdef WIN32
53 #undef EINTR
54 #define EINTR WSAEINTR
55 #undef errno
56 #define errno WSAGetLastError()
57 #undef strerror
58 #define strerror(x) StrError(x)
59 #endif
60
61 static struct olsr_socket_entry *olsr_socket_entries = NULL;
62
63 /**
64  * Add a socket and handler to the socketset
65  * beeing used in the main select(2) loop
66  * in listen_loop
67  *
68  *@param fd the socket
69  *@param pf the processing function
70  */
71 void
72 add_olsr_socket(int fd, socket_handler_func pf_pr, socket_handler_func pf_imm, void *data, unsigned int flags)
73 {
74   struct olsr_socket_entry *new_entry;
75
76   if (fd < 0 || (pf_pr == NULL && pf_imm == NULL)) {
77     olsr_syslog(OLSR_LOG_ERR, "%s: Bogus socket entry - not registering...", __func__);
78     return;
79   }
80   OLSR_PRINTF(2, "Adding OLSR socket entry %d\n", fd);
81
82   new_entry = olsr_malloc(sizeof(*new_entry), "Socket entry");
83
84   new_entry->fd = fd;
85   new_entry->process_immediate = pf_imm;
86   new_entry->process_pollrate = pf_pr;
87   new_entry->data = data;
88   new_entry->flags = flags;
89
90   /* Queue */
91   new_entry->next = olsr_socket_entries;
92   olsr_socket_entries = new_entry;
93 }
94
95 /**
96  * Remove a socket and handler to the socketset
97  * beeing used in the main select(2) loop
98  * in listen_loop
99  *
100  *@param fd the socket
101  *@param pf the processing function
102  */
103 int
104 remove_olsr_socket(int fd, socket_handler_func pf_pr, socket_handler_func pf_imm)
105 {
106   struct olsr_socket_entry *entry, *prev_entry;
107
108   if (fd < 0 || (pf_pr == NULL && pf_imm == NULL)) {
109     olsr_syslog(OLSR_LOG_ERR, "%s: Bogus socket entry - not processing...", __func__);
110     return 0;
111   }
112   OLSR_PRINTF(1, "Removing OLSR socket entry %d\n", fd);
113
114   for (entry = olsr_socket_entries, prev_entry = NULL;
115        entry != NULL;
116        prev_entry = entry, entry = entry->next) {
117     if (entry->fd == fd && entry->process_immediate == pf_imm && entry->process_pollrate == pf_pr) {
118       if (prev_entry == NULL) {
119         olsr_socket_entries = entry->next;
120       } else {
121         prev_entry->next = entry->next;
122       }
123       free(entry);
124       return 1;
125     }
126   }
127   return 0;
128 }
129
130
131 void
132 olsr_poll_sockets(void)
133 {
134   int n;
135   struct olsr_socket_entry *entry;
136   fd_set ibits, obits;
137   struct timeval tvp = { 0, 0 };
138   int hfd = 0, fdsets = 0;
139   const char * err_msg;
140   /* If there are no registered sockets we
141    * do not call select(2)
142    */
143   if (olsr_socket_entries == NULL) {
144     return;
145   }
146
147   FD_ZERO(&ibits);
148   FD_ZERO(&obits);
149   
150   /* Adding file-descriptors to FD set */
151   for (entry = olsr_socket_entries; entry != NULL; entry = entry->next) {
152     if (entry->process_pollrate == NULL) {
153       continue;
154     }
155     if ((entry->flags & SP_PR_READ) != 0) {
156       fdsets |= SP_PR_READ;
157       FD_SET((unsigned int)entry->fd, &ibits); /* And we cast here since we get a warning on Win32 */    
158     }
159     if ((entry->flags & SP_PR_WRITE) != 0) {
160       fdsets |= SP_PR_WRITE;
161       FD_SET((unsigned int)entry->fd, &obits); /* And we cast here since we get a warning on Win32 */    
162     }
163     if ((entry->flags & (SP_PR_READ|SP_PR_READ)) != 0) {
164       if (entry->fd >= hfd) {
165         hfd = entry->fd + 1;
166       }
167     }
168   }
169
170   if (hfd == 0) {
171     /* we didn't set anything - no need to continue */
172     return;
173   }
174       
175   /* Running select on the FD set */
176   do {
177     n = olsr_select(hfd, 
178                     fdsets & SP_PR_READ ? &ibits : NULL,
179                     fdsets & SP_PR_WRITE ? &obits : NULL,
180                     NULL,
181                     &tvp);
182   } while (n == -1 && (errno == EINTR || errno == EAGAIN));
183
184   switch (n) {
185   case 0:
186     break;
187
188   case -1:      /* Did somethig go wrong? */
189     err_msg = strerror(errno);
190     olsr_syslog(OLSR_LOG_ERR, "select: %s", err_msg);
191     OLSR_PRINTF(1, "Error select: %s", err_msg);
192     break;
193
194   default:      /* Update time since this is much used by the parsing functions */
195     now_times = olsr_times();
196     for (entry = olsr_socket_entries; entry != NULL; entry = entry->next) {
197       int rd, wr;
198       if (entry->process_pollrate == NULL) {
199         continue;
200       }
201       rd = (entry->flags & SP_PR_READ) != 0 && FD_ISSET(entry->fd, &ibits);
202       wr = (entry->flags & SP_PR_WRITE) != 0 && FD_ISSET(entry->fd, &obits);
203       if (rd && wr) {
204         entry->process_pollrate(entry->fd, entry->data, SP_PR_READ|SP_PR_WRITE);
205       } else if (wr) {
206         entry->process_pollrate(entry->fd, entry->data, SP_PR_READ);
207       } else if (rd) {
208         entry->process_pollrate(entry->fd, entry->data, SP_PR_WRITE);
209       }
210     }
211     break;
212   }
213 }
214
215 /*
216  * Local Variables:
217  * c-basic-offset: 2
218  * End:
219  */