全部博文(1144)
分类: LINUX
2010-05-20 01:52:50
Perl进程间通讯前几天做了一个比较简单的东西,但感觉比较意思就在这里记下来。大概的需求是这样的:有一个程序在会产生一些警告信息并保存在/var/log/alert中的。如果有新的警告信息时就要在服务器上发出警告的声音,但启动是通过web界面的而不是在命令行。
刚开始想到步骤是这样的:
1、 因为是要在后台运行,肯定是fork一个子进程,父进程退出而不会使web程序阻塞在此,
2、 然后再fork一个子进程,父进程负责监听日志文件的更新。子进程负责调用声音程序。两者之者用管道通讯。
3、 要实时监控日志文件的更新,因为不想搞太复杂就用管道的方式:open(LOG,"tail -n 2 -f /var/log/alert| ") ||die "Unable to open log file $!\n";这样就可以简单的得到文件的实时更新
4、 要考虑到退出,所以需要一个文件保存进程ID。
这样实现的代码很简单:
my $pidfile ="/tmp/alertsounds.pid";
sub StartAlertSounds{
my $pid;
if ($pid = fork()) {
return;
} elsif (defined $pid) {
system("echo pid=$$ > $pidfile'");#保存进程ID
&RunAlertSounds();
}
}else {
die "Can't fork: $!\n";
}
}
sub RunAlertSounds{
my $pid;
pipe(README, WRITEME);
if ($pid = fork()) { #父进程
close(README);
open(LOG,"tail -n 2 -f /var/log/alert| ") ||die "Unable to open log file $!\n";
while (
print WRITEME "Sounds\n";
}
close(LOG );
close(WRITEME);
}elsif (defined $pid) { #子进程
close(WRITEME);
while(
system("/usr/bin/sounds/AlertSounds >/dev/null 2>&1");
}
close(README);
}else {
die "Can't fork: $!\n";
}
}
但测试一下就发现了不少问题:
1、 退出时open(LOG,"tail -n 2 -f /var/log/snort/alert| ")所产生的子进程不会一起退出。
2、 当短时间内(比如1秒)有多条信息过来时,我们希望只产生一条警告声音,但现在会连续响多条。
3、 有时父进程往管道写数据时子进程不一定能马上收到;
分析一下就有下面的解决方法:
1、 使父进程成为进程组的头领进程,然后在父进程退出前给进程组发送退出信号。
2、 使用锁机制,父进程写数据前锁定。等子进程调用声音程序完成后再解锁。在解锁前如果父进程有监听到新的警告就忽略。简单的一点的锁机制就用一个文件:父进程创建一个空文件表示已经上锁,子进程删除这个文件表示解锁
3、 这个是因为缓存问题,把管道的缓存设置好就行了。
修改后的代码如下:
my $pidfile ="/tmp/alertsounds.pid";
my $lockfilename="/tmp/sounds.lock";
sub StartAlertSounds{
my $pid;
if ($pid = fork()) {
return;
} elsif (defined $pid) {
setpgrp(0,0);#当前进程成为进程组的头领
$SIG{INT} = \&catch_zap; # INT信号处理
$SIG{QUIT} = \&catch_zap; # QUIT信号处理
system("echo $$ > $pidfile'");#保存进程ID
&RunAlertSounds();
}
}else {
die "Can't fork: $!\n";
}
}
sub RunAlertSounds{
my $pid;
system("rm -f $lockfilename");
pipe(README, WRITEME);
if ($pid = fork()) { #父进程
close(README);#关闭读端
select( WRITEME ); #这里一定要选择,不然设置的缓存大小不会对WRITEME有效
$| = 1;
open(LOG,"tail -n 2 -f /var/log/alert| ") ||die "Unable to open log file $!\n";
while (
system(“touch $lockfilename");#加锁
print WRITEME "Sounds\n";
}
close(LOG );
close(WRITEME);
waitpid($pid, 0);
}elsif (defined $pid) { #子进程
close(WRITEME);#关闭写端
while(
system(“/usr/bin/sounds/AlertSounds >/dev/null 2>&1");
sleep(1);
system("rm -f $lockfilename");#解锁
}
close(README);
}else {
die "Can't fork: $!\n";
}
}
sub catch_zap{
local $SIG{QUIT} = 'IGNORE'; # 排除自己
kill(QUIT, -$$); # 通知进程组的所有进程退出
system("echo > $pidfile");
exit;
}
sub StoptAlertSounds{
if(-e $pidfile ){
@pid = `cat $pidfile `;
kill(QUIT,$pid[0]}) if(defined $pid[0] and $pid[0] ne '');
}
}