1 #include "private.h" // for ntdb_fcntl_unlock
3 #include "tap-interface.h"
10 static int mylock(int fd, int rw, off_t off, off_t len, bool waitflag,
24 fl.l_whence = SEEK_SET;
29 ret = fcntl(fd, F_SETLKW, &fl);
31 ret = fcntl(fd, F_SETLK, &fl);
32 } while (ret != 0 && errno == EINTR);
38 static int trav(struct ntdb_context *ntdb, NTDB_DATA k, NTDB_DATA d, int *terr)
44 int main(int argc, char *argv[])
47 struct ntdb_context *ntdb;
48 int flags[] = { NTDB_DEFAULT, NTDB_NOMMAP,
49 NTDB_CONVERT, NTDB_NOMMAP|NTDB_CONVERT };
50 union ntdb_attribute lock_attr;
51 NTDB_DATA key = ntdb_mkdata("key", 3);
52 NTDB_DATA data = ntdb_mkdata("data", 4);
55 lock_attr.base.attr = NTDB_ATTRIBUTE_FLOCK;
56 lock_attr.base.next = &tap_log_attr;
57 lock_attr.flock.lock = mylock;
58 lock_attr.flock.unlock = ntdb_fcntl_unlock;
59 lock_attr.flock.data = &lock_err;
61 plan_tests(sizeof(flags) / sizeof(flags[0]) * 81);
63 for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
66 /* Nonblocking open; expect no error message. */
68 ntdb = ntdb_open("run-82-lockattr.ntdb", flags[i]|MAYBE_NOSYNC,
69 O_RDWR|O_CREAT|O_TRUNC, 0600, &lock_attr);
70 ok(errno == lock_err, "Errno is %u", errno);
72 ok1(tap_log_messages == 0);
75 ntdb = ntdb_open("run-82-lockattr.ntdb", flags[i]|MAYBE_NOSYNC,
76 O_RDWR|O_CREAT|O_TRUNC, 0600, &lock_attr);
77 ok(errno == lock_err, "Errno is %u", errno);
79 ok1(tap_log_messages == 0);
81 /* Forced fail open. */
83 ntdb = ntdb_open("run-82-lockattr.ntdb", flags[i]|MAYBE_NOSYNC,
84 O_RDWR|O_CREAT|O_TRUNC, 0600, &lock_attr);
85 ok1(errno == lock_err);
87 ok1(tap_log_messages == 1);
91 ntdb = ntdb_open("run-82-lockattr.ntdb", flags[i]|MAYBE_NOSYNC,
92 O_RDWR|O_CREAT|O_TRUNC, 0600, &lock_attr);
95 ok1(tap_log_messages == 0);
97 /* Nonblocking store. */
99 ok1(ntdb_store(ntdb, key, data, NTDB_REPLACE) == NTDB_ERR_LOCK);
100 ok1(tap_log_messages == 0);
102 ok1(ntdb_store(ntdb, key, data, NTDB_REPLACE) == NTDB_ERR_LOCK);
103 ok1(tap_log_messages == 0);
105 ok1(ntdb_store(ntdb, key, data, NTDB_REPLACE) == NTDB_ERR_LOCK);
106 ok1(tap_log_messages == 1);
107 tap_log_messages = 0;
109 /* Nonblocking fetch. */
111 ok1(!ntdb_exists(ntdb, key));
112 ok1(tap_log_messages == 0);
114 ok1(!ntdb_exists(ntdb, key));
115 ok1(tap_log_messages == 0);
117 ok1(!ntdb_exists(ntdb, key));
118 ok1(tap_log_messages == 1);
119 tap_log_messages = 0;
122 ok1(ntdb_fetch(ntdb, key, &d) == NTDB_ERR_LOCK);
123 ok1(tap_log_messages == 0);
125 ok1(ntdb_fetch(ntdb, key, &d) == NTDB_ERR_LOCK);
126 ok1(tap_log_messages == 0);
128 ok1(ntdb_fetch(ntdb, key, &d) == NTDB_ERR_LOCK);
129 ok1(tap_log_messages == 1);
130 tap_log_messages = 0;
132 /* Nonblocking delete. */
134 ok1(ntdb_delete(ntdb, key) == NTDB_ERR_LOCK);
135 ok1(tap_log_messages == 0);
137 ok1(ntdb_delete(ntdb, key) == NTDB_ERR_LOCK);
138 ok1(tap_log_messages == 0);
140 ok1(ntdb_delete(ntdb, key) == NTDB_ERR_LOCK);
141 ok1(tap_log_messages == 1);
142 tap_log_messages = 0;
144 /* Nonblocking locks. */
146 ok1(ntdb_chainlock(ntdb, key) == NTDB_ERR_LOCK);
147 ok1(tap_log_messages == 0);
149 ok1(ntdb_chainlock(ntdb, key) == NTDB_ERR_LOCK);
150 ok1(tap_log_messages == 0);
152 ok1(ntdb_chainlock(ntdb, key) == NTDB_ERR_LOCK);
153 ok1(tap_log_messages == 1);
154 tap_log_messages = 0;
157 ok1(ntdb_chainlock_read(ntdb, key) == NTDB_ERR_LOCK);
158 ok1(tap_log_messages == 0);
160 ok1(ntdb_chainlock_read(ntdb, key) == NTDB_ERR_LOCK);
161 ok1(tap_log_messages == 0);
163 ok1(ntdb_chainlock_read(ntdb, key) == NTDB_ERR_LOCK);
164 ok1(tap_log_messages == 1);
165 tap_log_messages = 0;
168 ok1(ntdb_lockall(ntdb) == NTDB_ERR_LOCK);
169 ok1(tap_log_messages == 0);
171 ok1(ntdb_lockall(ntdb) == NTDB_ERR_LOCK);
172 ok1(tap_log_messages == 0);
174 ok1(ntdb_lockall(ntdb) == NTDB_ERR_LOCK);
175 /* This actually does divide and conquer. */
176 ok1(tap_log_messages > 0);
177 tap_log_messages = 0;
180 ok1(ntdb_lockall_read(ntdb) == NTDB_ERR_LOCK);
181 ok1(tap_log_messages == 0);
183 ok1(ntdb_lockall_read(ntdb) == NTDB_ERR_LOCK);
184 ok1(tap_log_messages == 0);
186 ok1(ntdb_lockall_read(ntdb) == NTDB_ERR_LOCK);
187 ok1(tap_log_messages > 0);
188 tap_log_messages = 0;
190 /* Nonblocking traverse; go nonblock partway through. */
192 ok1(ntdb_store(ntdb, key, data, NTDB_REPLACE) == 0);
193 /* Need two entries to ensure two lock attempts! */
194 ok1(ntdb_store(ntdb, ntdb_mkdata("key2", 4), data,
197 ok1(ntdb_traverse(ntdb, trav, &lock_err) == NTDB_ERR_LOCK);
198 ok1(tap_log_messages == 0);
201 ok1(ntdb_traverse(ntdb, trav, &lock_err) == NTDB_ERR_LOCK);
202 ok1(tap_log_messages == 0);
205 ok1(ntdb_traverse(ntdb, trav, &lock_err) == NTDB_ERR_LOCK);
206 ok1(tap_log_messages == 1);
207 tap_log_messages = 0;
209 /* Nonblocking transactions. */
211 ok1(ntdb_transaction_start(ntdb) == NTDB_ERR_LOCK);
212 ok1(tap_log_messages == 0);
214 ok1(ntdb_transaction_start(ntdb) == NTDB_ERR_LOCK);
215 ok1(tap_log_messages == 0);
217 ok1(ntdb_transaction_start(ntdb) == NTDB_ERR_LOCK);
218 ok1(tap_log_messages == 1);
219 tap_log_messages = 0;
221 /* Nonblocking transaction prepare. */
223 ok1(ntdb_transaction_start(ntdb) == 0);
224 ok1(ntdb_delete(ntdb, key) == 0);
227 ok1(ntdb_transaction_prepare_commit(ntdb) == NTDB_ERR_LOCK);
228 ok1(tap_log_messages == 0);
231 ok1(ntdb_transaction_prepare_commit(ntdb) == 0);
232 ok1(ntdb_transaction_commit(ntdb) == 0);
234 /* And the transaction was committed, right? */
235 ok1(!ntdb_exists(ntdb, key));
237 ok1(tap_log_messages == 0);
239 return exit_status();