src/test/test-user-util.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2015 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include "alloc-util.h"
  21 #include "macro.h"
  22 #include "string-util.h"
  23 #include "user-util.h"
  24 #include "util.h"
  25
  26 static void test_uid_to_name_one(uid_t uid, const char *name) {
  27         _cleanup_free_ char *t = NULL;
  28
  29         assert_se(t = uid_to_name(uid));
  30         assert_se(streq_ptr(t, name));
  31 }
  32
  33 static void test_gid_to_name_one(gid_t gid, const char *name) {
  34         _cleanup_free_ char *t = NULL;
  35
  36         assert_se(t = gid_to_name(gid));
  37         assert_se(streq_ptr(t, name));
  38 }
  39
  40 static void test_parse_uid(void) {
  41         int r;
  42         uid_t uid;
  43
  44         r = parse_uid("100", &uid);
  45         assert_se(r == 0);
  46         assert_se(uid == 100);
  47
  48         r = parse_uid("65535", &uid);
  49         assert_se(r == -ENXIO);
  50
  51         r = parse_uid("asdsdas", &uid);
  52         assert_se(r == -EINVAL);
  53 }
  54
  55 static void test_uid_ptr(void) {
  56
  57         assert_se(UID_TO_PTR(0) != NULL);
  58         assert_se(UID_TO_PTR(1000) != NULL);
  59
  60         assert_se(PTR_TO_UID(UID_TO_PTR(0)) == 0);
  61         assert_se(PTR_TO_UID(UID_TO_PTR(1000)) == 1000);
  62 }
  63
  64 static void test_valid_user_group_name(void) {
  65         assert_se(!valid_user_group_name(NULL));
  66         assert_se(!valid_user_group_name(""));
  67         assert_se(!valid_user_group_name("1"));
  68         assert_se(!valid_user_group_name("65535"));
  69         assert_se(!valid_user_group_name("-1"));
  70         assert_se(!valid_user_group_name("-kkk"));
  71         assert_se(!valid_user_group_name("rööt"));
  72         assert_se(!valid_user_group_name("."));
  73         assert_se(!valid_user_group_name("eff.eff"));
  74         assert_se(!valid_user_group_name("foo\nbar"));
  75         assert_se(!valid_user_group_name("0123456789012345678901234567890123456789"));
  76         assert_se(!valid_user_group_name_or_id("aaa:bbb"));
  77
  78         assert_se(valid_user_group_name("root"));
  79         assert_se(valid_user_group_name("lennart"));
  80         assert_se(valid_user_group_name("LENNART"));
  81         assert_se(valid_user_group_name("_kkk"));
  82         assert_se(valid_user_group_name("kkk-"));
  83         assert_se(valid_user_group_name("kk-k"));
  84
  85         assert_se(valid_user_group_name("some5"));
  86         assert_se(!valid_user_group_name("5some"));
  87         assert_se(valid_user_group_name("INNER5NUMBER"));
  88 }
  89
  90 static void test_valid_user_group_name_or_id(void) {
  91         assert_se(!valid_user_group_name_or_id(NULL));
  92         assert_se(!valid_user_group_name_or_id(""));
  93         assert_se(valid_user_group_name_or_id("0"));
  94         assert_se(valid_user_group_name_or_id("1"));
  95         assert_se(valid_user_group_name_or_id("65534"));
  96         assert_se(!valid_user_group_name_or_id("65535"));
  97         assert_se(valid_user_group_name_or_id("65536"));
  98         assert_se(!valid_user_group_name_or_id("-1"));
  99         assert_se(!valid_user_group_name_or_id("-kkk"));
 100         assert_se(!valid_user_group_name_or_id("rööt"));
 101         assert_se(!valid_user_group_name_or_id("."));
 102         assert_se(!valid_user_group_name_or_id("eff.eff"));
 103         assert_se(!valid_user_group_name_or_id("foo\nbar"));
 104         assert_se(!valid_user_group_name_or_id("0123456789012345678901234567890123456789"));
 105         assert_se(!valid_user_group_name_or_id("aaa:bbb"));
 106
 107         assert_se(valid_user_group_name_or_id("root"));
 108         assert_se(valid_user_group_name_or_id("lennart"));
 109         assert_se(valid_user_group_name_or_id("LENNART"));
 110         assert_se(valid_user_group_name_or_id("_kkk"));
 111         assert_se(valid_user_group_name_or_id("kkk-"));
 112         assert_se(valid_user_group_name_or_id("kk-k"));
 113
 114         assert_se(valid_user_group_name_or_id("some5"));
 115         assert_se(!valid_user_group_name_or_id("5some"));
 116         assert_se(valid_user_group_name_or_id("INNER5NUMBER"));
 117 }
 118
 119 static void test_valid_gecos(void) {
 120
 121         assert_se(!valid_gecos(NULL));
 122         assert_se(valid_gecos(""));
 123         assert_se(valid_gecos("test"));
 124         assert_se(valid_gecos("Ümläüt"));
 125         assert_se(!valid_gecos("In\nvalid"));
 126         assert_se(!valid_gecos("In:valid"));
 127 }
 128
 129 static void test_valid_home(void) {
 130
 131         assert_se(!valid_home(NULL));
 132         assert_se(!valid_home(""));
 133         assert_se(!valid_home("."));
 134         assert_se(!valid_home("/home/.."));
 135         assert_se(!valid_home("/home/../"));
 136         assert_se(!valid_home("/home\n/foo"));
 137         assert_se(!valid_home("./piep"));
 138         assert_se(!valid_home("piep"));
 139         assert_se(!valid_home("/home/user:lennart"));
 140
 141         assert_se(valid_home("/"));
 142         assert_se(valid_home("/home"));
 143         assert_se(valid_home("/home/foo"));
 144 }
 145
 146 int main(int argc, char*argv[]) {
 147
 148         test_uid_to_name_one(0, "root");
 149         test_uid_to_name_one(0xFFFF, "65535");
 150         test_uid_to_name_one(0xFFFFFFFF, "4294967295");
 151
 152         test_gid_to_name_one(0, "root");
 153         test_gid_to_name_one(TTY_GID, "tty");
 154         test_gid_to_name_one(0xFFFF, "65535");
 155         test_gid_to_name_one(0xFFFFFFFF, "4294967295");
 156
 157         test_parse_uid();
 158         test_uid_ptr();
 159
 160         test_valid_user_group_name();
 161         test_valid_user_group_name_or_id();
 162         test_valid_gecos();
 163         test_valid_home();
 164
 165         return 0;
 166 }