323 lines
9.3 KiB
Python
323 lines
9.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
|
||
|
|
# #
|
||
|
|
# software : ImpraStorage <http://imprastorage.sourceforge.net/> #
|
||
|
|
# version : 0.4 #
|
||
|
|
# date : 2012 #
|
||
|
|
# licence : GPLv3.0 <http://www.gnu.org/licenses/> #
|
||
|
|
# author : a-Sansara <http://www.a-sansara.net/> #
|
||
|
|
# copyright : pluie.org <http://www.pluie.org/> #
|
||
|
|
# #
|
||
|
|
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
|
||
|
|
#
|
||
|
|
# This file is part of ImpraStorage.
|
||
|
|
#
|
||
|
|
# ImpraStorage is free software (free as in speech) : 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 (at your option) any later version.
|
||
|
|
#
|
||
|
|
# ImpraStorage 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 ImpraStorage. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ package util ~~
|
||
|
|
|
||
|
|
from hashlib import sha256
|
||
|
|
from math import log, floor, ceil
|
||
|
|
from random import choice
|
||
|
|
from os import urandom, popen
|
||
|
|
from os.path import dirname, realpath
|
||
|
|
from time import time
|
||
|
|
from re import split as regsplit
|
||
|
|
from base64 import urlsafe_b64encode
|
||
|
|
from inspect import stack
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ methods ~~
|
||
|
|
|
||
|
|
def quote_escape(data):
|
||
|
|
"""Escape simple quote
|
||
|
|
:Returns: `str`
|
||
|
|
"""
|
||
|
|
return data.replace('\'', r'\'')
|
||
|
|
|
||
|
|
def get_file_content(fileName):
|
||
|
|
"""Get file content of `fileName`
|
||
|
|
:Returns: `str`
|
||
|
|
"""
|
||
|
|
r = open(fileName, "rt")
|
||
|
|
data = r.read()
|
||
|
|
r.close()
|
||
|
|
return data
|
||
|
|
|
||
|
|
def hash_sha256(data):
|
||
|
|
"""Get a sha256 hash of str `data`
|
||
|
|
:Returns: `str`
|
||
|
|
"""
|
||
|
|
return str(sha256(bytes(data,'utf-8')).hexdigest())
|
||
|
|
|
||
|
|
def randomFrom(val, sval=0):
|
||
|
|
"""Get a random number from range `sval=0` to `val`
|
||
|
|
:Returns: `int`
|
||
|
|
"""
|
||
|
|
lst = list(range(sval,val))
|
||
|
|
return choice(lst)
|
||
|
|
|
||
|
|
def formatBytes(b, p=2):
|
||
|
|
"""Give a human representation of bytes size `b`
|
||
|
|
:Returns: `str`
|
||
|
|
"""
|
||
|
|
units = ['B', 'KB', 'MB', 'GB', 'TB'];
|
||
|
|
b = max(b,0);
|
||
|
|
if b == 0 : lb= 0
|
||
|
|
else : lb = log(b)
|
||
|
|
p = floor(lb/log(1024))
|
||
|
|
p = min(p, len(units)- 1)
|
||
|
|
#Uncomment one of the following alternatives
|
||
|
|
b /= pow(1024,p)
|
||
|
|
#b /= (1 << (10 * p))
|
||
|
|
return str(round(b, p))+' '+units[p]
|
||
|
|
|
||
|
|
def bstr(b,enc='utf-8'):
|
||
|
|
""""""
|
||
|
|
return str(b, encoding=enc)
|
||
|
|
|
||
|
|
def __CALLER__(args=''):
|
||
|
|
"""Give basic information of caller method
|
||
|
|
usage ::
|
||
|
|
|
||
|
|
eval(__CALLER())
|
||
|
|
eval(__CALLER('"%s","%s"' % (arg1,arg2)))
|
||
|
|
|
||
|
|
:Returns: `str`
|
||
|
|
"""
|
||
|
|
#~ print(inspect.stack()[1][3])
|
||
|
|
#~ print(print(args))
|
||
|
|
#~ print('-----')
|
||
|
|
#~ print(inspect.stack())
|
||
|
|
#~ print('---------------')
|
||
|
|
val = "self.__class__.__name__+'.%s' % stack()[1][3]+'("+quote_escape(args)+") l:'+str(stack()[1][2])"
|
||
|
|
return val
|
||
|
|
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class Noiser ~~
|
||
|
|
|
||
|
|
class Noiser:
|
||
|
|
""""""
|
||
|
|
|
||
|
|
KEY_LEN = 64
|
||
|
|
""""""
|
||
|
|
|
||
|
|
def __init__(self, key, part=0):
|
||
|
|
""""""
|
||
|
|
if len(key)!=self.KEY_LEN :
|
||
|
|
raise Exception('Invalid Pass length')
|
||
|
|
else :
|
||
|
|
self.key = key
|
||
|
|
self.build(part)
|
||
|
|
|
||
|
|
def build(self, part):
|
||
|
|
""""""
|
||
|
|
if not part < self.KEY_LEN-1 : raise Exception('part exceed limit')
|
||
|
|
else :
|
||
|
|
self.part, v = part, 0
|
||
|
|
for i in self.key[::-2] : v += i
|
||
|
|
v = int(ceil(v/4.22))
|
||
|
|
self.lns = int(ceil(v/2))-self.key[self.part]
|
||
|
|
self.lne = int(v-self.lns-self.key[self.part+2])
|
||
|
|
|
||
|
|
def getNoise(self, l):
|
||
|
|
""""""
|
||
|
|
return urandom(l)
|
||
|
|
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class Randomiz ~~
|
||
|
|
|
||
|
|
class Randomiz:
|
||
|
|
""""""
|
||
|
|
|
||
|
|
def __init__(self,count):
|
||
|
|
""""""
|
||
|
|
self.lst = list(range(0,count))
|
||
|
|
self.count = len(self.lst)
|
||
|
|
|
||
|
|
def new(self,count=None):
|
||
|
|
""""""
|
||
|
|
if count : self.count = count
|
||
|
|
self.__init__(self.count)
|
||
|
|
|
||
|
|
def get(self):
|
||
|
|
""""""
|
||
|
|
pos = choice(self.lst)
|
||
|
|
del self.lst[self.lst.index(pos)]
|
||
|
|
return pos
|
||
|
|
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class RuTime ~~
|
||
|
|
|
||
|
|
class RuTime:
|
||
|
|
"""Give basics time stats"""
|
||
|
|
|
||
|
|
def __init__(self,label):
|
||
|
|
"""Initialize duration with appropriate label"""
|
||
|
|
self.label = label
|
||
|
|
self._start()
|
||
|
|
|
||
|
|
def _start(self):
|
||
|
|
print(' ==> '+self.label)
|
||
|
|
self.sc = time()
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
"""Stop duration and print basics stats duration on console"""
|
||
|
|
self.ec = time()
|
||
|
|
self._stats()
|
||
|
|
|
||
|
|
def _stats(self):
|
||
|
|
print(' <== '+self.label+(' [%.9f s]' % (self.ec - self.sc))+' <¤¤ ')
|
||
|
|
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class IniFile ~~
|
||
|
|
|
||
|
|
class IniFile:
|
||
|
|
"""Read a write inifile"""
|
||
|
|
|
||
|
|
def __init__(self,path):
|
||
|
|
""""""
|
||
|
|
self.path = path
|
||
|
|
self.dic = {}
|
||
|
|
self.read()
|
||
|
|
|
||
|
|
def isEmpty(self):
|
||
|
|
""""""
|
||
|
|
return len(self.dic)==0
|
||
|
|
|
||
|
|
def has(self, key, section='main'):
|
||
|
|
""""""
|
||
|
|
d = (key in self.dic[section])
|
||
|
|
return d
|
||
|
|
|
||
|
|
def hasSection(self, section):
|
||
|
|
""""""
|
||
|
|
d = (section in self.dic)
|
||
|
|
return d
|
||
|
|
|
||
|
|
def get(self, key, section='main'):
|
||
|
|
""""""
|
||
|
|
return self.dic[section][key]
|
||
|
|
|
||
|
|
def set(self, key, val, section='main'):
|
||
|
|
""""""
|
||
|
|
v = None
|
||
|
|
if not section in self.dic:
|
||
|
|
self.dic[section] = {}
|
||
|
|
if key in self.dic[section]:
|
||
|
|
v = self.dic[section].pop(key)
|
||
|
|
self.dic[section][key] = val
|
||
|
|
return v
|
||
|
|
|
||
|
|
def rem(self, key, section):
|
||
|
|
""""""
|
||
|
|
v = None
|
||
|
|
if section in self.dic :
|
||
|
|
if key == '*' :
|
||
|
|
v = self.dic.pop(section)
|
||
|
|
elif key in self.dic[section]:
|
||
|
|
v = self.dic[section].pop(key)
|
||
|
|
return v
|
||
|
|
|
||
|
|
def write(self,path=None):
|
||
|
|
""""""
|
||
|
|
if path == None : path = self.path
|
||
|
|
content = self.toString()
|
||
|
|
with open(path, mode='w', encoding='utf-8') as o:
|
||
|
|
o.write(content)
|
||
|
|
|
||
|
|
def toString(self,path=None):
|
||
|
|
""""""
|
||
|
|
if path == None : path = self.path
|
||
|
|
content = ''
|
||
|
|
main = ''
|
||
|
|
for s in self.dic:
|
||
|
|
if s!='main':
|
||
|
|
content += '\n['+s+']\n'
|
||
|
|
for k in sorted(self.dic[s]):
|
||
|
|
k = k.rstrip(' ')
|
||
|
|
if s!='main':
|
||
|
|
content += k+' = '+self.dic[s][k]+'\n'
|
||
|
|
else : main += k+' = '+self.dic[s][k]+'\n'
|
||
|
|
return main + content
|
||
|
|
|
||
|
|
def read(self):
|
||
|
|
""""""
|
||
|
|
try:
|
||
|
|
with open(self.path, encoding='utf-8') as o:
|
||
|
|
csection = 'main'
|
||
|
|
self.dic[csection] = {}
|
||
|
|
for l in o:
|
||
|
|
l = l.rstrip()
|
||
|
|
d = regsplit(' *= *',l,1)
|
||
|
|
if len(d)> 1:
|
||
|
|
self.dic[csection][d[0]] = d[1]
|
||
|
|
elif len(l)>0 and l[0]=='[':
|
||
|
|
csection = l.strip('[]')
|
||
|
|
self.dic[csection] = {}
|
||
|
|
except IOError : pass
|
||
|
|
|
||
|
|
|
||
|
|
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
# ~~ class Rsa ~~
|
||
|
|
|
||
|
|
class Rsa:
|
||
|
|
""""""
|
||
|
|
|
||
|
|
def __init__(self, prvKey=None, pubKey=None, dpath='./'):
|
||
|
|
""""""
|
||
|
|
self.cpath = dirname(realpath(__file__))+'/../desurveil/scripts/'
|
||
|
|
self.prvKey = prvKey
|
||
|
|
self.pubKey = pubKey
|
||
|
|
self.dpath = dpath
|
||
|
|
if prvKey == None or pubKey==None : self.key()
|
||
|
|
|
||
|
|
def key(self):
|
||
|
|
""""""
|
||
|
|
cmd = self.cpath+"desurveil key -a "+self.dpath+".impra_id_rsa -l "+self.dpath+".impra_id_rsa.pub"
|
||
|
|
#print(cmd)
|
||
|
|
try :
|
||
|
|
with open(self.dpath+'.impra_id_rsa','rt') as f: pass
|
||
|
|
except IOError as e:
|
||
|
|
d = popen(cmd).read()
|
||
|
|
#print(d)
|
||
|
|
self.prvKey = get_file_content(self.dpath+'.impra_id_rsa')
|
||
|
|
self.pubKey = get_file_content(self.dpath+'.impra_id_rsa.pub')
|
||
|
|
#print('pubKey : \n'+self.pubKey)
|
||
|
|
#print('prvKey : \n'+self.prvKey)
|
||
|
|
|
||
|
|
def encrypt(self,data):
|
||
|
|
""""""
|
||
|
|
key = ''
|
||
|
|
if self.pubKey != None : key = " -CI '"+self.pubKey+"'"
|
||
|
|
#if self.pubKey != None : key = " -C '"+self.dpath+".impra_id_rsa.pub'"
|
||
|
|
cmd = self.cpath+"desurveil encrypt -i '"+data+"'"+key
|
||
|
|
#print(cmd)
|
||
|
|
return popen(cmd).read()
|
||
|
|
|
||
|
|
def decrypt(self,data):
|
||
|
|
""""""
|
||
|
|
key = ''
|
||
|
|
if self.prvKey != None : key = " -CI '"+self.prvKey+"'"
|
||
|
|
#if self.prvKey != None : key = " -C '"+self.dpath+".impra_id_rsa'"
|
||
|
|
cmd = self.cpath+"desurveil decrypt -i '"+data+"'"+key
|
||
|
|
#print(cmd)
|
||
|
|
return popen(cmd).read()
|