md.c 47.6 KB
Newer Older
1 2
/*-------------------------------------------------------------------------
 *
3
 * md.c
4
 *	  This code manages relations that reside on magnetic disk.
5
 *
B
Bruce Momjian 已提交
6
 * Portions Copyright (c) 1996-2009, PostgreSQL Global Development Group
B
Add:  
Bruce Momjian 已提交
7
 * Portions Copyright (c) 1994, Regents of the University of California
8 9 10
 *
 *
 * IDENTIFICATION
11
 *	  $PostgreSQL: pgsql/src/backend/storage/smgr/md.c,v 1.146 2009/06/11 14:49:02 momjian Exp $
12 13 14
 *
 *-------------------------------------------------------------------------
 */
15 16
#include "postgres.h"

B
Bruce Momjian 已提交
17
#include <unistd.h>
B
Bruce Momjian 已提交
18
#include <fcntl.h>
19 20
#include <sys/file.h>

21
#include "catalog/catalog.h"
B
Bruce Momjian 已提交
22
#include "miscadmin.h"
23
#include "postmaster/bgwriter.h"
24
#include "storage/fd.h"
25
#include "storage/bufmgr.h"
26
#include "storage/relfilenode.h"
B
Bruce Momjian 已提交
27
#include "storage/smgr.h"
28
#include "utils/hsearch.h"
29
#include "utils/memutils.h"
30
#include "pg_trace.h"
31

32

33 34 35
/* interval for calling AbsorbFsyncRequests in mdsync */
#define FSYNCS_PER_ABSORB		10

36 37 38
/* special values for the segno arg to RememberFsyncRequest */
#define FORGET_RELATION_FSYNC	(InvalidBlockNumber)
#define FORGET_DATABASE_FSYNC	(InvalidBlockNumber-1)
B
Bruce Momjian 已提交
39
#define UNLINK_RELATION_REQUEST (InvalidBlockNumber-2)
40

41 42 43 44 45 46 47 48
/*
 * On Windows, we have to interpret EACCES as possibly meaning the same as
 * ENOENT, because if a file is unlinked-but-not-yet-gone on that platform,
 * that's what you get.  Ugh.  This code is designed so that we don't
 * actually believe these cases are okay without further evidence (namely,
 * a pending fsync request getting revoked ... see mdsync).
 */
#ifndef WIN32
B
Bruce Momjian 已提交
49
#define FILE_POSSIBLY_DELETED(err)	((err) == ENOENT)
50
#else
B
Bruce Momjian 已提交
51
#define FILE_POSSIBLY_DELETED(err)	((err) == ENOENT || (err) == EACCES)
52 53
#endif

54
/*
N
Neil Conway 已提交
55 56 57
 *	The magnetic disk storage manager keeps track of open file
 *	descriptors in its own descriptor pool.  This is done to make it
 *	easier to support relations that are larger than the operating
58
 *	system's file size limit (often 2GBytes).  In order to do that,
59 60
 *	we break relations up into "segment" files that are each shorter than
 *	the OS file size limit.  The segment size is set by the RELSEG_SIZE
61
 *	configuration constant in pg_config.h.
62 63 64 65 66 67 68 69 70 71 72
 *
 *	On disk, a relation must consist of consecutively numbered segment
 *	files in the pattern
 *		-- Zero or more full segments of exactly RELSEG_SIZE blocks each
 *		-- Exactly one partial segment of size 0 <= size < RELSEG_SIZE blocks
 *		-- Optionally, any number of inactive segments of size 0 blocks.
 *	The full and partial segments are collectively the "active" segments.
 *	Inactive segments are those that once contained data but are currently
 *	not needed because of an mdtruncate() operation.  The reason for leaving
 *	them present at size zero, rather than unlinking them, is that other
 *	backends and/or the bgwriter might be holding open file references to
B
Bruce Momjian 已提交
73
 *	such segments.	If the relation expands again after mdtruncate(), such
74 75 76 77
 *	that a deactivated segment becomes active again, it is important that
 *	such file references still be valid --- else data might get written
 *	out to an unlinked old copy of a segment file that will eventually
 *	disappear.
78
 *
79
 *	The file descriptor pointer (md_fd field) stored in the SMgrRelation
80 81 82
 *	cache is, therefore, just the head of a list of MdfdVec objects, one
 *	per segment.  But note the md_fd pointer can be NULL, indicating
 *	relation not open.
83
 *
84
 *	Also note that mdfd_chain == NULL does not necessarily mean the relation
85 86 87
 *	doesn't have another segment after this one; we may just not have
 *	opened the next segment yet.  (We could not have "all segments are
 *	in the chain" as an invariant anyway, since another backend could
88 89 90
 *	extend the relation when we weren't looking.)  We do not make chain
 *	entries for inactive segments, however; as soon as we find a partial
 *	segment, we assume that any subsequent segments are inactive.
91
 *
92
 *	All MdfdVec objects are palloc'd in the MdCxt memory context.
93 94
 */

95 96
typedef struct _MdfdVec
{
B
Bruce Momjian 已提交
97 98
	File		mdfd_vfd;		/* fd number in fd.c's pool */
	BlockNumber mdfd_segno;		/* segment number, from 0 */
99
	struct _MdfdVec *mdfd_chain;	/* next segment, or NULL */
100
} MdfdVec;
101

N
Neil Conway 已提交
102
static MemoryContext MdCxt;		/* context for all md.c allocations */
103

104

105 106 107 108 109
/*
 * In some contexts (currently, standalone backends and the bgwriter process)
 * we keep track of pending fsync operations: we need to remember all relation
 * segments that have been written since the last checkpoint, so that we can
 * fsync them down to disk before completing the next checkpoint.  This hash
110 111
 * table remembers the pending operations.	We use a hash table mostly as
 * a convenient way of eliminating duplicate requests.
112
 *
113 114 115 116
 * We use a similar mechanism to remember no-longer-needed files that can
 * be deleted after the next checkpoint, but we use a linked list instead of
 * a hash table, because we don't expect there to be any duplicate requests.
 *
117 118 119 120 121
 * (Regular backends do not track pending operations locally, but forward
 * them to the bgwriter.)
 */
typedef struct
{
B
Bruce Momjian 已提交
122
	RelFileNode rnode;			/* the targeted relation */
123
	ForkNumber	forknum;
B
Bruce Momjian 已提交
124
	BlockNumber segno;			/* which segment */
125
} PendingOperationTag;
126

127 128
typedef uint16 CycleCtr;		/* can be any convenient integer size */

129 130 131
typedef struct
{
	PendingOperationTag tag;	/* hash table key (must be first!) */
132 133
	bool		canceled;		/* T => request canceled, not yet removed */
	CycleCtr	cycle_ctr;		/* mdsync_cycle_ctr when request was made */
134 135
} PendingOperationEntry;

136 137 138
typedef struct
{
	RelFileNode rnode;			/* the dead relation to delete */
B
Bruce Momjian 已提交
139
	CycleCtr	cycle_ctr;		/* mdckpt_cycle_ctr when request was made */
140
} PendingUnlinkEntry;
141

142
static HTAB *pendingOpsTable = NULL;
143
static List *pendingUnlinks = NIL;
144

145
static CycleCtr mdsync_cycle_ctr = 0;
146
static CycleCtr mdckpt_cycle_ctr = 0;
147

148

149 150 151 152 153
typedef enum					/* behavior for mdopen & _mdfd_getseg */
{
	EXTENSION_FAIL,				/* ereport if segment not present */
	EXTENSION_RETURN_NULL,		/* return NULL if not present */
	EXTENSION_CREATE			/* create new segments as needed */
154
} ExtensionBehavior;
155

156
/* local routines */
157 158
static MdfdVec *mdopen(SMgrRelation reln, ForkNumber forknum,
	   ExtensionBehavior behavior);
159
static void register_dirty_segment(SMgrRelation reln, ForkNumber forknum,
160
					   MdfdVec *seg);
161
static void register_unlink(RelFileNode rnode);
162
static MdfdVec *_fdvec_alloc(void);
163
static MdfdVec *_mdfd_openseg(SMgrRelation reln, ForkNumber forkno,
164
			  BlockNumber segno, int oflags);
165 166 167
static MdfdVec *_mdfd_getseg(SMgrRelation reln, ForkNumber forkno,
			 BlockNumber blkno, bool isTemp, ExtensionBehavior behavior);
static BlockNumber _mdnblocks(SMgrRelation reln, ForkNumber forknum,
168
		   MdfdVec *seg);
169

170

171
/*
172
 *	mdinit() -- Initialize private state for magnetic disk storage manager.
173
 */
174
void
175
mdinit(void)
176
{
177 178 179 180 181
	MdCxt = AllocSetContextCreate(TopMemoryContext,
								  "MdSmgr",
								  ALLOCSET_DEFAULT_MINSIZE,
								  ALLOCSET_DEFAULT_INITSIZE,
								  ALLOCSET_DEFAULT_MAXSIZE);
182

183
	/*
B
Bruce Momjian 已提交
184 185 186
	 * Create pending-operations hashtable if we need it.  Currently, we need
	 * it if we are standalone (not under a postmaster) OR if we are a
	 * bootstrap-mode subprocess of a postmaster (that is, a startup or
B
Bruce Momjian 已提交
187
	 * bgwriter process).
188 189 190 191 192 193
	 */
	if (!IsUnderPostmaster || IsBootstrapProcessingMode())
	{
		HASHCTL		hash_ctl;

		MemSet(&hash_ctl, 0, sizeof(hash_ctl));
194
		hash_ctl.keysize = sizeof(PendingOperationTag);
195 196 197 198 199 200
		hash_ctl.entrysize = sizeof(PendingOperationEntry);
		hash_ctl.hash = tag_hash;
		hash_ctl.hcxt = MdCxt;
		pendingOpsTable = hash_create("Pending Ops Table",
									  100L,
									  &hash_ctl,
B
Bruce Momjian 已提交
201
								   HASH_ELEM | HASH_FUNCTION | HASH_CONTEXT);
202
		pendingUnlinks = NIL;
203
	}
204 205
}

206
/*
207
 *	mdexists() -- Does the physical file exist?
208 209 210 211 212 213 214
 *
 * Note: this will return true for lingering files, with pending deletions
 */
bool
mdexists(SMgrRelation reln, ForkNumber forkNum)
{
	/*
215 216
	 * Close it first, to ensure that we notice if the fork has been unlinked
	 * since we opened it.
217 218 219 220 221 222
	 */
	mdclose(reln, forkNum);

	return (mdopen(reln, forkNum, EXTENSION_RETURN_NULL) != NULL);
}

223 224 225 226 227
/*
 *	mdcreate() -- Create a new relation on magnetic disk.
 *
 * If isRedo is true, it's okay for the relation to exist already.
 */
228
void
229
mdcreate(SMgrRelation reln, ForkNumber forkNum, bool isRedo)
230
{
231
	char	   *path;
232
	File		fd;
233

234
	if (isRedo && reln->md_fd[forkNum] != NULL)
235
		return;					/* created and opened already... */
236

237
	Assert(reln->md_fd[forkNum] == NULL);
238

239
	path = relpath(reln->smgr_rnode, forkNum);
240

241
	fd = PathNameOpenFile(path, O_RDWR | O_CREAT | O_EXCL | PG_BINARY, 0600);
242 243 244

	if (fd < 0)
	{
B
Bruce Momjian 已提交
245
		int			save_errno = errno;
246

247
		/*
B
Bruce Momjian 已提交
248 249 250 251
		 * During bootstrap, there are cases where a system relation will be
		 * accessed (by internal backend processes) before the bootstrap
		 * script nominally creates it.  Therefore, allow the file to exist
		 * already, even if isRedo is not set.	(See also mdopen)
252
		 */
253
		if (isRedo || IsBootstrapProcessingMode())
254
			fd = PathNameOpenFile(path, O_RDWR | PG_BINARY, 0600);
255
		if (fd < 0)
256
		{
257
			/* be sure to report the error reported by create, not open */
258
			errno = save_errno;
259 260
			ereport(ERROR,
					(errcode_for_file_access(),
261
					 errmsg("could not create relation %s: %m", path)));
262
		}
263
	}
264 265

	pfree(path);
266

267
	reln->md_fd[forkNum] = _fdvec_alloc();
268

269 270 271
	reln->md_fd[forkNum]->mdfd_vfd = fd;
	reln->md_fd[forkNum]->mdfd_segno = 0;
	reln->md_fd[forkNum]->mdfd_chain = NULL;
272 273 274
}

/*
275
 *	mdunlink() -- Unlink a relation.
276 277 278 279
 *
 * Note that we're passed a RelFileNode --- by the time this is called,
 * there won't be an SMgrRelation hashtable entry anymore.
 *
280 281 282 283 284 285 286
 * Actually, we don't unlink the first segment file of the relation, but
 * just truncate it to zero length, and record a request to unlink it after
 * the next checkpoint.  Additional segments can be unlinked immediately,
 * however.  Leaving the empty file in place prevents that relfilenode
 * number from being reused.  The scenario this protects us from is:
 * 1. We delete a relation (and commit, and actually remove its file).
 * 2. We create a new relation, which by chance gets the same relfilenode as
B
Bruce Momjian 已提交
287
 *	  the just-deleted one (OIDs must've wrapped around for that to happen).
288 289 290 291 292
 * 3. We crash before another checkpoint occurs.
 * During replay, we would delete the file and then recreate it, which is fine
 * if the contents of the file were repopulated by subsequent WAL entries.
 * But if we didn't WAL-log insertions, but instead relied on fsyncing the
 * file after populating it (as for instance CLUSTER and CREATE INDEX do),
B
Bruce Momjian 已提交
293
 * the contents of the file would be lost forever.	By leaving the empty file
294 295 296 297
 * until after the next checkpoint, we prevent reassignment of the relfilenode
 * number until it's safe, because relfilenode assignment skips over any
 * existing file.
 *
298
 * If isRedo is true, it's okay for the relation to be already gone.
299 300 301 302 303
 * Also, we should remove the file immediately instead of queuing a request
 * for later, since during redo there's no possibility of creating a
 * conflicting relation.
 *
 * Note: any failure should be reported as WARNING not ERROR, because
304
 * we are usually not in a transaction anymore when this is called.
305
 */
306
void
307
mdunlink(RelFileNode rnode, ForkNumber forkNum, bool isRedo)
308
{
309
	char	   *path;
B
Bruce Momjian 已提交
310
	int			ret;
311

312
	/*
B
Bruce Momjian 已提交
313 314
	 * We have to clean out any pending fsync requests for the doomed
	 * relation, else the next mdsync() will fail.
315
	 */
316
	ForgetRelationFsyncRequests(rnode, forkNum);
317

318
	path = relpath(rnode, forkNum);
319

320
	/*
321
	 * Delete or truncate the first segment.
322
	 */
323
	if (isRedo || forkNum != MAIN_FORKNUM)
324 325
		ret = unlink(path);
	else
326 327
	{
		/* truncate(2) would be easier here, but Windows hasn't got it */
328
		int			fd;
329 330 331 332

		fd = BasicOpenFile(path, O_RDWR | PG_BINARY, 0);
		if (fd >= 0)
		{
333
			int			save_errno;
334 335 336 337 338 339 340 341 342

			ret = ftruncate(fd, 0);
			save_errno = errno;
			close(fd);
			errno = save_errno;
		}
		else
			ret = -1;
	}
343
	if (ret < 0)
344
	{
345
		if (!isRedo || errno != ENOENT)
346 347
			ereport(WARNING,
					(errcode_for_file_access(),
348
					 errmsg("could not remove relation %s: %m", path)));
349
	}
350

351 352 353
	/*
	 * Delete any additional segments.
	 */
354
	else
355
	{
356
		char	   *segpath = (char *) palloc(strlen(path) + 12);
357
		BlockNumber segno;
358

359
		/*
B
Bruce Momjian 已提交
360 361
		 * Note that because we loop until getting ENOENT, we will correctly
		 * remove all inactive segments as well as active ones.
362
		 */
B
Bruce Momjian 已提交
363
		for (segno = 1;; segno++)
364
		{
365
			sprintf(segpath, "%s.%u", path, segno);
366 367 368 369
			if (unlink(segpath) < 0)
			{
				/* ENOENT is expected after the last segment... */
				if (errno != ENOENT)
370 371
					ereport(WARNING,
							(errcode_for_file_access(),
372 373
					 errmsg("could not remove segment %u of relation %s: %m",
							segno, path)));
374 375 376 377
				break;
			}
		}
		pfree(segpath);
378
	}
379

380
	pfree(path);
381 382

	/* Register request to unlink first segment later */
383
	if (!isRedo && forkNum == MAIN_FORKNUM)
384
		register_unlink(rnode);
385 386 387
}

/*
388
 *	mdextend() -- Add a block to the specified relation.
389
 *
390 391 392 393 394
 *		The semantics are nearly the same as mdwrite(): write at the
 *		specified position.  However, this is to be used for the case of
 *		extending a relation (i.e., blocknum is at or beyond the current
 *		EOF).  Note that we assume writing a block beyond current EOF
 *		causes intervening file space to become filled with zeroes.
395
 */
396
void
397 398
mdextend(SMgrRelation reln, ForkNumber forknum, BlockNumber blocknum,
		 char *buffer, bool isTemp)
399
{
400
	off_t		seekpos;
401
	int			nbytes;
402
	MdfdVec    *v;
403

404 405
	/* This assert is too expensive to have on normally ... */
#ifdef CHECK_WRITE_VS_EXTEND
406
	Assert(blocknum >= mdnblocks(reln, forknum));
407 408 409
#endif

	/*
B
Bruce Momjian 已提交
410 411 412
	 * If a relation manages to grow to 2^32-1 blocks, refuse to extend it any
	 * more --- we mustn't create a block whose number actually is
	 * InvalidBlockNumber.
413 414 415 416
	 */
	if (blocknum == InvalidBlockNumber)
		ereport(ERROR,
				(errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
417 418
				 errmsg("cannot extend relation %s beyond %u blocks",
						relpath(reln->smgr_rnode, forknum),
419 420
						InvalidBlockNumber)));

421
	v = _mdfd_getseg(reln, forknum, blocknum, isTemp, EXTENSION_CREATE);
422

423 424
	seekpos = (off_t) BLCKSZ *(blocknum % ((BlockNumber) RELSEG_SIZE));

425
	Assert(seekpos < (off_t) BLCKSZ * RELSEG_SIZE);
426

427
	/*
428 429
	 * Note: because caller usually obtained blocknum by calling mdnblocks,
	 * which did a seek(SEEK_END), this seek is often redundant and will be
B
Bruce Momjian 已提交
430
	 * optimized away by fd.c.	It's not redundant, however, if there is a
431 432 433 434
	 * partial page at the end of the file. In that case we want to try to
	 * overwrite the partial page with a full page.  It's also not redundant
	 * if bufmgr.c had to dump another buffer of the same file to make room
	 * for the new page's buffer.
435 436
	 */
	if (FileSeek(v->mdfd_vfd, seekpos, SEEK_SET) != seekpos)
437 438
		ereport(ERROR,
				(errcode_for_file_access(),
439
				 errmsg("could not seek to block %u of relation %s: %m",
440
						blocknum,
441
						relpath(reln->smgr_rnode, forknum))));
442 443 444

	if ((nbytes = FileWrite(v->mdfd_vfd, buffer, BLCKSZ)) != BLCKSZ)
	{
445 446 447
		if (nbytes < 0)
			ereport(ERROR,
					(errcode_for_file_access(),
448 449
					 errmsg("could not extend relation %s: %m",
							relpath(reln->smgr_rnode, forknum)),
450 451 452 453
					 errhint("Check free disk space.")));
		/* short write: complain appropriately */
		ereport(ERROR,
				(errcode(ERRCODE_DISK_FULL),
454 455
				 errmsg("could not extend relation %s: wrote only %d of %d bytes at block %u",
						relpath(reln->smgr_rnode, forknum),
456 457
						nbytes, BLCKSZ, blocknum),
				 errhint("Check free disk space.")));
458
	}
459

460
	if (!isTemp)
461
		register_dirty_segment(reln, forknum, v);
462

463
	Assert(_mdnblocks(reln, forknum, v) <= ((BlockNumber) RELSEG_SIZE));
464 465 466
}

/*
467
 *	mdopen() -- Open the specified relation.
468 469
 *
 * Note we only open the first segment, when there are multiple segments.
470 471 472 473 474
 *
 * If first segment is not present, either ereport or return NULL according
 * to "behavior".  We treat EXTENSION_CREATE the same as EXTENSION_FAIL;
 * EXTENSION_CREATE means it's OK to extend an existing relation, not to
 * invent one out of whole cloth.
475
 */
476
static MdfdVec *
477
mdopen(SMgrRelation reln, ForkNumber forknum, ExtensionBehavior behavior)
478
{
B
Bruce Momjian 已提交
479
	MdfdVec    *mdfd;
480
	char	   *path;
481
	File		fd;
482

483
	/* No work if already open */
484 485
	if (reln->md_fd[forknum])
		return reln->md_fd[forknum];
486

487
	path = relpath(reln->smgr_rnode, forknum);
488

489
	fd = PathNameOpenFile(path, O_RDWR | PG_BINARY, 0600);
490

491
	if (fd < 0)
492
	{
493
		/*
B
Bruce Momjian 已提交
494 495 496 497
		 * During bootstrap, there are cases where a system relation will be
		 * accessed (by internal backend processes) before the bootstrap
		 * script nominally creates it.  Therefore, accept mdopen() as a
		 * substitute for mdcreate() in bootstrap mode only. (See mdcreate)
498
		 */
499
		if (IsBootstrapProcessingMode())
500
			fd = PathNameOpenFile(path, O_RDWR | O_CREAT | O_EXCL | PG_BINARY, 0600);
501 502
		if (fd < 0)
		{
503 504
			if (behavior == EXTENSION_RETURN_NULL &&
				FILE_POSSIBLY_DELETED(errno))
505 506
			{
				pfree(path);
507
				return NULL;
508
			}
509 510
			ereport(ERROR,
					(errcode_for_file_access(),
511
					 errmsg("could not open relation %s: %m", path)));
512 513
		}
	}
514 515

	pfree(path);
516

517
	reln->md_fd[forknum] = mdfd = _fdvec_alloc();
V
Vadim B. Mikheev 已提交
518

519 520 521
	mdfd->mdfd_vfd = fd;
	mdfd->mdfd_segno = 0;
	mdfd->mdfd_chain = NULL;
522
	Assert(_mdnblocks(reln, forknum, mdfd) <= ((BlockNumber) RELSEG_SIZE));
523

524
	return mdfd;
525 526 527
}

/*
528
 *	mdclose() -- Close the specified relation, if it isn't closed already.
529
 */
530
void
531
mdclose(SMgrRelation reln, ForkNumber forknum)
532
{
533
	MdfdVec    *v = reln->md_fd[forknum];
534

535 536
	/* No work if already closed */
	if (v == NULL)
537
		return;
538

539
	reln->md_fd[forknum] = NULL;	/* prevent dangling pointer after error */
540

541
	while (v != NULL)
V
Vadim B. Mikheev 已提交
542
	{
543 544
		MdfdVec    *ov = v;

545 546 547 548 549
		/* if not closed already */
		if (v->mdfd_vfd >= 0)
			FileClose(v->mdfd_vfd);
		/* Now free vector */
		v = v->mdfd_chain;
550
		pfree(ov);
551
	}
552 553
}

554 555 556 557 558 559 560 561 562 563 564 565
/*
 *	mdprefetch() -- Initiate asynchronous read of the specified block of a relation
 */
void
mdprefetch(SMgrRelation reln, ForkNumber forknum, BlockNumber blocknum)
{
#ifdef USE_PREFETCH
	off_t		seekpos;
	MdfdVec    *v;

	v = _mdfd_getseg(reln, forknum, blocknum, false, EXTENSION_FAIL);

566 567
	seekpos = (off_t) BLCKSZ *(blocknum % ((BlockNumber) RELSEG_SIZE));

568 569 570
	Assert(seekpos < (off_t) BLCKSZ * RELSEG_SIZE);

	(void) FilePrefetch(v->mdfd_vfd, seekpos, BLCKSZ);
571
#endif   /* USE_PREFETCH */
572 573 574
}


575
/*
576
 *	mdread() -- Read the specified block from a relation.
577
 */
578
void
579 580
mdread(SMgrRelation reln, ForkNumber forknum, BlockNumber blocknum,
	   char *buffer)
581
{
582
	off_t		seekpos;
583 584
	int			nbytes;
	MdfdVec    *v;
585

586 587 588 589
	TRACE_POSTGRESQL_SMGR_MD_READ_START(forknum, blocknum,
										reln->smgr_rnode.spcNode,
										reln->smgr_rnode.dbNode,
										reln->smgr_rnode.relNode);
590

591
	v = _mdfd_getseg(reln, forknum, blocknum, false, EXTENSION_FAIL);
592

593 594
	seekpos = (off_t) BLCKSZ *(blocknum % ((BlockNumber) RELSEG_SIZE));

595
	Assert(seekpos < (off_t) BLCKSZ * RELSEG_SIZE);
596

597
	if (FileSeek(v->mdfd_vfd, seekpos, SEEK_SET) != seekpos)
598 599
		ereport(ERROR,
				(errcode_for_file_access(),
600 601
				 errmsg("could not seek to block %u of relation %s: %m",
						blocknum, relpath(reln->smgr_rnode, forknum))));
602

603 604
	nbytes = FileRead(v->mdfd_vfd, buffer, BLCKSZ);

605 606 607 608 609 610
	TRACE_POSTGRESQL_SMGR_MD_READ_DONE(forknum, blocknum,
									   reln->smgr_rnode.spcNode,
									   reln->smgr_rnode.dbNode,
									   reln->smgr_rnode.relNode,
									   nbytes,
									   BLCKSZ);
611 612

	if (nbytes != BLCKSZ)
613
	{
614 615 616
		if (nbytes < 0)
			ereport(ERROR,
					(errcode_for_file_access(),
617 618
					 errmsg("could not read block %u of relation %s: %m",
							blocknum, relpath(reln->smgr_rnode, forknum))));
B
Bruce Momjian 已提交
619

620
		/*
621 622
		 * Short read: we are at or past EOF, or we read a partial block at
		 * EOF.  Normally this is an error; upper levels should never try to
B
Bruce Momjian 已提交
623 624
		 * read a nonexistent block.  However, if zero_damaged_pages is ON or
		 * we are InRecovery, we should instead return zeroes without
625 626
		 * complaining.  This allows, for example, the case of trying to
		 * update a block that was later truncated away.
627
		 */
628
		if (zero_damaged_pages || InRecovery)
629
			MemSet(buffer, 0, BLCKSZ);
630
		else
631 632
			ereport(ERROR,
					(errcode(ERRCODE_DATA_CORRUPTED),
633 634
					 errmsg("could not read block %u of relation %s: read only %d of %d bytes",
							blocknum, relpath(reln->smgr_rnode, forknum),
635
							nbytes, BLCKSZ)));
636 637 638 639
	}
}

/*
640
 *	mdwrite() -- Write the supplied block at the appropriate location.
641 642 643 644
 *
 *		This is to be used only for updating already-existing blocks of a
 *		relation (ie, those before the current EOF).  To extend a relation,
 *		use mdextend().
645
 */
646
void
647 648
mdwrite(SMgrRelation reln, ForkNumber forknum, BlockNumber blocknum,
		char *buffer, bool isTemp)
649
{
650
	off_t		seekpos;
651
	int			nbytes;
652
	MdfdVec    *v;
653

654 655
	/* This assert is too expensive to have on normally ... */
#ifdef CHECK_WRITE_VS_EXTEND
656
	Assert(blocknum < mdnblocks(reln, forknum));
657 658
#endif

659 660 661 662
	TRACE_POSTGRESQL_SMGR_MD_WRITE_START(forknum, blocknum,
										 reln->smgr_rnode.spcNode,
										 reln->smgr_rnode.dbNode,
										 reln->smgr_rnode.relNode);
663

664
	v = _mdfd_getseg(reln, forknum, blocknum, isTemp, EXTENSION_FAIL);
665

666 667
	seekpos = (off_t) BLCKSZ *(blocknum % ((BlockNumber) RELSEG_SIZE));

668
	Assert(seekpos < (off_t) BLCKSZ * RELSEG_SIZE);
669

670
	if (FileSeek(v->mdfd_vfd, seekpos, SEEK_SET) != seekpos)
671 672
		ereport(ERROR,
				(errcode_for_file_access(),
673 674
				 errmsg("could not seek to block %u of relation %s: %m",
						blocknum, relpath(reln->smgr_rnode, forknum))));
675

676 677
	nbytes = FileWrite(v->mdfd_vfd, buffer, BLCKSZ);

678 679 680 681 682 683
	TRACE_POSTGRESQL_SMGR_MD_WRITE_DONE(forknum, blocknum,
										reln->smgr_rnode.spcNode,
										reln->smgr_rnode.dbNode,
										reln->smgr_rnode.relNode,
										nbytes,
										BLCKSZ);
684 685

	if (nbytes != BLCKSZ)
686
	{
687 688 689
		if (nbytes < 0)
			ereport(ERROR,
					(errcode_for_file_access(),
690 691
					 errmsg("could not write block %u of relation %s: %m",
							blocknum, relpath(reln->smgr_rnode, forknum))));
692 693 694
		/* short write: complain appropriately */
		ereport(ERROR,
				(errcode(ERRCODE_DISK_FULL),
695
				 errmsg("could not write block %u of relation %s: wrote only %d of %d bytes",
696
						blocknum,
697
						relpath(reln->smgr_rnode, forknum),
698 699
						nbytes, BLCKSZ),
				 errhint("Check free disk space.")));
700
	}
701

702
	if (!isTemp)
703
		register_dirty_segment(reln, forknum, v);
704
}
705 706

/*
707
 *	mdnblocks() -- Get the number of blocks stored in a relation.
708
 *
709
 *		Important side effect: all active segments of the relation are opened
710 711
 *		and added to the mdfd_chain list.  If this routine has not been
 *		called, then only segments up to the last one actually touched
712
 *		are present in the chain.
713
 */
714
BlockNumber
715
mdnblocks(SMgrRelation reln, ForkNumber forknum)
716
{
717
	MdfdVec    *v = mdopen(reln, forknum, EXTENSION_FAIL);
718
	BlockNumber nblocks;
719
	BlockNumber segno = 0;
720 721

	/*
B
Bruce Momjian 已提交
722 723 724
	 * Skip through any segments that aren't the last one, to avoid redundant
	 * seeks on them.  We have previously verified that these segments are
	 * exactly RELSEG_SIZE long, and it's useless to recheck that each time.
725 726
	 *
	 * NOTE: this assumption could only be wrong if another backend has
B
Bruce Momjian 已提交
727
	 * truncated the relation.	We rely on higher code levels to handle that
728
	 * scenario by closing and re-opening the md fd, which is handled via
B
Bruce Momjian 已提交
729
	 * relcache flush.	(Since the bgwriter doesn't participate in relcache
730 731
	 * flush, it could have segment chain entries for inactive segments;
	 * that's OK because the bgwriter never needs to compute relation size.)
732
	 */
733
	while (v->mdfd_chain != NULL)
734 735 736 737 738
	{
		segno++;
		v = v->mdfd_chain;
	}

739 740
	for (;;)
	{
741
		nblocks = _mdnblocks(reln, forknum, v);
742
		if (nblocks > ((BlockNumber) RELSEG_SIZE))
743
			elog(FATAL, "segment too big");
744 745
		if (nblocks < ((BlockNumber) RELSEG_SIZE))
			return (segno * ((BlockNumber) RELSEG_SIZE)) + nblocks;
746

747 748 749 750
		/*
		 * If segment is exactly RELSEG_SIZE, advance to next one.
		 */
		segno++;
751

752
		if (v->mdfd_chain == NULL)
753 754
		{
			/*
B
Bruce Momjian 已提交
755 756
			 * Because we pass O_CREAT, we will create the next segment (with
			 * zero length) immediately, if the last segment is of length
757 758
			 * RELSEG_SIZE.  While perhaps not strictly necessary, this keeps
			 * the logic simple.
759
			 */
760
			v->mdfd_chain = _mdfd_openseg(reln, forknum, segno, O_CREAT);
761
			if (v->mdfd_chain == NULL)
762 763
				ereport(ERROR,
						(errcode_for_file_access(),
764 765 766
					   errmsg("could not open segment %u of relation %s: %m",
							  segno,
							  relpath(reln->smgr_rnode, forknum))));
767
		}
768 769

		v = v->mdfd_chain;
770 771 772
	}
}

773
/*
774
 *	mdtruncate() -- Truncate relation to specified number of blocks.
775
 */
776
void
777 778
mdtruncate(SMgrRelation reln, ForkNumber forknum, BlockNumber nblocks,
		   bool isTemp)
779
{
780
	MdfdVec    *v;
781 782
	BlockNumber curnblk;
	BlockNumber priorblocks;
783

784
	/*
B
Bruce Momjian 已提交
785 786
	 * NOTE: mdnblocks makes sure we have opened all active segments, so that
	 * truncation loop will get them all!
787
	 */
788
	curnblk = mdnblocks(reln, forknum);
789
	if (nblocks > curnblk)
790 791 792 793 794
	{
		/* Bogus request ... but no complaint if InRecovery */
		if (InRecovery)
			return;
		ereport(ERROR,
795 796
				(errmsg("could not truncate relation %s to %u blocks: it's only %u blocks now",
						relpath(reln->smgr_rnode, forknum),
797 798
						nblocks, curnblk)));
	}
799
	if (nblocks == curnblk)
800
		return;					/* no work */
801

802
	v = mdopen(reln, forknum, EXTENSION_FAIL);
803

804
	priorblocks = 0;
805
	while (v != NULL)
806
	{
807 808 809
		MdfdVec    *ov = v;

		if (priorblocks > nblocks)
810
		{
811
			/*
B
Bruce Momjian 已提交
812 813 814
			 * This segment is no longer active (and has already been unlinked
			 * from the mdfd_chain). We truncate the file, but do not delete
			 * it, for reasons explained in the header comments.
815
			 */
816
			if (FileTruncate(v->mdfd_vfd, 0) < 0)
817 818
				ereport(ERROR,
						(errcode_for_file_access(),
819 820 821
					errmsg("could not truncate relation %s to %u blocks: %m",
						   relpath(reln->smgr_rnode, forknum),
						   nblocks)));
822
			if (!isTemp)
823
				register_dirty_segment(reln, forknum, v);
824
			v = v->mdfd_chain;
825 826
			Assert(ov != reln->md_fd[forknum]); /* we never drop the 1st
												 * segment */
827
			pfree(ov);
828
		}
829
		else if (priorblocks + ((BlockNumber) RELSEG_SIZE) > nblocks)
830
		{
831
			/*
B
Bruce Momjian 已提交
832 833 834 835
			 * This is the last segment we want to keep. Truncate the file to
			 * the right length, and clear chain link that points to any
			 * remaining segments (which we shall zap). NOTE: if nblocks is
			 * exactly a multiple K of RELSEG_SIZE, we will truncate the K+1st
836 837
			 * segment to 0 length but keep it. This adheres to the invariant
			 * given in the header comments.
838
			 */
839
			BlockNumber lastsegblocks = nblocks - priorblocks;
840

841
			if (FileTruncate(v->mdfd_vfd, (off_t) lastsegblocks * BLCKSZ) < 0)
842 843
				ereport(ERROR,
						(errcode_for_file_access(),
844 845 846
					errmsg("could not truncate relation %s to %u blocks: %m",
						   relpath(reln->smgr_rnode, forknum),
						   nblocks)));
847
			if (!isTemp)
848
				register_dirty_segment(reln, forknum, v);
849
			v = v->mdfd_chain;
850
			ov->mdfd_chain = NULL;
851 852 853
		}
		else
		{
854
			/*
B
Bruce Momjian 已提交
855 856
			 * We still need this segment and 0 or more blocks beyond it, so
			 * nothing to do here.
857 858 859 860
			 */
			v = v->mdfd_chain;
		}
		priorblocks += RELSEG_SIZE;
861
	}
862
}
863

864 865
/*
 *	mdimmedsync() -- Immediately sync a relation to stable storage.
866 867 868
 *
 * Note that only writes already issued are synced; this routine knows
 * nothing of dirty buffers that may exist inside the buffer manager.
869
 */
870
void
871
mdimmedsync(SMgrRelation reln, ForkNumber forknum)
872 873 874 875 876
{
	MdfdVec    *v;
	BlockNumber curnblk;

	/*
B
Bruce Momjian 已提交
877 878
	 * NOTE: mdnblocks makes sure we have opened all active segments, so that
	 * fsync loop will get them all!
879
	 */
880
	curnblk = mdnblocks(reln, forknum);
881

882
	v = mdopen(reln, forknum, EXTENSION_FAIL);
883 884 885 886

	while (v != NULL)
	{
		if (FileSync(v->mdfd_vfd) < 0)
887 888
			ereport(ERROR,
					(errcode_for_file_access(),
889 890 891
					 errmsg("could not fsync segment %u of relation %s: %m",
							v->mdfd_segno,
							relpath(reln->smgr_rnode, forknum))));
892 893 894 895
		v = v->mdfd_chain;
	}
}

896
/*
897
 *	mdsync() -- Sync previous writes to stable storage.
898
 */
899
void
900
mdsync(void)
901
{
902 903 904 905 906
	static bool mdsync_in_progress = false;

	HASH_SEQ_STATUS hstat;
	PendingOperationEntry *entry;
	int			absorb_counter;
907

908 909 910 911
	/*
	 * This is only called during checkpoints, and checkpoints should only
	 * occur in processes that have created a pendingOpsTable.
	 */
912
	if (!pendingOpsTable)
913
		elog(ERROR, "cannot sync without a pendingOpsTable");
914

915
	/*
916
	 * If we are in the bgwriter, the sync had better include all fsync
B
Bruce Momjian 已提交
917
	 * requests that were queued by backends up to this point.	The tightest
918
	 * race condition that could occur is that a buffer that must be written
B
Bruce Momjian 已提交
919 920 921 922
	 * and fsync'd for the checkpoint could have been dumped by a backend just
	 * before it was visited by BufferSync().  We know the backend will have
	 * queued an fsync request before clearing the buffer's dirtybit, so we
	 * are safe as long as we do an Absorb after completing BufferSync().
923
	 */
924 925 926 927 928 929 930 931 932 933
	AbsorbFsyncRequests();

	/*
	 * To avoid excess fsync'ing (in the worst case, maybe a never-terminating
	 * checkpoint), we want to ignore fsync requests that are entered into the
	 * hashtable after this point --- they should be processed next time,
	 * instead.  We use mdsync_cycle_ctr to tell old entries apart from new
	 * ones: new ones will have cycle_ctr equal to the incremented value of
	 * mdsync_cycle_ctr.
	 *
B
Bruce Momjian 已提交
934 935
	 * In normal circumstances, all entries present in the table at this point
	 * will have cycle_ctr exactly equal to the current (about to be old)
936 937 938 939 940
	 * value of mdsync_cycle_ctr.  However, if we fail partway through the
	 * fsync'ing loop, then older values of cycle_ctr might remain when we
	 * come back here to try again.  Repeated checkpoint failures would
	 * eventually wrap the counter around to the point where an old entry
	 * might appear new, causing us to skip it, possibly allowing a checkpoint
B
Bruce Momjian 已提交
941 942
	 * to succeed that should not have.  To forestall wraparound, any time the
	 * previous mdsync() failed to complete, run through the table and
943 944 945 946
	 * forcibly set cycle_ctr = mdsync_cycle_ctr.
	 *
	 * Think not to merge this loop with the main loop, as the problem is
	 * exactly that that loop may fail before having visited all the entries.
B
Bruce Momjian 已提交
947 948
	 * From a performance point of view it doesn't matter anyway, as this path
	 * will never be taken in a system that's functioning normally.
949 950 951 952 953 954 955 956 957 958 959 960 961
	 */
	if (mdsync_in_progress)
	{
		/* prior try failed, so update any stale cycle_ctr values */
		hash_seq_init(&hstat, pendingOpsTable);
		while ((entry = (PendingOperationEntry *) hash_seq_search(&hstat)) != NULL)
		{
			entry->cycle_ctr = mdsync_cycle_ctr;
		}
	}

	/* Advance counter so that new hashtable entries are distinguishable */
	mdsync_cycle_ctr++;
962

963 964
	/* Set flag to detect failure if we don't reach the end of the loop */
	mdsync_in_progress = true;
965

966 967 968 969 970
	/* Now scan the hashtable for fsync requests to process */
	absorb_counter = FSYNCS_PER_ABSORB;
	hash_seq_init(&hstat, pendingOpsTable);
	while ((entry = (PendingOperationEntry *) hash_seq_search(&hstat)) != NULL)
	{
971
		/*
972 973
		 * If the entry is new then don't process it this time.  Note that
		 * "continue" bypasses the hash-remove call at the bottom of the loop.
974
		 */
975 976
		if (entry->cycle_ctr == mdsync_cycle_ctr)
			continue;
977

978 979 980 981
		/* Else assert we haven't missed it */
		Assert((CycleCtr) (entry->cycle_ctr + 1) == mdsync_cycle_ctr);

		/*
B
Bruce Momjian 已提交
982 983 984 985
		 * If fsync is off then we don't have to bother opening the file at
		 * all.  (We delay checking until this point so that changing fsync on
		 * the fly behaves sensibly.)  Also, if the entry is marked canceled,
		 * fall through to delete it.
986 987
		 */
		if (enableFsync && !entry->canceled)
988
		{
989 990
			int			failures;

991
			/*
992 993 994 995 996
			 * If in bgwriter, we want to absorb pending requests every so
			 * often to prevent overflow of the fsync request queue.  It is
			 * unspecified whether newly-added entries will be visited by
			 * hash_seq_search, but we don't care since we don't need to
			 * process them anyway.
997
			 */
998 999 1000 1001 1002 1003 1004 1005
			if (--absorb_counter <= 0)
			{
				AbsorbFsyncRequests();
				absorb_counter = FSYNCS_PER_ABSORB;
			}

			/*
			 * The fsync table could contain requests to fsync segments that
B
Bruce Momjian 已提交
1006 1007 1008 1009 1010 1011
			 * have been deleted (unlinked) by the time we get to them. Rather
			 * than just hoping an ENOENT (or EACCES on Windows) error can be
			 * ignored, what we do on error is absorb pending requests and
			 * then retry.	Since mdunlink() queues a "revoke" message before
			 * actually unlinking, the fsync request is guaranteed to be
			 * marked canceled after the absorb if it really was this case.
1012 1013 1014
			 * DROP DATABASE likewise has to tell us to forget fsync requests
			 * before it starts deletions.
			 */
B
Bruce Momjian 已提交
1015
			for (failures = 0;; failures++)		/* loop exits at "break" */
1016
			{
1017 1018
				SMgrRelation reln;
				MdfdVec    *seg;
1019
				char	   *path;
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040

				/*
				 * Find or create an smgr hash entry for this relation. This
				 * may seem a bit unclean -- md calling smgr?  But it's really
				 * the best solution.  It ensures that the open file reference
				 * isn't permanently leaked if we get an error here. (You may
				 * say "but an unreferenced SMgrRelation is still a leak!" Not
				 * really, because the only case in which a checkpoint is done
				 * by a process that isn't about to shut down is in the
				 * bgwriter, and it will periodically do smgrcloseall(). This
				 * fact justifies our not closing the reln in the success path
				 * either, which is a good thing since in non-bgwriter cases
				 * we couldn't safely do that.)  Furthermore, in many cases
				 * the relation will have been dirtied through this same smgr
				 * relation, and so we can save a file open/close cycle.
				 */
				reln = smgropen(entry->tag.rnode);

				/*
				 * It is possible that the relation has been dropped or
				 * truncated since the fsync request was entered.  Therefore,
B
Bruce Momjian 已提交
1041 1042 1043 1044
				 * allow ENOENT, but only if we didn't fail already on this
				 * file.  This applies both during _mdfd_getseg() and during
				 * FileSync, since fd.c might have closed the file behind our
				 * back.
1045
				 */
1046
				seg = _mdfd_getseg(reln, entry->tag.forknum,
B
Bruce Momjian 已提交
1047
							  entry->tag.segno * ((BlockNumber) RELSEG_SIZE),
1048
								   false, EXTENSION_RETURN_NULL);
1049 1050 1051 1052 1053 1054
				if (seg != NULL &&
					FileSync(seg->mdfd_vfd) >= 0)
					break;		/* success; break out of retry loop */

				/*
				 * XXX is there any point in allowing more than one retry?
B
Bruce Momjian 已提交
1055 1056
				 * Don't see one at the moment, but easy to change the test
				 * here if so.
1057
				 */
1058
				path = relpath(entry->tag.rnode, entry->tag.forknum);
1059 1060 1061 1062
				if (!FILE_POSSIBLY_DELETED(errno) ||
					failures > 0)
					ereport(ERROR,
							(errcode_for_file_access(),
1063 1064
					  errmsg("could not fsync segment %u of relation %s: %m",
							 entry->tag.segno, path)));
1065 1066 1067
				else
					ereport(DEBUG1,
							(errcode_for_file_access(),
1068 1069 1070
							 errmsg("could not fsync segment %u of relation %s but retrying: %m",
									entry->tag.segno, path)));
				pfree(path);
1071

1072 1073 1074 1075
				/*
				 * Absorb incoming requests and check to see if canceled.
				 */
				AbsorbFsyncRequests();
B
Bruce Momjian 已提交
1076
				absorb_counter = FSYNCS_PER_ABSORB;		/* might as well... */
1077 1078 1079

				if (entry->canceled)
					break;
B
Bruce Momjian 已提交
1080
			}					/* end retry loop */
1081
		}
1082 1083

		/*
B
Bruce Momjian 已提交
1084 1085 1086
		 * If we get here, either we fsync'd successfully, or we don't have to
		 * because enableFsync is off, or the entry is (now) marked canceled.
		 * Okay to delete it.
1087 1088 1089 1090
		 */
		if (hash_search(pendingOpsTable, &entry->tag,
						HASH_REMOVE, NULL) == NULL)
			elog(ERROR, "pendingOpsTable corrupted");
B
Bruce Momjian 已提交
1091
	}							/* end loop over hashtable entries */
1092 1093 1094

	/* Flag successful completion of mdsync */
	mdsync_in_progress = false;
1095 1096
}

1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
/*
 * mdpreckpt() -- Do pre-checkpoint work
 *
 * To distinguish unlink requests that arrived before this checkpoint
 * started from those that arrived during the checkpoint, we use a cycle
 * counter similar to the one we use for fsync requests. That cycle
 * counter is incremented here.
 *
 * This must be called *before* the checkpoint REDO point is determined.
 * That ensures that we won't delete files too soon.
 *
 * Note that we can't do anything here that depends on the assumption
 * that the checkpoint will be completed.
 */
void
mdpreckpt(void)
{
B
Bruce Momjian 已提交
1114
	ListCell   *cell;
1115 1116

	/*
B
Bruce Momjian 已提交
1117 1118 1119 1120
	 * In case the prior checkpoint wasn't completed, stamp all entries in the
	 * list with the current cycle counter.  Anything that's in the list at
	 * the start of checkpoint can surely be deleted after the checkpoint is
	 * finished, regardless of when the request was made.
1121 1122 1123 1124 1125 1126 1127 1128 1129
	 */
	foreach(cell, pendingUnlinks)
	{
		PendingUnlinkEntry *entry = (PendingUnlinkEntry *) lfirst(cell);

		entry->cycle_ctr = mdckpt_cycle_ctr;
	}

	/*
B
Bruce Momjian 已提交
1130 1131
	 * Any unlink requests arriving after this point will be assigned the next
	 * cycle counter, and won't be unlinked until next checkpoint.
1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146
	 */
	mdckpt_cycle_ctr++;
}

/*
 * mdpostckpt() -- Do post-checkpoint work
 *
 * Remove any lingering files that can now be safely removed.
 */
void
mdpostckpt(void)
{
	while (pendingUnlinks != NIL)
	{
		PendingUnlinkEntry *entry = (PendingUnlinkEntry *) linitial(pendingUnlinks);
B
Bruce Momjian 已提交
1147
		char	   *path;
1148 1149

		/*
B
Bruce Momjian 已提交
1150 1151
		 * New entries are appended to the end, so if the entry is new we've
		 * reached the end of old entries.
1152
		 */
1153
		if (entry->cycle_ctr == mdckpt_cycle_ctr)
1154 1155 1156 1157 1158 1159
			break;

		/* Else assert we haven't missed it */
		Assert((CycleCtr) (entry->cycle_ctr + 1) == mdckpt_cycle_ctr);

		/* Unlink the file */
1160
		path = relpath(entry->rnode, MAIN_FORKNUM);
1161 1162 1163
		if (unlink(path) < 0)
		{
			/*
1164 1165 1166 1167 1168
			 * There's a race condition, when the database is dropped at the
			 * same time that we process the pending unlink requests. If the
			 * DROP DATABASE deletes the file before we do, we will get ENOENT
			 * here. rmtree() also has to ignore ENOENT errors, to deal with
			 * the possibility that we delete the file first.
1169 1170 1171 1172
			 */
			if (errno != ENOENT)
				ereport(WARNING,
						(errcode_for_file_access(),
1173
						 errmsg("could not remove relation %s: %m", path)));
1174 1175 1176 1177 1178 1179 1180 1181
		}
		pfree(path);

		pendingUnlinks = list_delete_first(pendingUnlinks);
		pfree(entry);
	}
}

1182
/*
1183 1184 1185 1186 1187 1188 1189
 * register_dirty_segment() -- Mark a relation segment as needing fsync
 *
 * If there is a local pending-ops table, just make an entry in it for
 * mdsync to process later.  Otherwise, try to pass off the fsync request
 * to the background writer process.  If that fails, just do the fsync
 * locally before returning (we expect this will not happen often enough
 * to be a performance problem).
1190
 */
1191
static void
1192
register_dirty_segment(SMgrRelation reln, ForkNumber forknum, MdfdVec *seg)
1193
{
1194 1195
	if (pendingOpsTable)
	{
1196
		/* push it into local pending-ops table */
1197
		RememberFsyncRequest(reln->smgr_rnode, forknum, seg->mdfd_segno);
1198 1199 1200
	}
	else
	{
1201
		if (ForwardFsyncRequest(reln->smgr_rnode, forknum, seg->mdfd_segno))
1202
			return;				/* passed it off successfully */
1203

1204 1205 1206
		if (FileSync(seg->mdfd_vfd) < 0)
			ereport(ERROR,
					(errcode_for_file_access(),
1207 1208 1209
					 errmsg("could not fsync segment %u of relation %s: %m",
							seg->mdfd_segno,
							relpath(reln->smgr_rnode, forknum))));
1210
	}
1211 1212
}

1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
/*
 * register_unlink() -- Schedule a file to be deleted after next checkpoint
 *
 * As with register_dirty_segment, this could involve either a local or
 * a remote pending-ops table.
 */
static void
register_unlink(RelFileNode rnode)
{
	if (pendingOpsTable)
	{
		/* push it into local pending-ops table */
1225
		RememberFsyncRequest(rnode, MAIN_FORKNUM, UNLINK_RELATION_REQUEST);
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236
	}
	else
	{
		/*
		 * Notify the bgwriter about it.  If we fail to queue the request
		 * message, we have to sleep and try again, because we can't simply
		 * delete the file now.  Ugly, but hopefully won't happen often.
		 *
		 * XXX should we just leave the file orphaned instead?
		 */
		Assert(IsUnderPostmaster);
1237 1238
		while (!ForwardFsyncRequest(rnode, MAIN_FORKNUM,
									UNLINK_RELATION_REQUEST))
1239 1240 1241 1242
			pg_usleep(10000L);	/* 10 msec seems a good number */
	}
}

V
WAL  
Vadim B. Mikheev 已提交
1243
/*
1244 1245
 * RememberFsyncRequest() -- callback from bgwriter side of fsync request
 *
1246 1247 1248
 * We stuff most fsync requests into the local hash table for execution
 * during the bgwriter's next checkpoint.  UNLINK requests go into a
 * separate linked list, however, because they get processed separately.
1249
 *
1250 1251
 * The range of possible segment numbers is way less than the range of
 * BlockNumber, so we can reserve high values of segno for special purposes.
1252 1253 1254 1255
 * We define three:
 * - FORGET_RELATION_FSYNC means to cancel pending fsyncs for a relation
 * - FORGET_DATABASE_FSYNC means to cancel pending fsyncs for a whole database
 * - UNLINK_RELATION_REQUEST is a request to delete the file after the next
B
Bruce Momjian 已提交
1256
 *	 checkpoint.
1257 1258 1259 1260
 *
 * (Handling the FORGET_* requests is a tad slow because the hash table has
 * to be searched linearly, but it doesn't seem worth rethinking the table
 * structure for them.)
V
WAL  
Vadim B. Mikheev 已提交
1261
 */
1262
void
1263
RememberFsyncRequest(RelFileNode rnode, ForkNumber forknum, BlockNumber segno)
V
WAL  
Vadim B. Mikheev 已提交
1264
{
1265 1266
	Assert(pendingOpsTable);

1267 1268 1269 1270 1271 1272 1273 1274 1275
	if (segno == FORGET_RELATION_FSYNC)
	{
		/* Remove any pending requests for the entire relation */
		HASH_SEQ_STATUS hstat;
		PendingOperationEntry *entry;

		hash_seq_init(&hstat, pendingOpsTable);
		while ((entry = (PendingOperationEntry *) hash_seq_search(&hstat)) != NULL)
		{
1276
			if (RelFileNodeEquals(entry->tag.rnode, rnode) &&
1277
				entry->tag.forknum == forknum)
1278
			{
1279 1280
				/* Okay, cancel this entry */
				entry->canceled = true;
1281 1282 1283 1284 1285 1286 1287 1288
			}
		}
	}
	else if (segno == FORGET_DATABASE_FSYNC)
	{
		/* Remove any pending requests for the entire database */
		HASH_SEQ_STATUS hstat;
		PendingOperationEntry *entry;
1289
		ListCell   *cell,
1290 1291
				   *prev,
				   *next;
1292

1293
		/* Remove fsync requests */
1294 1295 1296 1297 1298
		hash_seq_init(&hstat, pendingOpsTable);
		while ((entry = (PendingOperationEntry *) hash_seq_search(&hstat)) != NULL)
		{
			if (entry->tag.rnode.dbNode == rnode.dbNode)
			{
1299 1300
				/* Okay, cancel this entry */
				entry->canceled = true;
1301 1302
			}
		}
1303

1304 1305 1306 1307 1308 1309 1310
		/* Remove unlink requests */
		prev = NULL;
		for (cell = list_head(pendingUnlinks); cell; cell = next)
		{
			PendingUnlinkEntry *entry = (PendingUnlinkEntry *) lfirst(cell);

			next = lnext(cell);
1311
			if (entry->rnode.dbNode == rnode.dbNode)
1312 1313 1314 1315 1316 1317 1318
			{
				pendingUnlinks = list_delete_cell(pendingUnlinks, cell, prev);
				pfree(entry);
			}
			else
				prev = cell;
		}
1319
	}
1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333
	else if (segno == UNLINK_RELATION_REQUEST)
	{
		/* Unlink request: put it in the linked list */
		MemoryContext oldcxt = MemoryContextSwitchTo(MdCxt);
		PendingUnlinkEntry *entry;

		entry = palloc(sizeof(PendingUnlinkEntry));
		entry->rnode = rnode;
		entry->cycle_ctr = mdckpt_cycle_ctr;

		pendingUnlinks = lappend(pendingUnlinks, entry);

		MemoryContextSwitchTo(oldcxt);
	}
1334
	else
1335
	{
1336
		/* Normal case: enter a request to fsync this segment */
1337 1338 1339 1340 1341 1342 1343
		PendingOperationTag key;
		PendingOperationEntry *entry;
		bool		found;

		/* ensure any pad bytes in the hash key are zeroed */
		MemSet(&key, 0, sizeof(key));
		key.rnode = rnode;
1344
		key.forknum = forknum;
1345 1346 1347 1348 1349 1350
		key.segno = segno;

		entry = (PendingOperationEntry *) hash_search(pendingOpsTable,
													  &key,
													  HASH_ENTER,
													  &found);
1351 1352 1353 1354 1355 1356
		/* if new or previously canceled entry, initialize it */
		if (!found || entry->canceled)
		{
			entry->canceled = false;
			entry->cycle_ctr = mdsync_cycle_ctr;
		}
B
Bruce Momjian 已提交
1357

1358 1359
		/*
		 * NB: it's intentional that we don't change cycle_ctr if the entry
B
Bruce Momjian 已提交
1360
		 * already exists.	The fsync request must be treated as old, even
1361 1362 1363 1364 1365 1366 1367
		 * though the new request will be satisfied too by any subsequent
		 * fsync.
		 *
		 * However, if the entry is present but is marked canceled, we should
		 * act just as though it wasn't there.  The only case where this could
		 * happen would be if a file had been deleted, we received but did not
		 * yet act on the cancel request, and the same relfilenode was then
B
Bruce Momjian 已提交
1368 1369
		 * assigned to a new file.	We mustn't lose the new request, but it
		 * should be considered new not old.
1370
		 */
1371
	}
1372 1373 1374
}

/*
1375
 * ForgetRelationFsyncRequests -- forget any fsyncs for a rel
1376 1377
 */
void
1378
ForgetRelationFsyncRequests(RelFileNode rnode, ForkNumber forknum)
1379 1380 1381 1382
{
	if (pendingOpsTable)
	{
		/* standalone backend or startup process: fsync state is local */
1383
		RememberFsyncRequest(rnode, forknum, FORGET_RELATION_FSYNC);
1384 1385
	}
	else if (IsUnderPostmaster)
1386 1387
	{
		/*
1388 1389 1390 1391
		 * Notify the bgwriter about it.  If we fail to queue the revoke
		 * message, we have to sleep and try again ... ugly, but hopefully
		 * won't happen often.
		 *
B
Bruce Momjian 已提交
1392 1393 1394 1395
		 * XXX should we CHECK_FOR_INTERRUPTS in this loop?  Escaping with an
		 * error would leave the no-longer-used file still present on disk,
		 * which would be bad, so I'm inclined to assume that the bgwriter
		 * will always empty the queue soon.
1396
		 */
1397
		while (!ForwardFsyncRequest(rnode, forknum, FORGET_RELATION_FSYNC))
1398
			pg_usleep(10000L);	/* 10 msec seems a good number */
B
Bruce Momjian 已提交
1399

1400
		/*
B
Bruce Momjian 已提交
1401 1402
		 * Note we don't wait for the bgwriter to actually absorb the revoke
		 * message; see mdsync() for the implications.
1403 1404 1405
		 */
	}
}
1406

1407
/*
1408
 * ForgetDatabaseFsyncRequests -- forget any fsyncs and unlinks for a DB
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421
 */
void
ForgetDatabaseFsyncRequests(Oid dbid)
{
	RelFileNode rnode;

	rnode.dbNode = dbid;
	rnode.spcNode = 0;
	rnode.relNode = 0;

	if (pendingOpsTable)
	{
		/* standalone backend or startup process: fsync state is local */
1422
		RememberFsyncRequest(rnode, InvalidForkNumber, FORGET_DATABASE_FSYNC);
1423 1424 1425 1426
	}
	else if (IsUnderPostmaster)
	{
		/* see notes in ForgetRelationFsyncRequests */
1427 1428
		while (!ForwardFsyncRequest(rnode, InvalidForkNumber,
									FORGET_DATABASE_FSYNC))
1429
			pg_usleep(10000L);	/* 10 msec seems a good number */
1430
	}
V
WAL  
Vadim B. Mikheev 已提交
1431 1432
}

1433

1434
/*
1435
 *	_fdvec_alloc() -- Make a MdfdVec object.
1436
 */
1437
static MdfdVec *
1438
_fdvec_alloc(void)
1439
{
1440
	return (MdfdVec *) MemoryContextAlloc(MdCxt, sizeof(MdfdVec));
V
Vadim B. Mikheev 已提交
1441 1442 1443
}

/*
1444 1445
 * Open the specified segment of the relation,
 * and make a MdfdVec object for it.  Returns NULL on failure.
V
Vadim B. Mikheev 已提交
1446
 */
1447
static MdfdVec *
1448 1449
_mdfd_openseg(SMgrRelation reln, ForkNumber forknum, BlockNumber segno,
			  int oflags)
1450
{
1451 1452 1453 1454
	MdfdVec    *v;
	int			fd;
	char	   *path,
			   *fullpath;
1455

1456
	path = relpath(reln->smgr_rnode, forknum);
1457 1458 1459

	if (segno > 0)
	{
1460
		/* be sure we have enough space for the '.segno' */
1461
		fullpath = (char *) palloc(strlen(path) + 12);
1462
		sprintf(fullpath, "%s.%u", path, segno);
1463
		pfree(path);
1464 1465 1466 1467 1468
	}
	else
		fullpath = path;

	/* open the file */
1469
	fd = PathNameOpenFile(fullpath, O_RDWR | PG_BINARY | oflags, 0600);
1470

1471
	pfree(fullpath);
1472 1473

	if (fd < 0)
1474
		return NULL;
1475 1476

	/* allocate an mdfdvec entry for it */
1477
	v = _fdvec_alloc();
1478 1479 1480

	/* fill the entry */
	v->mdfd_vfd = fd;
1481
	v->mdfd_segno = segno;
1482
	v->mdfd_chain = NULL;
1483
	Assert(_mdnblocks(reln, forknum, v) <= ((BlockNumber) RELSEG_SIZE));
1484

1485
	/* all done */
1486
	return v;
1487
}
1488

N
Neil Conway 已提交
1489 1490
/*
 *	_mdfd_getseg() -- Find the segment of the relation holding the
1491 1492 1493 1494 1495
 *		specified block.
 *
 * If the segment doesn't exist, we ereport, return NULL, or create the
 * segment, according to "behavior".  Note: isTemp need only be correct
 * in the EXTENSION_CREATE case.
N
Neil Conway 已提交
1496
 */
1497
static MdfdVec *
1498 1499
_mdfd_getseg(SMgrRelation reln, ForkNumber forknum, BlockNumber blkno,
			 bool isTemp, ExtensionBehavior behavior)
1500
{
1501
	MdfdVec    *v = mdopen(reln, forknum, behavior);
1502
	BlockNumber targetseg;
1503
	BlockNumber nextsegno;
1504

1505
	if (!v)
1506
		return NULL;			/* only possible if EXTENSION_RETURN_NULL */
1507

1508 1509
	targetseg = blkno / ((BlockNumber) RELSEG_SIZE);
	for (nextsegno = 1; nextsegno <= targetseg; nextsegno++)
1510
	{
1511 1512
		Assert(nextsegno == v->mdfd_segno + 1);

1513
		if (v->mdfd_chain == NULL)
1514
		{
1515
			/*
B
Bruce Momjian 已提交
1516 1517 1518
			 * Normally we will create new segments only if authorized by the
			 * caller (i.e., we are doing mdextend()).	But when doing WAL
			 * recovery, create segments anyway; this allows cases such as
1519 1520 1521
			 * replaying WAL data that has a write into a high-numbered
			 * segment of a relation that was later deleted.  We want to go
			 * ahead and create the segments so we can finish out the replay.
1522
			 *
B
Bruce Momjian 已提交
1523 1524 1525 1526 1527
			 * We have to maintain the invariant that segments before the last
			 * active segment are of size RELSEG_SIZE; therefore, pad them out
			 * with zeroes if needed.  (This only matters if caller is
			 * extending the relation discontiguously, but that can happen in
			 * hash indexes.)
1528
			 */
1529 1530
			if (behavior == EXTENSION_CREATE || InRecovery)
			{
1531
				if (_mdnblocks(reln, forknum, v) < RELSEG_SIZE)
1532
				{
B
Bruce Momjian 已提交
1533
					char	   *zerobuf = palloc0(BLCKSZ);
1534

1535 1536
					mdextend(reln, forknum,
							 nextsegno * ((BlockNumber) RELSEG_SIZE) - 1,
1537 1538 1539
							 zerobuf, isTemp);
					pfree(zerobuf);
				}
1540
				v->mdfd_chain = _mdfd_openseg(reln, forknum, +nextsegno, O_CREAT);
1541 1542 1543 1544
			}
			else
			{
				/* We won't create segment if not existent */
1545
				v->mdfd_chain = _mdfd_openseg(reln, forknum, nextsegno, 0);
1546
			}
1547
			if (v->mdfd_chain == NULL)
1548
			{
1549 1550
				if (behavior == EXTENSION_RETURN_NULL &&
					FILE_POSSIBLY_DELETED(errno))
1551
					return NULL;
1552 1553
				ereport(ERROR,
						(errcode_for_file_access(),
1554
						 errmsg("could not open segment %u of relation %s (target block %u): %m",
1555
								nextsegno,
1556
								relpath(reln->smgr_rnode, forknum),
1557
								blkno)));
1558
			}
1559 1560
		}
		v = v->mdfd_chain;
1561
	}
1562
	return v;
1563 1564
}

1565
/*
1566
 * Get number of blocks present in a single disk file
1567
 */
1568
static BlockNumber
1569
_mdnblocks(SMgrRelation reln, ForkNumber forknum, MdfdVec *seg)
1570
{
1571
	off_t		len;
1572

1573
	len = FileSeek(seg->mdfd_vfd, 0L, SEEK_END);
1574
	if (len < 0)
1575 1576
		ereport(ERROR,
				(errcode_for_file_access(),
1577 1578
			 errmsg("could not seek to end of segment %u of relation %s: %m",
					seg->mdfd_segno, relpath(reln->smgr_rnode, forknum))));
1579 1580
	/* note that this calculation will ignore any partial block at EOF */
	return (BlockNumber) (len / BLCKSZ);
1581
}