r23784: use the GPLv3 boilerplate as recommended by the FSF and the license text
[tprouty/samba.git] / source / lib / smbldap_util.c
1 /* 
2    Unix SMB/CIFS mplementation.
3    LDAP protocol helper functions for SAMBA
4    Copyright (C) Jean François Micouleau        1998
5    Copyright (C) Gerald Carter                  2001-2003
6    Copyright (C) Shahms King                    2001
7    Copyright (C) Andrew Bartlett                2002-2003
8    Copyright (C) Stefan (metze) Metzmacher      2002-2003
9     
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14    
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19    
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22    
23 */
24
25 #include "includes.h"
26 #include "smbldap.h"
27
28 /**********************************************************************
29  Add the account-policies below the sambaDomain object to LDAP, 
30 *********************************************************************/
31
32 static NTSTATUS add_new_domain_account_policies(struct smbldap_state *ldap_state,
33                                                 const char *domain_name)
34 {
35         NTSTATUS ntstatus = NT_STATUS_UNSUCCESSFUL;
36         int i, rc;
37         uint32 policy_default;
38         const char *policy_attr = NULL;
39         pstring dn;
40         LDAPMod **mods = NULL;
41         char *escape_domain_name;
42
43         DEBUG(3,("add_new_domain_account_policies: Adding new account policies for domain\n"));
44
45         escape_domain_name = escape_rdn_val_string_alloc(domain_name);
46         if (!escape_domain_name) {
47                 DEBUG(0, ("Out of memory!\n"));
48                 return NT_STATUS_NO_MEMORY;
49         }
50
51         pstr_sprintf(dn, "%s=%s,%s", 
52                 get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
53                 escape_domain_name, lp_ldap_suffix());
54
55         SAFE_FREE(escape_domain_name);
56
57         for (i=1; decode_account_policy_name(i) != NULL; i++) {
58
59                 pstring val;
60
61                 policy_attr = get_account_policy_attr(i);
62                 if (!policy_attr) {
63                         DEBUG(0,("add_new_domain_account_policies: ops. no policy!\n"));
64                         continue;
65                 }
66
67                 if (!account_policy_get_default(i, &policy_default)) {
68                         DEBUG(0,("add_new_domain_account_policies: failed to get default account policy\n"));
69                         return ntstatus;
70                 }
71
72                 DEBUG(10,("add_new_domain_account_policies: adding \"%s\" with value: %d\n", policy_attr, policy_default));
73
74                 pstr_sprintf(val, "%d", policy_default); 
75
76                 smbldap_set_mod( &mods, LDAP_MOD_REPLACE, policy_attr, val);
77
78                 rc = smbldap_modify(ldap_state, dn, mods);
79
80                 if (rc!=LDAP_SUCCESS) {
81                         char *ld_error = NULL;
82                         ldap_get_option(ldap_state->ldap_struct, LDAP_OPT_ERROR_STRING, &ld_error);
83                         DEBUG(1,("add_new_domain_account_policies: failed to add account policies to dn= %s with: %s\n\t%s\n",
84                                 dn, ldap_err2string(rc),
85                                 ld_error ? ld_error : "unknown"));
86                         SAFE_FREE(ld_error);
87                         ldap_mods_free(mods, True);
88                         return ntstatus;
89                 }
90         }
91
92         ldap_mods_free(mods, True);
93
94         return NT_STATUS_OK;
95 }
96
97 /**********************************************************************
98  Add the sambaDomain to LDAP, so we don't have to search for this stuff
99  again.  This is a once-add operation for now.
100
101  TODO:  Add other attributes, and allow modification.
102 *********************************************************************/
103
104 static NTSTATUS add_new_domain_info(struct smbldap_state *ldap_state, 
105                                     const char *domain_name) 
106 {
107         fstring sid_string;
108         fstring algorithmic_rid_base_string;
109         pstring filter, dn;
110         LDAPMod **mods = NULL;
111         int rc;
112         LDAPMessage *result = NULL;
113         int num_result;
114         const char **attr_list;
115         char *escape_domain_name;
116
117         /* escape for filter */
118         escape_domain_name = escape_ldap_string_alloc(domain_name);
119         if (!escape_domain_name) {
120                 DEBUG(0, ("Out of memory!\n"));
121                 return NT_STATUS_NO_MEMORY;
122         }
123
124         slprintf (filter, sizeof (filter) - 1, "(&(%s=%s)(objectclass=%s))", 
125                   get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN), 
126                   escape_domain_name, LDAP_OBJ_DOMINFO);
127
128         SAFE_FREE(escape_domain_name);
129
130         attr_list = get_attr_list( NULL, dominfo_attr_list );
131         rc = smbldap_search_suffix(ldap_state, filter, attr_list, &result);
132         TALLOC_FREE( attr_list );
133
134         if (rc != LDAP_SUCCESS) {
135                 return NT_STATUS_UNSUCCESSFUL;
136         }
137
138         num_result = ldap_count_entries(ldap_state->ldap_struct, result);
139         
140         if (num_result > 1) {
141                 DEBUG (0, ("add_new_domain_info: More than domain with that name exists: bailing "
142                            "out!\n"));
143                 ldap_msgfree(result);
144                 return NT_STATUS_UNSUCCESSFUL;
145         }
146         
147         /* Check if we need to add an entry */
148         DEBUG(3,("add_new_domain_info: Adding new domain\n"));
149
150         /* this time escape for DN */
151         escape_domain_name = escape_rdn_val_string_alloc(domain_name);
152         if (!escape_domain_name) {
153                 DEBUG(0, ("Out of memory!\n"));
154                 return NT_STATUS_NO_MEMORY;
155         }
156
157         pstr_sprintf(dn, "%s=%s,%s",
158                      get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN),
159                      escape_domain_name, lp_ldap_suffix());
160
161         SAFE_FREE(escape_domain_name);
162
163         /* Free original search */
164         ldap_msgfree(result);
165
166         /* make the changes - the entry *must* not already have samba
167          * attributes */
168
169         smbldap_set_mod(&mods, LDAP_MOD_ADD,
170                         get_attr_key2string(dominfo_attr_list,
171                                             LDAP_ATTR_DOMAIN), 
172                         domain_name);
173
174         /* If we don't have an entry, then ask secrets.tdb for what it thinks.
175            It may choose to make it up */
176
177         sid_to_string(sid_string, get_global_sam_sid());
178         smbldap_set_mod(&mods, LDAP_MOD_ADD,
179                         get_attr_key2string(dominfo_attr_list,
180                                             LDAP_ATTR_DOM_SID),
181                         sid_string);
182
183         slprintf(algorithmic_rid_base_string,
184                  sizeof(algorithmic_rid_base_string) - 1, "%i",
185                  algorithmic_rid_base());
186         smbldap_set_mod(&mods, LDAP_MOD_ADD,
187                         get_attr_key2string(dominfo_attr_list,
188                                             LDAP_ATTR_ALGORITHMIC_RID_BASE), 
189                         algorithmic_rid_base_string);
190         smbldap_set_mod(&mods, LDAP_MOD_ADD, "objectclass", LDAP_OBJ_DOMINFO);
191         
192         /* add the sambaNextUserRid attributes. */
193         
194         {
195                 uint32 rid = BASE_RID;
196                 fstring rid_str;
197                 
198                 fstr_sprintf( rid_str, "%i", rid );
199                 DEBUG(10,("add_new_domain_info: setting next available user rid [%s]\n", rid_str));
200                 smbldap_set_mod(&mods, LDAP_MOD_ADD, 
201                         get_attr_key2string(dominfo_attr_list,
202                                             LDAP_ATTR_NEXT_USERRID), 
203                         rid_str);
204         }
205
206
207         rc = smbldap_add(ldap_state, dn, mods);
208
209         if (rc!=LDAP_SUCCESS) {
210                 char *ld_error = NULL;
211                 ldap_get_option(ldap_state->ldap_struct,
212                                 LDAP_OPT_ERROR_STRING, &ld_error);
213                 DEBUG(1,("add_new_domain_info: failed to add domain dn= %s with: %s\n\t%s\n",
214                          dn, ldap_err2string(rc),
215                          ld_error?ld_error:"unknown"));
216                 SAFE_FREE(ld_error);
217
218                 ldap_mods_free(mods, True);
219                 return NT_STATUS_UNSUCCESSFUL;
220         }
221
222         DEBUG(2,("add_new_domain_info: added: domain = %s in the LDAP database\n", domain_name));
223         ldap_mods_free(mods, True);
224         return NT_STATUS_OK;
225 }
226
227 /**********************************************************************
228 Search for the domain info entry
229 *********************************************************************/
230
231 NTSTATUS smbldap_search_domain_info(struct smbldap_state *ldap_state,
232                                     LDAPMessage ** result, const char *domain_name,
233                                     BOOL try_add)
234 {
235         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
236         pstring filter;
237         int rc;
238         const char **attr_list;
239         int count;
240         char *escape_domain_name;
241         
242         escape_domain_name = escape_ldap_string_alloc(domain_name);
243         if (!escape_domain_name) {
244                 DEBUG(0, ("Out of memory!\n"));
245                 return NT_STATUS_NO_MEMORY;
246         }
247
248         pstr_sprintf(filter, "(&(objectClass=%s)(%s=%s))",
249                 LDAP_OBJ_DOMINFO,
250                 get_attr_key2string(dominfo_attr_list, LDAP_ATTR_DOMAIN), 
251                 escape_domain_name);
252
253         SAFE_FREE(escape_domain_name);
254
255         DEBUG(2, ("smbldap_search_domain_info: Searching for:[%s]\n", filter));
256
257         attr_list = get_attr_list( NULL, dominfo_attr_list );
258         rc = smbldap_search_suffix(ldap_state, filter, attr_list , result);
259         TALLOC_FREE( attr_list );
260
261         if (rc != LDAP_SUCCESS) {
262                 DEBUG(2,("smbldap_search_domain_info: Problem during LDAPsearch: %s\n", ldap_err2string (rc)));
263                 DEBUG(2,("smbldap_search_domain_info: Query was: %s, %s\n", lp_ldap_suffix(), filter));
264                 goto failed;
265         }
266
267         count = ldap_count_entries(ldap_state->ldap_struct, *result);
268
269         if (count == 1)
270                 return NT_STATUS_OK;
271
272         ldap_msgfree(*result);
273         *result = NULL;
274         
275         if (count < 1) {
276
277                 DEBUG(3, ("smbldap_search_domain_info: Got no domain info entries for domain\n"));
278
279                 if (!try_add)
280                         goto failed;
281
282                 status = add_new_domain_info(ldap_state, domain_name);
283                 if (!NT_STATUS_IS_OK(status)) {
284                         DEBUG(0, ("smbldap_search_domain_info: Adding domain info for %s failed with %s\n", 
285                                 domain_name, nt_errstr(status)));
286                         goto failed;
287                 }
288                         
289                 status = add_new_domain_account_policies(ldap_state, domain_name);
290                 if (!NT_STATUS_IS_OK(status)) {
291                         DEBUG(0, ("smbldap_search_domain_info: Adding domain account policies for %s failed with %s\n", 
292                                 domain_name, nt_errstr(status)));
293                         goto failed;
294                 }
295
296                 return smbldap_search_domain_info(ldap_state, result, domain_name, False);
297                 
298         } 
299         
300         if (count > 1 ) {
301         
302                 DEBUG(0, ("smbldap_search_domain_info: Got too many (%d) domain info entries for domain %s\n",
303                           count, domain_name));
304                 goto failed;
305         }
306
307 failed:
308         return status;
309 }