class NetworkNetworkType(db.Model):
"""Association table for Network Types and Networks."""
__tablename__ = 'network_networktype'
id = db.Column(db.Integer(), primary_key=True)
network_id = db.Column(db.Integer(),
db.ForeignKey('network.id', ondelete='CASCADE'))
networktype_id = db.Column(
db.Integer(), db.ForeignKey('networktype.id', ondelete='CASCADE'))
class SystemSystemType(db.Model):
"""Association table for System Types and Systems."""
__tablename__ = 'system_systemtype'
id = db.Column(db.Integer(), primary_key=True)
system_id = db.Column(db.Integer(),
db.ForeignKey('system.id', ondelete='CASCADE'))
systemtype_id = db.Column(
db.Integer(), db.ForeignKey('systemtype.id', ondelete='CASCADE'))
class WizardState(db.Model):
"""Model to represent the current state the wizard is in."""
__tablename__ = 'wizardstate'
id = db.Column(db.Integer(), primary_key=True)
state_id = db.Column(db.ForeignKey('wizardstatenames.id'), nullable=False)
def __repr__(self) -> str:
return f"Wizard State: {self.state.name}"
class BOX4securitySystemType(db.Model):
"""Association table for System Types and BOX4security."""
__tablename__ = 'box4security_systemtype'
id = db.Column(db.Integer(), primary_key=True)
box4security = db.Column(
db.Integer(), db.ForeignKey('box4security.id', ondelete='CASCADE'))
systemtype_id = db.Column(
db.Integer(), db.ForeignKey('systemtype.id', ondelete='CASCADE'))
class SystemType(db.Model):
"""Model class for System Types."""
__tablename__ = 'systemtype'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(100)) # System Type Name
def __repr__(self):
"""Print SystemType in human readable form."""
return '{} ({})'.format(self.name, self.id)
class UserRole(db.Model):
"""Association table for Users and Roles."""
__tablename__ = 'user_role'
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'))
role_id = db.Column(db.Integer(),
db.ForeignKey('role.id', ondelete='CASCADE'))
class ScanCategory(db.Model):
"""Model class for Vulnerability Scan Categories."""
__tablename__ = 'scancategory'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String())
networks = db.relationship('Network', backref='scan_category')
def __repr__(self):
"""Print ScanCategory in human readable form."""
return '{} ({})'.format(self.name, self.id)
class Role(db.Model):
"""Role class for defining permissions."""
__tablename__ = 'role'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), unique=True)
description = db.Column(db.String(255))
def __repr__(self):
"""Print Role in human readable form."""
return '"{}": "{}"'.format(self.name, self.description)
class Network(db.Model):
"""Network model class.
Explanation of non-trivial fields:
scan_category: Scan Category as defined before:
1: No restrictions
2: Scans only at weekend or non-busy times
3: Scans only when admins and 4s are available
scan_weekday: lower cased weekday for scans
scan_time: start time for the scan on `scan_weekday`
types: List of associated NetworkTypes
"""
__tablename__ = 'network'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(100)) # Network Name
ip_address = db.Column(db.String(24), nullable=False) # Network Address
cidr = db.Column(db.Integer(), nullable=False) # CIDR-Number
vlan = db.Column(db.String(50)) # VLAN Tag
types = db.relationship('NetworkType', secondary='network_networktype')
scancategory_id = db.Column(db.Integer, db.ForeignKey('scancategory.id'))
scan_weekday = db.Column(db.String(24)) # lower case
scan_time = db.Column(db.Time()) # Start time for scan
systems = db.relationship('System', backref='network')
boxes4security = db.relationship('BOX4security', backref='network')
def __repr__(self):
"""Print Network in human readable form."""
return '{} ({}): {}/{}'.format(self.name, self.id, self.ip_address,
self.cidr)
class LogstashRule(db.Model):
__tablename__ = 'blocks_by_logstashfilter'
id = db.Column(db.Integer, primary_key=True)
src_ip = db.Column(db.String)
src_port = db.Column(db.Integer)
dst_ip = db.Column(db.String)
dst_port = db.Column(db.Integer)
proto = db.Column(db.String(4))
signature_id = db.Column(db.String(10))
signature = db.Column(db.String(256))
class BOX4security(db.Model):
"""Extension of BOX4security model."""
__tablename__ = 'box4security'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(100), unique=True)
ip_address = db.Column(db.String(24)) # BOX4security IP Address
types = db.relationship('SystemType', secondary='box4security_systemtype')
location = db.Column(db.String(255)) # BOX4security Location
scan_enabled = db.Column(db.Boolean(), default=False) # Scans active
ids_enabled = db.Column(db.Boolean(), default=False) # IDS enabled
network_id = db.Column(db.Integer, db.ForeignKey('network.id'))
dns_id = db.Column(db.Integer, db.ForeignKey('system.id'))
gateway_id = db.Column(db.Integer, db.ForeignKey('system.id'))
dns = db.relationship('System', foreign_keys=[dns_id], uselist=False)
gateway = db.relationship('System',
foreign_keys=[gateway_id],
uselist=False)
dhcp_enabled = db.Column(db.Boolean(), default=False) # dhcp enabled
def __repr__(self):
return f"BOX4s ({self.ip_address}) DNS:{self.dns.ip_address} Gateway:{self.gateway.ip_address}"
class BPFRule(db.Model):
__tablename__ = 'blocks_by_bpffilter'
id = db.Column(db.Integer, primary_key=True)
src_ip = db.Column(db.String)
src_port = db.Column(db.Integer)
dst_ip = db.Column(db.String)
dst_port = db.Column(db.Integer)
proto = db.Column(db.String(4))
class System(db.Model):
"""System model class.
Explanation of non-trivial fields:
scan_enabled: True if vulnerability scans should be enabled for the system, else False
ids_enabled: True if IDS should be enabled for the system, else False
types: List of associated SystemTypes
"""
__tablename__ = 'system'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(100), unique=True)
ip_address = db.Column(db.String(24)) # System IP Address
types = db.relationship('SystemType', secondary='system_systemtype')
location = db.Column(db.String(255)) # System Location
scan_enabled = db.Column(db.Boolean(), default=True) # Scans active
ids_enabled = db.Column(db.Boolean(), default=True) # IDS enabled
network_id = db.Column(db.Integer, db.ForeignKey('network.id'))
class User(db.Model, UserMixin):
"""User class to handle authentication and authorization.
active: can login e.g. not banned
email_confirmed_at: Zeitstempel, an dem die E-Mail bestÃĪtigt wurde
roles: Liste der, dem User zugeorndeten Regeln
"""
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
active = db.Column('is_active',
db.Boolean(),
nullable=False,
server_default='1')
# User authentication information. The collation='NOCASE' is required
# to search case insensitively when USER_IFIND_MODE is 'nocase_collation'.
email = db.Column(db.String(255), nullable=False, unique=True)
email_confirmed_at = db.Column(db.DateTime())
password = db.Column(db.String(255), nullable=False, server_default='')
# User information
first_name = db.Column(db.String(100), nullable=False, server_default='')
last_name = db.Column(db.String(100), nullable=False, server_default='')
# Define the relationship to Role via UserRoles
roles = db.relationship('Role', secondary='user_role')
def has_role(self, role):
# For each role the user has, check, if the given role is actually in the users roles
# But if the user is a super admin, always return true, as they are allowed to to everything
for r in self.roles:
if r.name == role or r.name == "Super Admin":
return True
return False
def getName(self):
"""Return name of current user or email if no name exists."""
if self.last_name:
return " ".join(filter(None, [self.first_name, self.last_name]))
else:
return self.email
class WizardStateNames(db.Model):
"""Model to assign changeable names to state ids."""
__tablename__ = 'wizardstatenames'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String())
stateObj = db.relationship('WizardState', backref="state", uselist=False)
