r15543: New implementation of 'net ads join' to be more like Windows XP.
[kai/samba.git] / source3 / libads / cldap.c
1 /* 
2    Samba Unix/Linux SMB client library 
3    net ads cldap functions 
4    Copyright (C) 2001 Andrew Tridgell (tridge@samba.org)
5    Copyright (C) 2003 Jim McDonough (jmcd@us.ibm.com)
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 2 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, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  
20 */
21
22 #include "includes.h"
23
24 /*
25   These seem to be strings as described in RFC1035 4.1.4 and can be:
26
27    - a sequence of labels ending in a zero octet
28    - a pointer
29    - a sequence of labels ending with a pointer
30
31   A label is a byte where the first two bits must be zero and the remaining
32   bits represent the length of the label followed by the label itself.
33   Therefore, the length of a label is at max 64 bytes.  Under RFC1035, a
34   sequence of labels cannot exceed 255 bytes.
35
36   A pointer consists of a 14 bit offset from the beginning of the data.
37
38   struct ptr {
39     unsigned ident:2; // must be 11
40     unsigned offset:14; // from the beginning of data
41   };
42
43   This is used as a method to compress the packet by eliminated duplicate
44   domain components.  Since a UDP packet should probably be < 512 bytes and a
45   DNS name can be up to 255 bytes, this actually makes a lot of sense.
46 */
47 static unsigned pull_netlogon_string(char *ret, const char *ptr,
48                                      const char *data)
49 {
50         char *pret = ret;
51         int followed_ptr = 0;
52         unsigned ret_len = 0;
53
54         memset(pret, 0, MAX_DNS_LABEL);
55         do {
56                 if ((*ptr & 0xc0) == 0xc0) {
57                         uint16 len;
58
59                         if (!followed_ptr) {
60                                 ret_len += 2;
61                                 followed_ptr = 1;
62                         }
63                         len = ((ptr[0] & 0x3f) << 8) | ptr[1];
64                         ptr = data + len;
65                 } else if (*ptr) {
66                         uint8 len = (uint8)*(ptr++);
67
68                         if ((pret - ret + len + 1) >= MAX_DNS_LABEL) {
69                                 d_fprintf(stderr, "DC returning too long DNS name\n");
70                                 return 0;
71                         }
72
73                         if (pret != ret) {
74                                 *pret = '.';
75                                 pret++;
76                         }
77                         memcpy(pret, ptr, len);
78                         pret += len;
79                         ptr += len;
80
81                         if (!followed_ptr) {
82                                 ret_len += (len + 1);
83                         }
84                 }
85         } while (*ptr);
86
87         return followed_ptr ? ret_len : ret_len + 1;
88 }
89
90 /*
91   do a cldap netlogon query
92 */
93 static int send_cldap_netlogon(int sock, const char *domain, 
94                                const char *hostname, unsigned ntversion)
95 {
96         ASN1_DATA data;
97         char ntver[4];
98 #ifdef CLDAP_USER_QUERY
99         char aac[4];
100
101         SIVAL(aac, 0, 0x00000180);
102 #endif
103         SIVAL(ntver, 0, ntversion);
104
105         memset(&data, 0, sizeof(data));
106
107         asn1_push_tag(&data,ASN1_SEQUENCE(0));
108         asn1_write_Integer(&data, 4);
109         asn1_push_tag(&data, ASN1_APPLICATION(3));
110         asn1_write_OctetString(&data, NULL, 0);
111         asn1_write_enumerated(&data, 0);
112         asn1_write_enumerated(&data, 0);
113         asn1_write_Integer(&data, 0);
114         asn1_write_Integer(&data, 0);
115         asn1_write_BOOLEAN2(&data, False);
116         asn1_push_tag(&data, ASN1_CONTEXT(0));
117
118         asn1_push_tag(&data, ASN1_CONTEXT(3));
119         asn1_write_OctetString(&data, "DnsDomain", 9);
120         asn1_write_OctetString(&data, domain, strlen(domain));
121         asn1_pop_tag(&data);
122
123         asn1_push_tag(&data, ASN1_CONTEXT(3));
124         asn1_write_OctetString(&data, "Host", 4);
125         asn1_write_OctetString(&data, hostname, strlen(hostname));
126         asn1_pop_tag(&data);
127
128 #ifdef CLDAP_USER_QUERY
129         asn1_push_tag(&data, ASN1_CONTEXT(3));
130         asn1_write_OctetString(&data, "User", 4);
131         asn1_write_OctetString(&data, "SAMBA$", 6);
132         asn1_pop_tag(&data);
133
134         asn1_push_tag(&data, ASN1_CONTEXT(3));
135         asn1_write_OctetString(&data, "AAC", 4);
136         asn1_write_OctetString(&data, aac, 4);
137         asn1_pop_tag(&data);
138 #endif
139
140         asn1_push_tag(&data, ASN1_CONTEXT(3));
141         asn1_write_OctetString(&data, "NtVer", 5);
142         asn1_write_OctetString(&data, ntver, 4);
143         asn1_pop_tag(&data);
144
145         asn1_pop_tag(&data);
146
147         asn1_push_tag(&data,ASN1_SEQUENCE(0));
148         asn1_write_OctetString(&data, "NetLogon", 8);
149         asn1_pop_tag(&data);
150         asn1_pop_tag(&data);
151         asn1_pop_tag(&data);
152
153         if (data.has_error) {
154                 d_fprintf(stderr, "Failed to build cldap netlogon at offset %d\n", (int)data.ofs);
155                 asn1_free(&data);
156                 return -1;
157         }
158
159         if (write(sock, data.data, data.length) != (ssize_t)data.length) {
160                 d_fprintf(stderr, "failed to send cldap query (%s)\n", strerror(errno));
161         }
162
163         asn1_free(&data);
164
165         return 0;
166 }
167
168
169 /*
170   receive a cldap netlogon reply
171 */
172 static int recv_cldap_netlogon(int sock, struct cldap_netlogon_reply *reply)
173 {
174         int ret;
175         ASN1_DATA data;
176         DATA_BLOB blob;
177         DATA_BLOB os1, os2, os3;
178         int i1;
179         char *p;
180
181         blob = data_blob(NULL, 8192);
182
183         ret = read(sock, blob.data, blob.length);
184
185         if (ret <= 0) {
186                 d_fprintf(stderr, "no reply received to cldap netlogon\n");
187                 return -1;
188         }
189         blob.length = ret;
190
191         asn1_load(&data, blob);
192         asn1_start_tag(&data, ASN1_SEQUENCE(0));
193         asn1_read_Integer(&data, &i1);
194         asn1_start_tag(&data, ASN1_APPLICATION(4));
195         asn1_read_OctetString(&data, &os1);
196         asn1_start_tag(&data, ASN1_SEQUENCE(0));
197         asn1_start_tag(&data, ASN1_SEQUENCE(0));
198         asn1_read_OctetString(&data, &os2);
199         asn1_start_tag(&data, ASN1_SET);
200         asn1_read_OctetString(&data, &os3);
201         asn1_end_tag(&data);
202         asn1_end_tag(&data);
203         asn1_end_tag(&data);
204         asn1_end_tag(&data);
205         asn1_end_tag(&data);
206
207         if (data.has_error) {
208                 d_fprintf(stderr, "Failed to parse cldap reply\n");
209                 return -1;
210         }
211
212         p = (char *)os3.data;
213
214         reply->type = IVAL(p, 0); p += 4;
215         reply->flags = IVAL(p, 0); p += 4;
216
217         memcpy(&reply->guid.info, p, UUID_FLAT_SIZE);
218         p += UUID_FLAT_SIZE;
219
220         p += pull_netlogon_string(reply->forest, p, (const char *)os3.data);
221         p += pull_netlogon_string(reply->domain, p, (const char *)os3.data);
222         p += pull_netlogon_string(reply->hostname, p, (const char *)os3.data);
223         p += pull_netlogon_string(reply->netbios_domain, p, (const char *)os3.data);
224         p += pull_netlogon_string(reply->netbios_hostname, p, (const char *)os3.data);
225         p += pull_netlogon_string(reply->unk, p, (const char *)os3.data);
226
227         if (reply->type == SAMLOGON_AD_R) {
228                 p += pull_netlogon_string(reply->user_name, p, (const char *)os3.data);
229         } else {
230                 *reply->user_name = 0;
231         }
232
233         p += pull_netlogon_string(reply->site_name, p, (const char *)os3.data);
234         p += pull_netlogon_string(reply->site_name_2, p, (const char *)os3.data);
235
236         reply->version = IVAL(p, 0);
237         reply->lmnt_token = SVAL(p, 4);
238         reply->lm20_token = SVAL(p, 6);
239
240         data_blob_free(&os1);
241         data_blob_free(&os2);
242         data_blob_free(&os3);
243         data_blob_free(&blob);
244         
245         return 0;
246 }
247
248 /*******************************************************************
249   do a cldap netlogon query.  Always 389/udp
250 *******************************************************************/
251
252 BOOL ads_cldap_netlogon(const char *server, const char *realm,  struct cldap_netlogon_reply *reply)
253 {
254         int sock;
255         int ret;
256
257         sock = open_udp_socket(server, LDAP_PORT );
258         if (sock == -1) {
259                 DEBUG(2,("ads_cldap_netlogon: Failed to open udp socket to %s\n", 
260                          server));
261                 return False;
262         }
263
264         ret = send_cldap_netlogon(sock, realm, global_myname(), 6);
265         if (ret != 0) {
266                 return False;
267         }
268         ret = recv_cldap_netlogon(sock, reply);
269         close(sock);
270
271         if (ret == -1) {
272                 return False;
273         }
274
275         return True;
276 }
277
278