| @@ -1,130 +1,131 | |||
|
|
1 | 1 | # Redmine - project management software |
|
|
2 | 2 | # Copyright (C) 2006-2011 Jean-Philippe Lang |
|
|
3 | 3 | # |
|
|
4 | 4 | # This program is free software; you can redistribute it and/or |
|
|
5 | 5 | # modify it under the terms of the GNU General Public License |
|
|
6 | 6 | # as published by the Free Software Foundation; either version 2 |
|
|
7 | 7 | # of the License, or (at your option) any later version. |
|
|
8 | 8 | # |
|
|
9 | 9 | # This program is distributed in the hope that it will be useful, |
|
|
10 | 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
11 | 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
12 | 12 | # GNU General Public License for more details. |
|
|
13 | 13 | # |
|
|
14 | 14 | # You should have received a copy of the GNU General Public License |
|
|
15 | 15 | # along with this program; if not, write to the Free Software |
|
|
16 | 16 | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
|
|
17 | 17 | |
|
|
18 | 18 | require 'iconv' |
|
|
19 | require 'net/ldap' | |
|
|
19 | 20 | |
|
|
20 | 21 | class AuthSourceLdap < AuthSource |
|
|
21 | 22 | validates_presence_of :host, :port, :attr_login |
|
|
22 | 23 | validates_length_of :name, :host, :maximum => 60, :allow_nil => true |
|
|
23 | 24 | validates_length_of :account, :account_password, :base_dn, :maximum => 255, :allow_nil => true |
|
|
24 | 25 | validates_length_of :attr_login, :attr_firstname, :attr_lastname, :attr_mail, :maximum => 30, :allow_nil => true |
|
|
25 | 26 | validates_numericality_of :port, :only_integer => true |
|
|
26 | 27 | |
|
|
27 | 28 | before_validation :strip_ldap_attributes |
|
|
28 | 29 | |
|
|
29 | 30 | def initialize(attributes=nil, *args) |
|
|
30 | 31 | super |
|
|
31 | 32 | self.port = 389 if self.port == 0 |
|
|
32 | 33 | end |
|
|
33 | 34 | |
|
|
34 | 35 | def authenticate(login, password) |
|
|
35 | 36 | return nil if login.blank? || password.blank? |
|
|
36 | 37 | attrs = get_user_dn(login) |
|
|
37 | 38 | |
|
|
38 | 39 | if attrs && attrs[:dn] && authenticate_dn(attrs[:dn], password) |
|
|
39 | 40 | logger.debug "Authentication successful for '#{login}'" if logger && logger.debug? |
|
|
40 | 41 | return attrs.except(:dn) |
|
|
41 | 42 | end |
|
|
42 | 43 | rescue Net::LDAP::LdapError => text |
|
|
43 | 44 | raise "LdapError: " + text |
|
|
44 | 45 | end |
|
|
45 | 46 | |
|
|
46 | 47 | # test the connection to the LDAP |
|
|
47 | 48 | def test_connection |
|
|
48 | 49 | ldap_con = initialize_ldap_con(self.account, self.account_password) |
|
|
49 | 50 | ldap_con.open { } |
|
|
50 | 51 | rescue Net::LDAP::LdapError => text |
|
|
51 | 52 | raise "LdapError: " + text |
|
|
52 | 53 | end |
|
|
53 | 54 | |
|
|
54 | 55 | def auth_method_name |
|
|
55 | 56 | "LDAP" |
|
|
56 | 57 | end |
|
|
57 | 58 | |
|
|
58 | 59 | private |
|
|
59 | 60 | |
|
|
60 | 61 | def strip_ldap_attributes |
|
|
61 | 62 | [:attr_login, :attr_firstname, :attr_lastname, :attr_mail].each do |attr| |
|
|
62 | 63 | write_attribute(attr, read_attribute(attr).strip) unless read_attribute(attr).nil? |
|
|
63 | 64 | end |
|
|
64 | 65 | end |
|
|
65 | 66 | |
|
|
66 | 67 | def initialize_ldap_con(ldap_user, ldap_password) |
|
|
67 | 68 | options = { :host => self.host, |
|
|
68 | 69 | :port => self.port, |
|
|
69 | 70 | :encryption => (self.tls ? :simple_tls : nil) |
|
|
70 | 71 | } |
|
|
71 | 72 | options.merge!(:auth => { :method => :simple, :username => ldap_user, :password => ldap_password }) unless ldap_user.blank? && ldap_password.blank? |
|
|
72 | 73 | Net::LDAP.new options |
|
|
73 | 74 | end |
|
|
74 | 75 | |
|
|
75 | 76 | def get_user_attributes_from_ldap_entry(entry) |
|
|
76 | 77 | { |
|
|
77 | 78 | :dn => entry.dn, |
|
|
78 | 79 | :firstname => AuthSourceLdap.get_attr(entry, self.attr_firstname), |
|
|
79 | 80 | :lastname => AuthSourceLdap.get_attr(entry, self.attr_lastname), |
|
|
80 | 81 | :mail => AuthSourceLdap.get_attr(entry, self.attr_mail), |
|
|
81 | 82 | :auth_source_id => self.id |
|
|
82 | 83 | } |
|
|
83 | 84 | end |
|
|
84 | 85 | |
|
|
85 | 86 | # Return the attributes needed for the LDAP search. It will only |
|
|
86 | 87 | # include the user attributes if on-the-fly registration is enabled |
|
|
87 | 88 | def search_attributes |
|
|
88 | 89 | if onthefly_register? |
|
|
89 | 90 | ['dn', self.attr_firstname, self.attr_lastname, self.attr_mail] |
|
|
90 | 91 | else |
|
|
91 | 92 | ['dn'] |
|
|
92 | 93 | end |
|
|
93 | 94 | end |
|
|
94 | 95 | |
|
|
95 | 96 | # Check if a DN (user record) authenticates with the password |
|
|
96 | 97 | def authenticate_dn(dn, password) |
|
|
97 | 98 | if dn.present? && password.present? |
|
|
98 | 99 | initialize_ldap_con(dn, password).bind |
|
|
99 | 100 | end |
|
|
100 | 101 | end |
|
|
101 | 102 | |
|
|
102 | 103 | # Get the user's dn and any attributes for them, given their login |
|
|
103 | 104 | def get_user_dn(login) |
|
|
104 | 105 | ldap_con = initialize_ldap_con(self.account, self.account_password) |
|
|
105 | 106 | login_filter = Net::LDAP::Filter.eq( self.attr_login, login ) |
|
|
106 | 107 | object_filter = Net::LDAP::Filter.eq( "objectClass", "*" ) |
|
|
107 | 108 | attrs = {} |
|
|
108 | 109 | |
|
|
109 | 110 | ldap_con.search( :base => self.base_dn, |
|
|
110 | 111 | :filter => object_filter & login_filter, |
|
|
111 | 112 | :attributes=> search_attributes) do |entry| |
|
|
112 | 113 | |
|
|
113 | 114 | if onthefly_register? |
|
|
114 | 115 | attrs = get_user_attributes_from_ldap_entry(entry) |
|
|
115 | 116 | else |
|
|
116 | 117 | attrs = {:dn => entry.dn} |
|
|
117 | 118 | end |
|
|
118 | 119 | |
|
|
119 | 120 | logger.debug "DN found for #{login}: #{attrs[:dn]}" if logger && logger.debug? |
|
|
120 | 121 | end |
|
|
121 | 122 | |
|
|
122 | 123 | attrs |
|
|
123 | 124 | end |
|
|
124 | 125 | |
|
|
125 | 126 | def self.get_attr(entry, attr_name) |
|
|
126 | 127 | if !attr_name.blank? |
|
|
127 | 128 | entry[attr_name].is_a?(Array) ? entry[attr_name].first : entry[attr_name] |
|
|
128 | 129 | end |
|
|
129 | 130 | end |
|
|
130 | 131 | end |
General Comments 0
You need to be logged in to leave comments.
Login now