DragonProbe/src/main.c

/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2019 Ha Thach (tinyusb.org)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 */

#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#include "tusb_config.h"

#include "bsp/board.h" /* a tinyusb header */
#include "tusb.h"

#include "DAP_config.h" /* ARM code *assumes* this is included prior to DAP.h */
#include "DAP.h"

#include "util.h"
#include "protocfg.h"
#include "protos.h"
#include "libco.h"

#ifdef PICO_BOARD
#include <pico/binary_info.h>
/*#include <hardware/i2c.h>
#include "pinout.h"*/
#endif

static cothread_t mainthread;

void thread_yield(void) {
	co_switch(mainthread);
}

#define DEFAULT_STACK_SIZE 1024

#ifdef DBOARD_HAS_UART
static cothread_t uartthread;
static uint8_t uartstack[DEFAULT_STACK_SIZE];

static void uart_thread_fn(void) {
	cdc_uart_init();
	thread_yield();
	while (1) {
		cdc_uart_task();
		thread_yield();
	}
}
#endif

#ifdef DBOARD_HAS_SERPROG
static cothread_t serprogthread;
static uint8_t serprogstack[DEFAULT_STACK_SIZE];

static void serprog_thread_fn(void) {
  cdc_serprog_init();
  thread_yield();
  while (1) {
    cdc_serprog_task();
    thread_yield();
  }
}
#endif

// FIXME
extern uint32_t co_active_buffer[64];
uint32_t co_active_buffer[64];
extern cothread_t co_active_handle;
cothread_t co_active_handle;

int main(void) {
	mainthread = co_active();

	// TODO: split this out in a bsp-specific file
#if defined(PICO_BOARD) && !defined(USE_USBCDC_FOR_STDIO)
	// use hardcoded values from TinyUSB board.h
	bi_decl(bi_2pins_with_func(0, 1, GPIO_FUNC_UART));
	/*i2c_init(PINOUT_I2C_DEV, 100*1000);

	gpio_set_function(PINOUT_I2C_SCL, GPIO_FUNC_I2C);
	gpio_set_function(PINOUT_I2C_SDA, GPIO_FUNC_I2C);
	gpio_pull_up(PINOUT_I2C_SCL);
	gpio_pull_up(PINOUT_I2C_SDA);

	bi_decl(bi_2pins_with_func(PINOUT_I2C_SCL, PINOUT_I2C_SDA, GPIO_FUNC_I2C));*/
#endif
	board_init();

#ifdef DBOARD_HAS_UART
	uartthread = co_derive(uartstack, sizeof uartstack, uart_thread_fn);
	co_switch(uartthread); // will call cdc_uart_init() on correct thread
#endif
#ifdef DBOARD_HAS_SERPROG
	serprogthread = co_derive(serprogstack, sizeof serprogstack, serprog_thread_fn);
	co_switch(serprogthread); // will call cdc_serprog_init() on correct thread
#endif
#ifdef DBOARD_HAS_CMSISDAP
	DAP_Setup();
#endif

	tusb_init();

#ifdef USE_USBCDC_FOR_STDIO
	stdio_usb_init();
#endif

	while (1) {
		/*uint8_t val = 0x12;
		i2c_write_timeout_us(PINOUT_I2C_DEV, 0x13, &val, 1, false, 1000*1000);*/

		tud_task(); // tinyusb device task
#ifdef DBOARD_HAS_UART
		co_switch(uartthread);
#endif

		//i2c_write_timeout_us(PINOUT_I2C_DEV, 0x13, &val, 1, false, 1000*1000);

		tud_task(); // tinyusb device task
#ifdef DBOARD_HAS_SERPROG
		co_switch(serprogthread);
#endif
	}

	return 0;
}

//--------------------------------------------------------------------+
// USB HID
//--------------------------------------------------------------------+

// Invoked when received GET_REPORT control request
// Application must fill buffer report's content and return its length.
// Return zero will cause the stack to STALL request
uint16_t tud_hid_get_report_cb(uint8_t instance, uint8_t report_id,
		hid_report_type_t report_type, uint8_t* buffer, uint16_t reqlen) {
	// TODO not Implemented
	(void) instance;
	(void) report_id;
	(void) report_type;
	(void) buffer;
	(void) reqlen;

	return 0;
}

void tud_hid_set_report_cb(uint8_t instance, uint8_t report_id,
		hid_report_type_t report_type, uint8_t const* RxDataBuffer, uint16_t bufsize) {
	static uint8_t TxDataBuffer[CFG_TUD_HID_EP_BUFSIZE];
	uint32_t response_size = TU_MIN(CFG_TUD_HID_EP_BUFSIZE, bufsize);

	// This doesn't use multiple report and report ID
	(void) instance;
	(void) report_id;
	(void) report_type;

#ifdef DBOARD_HAS_CMSISDAP
	DAP_ProcessCommand(RxDataBuffer, TxDataBuffer);
#endif

	tud_hid_report(0, TxDataBuffer, response_size);
}