parent
448c853760
commit
4ca991f98b
@ -0,0 +1,62 @@
|
||||
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()
|
||||
@ -0,0 +1,23 @@
|
||||
|
||||
from twisted.internet import reactor
|
||||
from twisted.web.xmlrpc import Proxy
|
||||
|
||||
|
||||
def printValue(value):
|
||||
print(repr(value))
|
||||
reactor.stop()
|
||||
|
||||
|
||||
def printError(error):
|
||||
print("error", error)
|
||||
reactor.stop()
|
||||
|
||||
|
||||
def capitalize(value):
|
||||
print(value)
|
||||
|
||||
|
||||
proxy = Proxy(b"http://localhost:7080/xmlrpc")
|
||||
# The callRemote method accepts a method name and an argument list.
|
||||
proxy.callRemote("FD_API.reset", '2', '55555').addCallbacks(capitalize, printError)
|
||||
reactor.run()
|
||||
Loading…
Reference in new issue