-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathskyhookUpDown.py
84 lines (77 loc) · 2.37 KB
/
skyhookUpDown.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
try:
import ipfshttpclient
except:
print("[!] Module ipfshttpclient not installed")
exit()
try:
import skyhookfilecrypt
except:
print("[!] Module skyhookfilecrypt not installed")
exit()
from datetime import datetime
import os, skyhookDb, random, string, skyhookConfig
tmpDir = skyhookConfig.tmpDir.rstrip('/')
def getRandomString(length):
return("".join(random.choice(string.ascii_letters + string.digits) for i in range(length)))
try:
peer = ipfshttpclient.connect(skyhookConfig.host)
except:
print("[!] Cannot connect to {}".format(skyhookConfig.host))
exit()
def uploadFile(fileName):
currentDir = os.getcwd()
if not fileName in [f for f in os.listdir(".") if os.path.isfile(f)]:
return(1)
else:
password = getRandomString(32)
aesName = "{}.sky".format(fileName)
tmpPath = "{}/{}".format(tmpDir, aesName)
print("[+] Encrypting {}".format(fileName))
try:
skyhookfilecrypt.encryptFile(fileName, tmpPath, bytes(password, "ascii"))
except:
os.remove(tmpPath)
return(2)
os.chdir(tmpDir)
print("[+] Uploading {}".format(fileName))
try:
result = peer.add(aesName)
os.chdir(currentDir)
except:
os.chdir(currentDir)
os.remove(tmpPath)
return(3)
now = datetime.now()
currentDate = now.strftime("%d/%m/%Y-%H:%M:%S")
print("[+] Adding entry to history")
res = skyhookDb.addToHistory(fileName, result["Hash"], password, currentDate)
if res == 0:
pass
else:
os.remove(tmpPath)
return(4)
os.remove(tmpPath)
return(0)
def downloadFile(fileHash):
currentDir = os.getcwd()
fileName, password = skyhookDb.getEntry(fileHash)
if fileName == 1 and password == 1:
return(1)
saveFile = "{}/{}".format(currentDir, fileName)
os.chdir(tmpDir)
print("[+] Downloading {}".format(fileName))
try:
peer.get(fileHash)
except:
os.chdir(currentDir)
return(2)
print("[+] Decrypting {}".format(fileName))
try:
skyhookfilecrypt.decryptFile(fileHash, saveFile, bytes(password, "ascii"))
except:
os.remove(fileHash)
os.chdir(currentDir)
return(3)
os.remove(fileHash)
os.chdir(currentDir)
return(fileName)