mirror of
https://git.proxmox.com/git/mirror_zfs.git
synced 2024-12-26 19:19:32 +03:00
68301ba20e
This commit adds two features to zed, that macOS desires. The first is that when you unload the kernel module, zed would enter into a cpubusy loop calling zfs_events_next() repeatedly. We now look for ENODEV, returned by kernel, so zed can exit gracefully. Second feature is -I (idle) (alas -P persist was taken) is for the deamon to; 1; if started without ZFS kernel module, stick around waiting for it. 2; if kernel module is unloaded, go back to 1. This is due to daemons in macOS is started by launchctl, and is expected to stick around. Currently, the busy loop only exists when errno is ENODEV. This is to ensure that functionality that upstream expects is not changed. It did not care about errors before, and it still does not. (with the exception of ENODEV). However, it is probably better that all errors (ERESTART notwithstanding) exits the loop, and the issues complaining about zed taking all CPU will go away. Reviewed-by: Brian Behlendorf <behlendorf1@llnl.gov> Signed-off-by: Jorgen Lundman <lundman@lundman.net> Closes #10476
736 lines
18 KiB
C
736 lines
18 KiB
C
/*
|
|
* This file is part of the ZFS Event Daemon (ZED)
|
|
* for ZFS on Linux (ZoL) <http://zfsonlinux.org/>.
|
|
* Developed at Lawrence Livermore National Laboratory (LLNL-CODE-403049).
|
|
* Copyright (C) 2013-2014 Lawrence Livermore National Security, LLC.
|
|
* Refer to the ZoL git commit log for authoritative copyright attribution.
|
|
*
|
|
* The contents of this file are subject to the terms of the
|
|
* Common Development and Distribution License Version 1.0 (CDDL-1.0).
|
|
* You can obtain a copy of the license from the top-level file
|
|
* "OPENSOLARIS.LICENSE" or at <http://opensource.org/licenses/CDDL-1.0>.
|
|
* You may not use this file except in compliance with the license.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <ctype.h>
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <libgen.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/uio.h>
|
|
#include <unistd.h>
|
|
#include "zed.h"
|
|
#include "zed_conf.h"
|
|
#include "zed_file.h"
|
|
#include "zed_log.h"
|
|
#include "zed_strings.h"
|
|
|
|
/*
|
|
* Return a new configuration with default values.
|
|
*/
|
|
struct zed_conf *
|
|
zed_conf_create(void)
|
|
{
|
|
struct zed_conf *zcp;
|
|
|
|
zcp = calloc(1, sizeof (*zcp));
|
|
if (!zcp)
|
|
goto nomem;
|
|
|
|
zcp->syslog_facility = LOG_DAEMON;
|
|
zcp->min_events = ZED_MIN_EVENTS;
|
|
zcp->max_events = ZED_MAX_EVENTS;
|
|
zcp->pid_fd = -1;
|
|
zcp->zedlets = NULL; /* created via zed_conf_scan_dir() */
|
|
zcp->state_fd = -1; /* opened via zed_conf_open_state() */
|
|
zcp->zfs_hdl = NULL; /* opened via zed_event_init() */
|
|
zcp->zevent_fd = -1; /* opened via zed_event_init() */
|
|
|
|
if (!(zcp->conf_file = strdup(ZED_CONF_FILE)))
|
|
goto nomem;
|
|
|
|
if (!(zcp->pid_file = strdup(ZED_PID_FILE)))
|
|
goto nomem;
|
|
|
|
if (!(zcp->zedlet_dir = strdup(ZED_ZEDLET_DIR)))
|
|
goto nomem;
|
|
|
|
if (!(zcp->state_file = strdup(ZED_STATE_FILE)))
|
|
goto nomem;
|
|
|
|
return (zcp);
|
|
|
|
nomem:
|
|
zed_log_die("Failed to create conf: %s", strerror(errno));
|
|
return (NULL);
|
|
}
|
|
|
|
/*
|
|
* Destroy the configuration [zcp].
|
|
*
|
|
* Note: zfs_hdl & zevent_fd are destroyed via zed_event_fini().
|
|
*/
|
|
void
|
|
zed_conf_destroy(struct zed_conf *zcp)
|
|
{
|
|
if (!zcp)
|
|
return;
|
|
|
|
if (zcp->state_fd >= 0) {
|
|
if (close(zcp->state_fd) < 0)
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to close state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
zcp->state_fd = -1;
|
|
}
|
|
if (zcp->pid_file) {
|
|
if ((unlink(zcp->pid_file) < 0) && (errno != ENOENT))
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to remove PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
}
|
|
if (zcp->pid_fd >= 0) {
|
|
if (close(zcp->pid_fd) < 0)
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to close PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
zcp->pid_fd = -1;
|
|
}
|
|
if (zcp->conf_file) {
|
|
free(zcp->conf_file);
|
|
zcp->conf_file = NULL;
|
|
}
|
|
if (zcp->pid_file) {
|
|
free(zcp->pid_file);
|
|
zcp->pid_file = NULL;
|
|
}
|
|
if (zcp->zedlet_dir) {
|
|
free(zcp->zedlet_dir);
|
|
zcp->zedlet_dir = NULL;
|
|
}
|
|
if (zcp->state_file) {
|
|
free(zcp->state_file);
|
|
zcp->state_file = NULL;
|
|
}
|
|
if (zcp->zedlets) {
|
|
zed_strings_destroy(zcp->zedlets);
|
|
zcp->zedlets = NULL;
|
|
}
|
|
free(zcp);
|
|
}
|
|
|
|
/*
|
|
* Display command-line help and exit.
|
|
*
|
|
* If [got_err] is 0, output to stdout and exit normally;
|
|
* otherwise, output to stderr and exit with a failure status.
|
|
*/
|
|
static void
|
|
_zed_conf_display_help(const char *prog, int got_err)
|
|
{
|
|
FILE *fp = got_err ? stderr : stdout;
|
|
int w1 = 4; /* width of leading whitespace */
|
|
int w2 = 8; /* width of L-justified option field */
|
|
|
|
fprintf(fp, "Usage: %s [OPTION]...\n", (prog ? prog : "zed"));
|
|
fprintf(fp, "\n");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-h",
|
|
"Display help.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-L",
|
|
"Display license information.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-V",
|
|
"Display version information.");
|
|
fprintf(fp, "\n");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-v",
|
|
"Be verbose.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-f",
|
|
"Force daemon to run.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-F",
|
|
"Run daemon in the foreground.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-I",
|
|
"Idle daemon until kernel module is (re)loaded.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-M",
|
|
"Lock all pages in memory.");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-P",
|
|
"$PATH for ZED to use (only used by ZTS).");
|
|
fprintf(fp, "%*c%*s %s\n", w1, 0x20, -w2, "-Z",
|
|
"Zero state file.");
|
|
fprintf(fp, "\n");
|
|
#if 0
|
|
fprintf(fp, "%*c%*s %s [%s]\n", w1, 0x20, -w2, "-c FILE",
|
|
"Read configuration from FILE.", ZED_CONF_FILE);
|
|
#endif
|
|
fprintf(fp, "%*c%*s %s [%s]\n", w1, 0x20, -w2, "-d DIR",
|
|
"Read enabled ZEDLETs from DIR.", ZED_ZEDLET_DIR);
|
|
fprintf(fp, "%*c%*s %s [%s]\n", w1, 0x20, -w2, "-p FILE",
|
|
"Write daemon's PID to FILE.", ZED_PID_FILE);
|
|
fprintf(fp, "%*c%*s %s [%s]\n", w1, 0x20, -w2, "-s FILE",
|
|
"Write daemon's state to FILE.", ZED_STATE_FILE);
|
|
fprintf(fp, "\n");
|
|
|
|
exit(got_err ? EXIT_FAILURE : EXIT_SUCCESS);
|
|
}
|
|
|
|
/*
|
|
* Display license information to stdout and exit.
|
|
*/
|
|
static void
|
|
_zed_conf_display_license(void)
|
|
{
|
|
const char **pp;
|
|
const char *text[] = {
|
|
"The ZFS Event Daemon (ZED) is distributed under the terms of the",
|
|
" Common Development and Distribution License (CDDL-1.0)",
|
|
" <http://opensource.org/licenses/CDDL-1.0>.",
|
|
"",
|
|
"Developed at Lawrence Livermore National Laboratory"
|
|
" (LLNL-CODE-403049).",
|
|
"",
|
|
NULL
|
|
};
|
|
|
|
for (pp = text; *pp; pp++)
|
|
printf("%s\n", *pp);
|
|
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
/*
|
|
* Display version information to stdout and exit.
|
|
*/
|
|
static void
|
|
_zed_conf_display_version(void)
|
|
{
|
|
printf("%s-%s-%s\n",
|
|
ZFS_META_NAME, ZFS_META_VERSION, ZFS_META_RELEASE);
|
|
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
/*
|
|
* Copy the [path] string to the [resultp] ptr.
|
|
* If [path] is not an absolute path, prefix it with the current working dir.
|
|
* If [resultp] is non-null, free its existing string before assignment.
|
|
*/
|
|
static void
|
|
_zed_conf_parse_path(char **resultp, const char *path)
|
|
{
|
|
char buf[PATH_MAX];
|
|
|
|
assert(resultp != NULL);
|
|
assert(path != NULL);
|
|
|
|
if (*resultp)
|
|
free(*resultp);
|
|
|
|
if (path[0] == '/') {
|
|
*resultp = strdup(path);
|
|
} else if (!getcwd(buf, sizeof (buf))) {
|
|
zed_log_die("Failed to get current working dir: %s",
|
|
strerror(errno));
|
|
} else if (strlcat(buf, "/", sizeof (buf)) >= sizeof (buf)) {
|
|
zed_log_die("Failed to copy path: %s", strerror(ENAMETOOLONG));
|
|
} else if (strlcat(buf, path, sizeof (buf)) >= sizeof (buf)) {
|
|
zed_log_die("Failed to copy path: %s", strerror(ENAMETOOLONG));
|
|
} else {
|
|
*resultp = strdup(buf);
|
|
}
|
|
if (!*resultp)
|
|
zed_log_die("Failed to copy path: %s", strerror(ENOMEM));
|
|
}
|
|
|
|
/*
|
|
* Parse the command-line options into the configuration [zcp].
|
|
*/
|
|
void
|
|
zed_conf_parse_opts(struct zed_conf *zcp, int argc, char **argv)
|
|
{
|
|
const char * const opts = ":hLVc:d:p:P:s:vfFMZI";
|
|
int opt;
|
|
|
|
if (!zcp || !argv || !argv[0])
|
|
zed_log_die("Failed to parse options: Internal error");
|
|
|
|
opterr = 0; /* suppress default getopt err msgs */
|
|
|
|
while ((opt = getopt(argc, argv, opts)) != -1) {
|
|
switch (opt) {
|
|
case 'h':
|
|
_zed_conf_display_help(argv[0], EXIT_SUCCESS);
|
|
break;
|
|
case 'L':
|
|
_zed_conf_display_license();
|
|
break;
|
|
case 'V':
|
|
_zed_conf_display_version();
|
|
break;
|
|
case 'c':
|
|
_zed_conf_parse_path(&zcp->conf_file, optarg);
|
|
break;
|
|
case 'd':
|
|
_zed_conf_parse_path(&zcp->zedlet_dir, optarg);
|
|
break;
|
|
case 'I':
|
|
zcp->do_idle = 1;
|
|
break;
|
|
case 'p':
|
|
_zed_conf_parse_path(&zcp->pid_file, optarg);
|
|
break;
|
|
case 'P':
|
|
_zed_conf_parse_path(&zcp->path, optarg);
|
|
break;
|
|
case 's':
|
|
_zed_conf_parse_path(&zcp->state_file, optarg);
|
|
break;
|
|
case 'v':
|
|
zcp->do_verbose = 1;
|
|
break;
|
|
case 'f':
|
|
zcp->do_force = 1;
|
|
break;
|
|
case 'F':
|
|
zcp->do_foreground = 1;
|
|
break;
|
|
case 'M':
|
|
zcp->do_memlock = 1;
|
|
break;
|
|
case 'Z':
|
|
zcp->do_zero = 1;
|
|
break;
|
|
case '?':
|
|
default:
|
|
if (optopt == '?')
|
|
_zed_conf_display_help(argv[0], EXIT_SUCCESS);
|
|
|
|
fprintf(stderr, "%s: %s '-%c'\n\n", argv[0],
|
|
"Invalid option", optopt);
|
|
_zed_conf_display_help(argv[0], EXIT_FAILURE);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Parse the configuration file into the configuration [zcp].
|
|
*
|
|
* FIXME: Not yet implemented.
|
|
*/
|
|
void
|
|
zed_conf_parse_file(struct zed_conf *zcp)
|
|
{
|
|
if (!zcp)
|
|
zed_log_die("Failed to parse config: %s", strerror(EINVAL));
|
|
}
|
|
|
|
/*
|
|
* Scan the [zcp] zedlet_dir for files to exec based on the event class.
|
|
* Files must be executable by user, but not writable by group or other.
|
|
* Dotfiles are ignored.
|
|
*
|
|
* Return 0 on success with an updated set of zedlets,
|
|
* or -1 on error with errno set.
|
|
*
|
|
* FIXME: Check if zedlet_dir and all parent dirs are secure.
|
|
*/
|
|
int
|
|
zed_conf_scan_dir(struct zed_conf *zcp)
|
|
{
|
|
zed_strings_t *zedlets;
|
|
DIR *dirp;
|
|
struct dirent *direntp;
|
|
char pathname[PATH_MAX];
|
|
struct stat st;
|
|
int n;
|
|
|
|
if (!zcp) {
|
|
errno = EINVAL;
|
|
zed_log_msg(LOG_ERR, "Failed to scan zedlet dir: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
zedlets = zed_strings_create();
|
|
if (!zedlets) {
|
|
errno = ENOMEM;
|
|
zed_log_msg(LOG_WARNING, "Failed to scan dir \"%s\": %s",
|
|
zcp->zedlet_dir, strerror(errno));
|
|
return (-1);
|
|
}
|
|
dirp = opendir(zcp->zedlet_dir);
|
|
if (!dirp) {
|
|
int errno_bak = errno;
|
|
zed_log_msg(LOG_WARNING, "Failed to open dir \"%s\": %s",
|
|
zcp->zedlet_dir, strerror(errno));
|
|
zed_strings_destroy(zedlets);
|
|
errno = errno_bak;
|
|
return (-1);
|
|
}
|
|
while ((direntp = readdir(dirp))) {
|
|
if (direntp->d_name[0] == '.')
|
|
continue;
|
|
|
|
n = snprintf(pathname, sizeof (pathname),
|
|
"%s/%s", zcp->zedlet_dir, direntp->d_name);
|
|
if ((n < 0) || (n >= sizeof (pathname))) {
|
|
zed_log_msg(LOG_WARNING, "Failed to stat \"%s\": %s",
|
|
direntp->d_name, strerror(ENAMETOOLONG));
|
|
continue;
|
|
}
|
|
if (stat(pathname, &st) < 0) {
|
|
zed_log_msg(LOG_WARNING, "Failed to stat \"%s\": %s",
|
|
pathname, strerror(errno));
|
|
continue;
|
|
}
|
|
if (!S_ISREG(st.st_mode)) {
|
|
zed_log_msg(LOG_INFO,
|
|
"Ignoring \"%s\": not a regular file",
|
|
direntp->d_name);
|
|
continue;
|
|
}
|
|
if ((st.st_uid != 0) && !zcp->do_force) {
|
|
zed_log_msg(LOG_NOTICE,
|
|
"Ignoring \"%s\": not owned by root",
|
|
direntp->d_name);
|
|
continue;
|
|
}
|
|
if (!(st.st_mode & S_IXUSR)) {
|
|
zed_log_msg(LOG_INFO,
|
|
"Ignoring \"%s\": not executable by user",
|
|
direntp->d_name);
|
|
continue;
|
|
}
|
|
if ((st.st_mode & S_IWGRP) && !zcp->do_force) {
|
|
zed_log_msg(LOG_NOTICE,
|
|
"Ignoring \"%s\": writable by group",
|
|
direntp->d_name);
|
|
continue;
|
|
}
|
|
if ((st.st_mode & S_IWOTH) && !zcp->do_force) {
|
|
zed_log_msg(LOG_NOTICE,
|
|
"Ignoring \"%s\": writable by other",
|
|
direntp->d_name);
|
|
continue;
|
|
}
|
|
if (zed_strings_add(zedlets, NULL, direntp->d_name) < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to register \"%s\": %s",
|
|
direntp->d_name, strerror(errno));
|
|
continue;
|
|
}
|
|
if (zcp->do_verbose)
|
|
zed_log_msg(LOG_INFO,
|
|
"Registered zedlet \"%s\"", direntp->d_name);
|
|
}
|
|
if (closedir(dirp) < 0) {
|
|
int errno_bak = errno;
|
|
zed_log_msg(LOG_WARNING, "Failed to close dir \"%s\": %s",
|
|
zcp->zedlet_dir, strerror(errno));
|
|
zed_strings_destroy(zedlets);
|
|
errno = errno_bak;
|
|
return (-1);
|
|
}
|
|
if (zcp->zedlets)
|
|
zed_strings_destroy(zcp->zedlets);
|
|
|
|
zcp->zedlets = zedlets;
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* Write the PID file specified in [zcp].
|
|
* Return 0 on success, -1 on error.
|
|
*
|
|
* This must be called after fork()ing to become a daemon (so the correct PID
|
|
* is recorded), but before daemonization is complete and the parent process
|
|
* exits (for synchronization with systemd).
|
|
*/
|
|
int
|
|
zed_conf_write_pid(struct zed_conf *zcp)
|
|
{
|
|
const mode_t dirmode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH;
|
|
const mode_t filemode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
|
|
char buf[PATH_MAX];
|
|
int n;
|
|
char *p;
|
|
mode_t mask;
|
|
int rv;
|
|
|
|
if (!zcp || !zcp->pid_file) {
|
|
errno = EINVAL;
|
|
zed_log_msg(LOG_ERR, "Failed to create PID file: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
assert(zcp->pid_fd == -1);
|
|
/*
|
|
* Create PID file directory if needed.
|
|
*/
|
|
n = strlcpy(buf, zcp->pid_file, sizeof (buf));
|
|
if (n >= sizeof (buf)) {
|
|
errno = ENAMETOOLONG;
|
|
zed_log_msg(LOG_ERR, "Failed to create PID file: %s",
|
|
strerror(errno));
|
|
goto err;
|
|
}
|
|
p = strrchr(buf, '/');
|
|
if (p)
|
|
*p = '\0';
|
|
|
|
if ((mkdirp(buf, dirmode) < 0) && (errno != EEXIST)) {
|
|
zed_log_msg(LOG_ERR, "Failed to create directory \"%s\": %s",
|
|
buf, strerror(errno));
|
|
goto err;
|
|
}
|
|
/*
|
|
* Obtain PID file lock.
|
|
*/
|
|
mask = umask(0);
|
|
umask(mask | 022);
|
|
zcp->pid_fd = open(zcp->pid_file, (O_RDWR | O_CREAT), filemode);
|
|
umask(mask);
|
|
if (zcp->pid_fd < 0) {
|
|
zed_log_msg(LOG_ERR, "Failed to open PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
goto err;
|
|
}
|
|
rv = zed_file_lock(zcp->pid_fd);
|
|
if (rv < 0) {
|
|
zed_log_msg(LOG_ERR, "Failed to lock PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
goto err;
|
|
} else if (rv > 0) {
|
|
pid_t pid = zed_file_is_locked(zcp->pid_fd);
|
|
if (pid < 0) {
|
|
zed_log_msg(LOG_ERR,
|
|
"Failed to test lock on PID file \"%s\"",
|
|
zcp->pid_file);
|
|
} else if (pid > 0) {
|
|
zed_log_msg(LOG_ERR,
|
|
"Found PID %d bound to PID file \"%s\"",
|
|
pid, zcp->pid_file);
|
|
} else {
|
|
zed_log_msg(LOG_ERR,
|
|
"Inconsistent lock state on PID file \"%s\"",
|
|
zcp->pid_file);
|
|
}
|
|
goto err;
|
|
}
|
|
/*
|
|
* Write PID file.
|
|
*/
|
|
n = snprintf(buf, sizeof (buf), "%d\n", (int)getpid());
|
|
if ((n < 0) || (n >= sizeof (buf))) {
|
|
errno = ERANGE;
|
|
zed_log_msg(LOG_ERR, "Failed to write PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
} else if (zed_file_write_n(zcp->pid_fd, buf, n) != n) {
|
|
zed_log_msg(LOG_ERR, "Failed to write PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
} else if (fdatasync(zcp->pid_fd) < 0) {
|
|
zed_log_msg(LOG_ERR, "Failed to sync PID file \"%s\": %s",
|
|
zcp->pid_file, strerror(errno));
|
|
} else {
|
|
return (0);
|
|
}
|
|
|
|
err:
|
|
if (zcp->pid_fd >= 0) {
|
|
(void) close(zcp->pid_fd);
|
|
zcp->pid_fd = -1;
|
|
}
|
|
return (-1);
|
|
}
|
|
|
|
/*
|
|
* Open and lock the [zcp] state_file.
|
|
* Return 0 on success, -1 on error.
|
|
*
|
|
* FIXME: Move state information into kernel.
|
|
*/
|
|
int
|
|
zed_conf_open_state(struct zed_conf *zcp)
|
|
{
|
|
char dirbuf[PATH_MAX];
|
|
mode_t dirmode = S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH;
|
|
int n;
|
|
char *p;
|
|
int rv;
|
|
|
|
if (!zcp || !zcp->state_file) {
|
|
errno = EINVAL;
|
|
zed_log_msg(LOG_ERR, "Failed to open state file: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
n = strlcpy(dirbuf, zcp->state_file, sizeof (dirbuf));
|
|
if (n >= sizeof (dirbuf)) {
|
|
errno = ENAMETOOLONG;
|
|
zed_log_msg(LOG_WARNING, "Failed to open state file: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
p = strrchr(dirbuf, '/');
|
|
if (p)
|
|
*p = '\0';
|
|
|
|
if ((mkdirp(dirbuf, dirmode) < 0) && (errno != EEXIST)) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to create directory \"%s\": %s",
|
|
dirbuf, strerror(errno));
|
|
return (-1);
|
|
}
|
|
if (zcp->state_fd >= 0) {
|
|
if (close(zcp->state_fd) < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to close state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
}
|
|
}
|
|
if (zcp->do_zero)
|
|
(void) unlink(zcp->state_file);
|
|
|
|
zcp->state_fd = open(zcp->state_file,
|
|
(O_RDWR | O_CREAT), (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH));
|
|
if (zcp->state_fd < 0) {
|
|
zed_log_msg(LOG_WARNING, "Failed to open state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
}
|
|
rv = zed_file_lock(zcp->state_fd);
|
|
if (rv < 0) {
|
|
zed_log_msg(LOG_WARNING, "Failed to lock state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
}
|
|
if (rv > 0) {
|
|
pid_t pid = zed_file_is_locked(zcp->state_fd);
|
|
if (pid < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to test lock on state file \"%s\"",
|
|
zcp->state_file);
|
|
} else if (pid > 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Found PID %d bound to state file \"%s\"",
|
|
pid, zcp->state_file);
|
|
} else {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Inconsistent lock state on state file \"%s\"",
|
|
zcp->state_file);
|
|
}
|
|
return (-1);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* Read the opened [zcp] state_file to obtain the eid & etime of the last event
|
|
* processed. Write the state from the last event to the [eidp] & [etime] args
|
|
* passed by reference. Note that etime[] is an array of size 2.
|
|
* Return 0 on success, -1 on error.
|
|
*/
|
|
int
|
|
zed_conf_read_state(struct zed_conf *zcp, uint64_t *eidp, int64_t etime[])
|
|
{
|
|
ssize_t len;
|
|
struct iovec iov[3];
|
|
ssize_t n;
|
|
|
|
if (!zcp || !eidp || !etime) {
|
|
errno = EINVAL;
|
|
zed_log_msg(LOG_ERR,
|
|
"Failed to read state file: %s", strerror(errno));
|
|
return (-1);
|
|
}
|
|
if (lseek(zcp->state_fd, 0, SEEK_SET) == (off_t)-1) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to reposition state file offset: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
len = 0;
|
|
iov[0].iov_base = eidp;
|
|
len += iov[0].iov_len = sizeof (*eidp);
|
|
iov[1].iov_base = &etime[0];
|
|
len += iov[1].iov_len = sizeof (etime[0]);
|
|
iov[2].iov_base = &etime[1];
|
|
len += iov[2].iov_len = sizeof (etime[1]);
|
|
|
|
n = readv(zcp->state_fd, iov, 3);
|
|
if (n == 0) {
|
|
*eidp = 0;
|
|
} else if (n < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to read state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
} else if (n != len) {
|
|
errno = EIO;
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to read state file \"%s\": Read %d of %d bytes",
|
|
zcp->state_file, n, len);
|
|
return (-1);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* Write the [eid] & [etime] of the last processed event to the opened
|
|
* [zcp] state_file. Note that etime[] is an array of size 2.
|
|
* Return 0 on success, -1 on error.
|
|
*/
|
|
int
|
|
zed_conf_write_state(struct zed_conf *zcp, uint64_t eid, int64_t etime[])
|
|
{
|
|
ssize_t len;
|
|
struct iovec iov[3];
|
|
ssize_t n;
|
|
|
|
if (!zcp) {
|
|
errno = EINVAL;
|
|
zed_log_msg(LOG_ERR,
|
|
"Failed to write state file: %s", strerror(errno));
|
|
return (-1);
|
|
}
|
|
if (lseek(zcp->state_fd, 0, SEEK_SET) == (off_t)-1) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to reposition state file offset: %s",
|
|
strerror(errno));
|
|
return (-1);
|
|
}
|
|
len = 0;
|
|
iov[0].iov_base = &eid;
|
|
len += iov[0].iov_len = sizeof (eid);
|
|
iov[1].iov_base = &etime[0];
|
|
len += iov[1].iov_len = sizeof (etime[0]);
|
|
iov[2].iov_base = &etime[1];
|
|
len += iov[2].iov_len = sizeof (etime[1]);
|
|
|
|
n = writev(zcp->state_fd, iov, 3);
|
|
if (n < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to write state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
}
|
|
if (n != len) {
|
|
errno = EIO;
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to write state file \"%s\": Wrote %d of %d bytes",
|
|
zcp->state_file, n, len);
|
|
return (-1);
|
|
}
|
|
if (fdatasync(zcp->state_fd) < 0) {
|
|
zed_log_msg(LOG_WARNING,
|
|
"Failed to sync state file \"%s\": %s",
|
|
zcp->state_file, strerror(errno));
|
|
return (-1);
|
|
}
|
|
return (0);
|
|
}
|