home *** CD-ROM | disk | FTP | other *** search
/ ftp.f-secure.com / 2014.06.ftp.f-secure.com.tar / ftp.f-secure.com / support / hotfix / fsis / IS-SpamControl.fsfix / iufssc / lib / Mail / SpamAssassin / SubProcBackChannel.pm < prev    next >
Text File  |  2006-11-29  |  4KB  |  168 lines
  1. # back-channel for communication between a master and multiple slave processes.
  2. #
  3. # <@LICENSE>
  4. # Licensed to the Apache Software Foundation (ASF) under one or more
  5. # contributor license agreements.  See the NOTICE file distributed with
  6. # this work for additional information regarding copyright ownership.
  7. # The ASF licenses this file to you under the Apache License, Version 2.0
  8. # (the "License"); you may not use this file except in compliance with
  9. # the License.  You may obtain a copy of the License at:
  10. # 
  11. #     http://www.apache.org/licenses/LICENSE-2.0
  12. # 
  13. # Unless required by applicable law or agreed to in writing, software
  14. # distributed under the License is distributed on an "AS IS" BASIS,
  15. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16. # See the License for the specific language governing permissions and
  17. # limitations under the License.
  18. # </@LICENSE>
  19.  
  20. package Mail::SpamAssassin::SubProcBackChannel;
  21.  
  22. use strict;
  23. use warnings;
  24. use bytes;
  25.  
  26. use IO::Socket;
  27. use Mail::SpamAssassin::Util;
  28. use Mail::SpamAssassin::Constants qw(:sa);
  29.  
  30. use vars qw {
  31. };
  32.  
  33. my @ISA = qw();
  34.  
  35. =head1 NAME
  36.  
  37. Mail::SpamAssassin::SubProcBackChannel - back-channel for communication between a master and multiple slave processes
  38.  
  39. =head1 METHODS
  40.  
  41. =over 4
  42.  
  43. =cut
  44.  
  45.  
  46. ###########################################################################
  47.  
  48. sub new {
  49.   my $class = shift;
  50.   $class = ref($class) || $class;
  51.  
  52.   my $self = shift;
  53.   if (!defined $self) { $self = { }; }
  54.   bless ($self, $class);
  55.  
  56.   $self->{kids} = { };
  57.   $self->{fileno_to_fh} = { };
  58.  
  59.   $self;
  60. }
  61.  
  62. ###########################################################################
  63.  
  64. sub set_selector {
  65.   my ($self, $sel) = @_;
  66.   $self->{selector} = $sel;
  67. }
  68.  
  69. sub setup_backchannel_parent_pre_fork {
  70.   my ($self) = @_;
  71.  
  72.   my $io = IO::Socket->new();
  73.   ($self->{latest_kid_fh}, $self->{parent}) =
  74.             $io->socketpair(AF_UNIX,SOCK_STREAM,PF_UNSPEC)
  75.             or die "backchannel: socketpair failed: $!";
  76.  
  77.   # set those to use non-blocking I/O
  78.   $self->{parent}->blocking(0)
  79.             or die "backchannel: set non-blocking failed: $!";
  80.   $self->{latest_kid_fh}->blocking(0)
  81.             or die "backchannel: set non-blocking failed: $!";
  82. }
  83.  
  84. sub setup_backchannel_parent_post_fork {
  85.   my ($self, $pid) = @_;
  86.  
  87.   my $fh = $self->{latest_kid_fh};
  88.  
  89.   close $self->{parent};    # because it's us!
  90.  
  91.   # disable caching for parent<->child relations
  92.   my ($old) = select($fh);
  93.   $|++;
  94.   select($old);
  95.  
  96.   $self->{kids}->{$pid} = $fh;
  97.   $self->add_to_selector($fh);
  98. }
  99.  
  100. sub add_to_selector {
  101.   my ($self, $fh) = @_;
  102.   my $fno = fileno($fh);
  103.   $self->{fileno_to_fh}->{$fno} = $fh;
  104.   vec (${$self->{selector}}, $fno, 1) = 1;
  105. }
  106.  
  107. sub select_vec_to_fh_list {
  108.   my ($self, $vec) = @_;
  109.   my $i = -1;
  110.  
  111.   # grotesque hackery alert! ;)   turn the vec() map of fds into a list of
  112.   # filehandles.  note that filenos that don't have a filehandle in the
  113.   # {fileno_to_fh} hash will be ignored; this is by design, so that other fhs
  114.   # can be selected on using the same vec, and the caller can just check for
  115.   # those in their own code, before they fall back to using this method.
  116.  
  117.   return grep {
  118.         defined
  119.       } map {
  120.         $i++;
  121.         ($_ ? $self->{fileno_to_fh}->{$i} : undef);
  122.       } split (//, unpack ("b*", $vec));
  123. }
  124.  
  125. sub get_socket_for_child {
  126.   my ($self, $pid) = @_;
  127.   return $self->{kids}->{$pid};
  128. }
  129.  
  130. sub delete_socket_for_child {
  131.   my ($self, $pid) = @_;
  132.   delete $self->{kids}->{$pid};
  133. }
  134.  
  135. ###########################################################################
  136.  
  137. sub setup_backchannel_child_post_fork {
  138.   my ($self) = @_;
  139.  
  140.   close $self->{latest_kid_fh}; # because it's us!
  141.  
  142.   my $old = select($self->{parent});
  143.   $| = 1;   # print to parent by default, turn off buffering
  144.   select($old);
  145. }
  146.  
  147. sub get_parent_socket {
  148.   my ($self) = @_;
  149.   return $self->{parent};
  150. }
  151.  
  152. ############################################################################
  153.  
  154. 1;
  155.  
  156. __END__
  157.  
  158. =back
  159.  
  160. =head1 SEE ALSO
  161.  
  162. C<Mail::SpamAssassin>
  163. C<Mail::SpamAssassin::ArchiveIterator>
  164. C<Mail::SpamAssassin::SpamdPreforkScaling>
  165. C<spamassassin>
  166. C<spamd>
  167. C<mass-check>
  168.