429 lines
13 KiB
Perl
429 lines
13 KiB
Perl
#!/usr/bin/perl -wT
|
|
#
|
|
# ==========================================================================
|
|
#
|
|
# ZoneMinder Package Control Script, $Date$, $Revision$
|
|
# Copyright (C) 2001-2008 Philip Coombes
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
# ==========================================================================
|
|
|
|
=head1 NAME
|
|
|
|
zmpkg.pl - ZoneMinder Package Control Script
|
|
|
|
=head1 SYNOPSIS
|
|
|
|
zmpkg.pl {start|stop|restart|status|logrot|'state'|version}
|
|
|
|
=head1 DESCRIPTION
|
|
|
|
This script is used to start and stop the ZoneMinder package primarily to
|
|
allow command line control for automatic restart on reboot (see zm script)
|
|
|
|
=cut
|
|
use strict;
|
|
use bytes;
|
|
|
|
# ==========================================================================
|
|
#
|
|
# Don't change anything below here
|
|
#
|
|
# ==========================================================================
|
|
|
|
@EXTRA_PERL_LIB@
|
|
use ZoneMinder;
|
|
use DBI;
|
|
use POSIX;
|
|
use Time::HiRes qw/gettimeofday/;
|
|
use autouse 'Pod::Usage'=>qw(pod2usage);
|
|
|
|
# Detaint our environment
|
|
$ENV{PATH} = '/bin:/usr/bin';
|
|
$ENV{SHELL} = '/bin/sh' if exists $ENV{SHELL};
|
|
delete @ENV{qw(IFS CDPATH ENV BASH_ENV)};
|
|
my $store_state=""; # PP - will remember state name passed
|
|
|
|
logInit();
|
|
|
|
my $command = $ARGV[0]||'';
|
|
if ( $command eq 'version' ) {
|
|
print ZoneMinder::Base::ZM_VERSION . "\n";
|
|
exit(0);
|
|
}
|
|
|
|
my $state;
|
|
|
|
my $dbh;
|
|
|
|
if ( !$command || $command !~ /^(?:start|stop|restart|status|logrot|version)$/ )
|
|
{
|
|
if ( $command )
|
|
{
|
|
$dbh = zmDbConnect();
|
|
# Check to see if it's a valid run state
|
|
my $sql = 'select * from States where Name = ?';
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute( $command )
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
if ( $state = $sth->fetchrow_hashref() )
|
|
{
|
|
$state->{Name} = $command;
|
|
$state->{Definitions} = [];
|
|
foreach( split( /,/, $state->{Definition} ) )
|
|
{
|
|
my ( $id, $function, $enabled ) = split( /:/, $_ );
|
|
push( @{$state->{Definitions}},
|
|
{ Id=>$id, Function=>$function, Enabled=>$enabled }
|
|
);
|
|
}
|
|
$store_state=$command; # PP - Remember the name that was passed to search in DB
|
|
$command = 'state';
|
|
}
|
|
else
|
|
{
|
|
$command = undef;
|
|
}
|
|
}
|
|
if ( !$command )
|
|
{
|
|
pod2usage(-exitstatus => -1);
|
|
}
|
|
}
|
|
$dbh = zmDbConnect() if ! $dbh;
|
|
# PP - Sane state check
|
|
isActiveSanityCheck();
|
|
|
|
# Move to the right place
|
|
chdir( $Config{ZM_PATH_WEB} )
|
|
or Fatal( "Can't chdir to '".$Config{ZM_PATH_WEB}."': $!" );
|
|
|
|
my $dbg_id = "";
|
|
|
|
Info( "Command: $command\n" );
|
|
|
|
my $retval = 0;
|
|
|
|
if ( $command eq "state" )
|
|
{
|
|
Info( "Updating DB: $state->{Name}\n" );
|
|
my $sql = $Config{ZM_SERVER_ID} ? 'SELECT * FROM Monitors WHERE ServerId=? ORDER BY Id ASC' : 'SELECT * FROM Monitors ORDER BY Id ASC';
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute( $Config{ZM_SERVER_ID} ? $Config{ZM_SERVER_ID}: () )
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
while( my $monitor = $sth->fetchrow_hashref() )
|
|
{
|
|
foreach my $definition ( @{$state->{Definitions}} )
|
|
{
|
|
if ( $monitor->{Id} =~ /^$definition->{Id}$/ )
|
|
{
|
|
$monitor->{NewFunction} = $definition->{Function};
|
|
$monitor->{NewEnabled} = $definition->{Enabled};
|
|
}
|
|
}
|
|
#next if ( !$monitor->{NewFunction} );
|
|
$monitor->{NewFunction} = 'None'
|
|
if ( !$monitor->{NewFunction} );
|
|
$monitor->{NewEnabled} = 0
|
|
if ( !$monitor->{NewEnabled} );
|
|
if ( $monitor->{Function} ne $monitor->{NewFunction}
|
|
|| $monitor->{Enabled} ne $monitor->{NewEnabled}
|
|
)
|
|
{
|
|
my $sql = "update Monitors set Function = ?, Enabled = ? where Id = ?";
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute( $monitor->{NewFunction}, $monitor->{NewEnabled}, $monitor->{Id} )
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
}
|
|
}
|
|
$sth->finish();
|
|
|
|
# PP - Now mark a specific state as active
|
|
resetStates();
|
|
Info ("Marking $store_state as Enabled");
|
|
$sql = "update States set IsActive = '1' where Name = ?";
|
|
$sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
$res = $sth->execute( $store_state )
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
|
|
# PP - zero out other states isActive
|
|
$command = "restart";
|
|
}
|
|
|
|
# Check if we are running systemd and if we have been called by the system
|
|
if ( $command =~ /^(start|stop|restart)$/ )
|
|
{
|
|
# We have to detaint to keep perl from complaining
|
|
$command = $1;
|
|
|
|
|
|
if ( systemdRunning() && !calledBysystem() ) {
|
|
qx(@BINDIR@/zmsystemctl.pl $command);
|
|
$command = "";
|
|
}
|
|
}
|
|
|
|
if ( $command =~ /^(?:stop|restart)$/ )
|
|
{
|
|
my $status = runCommand( "zmdc.pl check" );
|
|
|
|
if ( $status eq "running" )
|
|
{
|
|
runCommand( "zmdc.pl shutdown" );
|
|
zmMemTidy();
|
|
}
|
|
else
|
|
{
|
|
$retval = 1;
|
|
}
|
|
}
|
|
|
|
#runCommand( "zmupdate.pl -f" );
|
|
|
|
if ( $command =~ /^(?:start|restart)$/ )
|
|
{
|
|
my $status = runCommand( "zmdc.pl check" );
|
|
|
|
if ( $status eq "stopped" )
|
|
{
|
|
if ( $Config{ZM_DYN_DB_VERSION}
|
|
and ( $Config{ZM_DYN_DB_VERSION} ne ZM_VERSION )
|
|
)
|
|
{
|
|
Fatal( "Version mismatch, system is version ".ZM_VERSION
|
|
.", database is ".$Config{ZM_DYN_DB_VERSION}
|
|
.", please run zmupdate.pl to update."
|
|
);
|
|
exit( -1 );
|
|
}
|
|
|
|
# Recreate the temporary directory if it's been wiped
|
|
verifyFolder("@ZM_TMPDIR@");
|
|
|
|
# Recreate the run directory if it's been wiped
|
|
verifyFolder("@ZM_RUNDIR@");
|
|
|
|
# Recreate the sock directory if it's been wiped
|
|
verifyFolder("@ZM_SOCKDIR@");
|
|
|
|
zmMemTidy();
|
|
runCommand( "zmdc.pl startup" );
|
|
|
|
Info( "Starting up services" . ( $Config{ZM_SERVER_ID} ? " for server $Config{ZM_SERVER_ID}\n" : "\n" ) );
|
|
my $sql = $Config{ZM_SERVER_ID} ? 'SELECT * FROM Monitors WHERE ServerId=?' : 'SELECT * FROM Monitors';
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute( $Config{ZM_SERVER_ID} ? $Config{ZM_SERVER_ID} : () )
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
while( my $monitor = $sth->fetchrow_hashref() )
|
|
{
|
|
if ( $monitor->{Function} ne 'None' )
|
|
{
|
|
if ( $monitor->{Type} eq 'Local' )
|
|
{
|
|
runCommand( "zmdc.pl start zmc -d $monitor->{Device}" );
|
|
}
|
|
else
|
|
{
|
|
runCommand( "zmdc.pl start zmc -m $monitor->{Id}" );
|
|
}
|
|
if ( $monitor->{Function} ne 'Monitor' )
|
|
{
|
|
if ( $Config{ZM_OPT_FRAME_SERVER} )
|
|
{
|
|
runCommand( "zmdc.pl start zmf -m $monitor->{Id}" );
|
|
}
|
|
runCommand( "zmdc.pl start zma -m $monitor->{Id}" );
|
|
}
|
|
if ( $Config{ZM_OPT_CONTROL} )
|
|
{
|
|
if ( $monitor->{Function} eq 'Modect' || $monitor->{Function} eq 'Mocord' )
|
|
{
|
|
if ( $monitor->{Controllable} && $monitor->{TrackMotion} )
|
|
{
|
|
runCommand( "zmdc.pl start zmtrack.pl -m $monitor->{Id}" );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
$sth->finish();
|
|
|
|
# This is now started unconditionally
|
|
runCommand( "zmdc.pl start zmfilter.pl" );
|
|
if ( $Config{ZM_RUN_AUDIT} )
|
|
{
|
|
runCommand( "zmdc.pl start zmaudit.pl -c" );
|
|
}
|
|
if ( $Config{ZM_OPT_TRIGGERS} )
|
|
{
|
|
runCommand( "zmdc.pl start zmtrigger.pl" );
|
|
}
|
|
if ( $Config{ZM_OPT_X10} )
|
|
{
|
|
runCommand( "zmdc.pl start zmx10.pl -c start" );
|
|
}
|
|
runCommand( "zmdc.pl start zmwatch.pl" );
|
|
if ( $Config{ZM_CHECK_FOR_UPDATES} )
|
|
{
|
|
runCommand( "zmdc.pl start zmupdate.pl -c" );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
$retval = 1;
|
|
}
|
|
}
|
|
|
|
if ( $command eq "status" )
|
|
{
|
|
my $status = runCommand( "zmdc.pl check" );
|
|
|
|
print( STDOUT $status."\n" );
|
|
}
|
|
|
|
if ( $command eq "logrot" )
|
|
{
|
|
runCommand( "zmdc.pl logrot" );
|
|
}
|
|
|
|
exit( $retval );
|
|
|
|
# PP - Make sure isActive is on and only one
|
|
sub isActiveSanityCheck
|
|
{
|
|
|
|
Info ("Sanity checking States table...");
|
|
$dbh = zmDbConnect() if ! $dbh;
|
|
|
|
# PP - First, make sure default exists and there is only one
|
|
my $sql = "select Name from States where Name = 'default'";
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
|
|
if ($sth->rows != 1) # PP - no row, or too many rows. Either case is an error
|
|
{
|
|
Info( "Fixing States table - either no default state or duplicate default states" );
|
|
$sql = "delete from States where Name = 'default'";
|
|
$sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
$res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
$sql = "insert into States (Name,Definition,IsActive) VALUES ('default','','1');";
|
|
$sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
$res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
}
|
|
|
|
|
|
# PP - Now make sure no two states have IsActive=1
|
|
$sql = "select Name from States where IsActive = '1'";
|
|
$sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
$res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
|
|
if ( $sth->rows != 1 )
|
|
{
|
|
Info( "Fixing States table so only one run state is active" );
|
|
resetStates();
|
|
$sql = "update States set IsActive='1' WHERE Name='default'";
|
|
$sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
$res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
# PP - zeroes out isActive for all states
|
|
sub resetStates
|
|
{
|
|
$dbh = zmDbConnect() if ! $dbh;
|
|
my $sql = "update States set IsActive = '0'";
|
|
my $sth = $dbh->prepare_cached( $sql )
|
|
or Fatal( "Can't prepare '$sql': ".$dbh->errstr() );
|
|
my $res = $sth->execute()
|
|
or Fatal( "Can't execute: ".$sth->errstr() );
|
|
|
|
}
|
|
|
|
sub systemdRunning
|
|
{
|
|
my $result = 0;
|
|
|
|
my $output = qx(ps -o comm="" -p 1);
|
|
chomp( $output );
|
|
|
|
if ($output =~ /systemd/) {
|
|
$result = 1;
|
|
}
|
|
|
|
return $result;
|
|
}
|
|
|
|
sub calledBysystem
|
|
{
|
|
my $result = 0;
|
|
my $ppid = getppid();
|
|
|
|
my $output = qx(ps -o comm="" -p $ppid);
|
|
chomp( $output );
|
|
|
|
if ($output =~ /^(?:systemd|init)$/) {
|
|
$result = 1;
|
|
}
|
|
|
|
return $result;
|
|
}
|
|
|
|
sub verifyFolder
|
|
{
|
|
my $folder = shift;
|
|
|
|
# Recreate the temporary directory if it's been wiped
|
|
if ( !-e $folder )
|
|
{
|
|
Debug( "Recreating directory '$folder'" );
|
|
mkdir( "$folder", 0774 )
|
|
or Fatal( "Can't create missing temporary directory '$folder': $!" );
|
|
my ( $runName ) = getpwuid( $> );
|
|
if ( $runName ne $Config{ZM_WEB_USER} )
|
|
{
|
|
# Not running as web user, so should be root in which case
|
|
# chown the directory
|
|
my ( $webName, $webPass, $webUid, $webGid ) = getpwnam( $Config{ZM_WEB_USER} )
|
|
or Fatal( "Can't get user details for web user '"
|
|
.$Config{ZM_WEB_USER}."': $!"
|
|
);
|
|
chown( $webUid, $webGid, "$folder" )
|
|
or Fatal( "Can't change ownership of directory '$folder' to '"
|
|
.$Config{ZM_WEB_USER}.":".$Config{ZM_WEB_GROUP}."': $!"
|
|
);
|
|
}
|
|
}
|
|
}
|
|
__END__
|