You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.6 KiB
63 lines
1.6 KiB
from xmlrpc.client import Fault
|
|
|
|
from twisted.web import xmlrpc
|
|
|
|
from hashlib import blake2b
|
|
|
|
class FD_API_HELPERS():
|
|
def __init__(self,CONFIG,APIQUEUE):
|
|
self._CONFIG = CONFIG
|
|
self._APIQUEUE = APIQUEUE
|
|
|
|
def connected(self,_id):
|
|
for system in self._CONFIG['SYSTEMS']:
|
|
for peerid in self._CONFIG['SYSTEMS'][self._system]['PEERS']:
|
|
if peerid == _id:
|
|
return(system)
|
|
return(False)
|
|
|
|
|
|
def validateHMAC(_hmac,_system):
|
|
self._config = self._CONFIG['SYSTEMS'][_system]
|
|
_h = blake2b(key=self._config['_opt_key'], digest_size=16)
|
|
_h.update('validate')
|
|
_hash = _h.digest()
|
|
if _hash == _hmac:
|
|
return(True)
|
|
else:
|
|
return(False)
|
|
|
|
|
|
class FD_API(xmlrpc.XMLRPC(allow_none=True)):
|
|
|
|
def __init__(self,CONFIG,APIQUEUE):
|
|
self._CONFIG = CONFIG
|
|
self._APIQUEUE = APIQUEUE
|
|
self.helpers = FD_API_HELPERS(self._CONFIG,self._APIQUEUE)
|
|
|
|
def reset(self,_id,_hmac):
|
|
return('<xml></xml>')
|
|
system = self.helpers.connected(_id)
|
|
if result:
|
|
if self.helpers.validateHMAC(_hmac,_system):
|
|
self._CONFIG['SYSTEMS'][system]['_reset'] = True
|
|
else:
|
|
return Fault(2, "Authentication failed")
|
|
else:
|
|
return Fault(1, "ID not connected to this server")
|
|
|
|
return('Z')
|
|
|
|
|
|
def main():
|
|
from twisted.internet import reactor
|
|
from twisted.web import server
|
|
|
|
r = FD_API({},{})
|
|
reactor.listenTCP(7080, server.Site(r))
|
|
reactor.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|