More strlcat/strlcpy truncate checks.
[amitay/samba.git] / source3 / libads / ads_struct.c
1 /* 
2    Unix SMB/CIFS implementation.
3    ads (active directory) utility library
4    Copyright (C) Andrew Tridgell 2001
5    Copyright (C) Andrew Bartlett 2001
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 #include "ads.h"
23
24 /* return a ldap dn path from a string, given separators and field name
25    caller must free
26 */
27 char *ads_build_path(const char *realm, const char *sep, const char *field, int reverse)
28 {
29         char *p, *r;
30         int numbits = 0;
31         char *ret;
32         int len;
33         char *saveptr;
34
35         r = SMB_STRDUP(realm);
36
37         if (!r || !*r) {
38                 return r;
39         }
40
41         for (p=r; *p; p++) {
42                 if (strchr(sep, *p)) {
43                         numbits++;
44                 }
45         }
46
47         len = (numbits+1)*(strlen(field)+1) + strlen(r) + 1;
48
49         ret = (char *)SMB_MALLOC(len);
50         if (!ret) {
51                 free(r);
52                 return NULL;
53         }
54
55         if (strlcpy(ret,field, len) >= len) {
56                 /* Truncate ! */
57                 free(r);
58                 return NULL;
59         }
60         p=strtok_r(r, sep, &saveptr);
61         if (p) {
62                 if (strlcat(ret, p, len) >= len) {
63                         free(r);
64                         return NULL;
65                 }
66
67                 while ((p=strtok_r(NULL, sep, &saveptr)) != NULL) {
68                         int retval;
69                         char *s = NULL;
70                         if (reverse)
71                                 retval = asprintf(&s, "%s%s,%s", field, p, ret);
72                         else
73                                 retval = asprintf(&s, "%s,%s%s", ret, field, p);
74                         free(ret);
75                         if (retval == -1) {
76                                 free(r);
77                                 return NULL;
78                         }
79                         ret = SMB_STRDUP(s);
80                         free(s);
81                 }
82         }
83
84         free(r);
85         return ret;
86 }
87
88 /* return a dn of the form "dc=AA,dc=BB,dc=CC" from a 
89    realm of the form AA.BB.CC 
90    caller must free
91 */
92 char *ads_build_dn(const char *realm)
93 {
94         return ads_build_path(realm, ".", "dc=", 0);
95 }
96
97 /* return a DNS name in the for aa.bb.cc from the DN  
98    "dc=AA,dc=BB,dc=CC".  caller must free
99 */
100 char *ads_build_domain(const char *dn)
101 {
102         char *dnsdomain = NULL;
103
104         /* result should always be shorter than the DN */
105
106         if ( (dnsdomain = SMB_STRDUP( dn )) == NULL ) {
107                 DEBUG(0,("ads_build_domain: malloc() failed!\n"));              
108                 return NULL;            
109         }       
110
111         strlower_m( dnsdomain );        
112         all_string_sub( dnsdomain, "dc=", "", 0);
113         all_string_sub( dnsdomain, ",", ".", 0 );
114
115         return dnsdomain;       
116 }
117
118
119
120 #ifndef LDAP_PORT
121 #define LDAP_PORT 389
122 #endif
123
124 /*
125   initialise a ADS_STRUCT, ready for some ads_ ops
126 */
127 ADS_STRUCT *ads_init(const char *realm, 
128                      const char *workgroup,
129                      const char *ldap_server)
130 {
131         ADS_STRUCT *ads;
132         int wrap_flags;
133
134         ads = SMB_XMALLOC_P(ADS_STRUCT);
135         ZERO_STRUCTP(ads);
136
137         ads->server.realm = realm? SMB_STRDUP(realm) : NULL;
138         ads->server.workgroup = workgroup ? SMB_STRDUP(workgroup) : NULL;
139         ads->server.ldap_server = ldap_server? SMB_STRDUP(ldap_server) : NULL;
140
141         /* we need to know if this is a foreign realm */
142         if (realm && *realm && !strequal(lp_realm(), realm)) {
143                 ads->server.foreign = 1;
144         }
145         if (workgroup && *workgroup && !strequal(lp_workgroup(), workgroup)) {
146                 ads->server.foreign = 1;
147         }
148
149         /* the caller will own the memory by default */
150         ads->is_mine = 1;
151
152         wrap_flags = lp_client_ldap_sasl_wrapping();
153         if (wrap_flags == -1) {
154                 wrap_flags = 0;
155         }
156
157         ads->auth.flags = wrap_flags;
158
159         return ads;
160 }
161
162 /****************************************************************
163 ****************************************************************/
164
165 bool ads_set_sasl_wrap_flags(ADS_STRUCT *ads, int flags)
166 {
167         if (!ads) {
168                 return false;
169         }
170
171         ads->auth.flags = flags;
172
173         return true;
174 }
175
176 /*
177   free the memory used by the ADS structure initialized with 'ads_init(...)'
178 */
179 void ads_destroy(ADS_STRUCT **ads)
180 {
181         if (ads && *ads) {
182                 bool is_mine;
183
184                 is_mine = (*ads)->is_mine;
185 #if HAVE_LDAP
186                 ads_disconnect(*ads);
187 #endif
188                 SAFE_FREE((*ads)->server.realm);
189                 SAFE_FREE((*ads)->server.workgroup);
190                 SAFE_FREE((*ads)->server.ldap_server);
191
192                 SAFE_FREE((*ads)->auth.realm);
193                 SAFE_FREE((*ads)->auth.password);
194                 SAFE_FREE((*ads)->auth.user_name);
195                 SAFE_FREE((*ads)->auth.kdc_server);
196
197                 SAFE_FREE((*ads)->config.realm);
198                 SAFE_FREE((*ads)->config.bind_path);
199                 SAFE_FREE((*ads)->config.ldap_server_name);
200                 SAFE_FREE((*ads)->config.server_site_name);
201                 SAFE_FREE((*ads)->config.client_site_name);
202                 SAFE_FREE((*ads)->config.schema_path);
203                 SAFE_FREE((*ads)->config.config_path);
204
205                 ZERO_STRUCTP(*ads);
206
207                 if ( is_mine )
208                         SAFE_FREE(*ads);
209         }
210 }