copyleftie/gajim-plural
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

better way to handle non supported hash algorithms

Browse source
This commit is contained in:
Jefry Lagrange 2012-01-24 16:51:26 -05:00
parent c02b43f88f
commit b2897e36bb
2 changed files with 8 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/common/jingle_ft.py
View file

@ -118,11 +118,15 @@ class JingleFileTransfer(JingleContent):
if self.session.hash_algo == None: if self.session.hash_algo == None:
return return
try: try:
file = open(self.file_props['file-name'], 'r') file_ = open(self.file_props['file-name'], 'r')
except: except:
# can't open file
return return
h = xmpp.Hashes() h = xmpp.Hashes()
hash_ = h.calculateHash(self.session.hash_algo, file) hash_ = h.calculateHash(self.session.hash_algo, file_)
if not hash_:
# Hash alogrithm not supported
return
self.file_props['hash'] = hash_ self.file_props['hash'] = hash_
h.addHash(hash_, self.session.hash_algo) h.addHash(hash_, self.session.hash_algo)
checksum = xmpp.Node(tag='checksum', checksum = xmpp.Node(tag='checksum',

10
src/common/xmpp/protocol.py
View file

@ -1078,10 +1078,7 @@ class Hashes(Node):
elif algo == 'sha-512': elif algo == 'sha-512':
hl = hashlib.sha512() hl = hashlib.sha512()
if hl == None: if hl:
# Raise exception
raise Exception('Hash algorithm not supported')
else:
hl.update(file_string) hl.update(file_string)
hash_ = hl.hexdigest() hash_ = hl.hexdigest()
else: # if it is a file else: # if it is a file
@ -1095,10 +1092,7 @@ class Hashes(Node):
elif algo == 'sha-512': elif algo == 'sha-512':
hl = hashlib.sha512() hl = hashlib.sha512()
if hl == None: if hl:
# Raise exception
raise Exception('Hash algorithm not supported')
else:
for line in file_string: for line in file_string:
hl.update(line) hl.update(line)
hash_ = hl.hexdigest() hash_ = hl.hexdigest()