4 Copyright (C) Simo Sorce 2004-2008
5 Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
6 Copyright (C) Andrew Tridgell 2005
7 Copyright (C) Stefan Metzmacher <metze@samba.org> 2007
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
26 * Component: ldb instancetype module
28 * Description: add an instanceType onto every new record
30 * Author: Andrew Bartlett
35 #include "ldb_module.h"
36 #include "librpc/gen_ndr/ndr_misc.h"
37 #include "dsdb/samdb/samdb.h"
38 #include "../libds/common/flags.h"
39 #include "dsdb/samdb/ldb_modules/util.h"
42 struct ldb_module
*module
;
43 struct ldb_request
*req
;
44 struct ldb_request
*add_req
;
47 static int it_add_callback(struct ldb_request
*req
, struct ldb_reply
*ares
)
49 struct ldb_context
*ldb
;
50 struct it_context
*ac
;
52 ac
= talloc_get_type(req
->context
, struct it_context
);
53 ldb
= ldb_module_get_ctx(ac
->module
);
56 return ldb_module_done(ac
->req
, NULL
, NULL
,
57 LDB_ERR_OPERATIONS_ERROR
);
60 if (ares
->type
== LDB_REPLY_REFERRAL
) {
61 return ldb_module_send_referral(ac
->req
, ares
->referral
);
64 if (ares
->error
!= LDB_SUCCESS
) {
65 return ldb_module_done(ac
->req
, ares
->controls
,
66 ares
->response
, ares
->error
);
69 if (ares
->type
!= LDB_REPLY_DONE
) {
70 ldb_set_errstring(ldb
, "Invalid reply type!");
71 return ldb_module_done(ac
->req
, NULL
, NULL
,
72 LDB_ERR_OPERATIONS_ERROR
);
75 /* Add the boilerplate entries */
77 return ldb_module_done(ac
->req
, ares
->controls
,
78 ares
->response
, ares
->error
);
81 /* add_record: add instancetype attribute */
82 static int instancetype_add(struct ldb_module
*module
, struct ldb_request
*req
)
84 struct ldb_context
*ldb
;
85 struct ldb_request
*down_req
;
86 struct ldb_message
*msg
;
87 struct it_context
*ac
;
88 uint32_t instance_type
;
91 ldb
= ldb_module_get_ctx(module
);
93 ldb_debug(ldb
, LDB_DEBUG_TRACE
, "instancetype_add_record\n");
95 /* do not manipulate our control entries */
96 if (ldb_dn_is_special(req
->op
.add
.message
->dn
)) {
97 return ldb_next_request(module
, req
);
100 if (ldb_msg_find_element(req
->op
.add
.message
, "instanceType")) {
101 unsigned int instanceType
= ldb_msg_find_attr_as_uint(req
->op
.add
.message
, "instanceType", 0);
102 if (!(instanceType
& INSTANCE_TYPE_IS_NC_HEAD
)) {
103 return ldb_next_request(module
, req
);
106 /* Forward the 'add' to the modules below, but if it
107 * succeeds, then we might need to add the boilerplate
108 * entries (lost+found, deleted objects) */
109 ac
= talloc(req
, struct it_context
);
111 return LDB_ERR_OPERATIONS_ERROR
;
116 ret
= ldb_build_add_req(&ac
->add_req
, ldb_module_get_ctx(ac
->module
), ac
,
117 ac
->req
->op
.add
.message
,
122 if (ret
!= LDB_SUCCESS
) {
126 /* Do the original add */
127 return ldb_next_request(ac
->module
, ac
->add_req
);
130 /* we have to copy the message as the caller might have it as a const */
131 msg
= ldb_msg_copy_shallow(req
, req
->op
.add
.message
);
134 return LDB_ERR_OPERATIONS_ERROR
;
138 * TODO: calculate correct instance type
140 instance_type
= INSTANCE_TYPE_WRITE
;
142 ret
= ldb_msg_add_fmt(msg
, "instanceType", "%u", instance_type
);
143 if (ret
!= LDB_SUCCESS
) {
145 return LDB_ERR_OPERATIONS_ERROR
;
148 ret
= ldb_build_add_req(&down_req
, ldb
, req
,
151 req
, dsdb_next_callback
,
153 if (ret
!= LDB_SUCCESS
) {
157 /* go on with the call chain */
158 return ldb_next_request(module
, down_req
);
161 _PUBLIC_
const struct ldb_module_ops ldb_instancetype_module_ops
= {
162 .name
= "instancetype",
163 .add
= instancetype_add
,