def __init__(self, keys_dirname=0, password=0):
self.server = ServerIdentity(keys_dirname, password)
self.ticket_manager = TicketManager(self.server)
self.storage = SafeBoxStorage(self.server)
def __init__(self, keys_dirname=0, password=0):
self.server = ServerIdentity(keys_dirname, password)
self.ticket_manager = TicketManager(self.server)
self.storage = SafeBoxStorage(self.server)
class AccessCtrlHandler(object):
def __init__(self, keys_dirname=0, password=0):
self.server = ServerIdentity(keys_dirname, password)
self.ticket_manager = TicketManager(self.server)
self.storage = SafeBoxStorage(self.server)
# Handling Session resource related operations:
#
def handleGetKey(self):
key = self.server.pub_key.exportKey('PEM')
print key
reply_dict = { 'status': "OK", 'key': key }
return json.dumps(reply_dict, sort_keys=True, encoding="utf-8")
# handleGetTicket: Checks if the userd ccid exists in the database,
# if it does returns a ticket.
def handleGetTicket(self, request):
def getTicket_cb(data):
if not data:
reply_dict = { 'status': {'error': "Invalid Request",
'message': 'User does not exist.'} }
else:
pboxid = data[0][0]
pubkey = data[0][1]
print pubkey
ticket = self.ticket_manager.generateTicket(pboxid, pubkey)
reply_dict = { 'status': "OK", 'ticket': ticket}
request.write( json.dumps(reply_dict, sort_keys=True, encoding="utf-8") )
request.finish()
d = self.storage.getClientData(request)
d.addCallback(getTicket_cb)
return NOT_DONE_YET
# handleValidation: handles the validation process for a given method
# only calls method if the provided ticket is valid.
def handleValidation(self, request, method, ticket=None):
# ticket will only exist from arguments when this method
# is being called to validate data transfer operations
if not ticket:
ticket = request.content.read(TICKET_SIZE)
print str(ticket)
if not ticket:
reply_dict = { 'status': {'error': "Invalid Request",
'message': "No ticket on request body."} }
return json.dumps(reply_dict, encoding="utf-8")
def handleValidation_cb(data):
if not data:
reply_dict = { 'status': {'error': "Invalid Request",
'message': 'User does not exist.'} }
else:
pboxid = data[0][0]
pubkey = data[0][1]
print pubkey
if self.ticket_manager.validateTicket(ticket, pboxid, pubkey):
print "Valid Ticket!"
d = method(request, pboxid, pubkey)
return NOT_DONE_YET
else:
print "Invalid Ticket!"
reply_dict = { 'status': {'error': "Invalid Ticket",
'message': 'N/A'} }
request.write( json.dumps(reply_dict, sort_keys=True, encoding="utf-8") )
request.finish()
d = self.storage.getClientData(request)
d.addCallback(handleValidation_cb)
return NOT_DONE_YET
# Handling PBoxes resource related operations:
#
#handle ListPBoxes
def handleListPBoxes(self, request):
return self.handleValidation(request, self.storage.listPBoxes)
def handleGetPBoxMData(self, request):
return self.handleValidation(request, self.storage.getPBoxMData)
# handleRegisterPBox: Checks if client exists, if so returns error, else registers the client.
def handleRegisterPBox(self, request):
# Checking if the client exists.
# pprint(request.__dict__)
def checkClientExists_cb(data, key_txt):
if data:
reply_dict = { 'status': {'error': "Invalid Request",
'message': 'User already exists.'} }
request.write( json.dumps(reply_dict, sort_keys=True, encoding="utf-8") )
request.finish()
else:
d = self.storage.registerPBox(request, key_txt)
return NOT_DONE_YET
# Validating key.
key_txt = request.content.read()
if not key_txt:
reply_dict = { 'status': {'error': "Invalid Request",
'message': "No key on request body."} }
return json.dumps(reply_dict, encoding="utf-8")
cli_key = RSA.importKey(key_txt)
if not cli_key.can_encrypt():
reply_dict = { 'status': {'error': "Invalid Request",
'message': "No key on request body."} }
return json.dumps(reply_dict, encoding="utf-8")
d = self.storage.getClientData(request)
d.addCallback(checkClientExists_cb, key_txt)
return NOT_DONE_YET
# Handling Files resource related operations:
#
def handleListFiles(self, request):
return self.handleValidation(request, self.storage.listFiles)
def handleGetFileMData(self, request):
return self.handleValidation(request, self.storage.getFileMData)
def handleGetFile(self, request):
return self.handleValidation(request, self.storage.getFile)
def handlePutFile(self, request):#
return self.handleValidation(request, self.storage.putFile)
def handleUpdateFile(self, request):#
return self.handleValidation(request, self.storage.updateFile)
def handleDeleteFile(self, request):
return self.handleValidation(request, self.storage.deleteFile)
# Handling Share resource related operations:
#
def handleShareFile(self, request):
return self.handleValidation(request, self.storage.shareFile)
def handleGetShareMData(self, request):
return self.handleValidation(request, self.storage.getShareMData)
def handleGetShared(self, request):
return self.handleValidation(request, self.storage.getShared)
def handleListShares(self, request):
return self.handleValidation(request, self.storage.listShares)
def handleUpdateShared(self, request):
return self.handleValidation(request, self.storage.updateShared)
def handleUpdateSharePerm(self, request):
return self.handleValidation(request, self.storage.updateSharePerm)
def handleDeleteShare(self, request):
return self.handleValidation(request, self.storage.deleteShare)
class AccessCtrlHandler(object):
def __init__(self, keys_dirname=0, password=0):
self.server = ServerIdentity(keys_dirname, password)
self.ticket_manager = TicketManager(self.server)
self.storage = SafeBoxStorage(self.server)
# Handling Session resource related operations:
#
def handleGetKey(self):
key = self.server.pub_key.exportKey('PEM')
print key
reply_dict = {'status': "OK", 'key': key}
return json.dumps(reply_dict, sort_keys=True, encoding="utf-8")
# handleGetTicket: Checks if the userd ccid exists in the database,
# if it does returns a ticket.
def handleGetTicket(self, request):
def getTicket_cb(data):
if not data:
reply_dict = {
'status': {
'error': "Invalid Request",
'message': 'User does not exist.'
}
}
else:
pboxid = data[0][0]
pubkey = data[0][1]
print pubkey
ticket = self.ticket_manager.generateTicket(pboxid, pubkey)
reply_dict = {'status': "OK", 'ticket': ticket}
request.write(
json.dumps(reply_dict, sort_keys=True, encoding="utf-8"))
request.finish()
d = self.storage.getClientData(request)
d.addCallback(getTicket_cb)
return NOT_DONE_YET
# handleValidation: handles the validation process for a given method
# only calls method if the provided ticket is valid.
def handleValidation(self, request, method, ticket=None):
# ticket will only exist from arguments when this method
# is being called to validate data transfer operations
if not ticket:
ticket = request.content.read(TICKET_SIZE)
print str(ticket)
if not ticket:
reply_dict = {
'status': {
'error': "Invalid Request",
'message': "No ticket on request body."
}
}
return json.dumps(reply_dict, encoding="utf-8")
def handleValidation_cb(data):
if not data:
reply_dict = {
'status': {
'error': "Invalid Request",
'message': 'User does not exist.'
}
}
else:
pboxid = data[0][0]
pubkey = data[0][1]
print pubkey
if self.ticket_manager.validateTicket(ticket, pboxid, pubkey):
print "Valid Ticket!"
d = method(request, pboxid, pubkey)
return NOT_DONE_YET
else:
print "Invalid Ticket!"
reply_dict = {
'status': {
'error': "Invalid Ticket",
'message': 'N/A'
}
}
request.write(
json.dumps(reply_dict, sort_keys=True, encoding="utf-8"))
request.finish()
d = self.storage.getClientData(request)
d.addCallback(handleValidation_cb)
return NOT_DONE_YET
# Handling PBoxes resource related operations:
#
#handle ListPBoxes
def handleListPBoxes(self, request):
return self.handleValidation(request, self.storage.listPBoxes)
def handleGetPBoxMData(self, request):
return self.handleValidation(request, self.storage.getPBoxMData)
# handleRegisterPBox: Checks if client exists, if so returns error, else registers the client.
def handleRegisterPBox(self, request):
# Checking if the client exists.
# pprint(request.__dict__)
def checkClientExists_cb(data, key_txt):
if data:
reply_dict = {
'status': {
'error': "Invalid Request",
'message': 'User already exists.'
}
}
request.write(
json.dumps(reply_dict, sort_keys=True, encoding="utf-8"))
request.finish()
else:
d = self.storage.registerPBox(request, key_txt)
return NOT_DONE_YET
# Validating key.
key_txt = request.content.read()
if not key_txt:
reply_dict = {
'status': {
'error': "Invalid Request",
'message': "No key on request body."
}
}
return json.dumps(reply_dict, encoding="utf-8")
cli_key = RSA.importKey(key_txt)
if not cli_key.can_encrypt():
reply_dict = {
'status': {
'error': "Invalid Request",
'message': "No key on request body."
}
}
return json.dumps(reply_dict, encoding="utf-8")
d = self.storage.getClientData(request)
d.addCallback(checkClientExists_cb, key_txt)
return NOT_DONE_YET
# Handling Files resource related operations:
#
def handleListFiles(self, request):
return self.handleValidation(request, self.storage.listFiles)
def handleGetFileMData(self, request):
return self.handleValidation(request, self.storage.getFileMData)
def handleGetFile(self, request):
return self.handleValidation(request, self.storage.getFile)
def handlePutFile(self, request): #
return self.handleValidation(request, self.storage.putFile)
def handleUpdateFile(self, request): #
return self.handleValidation(request, self.storage.updateFile)
def handleDeleteFile(self, request):
return self.handleValidation(request, self.storage.deleteFile)
# Handling Share resource related operations:
#
def handleShareFile(self, request):
return self.handleValidation(request, self.storage.shareFile)
def handleGetShareMData(self, request):
return self.handleValidation(request, self.storage.getShareMData)
def handleGetShared(self, request):
return self.handleValidation(request, self.storage.getShared)
def handleListShares(self, request):
return self.handleValidation(request, self.storage.listShares)
def handleUpdateShared(self, request):
return self.handleValidation(request, self.storage.updateShared)
def handleUpdateSharePerm(self, request):
return self.handleValidation(request, self.storage.updateSharePerm)
def handleDeleteShare(self, request):
return self.handleValidation(request, self.storage.deleteShare)
