avr/con-utils.c

   1 /*
   2  * (C) Copyright 2014 Leo C. <erbl259-lmu@yahoo.de>
   3  *
   4  * SPDX-License-Identifier:     GPL-2.0+
   5  */
   6
   7 #include "common.h"
   8 #include <string.h>
   9 #include <avr/wdt.h>
  10
  11 #include "config.h"
  12 #include "serial.h"
  13 #include "background.h"
  14 #include "con-utils.h"
  15
  16 uint_fast8_t tstc(void)
  17 {
  18         bg_shed();
  19         return serial_tstc();
  20 }
  21
  22 int my_getchar(uint_fast8_t waitforchar)
  23 {
  24         int c;
  25
  26         do {
  27                 bg_shed();
  28                 c = serial_getc();
  29         } while ((c < 0) && waitforchar);
  30
  31 #ifdef CONFIG_SYS_FBOOTSIG
  32         if (c < 0)
  33                 return c;
  34
  35         static const FLASH unsigned char bootsig[] = {CONFIG_SYS_FBOOTSIG};
  36         static uint8_t pb;
  37         unsigned char uc = c;
  38
  39
  40         if (bootsig[pb] == 0) {
  41                 if (uc == 0xff) {
  42                         wdt_enable(WDTO_15MS);
  43                         for(;;)
  44                                 ;
  45                 } else
  46                         pb = 0;
  47
  48         } else {
  49                 if (bootsig[pb] == uc)
  50                         pb++;
  51                 else
  52                         pb = 0;
  53         }
  54 #endif
  55
  56         return c;
  57 }
  58
  59
  60 /* test if ctrl-c was pressed */
  61
  62 static uint_fast8_t ctrlc_disabled;     /* see disable_ctrl() */
  63 static uint_fast8_t ctrlc_was_pressed;
  64
  65 uint_fast8_t ctrlc(void)
  66 {
  67         if (!ctrlc_disabled) {
  68                 switch (serial_getc()) {
  69                 case 0x03:              /* ^C - Control C */
  70                         ctrlc_was_pressed = 1;
  71                         return 1;
  72                 default:
  73                         break;
  74                 }
  75         }
  76         return 0;
  77 }
  78
  79 /* Reads user's confirmation.
  80    Returns 1 if user's input is "y", "Y", "yes" or "YES"
  81 */
  82 uint_fast8_t confirm_yesno(void)
  83 {
  84         unsigned int i;
  85         char str_input[5];
  86
  87         /* Flush input */
  88         while (serial_getc())
  89                 ;
  90         i = 0;
  91         while (i < sizeof(str_input)) {
  92                 str_input[i] = my_getchar(1);
  93                 putchar(str_input[i]);
  94                 if (str_input[i] == '\r')
  95                         break;
  96                 i++;
  97         }
  98         putchar('\n');
  99         if (strncmp(str_input, "y\r", 2) == 0 ||
 100             strncmp(str_input, "Y\r", 2) == 0 ||
 101             strncmp(str_input, "yes\r", 4) == 0 ||
 102             strncmp(str_input, "YES\r", 4) == 0)
 103                 return 1;
 104         return 0;
 105 }
 106
 107 /* pass 1 to disable ctrlc() checking, 0 to enable.
 108  * returns previous state
 109  */
 110 uint_fast8_t disable_ctrlc(uint_fast8_t disable)
 111 {
 112         uint_fast8_t prev = ctrlc_disabled;     /* save previous state */
 113
 114         ctrlc_disabled = disable;
 115         return prev;
 116 }
 117
 118 uint_fast8_t had_ctrlc (void)
 119 {
 120         return ctrlc_was_pressed;
 121 }
 122
 123 void clear_ctrlc(void)
 124 {
 125         ctrlc_was_pressed = 0;
 126 }