git.sur5r.net Git - openldap/blob - libraries/libldap/getdxbyname.c

   1 /* $OpenLDAP$ */
   2 /*
   3  * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
   4  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
   5  */
   6 /*  Portions
   7  *  Copyright (c) 1995 Regents of the University of Michigan.
   8  *  All rights reserved.
   9  *
  10  * ldap_getdxbyname - retrieve DX records from the DNS (from TXT records for now)
  11  */
  12
  13 #include "portable.h"
  14
  15 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
  16
  17 #include <stdio.h>
  18 #include <ac/stdlib.h>
  19
  20 #include <ac/ctype.h>
  21 #include <ac/socket.h>
  22 #include <ac/string.h>
  23 #include <ac/time.h>
  24
  25 #include "ldap-int.h"
  26
  27 static char ** decode_answer LDAP_P(( unsigned char *answer, ber_len_t len ));
  28
  29 #define MAX_TO_SORT     32
  30
  31
  32 /*
  33  * ldap_getdxbyname - lookup DNS DX records for domain and return an ordered
  34  *      array.
  35  */
  36 char **
  37 ldap_getdxbyname( const char *domain )
  38 {
  39     unsigned char       buf[ PACKETSZ ];
  40     char                **dxs;
  41     int                 rc;
  42
  43     Debug( LDAP_DEBUG_TRACE, "ldap_getdxbyname( %s )\n", domain, 0, 0 );
  44
  45     memset( buf, 0, sizeof( buf ));
  46
  47     if (( rc = res_search( domain, C_IN, T_TXT, buf, sizeof( buf ))) < 0
  48                 || ( dxs = decode_answer( buf, rc )) == NULL ) {
  49         /*
  50          * punt:  return list conisting of the original domain name only
  51          */
  52         if (( dxs = (char **)LDAP_MALLOC( 2 * sizeof( char * ))) == NULL ||
  53                 ( dxs[ 0 ] = LDAP_STRDUP( domain )) == NULL ) {
  54             if ( dxs != NULL ) {
  55                 LDAP_FREE( dxs );
  56             }
  57             dxs = NULL;
  58         } else {
  59             dxs[ 1 ] = NULL;
  60         }
  61     }
  62
  63     return( dxs );
  64 }
  65
  66
  67 static char **
  68 decode_answer( unsigned char *answer, ber_len_t len )
  69 {
  70     HEADER              *hp;
  71     char                buf[ 256 ], **dxs;
  72     unsigned char       *eom, *p;
  73     int                 ancount, err, rc, type, class, dx_count, rr_len;
  74     int                 dx_pref[ MAX_TO_SORT ];
  75
  76 #ifdef LDAP_DEBUG
  77 #ifdef notdef
  78     if ( ldap_debug & LDAP_DEBUG_PACKETS ) {
  79                 __p_query( answer );
  80     }
  81 #endif
  82 #endif /* LDAP_DEBUG */
  83
  84     dxs = NULL;
  85     hp = (HEADER *)answer;
  86     eom = answer + len;
  87
  88     if ( ntohs( hp->qdcount ) != 1 ) {
  89         h_errno = NO_RECOVERY;
  90         return( NULL );
  91     }
  92
  93     ancount = ntohs( hp->ancount );
  94     if ( ancount < 1 ) {
  95         h_errno = NO_DATA;
  96         return( NULL );
  97     }
  98
  99     /*
 100      * skip over the query
 101      */
 102     p = answer + HFIXEDSZ;
 103     if (( rc = dn_expand( answer, eom, p, buf, sizeof( buf ))) < 0 ) {
 104         h_errno = NO_RECOVERY;
 105         return( NULL );
 106     }
 107     p += ( rc + QFIXEDSZ );
 108
 109     /*
 110      * pull out the answers we are interested in
 111      */
 112     err = dx_count = 0;
 113     while ( ancount > 0 && err == 0 && p < eom ) {
 114         if (( rc = dn_expand( answer, eom, p, buf, sizeof( buf ))) < 0 ) {
 115             err = NO_RECOVERY;
 116             continue;
 117         }
 118         p += rc;        /* skip over name */
 119         type = _getshort( p );
 120         p += INT16SZ;
 121         class = _getshort( p );
 122         p += INT16SZ;
 123         p += INT32SZ;           /* skip over TTL */
 124         rr_len = _getshort( p );
 125         p += INT16SZ;
 126         if ( class == C_IN && type == T_TXT ) {
 127             int         pref, txt_len;
 128             char        *q, *r;
 129
 130             q = (char *)p;
 131             while ( q < (char *)p + rr_len && err == 0 ) {
 132                 if ( *q >= 3 && strncasecmp( q + 1, "dx:", 3 ) == 0 ) {
 133                     txt_len = *q - 3;
 134                     r = q + 4;
 135                     while ( isspace( (unsigned char) *r )) {
 136                         ++r;
 137                         --txt_len;
 138                     }
 139                     pref = 0;
 140                     while ( isdigit( (unsigned char) *r )) {
 141                         pref *= 10;
 142                         pref += ( *r - '0' );
 143                         ++r;
 144                         --txt_len;
 145                     }
 146                     if ( dx_count < MAX_TO_SORT - 1 ) {
 147                         dx_pref[ dx_count ] = pref;
 148                     }
 149                     while ( isspace( (unsigned char) *r )) {
 150                         ++r;
 151                         --txt_len;
 152                     }
 153                         dxs = (char **)LDAP_REALLOC( dxs,
 154                                 ( dx_count + 2 ) * sizeof( char * ));
 155                     if ( dxs == NULL || ( dxs[ dx_count ] =
 156                                 (char *)LDAP_CALLOC( 1, txt_len + 1 )) == NULL ) {
 157                         err = NO_RECOVERY;
 158                         continue;
 159                     }
 160                     memcpy( dxs[ dx_count ], r, txt_len );
 161                     dxs[ ++dx_count ] = NULL;
 162                 }
 163                 q += ( *q + 1 );        /* move past last TXT record */
 164             }
 165         }
 166         p += rr_len;
 167     }
 168
 169     if ( err == 0 ) {
 170         if ( dx_count == 0 ) {
 171             err = NO_DATA;
 172         } else {
 173             /*
 174              * sort records based on associated preference value
 175              */
 176             int         i, j, sort_count, tmp_pref;
 177             char        *tmp_dx;
 178
 179             sort_count = ( dx_count < MAX_TO_SORT ) ? dx_count : MAX_TO_SORT;
 180             for ( i = 0; i < sort_count; ++i ) {
 181                 for ( j = i + 1; j < sort_count; ++j ) {
 182                     if ( dx_pref[ i ] > dx_pref[ j ] ) {
 183                         tmp_pref = dx_pref[ i ];
 184                         dx_pref[ i ] = dx_pref[ j ];
 185                         dx_pref[ j ] = tmp_pref;
 186                         tmp_dx = dxs[ i ];
 187                         dxs[ i ] = dxs[ j ];
 188                         dxs[ j ] = tmp_dx;
 189                     }
 190                 }
 191             }
 192         }
 193     }
 194
 195     h_errno = err;
 196     return( dxs );
 197 }
 198
 199 #endif /* LDAP_API_FEATURE_X_OPENLDAP_V2_DNS */