setup.py

import fnmatch
import glob
import os
import setuptools
import re
import subprocess

from src.drozer import meta
from sys import platform

def find_files(src):
    matches = []

    for root, dirnames, filenames in os.walk(src):
        matches.extend(map(lambda f: os.path.join(root, f), filenames))

    return matches

def find_libs(src):
    matches = []

    for root, dirnames, filenames in os.walk(src):
        for filename in fnmatch.filter(dirnames, 'lib'):
            matches.extend(glob.glob(os.path.join(root, filename, "*", "*")))
        for filename in fnmatch.filter(dirnames, 'libs'):
            matches.extend(glob.glob(os.path.join(root, filename, "*", "*")))

    return map(lambda fn: os.path.basename(fn), filter(lambda fn: os.path.isfile(fn), matches))

# Do a system check when installing bash complete script
def get_install_data():
	install_data = []
	if platform == "linux" or platform == "linux2":
		install_data = [('/etc/bash_completion.d',['scripts/drozer'])]

	return install_data

def get_executable_scripts():
  scripts = ["bin/drozer", "bin/drozer-complete", "bin/drozer-repository"]
  if platform == "win32":
    scripts.append("bin/drozer.bat")

  return scripts

def get_pwd():
	pwd = ''
	
	if platform == "linux" or platform == "linux2":
		pwd = 'src/drozer'
	elif platform == "win32":
		pwd = 'src\\drozer'
	
	return pwd

def clear_apks():
	pwd = get_pwd()
	
	if platform == 'linux' or platform == 'linux2':
		pwd += '/modules'
	elif platform == 'win32':
		pwd += '\\modules'
	
	for root, dirnames, filenames in os.walk(pwd):
		for filename in filenames:
			if (fnmatch.fnmatch(filename, "*.class") or fnmatch.fnmatch(filename, "*.apk")):
				#print os.path.join(root, filename)
				os.remove(os.path.join(root, filename))

def make_apks():

	pwd = get_pwd()
	lib = os.path.dirname(os.path.realpath(__file__))
	dx =''

	if platform == 'linux' or 'linux2':
		pwd += '/modules'
		lib += '/src/drozer/lib/'
		dx = 'dx'
	elif platform == 'win32':
		pwd += '\\modules'
		lib += '\\src\\drozer\\lib\\'
		dx = 'dx.bat'

	#If apks exist, delete them and generate new ones
	clear_apks()

	# Generate apks
	for root, dirnames, filenames in os.walk(pwd):
		for filename in filenames:
			if (fnmatch.fnmatch(filename, "*.java")):
				#Compile java
				javac_cmd = ['javac', '-cp', lib+'android.jar', filename]

				#Build apk
				m = re.search('(.+?)(\.[^.]*$|$)',filename)
				dx_cmd = [lib+dx, '--dex', '--output='+m.group(1)+'.apk',m.group(1)+'*.class']

				if platform == "linux2" or platform == "linux":
					subprocess.call(' '.join(javac_cmd),shell=True,cwd=root)

					subprocess.call(' '.join(dx_cmd),shell=True,cwd=root)
				elif platform == "win32":
					subprocess.call(javac_cmd,shell=True,cwd=root)

					subprocess.call(dx_cmd,shell=True,cwd=root)
				
def get_package_data():
	data = {"":[]}
	pwd = get_pwd()

	#Make sure we build apks before generating a package
	make_apks()

	for root, dirnames, filenames in os.walk(pwd):
		for filename in filenames:
			if not (fnmatch.fnmatch(filename, "*.class") or fnmatch.fnmatch(filename, "*.pyc")):
				data[""].append(os.path.join(root, filename)[11:])
	return data

setuptools.setup(
  name = meta.name,
  version = str(meta.version),
  author = meta.vendor,
  author_email = meta.contact,
  description = meta.description,
  long_description = meta.long_description,
  license = meta.license,
  keywords = meta.keywords,
  url = meta.url,

  packages = setuptools.find_packages("src"),
  package_dir = {   "drozer": "src/drozer",
                    "mwr": "src/mwr",
                    "pydiesel": "src/pydiesel" },
  package_data = get_package_data(),
  scripts = get_executable_scripts(),
  install_requires = ["protobuf>=2.6.1","pyopenssl>=16.2", "pyyaml>=3.11"],
  data_files = get_install_data(),
  classifiers = [])