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 char * 22 ldns_tsig_algorithm(ldns_tsig_credentials *tc) 23 { 24 return tc->algorithm; 25 } 26 27 char * 28 ldns_tsig_keyname(ldns_tsig_credentials *tc) 29 { 30 return tc->keyname; 31 } 32 33 char * 34 ldns_tsig_keydata(ldns_tsig_credentials *tc) 35 { 36 return tc->keydata; 37 } 38 39 char * 40 ldns_tsig_keyname_clone(ldns_tsig_credentials *tc) 41 { 42 return strdup(tc->keyname); 43 } 44 45 char * 46 ldns_tsig_keydata_clone(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 uint8_t * 55 ldns_tsig_prepare_pkt_wire(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 (strlen(name) == 12 138 && strncasecmp(name, "hmac-sha256.", 11) == 0) { 139 #ifdef HAVE_EVP_SHA256 140 return EVP_sha256(); 141 #else 142 return NULL; 143 #endif 144 } else if (strlen(name) == 10 145 && strncasecmp(name, "hmac-sha1.", 9) == 0) { 146 return EVP_sha1(); 147 } else if (strlen(name) == 25 148 && strncasecmp(name, "hmac-md5.sig-alg.reg.int.", 25) 149 == 0) { 150 return EVP_md5(); 151 } else { 152 return NULL; 153 } 154 } 155 #endif 156 157 #ifdef HAVE_SSL 158 static ldns_status 159 ldns_tsig_mac_new(ldns_rdf **tsig_mac, uint8_t *pkt_wire, size_t pkt_wire_size, 160 const char *key_data, ldns_rdf *key_name_rdf, ldns_rdf *fudge_rdf, 161 ldns_rdf *algorithm_rdf, ldns_rdf *time_signed_rdf, ldns_rdf *error_rdf, 162 ldns_rdf *other_data_rdf, ldns_rdf *orig_mac_rdf, int tsig_timers_only) 163 { 164 ldns_status status; 165 char *wireformat; 166 int wiresize; 167 unsigned char *mac_bytes = NULL; 168 unsigned char *key_bytes = NULL; 169 int key_size; 170 const EVP_MD *digester; 171 char *algorithm_name = NULL; 172 unsigned int md_len = EVP_MAX_MD_SIZE; 173 ldns_rdf *result = NULL; 174 ldns_buffer *data_buffer = NULL; 175 ldns_rdf *canonical_key_name_rdf = NULL; 176 ldns_rdf *canonical_algorithm_rdf = NULL; 177 178 if (key_name_rdf == NULL || algorithm_rdf == NULL) { 179 return LDNS_STATUS_NULL; 180 } 181 canonical_key_name_rdf = ldns_rdf_clone(key_name_rdf); 182 canonical_algorithm_rdf = ldns_rdf_clone(algorithm_rdf); 183 184 if (canonical_key_name_rdf == NULL 185 || canonical_algorithm_rdf == NULL) { 186 return LDNS_STATUS_MEM_ERR; 187 } 188 /* 189 * prepare the digestable information 190 */ 191 data_buffer = ldns_buffer_new(LDNS_MAX_PACKETLEN); 192 if (!data_buffer) { 193 status = LDNS_STATUS_MEM_ERR; 194 goto clean; 195 } 196 /* if orig_mac is not NULL, add it too */ 197 if (orig_mac_rdf) { 198 (void) ldns_rdf2buffer_wire(data_buffer, orig_mac_rdf); 199 } 200 ldns_buffer_write(data_buffer, pkt_wire, pkt_wire_size); 201 if (!tsig_timers_only) { 202 ldns_dname2canonical(canonical_key_name_rdf); 203 (void)ldns_rdf2buffer_wire(data_buffer, 204 canonical_key_name_rdf); 205 ldns_buffer_write_u16(data_buffer, LDNS_RR_CLASS_ANY); 206 ldns_buffer_write_u32(data_buffer, 0); 207 ldns_dname2canonical(canonical_algorithm_rdf); 208 (void)ldns_rdf2buffer_wire(data_buffer, 209 canonical_algorithm_rdf); 210 } 211 (void)ldns_rdf2buffer_wire(data_buffer, time_signed_rdf); 212 (void)ldns_rdf2buffer_wire(data_buffer, fudge_rdf); 213 if (!tsig_timers_only) { 214 (void)ldns_rdf2buffer_wire(data_buffer, error_rdf); 215 (void)ldns_rdf2buffer_wire(data_buffer, other_data_rdf); 216 } 217 218 wireformat = (char *) data_buffer->_data; 219 wiresize = (int) ldns_buffer_position(data_buffer); 220 221 algorithm_name = ldns_rdf2str(algorithm_rdf); 222 if(!algorithm_name) { 223 status = LDNS_STATUS_MEM_ERR; 224 goto clean; 225 } 226 227 /* prepare the key */ 228 key_bytes = LDNS_XMALLOC(unsigned char, 229 ldns_b64_pton_calculate_size(strlen(key_data))); 230 if(!key_bytes) { 231 status = LDNS_STATUS_MEM_ERR; 232 goto clean; 233 } 234 key_size = ldns_b64_pton(key_data, key_bytes, 235 ldns_b64_pton_calculate_size(strlen(key_data))); 236 if (key_size < 0) { 237 status = LDNS_STATUS_INVALID_B64; 238 goto clean; 239 } 240 /* hmac it */ 241 /* 2 spare bytes for the length */ 242 mac_bytes = LDNS_XMALLOC(unsigned char, md_len+2); 243 if(!mac_bytes) { 244 status = LDNS_STATUS_MEM_ERR; 245 goto clean; 246 } 247 memset(mac_bytes, 0, md_len+2); 248 249 digester = ldns_digest_function(algorithm_name); 250 251 if (digester) { 252 (void) HMAC(digester, key_bytes, key_size, (void *)wireformat, 253 (size_t) wiresize, mac_bytes + 2, &md_len); 254 255 ldns_write_uint16(mac_bytes, md_len); 256 result = ldns_rdf_new_frm_data(LDNS_RDF_TYPE_INT16_DATA, md_len + 2, 257 mac_bytes); 258 } else { 259 status = LDNS_STATUS_CRYPTO_UNKNOWN_ALGO; 260 goto clean; 261 } 262 *tsig_mac = result; 263 status = LDNS_STATUS_OK; 264 clean: 265 LDNS_FREE(mac_bytes); 266 LDNS_FREE(key_bytes); 267 LDNS_FREE(algorithm_name); 268 ldns_buffer_free(data_buffer); 269 ldns_rdf_free(canonical_algorithm_rdf); 270 ldns_rdf_free(canonical_key_name_rdf); 271 return status; 272 } 273 #endif /* HAVE_SSL */ 274 275 276 #ifdef HAVE_SSL 277 bool 278 ldns_pkt_tsig_verify(ldns_pkt *pkt, uint8_t *wire, size_t wirelen, const char *key_name, 279 const char *key_data, ldns_rdf *orig_mac_rdf) 280 { 281 return ldns_pkt_tsig_verify_next(pkt, wire, wirelen, key_name, key_data, orig_mac_rdf, 0); 282 } 283 284 bool 285 ldns_pkt_tsig_verify_next(ldns_pkt *pkt, uint8_t *wire, size_t wirelen, const char* key_name, 286 const char *key_data, ldns_rdf *orig_mac_rdf, int tsig_timers_only) 287 { 288 ldns_rdf *fudge_rdf; 289 ldns_rdf *algorithm_rdf; 290 ldns_rdf *time_signed_rdf; 291 ldns_rdf *orig_id_rdf; 292 ldns_rdf *error_rdf; 293 ldns_rdf *other_data_rdf; 294 ldns_rdf *pkt_mac_rdf; 295 ldns_rdf *my_mac_rdf; 296 ldns_rdf *key_name_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, key_name); 297 uint16_t pkt_id, orig_pkt_id; 298 ldns_status status; 299 300 uint8_t *prepared_wire = NULL; 301 size_t prepared_wire_size = 0; 302 303 ldns_rr *orig_tsig = ldns_pkt_tsig(pkt); 304 305 if (!orig_tsig || ldns_rr_rd_count(orig_tsig) <= 6) { 306 ldns_rdf_deep_free(key_name_rdf); 307 return false; 308 } 309 algorithm_rdf = ldns_rr_rdf(orig_tsig, 0); 310 time_signed_rdf = ldns_rr_rdf(orig_tsig, 1); 311 fudge_rdf = ldns_rr_rdf(orig_tsig, 2); 312 pkt_mac_rdf = ldns_rr_rdf(orig_tsig, 3); 313 orig_id_rdf = ldns_rr_rdf(orig_tsig, 4); 314 error_rdf = ldns_rr_rdf(orig_tsig, 5); 315 other_data_rdf = ldns_rr_rdf(orig_tsig, 6); 316 317 /* remove temporarily */ 318 ldns_pkt_set_tsig(pkt, NULL); 319 /* temporarily change the id to the original id */ 320 pkt_id = ldns_pkt_id(pkt); 321 orig_pkt_id = ldns_rdf2native_int16(orig_id_rdf); 322 ldns_pkt_set_id(pkt, orig_pkt_id); 323 324 prepared_wire = ldns_tsig_prepare_pkt_wire(wire, wirelen, &prepared_wire_size); 325 326 status = ldns_tsig_mac_new(&my_mac_rdf, prepared_wire, prepared_wire_size, 327 key_data, key_name_rdf, fudge_rdf, algorithm_rdf, 328 time_signed_rdf, error_rdf, other_data_rdf, orig_mac_rdf, tsig_timers_only); 329 330 LDNS_FREE(prepared_wire); 331 332 if (status != LDNS_STATUS_OK) { 333 ldns_rdf_deep_free(key_name_rdf); 334 return false; 335 } 336 /* Put back the values */ 337 ldns_pkt_set_tsig(pkt, orig_tsig); 338 ldns_pkt_set_id(pkt, pkt_id); 339 340 ldns_rdf_deep_free(key_name_rdf); 341 342 if (ldns_rdf_compare(pkt_mac_rdf, my_mac_rdf) == 0) { 343 ldns_rdf_deep_free(my_mac_rdf); 344 return true; 345 } else { 346 ldns_rdf_deep_free(my_mac_rdf); 347 return false; 348 } 349 } 350 #endif /* HAVE_SSL */ 351 352 #ifdef HAVE_SSL 353 ldns_status 354 ldns_pkt_tsig_sign(ldns_pkt *pkt, const char *key_name, const char *key_data, 355 uint16_t fudge, const char *algorithm_name, ldns_rdf *query_mac) 356 { 357 return ldns_pkt_tsig_sign_next(pkt, key_name, key_data, fudge, algorithm_name, query_mac, 0); 358 } 359 360 ldns_status 361 ldns_pkt_tsig_sign_next(ldns_pkt *pkt, const char *key_name, const char *key_data, 362 uint16_t fudge, const char *algorithm_name, ldns_rdf *query_mac, int tsig_timers_only) 363 { 364 ldns_rr *tsig_rr; 365 ldns_rdf *key_name_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, key_name); 366 ldns_rdf *fudge_rdf = NULL; 367 ldns_rdf *orig_id_rdf = NULL; 368 ldns_rdf *algorithm_rdf; 369 ldns_rdf *error_rdf = NULL; 370 ldns_rdf *mac_rdf = NULL; 371 ldns_rdf *other_data_rdf = NULL; 372 373 ldns_status status = LDNS_STATUS_OK; 374 375 uint8_t *pkt_wire = NULL; 376 size_t pkt_wire_len; 377 378 struct timeval tv_time_signed; 379 uint8_t *time_signed = NULL; 380 ldns_rdf *time_signed_rdf = NULL; 381 382 algorithm_rdf = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_DNAME, algorithm_name); 383 if(!key_name_rdf || !algorithm_rdf) { 384 status = LDNS_STATUS_MEM_ERR; 385 goto clean; 386 } 387 388 /* eww don't have create tsigtime rdf yet :( */ 389 /* bleh :p */ 390 if (gettimeofday(&tv_time_signed, NULL) == 0) { 391 time_signed = LDNS_XMALLOC(uint8_t, 6); 392 if(!time_signed) { 393 status = LDNS_STATUS_MEM_ERR; 394 goto clean; 395 } 396 ldns_write_uint64_as_uint48(time_signed, 397 (uint64_t)tv_time_signed.tv_sec); 398 } else { 399 status = LDNS_STATUS_INTERNAL_ERR; 400 goto clean; 401 } 402 403 time_signed_rdf = ldns_rdf_new(LDNS_RDF_TYPE_TSIGTIME, 6, time_signed); 404 if(!time_signed_rdf) { 405 LDNS_FREE(time_signed); 406 status = LDNS_STATUS_MEM_ERR; 407 goto clean; 408 } 409 410 fudge_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, fudge); 411 412 orig_id_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, ldns_pkt_id(pkt)); 413 414 error_rdf = ldns_native2rdf_int16(LDNS_RDF_TYPE_INT16, 0); 415 416 other_data_rdf = ldns_native2rdf_int16_data(0, NULL); 417 418 if(!fudge_rdf || !orig_id_rdf || !error_rdf || !other_data_rdf) { 419 status = LDNS_STATUS_MEM_ERR; 420 goto clean; 421 } 422 423 if (ldns_pkt2wire(&pkt_wire, pkt, &pkt_wire_len) != LDNS_STATUS_OK) { 424 status = LDNS_STATUS_ERR; 425 goto clean; 426 } 427 428 status = ldns_tsig_mac_new(&mac_rdf, pkt_wire, pkt_wire_len, 429 key_data, key_name_rdf, fudge_rdf, algorithm_rdf, 430 time_signed_rdf, error_rdf, other_data_rdf, query_mac, tsig_timers_only); 431 432 if (!mac_rdf) { 433 goto clean; 434 } 435 436 LDNS_FREE(pkt_wire); 437 438 /* Create the TSIG RR */ 439 tsig_rr = ldns_rr_new(); 440 if(!tsig_rr) { 441 status = LDNS_STATUS_MEM_ERR; 442 goto clean; 443 } 444 ldns_rr_set_owner(tsig_rr, key_name_rdf); 445 ldns_rr_set_class(tsig_rr, LDNS_RR_CLASS_ANY); 446 ldns_rr_set_type(tsig_rr, LDNS_RR_TYPE_TSIG); 447 ldns_rr_set_ttl(tsig_rr, 0); 448 449 ldns_rr_push_rdf(tsig_rr, algorithm_rdf); 450 ldns_rr_push_rdf(tsig_rr, time_signed_rdf); 451 ldns_rr_push_rdf(tsig_rr, fudge_rdf); 452 ldns_rr_push_rdf(tsig_rr, mac_rdf); 453 ldns_rr_push_rdf(tsig_rr, orig_id_rdf); 454 ldns_rr_push_rdf(tsig_rr, error_rdf); 455 ldns_rr_push_rdf(tsig_rr, other_data_rdf); 456 457 ldns_pkt_set_tsig(pkt, tsig_rr); 458 459 return status; 460 461 clean: 462 LDNS_FREE(pkt_wire); 463 ldns_rdf_free(key_name_rdf); 464 ldns_rdf_free(algorithm_rdf); 465 ldns_rdf_free(time_signed_rdf); 466 ldns_rdf_free(fudge_rdf); 467 ldns_rdf_free(orig_id_rdf); 468 ldns_rdf_free(error_rdf); 469 ldns_rdf_free(other_data_rdf); 470 return status; 471 } 472 #endif /* HAVE_SSL */ 473