more stuffssss

README.md

@@ -5,10 +5,18 @@ A tool for simplifying power analysis attacks against other gadgets.
 Built on the Raspberry Pi Pico W, it runs a PWM channel to do manual clock control on the nugget you're hacking, and then reads an ADC channel to measure voltage used at each clock cycle.
 
 ## Screenshots
+Config page  
 ![index-page](resources/index.png)
   
+Running page  
 ![running-page](resources/running.png)
   
+Serial terminal at boot  
+![terminal-boot](resources/terminal0.png)
+  
+Serial terminal while sampling and looping  
+![terminal-running](resources/terminal-running.png)
+
 ## Installation
 Make sure you have Micropython installed on your Pico W.
 Open up your Pico W in your favorite IDE (Thonny and VS Code are commonly used) and upload main.py.
@@ -43,6 +51,12 @@ If you're having trouble, see the official [Raspberry Pi Pico W Getting Started
      Your wifi password  
      Dont forget the 's around the password  
   
+`power_on_pin`  
+     The pin that toggles 3.3v on and off to activate and reset the nugget board  
+  
+`seconds_awake`
+     The number of seconds to turn on power_on_pin before resetting  
+
 ### On the Webpage
 `Duty Cycle`  
      The Percentage of the time that the clock is on vs off  

main.py

@@ -1,10 +1,10 @@
 import network
 import socket
 import machine
+from machine import Timer
 import utime
 import _thread
 import math
-#import os
 
 
 # gpio pin to read power analysis off of (int)
@@ -15,9 +15,16 @@ power_analysis_pin = 28
 # output gpio pin generating clock pulse
 clock_pulse_pin = 20
 
+# this pin gets power and toggles off and on every X PWM cycles
+# it powers the board you're targeting
+power_on_pin = 2
+
+# how many seconds to power the board before reset
+seconds_awake = 3
+
 # Wi-Fi credentials
 ssid = 'your-wifi-name-here'
-password = 'yuour-wifi-password-here'
+password = 'your-wifi-password-here'
 
 # == Clock Pulse Genertor == #
 # starts a pwm
@@ -63,7 +70,7 @@ header = f"""
             <link rel="preload" href="http://10.0.0.80/ComicCode-Regular.woff2" as="font" type="font/woff2" crossorigin>
             <style>
             body {{
-                font-family: 'Comic Code';
+                font-family: 'Comic Code', 'Comic Sans', Monospace;
                 font-weight: normal;
                 font-style: normal;
             }}
@@ -137,6 +144,21 @@ def running_webpage(outpin = 20, adcpin = 28, duty_cycle = 0.5, frequency = 2000
 
 print("Board Started or Reset\n===========\n\n")
 
+toggle_power = machine.Pin(power_on_pin, machine.Pin.OUT)
+toggle_power.value(0)
+
+# I like to have the LED on :3
+led = machine.Pin('LED', machine.Pin.OUT)
+led.value(1)
+
+# do da thingggggssssss
+def reset_init(t):
+    print("==Looping==")
+
+    toggle_power.value(1)
+    utime.sleep(1)
+    toggle_power(0)
+
 # Connect to WLAN
 wlan = network.WLAN(network.STA_IF)
 wlan.active(True)
@@ -166,10 +188,6 @@ s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 s.bind(addr)
 s.listen()
 
-# I like to have the LED on :3
-led = machine.Pin('LED', machine.Pin.OUT)
-led.value(1)
-
 while True:
     try:
         conn, addr = s.accept()
@@ -193,16 +211,20 @@ while True:
                 if req_split[0] == 'duty_cycle':
                         duty_cycle_in = float(req_split[1])
                 elif req_split[0] == 'freq':
-                        freq_in = int(req_split[1])
+                    freq_in = int(req_split[1])
                 elif req_split[0] == 'samples_per_pulse':
                         samples_per_pulse_in =  int(req_split[1])
-            print(f"power_analysis_pin: {power_analysis_pin}\nduty_cycle_in: {duty_cycle_in}\nfreq_in: {freq_in}\nsamples_per_pulse_in:{samples_per_pulse_in}")
+            loop_time = 1000*(seconds_awake-1)
+
+            print(f"power_analysis_pin: {power_analysis_pin}\nduty_cycle_in: {duty_cycle_in}\nfreq_in: {freq_in}\nsamples_per_pulse_in: {samples_per_pulse_in}\nloop_time (ms): {loop_time}\npower_on_pin: {power_on_pin}")
 
             # run the clock pulse (PWM) task 
             do_pwm(clock_pulse_pin, duty_cycle_in, freq_in)
 
             # fork the voltage measuring ADC reading to the second core
-            _thread.start_new_thread(do_adc, (power_analysis_pin, duty_cycle_in, freq_in, samples_per_pulse_in))
+            adc_thread = _thread.start_new_thread(do_adc, (power_analysis_pin, duty_cycle_in, freq_in, samples_per_pulse_in))
+
+            Timer(mode=Timer.PERIODIC, period=loop_time, callback=reset_init)
 
             response = running_webpage()