main.c 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329
  1. /*
  2. * =====================================================================================
  3. *
  4. * ________ .__ __ ________ ____ ________
  5. * \_____ \ __ __|__| ____ | | __\______ \ _______ _/_ |/ _____/
  6. * / / \ \| | \ |/ ___\| |/ / | | \_/ __ \ \/ /| / __ \
  7. * / \_/. \ | / \ \___| < | ` \ ___/\ / | \ |__\ \
  8. * \_____\ \_/____/|__|\___ >__|_ \/_______ /\___ >\_/ |___|\_____ /
  9. * \__> \/ \/ \/ \/ \/
  10. *
  11. * www.optixx.org
  12. *
  13. *
  14. * Version: 1.0
  15. * Created: 07/21/2009 03:32:16 PM
  16. * Author: david@optixx.org
  17. *
  18. * =====================================================================================
  19. */
  20. #include <avr/interrupt.h>
  21. #include <util/delay.h>
  22. #include <stdlib.h>
  23. #include <avr/pgmspace.h>
  24. #include <avr/eeprom.h>
  25. #include <string.h>
  26. #include "usbdrv.h"
  27. #include "oddebug.h"
  28. #include "config.h"
  29. #include "requests.h"
  30. #include "uart.h"
  31. #include "sram.h"
  32. #include "debug.h"
  33. #include "info.h"
  34. #include "dump.h"
  35. #include "crc.h"
  36. #include "usb_bulk.h"
  37. #include "timer.h"
  38. #include "watchdog.h"
  39. #include "rle.h"
  40. #include "loader.h"
  41. #include "command.h"
  42. #include "shared_memory.h"
  43. #include "irq.h"
  44. #include "pwm.h"
  45. #include "testing.h"
  46. #include "shell.h"
  47. #include "system.h"
  48. #ifndef NO_DEBUG
  49. extern FILE uart_stdout;
  50. #endif
  51. extern system_t my_system;
  52. uint8_t debug_level = (DEBUG | DEBUG_USB | DEBUG_CRC);
  53. usb_transaction_t usb_trans;
  54. usbMsgLen_t usbFunctionSetup(uchar data[8])
  55. {
  56. usbRequest_t *rq = (void *) data;
  57. uint8_t ret_len = 0;
  58. if (rq->bRequest == USB_BULK_UPLOAD_INIT) {
  59. usb_trans.req_bank = 0;
  60. usb_trans.rx_remaining = 0;
  61. debug_P(DEBUG_USB, PSTR("USB_BULK_UPLOAD_INIT: %i %i\n"),
  62. rq->wValue.word, rq->wIndex.word);
  63. usb_trans.req_bank_size = (uint32_t) (1L << rq->wValue.word);
  64. usb_trans.req_bank_cnt = rq->wIndex.word;
  65. usb_trans.req_addr_end =
  66. (uint32_t) usb_trans.req_bank_size * usb_trans.req_bank_cnt;
  67. usb_trans.req_percent = 0;
  68. usb_trans.req_percent_last = 0;
  69. usb_trans.sync_errors = 0;
  70. debug_P(DEBUG_USB,
  71. PSTR
  72. ("USB_BULK_UPLOAD_INIT: bank_size=0x%08lx bank_cnt=0x%x end_addr=0x%08lx\n"),
  73. usb_trans.req_bank_size, usb_trans.req_bank_cnt,
  74. usb_trans.req_addr_end);
  75. shared_memory_write(SHARED_MEM_TX_CMD_UPLOAD_START, 0);
  76. shared_memory_write(SHARED_MEM_TX_CMD_BANK_COUNT,
  77. usb_trans.req_bank_cnt);
  78. #if DO_TIMER
  79. if (usb_trans.req_addr == 0x000000) {
  80. #ifndef NO_DEBUG
  81. timer_start();
  82. #endif
  83. }
  84. #endif
  85. /*
  86. * -------------------------------------------------------------------------
  87. */
  88. } else if (rq->bRequest == USB_BULK_UPLOAD_ADDR) {
  89. usb_trans.req_state = REQ_STATUS_BULK_UPLOAD;
  90. usb_trans.req_addr = rq->wValue.word;
  91. usb_trans.req_addr = usb_trans.req_addr << 16;
  92. usb_trans.req_addr = usb_trans.req_addr | rq->wIndex.word;
  93. usb_trans.rx_remaining = rq->wLength.word;
  94. if (usb_trans.req_addr
  95. && usb_trans.req_addr % usb_trans.req_bank_size == 0) {
  96. #if DO_TIMER
  97. #ifndef NO_DEBUG
  98. #ifdef FLT_DEBUG
  99. debug_P(DEBUG_USB,
  100. PSTR
  101. ("USB_BULK_UPLOAD_ADDR: req_bank=0x%02x addr=0x%08lx time=%.4f\n"),
  102. usb_trans.req_bank, usb_trans.req_addr, timer_stop());
  103. #else
  104. debug_P(DEBUG_USB,
  105. PSTR
  106. ("USB_BULK_UPLOAD_ADDR: req_bank=0x%02x addr=0x%08lx time=%i\n"),
  107. usb_trans.req_bank, usb_trans.req_addr, timer_stop_int());
  108. #endif
  109. timer_start();
  110. #endif
  111. #endif
  112. usb_trans.req_bank++;
  113. } else {
  114. sram_bulk_write_start(usb_trans.req_addr);
  115. }
  116. ret_len = USB_MAX_TRANS;
  117. /*
  118. * -------------------------------------------------------------------------
  119. */
  120. } else if (rq->bRequest == USB_BULK_UPLOAD_NEXT) {
  121. usb_trans.req_state = REQ_STATUS_BULK_UPLOAD;
  122. usb_trans.req_addr = rq->wValue.word;
  123. usb_trans.req_addr = usb_trans.req_addr << 16;
  124. usb_trans.req_addr = usb_trans.req_addr | rq->wIndex.word;
  125. usb_trans.rx_remaining = rq->wLength.word;
  126. #if DO_SHM
  127. usb_trans.req_percent =
  128. (uint32_t) (100 * usb_trans.req_addr) / usb_trans.req_addr_end;
  129. if (usb_trans.req_percent != usb_trans.req_percent_last) {
  130. shared_memory_write(SHARED_MEM_TX_CMD_UPLOAD_PROGESS,
  131. usb_trans.req_percent);
  132. }
  133. usb_trans.req_percent_last = usb_trans.req_percent;
  134. shared_memory_scratchpad_region_save_helper(usb_trans.req_addr);
  135. #endif
  136. if (usb_trans.req_addr
  137. && (usb_trans.req_addr % usb_trans.req_bank_size) == 0) {
  138. #if DO_TIMER
  139. #ifndef NO_DEBUG
  140. #ifdef FLT_DEBUG
  141. debug_P(DEBUG_USB,
  142. PSTR
  143. ("USB_BULK_UPLOAD_NEXT: req_bank=0x%02x addr=0x%08lx time=%.4f\n"),
  144. usb_trans.req_bank, usb_trans.req_addr, timer_stop());
  145. #else
  146. debug_P(DEBUG_USB,
  147. PSTR
  148. ("USB_BULK_UPLOAD_NEXT: req_bank=0x%02x addr=0x%08lx time=%i\n"),
  149. usb_trans.req_bank, usb_trans.req_addr, timer_stop_int());
  150. #endif
  151. timer_start();
  152. #endif
  153. #endif
  154. usb_trans.req_bank++;
  155. #if DO_SHM
  156. shared_memory_write(SHARED_MEM_TX_CMD_BANK_CURRENT,
  157. usb_trans.req_bank);
  158. #endif
  159. }
  160. ret_len = USB_MAX_TRANS;
  161. /*
  162. * -------------------------------------------------------------------------
  163. */
  164. } else if (rq->bRequest == USB_BULK_UPLOAD_END) {
  165. debug_P(DEBUG_USB, PSTR("USB_BULK_UPLOAD_END:\n"));
  166. usb_trans.req_state = REQ_STATUS_IDLE;
  167. sram_bulk_write_end();
  168. #if DO_SHM
  169. shared_memory_write(SHARED_MEM_TX_CMD_UPLOAD_END, 0);
  170. #endif
  171. ret_len = 0;
  172. /*
  173. * -------------------------------------------------------------------------
  174. */
  175. } else if (rq->bRequest == USB_CRC) {
  176. usb_trans.req_addr = rq->wValue.word;
  177. usb_trans.req_addr = usb_trans.req_addr << 16;
  178. usb_trans.req_addr = usb_trans.req_addr | rq->wIndex.word;
  179. debug_P(DEBUG_USB, PSTR("USB_CRC: addr=0x%08lx \n"),
  180. usb_trans.req_addr);
  181. #if DO_CRC_CHECK
  182. crc_check_bulk_memory(0x000000, usb_trans.req_addr,
  183. usb_trans.req_bank_size);
  184. #endif
  185. ret_len = 0;
  186. /*
  187. * -------------------------------------------------------------------------
  188. */
  189. } else if (rq->bRequest == USB_MODE_SNES) {
  190. usb_trans.req_state = REQ_STATUS_SNES;
  191. debug_P(DEBUG_USB, PSTR("USB_MODE_SNES:\n"));
  192. ret_len = 0;
  193. /*
  194. * -------------------------------------------------------------------------
  195. */
  196. } else if (rq->bRequest == USB_MODE_AVR) {
  197. usb_trans.req_state = REQ_STATUS_AVR;
  198. debug_P(DEBUG_USB, PSTR("USB_MODE_AVR:\n"));
  199. ret_len = 0;
  200. /*
  201. * -------------------------------------------------------------------------
  202. */
  203. } else if (rq->bRequest == USB_AVR_RESET) {
  204. debug_P(DEBUG_USB, PSTR("USB_AVR_RESET:\n"));
  205. soft_reset();
  206. ret_len = 0;
  207. /*
  208. * -------------------------------------------------------------------------
  209. */
  210. } else if (rq->bRequest == USB_SET_LAODER) {
  211. debug_P(DEBUG_USB, PSTR("USB_SET_LAODER:\n"));
  212. usb_trans.loader_enabled = rq->wValue.word;
  213. ret_len = 0;
  214. }
  215. usbMsgPtr = usb_trans.rx_buffer;
  216. return ret_len;
  217. }
  218. /*
  219. * -------------------------------------------------------------------------
  220. */
  221. void globals_init()
  222. {
  223. memset(&usb_trans, 0, sizeof(usb_transaction_t));
  224. usb_trans.req_addr = 0;
  225. usb_trans.req_addr_end = 0;
  226. usb_trans.req_state = REQ_STATUS_IDLE;
  227. usb_trans.rx_remaining = 0;
  228. usb_trans.tx_remaining = 0;
  229. usb_trans.sync_errors = 0;
  230. usb_trans.loader_enabled = 1;
  231. }
  232. int main(void)
  233. {
  234. #ifndef NO_DEBUG
  235. uart_init();
  236. stdout = &uart_stdout;
  237. banner();
  238. #endif
  239. shared_memory_init();
  240. system_init();
  241. sram_init();
  242. pwm_init();
  243. irq_init();
  244. boot_startup_rom(50);
  245. globals_init();
  246. pwm_stop();
  247. usbInit();
  248. usb_connect();
  249. sei();
  250. while (1) {
  251. system_set_bus_avr();
  252. system_set_wr_disable();
  253. while (usb_trans.req_state != REQ_STATUS_SNES) {
  254. usbPoll();
  255. #if DO_SHELL
  256. #ifndef NO_DEBUG
  257. shell_run();
  258. #endif
  259. #endif
  260. }
  261. #if DO_SHM
  262. shared_memory_write(SHARED_MEM_TX_CMD_TERMINATE, 0);
  263. #endif
  264. #if DO_SHM_SCRATCHPAD
  265. shared_memory_scratchpad_region_tx_restore();
  266. shared_memory_scratchpad_region_rx_restore();
  267. #endif
  268. #if DO_CRC_CHECK
  269. info_P(PSTR("-->CRC Check\n"));
  270. crc_check_bulk_memory(0x000000,
  271. usb_trans.req_bank_size * usb_trans.req_bank_cnt,
  272. usb_trans.req_bank_size);
  273. #endif
  274. system_set_rom_mode(&usb_trans);
  275. system_set_wr_disable();
  276. system_set_bus_snes();
  277. system_send_snes_reset();
  278. irq_stop();
  279. while ((usb_trans.req_state != REQ_STATUS_AVR)) {
  280. usbPoll();
  281. #if DO_SHELL
  282. #ifndef NO_DEBUG
  283. shell_run();
  284. #endif
  285. #endif
  286. }
  287. info_P(PSTR("-->Switch TO AVR\n"));
  288. shared_memory_init();
  289. irq_init();
  290. if (usb_trans.loader_enabled) {
  291. boot_startup_rom(50);
  292. } else {
  293. system_set_bus_avr();
  294. system_send_snes_reset();
  295. }
  296. globals_init();
  297. }
  298. return 0;
  299. }