mirror of https://github.com/GNS3/gns3-server
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
193 lines
6.5 KiB
193 lines
6.5 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2015 GNS3 Technologies Inc.
|
|
#
|
|
# 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/>.
|
|
|
|
"""
|
|
Set up and run the server.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import signal
|
|
import asyncio
|
|
import aiohttp
|
|
import functools
|
|
import types
|
|
import time
|
|
|
|
from .web.route import Route
|
|
from .web.request_handler import RequestHandler
|
|
from .config import Config
|
|
from .modules import MODULES
|
|
from .modules.port_manager import PortManager
|
|
|
|
# TODO: get rid of * have something generic to automatically import handlers so the routes can be found
|
|
from gns3server.handlers import *
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Server:
|
|
|
|
def __init__(self, host, port):
|
|
|
|
self._host = host
|
|
self._port = port
|
|
self._loop = None
|
|
self._start_time = time.time()
|
|
self._port_manager = PortManager(host)
|
|
|
|
@asyncio.coroutine
|
|
def _run_application(self, handler, ssl_context=None):
|
|
|
|
try:
|
|
server = yield from self._loop.create_server(handler, self._host, self._port, ssl=ssl_context)
|
|
except OSError as e:
|
|
log.critical("Could not start the server: {}".format(e))
|
|
self._loop.stop()
|
|
return
|
|
return server
|
|
|
|
@asyncio.coroutine
|
|
def _stop_application(self):
|
|
"""
|
|
Cleanup the modules (shutdown running emulators etc.)
|
|
"""
|
|
|
|
for module in MODULES:
|
|
log.debug("Unloading module {}".format(module.__name__))
|
|
m = module.instance()
|
|
yield from m.unload()
|
|
self._loop.stop()
|
|
|
|
def _signal_handling(self, handler):
|
|
|
|
@asyncio.coroutine
|
|
def signal_handler(signame):
|
|
log.warning("Server has got signal {}, exiting...".format(signame))
|
|
yield from handler.finish_connections()
|
|
yield from self._stop_application()
|
|
|
|
signals = ["SIGTERM", "SIGINT"]
|
|
if sys.platform.startswith("win"):
|
|
signals.extend(["SIGBREAK"])
|
|
else:
|
|
signals.extend(["SIGHUP", "SIGQUIT"])
|
|
|
|
for signal_name in signals:
|
|
callback = functools.partial(asyncio.async, signal_handler(signal_name))
|
|
if sys.platform.startswith("win"):
|
|
# add_signal_handler() is not yet supported on Windows
|
|
signal.signal(getattr(signal, signal_name), callback)
|
|
else:
|
|
self._loop.add_signal_handler(getattr(signal, signal_name), callback)
|
|
|
|
def _reload_hook(self):
|
|
|
|
@asyncio.coroutine
|
|
def reload():
|
|
|
|
log.info("Reloading")
|
|
yield from self._stop_application()
|
|
os.execv(sys.executable, [sys.executable] + sys.argv)
|
|
|
|
# code extracted from tornado
|
|
for module in sys.modules.values():
|
|
# Some modules play games with sys.modules (e.g. email/__init__.py
|
|
# in the standard library), and occasionally this can cause strange
|
|
# failures in getattr. Just ignore anything that's not an ordinary
|
|
# module.
|
|
if not isinstance(module, types.ModuleType):
|
|
continue
|
|
path = getattr(module, "__file__", None)
|
|
if not path:
|
|
continue
|
|
if path.endswith(".pyc") or path.endswith(".pyo"):
|
|
path = path[:-1]
|
|
modified = os.stat(path).st_mtime
|
|
if modified > self._start_time:
|
|
log.debug("File {} has been modified".format(path))
|
|
asyncio.async(reload())
|
|
self._loop.call_later(1, self._reload_hook)
|
|
|
|
def _create_ssl_context(self, server_config):
|
|
|
|
import ssl
|
|
ssl_context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
|
|
certfile = server_config["certfile"]
|
|
certkey = server_config["certkey"]
|
|
try:
|
|
ssl_context.load_cert_chain(certfile, certkey)
|
|
except FileNotFoundError:
|
|
log.critical("Could not find the SSL certfile or certkey")
|
|
raise SystemExit
|
|
except ssl.SSLError as e:
|
|
log.critical("SSL error: {}".format(e))
|
|
raise SystemExit
|
|
return ssl_context
|
|
|
|
@asyncio.coroutine
|
|
def start_shell(self):
|
|
from ptpython.repl import embed
|
|
from gns3server.modules import Qemu
|
|
|
|
yield from embed(globals(), locals(), return_asyncio_coroutine=True, patch_stdout=True)
|
|
|
|
def run(self):
|
|
"""
|
|
Starts the server.
|
|
"""
|
|
|
|
logger = logging.getLogger("asyncio")
|
|
logger.setLevel(logging.WARNING)
|
|
|
|
server_config = Config.instance().get_section_config("Server")
|
|
if sys.platform.startswith("win"):
|
|
# use the Proactor event loop on Windows
|
|
asyncio.set_event_loop(asyncio.ProactorEventLoop())
|
|
|
|
ssl_context = None
|
|
if server_config.getboolean("ssl"):
|
|
if sys.platform.startswith("win"):
|
|
log.critical("SSL mode is not supported on Windows")
|
|
raise SystemExit
|
|
ssl_context = self._create_ssl_context(server_config)
|
|
|
|
self._loop = asyncio.get_event_loop()
|
|
app = aiohttp.web.Application()
|
|
for method, route, handler in Route.get_routes():
|
|
log.debug("Adding route: {} {}".format(method, route))
|
|
app.router.add_route(method, route, handler)
|
|
for module in MODULES:
|
|
log.debug("Loading module {}".format(module.__name__))
|
|
m = module.instance()
|
|
m.port_manager = self._port_manager
|
|
|
|
log.info("Starting server on {}:{}".format(self._host, self._port))
|
|
handler = app.make_handler(handler=RequestHandler)
|
|
self._loop.run_until_complete(self._run_application(handler, ssl_context))
|
|
self._signal_handling(handler)
|
|
|
|
if server_config.getboolean("live"):
|
|
log.info("Code live reload is enabled, watching for file changes")
|
|
self._loop.call_later(1, self._reload_hook)
|
|
|
|
if server_config.getboolean("shell"):
|
|
asyncio.async(self.start_shell())
|
|
|
|
self._loop.run_forever()
|