mirror of
https://github.com/samba-team/samba.git
synced 2025-02-26 21:57:41 +03:00
s3-mdssvc: dalloc: dynamic object store based on talloc
dalloc is a hack with a bizarre API, but it does its job: it's a simple object store that allows for storing simple and complex data types. We'll use it for storing Spotlight query data. Signed-off-by: Ralph Boehme <slow@samba.org> Reviewed-by: Stefan Metzmacher <metze@samba.org>
This commit is contained in:
Ralph Boehme
Ralph Böhme
parent
25f56f5f40
commit
222e863d4e
230
source3/rpc_server/mdssvc/dalloc.c
Normal file
230
source3/rpc_server/mdssvc/dalloc.c
Normal file
@ -0,0 +1,230 @@
|
||||
/*
|
||||
Copyright (c) Ralph Boehme 2012-2014
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "replace.h"
|
||||
#include <talloc.h>
|
||||
#include "dalloc.h"
|
||||
|
||||
/**
|
||||
* Dynamic Datastore
|
||||
**/
|
||||
struct dalloc_ctx {
|
||||
void **dd_talloc_array;
|
||||
};
|
||||
|
||||
void *_dalloc_new(TALLOC_CTX *mem_ctx, const char *type)
|
||||
{
|
||||
void *p;
|
||||
|
||||
p = talloc_zero(mem_ctx, DALLOC_CTX);
|
||||
if (p == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
talloc_set_name_const(p, type);
|
||||
|
||||
return p;
|
||||
}
|
||||
|
||||
int _dalloc_add_talloc_chunk(DALLOC_CTX *dd, void *obj, const char *type, size_t size)
|
||||
{
|
||||
size_t array_len = talloc_array_length(dd->dd_talloc_array);
|
||||
|
||||
dd->dd_talloc_array = talloc_realloc(dd,
|
||||
dd->dd_talloc_array,
|
||||
void *,
|
||||
array_len + 1);
|
||||
if (dd->dd_talloc_array == NULL) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (size != 0) {
|
||||
void *p;
|
||||
|
||||
p = talloc_named_const(dd->dd_talloc_array, size, type);
|
||||
if (p == NULL) {
|
||||
return -1;
|
||||
}
|
||||
memcpy(p, obj, size);
|
||||
obj = p;
|
||||
} else {
|
||||
_talloc_get_type_abort(obj, type, __location__);
|
||||
}
|
||||
|
||||
dd->dd_talloc_array[array_len] = obj;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Get number of elements, returns 0 if the structure is empty or not initialized */
|
||||
size_t dalloc_size(const DALLOC_CTX *d)
|
||||
{
|
||||
if (d == NULL) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (d->dd_talloc_array == NULL) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
return talloc_array_length(d->dd_talloc_array);
|
||||
}
|
||||
|
||||
/* Return element at position */
|
||||
void *dalloc_get_object(const DALLOC_CTX *d, int i)
|
||||
{
|
||||
size_t size = dalloc_size(d);
|
||||
|
||||
if (i >= size) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return d->dd_talloc_array[i];
|
||||
}
|
||||
|
||||
/* Return typename of element at position */
|
||||
const char *dalloc_get_name(const DALLOC_CTX *d, int i)
|
||||
{
|
||||
void *o = dalloc_get_object(d, i);
|
||||
|
||||
if (o == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return talloc_get_name(o);
|
||||
}
|
||||
|
||||
/*
|
||||
* Get pointer to value from a DALLOC object
|
||||
*
|
||||
* Returns pointer to object from a DALLOC object. Nested object interation
|
||||
* is supported by using the type string "DALLOC_CTX". Any other type string
|
||||
* designates the requested objects type.
|
||||
*/
|
||||
void *dalloc_get(const DALLOC_CTX *d, ...)
|
||||
{
|
||||
int result = 0;
|
||||
void *p = NULL;
|
||||
va_list args;
|
||||
const char *type;
|
||||
int elem;
|
||||
|
||||
va_start(args, d);
|
||||
type = va_arg(args, const char *);
|
||||
|
||||
while (strcmp(type, "DALLOC_CTX") == 0) {
|
||||
elem = va_arg(args, int);
|
||||
if (elem >= talloc_array_length(d->dd_talloc_array)) {
|
||||
result = -1;
|
||||
goto done;
|
||||
}
|
||||
d = d->dd_talloc_array[elem];
|
||||
type = va_arg(args, const char *);
|
||||
}
|
||||
|
||||
elem = va_arg(args, int);
|
||||
if (elem >= talloc_array_length(d->dd_talloc_array)) {
|
||||
result = -1;
|
||||
goto done;
|
||||
}
|
||||
|
||||
p = talloc_check_name(d->dd_talloc_array[elem], type);
|
||||
if (p == NULL) {
|
||||
result = -1;
|
||||
goto done;
|
||||
}
|
||||
|
||||
done:
|
||||
va_end(args);
|
||||
if (result != 0) {
|
||||
p = NULL;
|
||||
}
|
||||
return p;
|
||||
}
|
||||
|
||||
void *dalloc_value_for_key(const DALLOC_CTX *d, ...)
|
||||
{
|
||||
int result = 0;
|
||||
void *p = NULL;
|
||||
va_list args;
|
||||
const char *type;
|
||||
int elem;
|
||||
size_t array_len;
|
||||
|
||||
va_start(args, d);
|
||||
type = va_arg(args, const char *);
|
||||
|
||||
while (strcmp(type, "DALLOC_CTX") == 0) {
|
||||
array_len = talloc_array_length(d->dd_talloc_array);
|
||||
elem = va_arg(args, int);
|
||||
if (elem >= array_len) {
|
||||
result = -1;
|
||||
goto done;
|
||||
}
|
||||
d = d->dd_talloc_array[elem];
|
||||
type = va_arg(args, const char *);
|
||||
}
|
||||
|
||||
array_len = talloc_array_length(d->dd_talloc_array);
|
||||
|
||||
for (elem = 0; elem + 1 < array_len; elem += 2) {
|
||||
if (strcmp(talloc_get_name(d->dd_talloc_array[elem]), "char *") != 0) {
|
||||
result = -1;
|
||||
goto done;
|
||||
}
|
||||
if (strcmp((char *)d->dd_talloc_array[elem],type) == 0) {
|
||||
p = d->dd_talloc_array[elem + 1];
|
||||
break;
|
||||
}
|
||||
}
|
||||
va_end(args);
|
||||
|
||||
done:
|
||||
if (result != 0) {
|
||||
p = NULL;
|
||||
}
|
||||
return p;
|
||||
}
|
||||
|
||||
static char *dalloc_strdup(TALLOC_CTX *mem_ctx, const char *string)
|
||||
{
|
||||
char *p;
|
||||
|
||||
p = talloc_strdup(mem_ctx, string);
|
||||
if (p == NULL) {
|
||||
return NULL;
|
||||
}
|
||||
talloc_set_name_const(p, "char *");
|
||||
return p;
|
||||
}
|
||||
|
||||
int dalloc_stradd(DALLOC_CTX *d, const char *string)
|
||||
{
|
||||
int result;
|
||||
char *p;
|
||||
|
||||
p = dalloc_strdup(d, string);
|
||||
if (p == NULL) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
result = dalloc_add(d, p, char *);
|
||||
if (result != 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
163
source3/rpc_server/mdssvc/dalloc.h
Normal file
163
source3/rpc_server/mdssvc/dalloc.h
Normal file
@ -0,0 +1,163 @@
|
||||
/*
|
||||
Copyright (c) Ralph Boehme 2012-2014
|
||||
|
||||
This program is free software; you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation; either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
/*
|
||||
Typesafe, dynamic object store based on talloc
|
||||
|
||||
Usage
|
||||
=====
|
||||
|
||||
Define some types:
|
||||
|
||||
A key/value store aka dictionary that supports retrieving elements
|
||||
by key:
|
||||
|
||||
typedef dict_t DALLOC_CTX;
|
||||
|
||||
An ordered set that can store different objects which can be
|
||||
retrieved by number:
|
||||
|
||||
typedef set_t DALLOC_CTX;
|
||||
|
||||
Create an dalloc object and add elementes of different type:
|
||||
|
||||
TALLOC_CTX *mem_ctx = talloc_new(NULL);
|
||||
DALLOC_CTX *d = dalloc_new(mem_ctx);
|
||||
|
||||
Store an int value in the object:
|
||||
|
||||
uint64_t i = 1;
|
||||
dalloc_add_copy(d, &i, uint64_t);
|
||||
|
||||
Store a string:
|
||||
|
||||
dalloc_stradd(d, "hello world");
|
||||
|
||||
Add a nested object:
|
||||
|
||||
DALLOC_CTX *nested = dalloc_new(d);
|
||||
dalloc_add(d, nested, DALLOC_CTX);
|
||||
|
||||
Add an int value to the nested object, this can be fetched:
|
||||
|
||||
i = 2;
|
||||
dalloc_add_copy(nested, &i, uint64_t);
|
||||
|
||||
Add a nested set:
|
||||
|
||||
set_t *set = dalloc_zero(nested, set_t);
|
||||
dalloc_add(nested, set, set_t);
|
||||
|
||||
Add an int value to the set:
|
||||
|
||||
i = 3;
|
||||
dalloc_add_copy(set, &i, uint64_t);
|
||||
|
||||
Add a dictionary (key/value store):
|
||||
|
||||
dict_t *dict = dalloc_zero(nested, dict_t);
|
||||
dalloc_add(nested, dict, dict_t);
|
||||
|
||||
Store a string as key in the dict:
|
||||
|
||||
dalloc_stradd(dict, "key");
|
||||
|
||||
Add a value for the key:
|
||||
|
||||
i = 4;
|
||||
dalloc_add_copy(dict, &i, uint64_t);
|
||||
|
||||
Fetching value references
|
||||
=========================
|
||||
|
||||
You can fetch anything that is not a DALLOC_CTXs, because passing
|
||||
"DALLOC_CTXs" as type to the functions dalloc_get() and
|
||||
dalloc_value_for_key() tells the function to step into that object
|
||||
and expect more arguments that specify which element to fetch.
|
||||
|
||||
Get reference to an objects element by position:
|
||||
|
||||
uint64_t *p = dalloc_get(d, "uint64_t", 0);
|
||||
|
||||
p now points to the first int with a value of 1.
|
||||
|
||||
Get reference to the "hello world" string:
|
||||
|
||||
str = dalloc_get(d, "char *", 1);
|
||||
|
||||
You can't fetch a DALLOC_CTX itself:
|
||||
|
||||
nested = dalloc_get(d, "DALLOC_CTX", 2);
|
||||
|
||||
But you can fetch elements from the neseted DALLOC_CTX:
|
||||
|
||||
p = dalloc_get(d, "DALLOC_CTX", 2, "uint64_t", 0);
|
||||
|
||||
p now points to the value 2.
|
||||
|
||||
You can fetch types that are typedefd DALLOC_CTXs:
|
||||
|
||||
set = dalloc_get(d, "DALLOC_CTX", 2, "set_t", 1);
|
||||
|
||||
Fetch int from set, must use DALLOC_CTX as type for the set:
|
||||
|
||||
p = dalloc_get(d, "DALLOC_CTX", 2, "DALLOC_CTX", 1, "uint64_t", 0);
|
||||
|
||||
p points to 3.
|
||||
|
||||
Fetch value by key from dictionary:
|
||||
|
||||
p = dalloc_value_for_key(d, "DALLOC_CTX", 2, "DALLOC_CTX", 2, "key");
|
||||
|
||||
p now points to 4.
|
||||
*/
|
||||
|
||||
#ifndef DALLOC_H
|
||||
#define DALLOC_H
|
||||
|
||||
#include <talloc.h>
|
||||
|
||||
struct dalloc_ctx;
|
||||
typedef struct dalloc_ctx DALLOC_CTX;
|
||||
|
||||
#define dalloc_new(mem_ctx) (DALLOC_CTX *)_dalloc_new((mem_ctx), "DALLOC_CTX")
|
||||
#define dalloc_zero(mem_ctx, type) (type *)_dalloc_new((mem_ctx), #type)
|
||||
|
||||
/**
|
||||
* talloc a chunk for obj of required size, copy the obj into the
|
||||
* chunk and add the chunk to the dalloc ctx
|
||||
**/
|
||||
#define dalloc_add_copy(d, obj, type) _dalloc_add_talloc_chunk((d), (obj), #type, sizeof(type))
|
||||
|
||||
/**
|
||||
* Add a pointer to a talloced object to the dalloc ctx. The object
|
||||
* must be a talloc child of the dalloc ctx.
|
||||
**/
|
||||
#define dalloc_add(d, obj, type) _dalloc_add_talloc_chunk((d), (obj), #type, 0)
|
||||
|
||||
|
||||
extern void *dalloc_get(const DALLOC_CTX *d, ...);
|
||||
extern void *dalloc_value_for_key(const DALLOC_CTX *d, ...);
|
||||
extern size_t dalloc_size(const DALLOC_CTX *d);
|
||||
extern void *dalloc_get_object(const DALLOC_CTX *d, int i);
|
||||
extern const char *dalloc_get_name(const DALLOC_CTX *d, int i);
|
||||
extern int dalloc_stradd(DALLOC_CTX *d, const char *string);
|
||||
|
||||
extern void *_dalloc_new(TALLOC_CTX *mem_ctx, const char *type);
|
||||
extern int _dalloc_add_talloc_chunk(DALLOC_CTX *d, void *obj, const char *type, size_t size);
|
||||
|
||||
#endif /* DALLOC_H */
|
||||
@ -130,6 +130,7 @@ bld.SAMBA3_SUBSYSTEM('RPC_WKSSVC',
|
||||
|
||||
bld.SAMBA3_SUBSYSTEM('RPC_MDSSVC',
|
||||
source='''mdssvc/mdssvc.c
|
||||
mdssvc/dalloc.c
|
||||
mdssvc/srv_mdssvc_nt.c
|
||||
../../librpc/gen_ndr/srv_mdssvc.c''',
|
||||
deps='samba-util ' + bld.env['libtracker'],
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user