r23784: use the GPLv3 boilerplate as recommended by the FSF and the license text
[tprouty/samba.git] / source / libads / ldap_schema.c
1 /* 
2    Unix SMB/CIFS implementation.
3    ads (active directory) utility library
4    Copyright (C) Guenther Deschner 2005-2006
5    Copyright (C) Gerald (Jerry) Carter 2006
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 3 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include "includes.h"
22
23 #ifdef HAVE_LDAP
24
25 ADS_STATUS ads_get_attrnames_by_oids(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx,
26                                      const char *schema_path,
27                                      const char **OIDs, size_t num_OIDs, 
28                                      char ***OIDs_out, char ***names, size_t *count)
29 {
30         ADS_STATUS status;
31         LDAPMessage *res = NULL;
32         LDAPMessage *msg;
33         char *expr = NULL;
34         const char *attrs[] = { "lDAPDisplayName", "attributeId", NULL };
35         int i = 0, p = 0;
36         
37         if (!ads || !mem_ctx || !names || !count || !OIDs || !OIDs_out) {
38                 return ADS_ERROR(LDAP_PARAM_ERROR);
39         }
40
41         if (num_OIDs == 0 || OIDs[0] == NULL) {
42                 return ADS_ERROR_NT(NT_STATUS_NONE_MAPPED);
43         }
44
45         if ((expr = talloc_asprintf(mem_ctx, "(|")) == NULL) {
46                 return ADS_ERROR(LDAP_NO_MEMORY);
47         }
48
49         for (i=0; i<num_OIDs; i++) {
50
51                 if ((expr = talloc_asprintf_append(expr, "(attributeId=%s)", 
52                                                    OIDs[i])) == NULL) {
53                         return ADS_ERROR(LDAP_NO_MEMORY);
54                 }
55         }
56
57         if ((expr = talloc_asprintf_append(expr, ")")) == NULL) {
58                 return ADS_ERROR(LDAP_NO_MEMORY);
59         }
60
61         status = ads_do_search_retry(ads, schema_path, 
62                                      LDAP_SCOPE_SUBTREE, expr, attrs, &res);
63         if (!ADS_ERR_OK(status)) {
64                 return status;
65         }
66
67         *count = ads_count_replies(ads, res);
68         if (*count == 0 || !res) {
69                 status = ADS_ERROR_NT(NT_STATUS_NONE_MAPPED);
70                 goto out;
71         }
72
73         if (((*names) = TALLOC_ARRAY(mem_ctx, char *, *count)) == NULL) {
74                 status = ADS_ERROR(LDAP_NO_MEMORY);
75                 goto out;
76         }
77         if (((*OIDs_out) = TALLOC_ARRAY(mem_ctx, char *, *count)) == NULL) {
78                 status = ADS_ERROR(LDAP_NO_MEMORY);
79                 goto out;
80         }
81
82         for (msg = ads_first_entry(ads, res); msg != NULL; 
83              msg = ads_next_entry(ads, msg)) {
84
85                 (*names)[p]     = ads_pull_string(ads, mem_ctx, msg, 
86                                                   "lDAPDisplayName");
87                 (*OIDs_out)[p]  = ads_pull_string(ads, mem_ctx, msg, 
88                                                   "attributeId");
89                 if (((*names)[p] == NULL) || ((*OIDs_out)[p] == NULL)) {
90                         status = ADS_ERROR(LDAP_NO_MEMORY);
91                         goto out;
92                 }
93
94                 p++;
95         }
96
97         if (*count < num_OIDs) {
98                 status = ADS_ERROR_NT(STATUS_SOME_UNMAPPED);
99                 goto out;
100         }
101
102         status = ADS_ERROR(LDAP_SUCCESS);
103 out:
104         ads_msgfree(ads, res);
105
106         return status;
107 }
108
109 const char *ads_get_attrname_by_oid(ADS_STRUCT *ads, const char *schema_path, TALLOC_CTX *mem_ctx, const char * OID)
110 {
111         ADS_STATUS rc;
112         int count = 0;
113         LDAPMessage *res = NULL;
114         char *expr = NULL;
115         const char *attrs[] = { "lDAPDisplayName", NULL };
116         char *result;
117
118         if (ads == NULL || mem_ctx == NULL || OID == NULL) {
119                 goto failed;
120         }
121
122         expr = talloc_asprintf(mem_ctx, "(attributeId=%s)", OID);
123         if (expr == NULL) {
124                 goto failed;
125         }
126
127         rc = ads_do_search_retry(ads, schema_path, LDAP_SCOPE_SUBTREE, 
128                 expr, attrs, &res);
129         if (!ADS_ERR_OK(rc)) {
130                 goto failed;
131         }
132
133         count = ads_count_replies(ads, res);
134         if (count == 0 || !res) {
135                 goto failed;
136         }
137
138         result = ads_pull_string(ads, mem_ctx, res, "lDAPDisplayName");
139         ads_msgfree(ads, res);
140
141         return result;
142         
143 failed:
144         DEBUG(0,("ads_get_attrname_by_oid: failed to retrieve name for oid: %s\n", 
145                 OID));
146         
147         ads_msgfree(ads, res);
148         return NULL;
149 }
150
151 /*********************************************************************
152 *********************************************************************/
153
154 static ADS_STATUS ads_schema_path(ADS_STRUCT *ads, TALLOC_CTX *mem_ctx, char **schema_path)
155 {
156         ADS_STATUS status;
157         LDAPMessage *res;
158         const char *schema;
159         const char *attrs[] = { "schemaNamingContext", NULL };
160
161         status = ads_do_search(ads, "", LDAP_SCOPE_BASE, "(objectclass=*)", attrs, &res);
162         if (!ADS_ERR_OK(status)) {
163                 return status;
164         }
165
166         if ( (schema = ads_pull_string(ads, mem_ctx, res, "schemaNamingContext")) == NULL ) {
167                 ads_msgfree(ads, res);
168                 return ADS_ERROR(LDAP_NO_RESULTS_RETURNED);
169         }
170
171         if ( (*schema_path = talloc_strdup(mem_ctx, schema)) == NULL ) {
172                 ads_msgfree(ads, res);
173                 return ADS_ERROR(LDAP_NO_MEMORY);
174         }
175
176         ads_msgfree(ads, res);
177
178         return status;
179 }
180
181 /**
182  * Check for "Services for Unix" or rfc2307 Schema and load some attributes into the ADS_STRUCT
183  * @param ads connection to ads server
184  * @param enum mapping type
185  * @return ADS_STATUS status of search (False if one or more attributes couldn't be
186  * found in Active Directory)
187  **/ 
188 ADS_STATUS ads_check_posix_schema_mapping(TALLOC_CTX *mem_ctx,
189                                           ADS_STRUCT *ads,
190                                           enum wb_posix_mapping map_type,
191                                           struct posix_schema **s ) 
192 {
193         TALLOC_CTX *ctx = NULL; 
194         ADS_STATUS status;
195         char **oids_out, **names_out;
196         size_t num_names;
197         char *schema_path = NULL;
198         int i;
199         struct posix_schema *schema = NULL;
200
201         const char *oids_sfu[] = {      ADS_ATTR_SFU_UIDNUMBER_OID,
202                                         ADS_ATTR_SFU_GIDNUMBER_OID,
203                                         ADS_ATTR_SFU_HOMEDIR_OID,
204                                         ADS_ATTR_SFU_SHELL_OID,
205                                         ADS_ATTR_SFU_GECOS_OID};
206
207         const char *oids_sfu20[] = {    ADS_ATTR_SFU20_UIDNUMBER_OID,
208                                         ADS_ATTR_SFU20_GIDNUMBER_OID,
209                                         ADS_ATTR_SFU20_HOMEDIR_OID,
210                                         ADS_ATTR_SFU20_SHELL_OID,
211                                         ADS_ATTR_SFU20_GECOS_OID};
212
213         const char *oids_rfc2307[] = {  ADS_ATTR_RFC2307_UIDNUMBER_OID,
214                                         ADS_ATTR_RFC2307_GIDNUMBER_OID,
215                                         ADS_ATTR_RFC2307_HOMEDIR_OID,
216                                         ADS_ATTR_RFC2307_SHELL_OID,
217                                         ADS_ATTR_RFC2307_GECOS_OID };
218
219         DEBUG(10,("ads_check_posix_schema_mapping for schema mode: %d\n", map_type));
220
221         switch (map_type) {
222         
223                 case WB_POSIX_MAP_TEMPLATE:
224                 case WB_POSIX_MAP_UNIXINFO:
225                         DEBUG(10,("ads_check_posix_schema_mapping: nothing to do\n"));
226                         return ADS_ERROR(LDAP_SUCCESS);
227
228                 case WB_POSIX_MAP_SFU:
229                 case WB_POSIX_MAP_SFU20:
230                 case WB_POSIX_MAP_RFC2307:
231                         break;
232
233                 default:
234                         DEBUG(0,("ads_check_posix_schema_mapping: "
235                                  "unknown enum %d\n", map_type));
236                         return ADS_ERROR(LDAP_PARAM_ERROR);
237         }
238
239         if ( (ctx = talloc_init("ads_check_posix_schema_mapping")) == NULL ) {
240                 return ADS_ERROR(LDAP_NO_MEMORY);
241         }
242
243         if ( (schema = TALLOC_P(mem_ctx, struct posix_schema)) == NULL ) {
244                 TALLOC_FREE( ctx );
245                 return ADS_ERROR(LDAP_NO_MEMORY);
246         }
247         
248         status = ads_schema_path(ads, ctx, &schema_path);
249         if (!ADS_ERR_OK(status)) {
250                 DEBUG(3,("ads_check_posix_mapping: Unable to retrieve schema DN!\n"));
251                 goto done;
252         }
253
254         switch (map_type) {
255                 case WB_POSIX_MAP_SFU:
256                         status = ads_get_attrnames_by_oids(ads, ctx, schema_path, oids_sfu, 
257                                                            ARRAY_SIZE(oids_sfu), 
258                                                            &oids_out, &names_out, &num_names);
259                         break;
260                 case WB_POSIX_MAP_SFU20:
261                         status = ads_get_attrnames_by_oids(ads, ctx, schema_path, oids_sfu20, 
262                                                            ARRAY_SIZE(oids_sfu20), 
263                                                            &oids_out, &names_out, &num_names);
264                         break;
265                 case WB_POSIX_MAP_RFC2307:
266                         status = ads_get_attrnames_by_oids(ads, ctx, schema_path, oids_rfc2307, 
267                                                            ARRAY_SIZE(oids_rfc2307), 
268                                                            &oids_out, &names_out, &num_names);
269                         break;
270                 default:
271                         status = ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
272                         break;
273         }
274
275         if (!ADS_ERR_OK(status)) {
276                 DEBUG(3,("ads_check_posix_schema_mapping: failed %s\n", 
277                         ads_errstr(status)));
278                 goto done;
279         }
280
281         for (i=0; i<num_names; i++) {
282
283                 DEBUGADD(10,("\tOID %s has name: %s\n", oids_out[i], names_out[i]));
284
285                 if (strequal(ADS_ATTR_RFC2307_UIDNUMBER_OID, oids_out[i]) ||
286                     strequal(ADS_ATTR_SFU_UIDNUMBER_OID, oids_out[i]) ||
287                     strequal(ADS_ATTR_SFU20_UIDNUMBER_OID, oids_out[i])) {
288                         schema->posix_uidnumber_attr = talloc_strdup(schema, names_out[i]);
289                         continue;                      
290                 }
291
292                 if (strequal(ADS_ATTR_RFC2307_GIDNUMBER_OID, oids_out[i]) ||
293                     strequal(ADS_ATTR_SFU_GIDNUMBER_OID, oids_out[i]) ||
294                     strequal(ADS_ATTR_SFU20_GIDNUMBER_OID, oids_out[i])) {
295                         schema->posix_gidnumber_attr = talloc_strdup(schema, names_out[i]);
296                         continue;               
297                 }
298
299                 if (strequal(ADS_ATTR_RFC2307_HOMEDIR_OID, oids_out[i]) ||
300                     strequal(ADS_ATTR_SFU_HOMEDIR_OID, oids_out[i]) ||
301                     strequal(ADS_ATTR_SFU20_HOMEDIR_OID, oids_out[i])) {
302                         schema->posix_homedir_attr = talloc_strdup(schema, names_out[i]);
303                         continue;                       
304                 }
305
306                 if (strequal(ADS_ATTR_RFC2307_SHELL_OID, oids_out[i]) ||
307                     strequal(ADS_ATTR_SFU_SHELL_OID, oids_out[i]) ||
308                     strequal(ADS_ATTR_SFU20_SHELL_OID, oids_out[i])) {
309                         schema->posix_shell_attr = talloc_strdup(schema, names_out[i]);
310                         continue;                       
311                 }
312
313                 if (strequal(ADS_ATTR_RFC2307_GECOS_OID, oids_out[i]) ||
314                     strequal(ADS_ATTR_SFU_GECOS_OID, oids_out[i]) ||
315                     strequal(ADS_ATTR_SFU20_GECOS_OID, oids_out[i])) {
316                         schema->posix_gecos_attr = talloc_strdup(schema, names_out[i]);
317                 }
318         }
319
320         if (!schema->posix_uidnumber_attr ||
321             !schema->posix_gidnumber_attr ||
322             !schema->posix_homedir_attr ||
323             !schema->posix_shell_attr ||
324             !schema->posix_gecos_attr) {
325                 status = ADS_ERROR(LDAP_NO_MEMORY);
326                 TALLOC_FREE( schema );          
327                 goto done;
328         }
329
330         *s = schema;
331         
332         status = ADS_ERROR(LDAP_SUCCESS);
333         
334 done:
335         TALLOC_FREE(ctx);
336
337         return status;
338 }
339
340 #endif