xref: /OK3568_Linux_fs/yocto/bitbake/bin/bitbake-hashserv (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1#! /usr/bin/env python3
2#
3# Copyright (C) 2018 Garmin Ltd.
4#
5# SPDX-License-Identifier: GPL-2.0-only
6#
7
8import os
9import sys
10import logging
11import argparse
12import sqlite3
13import warnings
14warnings.simplefilter("default")
15
16sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(__file__)), 'lib'))
17
18import hashserv
19
20VERSION = "1.0.0"
21
22DEFAULT_BIND = 'unix://./hashserve.sock'
23
24
25def main():
26    parser = argparse.ArgumentParser(description='Hash Equivalence Reference Server. Version=%s' % VERSION,
27                                     epilog='''The bind address is the path to a unix domain socket if it is
28                                               prefixed with "unix://". Otherwise, it is an IP address
29                                               and port in form ADDRESS:PORT. To bind to all addresses, leave
30                                               the ADDRESS empty, e.g. "--bind :8686". To bind to a specific
31                                               IPv6 address, enclose the address in "[]", e.g.
32                                               "--bind [::1]:8686"'''
33                                     )
34
35    parser.add_argument('-b', '--bind', default=DEFAULT_BIND, help='Bind address (default "%(default)s")')
36    parser.add_argument('-d', '--database', default='./hashserv.db', help='Database file (default "%(default)s")')
37    parser.add_argument('-l', '--log', default='WARNING', help='Set logging level')
38    parser.add_argument('-u', '--upstream', help='Upstream hashserv to pull hashes from')
39    parser.add_argument('-r', '--read-only', action='store_true', help='Disallow write operations from clients')
40
41    args = parser.parse_args()
42
43    logger = logging.getLogger('hashserv')
44
45    level = getattr(logging, args.log.upper(), None)
46    if not isinstance(level, int):
47        raise ValueError('Invalid log level: %s' % args.log)
48
49    logger.setLevel(level)
50    console = logging.StreamHandler()
51    console.setLevel(level)
52    logger.addHandler(console)
53
54    server = hashserv.create_server(args.bind, args.database, upstream=args.upstream, read_only=args.read_only)
55    server.serve_forever()
56    return 0
57
58
59if __name__ == '__main__':
60    try:
61        ret = main()
62    except Exception:
63        ret = 1
64        import traceback
65        traceback.print_exc()
66    sys.exit(ret)
67