tdb: Fix a "increases alignment" warning
[samba.git] / lib / tdb / test / run-transaction-expand.c
1 #include "../common/tdb_private.h"
2
3 /* Speed up the tests, but do the actual sync tests. */
4 static unsigned int sync_counts = 0;
5 static inline int fake_fsync(int fd)
6 {
7         sync_counts++;
8         return 0;
9 }
10 #define fsync fake_fsync
11
12 #ifdef MS_SYNC
13 static inline int fake_msync(void *addr, size_t length, int flags)
14 {
15         sync_counts++;
16         return 0;
17 }
18 #define msync fake_msync
19 #endif
20
21 #ifdef HAVE_FDATASYNC
22 static inline int fake_fdatasync(int fd)
23 {
24         sync_counts++;
25         return 0;
26 }
27 #define fdatasync fake_fdatasync
28 #endif
29
30 #include "../common/io.c"
31 #include "../common/tdb.c"
32 #include "../common/lock.c"
33 #include "../common/freelist.c"
34 #include "../common/traverse.c"
35 #include "../common/transaction.c"
36 #include "../common/error.c"
37 #include "../common/open.c"
38 #include "../common/check.c"
39 #include "../common/hash.c"
40 #include "../common/mutex.c"
41 #include "tap-interface.h"
42 #include <stdlib.h>
43 #include "logging.h"
44
45 static void write_record(struct tdb_context *tdb, size_t extra_len,
46                          TDB_DATA *data)
47 {
48         TDB_DATA key;
49         key.dsize = strlen("hi");
50         key.dptr = discard_const_p(uint8_t, "hi");
51
52         data->dsize += extra_len;
53         tdb_transaction_start(tdb);
54         tdb_store(tdb, key, *data, TDB_REPLACE);
55         tdb_transaction_commit(tdb);
56 }
57
58 int main(int argc, char *argv[])
59 {
60         struct tdb_context *tdb;
61         size_t i;
62         TDB_DATA data;
63         struct tdb_record rec;
64         tdb_off_t off;
65
66         /* Do *not* suppress sync for this test; we do it ourselves. */
67         unsetenv("TDB_NO_FSYNC");
68
69         plan_tests(5);
70         tdb = tdb_open_ex("run-transaction-expand.tdb",
71                           1024, TDB_CLEAR_IF_FIRST,
72                           O_CREAT|O_TRUNC|O_RDWR, 0600, &taplogctx, NULL);
73         ok1(tdb);
74
75         data.dsize = 0;
76         data.dptr = calloc(1000, getpagesize());
77         if (data.dptr == NULL) {
78                 diag("Unable to allocate memory for data.dptr");
79                 tdb_close(tdb);
80                 exit(1);
81         }
82
83         /* Simulate a slowly growing record. */
84         for (i = 0; i < 1000; i++)
85                 write_record(tdb, getpagesize(), &data);
86
87         tdb_ofs_read(tdb, TDB_RECOVERY_HEAD, &off);
88         tdb_read(tdb, off, &rec, sizeof(rec), DOCONV());
89         diag("TDB size = %zu, recovery = %llu-%llu",
90              (size_t)tdb->map_size, (unsigned long long)off, (unsigned long long)(off + sizeof(rec) + rec.rec_len));
91
92         /* We should only be about 5 times larger than largest record. */
93         ok1(tdb->map_size < 6 * i * getpagesize());
94         tdb_close(tdb);
95
96         tdb = tdb_open_ex("run-transaction-expand.tdb",
97                           1024, TDB_CLEAR_IF_FIRST,
98                           O_CREAT|O_TRUNC|O_RDWR, 0600, &taplogctx, NULL);
99         ok1(tdb);
100
101         data.dsize = 0;
102
103         /* Simulate a slowly growing record, repacking to keep
104          * recovery area at end. */
105         for (i = 0; i < 1000; i++) {
106                 write_record(tdb, getpagesize(), &data);
107                 if (i % 10 == 0)
108                         tdb_repack(tdb);
109         }
110
111         tdb_ofs_read(tdb, TDB_RECOVERY_HEAD, &off);
112         tdb_read(tdb, off, &rec, sizeof(rec), DOCONV());
113         diag("TDB size = %zu, recovery = %llu-%llu",
114              (size_t)tdb->map_size, (unsigned long long)off, (unsigned long long)(off + sizeof(rec) + rec.rec_len));
115
116         /* We should only be about 4 times larger than largest record. */
117         ok1(tdb->map_size < 5 * i * getpagesize());
118
119         /* We should have synchronized multiple times. */
120         ok1(sync_counts);
121         tdb_close(tdb);
122         free(data.dptr);
123
124         return exit_status();
125 }