gns3-server/gns3server/web/web_server.py

314 lines
11 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
#
2015-01-14 00:05:26 +00:00
# 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/>.
2013-12-06 04:39:27 +00:00
"""
Set up and run the server.
2013-12-06 04:39:27 +00:00
"""
import os
2015-01-14 00:05:26 +00:00
import sys
import signal
2015-01-14 00:05:26 +00:00
import asyncio
import aiohttp
2016-05-17 10:39:23 +00:00
import aiohttp_cors
2015-01-14 00:05:26 +00:00
import functools
import time
import atexit
# Import encoding now, to avoid implicit import later.
# Implicit import within threads may cause LookupError when standard library is in a ZIP
import encodings.idna
from .route import Route
from ..config import Config
2016-04-15 15:57:06 +00:00
from ..compute import MODULES
from ..compute.port_manager import PortManager
from ..compute.qemu import Qemu
2016-04-19 13:35:50 +00:00
from ..controller import Controller
from gns3server.utils.static import get_static_dir
# do not delete this import
import gns3server.handlers
2015-01-14 00:05:26 +00:00
import logging
log = logging.getLogger(__name__)
2017-10-26 14:29:01 +00:00
if not (aiohttp.__version__.startswith("2.2") or aiohttp.__version__.startswith("2.3")):
raise RuntimeError("aiohttp 2.2.x or 2.3.x is required to run the GNS3 server")
2017-05-31 14:56:28 +00:00
class WebServer:
2016-06-15 23:37:43 +00:00
def __init__(self, host, port):
self._host = host
self._port = port
2015-01-14 00:05:26 +00:00
self._loop = None
self._handler = None
self._server = None
self._app = None
2015-01-14 00:05:26 +00:00
self._start_time = time.time()
2016-06-15 23:37:43 +00:00
self._running = False
self._closing = False
@staticmethod
2016-06-15 23:37:43 +00:00
def instance(host=None, port=None):
"""
Singleton to return only one instance of Server.
:returns: instance of Server
"""
if not hasattr(WebServer, "_instance") or WebServer._instance is None:
assert host is not None
assert port is not None
2016-06-15 23:37:43 +00:00
WebServer._instance = WebServer(host, port)
return WebServer._instance
def _run_application(self, handler, ssl_context=None):
try:
srv = self._loop.create_server(handler, self._host, self._port, ssl=ssl_context)
self._server, startup_res = self._loop.run_until_complete(asyncio.gather(srv, self._app.startup(), loop=self._loop))
except (RuntimeError, OSError, asyncio.CancelledError) as e:
log.critical("Could not start the server: {}".format(e))
return False
return True
2015-01-14 00:05:26 +00:00
async def shutdown_server(self):
2013-12-06 04:39:27 +00:00
"""
Cleanly shutdown the server.
"""
if not self._closing:
self._closing = True
else:
log.warning("Close is already in progress")
return
if self._server:
self._server.close()
await self._server.wait_closed()
if self._app:
await self._app.shutdown()
if self._handler:
2017-10-26 14:29:01 +00:00
try:
# aiohttp < 2.3
await self._handler.finish_connections(2) # Parameter is timeout
2017-10-26 14:29:01 +00:00
except AttributeError:
# aiohttp >= 2.3
await self._handler.shutdown(2) # Parameter is timeout
if self._app:
await self._app.cleanup()
await Controller.instance().stop()
2015-01-22 10:49:22 +00:00
for module in MODULES:
log.debug("Unloading module {}".format(module.__name__))
m = module.instance()
await m.unload()
if PortManager.instance().tcp_ports:
log.warning("TCP ports are still used {}".format(PortManager.instance().tcp_ports))
if PortManager.instance().udp_ports:
log.warning("UDP ports are still used {}".format(PortManager.instance().udp_ports))
for task in asyncio.Task.all_tasks():
task.cancel()
try:
await asyncio.wait_for(task, 1)
except BaseException:
pass
2015-01-14 00:05:26 +00:00
self._loop.stop()
def _signal_handling(self):
def signal_handler(signame, *args):
2015-01-20 13:59:19 +00:00
log.warning("Server has got signal {}, exiting...".format(signame))
try:
asyncio.ensure_future(self.shutdown_server())
except asyncio.CancelledError:
pass
2015-01-14 00:05:26 +00:00
signals = ["SIGTERM", "SIGINT"]
if sys.platform.startswith("win"):
signals.extend(["SIGBREAK"])
else:
signals.extend(["SIGHUP", "SIGQUIT"])
2015-01-14 00:05:26 +00:00
for signal_name in signals:
callback = functools.partial(signal_handler, signal_name)
2015-01-14 00:05:26 +00:00
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)
2015-01-24 19:11:51 +00:00
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
log.info("SSL is enabled")
2015-01-24 19:11:51 +00:00
return ssl_context
async def start_shell(self):
2015-02-24 00:08:34 +00:00
try:
from ptpython.repl import embed
except ImportError:
log.error("Unable to start a shell: the ptpython module must be installed!")
return
await embed(globals(), locals(), return_asyncio_coroutine=True, patch_stdout=True, history_filename=".gns3_shell_history")
2015-02-20 21:40:20 +00:00
def _exit_handling(self):
"""
Makes sure the asyncio loop is closed.
"""
def close_asyncio_loop():
loop = None
try:
loop = asyncio.get_event_loop()
except AttributeError:
pass
if loop is not None:
loop.close()
atexit.register(close_asyncio_loop)
async def _on_startup(self, *args):
"""
Called when the HTTP server start
"""
await Controller.instance().start()
# Because with a large image collection
# without md5sum already computed we start the
# computing with server start
asyncio.ensure_future(Qemu.instance().list_images())
def run(self):
"""
2015-01-14 00:05:26 +00:00
Starts the server.
"""
server_logger = logging.getLogger('aiohttp.server')
# In debug mode we don't use the standard request log but a more complete in response.py
if log.getEffectiveLevel() == logging.DEBUG:
server_logger.setLevel(logging.CRITICAL)
2015-01-20 22:28:40 +00:00
logger = logging.getLogger("asyncio")
2015-10-14 16:10:05 +00:00
logger.setLevel(logging.ERROR)
2015-01-20 22:28:40 +00:00
if sys.platform.startswith("win"):
loop = asyncio.get_event_loop()
# Add a periodic callback to give a chance to process signals on Windows
# because asyncio.add_signal_handler() is not supported yet on that platform
# otherwise the loop runs outside of signal module's ability to trap signals.
def wakeup():
loop.call_later(0.5, wakeup)
loop.call_later(0.5, wakeup)
server_config = Config.instance().get_section_config("Server")
2015-01-24 19:11:51 +00:00
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)
2015-01-14 00:05:26 +00:00
self._loop = asyncio.get_event_loop()
2017-10-26 11:37:50 +00:00
if log.getEffectiveLevel() == logging.DEBUG:
# On debug version we enable info that
# coroutine is not called in a way await/await
self._loop.set_debug(True)
for key, val in os.environ.items():
log.debug("ENV %s=%s", key, val)
self._app = aiohttp.web.Application()
# Background task started with the server
self._app.on_startup.append(self._on_startup)
2016-05-17 10:39:23 +00:00
# Allow CORS for this domains
cors = aiohttp_cors.setup(self._app, defaults={
2016-05-17 10:39:23 +00:00
# Default web server for web gui dev
2016-05-25 09:36:39 +00:00
"http://127.0.0.1:8080": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*"),
2016-05-17 15:51:22 +00:00
"http://localhost:8080": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*"),
"http://127.0.0.1:4200": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*"),
"http://localhost:4200": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*"),
"http://gns3.github.io": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*"),
"https://gns3.github.io": aiohttp_cors.ResourceOptions(expose_headers="*", allow_headers="*")
2016-05-17 10:39:23 +00:00
})
PortManager.instance().console_host = self._host
2015-01-14 00:05:26 +00:00
for method, route, handler in Route.get_routes():
2015-01-20 13:59:19 +00:00
log.debug("Adding route: {} {}".format(method, route))
cors.add(self._app.router.add_route(method, route, handler))
2015-01-14 00:05:26 +00:00
for module in MODULES:
2015-01-20 13:59:19 +00:00
log.debug("Loading module {}".format(module.__name__))
2015-01-15 15:59:01 +00:00
m = module.instance()
m.port_manager = PortManager.instance()
# adding static route
#self._app.router.add_static(
# '/static/',
# path=get_static_dir(),
# name='static'
#)
2015-01-20 13:59:19 +00:00
log.info("Starting server on {}:{}".format(self._host, self._port))
self._handler = self._app.make_handler()
if self._run_application(self._handler, ssl_context) is False:
self._loop.stop()
return
self._signal_handling()
self._exit_handling()
2015-02-20 21:40:20 +00:00
if server_config.getboolean("shell"):
asyncio.ensure_future(self.start_shell())
2015-02-20 21:40:20 +00:00
try:
self._loop.run_forever()
except TypeError as e:
# This is to ignore an asyncio.windows_events exception
# on Windows when the process gets the SIGBREAK signal
# TypeError: async() takes 1 positional argument but 3 were given
log.warning("TypeError exception in the loop {}".format(e))
finally:
if self._loop.is_running():
try:
self._loop.run_until_complete(self.shutdown_server())
except asyncio.CancelledError:
pass