lustre/tests/flock_test.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <errno.h>
  40 #include <fcntl.h>
  41 #include <unistd.h>
  42 #include <string.h>
  43 #include <sys/types.h>
  44 #include <sys/stat.h>
  45 #include <sys/file.h>
  46 #include <sys/wait.h>
  47
  48 void chd_lock_unlock(int);
  49 char fname[1024];
  50
  51 int main(int argc, char **argv)
  52 {
  53     pid_t pid;
  54     int cfd, fd, rc;
  55
  56     if (argc != 2) {
  57         fprintf(stderr, "\nUSAGE: flock_test filepath\n");
  58         exit(2);
  59     }
  60     strncpy(fname, argv[1], 1023);
  61     fname[1023] ='\0';
  62     fd = open(fname, O_RDWR|O_CREAT, (mode_t)0666);
  63     if (fd == -1) {
  64         fprintf(stderr, "flock_test: failed to open %s : ", fname);
  65         perror("");
  66         exit(1);
  67     }
  68     if (flock(fd, LOCK_EX | LOCK_NB) == -1) {
  69         fprintf(stderr, "flock_test: parent attempt to lock %s failed : ", \
  70             fname);
  71         perror("");
  72         exit(1);
  73     }
  74
  75     pid = fork();
  76     if (pid == -1) {
  77         fprintf(stderr, "flock_test: fork failed : ");
  78         perror("");
  79         exit(1);
  80     }
  81
  82     if (pid == 0) {
  83         pid = getpid();
  84         sleep(2);
  85         if ((cfd = open(fname, O_RDWR)) == -1) {
  86             fprintf(stderr, "flock_test child (%d) cannot open %s: ", \
  87                 pid, fname);
  88             perror("");
  89             exit(1);
  90         }
  91         if(flock(cfd, LOCK_EX | LOCK_NB) != -1) {
  92             fprintf(stderr, "flock_test child (%d): %s not yet locked  : ", \
  93                 pid, fname);
  94             exit(1);
  95         }
  96         if(flock(fd, LOCK_UN) == -1) {
  97             fprintf(stderr, "flock_test child (%d): cannot unlock %s: ", \
  98                 pid, fname);
  99             perror("");
 100             exit(1);
 101         }
 102         if(flock(cfd, LOCK_EX | LOCK_NB) == -1 ) {
 103             fprintf(stderr, \
 104                 "flock_test: child (%d) cannot re-lock %s after unlocking : ", \
 105                 pid, fname);
 106             perror("");
 107             exit(1);
 108         }
 109         close(cfd);
 110         exit(0);
 111     }
 112
 113     waitpid(pid, &rc, 0);
 114     close(fd);
 115     unlink(fname);
 116     if (WIFEXITED(rc) && WEXITSTATUS(rc) != 0) {
 117         fprintf(stderr, "flock_test: child (%d) exit code = %d\n", \
 118             pid, WEXITSTATUS(rc));
 119         exit(1);
 120     }
 121     exit(0);
 122 }