212 lines
8.4 KiB
Python
212 lines
8.4 KiB
Python
# -*- coding: utf8 -*-
|
|
"""
|
|
********************************************************************************
|
|
* CNIRevelator *
|
|
* *
|
|
* Desc: Application launcher download stuff *
|
|
* *
|
|
* Copyright © 2018-2019 Adrien Bourmault (neox95) *
|
|
* *
|
|
* This file is part of CNIRevelator. *
|
|
* *
|
|
* CNIRevelator is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation, either version 3 of the License, or *
|
|
* any later version. *
|
|
* *
|
|
* CNIRevelator is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY*without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with CNIRevelator. If not, see <https:*www.gnu.org/licenses/>. *
|
|
********************************************************************************
|
|
"""
|
|
|
|
import base64, hashlib
|
|
import os
|
|
from pypac import PACSession
|
|
from requests.auth import HTTPProxyAuth
|
|
from Crypto import Random
|
|
from Crypto.Cipher import AES
|
|
from requests import Session
|
|
from time import time
|
|
|
|
import critical # critical.py
|
|
import logger # logger.py
|
|
import globs # globs.py
|
|
import ihm # ihm.py
|
|
import lang # lang.py
|
|
|
|
class AESCipher(object):
|
|
|
|
def __init__(self, key):
|
|
self.bs = 32
|
|
self.key = hashlib.sha256(key.encode()).digest()
|
|
|
|
def encrypt(self, raw):
|
|
raw = self._pad(raw)
|
|
iv = Random.new().read(AES.block_size)
|
|
cipher = AES.new(self.key, AES.MODE_CBC, iv)
|
|
return base64.b64encode(iv + cipher.encrypt(raw))
|
|
|
|
def decrypt(self, enc):
|
|
enc = base64.b64decode(enc)
|
|
iv = enc[:AES.block_size]
|
|
cipher = AES.new(self.key, AES.MODE_CBC, iv)
|
|
return self._unpad(cipher.decrypt(enc[AES.block_size:])).decode('utf-8')
|
|
|
|
def _pad(self, s):
|
|
return s + (self.bs - len(s) % self.bs) * chr(self.bs - len(s) % self.bs)
|
|
|
|
@staticmethod
|
|
def _unpad(s):
|
|
return s[:-ord(s[len(s) - 1:])]
|
|
|
|
|
|
class newcredentials:
|
|
def __init__(self):
|
|
|
|
logfile = logger.logCur
|
|
|
|
self.login = ''
|
|
self.password = ''
|
|
self.valid = False
|
|
self.readInTheBooks = False
|
|
self.trying = 0
|
|
|
|
while True:
|
|
session = PACSession(proxy_auth=(HTTPProxyAuth(self.login, self.password)))
|
|
self.trying += 1
|
|
|
|
try:
|
|
sessionAnswer = session.get('http://www.google.com')
|
|
except Exception as e:
|
|
logfile.printdbg('Network Error : ' + str(e))
|
|
self.login = "nointernet"
|
|
return
|
|
|
|
logfile.printdbg("Session Answer : " + str(sessionAnswer))
|
|
|
|
if str(sessionAnswer) == '<Response [200]>':
|
|
logfile.printdbg('Successfully connected to the Internet !')
|
|
self.sessionHandler = session
|
|
self.valid = True
|
|
return
|
|
|
|
if self.trying > 4:
|
|
logfile.printerr('Invalid credentials : access denied, a maximum of 3 trials have been raised !')
|
|
return
|
|
|
|
logfile.printdbg('Invalid credentials : access denied')
|
|
|
|
# Deleting the root of Evil if needed
|
|
if self.readInTheBooks:
|
|
os.remove(globs.CNIRConfig)
|
|
logfile.printdbg("Deleting the root of Evil")
|
|
|
|
try:
|
|
with open(globs.CNIRConfig, 'rb') as (configFile):
|
|
self.readInTheBooks = True
|
|
# Decrypt the config file
|
|
AESObj = AESCipher(globs.CNIRCryptoKey)
|
|
try:
|
|
# Reading it
|
|
reading = AESObj.decrypt(configFile.read())
|
|
# Parsing it
|
|
if reading != '||':
|
|
if reading.find('||') != -1:
|
|
# TADAAA
|
|
self.login, self.password = reading.split('||')[0:2]
|
|
else:
|
|
# UPS
|
|
logfile.printerr('Cryptokey is bad !')
|
|
return
|
|
|
|
except Exception as e:
|
|
raise IOError(str(e))
|
|
|
|
except FileNotFoundError:
|
|
logfile.printdbg('We will ask for credentials then')
|
|
|
|
launcherWindow = ihm.launcherWindowCur
|
|
|
|
# Parameters for the password invite
|
|
invite = ihm.LoginDialog(launcherWindow)
|
|
invite.transient(launcherWindow)
|
|
invite.grab_set()
|
|
|
|
launcherWindow.wait_window(invite)
|
|
|
|
# Getting the credentials
|
|
self.login = invite.login
|
|
self.password = invite.key
|
|
|
|
AESObj = AESCipher(globs.CNIRCryptoKey)
|
|
try:
|
|
os.mkdir(globs.CNIRFolder + '\\config')
|
|
except:
|
|
pass
|
|
with open(globs.CNIRConfig, 'wb+') as (configFile):
|
|
logfile.printdbg('Saving credentials in encrypted config file')
|
|
configFile.write(AESObj.encrypt(self.login + '||' + self.password))
|
|
|
|
|
|
return
|
|
|
|
class newdownload:
|
|
def __init__(self, credentials, urlFile, destinationFile, title):
|
|
self.urlFile = urlFile
|
|
self.destinationFile = destinationFile
|
|
self.session = credentials.sessionHandler
|
|
self.title = title
|
|
|
|
logfile = logger.logCur
|
|
launcherWindow = ihm.launcherWindowCur
|
|
|
|
logfile.printdbg('Requesting download of {}'.format(urlFile))
|
|
|
|
self.handler = self.session.get(self.urlFile, stream=True, headers={'Connection' : 'close', "Cache-Control": "no-cache", "Pragma": "no-cache"})
|
|
self.handler.raise_for_status()
|
|
|
|
self.filesize = int(self.handler.headers['Content-length'])
|
|
self.chunksize = int(self.filesize / 7)
|
|
self.count = 0
|
|
|
|
def download(self):
|
|
logfile = logger.logCur
|
|
launcherWindow = ihm.launcherWindowCur
|
|
url = self.urlFile
|
|
filename = self.destinationFile
|
|
title = self.title
|
|
|
|
reducedFilename = filename.split("\\")[-1]
|
|
|
|
launcherWindow.printmsg('{} {}'.format(lang.all[globs.CNIRlang]["Downloading"], title))
|
|
logfile.printdbg('{} {}'.format("Downloading", reducedFilename))
|
|
|
|
try:
|
|
os.remove(filename)
|
|
except:
|
|
pass
|
|
|
|
with open(filename, 'wb') as fh:
|
|
for chunk in self.handler.iter_content(chunk_size=self.chunksize):
|
|
fh.write(chunk)
|
|
|
|
self.count = os.path.getsize(self.destinationFile)
|
|
Percent = self.count / self.filesize * 100
|
|
|
|
launcherWindow.progressBar.stop()
|
|
launcherWindow.progressBar.configure(mode='determinate', value=(int(Percent)), maximum=100)
|
|
launcherWindow.printmsg('{} {}'.format(lang.all[globs.CNIRlang]["Downloading"], title) + ' : {:4.2f} %'.format(Percent))
|
|
|
|
launcherWindow.progressBar.configure(mode='indeterminate', value=0, maximum=20)
|
|
launcherWindow.progressBar.start()
|
|
|
|
logfile.printdbg('{} {}'.format(lang.all[globs.CNIRlang]["Successful retrieved"], filename))
|
|
|
|
return filename
|
|
|