def run(self, config_yaml):
is_config_loaded = self.load_config_from_yaml(config_yaml)
if not is_config_loaded:
base.fatal_fail("Unable to load config file, can't parse the YAML!")
server = websockets.serve(self.proxy_dispatcher, self.host, self.port)
self.logger.log("Initializing PROXY SERVER")
asyncio.get_event_loop().run_until_complete(server)
asyncio.get_event_loop().run_forever()
def run(self, config_yaml):
is_config_loaded = self.load_config_from_yaml(config_yaml)
if not is_config_loaded:
base.fatal_fail("Unable to load config file, can't parse the YAML!")
server = websockets.serve(self.proxy_dispatcher, self.host, self.port)
self.logger.log("Initializing PROXY SERVER")
asyncio.get_event_loop().run_until_complete(server)
asyncio.get_event_loop().run_forever()
def load_server_config_from_yaml(self, config_yaml):
server_config = config_yaml['configuration']['serverConfiguration']
self.host = server_config['listenHost']
self.port = int(server_config['port'])
self.serverType = server_config['type']
self.requests_per_connection = int(server_config['requestsPerConnection'])
if not self.has_valid_server_type():
self.logger.log("Server type value [" + self.serverType + "] in config is invalid. Can't start server")
base.fatal_fail(None)
if self.is_forced_url_server() or self.is_forced_url_no_password_server():
self.proxied_url = server_config['proxiedUrl']
if self.proxied_url is None or self.proxied_url == "":
error_message = "Proxied url in config missing--required when running in FORCED_URL mode."
self.logger.log(error_message)
base.fatal_fail(None)
def load_server_config_from_yaml(self, config_yaml):
server_config = config_yaml['configuration']['serverConfiguration']
self.host = server_config['listenHost']
self.port = int(server_config['port'])
self.serverType = server_config['type']
self.requests_per_connection = int(server_config['requestsPerConnection'])
if not self.has_valid_server_type():
self.logger.log("Server type value [" + self.serverType + "] in config is invalid. Can't start server")
base.fatal_fail(None)
if self.is_forced_url_server() or self.is_forced_url_no_password_server():
self.proxied_url = server_config['proxiedUrl']
if self.proxied_url is None or self.proxied_url == "":
error_message = "Proxied url in config missing--required when running in FORCED_URL mode."
self.logger.log(error_message)
base.fatal_fail(None)
from websocket_proxpy.util import base
try:
import websockets
except ImportError:
base.fatal_fail(
"'websockets' library required (pip install websockets). Exiting.")
from websocket_proxpy.util.jsonutils import get_json_status_response
import asyncio
import json
class WebSocketProxpy:
logger = None
host = "localhost"
port = 1111
serverType = "OPEN_URL"
proxied_url = ""
password = ""
send_suffix = ""
send_prefix = ""
requests_per_connection = 10000
def __init__(self, logger):
self.logger = logger
def is_open_url_server(self):
return self.serverType == "OPEN_URL"
def is_forced_url_server(self):
return self.serverType == "FORCED_URL"
from websocket_proxpy.proxy import WebSocketProxpy
from websocket_proxpy.util import loggers
from websocket_proxpy.util import base
try:
import yaml
except ImportError:
base.fatal_fail("'yaml' library required (pip install yaml). Exiting.")
CONFIG_FILE_NAME = "config.yaml"
config = yaml.load(open(CONFIG_FILE_NAME))
WebSocketProxpy(loggers.ConsoleDebugLogger()).run(config)
from websocket_proxpy.util import base
try:
import websockets
except ImportError:
base.fatal_fail("'websockets' library required (pip install websockets). Exiting.")
from websocket_proxpy.util.jsonutils import get_json_status_response
import asyncio
import json
class WebSocketProxpy:
logger = None
host = "localhost"
port = 1111
serverType = "OPEN_URL"
proxied_url = ""
password = ""
send_suffix = ""
send_prefix = ""
requests_per_connection = 10000
def __init__(self, logger):
self.logger = logger
def is_open_url_server(self):
return self.serverType == "OPEN_URL"
def is_forced_url_server(self):
return self.serverType == "FORCED_URL"
def is_forced_url_no_password_server(self):