1#!/usr/bin/perl -w
2
3# start-memcached
4# 2003/2004 - Jay Bonci <[email protected]>
5# This script handles the parsing of the /etc/memcached.conf file
6# and was originally created for the Debian distribution.
7# Anyone may use this little script under the same terms as
8# memcached itself.
9
10use POSIX qw(setsid);
11use strict;
12
13if($> != 0 and $< != 0)
14{
15    print STDERR "Only root wants to run start-memcached.\n";
16    exit;
17}
18
19my $params; my $etchandle; my $etcfile = "/etc/memcached.conf";
20
21# This script assumes that memcached is located at /usr/bin/memcached, and
22# that the pidfile is writable at /var/run/memcached.pid
23
24my $memcached = "/usr/bin/memcached";
25my $pidfile = "/var/run/memcached.pid";
26
27if (scalar(@ARGV) == 2) {
28    $etcfile = shift(@ARGV);
29    $pidfile = shift(@ARGV);
30}
31
32# If we don't get a valid logfile parameter in the /etc/memcached.conf file,
33# we'll just throw away all of our in-daemon output. We need to re-tie it so
34# that non-bash shells will not hang on logout. Thanks to Michael Renner for
35# the tip
36my $fd_reopened = "/dev/null";
37
38sub handle_logfile
39{
40    my ($logfile) = @_;
41    $fd_reopened = $logfile;
42}
43
44sub reopen_logfile
45{
46    my ($logfile) = @_;
47
48    open *STDERR, ">>$logfile";
49    open *STDOUT, ">>$logfile";
50    open *STDIN, ">>/dev/null";
51    $fd_reopened = $logfile;
52}
53
54# This is set up in place here to support other non -[a-z] directives
55
56my $conf_directives = {
57    "logfile" => \&handle_logfile,
58};
59
60if(open $etchandle, $etcfile)
61{
62    foreach my $line (<$etchandle>)
63    {
64        $line ||= "";
65        $line =~ s/\#.*//g;
66        $line =~ s/\s+$//g;
67        $line =~ s/^\s+//g;
68        next unless $line;
69        next if $line =~ /^\-[dh]/;
70
71        if($line =~ /^[^\-]/)
72        {
73            my ($directive, $arg) = $line =~ /^(.*?)\s+(.*)/;
74            $conf_directives->{$directive}->($arg);
75            next;
76        }
77
78        push @$params, $line;
79    }
80
81}else{
82    $params = [];
83}
84
85push @$params, "-u root" unless(grep "-u", @$params);
86$params = join " ", @$params;
87
88if(-e $pidfile)
89{
90    open PIDHANDLE, "$pidfile";
91    my $localpid = <PIDHANDLE>;
92    close PIDHANDLE;
93
94    chomp $localpid;
95    if(-d "/proc/$localpid")
96    {
97        print STDERR "memcached is already running.\n";
98        exit;
99    }else{
100        unlink $pidfile;
101    }
102
103}
104
105my $pid = fork();
106
107if($pid == 0)
108{
109    # setsid makes us the session leader
110    setsid();
111    reopen_logfile($fd_reopened);
112    # must fork again now that tty is closed
113    $pid = fork();
114    if ($pid) {
115      if(open PIDHANDLE,">$pidfile")
116      {
117          print PIDHANDLE $pid;
118          close PIDHANDLE;
119      }else{
120
121          print STDERR "Can't write pidfile to $pidfile.\n";
122      }
123      exit(0);
124    }
125    exec "$memcached $params";
126    exit(0);
127
128}
129