tests/virlockspacetest.c

   1 /*
   2  * Copyright (C) 2011, 2013, 2014 Red Hat, Inc.
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library;  If not, see
  16  * <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <config.h>
  20
  21 #include <signal.h>
  22 #include <sys/stat.h>
  23
  24 #include "testutils.h"
  25 #include "virutil.h"
  26 #include "virerror.h"
  27 #include "viralloc.h"
  28 #include "virfile.h"
  29 #include "virlog.h"
  30
  31 #include "virlockspace.h"
  32
  33 #define VIR_FROM_THIS VIR_FROM_RPC
  34
  35 VIR_LOG_INIT("tests.lockspacetest");
  36
  37 #define LOCKSPACE_DIR abs_builddir "/virlockspacedata"
  38
  39 static int testLockSpaceCreate(const void *args ATTRIBUTE_UNUSED)
  40 {
  41     virLockSpacePtr lockspace;
  42     int ret = -1;
  43
  44     rmdir(LOCKSPACE_DIR);
  45
  46     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
  47         goto cleanup;
  48
  49     if (!virFileIsDir(LOCKSPACE_DIR))
  50         goto cleanup;
  51
  52     ret = 0;
  53
  54  cleanup:
  55     virLockSpaceFree(lockspace);
  56     rmdir(LOCKSPACE_DIR);
  57     return ret;
  58 }
  59
  60
  61 static int testLockSpaceResourceLifecycle(const void *args ATTRIBUTE_UNUSED)
  62 {
  63     virLockSpacePtr lockspace;
  64     int ret = -1;
  65
  66     rmdir(LOCKSPACE_DIR);
  67
  68     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
  69         goto cleanup;
  70
  71     if (!virFileIsDir(LOCKSPACE_DIR))
  72         goto cleanup;
  73
  74     if (virLockSpaceCreateResource(lockspace, "foo") < 0)
  75         goto cleanup;
  76
  77     if (!virFileExists(LOCKSPACE_DIR "/foo"))
  78         goto cleanup;
  79
  80     if (virLockSpaceDeleteResource(lockspace, "foo") < 0)
  81         goto cleanup;
  82
  83     if (virFileExists(LOCKSPACE_DIR "/foo"))
  84         goto cleanup;
  85
  86     ret = 0;
  87
  88  cleanup:
  89     virLockSpaceFree(lockspace);
  90     rmdir(LOCKSPACE_DIR);
  91     return ret;
  92 }
  93
  94
  95 static int testLockSpaceResourceLockExcl(const void *args ATTRIBUTE_UNUSED)
  96 {
  97     virLockSpacePtr lockspace;
  98     int ret = -1;
  99
 100     rmdir(LOCKSPACE_DIR);
 101
 102     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
 103         goto cleanup;
 104
 105     if (!virFileIsDir(LOCKSPACE_DIR))
 106         goto cleanup;
 107
 108     if (virLockSpaceCreateResource(lockspace, "foo") < 0)
 109         goto cleanup;
 110
 111     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(), 0) < 0)
 112         goto cleanup;
 113
 114     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 115         goto cleanup;
 116
 117     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(), 0) == 0)
 118         goto cleanup;
 119
 120     if (virLockSpaceDeleteResource(lockspace, "foo") == 0)
 121         goto cleanup;
 122
 123     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 124         goto cleanup;
 125
 126     if (virLockSpaceDeleteResource(lockspace, "foo") < 0)
 127         goto cleanup;
 128
 129     if (virFileExists(LOCKSPACE_DIR "/foo"))
 130         goto cleanup;
 131
 132     ret = 0;
 133
 134  cleanup:
 135     virLockSpaceFree(lockspace);
 136     rmdir(LOCKSPACE_DIR);
 137     return ret;
 138 }
 139
 140
 141 static int testLockSpaceResourceLockExclAuto(const void *args ATTRIBUTE_UNUSED)
 142 {
 143     virLockSpacePtr lockspace;
 144     int ret = -1;
 145
 146     rmdir(LOCKSPACE_DIR);
 147
 148     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
 149         goto cleanup;
 150
 151     if (!virFileIsDir(LOCKSPACE_DIR))
 152         goto cleanup;
 153
 154     if (virLockSpaceCreateResource(lockspace, "foo") < 0)
 155         goto cleanup;
 156
 157     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 158                                     VIR_LOCK_SPACE_ACQUIRE_AUTOCREATE) < 0)
 159         goto cleanup;
 160
 161     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 162         goto cleanup;
 163
 164     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 165         goto cleanup;
 166
 167     if (virFileExists(LOCKSPACE_DIR "/foo"))
 168         goto cleanup;
 169
 170     ret = 0;
 171
 172  cleanup:
 173     virLockSpaceFree(lockspace);
 174     rmdir(LOCKSPACE_DIR);
 175     return ret;
 176 }
 177
 178
 179 static int testLockSpaceResourceLockShr(const void *args ATTRIBUTE_UNUSED)
 180 {
 181     virLockSpacePtr lockspace;
 182     int ret = -1;
 183
 184     rmdir(LOCKSPACE_DIR);
 185
 186     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
 187         goto cleanup;
 188
 189     if (!virFileIsDir(LOCKSPACE_DIR))
 190         goto cleanup;
 191
 192     if (virLockSpaceCreateResource(lockspace, "foo") < 0)
 193         goto cleanup;
 194
 195     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 196                                     VIR_LOCK_SPACE_ACQUIRE_SHARED) < 0)
 197         goto cleanup;
 198
 199     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(), 0) == 0)
 200         goto cleanup;
 201
 202     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 203                                     VIR_LOCK_SPACE_ACQUIRE_SHARED) < 0)
 204         goto cleanup;
 205
 206     if (virLockSpaceDeleteResource(lockspace, "foo") == 0)
 207         goto cleanup;
 208
 209     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 210         goto cleanup;
 211
 212     if (virLockSpaceDeleteResource(lockspace, "foo") == 0)
 213         goto cleanup;
 214
 215     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 216         goto cleanup;
 217
 218     if (virLockSpaceDeleteResource(lockspace, "foo") < 0)
 219         goto cleanup;
 220
 221     if (virFileExists(LOCKSPACE_DIR "/foo"))
 222         goto cleanup;
 223
 224     ret = 0;
 225
 226  cleanup:
 227     virLockSpaceFree(lockspace);
 228     rmdir(LOCKSPACE_DIR);
 229     return ret;
 230 }
 231
 232
 233 static int testLockSpaceResourceLockShrAuto(const void *args ATTRIBUTE_UNUSED)
 234 {
 235     virLockSpacePtr lockspace;
 236     int ret = -1;
 237
 238     rmdir(LOCKSPACE_DIR);
 239
 240     if (!(lockspace = virLockSpaceNew(LOCKSPACE_DIR)))
 241         goto cleanup;
 242
 243     if (!virFileIsDir(LOCKSPACE_DIR))
 244         goto cleanup;
 245
 246     if (virLockSpaceCreateResource(lockspace, "foo") < 0)
 247         goto cleanup;
 248
 249     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 250                                     VIR_LOCK_SPACE_ACQUIRE_SHARED |
 251                                     VIR_LOCK_SPACE_ACQUIRE_AUTOCREATE) < 0)
 252         goto cleanup;
 253
 254     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 255         goto cleanup;
 256
 257     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 258                                     VIR_LOCK_SPACE_ACQUIRE_AUTOCREATE) == 0)
 259         goto cleanup;
 260
 261     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 262         goto cleanup;
 263
 264     if (virLockSpaceAcquireResource(lockspace, "foo", geteuid(),
 265                                     VIR_LOCK_SPACE_ACQUIRE_SHARED |
 266                                     VIR_LOCK_SPACE_ACQUIRE_AUTOCREATE) < 0)
 267         goto cleanup;
 268
 269     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 270         goto cleanup;
 271
 272     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 273         goto cleanup;
 274
 275     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 276         goto cleanup;
 277
 278     if (virLockSpaceReleaseResource(lockspace, "foo", geteuid()) < 0)
 279         goto cleanup;
 280
 281     if (virFileExists(LOCKSPACE_DIR "/foo"))
 282         goto cleanup;
 283
 284     ret = 0;
 285
 286  cleanup:
 287     virLockSpaceFree(lockspace);
 288     rmdir(LOCKSPACE_DIR);
 289     return ret;
 290 }
 291
 292
 293 static int testLockSpaceResourceLockPath(const void *args ATTRIBUTE_UNUSED)
 294 {
 295     virLockSpacePtr lockspace;
 296     int ret = -1;
 297
 298     rmdir(LOCKSPACE_DIR);
 299
 300     if (!(lockspace = virLockSpaceNew(NULL)))
 301         goto cleanup;
 302
 303     if (mkdir(LOCKSPACE_DIR, 0700) < 0)
 304         goto cleanup;
 305
 306     if (virLockSpaceCreateResource(lockspace, LOCKSPACE_DIR "/foo") < 0)
 307         goto cleanup;
 308
 309     if (virLockSpaceAcquireResource(lockspace, LOCKSPACE_DIR "/foo", geteuid(), 0) < 0)
 310         goto cleanup;
 311
 312     if (!virFileExists(LOCKSPACE_DIR "/foo"))
 313         goto cleanup;
 314
 315     if (virLockSpaceAcquireResource(lockspace, LOCKSPACE_DIR "/foo", geteuid(), 0) == 0)
 316         goto cleanup;
 317
 318     if (virLockSpaceDeleteResource(lockspace, LOCKSPACE_DIR "/foo") == 0)
 319         goto cleanup;
 320
 321     if (virLockSpaceReleaseResource(lockspace, LOCKSPACE_DIR "/foo", geteuid()) < 0)
 322         goto cleanup;
 323
 324     if (virLockSpaceDeleteResource(lockspace, LOCKSPACE_DIR "/foo") < 0)
 325         goto cleanup;
 326
 327     if (virFileExists(LOCKSPACE_DIR "/foo"))
 328         goto cleanup;
 329
 330     ret = 0;
 331
 332  cleanup:
 333     virLockSpaceFree(lockspace);
 334     rmdir(LOCKSPACE_DIR);
 335     return ret;
 336 }
 337
 338
 339
 340 static int
 341 mymain(void)
 342 {
 343     int ret = 0;
 344
 345     signal(SIGPIPE, SIG_IGN);
 346
 347     if (virTestRun("Lockspace creation", testLockSpaceCreate, NULL) < 0)
 348         ret = -1;
 349
 350     if (virTestRun("Lockspace res lifecycle", testLockSpaceResourceLifecycle, NULL) < 0)
 351         ret = -1;
 352
 353     if (virTestRun("Lockspace res lock excl", testLockSpaceResourceLockExcl, NULL) < 0)
 354         ret = -1;
 355
 356     if (virTestRun("Lockspace res lock shr", testLockSpaceResourceLockShr, NULL) < 0)
 357         ret = -1;
 358
 359     if (virTestRun("Lockspace res lock excl auto", testLockSpaceResourceLockExclAuto, NULL) < 0)
 360         ret = -1;
 361
 362     if (virTestRun("Lockspace res lock shr auto", testLockSpaceResourceLockShrAuto, NULL) < 0)
 363         ret = -1;
 364
 365     if (virTestRun("Lockspace res full path", testLockSpaceResourceLockPath, NULL) < 0)
 366         ret = -1;
 367
 368     return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
 369 }
 370
 371 VIR_TEST_MAIN(mymain)