criu/test/zdtm/static/file_locks02.c
Pavel Emelyanov 2e13f1f029 test: Get rid of live directory
Move static and transition into zdtm top. We can't move all the micro
tests themselves, as we need to distinguish static from non static (zdtm.py
makes additional checks on static ones).

Signed-off-by: Pavel Emelyanov <xemul@virtuozzo.com>
2016-02-20 13:40:52 +03:00

99 lines
1.7 KiB
C

#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/file.h>
#include <string.h>
#include <sys/wait.h>
#include "zdtmtst.h"
const char *test_doc = "Check that 'shared' flocks work";
const char *test_author = "Pavel Emelyanov <xemul@parallels.com>";
char *filename;
TEST_OPTION(filename, string, "file name", 1);
static int check_file_locks(pid_t child)
{
FILE *fp_locks = NULL;
char buf[100], fl_flag[16], fl_type[16], fl_option[16];
pid_t pid = getpid();
int found = 0, num, fl_owner;
fp_locks = fopen("/proc/locks", "r");
if (!fp_locks)
return -1;
test_msg("C: %d\n", pid);
while (fgets(buf, sizeof(buf), fp_locks)) {
test_msg("c: %s", buf);
if (strstr(buf, "->"))
continue;
num = sscanf(buf,
"%*d:%s %s %s %d %*02x:%*02x:%*d %*d %*s",
fl_flag, fl_type, fl_option, &fl_owner);
if (num < 4) {
pr_perror("Invalid lock info.");
break;
}
if (fl_owner != pid && fl_owner != child)
continue;
if (!strcmp(fl_flag, "FLOCK") &&
!strcmp(fl_type, "ADVISORY") &&
!strcmp(fl_option, "WRITE"))
found++;
memset(fl_flag, 0, sizeof(fl_flag));
memset(fl_type, 0, sizeof(fl_type));
memset(fl_option, 0, sizeof(fl_option));
}
fclose(fp_locks);
return found == 1;
}
int main(int argc, char **argv)
{
int fd, pid;
test_init(argc, argv);
fd = open(filename, O_CREAT | O_RDWR, 0600);
if (fd < 0) {
pr_perror("No file");
return -1;
}
flock(fd, LOCK_EX);
pid = fork();
if (pid == 0) {
test_waitsig();
exit(0);
}
test_daemon();
test_waitsig();
if (check_file_locks(pid))
pass();
else
fail("Flock file locks check failed");
kill(pid, SIGTERM);
waitpid(pid, NULL, 0);
close(fd);
unlink(filename);
return 0;
}