Commit a89d00b5 authored by Ricard Vilalta's avatar Ricard Vilalta
Browse files

First commit

parents
Loading
Loading
Loading
Loading

.gitignore

0 → 100644
+139 −0
Original line number Diff line number Diff line
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST

# PyInstaller
#  Usually these files are written by a python script from a template
#  before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
cover/

# Translations
*.mo
*.pot

# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
.pybuilder/
target/

# Jupyter Notebook
.ipynb_checkpoints

# IPython
profile_default/
ipython_config.py

# pyenv
#   For a library or package, you might want to ignore these files since the code is
#   intended to run in multiple environments; otherwise, check them in:
# .python-version

# pipenv
#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
#   However, in case of collaboration, if having platform-specific dependencies or dependencies
#   having no cross-platform support, pipenv may install dependencies that don't work, or not
#   install all needed dependencies.
#Pipfile.lock

# PEP 582; used by e.g. github.com/David-OConnor/pyflow
__pypackages__/

# Celery stuff
celerybeat-schedule
celerybeat.pid

# SageMath parsed files
*.sage.py

# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/
.dmypy.json
dmypy.json

# Pyre type checker
.pyre/

# pytype static type analyzer
.pytype/

# Cython debug symbols
cython_debug/
+24 −0
Original line number Diff line number Diff line
// protocol buffers documentation: https://developers.google.com/protocol-buffers/docs/proto3
syntax = "proto3";
package attack_mitigator;

import "context.proto";

service AttackMitigator {
  rpc NotifyAttack (AttackDescription) returns (AttackResponse) {}
}

message AttackDescription {
  context.Uuid cs_id = 1;
  int32 attack_id = 2;
  float confidence = 3;
  string attack_description = 4;
}

message AttackResponse {
  context.Uuid cs_id = 1;
  int32 attack_id = 2;
  string attack_description = 3;
  int32 response_strategy_id = 4;
  string response_strategy_description = 5;
}

proto/automation.proto

0 → 100644
+51 −0
Original line number Diff line number Diff line
//Example of topology
syntax = "proto3";
package automation;

//import "google/protobuf/empty.proto";
import "context.proto";

service AutomationService {
  rpc ZtpGetDeviceRole(DeviceRoleId) returns (DeviceRole) {}
  rpc ZtpGetDeviceRolesByDeviceId(context.DeviceId) returns (DeviceRoleList) {}
  rpc ZtpAdd(DeviceRole) returns (DeviceRoleState) {}
  rpc ZtpUpdate(DeviceRole) returns (DeviceRoleState) {}
  rpc ZtpDelete(DeviceRole) returns (DeviceRoleState) {}
  rpc ZtpDeleteAllByDeviceId(context.DeviceId) returns (DeviceRoleState) {}
}

enum DeviceRoleType {
  NONE = 0;
  DEV_OPS = 1;
  DEV_CONF = 2;
  PIPELINE_CONF = 3;
}

message DeviceRoleId {
  context.Uuid devRoleId = 1;
  context.DeviceId devId = 2;
}

message DeviceRole {
  DeviceRoleId devRoleId = 1;
  DeviceRoleType devRoleType = 2;
}

message DeviceRoleList {
  repeated DeviceRole devRole = 1;
}

message DeviceRoleState {
  DeviceRoleId devRoleId = 1;
  ZtpDeviceState devRoleState = 2;
}

enum ZtpDeviceState {
  PLANNED  = 0;
  POTENCIAL_AVAILABLE = 1;
  POTENCIAL_BUSY = 2;
  INSTALLED = 3;
  PENDING_REMOVAL = 4;
}

+23 −0
Original line number Diff line number Diff line
// protocol buffers documentation: https://developers.google.com/protocol-buffers/docs/proto3
syntax = "proto3";
package centralized_cybersecurity;

import "google/protobuf/empty.proto";
import "service.proto";
import "monitoring.proto";


service CentralizedCyberSecurityService {

  rpc NotifyServiceUpdate (service.Service) returns (google.protobuf.Empty) {}
  
  // rpc that triggers the attack detection loop
  rpc DetectAttack (google.protobuf.Empty) returns (google.protobuf.Empty) {}

  // rpc called by the distributed component to report KPIs
  rpc ReportSummarizedKpi (monitoring.KpiList) returns (google.protobuf.Empty) {}

  rpc ReportKpi (monitoring.KpiList) returns (google.protobuf.Empty) {}

}

proto/compile.sh

0 → 100755
+10 −0
Original line number Diff line number Diff line
#!/bin/bash
python3 -m grpc_tools.protoc -I=./ --python_out=src/ --grpc_python_out=src/ *.proto
#requires installation of protoc-gen-uml
export PATH=/home/osboxes/protoc-gen-uml/target/universal/stage/bin:$PATH
protoc --uml_out=uml/ -I ./ *.proto
cd uml
java -jar plantuml.7997.jar *.puml
rm *.puml