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