src/test-install.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2011 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <errno.h>
  23 #include <string.h>
  24 #include <stdio.h>
  25 #include <fcntl.h>
  26
  27 #include "util.h"
  28 #include "install.h"
  29
  30 static void dump_changes(UnitFileChange *c, unsigned n) {
  31         unsigned i;
  32
  33         assert(n == 0 || c);
  34
  35         for (i = 0; i < n; i++) {
  36                 if (c[i].type == UNIT_FILE_UNLINK)
  37                         printf("rm '%s'\n", c[i].path);
  38                 else if (c[i].type == UNIT_FILE_SYMLINK)
  39                         printf("ln -s '%s' '%s'\n", c[i].source, c[i].path);
  40         }
  41 }
  42
  43 int main(int argc, char* argv[]) {
  44         Hashmap *h;
  45         UnitFileList *p;
  46         Iterator i;
  47         int r;
  48         const char *const files[] = { "avahi-daemon.service", NULL };
  49         const char *const files2[] = { "/home/lennart/test.service", NULL };
  50         UnitFileChange *changes = NULL;
  51         unsigned n_changes = 0;
  52
  53         h = hashmap_new(string_hash_func, string_compare_func);
  54         r = unit_file_get_list(UNIT_FILE_SYSTEM, NULL, h);
  55         assert_se(r == 0);
  56
  57         HASHMAP_FOREACH(p, h, i) {
  58                 UnitFileState s;
  59
  60                 s = unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(p->path));
  61
  62                 assert_se(p->state == s);
  63
  64                 fprintf(stderr, "%s (%s)\n",
  65                         p->path,
  66                         unit_file_state_to_string(p->state));
  67         }
  68
  69         unit_file_list_free(h);
  70
  71         log_error("enable");
  72
  73         r = unit_file_enable(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
  74         assert_se(r >= 0);
  75
  76         log_error("enable2");
  77
  78         r = unit_file_enable(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
  79         assert_se(r >= 0);
  80
  81         dump_changes(changes, n_changes);
  82         unit_file_changes_free(changes, n_changes);
  83
  84         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_ENABLED);
  85
  86         log_error("disable");
  87
  88         changes = NULL;
  89         n_changes = 0;
  90
  91         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
  92         assert_se(r >= 0);
  93
  94         dump_changes(changes, n_changes);
  95         unit_file_changes_free(changes, n_changes);
  96
  97         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_DISABLED);
  98
  99         log_error("mask");
 100         changes = NULL;
 101         n_changes = 0;
 102
 103         r = unit_file_mask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
 104         assert_se(r >= 0);
 105         log_error("mask2");
 106         r = unit_file_mask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
 107         assert_se(r >= 0);
 108
 109         dump_changes(changes, n_changes);
 110         unit_file_changes_free(changes, n_changes);
 111
 112         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_MASKED);
 113
 114         log_error("unmask");
 115         changes = NULL;
 116         n_changes = 0;
 117
 118         r = unit_file_unmask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
 119         assert_se(r >= 0);
 120         log_error("unmask2");
 121         r = unit_file_unmask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
 122         assert_se(r >= 0);
 123
 124         dump_changes(changes, n_changes);
 125         unit_file_changes_free(changes, n_changes);
 126
 127         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_DISABLED);
 128
 129         log_error("mask");
 130         changes = NULL;
 131         n_changes = 0;
 132
 133         r = unit_file_mask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
 134         assert_se(r >= 0);
 135
 136         dump_changes(changes, n_changes);
 137         unit_file_changes_free(changes, n_changes);
 138
 139         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_MASKED);
 140
 141         log_error("disable");
 142         changes = NULL;
 143         n_changes = 0;
 144
 145         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
 146         assert_se(r >= 0);
 147         log_error("disable2");
 148         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
 149         assert_se(r >= 0);
 150
 151         dump_changes(changes, n_changes);
 152         unit_file_changes_free(changes, n_changes);
 153
 154         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_MASKED);
 155
 156         log_error("umask");
 157         changes = NULL;
 158         n_changes = 0;
 159
 160         r = unit_file_unmask(UNIT_FILE_SYSTEM, false, NULL, (char**) files, &changes, &n_changes);
 161         assert_se(r >= 0);
 162
 163         dump_changes(changes, n_changes);
 164         unit_file_changes_free(changes, n_changes);
 165
 166         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, files[0]) == UNIT_FILE_DISABLED);
 167
 168         log_error("enable files2");
 169         changes = NULL;
 170         n_changes = 0;
 171
 172         r = unit_file_enable(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, false, &changes, &n_changes);
 173         assert_se(r >= 0);
 174
 175         dump_changes(changes, n_changes);
 176         unit_file_changes_free(changes, n_changes);
 177
 178         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == UNIT_FILE_ENABLED);
 179
 180         log_error("disable files2");
 181         changes = NULL;
 182         n_changes = 0;
 183
 184         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, &changes, &n_changes);
 185         assert_se(r >= 0);
 186
 187         dump_changes(changes, n_changes);
 188         unit_file_changes_free(changes, n_changes);
 189
 190         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == _UNIT_FILE_STATE_INVALID);
 191
 192         log_error("link files2");
 193         changes = NULL;
 194         n_changes = 0;
 195
 196         r = unit_file_link(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, false, &changes, &n_changes);
 197         assert_se(r >= 0);
 198
 199         dump_changes(changes, n_changes);
 200         unit_file_changes_free(changes, n_changes);
 201
 202         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == UNIT_FILE_LINKED);
 203
 204         log_error("disable files2");
 205         changes = NULL;
 206         n_changes = 0;
 207
 208         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, &changes, &n_changes);
 209         assert_se(r >= 0);
 210
 211         dump_changes(changes, n_changes);
 212         unit_file_changes_free(changes, n_changes);
 213
 214         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == _UNIT_FILE_STATE_INVALID);
 215
 216         log_error("link files2");
 217         changes = NULL;
 218         n_changes = 0;
 219
 220         r = unit_file_link(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, false, &changes, &n_changes);
 221         assert_se(r >= 0);
 222
 223         dump_changes(changes, n_changes);
 224         unit_file_changes_free(changes, n_changes);
 225
 226         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == UNIT_FILE_LINKED);
 227
 228         log_error("reenable files2");
 229         changes = NULL;
 230         n_changes = 0;
 231
 232         r = unit_file_reenable(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, false, &changes, &n_changes);
 233         assert_se(r >= 0);
 234
 235         dump_changes(changes, n_changes);
 236         unit_file_changes_free(changes, n_changes);
 237
 238         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == UNIT_FILE_ENABLED);
 239
 240         log_error("disable files2");
 241         changes = NULL;
 242         n_changes = 0;
 243
 244         r = unit_file_disable(UNIT_FILE_SYSTEM, false, NULL, (char**) files2, &changes, &n_changes);
 245         assert_se(r >= 0);
 246
 247         dump_changes(changes, n_changes);
 248         unit_file_changes_free(changes, n_changes);
 249
 250         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files2[0])) == _UNIT_FILE_STATE_INVALID);
 251         log_error("preset files");
 252         changes = NULL;
 253         n_changes = 0;
 254
 255         r = unit_file_preset(UNIT_FILE_SYSTEM, false, NULL, (char**) files, false, &changes, &n_changes);
 256         assert_se(r >= 0);
 257
 258         dump_changes(changes, n_changes);
 259         unit_file_changes_free(changes, n_changes);
 260
 261         assert_se(unit_file_get_state(UNIT_FILE_SYSTEM, NULL, file_name_from_path(files[0])) == UNIT_FILE_ENABLED);
 262
 263         return 0;
 264 }