Newer Chopstx
[gnuk/neug.git] / src / main.c
index c02a678..f3248c4 100644 (file)
@@ -2,10 +2,10 @@
  * main.c - main routine of neug
  *
  * Main routine:
- * Copyright (C) 2011, 2012 Free Software Initiative of Japan
+ * Copyright (C) 2011, 2012, 2013 Free Software Initiative of Japan
  * Author: NIIBE Yutaka <gniibe@fsij.org>
  *
- * This file is a part of NeuG, a Random Number Generator
+ * This file is a part of NeuG, a True Random Number Generator
  * implementation.
  *
  * NeuG is free software: you can redistribute it and/or modify it
@@ -13,7 +13,7 @@
  * the Free Software Foundation, either version 3 of the License, or
  * (at your option) any later version.
  *
- * Gnuk is distributed in the hope that it will be useful, but WITHOUT
+ * NeuG is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
  * License for more details.
  *
  */
 
+
+#include <stdint.h>
+#include <string.h>
+#include <chopstx.h>
+
 #include "config.h"
-#include "ch.h"
-#include "hal.h"
-#include "board.h"
 #include "neug.h"
 #include "usb_lld.h"
 #include "sys.h"
+#include "stm32f103.h"
+#include "adc.h"
 
-/*
- * We are trying to avoid dependency to C library. 
- * GCC built-in function(s) are declared here.
- */
-extern void *memcpy(void *dest, const void *src, size_t n);
-extern void *memset (void *s, int c, size_t n);
+chopstx_mutex_t usb_mtx;
+chopstx_cond_t cnd_usb;
+
+extern uint8_t __process0_stack_end__;
+static chopstx_t main_thd = (uint32_t)(&__process0_stack_end__ - 60);
 
-\f
 #define ENDP0_RXADDR        (0x40)
 #define ENDP0_TXADDR        (0x80)
 #define ENDP1_TXADDR        (0xc0)
@@ -168,7 +170,7 @@ static const uint8_t vcom_string0[4] = {
 static uint8_t vcom_string3[28] = {
   28,                    /* bLength.                         */
   USB_STRING_DESCRIPTOR_TYPE, /* bDescriptorType.                 */
-  '0', 0,  '.', 0,  '0', 0, '2', 0,    /* Version number of NeuG.          */
+  '0', 0,  '.', 0,  '0', 0, '6', 0,    /* Version number of NeuG.          */
   '-', 0,
   0, 0, 0, 0,  /* Filled by Unique device ID.      */
   0, 0, 0, 0,
@@ -176,41 +178,19 @@ static uint8_t vcom_string3[28] = {
   0, 0, 0, 0,
 };
 
-static const struct Descriptor device_desc = {
-  vcom_device_desc,
-  sizeof (vcom_device_desc)
-};
-
-static const struct Descriptor config_desc = {
-  vcom_configuration_desc,
-  sizeof (vcom_configuration_desc)
-};
-
-static const struct Descriptor string_descs[] = {
-  {vcom_string0, sizeof vcom_string0},
-  {neug_string_vendor, sizeof (neug_string_vendor)},
-  {neug_string_product, sizeof (neug_string_product)},
-  {vcom_string3, sizeof (vcom_string3)},
-  {neug_revision_detail, sizeof (neug_revision_detail)},
-  {neug_config_options, sizeof (neug_config_options)},
-  {sys_version, sizeof (sys_version)},
-};
-
-#define NUM_STRING_DESC (sizeof (string_descs)/sizeof (struct Descriptor))
 
 #define NUM_INTERFACES 2
 
 uint32_t bDeviceState = UNCONNECTED; /* USB device status */
 
-
-static void
-neug_device_reset (void)
+void
+usb_cb_device_reset (void)
 {
   /* Set DEVICE as not configured */
   usb_lld_set_configuration (0);
 
   /* Current Feature initialization */
-  usb_lld_set_feature (config_desc.Descriptor[7]);
+  usb_lld_set_feature (vcom_configuration_desc[7]);
 
   usb_lld_reset ();
 
@@ -218,19 +198,26 @@ neug_device_reset (void)
   usb_lld_setup_endpoint (ENDP0, EP_CONTROL, 0, ENDP0_RXADDR, ENDP0_TXADDR, 64);
 }
 
-#define USB_REGNUAL_MEMINFO    0
-#define USB_REGNUAL_SEND       1
-#define USB_REGNUAL_RESULT     2
-#define USB_REGNUAL_FLASH      3
-#define USB_REGNUAL_PROTECT    4
-#define USB_REGNUAL_FINISH     5
+extern uint8_t _regnual_start, __heap_end__;
+
+static const uint8_t *const mem_info[] = { &_regnual_start,  &__heap_end__, };
 
-static uint8_t mem[256];
-static uint32_t result;
+/* USB vendor requests to control pipe */
+#define USB_FSIJ_MEMINFO         0
+#define USB_FSIJ_DOWNLOAD        1
+#define USB_FSIJ_EXEC            2
+#define USB_NEUG_GET_INFO      254
+#define USB_NEUG_EXIT          255 /* Ask to exit and to receive reGNUal */
 
-extern uint8_t __flash_start__,  __flash_end__;
-static const uint8_t *const mem_info[] = { &__flash_start__,  &__flash_end__ };
+enum {
+  FSIJ_DEVICE_RUNNING = 0,
+  FSIJ_DEVICE_EXITED,
+  FSIJ_DEVICE_EXEC_REQUESTED,
+  /**/
+  FSIJ_DEVICE_NEUG_EXIT_REQUESTED = 255
+}; 
 
+static uint8_t fsij_device_state = FSIJ_DEVICE_RUNNING;
 
 static uint32_t rbit (uint32_t v)
 {
@@ -240,49 +227,49 @@ static uint32_t rbit (uint32_t v)
   return r;
 }
 
-static uint32_t fetch (int i)
+/* After calling this function, CRC module remain enabled.  */
+static int download_check_crc32 (const uint32_t *end_p)
 {
-  uint32_t v;
-
-  v = *(uint32_t *)(&mem[i*4]);
-  return rbit (v);
-}
-
-static uint32_t calc_crc32 (void)
-{
-  int i;
+  uint32_t crc32 = *end_p;
+  const uint32_t *p;
 
+  RCC->AHBENR |= RCC_AHBENR_CRCEN;
   CRC->CR = CRC_CR_RESET;
 
-  for (i = 0; i < 256/4; i++)
-    CRC->DR = fetch (i);
+  for (p = (const uint32_t *)&_regnual_start; p < end_p; p++)
+    CRC->DR = rbit (*p);
 
-  return rbit (CRC->DR);
-}
+  if ((rbit (CRC->DR) ^ crc32) == 0xffffffff)
+    return USB_SUCCESS;
 
+  return USB_UNSUPPORT;
+}
 
-static void neug_ctrl_write_finish (uint8_t req, uint8_t req_no,
-                                   uint16_t value, uint16_t index,
-                                   uint16_t len)
+void
+usb_cb_ctrl_write_finish (uint8_t req, uint8_t req_no, uint16_t value,
+                         uint16_t index, uint16_t len)
 {
   uint8_t type_rcp = req & (REQUEST_TYPE|RECIPIENT);
 
-  if (type_rcp == (VENDOR_REQUEST | DEVICE_RECIPIENT) && USB_SETUP_SET (req))
+  if (type_rcp == (VENDOR_REQUEST | DEVICE_RECIPIENT)
+      && USB_SETUP_SET (req) && len == 0)
     {
-      if (req_no == USB_REGNUAL_SEND && value == 0)
-       result = calc_crc32 ();
-      else if (req_no == USB_REGNUAL_FLASH && len == 0 && index == 0)
+      if (req_no == USB_FSIJ_EXEC)
        {
-         uint32_t dst_addr = (0x08000000 + value * 0x100);
+         if (fsij_device_state != FSIJ_DEVICE_EXITED)
+           return;
 
-         result = flash_write (dst_addr, mem, 256);
+         (void)value; (void)index;
+         usb_lld_prepare_shutdown (); /* No further USB communication */
+         fsij_device_state = FSIJ_DEVICE_EXEC_REQUESTED;
+       }
+      else if (req_no == USB_NEUG_EXIT)
+       {
+         /* Force exit from the main loop.  */
+         chopstx_mutex_lock (&usb_mtx);
+         chopstx_cond_signal (&cnd_usb);
+         chopstx_mutex_unlock (&usb_mtx);
        }
-      else if (req_no == USB_REGNUAL_PROTECT && len == 0
-              && value == 0 && index == 0)
-       result = flash_protect ();
-      else if (req_no == USB_REGNUAL_FINISH && len == 0
-              && value == 0 && index == 0)
-       nvic_system_reset ();
     }
 }
 
@@ -323,15 +310,25 @@ vcom_port_data_setup (uint8_t req, uint8_t req_no, uint16_t value)
        }
       else if (req_no == USB_CDC_REQ_SET_CONTROL_LINE_STATE)
        {
+         uint8_t connected_saved = connected;
+
          if (value != 0)
            {
-             if ((connected & 1) == 0)
+             if (connected == 0)
                /* It's Open call */
                connected++;
            }
          else
-           /* Close call */
-           connected++;
+           {
+             if (connected)
+               /* Close call */
+               connected = 0;
+           }
+
+         chopstx_mutex_lock (&usb_mtx);
+         if (connected != connected_saved)
+           chopstx_cond_signal (&cnd_usb);
+         chopstx_mutex_unlock (&usb_mtx);
 
          return USB_SUCCESS;
        }
@@ -340,8 +337,8 @@ vcom_port_data_setup (uint8_t req, uint8_t req_no, uint16_t value)
   return USB_UNSUPPORT;
 }
 
-static int
-neug_setup (uint8_t req, uint8_t req_no,
+int
+usb_cb_setup (uint8_t req, uint8_t req_no,
               uint16_t value, uint16_t index, uint16_t len)
 {
   uint8_t type_rcp = req & (REQUEST_TYPE|RECIPIENT);
@@ -350,43 +347,73 @@ neug_setup (uint8_t req, uint8_t req_no,
     {
       if (USB_SETUP_GET (req))
        {
-         if (req_no == USB_REGNUAL_MEMINFO)
+         if (req_no == USB_FSIJ_MEMINFO)
            {
              usb_lld_set_data_to_send (mem_info, sizeof (mem_info));
              return USB_SUCCESS;
            }
-         else if (req_no == USB_REGNUAL_RESULT)
+         else if (req_no == USB_NEUG_GET_INFO)
            {
-             usb_lld_set_data_to_send (&result, sizeof (uint32_t));
+             if (index == 0)
+               usb_lld_set_data_to_send (&neug_mode, sizeof (uint8_t));
+             else if (index == 1)
+               usb_lld_set_data_to_send (&neug_err_cnt, sizeof (uint16_t));
+             else if (index == 2)
+               usb_lld_set_data_to_send (&neug_err_cnt_rc, sizeof (uint16_t));
+             else if (index == 3)
+               usb_lld_set_data_to_send (&neug_err_cnt_p64, sizeof (uint16_t));
+             else if (index == 4)
+               usb_lld_set_data_to_send (&neug_err_cnt_p4k, sizeof (uint16_t));
+             else if (index == 5)
+               usb_lld_set_data_to_send (&neug_rc_max, sizeof (uint16_t));
+             else if (index == 6)
+               usb_lld_set_data_to_send (&neug_p64_max, sizeof (uint16_t));
+             else if (index == 7)
+               usb_lld_set_data_to_send (&neug_p4k_max, sizeof (uint16_t));
+             else
+               return USB_UNSUPPORT;
+
              return USB_SUCCESS;
            }
        }
       else /* SETUP_SET */
        {
-         if (req_no == USB_REGNUAL_SEND)
+         uint8_t *addr = (uint8_t *)(0x20000000 + value * 0x100 + index);
+
+         if (req_no == USB_FSIJ_DOWNLOAD)
            {
-             if (value != 0 || index + len > 256)
+             if (fsij_device_state != FSIJ_DEVICE_EXITED)
+               return USB_UNSUPPORT;
+
+             if (addr < &_regnual_start || addr + len > &__heap_end__)
                return USB_UNSUPPORT;
 
              if (index + len < 256)
-               memset (mem + index + len, 0xff, 256 - (index + len));
+               memset (addr + index + len, 0, 256 - (index + len));
 
-             usb_lld_set_data_to_recv (mem + index, len);
+             usb_lld_set_data_to_recv (addr, len);
              return USB_SUCCESS;
            }
-         else if (req_no == USB_REGNUAL_FLASH && len == 0 && index == 0)
+         else if (req_no == USB_FSIJ_EXEC && len == 0)
+           {
+             if (fsij_device_state != FSIJ_DEVICE_EXITED)
+               return USB_UNSUPPORT;
+
+             if (((uint32_t)addr & 0x03))
+               return USB_UNSUPPORT;
+
+             return download_check_crc32 ((uint32_t *)addr);
+           }
+         else if (req_no == USB_NEUG_EXIT && len == 0)
            {
-             uint32_t dst_addr = (0x08000000 + value * 0x100);
+             if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+               return USB_UNSUPPORT;
+
+             fsij_device_state = FSIJ_DEVICE_NEUG_EXIT_REQUESTED;
+             chopstx_wakeup_usec_wait (main_thd);
 
-             if (dst_addr + 256 <= (uint32_t)&__flash_end__)
-               return USB_SUCCESS;
+             return USB_SUCCESS;
            }
-         else if (req_no == USB_REGNUAL_PROTECT && len == 0
-                  && value == 0 && index == 0)
-           return USB_SUCCESS;
-         else if (req_no == USB_REGNUAL_FINISH && len == 0
-                  && value == 0 && index == 0)
-           return USB_SUCCESS;
        }
     }
   else if (type_rcp == (CLASS_REQUEST | INTERFACE_RECIPIENT))
@@ -396,32 +423,63 @@ neug_setup (uint8_t req, uint8_t req_no,
   return USB_UNSUPPORT;
 }
 
-static int
-neug_get_descriptor (uint8_t desc_type, uint16_t index, uint16_t value)
+int
+usb_cb_get_descriptor (uint8_t desc_type, uint16_t index, uint16_t value)
 {
   (void)index;
   if (desc_type == DEVICE_DESCRIPTOR)
     {
-      usb_lld_set_data_to_send (device_desc.Descriptor,
-                               device_desc.Descriptor_Size);
+      usb_lld_set_data_to_send (vcom_device_desc, sizeof (vcom_device_desc));
       return USB_SUCCESS;
     }
   else if (desc_type == CONFIG_DESCRIPTOR)
     {
-      usb_lld_set_data_to_send (config_desc.Descriptor,
-                               config_desc.Descriptor_Size);
+      usb_lld_set_data_to_send (vcom_configuration_desc,
+                               sizeof (vcom_configuration_desc));
       return USB_SUCCESS;
     }
   else if (desc_type == STRING_DESCRIPTOR)
     {
       uint8_t desc_index = value & 0xff;
+      const uint8_t *str;
+      int size;
 
-      if (desc_index < NUM_STRING_DESC)
+      switch (desc_index)
        {
-         usb_lld_set_data_to_send (string_descs[desc_index].Descriptor,
-                                   string_descs[desc_index].Descriptor_Size);
-         return USB_SUCCESS;
+       case 0:
+         str = vcom_string0;
+         size = sizeof (vcom_string0);
+         break;
+       case 1:
+         str = neug_string_vendor;
+         size = sizeof (neug_string_vendor);
+         break;
+       case 2:
+         str = neug_string_product;
+         size = sizeof (neug_string_product);
+         break;
+       case 3:
+         str = vcom_string3;
+         size = sizeof (vcom_string3);
+         break;
+       case 4:
+         str = neug_revision_detail;
+         size = sizeof (neug_revision_detail);
+         break;
+       case 5:
+         str = neug_config_options;
+         size = sizeof (neug_config_options);
+         break;
+       case 6:
+         str = sys_version;
+         size = sizeof (sys_version);
+         break;
+       default:
+         return USB_UNSUPPORT;
        }
+
+      usb_lld_set_data_to_send (str, size);
+      return USB_SUCCESS;
     }
 
   return USB_UNSUPPORT;
@@ -452,7 +510,7 @@ neug_setup_endpoints_for_interface (uint16_t interface, int stop)
     }
 }
 
-static int neug_usb_event (uint8_t event_type, uint16_t value)
+int usb_cb_handle_event (uint8_t event_type, uint16_t value)
 {
   int i;
   uint8_t current_conf;
@@ -496,7 +554,7 @@ static int neug_usb_event (uint8_t event_type, uint16_t value)
 }
 
 
-static int neug_interface (uint8_t cmd, uint16_t interface, uint16_t alt)
+int usb_cb_interface (uint8_t cmd, uint16_t interface, uint16_t alt)
 {
   static uint8_t zero = 0;
 
@@ -524,15 +582,29 @@ static int neug_interface (uint8_t cmd, uint16_t interface, uint16_t alt)
     }
 }
 
-const struct usb_device_method Device_Method = {
-  neug_device_reset,
-  neug_ctrl_write_finish,
-  neug_setup,
-  neug_get_descriptor,
-  neug_usb_event,
-  neug_interface,
-};
+#define INTR_REQ_USB 20
+#define PRIO_USB 4
+
+static void *
+usb_intr (void *arg)
+{
+  chopstx_intr_t interrupt;
+
+  (void)arg;
+  usb_lld_init (vcom_configuration_desc[7]);
+  chopstx_claim_irq (&interrupt, INTR_REQ_USB);
+  usb_interrupt_handler ();
+
+  while (1)
+    {
+      chopstx_intr_wait (&interrupt);
+
+      /* Process interrupt. */
+      usb_interrupt_handler ();
+    }
 
+  return NULL;
+}
 
 
 static void fill_serial_no_by_unique_id (void)
@@ -556,29 +628,12 @@ static void fill_serial_no_by_unique_id (void)
     }
 }
 \f
-static Thread *main_thread = NULL;
-
-CH_IRQ_HANDLER (Vector90)
-{
-  CH_IRQ_PROLOGUE();
-  chSysLockFromIsr();
-
-  usb_interrupt_handler ();
-
-  chSysUnlockFromIsr();
-  CH_IRQ_EPILOGUE();
-}
-
 void
 EP1_IN_Callback (void)
 {
-  if (main_thread != NULL)
-    {
-      chSysLockFromIsr ();
-      main_thread->p_u.rdymsg = RDY_OK;
-      chSchReadyI (main_thread);
-      chSysUnlockFromIsr ();
-    }
+  chopstx_mutex_lock (&usb_mtx);
+  chopstx_cond_signal (&cnd_usb);
+  chopstx_mutex_unlock (&usb_mtx);
 }
 
 void
@@ -592,41 +647,124 @@ EP3_OUT_Callback (void)
   usb_lld_rx_enable (ENDP3);
 }
 \f
-static WORKING_AREA(wa_led, 64);
+typedef uint32_t eventmask_t;
+#define ALL_EVENTS (~0)
+
+struct event_flag {
+  chopstx_mutex_t mutex;
+  chopstx_cond_t cond;
+  eventmask_t flag;
+};
+
+static void event_flag_init (struct event_flag *ev)
+{
+  ev->flag = 0;
+  chopstx_mutex_init (&ev->mutex);
+  chopstx_cond_init (&ev->cond);
+}
+
+
+static eventmask_t event_flag_waitone (struct event_flag *ev, eventmask_t m)
+{
+  int n;
+
+  chopstx_mutex_lock (&ev->mutex);
+  while (!(ev->flag & m))
+    chopstx_cond_wait (&ev->cond, &ev->mutex);
+
+  n = __builtin_ffs ((ev->flag & m));
+  ev->flag &= ~(1 << (n - 1));
+  chopstx_mutex_unlock (&ev->mutex);
+
+  return (1 << (n - 1));
+}
+
+static void event_flag_signal (struct event_flag *ev, eventmask_t m)
+{
+  chopstx_mutex_lock (&ev->mutex);
+  ev->flag |= m;
+  chopstx_cond_signal (&ev->cond);
+  chopstx_mutex_unlock (&ev->mutex);
+}
 
-#define LED_ONESHOT_SHORT ((eventmask_t)1)
-#define LED_ONESHOT_LONG  ((eventmask_t)2)
-static Thread *led_thread;
+extern uint8_t __process1_stack_base__, __process1_stack_size__;
+extern uint8_t __process3_stack_base__, __process3_stack_size__;
+
+const uint32_t __stackaddr_led = (uint32_t)&__process1_stack_base__;
+const size_t __stacksize_led = (size_t)&__process1_stack_size__;
+const uint32_t __stackaddr_usb = (uint32_t)&__process3_stack_base__;
+const size_t __stacksize_usb = (size_t)&__process3_stack_size__;
+
+
+#define PRIO_LED 1
+struct event_flag led_event;
+
+#define LED_ONESHOT_SHORT      ((eventmask_t)1)
+#define LED_TWOSHOTS           ((eventmask_t)2)
+#define LED_ONESHOT_LONG       ((eventmask_t)4)
 
 /*
  * LED blinker: When notified, let LED emit for 100ms.
  */
-static msg_t led_blinker (void *arg)
+static void *led_blinker (void *arg)
 {
   (void)arg;
 
-  led_thread = chThdSelf ();
   set_led (0);
 
   while (1)
     {
       eventmask_t m;
 
-      m = chEvtWaitOne (ALL_EVENTS);
+      m = event_flag_waitone (&led_event, ALL_EVENTS);
+      if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+       break;
+
       set_led (1);
       if (m == LED_ONESHOT_SHORT)
-       chThdSleep (MS2ST (100));
+       chopstx_usec_wait (100*1000);
+      else if (m == LED_TWOSHOTS)
+       {
+         chopstx_usec_wait (50*1000);
+         set_led (0);
+         chopstx_usec_wait (50*1000);
+         set_led (1);
+         chopstx_usec_wait (50*1000);
+       }
       else
-       chThdSleep (MS2ST (250));
+       chopstx_usec_wait (250*1000);
       set_led (0);
     }
 
   return 0;
 }
 \f
-#define RANDOM_BYTES_LENGTH 60
+#define RANDOM_BYTES_LENGTH 64
 static uint32_t random_word[RANDOM_BYTES_LENGTH/sizeof (uint32_t)];
 
+static void copy_to_tx (uint32_t v, int i)
+{
+  usb_lld_txcpy (&v, ENDP1, i * 4, 4);
+}
+
+/*
+ * In Gnuk 1.0.[12], reGNUal was not relocatable.
+ * Now, it's relocatable, but we need to calculate its entry address
+ * based on it's pre-defined address.
+ */
+#define REGNUAL_START_ADDRESS_COMPATIBLE 0x20001400
+static uint32_t
+calculate_regnual_entry_address (const uint8_t *addr)
+{
+  const uint8_t *p = addr + 4;
+  uint32_t v = p[0] + (p[1] << 8) + (p[2] << 16) + (p[3] << 24);
+
+  v -= REGNUAL_START_ADDRESS_COMPATIBLE;
+  v += (uint32_t)addr;
+  return v;
+}
+
+
 /*
  * Entry point.
  *
@@ -635,71 +773,166 @@ static uint32_t random_word[RANDOM_BYTES_LENGTH/sizeof (uint32_t)];
 int
 main (int argc, char **argv)
 {
+  uint32_t entry;
+  chopstx_t led_thread, usb_thd;
+
   (void)argc;
   (void)argv;
 
   fill_serial_no_by_unique_id ();
 
-  halInit();
-  chSysInit();
+  adc_init ();
 
-  main_thread = chThdSelf ();
+  event_flag_init (&led_event);
+  
+  led_thread = chopstx_create (PRIO_LED, __stackaddr_led, __stacksize_led,
+                              led_blinker, NULL);
 
-  usb_lld_init (config_desc.Descriptor[7]);
+  chopstx_mutex_init (&usb_mtx);
+  chopstx_cond_init (&cnd_usb);
 
-  chThdCreateStatic (wa_led, sizeof (wa_led), NORMALPRIO, led_blinker, NULL);
+  usb_thd = chopstx_create (PRIO_USB, __stackaddr_usb, __stacksize_usb,
+                           usb_intr, NULL);
 
   neug_init (random_word, RANDOM_BYTES_LENGTH/sizeof (uint32_t));
 
   while (1)
     {
       unsigned int count = 0;
+      int last_was_fullsizepacket = 0;
+
+      if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+       break;
 
-      connected = 0;
       while (count < NEUG_PRE_LOOP || bDeviceState != CONFIGURED)
        {
-         (void)neug_get (NEUG_KICK_FILLING);
-         if ((count & 0x000f) == 0)
-           chEvtSignalFlags (led_thread, LED_ONESHOT_SHORT);
-         chThdSleep (MS2ST (25));
+         if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+           break;
+
+         neug_wait_full ();
+         neug_flush ();
+
+         if ((count & 0x0007) == 0)
+           event_flag_signal (&led_event, LED_ONESHOT_SHORT);
+         chopstx_usec_wait (25*1000);
          count++;
        }
 
     waiting_connection:
-      while ((connected & 1) == 0)
+      while (connected == 0)
        {
+         if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+           break;
+
          neug_flush ();
-         chEvtSignalFlags (led_thread, LED_ONESHOT_LONG);
-         chThdSleep (MS2ST (2500));
+         event_flag_signal (&led_event, LED_TWOSHOTS);
+         chopstx_usec_wait (5000*1000);
        }
 
+      if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+       break;
+
       /* The connection opened.  */
       count = 0;
+      /*
+       * No parity is standard.  It means to provide conditioned output.
+       * When parity enabled, it means to provide raw output.
+       */
+      neug_mode_select (line_coding.paritytype); /* 0: None, 1: Odd, 2: Even */
 
       while (1)
        {
-         if ((connected & 1) == 0)
-           goto waiting_connection;
+         int i;
+
+         if (fsij_device_state != FSIJ_DEVICE_RUNNING)
+           break;
 
          if (bDeviceState != CONFIGURED)
            break;
 
-         neug_wait_full ();
+         if ((count & 0x03ff) == 0)
+           event_flag_signal (&led_event, LED_ONESHOT_SHORT);
 
-         if ((count & 0x00ff) == 0)
-           chEvtSignalFlags (led_thread, LED_ONESHOT_SHORT);
+         i = neug_consume_random (copy_to_tx);
 
-         usb_lld_txcpy (random_word, ENDP1, 0, RANDOM_BYTES_LENGTH);
-         neug_flush ();
+         if (i == 0 && !last_was_fullsizepacket)
+           {    /* Only send ZLP when the last packet was fullsize.  */
+             neug_wait_full ();
+             continue;
+           }
 
-         chSysLock ();
-         usb_lld_tx_enable (ENDP1, RANDOM_BYTES_LENGTH);
-         chSchGoSleepS (THD_STATE_SUSPENDED);
-         chSysUnlock();
+         if (i == 64/4)
+           last_was_fullsizepacket = 1;
+         else
+           last_was_fullsizepacket = 0;
+
+         chopstx_mutex_lock (&usb_mtx);
+         if (connected == 0)
+           {
+             chopstx_mutex_unlock (&usb_mtx);
+             goto waiting_connection;
+           }
+         else
+           {
+             usb_lld_tx_enable (ENDP1, i * 4);
+             chopstx_cond_wait (&cnd_usb, &usb_mtx);
+           }
+         chopstx_mutex_unlock (&usb_mtx);
 
          count++;
        }
     }
 
+  chopstx_cancel (led_thread);
+  chopstx_join (led_thread, NULL);
+
+  /*
+   * We come here, because of FSIJ_DEVICE_NEUG_EXIT_REQUESTED.
+   */
+  neug_fini ();
+
+  fsij_device_state = FSIJ_DEVICE_EXITED;
+
+  while (fsij_device_state == FSIJ_DEVICE_EXITED)
+    chopstx_usec_wait (500*1000);
+
+  flash_unlock ();             /* Flash unlock should be done here */
+  set_led (1);
+  usb_lld_shutdown ();
+
+  /* Finish application.  */
+  chopstx_cancel (usb_thd);
+  chopstx_join (usb_thd, NULL);
+
+  /* Set vector */
+  SCB->VTOR = (uint32_t)&_regnual_start;
+  entry = calculate_regnual_entry_address (&_regnual_start);
+#ifdef DFU_SUPPORT
+#define FLASH_SYS_START_ADDR 0x08000000
+#define FLASH_SYS_END_ADDR (0x08000000+0x1000)
+  {
+    extern uint8_t _sys;
+    uint32_t addr;
+    handler *new_vector = (handler *)FLASH_SYS_START_ADDR;
+    void (*func) (void (*)(void)) = (void (*)(void (*)(void)))new_vector[10];
+
+    /* Kill DFU */
+    for (addr = FLASH_SYS_START_ADDR; addr < FLASH_SYS_END_ADDR;
+        addr += FLASH_PAGE_SIZE)
+      flash_erase_page (addr);
+
+    /* copy system service routines */
+    flash_write (FLASH_SYS_START_ADDR, &_sys, 0x1000);
+
+    /* Leave NeuG to exec reGNUal */
+    (*func) ((void (*)(void))entry);
+    for (;;);
+  }
+#else
+  /* Leave NeuG to exec reGNUal */
+  flash_erase_all_and_exec ((void (*)(void))entry);
+#endif
+
+  /* Never reached */
   return 0;
 }