1 /* $NetBSD: journal.c,v 1.5 2020/05/24 19:46:23 christos Exp $ */ 2 3 /* 4 * Copyright (C) Internet Systems Consortium, Inc. ("ISC") 5 * 6 * This Source Code Form is subject to the terms of the Mozilla Public 7 * License, v. 2.0. If a copy of the MPL was not distributed with this 8 * file, You can obtain one at http://mozilla.org/MPL/2.0/. 9 * 10 * See the COPYRIGHT file distributed with this work for additional 11 * information regarding copyright ownership. 12 */ 13 14 #include <errno.h> 15 #include <inttypes.h> 16 #include <stdbool.h> 17 #include <stdlib.h> 18 #include <unistd.h> 19 20 #include <isc/file.h> 21 #include <isc/mem.h> 22 #include <isc/print.h> 23 #include <isc/stdio.h> 24 #include <isc/string.h> 25 #include <isc/util.h> 26 27 #include <dns/compress.h> 28 #include <dns/db.h> 29 #include <dns/dbiterator.h> 30 #include <dns/diff.h> 31 #include <dns/fixedname.h> 32 #include <dns/journal.h> 33 #include <dns/log.h> 34 #include <dns/rdataset.h> 35 #include <dns/rdatasetiter.h> 36 #include <dns/result.h> 37 #include <dns/soa.h> 38 39 /*! \file 40 * \brief Journaling. 41 * 42 * A journal file consists of 43 * 44 * \li A fixed-size header of type journal_rawheader_t. 45 * 46 * \li The index. This is an unordered array of index entries 47 * of type journal_rawpos_t giving the locations 48 * of some arbitrary subset of the journal's addressable 49 * transactions. The index entries are used as hints to 50 * speed up the process of locating a transaction with a given 51 * serial number. Unused index entries have an "offset" 52 * field of zero. The size of the index can vary between 53 * journal files, but does not change during the lifetime 54 * of a file. The size can be zero. 55 * 56 * \li The journal data. This consists of one or more transactions. 57 * Each transaction begins with a transaction header of type 58 * journal_rawxhdr_t. The transaction header is followed by a 59 * sequence of RRs, similar in structure to an IXFR difference 60 * sequence (RFC1995). That is, the pre-transaction SOA, 61 * zero or more other deleted RRs, the post-transaction SOA, 62 * and zero or more other added RRs. Unlike in IXFR, each RR 63 * is prefixed with a 32-bit length. 64 * 65 * The journal data part grows as new transactions are 66 * appended to the file. Only those transactions 67 * whose serial number is current-(2^31-1) to current 68 * are considered "addressable" and may be pointed 69 * to from the header or index. They may be preceded 70 * by old transactions that are no longer addressable, 71 * and they may be followed by transactions that were 72 * appended to the journal but never committed by updating 73 * the "end" position in the header. The latter will 74 * be overwritten when new transactions are added. 75 */ 76 77 /**************************************************************************/ 78 /* 79 * Miscellaneous utilities. 80 */ 81 82 #define JOURNAL_COMMON_LOGARGS \ 83 dns_lctx, DNS_LOGCATEGORY_GENERAL, DNS_LOGMODULE_JOURNAL 84 85 #define JOURNAL_DEBUG_LOGARGS(n) JOURNAL_COMMON_LOGARGS, ISC_LOG_DEBUG(n) 86 87 /*% 88 * It would be non-sensical (or at least obtuse) to use FAIL() with an 89 * ISC_R_SUCCESS code, but the test is there to keep the Solaris compiler 90 * from complaining about "end-of-loop code not reached". 91 */ 92 #define FAIL(code) \ 93 do { \ 94 result = (code); \ 95 if (result != ISC_R_SUCCESS) \ 96 goto failure; \ 97 } while (/*CONSTCOND*/0) 98 99 #define CHECK(op) \ 100 do { \ 101 result = (op); \ 102 if (result != ISC_R_SUCCESS) \ 103 goto failure; \ 104 } while (/*CONSTCOND*/0) 105 106 #define JOURNAL_SERIALSET 0x01U 107 108 static isc_result_t 109 index_to_disk(dns_journal_t *); 110 111 static inline uint32_t 112 decode_uint32(unsigned char *p) { 113 return ((p[0] << 24) + (p[1] << 16) + (p[2] << 8) + (p[3] << 0)); 114 } 115 116 static inline void 117 encode_uint32(uint32_t val, unsigned char *p) { 118 p[0] = (uint8_t)(val >> 24); 119 p[1] = (uint8_t)(val >> 16); 120 p[2] = (uint8_t)(val >> 8); 121 p[3] = (uint8_t)(val >> 0); 122 } 123 124 isc_result_t 125 dns_db_createsoatuple(dns_db_t *db, dns_dbversion_t *ver, isc_mem_t *mctx, 126 dns_diffop_t op, dns_difftuple_t **tp) { 127 isc_result_t result; 128 dns_dbnode_t *node; 129 dns_rdataset_t rdataset; 130 dns_rdata_t rdata = DNS_RDATA_INIT; 131 dns_fixedname_t fixed; 132 dns_name_t *zonename; 133 134 zonename = dns_fixedname_initname(&fixed); 135 dns_name_copynf(dns_db_origin(db), zonename); 136 137 node = NULL; 138 result = dns_db_findnode(db, zonename, false, &node); 139 if (result != ISC_R_SUCCESS) { 140 goto nonode; 141 } 142 143 dns_rdataset_init(&rdataset); 144 result = dns_db_findrdataset(db, node, ver, dns_rdatatype_soa, 0, 145 (isc_stdtime_t)0, &rdataset, NULL); 146 if (result != ISC_R_SUCCESS) { 147 goto freenode; 148 } 149 150 result = dns_rdataset_first(&rdataset); 151 if (result != ISC_R_SUCCESS) { 152 goto freenode; 153 } 154 155 dns_rdataset_current(&rdataset, &rdata); 156 dns_rdataset_getownercase(&rdataset, zonename); 157 158 result = dns_difftuple_create(mctx, op, zonename, rdataset.ttl, &rdata, 159 tp); 160 161 dns_rdataset_disassociate(&rdataset); 162 dns_db_detachnode(db, &node); 163 return (result); 164 165 freenode: 166 dns_db_detachnode(db, &node); 167 nonode: 168 UNEXPECTED_ERROR(__FILE__, __LINE__, "missing SOA"); 169 return (result); 170 } 171 172 /* Journaling */ 173 174 /*% 175 * On-disk representation of a "pointer" to a journal entry. 176 * These are used in the journal header to locate the beginning 177 * and end of the journal, and in the journal index to locate 178 * other transactions. 179 */ 180 typedef struct { 181 unsigned char serial[4]; /*%< SOA serial before update. */ 182 /* 183 * XXXRTH Should offset be 8 bytes? 184 * XXXDCL ... probably, since isc_offset_t is 8 bytes on many OSs. 185 * XXXAG ... but we will not be able to seek >2G anyway on many 186 * platforms as long as we are using fseek() rather 187 * than lseek(). 188 */ 189 unsigned char offset[4]; /*%< Offset from beginning of file. */ 190 } journal_rawpos_t; 191 192 /*% 193 * The header is of a fixed size, with some spare room for future 194 * extensions. 195 */ 196 #define JOURNAL_HEADER_SIZE 64 /* Bytes. */ 197 198 /*% 199 * The on-disk representation of the journal header. 200 * All numbers are stored in big-endian order. 201 */ 202 typedef union { 203 struct { 204 /*% File format version ID. */ 205 unsigned char format[16]; 206 /*% Position of the first addressable transaction */ 207 journal_rawpos_t begin; 208 /*% Position of the next (yet nonexistent) transaction. */ 209 journal_rawpos_t end; 210 /*% Number of index entries following the header. */ 211 unsigned char index_size[4]; 212 /*% Source serial number. */ 213 unsigned char sourceserial[4]; 214 unsigned char flags; 215 } h; 216 /* Pad the header to a fixed size. */ 217 unsigned char pad[JOURNAL_HEADER_SIZE]; 218 } journal_rawheader_t; 219 220 /*% 221 * The on-disk representation of the transaction header. 222 * There is one of these at the beginning of each transaction. 223 */ 224 typedef struct { 225 unsigned char size[4]; /*%< In bytes, excluding header. */ 226 unsigned char serial0[4]; /*%< SOA serial before update. */ 227 unsigned char serial1[4]; /*%< SOA serial after update. */ 228 } journal_rawxhdr_t; 229 230 /*% 231 * The on-disk representation of the RR header. 232 * There is one of these at the beginning of each RR. 233 */ 234 typedef struct { 235 unsigned char size[4]; /*%< In bytes, excluding header. */ 236 } journal_rawrrhdr_t; 237 238 /*% 239 * The in-core representation of the journal header. 240 */ 241 typedef struct { 242 uint32_t serial; 243 isc_offset_t offset; 244 } journal_pos_t; 245 246 #define POS_VALID(pos) ((pos).offset != 0) 247 #define POS_INVALIDATE(pos) ((pos).offset = 0, (pos).serial = 0) 248 249 typedef struct { 250 unsigned char format[16]; 251 journal_pos_t begin; 252 journal_pos_t end; 253 uint32_t index_size; 254 uint32_t sourceserial; 255 bool serialset; 256 } journal_header_t; 257 258 /*% 259 * The in-core representation of the transaction header. 260 */ 261 262 typedef struct { 263 uint32_t size; 264 uint32_t serial0; 265 uint32_t serial1; 266 } journal_xhdr_t; 267 268 /*% 269 * The in-core representation of the RR header. 270 */ 271 typedef struct { 272 uint32_t size; 273 } journal_rrhdr_t; 274 275 /*% 276 * Initial contents to store in the header of a newly created 277 * journal file. 278 * 279 * The header starts with the magic string ";BIND LOG V9\n" 280 * to identify the file as a BIND 9 journal file. An ASCII 281 * identification string is used rather than a binary magic 282 * number to be consistent with BIND 8 (BIND 8 journal files 283 * are ASCII text files). 284 */ 285 286 static journal_header_t initial_journal_header = { 287 ";BIND LOG V9\n", { 0, 0 }, { 0, 0 }, 0, 0, 0 288 }; 289 290 #define JOURNAL_EMPTY(h) ((h)->begin.offset == (h)->end.offset) 291 292 typedef enum { 293 JOURNAL_STATE_INVALID, 294 JOURNAL_STATE_READ, 295 JOURNAL_STATE_WRITE, 296 JOURNAL_STATE_TRANSACTION, 297 JOURNAL_STATE_INLINE 298 } journal_state_t; 299 300 struct dns_journal { 301 unsigned int magic; /*%< JOUR */ 302 isc_mem_t *mctx; /*%< Memory context */ 303 journal_state_t state; 304 char *filename; /*%< Journal file name */ 305 FILE *fp; /*%< File handle */ 306 isc_offset_t offset; /*%< Current file offset */ 307 journal_header_t header; /*%< In-core journal header */ 308 unsigned char *rawindex; /*%< In-core buffer for journal index 309 * in 310 * on-disk format */ 311 journal_pos_t *index; /*%< In-core journal index */ 312 313 /*% Current transaction state (when writing). */ 314 struct { 315 unsigned int n_soa; /*%< Number of SOAs seen */ 316 journal_pos_t pos[2]; /*%< Begin/end position */ 317 } x; 318 319 /*% Iteration state (when reading). */ 320 struct { 321 /* These define the part of the journal we iterate over. */ 322 journal_pos_t bpos; /*%< Position before first, */ 323 journal_pos_t epos; /*%< and after last transaction */ 324 /* The rest is iterator state. */ 325 uint32_t current_serial; /*%< Current SOA serial 326 * */ 327 isc_buffer_t source; /*%< Data from disk */ 328 isc_buffer_t target; /*%< Data from _fromwire check 329 * */ 330 dns_decompress_t dctx; /*%< Dummy decompression ctx */ 331 dns_name_t name; /*%< Current domain name */ 332 dns_rdata_t rdata; /*%< Current rdata */ 333 uint32_t ttl; /*%< Current TTL */ 334 unsigned int xsize; /*%< Size of transaction data */ 335 unsigned int xpos; /*%< Current position in it */ 336 isc_result_t result; /*%< Result of last call */ 337 } it; 338 }; 339 340 #define DNS_JOURNAL_MAGIC ISC_MAGIC('J', 'O', 'U', 'R') 341 #define DNS_JOURNAL_VALID(t) ISC_MAGIC_VALID(t, DNS_JOURNAL_MAGIC) 342 343 static void 344 journal_pos_decode(journal_rawpos_t *raw, journal_pos_t *cooked) { 345 cooked->serial = decode_uint32(raw->serial); 346 cooked->offset = decode_uint32(raw->offset); 347 } 348 349 static void 350 journal_pos_encode(journal_rawpos_t *raw, journal_pos_t *cooked) { 351 encode_uint32(cooked->serial, raw->serial); 352 encode_uint32(cooked->offset, raw->offset); 353 } 354 355 static void 356 journal_header_decode(journal_rawheader_t *raw, journal_header_t *cooked) { 357 INSIST(sizeof(cooked->format) == sizeof(raw->h.format)); 358 memmove(cooked->format, raw->h.format, sizeof(cooked->format)); 359 journal_pos_decode(&raw->h.begin, &cooked->begin); 360 journal_pos_decode(&raw->h.end, &cooked->end); 361 cooked->index_size = decode_uint32(raw->h.index_size); 362 cooked->sourceserial = decode_uint32(raw->h.sourceserial); 363 cooked->serialset = ((raw->h.flags & JOURNAL_SERIALSET) != 0); 364 } 365 366 static void 367 journal_header_encode(journal_header_t *cooked, journal_rawheader_t *raw) { 368 unsigned char flags = 0; 369 370 INSIST(sizeof(cooked->format) == sizeof(raw->h.format)); 371 memset(raw->pad, 0, sizeof(raw->pad)); 372 memmove(raw->h.format, cooked->format, sizeof(raw->h.format)); 373 journal_pos_encode(&raw->h.begin, &cooked->begin); 374 journal_pos_encode(&raw->h.end, &cooked->end); 375 encode_uint32(cooked->index_size, raw->h.index_size); 376 encode_uint32(cooked->sourceserial, raw->h.sourceserial); 377 if (cooked->serialset) { 378 flags |= JOURNAL_SERIALSET; 379 } 380 raw->h.flags = flags; 381 } 382 383 /* 384 * Journal file I/O subroutines, with error checking and reporting. 385 */ 386 static isc_result_t 387 journal_seek(dns_journal_t *j, uint32_t offset) { 388 isc_result_t result; 389 390 result = isc_stdio_seek(j->fp, (off_t)offset, SEEK_SET); 391 if (result != ISC_R_SUCCESS) { 392 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 393 "%s: seek: %s", j->filename, 394 isc_result_totext(result)); 395 return (ISC_R_UNEXPECTED); 396 } 397 j->offset = offset; 398 return (ISC_R_SUCCESS); 399 } 400 401 static isc_result_t 402 journal_read(dns_journal_t *j, void *mem, size_t nbytes) { 403 isc_result_t result; 404 405 result = isc_stdio_read(mem, 1, nbytes, j->fp, NULL); 406 if (result != ISC_R_SUCCESS) { 407 if (result == ISC_R_EOF) { 408 return (ISC_R_NOMORE); 409 } 410 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 411 "%s: read: %s", j->filename, 412 isc_result_totext(result)); 413 return (ISC_R_UNEXPECTED); 414 } 415 j->offset += (isc_offset_t)nbytes; 416 return (ISC_R_SUCCESS); 417 } 418 419 static isc_result_t 420 journal_write(dns_journal_t *j, void *mem, size_t nbytes) { 421 isc_result_t result; 422 423 result = isc_stdio_write(mem, 1, nbytes, j->fp, NULL); 424 if (result != ISC_R_SUCCESS) { 425 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 426 "%s: write: %s", j->filename, 427 isc_result_totext(result)); 428 return (ISC_R_UNEXPECTED); 429 } 430 j->offset += (isc_offset_t)nbytes; 431 return (ISC_R_SUCCESS); 432 } 433 434 static isc_result_t 435 journal_fsync(dns_journal_t *j) { 436 isc_result_t result; 437 result = isc_stdio_flush(j->fp); 438 if (result != ISC_R_SUCCESS) { 439 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 440 "%s: flush: %s", j->filename, 441 isc_result_totext(result)); 442 return (ISC_R_UNEXPECTED); 443 } 444 result = isc_stdio_sync(j->fp); 445 if (result != ISC_R_SUCCESS) { 446 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 447 "%s: fsync: %s", j->filename, 448 isc_result_totext(result)); 449 return (ISC_R_UNEXPECTED); 450 } 451 return (ISC_R_SUCCESS); 452 } 453 454 /* 455 * Read/write a transaction header at the current file position. 456 */ 457 458 static isc_result_t 459 journal_read_xhdr(dns_journal_t *j, journal_xhdr_t *xhdr) { 460 journal_rawxhdr_t raw; 461 isc_result_t result; 462 result = journal_read(j, &raw, sizeof(raw)); 463 if (result != ISC_R_SUCCESS) { 464 return (result); 465 } 466 xhdr->size = decode_uint32(raw.size); 467 xhdr->serial0 = decode_uint32(raw.serial0); 468 xhdr->serial1 = decode_uint32(raw.serial1); 469 return (ISC_R_SUCCESS); 470 } 471 472 static isc_result_t 473 journal_write_xhdr(dns_journal_t *j, uint32_t size, uint32_t serial0, 474 uint32_t serial1) { 475 journal_rawxhdr_t raw; 476 encode_uint32(size, raw.size); 477 encode_uint32(serial0, raw.serial0); 478 encode_uint32(serial1, raw.serial1); 479 return (journal_write(j, &raw, sizeof(raw))); 480 } 481 482 /* 483 * Read an RR header at the current file position. 484 */ 485 486 static isc_result_t 487 journal_read_rrhdr(dns_journal_t *j, journal_rrhdr_t *rrhdr) { 488 journal_rawrrhdr_t raw; 489 isc_result_t result; 490 result = journal_read(j, &raw, sizeof(raw)); 491 if (result != ISC_R_SUCCESS) { 492 return (result); 493 } 494 rrhdr->size = decode_uint32(raw.size); 495 return (ISC_R_SUCCESS); 496 } 497 498 static isc_result_t 499 journal_file_create(isc_mem_t *mctx, const char *filename) { 500 FILE *fp = NULL; 501 isc_result_t result; 502 journal_header_t header; 503 journal_rawheader_t rawheader; 504 int index_size = 56; /* XXX configurable */ 505 int size; 506 void *mem; /* Memory for temporary index image. */ 507 508 INSIST(sizeof(journal_rawheader_t) == JOURNAL_HEADER_SIZE); 509 510 result = isc_stdio_open(filename, "wb", &fp); 511 if (result != ISC_R_SUCCESS) { 512 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 513 "%s: create: %s", filename, 514 isc_result_totext(result)); 515 return (ISC_R_UNEXPECTED); 516 } 517 518 header = initial_journal_header; 519 header.index_size = index_size; 520 journal_header_encode(&header, &rawheader); 521 522 size = sizeof(journal_rawheader_t) + 523 index_size * sizeof(journal_rawpos_t); 524 525 mem = isc_mem_get(mctx, size); 526 memset(mem, 0, size); 527 memmove(mem, &rawheader, sizeof(rawheader)); 528 529 result = isc_stdio_write(mem, 1, (size_t)size, fp, NULL); 530 if (result != ISC_R_SUCCESS) { 531 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 532 "%s: write: %s", filename, 533 isc_result_totext(result)); 534 (void)isc_stdio_close(fp); 535 (void)isc_file_remove(filename); 536 isc_mem_put(mctx, mem, size); 537 return (ISC_R_UNEXPECTED); 538 } 539 isc_mem_put(mctx, mem, size); 540 541 result = isc_stdio_close(fp); 542 if (result != ISC_R_SUCCESS) { 543 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 544 "%s: close: %s", filename, 545 isc_result_totext(result)); 546 (void)isc_file_remove(filename); 547 return (ISC_R_UNEXPECTED); 548 } 549 550 return (ISC_R_SUCCESS); 551 } 552 553 static isc_result_t 554 journal_open(isc_mem_t *mctx, const char *filename, bool writable, bool create, 555 dns_journal_t **journalp) { 556 FILE *fp = NULL; 557 isc_result_t result; 558 journal_rawheader_t rawheader; 559 dns_journal_t *j; 560 561 INSIST(journalp != NULL && *journalp == NULL); 562 j = isc_mem_get(mctx, sizeof(*j)); 563 564 j->mctx = NULL; 565 isc_mem_attach(mctx, &j->mctx); 566 j->state = JOURNAL_STATE_INVALID; 567 j->fp = NULL; 568 j->filename = isc_mem_strdup(mctx, filename); 569 j->index = NULL; 570 j->rawindex = NULL; 571 572 if (j->filename == NULL) { 573 FAIL(ISC_R_NOMEMORY); 574 } 575 576 result = isc_stdio_open(j->filename, writable ? "rb+" : "rb", &fp); 577 578 if (result == ISC_R_FILENOTFOUND) { 579 if (create) { 580 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_DEBUG(1), 581 "journal file %s does not exist, " 582 "creating it", 583 j->filename); 584 CHECK(journal_file_create(mctx, filename)); 585 /* 586 * Retry. 587 */ 588 result = isc_stdio_open(j->filename, "rb+", &fp); 589 } else { 590 FAIL(ISC_R_NOTFOUND); 591 } 592 } 593 if (result != ISC_R_SUCCESS) { 594 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 595 "%s: open: %s", j->filename, 596 isc_result_totext(result)); 597 FAIL(ISC_R_UNEXPECTED); 598 } 599 600 j->fp = fp; 601 602 /* 603 * Set magic early so that seek/read can succeed. 604 */ 605 j->magic = DNS_JOURNAL_MAGIC; 606 607 CHECK(journal_seek(j, 0)); 608 CHECK(journal_read(j, &rawheader, sizeof(rawheader))); 609 610 if (memcmp(rawheader.h.format, initial_journal_header.format, 611 sizeof(initial_journal_header.format)) != 0) 612 { 613 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 614 "%s: journal format not recognized", j->filename); 615 FAIL(ISC_R_UNEXPECTED); 616 } 617 journal_header_decode(&rawheader, &j->header); 618 619 /* 620 * If there is an index, read the raw index into a dynamically 621 * allocated buffer and then convert it into a cooked index. 622 */ 623 if (j->header.index_size != 0) { 624 unsigned int i; 625 unsigned int rawbytes; 626 unsigned char *p; 627 628 rawbytes = j->header.index_size * sizeof(journal_rawpos_t); 629 j->rawindex = isc_mem_get(mctx, rawbytes); 630 631 CHECK(journal_read(j, j->rawindex, rawbytes)); 632 633 j->index = isc_mem_get(mctx, j->header.index_size * 634 sizeof(journal_pos_t)); 635 636 p = j->rawindex; 637 for (i = 0; i < j->header.index_size; i++) { 638 j->index[i].serial = decode_uint32(p); 639 p += 4; 640 j->index[i].offset = decode_uint32(p); 641 p += 4; 642 } 643 INSIST(p == j->rawindex + rawbytes); 644 } 645 j->offset = -1; /* Invalid, must seek explicitly. */ 646 647 /* 648 * Initialize the iterator. 649 */ 650 dns_name_init(&j->it.name, NULL); 651 dns_rdata_init(&j->it.rdata); 652 653 /* 654 * Set up empty initial buffers for unchecked and checked 655 * wire format RR data. They will be reallocated 656 * later. 657 */ 658 isc_buffer_init(&j->it.source, NULL, 0); 659 isc_buffer_init(&j->it.target, NULL, 0); 660 dns_decompress_init(&j->it.dctx, -1, DNS_DECOMPRESS_NONE); 661 662 j->state = writable ? JOURNAL_STATE_WRITE : JOURNAL_STATE_READ; 663 664 *journalp = j; 665 return (ISC_R_SUCCESS); 666 667 failure: 668 j->magic = 0; 669 if (j->rawindex != NULL) { 670 isc_mem_put(j->mctx, j->rawindex, 671 j->header.index_size * sizeof(journal_rawpos_t)); 672 } 673 if (j->index != NULL) { 674 isc_mem_put(j->mctx, j->index, 675 j->header.index_size * sizeof(journal_pos_t)); 676 } 677 if (j->filename != NULL) { 678 isc_mem_free(j->mctx, j->filename); 679 } 680 if (j->fp != NULL) { 681 (void)isc_stdio_close(j->fp); 682 } 683 isc_mem_putanddetach(&j->mctx, j, sizeof(*j)); 684 return (result); 685 } 686 687 isc_result_t 688 dns_journal_open(isc_mem_t *mctx, const char *filename, unsigned int mode, 689 dns_journal_t **journalp) { 690 isc_result_t result; 691 size_t namelen; 692 char backup[1024]; 693 bool writable, create; 694 695 create = ((mode & DNS_JOURNAL_CREATE) != 0); 696 writable = ((mode & (DNS_JOURNAL_WRITE | DNS_JOURNAL_CREATE)) != 0); 697 698 result = journal_open(mctx, filename, writable, create, journalp); 699 if (result == ISC_R_NOTFOUND) { 700 namelen = strlen(filename); 701 if (namelen > 4U && strcmp(filename + namelen - 4, ".jnl") == 0) 702 { 703 namelen -= 4; 704 } 705 706 result = snprintf(backup, sizeof(backup), "%.*s.jbk", 707 (int)namelen, filename); 708 if (result >= sizeof(backup)) { 709 return (ISC_R_NOSPACE); 710 } 711 result = journal_open(mctx, backup, writable, writable, 712 journalp); 713 } 714 return (result); 715 } 716 717 /* 718 * A comparison function defining the sorting order for 719 * entries in the IXFR-style journal file. 720 * 721 * The IXFR format requires that deletions are sorted before 722 * additions, and within either one, SOA records are sorted 723 * before others. 724 * 725 * Also sort the non-SOA records by type as a courtesy to the 726 * server receiving the IXFR - it may help reduce the amount of 727 * rdataset merging it has to do. 728 */ 729 static int 730 ixfr_order(const void *av, const void *bv) { 731 dns_difftuple_t const *const *ap = av; 732 dns_difftuple_t const *const *bp = bv; 733 dns_difftuple_t const *a = *ap; 734 dns_difftuple_t const *b = *bp; 735 int r; 736 int bop = 0, aop = 0; 737 738 switch (a->op) { 739 case DNS_DIFFOP_DEL: 740 case DNS_DIFFOP_DELRESIGN: 741 aop = 1; 742 break; 743 case DNS_DIFFOP_ADD: 744 case DNS_DIFFOP_ADDRESIGN: 745 aop = 0; 746 break; 747 default: 748 INSIST(0); 749 ISC_UNREACHABLE(); 750 } 751 752 switch (b->op) { 753 case DNS_DIFFOP_DEL: 754 case DNS_DIFFOP_DELRESIGN: 755 bop = 1; 756 break; 757 case DNS_DIFFOP_ADD: 758 case DNS_DIFFOP_ADDRESIGN: 759 bop = 0; 760 break; 761 default: 762 INSIST(0); 763 ISC_UNREACHABLE(); 764 } 765 766 r = bop - aop; 767 if (r != 0) { 768 return (r); 769 } 770 771 r = (b->rdata.type == dns_rdatatype_soa) - 772 (a->rdata.type == dns_rdatatype_soa); 773 if (r != 0) { 774 return (r); 775 } 776 777 r = (a->rdata.type - b->rdata.type); 778 return (r); 779 } 780 781 /* 782 * Advance '*pos' to the next journal transaction. 783 * 784 * Requires: 785 * *pos refers to a valid journal transaction. 786 * 787 * Ensures: 788 * When ISC_R_SUCCESS is returned, 789 * *pos refers to the next journal transaction. 790 * 791 * Returns one of: 792 * 793 * ISC_R_SUCCESS 794 * ISC_R_NOMORE *pos pointed at the last transaction 795 * Other results due to file errors are possible. 796 */ 797 static isc_result_t 798 journal_next(dns_journal_t *j, journal_pos_t *pos) { 799 isc_result_t result; 800 journal_xhdr_t xhdr; 801 REQUIRE(DNS_JOURNAL_VALID(j)); 802 803 result = journal_seek(j, pos->offset); 804 if (result != ISC_R_SUCCESS) { 805 return (result); 806 } 807 808 if (pos->serial == j->header.end.serial) { 809 return (ISC_R_NOMORE); 810 } 811 /* 812 * Read the header of the current transaction. 813 * This will return ISC_R_NOMORE if we are at EOF. 814 */ 815 result = journal_read_xhdr(j, &xhdr); 816 if (result != ISC_R_SUCCESS) { 817 return (result); 818 } 819 820 /* 821 * Check serial number consistency. 822 */ 823 if (xhdr.serial0 != pos->serial) { 824 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 825 "%s: journal file corrupt: " 826 "expected serial %u, got %u", 827 j->filename, pos->serial, xhdr.serial0); 828 return (ISC_R_UNEXPECTED); 829 } 830 831 /* 832 * Check for offset wraparound. 833 */ 834 if ((isc_offset_t)(pos->offset + sizeof(journal_rawxhdr_t) + 835 xhdr.size) < pos->offset) 836 { 837 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 838 "%s: offset too large", j->filename); 839 return (ISC_R_UNEXPECTED); 840 } 841 842 pos->offset += sizeof(journal_rawxhdr_t) + xhdr.size; 843 pos->serial = xhdr.serial1; 844 return (ISC_R_SUCCESS); 845 } 846 847 /* 848 * If the index of the journal 'j' contains an entry "better" 849 * than '*best_guess', replace '*best_guess' with it. 850 * 851 * "Better" means having a serial number closer to 'serial' 852 * but not greater than 'serial'. 853 */ 854 static void 855 index_find(dns_journal_t *j, uint32_t serial, journal_pos_t *best_guess) { 856 unsigned int i; 857 if (j->index == NULL) { 858 return; 859 } 860 for (i = 0; i < j->header.index_size; i++) { 861 if (POS_VALID(j->index[i]) && 862 DNS_SERIAL_GE(serial, j->index[i].serial) && 863 DNS_SERIAL_GT(j->index[i].serial, best_guess->serial)) 864 { 865 *best_guess = j->index[i]; 866 } 867 } 868 } 869 870 /* 871 * Add a new index entry. If there is no room, make room by removing 872 * the odd-numbered entries and compacting the others into the first 873 * half of the index. This decimates old index entries exponentially 874 * over time, so that the index always contains a much larger fraction 875 * of recent serial numbers than of old ones. This is deliberate - 876 * most index searches are for outgoing IXFR, and IXFR tends to request 877 * recent versions more often than old ones. 878 */ 879 static void 880 index_add(dns_journal_t *j, journal_pos_t *pos) { 881 unsigned int i; 882 if (j->index == NULL) { 883 return; 884 } 885 /* 886 * Search for a vacant position. 887 */ 888 for (i = 0; i < j->header.index_size; i++) { 889 if (!POS_VALID(j->index[i])) { 890 break; 891 } 892 } 893 if (i == j->header.index_size) { 894 unsigned int k = 0; 895 /* 896 * Found no vacant position. Make some room. 897 */ 898 for (i = 0; i < j->header.index_size; i += 2) { 899 j->index[k++] = j->index[i]; 900 } 901 i = k; /* 'i' identifies the first vacant position. */ 902 while (k < j->header.index_size) { 903 POS_INVALIDATE(j->index[k]); 904 k++; 905 } 906 } 907 INSIST(i < j->header.index_size); 908 INSIST(!POS_VALID(j->index[i])); 909 910 /* 911 * Store the new index entry. 912 */ 913 j->index[i] = *pos; 914 } 915 916 /* 917 * Invalidate any existing index entries that could become 918 * ambiguous when a new transaction with number 'serial' is added. 919 */ 920 static void 921 index_invalidate(dns_journal_t *j, uint32_t serial) { 922 unsigned int i; 923 if (j->index == NULL) { 924 return; 925 } 926 for (i = 0; i < j->header.index_size; i++) { 927 if (!DNS_SERIAL_GT(serial, j->index[i].serial)) { 928 POS_INVALIDATE(j->index[i]); 929 } 930 } 931 } 932 933 /* 934 * Try to find a transaction with initial serial number 'serial' 935 * in the journal 'j'. 936 * 937 * If found, store its position at '*pos' and return ISC_R_SUCCESS. 938 * 939 * If 'serial' is current (= the ending serial number of the 940 * last transaction in the journal), set '*pos' to 941 * the position immediately following the last transaction and 942 * return ISC_R_SUCCESS. 943 * 944 * If 'serial' is within the range of addressable serial numbers 945 * covered by the journal but that particular serial number is missing 946 * (from the journal, not just from the index), return ISC_R_NOTFOUND. 947 * 948 * If 'serial' is outside the range of addressable serial numbers 949 * covered by the journal, return ISC_R_RANGE. 950 * 951 */ 952 static isc_result_t 953 journal_find(dns_journal_t *j, uint32_t serial, journal_pos_t *pos) { 954 isc_result_t result; 955 journal_pos_t current_pos; 956 REQUIRE(DNS_JOURNAL_VALID(j)); 957 958 if (DNS_SERIAL_GT(j->header.begin.serial, serial)) { 959 return (ISC_R_RANGE); 960 } 961 if (DNS_SERIAL_GT(serial, j->header.end.serial)) { 962 return (ISC_R_RANGE); 963 } 964 if (serial == j->header.end.serial) { 965 *pos = j->header.end; 966 return (ISC_R_SUCCESS); 967 } 968 969 current_pos = j->header.begin; 970 index_find(j, serial, ¤t_pos); 971 972 while (current_pos.serial != serial) { 973 if (DNS_SERIAL_GT(current_pos.serial, serial)) { 974 return (ISC_R_NOTFOUND); 975 } 976 result = journal_next(j, ¤t_pos); 977 if (result != ISC_R_SUCCESS) { 978 return (result); 979 } 980 } 981 *pos = current_pos; 982 return (ISC_R_SUCCESS); 983 } 984 985 isc_result_t 986 dns_journal_begin_transaction(dns_journal_t *j) { 987 uint32_t offset; 988 isc_result_t result; 989 journal_rawxhdr_t hdr; 990 991 REQUIRE(DNS_JOURNAL_VALID(j)); 992 REQUIRE(j->state == JOURNAL_STATE_WRITE || 993 j->state == JOURNAL_STATE_INLINE); 994 995 /* 996 * Find the file offset where the new transaction should 997 * be written, and seek there. 998 */ 999 if (JOURNAL_EMPTY(&j->header)) { 1000 offset = sizeof(journal_rawheader_t) + 1001 j->header.index_size * sizeof(journal_rawpos_t); 1002 } else { 1003 offset = j->header.end.offset; 1004 } 1005 j->x.pos[0].offset = offset; 1006 j->x.pos[1].offset = offset; /* Initial value, will be incremented. */ 1007 j->x.n_soa = 0; 1008 1009 CHECK(journal_seek(j, offset)); 1010 1011 /* 1012 * Write a dummy transaction header of all zeroes to reserve 1013 * space. It will be filled in when the transaction is 1014 * finished. 1015 */ 1016 memset(&hdr, 0, sizeof(hdr)); 1017 CHECK(journal_write(j, &hdr, sizeof(hdr))); 1018 j->x.pos[1].offset = j->offset; 1019 1020 j->state = JOURNAL_STATE_TRANSACTION; 1021 result = ISC_R_SUCCESS; 1022 failure: 1023 return (result); 1024 } 1025 1026 isc_result_t 1027 dns_journal_writediff(dns_journal_t *j, dns_diff_t *diff) { 1028 dns_difftuple_t *t; 1029 isc_buffer_t buffer; 1030 void *mem = NULL; 1031 uint64_t size; 1032 isc_result_t result; 1033 isc_region_t used; 1034 1035 REQUIRE(DNS_DIFF_VALID(diff)); 1036 REQUIRE(j->state == JOURNAL_STATE_TRANSACTION); 1037 1038 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), "writing to journal"); 1039 (void)dns_diff_print(diff, NULL); 1040 1041 /* 1042 * Pass 1: determine the buffer size needed, and 1043 * keep track of SOA serial numbers. 1044 */ 1045 size = 0; 1046 for (t = ISC_LIST_HEAD(diff->tuples); t != NULL; 1047 t = ISC_LIST_NEXT(t, link)) { 1048 if (t->rdata.type == dns_rdatatype_soa) { 1049 if (j->x.n_soa < 2) { 1050 j->x.pos[j->x.n_soa].serial = 1051 dns_soa_getserial(&t->rdata); 1052 } 1053 j->x.n_soa++; 1054 } 1055 size += sizeof(journal_rawrrhdr_t); 1056 size += t->name.length; /* XXX should have access macro? */ 1057 size += 10; 1058 size += t->rdata.length; 1059 } 1060 1061 if (size >= DNS_JOURNAL_SIZE_MAX) { 1062 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1063 "dns_journal_writediff: %s: journal entry " 1064 "too big to be stored: %" PRIu64 " bytes", 1065 j->filename, size); 1066 return (ISC_R_NOSPACE); 1067 } 1068 1069 mem = isc_mem_get(j->mctx, size); 1070 1071 isc_buffer_init(&buffer, mem, size); 1072 1073 /* 1074 * Pass 2. Write RRs to buffer. 1075 */ 1076 for (t = ISC_LIST_HEAD(diff->tuples); t != NULL; 1077 t = ISC_LIST_NEXT(t, link)) { 1078 /* 1079 * Write the RR header. 1080 */ 1081 isc_buffer_putuint32(&buffer, 1082 t->name.length + 10 + t->rdata.length); 1083 /* 1084 * Write the owner name, RR header, and RR data. 1085 */ 1086 isc_buffer_putmem(&buffer, t->name.ndata, t->name.length); 1087 isc_buffer_putuint16(&buffer, t->rdata.type); 1088 isc_buffer_putuint16(&buffer, t->rdata.rdclass); 1089 isc_buffer_putuint32(&buffer, t->ttl); 1090 INSIST(t->rdata.length < 65536); 1091 isc_buffer_putuint16(&buffer, (uint16_t)t->rdata.length); 1092 INSIST(isc_buffer_availablelength(&buffer) >= t->rdata.length); 1093 isc_buffer_putmem(&buffer, t->rdata.data, t->rdata.length); 1094 } 1095 1096 isc_buffer_usedregion(&buffer, &used); 1097 INSIST(used.length == size); 1098 1099 j->x.pos[1].offset += used.length; 1100 1101 /* 1102 * Write the buffer contents to the journal file. 1103 */ 1104 CHECK(journal_write(j, used.base, used.length)); 1105 1106 result = ISC_R_SUCCESS; 1107 1108 failure: 1109 if (mem != NULL) { 1110 isc_mem_put(j->mctx, mem, size); 1111 } 1112 return (result); 1113 } 1114 1115 isc_result_t 1116 dns_journal_commit(dns_journal_t *j) { 1117 isc_result_t result; 1118 journal_rawheader_t rawheader; 1119 uint64_t total; 1120 1121 REQUIRE(DNS_JOURNAL_VALID(j)); 1122 REQUIRE(j->state == JOURNAL_STATE_TRANSACTION || 1123 j->state == JOURNAL_STATE_INLINE); 1124 1125 /* 1126 * Just write out a updated header. 1127 */ 1128 if (j->state == JOURNAL_STATE_INLINE) { 1129 CHECK(journal_fsync(j)); 1130 journal_header_encode(&j->header, &rawheader); 1131 CHECK(journal_seek(j, 0)); 1132 CHECK(journal_write(j, &rawheader, sizeof(rawheader))); 1133 CHECK(journal_fsync(j)); 1134 j->state = JOURNAL_STATE_WRITE; 1135 return (ISC_R_SUCCESS); 1136 } 1137 1138 /* 1139 * Perform some basic consistency checks. 1140 */ 1141 if (j->x.n_soa != 2) { 1142 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1143 "%s: malformed transaction: %d SOAs", j->filename, 1144 j->x.n_soa); 1145 return (ISC_R_UNEXPECTED); 1146 } 1147 if (!DNS_SERIAL_GT(j->x.pos[1].serial, j->x.pos[0].serial)) { 1148 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1149 "%s: malformed transaction: serial number " 1150 "did not increase", 1151 j->filename); 1152 return (ISC_R_UNEXPECTED); 1153 } 1154 if (!JOURNAL_EMPTY(&j->header)) { 1155 if (j->x.pos[0].serial != j->header.end.serial) { 1156 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1157 "malformed transaction: " 1158 "%s last serial %u != " 1159 "transaction first serial %u", 1160 j->filename, j->header.end.serial, 1161 j->x.pos[0].serial); 1162 return (ISC_R_UNEXPECTED); 1163 } 1164 } 1165 1166 /* 1167 * We currently don't support huge journal entries. 1168 */ 1169 total = j->x.pos[1].offset - j->x.pos[0].offset; 1170 if (total >= DNS_JOURNAL_SIZE_MAX) { 1171 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1172 "transaction too big to be stored in journal: " 1173 "%" PRIu64 "b (max is %" PRIu64 "b)", 1174 total, (uint64_t)DNS_JOURNAL_SIZE_MAX); 1175 return (ISC_R_UNEXPECTED); 1176 } 1177 1178 /* 1179 * Some old journal entries may become non-addressable 1180 * when we increment the current serial number. Purge them 1181 * by stepping header.begin forward to the first addressable 1182 * transaction. Also purge them from the index. 1183 */ 1184 if (!JOURNAL_EMPTY(&j->header)) { 1185 while (!DNS_SERIAL_GT(j->x.pos[1].serial, 1186 j->header.begin.serial)) { 1187 CHECK(journal_next(j, &j->header.begin)); 1188 } 1189 index_invalidate(j, j->x.pos[1].serial); 1190 } 1191 #ifdef notyet 1192 if (DNS_SERIAL_GT(last_dumped_serial, j->x.pos[1].serial)) { 1193 force_dump(...); 1194 } 1195 #endif /* ifdef notyet */ 1196 1197 /* 1198 * Commit the transaction data to stable storage. 1199 */ 1200 CHECK(journal_fsync(j)); 1201 1202 if (j->state == JOURNAL_STATE_TRANSACTION) { 1203 isc_offset_t offset; 1204 offset = (j->x.pos[1].offset - j->x.pos[0].offset) - 1205 sizeof(journal_rawxhdr_t); 1206 /* 1207 * Update the transaction header. 1208 */ 1209 CHECK(journal_seek(j, j->x.pos[0].offset)); 1210 CHECK(journal_write_xhdr(j, offset, j->x.pos[0].serial, 1211 j->x.pos[1].serial)); 1212 } 1213 1214 /* 1215 * Update the journal header. 1216 */ 1217 if (JOURNAL_EMPTY(&j->header)) { 1218 j->header.begin = j->x.pos[0]; 1219 } 1220 j->header.end = j->x.pos[1]; 1221 journal_header_encode(&j->header, &rawheader); 1222 CHECK(journal_seek(j, 0)); 1223 CHECK(journal_write(j, &rawheader, sizeof(rawheader))); 1224 1225 /* 1226 * Update the index. 1227 */ 1228 index_add(j, &j->x.pos[0]); 1229 1230 /* 1231 * Convert the index into on-disk format and write 1232 * it to disk. 1233 */ 1234 CHECK(index_to_disk(j)); 1235 1236 /* 1237 * Commit the header to stable storage. 1238 */ 1239 CHECK(journal_fsync(j)); 1240 1241 /* 1242 * We no longer have a transaction open. 1243 */ 1244 j->state = JOURNAL_STATE_WRITE; 1245 1246 result = ISC_R_SUCCESS; 1247 1248 failure: 1249 return (result); 1250 } 1251 1252 isc_result_t 1253 dns_journal_write_transaction(dns_journal_t *j, dns_diff_t *diff) { 1254 isc_result_t result; 1255 CHECK(dns_diff_sort(diff, ixfr_order)); 1256 CHECK(dns_journal_begin_transaction(j)); 1257 CHECK(dns_journal_writediff(j, diff)); 1258 CHECK(dns_journal_commit(j)); 1259 result = ISC_R_SUCCESS; 1260 failure: 1261 return (result); 1262 } 1263 1264 void 1265 dns_journal_destroy(dns_journal_t **journalp) { 1266 dns_journal_t *j = *journalp; 1267 *journalp = NULL; 1268 REQUIRE(DNS_JOURNAL_VALID(j)); 1269 1270 j->it.result = ISC_R_FAILURE; 1271 dns_name_invalidate(&j->it.name); 1272 dns_decompress_invalidate(&j->it.dctx); 1273 if (j->rawindex != NULL) { 1274 isc_mem_put(j->mctx, j->rawindex, 1275 j->header.index_size * sizeof(journal_rawpos_t)); 1276 } 1277 if (j->index != NULL) { 1278 isc_mem_put(j->mctx, j->index, 1279 j->header.index_size * sizeof(journal_pos_t)); 1280 } 1281 if (j->it.target.base != NULL) { 1282 isc_mem_put(j->mctx, j->it.target.base, j->it.target.length); 1283 } 1284 if (j->it.source.base != NULL) { 1285 isc_mem_put(j->mctx, j->it.source.base, j->it.source.length); 1286 } 1287 if (j->filename != NULL) { 1288 isc_mem_free(j->mctx, j->filename); 1289 } 1290 if (j->fp != NULL) { 1291 (void)isc_stdio_close(j->fp); 1292 } 1293 j->magic = 0; 1294 isc_mem_putanddetach(&j->mctx, j, sizeof(*j)); 1295 } 1296 1297 /* 1298 * Roll the open journal 'j' into the database 'db'. 1299 * A new database version will be created. 1300 */ 1301 1302 /* XXX Share code with incoming IXFR? */ 1303 1304 static isc_result_t 1305 roll_forward(dns_journal_t *j, dns_db_t *db, unsigned int options) { 1306 isc_buffer_t source; /* Transaction data from disk */ 1307 isc_buffer_t target; /* Ditto after _fromwire check */ 1308 uint32_t db_serial; /* Database SOA serial */ 1309 uint32_t end_serial; /* Last journal SOA serial */ 1310 isc_result_t result; 1311 dns_dbversion_t *ver = NULL; 1312 journal_pos_t pos; 1313 dns_diff_t diff; 1314 unsigned int n_soa = 0; 1315 unsigned int n_put = 0; 1316 dns_diffop_t op; 1317 1318 REQUIRE(DNS_JOURNAL_VALID(j)); 1319 REQUIRE(DNS_DB_VALID(db)); 1320 1321 dns_diff_init(j->mctx, &diff); 1322 1323 /* 1324 * Set up empty initial buffers for unchecked and checked 1325 * wire format transaction data. They will be reallocated 1326 * later. 1327 */ 1328 isc_buffer_init(&source, NULL, 0); 1329 isc_buffer_init(&target, NULL, 0); 1330 1331 /* 1332 * Create the new database version. 1333 */ 1334 CHECK(dns_db_newversion(db, &ver)); 1335 1336 /* 1337 * Get the current database SOA serial number. 1338 */ 1339 CHECK(dns_db_getsoaserial(db, ver, &db_serial)); 1340 1341 /* 1342 * Locate a journal entry for the current database serial. 1343 */ 1344 CHECK(journal_find(j, db_serial, &pos)); 1345 /* 1346 * XXX do more drastic things, like marking zone stale, 1347 * if this fails? 1348 */ 1349 /* 1350 * XXXRTH The zone code should probably mark the zone as bad and 1351 * scream loudly into the log if this is a dynamic update 1352 * log reply that failed. 1353 */ 1354 1355 end_serial = dns_journal_last_serial(j); 1356 if (db_serial == end_serial) { 1357 CHECK(DNS_R_UPTODATE); 1358 } 1359 1360 CHECK(dns_journal_iter_init(j, db_serial, end_serial)); 1361 1362 for (result = dns_journal_first_rr(j); result == ISC_R_SUCCESS; 1363 result = dns_journal_next_rr(j)) 1364 { 1365 dns_name_t *name; 1366 uint32_t ttl; 1367 dns_rdata_t *rdata; 1368 dns_difftuple_t *tuple = NULL; 1369 1370 name = NULL; 1371 rdata = NULL; 1372 dns_journal_current_rr(j, &name, &ttl, &rdata); 1373 1374 if (rdata->type == dns_rdatatype_soa) { 1375 n_soa++; 1376 if (n_soa == 2) { 1377 db_serial = j->it.current_serial; 1378 } 1379 } 1380 1381 if (n_soa == 3) { 1382 n_soa = 1; 1383 } 1384 if (n_soa == 0) { 1385 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1386 "%s: journal file corrupt: missing " 1387 "initial SOA", 1388 j->filename); 1389 FAIL(ISC_R_UNEXPECTED); 1390 } 1391 if ((options & DNS_JOURNALOPT_RESIGN) != 0) { 1392 op = (n_soa == 1) ? DNS_DIFFOP_DELRESIGN 1393 : DNS_DIFFOP_ADDRESIGN; 1394 } else { 1395 op = (n_soa == 1) ? DNS_DIFFOP_DEL : DNS_DIFFOP_ADD; 1396 } 1397 1398 CHECK(dns_difftuple_create(diff.mctx, op, name, ttl, rdata, 1399 &tuple)); 1400 dns_diff_append(&diff, &tuple); 1401 1402 if (++n_put > 100) { 1403 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), 1404 "%s: applying diff to database (%u)", 1405 j->filename, db_serial); 1406 (void)dns_diff_print(&diff, NULL); 1407 CHECK(dns_diff_apply(&diff, db, ver)); 1408 dns_diff_clear(&diff); 1409 n_put = 0; 1410 } 1411 } 1412 if (result == ISC_R_NOMORE) { 1413 result = ISC_R_SUCCESS; 1414 } 1415 CHECK(result); 1416 1417 if (n_put != 0) { 1418 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), 1419 "%s: applying final diff to database (%u)", 1420 j->filename, db_serial); 1421 (void)dns_diff_print(&diff, NULL); 1422 CHECK(dns_diff_apply(&diff, db, ver)); 1423 dns_diff_clear(&diff); 1424 } 1425 1426 failure: 1427 if (ver != NULL) { 1428 dns_db_closeversion(db, &ver, 1429 result == ISC_R_SUCCESS ? true : false); 1430 } 1431 1432 if (source.base != NULL) { 1433 isc_mem_put(j->mctx, source.base, source.length); 1434 } 1435 if (target.base != NULL) { 1436 isc_mem_put(j->mctx, target.base, target.length); 1437 } 1438 1439 dns_diff_clear(&diff); 1440 1441 INSIST(ver == NULL); 1442 1443 return (result); 1444 } 1445 1446 isc_result_t 1447 dns_journal_rollforward(isc_mem_t *mctx, dns_db_t *db, unsigned int options, 1448 const char *filename) { 1449 dns_journal_t *j; 1450 isc_result_t result; 1451 1452 REQUIRE(DNS_DB_VALID(db)); 1453 REQUIRE(filename != NULL); 1454 1455 j = NULL; 1456 result = dns_journal_open(mctx, filename, DNS_JOURNAL_READ, &j); 1457 if (result == ISC_R_NOTFOUND) { 1458 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), "no journal file, but " 1459 "that's OK"); 1460 return (DNS_R_NOJOURNAL); 1461 } 1462 if (result != ISC_R_SUCCESS) { 1463 return (result); 1464 } 1465 if (JOURNAL_EMPTY(&j->header)) { 1466 result = DNS_R_UPTODATE; 1467 } else { 1468 result = roll_forward(j, db, options); 1469 } 1470 1471 dns_journal_destroy(&j); 1472 1473 return (result); 1474 } 1475 1476 isc_result_t 1477 dns_journal_print(isc_mem_t *mctx, const char *filename, FILE *file) { 1478 dns_journal_t *j; 1479 isc_buffer_t source; /* Transaction data from disk */ 1480 isc_buffer_t target; /* Ditto after _fromwire check */ 1481 uint32_t start_serial; /* Database SOA serial */ 1482 uint32_t end_serial; /* Last journal SOA serial */ 1483 isc_result_t result; 1484 dns_diff_t diff; 1485 unsigned int n_soa = 0; 1486 unsigned int n_put = 0; 1487 1488 REQUIRE(filename != NULL); 1489 1490 j = NULL; 1491 result = dns_journal_open(mctx, filename, DNS_JOURNAL_READ, &j); 1492 if (result == ISC_R_NOTFOUND) { 1493 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), "no journal file"); 1494 return (DNS_R_NOJOURNAL); 1495 } 1496 1497 if (result != ISC_R_SUCCESS) { 1498 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1499 "journal open failure: %s: %s", 1500 isc_result_totext(result), filename); 1501 return (result); 1502 } 1503 1504 if (j->header.serialset) { 1505 fprintf(file, "Source serial = %u\n", j->header.sourceserial); 1506 } 1507 dns_diff_init(j->mctx, &diff); 1508 1509 /* 1510 * Set up empty initial buffers for unchecked and checked 1511 * wire format transaction data. They will be reallocated 1512 * later. 1513 */ 1514 isc_buffer_init(&source, NULL, 0); 1515 isc_buffer_init(&target, NULL, 0); 1516 1517 start_serial = dns_journal_first_serial(j); 1518 end_serial = dns_journal_last_serial(j); 1519 1520 CHECK(dns_journal_iter_init(j, start_serial, end_serial)); 1521 1522 for (result = dns_journal_first_rr(j); result == ISC_R_SUCCESS; 1523 result = dns_journal_next_rr(j)) 1524 { 1525 dns_name_t *name; 1526 uint32_t ttl; 1527 dns_rdata_t *rdata; 1528 dns_difftuple_t *tuple = NULL; 1529 1530 name = NULL; 1531 rdata = NULL; 1532 dns_journal_current_rr(j, &name, &ttl, &rdata); 1533 1534 if (rdata->type == dns_rdatatype_soa) { 1535 n_soa++; 1536 } 1537 1538 if (n_soa == 3) { 1539 n_soa = 1; 1540 } 1541 if (n_soa == 0) { 1542 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1543 "%s: journal file corrupt: missing " 1544 "initial SOA", 1545 j->filename); 1546 FAIL(ISC_R_UNEXPECTED); 1547 } 1548 CHECK(dns_difftuple_create( 1549 diff.mctx, n_soa == 1 ? DNS_DIFFOP_DEL : DNS_DIFFOP_ADD, 1550 name, ttl, rdata, &tuple)); 1551 dns_diff_append(&diff, &tuple); 1552 1553 if (++n_put > 100) { 1554 result = dns_diff_print(&diff, file); 1555 dns_diff_clear(&diff); 1556 n_put = 0; 1557 if (result != ISC_R_SUCCESS) { 1558 break; 1559 } 1560 } 1561 } 1562 if (result == ISC_R_NOMORE) { 1563 result = ISC_R_SUCCESS; 1564 } 1565 CHECK(result); 1566 1567 if (n_put != 0) { 1568 result = dns_diff_print(&diff, file); 1569 dns_diff_clear(&diff); 1570 } 1571 goto cleanup; 1572 1573 failure: 1574 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1575 "%s: cannot print: journal file corrupt", j->filename); 1576 1577 cleanup: 1578 if (source.base != NULL) { 1579 isc_mem_put(j->mctx, source.base, source.length); 1580 } 1581 if (target.base != NULL) { 1582 isc_mem_put(j->mctx, target.base, target.length); 1583 } 1584 1585 dns_diff_clear(&diff); 1586 dns_journal_destroy(&j); 1587 1588 return (result); 1589 } 1590 1591 /**************************************************************************/ 1592 /* 1593 * Miscellaneous accessors. 1594 */ 1595 bool 1596 dns_journal_empty(dns_journal_t *j) { 1597 return (JOURNAL_EMPTY(&j->header)); 1598 } 1599 1600 uint32_t 1601 dns_journal_first_serial(dns_journal_t *j) { 1602 return (j->header.begin.serial); 1603 } 1604 1605 uint32_t 1606 dns_journal_last_serial(dns_journal_t *j) { 1607 return (j->header.end.serial); 1608 } 1609 1610 void 1611 dns_journal_set_sourceserial(dns_journal_t *j, uint32_t sourceserial) { 1612 REQUIRE(j->state == JOURNAL_STATE_WRITE || 1613 j->state == JOURNAL_STATE_INLINE || 1614 j->state == JOURNAL_STATE_TRANSACTION); 1615 1616 j->header.sourceserial = sourceserial; 1617 j->header.serialset = true; 1618 if (j->state == JOURNAL_STATE_WRITE) { 1619 j->state = JOURNAL_STATE_INLINE; 1620 } 1621 } 1622 1623 bool 1624 dns_journal_get_sourceserial(dns_journal_t *j, uint32_t *sourceserial) { 1625 REQUIRE(sourceserial != NULL); 1626 1627 if (!j->header.serialset) { 1628 return (false); 1629 } 1630 *sourceserial = j->header.sourceserial; 1631 return (true); 1632 } 1633 1634 /**************************************************************************/ 1635 /* 1636 * Iteration support. 1637 * 1638 * When serving an outgoing IXFR, we transmit a part the journal starting 1639 * at the serial number in the IXFR request and ending at the serial 1640 * number that is current when the IXFR request arrives. The ending 1641 * serial number is not necessarily at the end of the journal: 1642 * the journal may grow while the IXFR is in progress, but we stop 1643 * when we reach the serial number that was current when the IXFR started. 1644 */ 1645 1646 static isc_result_t 1647 read_one_rr(dns_journal_t *j); 1648 1649 /* 1650 * Make sure the buffer 'b' is has at least 'size' bytes 1651 * allocated, and clear it. 1652 * 1653 * Requires: 1654 * Either b->base is NULL, or it points to b->length bytes of memory 1655 * previously allocated by isc_mem_get(). 1656 */ 1657 1658 static isc_result_t 1659 size_buffer(isc_mem_t *mctx, isc_buffer_t *b, unsigned size) { 1660 if (b->length < size) { 1661 void *mem = isc_mem_get(mctx, size); 1662 if (mem == NULL) { 1663 return (ISC_R_NOMEMORY); 1664 } 1665 if (b->base != NULL) { 1666 isc_mem_put(mctx, b->base, b->length); 1667 } 1668 b->base = mem; 1669 b->length = size; 1670 } 1671 isc_buffer_clear(b); 1672 return (ISC_R_SUCCESS); 1673 } 1674 1675 isc_result_t 1676 dns_journal_iter_init(dns_journal_t *j, uint32_t begin_serial, 1677 uint32_t end_serial) { 1678 isc_result_t result; 1679 1680 CHECK(journal_find(j, begin_serial, &j->it.bpos)); 1681 INSIST(j->it.bpos.serial == begin_serial); 1682 1683 CHECK(journal_find(j, end_serial, &j->it.epos)); 1684 INSIST(j->it.epos.serial == end_serial); 1685 1686 result = ISC_R_SUCCESS; 1687 failure: 1688 j->it.result = result; 1689 return (j->it.result); 1690 } 1691 1692 isc_result_t 1693 dns_journal_first_rr(dns_journal_t *j) { 1694 isc_result_t result; 1695 1696 /* 1697 * Seek to the beginning of the first transaction we are 1698 * interested in. 1699 */ 1700 CHECK(journal_seek(j, j->it.bpos.offset)); 1701 j->it.current_serial = j->it.bpos.serial; 1702 1703 j->it.xsize = 0; /* We have no transaction data yet... */ 1704 j->it.xpos = 0; /* ...and haven't used any of it. */ 1705 1706 return (read_one_rr(j)); 1707 1708 failure: 1709 return (result); 1710 } 1711 1712 static isc_result_t 1713 read_one_rr(dns_journal_t *j) { 1714 isc_result_t result; 1715 1716 dns_rdatatype_t rdtype; 1717 dns_rdataclass_t rdclass; 1718 unsigned int rdlen; 1719 uint32_t ttl; 1720 journal_xhdr_t xhdr; 1721 journal_rrhdr_t rrhdr; 1722 1723 if (j->offset > j->it.epos.offset) { 1724 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1725 "%s: journal corrupt: possible integer overflow", 1726 j->filename); 1727 return (ISC_R_UNEXPECTED); 1728 } 1729 if (j->offset == j->it.epos.offset) { 1730 return (ISC_R_NOMORE); 1731 } 1732 if (j->it.xpos == j->it.xsize) { 1733 /* 1734 * We are at a transaction boundary. 1735 * Read another transaction header. 1736 */ 1737 CHECK(journal_read_xhdr(j, &xhdr)); 1738 if (xhdr.size == 0) { 1739 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1740 "%s: journal corrupt: empty transaction", 1741 j->filename); 1742 FAIL(ISC_R_UNEXPECTED); 1743 } 1744 if (xhdr.serial0 != j->it.current_serial) { 1745 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1746 "%s: journal file corrupt: " 1747 "expected serial %u, got %u", 1748 j->filename, j->it.current_serial, 1749 xhdr.serial0); 1750 FAIL(ISC_R_UNEXPECTED); 1751 } 1752 j->it.xsize = xhdr.size; 1753 j->it.xpos = 0; 1754 } 1755 /* 1756 * Read an RR. 1757 */ 1758 CHECK(journal_read_rrhdr(j, &rrhdr)); 1759 /* 1760 * Perform a sanity check on the journal RR size. 1761 * The smallest possible RR has a 1-byte owner name 1762 * and a 10-byte header. The largest possible 1763 * RR has 65535 bytes of data, a header, and a maximum- 1764 * size owner name, well below 70 k total. 1765 */ 1766 if (rrhdr.size < 1 + 10 || rrhdr.size > 70000) { 1767 isc_log_write(JOURNAL_COMMON_LOGARGS, ISC_LOG_ERROR, 1768 "%s: journal corrupt: impossible RR size " 1769 "(%d bytes)", 1770 j->filename, rrhdr.size); 1771 FAIL(ISC_R_UNEXPECTED); 1772 } 1773 1774 CHECK(size_buffer(j->mctx, &j->it.source, rrhdr.size)); 1775 CHECK(journal_read(j, j->it.source.base, rrhdr.size)); 1776 isc_buffer_add(&j->it.source, rrhdr.size); 1777 1778 /* 1779 * The target buffer is made the same size 1780 * as the source buffer, with the assumption that when 1781 * no compression in present, the output of dns_*_fromwire() 1782 * is no larger than the input. 1783 */ 1784 CHECK(size_buffer(j->mctx, &j->it.target, rrhdr.size)); 1785 1786 /* 1787 * Parse the owner name. We don't know where it 1788 * ends yet, so we make the entire "remaining" 1789 * part of the buffer "active". 1790 */ 1791 isc_buffer_setactive(&j->it.source, 1792 j->it.source.used - j->it.source.current); 1793 CHECK(dns_name_fromwire(&j->it.name, &j->it.source, &j->it.dctx, 0, 1794 &j->it.target)); 1795 1796 /* 1797 * Check that the RR header is there, and parse it. 1798 */ 1799 if (isc_buffer_remaininglength(&j->it.source) < 10) { 1800 FAIL(DNS_R_FORMERR); 1801 } 1802 1803 rdtype = isc_buffer_getuint16(&j->it.source); 1804 rdclass = isc_buffer_getuint16(&j->it.source); 1805 ttl = isc_buffer_getuint32(&j->it.source); 1806 rdlen = isc_buffer_getuint16(&j->it.source); 1807 1808 /* 1809 * Parse the rdata. 1810 */ 1811 if (isc_buffer_remaininglength(&j->it.source) != rdlen) { 1812 FAIL(DNS_R_FORMERR); 1813 } 1814 isc_buffer_setactive(&j->it.source, rdlen); 1815 dns_rdata_reset(&j->it.rdata); 1816 CHECK(dns_rdata_fromwire(&j->it.rdata, rdclass, rdtype, &j->it.source, 1817 &j->it.dctx, 0, &j->it.target)); 1818 j->it.ttl = ttl; 1819 1820 j->it.xpos += sizeof(journal_rawrrhdr_t) + rrhdr.size; 1821 if (rdtype == dns_rdatatype_soa) { 1822 /* XXX could do additional consistency checks here */ 1823 j->it.current_serial = dns_soa_getserial(&j->it.rdata); 1824 } 1825 1826 result = ISC_R_SUCCESS; 1827 1828 failure: 1829 j->it.result = result; 1830 return (result); 1831 } 1832 1833 isc_result_t 1834 dns_journal_next_rr(dns_journal_t *j) { 1835 j->it.result = read_one_rr(j); 1836 return (j->it.result); 1837 } 1838 1839 void 1840 dns_journal_current_rr(dns_journal_t *j, dns_name_t **name, uint32_t *ttl, 1841 dns_rdata_t **rdata) { 1842 REQUIRE(j->it.result == ISC_R_SUCCESS); 1843 *name = &j->it.name; 1844 *ttl = j->it.ttl; 1845 *rdata = &j->it.rdata; 1846 } 1847 1848 /**************************************************************************/ 1849 /* 1850 * Generating diffs from databases 1851 */ 1852 1853 /* 1854 * Construct a diff containing all the RRs at the current name of the 1855 * database iterator 'dbit' in database 'db', version 'ver'. 1856 * Set '*name' to the current name, and append the diff to 'diff'. 1857 * All new tuples will have the operation 'op'. 1858 * 1859 * Requires: 'name' must have buffer large enough to hold the name. 1860 * Typically, a dns_fixedname_t would be used. 1861 */ 1862 static isc_result_t 1863 get_name_diff(dns_db_t *db, dns_dbversion_t *ver, isc_stdtime_t now, 1864 dns_dbiterator_t *dbit, dns_name_t *name, dns_diffop_t op, 1865 dns_diff_t *diff) { 1866 isc_result_t result; 1867 dns_dbnode_t *node = NULL; 1868 dns_rdatasetiter_t *rdsiter = NULL; 1869 dns_difftuple_t *tuple = NULL; 1870 1871 result = dns_dbiterator_current(dbit, &node, name); 1872 if (result != ISC_R_SUCCESS) { 1873 return (result); 1874 } 1875 1876 result = dns_db_allrdatasets(db, node, ver, now, &rdsiter); 1877 if (result != ISC_R_SUCCESS) { 1878 goto cleanup_node; 1879 } 1880 1881 for (result = dns_rdatasetiter_first(rdsiter); result == ISC_R_SUCCESS; 1882 result = dns_rdatasetiter_next(rdsiter)) 1883 { 1884 dns_rdataset_t rdataset; 1885 1886 dns_rdataset_init(&rdataset); 1887 dns_rdatasetiter_current(rdsiter, &rdataset); 1888 1889 for (result = dns_rdataset_first(&rdataset); 1890 result == ISC_R_SUCCESS; 1891 result = dns_rdataset_next(&rdataset)) 1892 { 1893 dns_rdata_t rdata = DNS_RDATA_INIT; 1894 dns_rdataset_current(&rdataset, &rdata); 1895 result = dns_difftuple_create(diff->mctx, op, name, 1896 rdataset.ttl, &rdata, 1897 &tuple); 1898 if (result != ISC_R_SUCCESS) { 1899 dns_rdataset_disassociate(&rdataset); 1900 goto cleanup_iterator; 1901 } 1902 dns_diff_append(diff, &tuple); 1903 } 1904 dns_rdataset_disassociate(&rdataset); 1905 if (result != ISC_R_NOMORE) { 1906 goto cleanup_iterator; 1907 } 1908 } 1909 if (result != ISC_R_NOMORE) { 1910 goto cleanup_iterator; 1911 } 1912 1913 result = ISC_R_SUCCESS; 1914 1915 cleanup_iterator: 1916 dns_rdatasetiter_destroy(&rdsiter); 1917 1918 cleanup_node: 1919 dns_db_detachnode(db, &node); 1920 1921 return (result); 1922 } 1923 1924 /* 1925 * Comparison function for use by dns_diff_subtract when sorting 1926 * the diffs to be subtracted. The sort keys are the rdata type 1927 * and the rdata itself. The owner name is ignored, because 1928 * it is known to be the same for all tuples. 1929 */ 1930 static int 1931 rdata_order(const void *av, const void *bv) { 1932 dns_difftuple_t const *const *ap = av; 1933 dns_difftuple_t const *const *bp = bv; 1934 dns_difftuple_t const *a = *ap; 1935 dns_difftuple_t const *b = *bp; 1936 int r; 1937 r = (b->rdata.type - a->rdata.type); 1938 if (r != 0) { 1939 return (r); 1940 } 1941 r = dns_rdata_compare(&a->rdata, &b->rdata); 1942 return (r); 1943 } 1944 1945 static isc_result_t 1946 dns_diff_subtract(dns_diff_t diff[2], dns_diff_t *r) { 1947 isc_result_t result; 1948 dns_difftuple_t *p[2]; 1949 int i, t; 1950 bool append; 1951 1952 CHECK(dns_diff_sort(&diff[0], rdata_order)); 1953 CHECK(dns_diff_sort(&diff[1], rdata_order)); 1954 1955 for (;;) { 1956 p[0] = ISC_LIST_HEAD(diff[0].tuples); 1957 p[1] = ISC_LIST_HEAD(diff[1].tuples); 1958 if (p[0] == NULL && p[1] == NULL) { 1959 break; 1960 } 1961 1962 for (i = 0; i < 2; i++) { 1963 if (p[!i] == NULL) { 1964 { 1965 ISC_LIST_UNLINK(diff[i].tuples, p[i], 1966 link); 1967 ISC_LIST_APPEND(r->tuples, p[i], link); 1968 goto next; 1969 } 1970 } 1971 } 1972 t = rdata_order(&p[0], &p[1]); 1973 if (t < 0) { 1974 ISC_LIST_UNLINK(diff[0].tuples, p[0], link); 1975 ISC_LIST_APPEND(r->tuples, p[0], link); 1976 goto next; 1977 } 1978 if (t > 0) { 1979 ISC_LIST_UNLINK(diff[1].tuples, p[1], link); 1980 ISC_LIST_APPEND(r->tuples, p[1], link); 1981 goto next; 1982 } 1983 INSIST(t == 0); 1984 /* 1985 * Identical RRs in both databases; skip them both 1986 * if the ttl differs. 1987 */ 1988 append = (p[0]->ttl != p[1]->ttl); 1989 for (i = 0; i < 2; i++) { 1990 ISC_LIST_UNLINK(diff[i].tuples, p[i], link); 1991 if (append) { 1992 ISC_LIST_APPEND(r->tuples, p[i], link); 1993 } else { 1994 dns_difftuple_free(&p[i]); 1995 } 1996 } 1997 next:; 1998 } 1999 result = ISC_R_SUCCESS; 2000 failure: 2001 return (result); 2002 } 2003 2004 static isc_result_t 2005 diff_namespace(dns_db_t *dba, dns_dbversion_t *dbvera, dns_db_t *dbb, 2006 dns_dbversion_t *dbverb, unsigned int options, 2007 dns_diff_t *resultdiff) { 2008 dns_db_t *db[2]; 2009 dns_dbversion_t *ver[2]; 2010 dns_dbiterator_t *dbit[2] = { NULL, NULL }; 2011 bool have[2] = { false, false }; 2012 dns_fixedname_t fixname[2]; 2013 isc_result_t result, itresult[2]; 2014 dns_diff_t diff[2]; 2015 int i, t; 2016 2017 db[0] = dba, db[1] = dbb; 2018 ver[0] = dbvera, ver[1] = dbverb; 2019 2020 dns_diff_init(resultdiff->mctx, &diff[0]); 2021 dns_diff_init(resultdiff->mctx, &diff[1]); 2022 2023 dns_fixedname_init(&fixname[0]); 2024 dns_fixedname_init(&fixname[1]); 2025 2026 result = dns_db_createiterator(db[0], options, &dbit[0]); 2027 if (result != ISC_R_SUCCESS) { 2028 return (result); 2029 } 2030 result = dns_db_createiterator(db[1], options, &dbit[1]); 2031 if (result != ISC_R_SUCCESS) { 2032 goto cleanup_iterator; 2033 } 2034 2035 itresult[0] = dns_dbiterator_first(dbit[0]); 2036 itresult[1] = dns_dbiterator_first(dbit[1]); 2037 2038 for (;;) { 2039 for (i = 0; i < 2; i++) { 2040 if (!have[i] && itresult[i] == ISC_R_SUCCESS) { 2041 CHECK(get_name_diff( 2042 db[i], ver[i], 0, dbit[i], 2043 dns_fixedname_name(&fixname[i]), 2044 i == 0 ? DNS_DIFFOP_ADD 2045 : DNS_DIFFOP_DEL, 2046 &diff[i])); 2047 itresult[i] = dns_dbiterator_next(dbit[i]); 2048 have[i] = true; 2049 } 2050 } 2051 2052 if (!have[0] && !have[1]) { 2053 INSIST(ISC_LIST_EMPTY(diff[0].tuples)); 2054 INSIST(ISC_LIST_EMPTY(diff[1].tuples)); 2055 break; 2056 } 2057 2058 for (i = 0; i < 2; i++) { 2059 if (!have[!i]) { 2060 ISC_LIST_APPENDLIST(resultdiff->tuples, 2061 diff[i].tuples, link); 2062 INSIST(ISC_LIST_EMPTY(diff[i].tuples)); 2063 have[i] = false; 2064 goto next; 2065 } 2066 } 2067 2068 t = dns_name_compare(dns_fixedname_name(&fixname[0]), 2069 dns_fixedname_name(&fixname[1])); 2070 if (t < 0) { 2071 ISC_LIST_APPENDLIST(resultdiff->tuples, diff[0].tuples, 2072 link); 2073 INSIST(ISC_LIST_EMPTY(diff[0].tuples)); 2074 have[0] = false; 2075 continue; 2076 } 2077 if (t > 0) { 2078 ISC_LIST_APPENDLIST(resultdiff->tuples, diff[1].tuples, 2079 link); 2080 INSIST(ISC_LIST_EMPTY(diff[1].tuples)); 2081 have[1] = false; 2082 continue; 2083 } 2084 INSIST(t == 0); 2085 CHECK(dns_diff_subtract(diff, resultdiff)); 2086 INSIST(ISC_LIST_EMPTY(diff[0].tuples)); 2087 INSIST(ISC_LIST_EMPTY(diff[1].tuples)); 2088 have[0] = have[1] = false; 2089 next:; 2090 } 2091 if (itresult[0] != ISC_R_NOMORE) { 2092 FAIL(itresult[0]); 2093 } 2094 if (itresult[1] != ISC_R_NOMORE) { 2095 FAIL(itresult[1]); 2096 } 2097 2098 INSIST(ISC_LIST_EMPTY(diff[0].tuples)); 2099 INSIST(ISC_LIST_EMPTY(diff[1].tuples)); 2100 2101 failure: 2102 dns_dbiterator_destroy(&dbit[1]); 2103 2104 cleanup_iterator: 2105 dns_dbiterator_destroy(&dbit[0]); 2106 dns_diff_clear(&diff[0]); 2107 dns_diff_clear(&diff[1]); 2108 return (result); 2109 } 2110 2111 /* 2112 * Compare the databases 'dba' and 'dbb' and generate a journal 2113 * entry containing the changes to make 'dba' from 'dbb' (note 2114 * the order). This journal entry will consist of a single, 2115 * possibly very large transaction. 2116 */ 2117 isc_result_t 2118 dns_db_diff(isc_mem_t *mctx, dns_db_t *dba, dns_dbversion_t *dbvera, 2119 dns_db_t *dbb, dns_dbversion_t *dbverb, const char *filename) { 2120 isc_result_t result; 2121 dns_diff_t diff; 2122 2123 dns_diff_init(mctx, &diff); 2124 2125 result = dns_db_diffx(&diff, dba, dbvera, dbb, dbverb, filename); 2126 2127 dns_diff_clear(&diff); 2128 2129 return (result); 2130 } 2131 2132 isc_result_t 2133 dns_db_diffx(dns_diff_t *diff, dns_db_t *dba, dns_dbversion_t *dbvera, 2134 dns_db_t *dbb, dns_dbversion_t *dbverb, const char *filename) { 2135 isc_result_t result; 2136 dns_journal_t *journal = NULL; 2137 2138 if (filename != NULL) { 2139 result = dns_journal_open(diff->mctx, filename, 2140 DNS_JOURNAL_CREATE, &journal); 2141 if (result != ISC_R_SUCCESS) { 2142 return (result); 2143 } 2144 } 2145 2146 CHECK(diff_namespace(dba, dbvera, dbb, dbverb, DNS_DB_NONSEC3, diff)); 2147 CHECK(diff_namespace(dba, dbvera, dbb, dbverb, DNS_DB_NSEC3ONLY, diff)); 2148 2149 if (journal != NULL) { 2150 if (ISC_LIST_EMPTY(diff->tuples)) { 2151 isc_log_write(JOURNAL_DEBUG_LOGARGS(3), "no changes"); 2152 } else { 2153 CHECK(dns_journal_write_transaction(journal, diff)); 2154 } 2155 } 2156 2157 failure: 2158 if (journal != NULL) { 2159 dns_journal_destroy(&journal); 2160 } 2161 return (result); 2162 } 2163 2164 isc_result_t 2165 dns_journal_compact(isc_mem_t *mctx, char *filename, uint32_t serial, 2166 uint32_t target_size) { 2167 unsigned int i; 2168 journal_pos_t best_guess; 2169 journal_pos_t current_pos; 2170 dns_journal_t *j1 = NULL; 2171 dns_journal_t *j2 = NULL; 2172 journal_rawheader_t rawheader; 2173 unsigned int copy_length; 2174 size_t namelen; 2175 char *buf = NULL; 2176 unsigned int size = 0; 2177 isc_result_t result; 2178 unsigned int indexend; 2179 char newname[PATH_MAX]; 2180 char backup[PATH_MAX]; 2181 bool is_backup = false; 2182 2183 REQUIRE(filename != NULL); 2184 2185 namelen = strlen(filename); 2186 if (namelen > 4U && strcmp(filename + namelen - 4, ".jnl") == 0) { 2187 namelen -= 4; 2188 } 2189 2190 result = snprintf(newname, sizeof(newname), "%.*s.jnw", (int)namelen, 2191 filename); 2192 RUNTIME_CHECK(result < sizeof(newname)); 2193 2194 result = snprintf(backup, sizeof(backup), "%.*s.jbk", (int)namelen, 2195 filename); 2196 RUNTIME_CHECK(result < sizeof(backup)); 2197 2198 result = journal_open(mctx, filename, false, false, &j1); 2199 if (result == ISC_R_NOTFOUND) { 2200 is_backup = true; 2201 result = journal_open(mctx, backup, false, false, &j1); 2202 } 2203 if (result != ISC_R_SUCCESS) { 2204 return (result); 2205 } 2206 2207 if (JOURNAL_EMPTY(&j1->header)) { 2208 dns_journal_destroy(&j1); 2209 return (ISC_R_SUCCESS); 2210 } 2211 2212 if (DNS_SERIAL_GT(j1->header.begin.serial, serial) || 2213 DNS_SERIAL_GT(serial, j1->header.end.serial)) 2214 { 2215 dns_journal_destroy(&j1); 2216 return (ISC_R_RANGE); 2217 } 2218 2219 /* 2220 * Cope with very small target sizes. 2221 */ 2222 indexend = sizeof(journal_rawheader_t) + 2223 j1->header.index_size * sizeof(journal_rawpos_t); 2224 if (target_size < DNS_JOURNAL_SIZE_MIN) { 2225 target_size = DNS_JOURNAL_SIZE_MIN; 2226 } 2227 if (target_size < indexend * 2) { 2228 target_size = target_size / 2 + indexend; 2229 } 2230 2231 /* 2232 * See if there is any work to do. 2233 */ 2234 if ((uint32_t)j1->header.end.offset < target_size) { 2235 dns_journal_destroy(&j1); 2236 return (ISC_R_SUCCESS); 2237 } 2238 2239 CHECK(journal_open(mctx, newname, true, true, &j2)); 2240 2241 /* 2242 * Remove overhead so space test below can succeed. 2243 */ 2244 if (target_size >= indexend) { 2245 target_size -= indexend; 2246 } 2247 2248 /* 2249 * Find if we can create enough free space. 2250 */ 2251 best_guess = j1->header.begin; 2252 for (i = 0; i < j1->header.index_size; i++) { 2253 if (POS_VALID(j1->index[i]) && 2254 DNS_SERIAL_GE(serial, j1->index[i].serial) && 2255 ((uint32_t)(j1->header.end.offset - j1->index[i].offset) >= 2256 target_size / 2) && 2257 j1->index[i].offset > best_guess.offset) 2258 { 2259 best_guess = j1->index[i]; 2260 } 2261 } 2262 2263 current_pos = best_guess; 2264 while (current_pos.serial != serial) { 2265 CHECK(journal_next(j1, ¤t_pos)); 2266 if (current_pos.serial == j1->header.end.serial) { 2267 break; 2268 } 2269 2270 if (DNS_SERIAL_GE(serial, current_pos.serial) && 2271 ((uint32_t)(j1->header.end.offset - current_pos.offset) >= 2272 (target_size / 2)) && 2273 current_pos.offset > best_guess.offset) 2274 { 2275 best_guess = current_pos; 2276 } else { 2277 break; 2278 } 2279 } 2280 2281 INSIST(best_guess.serial != j1->header.end.serial); 2282 if (best_guess.serial != serial) { 2283 CHECK(journal_next(j1, &best_guess)); 2284 } 2285 2286 /* 2287 * We should now be roughly half target_size provided 2288 * we did not reach 'serial'. If not we will just copy 2289 * all uncommitted deltas regardless of the size. 2290 */ 2291 copy_length = j1->header.end.offset - best_guess.offset; 2292 2293 if (copy_length != 0) { 2294 /* 2295 * Copy best_guess to end into space just freed. 2296 */ 2297 size = 64 * 1024; 2298 if (copy_length < size) { 2299 size = copy_length; 2300 } 2301 buf = isc_mem_get(mctx, size); 2302 2303 CHECK(journal_seek(j1, best_guess.offset)); 2304 CHECK(journal_seek(j2, indexend)); 2305 for (i = 0; i < copy_length; i += size) { 2306 unsigned int len = (copy_length - i) > size 2307 ? size 2308 : (copy_length - i); 2309 CHECK(journal_read(j1, buf, len)); 2310 CHECK(journal_write(j2, buf, len)); 2311 } 2312 2313 CHECK(journal_fsync(j2)); 2314 2315 /* 2316 * Compute new header. 2317 */ 2318 j2->header.begin.serial = best_guess.serial; 2319 j2->header.begin.offset = indexend; 2320 j2->header.end.serial = j1->header.end.serial; 2321 j2->header.end.offset = indexend + copy_length; 2322 j2->header.sourceserial = j1->header.sourceserial; 2323 j2->header.serialset = j1->header.serialset; 2324 2325 /* 2326 * Update the journal header. 2327 */ 2328 journal_header_encode(&j2->header, &rawheader); 2329 CHECK(journal_seek(j2, 0)); 2330 CHECK(journal_write(j2, &rawheader, sizeof(rawheader))); 2331 CHECK(journal_fsync(j2)); 2332 2333 /* 2334 * Build new index. 2335 */ 2336 current_pos = j2->header.begin; 2337 while (current_pos.serial != j2->header.end.serial) { 2338 index_add(j2, ¤t_pos); 2339 CHECK(journal_next(j2, ¤t_pos)); 2340 } 2341 2342 /* 2343 * Write index. 2344 */ 2345 CHECK(index_to_disk(j2)); 2346 CHECK(journal_fsync(j2)); 2347 2348 indexend = j2->header.end.offset; 2349 POST(indexend); 2350 } 2351 2352 /* 2353 * Close both journals before trying to rename files (this is 2354 * necessary on WIN32). 2355 */ 2356 dns_journal_destroy(&j1); 2357 dns_journal_destroy(&j2); 2358 2359 /* 2360 * With a UFS file system this should just succeed and be atomic. 2361 * Any IXFR outs will just continue and the old journal will be 2362 * removed on final close. 2363 * 2364 * With MSDOS / NTFS we need to do a two stage rename, triggered 2365 * by EEXIST. (If any IXFR's are running in other threads, however, 2366 * this will fail, and the journal will not be compacted. But 2367 * if so, hopefully they'll be finished by the next time we 2368 * compact.) 2369 */ 2370 if (rename(newname, filename) == -1) { 2371 if (errno == EEXIST && !is_backup) { 2372 result = isc_file_remove(backup); 2373 if (result != ISC_R_SUCCESS && 2374 result != ISC_R_FILENOTFOUND) { 2375 goto failure; 2376 } 2377 if (rename(filename, backup) == -1) { 2378 goto maperrno; 2379 } 2380 if (rename(newname, filename) == -1) { 2381 goto maperrno; 2382 } 2383 (void)isc_file_remove(backup); 2384 } else { 2385 maperrno: 2386 result = ISC_R_FAILURE; 2387 goto failure; 2388 } 2389 } 2390 2391 result = ISC_R_SUCCESS; 2392 2393 failure: 2394 (void)isc_file_remove(newname); 2395 if (buf != NULL) { 2396 isc_mem_put(mctx, buf, size); 2397 } 2398 if (j1 != NULL) { 2399 dns_journal_destroy(&j1); 2400 } 2401 if (j2 != NULL) { 2402 dns_journal_destroy(&j2); 2403 } 2404 return (result); 2405 } 2406 2407 static isc_result_t 2408 index_to_disk(dns_journal_t *j) { 2409 isc_result_t result = ISC_R_SUCCESS; 2410 2411 if (j->header.index_size != 0) { 2412 unsigned int i; 2413 unsigned char *p; 2414 unsigned int rawbytes; 2415 2416 rawbytes = j->header.index_size * sizeof(journal_rawpos_t); 2417 2418 p = j->rawindex; 2419 for (i = 0; i < j->header.index_size; i++) { 2420 encode_uint32(j->index[i].serial, p); 2421 p += 4; 2422 encode_uint32(j->index[i].offset, p); 2423 p += 4; 2424 } 2425 INSIST(p == j->rawindex + rawbytes); 2426 2427 CHECK(journal_seek(j, sizeof(journal_rawheader_t))); 2428 CHECK(journal_write(j, j->rawindex, rawbytes)); 2429 } 2430 failure: 2431 return (result); 2432 } 2433