diff --git a/src/l3_attackmitigator/service/__main__.py b/src/l3_attackmitigator/service/__main__.py
index 394d3c3f9846077c9f6ffc9463b369c89608da9b..3843c6c18e4dc6f72c07bad6e0a71880a73b82de 100644
--- a/src/l3_attackmitigator/service/__main__.py
+++ b/src/l3_attackmitigator/service/__main__.py
@@ -34,11 +34,11 @@ def main():
start_http_server(metrics_port)
# Get database instance
- database = get_database()
+ #database = get_database()
# Starting l3_attackmitigator service
grpc_service = l3_attackmitigatorService(
- database, port=service_port, max_workers=max_workers, grace_period=grace_period)
+ port=service_port, max_workers=max_workers, grace_period=grace_period)
grpc_service.start()
# Wait for Ctrl+C or termination signal
diff --git a/src/l3_attackmitigator/service/l3_attackmitigatorService.py b/src/l3_attackmitigator/service/l3_attackmitigatorService.py
index bfad98200ccfba99ffc1b56d5138b60122eb3a3a..254457c705e7311b34d5a2172fd208c3a3bf6657 100644
--- a/src/l3_attackmitigator/service/l3_attackmitigatorService.py
+++ b/src/l3_attackmitigator/service/l3_attackmitigatorService.py
@@ -23,13 +23,11 @@ LOGGER = logging.getLogger(__name__)
class l3_attackmitigatorService:
def __init__(
self,
- database,
address=BIND_ADDRESS,
port=GRPC_SERVICE_PORT,
max_workers=GRPC_MAX_WORKERS,
grace_period=GRPC_GRACE_PERIOD,
):
- self.database = database
self.address = address
self.port = port
self.endpoint = None
@@ -52,7 +50,7 @@ class l3_attackmitigatorService:
self.server = grpc.server(self.pool) # , interceptors=(tracer_interceptor,))
self.l3_attackmitigator_servicer = (
- l3_attackmitigatorServiceServicerImpl(self.database)
+ l3_attackmitigatorServiceServicerImpl()
)
add_L3AttackmitigatorServicer_to_server(
self.l3_attackmitigator_servicer, self.server
diff --git a/src/l3_attackmitigator/service/l3_attackmitigatorServiceServicerImpl.py b/src/l3_attackmitigator/service/l3_attackmitigatorServiceServicerImpl.py
index 0e2a64c1e9a89469784600ff154830436e634221..660d790bf827469904624189f7b4ec70be746b35 100644
--- a/src/l3_attackmitigator/service/l3_attackmitigatorServiceServicerImpl.py
+++ b/src/l3_attackmitigator/service/l3_attackmitigatorServiceServicerImpl.py
@@ -27,10 +27,8 @@ LAST_VALUE = -1
class l3_attackmitigatorServiceServicerImpl(L3AttackmitigatorServicer):
- def __init__(self, database: Database):
+ def __init__(self):
LOGGER.debug("Creating Servicer...")
- self.database = database
- LOGGER.debug("Servicer Created")
class Mitigator(L3AttackmitigatorServicer):
def send_output(self, request, context):
diff --git a/src/l3_attackmitigator/tests/test_unitary.py b/src/l3_attackmitigator/tests/test_unitary.py
index 4acecd94e81eb9efceba4789acb61359a0b5fa8c..54579472f4c6e03931c3622270ded5ae30f9a184 100644
--- a/src/l3_attackmitigator/tests/test_unitary.py
+++ b/src/l3_attackmitigator/tests/test_unitary.py
@@ -25,15 +25,9 @@ LOGGER.setLevel(logging.DEBUG)
@pytest.fixture(scope='session')
-def database():
- _database = get_database(engine=DatabaseEngineEnum.INMEMORY)
- return _database
-
-
-@pytest.fixture(scope='session')
-def l3_attackmitigator_service(database):
+def l3_attackmitigator_service():
_service = l3_attackmitigatorService(
- database, port=port, max_workers=GRPC_MAX_WORKERS, grace_period=GRPC_GRACE_PERIOD)
+ port=port, max_workers=GRPC_MAX_WORKERS, grace_period=GRPC_GRACE_PERIOD)
_service.start()
yield _service
_service.stop()
@@ -49,10 +43,10 @@ def test_demo():
print('Demo Test')
pass
-def test_grpc_server(database):
+def test_grpc_server():
print('Starting AM')
_service = l3_attackmitigatorService(
- database, port=port, max_workers=GRPC_MAX_WORKERS, grace_period=GRPC_GRACE_PERIOD)
+ port=port, max_workers=GRPC_MAX_WORKERS, grace_period=GRPC_GRACE_PERIOD)
p1 = multiprocessing.Process(target=_service.start, args=())
#_service.start()
p1.start()
diff --git a/src/l3_centralizedattackdetector/service/__main__.py b/src/l3_centralizedattackdetector/service/__main__.py
index eac7b5f22e285d5a4a63452fac1e45325f0325d3..1e593111bf580e692115de502c2817cea0a821dc 100644
--- a/src/l3_centralizedattackdetector/service/__main__.py
+++ b/src/l3_centralizedattackdetector/service/__main__.py
@@ -33,12 +33,10 @@ def main():
# Start metrics server
start_http_server(metrics_port)
- # Get database instance
- database = get_database()
# Starting l3_centralizedattackdetector service
grpc_service = l3_centralizedattackdetectorService(
- database, port=service_port, max_workers=max_workers, grace_period=grace_period)
+ port=service_port, max_workers=max_workers, grace_period=grace_period)
grpc_service.start()
# Wait for Ctrl+C or termination signal
diff --git a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorService.py b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorService.py
index 7c71fc77f53759b5c2f4ee9897f14b5117639d96..1bcb50e1e07a1115b457921b1a965791ba4437b7 100644
--- a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorService.py
+++ b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorService.py
@@ -23,13 +23,11 @@ LOGGER = logging.getLogger(__name__)
class l3_centralizedattackdetectorService:
def __init__(
self,
- database,
address=BIND_ADDRESS,
port=GRPC_SERVICE_PORT,
max_workers=GRPC_MAX_WORKERS,
grace_period=GRPC_GRACE_PERIOD,
):
- self.database = database
self.address = address
self.port = port
self.endpoint = None
@@ -52,7 +50,7 @@ class l3_centralizedattackdetectorService:
self.server = grpc.server(self.pool) # , interceptors=(tracer_interceptor,))
self.l3_centralizedattackdetector_servicer = (
- l3_centralizedattackdetectorServiceServicerImpl(self.database)
+ l3_centralizedattackdetectorServiceServicerImpl()
)
add_L3CentralizedattackdetectorServicer_to_server(
self.l3_centralizedattackdetector_servicer, self.server
diff --git a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
index 0e137f4249a0cd1aff97b3f940d73cd024a60b30..4ff4f58eebe1e1c5adbe5aa5cd696139a93d1641 100644
--- a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
+++ b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
@@ -25,7 +25,7 @@ from l3_centralizedattackdetector.proto.l3_attackmitigator_pb2 import (
Output,
)
from l3_centralizedattackdetector.proto.l3_attackmitigator_pb2_grpc import (
- l3_attackmitigatorStub,
+ L3AttackmitigatorStub,
)
LOGGER = logging.getLogger(__name__)
@@ -36,10 +36,8 @@ INFERENCE_VALUES = []
class l3_centralizedattackdetectorServiceServicerImpl(L3CentralizedattackdetectorServicer):
- def __init__(self, database: Database):
+ def __init__(self):
LOGGER.debug("Creating Servicer...")
- self.database = database
- LOGGER.debug("Servicer Created")
class CAD(L3CentralizedattackdetectorServicer):
def __init__(self, ml_model):
@@ -61,7 +59,7 @@ class l3_centralizedattackdetectorServiceServicerImpl(L3Centralizedattackdetecto
with grpc.insecure_channel("localhost:10002") as channel:
stub = l3_attackmitigatorStub(channel)
print("Sending to mitigator...")
- response = stub.send_output(output)
+ response = stub.SendOutput(output)
print("Sent output to mitigator and received: ", response.message)
# RETURN "OK" TO THE CALLER
diff --git a/src/l3_distributedattackdetector/service/l3_distributedattackdetectorServiceServicerImpl.py b/src/l3_distributedattackdetector/service/l3_distributedattackdetectorServiceServicerImpl.py
index e90fe3128fbd23c66190d0bd0fca6ccde32aa133..0f8cc3d31ba0e3883e0c879da87d322ce9b0fd69 100644
--- a/src/l3_distributedattackdetector/service/l3_distributedattackdetectorServiceServicerImpl.py
+++ b/src/l3_distributedattackdetector/service/l3_distributedattackdetectorServiceServicerImpl.py
@@ -11,12 +11,13 @@ from l3_distributedattackdetector.proto.l3_centralizedattackdetector_pb2 import
)
from l3_distributedattackdetector.proto.l3_centralizedattackdetector_pb2_grpc import (
L3CentralizedattackdetectorStub,
+ L3CentralizedattackdetectorServicer
)
LOGGER = logging.getLogger(__name__)
TSTAT_DIR_NAME = "piped/"
-class l3_distributedattackdetectorServiceServicerImpl():
+class l3_distributedattackdetectorServiceServicerImpl(L3CentralizedattackdetectorServicer):
def __init__(self):
LOGGER.debug("Creating Servicer...")
@@ -116,7 +117,7 @@ class l3_distributedattackdetectorServiceServicerImpl():
def open_channel(self, input_information):
with grpc.insecure_channel("localhost:10001") as channel:
stub = L3CentralizedattackdetectorStub(channel)
- response = stub.send_input(
+ response = stub.SendInput(
ModelInput(**input_information))
logging.debug("Inferencer send_input sent and received: ",
response.message)