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         bg_shed();
  68         if (!ctrlc_disabled) {
  69                 switch (serial_getc()) {
  70                 case 0x03:              /* ^C - Control C */
  71                         ctrlc_was_pressed = 1;
  72                         return 1;
  73                 default:
  74                         break;
  75                 }
  76         }
  77         return 0;
  78 }
  79
  80 /* Reads user's confirmation.
  81    Returns 1 if user's input is "y", "Y", "yes" or "YES"
  82 */
  83 uint_fast8_t confirm_yes(void)
  84 {
  85         uint_fast8_t checkch, ch;
  86
  87         checkch = ch = my_getchar(1);
  88         putchar(ch);
  89         while (ch != '\r') {
  90                 ch = my_getchar(1);
  91                 putchar(ch);
  92         }
  93         putchar('\n');
  94
  95         return (checkch  == 'y');
  96 }
  97
  98 #if 0
  99 uint_fast8_t confirm_yesno(void)
 100 {
 101         unsigned int i;
 102         char str_input[5];
 103
 104         /* Flush input */
 105         while (serial_getc() < 0)
 106                 ;
 107         i = 0;
 108         while (i < sizeof(str_input)) {
 109                 str_input[i] = my_getchar(1);
 110                 putchar(str_input[i]);
 111                 if (str_input[i] == '\r')
 112                         break;
 113                 i++;
 114         }
 115         putchar('\n');
 116         if (strncmp(str_input, "y\r", 2) == 0 ||
 117             strncmp(str_input, "Y\r", 2) == 0 ||
 118             strncmp(str_input, "yes\r", 4) == 0 ||
 119             strncmp(str_input, "YES\r", 4) == 0)
 120                 return 1;
 121         return 0;
 122 }
 123 #endif
 124
 125 /* pass 1 to disable ctrlc() checking, 0 to enable.
 126  * returns previous state
 127  */
 128 uint_fast8_t disable_ctrlc(uint_fast8_t disable)
 129 {
 130         uint_fast8_t prev = ctrlc_disabled;     /* save previous state */
 131
 132         ctrlc_disabled = disable;
 133         return prev;
 134 }
 135
 136 uint_fast8_t had_ctrlc (void)
 137 {
 138         return ctrlc_was_pressed;
 139 }
 140
 141 void clear_ctrlc(void)
 142 {
 143         ctrlc_was_pressed = 0;
 144 }