/[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.45 - (hide annotations)
Sun Jun 1 14:14:34 2025 UTC (9 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.44: +12 -9 lines
Content type: text/x-csrc
Skip pid cleanup from sockaddr pool for section list loader process

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 sysadm 1.45 if (siginfo.si_pid != section_list_loader_pid)
160 sysadm 1.44 {
161 sysadm 1.45 i = 0;
162     for (; i < BBS_max_client; i++)
163 sysadm 1.44 {
164 sysadm 1.45 if (process_sockaddr_pool[i].pid == siginfo.si_pid)
165     {
166     process_sockaddr_pool[i].pid = 0;
167     break;
168     }
169     }
170     if (i >= BBS_max_client)
171     {
172     log_error("Child process (%d) not found in process sockaddr pool\n", siginfo.si_pid);
173 sysadm 1.44 }
174     }
175 sysadm 1.19 }
176     else if (ret == 0)
177     {
178     break;
179     }
180     else if (ret < 0)
181     {
182     log_error("Error in waitid: %d\n", errno);
183     break;
184     }
185     }
186    
187 sysadm 1.22 if (SYS_server_exit && !SYS_child_exit && SYS_child_process_count > 0)
188 sysadm 1.21 {
189 sysadm 1.22 log_std("Notify %d child process to exit\n", SYS_child_process_count);
190 sysadm 1.21 if (kill(0, SIGTERM) < 0)
191     {
192     log_error("Send SIGTERM signal failed (%d)\n", errno);
193     }
194 sysadm 1.31
195     sd_notifyf(0, "STATUS=Waiting for %d child process to exit", SYS_child_process_count);
196 sysadm 1.21 }
197    
198 sysadm 1.44 if (SYS_conf_reload && !SYS_server_exit)
199 sysadm 1.19 {
200 sysadm 1.44 SYS_conf_reload = 0;
201 sysadm 1.37 sd_notify(0, "RELOADING=1");
202 sysadm 1.30
203 sysadm 1.44 // Reload configuration
204     if (load_conf(CONF_BBSD) < 0)
205     {
206     log_error("Reload conf failed\n");
207     }
208    
209 sysadm 1.34 p_bbs_menu_new = calloc(1, sizeof(MENU_SET));
210     if (p_bbs_menu_new == NULL)
211     {
212     log_error("OOM: calloc(MENU_SET)\n");
213     }
214     else if (load_menu(p_bbs_menu_new, CONF_MENU) < 0)
215     {
216     unload_menu(p_bbs_menu_new);
217     free(p_bbs_menu_new);
218 sysadm 1.38 p_bbs_menu_new = NULL;
219 sysadm 1.33
220 sysadm 1.19 log_error("Reload menu failed\n");
221     }
222     else
223     {
224 sysadm 1.34 unload_menu(p_bbs_menu);
225     free(p_bbs_menu);
226    
227     p_bbs_menu = p_bbs_menu_new;
228     p_bbs_menu_new = NULL;
229    
230 sysadm 1.19 log_std("Reload menu successfully\n");
231     }
232 sysadm 1.37
233     sd_notify(0, "READY=1");
234 sysadm 1.19 }
235    
236 sysadm 1.35 if (SYS_data_file_reload && !SYS_server_exit)
237     {
238     SYS_data_file_reload = 0;
239 sysadm 1.37 sd_notify(0, "RELOADING=1");
240 sysadm 1.35
241     for (int i = 0; i < data_files_load_startup_count; i++)
242     {
243 sysadm 1.39 if (load_file(data_files_load_startup[i]) < 0)
244 sysadm 1.35 {
245     log_error("load_file_mmap(%s) error\n", data_files_load_startup[i]);
246     }
247     }
248    
249     log_std("Reload data files successfully\n");
250 sysadm 1.37 sd_notify(0, "READY=1");
251 sysadm 1.35 }
252    
253 sysadm 1.42 if (SYS_section_list_reload && !SYS_server_exit)
254     {
255     SYS_section_list_reload = 0;
256    
257     if (section_list_loader_reload() < 0)
258     {
259     log_error("ksection_list_loader_reload() failed\n");
260     }
261     }
262    
263 sysadm 1.25 nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second
264 sysadm 1.22
265 sysadm 1.25 if (nfds < 0)
266 sysadm 1.19 {
267     if (errno != EINTR)
268     {
269 sysadm 1.25 log_error("epoll_wait() error (%d)\n", errno);
270     break;
271 sysadm 1.19 }
272     continue;
273     }
274 sysadm 1.13
275 sysadm 1.19 // Stop accept new connection on exit
276     if (SYS_server_exit)
277 sysadm 1.13 {
278     continue;
279     }
280    
281 sysadm 1.25 for (int i = 0; i < nfds; i++)
282 sysadm 1.13 {
283 sysadm 1.25 if (events[i].data.fd == socket_server)
284 sysadm 1.13 {
285 sysadm 1.29 while (!SYS_server_exit) // Accept all incoming connections until error
286 sysadm 1.13 {
287 sysadm 1.25 socket_client = accept(socket_server, (struct sockaddr *)&sin, &namelen);
288     if (socket_client < 0)
289     {
290 sysadm 1.28 if (errno == EAGAIN || errno == EWOULDBLOCK)
291     {
292     break;
293     }
294     else if (errno == EINTR)
295     {
296     continue;
297     }
298     else
299 sysadm 1.25 {
300     log_error("accept(socket_server) error (%d)\n", errno);
301 sysadm 1.28 break;
302 sysadm 1.25 }
303     }
304    
305     strncpy(hostaddr_client, inet_ntoa(sin.sin_addr), sizeof(hostaddr_client) - 1);
306     hostaddr_client[sizeof(hostaddr_client) - 1] = '\0';
307    
308     port_client = ntohs(sin.sin_port);
309    
310     log_std("Accept connection from %s:%d\n", hostaddr_client, port_client);
311    
312 sysadm 1.43 if (SYS_child_process_count - 1 < BBS_max_client)
313 sysadm 1.25 {
314 sysadm 1.44 j = 0;
315     for (i = 0; i < BBS_max_client; i++)
316     {
317     if (process_sockaddr_pool[i].pid != 0 && process_sockaddr_pool[i].s_addr == sin.sin_addr.s_addr)
318     {
319     j++;
320     if (j >= BBS_max_client_per_ip)
321     {
322     log_error("Too many client connections (%d) from %s\n", j, hostaddr_client);
323     break;
324     }
325     }
326     }
327    
328     if (j < BBS_max_client_per_ip)
329 sysadm 1.43 {
330 sysadm 1.44 if ((pid = fork_server()) < 0)
331     {
332     log_error("fork_server() error\n");
333     }
334     else if (pid > 0)
335     {
336     i = 0;
337     for (; i < BBS_max_client; i++)
338     {
339     if (process_sockaddr_pool[i].pid == 0)
340     {
341     break;
342     }
343     }
344    
345     if (i >= BBS_max_client)
346     {
347     log_error("Process sockaddr pool depleted\n");
348     }
349     else
350     {
351     process_sockaddr_pool[i].pid = pid;
352     process_sockaddr_pool[i].s_addr = sin.sin_addr.s_addr;
353     }
354     }
355 sysadm 1.43 }
356     }
357     else
358     {
359     log_error("Rejected client connection over limit (%d)\n", SYS_child_process_count - 1);
360 sysadm 1.25 }
361    
362     if (close(socket_client) == -1)
363     {
364     log_error("close(socket_lient) error (%d)\n", errno);
365     }
366 sysadm 1.13 }
367     }
368     }
369 sysadm 1.25 }
370 sysadm 1.13
371 sysadm 1.30 if (close(epollfd) < 0)
372     {
373     log_error("close(epoll) error (%d)\n");
374     }
375    
376 sysadm 1.25 fcntl(socket_server, F_SETFL, flags);
377 sysadm 1.13
378     if (close(socket_server) == -1)
379     {
380     log_error("Close server socket failed\n");
381     }
382 sysadm 1.7
383 sysadm 1.13 return 0;
384 sysadm 1.1 }

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