main.c

#include "pico/stdlib.h"
#include "hardware/pio.h"
#include "hardware/watchdog.h"
#include "pico/float.h"
// #include "hardware/regs/glitch_detector.h"
#include "hardware/regs/powman.h"
#include "hardware/structs/powman.h"
#include "hardware/structs/otp.h"
#include "rp2350_playground.h"

#include "display/display.h"

#include <stdio.h>

#pragma GCC push_options
#pragma GCC optimize("O0")
#define OUTER_LOOP_CNT 200
#define INNER_LOOP_CNT 200


#define nop() __asm__ volatile ("nop")

void __not_in_flash_func(glitch_loop)(void)
{
	// volatile register uint32_t i, j;
	// volatile register uint32_t cnt;
	uint32_t blink_status = 1;
	bool glitch_detector_triggered = (powman_hw->chip_reset & 0x04000000) != 0;

	if (glitch_detector_triggered)
	{
		uart_putc_raw(TARGET_UART_INSTANCE, 'G');
	}
	else
	{
		uart_putc_raw(TARGET_UART_INSTANCE, 'R');
	}

	// if (!glitch_detector_armed())
	// {
	// 	uart_putc_raw(TARGET_UART_INSTANCE, 'S');
	// }
	// else
	// {
	// 	uart_putc_raw(TARGET_UART_INSTANCE, 'F');
	// }

	watchdog_update();
	// cnt = 0;
	uint32_t cnt = 0, i ,j;

	watchdog_update();
	gpio_put(TRIGGER_PIN, 1);
	sleep_ms(1);
	watchdog_update();
	gpio_put(TRIGGER_PIN, 0);

	
	gpio_put(TRIGGER2_PIN, 1);
	for (i = 0; i < OUTER_LOOP_CNT; i++)
	{
		for (j = 0; j < INNER_LOOP_CNT; j++)
		{
			cnt++;
		}
	}
	gpio_put(TRIGGER2_PIN, 0);

	// const bool glitch_detected = i != OUTER_LOOP_CNT || j != INNER_LOOP_CNT || cnt != (OUTER_LOOP_CNT * INNER_LOOP_CNT);


	// Check for glitch
	if (i != OUTER_LOOP_CNT || j != INNER_LOOP_CNT || cnt != (OUTER_LOOP_CNT * INNER_LOOP_CNT))
	{
		// Q indicates successful glitch
		watchdog_update();
		uart_putc_raw(uart0, 'Q');
	}
	else
	{
		// N indicates regular execution
		uart_putc_raw(uart0, 'N');
	}
	watchdog_update();

	// char buffer[20];
	// snprintf(buffer, sizeof(buffer), "%d, %d, %d", i, j, cnt);
	// uart_puts(uart0, buffer);
	
	
}
#pragma GCC pop_options


void __not_in_flash_func(glitch_before_loading)(void)
{
	// int label = -1;
	volatile register float tmp1 = 0.0;
	volatile float features[11] = {4.6f, 0.12f, 0.0f, 0.9f, 0.01f, 1.0f, 6.0f, 0.99f, 2.74f, 0.33f, 8.4f};
	volatile register float correct_val = features[6];

	uint32_t blink_status = 1;
	bool glitch_detector_triggered = (powman_hw->chip_reset & 0x04000000) != 0;

	watchdog_update();
	if (glitch_detector_triggered)
	{
		uart_putc_raw(TARGET_UART_INSTANCE, 'G');
	}
	else
	{
		uart_putc_raw(TARGET_UART_INSTANCE, 'R');
	}

	watchdog_update();
	gpio_put(TRIGGER_PIN, 1);
	sleep_ms(1);
	watchdog_update();
	gpio_put(TRIGGER_PIN, 0);

	
	gpio_put(TRIGGER2_PIN, 1);
	delayNops(50);
	nop();
	tmp1 = features[6];
	delayNops(50);
	nop();
	gpio_put(TRIGGER2_PIN, 0);

	// Check for glitch
	if (features[6] != correct_val)
	{
		// Q indicates successful glitch
		watchdog_update();
		uart_putc_raw(uart0, 'Q');
	}
	else
	{
		// N indicates regular execution
		uart_putc_raw(uart0, 'N');
	}
	watchdog_update();

	watchdog_update();
    char buffer[12];
    snprintf(buffer, sizeof(buffer), "%.1f", tmp1); // Format tmp2 as a string
    uart_puts(uart0, buffer);
}
#pragma GCC pop_options

inline void delayNops(long nops)
{
    __asm__ volatile(
    "L_%=_delayNops:"   "\n\t"
    "subs   %0, #1"       "\n\t" // Subtract 1 from nops
    "bne    L_%=_delayNops"   "\n" // Loop if nops is not zero
    : "+r" (nops) :
    );
}

int main()
{

	init_uart();
	// while (true) {
    //     uart_puts(TARGET_UART_INSTANCE, "Hello from RP2350!\r\n");
    //     sleep_ms(1000);
    // }
	watchdog_enable(100, 1);
	gpio_init(TRIGGER_PIN);
	gpio_set_dir(TRIGGER_PIN, GPIO_OUT);
	gpio_put(TRIGGER_PIN, 0);

	gpio_init(TRIGGER2_PIN);
	gpio_set_dir(TRIGGER2_PIN, GPIO_OUT);
	gpio_put(TRIGGER2_PIN, 0);

	for(int i=0; i < 5; i++) {
		gpio_init(joy_io[i]);
		gpio_set_dir(joy_io[i], GPIO_IN);
		gpio_pull_up(joy_io[i]);
	}

	glitch_detector_disarm();
	glitch_loop();
	//glitch_before_loading();

	for (;;) { watchdog_update(); }
}