mirror of git://holbrook.no/eth-monitor.git
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
# standard imports
|
|
import logging
|
|
import uuid
|
|
|
|
# external imports
|
|
from chainlib.eth.address import is_same_address
|
|
|
|
logg = logging.getLogger()
|
|
|
|
|
|
class RuleMethod:
|
|
|
|
def __init__(self, methods, description=None):
|
|
self.methods = methods
|
|
self.description = description
|
|
if self.description == None:
|
|
self.description = str(uuid.uuid4())
|
|
|
|
|
|
def check(self, sender, recipient, data, tx_hash):
|
|
if len(self.methods) == 0:
|
|
return False
|
|
|
|
for method in self.methods:
|
|
l = len(method)
|
|
if len(method) > len(data):
|
|
continue
|
|
if data[:l] == method:
|
|
logg.debug('tx {} rule {} match in DATA {}'.format(tx_hash, self.description, method))
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def __str__(self):
|
|
return 'Method ' + self.description + ' {}'.format(
|
|
self.methods,
|
|
)
|
|
|
|
|
|
class RuleSimple:
|
|
|
|
def __init__(self, outputs, inputs, executables, description=None):
|
|
self.description = description
|
|
if self.description == None:
|
|
self.description = str(uuid.uuid4())
|
|
self.outputs = outputs
|
|
self.inputs = inputs
|
|
self.executables = executables
|
|
|
|
|
|
def check(self, sender, recipient, data, tx_hash):
|
|
for rule in self.outputs:
|
|
if rule != None and is_same_address(sender, rule):
|
|
logg.debug('tx {} rule {} match in SENDER {}'.format(tx_hash, self.description, sender))
|
|
return True
|
|
for rule in self.inputs:
|
|
if rule != None and is_same_address(recipient, rule):
|
|
logg.debug('tx {} rule {} match in RECIPIENT {}'.format(tx_hash, self.description, recipient))
|
|
return True
|
|
for rule in self.executables:
|
|
if rule != None and is_same_address(recipient, rule):
|
|
logg.debug('tx {} rule {} match in EXECUTABLE {}'.format(tx_hash, self.description, recipient))
|
|
return True
|
|
|
|
|
|
def __str__(self):
|
|
return 'Simple ' + self.description + ' outputs {} inputs {} execs {}'.format(
|
|
self.outputs,
|
|
self.inputs,
|
|
self.executables,
|
|
)
|
|
|
|
|
|
class AddressRules:
|
|
|
|
def __init__(self, include_by_default=False):
|
|
self.excludes = []
|
|
self.includes = []
|
|
self.include_by_default = include_by_default
|
|
|
|
|
|
def exclude(self, rule):
|
|
self.excludes.append(rule)
|
|
logg.info('cache filter added EXCLUDE rule {}'.format(rule))
|
|
|
|
|
|
def include(self, rule):
|
|
self.includes.append(rule)
|
|
logg.info('cache filter added INCLUDE rule {}'.format(rule))
|
|
|
|
|
|
def apply_rules(self, tx):
|
|
return self.apply_rules_addresses(tx.outputs[0], tx.inputs[0], tx.payload, tx.hash)
|
|
|
|
|
|
# TODO: rename
|
|
def apply_rules_addresses(self, sender, recipient, data, tx_hash):
|
|
v = self.include_by_default
|
|
|
|
for rule in self.includes:
|
|
if rule.check(sender, recipient, data, tx_hash):
|
|
v = True
|
|
logg.info('match in includes rule: {}'.format(rule))
|
|
break
|
|
|
|
for rule in self.excludes:
|
|
if rule.check(sender, recipient, data, tx_hash):
|
|
v = False
|
|
logg.info('match in excludes rule: {}'.format(rule))
|
|
break
|
|
|
|
return v
|