From a79f029a4a7b70941a5abf31e8d4023f2be2c8bb Mon Sep 17 00:00:00 2001 From: ramonn Date: Tue, 27 May 2014 12:32:38 +0000 Subject: [PATCH] 2014-05-27 Ramon Novoa * bin/tentacle_server: Updated to the latest version. git-svn-id: https://svn.code.sf.net/p/pandora/code/trunk@10017 c3f86ba8-e40f-0410-aaad-9ba5e7f4b01f --- pandora_server/ChangeLog | 4 + pandora_server/bin/tentacle_server | 378 +++++++++++++++-------------- 2 files changed, 205 insertions(+), 177 deletions(-) diff --git a/pandora_server/ChangeLog b/pandora_server/ChangeLog index 2e73ceea85..74ceafeeac 100644 --- a/pandora_server/ChangeLog +++ b/pandora_server/ChangeLog @@ -1,3 +1,7 @@ +2014-05-27 Ramon Novoa + + * bin/tentacle_server: Updated to the latest version. + 2014-05-25 Junichi Satoh * lib/PandoraFMS/Tools.pm: Fixed STD* file open error. diff --git a/pandora_server/bin/tentacle_server b/pandora_server/bin/tentacle_server index a24c63309a..a8146c563e 100755 --- a/pandora_server/bin/tentacle_server +++ b/pandora_server/bin/tentacle_server @@ -60,8 +60,14 @@ use strict; use warnings; use Getopt::Std; use IO::Select; +use threads; use Thread::Semaphore; -use POSIX qw(:sys_wait_h strftime); +use POSIX ":sys_wait_h"; +use Time::HiRes qw(usleep); + +# Constants for Win32 services. +use constant WIN32_SERVICE_STOPPED => 0x01; +use constant WIN32_SERVICE_RUNNING => 0x04; my $t_libwrap_installed = eval { require Authen::Libwrap } ? 1 : 0; @@ -77,6 +83,9 @@ my $SOCKET_MODULE = : eval { require IO::Socket::INET } ? 'IO::Socket::INET' : die $@; +# Service name for Win32. +my $SERVICE_NAME="Tentacle Server"; + # Program version our $VERSION = '0.4.0'; @@ -126,7 +135,7 @@ my $t_retries = 3; my $t_select; # Semaphore -my $t_sem; +my $t_sem :shared; # Server socket my @t_server_sockets; @@ -169,10 +178,6 @@ my $t_use_libwrap = 0; my $t_program_name = $0; $t_program_name =~ s/.*\///g; -# logfile name and it's file handle -my $t_logfile; -my $t_logfile_handle; - ################################################################################ ## SUB print_help ## Print help screen. @@ -192,12 +197,12 @@ sub print_help { print ("\t-h\t\tShow help.\n"); print ("\t-i\t\tFilters.\n"); print ("\t-k key\t\tOpenSSL private key file.\n"); - print ("\t-l file\t\tPath to the log file. Only used in daemon mode.\n"); print ("\t-m size\t\tMaximum file size in bytes (default ${t_max_size}b).\n"); print ("\t-o\t\tEnable file overwrite.\n"); print ("\t-p port\t\tPort to listen on (default $t_port).\n"); print ("\t-q\t\tQuiet. Do now print error messages.\n"); print ("\t-r number\tNumber of retries for network opertions (default $t_retries).\n"); + print ("\t-S (install|uninstall|run) Manage the win32 service.\n"); print ("\t-t time\t\tTime-out for network operations in seconds (default ${t_timeout}s).\n"); print ("\t-v\t\tBe verbose.\n"); print ("\t-w\t\tPrompt for OpenSSL private key password.\n"); @@ -222,13 +227,9 @@ sub daemonize { open (STDIN, '/dev/null') || error ("Cannot read /dev/null: $!."); - if ($t_logfile) { - open_logfile(); - } else { - # Do not be verbose when running as a daemon - open (STDOUT, '>/dev/null') || error ("Cannot write to /dev/null: $!."); - open (STDERR, '>/dev/null') || error ("Cannot write to /dev/null: $!."); - } + # Do not be verbose when running as a daemon + open (STDOUT, '>/dev/null') || error ("Cannot write to /dev/null: $!."); + open (STDERR, '>/dev/null') || error ("Cannot write to /dev/null: $!."); # Fork $pid = fork (); @@ -240,40 +241,11 @@ sub daemonize { if ($pid != 0) { exit; } - $SIG{TERM} = \&stop_server; - $SIG{HUP} = sub { - print_log ("SIGHUP received."); - if ($t_logfile_handle) { - close($t_logfile_handle); - open_logfile(); - } - }; # Child POSIX::setsid () || error ("Cannot start a new session: $!."); } -################################################################################ -## SUB start_win_service -## Turn the current process into a Windows service. -################################################################################ -#sub start_win_service { -# require Win32::Daemon; -# -# # Tell the OS to start the service -# Win32::Daemon::StartService (); -# -# # Wait until the service manager is ready -# while (SERVICE_START_PENDING != Win32::Daemon::State()) { -# sleep (1); -# } -# -# # Tell the service manager we are running -# Win32::Daemon::State (SERVICE_RUNNING); -# -# # Call Win32::Daemon::StopService() when done -#} - ################################################################################ ## SUB parse_options ## Parse command line options and initialize global variables. @@ -284,7 +256,7 @@ sub parse_options { my @t_addresses_tmp; # Get options - if (getopts ('a:c:de:f:hi:k:l:m:op:qr:s:t:vwx:b:g:T', \%opts) == 0 || defined ($opts{'h'})) { + if (getopts ('a:b:c:de:f:g:hi:k:m:op:qr:s:S:t:Tvwx:', \%opts) == 0 || defined ($opts{'h'})) { print_help (); exit 1; } @@ -481,24 +453,34 @@ sub parse_options { } } - if (defined ($opts{'l'})) { - $t_logfile = $opts{'l'}; + # Win32 service management + if (defined ($opts{'S'})) { + my $service_action = $opts{'S'}; + if ($^O ne 'MSWin32') { + error ("Windows services are only available on Win32."); + } else { + eval "use Win32::Daemon"; + die($@) if ($@); + + if ($service_action eq 'install') { + install_service(); + } elsif ($service_action eq 'uninstall') { + uninstall_service(); + } elsif ($service_action eq 'run') { + Win32::Daemon::RegisterCallbacks({ + start => \&callback_start, + running => \&callback_running, + stop => \&callback_stop, + }); + Win32::Daemon::StartService(); + exit 0; + } else { + error("Unknown action: $service_action"); + } + } } } -################################################################################ -## SUB sigchld_handler -## Handle child process termination. -################################################################################ -sub sigchld_handler { - - while (waitpid(-1, &WNOHANG) > 0) { - $t_sem->up (); - } - - $SIG{CHLD} = \&sigchld_handler; -} - ################################################################################ ## SUB start_proxy ## Open the server socket. @@ -527,11 +509,6 @@ sub open_proxy { ################################################################################ sub start_server { - # Show IPv6 status - if ($SOCKET_MODULE eq 'IO::Socket::INET') { - print_log ("IO::Socket::INET6 is not found. IPv6 is disabled."); - } - my $t_server_socket; foreach my $t_address (@t_addresses) { @@ -548,9 +525,8 @@ sub start_server { print_log ("Cannot open socket for address $t_address on port $t_port: $!."); next; } - my $msg = "Server listening on $t_address port $t_port"; - $msg .= " (press to stop)" unless $t_daemon; - print_log ($msg); + + print_log ("Server listening on $t_address port $t_port (press to stop)"); # Say message if tentacle proxy is enable if (defined ($t_proxy_ip)) { @@ -629,18 +605,6 @@ sub close_proxy { } -################################################################################ -## SUB open_logfile -## open logfile and assign log file handle to STDOUT and STDERR -################################################################################ -sub open_logfile { - if ($t_logfile) { - open ($t_logfile_handle, '>>', $t_logfile) || error ("Cannot open log file: $t_logfile: $!."); - open (STDOUT, '>&', $t_logfile_handle) || error ("Cannot dup log file handle: $!."); - open (STDERR, '>&', $t_logfile_handle) || error ("Cannot dup log file handle: $!."); - } -} - ################################################################################ ## SUB stop_server ## Close the server socket. @@ -651,9 +615,6 @@ sub stop_server { $t_server_socket->close (); } print_log ("Server going down"); - if ($t_logfile_handle) { - close($t_logfile_handle); - } exit 0; } @@ -698,87 +659,86 @@ sub start_ssl { } ################################################################################ -## SUB accept_connection -## Accept an incoming connection and fork. +## SUB accept_connections +## Manage incoming connections. ################################################################################ -sub accept_connection { +sub accept_connections { my $pid; my $t_server_socket; - my @ready = $select->can_read; + # Start server + start_server (); - foreach $t_server_socket (@ready) { + # Initialize semaphore + $t_sem = Thread::Semaphore->new ($t_max_conn); - # Accept connection - $t_client_socket = $t_server_socket->accept (); + while (1) { + my @ready = $select->can_read; + foreach $t_server_socket (@ready) { - if (! defined ($t_client_socket)) { + # Accept connection + $t_client_socket = $t_server_socket->accept (); - # EINTR - if ($! ne '') { - next; + if (! defined ($t_client_socket)) { + next if ($! ne ''); # EINTR + error ("accept: $!."); } - error ("accept: $!."); + print_log ("Client connected from " . $t_client_socket->sockhost ()); + + # Create a new thread and serve the client + $t_sem->down(); + my $thr = threads->create(\&serve_client); + if (! defined ($thr)) { + error ("Error creating thread: $!."); + } + $thr->detach(); } - print_log ("Client connected from " . $t_client_socket->sockhost ()); + usleep (1000); + } +} - # Fork and serve the client - $pid = fork (); - if (! defined ($pid)) { - error ("Cannot fork: $!."); +################################################################################ +## SUB serve_client +## Serve a connected client. +################################################################################ +sub serve_client() { + + if ($t_use_libwrap) { + if (! hosts_ctl($t_program_name, $t_client_socket)) { + print_log ("Connection from " . $t_client_socket->sockhost() . " is closed by tcpwrappers."); + $t_client_socket->close (); + $t_sem->up(); + return; + } + } + + eval { + # Add client socket to select queue + $t_select = IO::Select->new (); + $t_select->add ($t_client_socket); + + # Start SSL + if ($t_ssl == 1) { + start_ssl (); } - # Child - if ($pid == 0) { - - # We do not need the server socket - $t_server_socket->close (); - - if ($t_use_libwrap) { - if (! hosts_ctl($t_program_name, $t_client_socket)) { - print_log ("Connection from " . $t_client_socket->sockhost() . " is closed by tcpwrappers."); - $t_client_socket->close (); - - exit; - } - } - - # Add client socket to select queue - $t_select = IO::Select->new (); - $t_select->add ($t_client_socket); - - # Start SSL - if ($t_ssl == 1) { - start_ssl (); - } - - # Authenticate client - if ($t_pwd ne '') { - auth_pwd (); - } - - # Check if proxy mode is enable - if (defined ($t_proxy_ip)) { - - serve_proxy_connection (); - - } else { - - serve_connection (); - } - - $t_client_socket->close (); - - # Must exit now - exit; + # Authenticate client + if ($t_pwd ne '') { + auth_pwd (); } + + # Check if proxy mode is enable + if (defined ($t_proxy_ip)) { + serve_proxy_connection (); + } else { + serve_connection (); + } + }; - # Parent - $t_client_socket->close (); - - } + $t_client_socket->close (); + $t_sem->up(); } ################################################################################ @@ -1065,15 +1025,6 @@ sub send_file { # Common functions ################################################################################ -################################################################################ -## SUB format_log -## format log message -################################################################################ -sub format_log { - return strftime("%Y-%m-%d %H:%M:%S ", localtime()) - . $t_program_name. ' [' . $$ . ']: ' . $_[0]; -} - ################################################################################ ## SUB print_log ## Print log messages. @@ -1081,9 +1032,7 @@ sub format_log { sub print_log { if ($t_log == 1) { - my $msg = "[log] $_[0]\n"; - $msg = format_log($msg) if ($t_logfile_handle); - print (STDOUT $msg); + print (STDOUT "[log] $_[0]\n"); } } @@ -1094,12 +1043,8 @@ sub print_log { sub error { if ($t_quiet == 0) { - my $msg = "[err] $_[0]\n"; - $msg = format_log($msg) if ($t_logfile_handle); - print (STDERR $msg); + die("[err] $_[0]\n\n"); } - - exit 1; } ################################################################################ @@ -1464,6 +1409,91 @@ sub apply_filters ($) { return ''; } +################################################################################ +## SUB install_service +## Install the Windows service. +################################################################################ +sub install_service() { + + my $service_path = $0; + my $service_params = "-s \"$t_directory\" -S run"; + + my %service_hash = ( + machine => '', + name => 'TENTACLESRV', + display => $SERVICE_NAME, + path => $service_path, + user => '', + pwd => '', + description => 'Tentacle Server http://sourceforge.net/projects/tentacled/', + parameters => $service_params + ); + + if (Win32::Daemon::CreateService(\%service_hash)) { + print "Successfully added.\n"; + exit 0; + } else { + print "Failed to add service: " . Win32::FormatMessage(Win32::Daemon::GetLastError()) . "\n"; + exit 1; + } +} + +################################################################################ +## SUB uninstall_service +## Install the Windows service. +################################################################################ +sub uninstall_service() { + if (Win32::Daemon::DeleteService('', 'TENTACLESRV')) { + print "Successfully deleted.\n"; + exit 0; + } else { + print "Failed to delete service: " . Win32::FormatMessage(Win32::Daemon::GetLastError()) . "\n"; + exit 1; + } +} + +################################################################################ +## SUB callback_running +## Windows service callback function for the running event. +################################################################################ +sub callback_running { + + if (Win32::Daemon::State() == WIN32_SERVICE_RUNNING) { + } +} + +################################################################################ +## SUB callback_start +## Windows service callback function for the start event. +################################################################################ +sub callback_start { + + # Accept_connections (); + my $thr = threads->create(\&accept_connections); + if (!defined($thr)) { + Win32::Daemon::State(WIN32_SERVICE_STOPPED); + Win32::Daemon::StopService(); + return; + } + $thr->detach(); + + Win32::Daemon::State(WIN32_SERVICE_RUNNING); +} + +################################################################################ +## SUB callback_stop +## Windows service callback function for the stop event. +################################################################################ +sub callback_stop { + + foreach my $t_server_socket (@t_server_sockets) { + $t_server_socket->close (); + } + + Win32::Daemon::State(WIN32_SERVICE_STOPPED); + Win32::Daemon::StopService(); +} + ################################################################################ # Main ################################################################################ @@ -1483,6 +1513,11 @@ if ($#ARGV != -1) { exit 1; } +# Show IPv6 status +if ($SOCKET_MODULE eq 'IO::Socket::INET') { + print_log ("IO::Socket::INET6 is not found. IPv6 is disabled."); +} + # Run as daemon? if ($t_daemon == 1 && $^O ne 'MSWin32') { daemonize (); @@ -1490,25 +1525,16 @@ if ($t_daemon == 1 && $^O ne 'MSWin32') { # Handle ctr-c if ($^O eq 'MSWin32') { + no warnings; $SIG{INT2} = \&stop_server; + use warnings; } else { $SIG{INT} = \&stop_server; } -# Handle SIGCHLD -$SIG{CHLD} = \&sigchld_handler; - -start_server (); - -# Initialize semaphore -$t_sem = Thread::Semaphore->new ($t_max_conn); - # Accept connections -while (1) { - $t_sem->down (); - accept_connection (); -} +accept_connections(); __END__ @@ -1540,8 +1566,6 @@ __END__ =item I<-k key> B file. -=item I<-l file> Path to the B. Only used in daemon mode. - =item I<-m size> B in bytes (default I<2000000b>). =item I<-o> Enable file B.