07abb53ca47f51c8fbd0cea4772254dd76e61df0
[ira/wip.git] / source4 / lib / ldb / modules / rdn_name.c
1 /* 
2    ldb database library
3
4    Copyright (C) Andrew Bartlett 2005
5    Copyright (C) Simo Sorce 2006-2008
6
7      ** NOTE! The following LGPL license applies to the ldb
8      ** library. This does NOT imply that all of Samba is released
9      ** under the LGPL
10    
11    This library is free software; you can redistribute it and/or
12    modify it under the terms of the GNU Lesser General Public
13    License as published by the Free Software Foundation; either
14    version 3 of the License, or (at your option) any later version.
15
16    This library is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19    Lesser General Public License for more details.
20
21    You should have received a copy of the GNU Lesser General Public
22    License along with this library; if not, see <http://www.gnu.org/licenses/>.
23 */
24
25 /*
26  *  Name: rdn_name
27  *
28  *  Component: ldb rdn name module
29  *
30  *  Description: keep a consistent name attribute on objects manpulations
31  *
32  *  Author: Andrew Bartlett
33  *
34  *  Modifications:
35  *    - made the module async
36  *      Simo Sorce Mar 2006
37  */
38
39 #include "ldb_includes.h"
40 #include "ldb_module.h"
41
42 struct rename_context {
43
44         struct ldb_module *module;
45         struct ldb_request *req;
46
47         struct ldb_reply *ares;
48 };
49
50 static struct ldb_message_element *rdn_name_find_attribute(const struct ldb_message *msg, const char *name)
51 {
52         int i;
53
54         for (i = 0; i < msg->num_elements; i++) {
55                 if (ldb_attr_cmp(name, msg->elements[i].name) == 0) {
56                         return &msg->elements[i];
57                 }
58         }
59
60         return NULL;
61 }
62
63 static int rdn_name_add_callback(struct ldb_request *req,
64                                  struct ldb_reply *ares)
65 {
66         struct rename_context *ac;
67
68         ac = talloc_get_type(req->context, struct rename_context);
69
70         if (!ares) {
71                 return ldb_module_done(ac->req, NULL, NULL,
72                                         LDB_ERR_OPERATIONS_ERROR);
73         }
74         if (ares->error != LDB_SUCCESS) {
75                 return ldb_module_done(ac->req, ares->controls,
76                                         ares->response, ares->error);
77         }
78
79         if (ares->type != LDB_REPLY_DONE) {
80                 return ldb_module_done(ac->req, NULL, NULL,
81                                         LDB_ERR_OPERATIONS_ERROR);
82         }
83
84         return ldb_module_done(ac->req, ares->controls,
85                                         ares->response, LDB_SUCCESS);
86 }
87
88 static int rdn_name_add(struct ldb_module *module, struct ldb_request *req)
89 {
90         struct ldb_context *ldb;
91         struct ldb_request *down_req;
92         struct rename_context *ac;
93         struct ldb_message *msg;
94         struct ldb_message_element *attribute;
95         const struct ldb_schema_attribute *a;
96         const char *rdn_name;
97         struct ldb_val rdn_val;
98         int i, ret;
99
100         ldb = ldb_module_get_ctx(module);
101         ldb_debug(ldb, LDB_DEBUG_TRACE, "rdn_name_add_record");
102
103         /* do not manipulate our control entries */
104         if (ldb_dn_is_special(req->op.add.message->dn)) {
105                 return ldb_next_request(module, req);
106         }
107
108         ac = talloc_zero(req, struct rename_context);
109         if (ac == NULL) {
110                 return LDB_ERR_OPERATIONS_ERROR;
111         }
112
113         ac->module = module;
114         ac->req = req;
115
116         msg = ldb_msg_copy_shallow(req, req->op.add.message);
117         if (msg == NULL) {
118                 return LDB_ERR_OPERATIONS_ERROR;
119         }
120
121         rdn_name = ldb_dn_get_rdn_name(msg->dn);
122         if (rdn_name == NULL) {
123                 talloc_free(ac);
124                 return LDB_ERR_OPERATIONS_ERROR;
125         }
126         
127         rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(msg->dn));
128         
129         /* Perhaps someone above us tried to set this? */
130         if ((attribute = rdn_name_find_attribute(msg, "name")) != NULL ) {
131                 attribute->num_values = 0;
132         }
133
134         if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
135                 talloc_free(ac);
136                 return LDB_ERR_OPERATIONS_ERROR;
137         }
138
139         attribute = rdn_name_find_attribute(msg, rdn_name);
140
141         if (!attribute) {
142                 if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
143                         talloc_free(ac);
144                         return LDB_ERR_OPERATIONS_ERROR;
145                 }
146         } else {
147                 a = ldb_schema_attribute_by_name(ldb, rdn_name);
148
149                 for (i = 0; i < attribute->num_values; i++) {
150                         ret = a->syntax->comparison_fn(ldb, msg,
151                                         &rdn_val, &attribute->values[i]);
152                         if (ret == 0) {
153                                 /* overwrite so it matches in case */
154                                 attribute->values[i] = rdn_val;
155                                 break;
156                         }
157                 }
158                 if (i == attribute->num_values) {
159                         char *rdn_errstring = talloc_asprintf(ac,
160                                 "RDN mismatch on %s: %s (%.*s) should match one of:", 
161                                 ldb_dn_get_linearized(msg->dn), rdn_name, 
162                                 (int)rdn_val.length, (const char *)rdn_val.data);
163                         for (i = 0; i < attribute->num_values; i++) {
164                                 rdn_errstring = talloc_asprintf_append(
165                                         rdn_errstring, " (%.*s)",
166                                         (int)attribute->values[i].length, 
167                                         (const char *)attribute->values[i].data);
168                         }
169                         ldb_set_errstring(ldb, rdn_errstring);
170                         talloc_free(ac);
171                         /* Match AD's error here */
172                         return LDB_ERR_INVALID_DN_SYNTAX;
173                 }
174         }
175
176         ret = ldb_build_add_req(&down_req, ldb, req,
177                                 msg,
178                                 req->controls,
179                                 ac, rdn_name_add_callback,
180                                 req);
181         if (ret != LDB_SUCCESS) {
182                 return ret;
183         }
184
185         talloc_steal(down_req, msg);
186
187         /* go on with the call chain */
188         return ldb_next_request(module, down_req);
189 }
190
191 static int rdn_modify_callback(struct ldb_request *req, struct ldb_reply *ares)
192 {
193         struct rename_context *ac;
194
195         ac = talloc_get_type(req->context, struct rename_context);
196
197         if (!ares) {
198                 return ldb_module_done(ac->req, NULL, NULL,
199                                         LDB_ERR_OPERATIONS_ERROR);
200         }
201         if (ares->error != LDB_SUCCESS) {
202                 return ldb_module_done(ac->req, ares->controls,
203                                         ares->response, ares->error);
204         }
205
206         /* the only supported reply right now is a LDB_REPLY_DONE */
207         if (ares->type != LDB_REPLY_DONE) {
208                 return ldb_module_done(ac->req, NULL, NULL,
209                                         LDB_ERR_OPERATIONS_ERROR);
210         }
211
212         /* send saved controls eventually */
213         return ldb_module_done(ac->req, ac->ares->controls,
214                                 ac->ares->response, LDB_SUCCESS);
215 }
216
217 static int rdn_rename_callback(struct ldb_request *req, struct ldb_reply *ares)
218 {
219         struct ldb_context *ldb;
220         struct rename_context *ac;
221         struct ldb_request *mod_req;
222         const char *rdn_name;
223         struct ldb_val rdn_val;
224         struct ldb_message *msg;
225         int ret;
226
227         ac = talloc_get_type(req->context, struct rename_context);
228         ldb = ldb_module_get_ctx(ac->module);
229
230         if (!ares) {
231                 goto error;
232         }
233         if (ares->error != LDB_SUCCESS) {
234                 return ldb_module_done(ac->req, ares->controls,
235                                         ares->response, ares->error);
236         }
237
238         /* the only supported reply right now is a LDB_REPLY_DONE */
239         if (ares->type != LDB_REPLY_DONE) {
240                 goto error;
241         }
242
243         /* save reply for caller */
244         ac->ares = talloc_steal(ac, ares);
245
246         msg = ldb_msg_new(ac);
247         if (msg == NULL) {
248                 goto error;
249         }
250         msg->dn = ldb_dn_copy(msg, ac->req->op.rename.newdn);
251         if (msg->dn == NULL) {
252                 goto error;
253         }
254         rdn_name = ldb_dn_get_rdn_name(ac->req->op.rename.newdn);
255         if (rdn_name == NULL) {
256                 goto error;
257         }
258         
259         rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(ac->req->op.rename.newdn));
260         
261         if (ldb_msg_add_empty(msg, rdn_name, LDB_FLAG_MOD_REPLACE, NULL) != 0) {
262                 goto error;
263         }
264         if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
265                 goto error;
266         }
267         if (ldb_msg_add_empty(msg, "name", LDB_FLAG_MOD_REPLACE, NULL) != 0) {
268                 goto error;
269         }
270         if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
271                 goto error;
272         }
273
274         ret = ldb_build_mod_req(&mod_req, ldb,
275                                 ac, msg, NULL,
276                                 ac, rdn_modify_callback,
277                                 req);
278         if (ret != LDB_SUCCESS) {
279                 return ldb_module_done(ac->req, NULL, NULL, ret);
280         }
281         talloc_steal(mod_req, msg);
282
283         /* do the mod call */
284         return ldb_request(ldb, mod_req);
285
286 error:
287         return ldb_module_done(ac->req, NULL, NULL,
288                                                 LDB_ERR_OPERATIONS_ERROR);
289 }
290
291 static int rdn_name_rename(struct ldb_module *module, struct ldb_request *req)
292 {
293         struct ldb_context *ldb;
294         struct rename_context *ac;
295         struct ldb_request *down_req;
296         int ret;
297
298         ldb = ldb_module_get_ctx(module);
299         ldb_debug(ldb, LDB_DEBUG_TRACE, "rdn_name_rename");
300
301         /* do not manipulate our control entries */
302         if (ldb_dn_is_special(req->op.rename.newdn)) {
303                 return ldb_next_request(module, req);
304         }
305
306         ac = talloc_zero(req, struct rename_context);
307         if (ac == NULL) {
308                 return LDB_ERR_OPERATIONS_ERROR;
309         }
310
311         ac->module = module;
312         ac->req = req;
313
314         ret = ldb_build_rename_req(&down_req,
315                                    ldb,
316                                    ac,
317                                    req->op.rename.olddn,
318                                    req->op.rename.newdn,
319                                    req->controls,
320                                    ac,
321                                    rdn_rename_callback,
322                                    req);
323
324         if (ret != LDB_SUCCESS) {
325                 return LDB_ERR_OPERATIONS_ERROR;
326         }
327
328         /* rename first, modify "name" if rename is ok */
329         return ldb_next_request(module, down_req);
330 }
331
332 const struct ldb_module_ops ldb_rdn_name_module_ops = {
333         .name              = "rdn_name",
334         .add               = rdn_name_add,
335         .rename            = rdn_name_rename,
336 };