X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=src%2FShm.pl;h=efdfced295665d79c98598e3a26facd549272e38;hb=f7cae48a17d6decd0a9bd997188271daa0a885b1;hp=0c6c4f6fe5f5bfccf7de9d5f8f6618104a4b5dcd;hpb=868ede9718cb77f516e6620a5225347d4e19e468;p=infobot.git

diff --git a/src/Shm.pl b/src/Shm.pl
index 0c6c4f6..efdfced 100644
--- a/src/Shm.pl
+++ b/src/Shm.pl
@@ -5,7 +5,7 @@
 #   Created: 20000124
 #
 
-if (&IsParam("useStrict")) { use strict; }
+# use strict;	# TODO
 
 use POSIX qw(_exit);
 
@@ -13,6 +13,11 @@ sub openSHM {
     my $IPC_PRIVATE = 0;
     my $size = 2000;
 
+    if (&IsParam("noSHM")) {
+	&status("Shared memory: Disabled. WARNING: bot may become unreliable");
+	return 0;
+    }
+
     if (defined( $_ = shmget($IPC_PRIVATE, $size, 0777) )) {
 	&status("Created shared memory (shm) key: [$_]");
 	return $_;
@@ -40,7 +45,10 @@ sub shmRead {
     my $size = 3*80;
     my $retval = '';
 
+    return '' if (&IsParam("noSHM"));
+
     if (shmread($key,$retval,$position,$size)) {
+	&DEBUG("shmRead($key): $retval");
 	return $retval;
     } else {
 	&ERROR("shmRead: failed: $!");
@@ -55,22 +63,32 @@ sub shmWrite {
     my $position = 0;
     my $size = 80*3;
 
-    # NULL hack.
-    ### TODO: create shmClear to deal with this.
-    if ($str !~ /^$/) {
-	my $read = &shmRead($key);
-	$read =~ s/\0+//g;
+    return if (&IsParam("noSHM"));
 
-	if ($str eq "") {
-	    $str = time().": ";		# time stamping, null.
-	} elsif ($read eq "") {
-	    $str = time().": ";		# timestamping.
-	} else {
-	    $str = $read ."||". $str;
+    if (length($str) > $size) {
+	&status("ERROR: length(str) (..)>$size...");
+	return;
+    }
+
+    if (length($str) == 0) {
+	# does $size overwrite the whole lot?
+	# if not, set to 2000.
+	if (!shmwrite($key, '', $position, $size)) {
+	    &ERROR("shmWrite: failed: $!");
 	}
+	return;
     }
 
-    if (!shmwrite($key,$str,$position,$size)) {
+    my $read = &shmRead($key);
+    $read =~ s/\0+//g;
+    if ($read eq "") {
+	$str = sprintf("%s:%d:%d: ", $param{ircNick}, $bot_pid, time());
+    } else {
+	$str = $read ."||". $str;
+    }
+
+    if (!shmwrite($key, $str, $position, $size)) {
+	&DEBUG("shmWrite($key, $str)");
 	&ERROR("shmWrite: failed: $!");
     }
 }
@@ -92,31 +110,30 @@ sub addForked {
     }
 
     foreach (keys %forked) {
-	my $time = time() - $forked{$_}{Time};
+	my $n = $_;
+	my $time = time() - $forked{$n}{Time};
 	next unless ($time > $forker_timeout);
 
 	### TODO: use &time2string()?
-	&WARN("Fork: looks like we lost '$_', executed $time ago");
+	&WARN("Fork: looks like we lost '$n', executed $time ago");
 
-	my $pid = $forked{$name}{PID};
+	my $pid = $forked{$n}{PID};
 	if (!defined $pid) {
-	    &WARN("Fork: no pid for $name.");
-	    delete $forked{$name};
+	    &WARN("Fork: no pid for $n.");
+	    delete $forked{$n};
 	    next;
 	}
 
-	# don't kill parent!
-	if ($pid == $$) {
-	    &status("Fork: pid == \$\$ ($$)");
-	    next;
-	}
+	if ($pid == $bot_pid) {
+	    # don't kill parent, just warn.
+	    &status("Fork: pid == \$bot_pid == \$\$ ($bot_pid)");
 
-	if ( -d "/proc/$pid") {
+	} elsif ( -d "/proc/$pid") {	# pid != bot_pid.
 	    &status("Fork: killing $name ($pid)");
 	    kill 9, $pid;
 	}
 
-	delete $forked{$name};
+	delete $forked{$n};
     }
 
     my $count = 0;
@@ -126,10 +143,11 @@ sub addForked {
 	if ($count > 3) {	# 3 seconds.
 	    my $list = join(', ', keys %forked);
 	    if (defined $who) {
-		&msg($who, "already running ($list) => exceeded allowed forked processes count (1?).");
+		&msg($who, "exceeded allowed forked count (shm $shm): $list");
 	    } else {
-		&status("Fork: I ran too many forked processes :) Giving up $name.");
+		&status("Fork: I ran too many forked processes :) Giving up $name. Shm: $shm");
 	    }
+
 	    return 0;
 	}
 
@@ -207,4 +225,54 @@ sub delForked {
     POSIX::_exit(0);
 }
 
+sub shmFlush {
+    return if ($$ != $::bot_pid); # fork protection.
+
+    if (@_) {
+	&ScheduleThis(5, "shmFlush");
+	return if ($_[0] eq "2");
+    }
+
+    my $time;
+    my $shmmsg = &shmRead($shm);
+    # remove padded \0's.
+    $shmmsg =~ s/\0//g;
+    return if (length($shmmsg) == 0);
+    if ($shmmsg =~ s/^(\S+):(\d+):(\d+): //) {
+	my $n	= $1;
+	my $pid	= $2;
+	$time	= $3;
+    } else {
+	&status("warn: shmmsg='$shmmsg'.");
+	return;
+    }
+
+    foreach (split '\|\|', $shmmsg) {
+	next if (/^$/);
+	&VERB("shm: Processing '$_'.",2);
+
+	if (/^DCC SEND (\S+) (\S+)$/) {
+	    my ($nick,$file) = ($1,$2);
+	    if (exists $dcc{'SEND'}{$who}) {
+		&msg($nick, "DCC already active.");
+	    } else {
+		&DEBUG("shm: dcc sending $2 to $1.");
+		$conn->new_send($1,$2);
+		$dcc{'SEND'}{$who} = time();
+	    }
+	} elsif (/^SET FORKPID (\S+) (\S+)/) {
+	    $forked{$1}{PID} = $2;
+	} elsif (/^DELETE FORK (\S+)$/) {
+	    delete $forked{$1};
+	} elsif (/^EVAL (.*)$/) {
+	    &DEBUG("evaling '$1'.");
+	    eval $1;
+	} else {
+	    &DEBUG("shm: unknown msg. ($_)");
+	}
+    }
+
+    &shmWrite($shm,"") if ($shmmsg ne "");
+}
+
 1;