src/test/test-terminal-util.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <fcntl.h>
   4 #include <stdbool.h>
   5 #include <stdio.h>
   6 #include <unistd.h>
   7
   8 #include "alloc-util.h"
   9 #include "fd-util.h"
  10 #include "macro.h"
  11 #include "path-util.h"
  12 #include "strv.h"
  13 #include "terminal-util.h"
  14 #include "tests.h"
  15 #include "tmpfile-util.h"
  16 #include "util.h"
  17
  18 static void test_default_term_for_tty(void) {
  19         puts(default_term_for_tty("/dev/tty23"));
  20         puts(default_term_for_tty("/dev/ttyS23"));
  21         puts(default_term_for_tty("/dev/tty0"));
  22         puts(default_term_for_tty("/dev/pty0"));
  23         puts(default_term_for_tty("/dev/pts/0"));
  24         puts(default_term_for_tty("/dev/console"));
  25         puts(default_term_for_tty("tty23"));
  26         puts(default_term_for_tty("ttyS23"));
  27         puts(default_term_for_tty("tty0"));
  28         puts(default_term_for_tty("pty0"));
  29         puts(default_term_for_tty("pts/0"));
  30         puts(default_term_for_tty("console"));
  31 }
  32
  33 static void test_read_one_char(void) {
  34         _cleanup_fclose_ FILE *file = NULL;
  35         char r;
  36         bool need_nl;
  37         char name[] = "/tmp/test-read_one_char.XXXXXX";
  38
  39         assert(fmkostemp_safe(name, "r+", &file) == 0);
  40
  41         assert_se(fputs("c\n", file) >= 0);
  42         rewind(file);
  43         assert_se(read_one_char(file, &r, 1000000, &need_nl) >= 0);
  44         assert_se(!need_nl);
  45         assert_se(r == 'c');
  46         assert_se(read_one_char(file, &r, 1000000, &need_nl) < 0);
  47
  48         rewind(file);
  49         assert_se(fputs("foobar\n", file) >= 0);
  50         rewind(file);
  51         assert_se(read_one_char(file, &r, 1000000, &need_nl) < 0);
  52
  53         rewind(file);
  54         assert_se(fputs("\n", file) >= 0);
  55         rewind(file);
  56         assert_se(read_one_char(file, &r, 1000000, &need_nl) < 0);
  57
  58         assert_se(unlink(name) >= 0);
  59 }
  60
  61 static void test_getttyname_malloc(void) {
  62         _cleanup_free_ char *ttyname = NULL;
  63         _cleanup_close_ int master = -1;
  64
  65         assert_se((master = posix_openpt(O_RDWR|O_NOCTTY)) >= 0);
  66         assert_se(getttyname_malloc(master, &ttyname) >= 0);
  67         assert_se(PATH_IN_SET(ttyname, "ptmx", "pts/ptmx"));
  68 }
  69
  70 int main(int argc, char *argv[]) {
  71         test_setup_logging(LOG_INFO);
  72
  73         test_default_term_for_tty();
  74         test_read_one_char();
  75         test_getttyname_malloc();
  76
  77         return 0;
  78 }