123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- #include "util/interprocess_sync_file.h"
- #include <gtest/gtest.h>
- #include <unistd.h>
- using namespace std;
- namespace isc {
- namespace util {
- namespace {
- unsigned char
- parentReadLockedState (int fd) {
- unsigned char locked = 0xff;
- fd_set rfds;
- FD_ZERO(&rfds);
- FD_SET(fd, &rfds);
-
-
-
-
-
- struct timeval tv;
- tv.tv_sec = 5;
- tv.tv_usec = 0;
- const int nfds = select(fd + 1, &rfds, NULL, NULL, &tv);
- EXPECT_EQ(1, nfds);
- if (nfds == 1) {
-
- read(fd, &locked, sizeof(locked));
- }
- return (locked);
- }
- TEST(InterprocessSyncFileTest, TestLock) {
- InterprocessSyncFile sync("test");
- InterprocessSyncLocker locker(sync);
- EXPECT_FALSE(locker.isLocked());
- EXPECT_TRUE(locker.lock());
- EXPECT_TRUE(locker.isLocked());
- int fds[2];
-
-
-
-
-
-
- pipe(fds);
- if (fork() == 0) {
- unsigned char locked = 0;
-
- close(fds[0]);
- InterprocessSyncFile sync2("test");
- InterprocessSyncLocker locker2(sync2);
- if (!locker2.tryLock()) {
- EXPECT_FALSE(locker2.isLocked());
- locked = 1;
- } else {
- EXPECT_TRUE(locker2.isLocked());
- }
- write(fds[1], &locked, sizeof(locked));
- close(fds[1]);
- exit(0);
- } else {
-
- close(fds[1]);
- const unsigned char locked = parentReadLockedState(fds[0]);
- close(fds[0]);
- EXPECT_EQ(1, locked);
- }
- EXPECT_TRUE(locker.unlock());
- EXPECT_FALSE(locker.isLocked());
- EXPECT_EQ (0, unlink(TEST_DATA_TOPBUILDDIR "/test_lockfile"));
- }
- TEST(InterprocessSyncFileTest, TestMultipleFilesDirect) {
- InterprocessSyncFile sync("test1");
- InterprocessSyncLocker locker(sync);
- EXPECT_TRUE(locker.lock());
- InterprocessSyncFile sync2("test2");
- InterprocessSyncLocker locker2(sync2);
- EXPECT_TRUE(locker2.lock());
- EXPECT_TRUE(locker2.unlock());
- EXPECT_TRUE(locker.unlock());
- EXPECT_EQ (0, unlink(TEST_DATA_TOPBUILDDIR "/test1_lockfile"));
- EXPECT_EQ (0, unlink(TEST_DATA_TOPBUILDDIR "/test2_lockfile"));
- }
- TEST(InterprocessSyncFileTest, TestMultipleFilesForked) {
- InterprocessSyncFile sync("test1");
- InterprocessSyncLocker locker(sync);
- EXPECT_TRUE(locker.lock());
- int fds[2];
- pipe(fds);
- if (fork() == 0) {
- unsigned char locked = 0xff;
-
- close(fds[0]);
- InterprocessSyncFile sync2("test2");
- InterprocessSyncLocker locker2(sync2);
- if (locker2.tryLock()) {
- locked = 0;
- }
- write(fds[1], &locked, sizeof(locked));
- close(fds[1]);
- exit(0);
- } else {
-
- close(fds[1]);
- const unsigned char locked = parentReadLockedState(fds[0]);
- close(fds[0]);
- EXPECT_EQ(0, locked);
- }
- EXPECT_TRUE(locker.unlock());
- EXPECT_EQ (0, unlink(TEST_DATA_TOPBUILDDIR "/test1_lockfile"));
- EXPECT_EQ (0, unlink(TEST_DATA_TOPBUILDDIR "/test2_lockfile"));
- }
- }
- }
- }
|