X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=blootbot%2Fsrc%2Fcore.pl;h=9cf0c802375862491cd26d90e5b75a52852ce6a3;hb=5a55d562a05391c957f9a06bf01c5638bedf3d3c;hp=cfd304f9829f53bc355ebc379e89b87ce0c98f73;hpb=88c61620c7483afb1d43c261060cebacc98d8ef5;p=infobot.git diff --git a/blootbot/src/core.pl b/blootbot/src/core.pl index cfd304f..9cf0c80 100644 --- a/blootbot/src/core.pl +++ b/blootbot/src/core.pl @@ -7,25 +7,30 @@ 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 @@ -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,8 +58,25 @@ $ucount_userfile = 0; $utime_chanfile = 0; $wtime_chanfile = 0; $ucount_chanfile = 0; - -$bot_version = "blootbot cvs (20010214) -- $^O"; +$running = 0; +### more variables... +$msgtime = time(); +$msgsize = 0; +$msgcount = 0; +$pubtime = 0; +$pubsize = 0; +$pubcount = 0; +$nottime = 0; +$notsize = 0; +$notcount = 0; +### +if ( -d "CVS" ) { + use POSIX qw(strftime); + $bot_release = strftime("cvs (%Y%m%d)", localtime( (stat("CVS"))[9] ) ); +} else { + $bot_release = "1.0.10 (2001xxxx)"; +} +$bot_version = "blootbot $bot_release -- $^O"; $noreply = "NOREPLY"; ########## @@ -77,17 +98,26 @@ sub doExit { &status("parent caught SIG$sig (pid $$).") if (defined $sig); &status("--- Start of quit."); + $ident ||= "blootbot"; # lame hack. + + &status("Memory Usage: $memusage kB"); - &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")); + &uptimeWriteFile() if (&IsChanConf("uptime")); + &News::writeNews() if (&ChanConfList("news")); &closeDB(); &closeSHM($shm); &dumpallvars() if (&IsParam("dumpvarsAtExit")); + &symdumpAll() if (&IsParam("symdumpAtExit")); &closeLog(); &closeSQLDebug() if (&IsParam("SQLDebug")); @@ -130,14 +160,9 @@ sub ChanConfList { return unless (defined $param); my %chan = &getChanConfList($param); - # what if we have it set on _default and a few channels with - # negative set? perhaps we should disable -blah settings to remove - # this mess. - if (exists $chan{_default}) { return keys %chanconf; } else { - ### TODO: -option is included aswell though. return keys %chan; } } @@ -179,24 +204,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. @@ -205,6 +241,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; } @@ -213,60 +287,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 /^$c$/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("conf: backward-compat: found param{$what} ($param{$what}) for $chan instead."); + $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("conf: $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; } - ### TODO: FreeBSD/*BSD support. + + 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."; + } + + &status($str); + } + $memusageOld = $memusage; } ###### @@ -278,15 +418,8 @@ 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"); + &loadLang($bot_data_dir. "/blootbot.lang"); &loadIRCServers(); &readUserFile(); &readChanFile(); @@ -294,20 +427,30 @@ sub setup { $shm = &openSHM(); &openSQLDebug() if (&IsParam("SQLDebug")); - &openDB($param{'DBName'}, $param{'SQLUser'}, $param{'SQLPass'}); + &openDB($param{'DBName'}, $param{'DBType'}, $param{'SQLUser'}, + $param{'SQLPass'}); + &checkTables(); &status("Setup: ". &countKeys("factoids") ." factoids."); + &News::readNews() if (&ChanConfList("news")); + &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("-------------------------------------------------------"); } 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; @@ -324,12 +467,12 @@ 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 { @@ -347,7 +490,15 @@ sub startup { sub shutdown { # reverse order of &setup(). - &DEBUG("shutdown called."); + &status("--- shutdown called."); + + $ident ||= "blootbot"; # hack. + + # opened files must be written to on shutdown/hup/whatever + # unless they're write-only, like uptime. + &writeUserFile(); + &writeChanFile(); + &News::writeNews() if (&ChanConfList("news")); &closeDB(); &closeSHM($shm); # aswell. TODO: use this in &doExit? @@ -361,16 +512,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(localtime) ); +### $irc->start; + &clearIRCVars(); $conn->connect(); - return; +### return; } + &ircCheck(); # heh, evil! + &DCCBroadcast("-HUP called.","m"); &shutdown(); - &loadConfig($bot_misc_dir."/blootbot.config"); + &loadConfig($bot_config_dir."/blootbot.config"); &reloadAllModules() if (&IsParam("DEBUG")); &setup(); @@ -385,9 +541,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; }