sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-02-23 09:57:40 +03:00
Code

util: New functions pidfile_path_create(), pidfile_fd_close()

Browse Source 
Uses the core of CTDB's create_pidfile_context() for
pidfile_path_create(). pidfile_fd_close() is a subset of CTDB's
pidfile_context_destructor().

Signed-off-by: Martin Schwenke <martin@meltin.net>
Reviewed-by: Volker Lendecke <vl@samba.org>
This commit is contained in:
Martin Schwenke 2017-07-31 15:11:33 +10:00 committed by Martin Schwenke
parent 59ebb29e6a
commit 411b7c8753
2 changed files with 102 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
lib/util/pidfile.c
View File

@ -2,6 +2,7 @@
Unix SMB/CIFS implementation.
pidfile handling
Copyright (C) Andrew Tridgell 1998
Copyright (C) Amitay Isaccs 2016
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
@ -31,6 +32,104 @@
* @brief Pid file handling
*/
int pidfile_path_create(const char *path, int *outfd)
{
struct flock lck;
char tmp[64] = { 0 };
pid_t pid;
int fd, ret = 0;
int len;
ssize_t nwritten;
pid = getpid();
fd = open(path, O_CREAT|O_WRONLY|O_NONBLOCK, 0644);
if (fd == -1) {
return errno;
}
if (! set_close_on_exec(fd)) {
close(fd);
return EIO;
}
lck = (struct flock) {
.l_type = F_WRLCK,
.l_whence = SEEK_SET,
};
do {
ret = fcntl(fd, F_SETLK, &lck);
} while ((ret == -1) && (errno == EINTR));
if (ret != 0) {
ret = errno;
close(fd);
return ret;
}
/*
* PID file is locked by us so from here on we should unlink
* on failure
*/
do {
ret = ftruncate(fd, 0);
} while ((ret == -1) && (errno == EINTR));
if (ret == -1) {
ret = EIO;
goto fail_unlink;
}
len = snprintf(tmp, sizeof(tmp), "%u\n", pid);
if (len < 0) {
ret = errno;
goto fail_unlink;
}
if (len >= sizeof(tmp)) {
ret = ENOSPC;
goto fail_unlink;
}
do {
nwritten = write(fd, tmp, len);
} while ((nwritten == -1) && (errno == EINTR));
if ((nwritten == -1) || (nwritten != len)) {
ret = EIO;
goto fail_unlink;
}
if (outfd != NULL) {
*outfd = fd;
}
return 0;
fail_unlink:
unlink(path);
close(fd);
return ret;
}
void pidfile_fd_close(int fd)
{
struct flock lck = {
.l_type = F_UNLCK,
.l_whence = SEEK_SET,
};
int ret;
do {
ret = fcntl(fd, F_SETLK, &lck);
} while ((ret == -1) && (errno == EINTR));
do {
ret = close(fd);
} while ((ret == -1) && (errno == EINTR));
}
/**
* return the pid in a pidfile. return 0 if the process (or pidfile)
* does not exist

3
lib/util/pidfile.h
View File

@ -20,6 +20,9 @@
#ifndef _SAMBA_PIDFILE_H_
#define _SAMBA_PIDFILE_H_
int pidfile_path_create(const char *path, int *outfd);
void pidfile_fd_close(int fd);
pid_t pidfile_pid(const char *piddir, const char *name);
void pidfile_create(const char *piddir, const char *program_name);
void pidfile_unlink(const char *piddir, const char *program_name);