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 "def.h"
  22 #include "macro.h"
  23 #include "string-util.h"
  24 #include "user-util.h"
  25 #include "util.h"
  26
  27 static void test_uid_to_name_one(uid_t uid, const char *name) {
  28         _cleanup_free_ char *t = NULL;
  29
  30         assert_se(t = uid_to_name(uid));
  31         assert_se(streq_ptr(t, name));
  32 }
  33
  34 static void test_gid_to_name_one(gid_t gid, const char *name) {
  35         _cleanup_free_ char *t = NULL;
  36
  37         assert_se(t = gid_to_name(gid));
  38         assert_se(streq_ptr(t, name));
  39 }
  40
  41 static void test_parse_uid(void) {
  42         int r;
  43         uid_t uid;
  44
  45         r = parse_uid("100", &uid);
  46         assert_se(r == 0);
  47         assert_se(uid == 100);
  48
  49         r = parse_uid("65535", &uid);
  50         assert_se(r == -ENXIO);
  51
  52         r = parse_uid("asdsdas", &uid);
  53         assert_se(r == -EINVAL);
  54 }
  55
  56 static void test_uid_ptr(void) {
  57
  58         assert_se(UID_TO_PTR(0) != NULL);
  59         assert_se(UID_TO_PTR(1000) != NULL);
  60
  61         assert_se(PTR_TO_UID(UID_TO_PTR(0)) == 0);
  62         assert_se(PTR_TO_UID(UID_TO_PTR(1000)) == 1000);
  63 }
  64
  65 static void test_valid_user_group_name(void) {
  66         assert_se(!valid_user_group_name(NULL));
  67         assert_se(!valid_user_group_name(""));
  68         assert_se(!valid_user_group_name("1"));
  69         assert_se(!valid_user_group_name("65535"));
  70         assert_se(!valid_user_group_name("-1"));
  71         assert_se(!valid_user_group_name("-kkk"));
  72         assert_se(!valid_user_group_name("rööt"));
  73         assert_se(!valid_user_group_name("."));
  74         assert_se(!valid_user_group_name("eff.eff"));
  75         assert_se(!valid_user_group_name("foo\nbar"));
  76         assert_se(!valid_user_group_name("0123456789012345678901234567890123456789"));
  77         assert_se(!valid_user_group_name_or_id("aaa:bbb"));
  78
  79         assert_se(valid_user_group_name("root"));
  80         assert_se(valid_user_group_name("lennart"));
  81         assert_se(valid_user_group_name("LENNART"));
  82         assert_se(valid_user_group_name("_kkk"));
  83         assert_se(valid_user_group_name("kkk-"));
  84         assert_se(valid_user_group_name("kk-k"));
  85
  86         assert_se(valid_user_group_name("some5"));
  87         assert_se(!valid_user_group_name("5some"));
  88         assert_se(valid_user_group_name("INNER5NUMBER"));
  89 }
  90
  91 static void test_valid_user_group_name_or_id(void) {
  92         assert_se(!valid_user_group_name_or_id(NULL));
  93         assert_se(!valid_user_group_name_or_id(""));
  94         assert_se(valid_user_group_name_or_id("0"));
  95         assert_se(valid_user_group_name_or_id("1"));
  96         assert_se(valid_user_group_name_or_id("65534"));
  97         assert_se(!valid_user_group_name_or_id("65535"));
  98         assert_se(valid_user_group_name_or_id("65536"));
  99         assert_se(!valid_user_group_name_or_id("-1"));
 100         assert_se(!valid_user_group_name_or_id("-kkk"));
 101         assert_se(!valid_user_group_name_or_id("rööt"));
 102         assert_se(!valid_user_group_name_or_id("."));
 103         assert_se(!valid_user_group_name_or_id("eff.eff"));
 104         assert_se(!valid_user_group_name_or_id("foo\nbar"));
 105         assert_se(!valid_user_group_name_or_id("0123456789012345678901234567890123456789"));
 106         assert_se(!valid_user_group_name_or_id("aaa:bbb"));
 107
 108         assert_se(valid_user_group_name_or_id("root"));
 109         assert_se(valid_user_group_name_or_id("lennart"));
 110         assert_se(valid_user_group_name_or_id("LENNART"));
 111         assert_se(valid_user_group_name_or_id("_kkk"));
 112         assert_se(valid_user_group_name_or_id("kkk-"));
 113         assert_se(valid_user_group_name_or_id("kk-k"));
 114
 115         assert_se(valid_user_group_name_or_id("some5"));
 116         assert_se(!valid_user_group_name_or_id("5some"));
 117         assert_se(valid_user_group_name_or_id("INNER5NUMBER"));
 118 }
 119
 120 static void test_valid_gecos(void) {
 121
 122         assert_se(!valid_gecos(NULL));
 123         assert_se(valid_gecos(""));
 124         assert_se(valid_gecos("test"));
 125         assert_se(valid_gecos("Ümläüt"));
 126         assert_se(!valid_gecos("In\nvalid"));
 127         assert_se(!valid_gecos("In:valid"));
 128 }
 129
 130 static void test_valid_home(void) {
 131
 132         assert_se(!valid_home(NULL));
 133         assert_se(!valid_home(""));
 134         assert_se(!valid_home("."));
 135         assert_se(!valid_home("/home/.."));
 136         assert_se(!valid_home("/home/../"));
 137         assert_se(!valid_home("/home\n/foo"));
 138         assert_se(!valid_home("./piep"));
 139         assert_se(!valid_home("piep"));
 140         assert_se(!valid_home("/home/user:lennart"));
 141
 142         assert_se(valid_home("/"));
 143         assert_se(valid_home("/home"));
 144         assert_se(valid_home("/home/foo"));
 145 }
 146
 147 int main(int argc, char*argv[]) {
 148
 149         test_uid_to_name_one(0, "root");
 150         test_uid_to_name_one(0xFFFF, "65535");
 151         test_uid_to_name_one(0xFFFFFFFF, "4294967295");
 152
 153         test_gid_to_name_one(0, "root");
 154         test_gid_to_name_one(TTY_GID, "tty");
 155         test_gid_to_name_one(0xFFFF, "65535");
 156         test_gid_to_name_one(0xFFFFFFFF, "4294967295");
 157
 158         test_parse_uid();
 159         test_uid_ptr();
 160
 161         test_valid_user_group_name();
 162         test_valid_user_group_name_or_id();
 163         test_valid_gecos();
 164         test_valid_home();
 165
 166         return 0;
 167 }