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

Diff of /lbbs/src/net_server.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.24 by sysadm, Sun May 11 04:09:08 2025 UTC Revision 1.30 by sysadm, Tue May 13 07:30:53 2025 UTC
# Line 33  Line 33 
33  #include <sys/syscall.h>  #include <sys/syscall.h>
34  #include <sys/socket.h>  #include <sys/socket.h>
35  #include <sys/wait.h>  #include <sys/wait.h>
36    #include <sys/epoll.h>
37  #include <arpa/inet.h>  #include <arpa/inet.h>
38    
39  int net_server(const char *hostaddr, in_port_t port)  int net_server(const char *hostaddr, in_port_t port)
# Line 41  int net_server(const char *hostaddr, in_ Line 42  int net_server(const char *hostaddr, in_
42          int ret;          int ret;
43          int flags;          int flags;
44          struct sockaddr_in sin;          struct sockaddr_in sin;
45          fd_set testfds;          struct epoll_event ev, events[MAX_EVENTS];
46          struct timeval timeout;          int nfds, epollfd;
         sigset_t nsigset;  
         sigset_t osigset;  
47          siginfo_t siginfo;          siginfo_t siginfo;
48    
49          socket_server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);          socket_server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
# Line 52  int net_server(const char *hostaddr, in_ Line 51  int net_server(const char *hostaddr, in_
51          if (socket_server < 0)          if (socket_server < 0)
52          {          {
53                  log_error("Create socket failed\n");                  log_error("Create socket failed\n");
54                  exit(1);                  return -1;
55          }          }
56    
57          sin.sin_family = AF_INET;          sin.sin_family = AF_INET;
# Line 74  int net_server(const char *hostaddr, in_ Line 73  int net_server(const char *hostaddr, in_
73          {          {
74                  log_error("Bind address %s:%u failed (%d)\n",                  log_error("Bind address %s:%u failed (%d)\n",
75                                    inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), errno);                                    inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), errno);
76                  exit(2);                  return -1;
77          }          }
78    
79          if (listen(socket_server, 10) < 0)          if (listen(socket_server, 10) < 0)
80          {          {
81                  log_error("Socket listen failed (%d)\n", errno);                  log_error("Socket listen failed (%d)\n", errno);
82                  exit(3);                  return -1;
83          }          }
84    
85          strncpy(hostaddr_server, inet_ntoa(sin.sin_addr), sizeof(hostaddr_server) - 1);          strncpy(hostaddr_server, inet_ntoa(sin.sin_addr), sizeof(hostaddr_server) - 1);
# Line 91  int net_server(const char *hostaddr, in_ Line 90  int net_server(const char *hostaddr, in_
90    
91          log_std("Listening at %s:%d\n", hostaddr_server, port_server);          log_std("Listening at %s:%d\n", hostaddr_server, port_server);
92    
93          sigemptyset(&nsigset);          epollfd = epoll_create1(0);
94          sigaddset(&nsigset, SIGHUP);          if (epollfd < 0)
95          sigaddset(&nsigset, SIGCHLD);          {
96          sigaddset(&nsigset, SIGTERM);                  log_error("epoll_create1() error (%d)\n", errno);
97                    return -1;
98            }
99    
100          while (!SYS_server_exit || SYS_child_process_count > 0)          ev.events = EPOLLIN;
101            ev.data.fd = socket_server;
102            if (epoll_ctl(epollfd, EPOLL_CTL_ADD, socket_server, &ev) == -1)
103          {          {
104                  sigprocmask(SIG_BLOCK, &nsigset, &osigset);                  log_error("epoll_ctl(socket_server) error (%d)\n", errno);
105                    if (close(epollfd) < 0)
106                    {
107                            log_error("close(epoll) error (%d)\n");
108                    }
109                    return -1;
110            }
111    
112            flags = fcntl(socket_server, F_GETFL, 0);
113            fcntl(socket_server, F_SETFL, flags | O_NONBLOCK);
114    
115            while (!SYS_server_exit || SYS_child_process_count > 0)
116            {
117                  while ((SYS_child_exit || SYS_server_exit) && SYS_child_process_count > 0)                  while ((SYS_child_exit || SYS_server_exit) && SYS_child_process_count > 0)
118                  {                  {
119                            SYS_child_exit = 0;
120    
121                          siginfo.si_pid = 0;                          siginfo.si_pid = 0;
122                          ret = waitid(P_ALL, 0, &siginfo, WEXITED | WNOHANG);                          ret = waitid(P_ALL, 0, &siginfo, WEXITED | WNOHANG);
123                          if (ret == 0 && siginfo.si_pid > 0)                          if (ret == 0 && siginfo.si_pid > 0)
124                          {                          {
125                                    SYS_child_exit = 1; // Retry waitid
126    
127                                  SYS_child_process_count--;                                  SYS_child_process_count--;
128                                  log_std("Child process (%d) exited\n", siginfo.si_pid);                                  log_std("Child process (%d) exited\n", siginfo.si_pid);
129                          }                          }
130                          else if (ret == 0)                          else if (ret == 0)
131                          {                          {
                                 SYS_child_exit = 0;  
132                                  break;                                  break;
133                          }                          }
134                          else if (ret < 0)                          else if (ret < 0)
# Line 132  int net_server(const char *hostaddr, in_ Line 149  int net_server(const char *hostaddr, in_
149    
150                  if (SYS_menu_reload && !SYS_server_exit)                  if (SYS_menu_reload && !SYS_server_exit)
151                  {                  {
152                            SYS_menu_reload = 0;
153    
154                          if (reload_menu(&bbs_menu) < 0)                          if (reload_menu(&bbs_menu) < 0)
155                          {                          {
156                                  log_error("Reload menu failed\n");                                  log_error("Reload menu failed\n");
# Line 140  int net_server(const char *hostaddr, in_ Line 159  int net_server(const char *hostaddr, in_
159                          {                          {
160                                  log_std("Reload menu successfully\n");                                  log_std("Reload menu successfully\n");
161                          }                          }
                         SYS_menu_reload = 0;  
162                  }                  }
163    
164                  sigprocmask(SIG_SETMASK, &osigset, NULL);                  nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second
   
                 FD_ZERO(&testfds);  
                 FD_SET(socket_server, &testfds);  
165    
166                  timeout.tv_sec = 0;                  if (nfds < 0)
                 timeout.tv_usec = 100 * 1000; // 0.1 second  
   
                 ret = select(socket_server + 1, &testfds, NULL, NULL, &timeout);  
   
                 if (ret < 0)  
167                  {                  {
168                          if (errno != EINTR)                          if (errno != EINTR)
169                          {                          {
170                                  log_error("Accept connection error: %d\n", errno);                                  log_error("epoll_wait() error (%d)\n", errno);
171                                    break;
172                          }                          }
173                          continue;                          continue;
174                  }                  }
                 else if (ret == 0) // timeout  
                 {  
                         continue;  
                 }  
175    
176                  // Stop accept new connection on exit                  // Stop accept new connection on exit
177                  if (SYS_server_exit)                  if (SYS_server_exit)
# Line 172  int net_server(const char *hostaddr, in_ Line 179  int net_server(const char *hostaddr, in_
179                          continue;                          continue;
180                  }                  }
181    
182                  if (FD_ISSET(socket_server, &testfds))                  for (int i = 0; i < nfds; i++)
183                  {                  {
184                          flags = fcntl(socket_server, F_GETFL, 0);                          if (events[i].data.fd == socket_server)
                         fcntl(socket_server, F_SETFL, flags | O_NONBLOCK);  
   
                         while ((socket_client =  
                                                 accept(socket_server, (struct sockaddr *)&sin, &namelen)) < 0)  
185                          {                          {
186                                  if (errno != EWOULDBLOCK && errno != ECONNABORTED && errno != EINTR)                                  while (!SYS_server_exit) // Accept all incoming connections until error
187                                  {                                  {
188                                          log_error("Accept connection error\n");                                          socket_client = accept(socket_server, (struct sockaddr *)&sin, &namelen);
189                                          break;                                          if (socket_client < 0)
190                                            {
191                                                    if (errno == EAGAIN || errno == EWOULDBLOCK)
192                                                    {
193                                                            break;
194                                                    }
195                                                    else if (errno == EINTR)
196                                                    {
197                                                            continue;
198                                                    }
199                                                    else
200                                                    {
201                                                            log_error("accept(socket_server) error (%d)\n", errno);
202                                                            break;
203                                                    }
204                                            }
205    
206                                            strncpy(hostaddr_client, inet_ntoa(sin.sin_addr), sizeof(hostaddr_client) - 1);
207                                            hostaddr_client[sizeof(hostaddr_client) - 1] = '\0';
208    
209                                            port_client = ntohs(sin.sin_port);
210    
211                                            log_std("Accept connection from %s:%d\n", hostaddr_client, port_client);
212    
213                                            if (fork_server() < 0)
214                                            {
215                                                    log_error("fork_server() error\n");
216                                            }
217    
218                                            if (close(socket_client) == -1)
219                                            {
220                                                    log_error("close(socket_lient) error (%d)\n", errno);
221                                            }
222                                  }                                  }
223                          }                          }
   
                         fcntl(socket_server, F_SETFL, flags);  
                 }  
   
                 if (socket_client < 0)  
                 {  
                         log_error("Accept connection error\n");  
                         continue;  
                 }  
   
                 strncpy(hostaddr_client, inet_ntoa(sin.sin_addr), sizeof(hostaddr_client) - 1);  
                 hostaddr_client[sizeof(hostaddr_client) - 1] = '\0';  
   
                 port_client = ntohs(sin.sin_port);  
   
                 log_std("Accept connection from %s:%d\n", hostaddr_client,  
                                 port_client);  
   
                 if (fork_server() < 0)  
                 {  
                         log_error("Fork error\n");  
224                  }                  }
225            }
226    
227                  if (close(socket_client) == -1)          if (close(epollfd) < 0)
228                  {          {
229                          log_error("Close client socket failed\n");                  log_error("close(epoll) error (%d)\n");
                 }  
230          }          }
231    
232            fcntl(socket_server, F_SETFL, flags);
233    
234          if (close(socket_server) == -1)          if (close(socket_server) == -1)
235          {          {
236                  log_error("Close server socket failed\n");                  log_error("Close server socket failed\n");

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1