contrib/ldns/tsig.c

   1 /*
   2  * tsig.c
   3  *
   4  * contains the functions needed for TSIG [RFC2845]
   5  *
   6  * (c) 2005-2006 NLnet Labs
   7  * See the file LICENSE for the license
   8  */
   9
  10 #include <ldns/config.h>
  11
  12 #include <ldns/ldns.h>
  13
  14 #include <strings.h>
  15
  16 #ifdef HAVE_SSL
  17 #include <openssl/hmac.h>
  18 #include <openssl/md5.h>
  19 #endif /* HAVE_SSL */
  20
  21 const char *
  22 ldns_tsig_algorithm(const ldns_tsig_credentials *tc)
  23 {
  24         return tc->algorithm;
  25 }
  26
  27 const char *
  28 ldns_tsig_keyname(const ldns_tsig_credentials *tc)
  29 {
  30         return tc->keyname;
  31 }
  32
  33 const char *
  34 ldns_tsig_keydata(const ldns_tsig_credentials *tc)
  35 {
  36         return tc->keydata;
  37 }
  38
  39 char *
  40 ldns_tsig_keyname_clone(const ldns_tsig_credentials *tc)
  41 {
  42         return strdup(tc->keyname);
  43 }
  44
  45 char *
  46 ldns_tsig_keydata_clone(const ldns_tsig_credentials *tc)
  47 {
  48         return strdup(tc->keydata);
  49 }
  50
  51 /*
  52  *  Makes an exact copy of the wire, but with the tsig rr removed
  53  */
  54 static uint8_t *
  55 ldns_tsig_prepare_pkt_wire(const uint8_t *wire, size_t wire_len, size_t *result_len)
  56 {
  57         uint8_t *wire2 = NULL;
  58         uint16_t qd_count;
  59         uint16_t an_count;
  60         uint16_t ns_count;
  61         uint16_t ar_count;
  62         ldns_rr *rr;
  63
  64         size_t pos;
  65         uint16_t i;
  66
  67         ldns_status status;
  68
  69         if(wire_len < LDNS_HEADER_SIZE) {
  70                 return NULL;
  71         }
  72         /* fake parse the wire */
  73         qd_count = LDNS_QDCOUNT(wire);
  74         an_count = LDNS_ANCOUNT(wire);
  75         ns_count = LDNS_NSCOUNT(wire);
  76         ar_count = LDNS_ARCOUNT(wire);
  77
  78         if (ar_count > 0) {
  79                 ar_count--;
  80         } else {
  81                 return NULL;
  82         }
  83
  84         pos = LDNS_HEADER_SIZE;
  85
  86         for (i = 0; i < qd_count; i++) {
  87                 status = ldns_wire2rr(&rr, wire, wire_len, &pos, LDNS_SECTION_QUESTION);
  88                 if (status != LDNS_STATUS_OK) {
  89                         return NULL;
  90                 }
  91                 ldns_rr_free(rr);
  92         }
  93
  94         for (i = 0; i < an_count; i++) {
  95                 status = ldns_wire2rr(&rr, wire, wire_len, &pos, LDNS_SECTION_ANSWER);
  96                 if (status != LDNS_STATUS_OK) {
  97                         return NULL;
  98                 }
  99                 ldns_rr_free(rr);
 100         }
 101
 102         for (i = 0; i < ns_count; i++) {
 103                 status = ldns_wire2rr(&rr, wire, wire_len, &pos, LDNS_SECTION_AUTHORITY);
 104                 if (status != LDNS_STATUS_OK) {
 105                         return NULL;
 106                 }
 107                 ldns_rr_free(rr);
 108         }
 109
 110         for (i = 0; i < ar_count; i++) {
 111                 status = ldns_wire2rr(&rr, wire, wire_len, &pos,
 112                                 LDNS_SECTION_ADDITIONAL);
 113                 if (status != LDNS_STATUS_OK) {
 114                         return NULL;
 115                 }
 116                 ldns_rr_free(rr);
 117         }
 118
 119         *result_len = pos;
 120         wire2 = LDNS_XMALLOC(uint8_t, *result_len);
 121         if(!wire2) {
 122                 return NULL;
 123         }
 124         memcpy(wire2, wire, *result_len);
 125
 126         ldns_write_uint16(wire2 + LDNS_ARCOUNT_OFF, ar_count);
 127
 128         return wire2;
 129 }
 130
 131 #ifdef HAVE_SSL
 132 static const EVP_MD *
 133 ldns_digest_function(char *name)
 134 {
 135         /* these are the mandatory algorithms from RFC4635 */
 136         /* The optional algorithms are not yet implemented */
 137         if (strcasecmp(name, "hmac-sha512.") == 0) {
 138 #ifdef HAVE_EVP_SHA512
 139                 return EVP_sha512();
 140 #else
 141                 return NULL;
 142 #endif
 143         } else if (strcasecmp(name, "hmac-shac384.") == 0) {
 144 #ifdef HAVE_EVP_SHA384
 145                 return EVP_sha384();
 146 #else
 147                 return NULL;
 148 #endif
 149         } else if (strcasecmp(name, "hmac-sha256.") == 0) {
 150 #ifdef HAVE_EVP_SHA256
 151                 return EVP_sha256();
 152 #else
 153                 return NULL;
 154 #endif
 155         } else if (strcasecmp(name, "hmac-sha1.") == 0) {
 156                 return EVP_sha1();
 157         } else if (strcasecmp(name, "hmac-md5.sig-alg.reg.int.") == 0) {
 158                 return EVP_md5();
 159         } else {
 160                 return NULL;
 161         }
 162 }
 163 #endif
 164
 165 #ifdef HAVE_SSL
 166 static ldns_status
 167 ldns_tsig_mac_new(ldns_rdf **tsig_mac, const uint8_t *pkt_wire, size_t pkt_wire_size,
 168                 const char *key_data, const ldns_rdf *key_name_rdf, const ldns_rdf *fudge_rdf,
 169                 const ldns_rdf *algorithm_rdf, const ldns_rdf *time_signed_rdf, const ldns_rdf *error_rdf,
 170                 const ldns_rdf *other_data_rdf, const ldns_rdf *orig_mac_rdf, int tsig_timers_only)
 171 {
 172         ldns_status status;
 173         char *wireformat;
 174         int wiresize;
 175         unsigned char *mac_bytes = NULL;
 176         unsigned char *key_bytes = NULL;
 177         int key_size;
 178         const EVP_MD *digester;
 179         char *algorithm_name = NULL;
 180         unsigned int md_len = EVP_MAX_MD_SIZE;
 181         ldns_rdf *result = NULL;
 182         ldns_buffer *data_buffer = NULL;
 183         ldns_rdf *canonical_key_name_rdf = NULL;
 184         ldns_rdf *canonical_algorithm_rdf = NULL;
 185
 186         if (key_name_rdf == NULL || algorithm_rdf == NULL) {
 187                 return LDNS_STATUS_NULL;
 188         }
 189         canonical_key_name_rdf  = ldns_rdf_clone(key_name_rdf);
 190         if (canonical_key_name_rdf == NULL) {
 191                 return LDNS_STATUS_MEM_ERR;
 192         }
 193         canonical_algorithm_rdf = ldns_rdf_clone(algorithm_rdf);
 194         if (canonical_algorithm_rdf == NULL) {
 195                 ldns_rdf_deep_free(canonical_key_name_rdf);
 196                 return LDNS_STATUS_MEM_ERR;
 197         }
 198         /*
 199          * prepare the digestable information
 200          */
 201         data_buffer = ldns_buffer_new(LDNS_MAX_PACKETLEN);
 202         if (!data_buffer) {
 203                 status = LDNS_STATUS_MEM_ERR;
 204                 goto clean;
 205         }
 206         /* if orig_mac is not NULL, add it too */
 207         if (orig_mac_rdf) {
 208                 (void) ldns_rdf2buffer_wire(data_buffer, orig_mac_rdf);
 209         }
 210         ldns_buffer_write(data_buffer, pkt_wire, pkt_wire_size);
 211         if (!tsig_timers_only) {
 212                 ldns_dname2canonical(canonical_key_name_rdf);
 213                 (void)ldns_rdf2buffer_wire(data_buffer,
 214                                 canonical_key_name_rdf);
 215                 ldns_buffer_write_u16(data_buffer, LDNS_RR_CLASS_ANY);
 216                 ldns_buffer_write_u32(data_buffer, 0);
 217                 ldns_dname2canonical(canonical_algorithm_rdf);
 218                 (void)ldns_rdf2buffer_wire(data_buffer,
 219                                 canonical_algorithm_rdf);
 220         }
 221         (void)ldns_rdf2buffer_wire(data_buffer, time_signed_rdf);
 222         (void)ldns_rdf2buffer_wire(data_buffer, fudge_rdf);
 223         if (!tsig_timers_only) {
 224                 (void)ldns_rdf2buffer_wire(data_buffer, error_rdf);
 225                 (void)ldns_rdf2buffer_wire(data_buffer, other_data_rdf);
 226         }
 227
 228         wireformat = (char *) data_buffer->_data;
 229         wiresize = (int) ldns_buffer_position(data_buffer);
 230
 231         algorithm_name = ldns_rdf2str(algorithm_rdf);
 232         if(!algorithm_name) {
 233                 status = LDNS_STATUS_MEM_ERR;
 234                 goto clean;
 235         }
 236
 237         /* prepare the key */
 238         key_bytes = LDNS_XMALLOC(unsigned char,
 239                         ldns_b64_pton_calculate_size(strlen(key_data)));
 240         if(!key_bytes) {
 241                 status = LDNS_STATUS_MEM_ERR;
 242                 goto clean;
 243         }
 244         key_size = ldns_b64_pton(key_data, key_bytes,
 245         ldns_b64_pton_calculate_size(strlen(key_data)));
 246         if (key_size < 0) {
 247                 status = LDNS_STATUS_INVALID_B64;
 248                 goto clean;
 249         }
 250         /* hmac it */
 251         /* 2 spare bytes for the length */
 252         mac_bytes = LDNS_XMALLOC(unsigned char, md_len+2);
 253         if(!mac_bytes) {
 254                 status = LDNS_STATUS_MEM_ERR;
 255                 goto clean;
 256         }
 257         memset(mac_bytes, 0, md_len+2);
 258
 259         digester = ldns_digest_function(algorithm_name);
 260
 261         if (digester) {
 262                 (void) HMAC(digester, key_bytes, key_size, (void *)wireformat,
 263                             (size_t) wiresize, mac_bytes + 2, &md_len);
 264
 265                 ldns_write_uint16(mac_bytes, md_len);
 266                 result = ldns_rdf_new_frm_data(LDNS_RDF_TYPE_INT16_DATA, md_len + 2,
 267                                 mac_bytes);
 268         } else {
 269                 status = LDNS_STATUS_CRYPTO_UNKNOWN_ALGO;
 270                 goto clean;
 271         }
 272         *tsig_mac = result;
 273         status = LDNS_STATUS_OK;
 274   clean:
 275         LDNS_FREE(mac_bytes);
 276         LDNS_FREE(key_bytes);
 277         LDNS_FREE(algorithm_name);
 278         ldns_buffer_free(data_buffer);
 279         ldns_rdf_deep_free(canonical_algorithm_rdf);
 280         ldns_rdf_deep_free(canonical_key_name_rdf);
 281         return status;
 282 }
 283 #endif /*  HAVE_SSL */
 284
 285
 286 #ifdef HAVE_SSL
 287 bool
 288 ldns_pkt_tsig_verify(ldns_pkt *pkt, const uint8_t *wire, size_t wirelen, const char *key_name,
 289         const char *key_data, const ldns_rdf *orig_mac_rdf)
 290 {
 291         return ldns_pkt_tsig_verify_next(pkt, wire, wirelen, key_name, key_data, orig_mac_rdf, 0);
 292 }
 293
 294 bool
 295 ldns_pkt_tsig_verify_next(ldns_pkt *pkt, const uint8_t *wire, size_t wirelen, const char* key_name,
 296         const char *key_data, const ldns_rdf *orig_mac_rdf, int tsig_timers_only)
 297 {
 298         ldns_rdf *fudge_rdf;
 299         ldns_rdf *algorithm_rdf;
 300         ldns_rdf *time_signed_rdf;
 301         ldns_rdf *orig_id_rdf;
 302         ldns_rdf *error_rdf;
 303         ldns_rdf *other_data_rdf;
 304         ldns_rdf *pkt_mac_rdf;
 305         ldns_rdf *my_mac_rdf;
 306         ldns_rdf *key_name_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, key_name);
 307         uint16_t pkt_id, orig_pkt_id;
 308         ldns_status status;
 309
 310         uint8_t *prepared_wire = NULL;
 311         size_t prepared_wire_size = 0;
 312
 313         ldns_rr *orig_tsig = ldns_pkt_tsig(pkt);
 314
 315         if (!orig_tsig || ldns_rr_rd_count(orig_tsig) <= 6) {
 316                 ldns_rdf_deep_free(key_name_rdf);
 317                 return false;
 318         }
 319         algorithm_rdf = ldns_rr_rdf(orig_tsig, 0);
 320         time_signed_rdf = ldns_rr_rdf(orig_tsig, 1);
 321         fudge_rdf = ldns_rr_rdf(orig_tsig, 2);
 322         pkt_mac_rdf = ldns_rr_rdf(orig_tsig, 3);
 323         orig_id_rdf = ldns_rr_rdf(orig_tsig, 4);
 324         error_rdf = ldns_rr_rdf(orig_tsig, 5);
 325         other_data_rdf = ldns_rr_rdf(orig_tsig, 6);
 326
 327         /* remove temporarily */
 328         ldns_pkt_set_tsig(pkt, NULL);
 329         /* temporarily change the id to the original id */
 330         pkt_id = ldns_pkt_id(pkt);
 331         orig_pkt_id = ldns_rdf2native_int16(orig_id_rdf);
 332         ldns_pkt_set_id(pkt, orig_pkt_id);
 333
 334         prepared_wire = ldns_tsig_prepare_pkt_wire(wire, wirelen, &prepared_wire_size);
 335
 336         status = ldns_tsig_mac_new(&my_mac_rdf, prepared_wire, prepared_wire_size,
 337                         key_data, key_name_rdf, fudge_rdf, algorithm_rdf,
 338                         time_signed_rdf, error_rdf, other_data_rdf, orig_mac_rdf, tsig_timers_only);
 339
 340         LDNS_FREE(prepared_wire);
 341
 342         if (status != LDNS_STATUS_OK) {
 343                 ldns_rdf_deep_free(key_name_rdf);
 344                 return false;
 345         }
 346         /* Put back the values */
 347         ldns_pkt_set_tsig(pkt, orig_tsig);
 348         ldns_pkt_set_id(pkt, pkt_id);
 349
 350         ldns_rdf_deep_free(key_name_rdf);
 351
 352         if (ldns_rdf_compare(pkt_mac_rdf, my_mac_rdf) == 0) {
 353                 ldns_rdf_deep_free(my_mac_rdf);
 354                 return true;
 355         } else {
 356                 ldns_rdf_deep_free(my_mac_rdf);
 357                 return false;
 358         }
 359 }
 360 #endif /* HAVE_SSL */
 361
 362 #ifdef HAVE_SSL
 363 ldns_status
 364 ldns_pkt_tsig_sign(ldns_pkt *pkt, const char *key_name, const char *key_data,
 365         uint16_t fudge, const char *algorithm_name, const ldns_rdf *query_mac)
 366 {
 367         return ldns_pkt_tsig_sign_next(pkt, key_name, key_data, fudge, algorithm_name, query_mac, 0);
 368 }
 369
 370 ldns_status
 371 ldns_pkt_tsig_sign_next(ldns_pkt *pkt, const char *key_name, const char *key_data,
 372         uint16_t fudge, const char *algorithm_name, const ldns_rdf *query_mac, int tsig_timers_only)
 373 {
 374         ldns_rr *tsig_rr;
 375         ldns_rdf *key_name_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, key_name);
 376         ldns_rdf *fudge_rdf = NULL;
 377         ldns_rdf *orig_id_rdf = NULL;
 378         ldns_rdf *algorithm_rdf;
 379         ldns_rdf *error_rdf = NULL;
 380         ldns_rdf *mac_rdf = NULL;
 381         ldns_rdf *other_data_rdf = NULL;
 382
 383         ldns_status status = LDNS_STATUS_OK;
 384
 385         uint8_t *pkt_wire = NULL;
 386         size_t pkt_wire_len;
 387
 388         struct timeval tv_time_signed;
 389         uint8_t *time_signed = NULL;
 390         ldns_rdf *time_signed_rdf = NULL;
 391
 392         algorithm_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, algorithm_name);
 393         if(!key_name_rdf || !algorithm_rdf) {
 394                 status = LDNS_STATUS_MEM_ERR;
 395                 goto clean;
 396         }
 397
 398         /* eww don't have create tsigtime rdf yet :( */
 399         /* bleh :p */
 400         if (gettimeofday(&tv_time_signed, NULL) == 0) {
 401                 time_signed = LDNS_XMALLOC(uint8_t, 6);
 402                 if(!time_signed) {
 403                         status = LDNS_STATUS_MEM_ERR;
 404                         goto clean;
 405                 }
 406                 ldns_write_uint64_as_uint48(time_signed,
 407                                 (uint64_t)tv_time_signed.tv_sec);
 408         } else {
 409                 status = LDNS_STATUS_INTERNAL_ERR;
 410                 goto clean;
 411         }
 412
 413         time_signed_rdf = ldns_rdf_new(LDNS_RDF_TYPE_TSIGTIME, 6, time_signed);
 414         if(!time_signed_rdf) {
 415                 LDNS_FREE(time_signed);
 416                 status = LDNS_STATUS_MEM_ERR;
 417                 goto clean;
 418         }
 419
 420         fudge_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, fudge);
 421
 422         orig_id_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, ldns_pkt_id(pkt));
 423
 424         error_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, 0);
 425
 426         other_data_rdf = ldns_native2rdf_int16_data(0, NULL);
 427
 428         if(!fudge_rdf || !orig_id_rdf || !error_rdf || !other_data_rdf) {
 429                 status = LDNS_STATUS_MEM_ERR;
 430                 goto clean;
 431         }
 432
 433         if (ldns_pkt2wire(&pkt_wire, pkt, &pkt_wire_len) != LDNS_STATUS_OK) {
 434                 status = LDNS_STATUS_ERR;
 435                 goto clean;
 436         }
 437
 438         status = ldns_tsig_mac_new(&mac_rdf, pkt_wire, pkt_wire_len,
 439                         key_data, key_name_rdf, fudge_rdf, algorithm_rdf,
 440                         time_signed_rdf, error_rdf, other_data_rdf, query_mac, tsig_timers_only);
 441
 442         if (!mac_rdf) {
 443                 goto clean;
 444         }
 445
 446         LDNS_FREE(pkt_wire);
 447
 448         /* Create the TSIG RR */
 449         tsig_rr = ldns_rr_new();
 450         if(!tsig_rr) {
 451                 status = LDNS_STATUS_MEM_ERR;
 452                 goto clean;
 453         }
 454         ldns_rr_set_owner(tsig_rr, key_name_rdf);
 455         ldns_rr_set_class(tsig_rr, LDNS_RR_CLASS_ANY);
 456         ldns_rr_set_type(tsig_rr, LDNS_RR_TYPE_TSIG);
 457         ldns_rr_set_ttl(tsig_rr, 0);
 458
 459         ldns_rr_push_rdf(tsig_rr, algorithm_rdf);
 460         ldns_rr_push_rdf(tsig_rr, time_signed_rdf);
 461         ldns_rr_push_rdf(tsig_rr, fudge_rdf);
 462         ldns_rr_push_rdf(tsig_rr, mac_rdf);
 463         ldns_rr_push_rdf(tsig_rr, orig_id_rdf);
 464         ldns_rr_push_rdf(tsig_rr, error_rdf);
 465         ldns_rr_push_rdf(tsig_rr, other_data_rdf);
 466
 467         ldns_pkt_set_tsig(pkt, tsig_rr);
 468
 469         return status;
 470
 471   clean:
 472         LDNS_FREE(pkt_wire);
 473         ldns_rdf_free(key_name_rdf);
 474         ldns_rdf_free(algorithm_rdf);
 475         ldns_rdf_free(time_signed_rdf);
 476         ldns_rdf_free(fudge_rdf);
 477         ldns_rdf_free(orig_id_rdf);
 478         ldns_rdf_free(error_rdf);
 479         ldns_rdf_free(other_data_rdf);
 480         return status;
 481 }
 482 #endif /* HAVE_SSL */