src/test/test-hostname-util.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include "alloc-util.h"
   4 #include "fileio.h"
   5 #include "hostname-util.h"
   6 #include "string-util.h"
   7 #include "tmpfile-util.h"
   8 #include "util.h"
   9
  10 static void test_hostname_is_valid(void) {
  11         assert_se(hostname_is_valid("foobar", false));
  12         assert_se(hostname_is_valid("foobar.com", false));
  13         assert_se(!hostname_is_valid("foobar.com.", false));
  14         assert_se(hostname_is_valid("fooBAR", false));
  15         assert_se(hostname_is_valid("fooBAR.com", false));
  16         assert_se(!hostname_is_valid("fooBAR.", false));
  17         assert_se(!hostname_is_valid("fooBAR.com.", false));
  18         assert_se(!hostname_is_valid("fööbar", false));
  19         assert_se(!hostname_is_valid("", false));
  20         assert_se(!hostname_is_valid(".", false));
  21         assert_se(!hostname_is_valid("..", false));
  22         assert_se(!hostname_is_valid("foobar.", false));
  23         assert_se(!hostname_is_valid(".foobar", false));
  24         assert_se(!hostname_is_valid("foo..bar", false));
  25         assert_se(!hostname_is_valid("foo.bar..", false));
  26         assert_se(!hostname_is_valid("xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", false));
  27         assert_se(!hostname_is_valid("au-xph5-rvgrdsb5hcxc-47et3a5vvkrc-server-wyoz4elpdpe3.openstack.local", false));
  28
  29         assert_se(hostname_is_valid("foobar", true));
  30         assert_se(hostname_is_valid("foobar.com", true));
  31         assert_se(hostname_is_valid("foobar.com.", true));
  32         assert_se(hostname_is_valid("fooBAR", true));
  33         assert_se(hostname_is_valid("fooBAR.com", true));
  34         assert_se(!hostname_is_valid("fooBAR.", true));
  35         assert_se(hostname_is_valid("fooBAR.com.", true));
  36         assert_se(!hostname_is_valid("fööbar", true));
  37         assert_se(!hostname_is_valid("", true));
  38         assert_se(!hostname_is_valid(".", true));
  39         assert_se(!hostname_is_valid("..", true));
  40         assert_se(!hostname_is_valid("foobar.", true));
  41         assert_se(!hostname_is_valid(".foobar", true));
  42         assert_se(!hostname_is_valid("foo..bar", true));
  43         assert_se(!hostname_is_valid("foo.bar..", true));
  44         assert_se(!hostname_is_valid("xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", true));
  45 }
  46
  47 static void test_hostname_cleanup(void) {
  48         char *s;
  49
  50         s = strdupa("foobar");
  51         assert_se(streq(hostname_cleanup(s), "foobar"));
  52         s = strdupa("foobar.com");
  53         assert_se(streq(hostname_cleanup(s), "foobar.com"));
  54         s = strdupa("foobar.com.");
  55         assert_se(streq(hostname_cleanup(s), "foobar.com"));
  56         s = strdupa("fooBAR");
  57         assert_se(streq(hostname_cleanup(s), "fooBAR"));
  58         s = strdupa("fooBAR.com");
  59         assert_se(streq(hostname_cleanup(s), "fooBAR.com"));
  60         s = strdupa("fooBAR.");
  61         assert_se(streq(hostname_cleanup(s), "fooBAR"));
  62         s = strdupa("fooBAR.com.");
  63         assert_se(streq(hostname_cleanup(s), "fooBAR.com"));
  64         s = strdupa("fööbar");
  65         assert_se(streq(hostname_cleanup(s), "fbar"));
  66         s = strdupa("");
  67         assert_se(isempty(hostname_cleanup(s)));
  68         s = strdupa(".");
  69         assert_se(isempty(hostname_cleanup(s)));
  70         s = strdupa("..");
  71         assert_se(isempty(hostname_cleanup(s)));
  72         s = strdupa("foobar.");
  73         assert_se(streq(hostname_cleanup(s), "foobar"));
  74         s = strdupa(".foobar");
  75         assert_se(streq(hostname_cleanup(s), "foobar"));
  76         s = strdupa("foo..bar");
  77         assert_se(streq(hostname_cleanup(s), "foo.bar"));
  78         s = strdupa("foo.bar..");
  79         assert_se(streq(hostname_cleanup(s), "foo.bar"));
  80         s = strdupa("xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx");
  81         assert_se(streq(hostname_cleanup(s), "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"));
  82 }
  83
  84 static void test_read_etc_hostname(void) {
  85         char path[] = "/tmp/hostname.XXXXXX";
  86         char *hostname;
  87         int fd;
  88
  89         fd = mkostemp_safe(path);
  90         assert(fd > 0);
  91         close(fd);
  92
  93         /* simple hostname */
  94         assert_se(write_string_file(path, "foo", WRITE_STRING_FILE_CREATE) == 0);
  95         assert_se(read_etc_hostname(path, &hostname) == 0);
  96         assert_se(streq(hostname, "foo"));
  97         hostname = mfree(hostname);
  98
  99         /* with comment */
 100         assert_se(write_string_file(path, "# comment\nfoo", WRITE_STRING_FILE_CREATE) == 0);
 101         assert_se(read_etc_hostname(path, &hostname) == 0);
 102         assert_se(hostname);
 103         assert_se(streq(hostname, "foo"));
 104         hostname = mfree(hostname);
 105
 106         /* with comment and extra whitespace */
 107         assert_se(write_string_file(path, "# comment\n\n foo ", WRITE_STRING_FILE_CREATE) == 0);
 108         assert_se(read_etc_hostname(path, &hostname) == 0);
 109         assert_se(hostname);
 110         assert_se(streq(hostname, "foo"));
 111         hostname = mfree(hostname);
 112
 113         /* cleans up name */
 114         assert_se(write_string_file(path, "!foo/bar.com", WRITE_STRING_FILE_CREATE) == 0);
 115         assert_se(read_etc_hostname(path, &hostname) == 0);
 116         assert_se(hostname);
 117         assert_se(streq(hostname, "foobar.com"));
 118         hostname = mfree(hostname);
 119
 120         /* no value set */
 121         hostname = (char*) 0x1234;
 122         assert_se(write_string_file(path, "# nothing here\n", WRITE_STRING_FILE_CREATE) == 0);
 123         assert_se(read_etc_hostname(path, &hostname) == -ENOENT);
 124         assert_se(hostname == (char*) 0x1234);  /* does not touch argument on error */
 125
 126         /* nonexisting file */
 127         assert_se(read_etc_hostname("/non/existing", &hostname) == -ENOENT);
 128         assert_se(hostname == (char*) 0x1234);  /* does not touch argument on error */
 129
 130         unlink(path);
 131 }
 132
 133 int main(int argc, char *argv[]) {
 134         log_parse_environment();
 135         log_open();
 136
 137         test_hostname_is_valid();
 138         test_hostname_cleanup();
 139         test_read_etc_hostname();
 140
 141         return 0;
 142 }