← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/daemon.c

  • Committer: Lee Bieber
  • Date: 2010-11-14 23:15:42 UTC
  • mfrom: (1929.1.42 warning-stack-frame)
  • Revision ID: kalebral@gmail.com-20101114231542-fnnu6ydd2p17n582
Merge Monty - fix bug 672372: some functions use > 32k stack

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/daemon.c
1
 
/*    $Header: /cvsroot/wikipedia/willow/src/bin/willow/daemon.c,v 1.1 2005/05/02 19:15:21 kateturner Exp $    */
2
 
/*    $NetBSD: daemon.c,v 1.9 2003/08/07 16:42:46 agc Exp $    */
3
 
/*-
4
 
 * Copyright (c) 1990, 1993
5
 
 *    The Regents of the University of California.  All rights reserved.
6
 
 * Copyright (c) 2010
7
 
 *    Stewart Smith
8
 
 *
9
 
 * Redistribution and use in source and binary forms, with or without
10
 
 * modification, are permitted provided that the following conditions
11
 
 * are met:
12
 
 * 1. Redistributions of source code must retain the above copyright
13
 
 *    notice, this list of conditions and the following disclaimer.
14
 
 * 2. Redistributions in binary form must reproduce the above copyright
15
 
 *    notice, this list of conditions and the following disclaimer in the
16
 
 *    documentation and/or other materials provided with the distribution.
17
 
 * 3. Neither the name of the University nor the names of its contributors
18
 
 *    may be used to endorse or promote products derived from this software
19
 
 *    without specific prior written permission.
20
 
 *
21
 
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22
 
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23
 
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24
 
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25
 
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26
 
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27
 
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28
 
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29
 
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30
 
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31
 
 * SUCH DAMAGE.
32
 
 */
33
 
 
34
 
#if defined __SUNPRO_C || defined __DECC || defined __HP_cc
35
 
# pragma ident "@(#)$Header: /cvsroot/wikipedia/willow/src/bin/willow/daemon.c,v 1.1 2005/05/02 19:15:21 kateturner Exp $"
36
 
# pragma ident "$NetBSD: daemon.c,v 1.9 2003/08/07 16:42:46 agc Exp $"
37
 
#endif
38
 
 
39
 
#include <fcntl.h>
40
 
#include <stdio.h>
41
 
#include <stdlib.h>
42
 
#include <sys/types.h>
43
 
#include <sys/wait.h>
44
 
#include <signal.h>
45
 
#include <unistd.h>
46
 
#include <sys/select.h>
47
 
 
48
 
int daemonize(int nochdir, int noclose, int wait_sigusr1);
49
 
int daemon_is_ready(void);
50
 
void sigusr1_handler(int sig);
51
 
 
52
 
pid_t parent_pid;
53
 
 
54
 
void sigusr1_handler(int sig)
55
 
{
56
 
  if (sig == SIGUSR1)
57
 
    _exit(EXIT_SUCCESS);
58
 
}
59
 
 
60
 
int daemon_is_ready()
61
 
{
62
 
  kill(parent_pid, SIGUSR1);
63
 
  return 0;
64
 
}
65
 
 
66
 
int daemonize(int nochdir, int noclose, int wait_sigusr1)
67
 
{
68
 
    int fd;
69
 
    pid_t child= -1;
70
 
 
71
 
    parent_pid= getpid();
72
 
    signal(SIGUSR1, sigusr1_handler);
73
 
 
74
 
    child= fork();
75
 
 
76
 
    switch (child)
77
 
    {
78
 
    case -1:
79
 
        return (-1);
80
 
    case 0:
81
 
        break;
82
 
    default:
83
 
      if (wait_sigusr1)
84
 
      {
85
 
        /* parent */
86
 
        int exit_code= -1;
87
 
        int status;
88
 
        while (waitpid(child, &status, 0) != child);
89
 
        if (WIFEXITED(status))
90
 
        {
91
 
          exit_code= WEXITSTATUS(status);
92
 
        }
93
 
        if (WIFSIGNALED(status))
94
 
        {
95
 
          exit_code= -1;
96
 
        }
97
 
        _exit(exit_code);
98
 
      }
99
 
      else
100
 
      {
101
 
        _exit(EXIT_SUCCESS);
102
 
      }
103
 
    }
104
 
 
105
 
    /* child */
106
 
    if (setsid() == -1)
107
 
        return (-1);
108
 
 
109
 
    if (nochdir == 0) {
110
 
        if(chdir("/") != 0) {
111
 
            perror("chdir");
112
 
            return (-1);
113
 
        }
114
 
    }
115
 
 
116
 
    if (noclose == 0 && (fd = open("/dev/null", O_RDWR, 0)) != -1) {
117
 
        if(dup2(fd, STDIN_FILENO) < 0) {
118
 
            perror("dup2 stdin");
119
 
            return (-1);
120
 
        }
121
 
        if(dup2(fd, STDOUT_FILENO) < 0) {
122
 
            perror("dup2 stdout");
123
 
            return (-1);
124
 
        }
125
 
        if(dup2(fd, STDERR_FILENO) < 0) {
126
 
            perror("dup2 stderr");
127
 
            return (-1);
128
 
        }
129
 
 
130
 
        if (fd > STDERR_FILENO) {
131
 
            if(close(fd) < 0) {
132
 
                perror("close");
133
 
                return (-1);
134
 
            }
135
 
        }
136
 
    }
137
 
    return (0);
138
 
}