X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=src%2Fcore.pl;h=77c8f9c4fc3767d981a0d99290c6f14e5ae75532;hb=22a73ba6205f6120785f2e6a17921a76774ec3f7;hp=b075235c7b9d8c023b1b5eaed393e6053d9e654f;hpb=dfdd8e6e52183b27328c71609b478c0b9aa41e00;p=infobot.git diff --git a/src/core.pl b/src/core.pl index b075235..77c8f9c 100644 --- a/src/core.pl +++ b/src/core.pl @@ -10,15 +10,20 @@ use strict; # dynamic 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. @@ -29,7 +34,7 @@ use vars qw(@joinchan @ircServers @wingateBad @wingateNow @wingateCache # use vars qw(%count %netsplit %netsplitservers %flood %dcc %orig %nuh %talkWho %seen %floodwarn %param %dbh %ircPort - %jointime %topic %joinverb %moduleAge %last %time %mask %file + %topic %moduleAge %last %time %mask %file %forked %chanconf %channels ); @@ -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,9 +58,25 @@ $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... +$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"; ########## @@ -78,17 +98,24 @@ sub doExit { &status("parent caught SIG$sig (pid $$).") if (defined $sig); &status("--- Start of quit."); + $ident ||= "blootbot"; # lame hack. - &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 (&IsParam("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")); @@ -131,8 +158,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,31 +202,81 @@ 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; } + # should we use IsParam() externally where needed or hack it in + # here just in case? fix it later. + if (&IsParam($param)) { + &status("ICC: found '$param' option in main config file."); + return 1; + } + + $chan ||= "_default"; + + my $old = $chan; + if ($chan =~ tr/A-Z/a-z/) { + &WARN("IsChanConf: lowercased chan. ($old)"); + } + ### TODO: VERBOSITY on how chanconf returned 1 or 0 or -1. my %chan = &getChanConfList($param); + my $nomatch = 0; if (!defined $msgType) { - return $chan{_default} || 0; + $nomatch++; + } else { + $nomatch++ if ($msgType eq ""); + $nomatch++ unless ($msgType =~ /^(public|private)$/i); + } + +### debug purposes only. +# &DEBUG("param => $param, msgType => $msgType."); +# foreach (keys %chan) { +# &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") { - return $chan{lc $chan} || $chan{_default} || 0; + 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") { - return $chan{_default} || 0; - } + 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); + } - &DEBUG("param => $param, msgType => $msgType."); - foreach (keys %chan) { - &DEBUG(" $_ => $chan{$_}"); + return $chan{$chan} || $chan{_default} || 0; } + &DEBUG("ICC: no-match: 0/$param (msgType = $msgType)"); + return 0; } @@ -205,62 +285,125 @@ 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("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) = @_; + + if (exists $param{$what}) { + if (!exists $cache{config}{$what}) { + &status("conf: backward-compat: found param{$what} ($param{$what}) instead."); + $cache{config}{$what} = 1; + } + + return $param{$what}; + } + my $val = &getChanConf($what, $chan); + return $val if (defined $val); + + $param{$what} = $default; + &status("conf: auto-setting param{$what} = $default"); + $cache{config}{$what} = 1; + return $default; +} + + +##### +# 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; } - $chan ||= "_default"; - my @c = grep /^$chan$/i, keys %chanconf; + my $c; + foreach $c (keys %chanconf) { + foreach (keys %{ $chanconf{$c} }) { + next unless (/^$param$/); - if ($c[0] ne $chan) { - &WARN("c ne chan ($c[0] ne $chan)"); + &DEBUG("chanconf{$c}{$_} ..."); + return $chanconf{$c}{$_}; + } } - return $chanconf{$c[0] || "_default"}{$param}; + return; } sub showProc { my ($prefix) = $_[0] || ""; - if (!open(IN, "/proc/$$/status")) { - &ERROR("cannot open '/proc/$$/status'."); - return; - } - 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); - &DCCBroadcast($str) if (&whatInterface() =~ /IRC/ && - grep(/Irc.pl/, keys %moduleAge)); - } - $memusageOld = $memusage; + } elsif ($^O eq "netbsd") { + $memusage = (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 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); } - ### TODO: FreeBSD/*BSD support. + $memusageOld = $memusage; } ###### @@ -272,16 +415,9 @@ 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. @@ -289,19 +425,28 @@ sub setup { $shm = &openSHM(); &openSQLDebug() if (&IsParam("SQLDebug")); &openDB($param{'DBName'}, $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; @@ -318,12 +463,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 { @@ -336,18 +481,24 @@ sub startup { $count{'Question'} = 0; $count{'Update'} = 0; $count{'Dunno'} = 0; + $count{'Moron'} = 0; } 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? &closeLog(); - ### is this valid? - &writeUserFile(); - &writeChanFile(); } sub restart { @@ -358,14 +509,18 @@ sub restart { ### crappy bug in Net::IRC? if (!$conn->connected and time - $msgtime > 900) { - &status("reconnecting because of uncaught disconnect."); -## $irc->start; + &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(); @@ -380,9 +535,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; }