summaryrefslogtreecommitdiff
path: root/server/util
diff options
context:
space:
mode:
authorSimo Sorce <idra@samba.org>2008-11-25 16:42:38 -0500
committerSimo Sorce <idra@samba.org>2008-11-25 16:44:17 -0500
commitd4b9a588bade582ba2968f270101032a11981e17 (patch)
tree41e817fbc39d3b5b0310b5f0d8670ca91ad37783 /server/util
parentf85ae5fce84ad7bdc515c1ec324b9c56cb1a39b2 (diff)
downloadsssd-d4b9a588bade582ba2968f270101032a11981e17.tar.gz
sssd-d4b9a588bade582ba2968f270101032a11981e17.tar.bz2
sssd-d4b9a588bade582ba2968f270101032a11981e17.zip
Move all server helpers in util/server.c
Diffstat (limited to 'server/util')
-rw-r--r--server/util/become_daemon.c161
-rw-r--r--server/util/server.c400
2 files changed, 400 insertions, 161 deletions
diff --git a/server/util/become_daemon.c b/server/util/become_daemon.c
deleted file mode 100644
index 4a940eaa..00000000
--- a/server/util/become_daemon.c
+++ /dev/null
@@ -1,161 +0,0 @@
-/*
- Unix SMB/CIFS implementation.
- Samba utility functions
- Copyright (C) Andrew Tridgell 1992-1998
- Copyright (C) Jeremy Allison 2001-2002
- Copyright (C) Simo Sorce 2001
- Copyright (C) Jim McDonough (jmcd@us.ibm.com) 2003.
- Copyright (C) James J Myers 2003
-
- 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
- the Free Software Foundation; either version 3 of the License, or
- (at your option) any later version.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
-
- You should have received a copy of the GNU General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>.
-*/
-
-#define _GNU_SOURCE
-#include <stdio.h>
-#include <stdbool.h>
-#include <unistd.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <fcntl.h>
-#include "util/util.h"
-
-
-/*******************************************************************
- Close the low 3 fd's and open dev/null in their place.
-********************************************************************/
-static void close_low_fds(bool stderr_too)
-{
-#ifndef VALGRIND
- int fd;
- int i;
-
- close(0);
- close(1);
-
- if (stderr_too)
- close(2);
-
- /* try and use up these file descriptors, so silly
- library routines writing to stdout etc won't cause havoc */
- for (i=0;i<3;i++) {
- if (i == 2 && !stderr_too)
- continue;
-
- fd = open("/dev/null",O_RDWR,0);
- if (fd < 0)
- fd = open("/dev/null",O_WRONLY,0);
- if (fd < 0) {
- DEBUG(0,("Can't open /dev/null\n"));
- return;
- }
- if (fd != i) {
- DEBUG(0,("Didn't get file descriptor %d\n",i));
- return;
- }
- }
-#endif
-}
-
-/**
- Become a daemon, discarding the controlling terminal.
-**/
-
-void become_daemon(bool Fork)
-{
- if (Fork) {
- if (fork()) {
- _exit(0);
- }
- }
-
- /* detach from the terminal */
-#ifdef HAVE_SETSID
- setsid();
-#elif defined(TIOCNOTTY)
- {
- int i = open("/dev/tty", O_RDWR, 0);
- if (i != -1) {
- ioctl(i, (int) TIOCNOTTY, (char *)0);
- close(i);
- }
- }
-#endif /* HAVE_SETSID */
-
- /* Close fd's 0,1,2. Needed if started by rsh */
- close_low_fds(false); /* Don't close stderr, let the debug system
- attach it to the logfile */
-}
-
-int pidfile(const char *path, const char *name)
-{
- char pid_str[32];
- pid_t pid;
- char *file;
- int fd;
- int ret;
-
- asprintf(&file, "%s/%s.pid", path, name);
-
- fd = open(file, O_RDONLY, 0644);
- if (fd != -1) {
-
- pid_str[sizeof(pid_str) -1] = '\0';
- ret = read(fd, pid_str, sizeof(pid_str) -1);
- if (ret > 0) {
- /* let's check the pid */
-
- pid = (pid_t)atoi(pid_str);
- if (pid != 0) {
- errno = 0;
- ret = kill(pid, 0);
- if (ret != 0 && errno != ESRCH) {
- close(fd);
- free(file);
- return EEXIST;
- }
- }
- }
-
- /* notihng in the file or no process */
- close(fd);
- unlink(file);
-
- } else {
- if (errno != ENOENT) {
- free(file);
- return EIO;
- }
- }
-
- fd = open(file, O_CREAT | O_WRONLY | O_EXCL, 0644);
- if (fd == -1) {
- free(file);
- return EIO;
- }
- free(file);
-
- memset(pid_str, 0, sizeof(pid_str));
- snprintf(pid_str, sizeof(pid_str) -1, "%u\n", (unsigned int) getpid());
-
- ret = write(fd, pid_str, strlen(pid_str));
- if (ret != strlen(pid_str)) {
- close(fd);
- return EIO;
- }
-
- close(fd);
-
- return 0;
-}
-
diff --git a/server/util/server.c b/server/util/server.c
new file mode 100644
index 00000000..45a6948e
--- /dev/null
+++ b/server/util/server.c
@@ -0,0 +1,400 @@
+/*
+ SSSD
+
+ Servers setup routines
+
+ Copyright (C) Andrew Tridgell 1992-2005
+ Copyright (C) Martin Pool 2002
+ Copyright (C) Jelmer Vernooij 2002
+ Copyright (C) James J Myers 2003 <myersjj@samba.org>
+ Copyright (C) Simo Sorce 2008
+
+ 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
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdio.h>
+#include <stdbool.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include "../events/events.h"
+#include "../ldb/include/ldb.h"
+#include "confdb/confdb.h"
+#include "util/util.h"
+
+/*******************************************************************
+ Close the low 3 fd's and open dev/null in their place.
+********************************************************************/
+static void close_low_fds(bool stderr_too)
+{
+#ifndef VALGRIND
+ int fd;
+ int i;
+
+ close(0);
+ close(1);
+
+ if (stderr_too)
+ close(2);
+
+ /* try and use up these file descriptors, so silly
+ library routines writing to stdout etc won't cause havoc */
+ for (i=0;i<3;i++) {
+ if (i == 2 && !stderr_too)
+ continue;
+
+ fd = open("/dev/null",O_RDWR,0);
+ if (fd < 0)
+ fd = open("/dev/null",O_WRONLY,0);
+ if (fd < 0) {
+ DEBUG(0,("Can't open /dev/null\n"));
+ return;
+ }
+ if (fd != i) {
+ DEBUG(0,("Didn't get file descriptor %d\n",i));
+ return;
+ }
+ }
+#endif
+}
+
+/**
+ Become a daemon, discarding the controlling terminal.
+**/
+
+void become_daemon(bool Fork)
+{
+ if (Fork) {
+ if (fork()) {
+ _exit(0);
+ }
+ }
+
+ /* detach from the terminal */
+#ifdef HAVE_SETSID
+ setsid();
+#elif defined(TIOCNOTTY)
+ {
+ int i = open("/dev/tty", O_RDWR, 0);
+ if (i != -1) {
+ ioctl(i, (int) TIOCNOTTY, (char *)0);
+ close(i);
+ }
+ }
+#endif /* HAVE_SETSID */
+
+ /* Close fd's 0,1,2. Needed if started by rsh */
+ close_low_fds(false); /* Don't close stderr, let the debug system
+ attach it to the logfile */
+}
+
+int pidfile(const char *path, const char *name)
+{
+ char pid_str[32];
+ pid_t pid;
+ char *file;
+ int fd;
+ int ret;
+
+ asprintf(&file, "%s/%s.pid", path, name);
+
+ fd = open(file, O_RDONLY, 0644);
+ if (fd != -1) {
+
+ pid_str[sizeof(pid_str) -1] = '\0';
+ ret = read(fd, pid_str, sizeof(pid_str) -1);
+ if (ret > 0) {
+ /* let's check the pid */
+
+ pid = (pid_t)atoi(pid_str);
+ if (pid != 0) {
+ errno = 0;
+ ret = kill(pid, 0);
+ if (ret != 0 && errno != ESRCH) {
+ close(fd);
+ free(file);
+ return EEXIST;
+ }
+ }
+ }
+
+ /* notihng in the file or no process */
+ close(fd);
+ unlink(file);
+
+ } else {
+ if (errno != ENOENT) {
+ free(file);
+ return EIO;
+ }
+ }
+
+ fd = open(file, O_CREAT | O_WRONLY | O_EXCL, 0644);
+ if (fd == -1) {
+ free(file);
+ return EIO;
+ }
+ free(file);
+
+ memset(pid_str, 0, sizeof(pid_str));
+ snprintf(pid_str, sizeof(pid_str) -1, "%u\n", (unsigned int) getpid());
+
+ ret = write(fd, pid_str, strlen(pid_str));
+ if (ret != strlen(pid_str)) {
+ close(fd);
+ return EIO;
+ }
+
+ close(fd);
+
+ return 0;
+}
+
+static void sig_hup(int sig)
+{
+ /* cycle log/debug files */
+ return;
+}
+
+static void sig_term(int sig)
+{
+#if HAVE_GETPGRP
+ static int done_sigterm;
+ if (done_sigterm == 0 && getpgrp() == getpid()) {
+ DEBUG(0,("SIGTERM: killing children\n"));
+ done_sigterm = 1;
+ kill(-getpgrp(), SIGTERM);
+ }
+#endif
+ exit(0);
+}
+
+/*
+ setup signal masks
+*/
+static void setup_signals(void)
+{
+ /* we are never interested in SIGPIPE */
+ BlockSignals(true, SIGPIPE);
+
+#if defined(SIGFPE)
+ /* we are never interested in SIGFPE */
+ BlockSignals(true, SIGFPE);
+#endif
+
+ /* We are no longer interested in USR1 */
+ BlockSignals(true, SIGUSR1);
+
+#if defined(SIGUSR2)
+ /* We are no longer interested in USR2 */
+ BlockSignals(true, SIGUSR2);
+#endif
+
+ /* POSIX demands that signals are inherited. If the invoking process has
+ * these signals masked, we will have problems, as we won't recieve them. */
+ BlockSignals(false, SIGHUP);
+ BlockSignals(false, SIGTERM);
+
+ CatchSignal(SIGHUP, sig_hup);
+ CatchSignal(SIGTERM, sig_term);
+}
+
+/*
+ handle io on stdin
+*/
+static void server_stdin_handler(struct event_context *event_ctx, struct fd_event *fde,
+ uint16_t flags, void *private)
+{
+ const char *binary_name = (const char *)private;
+ uint8_t c;
+ if (read(0, &c, 1) == 0) {
+ DEBUG(0,("%s: EOF on stdin - terminating\n", binary_name));
+#if HAVE_GETPGRP
+ if (getpgrp() == getpid()) {
+ kill(-getpgrp(), SIGTERM);
+ }
+#endif
+ exit(0);
+ }
+}
+
+/*
+ main server.
+*/
+int main(int argc, const char *argv[])
+{
+ char *service = NULL;
+ bool opt_daemon = false;
+ bool opt_interactive = false;
+ int opt;
+ poptContext pc;
+ struct event_context *event_ctx;
+ struct confdb_ctx *confdb_ctx;
+ TALLOC_CTX *mem_ctx;
+ uint16_t stdin_event_flags;
+ int ret = EOK;
+ bool is_monitor = false;
+
+ debug_prg_name = argv[0];
+
+ enum {
+ OPT_DAEMON = 1000,
+ OPT_INTERACTIVE
+ };
+ struct poptOption long_options[] = {
+ POPT_AUTOHELP
+ {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON,
+ "Become a daemon (default)", NULL },
+ {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE,
+ "Run interactive (not a daemon)", NULL},
+ {"service", 's', POPT_ARG_STRING, &service, 0,
+ "Executes a specific service instead of the monitor", NULL},
+ {"debug-level", 'd', POPT_ARG_INT, &debug_level, 0,
+ "Executes a specific service instead of the monitor", NULL},
+ { NULL }
+ };
+
+ pc = poptGetContext(argv[0], argc, argv, long_options, 0);
+ while((opt = poptGetNextOpt(pc)) != -1) {
+ switch(opt) {
+ case OPT_DAEMON:
+ opt_daemon = true;
+ break;
+ case OPT_INTERACTIVE:
+ opt_interactive = true;
+ break;
+ default:
+ fprintf(stderr, "\nInvalid option %s: %s\n\n",
+ poptBadOption(pc, 0), poptStrerror(opt));
+ poptPrintUsage(pc, stderr, 0);
+ return 1;
+ }
+ }
+
+ if (!service) {
+ fprintf(stderr,"\nERROR: No service specified\n\n");
+ return 5;
+ }
+
+ if (strcmp(service, "monitor") == 0) is_monitor = true;
+
+ if (is_monitor) {
+ if (opt_daemon && opt_interactive) {
+ fprintf(stderr,"\nERROR: "
+ "Option -i|--interactive is not allowed together with -D|--daemon\n\n");
+ poptPrintUsage(pc, stderr, 0);
+ return 1;
+ } else if (!opt_interactive) {
+ /* default is --daemon */
+ opt_daemon = true;
+ }
+ } else {
+ if (opt_daemon || opt_interactive) {
+ fprintf(stderr,"\nERROR: "
+ "Options -i or -D not allowed with -s (service)\n\n");
+ poptPrintUsage(pc, stderr, 0);
+ return 1;
+ }
+
+ }
+
+ poptFreeContext(pc);
+
+ setup_signals();
+
+ /* we want default permissions on created files to be very strict,
+ so set our umask to 0177 */
+ umask(0177);
+
+ if (opt_daemon) {
+ DEBUG(3,("Becoming a daemon.\n"));
+ become_daemon(true);
+
+ ret = pidfile(PID_PATH, "sssd");
+ if (ret != EOK) {
+ fprintf(stderr, "\nERROR: PID File reports daemon already running!\n");
+ return 1;
+ }
+ }
+
+ /* the event context is the top level structure.
+ * Everything else should hang off that */
+ event_ctx = event_context_init(talloc_autofree_context());
+ if (event_ctx == NULL) {
+ DEBUG(0,("The event context initialiaziton failed\n"));
+ return 1;
+ }
+
+ mem_ctx = talloc_new(event_ctx);
+ if (mem_ctx == NULL) {
+ DEBUG(0,("Out of memory, aborting!\n"));
+ return 1;
+ }
+
+ ret = confdb_init(mem_ctx, event_ctx, &confdb_ctx);
+ if (ret != EOK) {
+ DEBUG(0,("The confdb initialization failed\n"));
+ return 1;
+ }
+
+ if (opt_interactive) {
+ /* terminate when stdin goes away */
+ stdin_event_flags = EVENT_FD_READ;
+ } else {
+ /* stay alive forever */
+ stdin_event_flags = 0;
+ }
+
+ /* catch EOF on stdin */
+#ifdef SIGTTIN
+ signal(SIGTTIN, SIG_IGN);
+#endif
+ event_add_fd(event_ctx, event_ctx, 0, stdin_event_flags,
+ server_stdin_handler,
+ discard_const(argv[0]));
+
+ /* What are we asked to run ? */
+ if (is_monitor) {
+ /* the monitor */
+ ret = monitor_process_init(mem_ctx, event_ctx, confdb_ctx);
+
+ } else {
+
+ if (strcmp(service, "nss") == 0) {
+ ret = nss_process_init(mem_ctx, event_ctx, confdb_ctx);
+
+ } else if (strcmp(service, "dp") == 0) {
+ ret = dp_process_init(mem_ctx, event_ctx, confdb_ctx);
+
+ } else {
+ fprintf(stderr,
+ "\nERROR: Unknown Service specified [%s]\n",
+ service);
+ ret = EINVAL;
+ }
+ }
+
+ if (ret != EOK) return 3;
+
+ /* wait for events - this is where smbd sits for most of its
+ life */
+ event_loop_wait(event_ctx);
+
+ /* as everything hangs off this event context, freeing it
+ should initiate a clean shutdown of all services */
+ talloc_free(event_ctx);
+
+ return 0;
+}