def display(self):
db = Database()
if 'file_id' in self.request.POST:
file_id = self.request.POST['file_id']
# Check to see if we already have strings stored.
new_strings = db.get_strings(file_id)
if new_strings:
string_id = new_strings._id
else:
string_id = False
print string_id
self.render_data = {'ExtractStrings': {'string_id': string_id}}
def display(self):
db = Database()
if 'file_id' in self.request.POST:
file_id = self.request.POST['file_id']
# Check to see if we already have strings stored.
new_strings = db.get_strings(file_id)
if new_strings:
string_id = new_strings._id
else:
string_id = False
print string_id
self.render_data = {'ExtractStrings': {'string_id': string_id}}
def run(self):
db = Database()
# Get Options
if "min_length" in self.request.POST:
min_len = self.request.POST['min_length']
else:
min_len = 4
if 'file_id' in self.request.POST:
file_id = self.request.POST['file_id']
# Check to see if we already have strings stored.
new_strings = db.get_strings(file_id)
if new_strings:
string_id = new_strings._id
else:
file_object = db.get_filebyid(file_id)
# Always get ASCII and Unicode
file_data = file_object.read()
ascii_strings = self.ascii_strings(file_data, 4)
unicode_strings = self.unicode_strings(file_data, 4)
if HAVE_FLOSS:
# Advacned Floss needs a file on disk
with tempfile.NamedTemporaryFile() as tmp:
tmp.write(file_data)
file_path = tmp.name
if self.is_supported_file_type(file_path):
try:
vw = viv_utils.getWorkspace(file_path, should_save=False)
except Exception:
print "ahhhhhhhhhhhhhh"
raise
# Decode Strings
#decoding_functions_candidates = im.identify_decoding_functions(vw, selected_plugins, selected_functions)
#function_index = viv_utils.InstructionFunctionIndex(vw)
#decoded_strings = decode_strings(vw, function_index, decoding_functions_candidates)
# Stack Strings
# Generate the final output file
string_list = '##### ASCII Strings #####\n {0} \n ##### Unicode Strings #####\n {1}'.format(ascii_strings, unicode_strings)
'''
String lists can get larger than the 16Mb bson limit
Need to store in GridFS
'''
store_data = {'file_id': file_id, 'string_list': string_list}
string_id = db.create_file(string_list, 'session_id', 'sha256', '{0}_strings.txt'.format(file_id))
print string_id
self.render_type = 'html'
self.render_data = '<td><a class="btn btn-success" role="button" href="/download/file/{0}">Download</a></td>'.format(
string_id)
def run(self):
db = Database()
# Get Options
if "min_length" in self.request.POST:
min_len = self.request.POST['min_length']
else:
min_len = 4
if 'file_id' in self.request.POST:
file_id = self.request.POST['file_id']
# Check to see if we already have strings stored.
new_strings = db.get_strings(file_id)
if new_strings:
string_id = new_strings._id
else:
file_object = db.get_filebyid(file_id)
# Always get ASCII and Unicode
file_data = file_object.read()
ascii_strings = self.ascii_strings(file_data, 4)
unicode_strings = self.unicode_strings(file_data, 4)
if HAVE_FLOSS:
# Advacned Floss needs a file on disk
with tempfile.NamedTemporaryFile() as tmp:
tmp.write(file_data)
file_path = tmp.name
if self.is_supported_file_type(file_path):
try:
vw = viv_utils.getWorkspace(file_path,
should_save=False)
except Exception:
print "ahhhhhhhhhhhhhh"
raise
# Decode Strings
#decoding_functions_candidates = im.identify_decoding_functions(vw, selected_plugins, selected_functions)
#function_index = viv_utils.InstructionFunctionIndex(vw)
#decoded_strings = decode_strings(vw, function_index, decoding_functions_candidates)
# Stack Strings
# Generate the final output file
string_list = '##### ASCII Strings #####\n {0} \n ##### Unicode Strings #####\n {1}'.format(
ascii_strings, unicode_strings)
'''
String lists can get larger than the 16Mb bson limit
Need to store in GridFS
'''
store_data = {'file_id': file_id, 'string_list': string_list}
string_id = db.create_file(string_list, 'session_id', 'sha256',
'{0}_strings.txt'.format(file_id))
print string_id
self.render_type = 'html'
self.render_data = '<td><a class="btn btn-success" role="button" href="/download/file/{0}">Download</a></td>'.format(
string_id)