granicus.if.org Git - postgresql/blob - src/backend/replication/walsender.c

   1 /*-------------------------------------------------------------------------
   2  *
   3  * walsender.c
   4  *
   5  * The WAL sender process (walsender) is new as of Postgres 9.0. It takes
   6  * charge of XLOG streaming sender in the primary server. At first, it is
   7  * started by the postmaster when the walreceiver in the standby server
   8  * connects to the primary server and requests XLOG streaming replication,
   9  * i.e., unlike any auxiliary process, it is not an always-running process.
  10  * It attempts to keep reading XLOG records from the disk and sending them
  11  * to the standby server, as long as the connection is alive (i.e., like
  12  * any backend, there is an one to one relationship between a connection
  13  * and a walsender process).
  14  *
  15  * Normal termination is by SIGTERM, which instructs the walsender to
  16  * close the connection and exit(0) at next convenient moment. Emergency
  17  * termination is by SIGQUIT; like any backend, the walsender will simply
  18  * abort and exit on SIGQUIT. A close of the connection and a FATAL error
  19  * are treated as not a crash but approximately normal termination;
  20  * the walsender will exit quickly without sending any more XLOG records.
  21  *
  22  * If the server is shut down, postmaster sends us SIGUSR2 after all
  23  * regular backends have exited and the shutdown checkpoint has been written.
  24  * This instruct walsender to send any outstanding WAL, including the
  25  * shutdown checkpoint record, and then exit.
  26  *
  27  * Note that there can be more than one walsender process concurrently.
  28  *
  29  * Portions Copyright (c) 2010-2010, PostgreSQL Global Development Group
  30  *
  31  *
  32  * IDENTIFICATION
  33  *        $PostgreSQL: pgsql/src/backend/replication/walsender.c,v 1.11 2010/03/24 20:11:12 sriggs Exp $
  34  *
  35  *-------------------------------------------------------------------------
  36  */
  37 #include "postgres.h"
  38
  39 #include <unistd.h>
  40
  41 #include "access/xlog_internal.h"
  42 #include "catalog/pg_type.h"
  43 #include "libpq/libpq.h"
  44 #include "libpq/pqformat.h"
  45 #include "libpq/pqsignal.h"
  46 #include "miscadmin.h"
  47 #include "replication/walsender.h"
  48 #include "storage/fd.h"
  49 #include "storage/ipc.h"
  50 #include "storage/lock.h"
  51 #include "storage/pmsignal.h"
  52 #include "tcop/tcopprot.h"
  53 #include "utils/guc.h"
  54 #include "utils/memutils.h"
  55 #include "utils/ps_status.h"
  56
  57 /* Array of WalSnds in shared memory */
  58 WalSndCtlData *WalSndCtl = NULL;
  59
  60 /* My slot in the shared memory array */
  61 static WalSnd *MyWalSnd = NULL;
  62
  63 /* Global state */
  64 bool            am_walsender = false;           /* Am I a walsender process ? */
  65
  66 /* User-settable parameters for walsender */
  67 int                     MaxWalSenders = 0;      /* the maximum number of concurrent walsenders */
  68 int                     WalSndDelay = 200;      /* max sleep time between some actions */
  69
  70 #define NAPTIME_PER_CYCLE 100000L       /* max sleep time between cycles (100ms) */
  71
  72 /*
  73  * These variables are used similarly to openLogFile/Id/Seg/Off,
  74  * but for walsender to read the XLOG.
  75  */
  76 static int      sendFile = -1;
  77 static uint32 sendId = 0;
  78 static uint32 sendSeg = 0;
  79 static uint32 sendOff = 0;
  80
  81 /*
  82  * How far have we sent WAL already? This is also advertised in
  83  * MyWalSnd->sentPtr.
  84  */
  85 static XLogRecPtr sentPtr = {0, 0};
  86
  87 /* Flags set by signal handlers for later service in main loop */
  88 static volatile sig_atomic_t got_SIGHUP = false;
  89 static volatile sig_atomic_t shutdown_requested = false;
  90 static volatile sig_atomic_t ready_to_stop = false;
  91
  92 /* Signal handlers */
  93 static void WalSndSigHupHandler(SIGNAL_ARGS);
  94 static void WalSndShutdownHandler(SIGNAL_ARGS);
  95 static void WalSndQuickDieHandler(SIGNAL_ARGS);
  96
  97 /* Prototypes for private functions */
  98 static int      WalSndLoop(void);
  99 static void InitWalSnd(void);
 100 static void WalSndHandshake(void);
 101 static void WalSndKill(int code, Datum arg);
 102 static void XLogRead(char *buf, XLogRecPtr recptr, Size nbytes);
 103 static bool XLogSend(StringInfo outMsg);
 104 static void CheckClosedConnection(void);
 105
 106 /*
 107  * How much WAL to send in one message? Must be >= XLOG_BLCKSZ.
 108  */
 109 #define MAX_SEND_SIZE (XLOG_SEG_SIZE / 2)
 110
 111 /* Main entry point for walsender process */
 112 int
 113 WalSenderMain(void)
 114 {
 115         MemoryContext walsnd_context;
 116
 117         if (!superuser())
 118                 ereport(FATAL,
 119                                 (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
 120                                  errmsg("must be superuser to start walsender")));
 121
 122         if (RecoveryInProgress())
 123                 ereport(FATAL,
 124                                 (errcode(ERRCODE_CANNOT_CONNECT_NOW),
 125                                  errmsg("recovery is still in progress, can't accept WAL streaming connections")));
 126
 127         /* Create a per-walsender data structure in shared memory */
 128         InitWalSnd();
 129
 130         /*
 131          * Create a memory context that we will do all our work in.  We do this so
 132          * that we can reset the context during error recovery and thereby avoid
 133          * possible memory leaks.  Formerly this code just ran in
 134          * TopMemoryContext, but resetting that would be a really bad idea.
 135          *
 136          * XXX: we don't actually attempt error recovery in walsender, we just
 137          * close the connection and exit.
 138          */
 139         walsnd_context = AllocSetContextCreate(TopMemoryContext,
 140                                                                                    "Wal Sender",
 141                                                                                    ALLOCSET_DEFAULT_MINSIZE,
 142                                                                                    ALLOCSET_DEFAULT_INITSIZE,
 143                                                                                    ALLOCSET_DEFAULT_MAXSIZE);
 144         MemoryContextSwitchTo(walsnd_context);
 145
 146         /* Unblock signals (they were blocked when the postmaster forked us) */
 147         PG_SETMASK(&UnBlockSig);
 148
 149         /* Tell the standby that walsender is ready for receiving commands */
 150         ReadyForQuery(DestRemote);
 151
 152         /* Handle handshake messages before streaming */
 153         WalSndHandshake();
 154
 155         /* Main loop of walsender */
 156         return WalSndLoop();
 157 }
 158
 159 static void
 160 WalSndHandshake(void)
 161 {
 162         StringInfoData input_message;
 163         bool            replication_started = false;
 164
 165         initStringInfo(&input_message);
 166
 167         while (!replication_started)
 168         {
 169                 int                     firstchar;
 170
 171                 /* Wait for a command to arrive */
 172                 firstchar = pq_getbyte();
 173
 174                 /*
 175                  * Check for any other interesting events that happened while we
 176                  * slept.
 177                  */
 178                 if (got_SIGHUP)
 179                 {
 180                         got_SIGHUP = false;
 181                         ProcessConfigFile(PGC_SIGHUP);
 182                 }
 183
 184                 if (firstchar != EOF)
 185                 {
 186                         /*
 187                          * Read the message contents. This is expected to be done without
 188                          * blocking because we've been able to get message type code.
 189                          */
 190                         if (pq_getmessage(&input_message, 0))
 191                                 firstchar = EOF;        /* suitable message already logged */
 192                 }
 193
 194                 /* Handle the very limited subset of commands expected in this phase */
 195                 switch (firstchar)
 196                 {
 197                         case 'Q':                       /* Query message */
 198                                 {
 199                                         const char *query_string;
 200                                         XLogRecPtr      recptr;
 201
 202                                         query_string = pq_getmsgstring(&input_message);
 203                                         pq_getmsgend(&input_message);
 204
 205                                         if (strcmp(query_string, "IDENTIFY_SYSTEM") == 0)
 206                                         {
 207                                                 StringInfoData buf;
 208                                                 char            sysid[32];
 209                                                 char            tli[11];
 210
 211                                                 /*
 212                                                  * Reply with a result set with one row, two columns.
 213                                                  * First col is system ID, and second if timeline ID
 214                                                  */
 215
 216                                                 snprintf(sysid, sizeof(sysid), UINT64_FORMAT,
 217                                                                  GetSystemIdentifier());
 218                                                 snprintf(tli, sizeof(tli), "%u", ThisTimeLineID);
 219
 220                                                 /* Send a RowDescription message */
 221                                                 pq_beginmessage(&buf, 'T');
 222                                                 pq_sendint(&buf, 2, 2); /* 2 fields */
 223
 224                                                 /* first field */
 225                                                 pq_sendstring(&buf, "systemid");                /* col name */
 226                                                 pq_sendint(&buf, 0, 4); /* table oid */
 227                                                 pq_sendint(&buf, 0, 2); /* attnum */
 228                                                 pq_sendint(&buf, TEXTOID, 4);   /* type oid */
 229                                                 pq_sendint(&buf, -1, 2);                /* typlen */
 230                                                 pq_sendint(&buf, 0, 4); /* typmod */
 231                                                 pq_sendint(&buf, 0, 2); /* format code */
 232
 233                                                 /* second field */
 234                                                 pq_sendstring(&buf, "timeline");                /* col name */
 235                                                 pq_sendint(&buf, 0, 4); /* table oid */
 236                                                 pq_sendint(&buf, 0, 2); /* attnum */
 237                                                 pq_sendint(&buf, INT4OID, 4);   /* type oid */
 238                                                 pq_sendint(&buf, 4, 2); /* typlen */
 239                                                 pq_sendint(&buf, 0, 4); /* typmod */
 240                                                 pq_sendint(&buf, 0, 2); /* format code */
 241                                                 pq_endmessage(&buf);
 242
 243                                                 /* Send a DataRow message */
 244                                                 pq_beginmessage(&buf, 'D');
 245                                                 pq_sendint(&buf, 2, 2); /* # of columns */
 246                                                 pq_sendint(&buf, strlen(sysid), 4);             /* col1 len */
 247                                                 pq_sendbytes(&buf, (char *) &sysid, strlen(sysid));
 248                                                 pq_sendint(&buf, strlen(tli), 4);               /* col2 len */
 249                                                 pq_sendbytes(&buf, (char *) tli, strlen(tli));
 250                                                 pq_endmessage(&buf);
 251
 252                                                 /* Send CommandComplete and ReadyForQuery messages */
 253                                                 EndCommand("SELECT", DestRemote);
 254                                                 ReadyForQuery(DestRemote);
 255                                         }
 256                                         else if (sscanf(query_string, "START_REPLICATION %X/%X",
 257                                                                         &recptr.xlogid, &recptr.xrecoff) == 2)
 258                                         {
 259                                                 StringInfoData buf;
 260
 261                                                 /* Send a CopyOutResponse message, and start streaming */
 262                                                 pq_beginmessage(&buf, 'H');
 263                                                 pq_sendbyte(&buf, 0);
 264                                                 pq_sendint(&buf, 0, 2);
 265                                                 pq_endmessage(&buf);
 266
 267                                                 /*
 268                                                  * Initialize position to the received one, then the
 269                                                  * xlog records begin to be shipped from that position
 270                                                  */
 271                                                 sentPtr = recptr;
 272
 273                                                 /* break out of the loop */
 274                                                 replication_started = true;
 275                                         }
 276                                         else
 277                                         {
 278                                                 ereport(FATAL,
 279                                                                 (errcode(ERRCODE_PROTOCOL_VIOLATION),
 280                                                                  errmsg("invalid standby query string: %s", query_string)));
 281                                         }
 282                                         break;
 283                                 }
 284
 285                         case 'X':
 286                                 /* standby is closing the connection */
 287                                 proc_exit(0);
 288
 289                         case EOF:
 290                                 /* standby disconnected unexpectedly */
 291                                 ereport(COMMERROR,
 292                                                 (errcode(ERRCODE_PROTOCOL_VIOLATION),
 293                                                  errmsg("unexpected EOF on standby connection")));
 294                                 proc_exit(0);
 295
 296                         default:
 297                                 ereport(FATAL,
 298                                                 (errcode(ERRCODE_PROTOCOL_VIOLATION),
 299                                                  errmsg("invalid standby handshake message type %d", firstchar)));
 300                 }
 301         }
 302 }
 303
 304 /*
 305  * Check if the remote end has closed the connection.
 306  */
 307 static void
 308 CheckClosedConnection(void)
 309 {
 310         unsigned char firstchar;
 311         int                     r;
 312
 313         r = pq_getbyte_if_available(&firstchar);
 314         if (r < 0)
 315         {
 316                 /* unexpected error or EOF */
 317                 ereport(COMMERROR,
 318                                 (errcode(ERRCODE_PROTOCOL_VIOLATION),
 319                                  errmsg("unexpected EOF on standby connection")));
 320                 proc_exit(0);
 321         }
 322         if (r == 0)
 323         {
 324                 /* no data available without blocking */
 325                 return;
 326         }
 327
 328         /* Handle the very limited subset of commands expected in this phase */
 329         switch (firstchar)
 330         {
 331                         /*
 332                          * 'X' means that the standby is closing down the socket.
 333                          */
 334                 case 'X':
 335                         proc_exit(0);
 336
 337                 default:
 338                         ereport(FATAL,
 339                                         (errcode(ERRCODE_PROTOCOL_VIOLATION),
 340                                          errmsg("invalid standby closing message type %d",
 341                                                         firstchar)));
 342         }
 343 }
 344
 345 /* Main loop of walsender process */
 346 static int
 347 WalSndLoop(void)
 348 {
 349         StringInfoData output_message;
 350
 351         initStringInfo(&output_message);
 352
 353         /* Loop forever */
 354         for (;;)
 355         {
 356                 int                     remain;         /* remaining time (ms) */
 357
 358                 /*
 359                  * Emergency bailout if postmaster has died.  This is to avoid the
 360                  * necessity for manual cleanup of all postmaster children.
 361                  */
 362                 if (!PostmasterIsAlive(true))
 363                         exit(1);
 364                 /* Process any requests or signals received recently */
 365                 if (got_SIGHUP)
 366                 {
 367                         got_SIGHUP = false;
 368                         ProcessConfigFile(PGC_SIGHUP);
 369                 }
 370
 371                 /*
 372                  * When SIGUSR2 arrives, we send all outstanding logs up to the
 373                  * shutdown checkpoint record (i.e., the latest record) and exit.
 374                  */
 375                 if (ready_to_stop)
 376                 {
 377                         XLogSend(&output_message);
 378                         shutdown_requested = true;
 379                 }
 380
 381                 /* Normal exit from the walsender is here */
 382                 if (shutdown_requested)
 383                 {
 384                         /* Inform the standby that XLOG streaming was done */
 385                         pq_puttextmessage('C', "COPY 0");
 386                         pq_flush();
 387
 388                         proc_exit(0);
 389                 }
 390
 391                 /*
 392                  * Nap for the configured time or until a message arrives.
 393                  *
 394                  * On some platforms, signals won't interrupt the sleep.  To ensure we
 395                  * respond reasonably promptly when someone signals us, break down the
 396                  * sleep into NAPTIME_PER_CYCLE (ms) increments, and check for
 397                  * interrupts after each nap.
 398                  */
 399                 remain = WalSndDelay * 1000L;
 400                 while (remain > 0)
 401                 {
 402                         if (got_SIGHUP || shutdown_requested || ready_to_stop)
 403                                 break;
 404
 405                         /*
 406                          * Check to see whether a message from the standby or an interrupt
 407                          * from other processes has arrived.
 408                          */
 409                         pg_usleep(remain > NAPTIME_PER_CYCLE ? NAPTIME_PER_CYCLE : remain);
 410                         CheckClosedConnection();
 411
 412                         remain -= NAPTIME_PER_CYCLE;
 413                 }
 414
 415                 /* Attempt to send the log once every loop */
 416                 if (!XLogSend(&output_message))
 417                         goto eof;
 418         }
 419
 420         /* can't get here because the above loop never exits */
 421         return 1;
 422
 423 eof:
 424
 425         /*
 426          * Reset whereToSendOutput to prevent ereport from attempting to send any
 427          * more messages to the standby.
 428          */
 429         if (whereToSendOutput == DestRemote)
 430                 whereToSendOutput = DestNone;
 431
 432         proc_exit(0);
 433         return 1;                                       /* keep the compiler quiet */
 434 }
 435
 436 /* Initialize a per-walsender data structure for this walsender process */
 437 static void
 438 InitWalSnd(void)
 439 {
 440         /* use volatile pointer to prevent code rearrangement */
 441         int                     i;
 442
 443         /*
 444          * WalSndCtl should be set up already (we inherit this by fork() or
 445          * EXEC_BACKEND mechanism from the postmaster).
 446          */
 447         Assert(WalSndCtl != NULL);
 448         Assert(MyWalSnd == NULL);
 449
 450         /*
 451          * Find a free walsender slot and reserve it. If this fails, we must be
 452          * out of WalSnd structures.
 453          */
 454         for (i = 0; i < MaxWalSenders; i++)
 455         {
 456                 volatile WalSnd *walsnd = &WalSndCtl->walsnds[i];
 457
 458                 SpinLockAcquire(&walsnd->mutex);
 459
 460                 if (walsnd->pid != 0)
 461                 {
 462                         SpinLockRelease(&walsnd->mutex);
 463                         continue;
 464                 }
 465                 else
 466                 {
 467                         /* found */
 468                         MyWalSnd = (WalSnd *) walsnd;
 469                         walsnd->pid = MyProcPid;
 470                         MemSet(&MyWalSnd->sentPtr, 0, sizeof(XLogRecPtr));
 471                         SpinLockRelease(&walsnd->mutex);
 472                         break;
 473                 }
 474         }
 475         if (MyWalSnd == NULL)
 476                 ereport(FATAL,
 477                                 (errcode(ERRCODE_TOO_MANY_CONNECTIONS),
 478                                  errmsg("sorry, too many standbys already")));
 479
 480         /* Arrange to clean up at walsender exit */
 481         on_shmem_exit(WalSndKill, 0);
 482 }
 483
 484 /* Destroy the per-walsender data structure for this walsender process */
 485 static void
 486 WalSndKill(int code, Datum arg)
 487 {
 488         Assert(MyWalSnd != NULL);
 489
 490         /*
 491          * Mark WalSnd struct no longer in use. Assume that no lock is required
 492          * for this.
 493          */
 494         MyWalSnd->pid = 0;
 495
 496         /* WalSnd struct isn't mine anymore */
 497         MyWalSnd = NULL;
 498 }
 499
 500 /*
 501  * Read 'nbytes' bytes from WAL into 'buf', starting at location 'recptr'
 502  */
 503 void
 504 XLogRead(char *buf, XLogRecPtr recptr, Size nbytes)
 505 {
 506         char            path[MAXPGPATH];
 507         uint32          startoff;
 508
 509         while (nbytes > 0)
 510         {
 511                 int                     segbytes;
 512                 int                     readbytes;
 513
 514                 startoff = recptr.xrecoff % XLogSegSize;
 515
 516                 if (sendFile < 0 || !XLByteInSeg(recptr, sendId, sendSeg))
 517                 {
 518                         /* Switch to another logfile segment */
 519                         if (sendFile >= 0)
 520                                 close(sendFile);
 521
 522                         XLByteToSeg(recptr, sendId, sendSeg);
 523                         XLogFilePath(path, ThisTimeLineID, sendId, sendSeg);
 524
 525                         sendFile = BasicOpenFile(path, O_RDONLY | PG_BINARY, 0);
 526                         if (sendFile < 0)
 527                                 ereport(FATAL,  /* XXX: Why FATAL? */
 528                                                 (errcode_for_file_access(),
 529                                                  errmsg("could not open file \"%s\" (log file %u, segment %u): %m",
 530                                                                 path, sendId, sendSeg)));
 531                         sendOff = 0;
 532                 }
 533
 534                 /* Need to seek in the file? */
 535                 if (sendOff != startoff)
 536                 {
 537                         if (lseek(sendFile, (off_t) startoff, SEEK_SET) < 0)
 538                                 ereport(FATAL,
 539                                                 (errcode_for_file_access(),
 540                                                  errmsg("could not seek in log file %u, segment %u to offset %u: %m",
 541                                                                 sendId, sendSeg, startoff)));
 542                         sendOff = startoff;
 543                 }
 544
 545                 /* How many bytes are within this segment? */
 546                 if (nbytes > (XLogSegSize - startoff))
 547                         segbytes = XLogSegSize - startoff;
 548                 else
 549                         segbytes = nbytes;
 550
 551                 readbytes = read(sendFile, buf, segbytes);
 552                 if (readbytes <= 0)
 553                         ereport(FATAL,
 554                                         (errcode_for_file_access(),
 555                         errmsg("could not read from log file %u, segment %u, offset %u, "
 556                                    "length %lu: %m",
 557                                    sendId, sendSeg, sendOff, (unsigned long) segbytes)));
 558
 559                 /* Update state for read */
 560                 XLByteAdvance(recptr, readbytes);
 561
 562                 sendOff += readbytes;
 563                 nbytes -= readbytes;
 564                 buf += readbytes;
 565         }
 566 }
 567
 568 /*
 569  * Read all WAL that's been written (and flushed) since last cycle, and send
 570  * it to client.
 571  *
 572  * Returns true if OK, false if trouble.
 573  */
 574 static bool
 575 XLogSend(StringInfo outMsg)
 576 {
 577         XLogRecPtr      SendRqstPtr;
 578         char            activitymsg[50];
 579
 580         /* use volatile pointer to prevent code rearrangement */
 581         volatile WalSnd *walsnd = MyWalSnd;
 582
 583         /* Attempt to send all records flushed to the disk already */
 584         SendRqstPtr = GetWriteRecPtr();
 585
 586         /* Quick exit if nothing to do */
 587         if (!XLByteLT(sentPtr, SendRqstPtr))
 588                 return true;
 589
 590         /*
 591          * We gather multiple records together by issuing just one XLogRead() of a
 592          * suitable size, and send them as one CopyData message. Repeat until
 593          * we've sent everything we can.
 594          */
 595         while (XLByteLT(sentPtr, SendRqstPtr))
 596         {
 597                 XLogRecPtr      startptr;
 598                 XLogRecPtr      endptr;
 599                 Size            nbytes;
 600
 601                 /*
 602                  * Figure out how much to send in one message. If there's less than
 603                  * MAX_SEND_SIZE bytes to send, send everything. Otherwise send
 604                  * MAX_SEND_SIZE bytes, but round to page boundary.
 605                  *
 606                  * The rounding is not only for performance reasons. Walreceiver
 607                  * relies on the fact that we never split a WAL record across two
 608                  * messages. Since a long WAL record is split at page boundary into
 609                  * continuation records, page boundary is always a safe cut-off point.
 610                  * We also assume that SendRqstPtr never points in the middle of a WAL
 611                  * record.
 612                  */
 613                 startptr = sentPtr;
 614                 if (startptr.xrecoff >= XLogFileSize)
 615                 {
 616                         /*
 617                          * crossing a logid boundary, skip the non-existent last log
 618                          * segment in previous logical log file.
 619                          */
 620                         startptr.xlogid += 1;
 621                         startptr.xrecoff = 0;
 622                 }
 623
 624                 endptr = startptr;
 625                 XLByteAdvance(endptr, MAX_SEND_SIZE);
 626                 /* round down to page boundary. */
 627                 endptr.xrecoff -= (endptr.xrecoff % XLOG_BLCKSZ);
 628                 /* if we went beyond SendRqstPtr, back off */
 629                 if (XLByteLT(SendRqstPtr, endptr))
 630                         endptr = SendRqstPtr;
 631
 632                 /*
 633                  * OK to read and send the slice.
 634                  *
 635                  * We don't need to convert the xlogid/xrecoff from host byte order to
 636                  * network byte order because the both server can be expected to have
 637                  * the same byte order. If they have different byte order, we don't
 638                  * reach here.
 639                  */
 640                 pq_sendbyte(outMsg, 'w');
 641                 pq_sendbytes(outMsg, (char *) &startptr, sizeof(startptr));
 642
 643                 if (endptr.xlogid != startptr.xlogid)
 644                 {
 645                         Assert(endptr.xlogid == startptr.xlogid + 1);
 646                         nbytes = endptr.xrecoff + XLogFileSize - startptr.xrecoff;
 647                 }
 648                 else
 649                         nbytes = endptr.xrecoff - startptr.xrecoff;
 650
 651                 sentPtr = endptr;
 652
 653                 /*
 654                  * Read the log directly into the output buffer to prevent extra
 655                  * memcpy calls.
 656                  */
 657                 enlargeStringInfo(outMsg, nbytes);
 658
 659                 XLogRead(&outMsg->data[outMsg->len], startptr, nbytes);
 660                 outMsg->len += nbytes;
 661                 outMsg->data[outMsg->len] = '\0';
 662
 663                 pq_putmessage('d', outMsg->data, outMsg->len);
 664                 resetStringInfo(outMsg);
 665         }
 666
 667         /* Update shared memory status */
 668         SpinLockAcquire(&walsnd->mutex);
 669         walsnd->sentPtr = sentPtr;
 670         SpinLockRelease(&walsnd->mutex);
 671
 672         /* Flush pending output */
 673         if (pq_flush())
 674                 return false;
 675
 676         /* Report progress of XLOG streaming in PS display */
 677         snprintf(activitymsg, sizeof(activitymsg), "streaming %X/%X",
 678                          sentPtr.xlogid, sentPtr.xrecoff);
 679         set_ps_display(activitymsg, false);
 680
 681         return true;
 682 }
 683
 684 /* SIGHUP: set flag to re-read config file at next convenient time */
 685 static void
 686 WalSndSigHupHandler(SIGNAL_ARGS)
 687 {
 688         got_SIGHUP = true;
 689 }
 690
 691 /* SIGTERM: set flag to shut down */
 692 static void
 693 WalSndShutdownHandler(SIGNAL_ARGS)
 694 {
 695         shutdown_requested = true;
 696 }
 697
 698 /*
 699  * WalSndQuickDieHandler() occurs when signalled SIGQUIT by the postmaster.
 700  *
 701  * Some backend has bought the farm,
 702  * so we need to stop what we're doing and exit.
 703  */
 704 static void
 705 WalSndQuickDieHandler(SIGNAL_ARGS)
 706 {
 707         PG_SETMASK(&BlockSig);
 708
 709         /*
 710          * We DO NOT want to run proc_exit() callbacks -- we're here because
 711          * shared memory may be corrupted, so we don't want to try to clean up our
 712          * transaction.  Just nail the windows shut and get out of town.  Now that
 713          * there's an atexit callback to prevent third-party code from breaking
 714          * things by calling exit() directly, we have to reset the callbacks
 715          * explicitly to make this work as intended.
 716          */
 717         on_exit_reset();
 718
 719         /*
 720          * Note we do exit(2) not exit(0).      This is to force the postmaster into a
 721          * system reset cycle if some idiot DBA sends a manual SIGQUIT to a random
 722          * backend.  This is necessary precisely because we don't clean up our
 723          * shared memory state.  (The "dead man switch" mechanism in pmsignal.c
 724          * should ensure the postmaster sees this as a crash, too, but no harm in
 725          * being doubly sure.)
 726          */
 727         exit(2);
 728 }
 729
 730 /* SIGUSR2: set flag to do a last cycle and shut down afterwards */
 731 static void
 732 WalSndLastCycleHandler(SIGNAL_ARGS)
 733 {
 734         ready_to_stop = true;
 735 }
 736
 737 /* Set up signal handlers */
 738 void
 739 WalSndSignals(void)
 740 {
 741         /* Set up signal handlers */
 742         pqsignal(SIGHUP, WalSndSigHupHandler);          /* set flag to read config
 743                                                                                                  * file */
 744         pqsignal(SIGINT, SIG_IGN);      /* not used */
 745         pqsignal(SIGTERM, WalSndShutdownHandler);       /* request shutdown */
 746         pqsignal(SIGQUIT, WalSndQuickDieHandler);       /* hard crash time */
 747         pqsignal(SIGALRM, SIG_IGN);
 748         pqsignal(SIGPIPE, SIG_IGN);
 749         pqsignal(SIGUSR1, SIG_IGN); /* not used */
 750         pqsignal(SIGUSR2, WalSndLastCycleHandler);      /* request a last cycle and
 751                                                                                                  * shutdown */
 752
 753         /* Reset some signals that are accepted by postmaster but not here */
 754         pqsignal(SIGCHLD, SIG_DFL);
 755         pqsignal(SIGTTIN, SIG_DFL);
 756         pqsignal(SIGTTOU, SIG_DFL);
 757         pqsignal(SIGCONT, SIG_DFL);
 758         pqsignal(SIGWINCH, SIG_DFL);
 759 }
 760
 761 /* Report shared-memory space needed by WalSndShmemInit */
 762 Size
 763 WalSndShmemSize(void)
 764 {
 765         Size            size = 0;
 766
 767         size = offsetof(WalSndCtlData, walsnds);
 768         size = add_size(size, mul_size(MaxWalSenders, sizeof(WalSnd)));
 769
 770         return size;
 771 }
 772
 773 /* Allocate and initialize walsender-related shared memory */
 774 void
 775 WalSndShmemInit(void)
 776 {
 777         bool            found;
 778         int                     i;
 779
 780         WalSndCtl = (WalSndCtlData *)
 781                 ShmemInitStruct("Wal Sender Ctl", WalSndShmemSize(), &found);
 782
 783         if (WalSndCtl == NULL)
 784                 ereport(FATAL,
 785                                 (errcode(ERRCODE_OUT_OF_MEMORY),
 786                                  errmsg("not enough shared memory for walsender")));
 787         if (found)
 788                 return;                                 /* already initialized */
 789
 790         /* Initialize the data structures */
 791         MemSet(WalSndCtl, 0, WalSndShmemSize());
 792
 793         for (i = 0; i < MaxWalSenders; i++)
 794         {
 795                 WalSnd     *walsnd = &WalSndCtl->walsnds[i];
 796
 797                 SpinLockInit(&walsnd->mutex);
 798         }
 799 }
 800
 801 /*
 802  * Returns the oldest Send position among walsenders. Or InvalidXLogRecPtr
 803  * if none.
 804  */
 805 XLogRecPtr
 806 GetOldestWALSendPointer(void)
 807 {
 808         XLogRecPtr      oldest = {0, 0};
 809         int                     i;
 810         bool            found = false;
 811
 812         for (i = 0; i < MaxWalSenders; i++)
 813         {
 814                 /* use volatile pointer to prevent code rearrangement */
 815                 volatile WalSnd *walsnd = &WalSndCtl->walsnds[i];
 816                 XLogRecPtr      recptr;
 817
 818                 if (walsnd->pid == 0)
 819                         continue;
 820
 821                 SpinLockAcquire(&walsnd->mutex);
 822                 recptr = walsnd->sentPtr;
 823                 SpinLockRelease(&walsnd->mutex);
 824
 825                 if (recptr.xlogid == 0 && recptr.xrecoff == 0)
 826                         continue;
 827
 828                 if (!found || XLByteLT(recptr, oldest))
 829                         oldest = recptr;
 830                 found = true;
 831         }
 832         return oldest;
 833 }