s4:dsdb Add error string in 'no such object' because of 0 replies case
[kai/samba-autobuild/.git] / source4 / dsdb / samdb / ldb_modules / util.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Samba utility functions
4
5    Copyright (C) Andrew Tridgell 2009
6    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2009
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "ldb.h"
23 #include "ldb_module.h"
24
25 /*
26   search for attrs on one DN, in the modules below
27  */
28 int dsdb_module_search_dn(struct ldb_module *module,
29                           TALLOC_CTX *mem_ctx,
30                           struct ldb_result **_res,
31                           struct ldb_dn *basedn,
32                           const char * const *attrs)
33 {
34         int ret;
35         struct ldb_request *req;
36         TALLOC_CTX *tmp_ctx;
37         struct ldb_result *res;
38
39         tmp_ctx = talloc_new(mem_ctx);
40
41         res = talloc_zero(tmp_ctx, struct ldb_result);
42         if (!res) {
43                 return LDB_ERR_OPERATIONS_ERROR;
44         }
45
46         ret = ldb_build_search_req(&req, ldb_module_get_ctx(module), tmp_ctx,
47                                    basedn,
48                                    LDB_SCOPE_BASE,
49                                    NULL,
50                                    attrs,
51                                    NULL,
52                                    res,
53                                    ldb_search_default_callback,
54                                    NULL);
55         if (ret != LDB_SUCCESS) {
56                 talloc_free(tmp_ctx);
57                 return ret;
58         }
59
60         ret = ldb_next_request(module, req);
61         if (ret == LDB_SUCCESS) {
62                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
63         }
64
65         if (ret != LDB_SUCCESS) {
66                 talloc_free(tmp_ctx);
67                 return ret;
68         }
69
70         if (res->count != 1) {
71                 /* we may be reading a DB that does not have the 'check base on search' option... */
72                 ret = LDB_ERR_NO_SUCH_OBJECT;
73                 ldb_asprintf_errstring(ldb_module_get_ctx(module), 
74                                        "dsdb_module_search_dn: did not find base dn %s (%d results)", 
75                                        ldb_dn_get_linearized(basedn), res->count);
76         } else {
77                 *_res = talloc_steal(mem_ctx, res);
78         }
79         talloc_free(tmp_ctx);
80         return ret;
81 }
82
83 /*
84   search for attrs in the modules below
85  */
86 int dsdb_module_search(struct ldb_module *module,
87                        TALLOC_CTX *mem_ctx,
88                        struct ldb_result **_res,
89                        struct ldb_dn *basedn, enum ldb_scope scope, 
90                        const char * const *attrs,
91                        const char *expression)
92 {
93         int ret;
94         struct ldb_request *req;
95         TALLOC_CTX *tmp_ctx;
96         struct ldb_result *res;
97
98         tmp_ctx = talloc_new(mem_ctx);
99
100         res = talloc_zero(tmp_ctx, struct ldb_result);
101         if (!res) {
102                 return LDB_ERR_OPERATIONS_ERROR;
103         }
104
105         ret = ldb_build_search_req(&req, ldb_module_get_ctx(module), tmp_ctx,
106                                    basedn,
107                                    scope,
108                                    expression,
109                                    attrs,
110                                    NULL,
111                                    res,
112                                    ldb_search_default_callback,
113                                    NULL);
114         if (ret != LDB_SUCCESS) {
115                 talloc_free(tmp_ctx);
116                 return ret;
117         }
118
119         ret = ldb_next_request(module, req);
120         if (ret == LDB_SUCCESS) {
121                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
122         }
123
124         talloc_free(req);
125         if (ret == LDB_SUCCESS) {
126                 *_res = talloc_steal(mem_ctx, res);
127         }
128         talloc_free(tmp_ctx);
129         return ret;
130 }
131