290N/A

454N/A

395N/A

290N/A

290N/ACONTENT_PATH_DEFAULT = "/usr/share/lib/pkg"

383N/AMAX_REQUEST_BODY_SIZE = 128 * 1024 * 1024

290N/AHOST_DEFAULT = "0.0.0.0"

2339N/APORT_DEFAULT = 80

2535N/ASSL_PORT_DEFAULT = 443

290N/ATHREADS_MIN = 1

2535N/ATHREADS_DEFAULT = 60

290N/ATHREADS_MAX = 5000

1660N/ASOCKET_TIMEOUT_DEFAULT = 60

1660N/A

1660N/Aimport getopt

1660N/Aimport gettext

1660N/Aimport locale

1660N/Aimport logging

1660N/Aimport os

1660N/Aimport os.path

1660N/Aimport OpenSSL.crypto as crypto

1660N/Aimport subprocess

1660N/Aimport sys

1660N/Aimport tempfile

1660N/Aimport urlparse

1660N/A

1660N/Atry:

1660N/A import cherrypy

1660N/A version = cherrypy.__version__.split('.')

1660N/A if map(int, version) < [3, 1, 0]:

448N/A raise ImportError

448N/A elif map(int, version) >= [3, 2, 0]:

534N/A raise ImportError

534N/Aexcept ImportError:

534N/A print >> sys.stderr, """cherrypy 3.1.0 or greater (but less than """ \

534N/A """3.2.0) is required to use this program."""

534N/A sys.exit(2)

534N/A

534N/Aimport cherrypy.process.servers

290N/Afrom cherrypy.process.plugins import Daemonizer

290N/A

954N/Afrom pkg.misc import msg, emsg, setlocale

954N/Aimport pkg.client.api_errors as api_errors

954N/Aimport pkg.config as cfg

954N/Aimport pkg.portable.util as os_util

534N/Aimport pkg.search_errors as search_errors

1099N/Aimport pkg.server.depot as ds

290N/Aimport pkg.server.depotresponse as dr

1516N/Aimport pkg.server.repository as sr

290N/A

290N/A

290N/Aclass LogSink(object):

661N/A """This is a dummy object that we can use to discard log entries

2494N/A

2494N/A def write(self, *args, **kwargs):

2494N/A """Discard the bits."""

2516N/A pass

2516N/A

2516N/A def flush(self, *args, **kwargs):

2516N/A """Discard the bits."""

2516N/A pass

2516N/A

2516N/A

290N/Adef usage(text=None, retcode=2, full=False):

2523N/A """Optionally emit a usage message and then exit using the specified

1498N/A

1498N/A if text:

2310N/A emsg(text)

2310N/A

2310N/A if not full:

2535N/A # The full usage message isn't desired.

2535N/A emsg(_("Try `pkg.depotd --help or -?' for more "

2535N/A "information."))

2535N/A sys.exit(retcode)

2535N/A

2535N/A print """\

1544N/A sys.exit(retcode)

1557N/A

1903N/Aclass OptionError(Exception):

2046N/A """Option exception. """

2240N/A

1506N/A def __init__(self, *args):

413N/A Exception.__init__(self, *args)

2026N/A

413N/Aif __name__ == "__main__":

1978N/A

1024N/A setlocale(locale.LC_ALL, "")

395N/A gettext.install("pkg", "/usr/share/locale",

395N/A codeset=locale.getpreferredencoding())

2310N/A

2310N/A add_content = False

395N/A exit_ready = False

395N/A rebuild = False

413N/A reindex = False

395N/A nasty = False

2516N/A nasty_value = 0

2516N/A

2516N/A # Track initial configuration values.

2516N/A ivalues = { "pkg": {} }

2516N/A if "PKG_REPO" in os.environ:

2516N/A ivalues["pkg"]["inst_root"] = os.environ["PKG_REPO"]

2516N/A

2516N/A try:

2516N/A content_root = os.environ["PKG_DEPOT_CONTENT"]

2516N/A ivalues["pkg"]["content_root"] = content_root

2516N/A except KeyError:

2516N/A try:

2516N/A content_root = os.path.join(os.environ['PKG_HOME'],

2516N/A 'share/lib/pkg')

2516N/A ivalues["pkg"]["content_root"] = content_root

2516N/A except KeyError:

2516N/A pass

2516N/A

2516N/A opt = None

2516N/A addresses = set()

2516N/A debug_features = []

2516N/A disable_ops = []

2516N/A repo_props = {}

2516N/A socket_path = ""

2516N/A user_cfg = None

2516N/A try:

2516N/A long_opts = ["add-content", "cfg=", "cfg-file=",

2516N/A "content-root=", "debug=", "disable-ops=", "exit-ready",

2516N/A "help", "image-root=", "log-access=", "log-errors=",

2516N/A "llmirror", "mirror", "nasty=", "proxy-base=", "readonly",

2516N/A "rebuild", "refresh-index", "set-property=",

2516N/A "ssl-cert-file=", "ssl-dialog=", "ssl-key-file=",

2516N/A "sort-file-max-size=", "writable-root="]

2516N/A

2516N/A opts, pargs = getopt.getopt(sys.argv[1:], "a:d:np:s:t:?",

2516N/A long_opts)

2516N/A

2516N/A show_usage = False

2516N/A for opt, arg in opts:

2516N/A if opt == "-a":

2516N/A addresses.add(arg)

2516N/A elif opt == "-n":

2516N/A sys.exit(0)

2516N/A elif opt == "-d":

2516N/A ivalues["pkg"]["inst_root"] = arg

2516N/A elif opt == "-p":

2516N/A ivalues["pkg"]["port"] = arg

395N/A elif opt == "-s":

395N/A threads = int(arg)

395N/A if threads < THREADS_MIN:

395N/A raise OptionError, \

395N/A "minimum value is %d" % THREADS_MIN

2339N/A if threads > THREADS_MAX:

1191N/A raise OptionError, \

1452N/A "maximum value is %d" % THREADS_MAX

1231N/A ivalues["pkg"]["threads"] = threads

2535N/A elif opt == "-t":

2046N/A ivalues["pkg"]["socket_timeout"] = arg

395N/A elif opt == "--add-content":

395N/A add_content = True

424N/A elif opt == "--cfg":

395N/A user_cfg = arg

742N/A elif opt == "--cfg-file":

2339N/A ivalues["pkg"]["cfg_file"] = arg

2339N/A elif opt == "--content-root":

2339N/A ivalues["pkg"]["content_root"] = arg

2339N/A elif opt == "--debug":

2339N/A if arg is None or arg == "":

2339N/A continue

742N/A

742N/A # A list of features can be specified using a

742N/A # "," or any whitespace character as separators.

742N/A if "," in arg:

742N/A features = arg.split(",")

742N/A else:

742N/A features = arg.split()

742N/A debug_features.extend(features)

742N/A elif opt == "--disable-ops":

742N/A if arg is None or arg == "":

2310N/A raise OptionError, \

1902N/A "An argument must be specified."

1099N/A

2390N/A disableops = arg.split(",")

2335N/A for s in disableops:

2338N/A if "/" in s:

2338N/A op, ver = s.rsplit("/", 1)

2310N/A else:

2046N/A op = s

2223N/A ver = "*"

2046N/A

2046N/A if op not in \

2523N/A ds.DepotHTTP.REPO_OPS_DEFAULT:

2523N/A raise OptionError(

2523N/A "Invalid operation "

2523N/A "'%s'." % s)

2523N/A disable_ops.append(s)

2523N/A elif opt == "--exit-ready":

2310N/A exit_ready = True

2310N/A elif opt == "--image-root":

2310N/A ivalues["pkg"]["image_root"] = arg

2310N/A elif opt.startswith("--log-"):

2310N/A prop = "log_%s" % opt.lstrip("--log-")

2310N/A ivalues["pkg"][prop] = arg

2310N/A elif opt in ("--help", "-?"):

2310N/A show_usage = True

2508N/A elif opt == "--mirror":

2508N/A ivalues["pkg"]["mirror"] = True

2508N/A elif opt == "--llmirror":

2508N/A ivalues["pkg"]["mirror"] = True

2508N/A ivalues["pkg"]["ll_mirror"] = True

2535N/A ivalues["pkg"]["readonly"] = True

2535N/A elif opt == "--nasty":

2535N/A value_err = None

2535N/A try:

2535N/A nasty_value = int(arg)

2535N/A except ValueError, e:

2535N/A value_err = e

2535N/A

2535N/A if value_err or (nasty_value > 100 or

2535N/A nasty_value < 1):

2535N/A raise OptionError, "Invalid value " \

2535N/A "for nasty option.\n Please " \

2535N/A "choose a value between 1 and 100."

2535N/A nasty = True

2535N/A elif opt == "--proxy-base":

2535N/A # Attempt to decompose the url provided into

2535N/A # its base parts. This is done so we can

2535N/A # remove any scheme information since we

2535N/A # don't need it.

2535N/A scheme, netloc, path, params, query, \

2535N/A fragment = urlparse.urlparse(arg,

2535N/A "http", allow_fragments=0)

2535N/A

2535N/A if not netloc:

2535N/A raise OptionError, "Unable to " \

2535N/A "determine the hostname from " \

2535N/A "the provided URL; please use a " \

2535N/A "fully qualified URL."

2535N/A

2535N/A scheme = scheme.lower()

2535N/A if scheme not in ("http", "https"):

2535N/A raise OptionError, "Invalid URL; http " \

2535N/A "and https are the only supported " \

2535N/A "schemes."

2535N/A

2535N/A # Rebuild the url with the sanitized components.

2535N/A ivalues["pkg"]["proxy_base"] = \

2535N/A urlparse.urlunparse((scheme, netloc, path,

2535N/A params, query, fragment))

2535N/A elif opt == "--readonly":

2535N/A ivalues["pkg"]["readonly"] = True

2535N/A elif opt == "--rebuild":

2535N/A rebuild = True

2535N/A elif opt == "--refresh-index":

2535N/A # Note: This argument is for internal use

2535N/A # only.

2535N/A #

2535N/A # This flag is purposefully omitted in usage.

2535N/A # The supported way to forcefully reindex is to

2535N/A # kill any pkg.depot using that directory,

2535N/A # remove the index directory, and restart the

2535N/A # pkg.depot process. The index will be rebuilt

2535N/A # automatically on startup.

2535N/A reindex = True

2535N/A exit_ready = True

2535N/A elif opt == "--set-property":

2535N/A try:

2535N/A prop, p_value = arg.split("=", 1)

2339N/A p_sec, p_name = prop.split(".", 1)

2339N/A except ValueError:

2339N/A usage(_("property arguments must be of "

691N/A "the form '<section.property>="

691N/A "<value>'."))

691N/A repo_props.setdefault(p_sec, {})

395N/A repo_props[p_sec][p_name] = p_value

395N/A elif opt == "--ssl-cert-file":

395N/A if arg == "none" or arg == "":

395N/A # Assume this is an override to clear

395N/A # the value.

290N/A arg = ""

395N/A elif not os.path.isabs(arg):

395N/A raise OptionError, "The path to " \

591N/A "the Certificate file must be " \

591N/A "absolute."

591N/A elif not os.path.exists(arg):

1505N/A raise OptionError, "The specified " \

2516N/A "file does not exist."

1505N/A elif not os.path.isfile(arg):

1505N/A raise OptionError, "The specified " \

1632N/A "pathname is not a file."

1632N/A ivalues["pkg"]["ssl_cert_file"] = arg

1632N/A elif opt == "--ssl-key-file":

1632N/A if arg == "none" or arg == "":

2339N/A # Assume this is an override to clear

2339N/A # the value.

2339N/A arg = ""

2339N/A elif not os.path.isabs(arg):

2339N/A raise OptionError, "The path to " \

2339N/A "the Private Key file must be " \

2339N/A "absolute."

2339N/A elif not os.path.exists(arg):

2339N/A raise OptionError, "The specified " \

2339N/A "file does not exist."

2339N/A elif not os.path.isfile(arg):

2339N/A raise OptionError, "The specified " \

2339N/A "pathname is not a file."

2339N/A ivalues["pkg"]["ssl_key_file"] = arg

2339N/A elif opt == "--ssl-dialog":

2339N/A if arg != "builtin" and \

2364N/A arg != "smf" and not \

2339N/A arg.startswith("exec:/") and not \

2339N/A arg.startswith("svc:"):

2339N/A raise OptionError, "Invalid value " \

2339N/A "specified. Expected: builtin, " \

2339N/A "exec:/path/to/program, smf, or " \

2339N/A "an SMF FMRI."

2339N/A

2339N/A if arg.startswith("exec:"):

2339N/A if os_util.get_canonical_os_type() != \

2339N/A "unix":

2339N/A # Don't allow a somewhat

2339N/A # insecure authentication method

2339N/A # on some platforms.

2339N/A raise OptionError, "exec is " \

2339N/A "not a supported dialog " \

2339N/A "type for this operating " \

2339N/A "system."

2339N/A

2339N/A f = os.path.abspath(arg.split(

2339N/A "exec:")[1])

2364N/A if not os.path.isfile(f):

2364N/A raise OptionError, "Invalid " \

2364N/A "file path specified for " \

2364N/A "exec."

2364N/A ivalues["pkg"]["ssl_dialog"] = arg

2364N/A elif opt == "--sort-file-max-size":

2364N/A ivalues["pkg"]["sort_file_max_size"] = arg

2364N/A elif opt == "--writable-root":

2364N/A ivalues["pkg"]["writable_root"] = arg

2364N/A

2364N/A # Set accumulated values.

2364N/A if debug_features:

2364N/A ivalues["pkg"]["debug"] = debug_features

2339N/A if disable_ops:

395N/A ivalues["pkg"]["disable_ops"] = disable_ops

395N/A if addresses:

290N/A ivalues["pkg"]["address"] = list(addresses)

290N/A

2339N/A # Build configuration object.

2339N/A dconf = ds.DepotConfig(target=user_cfg, overrides=ivalues)

290N/A except getopt.GetoptError, _e:

290N/A usage("pkg.depotd: %s" % _e.msg)

290N/A except api_errors.ApiException, _e:

290N/A usage("pkg.depotd: %s" % str(_e))

290N/A except OptionError, _e:

290N/A usage("pkg.depotd: option: %s -- %s" % (opt, _e))

290N/A except (ArithmeticError, ValueError):

290N/A usage("pkg.depotd: illegal option value: %s specified " \

290N/A "for option: %s" % (arg, opt))

290N/A

395N/A if show_usage:

395N/A usage(retcode=0, full=True)

290N/A

290N/A if not dconf.get_property("pkg", "log_errors"):

290N/A dconf.set_property("pkg", "log_errors", "stderr")

290N/A

290N/A # If stdout is a tty, then send access output there by default instead

395N/A # of discarding it.

395N/A if not dconf.get_property("pkg", "log_access"):

395N/A if os.isatty(sys.stdout.fileno()):

290N/A dconf.set_property("pkg", "log_access", "stdout")

395N/A else:

395N/A dconf.set_property("pkg", "log_access", "none")

395N/A

395N/A # Check for invalid option combinations.

591N/A image_root = dconf.get_property("pkg", "image_root")

591N/A inst_root = dconf.get_property("pkg", "inst_root")

591N/A mirror = dconf.get_property("pkg", "mirror")

591N/A ll_mirror = dconf.get_property("pkg", "ll_mirror")

691N/A readonly = dconf.get_property("pkg", "readonly")

691N/A writable_root = dconf.get_property("pkg", "writable_root")

691N/A if rebuild and add_content:

691N/A usage("--add-content cannot be used with --rebuild")

2339N/A if rebuild and reindex:

2339N/A usage("--refresh-index cannot be used with --rebuild")

2339N/A if (rebuild or add_content) and (readonly or mirror):

2339N/A usage("--readonly and --mirror cannot be used with --rebuild "

290N/A "or --add-content")

290N/A if reindex and mirror:

290N/A usage("--mirror cannot be used with --refresh-index")

290N/A if reindex and readonly and not writable_root:

290N/A usage("--readonly can only be used with --refresh-index if "

591N/A "--writable-root is used")

591N/A if image_root and not ll_mirror:

691N/A usage("--image-root can only be used with --llmirror.")

691N/A if image_root and writable_root:

2339N/A usage("--image_root and --writable-root cannot be used "

2339N/A "together.")

290N/A if image_root and inst_root:

290N/A usage("--image-root and -d cannot be used together.")

2339N/A

2339N/A # If the image format changes this may need to be reexamined.

2339N/A if image_root:

2339N/A inst_root = os.path.join(image_root, "var", "pkg")

2339N/A

2339N/A # Set any values using defaults if they weren't provided.

2339N/A

290N/A # Only use the first value for now; multiple bind addresses may be

2339N/A # supported later.

2339N/A address = dconf.get_property("pkg", "address")

290N/A if address:

2339N/A address = address[0]

2339N/A elif not address:

2339N/A dconf.set_property("pkg", "address", [HOST_DEFAULT])

2339N/A address = dconf.get_property("pkg", "address")[0]

2339N/A

2339N/A if not inst_root:

2339N/A usage("Either PKG_REPO or -d must be provided")

2339N/A

290N/A content_root = dconf.get_property("pkg", "content_root")

395N/A if not content_root:

290N/A dconf.set_property("pkg", "content_root", CONTENT_PATH_DEFAULT)

395N/A content_root = dconf.get_property("pkg", "content_root")

506N/A

506N/A port = dconf.get_property("pkg", "port")

506N/A ssl_cert_file = dconf.get_property("pkg", "ssl_cert_file")

506N/A ssl_key_file = dconf.get_property("pkg", "ssl_key_file")

506N/A if (ssl_cert_file and not ssl_key_file) or (ssl_key_file and not

506N/A ssl_cert_file):

506N/A usage("The --ssl-cert-file and --ssl-key-file options must "

506N/A "must both be provided when using either option.")

834N/A elif not port:

506N/A if ssl_cert_file and ssl_key_file:

506N/A dconf.set_property("pkg", "port", SSL_PORT_DEFAULT)

506N/A else:

513N/A dconf.set_property("pkg", "port", PORT_DEFAULT)

506N/A port = dconf.get_property("pkg", "port")

506N/A

506N/A socket_timeout = dconf.get_property("pkg", "socket_timeout")

506N/A if not socket_timeout:

290N/A dconf.set_property("pkg", "socket_timeout",

290N/A SOCKET_TIMEOUT_DEFAULT)

2535N/A socket_timeout = dconf.get_property("pkg", "socket_timeout")

2535N/A

2535N/A threads = dconf.get_property("pkg", "threads")

395N/A if not threads:

413N/A dconf.set_property("pkg", "threads", THREADS_DEFAULT)

395N/A threads = dconf.get_property("pkg", "threads")

290N/A

1674N/A # If the program is going to reindex, the port is irrelevant since

1674N/A # the program will not bind to a port.

1674N/A if not exit_ready:

1674N/A try:

1674N/A cherrypy.process.servers.check_port(address, port)

1674N/A except Exception, e:

1674N/A emsg("pkg.depotd: unable to bind to the specified "

1674N/A "port: %d. Reason: %s" % (port, e))

1674N/A sys.exit(1)

1674N/A else:

1674N/A # Not applicable if we're not going to serve content

1674N/A dconf.set_property("pkg", "content_root", "")

1674N/A

1674N/A # Any relative paths should be made absolute using pkg_root. 'pkg_root'

1674N/A # is a special property that was added to enable internal deployment of

395N/A # multiple disparate versions of the pkg.depotd software.

395N/A pkg_root = dconf.get_property("pkg", "pkg_root")

506N/A

506N/A repo_config_file = dconf.get_property("pkg", "cfg_file")

395N/A if repo_config_file and not os.path.isabs(repo_config_file):

2535N/A repo_config_file = os.path.join(pkg_root, repo_config_file)

2535N/A

395N/A if content_root and not os.path.isabs(content_root):

430N/A content_root = os.path.join(pkg_root, content_root)

849N/A

834N/A if inst_root and not os.path.isabs(inst_root):

290N/A inst_root = os.path.join(pkg_root, inst_root)

2535N/A

2535N/A if ssl_cert_file:

2535N/A if ssl_cert_file == "none":

2535N/A ssl_cert_file = None

2535N/A elif not os.path.isabs(ssl_cert_file):

1099N/A ssl_cert_file = os.path.join(pkg_root, ssl_cert_file)

2535N/A

2535N/A if ssl_key_file:

2535N/A if ssl_key_file == "none":

2535N/A ssl_key_file = None

2535N/A elif not os.path.isabs(ssl_key_file):

2535N/A ssl_key_file = os.path.join(pkg_root, ssl_key_file)

2535N/A

2535N/A if writable_root and not os.path.isabs(writable_root):

2535N/A writable_root = os.path.join(pkg_root, writable_root)

2535N/A

1099N/A # Setup SSL if requested.

2535N/A key_data = None

2535N/A ssl_dialog = dconf.get_property("pkg", "ssl_dialog")

2535N/A if not exit_ready and ssl_cert_file and ssl_key_file and \

2535N/A ssl_dialog != "builtin":

2535N/A cmdline = None

2535N/A def get_ssl_passphrase(*ignored):

2535N/A p = None

2535N/A try:

2535N/A p = subprocess.Popen(cmdline, shell=True,

2535N/A stdout=subprocess.PIPE,

2535N/A stderr=None)

2535N/A p.wait()

2535N/A except Exception, __e:

2535N/A emsg("pkg.depotd: an error occurred while "

2535N/A "executing [%s]; unable to obtain the "

2535N/A "passphrase needed to decrypt the SSL "

2535N/A "private key file: %s" % (cmdline, __e))

1099N/A sys.exit(1)

2535N/A return p.stdout.read().strip("\n")

2535N/A

2535N/A if ssl_dialog.startswith("exec:"):

2535N/A exec_path = ssl_dialog.split("exec:")[1]

395N/A if not os.path.isabs(exec_path):

2535N/A exec_path = os.path.join(pkg_root, exec_path)

2535N/A cmdline = "%s %s %d" % (exec_path, "''", port)

2535N/A elif ssl_dialog == "smf" or ssl_dialog.startswith("svc:"):

2535N/A if ssl_dialog == "smf":

2535N/A # Assume the configuration target was an SMF

2535N/A # FMRI and let svcprop fail with an error if

1191N/A # it wasn't.

2535N/A svc_fmri = dconf.target

2535N/A else:

2535N/A svc_fmri = ssl_dialog

2535N/A cmdline = "/usr/bin/svcprop -p " \

1191N/A "pkg_secure/ssl_key_passphrase %s" % svc_fmri

1660N/A

1660N/A # The key file requires decryption, but the user has requested

1660N/A # exec-based authentication, so it will have to be decoded first

1660N/A # to an un-named temporary file.

849N/A try:

1208N/A with file(ssl_key_file, "rb") as key_file:

1208N/A pkey = crypto.load_privatekey(

1208N/A crypto.FILETYPE_PEM, key_file.read(),

1208N/A get_ssl_passphrase)

849N/A

290N/A key_data = tempfile.TemporaryFile()

2535N/A key_data.write(crypto.dump_privatekey(

2535N/A crypto.FILETYPE_PEM, pkey))

2535N/A key_data.seek(0)

2535N/A except EnvironmentError, _e:

2535N/A emsg("pkg.depotd: unable to read the SSL private key "

2535N/A "file: %s" % _e)

2535N/A sys.exit(1)

2535N/A except crypto.Error, _e:

2535N/A emsg("pkg.depotd: authentication or cryptography "

2535N/A "failure while attempting to decode\nthe SSL "

2535N/A "private key file: %s" % _e)

2535N/A sys.exit(1)

2535N/A else:

2535N/A # Redirect the server to the decrypted key file.

2535N/A ssl_key_file = "/dev/fd/%d" % key_data.fileno()

2535N/A key_data.close()

2535N/A

2535N/A # Setup our global configuration.

2535N/A gconf = {

2535N/A "checker.on": True,

2535N/A "environment": "production",

2535N/A "log.screen": False,

2535N/A "server.max_request_body_size": MAX_REQUEST_BODY_SIZE,

2535N/A "server.shutdown_timeout": 0,

2535N/A "server.socket_host": address,

2535N/A "server.socket_port": port,

2535N/A "server.socket_timeout": socket_timeout,

2535N/A "server.ssl_certificate": ssl_cert_file,

2535N/A "server.ssl_private_key": ssl_key_file,

2535N/A "server.thread_pool": threads,

2535N/A "tools.log_headers.on": True,

2535N/A "tools.encode.on": True

2535N/A }

2535N/A

2535N/A if "headers" in dconf.get_property("pkg", "debug"):

2535N/A # Despite its name, this only logs headers when there is an

2535N/A # error; it's redundant with the debug feature enabled.

2535N/A gconf["tools.log_headers.on"] = False

2535N/A

2535N/A # Causes the headers of every request to be logged to the error

2535N/A # log; even if an exception occurs.

2535N/A gconf["tools.log_headers_always.on"] = True

2535N/A cherrypy.tools.log_headers_always = cherrypy.Tool(

2535N/A "on_start_resource",

2535N/A cherrypy.lib.cptools.log_request_headers)

2535N/A

2535N/A log_type_map = {

2535N/A "errors": {

2535N/A "param": "log.error_file",

2535N/A "attr": "error_log"

2535N/A },

2535N/A "access": {

2535N/A "param": "log.access_file",

2535N/A "attr": "access_log"

2535N/A }

2535N/A }

2535N/A

2535N/A for log_type in log_type_map:

2535N/A dest = dconf.get_property("pkg", "log_%s" % log_type)

2535N/A if dest in ("stdout", "stderr", "none"):

2535N/A if dest == "none":

2535N/A h = logging.StreamHandler(LogSink())

2535N/A else:

2535N/A h = logging.StreamHandler(eval("sys.%s" % \

2535N/A dest))

2535N/A

2535N/A h.setLevel(logging.DEBUG)

2535N/A h.setFormatter(cherrypy._cplogging.logfmt)

2535N/A log_obj = eval("cherrypy.log.%s" % \

2535N/A log_type_map[log_type]["attr"])

2535N/A log_obj.addHandler(h)

2535N/A # Since we've replaced cherrypy's log handler with our

2535N/A # own, we don't want the output directed to a file.

2535N/A dest = ""

2535N/A elif dest:

2535N/A if not os.path.isabs(dest):

2535N/A dest = os.path.join(pkg_root, dest)

2535N/A gconf[log_type_map[log_type]["param"]] = dest

2535N/A

2535N/A cherrypy.config.update(gconf)

2535N/A

2535N/A # Now that our logging, etc. has been setup, it's safe to perform any

2535N/A # remaining preparation.

2535N/A

2535N/A # Initialize repository state.

2535N/A if not readonly:

2535N/A # Not readonly, so assume a new repository should be created.

2535N/A try:

2535N/A sr.repository_create(inst_root, properties=repo_props)

2535N/A except sr.RepositoryExistsError:

2535N/A # Already exists, nothing to do.

849N/A pass

290N/A except (api_errors.ApiException, sr.RepositoryError), _e:

2535N/A emsg("pkg.depotd: %s" % _e)

2535N/A sys.exit(1)

430N/A

395N/A try:

395N/A sort_file_max_size = dconf.get_property("pkg",

290N/A "sort_file_max_size")

383N/A

383N/A repo = sr.Repository(cfgpathname=repo_config_file,

395N/A log_obj=cherrypy, mirror=mirror, properties=repo_props,

383N/A read_only=readonly, root=inst_root,

383N/A sort_file_max_size=sort_file_max_size,

384N/A writable_root=writable_root)

383N/A except (RuntimeError, sr.RepositoryError), _e:

849N/A emsg("pkg.depotd: %s" % _e)

849N/A sys.exit(1)

849N/A except search_errors.IndexingException, _e:

849N/A emsg("pkg.depotd: %s" % str(_e), "INDEX")

849N/A sys.exit(1)

849N/A except api_errors.ApiException, _e:

849N/A emsg("pkg.depotd: %s" % str(_e))

849N/A sys.exit(1)

849N/A

2242N/A if not rebuild and not add_content and not repo.mirror and \

2242N/A not (repo.read_only and not repo.writable_root):

2242N/A # Automatically update search indexes on startup if not already

2242N/A # told to, and not in readonly/mirror mode.

2242N/A reindex = True

2242N/A

2242N/A if reindex:

2242N/A try:

2337N/A # Only execute a index refresh here if --exit-ready was

2337N/A # requested; it will be handled later in the setup

2242N/A # process for other cases.

2242N/A if repo.root and exit_ready:

849N/A repo.refresh_index()

2508N/A except (sr.RepositoryError, search_errors.IndexingException,

2508N/A api_errors.ApiException), e:

2508N/A emsg(str(e), "INDEX")

2508N/A sys.exit(1)

2508N/A elif rebuild:

2508N/A try:

2508N/A repo.rebuild(build_index=True)

2508N/A except sr.RepositoryError, e:

2508N/A emsg(str(e), "REBUILD")

2508N/A sys.exit(1)

2508N/A except (search_errors.IndexingException,

2508N/A api_errors.UnknownErrors,

2508N/A api_errors.PermissionsException), e:

2508N/A emsg(str(e), "INDEX")

2508N/A sys.exit(1)

2508N/A elif add_content:

2508N/A try:

2508N/A repo.add_content()

2508N/A repo.refresh_index()

2508N/A except sr.RepositoryError, e:

2508N/A emsg(str(e), "ADD_CONTENT")

2508N/A sys.exit(1)

2508N/A except (search_errors.IndexingException,

2508N/A api_errors.UnknownErrors,

2508N/A api_errors.PermissionsException), e:

2508N/A emsg(str(e), "INDEX")

2508N/A sys.exit(1)

2508N/A

2508N/A # Ready to start depot; exit now if requested.

2508N/A if exit_ready:

2508N/A sys.exit(0)

2508N/A

2508N/A # Next, initialize depot.

2508N/A if nasty:

2508N/A depot = ds.NastyDepotHTTP(repo, dconf)

2508N/A depot.set_nasty(nasty_value)

2508N/A else:

2508N/A depot = ds.DepotHTTP(repo, dconf)

2508N/A

2508N/A # Now build our site configuration.

2508N/A conf = {

2508N/A "/": {

2508N/A # We have to override cherrypy's default response_class so that

2508N/A # we have access to the write() callable to stream data

2508N/A # directly to the client.

2508N/A "wsgi.response_class": dr.DepotResponse,

2508N/A },

2508N/A "/robots.txt": {

2508N/A "tools.staticfile.on": True,

2508N/A "tools.staticfile.filename": os.path.join(depot.web_root,

2508N/A "robots.txt")

849N/A },

383N/A }

2508N/A

2508N/A proxy_base = dconf.get_property("pkg", "proxy_base")

2508N/A if proxy_base:

2508N/A # This changes the base URL for our server, and is primarily

2508N/A # intended to allow our depot process to operate behind Apache

2508N/A # or some other webserver process.

2508N/A #

2508N/A # Visit the following URL for more information:

2508N/A # http://cherrypy.org/wiki/BuiltinTools#tools.proxy

2508N/A proxy_conf = {

2508N/A "tools.proxy.on": True,

2508N/A "tools.proxy.local": "",

2508N/A "tools.proxy.base": proxy_base

2508N/A }

2508N/A

2508N/A # Now merge or add our proxy configuration information into the

2508N/A # existing configuration.

2508N/A for entry in proxy_conf:

2508N/A conf["/"][entry] = proxy_conf[entry]

2508N/A

2508N/A if ll_mirror:

2508N/A ds.DNSSD_Plugin(cherrypy.engine, gconf).subscribe()

2508N/A

2508N/A if reindex:

2508N/A # Tell depot to update search indexes when possible;

2508N/A # this is done as a background task so that packages

2508N/A # can be served immediately while search indexes are

2535N/A # still being updated.

2535N/A depot._queue_refresh_index()

383N/A

849N/A # If stdin is not a tty and the pkgdepot controller isn't being used,

383N/A # then assume process should be daemonized.

422N/A if not os.environ.get("PKGDEPOT_CONTROLLER") and \

422N/A not os.isatty(sys.stdin.fileno()):

422N/A Daemonizer(cherrypy.engine).subscribe()

422N/A

422N/A try:

422N/A root = cherrypy.Application(depot)

422N/A cherrypy.quickstart(root, config=conf)

422N/A except Exception, _e:

422N/A emsg("pkg.depotd: unknown error starting depot server, " \

422N/A "illegal option value specified?")

422N/A emsg(_e)

422N/A sys.exit(1)

422N/A