/[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.48 - (hide annotations)
Wed Jun 4 13:27:09 2025 UTC (9 months, 1 week ago) by sysadm
Branch: MAIN
Changes since 1.47: +1 -1 lines
Content type: text/x-csrc
Refact config parser

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

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