[lttng-tools.git] / src / common / hashtable / hashtable.h

/*
 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2 only,
 * as published by the Free Software Foundation.
 *
 * 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef _LTT_HT_H
#define _LTT_HT_H

#include <urcu.h>
#include <stdint.h>

#include <common/macros.h>
#include <urcu/rculfhash.h>

extern unsigned long lttng_ht_seed;

typedef unsigned long (*hash_fct)(void *_key, unsigned long seed);
typedef cds_lfht_match_fct hash_match_fct;

enum lttng_ht_type {
	LTTNG_HT_TYPE_STRING,
	LTTNG_HT_TYPE_ULONG,
	LTTNG_HT_TYPE_U64,
	LTTNG_HT_TYPE_TWO_U64,
};

struct lttng_ht {
	struct cds_lfht *ht;
	cds_lfht_match_fct match_fct;
	hash_fct hash_fct;
};

struct lttng_ht_iter {
	struct cds_lfht_iter iter;
};

struct lttng_ht_node_str {
	char *key;
	struct cds_lfht_node node;
	struct rcu_head head;
};

struct lttng_ht_node_ulong {
	unsigned long key;
	struct cds_lfht_node node;
	struct rcu_head head;
};

struct lttng_ht_node_u64 {
	uint64_t key;
	struct cds_lfht_node node;
	struct rcu_head head;
};

struct lttng_ht_two_u64 {
	uint64_t key1;
	uint64_t key2;
};

struct lttng_ht_node_two_u64 {
	struct lttng_ht_two_u64 key;
	struct cds_lfht_node node;
	struct rcu_head head;
};

/* Hashtable new and destroy */
LTTNG_HIDDEN
struct lttng_ht *lttng_ht_new(unsigned long size, int type);
LTTNG_HIDDEN
void lttng_ht_destroy(struct lttng_ht *ht);

/* Specialized node init and free functions */
LTTNG_HIDDEN
void lttng_ht_node_init_str(struct lttng_ht_node_str *node, char *key);
LTTNG_HIDDEN
void lttng_ht_node_init_ulong(struct lttng_ht_node_ulong *node,
		unsigned long key);
LTTNG_HIDDEN
void lttng_ht_node_init_u64(struct lttng_ht_node_u64 *node,
		uint64_t key);
LTTNG_HIDDEN
void lttng_ht_node_init_two_u64(struct lttng_ht_node_two_u64 *node,
		uint64_t key1, uint64_t key2);
LTTNG_HIDDEN
void lttng_ht_node_free_str(struct lttng_ht_node_str *node);
LTTNG_HIDDEN
void lttng_ht_node_free_ulong(struct lttng_ht_node_ulong *node);
LTTNG_HIDDEN
void lttng_ht_node_free_u64(struct lttng_ht_node_u64 *node);
LTTNG_HIDDEN
void lttng_ht_node_free_two_u64(struct lttng_ht_node_two_u64 *node);

LTTNG_HIDDEN
void lttng_ht_lookup(struct lttng_ht *ht, void *key,
		struct lttng_ht_iter *iter);

/* Specialized add unique functions */
LTTNG_HIDDEN
void lttng_ht_add_unique_str(struct lttng_ht *ht,
		struct lttng_ht_node_str *node);
LTTNG_HIDDEN
void lttng_ht_add_unique_ulong(struct lttng_ht *ht,
		struct lttng_ht_node_ulong *node);
LTTNG_HIDDEN
void lttng_ht_add_unique_u64(struct lttng_ht *ht,
		struct lttng_ht_node_u64 *node);
LTTNG_HIDDEN
void lttng_ht_add_unique_two_u64(struct lttng_ht *ht,
		struct lttng_ht_node_two_u64 *node);
LTTNG_HIDDEN
struct lttng_ht_node_ulong *lttng_ht_add_replace_ulong(
		struct lttng_ht *ht, struct lttng_ht_node_ulong *node);
LTTNG_HIDDEN
struct lttng_ht_node_u64 *lttng_ht_add_replace_u64(
		struct lttng_ht *ht, struct lttng_ht_node_u64 *node);
LTTNG_HIDDEN
void lttng_ht_add_str(struct lttng_ht *ht,
		struct lttng_ht_node_str *node);
LTTNG_HIDDEN
void lttng_ht_add_ulong(struct lttng_ht *ht,
		struct lttng_ht_node_ulong *node);
LTTNG_HIDDEN
void lttng_ht_add_u64(struct lttng_ht *ht,
		struct lttng_ht_node_u64 *node);

LTTNG_HIDDEN
int lttng_ht_del(struct lttng_ht *ht, struct lttng_ht_iter *iter);

LTTNG_HIDDEN
void lttng_ht_get_first(struct lttng_ht *ht,
		struct lttng_ht_iter *iter);
LTTNG_HIDDEN
void lttng_ht_get_next(struct lttng_ht *ht, struct lttng_ht_iter *iter);

LTTNG_HIDDEN
unsigned long lttng_ht_get_count(struct lttng_ht *ht);

LTTNG_HIDDEN
struct lttng_ht_node_str *lttng_ht_iter_get_node_str(
		struct lttng_ht_iter *iter);
LTTNG_HIDDEN
struct lttng_ht_node_ulong *lttng_ht_iter_get_node_ulong(
		struct lttng_ht_iter *iter);
LTTNG_HIDDEN
struct lttng_ht_node_u64 *lttng_ht_iter_get_node_u64(
		struct lttng_ht_iter *iter);
LTTNG_HIDDEN
struct lttng_ht_node_two_u64 *lttng_ht_iter_get_node_two_u64(
		struct lttng_ht_iter *iter);

#endif /* _LTT_HT_H */
Commit	Line	Data
	1	/*
	2	* Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License, version 2 only,
	6	* as published by the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope that it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	11	* more details.
	12	*
	13	* You should have received a copy of the GNU General Public License along
	14	* with this program; if not, write to the Free Software Foundation, Inc.,
	15	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	16	*/
	17
	18	#ifndef _LTT_HT_H
	19	#define _LTT_HT_H
	20
	21	#include <urcu.h>
	22	#include <stdint.h>
	23
	24	#include <common/macros.h>
	25	#include <urcu/rculfhash.h>
	26
	27	extern unsigned long lttng_ht_seed;
	28
	29	typedef unsigned long (hash_fct)(void _key, unsigned long seed);
	30	typedef cds_lfht_match_fct hash_match_fct;
	31
	32	enum lttng_ht_type {
	33	LTTNG_HT_TYPE_STRING,
	34	LTTNG_HT_TYPE_ULONG,
	35	LTTNG_HT_TYPE_U64,
	36	LTTNG_HT_TYPE_TWO_U64,
	37	};
	38
	39	struct lttng_ht {
	40	struct cds_lfht *ht;
	41	cds_lfht_match_fct match_fct;
	42	hash_fct hash_fct;
	43	};
	44
	45	struct lttng_ht_iter {
	46	struct cds_lfht_iter iter;
	47	};
	48
	49	struct lttng_ht_node_str {
	50	char *key;
	51	struct cds_lfht_node node;
	52	struct rcu_head head;
	53	};
	54
	55	struct lttng_ht_node_ulong {
	56	unsigned long key;
	57	struct cds_lfht_node node;
	58	struct rcu_head head;
	59	};
	60
	61	struct lttng_ht_node_u64 {
	62	uint64_t key;
	63	struct cds_lfht_node node;
	64	struct rcu_head head;
	65	};
	66
	67	struct lttng_ht_two_u64 {
	68	uint64_t key1;
	69	uint64_t key2;
	70	};
	71
	72	struct lttng_ht_node_two_u64 {
	73	struct lttng_ht_two_u64 key;
	74	struct cds_lfht_node node;
	75	struct rcu_head head;
	76	};
	77
	78	/* Hashtable new and destroy */
	79	LTTNG_HIDDEN
	80	struct lttng_ht *lttng_ht_new(unsigned long size, int type);
	81	LTTNG_HIDDEN
	82	void lttng_ht_destroy(struct lttng_ht *ht);
	83
	84	/* Specialized node init and free functions */
	85	LTTNG_HIDDEN
	86	void lttng_ht_node_init_str(struct lttng_ht_node_str node, char key);
	87	LTTNG_HIDDEN
	88	void lttng_ht_node_init_ulong(struct lttng_ht_node_ulong *node,
	89	unsigned long key);
	90	LTTNG_HIDDEN
	91	void lttng_ht_node_init_u64(struct lttng_ht_node_u64 *node,
	92	uint64_t key);
	93	LTTNG_HIDDEN
	94	void lttng_ht_node_init_two_u64(struct lttng_ht_node_two_u64 *node,
	95	uint64_t key1, uint64_t key2);
	96	LTTNG_HIDDEN
	97	void lttng_ht_node_free_str(struct lttng_ht_node_str *node);
	98	LTTNG_HIDDEN
	99	void lttng_ht_node_free_ulong(struct lttng_ht_node_ulong *node);
	100	LTTNG_HIDDEN
	101	void lttng_ht_node_free_u64(struct lttng_ht_node_u64 *node);
	102	LTTNG_HIDDEN
	103	void lttng_ht_node_free_two_u64(struct lttng_ht_node_two_u64 *node);
	104
	105	LTTNG_HIDDEN
	106	void lttng_ht_lookup(struct lttng_ht ht, void key,
	107	struct lttng_ht_iter *iter);
	108
	109	/* Specialized add unique functions */
	110	LTTNG_HIDDEN
	111	void lttng_ht_add_unique_str(struct lttng_ht *ht,
	112	struct lttng_ht_node_str *node);
	113	LTTNG_HIDDEN
	114	void lttng_ht_add_unique_ulong(struct lttng_ht *ht,
	115	struct lttng_ht_node_ulong *node);
	116	LTTNG_HIDDEN
	117	void lttng_ht_add_unique_u64(struct lttng_ht *ht,
	118	struct lttng_ht_node_u64 *node);
	119	LTTNG_HIDDEN
	120	void lttng_ht_add_unique_two_u64(struct lttng_ht *ht,
	121	struct lttng_ht_node_two_u64 *node);
	122	LTTNG_HIDDEN
	123	struct lttng_ht_node_ulong *lttng_ht_add_replace_ulong(
	124	struct lttng_ht ht, struct lttng_ht_node_ulong node);
	125	LTTNG_HIDDEN
	126	struct lttng_ht_node_u64 *lttng_ht_add_replace_u64(
	127	struct lttng_ht ht, struct lttng_ht_node_u64 node);
	128	LTTNG_HIDDEN
	129	void lttng_ht_add_str(struct lttng_ht *ht,
	130	struct lttng_ht_node_str *node);
	131	LTTNG_HIDDEN
	132	void lttng_ht_add_ulong(struct lttng_ht *ht,
	133	struct lttng_ht_node_ulong *node);
	134	LTTNG_HIDDEN
	135	void lttng_ht_add_u64(struct lttng_ht *ht,
	136	struct lttng_ht_node_u64 *node);
	137
	138	LTTNG_HIDDEN
	139	int lttng_ht_del(struct lttng_ht ht, struct lttng_ht_iter iter);
	140
	141	LTTNG_HIDDEN
	142	void lttng_ht_get_first(struct lttng_ht *ht,
	143	struct lttng_ht_iter *iter);
	144	LTTNG_HIDDEN
	145	void lttng_ht_get_next(struct lttng_ht ht, struct lttng_ht_iter iter);
	146
	147	LTTNG_HIDDEN
	148	unsigned long lttng_ht_get_count(struct lttng_ht *ht);
	149
	150	LTTNG_HIDDEN
	151	struct lttng_ht_node_str *lttng_ht_iter_get_node_str(
	152	struct lttng_ht_iter *iter);
	153	LTTNG_HIDDEN
	154	struct lttng_ht_node_ulong *lttng_ht_iter_get_node_ulong(
	155	struct lttng_ht_iter *iter);
	156	LTTNG_HIDDEN
	157	struct lttng_ht_node_u64 *lttng_ht_iter_get_node_u64(
	158	struct lttng_ht_iter *iter);
	159	LTTNG_HIDDEN
	160	struct lttng_ht_node_two_u64 *lttng_ht_iter_get_node_two_u64(
	161	struct lttng_ht_iter *iter);
	162
	163	#endif /* _LTT_HT_H */