96bef763-76aa-4452-b071-5b849b31cd38
/
PolarDBforPostgreSQL
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
PolarDBforPostgreSQL/external/polar_worker/polar_worker.c

982 lines
24 KiB
C
Raw Blame History

/*-------------------------------------------------------------------------
*
* polar_worker.c
* Do some backgroud things for PolarDB periodically. Such as:
* (1) auto prealloc wal files,
* (2) auto clean core dump files,
* (3) auto clean xlog temp files.
*
* Copyright (c) 2020, Alibaba Group Holding Limited
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* IDENTIFICATION
* external/polar_worker/polar_worker.c
*
*-------------------------------------------------------------------------
*/
#include "postgres.h"
#include <unistd.h>
#include "miscadmin.h"
#include "pgstat.h"
#include "access/polar_logindex_redo.h"
#include "access/xlog.h"
#include "access/xlog_internal.h"
#include "common/file_perm.h"
#include "postmaster/bgworker.h"
#include "postmaster/startup.h"
#include "storage/ipc.h"
#include "storage/latch.h"
#include "storage/lwlock.h"
#include "storage/polar_fd.h"
#include "storage/proc.h"
#include "storage/procsignal.h"
#include "storage/polar_io_fencing.h"
#include "utils/guc.h"
#include "utils/polar_coredump.h"
#include "utils/timestamp.h"
#include "polar_flashback/polar_flashback_log_file.h"
#include "polar_flashback/polar_flashback_log.h"
PG_MODULE_MAGIC;
#define POLAR_MAX_BUFF_SIZE 1024
#define POLAR_NUM_FILE_INTERVAL 32
#define POLAR_XLOG_TEMP_FILE_SUFFIX "xlogtemp"
#define POLAR_DATA_DIR() (POLAR_FILE_IN_SHARED_STORAGE() ? polar_datadir : DataDir)
#define POLAR_FILE_PATH(path, orign) \
snprintf((path), MAXPGPATH, "%s/%s", POLAR_DATA_DIR(), (orign));
typedef struct
{
time_t mtime;
char file_path[MAXPGPATH];
} polar_file_descriptor;
void _PG_init(void);
void polar_worker_handler_main(Datum main_arg);
static void start_polar_worker(void);
static void prealloc_wal_files(void);
static void polar_worker_sighup_handler(SIGNAL_ARGS);
static void polar_worker_sigterm_handler(SIGNAL_ARGS);
static void polar_worker_sigusr2_handler(SIGNAL_ARGS);
static void polar_xlog_temp_file_clean(void);
static void polar_remove_old_core_files(void);
static bool polar_remove_file(const char *path);
static bool polar_transfer_coretmp_to_log(const char *path);
static void polar_transfer_core_tmp_to_log_with_remove(void);
static char *polar_addr2line(char const *const program_name, void const *const addr);
static void polar_coredump_handler(void);
static int cmp_file_mtime(const void *a, const void *b);
static bool polar_address_to_stack_info(const void *addr, int line_number);
/* Flags set by signal handlers */
static volatile sig_atomic_t got_sigterm = false;
static volatile sig_atomic_t got_sighup = false;
/* GUC variables. */
static bool enable_polar_worker = true; /* Start prealloc wal file worker? */
static int polar_worker_check_interval; /* Check interval */
static int prealloc_wal_file_num; /* How many files needed to prealloc */
static char *polar_core_file_path;
static char *core_name_suffix;
static int core_file_outdate_time;
static int num_corefile_reserved_old;
static int num_corefile_reserved_new;
static int xlog_temp_outdate_time;
static int prealloc_flashback_log_file_num; /* How many flashback log files needed to prealloc */
/*
* Module load callback.
*/
void
_PG_init(void)
{
DefineCustomIntVariable("polar_worker.polar_worker_check_interval",
"Sets the interval between polar worker check",
"If set to zero, polar worker is disabled.",
&polar_worker_check_interval,
5,
0, INT_MAX / 1000,
PGC_SIGHUP,
GUC_UNIT_S,
NULL,
NULL,
NULL);
DefineCustomIntVariable("polar_worker.prealloc_wal_file_num",
"Sets the num of how many prealloc wal file",
NULL,
&prealloc_wal_file_num,
2,
1, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL);
/* can't define PGC_POSTMASTER variable after startup */
DefineCustomBoolVariable("polar_worker.enable_polar_worker",
"Starts worker or not.",
NULL,
&enable_polar_worker,
true,
PGC_POSTMASTER,
0,
NULL,
NULL,
NULL);
DefineCustomStringVariable(
"polar_worker.core_file_path",
"path of core file",
"path of core file",
&polar_core_file_path,
".",
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomStringVariable(
"polar_worker.core_name_suffix",
"To assign corefile name.",
"To assign corefile name.",
&core_name_suffix,
"core",
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomIntVariable(
"polar_worker.core_file_outdate_time",
"outdate time of core file.",
NULL,
&core_file_outdate_time,
-1,
-1, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomIntVariable(
"polar_worker.num_corefile_reserved_old",
"num of oldest reserverd corefile.",
NULL,
&num_corefile_reserved_old,
32,
0, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomIntVariable(
"polar_worker.num_corefile_reserved_new",
"num of newest reserverd corefile.",
NULL,
&num_corefile_reserved_new,
32,
0, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomIntVariable(
"polar_worker.xlog_temp_outdate_time",
"outdate time of xlog temp.",
NULL,
&xlog_temp_outdate_time,
-1,
-1, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL
);
DefineCustomIntVariable("polar_worker.prealloc_flashback_log_file_num",
"Sets the num of how many prealloc flashback log file",
NULL,
&prealloc_flashback_log_file_num,
2,
1, INT_MAX / 1000,
PGC_SIGHUP,
0,
NULL,
NULL,
NULL);
EmitWarningsOnPlaceholders("polar_worker");
/* Register polar worker. */
start_polar_worker();
}
/*
* Start autoprewarm master worker process.
*/
static void
start_polar_worker(void)
{
BackgroundWorker worker;
BackgroundWorkerHandle *handle;
BgwHandleStatus status;
pid_t pid;
MemSet(&worker, 0, sizeof(BackgroundWorker));
worker.bgw_flags = BGWORKER_SHMEM_ACCESS;
worker.bgw_start_time = BgWorkerStart_PostmasterStart;
worker.bgw_restart_time = 30;
strcpy(worker.bgw_library_name, "polar_worker");
strcpy(worker.bgw_function_name, "polar_worker_handler_main");
strcpy(worker.bgw_name, POLAR_WORKER_PROCESS_NAME);
strcpy(worker.bgw_type, POLAR_WORKER_PROCESS_NAME);
if (process_shared_preload_libraries_in_progress)
{
RegisterBackgroundWorker(&worker);
return;
}
/* must set notify PID to wait for startup */
worker.bgw_notify_pid = MyProcPid;
if (!RegisterDynamicBackgroundWorker(&worker, &handle))
ereport(ERROR,
(errcode(ERRCODE_INSUFFICIENT_RESOURCES),
errmsg("could not register background process"),
errhint("You may need to increase max_worker_processes.")));
status = WaitForBackgroundWorkerStartup(handle, &pid);
if (status != BGWH_STARTED)
ereport(ERROR,
(errcode(ERRCODE_INSUFFICIENT_RESOURCES),
errmsg("could not start background process"),
errhint("More details may be available in the server log.")));
}
/*
* Preallocate wal files beyond last redolog endpoint.
*/
static void
prealloc_wal_files(void)
{
XLogRecPtr insert_ptr = GetXLogInsertRecPtr();
XLogSegNo _log_seg_no;
int lf;
bool use_existent;
int count = 0;
XLByteToPrevSeg(insert_ptr, _log_seg_no, wal_segment_size);
while (count < prealloc_wal_file_num)
{
_log_seg_no++;
count++;
use_existent = true;
lf = XLogFileInit(_log_seg_no, &use_existent, true);
polar_close(lf);
if (!use_existent)
CheckpointStats.ckpt_segs_added++;
}
}
/*
* Main entry point for polar worker process.
*/
void
polar_worker_handler_main(Datum main_arg)
{
TimestampTz last_time = GetCurrentTimestamp();
/* Establish signal handlers; once that's done, unblock signals. */
pqsignal(SIGTERM, polar_worker_sigterm_handler);
pqsignal(SIGHUP, polar_worker_sighup_handler);
pqsignal(SIGUSR1, procsignal_sigusr1_handler);
pqsignal(SIGUSR2, polar_worker_sigusr2_handler);
/* ignore the child signal in polar_worker */
pqsignal(SIGCHLD, SIG_IGN);
BackgroundWorkerUnblockSignals();
/* Add polar worker into backends for showing them in pg_stat_activity */
polar_init_dynamic_bgworker_in_backends();
/*
* POLAR: if some process coredump happens, polar worder will be reset too,
* so do it at the begining.
*/
polar_coredump_handler();
/* Periodically prealloc wal file until terminated. */
while (!got_sigterm)
{
int rc = 0;
/* In case of a SIGHUP, just reload the configuration. */
if (got_sighup)
{
got_sighup = false;
ProcessConfigFile(PGC_SIGHUP);
}
/*
* check for any other interesting events that happened while we
* slept.
*/
if (ConfigReloadPending)
{
ConfigReloadPending = false;
ProcessConfigFile(PGC_SIGHUP);
}
if (!enable_polar_worker)
{
/* We don't want to prealloc wal file now, so just wait forever. */
rc = WaitLatch(&MyProc->procLatch,
WL_LATCH_SET | WL_POSTMASTER_DEATH,
-1L,
PG_WAIT_EXTENSION);
}
/* POLAR: in replica mode, polar worker only do fullpage snapshot work */
else
{
long delay_in_ms = 1000;
if (POLAR_LOGINDEX_ENABLE_FULLPAGE() && polar_in_replica_mode())
{
polar_bgworker_fullpage_snapshot_replay(polar_logindex_redo_instance->fullpage_ctl);
polar_fullpage_bgworker_wait_notify(polar_logindex_redo_instance->fullpage_ctl);
}
/*POLAR: coredump/wal prefetch only in master/standby take affect */
if (!polar_in_replica_mode())
{
TimestampTz next_time = 0;
long secs = 0;
int usecs = 0;
/* FATAL after detection of double write in one shared storage. */
polar_check_double_write();
/* Compute the next prealloc wal file time. */
next_time =
TimestampTzPlusMilliseconds(last_time,
polar_worker_check_interval * 1000);
TimestampDifference(GetCurrentTimestamp(), next_time,
&secs, &usecs);
delay_in_ms = secs * 1000 + (usecs / 1000);
/* If we are in Recovery mode, sleep util consistant */
if (RecoveryInProgress())
delay_in_ms = polar_worker_check_interval * 1000;
else if (delay_in_ms <= 0)
{
last_time = GetCurrentTimestamp();
/* Perform a prealloc wal and fullpage file operation if it's time. */
if (!RecoveryInProgress())
{
prealloc_wal_files();
if (POLAR_LOGINDEX_ENABLE_FULLPAGE())
polar_prealloc_fullpage_files(polar_logindex_redo_instance->fullpage_ctl);
}
/* Perform xlog temp file clean */
if (xlog_temp_outdate_time >= 0)
polar_xlog_temp_file_clean();
/*
* Perform a prealloc flashback log operation after
* the control info has been filled by polar_startup_flog.
*/
if (polar_is_flog_enabled(flog_instance) && polar_has_flog_startup(flog_instance))
polar_prealloc_flog_files(flog_instance->buf_ctl, prealloc_flashback_log_file_num);
continue;
}
}
rc = WaitLatch(&MyProc->procLatch,
WL_LATCH_SET | WL_TIMEOUT | WL_POSTMASTER_DEATH,
delay_in_ms < POLAR_IO_FENCING_INTERVAL ? delay_in_ms : POLAR_IO_FENCING_INTERVAL,
PG_WAIT_EXTENSION);
}
/* Reset the latch, bail out if postmaster died, otherwise loop. */
ResetLatch(&MyProc->procLatch);
if (rc & WL_POSTMASTER_DEATH)
proc_exit(1);
}
/*
* Don't allow to exit in signal handler,maybe process
* is IO pending now, file cannot close normally, so
* exit here for safety
*/
if (got_sigterm)
{
/*
* From here on, elog(ERROR) should end with exit(1), not send
* control back to the sigsetjmp block above
*/
ExitOnAnyError = true;
proc_exit(0);
}
}
/*
* Signal handler for SIGTERM
*/
static void
polar_worker_sigterm_handler(SIGNAL_ARGS)
{
int save_errno = errno;
got_sigterm = true;
if (MyProc)
SetLatch(&MyProc->procLatch);
polar_set_shutdown_requested_flag();
errno = save_errno;
}
/*
* Signal handler for SIGHUP
*/
static void
polar_worker_sighup_handler(SIGNAL_ARGS)
{
int save_errno = errno;
got_sighup = true;
ConfigReloadPending = true;
if (MyProc)
SetLatch(&MyProc->procLatch);
errno = save_errno;
}
/*
* Signal handler for SIGUSR2
*/
static void
polar_worker_sigusr2_handler(SIGNAL_ARGS)
{
polar_fullpage_set_online_promote(true);
}
/*
* POLAR: handler for coredump
*/
static void
polar_coredump_handler()
{
/*
* POLAR: process_coredump_info:
* 1. clean outdate core file
* 2. print stack info in core.tmp and clean it
*/
if (polar_enable_coredump_handler & CORE_DUMP_CLEAR_MASK)
polar_remove_old_core_files();
if (polar_enable_coredump_handler & CORE_DUMP_PRINT_MASK)
polar_transfer_core_tmp_to_log_with_remove();
}
/*
* POLAR: remove old system core files
* These files are ordered by create-time.
*/
static void
polar_remove_old_core_files(void)
{
struct dirent *corede;
struct stat st;
polar_file_descriptor *file_set;
char current_core_path[MAXPGPATH] = {0};
char core_file_path[MAXPGPATH] = {0};
DIR *coredir;
int num_core_files = 0;
int num_alloc_files = POLAR_NUM_FILE_INTERVAL;
time_t timestamp;
/* get the core-file path from the system core-pattern file */
if (!polar_read_core_pattern(POLAR_CORE_DUMP_PATTERN_FILE, core_file_path))
snprintf(core_file_path, MAXPGPATH, "%s", polar_core_file_path);
/* traverse the core file directory */
time(&timestamp);
file_set = (polar_file_descriptor *) palloc(num_alloc_files * sizeof(polar_file_descriptor));
coredir = AllocateDir(core_file_path, false);
while ((corede = ReadDir(coredir, core_file_path)) != NULL)
{
snprintf(current_core_path, MAXPGPATH, "%s/%s", core_file_path, corede->d_name);
/* Skip special stuff */
if (strcmp(corede->d_name, ".") == 0 || strcmp(corede->d_name, "..") == 0)
continue;
/* skip file could not open valid */
if (stat(current_core_path, &st))
continue;
/* skip file not belong to itself */
if (st.st_uid != geteuid())
continue;
/* skip dir */
if (S_ISDIR(st.st_mode))
continue;
/* skip file name invalid */
if (strncmp(corede->d_name, core_name_suffix, strlen(core_name_suffix)))
continue;
/* clean file when the file is out of date */
if (core_file_outdate_time >= 0 && timestamp - st.st_mtime >= core_file_outdate_time)
{
elog(DEBUG2, "attempting to remove core file %s", current_core_path);
polar_remove_file(current_core_path);
}
/* expand the file set when necessary */
if (num_core_files >= num_alloc_files)
{
num_alloc_files += POLAR_NUM_FILE_INTERVAL;
file_set = (polar_file_descriptor *) repalloc(file_set,
num_alloc_files * sizeof(polar_file_descriptor));
}
/* save [m_time, file_name] to file_set */
file_set[num_core_files].mtime = st.st_mtime;
strncpy(file_set[num_core_files].file_path, current_core_path, MAXPGPATH);
num_core_files++;
}
FreeDir(coredir);
if (num_core_files > 0)
{
int i = 0;
qsort(file_set, num_core_files, sizeof(polar_file_descriptor), cmp_file_mtime);
for (i = 0; i < num_core_files; i++)
{
if (i >= num_corefile_reserved_old &&
i < num_core_files - num_corefile_reserved_new)
{
elog(DEBUG2, "attempting to remove core file %s", file_set[i].file_path);
polar_remove_file(file_set[i].file_path);
}
}
}
pfree(file_set);
}
/*
* POLAR: transfer temp core files to log and remove it.
*/
static void
polar_transfer_core_tmp_to_log_with_remove(void)
{
DIR *coredir;
struct dirent *corede;
struct stat st;
char current_core_path[MAXPGPATH] = {0};
char *current_dir = ".";
/* search core stack info in the PG DIR */
coredir = AllocateDir(current_dir, false);
while ((corede = ReadDir(coredir, current_dir)) != NULL)
{
snprintf(current_core_path, MAXPGPATH, "%s/%s", current_dir, corede->d_name);
/* Skip special stuff */
if (strcmp(corede->d_name, ".") == 0 || strcmp(corede->d_name, "..") == 0)
continue;
/* skip file could not open valid */
if (stat(current_core_path, &st))
continue;
/* skip file not belong to itself */
if (getuid() != geteuid() || st.st_uid != geteuid())
continue;
/* skip file name invalid and skip dir */
if (strncmp(corede->d_name, POLAR_CORE_DUMP_FILE_SUFFIX, strlen(POLAR_CORE_DUMP_FILE_SUFFIX)) || S_ISDIR(st.st_mode))
continue;
/* could open file valid */
if (polar_transfer_coretmp_to_log(current_core_path))
{
/* delete temp file */
polar_remove_file(current_core_path);
}
}
FreeDir(coredir);
}
/*
* POLAR: remove file function
* The fucntion is same as drop of replication slot on disk.
*/
static bool
polar_remove_file(const char *path)
{
int rc = 0;
#ifdef WIN32
char newpath[MAXPGPATH] = {0};
snprintf(newpath, MAXPGPATH, "%s.deleted", path);
if (rename(path, newpath) != 0)
{
ereport(LOG,
(errcode_for_file_access(),
errmsg("could not rename deleted file \"%s\": %m",
path)));
return;
}
rc = unlink(newpath);
#else
rc = unlink(path);
#endif
if (rc != 0)
{
ereport(LOG,
(errcode_for_file_access(),
errmsg("could not remove old file \"%s\": %m",
path)));
return false;
}
return true;
}
/*
* POLAR: transfer temp core files to LOG output
* 1. get stacktrace info in temp core files
* 2. print stack info to LOG using add2line
*/
static bool
polar_transfer_coretmp_to_log(const char *path)
{
StackInfoOnDisk stack_info;
int readBytes = 0;
int i = 0, fd = 0;
/* open and read file */
fd = open(path, O_RDONLY | PG_BINARY);
if (fd < 0)
return false;
readBytes = read(fd, &stack_info, sizeof(stack_info));
if (readBytes != sizeof(stack_info) ||
stack_info.magic_number != POLAR_CORE_MAGIC_NUMBER)
{
int saved_errno = errno;
close(fd);
errno = saved_errno;
ereport(WARNING,
(errcode_for_file_access(),
errmsg("could not read file \"%s\", read %d of %u: %m",
path, readBytes,
(uint32) sizeof(stack_info))));
return true;
}
close(fd);
/* print Stack Info using add2line */
for (i = 0; i < stack_info.stack_size && i < POLAR_MAX_STACK_FRAMES; i++)
{
if (!polar_address_to_stack_info(stack_info.stack_traces[i], i))
return false;
}
return true;
}
/* POLAR: print stack info by address */
static bool
polar_address_to_stack_info(void const *addr, int line_number)
{
int maps_column_num = 0;
void *offset_start = NULL;
void *offset_end = NULL;
FILE *fd_maps = NULL;
/* initialize string from /proc/pid/maps */
char library_path[MAXPGPATH] = {0};
char proc_map_path[MAXPGPATH] = {0};
char buf[POLAR_MAX_BUFF_SIZE] = {0};
char maps_line[POLAR_MAX_BUFF_SIZE] = {0};
/* open file /proc/pid/maps */
sprintf(proc_map_path, "/proc/%d/maps", MyProcPid);
fd_maps = fopen(proc_map_path, "r");
if (NULL == fd_maps)
return false;
/* get the start address, end address and file path from /proc/pid/maps */
while (fgets(maps_line, sizeof(maps_line), fd_maps) != NULL)
{
maps_column_num = sscanf(maps_line, "%p-%p\t%*s\t%*s\t%*s\t%*s\t%s",
&offset_start, &offset_end, library_path);
if(maps_column_num == 3 && addr >= offset_start && addr <= offset_end)
break;
}
fclose(fd_maps);
/*
* POLAR: error case
* It is the oppsite case to find the right address.
* Note that, when maps_column_num == 3, offset_start and offset_end can't be null.
*/
if (maps_column_num != 3 || addr < offset_start || addr > offset_end)
return false;
/* get the real address for the .so file by offset_start */
if (strcmp(my_exec_path, library_path))
addr = (void *)((unsigned long) addr - (unsigned long) offset_start);
/* get the stack info */
snprintf(buf, POLAR_MAX_BUFF_SIZE, "%s", polar_addr2line(library_path, addr));
/* print the info if it is valid */
elog(LOG, "[Stack Info] line : %d, file %s, info : %s", line_number, library_path, buf);
return true;
}
/*
* POLAR: get stack info using os function add2line
*/
static char *
polar_addr2line(char const *const program_name, void const *const addr)
{
FILE *fpRead;
static char buf[POLAR_MAX_BUFF_SIZE];
char addr2line_cmd[POLAR_MAX_BUFF_SIZE] = {0};
char *ret = NULL;
memset(buf, '\0', sizeof(buf));
sprintf(addr2line_cmd, "addr2line -f -p -e %.256s %p", program_name, addr);
fpRead = popen(addr2line_cmd, "r");
if (fpRead == NULL)
return buf;
ret = fgets(buf, POLAR_MAX_BUFF_SIZE, fpRead);
if (ret == NULL)
{
/* buf may not be null after fgets */
memset(buf, '\0', sizeof(buf));
return buf;
}
if (buf[strlen(buf) - 1] == '\n')
buf[strlen(buf) - 1] = '\0';
if (fpRead != NULL)
pclose(fpRead);
return buf;
}
/*
* Compare mtime of file in order to sort array in descending order.
*/
static int
cmp_file_mtime(const void *a, const void *b)
{
polar_file_descriptor st1 = *((const polar_file_descriptor *) a);
polar_file_descriptor st2 = *((const polar_file_descriptor *) b);
if (st1.mtime < st2.mtime)
return -1;
else if (st1.mtime == st2.mtime)
return 0;
else
return 1;
}
/*
* POLAR: remove xlog temp file if it is outdate.
*/
static void
polar_xlog_temp_file_clean(void)
{
DIR *waldir;
struct dirent *walde;
struct stat st;
char current_xlog_temp_file_path[MAXPGPATH] = {0};
char current_dir[MAXPGPATH] = {0};
time_t timestamp;
static time_t last_clean_timestamp;
time(&timestamp);
/* to scan disk periodically */
if (last_clean_timestamp && timestamp - last_clean_timestamp < xlog_temp_outdate_time)
return ;
last_clean_timestamp = timestamp;
/* search xlog info in the PG WAL DIR */
POLAR_FILE_PATH(current_dir, "pg_wal");
waldir = polar_allocate_dir(current_dir);
while ((walde = polar_readdir(waldir)) != NULL)
{
snprintf(current_xlog_temp_file_path, MAXPGPATH, "%s/%s", current_dir, walde->d_name);
/* Skip special stuff */
if (strcmp(walde->d_name, ".") == 0 || strcmp(walde->d_name, "..") == 0)
continue;
/* skip file could not open valid */
if (stat(current_xlog_temp_file_path, &st))
continue;
/* skip file not belong to itself */
if (st.st_uid != geteuid())
continue;
/* skip dir */
if (S_ISDIR(st.st_mode))
continue;
/* skip file name invalid */
if (strncmp(walde->d_name, POLAR_XLOG_TEMP_FILE_SUFFIX, strlen(POLAR_XLOG_TEMP_FILE_SUFFIX)))
continue;
/* delete file if it is outdate */
if (timestamp - st.st_mtime >= xlog_temp_outdate_time)
polar_unlink(current_xlog_temp_file_path);
}
FreeDir(waldir);
}
/*
* POLAR: Get the core-file path from the system core-pattern file
*/
bool
polar_read_core_pattern(const char *core_pattern_path, char *core_file_path)
{
int fd = 0;
char *ret = NULL;
struct stat st;
char cwd[MAXPGPATH] = {0};
char buf[MAXPGPATH] = {0};
/* open system file and read core pattern path */
fd = open(core_pattern_path, O_RDONLY | PG_BINARY);
if (fd < 0)
{
ereport(LOG, (errcode_for_file_access(),
errmsg("could not open file \"%s\".", core_pattern_path)));
return false;
}
if (read(fd, (char *) buf, sizeof(buf)) < 0)
{
ereport(LOG, (errcode_for_file_access(),
errmsg("could not read file \"%s\".", core_pattern_path)));
return false;
}
close(fd);
/* get the current working path */
if (!getcwd(cwd, MAXPGPATH))
{
elog(LOG, "could not determine current directory");
return false;
}
/* get the rightmost / if possible */
ret = strrchr(buf, '/');
/* first case: / did not exsit in path (e.g. core) the practical path is the current working path */
if (ret == NULL)
{
snprintf(core_file_path, MAXPGPATH, "%s", cwd);
return true;
}
/* get the core directory from the core pattern */
if (ret - buf + 1 > MAXPGPATH)
{
elog(LOG, "core pattern length exceeds MAXPGPATH");
return false;
}
buf[ret - buf + 1] = '\0';
/* second case: absolute path (e.g. /tmp/corefile/core) */
if (buf[0] == '/')
snprintf(core_file_path, MAXPGPATH, "%s", buf);
else
{
/* third case: relative path (e.g. corefile/core) */
snprintf(core_file_path, MAXPGPATH, "%s/%s", cwd, buf);
}
/* judge whether core_file_path could be opened valid */
if (stat(core_file_path, &st))
return false;
return true;
}
Reference in New Issue View Git Blame Copy Permalink