/[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

Contents of /bin/ldap-rewrite.pl

Parent Directory Parent Directory | Revision Log Revision Log


Revision 24 - (show annotations)
Mon Mar 16 13:34:32 2009 UTC (10 years, 2 months ago) by dpavlin
File MIME type: text/plain
File size: 3991 byte(s)
we need to unset LANG if we don't want to corrupt encoding

1 #!/usr/bin/perl
2 # Copyright (c) 2006 Hans Klunder <hans.klunder@bigfoot.com>. All rights reserved.
3 # This program is free software; you can redistribute it and/or
4 # modify it under the same terms as Perl itself.
5
6
7 use strict;
8 use warnings;
9
10 use IO::Select;
11 use IO::Socket;
12 use IO::Socket::SSL;
13 use warnings;
14 use Data::Dump qw/dump/;
15 use Convert::ASN1 qw(asn_read);
16 use Net::LDAP::ASN qw(LDAPRequest LDAPResponse);
17 our $VERSION = '0.2';
18 use fields qw(socket target);
19 use YAML qw/LoadFile/;
20
21 my $config = {
22 yaml_dir => './yaml/',
23 listen => 'localhost:1389',
24 upstream_ldap => 'ldap.ffzg.hr',
25 upstream_ssl => 1,
26 overlay_prefix => 'ffzg-',
27
28 };
29
30 if ( ! -d $config->{yaml_dir} ) {
31 warn "DISABLE ", $config->{yaml_dir}," data overlay";
32 }
33
34 warn "# config = ",dump( $config );
35
36 sub handle {
37 my $clientsocket=shift;
38 my $serversocket=shift;
39
40 # read from client
41 asn_read($clientsocket, my $reqpdu);
42 log_request($reqpdu);
43
44 return 1 unless $reqpdu;
45
46 # send to server
47 print $serversocket $reqpdu or die "Could not send PDU to server\n ";
48
49 # read from server
50 my $ready;
51 my $sel = IO::Select->new($serversocket);
52 for( $ready = 1 ; $ready ; $ready = $sel->can_read(0)) {
53 asn_read($serversocket, my $respdu) or return 1;
54 $respdu = log_response($respdu);
55 # and send the result to the client
56 print $clientsocket $respdu;
57 }
58
59 return 0;
60 }
61
62
63 sub log_request {
64 my $pdu=shift;
65
66 print '-' x 80,"\n";
67 print "Request ASN 1:\n";
68 Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
69 print "Request Perl:\n";
70 my $request = $LDAPRequest->decode($pdu);
71 print dump($request);
72 }
73
74 sub log_response {
75 my $pdu=shift;
76
77 print '-' x 80,"\n";
78 print "Response ASN 1:\n";
79 Convert::ASN1::asn_hexdump(\*STDOUT,$pdu);
80 print "Response Perl:\n";
81 my $response = $LDAPResponse->decode($pdu);
82
83 if ( defined $response->{protocolOp}->{searchResEntry} ) {
84 my $uid = $response->{protocolOp}->{searchResEntry}->{objectName};
85 warn "## SEARCH $uid";
86
87 my @attrs;
88
89 map {
90 if ( $_->{type} eq 'hrEduPersonUniqueNumber' ) {
91 foreach my $val ( @{ $_->{vals} } ) {
92 next if $val !~ m{.+:.+};
93 my ( $n, $v ) = split(/\s*:\s*/, $val );
94 push @attrs, { type => $_->{type} . '_' . $n, vals => [ $v ] };
95 }
96 }
97 } @{ $response->{protocolOp}->{searchResEntry}->{attributes} };
98
99 warn "# ++ attrs ",dump( @attrs );
100
101 push @{ $response->{protocolOp}->{searchResEntry}->{attributes} }, $_ foreach @attrs;
102
103 my $path = $config->{yaml_dir} . "$uid.yaml";
104 if ( -e $path ) {
105 my $data = LoadFile($path);
106 warn "# yaml = ",dump($data);
107
108 foreach my $type ( keys %$data ) {
109
110 my $vals = $data->{$type};
111
112 push @{ $response->{protocolOp}->{searchResEntry}->{attributes} }, {
113 type => $config->{overlay_prefix} . $type,
114 vals => ref($vals) eq 'ARRAY' ? $vals : [ $vals ],
115 };
116 }
117 }
118
119 $pdu = $LDAPResponse->encode($response);
120 }
121
122 print dump($response);
123
124 return $pdu;
125 }
126
127 sub run_proxy {
128 my $listenersock = shift;
129 my $targetsock=shift;
130
131 die "Could not create listener socket: $!\n" unless $listenersock;
132 die "Could not create connection to server: $!\n" unless $targetsock;
133
134 my $sel = IO::Select->new($listenersock);
135 my %Handlers;
136 while (my @ready = $sel->can_read) {
137 foreach my $fh (@ready) {
138 if ($fh == $listenersock) {
139 # let's create a new socket
140 my $psock = $listenersock->accept;
141 $sel->add($psock);
142 } else {
143 my $result = handle($fh,$targetsock);
144 if ($result) {
145 # we have finished with the socket
146 $sel->remove($fh);
147 $fh->close;
148 delete $Handlers{*$fh};
149 }
150 }
151 }
152 }
153 }
154
155
156 $ENV{LANG} = 'C'; # so we don't double-encode utf-8 if LANG is utf-8
157
158 my $listenersock = IO::Socket::INET->new(
159 Listen => 5,
160 Proto => 'tcp',
161 Reuse => 1,
162 LocalAddr => $config->{listen},
163 );
164
165
166 my $targetsock = $config->{upstream_ssl}
167 ? IO::Socket::INET->new(
168 Proto => 'tcp',
169 PeerAddr => $config->{upstream_ldap},
170 PeerPort => 389,
171 )
172 : IO::Socket::SSL->new( $config->{upstream_ldap} . ':ldaps')
173 ;
174
175 run_proxy($listenersock,$targetsock);
176
177 1;

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26