use one module for the complete ES communication
This commit is contained in:
52
aman/com/Euroscope.py
Normal file
52
aman/com/Euroscope.py
Normal file
@@ -0,0 +1,52 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import glob
|
||||
import os
|
||||
import sys
|
||||
|
||||
import zmq
|
||||
from zmq.asyncio import Context
|
||||
import zmq.auth
|
||||
from zmq.auth.asyncio import AsyncioAuthenticator
|
||||
|
||||
from aman.config import Server
|
||||
|
||||
# @brief Receives and sends messages to EuroScope plugins
|
||||
class Euroscope:
|
||||
# @brief Initializes the ZMQ socket
|
||||
# @param[in] config The server configuration
|
||||
def __init__(self, config : Server.Server):
|
||||
self.context = Context.instance()
|
||||
|
||||
# initialize the authentication module
|
||||
authLocation = (
|
||||
str(config.ClientKeyPath)
|
||||
)
|
||||
self.auth = AsyncioAuthenticator(context = self.context)
|
||||
self.auth.configure_curve(domain='*', location = authLocation)
|
||||
self.auth.allow('127.0.0.1')
|
||||
self.auth.start()
|
||||
|
||||
# read the certificates
|
||||
keyPairPath = glob.glob(os.path.join(config.ServerKeyPath, '*.key_secret'))
|
||||
if 1 != len(keyPairPath):
|
||||
sys.stderr.write('No public-private keypair found for the server certificate')
|
||||
sys.exit(-1)
|
||||
keyPair = zmq.auth.load_certificate(keyPairPath[0])
|
||||
|
||||
# initialize the receiver
|
||||
self.receiverSocket = self.context.socket(zmq.ROUTER)
|
||||
self.receiverSocket.setsockopt(zmq.CURVE_PUBLICKEY, keyPair[0])
|
||||
self.receiverSocket.setsockopt(zmq.CURVE_SECRETKEY, keyPair[1])
|
||||
self.receiverSocket.setsockopt(zmq.CURVE_SERVER, True)
|
||||
self.receiverSocket.bind('tcp://' + config.Address + ':' + str(config.PortReceiver))
|
||||
|
||||
# initialize the notification
|
||||
self.notificationSocket = self.context.socket(zmq.DEALER)
|
||||
self.notificationSocket.setsockopt(zmq.CURVE_PUBLICKEY, keyPair[0])
|
||||
self.notificationSocket.setsockopt(zmq.CURVE_SECRETKEY, keyPair[1])
|
||||
self.notificationSocket.setsockopt(zmq.CURVE_SERVER, True)
|
||||
self.notificationSocket.bind('tcp://' + config.Address + ':' + str(config.PortNotification))
|
||||
|
||||
def __exit__(self, *_exc):
|
||||
self.auth.stop()
|
||||
Reference in New Issue
Block a user