1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
|
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <sys/wait.h>
#include "zdtmtst.h"
const char *test_doc = "Open r/w and unlink file, and fork before migration;\n"
"check that the child can write to it and the parent\n"
"can read from it after migration";
const char *test_author = "Roman Kagan <rkagan@parallels.com>";
char *filename;
TEST_OPTION(filename, string, "file name", 1);
int main(int argc, char **argv)
{
int fd, child_fd, ret;
pid_t pid;
uint32_t crc;
uint8_t buf[1000000];
task_waiter_t t;
test_init(argc, argv);
fd = open(filename, O_RDWR | O_CREAT | O_TRUNC, 0644);
if (fd < 0) {
pr_perror("can't open %s", filename);
exit(1);
}
child_fd = open(filename, O_RDWR | O_CREAT | O_TRUNC, 0644);
if (child_fd < 0) {
pr_perror("can't open %s", filename);
exit(1);
}
if (unlink(filename)) {
pr_perror("can't unlink %s", filename);
exit(1);
}
task_waiter_init(&t);
pid = fork();
if (pid < 0) {
pr_perror("can't fork");
exit(1);
}
if (pid == 0) { /* child writes to the unlinked file and returns */
close(fd);
task_waiter_complete_current(&t);
test_waitsig();
crc = ~0;
datagen(buf, sizeof(buf), &crc);
if (write(child_fd, buf, sizeof(buf)) != sizeof(buf))
_exit(errno);
close(child_fd);
_exit(0);
} else
task_waiter_wait4(&t, pid);
close(child_fd);
test_daemon();
test_waitsig();
if (kill(pid, SIGTERM)) {
fail("terminating the child failed");
goto out;
}
if (wait(&ret) != pid) {
fail("wait() returned wrong pid %d", pid);
goto out;
}
if (WIFEXITED(ret)) {
ret = WEXITSTATUS(ret);
if (ret) {
fail("child exited with nonzero code %d (%s)", ret, strerror(ret));
goto out;
}
}
if (WIFSIGNALED(ret)) {
fail("child exited on unexpected signal %d", WTERMSIG(ret));
goto out;
}
if (lseek(fd, 0, SEEK_SET) < 0) {
fail("lseeking to the beginning of file failed");
goto out;
}
if (read(fd, buf, sizeof(buf)) != sizeof(buf)) {
fail("can't read %s", filename);
goto out;
}
crc = ~0;
if (datachk(buf, sizeof(buf), &crc)) {
fail("CRC mismatch");
goto out;
}
if (close(fd)) {
fail("close failed");
goto out_noclose;
}
if (unlink(filename) != -1 || errno != ENOENT) {
fail("file %s should have been deleted before migration: unlink", filename);
goto out_noclose;
}
pass();
out:
close(fd);
out_noclose:
return 0;
}
|