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                                 DEBUG(1,("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         if (domain) {
 119                 asn1_push_tag(&data, ASN1_CONTEXT(3));
 120                 asn1_write_OctetString(&data, "DnsDomain", 9);
 121                 asn1_write_OctetString(&data, domain, strlen(domain));
 122                 asn1_pop_tag(&data);
 123         }
 124
 125         asn1_push_tag(&data, ASN1_CONTEXT(3));
 126         asn1_write_OctetString(&data, "Host", 4);
 127         asn1_write_OctetString(&data, hostname, strlen(hostname));
 128         asn1_pop_tag(&data);
 129
 130 #ifdef CLDAP_USER_QUERY
 131         asn1_push_tag(&data, ASN1_CONTEXT(3));
 132         asn1_write_OctetString(&data, "User", 4);
 133         asn1_write_OctetString(&data, "SAMBA$", 6);
 134         asn1_pop_tag(&data);
 135
 136         asn1_push_tag(&data, ASN1_CONTEXT(3));
 137         asn1_write_OctetString(&data, "AAC", 4);
 138         asn1_write_OctetString(&data, aac, 4);
 139         asn1_pop_tag(&data);
 140 #endif
 141
 142         asn1_push_tag(&data, ASN1_CONTEXT(3));
 143         asn1_write_OctetString(&data, "NtVer", 5);
 144         asn1_write_OctetString(&data, ntver, 4);
 145         asn1_pop_tag(&data);
 146
 147         asn1_pop_tag(&data);
 148
 149         asn1_push_tag(&data,ASN1_SEQUENCE(0));
 150         asn1_write_OctetString(&data, "NetLogon", 8);
 151         asn1_pop_tag(&data);
 152         asn1_pop_tag(&data);
 153         asn1_pop_tag(&data);
 154
 155         if (data.has_error) {
 156                 DEBUG(2,("Failed to build cldap netlogon at offset %d\n", (int)data.ofs));
 157                 asn1_free(&data);
 158                 return -1;
 159         }
 160
 161         if (write(sock, data.data, data.length) != (ssize_t)data.length) {
 162                 DEBUG(2,("failed to send cldap query (%s)\n", strerror(errno)));
 163                 asn1_free(&data);
 164                 return -1;
 165         }
 166
 167         asn1_free(&data);
 168
 169         return 0;
 170 }
 171
 172 static SIG_ATOMIC_T gotalarm;
 173
 174 /***************************************************************
 175  Signal function to tell us we timed out.
 176 ****************************************************************/
 177
 178 static void gotalarm_sig(void)
 179 {
 180         gotalarm = 1;
 181 }
 182
 183 /*
 184   receive a cldap netlogon reply
 185 */
 186 static int recv_cldap_netlogon(int sock, struct cldap_netlogon_reply *reply)
 187 {
 188         int ret;
 189         ASN1_DATA data;
 190         DATA_BLOB blob;
 191         DATA_BLOB os1, os2, os3;
 192         int i1;
 193         /* half the time of a regular ldap timeout, not less than 3 seconds. */
 194         unsigned int al_secs = MAX(3,lp_ldap_timeout()/2);
 195         char *p;
 196
 197         blob = data_blob(NULL, 8192);
 198         if (blob.data == NULL) {
 199                 DEBUG(1, ("data_blob failed\n"));
 200                 errno = ENOMEM;
 201                 return -1;
 202         }
 203
 204         /* Setup timeout */
 205         gotalarm = 0;
 206         CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
 207         alarm(al_secs);
 208         /* End setup timeout. */
 209
 210         ret = read(sock, blob.data, blob.length);
 211
 212         /* Teardown timeout. */
 213         CatchSignal(SIGALRM, SIGNAL_CAST SIG_IGN);
 214         alarm(0);
 215
 216         if (ret <= 0) {
 217                 DEBUG(1,("no reply received to cldap netlogon\n"));
 218                 data_blob_free(&blob);
 219                 return -1;
 220         }
 221         blob.length = ret;
 222
 223         asn1_load(&data, blob);
 224         asn1_start_tag(&data, ASN1_SEQUENCE(0));
 225         asn1_read_Integer(&data, &i1);
 226         asn1_start_tag(&data, ASN1_APPLICATION(4));
 227         asn1_read_OctetString(&data, &os1);
 228         asn1_start_tag(&data, ASN1_SEQUENCE(0));
 229         asn1_start_tag(&data, ASN1_SEQUENCE(0));
 230         asn1_read_OctetString(&data, &os2);
 231         asn1_start_tag(&data, ASN1_SET);
 232         asn1_read_OctetString(&data, &os3);
 233         asn1_end_tag(&data);
 234         asn1_end_tag(&data);
 235         asn1_end_tag(&data);
 236         asn1_end_tag(&data);
 237         asn1_end_tag(&data);
 238
 239         if (data.has_error) {
 240                 data_blob_free(&blob);
 241                 asn1_free(&data);
 242                 DEBUG(1,("Failed to parse cldap reply\n"));
 243                 return -1;
 244         }
 245
 246         p = (char *)os3.data;
 247
 248         reply->type = IVAL(p, 0); p += 4;
 249         reply->flags = IVAL(p, 0); p += 4;
 250
 251         memcpy(&reply->guid.info, p, UUID_FLAT_SIZE);
 252         p += UUID_FLAT_SIZE;
 253
 254         p += pull_netlogon_string(reply->forest, p, (const char *)os3.data);
 255         p += pull_netlogon_string(reply->domain, p, (const char *)os3.data);
 256         p += pull_netlogon_string(reply->hostname, p, (const char *)os3.data);
 257         p += pull_netlogon_string(reply->netbios_domain, p, (const char *)os3.data);
 258         p += pull_netlogon_string(reply->netbios_hostname, p, (const char *)os3.data);
 259         p += pull_netlogon_string(reply->unk, p, (const char *)os3.data);
 260
 261         if (reply->type == SAMLOGON_AD_R) {
 262                 p += pull_netlogon_string(reply->user_name, p, (const char *)os3.data);
 263         } else {
 264                 *reply->user_name = 0;
 265         }
 266
 267         p += pull_netlogon_string(reply->server_site_name, p, (const char *)os3.data);
 268         p += pull_netlogon_string(reply->client_site_name, p, (const char *)os3.data);
 269
 270         reply->version = IVAL(p, 0);
 271         reply->lmnt_token = SVAL(p, 4);
 272         reply->lm20_token = SVAL(p, 6);
 273
 274         data_blob_free(&os1);
 275         data_blob_free(&os2);
 276         data_blob_free(&os3);
 277         data_blob_free(&blob);
 278
 279         asn1_free(&data);
 280
 281         return 0;
 282 }
 283
 284 /*******************************************************************
 285   do a cldap netlogon query.  Always 389/udp
 286 *******************************************************************/
 287
 288 BOOL ads_cldap_netlogon(const char *server, const char *realm,  struct cldap_netlogon_reply *reply)
 289 {
 290         int sock;
 291         int ret;
 292
 293         sock = open_udp_socket(server, LDAP_PORT );
 294         if (sock == -1) {
 295                 DEBUG(2,("ads_cldap_netlogon: Failed to open udp socket to %s\n",
 296                          server));
 297                 return False;
 298         }
 299
 300         ret = send_cldap_netlogon(sock, realm, global_myname(), 6);
 301         if (ret != 0) {
 302                 close(sock);
 303                 return False;
 304         }
 305         ret = recv_cldap_netlogon(sock, reply);
 306         close(sock);
 307
 308         if (ret == -1) {
 309                 return False;
 310         }
 311
 312         return True;
 313 }