/[LeafOK_CVS]/lbbs/src/net_server.c
ViewVC logotype

Annotation of /lbbs/src/net_server.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.44 - (hide annotations)
Sun Jun 1 03:07:42 2025 UTC (9 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.43: +74 -4 lines
Content type: text/x-csrc
Add max connection and connection per ip limit

1 sysadm 1.1 /***************************************************************************
2 sysadm 1.13 net_server.c - description
3     -------------------
4 sysadm 1.18 Copyright : (C) 2004-2025 by Leaflet
5     Email : leaflet@leafok.com
6 sysadm 1.1 ***************************************************************************/
7    
8     /***************************************************************************
9     * *
10     * This program is free software; you can redistribute it and/or modify *
11     * it under the terms of the GNU General Public License as published by *
12 sysadm 1.18 * the Free Software Foundation; either version 3 of the License, or *
13 sysadm 1.1 * (at your option) any later version. *
14     * *
15     ***************************************************************************/
16    
17 sysadm 1.19 #define _XOPEN_SOURCE 500
18     #define _POSIX_C_SOURCE 200809L
19     #define _GNU_SOURCE
20    
21 sysadm 1.14 #include "net_server.h"
22 sysadm 1.1 #include "common.h"
23 sysadm 1.14 #include "log.h"
24 sysadm 1.7 #include "io.h"
25 sysadm 1.44 #include "init.h"
26 sysadm 1.14 #include "fork.h"
27 sysadm 1.19 #include "menu.h"
28 sysadm 1.35 #include "file_loader.h"
29 sysadm 1.42 #include "section_list_loader.h"
30 sysadm 1.20 #include <errno.h>
31     #include <fcntl.h>
32     #include <string.h>
33 sysadm 1.19 #include <signal.h>
34 sysadm 1.20 #include <stdlib.h>
35 sysadm 1.19 #include <unistd.h>
36     #include <sys/syscall.h>
37 sysadm 1.7 #include <sys/socket.h>
38 sysadm 1.19 #include <sys/wait.h>
39 sysadm 1.25 #include <sys/epoll.h>
40 sysadm 1.9 #include <arpa/inet.h>
41 sysadm 1.31 #include <systemd/sd-daemon.h>
42 sysadm 1.3
43 sysadm 1.44 struct process_sockaddr_t
44     {
45     pid_t pid;
46     in_addr_t s_addr;
47     };
48     typedef struct process_sockaddr_t PROCESS_SOCKADDR;
49    
50     static PROCESS_SOCKADDR process_sockaddr_pool[MAX_CLIENTS_LIMIT];
51    
52 sysadm 1.17 int net_server(const char *hostaddr, in_port_t port)
53 sysadm 1.1 {
54 sysadm 1.15 unsigned int namelen;
55 sysadm 1.19 int ret;
56 sysadm 1.15 int flags;
57 sysadm 1.13 struct sockaddr_in sin;
58 sysadm 1.25 struct epoll_event ev, events[MAX_EVENTS];
59     int nfds, epollfd;
60 sysadm 1.19 siginfo_t siginfo;
61 sysadm 1.31 int sd_notify_stopping = 0;
62 sysadm 1.34 MENU_SET *p_bbs_menu_new;
63 sysadm 1.44 int i, j;
64     pid_t pid;
65 sysadm 1.10
66 sysadm 1.13 socket_server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
67 sysadm 1.10
68 sysadm 1.13 if (socket_server < 0)
69     {
70     log_error("Create socket failed\n");
71 sysadm 1.27 return -1;
72 sysadm 1.13 }
73 sysadm 1.10
74 sysadm 1.13 sin.sin_family = AF_INET;
75 sysadm 1.23 sin.sin_addr.s_addr = (hostaddr[0] != '\0' ? inet_addr(hostaddr) : INADDR_ANY);
76 sysadm 1.13 sin.sin_port = htons(port);
77    
78 sysadm 1.23 // Reuse address and port
79     flags = 1;
80     if (setsockopt(socket_server, SOL_SOCKET, SO_REUSEADDR, &flags, sizeof(flags)) < 0)
81     {
82     log_error("setsockopt SO_REUSEADDR error (%d)\n", errno);
83     }
84     if (setsockopt(socket_server, SOL_SOCKET, SO_REUSEPORT, &flags, sizeof(flags)) < 0)
85     {
86     log_error("setsockopt SO_REUSEPORT error (%d)\n", errno);
87     }
88    
89 sysadm 1.13 if (bind(socket_server, (struct sockaddr *)&sin, sizeof(sin)) < 0)
90 sysadm 1.10 {
91 sysadm 1.23 log_error("Bind address %s:%u failed (%d)\n",
92     inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), errno);
93 sysadm 1.27 return -1;
94 sysadm 1.10 }
95    
96 sysadm 1.13 if (listen(socket_server, 10) < 0)
97 sysadm 1.10 {
98 sysadm 1.23 log_error("Socket listen failed (%d)\n", errno);
99 sysadm 1.27 return -1;
100 sysadm 1.10 }
101 sysadm 1.1
102 sysadm 1.16 strncpy(hostaddr_server, inet_ntoa(sin.sin_addr), sizeof(hostaddr_server) - 1);
103     hostaddr_server[sizeof(hostaddr_server) - 1] = '\0';
104    
105 sysadm 1.13 port_server = ntohs(sin.sin_port);
106 sysadm 1.19 namelen = sizeof(sin);
107 sysadm 1.13
108     log_std("Listening at %s:%d\n", hostaddr_server, port_server);
109    
110 sysadm 1.25 epollfd = epoll_create1(0);
111     if (epollfd < 0)
112     {
113     log_error("epoll_create1() error (%d)\n", errno);
114     return -1;
115     }
116    
117 sysadm 1.28 ev.events = EPOLLIN;
118 sysadm 1.25 ev.data.fd = socket_server;
119     if (epoll_ctl(epollfd, EPOLL_CTL_ADD, socket_server, &ev) == -1)
120     {
121     log_error("epoll_ctl(socket_server) error (%d)\n", errno);
122 sysadm 1.30 if (close(epollfd) < 0)
123     {
124     log_error("close(epoll) error (%d)\n");
125     }
126 sysadm 1.25 return -1;
127     }
128    
129     flags = fcntl(socket_server, F_GETFL, 0);
130     fcntl(socket_server, F_SETFL, flags | O_NONBLOCK);
131    
132 sysadm 1.31 // Startup complete
133     sd_notifyf(0, "READY=1\n"
134 sysadm 1.32 "STATUS=Listening at %s:%d\n"
135 sysadm 1.31 "MAINPID=%d",
136 sysadm 1.32 hostaddr_server, port_server, getpid());
137 sysadm 1.31
138 sysadm 1.19 while (!SYS_server_exit || SYS_child_process_count > 0)
139 sysadm 1.13 {
140 sysadm 1.31 if (SYS_server_exit && !sd_notify_stopping)
141     {
142     sd_notify(0, "STOPPING=1");
143     sd_notify_stopping = 1;
144     }
145 sysadm 1.37
146 sysadm 1.22 while ((SYS_child_exit || SYS_server_exit) && SYS_child_process_count > 0)
147 sysadm 1.19 {
148 sysadm 1.30 SYS_child_exit = 0;
149    
150 sysadm 1.19 siginfo.si_pid = 0;
151     ret = waitid(P_ALL, 0, &siginfo, WEXITED | WNOHANG);
152     if (ret == 0 && siginfo.si_pid > 0)
153     {
154 sysadm 1.30 SYS_child_exit = 1; // Retry waitid
155    
156 sysadm 1.19 SYS_child_process_count--;
157     log_std("Child process (%d) exited\n", siginfo.si_pid);
158 sysadm 1.44
159     i = 0;
160     for (; i < BBS_max_client; i++)
161     {
162     if (process_sockaddr_pool[i].pid == siginfo.si_pid)
163     {
164     process_sockaddr_pool[i].pid = 0;
165     break;
166     }
167     }
168     if (i >= BBS_max_client)
169     {
170     log_error("Child process (%d) not found in process sockaddr pool\n", siginfo.si_pid);
171     }
172 sysadm 1.19 }
173     else if (ret == 0)
174     {
175     break;
176     }
177     else if (ret < 0)
178     {
179     log_error("Error in waitid: %d\n", errno);
180     break;
181     }
182     }
183    
184 sysadm 1.22 if (SYS_server_exit && !SYS_child_exit && SYS_child_process_count > 0)
185 sysadm 1.21 {
186 sysadm 1.22 log_std("Notify %d child process to exit\n", SYS_child_process_count);
187 sysadm 1.21 if (kill(0, SIGTERM) < 0)
188     {
189     log_error("Send SIGTERM signal failed (%d)\n", errno);
190     }
191 sysadm 1.31
192     sd_notifyf(0, "STATUS=Waiting for %d child process to exit", SYS_child_process_count);
193 sysadm 1.21 }
194    
195 sysadm 1.44 if (SYS_conf_reload && !SYS_server_exit)
196 sysadm 1.19 {
197 sysadm 1.44 SYS_conf_reload = 0;
198 sysadm 1.37 sd_notify(0, "RELOADING=1");
199 sysadm 1.30
200 sysadm 1.44 // Reload configuration
201     if (load_conf(CONF_BBSD) < 0)
202     {
203     log_error("Reload conf failed\n");
204     }
205    
206 sysadm 1.34 p_bbs_menu_new = calloc(1, sizeof(MENU_SET));
207     if (p_bbs_menu_new == NULL)
208     {
209     log_error("OOM: calloc(MENU_SET)\n");
210     }
211     else if (load_menu(p_bbs_menu_new, CONF_MENU) < 0)
212     {
213     unload_menu(p_bbs_menu_new);
214     free(p_bbs_menu_new);
215 sysadm 1.38 p_bbs_menu_new = NULL;
216 sysadm 1.33
217 sysadm 1.19 log_error("Reload menu failed\n");
218     }
219     else
220     {
221 sysadm 1.34 unload_menu(p_bbs_menu);
222     free(p_bbs_menu);
223    
224     p_bbs_menu = p_bbs_menu_new;
225     p_bbs_menu_new = NULL;
226    
227 sysadm 1.19 log_std("Reload menu successfully\n");
228     }
229 sysadm 1.37
230     sd_notify(0, "READY=1");
231 sysadm 1.19 }
232    
233 sysadm 1.35 if (SYS_data_file_reload && !SYS_server_exit)
234     {
235     SYS_data_file_reload = 0;
236 sysadm 1.37 sd_notify(0, "RELOADING=1");
237 sysadm 1.35
238     for (int i = 0; i < data_files_load_startup_count; i++)
239     {
240 sysadm 1.39 if (load_file(data_files_load_startup[i]) < 0)
241 sysadm 1.35 {
242     log_error("load_file_mmap(%s) error\n", data_files_load_startup[i]);
243     }
244     }
245    
246     log_std("Reload data files successfully\n");
247 sysadm 1.37 sd_notify(0, "READY=1");
248 sysadm 1.35 }
249    
250 sysadm 1.42 if (SYS_section_list_reload && !SYS_server_exit)
251     {
252     SYS_section_list_reload = 0;
253    
254     if (section_list_loader_reload() < 0)
255     {
256     log_error("ksection_list_loader_reload() failed\n");
257     }
258     }
259    
260 sysadm 1.25 nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second
261 sysadm 1.22
262 sysadm 1.25 if (nfds < 0)
263 sysadm 1.19 {
264     if (errno != EINTR)
265     {
266 sysadm 1.25 log_error("epoll_wait() error (%d)\n", errno);
267     break;
268 sysadm 1.19 }
269     continue;
270     }
271 sysadm 1.13
272 sysadm 1.19 // Stop accept new connection on exit
273     if (SYS_server_exit)
274 sysadm 1.13 {
275     continue;
276     }
277    
278 sysadm 1.25 for (int i = 0; i < nfds; i++)
279 sysadm 1.13 {
280 sysadm 1.25 if (events[i].data.fd == socket_server)
281 sysadm 1.13 {
282 sysadm 1.29 while (!SYS_server_exit) // Accept all incoming connections until error
283 sysadm 1.13 {
284 sysadm 1.25 socket_client = accept(socket_server, (struct sockaddr *)&sin, &namelen);
285     if (socket_client < 0)
286     {
287 sysadm 1.28 if (errno == EAGAIN || errno == EWOULDBLOCK)
288     {
289     break;
290     }
291     else if (errno == EINTR)
292     {
293     continue;
294     }
295     else
296 sysadm 1.25 {
297     log_error("accept(socket_server) error (%d)\n", errno);
298 sysadm 1.28 break;
299 sysadm 1.25 }
300     }
301    
302     strncpy(hostaddr_client, inet_ntoa(sin.sin_addr), sizeof(hostaddr_client) - 1);
303     hostaddr_client[sizeof(hostaddr_client) - 1] = '\0';
304    
305     port_client = ntohs(sin.sin_port);
306    
307     log_std("Accept connection from %s:%d\n", hostaddr_client, port_client);
308    
309 sysadm 1.43 if (SYS_child_process_count - 1 < BBS_max_client)
310 sysadm 1.25 {
311 sysadm 1.44 j = 0;
312     for (i = 0; i < BBS_max_client; i++)
313     {
314     if (process_sockaddr_pool[i].pid != 0 && process_sockaddr_pool[i].s_addr == sin.sin_addr.s_addr)
315     {
316     j++;
317     if (j >= BBS_max_client_per_ip)
318     {
319     log_error("Too many client connections (%d) from %s\n", j, hostaddr_client);
320     break;
321     }
322     }
323     }
324    
325     if (j < BBS_max_client_per_ip)
326 sysadm 1.43 {
327 sysadm 1.44 if ((pid = fork_server()) < 0)
328     {
329     log_error("fork_server() error\n");
330     }
331     else if (pid > 0)
332     {
333     i = 0;
334     for (; i < BBS_max_client; i++)
335     {
336     if (process_sockaddr_pool[i].pid == 0)
337     {
338     break;
339     }
340     }
341    
342     if (i >= BBS_max_client)
343     {
344     log_error("Process sockaddr pool depleted\n");
345     }
346     else
347     {
348     process_sockaddr_pool[i].pid = pid;
349     process_sockaddr_pool[i].s_addr = sin.sin_addr.s_addr;
350     }
351     }
352 sysadm 1.43 }
353     }
354     else
355     {
356     log_error("Rejected client connection over limit (%d)\n", SYS_child_process_count - 1);
357 sysadm 1.25 }
358    
359     if (close(socket_client) == -1)
360     {
361     log_error("close(socket_lient) error (%d)\n", errno);
362     }
363 sysadm 1.13 }
364     }
365     }
366 sysadm 1.25 }
367 sysadm 1.13
368 sysadm 1.30 if (close(epollfd) < 0)
369     {
370     log_error("close(epoll) error (%d)\n");
371     }
372    
373 sysadm 1.25 fcntl(socket_server, F_SETFL, flags);
374 sysadm 1.13
375     if (close(socket_server) == -1)
376     {
377     log_error("Close server socket failed\n");
378     }
379 sysadm 1.7
380 sysadm 1.13 return 0;
381 sysadm 1.1 }

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1