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 ''); } }
|
阅读(1739) | 评论(0) | 转发(0) |