lib/cmd/daemon-cmd.py

   1 #!/bin/sh
   2 """": # -*-python-*-
   3 # https://sourceware.org/bugzilla/show_bug.cgi?id=26034
   4 export "BUP_ARGV_0"="$0"
   5 arg_i=1
   6 for arg in "$@"; do
   7     export "BUP_ARGV_${arg_i}"="$arg"
   8     shift
   9     arg_i=$((arg_i + 1))
  10 done
  11 # Here to end of preamble replaced during install
  12 bup_python="$(dirname "$0")/../../config/bin/python" || exit $?
  13 exec "$bup_python" "$0"
  14 """
  15 # end of bup preamble
  16
  17 from __future__ import absolute_import
  18
  19 # Intentionally replace the dirname "$0" that python prepends
  20 import os, sys
  21 sys.path[0] = os.path.dirname(os.path.realpath(__file__)) + '/..'
  22
  23 import getopt, socket, subprocess, fcntl
  24
  25 from bup import compat, options, path
  26 from bup.helpers import *
  27
  28 optspec = """
  29 bup daemon [options...] -- [bup-server options...]
  30 --
  31 l,listen  ip address to listen on, defaults to *
  32 p,port    port to listen on, defaults to 1982
  33 """
  34 o = options.Options(optspec, optfunc=getopt.getopt)
  35 opt, flags, extra = o.parse(compat.argv[1:])
  36
  37 host = opt.listen
  38 port = opt.port and int(opt.port) or 1982
  39
  40 import socket
  41 import sys
  42
  43 socks = []
  44 e = None
  45 for res in socket.getaddrinfo(host, port, socket.AF_UNSPEC,
  46                               socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
  47     af, socktype, proto, canonname, sa = res
  48     try:
  49         s = socket.socket(af, socktype, proto)
  50     except socket.error as e:
  51         continue
  52     try:
  53         if af == socket.AF_INET6:
  54             log("bup daemon: listening on [%s]:%s\n" % sa[:2])
  55         else:
  56             log("bup daemon: listening on %s:%s\n" % sa[:2])
  57         s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  58         s.bind(sa)
  59         s.listen(1)
  60         fcntl.fcntl(s.fileno(), fcntl.F_SETFD, fcntl.FD_CLOEXEC)
  61     except socket.error as e:
  62         s.close()
  63         continue
  64     socks.append(s)
  65
  66 if not socks:
  67     log('bup daemon: listen socket: %s\n' % e.args[1])
  68     sys.exit(1)
  69
  70 try:
  71     while True:
  72         [rl,wl,xl] = select.select(socks, [], [], 60)
  73         for l in rl:
  74             s, src = l.accept()
  75             try:
  76                 log("Socket accepted connection from %s\n" % (src,))
  77                 fd1 = os.dup(s.fileno())
  78                 fd2 = os.dup(s.fileno())
  79                 s.close()
  80                 sp = subprocess.Popen([path.exe(), 'mux', '--',
  81                                        path.exe(), 'server']
  82                                       + extra, stdin=fd1, stdout=fd2)
  83             finally:
  84                 os.close(fd1)
  85                 os.close(fd2)
  86 finally:
  87     for l in socks:
  88         l.shutdown(socket.SHUT_RDWR)
  89         l.close()
  90
  91 debug1("bup daemon: done")