4 Copyright (C) Andrew Tridgell 2004
6 ** NOTE! The following LGPL license applies to the ldb
7 ** library. This does NOT imply that all of Samba is released
10 This library is free software; you can redistribute it and/or
11 modify it under the terms of the GNU Lesser General Public
12 License as published by the Free Software Foundation; either
13 version 3 of the License, or (at your option) any later version.
15 This library is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 Lesser General Public License for more details.
20 You should have received a copy of the GNU Lesser General Public
21 License along with this library; if not, see <http://www.gnu.org/licenses/>.
27 * Component: ldb pack/unpack
29 * Description: pack/unpack routines for ldb messages as key/value blobs
31 * Author: Andrew Tridgell
35 #include "ldb/include/includes.h"
37 #include "ldb/ldb_tdb/ldb_tdb.h"
39 /* change this if the data format ever changes */
40 #define LTDB_PACKING_FORMAT 0x26011967
42 /* old packing formats */
43 #define LTDB_PACKING_FORMAT_NODN 0x26011966
45 /* use a portable integer format */
46 static void put_uint32(uint8_t *p
, int ofs
, unsigned int val
)
50 p
[1] = (val
>>8) & 0xFF;
51 p
[2] = (val
>>16) & 0xFF;
52 p
[3] = (val
>>24) & 0xFF;
55 static unsigned int pull_uint32(uint8_t *p
, int ofs
)
58 return p
[0] | (p
[1]<<8) | (p
[2]<<16) | (p
[3]<<24);
61 static int attribute_storable_values(const struct ldb_message_element
*el
)
63 if (el
->num_values
== 0) return 0;
65 if (ldb_attr_cmp(el
->name
, "dn") == 0) return 0;
67 if (ldb_attr_cmp(el
->name
, "distinguishedName") == 0) return 0;
69 return el
->num_values
;
73 pack a ldb message into a linear buffer in a TDB_DATA
75 note that this routine avoids saving elements with zero values,
76 as these are equivalent to having no element
78 caller frees the data buffer after use
80 int ltdb_pack_data(struct ldb_module
*module
,
81 const struct ldb_message
*message
,
82 struct TDB_DATA
*data
)
84 struct ldb_context
*ldb
= module
->ldb
;
85 unsigned int i
, j
, real_elements
=0;
91 dn
= ldb_dn_linearize(ldb
, message
->dn
);
97 /* work out how big it needs to be */
100 size
+= 1 + strlen(dn
);
102 for (i
=0;i
<message
->num_elements
;i
++) {
103 if (attribute_storable_values(&message
->elements
[i
]) == 0) {
109 size
+= 1 + strlen(message
->elements
[i
].name
) + 4;
110 for (j
=0;j
<message
->elements
[i
].num_values
;j
++) {
111 size
+= 4 + message
->elements
[i
].values
[j
].length
+ 1;
116 data
->dptr
= talloc_array(ldb
, uint8_t, size
);
124 p
= (uint8_t *)data
->dptr
;
125 put_uint32(p
, 0, LTDB_PACKING_FORMAT
);
126 put_uint32(p
, 4, real_elements
);
129 /* the dn needs to be packed so we can be case preserving
130 while hashing on a case folded dn */
132 memcpy(p
, dn
, len
+1);
135 for (i
=0;i
<message
->num_elements
;i
++) {
136 if (attribute_storable_values(&message
->elements
[i
]) == 0) {
139 len
= strlen(message
->elements
[i
].name
);
140 memcpy(p
, message
->elements
[i
].name
, len
+1);
142 put_uint32(p
, 0, message
->elements
[i
].num_values
);
144 for (j
=0;j
<message
->elements
[i
].num_values
;j
++) {
145 put_uint32(p
, 0, message
->elements
[i
].values
[j
].length
);
146 memcpy(p
+4, message
->elements
[i
].values
[j
].data
,
147 message
->elements
[i
].values
[j
].length
);
148 p
[4+message
->elements
[i
].values
[j
].length
] = 0;
149 p
+= 4 + message
->elements
[i
].values
[j
].length
+ 1;
158 unpack a ldb message from a linear buffer in TDB_DATA
160 Free with ltdb_unpack_data_free()
162 int ltdb_unpack_data(struct ldb_module
*module
,
163 const struct TDB_DATA
*data
,
164 struct ldb_message
*message
)
166 struct ldb_context
*ldb
= module
->ldb
;
168 unsigned int remaining
;
173 message
->elements
= NULL
;
175 p
= (uint8_t *)data
->dptr
;
176 if (data
->dsize
< 8) {
181 format
= pull_uint32(p
, 0);
182 message
->num_elements
= pull_uint32(p
, 4);
185 remaining
= data
->dsize
- 8;
188 case LTDB_PACKING_FORMAT_NODN
:
192 case LTDB_PACKING_FORMAT
:
193 len
= strnlen((char *)p
, remaining
);
194 if (len
== remaining
) {
198 message
->dn
= ldb_dn_explode(message
, (char *)p
);
199 if (message
->dn
== NULL
) {
203 remaining
-= len
+ 1;
212 if (message
->num_elements
== 0) {
213 message
->elements
= NULL
;
217 if (message
->num_elements
> remaining
/ 6) {
222 message
->elements
= talloc_array(message
, struct ldb_message_element
, message
->num_elements
);
223 if (!message
->elements
) {
228 memset(message
->elements
, 0,
229 message
->num_elements
* sizeof(struct ldb_message_element
));
231 for (i
=0;i
<message
->num_elements
;i
++) {
232 if (remaining
< 10) {
236 len
= strnlen((char *)p
, remaining
-6);
237 if (len
== remaining
-6) {
241 message
->elements
[i
].flags
= 0;
242 message
->elements
[i
].name
= talloc_strndup(message
->elements
, (char *)p
, len
);
243 if (message
->elements
[i
].name
== NULL
) {
247 remaining
-= len
+ 1;
249 message
->elements
[i
].num_values
= pull_uint32(p
, 0);
250 message
->elements
[i
].values
= NULL
;
251 if (message
->elements
[i
].num_values
!= 0) {
252 message
->elements
[i
].values
= talloc_array(message
->elements
,
254 message
->elements
[i
].num_values
);
255 if (!message
->elements
[i
].values
) {
262 for (j
=0;j
<message
->elements
[i
].num_values
;j
++) {
263 len
= pull_uint32(p
, 0);
264 if (len
> remaining
-5) {
269 message
->elements
[i
].values
[j
].length
= len
;
270 message
->elements
[i
].values
[j
].data
= (uint8_t *)talloc_size(message
->elements
[i
].values
, len
+1);
271 if (message
->elements
[i
].values
[j
].data
== NULL
) {
275 memcpy(message
->elements
[i
].values
[j
].data
, p
+4, len
);
276 message
->elements
[i
].values
[j
].data
[len
] = 0;
278 remaining
-= len
+4+1;
283 if (remaining
!= 0) {
284 ldb_debug(ldb
, LDB_DEBUG_ERROR
,
285 "Error: %d bytes unread in ltdb_unpack_data\n", remaining
);
291 talloc_free(message
->elements
);