setup.py

# NOTE: See the README file for a list of dependencies to install.

from __future__ import print_function
import sys

try:
 from setuptools import setup, Extension
except ImportError:
 print("No setuptools found, attempting to use distutils instead.")
 from distutils.core import setup, Extension

err = []
warn = []
apt_get_pkgs = []
pip_pkgs = []

#Check pyUSB 1.x dependency present.
#Support for 0.x is deprecated and being removed.
try:
 import usb
except ImportError:
 err.append("usb")
 apt_get_pkgs.append("python-usb")

try:
 import usb.core
except ImportError:
 err.append("You are using pyUSB 0.x. Upgrade to pyUSB 1.x.")

# TODO: Ideally we would detect missing python-dev and libgcrypt-dev to give better errors.

# Dot15d4 is a dep of some of the newer tools
try:
 from scapy.all import Dot15d4
except ImportError:
 warn.append("Scapy (with dot15d4 layer)")
 pip_pkgs.append("scapy")

if len(err) > 0:
 print("""
Library requirements not met. Install the following libraries, then re-run the setup script.

{}\n""".format('\n'.join(err)), file=sys.stderr)

if len(warn) > 0:
 print("""
Library recommendations not met. For full support, install the following libraries, then re-run the setup script.

{}\n""".format('\n'.join(warn)), file=sys.stderr)

if len(apt_get_pkgs) > 0 or len(pip_pkgs) > 0:
 print("The following commands should install these dependencies on Ubuntu, and can be adapted for other OSs:", file=sys.stderr)
 if len(apt_get_pkgs) > 0:
 print("\tsudo apt-get install -y {}".format(' '.join(apt_get_pkgs)), file=sys.stderr)
 if len(pip_pkgs) > 0:
 print("\tpip3 install {}".format(' '.join(pip_pkgs)), file=sys.stderr)

if len(err) > 0:
 sys.exit(1)

zigbee_crypt = Extension('zigbee_crypt',
 sources = ['zigbee_crypt/zigbee_crypt.c'],
 libraries = ['gcrypt'],
 include_dirs = ['/usr/local/include', '/usr/include', '/sw/include/', 'zigbee_crypt'],
 library_dirs = ['/usr/local/lib', '/usr/lib','/sw/var/lib/']
 )

setup(name = 'killerbee',
 version = '3.0.0-beta.2',
 description = 'ZigBee and IEEE 802.15.4 Attack Framework and Tools',
 author = 'Joshua Wright, Ryan Speers',
 author_email = 'jwright@willhackforsushi.com, ryan@riverloopsecurity.com',
 license = 'LICENSE.txt',
 packages = ['killerbee'],
 scripts = ['tools/zbdump', 'tools/zbgoodfind', 'tools/zbid', 'tools/zbreplay',
 'tools/zbconvert', 'tools/zbdsniff', 'tools/zbstumbler', 'tools/zbassocflood',
 'tools/zbscapy', 'tools/zbwireshark', 'tools/zbkey',
 'tools/zbwardrive', 'tools/zbopenear', 'tools/zbfakebeacon',
 'tools/zborphannotify', 'tools/zbpanidconflictflood', 'tools/zbrealign', 'tools/zbcat',
 'tools/zbjammer', 'tools/kbbootloader'],
 install_requires=['pyserial>=2.0', 'pyusb', 'pycrypto', 'rangeparser', 'scapy'],
 # NOTE: pygtk doesn't install via distutils on non-Windows hosts
 ext_modules = [zigbee_crypt],
 )