| 38 |
#include <libssh/libssh.h> |
#include <libssh/libssh.h> |
| 39 |
#include <libssh/server.h> |
#include <libssh/server.h> |
| 40 |
#include <netinet/in.h> |
#include <netinet/in.h> |
| 41 |
#include <sys/epoll.h> |
#include <sys/ioctl.h> |
| 42 |
#include <sys/socket.h> |
#include <sys/socket.h> |
|
#include <sys/syscall.h> |
|
| 43 |
#include <sys/types.h> |
#include <sys/types.h> |
| 44 |
#include <sys/wait.h> |
#include <sys/wait.h> |
| 45 |
|
|
| 46 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 47 |
|
#include <sys/epoll.h> |
| 48 |
|
#else |
| 49 |
|
#include <poll.h> |
| 50 |
|
#endif |
| 51 |
|
|
| 52 |
#ifdef HAVE_SYSTEMD_SD_DAEMON_H |
#ifdef HAVE_SYSTEMD_SD_DAEMON_H |
| 53 |
#include <systemd/sd-daemon.h> |
#include <systemd/sd-daemon.h> |
| 54 |
#endif |
#endif |
| 61 |
SSH_AUTH_MAX_DURATION = 60 * 1000, // milliseconds |
SSH_AUTH_MAX_DURATION = 60 * 1000, // milliseconds |
| 62 |
}; |
}; |
| 63 |
|
|
|
static const char SFTP_SERVER_PATH[] = "/usr/lib/sftp-server"; |
|
|
|
|
| 64 |
/* A userdata struct for session. */ |
/* A userdata struct for session. */ |
| 65 |
struct session_data_struct |
struct session_data_struct |
| 66 |
{ |
{ |
| 87 |
struct winsize *winsize; |
struct winsize *winsize; |
| 88 |
}; |
}; |
| 89 |
|
|
| 90 |
|
static int socket_server[2]; |
| 91 |
|
static int socket_client; |
| 92 |
|
|
| 93 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 94 |
|
static int epollfd_server = -1; |
| 95 |
|
#endif |
| 96 |
|
|
| 97 |
|
static ssh_bind sshbind; |
| 98 |
|
|
| 99 |
|
static HASH_DICT *hash_dict_pid_sockaddr = NULL; |
| 100 |
|
static HASH_DICT *hash_dict_sockaddr_count = NULL; |
| 101 |
|
|
| 102 |
|
static const char SFTP_SERVER_PATH[] = "/usr/lib/sftp-server"; |
| 103 |
|
|
| 104 |
static int auth_password(ssh_session session, const char *user, |
static int auth_password(ssh_session session, const char *user, |
| 105 |
const char *password, void *userdata) |
const char *password, void *userdata) |
| 106 |
{ |
{ |
| 327 |
} |
} |
| 328 |
|
|
| 329 |
// Child process |
// Child process |
| 330 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 331 |
|
if (close(epollfd_server) < 0) |
| 332 |
|
{ |
| 333 |
|
log_error("close(epollfd_server) error (%d)\n"); |
| 334 |
|
} |
| 335 |
|
#endif |
| 336 |
|
|
| 337 |
if (close(socket_server[0]) == -1 || close(socket_server[1]) == -1) |
for (i = 0; i < 2; i++) |
| 338 |
{ |
{ |
| 339 |
log_error("Close server socket failed\n"); |
if (close(socket_server[i]) == -1) |
| 340 |
|
{ |
| 341 |
|
log_error("Close server socket failed\n"); |
| 342 |
|
} |
| 343 |
} |
} |
| 344 |
|
|
| 345 |
|
hash_dict_destroy(hash_dict_pid_sockaddr); |
| 346 |
|
hash_dict_destroy(hash_dict_sockaddr_count); |
| 347 |
|
|
| 348 |
SSH_session = ssh_new(); |
SSH_session = ssh_new(); |
| 349 |
|
|
| 350 |
if (SSH_v2) |
if (SSH_v2) |
| 422 |
log_error("Error setting SSH options: %s\n", ssh_get_error(SSH_session)); |
log_error("Error setting SSH options: %s\n", ssh_get_error(SSH_session)); |
| 423 |
goto cleanup; |
goto cleanup; |
| 424 |
} |
} |
| 425 |
|
|
| 426 |
|
ssh_set_blocking(SSH_session, 0); |
| 427 |
} |
} |
| 428 |
|
|
| 429 |
// Redirect Input |
// Redirect Input |
| 440 |
goto cleanup; |
goto cleanup; |
| 441 |
} |
} |
| 442 |
|
|
| 443 |
|
if (io_init() < 0) |
| 444 |
|
{ |
| 445 |
|
log_error("io_init() error\n"); |
| 446 |
|
goto cleanup; |
| 447 |
|
} |
| 448 |
|
|
| 449 |
SYS_child_process_count = 0; |
SYS_child_process_count = 0; |
| 450 |
|
|
| 451 |
|
// BWF compile |
| 452 |
|
if (bwf_compile() < 0) |
| 453 |
|
{ |
| 454 |
|
log_error("bwf_compile() error\n"); |
| 455 |
|
goto cleanup; |
| 456 |
|
} |
| 457 |
|
|
| 458 |
bbs_main(); |
bbs_main(); |
| 459 |
|
|
| 460 |
cleanup: |
cleanup: |
| 463 |
|
|
| 464 |
if (SSH_v2) |
if (SSH_v2) |
| 465 |
{ |
{ |
| 466 |
close(cdata.pty_master); |
if (cdata.pty_master != -1) |
| 467 |
close(cdata.child_stdin); |
{ |
| 468 |
close(cdata.child_stdout); |
close(cdata.pty_master); |
| 469 |
close(cdata.child_stderr); |
} |
| 470 |
|
if (cdata.child_stdin != -1) |
| 471 |
|
{ |
| 472 |
|
close(cdata.child_stdin); |
| 473 |
|
} |
| 474 |
|
if (cdata.child_stdout != -1) |
| 475 |
|
{ |
| 476 |
|
close(cdata.child_stdout); |
| 477 |
|
} |
| 478 |
|
if (cdata.child_stderr != -1) |
| 479 |
|
{ |
| 480 |
|
close(cdata.child_stderr); |
| 481 |
|
} |
| 482 |
|
|
| 483 |
ssh_channel_free(SSH_channel); |
ssh_channel_free(SSH_channel); |
| 484 |
ssh_disconnect(SSH_session); |
ssh_disconnect(SSH_session); |
| 491 |
ssh_free(SSH_session); |
ssh_free(SSH_session); |
| 492 |
ssh_finalize(); |
ssh_finalize(); |
| 493 |
|
|
| 494 |
|
// BWF cleanup |
| 495 |
|
bwf_cleanup(); |
| 496 |
|
|
| 497 |
// Close Input and Output for client |
// Close Input and Output for client |
| 498 |
|
io_cleanup(); |
| 499 |
close(STDIN_FILENO); |
close(STDIN_FILENO); |
| 500 |
close(STDOUT_FILENO); |
close(STDOUT_FILENO); |
| 501 |
|
|
| 509 |
|
|
| 510 |
int net_server(const char *hostaddr, in_port_t port[]) |
int net_server(const char *hostaddr, in_port_t port[]) |
| 511 |
{ |
{ |
|
HASH_DICT *hash_dict_pid_sockaddr = NULL; |
|
|
HASH_DICT *hash_dict_sockaddr_count = NULL; |
|
|
|
|
| 512 |
unsigned int addrlen; |
unsigned int addrlen; |
| 513 |
int ret; |
int ret; |
| 514 |
int flags[2]; |
int flags_server[2]; |
| 515 |
struct sockaddr_in sin; |
struct sockaddr_in sin; |
| 516 |
|
|
| 517 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 518 |
struct epoll_event ev, events[MAX_EVENTS]; |
struct epoll_event ev, events[MAX_EVENTS]; |
| 519 |
int nfds, epollfd; |
#else |
| 520 |
siginfo_t siginfo; |
struct pollfd pfds[2]; |
| 521 |
|
#endif |
| 522 |
|
|
| 523 |
|
int nfds; |
| 524 |
int notify_child_exit = 0; |
int notify_child_exit = 0; |
| 525 |
time_t tm_notify_child_exit = time(NULL); |
time_t tm_notify_child_exit = time(NULL); |
|
MENU_SET bbs_menu_new; |
|
|
MENU_SET top10_menu_new; |
|
| 526 |
int i, j; |
int i, j; |
| 527 |
pid_t pid; |
pid_t pid; |
| 528 |
|
int ssh_key_valid = 0; |
| 529 |
int ssh_log_level = SSH_LOG_NOLOG; |
int ssh_log_level = SSH_LOG_NOLOG; |
| 530 |
|
|
| 531 |
#ifdef HAVE_SYSTEMD_SD_DAEMON_H |
#ifdef HAVE_SYSTEMD_SD_DAEMON_H |
| 532 |
int sd_notify_stopping = 0; |
int sd_notify_stopping = 0; |
| 533 |
#endif |
#endif |
| 536 |
|
|
| 537 |
sshbind = ssh_bind_new(); |
sshbind = ssh_bind_new(); |
| 538 |
|
|
| 539 |
|
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_RSA_KEY_FILE) < 0) |
| 540 |
|
{ |
| 541 |
|
log_error("Error loading SSH RSA key: %s\n", SSH_HOST_RSA_KEY_FILE); |
| 542 |
|
} |
| 543 |
|
else |
| 544 |
|
{ |
| 545 |
|
ssh_key_valid = 1; |
| 546 |
|
} |
| 547 |
|
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_ED25519_KEY_FILE) < 0) |
| 548 |
|
{ |
| 549 |
|
log_error("Error loading SSH ED25519 key: %s\n", SSH_HOST_ED25519_KEY_FILE); |
| 550 |
|
} |
| 551 |
|
else |
| 552 |
|
{ |
| 553 |
|
ssh_key_valid = 1; |
| 554 |
|
} |
| 555 |
|
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_ECDSA_KEY_FILE) < 0) |
| 556 |
|
{ |
| 557 |
|
log_error("Error loading SSH ECDSA key: %s\n", SSH_HOST_ECDSA_KEY_FILE); |
| 558 |
|
} |
| 559 |
|
else |
| 560 |
|
{ |
| 561 |
|
ssh_key_valid = 1; |
| 562 |
|
} |
| 563 |
|
|
| 564 |
|
if (!ssh_key_valid) |
| 565 |
|
{ |
| 566 |
|
log_error("Error: no valid SSH host key\n"); |
| 567 |
|
ssh_bind_free(sshbind); |
| 568 |
|
return -1; |
| 569 |
|
} |
| 570 |
|
|
| 571 |
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 || |
if (ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDADDR, hostaddr) < 0 || |
| 572 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 || |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_BINDPORT, &port) < 0 || |
| 573 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY, SSH_HOST_KEYFILE) < 0 || |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY_ALGORITHMS, "+ssh-rsa") < 0 || |
|
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_HOSTKEY_ALGORITHMS, "ssh-rsa,rsa-sha2-512,rsa-sha2-256") < 0 || |
|
| 574 |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0) |
ssh_bind_options_set(sshbind, SSH_BIND_OPTIONS_LOG_VERBOSITY, &ssh_log_level) < 0) |
| 575 |
{ |
{ |
| 576 |
log_error("Error setting SSH bind options: %s\n", ssh_get_error(sshbind)); |
log_error("Error setting SSH bind options: %s\n", ssh_get_error(sshbind)); |
| 578 |
return -1; |
return -1; |
| 579 |
} |
} |
| 580 |
|
|
| 581 |
epollfd = epoll_create1(0); |
#ifdef HAVE_SYS_EPOLL_H |
| 582 |
if (epollfd < 0) |
epollfd_server = epoll_create1(0); |
| 583 |
|
if (epollfd_server == -1) |
| 584 |
{ |
{ |
| 585 |
log_error("epoll_create1() error (%d)\n", errno); |
log_error("epoll_create1() error (%d)\n", errno); |
| 586 |
return -1; |
return -1; |
| 587 |
} |
} |
| 588 |
|
#endif |
| 589 |
|
|
| 590 |
// Server socket |
// Server socket |
| 591 |
for (i = 0; i < 2; i++) |
for (i = 0; i < 2; i++) |
| 603 |
sin.sin_port = htons(port[i]); |
sin.sin_port = htons(port[i]); |
| 604 |
|
|
| 605 |
// Reuse address and port |
// Reuse address and port |
| 606 |
flags[i] = 1; |
flags_server[i] = 1; |
| 607 |
if (setsockopt(socket_server[i], SOL_SOCKET, SO_REUSEADDR, &flags[i], sizeof(flags[i])) < 0) |
if (setsockopt(socket_server[i], SOL_SOCKET, SO_REUSEADDR, &flags_server[i], sizeof(flags_server[i])) < 0) |
| 608 |
{ |
{ |
| 609 |
log_error("setsockopt SO_REUSEADDR error (%d)\n", errno); |
log_error("setsockopt SO_REUSEADDR error (%d)\n", errno); |
| 610 |
} |
} |
| 611 |
if (setsockopt(socket_server[i], SOL_SOCKET, SO_REUSEPORT, &flags[i], sizeof(flags[i])) < 0) |
#if defined(SO_REUSEPORT) |
| 612 |
|
if (setsockopt(socket_server[i], SOL_SOCKET, SO_REUSEPORT, &flags_server[i], sizeof(flags_server[i])) < 0) |
| 613 |
{ |
{ |
| 614 |
log_error("setsockopt SO_REUSEPORT error (%d)\n", errno); |
log_error("setsockopt SO_REUSEPORT error (%d)\n", errno); |
| 615 |
} |
} |
| 616 |
|
#endif |
| 617 |
|
|
| 618 |
if (bind(socket_server[i], (struct sockaddr *)&sin, sizeof(sin)) < 0) |
if (bind(socket_server[i], (struct sockaddr *)&sin, sizeof(sin)) < 0) |
| 619 |
{ |
{ |
| 630 |
|
|
| 631 |
log_common("Listening at %s:%u\n", inet_ntoa(sin.sin_addr), ntohs(sin.sin_port)); |
log_common("Listening at %s:%u\n", inet_ntoa(sin.sin_addr), ntohs(sin.sin_port)); |
| 632 |
|
|
| 633 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 634 |
ev.events = EPOLLIN; |
ev.events = EPOLLIN; |
| 635 |
ev.data.fd = socket_server[i]; |
ev.data.fd = socket_server[i]; |
| 636 |
if (epoll_ctl(epollfd, EPOLL_CTL_ADD, socket_server[i], &ev) == -1) |
if (epoll_ctl(epollfd_server, EPOLL_CTL_ADD, socket_server[i], &ev) == -1) |
| 637 |
{ |
{ |
| 638 |
log_error("epoll_ctl(socket_server[%d]) error (%d)\n", i, errno); |
log_error("epoll_ctl(socket_server[%d]) error (%d)\n", i, errno); |
| 639 |
if (close(epollfd) < 0) |
if (close(epollfd_server) < 0) |
| 640 |
{ |
{ |
| 641 |
log_error("close(epoll) error (%d)\n"); |
log_error("close(epoll) error (%d)\n"); |
| 642 |
} |
} |
| 643 |
return -1; |
return -1; |
| 644 |
} |
} |
| 645 |
|
#endif |
| 646 |
|
|
| 647 |
flags[i] = fcntl(socket_server[i], F_GETFL, 0); |
flags_server[i] = fcntl(socket_server[i], F_GETFL, 0); |
| 648 |
fcntl(socket_server[i], F_SETFL, flags[i] | O_NONBLOCK); |
fcntl(socket_server[i], F_SETFL, flags_server[i] | O_NONBLOCK); |
| 649 |
} |
} |
| 650 |
|
|
| 651 |
|
ssh_bind_set_blocking(sshbind, 0); |
| 652 |
|
|
| 653 |
hash_dict_pid_sockaddr = hash_dict_create(MAX_CLIENT_LIMIT); |
hash_dict_pid_sockaddr = hash_dict_create(MAX_CLIENT_LIMIT); |
| 654 |
if (hash_dict_pid_sockaddr == NULL) |
if (hash_dict_pid_sockaddr == NULL) |
| 655 |
{ |
{ |
| 685 |
{ |
{ |
| 686 |
SYS_child_exit = 0; |
SYS_child_exit = 0; |
| 687 |
|
|
| 688 |
siginfo.si_pid = 0; |
pid = waitpid(-1, &ret, WNOHANG); |
| 689 |
ret = waitid(P_ALL, 0, &siginfo, WEXITED | WNOHANG); |
if (pid > 0) |
|
if (ret == 0 && siginfo.si_pid > 0) |
|
| 690 |
{ |
{ |
| 691 |
SYS_child_exit = 1; // Retry waitid |
SYS_child_exit = 1; // Retry waitid |
|
|
|
| 692 |
SYS_child_process_count--; |
SYS_child_process_count--; |
|
log_common("Child process (%d) exited\n", siginfo.si_pid); |
|
| 693 |
|
|
| 694 |
if (siginfo.si_pid != section_list_loader_pid) |
if (WIFEXITED(ret)) |
| 695 |
|
{ |
| 696 |
|
log_common("Child process (%d) exited, status=%d\n", pid, WEXITSTATUS(ret)); |
| 697 |
|
} |
| 698 |
|
else if (WIFSIGNALED(ret)) |
| 699 |
|
{ |
| 700 |
|
log_common("Child process (%d) is killed, status=%d\n", pid, WTERMSIG(ret)); |
| 701 |
|
} |
| 702 |
|
else |
| 703 |
|
{ |
| 704 |
|
log_common("Child process (%d) exited abnormally, status=%d\n", pid, ret); |
| 705 |
|
} |
| 706 |
|
|
| 707 |
|
if (pid != section_list_loader_pid) |
| 708 |
{ |
{ |
| 709 |
j = 0; |
j = 0; |
| 710 |
ret = hash_dict_get(hash_dict_pid_sockaddr, (uint64_t)siginfo.si_pid, (int64_t *)&j); |
ret = hash_dict_get(hash_dict_pid_sockaddr, (uint64_t)pid, (int64_t *)&j); |
| 711 |
if (ret < 0) |
if (ret < 0) |
| 712 |
{ |
{ |
| 713 |
log_error("hash_dict_get(hash_dict_pid_sockaddr, %d) error\n", siginfo.si_pid); |
log_error("hash_dict_get(hash_dict_pid_sockaddr, %d) error\n", pid); |
| 714 |
} |
} |
| 715 |
else |
else |
| 716 |
{ |
{ |
| 720 |
log_error("hash_dict_inc(hash_dict_sockaddr_count, %d, -1) error\n", j); |
log_error("hash_dict_inc(hash_dict_sockaddr_count, %d, -1) error\n", j); |
| 721 |
} |
} |
| 722 |
|
|
| 723 |
ret = hash_dict_del(hash_dict_pid_sockaddr, (uint64_t)siginfo.si_pid); |
ret = hash_dict_del(hash_dict_pid_sockaddr, (uint64_t)pid); |
| 724 |
if (ret < 0) |
if (ret < 0) |
| 725 |
{ |
{ |
| 726 |
log_error("hash_dict_del(hash_dict_pid_sockaddr, %d) error\n", siginfo.si_pid); |
log_error("hash_dict_del(hash_dict_pid_sockaddr, %d) error\n", pid); |
| 727 |
} |
} |
| 728 |
} |
} |
| 729 |
} |
} |
| 730 |
} |
} |
| 731 |
else if (ret == 0) |
else if (pid == 0) |
| 732 |
{ |
{ |
| 733 |
break; |
break; |
| 734 |
} |
} |
| 735 |
else if (ret < 0) |
else if (pid < 0) |
| 736 |
{ |
{ |
| 737 |
log_error("Error in waitid: %d\n", errno); |
log_error("Error in waitpid(): %d\n", errno); |
| 738 |
break; |
break; |
| 739 |
} |
} |
| 740 |
} |
} |
| 748 |
log_common("Notify %d child process to exit\n", SYS_child_process_count); |
log_common("Notify %d child process to exit\n", SYS_child_process_count); |
| 749 |
#endif |
#endif |
| 750 |
|
|
| 751 |
if (kill(-getpid(), SIGTERM) < 0) |
if (kill(0, SIGTERM) < 0) |
| 752 |
{ |
{ |
| 753 |
log_error("Send SIGTERM signal failed (%d)\n", errno); |
log_error("Send SIGTERM signal failed (%d)\n", errno); |
| 754 |
} |
} |
| 762 |
sd_notifyf(0, "STATUS=Kill %d child process", SYS_child_process_count); |
sd_notifyf(0, "STATUS=Kill %d child process", SYS_child_process_count); |
| 763 |
#endif |
#endif |
| 764 |
|
|
| 765 |
if (kill(-getpid(), SIGKILL) < 0) |
if (kill(0, SIGKILL) < 0) |
| 766 |
{ |
{ |
| 767 |
log_error("Send SIGKILL signal failed (%d)\n", errno); |
log_error("Send SIGKILL signal failed (%d)\n", errno); |
| 768 |
} |
} |
| 803 |
log_error("Reload BWF conf failed\n"); |
log_error("Reload BWF conf failed\n"); |
| 804 |
} |
} |
| 805 |
|
|
| 806 |
if (load_menu(&bbs_menu_new, CONF_MENU) < 0) |
if (detach_menu_shm(&bbs_menu) < 0) |
| 807 |
{ |
{ |
| 808 |
unload_menu(&bbs_menu_new); |
log_error("detach_menu_shm(bbs_menu) error\n"); |
|
log_error("Reload bbs menu failed\n"); |
|
| 809 |
} |
} |
| 810 |
else |
if (load_menu(&bbs_menu, CONF_MENU) < 0) |
| 811 |
{ |
{ |
| 812 |
|
log_error("load_menu(bbs_menu) error\n"); |
| 813 |
unload_menu(&bbs_menu); |
unload_menu(&bbs_menu); |
|
memcpy(&bbs_menu, &bbs_menu_new, sizeof(bbs_menu_new)); |
|
|
log_common("Reload bbs menu successfully\n"); |
|
| 814 |
} |
} |
| 815 |
|
|
| 816 |
if (load_menu(&top10_menu_new, CONF_TOP10_MENU) < 0) |
if (detach_menu_shm(&top10_menu) < 0) |
| 817 |
{ |
{ |
| 818 |
unload_menu(&top10_menu_new); |
log_error("detach_menu_shm(top10_menu) error\n"); |
|
log_error("Reload top10 menu failed\n"); |
|
| 819 |
} |
} |
| 820 |
else |
if (load_menu(&top10_menu, CONF_TOP10_MENU) < 0) |
| 821 |
{ |
{ |
| 822 |
|
log_error("load_menu(top10_menu) error\n"); |
| 823 |
unload_menu(&top10_menu); |
unload_menu(&top10_menu); |
|
top10_menu_new.allow_exit = 1; |
|
|
memcpy(&top10_menu, &top10_menu_new, sizeof(top10_menu_new)); |
|
|
log_common("Reload top10 menu successfully\n"); |
|
| 824 |
} |
} |
| 825 |
|
|
| 826 |
for (int i = 0; i < data_files_load_startup_count; i++) |
for (int i = 0; i < data_files_load_startup_count; i++) |
| 830 |
log_error("load_file(%s) error\n", data_files_load_startup[i]); |
log_error("load_file(%s) error\n", data_files_load_startup[i]); |
| 831 |
} |
} |
| 832 |
} |
} |
|
log_common("Reload data files successfully\n"); |
|
| 833 |
|
|
| 834 |
// Load section config and gen_ex |
// Load section config and gen_ex |
| 835 |
if (load_section_config_from_db(1) < 0) |
if (load_section_config_from_db(1) < 0) |
| 836 |
{ |
{ |
| 837 |
log_error("load_section_config_from_db(1) error\n"); |
log_error("load_section_config_from_db(1) error\n"); |
| 838 |
} |
} |
|
else |
|
|
{ |
|
|
log_common("Reload section config and gen_ex successfully\n"); |
|
|
} |
|
| 839 |
|
|
| 840 |
// Notify child processes to reload configuration |
// Notify child processes to reload configuration |
| 841 |
if (kill(-getpid(), SIGUSR1) < 0) |
if (kill(0, SIGUSR1) < 0) |
| 842 |
{ |
{ |
| 843 |
log_error("Send SIGUSR1 signal failed (%d)\n", errno); |
log_error("Send SIGUSR1 signal failed (%d)\n", errno); |
| 844 |
} |
} |
| 848 |
#endif |
#endif |
| 849 |
} |
} |
| 850 |
|
|
| 851 |
nfds = epoll_wait(epollfd, events, MAX_EVENTS, 100); // 0.1 second |
#ifdef HAVE_SYS_EPOLL_H |
| 852 |
|
nfds = epoll_wait(epollfd_server, events, MAX_EVENTS, 100); // 0.1 second |
| 853 |
if (nfds < 0) |
ret = nfds; |
| 854 |
|
#else |
| 855 |
|
pfds[0].fd = socket_server[0]; |
| 856 |
|
pfds[0].events = POLLIN; |
| 857 |
|
pfds[1].fd = socket_server[1]; |
| 858 |
|
pfds[1].events = POLLIN; |
| 859 |
|
nfds = 2; |
| 860 |
|
ret = poll(pfds, (nfds_t)nfds, 100); // 0.1 second |
| 861 |
|
#endif |
| 862 |
|
if (ret < 0) |
| 863 |
{ |
{ |
| 864 |
if (errno != EINTR) |
if (errno != EINTR) |
| 865 |
{ |
{ |
| 866 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 867 |
log_error("epoll_wait() error (%d)\n", errno); |
log_error("epoll_wait() error (%d)\n", errno); |
| 868 |
|
#else |
| 869 |
|
log_error("poll() error (%d)\n", errno); |
| 870 |
|
#endif |
| 871 |
break; |
break; |
| 872 |
} |
} |
| 873 |
continue; |
continue; |
| 881 |
|
|
| 882 |
for (int i = 0; i < nfds; i++) |
for (int i = 0; i < nfds; i++) |
| 883 |
{ |
{ |
| 884 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 885 |
if (events[i].data.fd == socket_server[0] || events[i].data.fd == socket_server[1]) |
if (events[i].data.fd == socket_server[0] || events[i].data.fd == socket_server[1]) |
| 886 |
|
#else |
| 887 |
|
if ((pfds[i].fd == socket_server[0] || pfds[i].fd == socket_server[1]) && (pfds[i].revents & POLLIN)) |
| 888 |
|
#endif |
| 889 |
{ |
{ |
| 890 |
|
#ifdef HAVE_SYS_EPOLL_H |
| 891 |
SSH_v2 = (events[i].data.fd == socket_server[1] ? 1 : 0); |
SSH_v2 = (events[i].data.fd == socket_server[1] ? 1 : 0); |
| 892 |
|
#else |
| 893 |
|
SSH_v2 = (pfds[i].fd == socket_server[1] ? 1 : 0); |
| 894 |
|
#endif |
| 895 |
|
|
| 896 |
while (!SYS_server_exit) // Accept all incoming connections until error |
while (!SYS_server_exit) // Accept all incoming connections until error |
| 897 |
{ |
{ |
| 898 |
addrlen = sizeof(sin); |
addrlen = sizeof(sin); |
| 899 |
socket_client = accept(socket_server[SSH_v2], (struct sockaddr *)&sin, &addrlen); |
socket_client = accept(socket_server[SSH_v2], (struct sockaddr *)&sin, (socklen_t *)&addrlen); |
| 900 |
if (socket_client < 0) |
if (socket_client < 0) |
| 901 |
{ |
{ |
| 902 |
if (errno == EAGAIN || errno == EWOULDBLOCK) |
if (errno == EAGAIN || errno == EWOULDBLOCK) |
| 970 |
} |
} |
| 971 |
} |
} |
| 972 |
|
|
| 973 |
if (close(epollfd) < 0) |
#ifdef HAVE_SYS_EPOLL_H |
| 974 |
|
if (close(epollfd_server) < 0) |
| 975 |
{ |
{ |
| 976 |
log_error("close(epoll) error (%d)\n"); |
log_error("close(epollfd_server) error (%d)\n"); |
| 977 |
} |
} |
| 978 |
|
#endif |
| 979 |
|
|
| 980 |
for (i = 0; i < 2; i++) |
for (i = 0; i < 2; i++) |
| 981 |
{ |
{ |
|
fcntl(socket_server[i], F_SETFL, flags[i]); |
|
|
|
|
| 982 |
if (close(socket_server[i]) == -1) |
if (close(socket_server[i]) == -1) |
| 983 |
{ |
{ |
| 984 |
log_error("Close server socket failed\n"); |
log_error("Close server socket failed\n"); |