Chinaunix首页 | 论坛 | 博客
  • 博客访问: 151901
  • 博文数量: 20
  • 博客积分: 1515
  • 博客等级: 上尉
  • 技术积分: 305
  • 用 户 组: 普通用户
  • 注册时间: 2009-05-14 10:06
文章分类

全部博文(20)

文章存档

2011年(2)

2010年(4)

2009年(14)

我的朋友

分类:

2009-12-30 11:13:29

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) |
给主人留下些什么吧!~~