/[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.46 - (hide annotations)
Tue Jun 3 05:33:06 2025 UTC (9 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.45: +2 -0 lines
Content type: text/x-csrc
Update include / lib path

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

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