Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * xlogreader.c
4 : : * Generic XLog reading facility
5 : : *
6 : : * Portions Copyright (c) 2013-2026, PostgreSQL Global Development Group
7 : : *
8 : : * IDENTIFICATION
9 : : * src/backend/access/transam/xlogreader.c
10 : : *
11 : : * NOTES
12 : : * See xlogreader.h for more notes on this facility.
13 : : *
14 : : * This file is compiled as both front-end and backend code, so it
15 : : * may not use ereport, server-defined static variables, etc.
16 : : *-------------------------------------------------------------------------
17 : : */
18 : : #include "postgres.h"
19 : :
20 : : #include <unistd.h>
21 : : #ifdef USE_LZ4
22 : : #include <lz4.h>
23 : : #endif
24 : : #ifdef USE_ZSTD
25 : : #include <zstd.h>
26 : : #endif
27 : :
28 : : #include "access/transam.h"
29 : : #include "access/xlog_internal.h"
30 : : #include "access/xlogreader.h"
31 : : #include "access/xlogrecord.h"
32 : : #include "catalog/pg_control.h"
33 : : #include "common/pg_lzcompress.h"
34 : : #include "replication/origin.h"
35 : :
36 : : #ifndef FRONTEND
37 : : #include "pgstat.h"
38 : : #include "storage/bufmgr.h"
39 : : #include "utils/wait_event.h"
40 : : #else
41 : : #include "common/logging.h"
42 : : #endif
43 : :
44 : : static void report_invalid_record(XLogReaderState *state, const char *fmt,...)
45 : : pg_attribute_printf(2, 3);
46 : : static void allocate_recordbuf(XLogReaderState *state, uint32 reclength);
47 : : static int ReadPageInternal(XLogReaderState *state, XLogRecPtr pageptr,
48 : : int reqLen);
49 : : static void XLogReaderInvalReadState(XLogReaderState *state);
50 : : static XLogPageReadResult XLogDecodeNextRecord(XLogReaderState *state, bool nonblocking);
51 : : static bool ValidXLogRecordHeader(XLogReaderState *state, XLogRecPtr RecPtr,
52 : : XLogRecPtr PrevRecPtr, XLogRecord *record, bool randAccess);
53 : : static bool ValidXLogRecord(XLogReaderState *state, XLogRecord *record,
54 : : XLogRecPtr recptr);
55 : : static void ResetDecoder(XLogReaderState *state);
56 : : static void WALOpenSegmentInit(WALOpenSegment *seg, WALSegmentContext *segcxt,
57 : : int segsize, const char *waldir);
58 : :
59 : : /* size of the buffer allocated for error message. */
60 : : #define MAX_ERRORMSG_LEN 1000
61 : :
62 : : /*
63 : : * Default size; large enough that typical users of XLogReader won't often need
64 : : * to use the 'oversized' memory allocation code path.
65 : : */
66 : : #define DEFAULT_DECODE_BUFFER_SIZE (64 * 1024)
67 : :
68 : : /*
69 : : * Construct a string in state->errormsg_buf explaining what's wrong with
70 : : * the current record being read.
71 : : */
72 : : static void
4857 alvherre@alvh.no-ip. 73 :CBC 286 : report_invalid_record(XLogReaderState *state, const char *fmt,...)
74 : : {
75 : : va_list args;
76 : :
77 : 286 : fmt = _(fmt);
78 : :
79 : 286 : va_start(args, fmt);
80 : 286 : vsnprintf(state->errormsg_buf, MAX_ERRORMSG_LEN, fmt, args);
81 : 286 : va_end(args);
82 : :
1509 tmunro@postgresql.or 83 : 286 : state->errormsg_deferred = true;
84 : 286 : }
85 : :
86 : : /*
87 : : * Set the size of the decoding buffer. A pointer to a caller supplied memory
88 : : * region may also be passed in, in which case non-oversized records will be
89 : : * decoded there.
90 : : */
91 : : void
92 : 1081 : XLogReaderSetDecodeBuffer(XLogReaderState *state, void *buffer, size_t size)
93 : : {
94 [ - + ]: 1081 : Assert(state->decode_buffer == NULL);
95 : :
96 : 1081 : state->decode_buffer = buffer;
97 : 1081 : state->decode_buffer_size = size;
98 : 1081 : state->decode_buffer_tail = buffer;
99 : 1081 : state->decode_buffer_head = buffer;
4857 alvherre@alvh.no-ip. 100 : 1081 : }
101 : :
102 : : /*
103 : : * Allocate and initialize a new XLogReader.
104 : : *
105 : : * Returns NULL if the xlogreader couldn't be allocated.
106 : : */
107 : : XLogReaderState *
2415 108 : 3154 : XLogReaderAllocate(int wal_segment_size, const char *waldir,
109 : : XLogReaderRoutine *routine, void *private_data)
110 : : {
111 : : XLogReaderState *state;
112 : :
113 : : state = (XLogReaderState *)
4050 fujii@postgresql.org 114 : 3154 : palloc_extended(sizeof(XLogReaderState),
115 : : MCXT_ALLOC_NO_OOM | MCXT_ALLOC_ZERO);
116 [ - + ]: 3154 : if (!state)
4050 fujii@postgresql.org 117 :UBC 0 : return NULL;
118 : :
119 : : /* initialize caller-provided support functions */
1821 tmunro@postgresql.or 120 :CBC 3154 : state->routine = *routine;
121 : :
122 : : /*
123 : : * Permanently allocate readBuf. We do it this way, rather than just
124 : : * making a static array, for two reasons: (1) no need to waste the
125 : : * storage in most instantiations of the backend; (2) a static char array
126 : : * isn't guaranteed to have any particular alignment, whereas
127 : : * palloc_extended() will provide MAXALIGN'd storage.
128 : : */
4050 fujii@postgresql.org 129 : 3154 : state->readBuf = (char *) palloc_extended(XLOG_BLCKSZ,
130 : : MCXT_ALLOC_NO_OOM);
131 [ - + ]: 3154 : if (!state->readBuf)
132 : : {
4050 fujii@postgresql.org 133 :UBC 0 : pfree(state);
134 : 0 : return NULL;
135 : : }
136 : :
137 : : /* Initialize segment info. */
2415 alvherre@alvh.no-ip. 138 :CBC 3154 : WALOpenSegmentInit(&state->seg, &state->segcxt, wal_segment_size,
139 : : waldir);
140 : :
141 : : /* system_identifier initialized to zeroes above */
1821 tmunro@postgresql.or 142 : 3154 : state->private_data = private_data;
143 : : /* ReadRecPtr, EndRecPtr and readLen initialized to zeroes above */
4050 fujii@postgresql.org 144 : 3154 : state->errormsg_buf = palloc_extended(MAX_ERRORMSG_LEN + 1,
145 : : MCXT_ALLOC_NO_OOM);
146 [ - + ]: 3154 : if (!state->errormsg_buf)
147 : : {
4050 fujii@postgresql.org 148 :UBC 0 : pfree(state->readBuf);
149 : 0 : pfree(state);
150 : 0 : return NULL;
151 : : }
4857 alvherre@alvh.no-ip. 152 :CBC 3154 : state->errormsg_buf[0] = '\0';
153 : :
154 : : /*
155 : : * Allocate an initial readRecordBuf of minimal size, which can later be
156 : : * enlarged if necessary.
157 : : */
945 michael@paquier.xyz 158 : 3154 : allocate_recordbuf(state, 0);
4857 alvherre@alvh.no-ip. 159 : 3154 : return state;
160 : : }
161 : :
162 : : void
163 : 2530 : XLogReaderFree(XLogReaderState *state)
164 : : {
1821 tmunro@postgresql.or 165 [ + + ]: 2530 : if (state->seg.ws_file != -1)
166 : 1388 : state->routine.segment_close(state);
167 : :
1509 168 [ + + + - ]: 2530 : if (state->decode_buffer && state->free_decode_buffer)
169 : 2496 : pfree(state->decode_buffer);
170 : :
4184 heikki.linnakangas@i 171 : 2530 : pfree(state->errormsg_buf);
4857 alvherre@alvh.no-ip. 172 [ + - ]: 2530 : if (state->readRecordBuf)
4184 heikki.linnakangas@i 173 : 2530 : pfree(state->readRecordBuf);
174 : 2530 : pfree(state->readBuf);
175 : 2530 : pfree(state);
4857 alvherre@alvh.no-ip. 176 : 2530 : }
177 : :
178 : : /*
179 : : * Allocate readRecordBuf to fit a record of at least the given length.
180 : : *
181 : : * readRecordBufSize is set to the new buffer size.
182 : : *
183 : : * To avoid useless small increases, round its size to a multiple of
184 : : * XLOG_BLCKSZ, and make sure it's at least 5*Max(BLCKSZ, XLOG_BLCKSZ) to start
185 : : * with. (That is enough for all "normal" records, but very large commit or
186 : : * abort records might need more space.)
187 : : *
188 : : * Note: This routine should *never* be called for xl_tot_len until the header
189 : : * of the record has been fully validated.
190 : : */
191 : : static void
192 : 3236 : allocate_recordbuf(XLogReaderState *state, uint32 reclength)
193 : : {
194 : 3236 : uint32 newSize = reclength;
195 : :
196 : 3236 : newSize += XLOG_BLCKSZ - (newSize % XLOG_BLCKSZ);
197 : 3236 : newSize = Max(newSize, 5 * Max(BLCKSZ, XLOG_BLCKSZ));
198 : :
199 [ + + ]: 3236 : if (state->readRecordBuf)
4184 heikki.linnakangas@i 200 : 82 : pfree(state->readRecordBuf);
945 michael@paquier.xyz 201 : 3236 : state->readRecordBuf = (char *) palloc(newSize);
4857 alvherre@alvh.no-ip. 202 : 3236 : state->readRecordBufSize = newSize;
203 : 3236 : }
204 : :
205 : : /*
206 : : * Initialize the passed segment structs.
207 : : */
208 : : static void
2415 209 : 3154 : WALOpenSegmentInit(WALOpenSegment *seg, WALSegmentContext *segcxt,
210 : : int segsize, const char *waldir)
211 : : {
212 : 3154 : seg->ws_file = -1;
213 : 3154 : seg->ws_segno = 0;
214 : 3154 : seg->ws_tli = 0;
215 : :
216 : 3154 : segcxt->ws_segsize = segsize;
217 [ + + ]: 3154 : if (waldir)
218 : 172 : snprintf(segcxt->ws_dir, MAXPGPATH, "%s", waldir);
219 : 3154 : }
220 : :
221 : : /*
222 : : * Begin reading WAL at 'RecPtr'.
223 : : *
224 : : * 'RecPtr' should point to the beginning of a valid WAL record. Pointing at
225 : : * the beginning of a page is also OK, if there is a new record right after
226 : : * the page header, i.e. not a continuation.
227 : : *
228 : : * This does not make any attempt to read the WAL yet, and hence cannot fail.
229 : : * If the starting address is not correct, the first call to XLogReadRecord()
230 : : * will error out.
231 : : */
232 : : void
2291 heikki.linnakangas@i 233 : 7018 : XLogBeginRead(XLogReaderState *state, XLogRecPtr RecPtr)
234 : : {
180 alvherre@kurilemu.de 235 [ - + ]:GNC 7018 : Assert(XLogRecPtrIsValid(RecPtr));
236 : :
2291 heikki.linnakangas@i 237 :CBC 7018 : ResetDecoder(state);
238 : :
239 : : /* Begin at the passed-in record pointer. */
240 : 7018 : state->EndRecPtr = RecPtr;
1509 tmunro@postgresql.or 241 : 7018 : state->NextRecPtr = RecPtr;
2291 heikki.linnakangas@i 242 : 7018 : state->ReadRecPtr = InvalidXLogRecPtr;
1509 tmunro@postgresql.or 243 : 7018 : state->DecodeRecPtr = InvalidXLogRecPtr;
244 : 7018 : }
245 : :
246 : : /*
247 : : * Release the last record that was returned by XLogNextRecord(), if any, to
248 : : * free up space. Returns the LSN past the end of the record.
249 : : */
250 : : XLogRecPtr
251 : 13879808 : XLogReleasePreviousRecord(XLogReaderState *state)
252 : : {
253 : : DecodedXLogRecord *record;
254 : : XLogRecPtr next_lsn;
255 : :
256 [ + + ]: 13879808 : if (!state->record)
1335 257 : 6950091 : return InvalidXLogRecPtr;
258 : :
259 : : /*
260 : : * Remove it from the decoded record queue. It must be the oldest item
261 : : * decoded, decode_queue_head.
262 : : */
1509 263 : 6929717 : record = state->record;
1335 264 : 6929717 : next_lsn = record->next_lsn;
1509 265 [ - + ]: 6929717 : Assert(record == state->decode_queue_head);
266 : 6929717 : state->record = NULL;
267 : 6929717 : state->decode_queue_head = record->next;
268 : :
269 : : /* It might also be the newest item decoded, decode_queue_tail. */
270 [ + + ]: 6929717 : if (state->decode_queue_tail == record)
271 : 4022350 : state->decode_queue_tail = NULL;
272 : :
273 : : /* Release the space. */
274 [ + + ]: 6929717 : if (unlikely(record->oversized))
275 : : {
276 : : /* It's not in the decode buffer, so free it to release space. */
277 : 100 : pfree(record);
278 : : }
279 : : else
280 : : {
281 : : /* It must be the head (oldest) record in the decode buffer. */
282 [ - + ]: 6929617 : Assert(state->decode_buffer_head == (char *) record);
283 : :
284 : : /*
285 : : * We need to update head to point to the next record that is in the
286 : : * decode buffer, if any, being careful to skip oversized ones
287 : : * (they're not in the decode buffer).
288 : : */
289 : 6929617 : record = record->next;
290 [ + + - + : 6929617 : while (unlikely(record && record->oversized))
- + ]
1509 tmunro@postgresql.or 291 :UBC 0 : record = record->next;
292 : :
1509 tmunro@postgresql.or 293 [ + + ]:CBC 6929617 : if (record)
294 : : {
295 : : /* Adjust head to release space up to the next record. */
296 : 2907367 : state->decode_buffer_head = (char *) record;
297 : : }
298 : : else
299 : : {
300 : : /*
301 : : * Otherwise we might as well just reset head and tail to the
302 : : * start of the buffer space, because we're empty. This means
303 : : * we'll keep overwriting the same piece of memory if we're not
304 : : * doing any prefetching.
305 : : */
306 : 4022250 : state->decode_buffer_head = state->decode_buffer;
307 : 4022250 : state->decode_buffer_tail = state->decode_buffer;
308 : : }
309 : : }
310 : :
1335 311 : 6929717 : return next_lsn;
312 : : }
313 : :
314 : : /*
315 : : * Attempt to read an XLOG record.
316 : : *
317 : : * XLogBeginRead() or XLogFindNextRecord() and then XLogReadAhead() must be
318 : : * called before the first call to XLogNextRecord(). This functions returns
319 : : * records and errors that were put into an internal queue by XLogReadAhead().
320 : : *
321 : : * On success, a record is returned.
322 : : *
323 : : * The returned record (or *errormsg) points to an internal buffer that's
324 : : * valid until the next call to XLogNextRecord.
325 : : */
326 : : DecodedXLogRecord *
1509 327 : 6939764 : XLogNextRecord(XLogReaderState *state, char **errormsg)
328 : : {
329 : : /* Release the last record returned by XLogNextRecord(). */
330 : 6939764 : XLogReleasePreviousRecord(state);
331 : :
332 [ + + ]: 6939764 : if (state->decode_queue_head == NULL)
333 : : {
334 : 3815 : *errormsg = NULL;
335 [ + + ]: 3815 : if (state->errormsg_deferred)
336 : : {
337 [ + + ]: 285 : if (state->errormsg_buf[0] != '\0')
338 : 280 : *errormsg = state->errormsg_buf;
339 : 285 : state->errormsg_deferred = false;
340 : : }
341 : :
342 : : /*
343 : : * state->EndRecPtr is expected to have been set by the last call to
344 : : * XLogBeginRead() or XLogNextRecord(), and is the location of the
345 : : * error.
346 : : */
180 alvherre@kurilemu.de 347 [ - + ]:GNC 3815 : Assert(XLogRecPtrIsValid(state->EndRecPtr));
348 : :
1509 tmunro@postgresql.or 349 :CBC 3815 : return NULL;
350 : : }
351 : :
352 : : /*
353 : : * Record this as the most recent record returned, so that we'll release
354 : : * it next time. This also exposes it to the traditional
355 : : * XLogRecXXX(xlogreader) macros, which work with the decoder rather than
356 : : * the record for historical reasons.
357 : : */
358 : 6935949 : state->record = state->decode_queue_head;
359 : :
360 : : /*
361 : : * Update the pointers to the beginning and one-past-the-end of this
362 : : * record, again for the benefit of historical code that expected the
363 : : * decoder to track this rather than accessing these fields of the record
364 : : * itself.
365 : : */
366 : 6935949 : state->ReadRecPtr = state->record->lsn;
367 : 6935949 : state->EndRecPtr = state->record->next_lsn;
368 : :
369 : 6935949 : *errormsg = NULL;
370 : :
371 : 6935949 : return state->record;
372 : : }
373 : :
374 : : /*
375 : : * Attempt to read an XLOG record.
376 : : *
377 : : * XLogBeginRead() or XLogFindNextRecord() must be called before the first call
378 : : * to XLogReadRecord().
379 : : *
380 : : * If the page_read callback fails to read the requested data, NULL is
381 : : * returned. The callback is expected to have reported the error; errormsg
382 : : * is set to NULL.
383 : : *
384 : : * If the reading fails for some other reason, NULL is also returned, and
385 : : * *errormsg is set to a string with details of the failure.
386 : : *
387 : : * The returned pointer (or *errormsg) points to an internal buffer that's
388 : : * valid until the next call to XLogReadRecord.
389 : : */
390 : : XLogRecord *
1821 391 : 3993510 : XLogReadRecord(XLogReaderState *state, char **errormsg)
392 : : {
393 : : DecodedXLogRecord *decoded;
394 : :
395 : : /*
396 : : * Release last returned record, if there is one. We need to do this so
397 : : * that we can check for empty decode queue accurately.
398 : : */
1509 399 : 3993510 : XLogReleasePreviousRecord(state);
400 : :
401 : : /*
402 : : * Call XLogReadAhead() in blocking mode to make sure there is something
403 : : * in the queue, though we don't use the result.
404 : : */
405 [ + - ]: 3993510 : if (!XLogReaderHasQueuedRecordOrError(state))
406 : 3993510 : XLogReadAhead(state, false /* nonblocking */ );
407 : :
408 : : /* Consume the head record or error. */
409 : 3993294 : decoded = XLogNextRecord(state, errormsg);
410 [ + + ]: 3993294 : if (decoded)
411 : : {
412 : : /*
413 : : * This function returns a pointer to the record's header, not the
414 : : * actual decoded record. The caller will access the decoded record
415 : : * through the XLogRecGetXXX() macros, which reach the decoded
416 : : * recorded as xlogreader->record.
417 : : */
418 [ - + ]: 3989779 : Assert(state->record == decoded);
419 : 3989779 : return &decoded->header;
420 : : }
421 : :
422 : 3515 : return NULL;
423 : : }
424 : :
425 : : /*
426 : : * Allocate space for a decoded record. The only member of the returned
427 : : * object that is initialized is the 'oversized' flag, indicating that the
428 : : * decoded record wouldn't fit in the decode buffer and must eventually be
429 : : * freed explicitly.
430 : : *
431 : : * The caller is responsible for adjusting decode_buffer_tail with the real
432 : : * size after successfully decoding a record into this space. This way, if
433 : : * decoding fails, then there is nothing to undo unless the 'oversized' flag
434 : : * was set and pfree() must be called.
435 : : *
436 : : * Return NULL if there is no space in the decode buffer and allow_oversized
437 : : * is false, or if memory allocation fails for an oversized buffer.
438 : : */
439 : : static DecodedXLogRecord *
440 : 6942679 : XLogReadRecordAlloc(XLogReaderState *state, size_t xl_tot_len, bool allow_oversized)
441 : : {
442 : 6942679 : size_t required_space = DecodeXLogRecordRequiredSpace(xl_tot_len);
443 : 6942679 : DecodedXLogRecord *decoded = NULL;
444 : :
445 : : /* Allocate a circular decode buffer if we don't have one already. */
446 [ + + ]: 6942679 : if (unlikely(state->decode_buffer == NULL))
447 : : {
448 [ + + ]: 2814 : if (state->decode_buffer_size == 0)
449 : 1734 : state->decode_buffer_size = DEFAULT_DECODE_BUFFER_SIZE;
450 : 2814 : state->decode_buffer = palloc(state->decode_buffer_size);
451 : 2814 : state->decode_buffer_head = state->decode_buffer;
452 : 2814 : state->decode_buffer_tail = state->decode_buffer;
453 : 2814 : state->free_decode_buffer = true;
454 : : }
455 : :
456 : : /* Try to allocate space in the circular decode buffer. */
457 [ + + ]: 6942679 : if (state->decode_buffer_tail >= state->decode_buffer_head)
458 : : {
459 : : /* Empty, or tail is to the right of head. */
879 460 : 6076444 : if (required_space <=
461 : 6076444 : state->decode_buffer_size -
462 [ + + ]: 6076444 : (state->decode_buffer_tail - state->decode_buffer))
463 : : {
464 : : /*-
465 : : * There is space between tail and end.
466 : : *
467 : : * +-----+--------------------+-----+
468 : : * | |////////////////////|here!|
469 : : * +-----+--------------------+-----+
470 : : * ^ ^
471 : : * | |
472 : : * h t
473 : : */
1509 474 : 6057783 : decoded = (DecodedXLogRecord *) state->decode_buffer_tail;
475 : 6057783 : decoded->oversized = false;
476 : 6057783 : return decoded;
477 : : }
879 478 : 18661 : else if (required_space <
479 [ + + ]: 18661 : state->decode_buffer_head - state->decode_buffer)
480 : : {
481 : : /*-
482 : : * There is space between start and head.
483 : : *
484 : : * +-----+--------------------+-----+
485 : : * |here!|////////////////////| |
486 : : * +-----+--------------------+-----+
487 : : * ^ ^
488 : : * | |
489 : : * h t
490 : : */
1509 491 : 16216 : decoded = (DecodedXLogRecord *) state->decode_buffer;
492 : 16216 : decoded->oversized = false;
493 : 16216 : return decoded;
494 : : }
495 : : }
496 : : else
497 : : {
498 : : /* Tail is to the left of head. */
879 499 : 866235 : if (required_space <
500 [ + + ]: 866235 : state->decode_buffer_head - state->decode_buffer_tail)
501 : : {
502 : : /*-
503 : : * There is space between tail and head.
504 : : *
505 : : * +-----+--------------------+-----+
506 : : * |/////|here! |/////|
507 : : * +-----+--------------------+-----+
508 : : * ^ ^
509 : : * | |
510 : : * t h
511 : : */
1509 512 : 866028 : decoded = (DecodedXLogRecord *) state->decode_buffer_tail;
513 : 866028 : decoded->oversized = false;
514 : 866028 : return decoded;
515 : : }
516 : : }
517 : :
518 : : /* Not enough space in the decode buffer. Are we allowed to allocate? */
519 [ + + ]: 2652 : if (allow_oversized)
520 : : {
945 michael@paquier.xyz 521 : 103 : decoded = palloc(required_space);
1509 tmunro@postgresql.or 522 : 103 : decoded->oversized = true;
523 : 103 : return decoded;
524 : : }
525 : :
526 : 2549 : return NULL;
527 : : }
528 : :
529 : : static XLogPageReadResult
530 : 6958278 : XLogDecodeNextRecord(XLogReaderState *state, bool nonblocking)
531 : : {
532 : : XLogRecPtr RecPtr;
533 : : XLogRecord *record;
534 : : XLogRecPtr targetPagePtr;
535 : : bool randAccess;
536 : : uint32 len,
537 : : total_len;
538 : : uint32 targetRecOff;
539 : : uint32 pageHeaderSize;
540 : : bool assembled;
541 : : bool gotheader;
542 : : int readOff;
543 : : DecodedXLogRecord *decoded;
544 : : char *errormsg; /* not used */
545 : :
546 : : /*
547 : : * randAccess indicates whether to verify the previous-record pointer of
548 : : * the record we're reading. We only do this if we're reading
549 : : * sequentially, which is what we initially assume.
550 : : */
1821 551 : 6958278 : randAccess = false;
552 : :
553 : : /* reset error state */
554 : 6958278 : state->errormsg_buf[0] = '\0';
1509 555 : 6958278 : decoded = NULL;
556 : :
1679 alvherre@alvh.no-ip. 557 : 6958278 : state->abortedRecPtr = InvalidXLogRecPtr;
558 : 6958278 : state->missingContrecPtr = InvalidXLogRecPtr;
559 : :
1509 tmunro@postgresql.or 560 : 6958278 : RecPtr = state->NextRecPtr;
561 : :
180 alvherre@kurilemu.de 562 [ + + ]:GNC 6958278 : if (XLogRecPtrIsValid(state->DecodeRecPtr))
563 : : {
564 : : /* read the record after the one we just read */
565 : :
566 : : /*
567 : : * NextRecPtr is pointing to end+1 of the previous WAL record. If
568 : : * we're at a page boundary, no more records can fit on the current
569 : : * page. We must skip over the page header, but we can't do that until
570 : : * we've read in the page, since the header size is variable.
571 : : */
572 : : }
573 : : else
574 : : {
575 : : /*
576 : : * Caller supplied a position to start at.
577 : : *
578 : : * In this case, NextRecPtr should already be pointing either to a
579 : : * valid record starting position or alternatively to the beginning of
580 : : * a page. See the header comments for XLogBeginRead.
581 : : */
1356 rhaas@postgresql.org 582 [ + + - + ]:CBC 6993 : Assert(RecPtr % XLOG_BLCKSZ == 0 || XRecOffIsValid(RecPtr));
1821 tmunro@postgresql.or 583 : 6993 : randAccess = true;
584 : : }
585 : :
1679 alvherre@alvh.no-ip. 586 : 6958278 : restart:
1509 tmunro@postgresql.or 587 : 6958280 : state->nonblocking = nonblocking;
1821 588 : 6958280 : state->currRecPtr = RecPtr;
1679 alvherre@alvh.no-ip. 589 : 6958280 : assembled = false;
590 : :
1821 tmunro@postgresql.or 591 : 6958280 : targetPagePtr = RecPtr - (RecPtr % XLOG_BLCKSZ);
592 : 6958280 : targetRecOff = RecPtr % XLOG_BLCKSZ;
593 : :
594 : : /*
595 : : * Read the page containing the record into state->readBuf. Request enough
596 : : * byte to cover the whole record header, or at least the part of it that
597 : : * fits on the same page.
598 : : */
599 : 6958280 : readOff = ReadPageInternal(state, targetPagePtr,
600 : 6958280 : Min(targetRecOff + SizeOfXLogRecord, XLOG_BLCKSZ));
1509 601 [ + + ]: 6958002 : if (readOff == XLREAD_WOULDBLOCK)
602 : 11608 : return XLREAD_WOULDBLOCK;
603 [ + + ]: 6946394 : else if (readOff < 0)
1821 604 : 3554 : goto err;
605 : :
606 : : /*
607 : : * ReadPageInternal always returns at least the page header, so we can
608 : : * examine it now.
609 : : */
610 [ + + ]: 6942840 : pageHeaderSize = XLogPageHeaderSize((XLogPageHeader) state->readBuf);
611 [ + + ]: 6942840 : if (targetRecOff == 0)
612 : : {
613 : : /*
614 : : * At page start, so skip over page header.
615 : : */
616 : 7205 : RecPtr += pageHeaderSize;
617 : 7205 : targetRecOff = pageHeaderSize;
618 : : }
619 [ - + ]: 6935635 : else if (targetRecOff < pageHeaderSize)
620 : : {
302 alvherre@kurilemu.de 621 :UNC 0 : report_invalid_record(state, "invalid record offset at %X/%08X: expected at least %u, got %u",
1160 peter@eisentraut.org 622 :UBC 0 : LSN_FORMAT_ARGS(RecPtr),
623 : : pageHeaderSize, targetRecOff);
1821 tmunro@postgresql.or 624 : 0 : goto err;
625 : : }
626 : :
1821 tmunro@postgresql.or 627 [ + + - + ]:CBC 6942840 : if ((((XLogPageHeader) state->readBuf)->xlp_info & XLP_FIRST_IS_CONTRECORD) &&
628 : : targetRecOff == pageHeaderSize)
629 : : {
302 alvherre@kurilemu.de 630 :UNC 0 : report_invalid_record(state, "contrecord is requested by %X/%08X",
1821 tmunro@postgresql.or 631 :UBC 0 : LSN_FORMAT_ARGS(RecPtr));
632 : 0 : goto err;
633 : : }
634 : :
635 : : /* ReadPageInternal has verified the page header */
1821 tmunro@postgresql.or 636 [ - + ]:CBC 6942840 : Assert(pageHeaderSize <= readOff);
637 : :
638 : : /*
639 : : * Read the record length.
640 : : *
641 : : * NB: Even though we use an XLogRecord pointer here, the whole record
642 : : * header might not fit on this page. xl_tot_len is the first field of the
643 : : * struct, so it must be on this page (the records are MAXALIGNed), but we
644 : : * cannot access any other fields until we've verified that we got the
645 : : * whole header.
646 : : */
647 : 6942840 : record = (XLogRecord *) (state->readBuf + RecPtr % XLOG_BLCKSZ);
648 : 6942840 : total_len = record->xl_tot_len;
649 : :
650 : : /*
651 : : * If the whole record header is on this page, validate it immediately.
652 : : * Otherwise do just a basic sanity check on xl_tot_len, and validate the
653 : : * rest of the header after reading it from the next page. The xl_tot_len
654 : : * check is necessary here to ensure that we enter the "Need to reassemble
655 : : * record" code path below; otherwise we might fail to apply
656 : : * ValidXLogRecordHeader at all.
657 : : */
658 [ + + ]: 6942840 : if (targetRecOff <= XLOG_BLCKSZ - SizeOfXLogRecord)
659 : : {
1509 660 [ + + ]: 6929426 : if (!ValidXLogRecordHeader(state, RecPtr, state->DecodeRecPtr, record,
661 : : randAccess))
1821 662 : 263 : goto err;
663 : 6929163 : gotheader = true;
664 : : }
665 : : else
666 : : {
667 : : /* There may be no next page if it's too small. */
952 668 [ + + ]: 13414 : if (total_len < SizeOfXLogRecord)
669 : : {
670 : 1 : report_invalid_record(state,
671 : : "invalid record length at %X/%08X: expected at least %u, got %u",
672 : 1 : LSN_FORMAT_ARGS(RecPtr),
673 : : (uint32) SizeOfXLogRecord, total_len);
674 : 1 : goto err;
675 : : }
676 : : /* We'll validate the header once we have the next page. */
1821 677 : 13413 : gotheader = false;
678 : : }
679 : :
680 : : /*
681 : : * Try to find space to decode this record, if we can do so without
682 : : * calling palloc. If we can't, we'll try again below after we've
683 : : * validated that total_len isn't garbage bytes from a recycled WAL page.
684 : : */
1509 685 : 6942576 : decoded = XLogReadRecordAlloc(state,
686 : : total_len,
687 : : false /* allow_oversized */ );
955 688 [ + + + + ]: 6942576 : if (decoded == NULL && nonblocking)
689 : : {
690 : : /*
691 : : * There is no space in the circular decode buffer, and the caller is
692 : : * only reading ahead. The caller should consume existing records to
693 : : * make space.
694 : : */
695 : 2432 : return XLREAD_WOULDBLOCK;
696 : : }
697 : :
1821 698 : 6940144 : len = XLOG_BLCKSZ - RecPtr % XLOG_BLCKSZ;
699 [ + + ]: 6940144 : if (total_len > len)
700 : : {
701 : : /* Need to reassemble record */
702 : : char *contdata;
703 : : XLogPageHeader pageHeader;
704 : : char *buffer;
705 : : uint32 gotlen;
706 : :
1679 alvherre@alvh.no-ip. 707 : 1526569 : assembled = true;
708 : :
709 : : /*
710 : : * We always have space for a couple of pages, enough to validate a
711 : : * boundary-spanning record header.
712 : : */
955 tmunro@postgresql.or 713 [ - + ]: 1526569 : Assert(state->readRecordBufSize >= XLOG_BLCKSZ * 2);
714 [ - + ]: 1526569 : Assert(state->readRecordBufSize >= len);
715 : :
716 : : /* Copy the first fragment of the record from the first page. */
1821 717 : 1526569 : memcpy(state->readRecordBuf,
718 : 1526569 : state->readBuf + RecPtr % XLOG_BLCKSZ, len);
719 : 1526569 : buffer = state->readRecordBuf + len;
720 : 1526569 : gotlen = len;
721 : :
722 : : do
723 : : {
724 : : /* Calculate pointer to beginning of next page */
725 : 1562425 : targetPagePtr += XLOG_BLCKSZ;
726 : :
727 : : /*
728 : : * Read the page header before processing the record data, so we
729 : : * can handle the case where the previous record ended as being a
730 : : * partial one.
731 : : */
290 akorotkov@postgresql 732 : 1562425 : readOff = ReadPageInternal(state, targetPagePtr, SizeOfXLogShortPHD);
1509 tmunro@postgresql.or 733 [ + + ]: 1562423 : if (readOff == XLREAD_WOULDBLOCK)
734 : 4098 : return XLREAD_WOULDBLOCK;
735 [ + + ]: 1558325 : else if (readOff < 0)
1821 736 : 17 : goto err;
737 : :
738 [ - + ]: 1558308 : Assert(SizeOfXLogShortPHD <= readOff);
739 : :
740 : 1558308 : pageHeader = (XLogPageHeader) state->readBuf;
741 : :
742 : : /*
743 : : * If we were expecting a continuation record and got an
744 : : * "overwrite contrecord" flag, that means the continuation record
745 : : * was overwritten with a different record. Restart the read by
746 : : * assuming the address to read is the location where we found
747 : : * this flag; but keep track of the LSN of the record we were
748 : : * reading, for later verification.
749 : : */
1679 alvherre@alvh.no-ip. 750 [ + + ]: 1558308 : if (pageHeader->xlp_info & XLP_FIRST_IS_OVERWRITE_CONTRECORD)
751 : : {
1621 752 : 2 : state->overwrittenRecPtr = RecPtr;
1679 753 : 2 : RecPtr = targetPagePtr;
754 : 2 : goto restart;
755 : : }
756 : :
757 : : /* Check that the continuation on next page looks valid */
1821 tmunro@postgresql.or 758 [ + + ]: 1558306 : if (!(pageHeader->xlp_info & XLP_FIRST_IS_CONTRECORD))
759 : : {
760 : 1 : report_invalid_record(state,
761 : : "there is no contrecord flag at %X/%08X",
762 : 1 : LSN_FORMAT_ARGS(RecPtr));
763 : 1 : goto err;
764 : : }
765 : :
766 : : /*
767 : : * Cross-check that xlp_rem_len agrees with how much of the record
768 : : * we expect there to be left.
769 : : */
770 [ + - ]: 1558305 : if (pageHeader->xlp_rem_len == 0 ||
771 [ + + ]: 1558305 : total_len != (pageHeader->xlp_rem_len + gotlen))
772 : : {
773 : 2 : report_invalid_record(state,
774 : : "invalid contrecord length %u (expected %lld) at %X/%08X",
775 : : pageHeader->xlp_rem_len,
776 : 2 : ((long long) total_len) - gotlen,
777 : 2 : LSN_FORMAT_ARGS(RecPtr));
778 : 2 : goto err;
779 : : }
780 : :
781 : : /* Wait for the next page to become available */
290 akorotkov@postgresql 782 : 1558303 : readOff = ReadPageInternal(state, targetPagePtr,
783 : 1558303 : Min(total_len - gotlen + SizeOfXLogShortPHD,
784 : : XLOG_BLCKSZ));
785 [ - + ]: 1558303 : if (readOff == XLREAD_WOULDBLOCK)
290 akorotkov@postgresql 786 :UBC 0 : return XLREAD_WOULDBLOCK;
290 akorotkov@postgresql 787 [ - + ]:CBC 1558303 : else if (readOff < 0)
290 akorotkov@postgresql 788 :UBC 0 : goto err;
789 : :
790 : : /* Append the continuation from this page to the buffer */
1821 tmunro@postgresql.or 791 [ + + ]:CBC 1558303 : pageHeaderSize = XLogPageHeaderSize(pageHeader);
792 : :
793 [ - + ]: 1558303 : if (readOff < pageHeaderSize)
1821 tmunro@postgresql.or 794 :UBC 0 : readOff = ReadPageInternal(state, targetPagePtr,
795 : : pageHeaderSize);
796 : :
1821 tmunro@postgresql.or 797 [ - + ]:CBC 1558303 : Assert(pageHeaderSize <= readOff);
798 : :
799 : 1558303 : contdata = (char *) state->readBuf + pageHeaderSize;
800 : 1558303 : len = XLOG_BLCKSZ - pageHeaderSize;
801 [ + + ]: 1558303 : if (pageHeader->xlp_rem_len < len)
802 : 1522433 : len = pageHeader->xlp_rem_len;
803 : :
804 [ - + ]: 1558303 : if (readOff < pageHeaderSize + len)
1821 tmunro@postgresql.or 805 :UBC 0 : readOff = ReadPageInternal(state, targetPagePtr,
806 : 0 : pageHeaderSize + len);
807 : :
447 peter@eisentraut.org 808 :CBC 1558303 : memcpy(buffer, contdata, len);
1821 tmunro@postgresql.or 809 : 1558303 : buffer += len;
810 : 1558303 : gotlen += len;
811 : :
812 : : /* If we just reassembled the record header, validate it. */
813 [ + + ]: 1558303 : if (!gotheader)
814 : : {
815 : 13378 : record = (XLogRecord *) state->readRecordBuf;
1509 816 [ - + ]: 13378 : if (!ValidXLogRecordHeader(state, RecPtr, state->DecodeRecPtr,
817 : : record, randAccess))
4857 alvherre@alvh.no-ip. 818 :UBC 0 : goto err;
1821 tmunro@postgresql.or 819 :CBC 13378 : gotheader = true;
820 : : }
821 : :
822 : : /*
823 : : * We might need a bigger buffer. We have validated the record
824 : : * header, in the case that it split over a page boundary. We've
825 : : * also cross-checked total_len against xlp_rem_len on the second
826 : : * page, and verified xlp_pageaddr on both.
827 : : */
955 828 [ + + ]: 1558303 : if (total_len > state->readRecordBufSize)
829 : : {
830 : : char save_copy[XLOG_BLCKSZ * 2];
831 : :
832 : : /*
833 : : * Save and restore the data we already had. It can't be more
834 : : * than two pages.
835 : : */
836 [ - + ]: 82 : Assert(gotlen <= lengthof(save_copy));
837 [ - + ]: 82 : Assert(gotlen <= state->readRecordBufSize);
838 : 82 : memcpy(save_copy, state->readRecordBuf, gotlen);
945 michael@paquier.xyz 839 : 82 : allocate_recordbuf(state, total_len);
955 tmunro@postgresql.or 840 : 82 : memcpy(state->readRecordBuf, save_copy, gotlen);
841 : 82 : buffer = state->readRecordBuf + gotlen;
842 : : }
843 [ + + ]: 1558303 : } while (gotlen < total_len);
1821 844 [ - + ]: 1522447 : Assert(gotheader);
845 : :
846 : 1522447 : record = (XLogRecord *) state->readRecordBuf;
847 [ - + ]: 1522447 : if (!ValidXLogRecord(state, record, RecPtr))
1821 tmunro@postgresql.or 848 :UBC 0 : goto err;
849 : :
1821 tmunro@postgresql.or 850 [ + + ]:CBC 1522447 : pageHeaderSize = XLogPageHeaderSize((XLogPageHeader) state->readBuf);
1509 851 : 1522447 : state->DecodeRecPtr = RecPtr;
852 : 1522447 : state->NextRecPtr = targetPagePtr + pageHeaderSize
1821 853 : 1522447 : + MAXALIGN(pageHeader->xlp_rem_len);
854 : : }
855 : : else
856 : : {
857 : : /* Wait for the record data to become available */
858 : 5413575 : readOff = ReadPageInternal(state, targetPagePtr,
859 : 5413575 : Min(targetRecOff + total_len, XLOG_BLCKSZ));
1509 860 [ - + ]: 5413575 : if (readOff == XLREAD_WOULDBLOCK)
1509 tmunro@postgresql.or 861 :UBC 0 : return XLREAD_WOULDBLOCK;
1509 tmunro@postgresql.or 862 [ - + ]:CBC 5413575 : else if (readOff < 0)
1821 tmunro@postgresql.or 863 :UBC 0 : goto err;
864 : :
865 : : /* Record does not cross a page boundary */
1821 tmunro@postgresql.or 866 [ + + ]:CBC 5413575 : if (!ValidXLogRecord(state, record, RecPtr))
867 : 1 : goto err;
868 : :
1509 869 : 5413574 : state->NextRecPtr = RecPtr + MAXALIGN(total_len);
870 : :
871 : 5413574 : state->DecodeRecPtr = RecPtr;
872 : : }
873 : :
874 : : /*
875 : : * Special processing if it's an XLOG SWITCH record
876 : : */
1853 877 [ + + ]: 6936021 : if (record->xl_rmid == RM_XLOG_ID &&
878 [ + + ]: 171656 : (record->xl_info & ~XLR_INFO_MASK) == XLOG_SWITCH)
879 : : {
880 : : /* Pretend it extends to end of segment */
1509 881 : 571 : state->NextRecPtr += state->segcxt.ws_segsize - 1;
882 : 571 : state->NextRecPtr -= XLogSegmentOffset(state->NextRecPtr, state->segcxt.ws_segsize);
883 : : }
884 : :
885 : : /*
886 : : * If we got here without a DecodedXLogRecord, it means we needed to
887 : : * validate total_len before trusting it, but by now we've done that.
888 : : */
955 889 [ + + ]: 6936021 : if (decoded == NULL)
890 : : {
891 [ - + ]: 103 : Assert(!nonblocking);
892 : 103 : decoded = XLogReadRecordAlloc(state,
893 : : total_len,
894 : : true /* allow_oversized */ );
895 : : /* allocation should always happen under allow_oversized */
945 michael@paquier.xyz 896 [ - + ]: 103 : Assert(decoded != NULL);
897 : : }
898 : :
1509 tmunro@postgresql.or 899 [ + - ]: 6936021 : if (DecodeXLogRecord(state, decoded, record, RecPtr, &errormsg))
900 : : {
901 : : /* Record the location of the next record. */
902 : 6936021 : decoded->next_lsn = state->NextRecPtr;
903 : :
904 : : /*
905 : : * If it's in the decode buffer, mark the decode buffer space as
906 : : * occupied.
907 : : */
908 [ + + ]: 6936021 : if (!decoded->oversized)
909 : : {
910 : : /* The new decode buffer head must be MAXALIGNed. */
911 [ - + ]: 6935918 : Assert(decoded->size == MAXALIGN(decoded->size));
912 [ + + ]: 6935918 : if ((char *) decoded == state->decode_buffer)
913 : 4044669 : state->decode_buffer_tail = state->decode_buffer + decoded->size;
914 : : else
915 : 2891249 : state->decode_buffer_tail += decoded->size;
916 : : }
917 : :
918 : : /* Insert it into the queue of decoded records. */
919 [ - + ]: 6936021 : Assert(state->decode_queue_tail != decoded);
920 [ + + ]: 6936021 : if (state->decode_queue_tail)
921 : 2907439 : state->decode_queue_tail->next = decoded;
922 : 6936021 : state->decode_queue_tail = decoded;
923 [ + + ]: 6936021 : if (!state->decode_queue_head)
924 : 4028582 : state->decode_queue_head = decoded;
925 : 6936021 : return XLREAD_SUCCESS;
926 : : }
927 : :
4857 alvherre@alvh.no-ip. 928 :UBC 0 : err:
1679 alvherre@alvh.no-ip. 929 [ + + ]:CBC 3839 : if (assembled)
930 : : {
931 : : /*
932 : : * We get here when a record that spans multiple pages needs to be
933 : : * assembled, but something went wrong -- perhaps a contrecord piece
934 : : * was lost. If caller is WAL replay, it will know where the aborted
935 : : * record was and where to direct followup WAL to be written, marking
936 : : * the next piece with XLP_FIRST_IS_OVERWRITE_CONTRECORD, which will
937 : : * in turn signal downstream WAL consumers that the broken WAL record
938 : : * is to be ignored.
939 : : */
940 : 20 : state->abortedRecPtr = RecPtr;
941 : 20 : state->missingContrecPtr = targetPagePtr;
942 : :
943 : : /*
944 : : * If we got here without reporting an error, make sure an error is
945 : : * queued so that XLogPrefetcherReadRecord() doesn't bring us back a
946 : : * second time and clobber the above state.
947 : : */
1037 tmunro@postgresql.or 948 : 20 : state->errormsg_deferred = true;
949 : : }
950 : :
1509 951 [ + + - + ]: 3839 : if (decoded && decoded->oversized)
1509 tmunro@postgresql.or 952 :UBC 0 : pfree(decoded);
953 : :
954 : : /*
955 : : * Invalidate the read state. We might read from a different source after
956 : : * failure.
957 : : */
3688 alvherre@alvh.no-ip. 958 :CBC 3839 : XLogReaderInvalReadState(state);
959 : :
960 : : /*
961 : : * If an error was written to errormsg_buf, it'll be returned to the
962 : : * caller of XLogReadRecord() after all successfully decoded records from
963 : : * the read queue.
964 : : */
965 : :
1509 tmunro@postgresql.or 966 : 3839 : return XLREAD_FAIL;
967 : : }
968 : :
969 : : /*
970 : : * Try to decode the next available record, and return it. The record will
971 : : * also be returned to XLogNextRecord(), which must be called to 'consume'
972 : : * each record.
973 : : *
974 : : * If nonblocking is true, may return NULL due to lack of data or WAL decoding
975 : : * space.
976 : : */
977 : : DecodedXLogRecord *
978 : 6958278 : XLogReadAhead(XLogReaderState *state, bool nonblocking)
979 : : {
980 : : XLogPageReadResult result;
981 : :
982 [ - + ]: 6958278 : if (state->errormsg_deferred)
1509 tmunro@postgresql.or 983 :UBC 0 : return NULL;
984 : :
1509 tmunro@postgresql.or 985 :CBC 6958278 : result = XLogDecodeNextRecord(state, nonblocking);
986 [ + + ]: 6957998 : if (result == XLREAD_SUCCESS)
987 : : {
988 [ - + ]: 6936021 : Assert(state->decode_queue_tail != NULL);
989 : 6936021 : return state->decode_queue_tail;
990 : : }
991 : :
1821 992 : 21977 : return NULL;
993 : : }
994 : :
995 : : /*
996 : : * Read a single xlog page including at least [pageptr, reqLen] of valid data
997 : : * via the page_read() callback.
998 : : *
999 : : * Returns XLREAD_FAIL if the required page cannot be read for some
1000 : : * reason; errormsg_buf is set in that case (unless the error occurs in the
1001 : : * page_read callback).
1002 : : *
1003 : : * Returns XLREAD_WOULDBLOCK if the requested data can't be read without
1004 : : * waiting. This can be returned only if the installed page_read callback
1005 : : * respects the state->nonblocking flag, and cannot read the requested data
1006 : : * immediately.
1007 : : *
1008 : : * We fetch the page from a reader-local cache if we know we have the required
1009 : : * data and if there hasn't been any error since caching the data.
1010 : : */
1011 : : static int
1012 : 15492876 : ReadPageInternal(XLogReaderState *state, XLogRecPtr pageptr, int reqLen)
1013 : : {
1014 : : int readLen;
1015 : : uint32 targetPageOff;
1016 : : XLogSegNo targetSegNo;
1017 : : XLogPageHeader hdr;
1018 : :
1019 [ - + ]: 15492876 : Assert((pageptr % XLOG_BLCKSZ) == 0);
1020 : :
1021 : 15492876 : XLByteToSeg(pageptr, targetSegNo, state->segcxt.ws_segsize);
1022 : 15492876 : targetPageOff = XLogSegmentOffset(pageptr, state->segcxt.ws_segsize);
1023 : :
1024 : : /* check whether we have all the requested data already */
1025 [ + + ]: 15492876 : if (targetSegNo == state->seg.ws_segno &&
1026 [ + + + + ]: 15484879 : targetPageOff == state->segoff && reqLen <= state->readLen)
1027 : 13886269 : return state->readLen;
1028 : :
1029 : : /*
1030 : : * Invalidate contents of internal buffer before read attempt. Just set
1031 : : * the length to 0, rather than a full XLogReaderInvalReadState(), so we
1032 : : * don't forget the segment we last successfully read.
1033 : : */
1340 1034 : 1606607 : state->readLen = 0;
1035 : :
1036 : : /*
1037 : : * Data is not in our buffer.
1038 : : *
1039 : : * Every time we actually read the segment, even if we looked at parts of
1040 : : * it before, we need to do verification as the page_read callback might
1041 : : * now be rereading data from a different source.
1042 : : *
1043 : : * Whenever switching to a new WAL segment, we read the first page of the
1044 : : * file and validate its header, even if that's not where the target
1045 : : * record is. This is so that we can check the additional identification
1046 : : * info that is present in the first page's "long" header.
1047 : : */
1821 1048 [ + + + + ]: 1606607 : if (targetSegNo != state->seg.ws_segno && targetPageOff != 0)
1049 : : {
1050 : 5852 : XLogRecPtr targetSegmentPtr = pageptr - targetPageOff;
1051 : :
1052 : 5852 : readLen = state->routine.page_read(state, targetSegmentPtr, XLOG_BLCKSZ,
1053 : : state->currRecPtr,
1054 : : state->readBuf);
1509 1055 [ - + ]: 5846 : if (readLen == XLREAD_WOULDBLOCK)
1509 tmunro@postgresql.or 1056 :UBC 0 : return XLREAD_WOULDBLOCK;
1509 tmunro@postgresql.or 1057 [ + + ]:CBC 5846 : else if (readLen < 0)
1821 tmunro@postgresql.or 1058 :GBC 4 : goto err;
1059 : :
1060 : : /* we can be sure to have enough WAL available, we scrolled back */
1821 tmunro@postgresql.or 1061 [ - + ]:CBC 5842 : Assert(readLen == XLOG_BLCKSZ);
1062 : :
1063 [ - + ]: 5842 : if (!XLogReaderValidatePageHeader(state, targetSegmentPtr,
1064 : : state->readBuf))
1821 tmunro@postgresql.or 1065 :UBC 0 : goto err;
1066 : : }
1067 : :
1068 : : /*
1069 : : * First, read the requested data length, but at least a short page header
1070 : : * so that we can validate it.
1071 : : */
1821 tmunro@postgresql.or 1072 :CBC 1606597 : readLen = state->routine.page_read(state, pageptr, Max(reqLen, SizeOfXLogShortPHD),
1073 : : state->currRecPtr,
1074 : : state->readBuf);
1509 1075 [ + + ]: 1606323 : if (readLen == XLREAD_WOULDBLOCK)
1076 : 15706 : return XLREAD_WOULDBLOCK;
1077 [ + + ]: 1590617 : else if (readLen < 0)
1821 1078 : 3555 : goto err;
1079 : :
1080 [ - + ]: 1587062 : Assert(readLen <= XLOG_BLCKSZ);
1081 : :
1082 : : /* Do we have enough data to check the header length? */
1083 [ - + ]: 1587062 : if (readLen <= SizeOfXLogShortPHD)
1821 tmunro@postgresql.or 1084 :UBC 0 : goto err;
1085 : :
1821 tmunro@postgresql.or 1086 [ - + ]:CBC 1587062 : Assert(readLen >= reqLen);
1087 : :
1088 : 1587062 : hdr = (XLogPageHeader) state->readBuf;
1089 : :
1090 : : /* still not enough */
1091 [ + + - + ]: 1587062 : if (readLen < XLogPageHeaderSize(hdr))
1092 : : {
1821 tmunro@postgresql.or 1093 [ # # ]:UBC 0 : readLen = state->routine.page_read(state, pageptr, XLogPageHeaderSize(hdr),
1094 : : state->currRecPtr,
1095 : : state->readBuf);
1509 1096 [ # # ]: 0 : if (readLen == XLREAD_WOULDBLOCK)
1097 : 0 : return XLREAD_WOULDBLOCK;
1098 [ # # ]: 0 : else if (readLen < 0)
1821 1099 : 0 : goto err;
1100 : : }
1101 : :
1102 : : /*
1103 : : * Now that we know we have the full header, validate it.
1104 : : */
1821 tmunro@postgresql.or 1105 [ + + ]:CBC 1587062 : if (!XLogReaderValidatePageHeader(state, pageptr, (char *) hdr))
1106 : 13 : goto err;
1107 : :
1108 : : /* update read state information */
1109 : 1587049 : state->seg.ws_segno = targetSegNo;
1110 : 1587049 : state->segoff = targetPageOff;
1111 : 1587049 : state->readLen = readLen;
1112 : :
1113 : 1587049 : return readLen;
1114 : :
1115 : 3572 : err:
1340 1116 : 3572 : XLogReaderInvalReadState(state);
1117 : :
1509 1118 : 3572 : return XLREAD_FAIL;
1119 : : }
1120 : :
1121 : : /*
1122 : : * Invalidate the xlogreader's read state to force a re-read.
1123 : : */
1124 : : static void
3688 alvherre@alvh.no-ip. 1125 : 7412 : XLogReaderInvalReadState(XLogReaderState *state)
1126 : : {
1821 tmunro@postgresql.or 1127 : 7412 : state->seg.ws_segno = 0;
1128 : 7412 : state->segoff = 0;
1129 : 7412 : state->readLen = 0;
4857 alvherre@alvh.no-ip. 1130 : 7412 : }
1131 : :
1132 : : /*
1133 : : * Validate an XLOG record header.
1134 : : *
1135 : : * This is just a convenience subroutine to avoid duplicated code in
1136 : : * XLogReadRecord. It's not intended for use from anywhere else.
1137 : : */
1138 : : static bool
1139 : 6942804 : ValidXLogRecordHeader(XLogReaderState *state, XLogRecPtr RecPtr,
1140 : : XLogRecPtr PrevRecPtr, XLogRecord *record,
1141 : : bool randAccess)
1142 : : {
4184 heikki.linnakangas@i 1143 [ + + ]: 6942804 : if (record->xl_tot_len < SizeOfXLogRecord)
1144 : : {
4857 alvherre@alvh.no-ip. 1145 : 252 : report_invalid_record(state,
1146 : : "invalid record length at %X/%08X: expected at least %u, got %u",
1897 peter@eisentraut.org 1147 : 252 : LSN_FORMAT_ARGS(RecPtr),
1148 : : (uint32) SizeOfXLogRecord, record->xl_tot_len);
4857 alvherre@alvh.no-ip. 1149 : 252 : return false;
1150 : : }
1489 jdavis@postgresql.or 1151 [ + + - + ]: 6942552 : if (!RmgrIdIsValid(record->xl_rmid))
1152 : : {
4857 alvherre@alvh.no-ip. 1153 :LBC (1) : report_invalid_record(state,
1154 : : "invalid resource manager ID %u at %X/%08X",
1897 peter@eisentraut.org 1155 : (1) : record->xl_rmid, LSN_FORMAT_ARGS(RecPtr));
4857 alvherre@alvh.no-ip. 1156 : (1) : return false;
1157 : : }
1821 tmunro@postgresql.or 1158 [ + + ]:CBC 6942552 : if (randAccess)
1159 : : {
1160 : : /*
1161 : : * We can't exactly verify the prev-link, but surely it should be less
1162 : : * than the record's own address.
1163 : : */
4857 alvherre@alvh.no-ip. 1164 [ - + ]: 6989 : if (!(record->xl_prev < RecPtr))
1165 : : {
4857 alvherre@alvh.no-ip. 1166 :UBC 0 : report_invalid_record(state,
1167 : : "record with incorrect prev-link %X/%08X at %X/%08X",
1897 peter@eisentraut.org 1168 : 0 : LSN_FORMAT_ARGS(record->xl_prev),
1169 : 0 : LSN_FORMAT_ARGS(RecPtr));
4857 alvherre@alvh.no-ip. 1170 : 0 : return false;
1171 : : }
1172 : : }
1173 : : else
1174 : : {
1175 : : /*
1176 : : * Record's prev-link should exactly match our previous location. This
1177 : : * check guards against torn WAL pages where a stale but valid-looking
1178 : : * WAL record starts on a sector boundary.
1179 : : */
4857 alvherre@alvh.no-ip. 1180 [ + + ]:CBC 6935563 : if (record->xl_prev != PrevRecPtr)
1181 : : {
1182 : 11 : report_invalid_record(state,
1183 : : "record with incorrect prev-link %X/%08X at %X/%08X",
1897 peter@eisentraut.org 1184 : 11 : LSN_FORMAT_ARGS(record->xl_prev),
1185 : 11 : LSN_FORMAT_ARGS(RecPtr));
4857 alvherre@alvh.no-ip. 1186 : 11 : return false;
1187 : : }
1188 : : }
1189 : :
1190 : 6942541 : return true;
1191 : : }
1192 : :
1193 : :
1194 : : /*
1195 : : * CRC-check an XLOG record. We do not believe the contents of an XLOG
1196 : : * record (other than to the minimal extent of computing the amount of
1197 : : * data to read in) until we've checked the CRCs.
1198 : : *
1199 : : * We assume all of the record (that is, xl_tot_len bytes) has been read
1200 : : * into memory at *record. Also, ValidXLogRecordHeader() has accepted the
1201 : : * record's header, which means in particular that xl_tot_len is at least
1202 : : * SizeOfXLogRecord.
1203 : : */
1204 : : static bool
1205 : 6936022 : ValidXLogRecord(XLogReaderState *state, XLogRecord *record, XLogRecPtr recptr)
1206 : : {
1207 : : pg_crc32c crc;
1208 : :
952 tmunro@postgresql.or 1209 [ - + ]: 6936022 : Assert(record->xl_tot_len >= SizeOfXLogRecord);
1210 : :
1211 : : /* Calculate the CRC */
4200 heikki.linnakangas@i 1212 : 6936022 : INIT_CRC32C(crc);
4184 1213 : 6936022 : COMP_CRC32C(crc, ((char *) record) + SizeOfXLogRecord, record->xl_tot_len - SizeOfXLogRecord);
1214 : : /* include the record header last */
4200 1215 : 6936022 : COMP_CRC32C(crc, (char *) record, offsetof(XLogRecord, xl_crc));
1216 : 6936022 : FIN_CRC32C(crc);
1217 : :
1218 [ + + ]: 6936022 : if (!EQ_CRC32C(record->xl_crc, crc))
1219 : : {
4857 alvherre@alvh.no-ip. 1220 : 1 : report_invalid_record(state,
1221 : : "incorrect resource manager data checksum in record at %X/%08X",
1897 peter@eisentraut.org 1222 : 1 : LSN_FORMAT_ARGS(recptr));
4857 alvherre@alvh.no-ip. 1223 : 1 : return false;
1224 : : }
1225 : :
1226 : 6936021 : return true;
1227 : : }
1228 : :
1229 : : /*
1230 : : * Validate a page header.
1231 : : *
1232 : : * Check if 'phdr' is valid as the header of the XLog page at position
1233 : : * 'recptr'.
1234 : : */
1235 : : bool
2922 heikki.linnakangas@i 1236 : 1594360 : XLogReaderValidatePageHeader(XLogReaderState *state, XLogRecPtr recptr,
1237 : : char *phdr)
1238 : : {
1239 : : XLogSegNo segno;
1240 : : int32 offset;
1241 : 1594360 : XLogPageHeader hdr = (XLogPageHeader) phdr;
1242 : :
4857 alvherre@alvh.no-ip. 1243 [ - + ]: 1594360 : Assert((recptr % XLOG_BLCKSZ) == 0);
1244 : :
2415 1245 : 1594360 : XLByteToSeg(recptr, segno, state->segcxt.ws_segsize);
1246 : 1594360 : offset = XLogSegmentOffset(recptr, state->segcxt.ws_segsize);
1247 : :
4857 1248 [ + + ]: 1594360 : if (hdr->xlp_magic != XLOG_PAGE_MAGIC)
1249 : : {
1250 : : char fname[MAXFNAMELEN];
1251 : :
2415 1252 : 10 : XLogFileName(fname, state->seg.ws_tli, segno, state->segcxt.ws_segsize);
1253 : :
4857 1254 : 10 : report_invalid_record(state,
1255 : : "invalid magic number %04X in WAL segment %s, LSN %X/%08X, offset %u",
1256 : 10 : hdr->xlp_magic,
1257 : : fname,
1247 michael@paquier.xyz 1258 : 10 : LSN_FORMAT_ARGS(recptr),
1259 : : offset);
4857 alvherre@alvh.no-ip. 1260 : 10 : return false;
1261 : : }
1262 : :
1263 [ + + ]: 1594350 : if ((hdr->xlp_info & ~XLP_ALL_FLAGS) != 0)
1264 : : {
1265 : : char fname[MAXFNAMELEN];
1266 : :
2415 1267 : 1 : XLogFileName(fname, state->seg.ws_tli, segno, state->segcxt.ws_segsize);
1268 : :
4857 1269 : 1 : report_invalid_record(state,
1270 : : "invalid info bits %04X in WAL segment %s, LSN %X/%08X, offset %u",
1271 : 1 : hdr->xlp_info,
1272 : : fname,
1247 michael@paquier.xyz 1273 : 1 : LSN_FORMAT_ARGS(recptr),
1274 : : offset);
4857 alvherre@alvh.no-ip. 1275 : 1 : return false;
1276 : : }
1277 : :
1278 [ + + ]: 1594349 : if (hdr->xlp_info & XLP_LONG_HEADER)
1279 : : {
1280 : 9135 : XLogLongPageHeader longhdr = (XLogLongPageHeader) hdr;
1281 : :
1282 [ + + ]: 9135 : if (state->system_identifier &&
1283 [ - + ]: 4039 : longhdr->xlp_sysid != state->system_identifier)
1284 : : {
4857 alvherre@alvh.no-ip. 1285 :UBC 0 : report_invalid_record(state,
1286 : : "WAL file is from different database system: WAL file database system identifier is %" PRIu64 ", pg_control database system identifier is %" PRIu64,
1287 : : longhdr->xlp_sysid,
1288 : : state->system_identifier);
1289 : 0 : return false;
1290 : : }
2415 alvherre@alvh.no-ip. 1291 [ - + ]:CBC 9135 : else if (longhdr->xlp_seg_size != state->segcxt.ws_segsize)
1292 : : {
4857 alvherre@alvh.no-ip. 1293 :UBC 0 : report_invalid_record(state,
1294 : : "WAL file is from different database system: incorrect segment size in page header");
1295 : 0 : return false;
1296 : : }
4857 alvherre@alvh.no-ip. 1297 [ - + ]:CBC 9135 : else if (longhdr->xlp_xlog_blcksz != XLOG_BLCKSZ)
1298 : : {
4857 alvherre@alvh.no-ip. 1299 :UBC 0 : report_invalid_record(state,
1300 : : "WAL file is from different database system: incorrect XLOG_BLCKSZ in page header");
1301 : 0 : return false;
1302 : : }
1303 : : }
4857 alvherre@alvh.no-ip. 1304 [ - + ]:CBC 1585214 : else if (offset == 0)
1305 : : {
1306 : : char fname[MAXFNAMELEN];
1307 : :
2415 alvherre@alvh.no-ip. 1308 :UBC 0 : XLogFileName(fname, state->seg.ws_tli, segno, state->segcxt.ws_segsize);
1309 : :
1310 : : /* hmm, first page of file doesn't have a long header? */
4857 1311 : 0 : report_invalid_record(state,
1312 : : "invalid info bits %04X in WAL segment %s, LSN %X/%08X, offset %u",
1313 : 0 : hdr->xlp_info,
1314 : : fname,
1247 michael@paquier.xyz 1315 : 0 : LSN_FORMAT_ARGS(recptr),
1316 : : offset);
4857 alvherre@alvh.no-ip. 1317 : 0 : return false;
1318 : : }
1319 : :
1320 : : /*
1321 : : * Check that the address on the page agrees with what we expected. This
1322 : : * check typically fails when an old WAL segment is recycled, and hasn't
1323 : : * yet been overwritten with new data yet.
1324 : : */
1301 michael@paquier.xyz 1325 [ + + ]:CBC 1594349 : if (hdr->xlp_pageaddr != recptr)
1326 : : {
1327 : : char fname[MAXFNAMELEN];
1328 : :
2415 alvherre@alvh.no-ip. 1329 : 7 : XLogFileName(fname, state->seg.ws_tli, segno, state->segcxt.ws_segsize);
1330 : :
4857 1331 : 7 : report_invalid_record(state,
1332 : : "unexpected pageaddr %X/%08X in WAL segment %s, LSN %X/%08X, offset %u",
1897 peter@eisentraut.org 1333 : 7 : LSN_FORMAT_ARGS(hdr->xlp_pageaddr),
1334 : : fname,
1247 michael@paquier.xyz 1335 : 7 : LSN_FORMAT_ARGS(recptr),
1336 : : offset);
4857 alvherre@alvh.no-ip. 1337 : 7 : return false;
1338 : : }
1339 : :
1340 : : /*
1341 : : * Since child timelines are always assigned a TLI greater than their
1342 : : * immediate parent's TLI, we should never see TLI go backwards across
1343 : : * successive pages of a consistent WAL sequence.
1344 : : *
1345 : : * Sometimes we re-read a segment that's already been (partially) read. So
1346 : : * we only verify TLIs for pages that are later than the last remembered
1347 : : * LSN.
1348 : : */
1349 [ + + ]: 1594342 : if (recptr > state->latestPagePtr)
1350 : : {
1351 [ - + ]: 1570963 : if (hdr->xlp_tli < state->latestPageTLI)
1352 : : {
1353 : : char fname[MAXFNAMELEN];
1354 : :
2415 alvherre@alvh.no-ip. 1355 :UBC 0 : XLogFileName(fname, state->seg.ws_tli, segno, state->segcxt.ws_segsize);
1356 : :
4857 1357 : 0 : report_invalid_record(state,
1358 : : "out-of-sequence timeline ID %u (after %u) in WAL segment %s, LSN %X/%08X, offset %u",
1359 : : hdr->xlp_tli,
1360 : : state->latestPageTLI,
1361 : : fname,
1247 michael@paquier.xyz 1362 : 0 : LSN_FORMAT_ARGS(recptr),
1363 : : offset);
4857 alvherre@alvh.no-ip. 1364 : 0 : return false;
1365 : : }
1366 : : }
4857 alvherre@alvh.no-ip. 1367 :CBC 1594342 : state->latestPagePtr = recptr;
1368 : 1594342 : state->latestPageTLI = hdr->xlp_tli;
1369 : :
1370 : 1594342 : return true;
1371 : : }
1372 : :
1373 : : /*
1374 : : * Forget about an error produced by XLogReaderValidatePageHeader().
1375 : : */
1376 : : void
1340 tmunro@postgresql.or 1377 : 5 : XLogReaderResetError(XLogReaderState *state)
1378 : : {
1379 : 5 : state->errormsg_buf[0] = '\0';
1380 : 5 : state->errormsg_deferred = false;
1381 : 5 : }
1382 : :
1383 : : /*
1384 : : * Find the first record with an lsn >= RecPtr.
1385 : : *
1386 : : * This is different from XLogBeginRead() in that RecPtr doesn't need to point
1387 : : * to a valid record boundary. Useful for checking whether RecPtr is a valid
1388 : : * xlog address for reading, and to find the first valid address after some
1389 : : * address when dumping records for debugging purposes.
1390 : : *
1391 : : * This positions the reader, like XLogBeginRead(), so that the next call to
1392 : : * XLogReadRecord() will read the next valid record.
1393 : : *
1394 : : * On failure, InvalidXLogRecPtr is returned, and *errormsg is set to a string
1395 : : * with details of the failure.
1396 : : *
1397 : : * When set, *errormsg points to an internal buffer that's valid until the next
1398 : : * call to XLogReadRecord.
1399 : : */
1400 : : XLogRecPtr
42 fujii@postgresql.org 1401 :GNC 147 : XLogFindNextRecord(XLogReaderState *state, XLogRecPtr RecPtr, char **errormsg)
1402 : : {
1403 : : XLogRecPtr tmpRecPtr;
1821 tmunro@postgresql.or 1404 :CBC 147 : XLogRecPtr found = InvalidXLogRecPtr;
1405 : : XLogPageHeader header;
1406 : :
42 fujii@postgresql.org 1407 :GNC 147 : *errormsg = NULL;
1408 : :
180 alvherre@kurilemu.de 1409 [ - + ]: 147 : Assert(XLogRecPtrIsValid(RecPtr));
1410 : :
1411 : : /* Make sure ReadPageInternal() can't return XLREAD_WOULDBLOCK. */
1509 tmunro@postgresql.or 1412 :CBC 147 : state->nonblocking = false;
1413 : :
1414 : : /*
1415 : : * skip over potential continuation data, keeping in mind that it may span
1416 : : * multiple pages
1417 : : */
1821 1418 : 147 : tmpRecPtr = RecPtr;
1419 : : while (true)
3536 fujii@postgresql.org 1420 :UBC 0 : {
1421 : : XLogRecPtr targetPagePtr;
1422 : : int targetRecOff;
1423 : : uint32 pageHeaderSize;
1424 : : int readLen;
1425 : :
1426 : : /*
1427 : : * Compute targetRecOff. It should typically be equal or greater than
1428 : : * short page-header since a valid record can't start anywhere before
1429 : : * that, except when caller has explicitly specified the offset that
1430 : : * falls somewhere there or when we are skipping multi-page
1431 : : * continuation record. It doesn't matter though because
1432 : : * ReadPageInternal() is prepared to handle that and will read at
1433 : : * least short page-header worth of data
1434 : : */
1821 tmunro@postgresql.or 1435 :CBC 147 : targetRecOff = tmpRecPtr % XLOG_BLCKSZ;
1436 : :
1437 : : /* scroll back to page boundary */
1438 : 147 : targetPagePtr = tmpRecPtr - targetRecOff;
1439 : :
1440 : : /* Read the page containing the record */
1441 : 147 : readLen = ReadPageInternal(state, targetPagePtr, targetRecOff);
1442 [ + + ]: 147 : if (readLen < 0)
3536 fujii@postgresql.org 1443 :GBC 1 : goto err;
1444 : :
1821 tmunro@postgresql.or 1445 :CBC 146 : header = (XLogPageHeader) state->readBuf;
1446 : :
3536 fujii@postgresql.org 1447 [ + + ]: 146 : pageHeaderSize = XLogPageHeaderSize(header);
1448 : :
1449 : : /* make sure we have enough data for the page header */
1821 tmunro@postgresql.or 1450 : 146 : readLen = ReadPageInternal(state, targetPagePtr, pageHeaderSize);
1451 [ - + ]: 146 : if (readLen < 0)
1821 tmunro@postgresql.or 1452 :UBC 0 : goto err;
1453 : :
1454 : : /* skip over potential continuation data */
3536 fujii@postgresql.org 1455 [ + + ]:CBC 146 : if (header->xlp_info & XLP_FIRST_IS_CONTRECORD)
1456 : : {
1457 : : /*
1458 : : * If the length of the remaining continuation data is more than
1459 : : * what can fit in this page, the continuation record crosses over
1460 : : * this page. Read the next page and try again. xlp_rem_len in the
1461 : : * next page header will contain the remaining length of the
1462 : : * continuation data
1463 : : *
1464 : : * Note that record headers are MAXALIGN'ed
1465 : : */
2371 1466 [ - + ]: 62 : if (MAXALIGN(header->xlp_rem_len) >= (XLOG_BLCKSZ - pageHeaderSize))
1821 tmunro@postgresql.or 1467 :UBC 0 : tmpRecPtr = targetPagePtr + XLOG_BLCKSZ;
1468 : : else
1469 : : {
1470 : : /*
1471 : : * The previous continuation record ends in this page. Set
1472 : : * tmpRecPtr to point to the first valid record
1473 : : */
1821 tmunro@postgresql.or 1474 :CBC 62 : tmpRecPtr = targetPagePtr + pageHeaderSize
3536 fujii@postgresql.org 1475 : 62 : + MAXALIGN(header->xlp_rem_len);
1476 : 62 : break;
1477 : : }
1478 : : }
1479 : : else
1480 : : {
1821 tmunro@postgresql.or 1481 : 84 : tmpRecPtr = targetPagePtr + pageHeaderSize;
3536 fujii@postgresql.org 1482 : 84 : break;
1483 : : }
1484 : : }
1485 : :
1486 : : /*
1487 : : * we know now that tmpRecPtr is an address pointing to a valid XLogRecord
1488 : : * because either we're at the first record after the beginning of a page
1489 : : * or we just jumped over the remaining data of a continuation.
1490 : : */
1821 tmunro@postgresql.or 1491 : 146 : XLogBeginRead(state, tmpRecPtr);
42 fujii@postgresql.org 1492 [ + - ]:GNC 1456 : while (XLogReadRecord(state, errormsg) != NULL)
1493 : : {
1494 : : /* past the record we've found, break out */
1821 tmunro@postgresql.or 1495 [ + + ]:CBC 1456 : if (RecPtr <= state->ReadRecPtr)
1496 : : {
1497 : : /* Rewind the reader to the beginning of the last record. */
1498 : 146 : found = state->ReadRecPtr;
1499 : 146 : XLogBeginRead(state, found);
1500 : 146 : return found;
1501 : : }
1502 : : }
1503 : :
4857 alvherre@alvh.no-ip. 1504 :UBC 0 : err:
1821 tmunro@postgresql.or 1505 :GBC 1 : XLogReaderInvalReadState(state);
1506 : :
1507 : : /*
1508 : : * We may have reported errors due to invalid WAL header, propagate the
1509 : : * error message to the caller.
1510 : : */
42 fujii@postgresql.org 1511 [ + - ]:GNC 1 : if (state->errormsg_deferred)
1512 : : {
1513 [ + - ]: 1 : if (state->errormsg_buf[0] != '\0')
1514 : 1 : *errormsg = state->errormsg_buf;
1515 : 1 : state->errormsg_deferred = false;
1516 : : }
1517 : :
1821 tmunro@postgresql.or 1518 :GBC 1 : return InvalidXLogRecPtr;
1519 : : }
1520 : :
1521 : : /*
1522 : : * Helper function to ease writing of XLogReaderRoutine->page_read callbacks.
1523 : : * If this function is used, caller must supply a segment_open callback in
1524 : : * 'state', as that is used here.
1525 : : *
1526 : : * Read 'count' bytes into 'buf', starting at location 'startptr', from WAL
1527 : : * fetched from timeline 'tli'.
1528 : : *
1529 : : * Returns true if succeeded, false if an error occurs, in which case
1530 : : * 'errinfo' receives error details.
1531 : : */
1532 : : bool
2188 alvherre@alvh.no-ip. 1533 :CBC 162549 : WALRead(XLogReaderState *state,
1534 : : char *buf, XLogRecPtr startptr, Size count, TimeLineID tli,
1535 : : WALReadError *errinfo)
1536 : : {
1537 : : char *p;
1538 : : XLogRecPtr recptr;
1539 : : Size nbytes;
1540 : : #ifndef FRONTEND
1541 : : instr_time io_start;
1542 : : #endif
1543 : :
2353 1544 : 162549 : p = buf;
1545 : 162549 : recptr = startptr;
1546 : 162549 : nbytes = count;
1547 : :
1548 [ + + ]: 325526 : while (nbytes > 0)
1549 : : {
1550 : : uint32 startoff;
1551 : : int segbytes;
1552 : : int readbytes;
1553 : :
2183 1554 : 162978 : startoff = XLogSegmentOffset(recptr, state->segcxt.ws_segsize);
1555 : :
1556 : : /*
1557 : : * If the data we want is not in a segment we have open, close what we
1558 : : * have (if anything) and open the next one, using the caller's
1559 : : * provided segment_open callback.
1560 : : */
1561 [ + + ]: 162978 : if (state->seg.ws_file < 0 ||
1562 [ + + ]: 161061 : !XLByteInSeg(recptr, state->seg.ws_segno, state->segcxt.ws_segsize) ||
1563 [ - + ]: 157038 : tli != state->seg.ws_tli)
1564 : : {
1565 : : XLogSegNo nextSegNo;
1566 : :
1567 [ + + ]: 5940 : if (state->seg.ws_file >= 0)
1821 tmunro@postgresql.or 1568 : 4023 : state->routine.segment_close(state);
1569 : :
2183 alvherre@alvh.no-ip. 1570 : 5940 : XLByteToSeg(recptr, nextSegNo, state->segcxt.ws_segsize);
1821 tmunro@postgresql.or 1571 : 5940 : state->routine.segment_open(state, nextSegNo, &tli);
1572 : :
1573 : : /* This shouldn't happen -- indicates a bug in segment_open */
2183 alvherre@alvh.no-ip. 1574 [ - + ]: 5939 : Assert(state->seg.ws_file >= 0);
1575 : :
1576 : : /* Update the current segment info. */
1577 : 5939 : state->seg.ws_tli = tli;
1578 : 5939 : state->seg.ws_segno = nextSegNo;
1579 : : }
1580 : :
1581 : : /* How many bytes are within this segment? */
1582 [ + + ]: 162977 : if (nbytes > (state->segcxt.ws_segsize - startoff))
1583 : 429 : segbytes = state->segcxt.ws_segsize - startoff;
1584 : : else
2353 1585 : 162548 : segbytes = nbytes;
1586 : :
1587 : : #ifndef FRONTEND
1588 : : /* Measure I/O timing when reading segment */
433 michael@paquier.xyz 1589 : 141832 : io_start = pgstat_prepare_io_time(track_wal_io_timing);
1590 : :
2353 alvherre@alvh.no-ip. 1591 : 141832 : pgstat_report_wait_start(WAIT_EVENT_WAL_READ);
1592 : : #endif
1593 : :
1594 : : /* Reset errno first; eases reporting non-errno-affecting errors */
1595 : 162977 : errno = 0;
173 michael@paquier.xyz 1596 :GNC 162977 : readbytes = pg_pread(state->seg.ws_file, p, segbytes, (pgoff_t) startoff);
1597 : :
1598 : : #ifndef FRONTEND
2353 alvherre@alvh.no-ip. 1599 :CBC 141832 : pgstat_report_wait_end();
1600 : :
455 michael@paquier.xyz 1601 : 141832 : pgstat_count_io_op_time(IOOBJECT_WAL, IOCONTEXT_NORMAL, IOOP_READ,
1602 : : io_start, 1, readbytes);
1603 : : #endif
1604 : :
2353 alvherre@alvh.no-ip. 1605 [ - + ]: 162977 : if (readbytes <= 0)
1606 : : {
2353 alvherre@alvh.no-ip. 1607 :UBC 0 : errinfo->wre_errno = errno;
1608 : 0 : errinfo->wre_req = segbytes;
1609 : 0 : errinfo->wre_read = readbytes;
1610 : 0 : errinfo->wre_off = startoff;
2183 1611 : 0 : errinfo->wre_seg = state->seg;
2353 1612 : 0 : return false;
1613 : : }
1614 : :
1615 : : /* Update state for read */
2353 alvherre@alvh.no-ip. 1616 :CBC 162977 : recptr += readbytes;
1617 : 162977 : nbytes -= readbytes;
1618 : 162977 : p += readbytes;
1619 : : }
1620 : :
1621 : 162548 : return true;
1622 : : }
1623 : :
1624 : : /* ----------------------------------------
1625 : : * Functions for decoding the data and block references in a record.
1626 : : * ----------------------------------------
1627 : : */
1628 : :
1629 : : /*
1630 : : * Private function to reset the state, forgetting all decoded records, if we
1631 : : * are asked to move to a new read position.
1632 : : */
1633 : : static void
4184 heikki.linnakangas@i 1634 : 7018 : ResetDecoder(XLogReaderState *state)
1635 : : {
1636 : : DecodedXLogRecord *r;
1637 : :
1638 : : /* Reset the decoded record queue, freeing any oversized records. */
1509 tmunro@postgresql.or 1639 [ + + ]: 18127 : while ((r = state->decode_queue_head) != NULL)
1640 : : {
1641 : 4091 : state->decode_queue_head = r->next;
1642 [ + + ]: 4091 : if (r->oversized)
1509 tmunro@postgresql.or 1643 :GBC 3 : pfree(r);
1644 : : }
1509 tmunro@postgresql.or 1645 :CBC 7018 : state->decode_queue_tail = NULL;
1646 : 7018 : state->decode_queue_head = NULL;
1647 : 7018 : state->record = NULL;
1648 : :
1649 : : /* Reset the decode buffer to empty. */
1650 : 7018 : state->decode_buffer_tail = state->decode_buffer;
1651 : 7018 : state->decode_buffer_head = state->decode_buffer;
1652 : :
1653 : : /* Clear error state. */
1654 : 7018 : state->errormsg_buf[0] = '\0';
1655 : 7018 : state->errormsg_deferred = false;
4184 heikki.linnakangas@i 1656 : 7018 : }
1657 : :
1658 : : /*
1659 : : * Compute the maximum possible amount of padding that could be required to
1660 : : * decode a record, given xl_tot_len from the record's header. This is the
1661 : : * amount of output buffer space that we need to decode a record, though we
1662 : : * might not finish up using it all.
1663 : : *
1664 : : * This computation is pessimistic and assumes the maximum possible number of
1665 : : * blocks, due to lack of better information.
1666 : : */
1667 : : size_t
1509 tmunro@postgresql.or 1668 : 13901532 : DecodeXLogRecordRequiredSpace(size_t xl_tot_len)
1669 : : {
1670 : 13901532 : size_t size = 0;
1671 : :
1672 : : /* Account for the fixed size part of the decoded record struct. */
1673 : 13901532 : size += offsetof(DecodedXLogRecord, blocks[0]);
1674 : : /* Account for the flexible blocks array of maximum possible size. */
1675 : 13901532 : size += sizeof(DecodedBkpBlock) * (XLR_MAX_BLOCK_ID + 1);
1676 : : /* Account for all the raw main and block data. */
1677 : 13901532 : size += xl_tot_len;
1678 : : /* We might insert padding before main_data. */
1679 : 13901532 : size += (MAXIMUM_ALIGNOF - 1);
1680 : : /* We might insert padding before each block's data. */
1681 : 13901532 : size += (MAXIMUM_ALIGNOF - 1) * (XLR_MAX_BLOCK_ID + 1);
1682 : : /* We might insert padding at the end. */
1683 : 13901532 : size += (MAXIMUM_ALIGNOF - 1);
1684 : :
1685 : 13901532 : return size;
1686 : : }
1687 : :
1688 : : /*
1689 : : * Decode a record. "decoded" must point to a MAXALIGNed memory area that has
1690 : : * space for at least DecodeXLogRecordRequiredSpace(record) bytes. On
1691 : : * success, decoded->size contains the actual space occupied by the decoded
1692 : : * record, which may turn out to be less.
1693 : : *
1694 : : * Only decoded->oversized member must be initialized already, and will not be
1695 : : * modified. Other members will be initialized as required.
1696 : : *
1697 : : * On error, a human-readable error message is returned in *errormsg, and
1698 : : * the return value is false.
1699 : : */
1700 : : bool
1701 : 6936021 : DecodeXLogRecord(XLogReaderState *state,
1702 : : DecodedXLogRecord *decoded,
1703 : : XLogRecord *record,
1704 : : XLogRecPtr lsn,
1705 : : char **errormsg)
1706 : : {
1707 : : /*
1708 : : * read next _size bytes from record buffer, but check for overrun first.
1709 : : */
1710 : : #define COPY_HEADER_FIELD(_dst, _size) \
1711 : : do { \
1712 : : if (remaining < _size) \
1713 : : goto shortdata_err; \
1714 : : memcpy(_dst, ptr, _size); \
1715 : : ptr += _size; \
1716 : : remaining -= _size; \
1717 : : } while(0)
1718 : :
1719 : : char *ptr;
1720 : : char *out;
1721 : : uint32 remaining;
1722 : : uint32 datatotal;
1399 rhaas@postgresql.org 1723 : 6936021 : RelFileLocator *rlocator = NULL;
1724 : : uint8 block_id;
1725 : :
1509 tmunro@postgresql.or 1726 : 6936021 : decoded->header = *record;
1727 : 6936021 : decoded->lsn = lsn;
1728 : 6936021 : decoded->next = NULL;
97 msawada@postgresql.o 1729 :GNC 6936021 : decoded->record_origin = InvalidReplOriginId;
1509 tmunro@postgresql.or 1730 :CBC 6936021 : decoded->toplevel_xid = InvalidTransactionId;
1731 : 6936021 : decoded->main_data = NULL;
1732 : 6936021 : decoded->main_data_len = 0;
1733 : 6936021 : decoded->max_block_id = -1;
4184 heikki.linnakangas@i 1734 : 6936021 : ptr = (char *) record;
1735 : 6936021 : ptr += SizeOfXLogRecord;
1736 : 6936021 : remaining = record->xl_tot_len - SizeOfXLogRecord;
1737 : :
1738 : : /* Decode the headers */
1739 : 6936021 : datatotal = 0;
1740 [ + + ]: 14287000 : while (remaining > datatotal)
1741 : : {
1742 [ - + ]: 14182282 : COPY_HEADER_FIELD(&block_id, sizeof(uint8));
1743 : :
1744 [ + + ]: 14182282 : if (block_id == XLR_BLOCK_ID_DATA_SHORT)
1745 : : {
1746 : : /* XLogRecordDataHeaderShort */
1747 : : uint8 main_data_len;
1748 : :
1749 [ - + ]: 6818377 : COPY_HEADER_FIELD(&main_data_len, sizeof(uint8));
1750 : :
1509 tmunro@postgresql.or 1751 : 6818377 : decoded->main_data_len = main_data_len;
4184 heikki.linnakangas@i 1752 : 6818377 : datatotal += main_data_len;
1753 : 6818377 : break; /* by convention, the main data fragment is
1754 : : * always last */
1755 : : }
1756 [ + + ]: 7363905 : else if (block_id == XLR_BLOCK_ID_DATA_LONG)
1757 : : {
1758 : : /* XLogRecordDataHeaderLong */
1759 : : uint32 main_data_len;
1760 : :
1761 [ - + ]: 12926 : COPY_HEADER_FIELD(&main_data_len, sizeof(uint32));
1509 tmunro@postgresql.or 1762 : 12926 : decoded->main_data_len = main_data_len;
4184 heikki.linnakangas@i 1763 : 12926 : datatotal += main_data_len;
1764 : 12926 : break; /* by convention, the main data fragment is
1765 : : * always last */
1766 : : }
4024 andres@anarazel.de 1767 [ + + ]: 7350979 : else if (block_id == XLR_BLOCK_ID_ORIGIN)
1768 : : {
97 msawada@postgresql.o 1769 [ - + ]:GNC 13061 : COPY_HEADER_FIELD(&decoded->record_origin, sizeof(ReplOriginId));
1770 : : }
2115 akapila@postgresql.o 1771 [ + + ]:CBC 7337918 : else if (block_id == XLR_BLOCK_ID_TOPLEVEL_XID)
1772 : : {
1509 tmunro@postgresql.or 1773 [ - + ]: 647 : COPY_HEADER_FIELD(&decoded->toplevel_xid, sizeof(TransactionId));
1774 : : }
4184 heikki.linnakangas@i 1775 [ + - ]: 7337271 : else if (block_id <= XLR_MAX_BLOCK_ID)
1776 : : {
1777 : : /* XLogRecordBlockHeader */
1778 : : DecodedBkpBlock *blk;
1779 : : uint8 fork_flags;
1780 : :
1781 : : /* mark any intervening block IDs as not in use */
1509 tmunro@postgresql.or 1782 [ + + ]: 7340641 : for (int i = decoded->max_block_id + 1; i < block_id; ++i)
1783 : 3370 : decoded->blocks[i].in_use = false;
1784 : :
1785 [ - + ]: 7337271 : if (block_id <= decoded->max_block_id)
1786 : : {
4184 heikki.linnakangas@i 1787 :UBC 0 : report_invalid_record(state,
1788 : : "out-of-order block_id %u at %X/%08X",
1789 : : block_id,
1897 peter@eisentraut.org 1790 : 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4184 heikki.linnakangas@i 1791 : 0 : goto err;
1792 : : }
1509 tmunro@postgresql.or 1793 :CBC 7337271 : decoded->max_block_id = block_id;
1794 : :
1795 : 7337271 : blk = &decoded->blocks[block_id];
4184 heikki.linnakangas@i 1796 : 7337271 : blk->in_use = true;
3373 rhaas@postgresql.org 1797 : 7337271 : blk->apply_image = false;
1798 : :
4184 heikki.linnakangas@i 1799 [ - + ]: 7337271 : COPY_HEADER_FIELD(&fork_flags, sizeof(uint8));
1800 : 7337271 : blk->forknum = fork_flags & BKPBLOCK_FORK_MASK;
1801 : 7337271 : blk->flags = fork_flags;
1802 : 7337271 : blk->has_image = ((fork_flags & BKPBLOCK_HAS_IMAGE) != 0);
1803 : 7337271 : blk->has_data = ((fork_flags & BKPBLOCK_HAS_DATA) != 0);
1804 : :
1489 tmunro@postgresql.or 1805 : 7337271 : blk->prefetch_buffer = InvalidBuffer;
1806 : :
4184 heikki.linnakangas@i 1807 [ - + ]: 7337271 : COPY_HEADER_FIELD(&blk->data_len, sizeof(uint16));
1808 : : /* cross-check that the HAS_DATA flag is set iff data_length > 0 */
1809 [ + + - + ]: 7337271 : if (blk->has_data && blk->data_len == 0)
1810 : : {
4184 heikki.linnakangas@i 1811 :UBC 0 : report_invalid_record(state,
1812 : : "BKPBLOCK_HAS_DATA set, but no data included at %X/%08X",
1897 peter@eisentraut.org 1813 : 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4075 fujii@postgresql.org 1814 : 0 : goto err;
1815 : : }
4184 heikki.linnakangas@i 1816 [ + + - + ]:CBC 7337271 : if (!blk->has_data && blk->data_len != 0)
1817 : : {
4184 heikki.linnakangas@i 1818 :UBC 0 : report_invalid_record(state,
1819 : : "BKPBLOCK_HAS_DATA not set, but data length is %d at %X/%08X",
147 peter@eisentraut.org 1820 :UNC 0 : blk->data_len,
1897 peter@eisentraut.org 1821 :UBC 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4075 fujii@postgresql.org 1822 : 0 : goto err;
1823 : : }
4184 heikki.linnakangas@i 1824 :CBC 7337271 : datatotal += blk->data_len;
1825 : :
1826 [ + + ]: 7337271 : if (blk->has_image)
1827 : : {
4073 fujii@postgresql.org 1828 [ - + ]: 2560993 : COPY_HEADER_FIELD(&blk->bimg_len, sizeof(uint16));
4184 heikki.linnakangas@i 1829 [ - + ]: 2560993 : COPY_HEADER_FIELD(&blk->hole_offset, sizeof(uint16));
4073 fujii@postgresql.org 1830 [ - + ]: 2560993 : COPY_HEADER_FIELD(&blk->bimg_info, sizeof(uint8));
1831 : :
3373 rhaas@postgresql.org 1832 : 2560993 : blk->apply_image = ((blk->bimg_info & BKPIMAGE_APPLY) != 0);
1833 : :
1771 michael@paquier.xyz 1834 [ - + ]: 2560993 : if (BKPIMAGE_COMPRESSED(blk->bimg_info))
1835 : : {
4073 fujii@postgresql.org 1836 [ # # ]:UBC 0 : if (blk->bimg_info & BKPIMAGE_HAS_HOLE)
1837 [ # # ]: 0 : COPY_HEADER_FIELD(&blk->hole_length, sizeof(uint16));
1838 : : else
1839 : 0 : blk->hole_length = 0;
1840 : : }
1841 : : else
4073 fujii@postgresql.org 1842 :CBC 2560993 : blk->hole_length = BLCKSZ - blk->bimg_len;
1843 : 2560993 : datatotal += blk->bimg_len;
1844 : :
1845 : : /*
1846 : : * cross-check that hole_offset > 0, hole_length > 0 and
1847 : : * bimg_len < BLCKSZ if the HAS_HOLE flag is set.
1848 : : */
1849 [ + + ]: 2560993 : if ((blk->bimg_info & BKPIMAGE_HAS_HOLE) &&
1850 [ + - ]: 2532410 : (blk->hole_offset == 0 ||
1851 [ + - ]: 2532410 : blk->hole_length == 0 ||
1852 [ - + ]: 2532410 : blk->bimg_len == BLCKSZ))
1853 : : {
4073 fujii@postgresql.org 1854 :UBC 0 : report_invalid_record(state,
1855 : : "BKPIMAGE_HAS_HOLE set, but hole offset %d length %d block image length %d at %X/%08X",
147 peter@eisentraut.org 1856 :UNC 0 : blk->hole_offset,
1857 : 0 : blk->hole_length,
1858 : 0 : blk->bimg_len,
1897 peter@eisentraut.org 1859 :UBC 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4073 fujii@postgresql.org 1860 : 0 : goto err;
1861 : : }
1862 : :
1863 : : /*
1864 : : * cross-check that hole_offset == 0 and hole_length == 0 if
1865 : : * the HAS_HOLE flag is not set.
1866 : : */
4073 fujii@postgresql.org 1867 [ + + ]:CBC 2560993 : if (!(blk->bimg_info & BKPIMAGE_HAS_HOLE) &&
1868 [ + - - + ]: 28583 : (blk->hole_offset != 0 || blk->hole_length != 0))
1869 : : {
4073 fujii@postgresql.org 1870 :UBC 0 : report_invalid_record(state,
1871 : : "BKPIMAGE_HAS_HOLE not set, but hole offset %d length %d at %X/%08X",
147 peter@eisentraut.org 1872 :UNC 0 : blk->hole_offset,
1873 : 0 : blk->hole_length,
1897 peter@eisentraut.org 1874 :UBC 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4073 fujii@postgresql.org 1875 : 0 : goto err;
1876 : : }
1877 : :
1878 : : /*
1879 : : * Cross-check that bimg_len < BLCKSZ if it is compressed.
1880 : : */
1771 michael@paquier.xyz 1881 [ - + ]:CBC 2560993 : if (BKPIMAGE_COMPRESSED(blk->bimg_info) &&
4073 fujii@postgresql.org 1882 [ # # ]:UBC 0 : blk->bimg_len == BLCKSZ)
1883 : : {
1884 : 0 : report_invalid_record(state,
1885 : : "BKPIMAGE_COMPRESSED set, but block image length %d at %X/%08X",
147 peter@eisentraut.org 1886 :UNC 0 : blk->bimg_len,
1897 peter@eisentraut.org 1887 :UBC 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4073 fujii@postgresql.org 1888 : 0 : goto err;
1889 : : }
1890 : :
1891 : : /*
1892 : : * cross-check that bimg_len = BLCKSZ if neither HAS_HOLE is
1893 : : * set nor COMPRESSED().
1894 : : */
4073 fujii@postgresql.org 1895 [ + + ]:CBC 2560993 : if (!(blk->bimg_info & BKPIMAGE_HAS_HOLE) &&
1771 michael@paquier.xyz 1896 [ + - ]: 28583 : !BKPIMAGE_COMPRESSED(blk->bimg_info) &&
4073 fujii@postgresql.org 1897 [ - + ]: 28583 : blk->bimg_len != BLCKSZ)
1898 : : {
4073 fujii@postgresql.org 1899 :UBC 0 : report_invalid_record(state,
1900 : : "neither BKPIMAGE_HAS_HOLE nor BKPIMAGE_COMPRESSED set, but block image length is %d at %X/%08X",
147 peter@eisentraut.org 1901 :UNC 0 : blk->data_len,
1897 peter@eisentraut.org 1902 :UBC 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4073 fujii@postgresql.org 1903 : 0 : goto err;
1904 : : }
1905 : : }
4184 heikki.linnakangas@i 1906 [ + + ]:CBC 7337271 : if (!(fork_flags & BKPBLOCK_SAME_REL))
1907 : : {
1399 rhaas@postgresql.org 1908 [ - + ]: 6699707 : COPY_HEADER_FIELD(&blk->rlocator, sizeof(RelFileLocator));
1909 : 6699707 : rlocator = &blk->rlocator;
1910 : : }
1911 : : else
1912 : : {
1913 [ - + ]: 637564 : if (rlocator == NULL)
1914 : : {
4184 heikki.linnakangas@i 1915 :UBC 0 : report_invalid_record(state,
1916 : : "BKPBLOCK_SAME_REL set but no previous rel at %X/%08X",
1897 peter@eisentraut.org 1917 : 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4184 heikki.linnakangas@i 1918 : 0 : goto err;
1919 : : }
1920 : :
1399 rhaas@postgresql.org 1921 :CBC 637564 : blk->rlocator = *rlocator;
1922 : : }
4184 heikki.linnakangas@i 1923 [ - + ]: 7337271 : COPY_HEADER_FIELD(&blk->blkno, sizeof(BlockNumber));
1924 : : }
1925 : : else
1926 : : {
4184 heikki.linnakangas@i 1927 :UBC 0 : report_invalid_record(state,
1928 : : "invalid block_id %u at %X/%08X",
1897 peter@eisentraut.org 1929 : 0 : block_id, LSN_FORMAT_ARGS(state->ReadRecPtr));
4184 heikki.linnakangas@i 1930 : 0 : goto err;
1931 : : }
1932 : : }
1933 : :
4184 heikki.linnakangas@i 1934 [ - + ]:CBC 6936021 : if (remaining != datatotal)
4184 heikki.linnakangas@i 1935 :UBC 0 : goto shortdata_err;
1936 : :
1937 : : /*
1938 : : * Ok, we've parsed the fragment headers, and verified that the total
1939 : : * length of the payload in the fragments is equal to the amount of data
1940 : : * left. Copy the data of each fragment to contiguous space after the
1941 : : * blocks array, inserting alignment padding before the data fragments so
1942 : : * they can be cast to struct pointers by REDO routines.
1943 : : */
1509 tmunro@postgresql.or 1944 :CBC 6936021 : out = ((char *) decoded) +
1945 : 6936021 : offsetof(DecodedXLogRecord, blocks) +
1946 : 6936021 : sizeof(decoded->blocks[0]) * (decoded->max_block_id + 1);
1947 : :
1948 : : /* block data first */
1949 [ + + ]: 14276662 : for (block_id = 0; block_id <= decoded->max_block_id; block_id++)
1950 : : {
1951 : 7340641 : DecodedBkpBlock *blk = &decoded->blocks[block_id];
1952 : :
4184 heikki.linnakangas@i 1953 [ + + ]: 7340641 : if (!blk->in_use)
1954 : 3370 : continue;
1955 : :
3373 rhaas@postgresql.org 1956 [ + + - + ]: 7337271 : Assert(blk->has_image || !blk->apply_image);
1957 : :
4184 heikki.linnakangas@i 1958 [ + + ]: 7337271 : if (blk->has_image)
1959 : : {
1960 : : /* no need to align image */
1509 tmunro@postgresql.or 1961 : 2560993 : blk->bkp_image = out;
1962 : 2560993 : memcpy(out, ptr, blk->bimg_len);
4073 fujii@postgresql.org 1963 : 2560993 : ptr += blk->bimg_len;
1509 tmunro@postgresql.or 1964 : 2560993 : out += blk->bimg_len;
1965 : : }
4184 heikki.linnakangas@i 1966 [ + + ]: 7337271 : if (blk->has_data)
1967 : : {
1509 tmunro@postgresql.or 1968 : 5423674 : out = (char *) MAXALIGN(out);
1969 : 5423674 : blk->data = out;
4184 heikki.linnakangas@i 1970 : 5423674 : memcpy(blk->data, ptr, blk->data_len);
1971 : 5423674 : ptr += blk->data_len;
1509 tmunro@postgresql.or 1972 : 5423674 : out += blk->data_len;
1973 : : }
1974 : : }
1975 : :
1976 : : /* and finally, the main data */
1977 [ + + ]: 6936021 : if (decoded->main_data_len > 0)
1978 : : {
1979 : 6831303 : out = (char *) MAXALIGN(out);
1980 : 6831303 : decoded->main_data = out;
1981 : 6831303 : memcpy(decoded->main_data, ptr, decoded->main_data_len);
1982 : 6831303 : ptr += decoded->main_data_len;
1983 : 6831303 : out += decoded->main_data_len;
1984 : : }
1985 : :
1986 : : /* Report the actual size we used. */
1987 : 6936021 : decoded->size = MAXALIGN(out - (char *) decoded);
1988 [ - + ]: 6936021 : Assert(DecodeXLogRecordRequiredSpace(record->xl_tot_len) >=
1989 : : decoded->size);
1990 : :
4184 heikki.linnakangas@i 1991 : 6936021 : return true;
1992 : :
4184 heikki.linnakangas@i 1993 :UBC 0 : shortdata_err:
1994 : 0 : report_invalid_record(state,
1995 : : "record with invalid length at %X/%08X",
1897 peter@eisentraut.org 1996 : 0 : LSN_FORMAT_ARGS(state->ReadRecPtr));
4184 heikki.linnakangas@i 1997 : 0 : err:
1998 : 0 : *errormsg = state->errormsg_buf;
1999 : :
2000 : 0 : return false;
2001 : : }
2002 : :
2003 : : /*
2004 : : * Returns information about the block that a block reference refers to.
2005 : : *
2006 : : * This is like XLogRecGetBlockTagExtended, except that the block reference
2007 : : * must exist and there's no access to prefetch_buffer.
2008 : : */
2009 : : void
4184 heikki.linnakangas@i 2010 :CBC 3265780 : XLogRecGetBlockTag(XLogReaderState *record, uint8 block_id,
2011 : : RelFileLocator *rlocator, ForkNumber *forknum,
2012 : : BlockNumber *blknum)
2013 : : {
1399 rhaas@postgresql.org 2014 [ - + ]: 3265780 : if (!XLogRecGetBlockTagExtended(record, block_id, rlocator, forknum,
2015 : : blknum, NULL))
2016 : : {
2017 : : #ifndef FRONTEND
1319 peter@eisentraut.org 2018 [ # # ]:UBC 0 : elog(ERROR, "could not locate backup block with ID %d in WAL record",
2019 : : block_id);
2020 : : #else
2021 : 0 : pg_fatal("could not locate backup block with ID %d in WAL record",
2022 : : block_id);
2023 : : #endif
2024 : : }
1489 tmunro@postgresql.or 2025 :CBC 3265780 : }
2026 : :
2027 : : /*
2028 : : * Returns information about the block that a block reference refers to,
2029 : : * optionally including the buffer that the block may already be in.
2030 : : *
2031 : : * If the WAL record contains a block reference with the given ID, *rlocator,
2032 : : * *forknum, *blknum and *prefetch_buffer are filled in (if not NULL), and
2033 : : * returns true. Otherwise returns false.
2034 : : */
2035 : : bool
2036 : 10049172 : XLogRecGetBlockTagExtended(XLogReaderState *record, uint8 block_id,
2037 : : RelFileLocator *rlocator, ForkNumber *forknum,
2038 : : BlockNumber *blknum,
2039 : : Buffer *prefetch_buffer)
2040 : : {
2041 : : DecodedBkpBlock *bkpb;
2042 : :
1485 tgl@sss.pgh.pa.us 2043 [ + + + + ]: 10049172 : if (!XLogRecHasBlockRef(record, block_id))
4184 heikki.linnakangas@i 2044 : 44905 : return false;
2045 : :
1509 tmunro@postgresql.or 2046 : 10004267 : bkpb = &record->record->blocks[block_id];
1399 rhaas@postgresql.org 2047 [ + + ]: 10004267 : if (rlocator)
2048 : 9943494 : *rlocator = bkpb->rlocator;
4184 heikki.linnakangas@i 2049 [ + + ]: 10004267 : if (forknum)
2050 : 6682711 : *forknum = bkpb->forknum;
2051 [ + + ]: 10004267 : if (blknum)
2052 : 8567548 : *blknum = bkpb->blkno;
1489 tmunro@postgresql.or 2053 [ + + ]: 10004267 : if (prefetch_buffer)
2054 : 3029949 : *prefetch_buffer = bkpb->prefetch_buffer;
4184 heikki.linnakangas@i 2055 : 10004267 : return true;
2056 : : }
2057 : :
2058 : : /*
2059 : : * Returns the data associated with a block reference, or NULL if there is
2060 : : * no data (e.g. because a full-page image was taken instead). The returned
2061 : : * pointer points to a MAXALIGNed buffer.
2062 : : */
2063 : : char *
2064 : 3463037 : XLogRecGetBlockData(XLogReaderState *record, uint8 block_id, Size *len)
2065 : : {
2066 : : DecodedBkpBlock *bkpb;
2067 : :
1509 tmunro@postgresql.or 2068 [ + - ]: 3463037 : if (block_id > record->record->max_block_id ||
2069 [ - + ]: 3463037 : !record->record->blocks[block_id].in_use)
4184 heikki.linnakangas@i 2070 :UBC 0 : return NULL;
2071 : :
1509 tmunro@postgresql.or 2072 :CBC 3463037 : bkpb = &record->record->blocks[block_id];
2073 : :
4184 heikki.linnakangas@i 2074 [ + + ]: 3463037 : if (!bkpb->has_data)
2075 : : {
2076 [ + - ]: 2132 : if (len)
2077 : 2132 : *len = 0;
2078 : 2132 : return NULL;
2079 : : }
2080 : : else
2081 : : {
2082 [ + + ]: 3460905 : if (len)
2083 : 3456387 : *len = bkpb->data_len;
2084 : 3460905 : return bkpb->data;
2085 : : }
2086 : : }
2087 : :
2088 : : /*
2089 : : * Restore a full-page image from a backup block attached to an XLOG record.
2090 : : *
2091 : : * Returns true if a full-page image is restored, and false on failure with
2092 : : * an error to be consumed by the caller.
2093 : : */
2094 : : bool
2095 : 2491159 : RestoreBlockImage(XLogReaderState *record, uint8 block_id, char *page)
2096 : : {
2097 : : DecodedBkpBlock *bkpb;
2098 : : char *ptr;
2099 : : PGAlignedBlock tmp;
2100 : :
1509 tmunro@postgresql.or 2101 [ + - ]: 2491159 : if (block_id > record->record->max_block_id ||
2102 [ - + ]: 2491159 : !record->record->blocks[block_id].in_use)
2103 : : {
1334 michael@paquier.xyz 2104 :UBC 0 : report_invalid_record(record,
2105 : : "could not restore image at %X/%08X with invalid block %d specified",
2106 : 0 : LSN_FORMAT_ARGS(record->ReadRecPtr),
2107 : : block_id);
4184 heikki.linnakangas@i 2108 : 0 : return false;
2109 : : }
1509 tmunro@postgresql.or 2110 [ - + ]:CBC 2491159 : if (!record->record->blocks[block_id].has_image)
2111 : : {
302 alvherre@kurilemu.de 2112 :UNC 0 : report_invalid_record(record, "could not restore image at %X/%08X with invalid state, block %d",
1334 michael@paquier.xyz 2113 :UBC 0 : LSN_FORMAT_ARGS(record->ReadRecPtr),
2114 : : block_id);
4184 heikki.linnakangas@i 2115 : 0 : return false;
2116 : : }
2117 : :
1509 tmunro@postgresql.or 2118 :CBC 2491159 : bkpb = &record->record->blocks[block_id];
4073 fujii@postgresql.org 2119 : 2491159 : ptr = bkpb->bkp_image;
2120 : :
1771 michael@paquier.xyz 2121 [ - + ]: 2491159 : if (BKPIMAGE_COMPRESSED(bkpb->bimg_info))
2122 : : {
2123 : : /* If a backup block image is compressed, decompress it */
1771 michael@paquier.xyz 2124 :UBC 0 : bool decomp_success = true;
2125 : :
2126 [ # # ]: 0 : if ((bkpb->bimg_info & BKPIMAGE_COMPRESS_PGLZ) != 0)
2127 : : {
2128 [ # # ]: 0 : if (pglz_decompress(ptr, bkpb->bimg_len, tmp.data,
2129 : 0 : BLCKSZ - bkpb->hole_length, true) < 0)
2130 : 0 : decomp_success = false;
2131 : : }
2132 [ # # ]: 0 : else if ((bkpb->bimg_info & BKPIMAGE_COMPRESS_LZ4) != 0)
2133 : : {
2134 : : #ifdef USE_LZ4
2135 [ # # ]: 0 : if (LZ4_decompress_safe(ptr, tmp.data,
2136 : 0 : bkpb->bimg_len, BLCKSZ - bkpb->hole_length) <= 0)
2137 : 0 : decomp_success = false;
2138 : : #else
2139 : : report_invalid_record(record, "could not restore image at %X/%08X compressed with %s not supported by build, block %d",
2140 : : LSN_FORMAT_ARGS(record->ReadRecPtr),
2141 : : "LZ4",
2142 : : block_id);
2143 : : return false;
2144 : : #endif
2145 : : }
1516 2146 [ # # ]: 0 : else if ((bkpb->bimg_info & BKPIMAGE_COMPRESS_ZSTD) != 0)
2147 : : {
2148 : : #ifdef USE_ZSTD
2149 : 0 : size_t decomp_result = ZSTD_decompress(tmp.data,
2150 : 0 : BLCKSZ - bkpb->hole_length,
2151 : 0 : ptr, bkpb->bimg_len);
2152 : :
2153 [ # # ]: 0 : if (ZSTD_isError(decomp_result))
2154 : 0 : decomp_success = false;
2155 : : #else
2156 : : report_invalid_record(record, "could not restore image at %X/%08X compressed with %s not supported by build, block %d",
2157 : : LSN_FORMAT_ARGS(record->ReadRecPtr),
2158 : : "zstd",
2159 : : block_id);
2160 : : return false;
2161 : : #endif
2162 : : }
2163 : : else
2164 : : {
302 alvherre@kurilemu.de 2165 :UNC 0 : report_invalid_record(record, "could not restore image at %X/%08X compressed with unknown method, block %d",
1761 michael@paquier.xyz 2166 :UBC 0 : LSN_FORMAT_ARGS(record->ReadRecPtr),
2167 : : block_id);
1771 2168 : 0 : return false;
2169 : : }
2170 : :
2171 [ # # ]: 0 : if (!decomp_success)
2172 : : {
302 alvherre@kurilemu.de 2173 :UNC 0 : report_invalid_record(record, "could not decompress image at %X/%08X, block %d",
1897 peter@eisentraut.org 2174 :UBC 0 : LSN_FORMAT_ARGS(record->ReadRecPtr),
2175 : : block_id);
4073 fujii@postgresql.org 2176 : 0 : return false;
2177 : : }
2178 : :
2803 tgl@sss.pgh.pa.us 2179 : 0 : ptr = tmp.data;
2180 : : }
2181 : :
2182 : : /* generate page, taking into account hole if necessary */
4184 heikki.linnakangas@i 2183 [ + + ]:CBC 2491159 : if (bkpb->hole_length == 0)
2184 : : {
4073 fujii@postgresql.org 2185 : 22532 : memcpy(page, ptr, BLCKSZ);
2186 : : }
2187 : : else
2188 : : {
2189 : 2468627 : memcpy(page, ptr, bkpb->hole_offset);
2190 : : /* must zero-fill the hole */
4184 heikki.linnakangas@i 2191 [ + + + - : 13062478 : MemSet(page + bkpb->hole_offset, 0, bkpb->hole_length);
+ - + + +
+ ]
2192 : 2468627 : memcpy(page + (bkpb->hole_offset + bkpb->hole_length),
4073 fujii@postgresql.org 2193 : 2468627 : ptr + bkpb->hole_offset,
4184 heikki.linnakangas@i 2194 : 2468627 : BLCKSZ - (bkpb->hole_offset + bkpb->hole_length));
2195 : : }
2196 : :
2197 : 2491159 : return true;
2198 : : }
2199 : :
2200 : : #ifndef FRONTEND
2201 : :
2202 : : /*
2203 : : * Extract the FullTransactionId from a WAL record.
2204 : : */
2205 : : FullTransactionId
2486 tmunro@postgresql.or 2206 :UBC 0 : XLogRecGetFullXid(XLogReaderState *record)
2207 : : {
2208 : : /*
2209 : : * This function is only safe during replay, because it depends on the
2210 : : * replay state. See AdvanceNextFullTransactionIdPastXid() for more.
2211 : : */
2212 [ # # # # ]: 0 : Assert(AmStartupProcess() || !IsUnderPostmaster);
2213 : :
465 noah@leadboat.com 2214 : 0 : return FullTransactionIdFromAllowableAt(TransamVariables->nextXid,
2215 : 0 : XLogRecGetXid(record));
2216 : : }
2217 : :
2218 : : #endif
|
