def postMethod(url, clientId, clientInfo, content, key):
'''
Função que cria arquivos no servidor
Cria um arquivo com o conteudo recebido do cliente
Para manter a autoridade dos arquivos, também é criado um arquivo, com mesmo nome,
porém, concatenado com a identificação e informação do cliente.
:param url: Url onde o arquivo sera salvo
:param clientId: Identificação do cliente
:param clientInfo: Informação sobre o cliente, neste caso o hostname
:return: Estrutura Protobuf já pronta para o envio
'''
message = response.Response()
archivePath = str(Path().absolute())
archivePath += '/contents/'
if not os.path.exists(archivePath):
os.makedirs("contents")
message = setDefaultServer(message)
archivePath += clientInfo + "/"
if not os.path.exists(archivePath):
os.makedirs("contents/" + clientInfo)
if url == '/':
archivePath += 'index.html'
else:
# if url[0] == '/':
# archivePath += str(''.join(url.split('/')))
# else:
# archivePath += url
archivePath += str(''.join(url.split('/')[-1]))
logging.info(" POST {0}".format(url))
if os.path.exists(archivePath):
logging.info(" POST in a existent file")
message.status = "FAIL - 403"
else:
archive = open(archivePath, 'w+')
archive.write(content)
logging.info(" POST Sucessful")
message.status = "OK - 200"
archive.close()
message.url = url
message.content = content
message.signature = communication.hmacFromResponse(message, key)
return message
def getResponse(communication, message, sock, key, flag):
responseFromServer = communication.recvMessage(sock)
if responseFromServer:
signature = communication.hmacFromResponse(responseFromServer, key)
if signature == responseFromServer['signature']:
print("\n######## RESPOSTA DO SERVIDOR ########")
if message['command'] == "GET":
print("STATUS:", responseFromServer['status'])
if ("OK" in responseFromServer['status']):
print("CONTEUDO:")
print(responseFromServer['content'])
if flag:
if not os.path.exists("contents"):
os.makedirs("contents")
if (responseFromServer['url'] in ["/", ""]):
archive = open("./contents/index.html", 'w')
archive.write(responseFromServer['content'])
archive.close()
else:
archive = open(
"./contents/{}".format(
responseFromServer['url'].split('/')[-1]),
'w')
archive.write(responseFromServer['content'])
archive.close()
else:
print("Arquivo não encontrado")
elif message['command'] == "POST":
print("STATUS:", responseFromServer['status'])
if ("OK" in responseFromServer['status']):
print("Arquivo {0} criado com sucesso!".format(
responseFromServer['url']))
else:
print("Falha ao criar o arquivo!")
elif message['command'] == "DELETE":
print("STATUS:", responseFromServer['status'])
if ("OK" in responseFromServer['status']):
print("Arquivo {0} deletado com sucesso!".format(
responseFromServer['url']))
else:
print("Falha ao deletar o arquivo!")
else:
print("STATUS:", responseFromServer['status'])
print("Comando Desconhecido")
return responseFromServer
def unknownMethod(key):
'''
Função que é chamada quando o comando especificado pelo cliente não é conhecido
Ex: "gets", "trace"
'''
logging.info(" Unknown command")
message = response.Response()
message = setDefaultServer(message)
message.status = "FAIL - 401"
message.signature = communication.hmacFromResponse(message, key)
return message
def getMethod(url, clientId, clientInfo, key):
'''
Função que popula o protobuf enviando como conteudo, o arquivo que o cliente quer
:param url: Url do arquivo desejado (ex: "index.html", "teste.txt")
:return: Protobuf já pronto para o envio
'''
message = response.Response()
archivePath = str(Path().absolute())
archivePath += '/contents/'
if not os.path.exists(archivePath):
os.makedirs("contents")
message = setDefaultServer(message)
if (url in ['/', '']):
archivePath += 'index.html'
else:
archivePath += clientInfo + "/"
url = url.split('/')[-1]
if url[0] != '/':
archivePath += "/" + url
else:
archivePath += url
message.url = url
logging.info(" GET {0}".format(url))
try:
archive = open(archivePath, 'r')
nameFile = archive.name.split('/')[-1]
message.content += archive.read()
logging.info(" GET Sucessful")
message.status = "OK - 200"
message.url = nameFile
archive.close()
except FileNotFoundError:
message.status = "FAIL - 404"
logging.info(" Archive not found")
except OSError:
message.status = "FAIL - 403"
logging.info(" Error on archive")
message.signature = communication.hmacFromResponse(message, key)
return message
def deleteMethod(url, clientId, clientInfo, key):
'''
Função de remoção de arquivos
Remove arquivos localizados no servidor
Para segurança dos arquivos, faz o teste antes de remover
O teste basicamente é uma busca pelo arquivo de lock (url + . + clientId + clientInfo)
:param url: Url do arquivo que será removido
:param clientId: Identificação do cliente
:param clientInfo: Informação do cliente, neste caso é o hostname
:return: Estrutura protobuf de resposta já pronta para envio
'''
message = response.Response()
archivePath = str(Path().absolute())
archivePath += '/contents/'
if not os.path.exists(archivePath):
os.makedirs("contents")
archivePath += clientInfo + "/"
message = setDefaultServer(message)
if url == '/':
archivePath += 'index.html'
else:
# if url[0] == '/':
# archivePath += str(''.join(url.split('/')))
# else:
# archivePath += url
archivePath += str(''.join(url.split('/')[-1]))
logging.info(" DELETE {0}".format(url))
if os.path.exists(archivePath):
os.remove(archivePath)
logging.info(" DELETE Sucessful")
message.status = "OK - 200"
else:
logging.info(" DELETE Unsucessful")
message.status = "FAIL - 403"
message.url = url
message.content = ""
message.signature = communication.hmacFromResponse(message, key)
return message
def getResponse(communication, message, sock,key):
responseFromServer = communication.recvMessage(sock, response.Response)
if responseFromServer:
signature = communication.hmacFromResponse(responseFromServer, key)
if signature == responseFromServer.signature:
print("\n######## RESPOSTA DO SERVIDOR ########")
if message.command == "GET":
print("STATUS:", responseFromServer.status)
if("OK" in responseFromServer.status):
print("CONTEUDO:")
print(responseFromServer.content)
if(responseFromServer.url in ["/", ""]):
archive = open("index.html", 'w')
archive.write(responseFromServer.content)
archive.close()
else:
archive = open(responseFromServer.url, 'w')
archive.write(responseFromServer.content)
archive.close()
else:
print("Arquivo não encontrado")
elif message.command == "POST":
print("STATUS:", responseFromServer.status)
if("OK" in responseFromServer.status):
print("Arquivo {0} criado com sucesso!".format(responseFromServer.url))
else:
print("Falha ao criar o arquivo!")
elif message.command == "DELETE":
print("STATUS:", responseFromServer.status)
if("OK" in responseFromServer.status):
print("Arquivo {0} deletado com sucesso!".format(responseFromServer.url))
else:
print("Falha ao deletar o arquivo!")
else:
print("STATUS:", responseFromServer.status)
print("Comando Desconhecido")