/[virtual-ldap]/bin/ldap-rewrite.pl
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /bin/ldap-rewrite.pl

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 72 by dpavlin, Mon Feb 22 22:18:55 2010 UTC revision 73 by dpavlin, Tue Feb 23 00:11:35 2010 UTC
# Line 67  sub handle { Line 67  sub handle {
67          # read from client          # read from client
68          asn_read($clientsocket, my $reqpdu);          asn_read($clientsocket, my $reqpdu);
69          if ( ! $reqpdu ) {          if ( ! $reqpdu ) {
70                  warn "WARNING no reqpdu\n";                  warn "client closed connection\n";
71                  return 1;                  return 0;
72          }          }
73          $reqpdu = log_request($reqpdu);          $reqpdu = log_request($reqpdu);
74    
# Line 79  sub handle { Line 79  sub handle {
79          my $ready;          my $ready;
80          my $sel = IO::Select->new($serversocket);          my $sel = IO::Select->new($serversocket);
81          for( $ready = 1 ; $ready ; $ready = $sel->can_read(0)) {          for( $ready = 1 ; $ready ; $ready = $sel->can_read(0)) {
82                  asn_read($serversocket, my $respdu) or return 1;                  asn_read($serversocket, my $respdu);
83                    if ( ! $respdu ) {
84                            warn "server closed connection\n";
85                            return 0;
86                    }
87                  $respdu = log_response($respdu);                  $respdu = log_response($respdu);
88                  # and send the result to the client                  # and send the result to the client
89                  print $clientsocket $respdu;                  print $clientsocket $respdu || return 0;
90          }          }
91    
92          return 0;          return 1;
93  }  }
94    
95    
96  sub log_request {  sub log_request {
97          my $pdu=shift;          my $pdu=shift;
98    
99            die "empty pdu" unless $pdu;
100    
101  #       print '-' x 80,"\n";  #       print '-' x 80,"\n";
102  #       print "Request ASN 1:\n";  #       print "Request ASN 1:\n";
103  #       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);  #       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
# Line 116  sub log_request { Line 122  sub log_request {
122    
123  sub log_response {  sub log_response {
124          my $pdu=shift;          my $pdu=shift;
125            die "empty pdu" unless $pdu;
126    
127  #       print '-' x 80,"\n";  #       print '-' x 80,"\n";
128  #       print "Response ASN 1:\n";  #       print "Response ASN 1:\n";
# Line 167  sub log_response { Line 174  sub log_response {
174          return $pdu;          return $pdu;
175  }  }
176    
 sub run_proxy {  
         my $listenersock = shift;  
         my $targetsock=shift;  
   
         die "Could not create listener socket: $!\n" unless $listenersock;  
         die "Could not create connection to server: $!\n" unless $targetsock;  
   
         my $sel = IO::Select->new($listenersock);  
         my %Handlers;  
         while (my @ready = $sel->can_read) {  
                 foreach my $fh (@ready) {  
                         if ($fh == $listenersock) {  
                                 # let's create a new socket  
                                 my $psock = $listenersock->accept;  
                                 $sel->add($psock);  
                         } else {  
                                 my $result = handle($fh,$targetsock);  
                                 if ($result) {  
                                         # we have finished with the socket  
                                         $sel->remove($fh);  
                                         $fh->close;  
                                         delete $Handlers{*$fh};  
                                 }  
                         }  
                 }  
         }  
 }  
   
177    
178  my $listenersock = IO::Socket::INET->new(  my $listenersock = IO::Socket::INET->new(
179          Listen => 5,          Listen => 5,
# Line 203  my $listenersock = IO::Socket::INET->new Line 182  my $listenersock = IO::Socket::INET->new
182          LocalAddr => $config->{listen},          LocalAddr => $config->{listen},
183  ) || die "can't open listen socket: $!";  ) || die "can't open listen socket: $!";
184    
185    our $server_sock;
186    
187  my $targetsock = $config->{upstream_ssl}  sub connect_to_server {
188          ? IO::Socket::INET->new(          my $sock;
189                  Proto => 'tcp',          if ( $config->{upstream_ssl} ) {
190                  PeerAddr => $config->{upstream_ldap},                  $sock = IO::Socket::SSL->new( $config->{upstream_ldap} . ':ldaps' );
191                  PeerPort => 389,          } else {
192          )                  $sock = IO::Socket::INET->new(
193          : IO::Socket::SSL->new( $config->{upstream_ldap} . ':ldaps')                          Proto => 'tcp',
194          || die "can't open upstream socket: $!";                          PeerAddr => $config->{upstream_ldap},
195                            PeerPort => 389,
196                    );
197            }
198            die "can't open ", $config->{upstream_ldap}, " $!\n" unless $sock;
199            warn "## connected to ", $sock->peerhost, ":", $sock->peerport, "\n";
200            return $sock;
201    }
202    
203  run_proxy($listenersock,$targetsock);  my $sel = IO::Select->new($listenersock);
204    while (my @ready = $sel->can_read) {
205            foreach my $fh (@ready) {
206                    if ($fh == $listenersock) {
207                            # let's create a new socket
208                            my $psock = $listenersock->accept;
209                            $sel->add($psock);
210                            warn "## add $psock " . time;
211                    } else {
212                            $server_sock->{$fh} ||= connect_to_server;
213                            if ( ! handle($fh,$server_sock->{$fh}) ) {
214                                    warn "## remove $fh " . time;
215                                    $sel->remove($server_sock->{$fh});
216                                    $server_sock->{$fh}->close;
217                                    delete $server_sock->{$fh};
218                                    # we have finished with the socket
219                                    $sel->remove($fh);
220                                    $fh->close;
221                            }
222                    }
223            }
224    }
225    
226  1;  1;
Legend:
Removed from v.72  
changed lines
  Added in v.73
  ViewVC Help
Powered by ViewVC 1.1.26