clowwindy
10 years ago
4 changed files with 242 additions and 38 deletions
@ -0,0 +1,173 @@ |
|||
#!/usr/bin/python |
|||
# -*- coding: utf-8 -*- |
|||
|
|||
# Copyright (c) 2014 clowwindy |
|||
# |
|||
# Permission is hereby granted, free of charge, to any person obtaining a copy |
|||
# of this software and associated documentation files (the "Software"), to deal |
|||
# in the Software without restriction, including without limitation the rights |
|||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|||
# copies of the Software, and to permit persons to whom the Software is |
|||
# furnished to do so, subject to the following conditions: |
|||
# |
|||
# The above copyright notice and this permission notice shall be included in |
|||
# all copies or substantial portions of the Software. |
|||
# |
|||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|||
# SOFTWARE. |
|||
|
|||
from __future__ import absolute_import, division, print_function, \ |
|||
with_statement |
|||
|
|||
import os |
|||
import sys |
|||
import logging |
|||
import signal |
|||
import time |
|||
from shadowsocks import common |
|||
|
|||
# this module is ported from ShadowVPN daemon.c |
|||
|
|||
|
|||
def daemon_exec(config): |
|||
if 'daemon' in config: |
|||
if os.name != 'posix': |
|||
raise Exception('daemon mode is only supported in unix') |
|||
command = config['daemon'] |
|||
if not command: |
|||
command = 'start' |
|||
pid_file = config['pid-file'] |
|||
log_file = config['log-file'] |
|||
command = common.to_str(command) |
|||
pid_file = common.to_str(pid_file) |
|||
log_file = common.to_str(log_file) |
|||
if command == 'start': |
|||
daemon_start(pid_file, log_file) |
|||
elif command == 'stop': |
|||
daemon_stop(pid_file) |
|||
# always exit after daemon_stop |
|||
sys.exit(0) |
|||
elif command == 'restart': |
|||
daemon_stop(pid_file) |
|||
daemon_start(pid_file, log_file) |
|||
else: |
|||
raise Exception('unsupported daemon command %s' % command) |
|||
|
|||
|
|||
def write_pid_file(pid_file, pid): |
|||
import fcntl |
|||
import stat |
|||
|
|||
try: |
|||
fd = os.open(pid_file, os.O_RDWR | os.O_CREAT, |
|||
stat.S_IRUSR | stat.S_IWUSR) |
|||
except OSError as e: |
|||
logging.error(e) |
|||
return -1 |
|||
flags = fcntl.fcntl(fd, fcntl.F_GETFD) |
|||
assert flags != -1 |
|||
flags |= fcntl.FD_CLOEXEC |
|||
r = fcntl.fcntl(fd, fcntl.F_SETFD, flags) |
|||
assert r != -1 |
|||
# There is no platform independent way to implement fcntl(fd, F_SETLK, &fl) |
|||
# via fcntl.fcntl. So use lockf instead |
|||
try: |
|||
fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB, 0, 0, os.SEEK_SET) |
|||
except IOError: |
|||
r = os.read(fd, 32) |
|||
if r: |
|||
logging.error('already started at pid %s' % common.to_str(r)) |
|||
else: |
|||
logging.error('already started') |
|||
os.close(fd) |
|||
return -1 |
|||
os.ftruncate(fd, 0) |
|||
os.write(fd, common.to_bytes(str(pid))) |
|||
return 0 |
|||
|
|||
|
|||
def freopen(f, mode, stream): |
|||
oldf = open(f, mode) |
|||
oldfd = oldf.fileno() |
|||
newfd = stream.fileno() |
|||
os.close(newfd) |
|||
os.dup2(oldfd, newfd) |
|||
|
|||
|
|||
def daemon_start(pid_file, log_file): |
|||
# fork only once because we are sure parent will exit |
|||
pid = os.fork() |
|||
assert pid != -1 |
|||
|
|||
def handle_exit(signum, _): |
|||
sys.exit(0) |
|||
|
|||
if pid > 0: |
|||
# parent waits for its child |
|||
signal.signal(signal.SIGINT, handle_exit) |
|||
time.sleep(5) |
|||
sys.exit(0) |
|||
|
|||
# child signals its parent to exit |
|||
ppid = os.getppid() |
|||
pid = os.getpid() |
|||
if write_pid_file(pid_file, pid) != 0: |
|||
os.kill(ppid, signal.SIGINT) |
|||
sys.exit(1) |
|||
|
|||
print('started') |
|||
os.kill(ppid, signal.SIGINT) |
|||
|
|||
sys.stdin.close() |
|||
freopen(log_file, 'a', sys.stdout) |
|||
freopen(log_file, 'a', sys.stderr) |
|||
|
|||
|
|||
def daemon_stop(pid_file): |
|||
import errno |
|||
try: |
|||
with open(pid_file) as f: |
|||
buf = f.read() |
|||
pid = common.to_str(buf) |
|||
if not buf: |
|||
logging.error('not running') |
|||
except IOError as e: |
|||
logging.error(e) |
|||
if e.errno == errno.ENOENT: |
|||
# always exit 0 if we are sure daemon is not running |
|||
logging.error('not running') |
|||
return |
|||
sys.exit(1) |
|||
pid = int(pid) |
|||
if pid > 0: |
|||
try: |
|||
os.kill(pid, signal.SIGTERM) |
|||
except OSError as e: |
|||
if e.errno == errno.ESRCH: |
|||
logging.error('not running') |
|||
# always exit 0 if we are sure daemon is not running |
|||
return |
|||
logging.error(e) |
|||
sys.exit(1) |
|||
else: |
|||
logging.error('pid is not positive: %d', pid) |
|||
|
|||
# sleep for maximum 10s |
|||
for i in range(0, 200): |
|||
try: |
|||
# query for the pid |
|||
os.kill(pid, 0) |
|||
except OSError as e: |
|||
if e.errno == errno.ESRCH: |
|||
break |
|||
time.sleep(0.05) |
|||
else: |
|||
logging.error('timed out when stopping pid %d', pid) |
|||
sys.exit(1) |
|||
print('stopped') |
|||
os.unlink(pid_file) |
|||
Loading…
Reference in new issue