added storage of extended ltdb header information
[vlendec/samba-autobuild/.git] / ctdb / common / ctdb_ltdb.c
1 /* 
2    ctdb ltdb code
3
4    Copyright (C) Andrew Tridgell  2006
5
6    This library is free software; you can redistribute it and/or
7    modify it under the terms of the GNU Lesser General Public
8    License as published by the Free Software Foundation; either
9    version 2 of the License, or (at your option) any later version.
10
11    This library is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14    Lesser General Public License for more details.
15
16    You should have received a copy of the GNU Lesser General Public
17    License along with this library; if not, write to the Free Software
18    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19 */
20
21 #include "includes.h"
22 #include "lib/events/events.h"
23 #include "system/network.h"
24 #include "system/filesys.h"
25 #include "ctdb_private.h"
26
27 /*
28   attach to a specific database
29 */
30 int ctdb_attach(struct ctdb_context *ctdb, const char *name, int tdb_flags, 
31                 int open_flags, mode_t mode)
32 {
33         /* when we have a separate daemon this will need to be a real
34            file, not a TDB_INTERNAL, so the parent can access it to
35            for ltdb bypass */
36         ctdb->ltdb = tdb_open(name, 0, /* tdb_flags */ TDB_INTERNAL, open_flags, mode);
37         if (ctdb->ltdb == NULL) {
38                 ctdb_set_error(ctdb, "Failed to open tdb %s\n", name);
39                 return -1;
40         }
41         return 0;
42 }
43
44
45 /*
46   construct an initial header for a record with no ltdb header yet
47 */
48 static void ltdb_initial_header(struct ctdb_context *ctdb, 
49                                 TDB_DATA key,
50                                 struct ctdb_ltdb_header *header)
51 {
52         header->rsn = 0;
53         header->dmaster = ctdb_hash(&key) % ctdb->num_nodes;
54         header->laccessor = header->dmaster;
55         header->lacount = 0;
56 }
57
58
59 /*
60   fetch a record from the ltdb, separating out the header information
61   and returning the body of the record. A valid (initial) header is
62   returned if the record is not present
63 */
64 int ctdb_ltdb_fetch(struct ctdb_context *ctdb, TALLOC_CTX *mem_ctx,
65                     TDB_DATA key, struct ctdb_ltdb_header *header, TDB_DATA *data)
66 {
67         TDB_DATA rec;
68
69         rec = tdb_fetch(ctdb->ltdb, key);
70         if (rec.dsize < sizeof(*header)) {
71                 /* return an initial header */
72                 ltdb_initial_header(ctdb, key, header);
73                 data->dptr = NULL;
74                 data->dsize = 0;
75                 return 0;
76         }
77
78         *header = *(struct ctdb_ltdb_header *)rec.dptr;
79
80         data->dsize = rec.dsize - sizeof(struct ctdb_ltdb_header);
81         data->dptr = talloc_memdup(mem_ctx, sizeof(struct ctdb_ltdb_header)+rec.dptr,
82                                    data->dsize);
83         CTDB_NO_MEMORY(ctdb, data->dptr);
84
85         return 0;
86 }
87
88
89 /*
90   fetch a record from the ltdb, separating out the header information
91   and returning the body of the record. A valid (initial) header is
92   returned if the record is not present
93 */
94 int ctdb_ltdb_store(struct ctdb_context *ctdb, TDB_DATA key, 
95                     struct ctdb_ltdb_header *header, TDB_DATA data)
96 {
97         TDB_DATA rec;
98         int ret;
99
100         rec.dsize = sizeof(struct ctdb_ltdb_header) + data.dsize;
101         rec.dptr = talloc_size(ctdb, rec.dsize);
102         CTDB_NO_MEMORY(ctdb, rec.dptr);
103
104         memcpy(rec.dptr, header, sizeof(*header));
105         memcpy(rec.dptr + sizeof(*header), data.dptr, data.dsize);
106         
107         ret = tdb_store(ctdb->ltdb, key, rec, TDB_REPLACE);
108         talloc_free(rec.dptr);
109
110         return ret;
111 }