# LiMEaide
# Copyright (c) 2011-2018 Daryl Bennett
# Author:
# Daryl Bennett - kd8bny@gmail.com
# This program 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 2 of the License, or (at
# your option) any later version.
# This program 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 this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import configparser
import logging
import os
import shutil
import sys
import urllib.request
import zipfile
from datetime import datetime
from termcolor import cprint, colored
class Config:
__version__ = "1"
def __init__(self):
super(Config, self).__init__()
# Internal Resources
self.lime_version = '1.8.1'
self.config_file = '.limeaide'
self.manifest = 'manifest.json'
self.lime_dir = './tools/LiME/src/'
self.tools_dir = './tools/'
self.output_dir = './output/'
self.profile_dir = './profiles/'
self.log_dir = './logs/'
# External Resources
self.lime_rdir = './.limeaide_working/'
self.date = None
self.volatility_dir = None
self.output = None
self.compress = None
self.format = None
self.digest = None
def set_date(self):
self.date = datetime.strftime(datetime.today(), "%Y_%m_%dT%H_%M_%S")
def setup_logging(self):
"""Setup logging to file and initial logger"""
logging.basicConfig(
level=logging.INFO, filename='{0}{1}.log'.format(
self.log_dir, self.date))
self.logger = logging.getLogger(__name__)
def __download_lime__(self):
"""Download LiME from GitHub"""
cprint("Downloading LiME", 'green')
try:
urllib.request.urlretrieve(
"https://github.com/504ensicsLabs/LiME/archive/" +
"v{}.zip".format(
self.lime_version), filename="./tools/lime_master.zip")
zip_lime = zipfile.ZipFile("./tools/lime_master.zip", 'r')
zip_lime.extractall(self.tools_dir)
zip_lime.close()
shutil.move(
'./tools/LiME-{}'.format(self.lime_version), './tools/LiME/')
except urllib.error.URLError:
sys.exit(colored(
"LiME failed to download. Check your internet connection" +
" or place manually", 'red'))
def __write_new_config__(self):
"""Write a new configuration file on first run."""
default_config = configparser.ConfigParser()
default_config['MANIFEST'] = {}
default_config['MANIFEST']['version'] = self.__version__
default_config.set('DEFAULT', 'volatility', '')
default_config.set('DEFAULT', 'output', 'dump.lime')
default_config.set('DEFAULT', 'compress', 'False')
default_config.set('DEFAULT', 'format', 'lime')
default_config.set('DEFAULT', 'digest', 'sha1')
with open(self.config_file, 'w') as configfile:
default_config.write(configfile)
def __update_vol_dir__(self):
"""Check if Volatility exists. If not prompt the user."""
cprint("Volatility directory missing.", 'red')
cprint("Please provide a path to your Volatility directory." +
"\ne.g. '~/volatility/'" +
"\n[q] to never ask again: ", 'green')
path_ext = '/volatility/plugins/overlays/linux/'
while True:
path = input(":")
if path == 'q':
path = 'None'
path_ext = ''
break
elif not os.path.isdir(path):
cprint(
"Entered directory does not exist. Please enter" +
" again", 'red')
elif not os.path.isdir(path + path_ext):
cprint(
"This is a directory but it".format(path) +
"doesn't include the required " +
"subdirectories: \n{0}".format(path_ext), 'red')
else:
break
self.volatility_dir = path + path_ext
default_config = configparser.ConfigParser()
default_config.read(self.config_file)
default_config.set(
'DEFAULT', 'volatility', self.volatility_dir)
with open(self.config_file, 'w+') as configfile:
default_config.write(configfile)
def check_directories(self):
"""Create necessary directories."""
if not os.path.isdir(self.output_dir):
os.mkdir(self.output_dir)
if not os.path.isdir(self.profile_dir):
os.mkdir(self.profile_dir)
if not os.path.isdir(self.tools_dir):
os.mkdir(self.tools_dir)
if not os.path.isdir(self.log_dir):
os.mkdir(self.log_dir)
if not os.path.isdir(self.lime_dir):
self.__download_lime__()
def read_config(self):
"""Read default configuration."""
if not os.path.isfile(self.config_file):
self.__write_new_config__()
default_config = configparser.ConfigParser()
default_config.read(self.config_file)
self.volatility_dir = default_config['DEFAULT']['volatility']
self.output = default_config['DEFAULT']['output']
self.compress = default_config['DEFAULT']['compress']
self.format = default_config['DEFAULT']['format']
self.digest = default_config['DEFAULT']['digest']
if self.volatility_dir != 'None':
if not self.volatility_dir or not os.path.isdir(
self.volatility_dir):
self.__update_vol_dir__()
def configure(self):
"""Configure LiMEaide."""
self.set_date()
self.check_directories()
self.setup_logging()
self.read_config()
