lib/yesno.c

   1 /* yesno.c -- read a yes/no response from stdin
   2
   3    Copyright (C) 1990, 1998, 2001, 2003, 2004, 2005, 2006, 2007 Free
   4    Software Foundation, Inc.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2, or (at your option)
   9    any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software Foundation,
  18    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  19
  20 #include <config.h>
  21
  22 #include "yesno.h"
  23
  24 #include <stdlib.h>
  25 #include <stdio.h>
  26
  27 #if ENABLE_NLS
  28 # include "getline.h"
  29 #endif
  30
  31 extern int rpmatch (char const *response);
  32
  33 /* Return true if we read an affirmative line from standard input.
  34
  35    Since this function uses stdin, it is suggested that the caller not
  36    use STDIN_FILENO directly, and also that the line
  37    atexit(close_stdin) be added to main().  */
  38
  39 bool
  40 yesno (void)
  41 {
  42   bool yes;
  43
  44 #if ENABLE_NLS
  45   char *response = NULL;
  46   size_t response_size = 0;
  47   ssize_t response_len = getline (&response, &response_size, stdin);
  48
  49   if (response_len <= 0)
  50     yes = false;
  51   else
  52     {
  53       response[response_len - 1] = '\0';
  54       yes = (0 < rpmatch (response));
  55     }
  56
  57   free (response);
  58 #else
  59   /* Test against "^[yY]", hardcoded to avoid requiring getline,
  60      regex, and rpmatch.  */
  61   int c = getchar ();
  62   yes = (c == 'y' || c == 'Y');
  63   while (c != '\n' && c != EOF)
  64     c = getchar ();
  65 #endif
  66
  67   return yes;
  68 }