Renaming lib/core/purge to lib/utils/purge

lib/core/option.py

@@ -92,7 +92,6 @@ from lib.core.exception import SqlmapUnsupportedDBMSException
 from lib.core.exception import SqlmapUserQuitException
 from lib.core.log import FORMATTER
 from lib.core.optiondict import optDict
-from lib.core.purge import purge
 from lib.core.settings import ACCESS_ALIASES
 from lib.core.settings import BURP_REQUEST_REGEX
 from lib.core.settings import BURP_XML_HISTORY_REGEX
@@ -146,6 +145,7 @@ from lib.request.templates import getPageTemplate
 from lib.utils.crawler import crawl
 from lib.utils.deps import checkDependencies
 from lib.utils.google import Google
+from lib.utils.purge import purge
 from thirdparty.colorama.initialise import init as coloramainit
 from thirdparty.keepalive import keepalive
 from thirdparty.oset.pyoset import oset

lib/core/purge.py

@@ -1,82 +0,0 @@
-#!/usr/bin/env python
-
-"""
-Copyright (c) 2006-2014 sqlmap developers (http://sqlmap.org/)
-See the file 'doc/COPYING' for copying permission
-"""
-
-import os
-import random
-import shutil
-import stat
-import string
-
-from lib.core.data import logger
-
-def purge(directory):
-    """
-    Safely removes content from a given directory
-    """
-
-    if not os.path.isdir(directory):
-        warnMsg = "skipping purging of directory '%s' as it does not exist" % directory
-        logger.warn(warnMsg)
-        return
-
-    infoMsg = "purging content of directory '%s'..." % directory
-    logger.info(infoMsg)
-
-    filepaths = []
-    dirpaths = []
-
-    for rootpath, directories, filenames in os.walk(directory):
-        dirpaths.extend([os.path.abspath(os.path.join(rootpath, _)) for _ in directories])
-        filepaths.extend([os.path.abspath(os.path.join(rootpath, _)) for _ in filenames])
-
-    logger.debug("changing file attributes")
-    for filepath in filepaths:
-        try:
-            os.chmod(filepath, stat.S_IREAD | stat.S_IWRITE)
-        except:
-            pass
-
-    logger.debug("writing random data to files")
-    for filepath in filepaths:
-        try:
-            filesize = os.path.getsize(filepath)
-            with open(filepath, 'w+b') as f:
-                f.write("".join(chr(random.randint(0, 255)) for _ in xrange(filesize)))
-        except:
-            pass
-
-    logger.debug("truncating files")
-    for filepath in filepaths:
-        try:
-            with open(filepath, 'w') as f:
-                pass
-        except:
-            pass
-
-    logger.debug("renaming filenames to random values")
-    for filepath in filepaths:
-        try:
-            os.rename(filepath, os.path.join(os.path.dirname(filepath), "".join(random.sample(string.ascii_letters, random.randint(4, 8)))))
-        except:
-            pass
-
-    dirpaths.sort(cmp=lambda x, y: y.count(os.path.sep) - x.count(os.path.sep))
-
-    logger.debug("renaming directory names to random values")
-    for dirpath in dirpaths:
-        try:
-            os.rename(dirpath, os.path.join(os.path.dirname(dirpath), "".join(random.sample(string.ascii_letters, random.randint(4, 8)))))
-        except:
-            pass
-
-    logger.debug("deleting the whole directory tree")
-    os.chdir(os.path.join(directory, ".."))
-
-    try:
-        shutil.rmtree(directory)
-    except OSError, ex:
-        logger.error("problem occurred while removing directory '%s' ('%s')" % (directory, ex))