X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=src%2Fcore.pl;h=802e4c20a72da8c33dd51a4a3440f5ff4d42499d;hb=f7cae48a17d6decd0a9bd997188271daa0a885b1;hp=383210c1b606a0a063e19bac93265b45cd9ad20b;hpb=d4e35c061b1750a518dd92b58438192da3c56047;p=infobot.git diff --git a/src/core.pl b/src/core.pl index 383210c..802e4c2 100644 --- a/src/core.pl +++ b/src/core.pl @@ -7,26 +7,31 @@ use strict; -# dynamic scalar. MUST BE REDUCED IN SIZE!!! +# scalar. MUST BE REDUCED IN SIZE!!! ### TODO: reorder. use vars qw( - $answer $correction_plausible $talkchannel + $bot_misc_dir $bot_pid $bot_base_dir $bot_src_dir + $bot_data_dir $bot_config_dir $bot_state_dir $bot_run_dir + $answer $correction_plausible $talkchannel $bot_release $statcount $memusage $user $memusageOld $bot_version $dbh - $shm $host $msg $bot_misc_dir $bot_pid $bot_base_dir $noreply - $bot_src_dir $conn $irc $learnok $nick $ident $no_syscall + $shm $host $msg $noreply $conn $irc $learnok $nick $ident $force_public_reply $addrchar $userHandle $addressedother $floodwho $chan $msgtime $server $firsttime $wingaterun - $flag_quit $msgType + $flag_quit $msgType $no_syscall $utime_userfile $wtime_userfile $ucount_userfile $utime_chanfile $wtime_chanfile $ucount_chanfile + $pubsize $pubcount $pubtime + $msgsize $msgcount $msgtime + $notsize $notcount $nottime + $running ); -# dynamic hash. +# array. use vars qw(@joinchan @ircServers @wingateBad @wingateNow @wingateCache ); -### dynamic hash. MUST BE REDUCED IN SIZE!!! -# +### hash. MUST BE REDUCED IN SIZE!!! +# use vars qw(%count %netsplit %netsplitservers %flood %dcc %orig %nuh %talkWho %seen %floodwarn %param %dbh %ircPort %topic %moduleAge %last %time %mask %file @@ -44,8 +49,7 @@ $SIG{'__WARN__'} = 'doWarn'; $last{buflen} = 0; $last{say} = ""; $last{msg} = ""; -$userHandle = "default"; -$msgtime = time(); +$userHandle = "_default"; $wingaterun = time(); $firsttime = 1; $utime_userfile = 0; @@ -54,15 +58,46 @@ $ucount_userfile = 0; $utime_chanfile = 0; $wtime_chanfile = 0; $ucount_chanfile = 0; - -### CHANGE TO STATIC. -$bot_version = "blootbot cvs (20001212) -- $^O"; +$running = 0; +### more variables... +# static scalar variables. +$mask{ip} = '(\d+)\.(\d+)\.(\d+)\.(\d+)'; +$mask{host} = '[\d\w\_\-\/]+\.[\.\d\w\_\-\/]+'; +$mask{chan} = '[\#\&]\S*|_default'; +my $isnick1 = 'a-zA-Z\[\]\{\}\_\`\^\|\\\\'; +my $isnick2 = '0-9\-'; +$mask{nick} = "[$isnick1]{1}[$isnick1$isnick2]*"; +$mask{nuh} = '\S*!\S*\@\S*'; +$msgtime = time(); +$msgsize = 0; +$msgcount = 0; +$pubtime = 0; +$pubsize = 0; +$pubcount = 0; +$nottime = 0; +$notsize = 0; +$notcount = 0; +### +$bot_release = "1.1.0"; +if ( -d "CVS" ) { + use POSIX qw(strftime); + $bot_release .= strftime(" cvs (%Y%m%d)", gmtime( (stat("CVS"))[9] ) ); +} +$bot_version = "blootbot $bot_release -- $^O"; $noreply = "NOREPLY"; ########## ### misc commands. ### +sub whatInterface { + if (!&IsParam("Interface") or $param{'Interface'} =~ /IRC/) { + return "IRC"; + } else { + return "CLI"; + } +} + sub doExit { my ($sig) = @_; @@ -78,17 +113,25 @@ sub doExit { &status("parent caught SIG$sig (pid $$).") if (defined $sig); &status("--- Start of quit."); + $ident ||= "blootbot"; # lame hack. + + &status("Memory Usage: $memusage KiB"); - &closeDCC(); &closePID(); - &seenFlush(); - &quit($param{'quitMsg'}) if (&whatInterface() =~ /IRC/); + &closeStats(); + # shutdown IRC and related components. + if (&whatInterface() =~ /IRC/) { + &closeDCC(); + &seenFlush(); + &quit($param{'quitMsg'}); + } &writeUserFile(); &writeChanFile(); - &uptimeWriteFile() if (&ChanConfList("uptime")); - &closeDB(); + &uptimeWriteFile() if (&IsChanConf("uptime")); + &sqlCloseDB(); &closeSHM($shm); &dumpallvars() if (&IsParam("dumpvarsAtExit")); + &symdumpAll() if (&IsParam("symdumpAtExit")); &closeLog(); &closeSQLDebug() if (&IsParam("SQLDebug")); @@ -131,8 +174,11 @@ sub ChanConfList { return unless (defined $param); my %chan = &getChanConfList($param); - ### TODO: -option is included aswell though. - return keys %chan; + if (exists $chan{_default}) { + return keys %chanconf; + } else { + return keys %chan; + } } ##### @@ -172,24 +218,35 @@ sub getChanConfList { # Return: 1 for enabled, 0 for passive disable, -1 for active disable. sub IsChanConf { my($param) = shift; + my $debug = 0; # knocked tons of bugs with this! :) if (!defined $param) { - &WARN("param == NULL."); + &WARN("IsChanConf: param == NULL."); return 0; } - ### TODO: VERBOSITY on how chanconf returned 1 or 0 or -1. - my %chan = &getChanConfList($param); - if (!defined $msgType) { - return $chan{_default} || 0; + # should we use IsParam() externally where needed or hack it in + # here just in case? fix it later. + if (&IsParam($param)) { + &DEBUG("ICC: found '$param' option in main config file."); + return 1; } - if ($msgType eq "public") { - return $chan{lc $chan} || $chan{_default} || 0; + $chan ||= "_default"; + + my $old = $chan; + if ($chan =~ tr/A-Z/a-z/) { + &WARN("IsChanConf: lowercased chan. ($old)"); } - if ($msgType eq "private") { - return $chan{_default} || 0; + ### TODO: VERBOSITY on how chanconf returned 1 or 0 or -1. + my %chan = &getChanConfList($param); + my $nomatch = 0; + if (!defined $msgType) { + $nomatch++; + } else { + $nomatch++ if ($msgType eq ""); + $nomatch++ unless ($msgType =~ /^(public|private)$/i); } ### debug purposes only. @@ -198,6 +255,44 @@ sub IsChanConf { # &DEBUG(" $_ => $chan{$_}"); # } + if ($nomatch) { + if ($chan{$chan}) { + &DEBUG("ICC: other: $chan{$chan} (_default/$param)") if ($debug); + } elsif ($chan{_default}) { + &DEBUG("ICC: other: $chan{_default} (_default/$param)") if ($debug); + } else { + &DEBUG("ICC: other: 0 ($param)") if ($debug); + } + + return $chan{$chan} || $chan{_default} || 0; + } + + if ($msgType eq "public") { + if ($chan{$chan}) { + &DEBUG("ICC: public: $chan{$chan} ($chan/$param)") if ($debug); + } elsif ($chan{_default}) { + &DEBUG("ICC: public: $chan{_default} (_default/$param)") if ($debug); + } else { + &DEBUG("ICC: public: 0 ($param)") if ($debug); + } + + return $chan{$chan} || $chan{_default} || 0; + } + + if ($msgType eq "private") { + if ($chan{_default}) { + &DEBUG("ICC: private: $chan{_default} (_default/$param)") if ($debug); + } elsif ($chan{$chan}) { + &DEBUG("ICC: private: $chan{$chan} ($chan/$param) (hack)") if ($debug); + } else { + &DEBUG("ICC: private: 0 ($param)") if ($debug); + } + + return $chan{$chan} || $chan{_default} || 0; + } + + &DEBUG("ICC: no-match: 0/$param (msgType = $msgType)"); + return 0; } @@ -206,60 +301,126 @@ sub IsChanConf { # About: Retrieve value for 'param' value in current/default chan. # Return: scalar for success, undef for failure. sub getChanConf { - my($param,$chan) = @_; + my($param,$c) = @_; if (!defined $param) { - &WARN("param == NULL."); + &WARN("gCC: param == NULL."); return 0; } + # this looks evil... + if (0 and !defined $chan) { + &DEBUG("gCC: ok !chan... doing _default instead."); + } + + $c ||= $chan; + $c ||= "_default"; + $c = "_default" if ($c eq "*"); # fix! + my @c = grep /^\Q$c\E$/i, keys %chanconf; + + if (@c) { + if (0 and $c[0] ne $c) { + &WARN("c ne chan ($c[0] ne $chan)"); + } + return $chanconf{$c[0]}{$param}; + } + +# &DEBUG("gCC: returning _default... "); + return $chanconf{"_default"}{$param}; +} + +sub getChanConfDefault { + my($what, $default, $chan) = @_; + $chan ||= "_default"; - my @c = grep /^$chan$/i, keys %chanconf; - if (@c and $c[0] ne $chan) { - &WARN("c ne chan ($c[0] ne $chan)"); + if (exists $param{$what}) { + if (!exists $cache{config}{$what}) { + &status("config ($chan): backward-compatible option: found param{$what} ($param{$what}) instead of chan option"); + $cache{config}{$what} = 1; + } + + return $param{$what}; } + my $val = &getChanConf($what, $chan); + return $val if (defined $val); - return $chanconf{$c[0] || "_default"}{$param}; + $param{$what} = $default; + &status("config ($chan): auto-setting param{$what} = $default"); + $cache{config}{$what} = 1; + return $default; } -sub showProc { - my ($prefix) = $_[0] || ""; - if (!open(IN, "/proc/$$/status")) { - &ERROR("cannot open '/proc/$$/status'."); - return; +##### +# Usage: &findChanConf($param); +# About: Retrieve value for 'param' value from any chan. +# Return: scalar for success, undef for failure. +sub findChanConf { + my($param) = @_; + + if (!defined $param) { + &WARN("param == NULL."); + return 0; + } + + my $c; + foreach $c (keys %chanconf) { + foreach (keys %{ $chanconf{$c} }) { + next unless (/^$param$/); + + return $chanconf{$c}{$_}; + } } + return; +} + +sub showProc { + my ($prefix) = $_[0] || ""; + if ($^O eq "linux") { + if (!open(IN, "/proc/$$/status")) { + &ERROR("cannot open '/proc/$$/status'."); + return; + } + while () { $memusage = $1 if (/^VmSize:\s+(\d+) kB/); } close IN; - if (defined $memusageOld and &IsParam("DEBUG")) { - # it's always going to be increase. - my $delta = $memusage - $memusageOld; - my $str; - if ($delta == 0) { - return; - } elsif ($delta > 500) { - $str = "MEM:$prefix increased by $delta kB. (total: $memusage kB)"; - } elsif ($delta > 0) { - $str = "MEM:$prefix increased by $delta kB"; - } else { # delta < 0. - $delta = -$delta; - # never knew RSS could decrease, probably Size can't? - $str = "MEM:$prefix decreased by $delta kB. YES YES YES"; - } - - &status($str); - } - $memusageOld = $memusage; + } elsif ($^O eq "netbsd") { + $memusage = int( (stat "/proc/$$/mem")[7]/1024 ); + + } elsif ($^O =~ /^(free|open)bsd$/) { + my @info = split /\s+/, `/bin/ps -l -p $$`; + $memusage = $info[20]; + } else { $memusage = "UNKNOWN"; + return; + } + + if (defined $memusageOld and &IsParam("DEBUG")) { + # it's always going to be increase. + my $delta = $memusage - $memusageOld; + my $str; + if ($delta == 0) { + return; + } elsif ($delta > 500) { + $str = "MEM:$prefix increased by $delta KiB. (total: $memusage KiB)"; + } elsif ($delta > 0) { + $str = "MEM:$prefix increased by $delta KiB"; + } else { # delta < 0. + $delta = -$delta; + # never knew RSS could decrease, probably Size can't? + $str = "MEM:$prefix decreased by $delta KiB."; + } + + &status($str); } - ### TODO: FreeBSD/*BSD support. + $memusageOld = $memusage; } ###### @@ -271,36 +432,38 @@ sub setup { &openLog(); # write, append. &status("--- Started logging."); - foreach ("debian") { - my $dir = "$bot_base_dir/$_/"; - next if ( -d $dir); - &status("Making dir $_"); - mkdir $dir, 0755; - } - # read. - &loadLang($bot_misc_dir. "/blootbot.lang"); - &loadIRCServers($bot_misc_dir. "/ircII.servers"); + &loadLang($bot_data_dir. "/blootbot.lang"); + &loadIRCServers(); &readUserFile(); &readChanFile(); &loadMyModulesNow(); # must be after chan file. $shm = &openSHM(); &openSQLDebug() if (&IsParam("SQLDebug")); - &openDB($param{'DBName'}, $param{'SQLUser'}, $param{'SQLPass'}); + &sqlOpenDB($param{'DBName'}, $param{'DBType'}, $param{'SQLUser'}, + $param{'SQLPass'}); + &checkTables(); &status("Setup: ". &countKeys("factoids") ." factoids."); + &getChanConfDefault("sendPrivateLimitLines", 3); + &getChanConfDefault("sendPrivateLimitBytes", 1000); + &getChanConfDefault("sendPublicLimitLines", 3); + &getChanConfDefault("sendPublicLimitBytes", 1000); + &getChanConfDefault("sendNoticeLimitLines", 3); + &getChanConfDefault("sendNoticeLimitBytes", 1000); $param{tempDir} =~ s#\~/#$ENV{HOME}/#; - &status("Initial memory usage: $memusage kB"); + &status("Initial memory usage: $memusage KiB"); + &status("-------------------------------------------------------"); } sub setupConfig { $param{'VERBOSITY'} = 1; - &loadConfig($bot_misc_dir."/blootbot.config"); + &loadConfig($bot_config_dir."/blootbot.config"); - foreach ("ircNick", "ircUser", "ircName", "DBType", "tempDir") { + foreach ( qw(ircNick ircUser ircName DBType tempDir) ) { next if &IsParam($_); &ERROR("Parameter $_ has not been defined."); exit 1; @@ -317,18 +480,18 @@ sub setupConfig { if (! -d $param{tempDir}) { &status("making $param{tempDir}..."); - system("mkdir $param{tempDir}"); + mkdir $param{tempDir}, 0755; } # static scalar variables. - $file{utm} = "$bot_base_dir/$param{'ircUser'}.uptime"; - $file{PID} = "$bot_base_dir/$param{'ircUser'}.pid"; + $file{utm} = "$bot_state_dir/$param{'ircUser'}.uptime"; + $file{PID} = "$bot_run_dir/$param{'ircUser'}.pid"; } sub startup { if (&IsParam("DEBUG")) { &status("enabling debug diagnostics."); - ### I thought disabling this reduced memory usage by 1000 kB. + ### I thought disabling this reduced memory usage by 1000 KiB. use diagnostics; } @@ -339,15 +502,23 @@ sub startup { } sub shutdown { + my ($sig) = @_; # reverse order of &setup(). - &DEBUG("shutdown called."); + &status("--- shutdown called."); + + $ident ||= "blootbot"; # hack. + + if (!&isFileUpdated("$bot_state_dir/blootbot.users", $wtime_userfile)) { + &writeUserFile() + } + + if (!&isFileUpdated("$bot_state_dir/blootbot.chan", $wtime_chanfile)) { + &writeChanFile(); + } - &closeDB(); + &sqlCloseDB(); &closeSHM($shm); # aswell. TODO: use this in &doExit? &closeLog(); - ### is this valid? - &writeUserFile(); - &writeChanFile(); } sub restart { @@ -357,16 +528,21 @@ sub restart { &status("--- $sig called."); ### crappy bug in Net::IRC? - if (!$conn->connected and time - $msgtime > 900) { - &status("reconnecting because of uncaught disconnect."); -## $irc->start; + my $delta = time() - $msgtime; + &DEBUG("restart: dtime = $delta"); + if (!$conn->connected or time() - $msgtime > 900) { + &status("reconnecting because of uncaught disconnect \@ ".scalar(gmtime) ); +### $irc->start; + &clearIRCVars(); $conn->connect(); - return; +### return; } + &ircCheck(); # heh, evil! + &DCCBroadcast("-HUP called.","m"); - &shutdown(); - &loadConfig($bot_misc_dir."/blootbot.config"); + &shutdown($sig); + &loadConfig($bot_config_dir."/blootbot.config"); &reloadAllModules() if (&IsParam("DEBUG")); &setup(); @@ -381,9 +557,8 @@ sub loadConfig { my ($file) = @_; if (!open(FILE, $file)) { - &ERROR("FAILED loadConfig ($file): $!"); - &status("Please copy files/sample.config to files/blootbot.config"); - &status(" and edit files/blootbot.config, modify to tastes."); + &ERROR("Failed to read configuration file ($file): $!"); + &status("Please read the INSTALL file on how to install and setup this file."); exit 0; }