4 Copyright (C) Simo Sorce 2004-2006
5 Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
7 ** NOTE! The following LGPL license applies to the ldb
8 ** library. This does NOT imply that all of Samba is released
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 2 of the License, or (at your option) any later version.
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.
21 You should have received a copy of the GNU Lesser General Public
22 License along with this library; if not, write to the Free Software
23 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
29 * Component: ldb objectguid module
31 * Description: add a unique objectGUID onto every new record
37 #include "ldb/include/includes.h"
38 #include "librpc/gen_ndr/ndr_misc.h"
40 static struct ldb_message_element *objectguid_find_attribute(const struct ldb_message *msg, const char *name)
44 for (i = 0; i < msg->num_elements; i++) {
45 if (ldb_attr_cmp(name, msg->elements[i].name) == 0) {
46 return &msg->elements[i];
53 /* add_record: add objectGUID attribute */
54 static int objectguid_add(struct ldb_module *module, struct ldb_request *req)
56 struct ldb_message *msg = req->op.add.message;
58 struct ldb_message *msg2;
59 struct ldb_message_element *attribute;
64 ldb_debug(module->ldb, LDB_DEBUG_TRACE, "objectguid_add_record\n");
66 if (ldb_dn_is_special(msg->dn)) { /* do not manipulate our control entries */
67 return ldb_next_request(module, req);
70 if ((attribute = objectguid_find_attribute(msg, "objectGUID")) != NULL ) {
71 return ldb_next_request(module, req);
74 msg2 = talloc(module, struct ldb_message);
80 msg2->num_elements = msg->num_elements;
81 msg2->private_data = msg->private_data;
82 msg2->elements = talloc_array(msg2, struct ldb_message_element, msg2->num_elements);
83 for (i = 0; i < msg2->num_elements; i++) {
84 msg2->elements[i] = msg->elements[i];
90 nt_status = ndr_push_struct_blob(&v, msg2, &guid,
91 (ndr_push_flags_fn_t)ndr_push_GUID);
92 if (!NT_STATUS_IS_OK(nt_status)) {
96 ret = ldb_msg_add_value(msg2, "objectGUID", &v);
101 req->op.add.message = msg2;
102 ret = ldb_next_request(module, req);
103 req->op.add.message = msg;
110 static int objectguid_add_async(struct ldb_module *module, struct ldb_request *req)
112 struct ldb_request *down_req;
113 struct ldb_message_element *attribute;
114 struct ldb_message *msg;
120 ldb_debug(module->ldb, LDB_DEBUG_TRACE, "objectguid_add_record\n");
122 /* do not manipulate our control entries */
123 if (ldb_dn_is_special(req->op.add.message->dn)) {
124 return ldb_next_request(module, req);
127 if ((attribute = objectguid_find_attribute(req->op.add.message, "objectGUID")) != NULL ) {
128 return ldb_next_request(module, req);
131 down_req = talloc(req, struct ldb_request);
132 if (down_req == NULL) {
133 return LDB_ERR_OPERATIONS_ERROR;
138 /* we have to copy the message as the caller might have it as a const */
139 down_req->op.add.message = msg = ldb_msg_copy_shallow(down_req, req->op.add.message);
141 return LDB_ERR_OPERATIONS_ERROR;
145 guid = GUID_random();
147 nt_status = ndr_push_struct_blob(&v, msg, &guid,
148 (ndr_push_flags_fn_t)ndr_push_GUID);
149 if (!NT_STATUS_IS_OK(nt_status)) {
153 ret = ldb_msg_add_value(msg, "objectGUID", &v);
158 /* go on with the call chain */
159 ret = ldb_next_request(module, down_req);
161 /* do not free down_req as the call results may be linked to it,
162 * it will be freed when the upper level request get freed */
163 if (ret == LDB_SUCCESS) {
164 req->async.handle = down_req->async.handle;
170 static int objectguid_request(struct ldb_module *module, struct ldb_request *req)
172 switch (req->operation) {
175 return objectguid_add(module, req);
178 return ldb_next_request(module, req);
183 static const struct ldb_module_ops objectguid_ops = {
184 .name = "objectguid",
185 .add = objectguid_add_async,
186 .request = objectguid_request
190 int objectguid_module_init(void)
192 return ldb_register_module(&objectguid_ops);