65c044c0f4127c2ef0cf230854904adbc69660e2
[kai/samba.git] / source4 / lib / ldb / modules / rdn_name.c
1 /* 
2    ldb database library
3
4    Copyright (C) Andrew Bartlet 2005
5    Copyright (C) Simo Sorce     2006
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: rdb_name
27  *
28  *  Component: ldb rdn name module
29  *
30  *  Description: keep a consistent name attribute on objects manpulations
31  *
32  *  Author: Andrew Bartlet
33  *
34  *  Modifications:
35  *    - made the module async
36  *      Simo Sorce Mar 2006
37  */
38
39 #include "ldb_includes.h"
40
41 static struct ldb_message_element *rdn_name_find_attribute(const struct ldb_message *msg, const char *name)
42 {
43         int i;
44
45         for (i = 0; i < msg->num_elements; i++) {
46                 if (ldb_attr_cmp(name, msg->elements[i].name) == 0) {
47                         return &msg->elements[i];
48                 }
49         }
50
51         return NULL;
52 }
53
54 static int rdn_name_add(struct ldb_module *module, struct ldb_request *req)
55 {
56         struct ldb_request *down_req;
57         struct ldb_message *msg;
58         struct ldb_message_element *attribute;
59         const char *rdn_name;
60         struct ldb_val rdn_val;
61         int i, ret;
62
63         ldb_debug(module->ldb, LDB_DEBUG_TRACE, "rdn_name_add_record\n");
64
65         /* do not manipulate our control entries */
66         if (ldb_dn_is_special(req->op.add.message->dn)) {
67                 return ldb_next_request(module, req);
68         }
69
70         down_req = talloc(req, struct ldb_request);
71         if (down_req == NULL) {
72                 return LDB_ERR_OPERATIONS_ERROR;
73         }
74
75         *down_req = *req;
76
77         down_req->op.add.message = msg = ldb_msg_copy_shallow(down_req, req->op.add.message);
78         if (msg == NULL) {
79                 return LDB_ERR_OPERATIONS_ERROR;
80         }
81
82         rdn_name = ldb_dn_get_rdn_name(msg->dn);
83         if (rdn_name == NULL) {
84                 talloc_free(down_req);
85                 return LDB_ERR_OPERATIONS_ERROR;
86         }
87         
88         rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(msg->dn));
89         
90         /* Perhaps someone above us tried to set this? */
91         if ((attribute = rdn_name_find_attribute(msg, "name")) != NULL ) {
92                 attribute->num_values = 0;
93         }
94
95         if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
96                 talloc_free(down_req);
97                 return LDB_ERR_OPERATIONS_ERROR;
98         }
99
100         attribute = rdn_name_find_attribute(msg, rdn_name);
101
102         if (!attribute) {
103                 if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
104                         talloc_free(down_req);
105                         return LDB_ERR_OPERATIONS_ERROR;
106                 }
107         } else {
108                 const struct ldb_schema_attribute *a = ldb_schema_attribute_by_name(module->ldb, rdn_name);
109
110                 for (i = 0; i < attribute->num_values; i++) {
111                         if (a->syntax->comparison_fn(module->ldb, msg, &rdn_val, &attribute->values[i]) == 0) {
112                                 /* overwrite so it matches in case */
113                                 attribute->values[i] = rdn_val;
114                                 break;
115                         }
116                 }
117                 if (i == attribute->num_values) {
118                         ldb_debug_set(module->ldb, LDB_DEBUG_FATAL, 
119                                       "RDN mismatch on %s: %s (%s)", 
120                                       ldb_dn_get_linearized(msg->dn), rdn_name, rdn_val.data);
121                         talloc_free(down_req);
122                         /* Match AD's error here */
123                         return LDB_ERR_INVALID_DN_SYNTAX;
124                 }
125         }
126
127         /* go on with the call chain */
128         ret = ldb_next_request(module, down_req);
129
130         /* do not free down_req as the call results may be linked to it,
131          * it will be freed when the upper level request get freed */
132         if (ret == LDB_SUCCESS) {
133                 req->handle = down_req->handle;
134         }
135
136         return ret;
137 }
138
139 struct rename_context {
140
141         enum {RENAME_RENAME, RENAME_MODIFY} step;
142         struct ldb_request *orig_req;
143         struct ldb_request *down_req;
144         struct ldb_request *mod_req;
145 };
146
147 static int rdn_name_rename(struct ldb_module *module, struct ldb_request *req)
148 {
149         struct ldb_handle *h;
150         struct rename_context *ac;
151
152         ldb_debug(module->ldb, LDB_DEBUG_TRACE, "rdn_name_rename\n");
153
154         /* do not manipulate our control entries */
155         if (ldb_dn_is_special(req->op.rename.newdn)) {
156                 return ldb_next_request(module, req);
157         }
158
159         h = talloc_zero(req, struct ldb_handle);
160         if (h == NULL) {
161                 return LDB_ERR_OPERATIONS_ERROR;
162         }
163
164         h->module = module;
165
166         ac = talloc_zero(h, struct rename_context);
167         if (ac == NULL) {
168                 return LDB_ERR_OPERATIONS_ERROR;
169         }
170
171         h->private_data = (void *)ac;
172
173         h->state = LDB_ASYNC_INIT;
174         h->status = LDB_SUCCESS;
175
176         ac->orig_req = req;
177         ac->down_req = talloc(req, struct ldb_request);
178         if (ac->down_req == NULL) {
179                 return LDB_ERR_OPERATIONS_ERROR;
180         }
181
182         *(ac->down_req) = *req;
183
184         ac->step = RENAME_RENAME;
185
186         req->handle = h;
187
188         /* rename first, modify "name" if rename is ok */
189         return ldb_next_request(module, ac->down_req);
190 }
191
192 static int rdn_name_rename_do_mod(struct ldb_handle *h) {
193
194         struct rename_context *ac;
195         const char *rdn_name;
196         struct ldb_val rdn_val;
197         struct ldb_message *msg;
198
199         ac = talloc_get_type(h->private_data, struct rename_context);
200
201         ac->mod_req = talloc_zero(ac, struct ldb_request);
202
203         ac->mod_req->operation = LDB_MODIFY;
204         ac->mod_req->op.mod.message = msg = ldb_msg_new(ac->mod_req);
205         if (msg == NULL) {
206                 return LDB_ERR_OPERATIONS_ERROR;
207         }
208
209         msg->dn = ldb_dn_copy(msg, ac->orig_req->op.rename.newdn);
210         if (msg->dn == NULL) {
211                 return LDB_ERR_OPERATIONS_ERROR;
212         }
213
214         rdn_name = ldb_dn_get_rdn_name(ac->orig_req->op.rename.newdn);
215         if (rdn_name == NULL) {
216                 return LDB_ERR_OPERATIONS_ERROR;
217         }
218         
219         rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(ac->orig_req->op.rename.newdn));
220         
221         if (ldb_msg_add_empty(msg, rdn_name, LDB_FLAG_MOD_REPLACE, NULL) != 0) {
222                 return LDB_ERR_OPERATIONS_ERROR;
223         }
224         if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
225                 return LDB_ERR_OPERATIONS_ERROR;
226         }
227         if (ldb_msg_add_empty(msg, "name", LDB_FLAG_MOD_REPLACE, NULL) != 0) {
228                 return LDB_ERR_OPERATIONS_ERROR;
229         }
230         if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
231                 return LDB_ERR_OPERATIONS_ERROR;
232         }
233
234         ldb_set_timeout_from_prev_req(h->module->ldb, ac->orig_req, ac->mod_req);
235
236         ac->step = RENAME_MODIFY;
237
238         /* do the mod call */
239         return ldb_request(h->module->ldb, ac->mod_req);
240 }
241
242 static int rdn_name_wait_once(struct ldb_handle *handle)
243 {
244         struct rename_context *ac;
245         int ret;
246     
247         if (!handle || !handle->private_data) {
248                 return LDB_ERR_OPERATIONS_ERROR;
249         }
250
251         if (handle->state == LDB_ASYNC_DONE) {
252                 return handle->status;
253         }
254
255         handle->state = LDB_ASYNC_PENDING;
256         handle->status = LDB_SUCCESS;
257
258         ac = talloc_get_type(handle->private_data, struct rename_context);
259
260         switch(ac->step) {
261         case RENAME_RENAME:
262                 ret = ldb_wait(ac->down_req->handle, LDB_WAIT_NONE);
263                 if (ret != LDB_SUCCESS) {
264                         handle->status = ret;
265                         goto done;
266                 }
267                 if (ac->down_req->handle->status != LDB_SUCCESS) {
268                         handle->status = ac->down_req->handle->status;
269                         goto done;
270                 }
271
272                 if (ac->down_req->handle->state != LDB_ASYNC_DONE) {
273                         return LDB_SUCCESS;
274                 }
275
276                 /* rename operation done */
277                 return rdn_name_rename_do_mod(handle);
278
279         case RENAME_MODIFY:
280                 ret = ldb_wait(ac->mod_req->handle, LDB_WAIT_NONE);
281                 if (ret != LDB_SUCCESS) {
282                         handle->status = ret;
283                         goto done;
284                 }
285                 if (ac->mod_req->handle->status != LDB_SUCCESS) {
286                         handle->status = ac->mod_req->handle->status;
287                         goto done;
288                 }
289
290                 if (ac->mod_req->handle->state != LDB_ASYNC_DONE) {
291                         return LDB_SUCCESS;
292                 }
293
294                 break;
295
296         default:
297                 ret = LDB_ERR_OPERATIONS_ERROR;
298                 goto done;
299         }
300
301         ret = LDB_SUCCESS;
302
303 done:
304         handle->state = LDB_ASYNC_DONE;
305         return ret;
306 }
307
308 static int rdn_name_wait(struct ldb_handle *handle, enum ldb_wait_type type)
309 {
310         int ret;
311  
312         if (!handle || !handle->private_data) {
313                 return LDB_ERR_OPERATIONS_ERROR;
314         }
315
316         if (type == LDB_WAIT_ALL) {
317                 while (handle->state != LDB_ASYNC_DONE) {
318                         ret = rdn_name_wait_once(handle);
319                         if (ret != LDB_SUCCESS) {
320                                 return ret;
321                         }
322                 }
323
324                 return handle->status;
325         }
326
327         return rdn_name_wait_once(handle);
328 }
329
330 const struct ldb_module_ops ldb_rdn_name_module_ops = {
331         .name              = "rdn_name",
332         .add               = rdn_name_add,
333         .rename            = rdn_name_rename,
334         .wait              = rdn_name_wait
335 };