arch/all-pc/boot/grub2-aros/grub-core/tests/fake_input.c

   1 /*
   2  *  GRUB  --  GRand Unified Bootloader
   3  *  Copyright (C) 2013  Free Software Foundation, Inc.
   4  *
   5  *  GRUB is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  GRUB is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <grub/term.h>
  20 #include <grub/mm.h>
  21 #include <grub/dl.h>
  22 #include <grub/test.h>
  23
  24 GRUB_MOD_LICENSE ("GPLv3+");
  25
  26 static int *seq;
  27 static int seqptr, nseq;
  28 static struct grub_term_input *saved;
  29 static int fake_input;
  30
  31 static int
  32 fake_getkey (struct grub_term_input *term __attribute__ ((unused)))
  33 {
  34   if (seq && seqptr < nseq)
  35     return seq[seqptr++];
  36   return -1;
  37 }
  38
  39 static struct grub_term_input fake_input_term =
  40   {
  41     .name = "fake",
  42     .getkey = fake_getkey
  43   };
  44
  45 void
  46 grub_terminal_input_fake_sequence (int *seq_in, int nseq_in)
  47 {
  48   if (!fake_input)
  49     saved = grub_term_inputs;
  50   if (seq)
  51     grub_free (seq);
  52   seq = grub_malloc (nseq_in * sizeof (seq[0]));
  53   if (!seq)
  54     return;
  55
  56   grub_term_inputs = &fake_input_term;
  57   grub_memcpy (seq, seq_in, nseq_in * sizeof (seq[0]));
  58
  59   nseq = nseq_in;
  60   seqptr = 0;
  61   fake_input = 1;
  62 }
  63
  64 void
  65 grub_terminal_input_fake_sequence_end (void)
  66 {
  67   if (!fake_input)
  68     return;
  69   fake_input = 0;
  70   grub_term_inputs = saved;
  71   grub_free (seq);
  72   seq = 0;
  73   nseq = 0;
  74   seqptr = 0;
  75 }