mirror of
https://github.com/dbisu/pico-ducky.git
synced 2025-12-06 02:41:45 +00:00
b99ddf37e5
* Refactor code to support Pico W board * Updates for Pico W support * Update based on changes in CircuitPython v8.0.0 beta 6 * Updated bundle download to 8.x, fixed a few typos * Updating documentation for web interface * Update support for CircuitPython 8.0.0, fix many bugs * Document multiple payload options * Update copyright dates * Fixed typo
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
# License : GPLv2.0
|
|
# copyright (c) 2023 Dave Bailey
|
|
# Author: Dave Bailey (dbisu, @daveisu)
|
|
# Pico and Pico W board support
|
|
|
|
|
|
import usb_hid
|
|
from adafruit_hid.keyboard import Keyboard
|
|
|
|
# comment out these lines for non_US keyboards
|
|
from adafruit_hid.keyboard_layout_us import KeyboardLayoutUS as KeyboardLayout
|
|
from adafruit_hid.keycode import Keycode
|
|
|
|
# uncomment these lines for non_US keyboards
|
|
# replace LANG with appropriate language
|
|
#from keyboard_layout_win_LANG import KeyboardLayout
|
|
#from keycode_win_LANG import Keycode
|
|
|
|
import supervisor
|
|
|
|
|
|
import time
|
|
import digitalio
|
|
from board import *
|
|
import board
|
|
from duckyinpython import *
|
|
if(board.board_id == 'raspberry_pi_pico_w'):
|
|
import wifi
|
|
from webapp import *
|
|
|
|
|
|
# sleep at the start to allow the device to be recognized by the host computer
|
|
time.sleep(.5)
|
|
|
|
def startWiFi():
|
|
import ipaddress
|
|
# Get wifi details and more from a secrets.py file
|
|
try:
|
|
from secrets import secrets
|
|
except ImportError:
|
|
print("WiFi secrets are kept in secrets.py, please add them there!")
|
|
raise
|
|
|
|
print("Connect wifi")
|
|
#wifi.radio.connect(secrets['ssid'],secrets['password'])
|
|
wifi.radio.start_ap(secrets['ssid'],secrets['password'])
|
|
|
|
HOST = repr(wifi.radio.ipv4_address_ap)
|
|
PORT = 80 # Port to listen on
|
|
print(HOST,PORT)
|
|
|
|
# turn off automatically reloading when files are written to the pico
|
|
#supervisor.disable_autoreload()
|
|
supervisor.runtime.autoreload = False
|
|
|
|
if(board.board_id == 'raspberry_pi_pico'):
|
|
led = pwmio.PWMOut(board.LED, frequency=5000, duty_cycle=0)
|
|
elif(board.board_id == 'raspberry_pi_pico_w'):
|
|
led = digitalio.DigitalInOut(board.LED)
|
|
led.switch_to_output()
|
|
|
|
|
|
progStatus = False
|
|
progStatus = getProgrammingStatus()
|
|
print("progStatus", progStatus)
|
|
if(progStatus == False):
|
|
print("Finding payload")
|
|
# not in setup mode, inject the payload
|
|
payload = selectPayload()
|
|
print("Running ", payload)
|
|
runScript(payload)
|
|
|
|
print("Done")
|
|
else:
|
|
print("Update your payload")
|
|
|
|
led_state = False
|
|
|
|
async def main_loop():
|
|
global led,button1
|
|
|
|
button_task = asyncio.create_task(monitor_buttons(button1))
|
|
if(board.board_id == 'raspberry_pi_pico_w'):
|
|
pico_led_task = asyncio.create_task(blink_pico_w_led(led))
|
|
print("Starting Wifi")
|
|
startWiFi()
|
|
print("Starting Web Service")
|
|
webservice_task = asyncio.create_task(startWebService())
|
|
await asyncio.gather(pico_led_task, button_task, webservice_task)
|
|
else:
|
|
pico_led_task = asyncio.create_task(blink_pico_led(led))
|
|
await asyncio.gather(pico_led_task, button_task)
|
|
|
|
asyncio.run(main_loop())
|