change feedback
[gnuk/neug.git] / src / main.c
1 /*
2  * main.c - main routine of neug
3  *
4  * Main routine:
5  * Copyright (C) 2011, 2012 Free Software Initiative of Japan
6  * Author: NIIBE Yutaka <gniibe@fsij.org>
7  *
8  * This file is a part of NeuG, a Random Number Generator
9  * implementation.
10  *
11  * NeuG is free software: you can redistribute it and/or modify it
12  * under the terms of the GNU General Public License as published by
13  * the Free Software Foundation, either version 3 of the License, or
14  * (at your option) any later version.
15  *
16  * Gnuk is distributed in the hope that it will be useful, but WITHOUT
17  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
18  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
19  * License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
23  *
24  */
25
26 #include "config.h"
27 #include "ch.h"
28 #include "hal.h"
29 #include "board.h"
30 #include "neug.h"
31 #include "usb_lld.h"
32 #include "sys.h"
33
34 /*
35  * We are trying to avoid dependency to C library. 
36  * GCC built-in function(s) are declared here.
37  */
38 extern void *memcpy(void *dest, const void *src, size_t n);
39 extern void *memset (void *s, int c, size_t n);
40
41 static Thread *main_thread = NULL;
42
43 \f
44 #define ENDP0_RXADDR        (0x40)
45 #define ENDP0_TXADDR        (0x80)
46 #define ENDP1_TXADDR        (0xc0)
47 #define ENDP2_TXADDR        (0x100)
48 #define ENDP3_RXADDR        (0x140)
49
50 #define USB_CDC_REQ_SET_LINE_CODING             0x20
51 #define USB_CDC_REQ_GET_LINE_CODING             0x21
52 #define USB_CDC_REQ_SET_CONTROL_LINE_STATE      0x22
53 #define USB_CDC_REQ_SEND_BREAK                  0x23
54
55 /* USB Device Descriptor */
56 static const uint8_t vcom_device_desc[18] = {
57   18,   /* bLength */
58   USB_DEVICE_DESCRIPTOR_TYPE,   /* bDescriptorType */
59   0x10, 0x01,                   /* bcdUSB = 1.1 */
60   0x02,                         /* bDeviceClass (CDC).              */
61   0x00,                         /* bDeviceSubClass.                 */
62   0x00,                         /* bDeviceProtocol.                 */
63   0x40,                         /* bMaxPacketSize.                  */
64 #include "usb-vid-pid-ver.c.inc"
65   1,                            /* iManufacturer.                   */
66   2,                            /* iProduct.                        */
67   3,                            /* iSerialNumber.                   */
68   1                             /* bNumConfigurations.              */
69 };
70
71 /* Configuration Descriptor tree for a CDC.*/
72 static const uint8_t vcom_configuration_desc[67] = {
73   9,
74   USB_CONFIGURATION_DESCRIPTOR_TYPE, /* bDescriptorType: Configuration */
75   /* Configuration Descriptor.*/
76   67, 0x00,                     /* wTotalLength.                    */
77   0x02,                         /* bNumInterfaces.                  */
78   0x01,                         /* bConfigurationValue.             */
79   0,                            /* iConfiguration.                  */
80   0x80,                         /* bmAttributes (bus powered).      */
81   50,                           /* bMaxPower (100mA).               */
82   /* Interface Descriptor.*/
83   9,
84   USB_INTERFACE_DESCRIPTOR_TYPE,
85   0x00,            /* bInterfaceNumber.                */
86   0x00,            /* bAlternateSetting.               */
87   0x01,            /* bNumEndpoints.                   */
88   0x02,            /* bInterfaceClass (Communications Interface Class,
89                       CDC section 4.2).  */
90   0x02,            /* bInterfaceSubClass (Abstract Control Model, CDC
91                       section 4.3).  */
92   0x01,            /* bInterfaceProtocol (AT commands, CDC section
93                       4.4).  */
94   0,               /* iInterface.                      */
95   /* Header Functional Descriptor (CDC section 5.2.3).*/
96   5,          /* bLength.                         */
97   0x24,       /* bDescriptorType (CS_INTERFACE).  */
98   0x00,       /* bDescriptorSubtype (Header Functional Descriptor). */
99   0x10, 0x01, /* bcdCDC.                          */
100   /* Call Management Functional Descriptor. */
101   5,            /* bFunctionLength.                 */
102   0x24,         /* bDescriptorType (CS_INTERFACE).  */
103   0x01,         /* bDescriptorSubtype (Call Management Functional
104                    Descriptor). */
105   0x03,         /* bmCapabilities (D0+D1).          */
106   0x01,         /* bDataInterface.                  */
107   /* ACM Functional Descriptor.*/
108   4,            /* bFunctionLength.                 */
109   0x24,         /* bDescriptorType (CS_INTERFACE).  */
110   0x02,         /* bDescriptorSubtype (Abstract Control Management
111                    Descriptor).  */
112   0x02,         /* bmCapabilities.                  */
113   /* Union Functional Descriptor.*/
114   5,            /* bFunctionLength.                 */
115   0x24,         /* bDescriptorType (CS_INTERFACE).  */
116   0x06,         /* bDescriptorSubtype (Union Functional
117                    Descriptor).  */
118   0x00,         /* bMasterInterface (Communication Class
119                    Interface).  */
120   0x01,         /* bSlaveInterface0 (Data Class Interface).  */
121   /* Endpoint 2 Descriptor.*/
122   7,
123   USB_ENDPOINT_DESCRIPTOR_TYPE,
124   ENDP2|0x80,    /* bEndpointAddress.    */
125   0x03,          /* bmAttributes (Interrupt).        */
126   0x08, 0x00,    /* wMaxPacketSize.                  */
127   0xFF,          /* bInterval.                       */
128   /* Interface Descriptor.*/
129   9,
130   USB_INTERFACE_DESCRIPTOR_TYPE, /* bDescriptorType: */
131   0x01,          /* bInterfaceNumber.                */
132   0x00,          /* bAlternateSetting.               */
133   0x02,          /* bNumEndpoints.                   */
134   0x0A,          /* bInterfaceClass (Data Class Interface, CDC section 4.5). */
135   0x00,          /* bInterfaceSubClass (CDC section 4.6). */
136   0x00,          /* bInterfaceProtocol (CDC section 4.7). */
137   0x00,          /* iInterface.                      */
138   /* Endpoint 3 Descriptor.*/
139   7,
140   USB_ENDPOINT_DESCRIPTOR_TYPE, /* bDescriptorType: Endpoint */
141   ENDP3,    /* bEndpointAddress. */
142   0x02,                         /* bmAttributes (Bulk).             */
143   0x40, 0x00,                   /* wMaxPacketSize.                  */
144   0x00,                         /* bInterval.                       */
145   /* Endpoint 1 Descriptor.*/
146   7,
147   USB_ENDPOINT_DESCRIPTOR_TYPE, /* bDescriptorType: Endpoint */
148   ENDP1|0x80,                   /* bEndpointAddress. */
149   0x02,                         /* bmAttributes (Bulk).             */
150   0x40, 0x00,                   /* wMaxPacketSize.                  */
151   0x00                          /* bInterval.                       */
152 };
153
154
155 /*
156  * U.S. English language identifier.
157  */
158 static const uint8_t vcom_string0[4] = {
159   4,                            /* bLength */
160   USB_STRING_DESCRIPTOR_TYPE,
161   0x09, 0x04                    /* LangID = 0x0409: US-English */
162 };
163
164 #define USB_STRINGS_FOR_NEUG 1
165 #include "usb-strings.c.inc"
166
167 /*
168  * Serial Number string.  NOTE: This does not have CONST qualifier.
169  */
170 static uint8_t vcom_string3[28] = {
171   28,                    /* bLength.                         */
172   USB_STRING_DESCRIPTOR_TYPE, /* bDescriptorType.                 */
173   '0', 0,  '.', 0,  '0', 0, '2', 0,    /* Version number of NeuG.          */
174   '-', 0,
175   0, 0, 0, 0,   /* Filled by Unique device ID.      */
176   0, 0, 0, 0,
177   0, 0, 0, 0,
178   0, 0, 0, 0,
179 };
180
181 static const struct Descriptor device_desc = {
182   vcom_device_desc,
183   sizeof (vcom_device_desc)
184 };
185
186 static const struct Descriptor config_desc = {
187   vcom_configuration_desc,
188   sizeof (vcom_configuration_desc)
189 };
190
191 static const struct Descriptor string_descs[] = {
192   {vcom_string0, sizeof vcom_string0},
193   {neug_string_vendor, sizeof (neug_string_vendor)},
194   {neug_string_product, sizeof (neug_string_product)},
195   {vcom_string3, sizeof (vcom_string3)},
196   {neug_revision_detail, sizeof (neug_revision_detail)},
197   {neug_config_options, sizeof (neug_config_options)},
198   {sys_version, sizeof (sys_version)},
199 };
200
201 #define NUM_STRING_DESC (sizeof (string_descs)/sizeof (struct Descriptor))
202
203 #define NUM_INTERFACES 2
204
205 uint32_t bDeviceState = UNCONNECTED; /* USB device status */
206
207 #define NEUG_WAIT_FOR_TX_READY 1
208 static uint8_t neug_state;
209
210 static void
211 neug_device_reset (void)
212 {
213   /* Set DEVICE as not configured */
214   usb_lld_set_configuration (0);
215
216   /* Current Feature initialization */
217   usb_lld_set_feature (config_desc.Descriptor[7]);
218
219   usb_lld_reset ();
220
221   /* Initialize Endpoint 0 */
222   usb_lld_setup_endpoint (ENDP0, EP_CONTROL, 0, ENDP0_RXADDR, ENDP0_TXADDR, 64);
223 }
224
225 extern uint8_t _regnual_start, __heap_end__;
226
227 static const uint8_t *const mem_info[] = { &_regnual_start,  &__heap_end__, };
228
229 /* USB vendor requests to control pipe */
230 #define USB_FSIJ_MEMINFO          0
231 #define USB_FSIJ_DOWNLOAD         1
232 #define USB_FSIJ_EXEC             2
233 #define USB_NEUG_EXIT           255 /* Ask to exit and to receive reGNUal */
234
235 enum {
236   FSIJ_DEVICE_RUNNING = 0,
237   FSIJ_DEVICE_EXITED,
238   FSIJ_DEVICE_EXEC_REQUESTED,
239   /**/
240   FSIJ_DEVICE_NEUG_EXIT_REQUESTED = 255
241 }; 
242
243 static uint8_t fsij_device_state = FSIJ_DEVICE_RUNNING;
244
245 static uint32_t rbit (uint32_t v)
246 {
247   uint32_t r;
248
249   asm ("rbit    %0, %1" : "=r" (r) : "r" (v));
250   return r;
251 }
252
253 /* After calling this function, CRC module remain enabled.  */
254 static int download_check_crc32 (const uint32_t *end_p)
255 {
256   uint32_t crc32 = *end_p;
257   const uint32_t *p;
258
259   RCC->AHBENR |= RCC_AHBENR_CRCEN;
260   CRC->CR = CRC_CR_RESET;
261
262   for (p = (const uint32_t *)&_regnual_start; p < end_p; p++)
263     CRC->DR = rbit (*p);
264
265   if ((rbit (CRC->DR) ^ crc32) == 0xffffffff)
266     return USB_SUCCESS;
267
268   return USB_UNSUPPORT;
269 }
270
271 static void neug_ctrl_write_finish (uint8_t req, uint8_t req_no,
272                                     uint16_t value, uint16_t index,
273                                     uint16_t len)
274 {
275   uint8_t type_rcp = req & (REQUEST_TYPE|RECIPIENT);
276
277   if (type_rcp == (VENDOR_REQUEST | DEVICE_RECIPIENT)
278       && USB_SETUP_SET (req) && len == 0)
279     {
280       if (req_no == USB_FSIJ_EXEC)
281         {
282           if (fsij_device_state != FSIJ_DEVICE_EXITED)
283             return;
284
285           (void)value; (void)index;
286           usb_lld_prepare_shutdown (); /* No further USB communication */
287           fsij_device_state = FSIJ_DEVICE_EXEC_REQUESTED;
288         }
289       else if (req_no == USB_NEUG_EXIT)
290         {
291           if (neug_state == NEUG_WAIT_FOR_TX_READY)
292             {
293               chSysLockFromIsr ();
294               main_thread->p_u.rdymsg = RDY_OK;
295               chSchReadyI (main_thread);
296               chSysUnlockFromIsr ();
297             }
298           else
299             chEvtSignalFlagsI (main_thread, 1);
300         }
301     }
302 }
303
304 struct line_coding
305 {
306   uint32_t bitrate;
307   uint8_t format;
308   uint8_t paritytype;
309   uint8_t datatype;
310 };
311
312 static struct line_coding line_coding = {
313   115200, /* baud rate: 115200    */
314   0x00,   /* stop bits: 1         */
315   0x00,   /* parity:    none      */
316   0x08    /* bits:      8         */
317 };
318
319 static uint8_t connected;
320
321 static int
322 vcom_port_data_setup (uint8_t req, uint8_t req_no, uint16_t value)
323 {
324   if (USB_SETUP_GET (req))
325     {
326       if (req_no == USB_CDC_REQ_GET_LINE_CODING)
327         {
328           usb_lld_set_data_to_send (&line_coding, sizeof(line_coding));
329           return USB_SUCCESS;
330         }
331     }
332   else  /* USB_SETUP_SET (req) */
333     {
334       if (req_no == USB_CDC_REQ_SET_LINE_CODING)
335         {
336           usb_lld_set_data_to_recv (&line_coding, sizeof(line_coding));
337           return USB_SUCCESS;
338         }
339       else if (req_no == USB_CDC_REQ_SET_CONTROL_LINE_STATE)
340         {
341           if (value != 0)
342             {
343               if (connected == 0)
344                 /* It's Open call */
345                 connected++;
346             }
347           else
348             {
349               if (connected)
350                 {
351                   /* Close call */
352                   connected = 0;
353                   if (neug_state == NEUG_WAIT_FOR_TX_READY)
354                     {
355                       chSysLockFromIsr ();
356                       main_thread->p_u.rdymsg = RDY_OK;
357                       chSchReadyI (main_thread);
358                       chSysUnlockFromIsr ();
359                     }
360                 }
361             }
362
363           return USB_SUCCESS;
364         }
365     }
366
367   return USB_UNSUPPORT;
368 }
369
370 static int
371 neug_setup (uint8_t req, uint8_t req_no,
372                uint16_t value, uint16_t index, uint16_t len)
373 {
374   uint8_t type_rcp = req & (REQUEST_TYPE|RECIPIENT);
375
376   if (type_rcp == (VENDOR_REQUEST | DEVICE_RECIPIENT))
377     {
378       if (USB_SETUP_GET (req))
379         {
380           if (req_no == USB_FSIJ_MEMINFO)
381             {
382               usb_lld_set_data_to_send (mem_info, sizeof (mem_info));
383               return USB_SUCCESS;
384             }
385         }
386       else /* SETUP_SET */
387         {
388           uint8_t *addr = (uint8_t *)(0x20000000 + value * 0x100 + index);
389
390           if (req_no == USB_FSIJ_DOWNLOAD)
391             {
392               if (fsij_device_state != FSIJ_DEVICE_EXITED)
393                 return USB_UNSUPPORT;
394
395               if (addr < &_regnual_start || addr + len > &__heap_end__)
396                 return USB_UNSUPPORT;
397
398               if (index + len < 256)
399                 memset (addr + index + len, 0, 256 - (index + len));
400
401               usb_lld_set_data_to_recv (addr, len);
402               return USB_SUCCESS;
403             }
404           else if (req_no == USB_FSIJ_EXEC && len == 0)
405             {
406               if (fsij_device_state != FSIJ_DEVICE_EXITED)
407                 return USB_UNSUPPORT;
408
409               if (((uint32_t)addr & 0x03))
410                 return USB_UNSUPPORT;
411
412               return download_check_crc32 ((uint32_t *)addr);
413             }
414           else if (req_no == USB_NEUG_EXIT && len == 0)
415             {
416               if (fsij_device_state != FSIJ_DEVICE_RUNNING)
417                 return USB_UNSUPPORT;
418
419               fsij_device_state = FSIJ_DEVICE_NEUG_EXIT_REQUESTED;
420               return USB_SUCCESS;
421             }
422         }
423     }
424   else if (type_rcp == (CLASS_REQUEST | INTERFACE_RECIPIENT))
425     if (index == 0)
426       return vcom_port_data_setup (req, req_no, value);
427
428   return USB_UNSUPPORT;
429 }
430
431 static int
432 neug_get_descriptor (uint8_t desc_type, uint16_t index, uint16_t value)
433 {
434   (void)index;
435   if (desc_type == DEVICE_DESCRIPTOR)
436     {
437       usb_lld_set_data_to_send (device_desc.Descriptor,
438                                 device_desc.Descriptor_Size);
439       return USB_SUCCESS;
440     }
441   else if (desc_type == CONFIG_DESCRIPTOR)
442     {
443       usb_lld_set_data_to_send (config_desc.Descriptor,
444                                 config_desc.Descriptor_Size);
445       return USB_SUCCESS;
446     }
447   else if (desc_type == STRING_DESCRIPTOR)
448     {
449       uint8_t desc_index = value & 0xff;
450
451       if (desc_index < NUM_STRING_DESC)
452         {
453           usb_lld_set_data_to_send (string_descs[desc_index].Descriptor,
454                                     string_descs[desc_index].Descriptor_Size);
455           return USB_SUCCESS;
456         }
457     }
458
459   return USB_UNSUPPORT;
460 }
461
462 static void
463 neug_setup_endpoints_for_interface (uint16_t interface, int stop)
464 {
465   if (interface == 0)
466     {
467       if (!stop)
468         usb_lld_setup_endpoint (ENDP2, EP_INTERRUPT, 0, 0, ENDP2_TXADDR, 0);
469       else
470         usb_lld_stall_tx (ENDP2);
471     }
472   else if (interface == 1)
473     {
474       if (!stop)
475         {
476           usb_lld_setup_endpoint (ENDP1, EP_BULK, 0, 0, ENDP1_TXADDR, 0);
477           usb_lld_setup_endpoint (ENDP3, EP_BULK, 0, ENDP3_RXADDR, 0, 64);
478         }
479       else
480         {
481           usb_lld_stall_tx (ENDP1);
482           usb_lld_stall_rx (ENDP3);
483         }
484     }
485 }
486
487 static int neug_usb_event (uint8_t event_type, uint16_t value)
488 {
489   int i;
490   uint8_t current_conf;
491
492   switch (event_type)
493     {
494     case USB_EVENT_ADDRESS:
495       bDeviceState = ADDRESSED;
496       return USB_SUCCESS;
497     case USB_EVENT_CONFIG:
498       current_conf = usb_lld_current_configuration ();
499       if (current_conf == 0)
500         {
501           if (value != 1)
502             return USB_UNSUPPORT;
503
504           usb_lld_set_configuration (1);
505           for (i = 0; i < NUM_INTERFACES; i++)
506             neug_setup_endpoints_for_interface (i, 0);
507           bDeviceState = CONFIGURED;
508         }
509       else if (current_conf != value)
510         {
511           if (value != 0)
512             return USB_UNSUPPORT;
513
514           usb_lld_set_configuration (0);
515           for (i = 0; i < NUM_INTERFACES; i++)
516             neug_setup_endpoints_for_interface (i, 1);
517           bDeviceState = ADDRESSED;
518         }
519       /* Do nothing when current_conf == value */
520       return USB_SUCCESS;
521
522       return USB_SUCCESS;
523     default:
524       break;
525     }
526
527   return USB_UNSUPPORT;
528 }
529
530
531 static int neug_interface (uint8_t cmd, uint16_t interface, uint16_t alt)
532 {
533   static uint8_t zero = 0;
534
535   if (interface >= NUM_INTERFACES)
536     return USB_UNSUPPORT;
537
538   switch (cmd)
539     {
540     case USB_SET_INTERFACE:
541       if (alt != 0)
542         return USB_UNSUPPORT;
543       else
544         {
545           neug_setup_endpoints_for_interface (interface, 0);
546           return USB_SUCCESS;
547         }
548
549     case USB_GET_INTERFACE:
550       usb_lld_set_data_to_send (&zero, 1);
551       return USB_SUCCESS;
552
553     default:
554     case USB_QUERY_INTERFACE:
555       return USB_SUCCESS;
556     }
557 }
558
559 const struct usb_device_method Device_Method = {
560   neug_device_reset,
561   neug_ctrl_write_finish,
562   neug_setup,
563   neug_get_descriptor,
564   neug_usb_event,
565   neug_interface,
566 };
567
568
569
570 static void fill_serial_no_by_unique_id (void)
571 {
572   extern const uint8_t * unique_device_id (void);
573   uint8_t *p = &vcom_string3[12];
574   const uint8_t *u = unique_device_id ();
575   int i;
576
577   for (i = 0; i < 4; i++)
578     {
579       uint8_t b = u[i];
580       uint8_t nibble; 
581
582       nibble = (b >> 4);
583       nibble += (nibble >= 10 ? ('A' - 10) : '0');
584       p[i*4] = nibble;
585       nibble = (b & 0x0f);
586       nibble += (nibble >= 10 ? ('A' - 10) : '0');
587       p[i*4+2] = nibble;
588     }
589 }
590 \f
591 CH_IRQ_HANDLER (Vector90)
592 {
593   CH_IRQ_PROLOGUE();
594   chSysLockFromIsr();
595
596   usb_interrupt_handler ();
597
598   chSysUnlockFromIsr();
599   CH_IRQ_EPILOGUE();
600 }
601
602 void
603 EP1_IN_Callback (void)
604 {
605   if (main_thread != NULL)
606     {
607       chSysLockFromIsr ();
608       main_thread->p_u.rdymsg = RDY_OK;
609       chSchReadyI (main_thread);
610       chSysUnlockFromIsr ();
611     }
612 }
613
614 void
615 EP2_IN_Callback (void)
616 {
617 }
618
619 void
620 EP3_OUT_Callback (void)
621 {
622   usb_lld_rx_enable (ENDP3);
623 }
624 \f
625 static WORKING_AREA(wa_led, 64);
626
627 #define LED_ONESHOT_SHORT       ((eventmask_t)1)
628 #define LED_TWOSHOTS            ((eventmask_t)2)
629 #define LED_ONESHOT_LONG        ((eventmask_t)4)
630 static Thread *led_thread;
631
632 /*
633  * LED blinker: When notified, let LED emit for 100ms.
634  */
635 static msg_t led_blinker (void *arg)
636 {
637   (void)arg;
638
639   led_thread = chThdSelf ();
640   set_led (0);
641
642   while (1)
643     {
644       eventmask_t m;
645
646       m = chEvtWaitOne (ALL_EVENTS);
647       if (fsij_device_state != FSIJ_DEVICE_RUNNING)
648         break;
649
650       set_led (1);
651       if (m == LED_ONESHOT_SHORT)
652         chThdSleep (MS2ST (100));
653       else if (m == LED_TWOSHOTS)
654         {
655           chThdSleep (MS2ST (50));
656           set_led (0);
657           chThdSleep (MS2ST (50));
658           set_led (1);
659           chThdSleep (MS2ST (50));
660         }
661       else
662         chThdSleep (MS2ST (250));
663       set_led (0);
664     }
665
666   return 0;
667 }
668 \f
669 #define RANDOM_BYTES_LENGTH 32
670 static uint32_t random_word[RANDOM_BYTES_LENGTH/sizeof (uint32_t)];
671
672 extern void adc2_init (void);
673
674 /*
675  * Entry point.
676  *
677  * NOTE: the main function is already a thread in the system on entry.
678  */
679 int
680 main (int argc, char **argv)
681 {
682   (void)argc;
683   (void)argv;
684
685   fill_serial_no_by_unique_id ();
686
687   halInit ();
688   adc2_init ();
689   chSysInit ();
690
691   main_thread = chThdSelf ();
692
693   chThdCreateStatic (wa_led, sizeof (wa_led), NORMALPRIO, led_blinker, NULL);
694
695   neug_init (random_word, RANDOM_BYTES_LENGTH/sizeof (uint32_t));
696
697   usb_lld_init (config_desc.Descriptor[7]);
698
699   while (1)
700     {
701       unsigned int count = 0;
702
703       if (fsij_device_state != FSIJ_DEVICE_RUNNING)
704         break;
705
706       while (count < NEUG_PRE_LOOP || bDeviceState != CONFIGURED)
707         {
708           if (fsij_device_state != FSIJ_DEVICE_RUNNING)
709             break;
710
711           neug_wait_full ();
712           neug_flush ();
713
714           if ((count & 0x0007) == 0)
715             chEvtSignalFlags (led_thread, LED_ONESHOT_SHORT);
716           chEvtWaitOneTimeout (ALL_EVENTS, MS2ST (25));
717           count++;
718         }
719
720     waiting_connection:
721       while (connected == 0)
722         {
723           if (fsij_device_state != FSIJ_DEVICE_RUNNING)
724             break;
725
726           neug_flush ();
727           chEvtSignalFlags (led_thread, LED_TWOSHOTS);
728           chEvtWaitOneTimeout (ALL_EVENTS, MS2ST (5000));
729         }
730
731       if (fsij_device_state != FSIJ_DEVICE_RUNNING)
732         break;
733
734       /* The connection opened.  */
735       count = 0;
736       /*
737        * No parity is standard.  It means to provide conditioned output.
738        * When parity enabled, it means to provide raw output.
739        */
740       neug_select (line_coding.paritytype != 0);
741
742       while (1)
743         {
744           if (fsij_device_state != FSIJ_DEVICE_RUNNING)
745             break;
746
747           if (bDeviceState != CONFIGURED)
748             break;
749
750           neug_wait_full ();
751
752           if ((count & 0x03ff) == 0)
753             chEvtSignalFlags (led_thread, LED_ONESHOT_SHORT);
754
755           usb_lld_txcpy (random_word, ENDP1, 0, RANDOM_BYTES_LENGTH);
756           neug_flush ();
757
758           chSysLock ();
759           if (connected == 0)
760             {
761               chSysUnlock();
762               goto waiting_connection;
763             }
764           else
765             {
766               neug_state = NEUG_WAIT_FOR_TX_READY;
767               usb_lld_tx_enable (ENDP1, RANDOM_BYTES_LENGTH);
768               chSchGoSleepS (THD_STATE_SUSPENDED);
769               neug_state = 0;
770             }
771           chSysUnlock();
772
773           count++;
774         }
775     }
776
777   chEvtSignalFlags (led_thread, LED_ONESHOT_SHORT);
778   chThdWait (led_thread);
779
780   /*
781    * We come here, because of FSIJ_DEVICE_NEUG_EXIT_REQUESTED.
782    */
783   neug_fini ();
784
785   fsij_device_state = FSIJ_DEVICE_EXITED;
786
787   while (fsij_device_state == FSIJ_DEVICE_EXITED)
788     chThdSleep (MS2ST (500));
789
790   flash_unlock ();              /* Flash unlock should be done here */
791   set_led (1);
792   usb_lld_shutdown ();
793   /* Disable SysTick */
794   SysTick->CTRL = 0;
795   /* Disable all interrupts */
796   port_disable ();
797   /* Set vector */
798   SCB->VTOR = (uint32_t)&_regnual_start;
799 #ifdef DFU_SUPPORT
800 #define FLASH_SYS_START_ADDR 0x08000000
801 #define FLASH_SYS_END_ADDR (0x08000000+0x1000)
802   {
803     extern uint8_t _sys;
804     uint32_t addr;
805     handler *new_vector = (handler *)FLASH_SYS_START_ADDR;
806     void (*func) (void (*)(void)) = (void (*)(void (*)(void)))new_vector[10];
807
808     /* Kill DFU */
809     for (addr = FLASH_SYS_START_ADDR; addr < FLASH_SYS_END_ADDR;
810          addr += FLASH_PAGE_SIZE)
811       flash_erase_page (addr);
812
813     /* copy system service routines */
814     flash_write (FLASH_SYS_START_ADDR, &_sys, 0x1000);
815
816     /* Leave NeuG to exec reGNUal */
817     (*func) (*((void (**)(void))(&_regnual_start+4)));
818     for (;;);
819   }
820 #else
821   /* Leave NeuG to exec reGNUal */
822   flash_erase_all_and_exec (*((void (**)(void))(&_regnual_start+4)));
823 #endif
824
825   /* Never reached */
826   return 0;
827 }