added rewrite of LDAP bind CN username@domain.com -> uid=username,dc=domain,dc=com

[virtual-ldap] / bin / ldap-rewrite.pl

diff --git a/bin/ldap-rewrite.pl b/bin/ldap-rewrite.pl
index ebf25b2..cdc9534 100755 (executable)
--- a/bin/ldap-rewrite.pl
+++ b/bin/ldap-rewrite.pl
@@ -3,18 +3,60 @@
 # This program is free software; you can redistribute it and/or
 # modify it under the same terms as Perl itself.
 
 # This program is free software; you can redistribute it and/or
 # modify it under the same terms as Perl itself.
 

+# It's modified by Dobrica Pavlinusic <dpavlin@rot13.org> to include following:
+#
+# * rewrite LDAP bind request cn: username@domain.com -> uid=username,dc=domain,dc=com
+# * rewrite search responses:
+# ** expand key:value pairs from hrEduPersonUniqueNumber into hrEduPersonUniqueNumber_key
+# ** augment response with yaml/dn.yaml data (for external data import)

 
 use strict;
 use warnings;
 
 use IO::Select;
 use IO::Socket;
 
 use strict;
 use warnings;
 
 use IO::Select;
 use IO::Socket;

+use IO::Socket::SSL;

 use warnings;
 use Data::Dump qw/dump/;
 use Convert::ASN1 qw(asn_read);
 use Net::LDAP::ASN qw(LDAPRequest LDAPResponse);
 use warnings;
 use Data::Dump qw/dump/;
 use Convert::ASN1 qw(asn_read);
 use Net::LDAP::ASN qw(LDAPRequest LDAPResponse);

-our $VERSION = '0.2';
+our $VERSION = '0.3';

 use fields qw(socket target);
 use fields qw(socket target);

+use YAML qw/LoadFile/;
+
+my $debug = 1;
+
+my $config = {
+       yaml_dir => './yaml/',
+       listen => shift @ARGV || 'localhost:1389',
+       upstream_ldap => 'ldap.ffzg.hr',
+       upstream_ssl => 1,
+       overlay_prefix => 'ffzg-',
+       log_file => 'log',
+
+};
+
+my $log_fh;
+
+sub log {
+       if ( ! $log_fh ) {
+               open($log_fh, '>>', $config->{log_file}) || die "can't open ", $config->{log_file},": $!";
+               print $log_fh "# " . time;
+       }
+       $log_fh->autoflush(1);
+       print $log_fh join("\n", @_),"\n";
+}
+
+BEGIN {
+       $SIG{'__WARN__'} = sub { warn @_; main::log(@_); }
+}
+
+
+if ( ! -d $config->{yaml_dir} ) {
+       warn "DISABLE ", $config->{yaml_dir}," data overlay";
+}
+
+warn "# config = ",dump( $config );

 
 sub handle {
        my $clientsocket=shift;
 
 sub handle {
        my $clientsocket=shift;


@@ -22,9 +64,11 @@ sub handle {
 
        # read from client
        asn_read($clientsocket, my $reqpdu);
 
        # read from client
        asn_read($clientsocket, my $reqpdu);

-       log_request($reqpdu);
-
-       return 1 unless $reqpdu;
+       if ( ! $reqpdu ) {
+               warn "WARNING no reqpdu\n";
+               return 1;
+       }
+       $reqpdu = log_request($reqpdu);

 
        # send to server
        print $serversocket $reqpdu or die "Could not send PDU to server\n ";
 
        # send to server
        print $serversocket $reqpdu or die "Could not send PDU to server\n ";


@@ -46,40 +90,78 @@ sub handle {
 sub log_request {
        my $pdu=shift;
 
 sub log_request {
        my $pdu=shift;
 

-       print '-' x 80,"\n";
-       print "Request ASN 1:\n";
-       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
-       print "Request Perl:\n";
+#      print '-' x 80,"\n";
+#      print "Request ASN 1:\n";
+#      Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
+#      print "Request Perl:\n";

        my $request = $LDAPRequest->decode($pdu);
        my $request = $LDAPRequest->decode($pdu);

-       print dump($request);
+       warn "## request = ",dump($request);
+
+       if ( defined $request->{bindRequest} ) {
+               if ( $request->{bindRequest}->{name} =~ m{@} ) {
+                       my $old = $request->{bindRequest}->{name};
+                       $request->{bindRequest}->{name} =~ s/[@\.]/,dc=/g;
+                       $request->{bindRequest}->{name} =~ s/^/uid=/;
+                       warn "rewrite bind cn $old -> ", $request->{bindRequest}->{name};
+                       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu) if $debug;
+                       $pdu = $LDAPRequest->encode($request);
+                       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu) if $debug;
+               }
+       }
+
+       return $pdu;

 }
 
 sub log_response {
        my $pdu=shift;
 
 }
 
 sub log_response {
        my $pdu=shift;
 

-       print '-' x 80,"\n";
-       print "Response ASN 1:\n";
-       Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
-       print "Response Perl:\n";
+#      print '-' x 80,"\n";
+#      print "Response ASN 1:\n";
+#      Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
+#      print "Response Perl:\n";

        my $response = $LDAPResponse->decode($pdu);
        my $response = $LDAPResponse->decode($pdu);

-       print dump($response);

 
        if ( defined $response->{protocolOp}->{searchResEntry} ) {
                my $uid = $response->{protocolOp}->{searchResEntry}->{objectName};
 
        if ( defined $response->{protocolOp}->{searchResEntry} ) {
                my $uid = $response->{protocolOp}->{searchResEntry}->{objectName};

-               warn "## SEARCH $uid";
+               warn "## objectName $uid";
+
+               my @attrs;
+

                map {
                map {

-                       if ( $_->{type} eq 'postalAddress' ) {
-                               $_->{vals} = [ 'foobar' ];
+                       if ( $_->{type} eq 'hrEduPersonUniqueNumber' ) {
+                               foreach my $val ( @{ $_->{vals} } ) {
+                                       next if $val !~ m{.+:.+};
+                                       my ( $n, $v ) = split(/\s*:\s*/, $val );
+                                       push @attrs, { type => $_->{type} . '_' . $n, vals => [ $v ] };
+                               }

                        }
                } @{ $response->{protocolOp}->{searchResEntry}->{attributes} };
 
                        }
                } @{ $response->{protocolOp}->{searchResEntry}->{attributes} };
 

-               push @{ $response->{protocolOp}->{searchResEntry}->{attributes} },
-                       { type => 'ffzg-datum_rodjenja', vals => [ '2009-01-01' ], }
-               ;
+               warn "# ++ attrs ",dump( @attrs );
+
+               push @{ $response->{protocolOp}->{searchResEntry}->{attributes} }, $_ foreach @attrs;
+
+               my $path = $config->{yaml_dir} . "$uid.yaml";
+               if ( -e $path ) {
+                       my $data = LoadFile($path);
+                       warn "# yaml = ",dump($data);
+
+                       foreach my $type ( keys %$data ) {
+
+                               my $vals = $data->{$type};
+
+                               push @{ $response->{protocolOp}->{searchResEntry}->{attributes} }, {
+                                       type => $config->{overlay_prefix} . $type,
+                                       vals => ref($vals) eq 'ARRAY' ? $vals : [ $vals ],
+                               };
+                       }
+               }

 
                $pdu = $LDAPResponse->encode($response);
        }
 
 
                $pdu = $LDAPResponse->encode($response);
        }
 

+       warn "## response = ", dump($response);
+

        return $pdu;
 }
 
        return $pdu;
 }
 


@@ -116,15 +198,18 @@ my $listenersock = IO::Socket::INET->new(
        Listen => 5,
        Proto => 'tcp',
        Reuse => 1,
        Listen => 5,
        Proto => 'tcp',
        Reuse => 1,

-       LocalPort => 1389
-);
-
-
-my $targetsock = new IO::Socket::INET (
-       Proto => 'tcp',
-       PeerAddr => 'ldap.ffzg.hr',
-       PeerPort => 389,
-);
+       LocalAddr => $config->{listen},
+) || die "can't open listen socket: $!";
+
+
+my $targetsock = $config->{upstream_ssl}
+       ? IO::Socket::INET->new(
+               Proto => 'tcp',
+               PeerAddr => $config->{upstream_ldap},
+               PeerPort => 389,
+       )
+       : IO::Socket::SSL->new( $config->{upstream_ldap} . ':ldaps')
+       || die "can't open upstream socket: $!";

 
 run_proxy($listenersock,$targetsock);
 
 
 run_proxy($listenersock,$targetsock);