summaryrefslogtreecommitdiff
path: root/src/leap/bitmask/core/launcher.py
blob: 14d8e6077e2b2eb69991a2a3f51c286d8209a792 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# -*- coding: utf-8 -*-
# launcher.py
# Copyright (C) 2016 LEAP
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""
Run bitmask daemon.
"""
from os.path import join, abspath, dirname
import platform
import sys

from twisted.scripts.twistd import run

from leap.bitmask import core
from leap.bitmask.core import flags
from leap.bitmask.core.logs import getLogPath
from leap.common.config import get_path_prefix

pid = abspath(join(get_path_prefix(), 'leap', 'bitmaskd.pid'))

STANDALONE = getattr(sys, 'frozen', False)


def here(module=None):
    if STANDALONE:
        # we are running in a |PyInstaller| bundle
        return sys._MEIPASS
    else:
        if module:
            return dirname(module.__file__)
        else:
            return dirname(__file__)


def run_bitmaskd():

    # TODO --- configure where to put the logs... (get --logfile, --logdir
    # from bitmaskctl
    for (index, arg) in enumerate(sys.argv):
        if arg == '--backend':
            flags.BACKEND = sys.argv[index + 1]
        elif arg == '--verbose' or arg == '-v':
            flags.VERBOSE = True
        if STANDALONE:
            flags.VERBOSE = True
    args = [
        '-y', join(here(core), "bitmaskd.tac"),
        '--logfile', getLogPath(),
    ]
    if platform.system() != 'Windows':
        args.extend([
            '--pidfile', pid,
            '--umask', '0022'])
    sys.argv[1:] = args
    print '[+] launching bitmaskd...'
    run()


if __name__ == "__main__":
    run_bitmaskd()