1 files changed, 260 insertions, 0 deletions

diff --git a/lib/libtle_l4p/stream_table.h b/lib/libtle_l4p/stream_table.h
new file mode 100644
index 0000000..8ad1103
--- /dev/null
+++ b/lib/libtle_l4p/stream_table.h
@@ -0,0 +1,260 @@
+/*
+ * Copyright (c) 2016  Intel Corporation.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef _STREAM_TABLE_H_
+#define _STREAM_TABLE_H_
+
+#include <rte_hash.h>
+#include "tcp_misc.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/* current stbl entry contains packet. */
+#define	STE_PKT	1
+
+struct stbl_entry {
+	void *data;
+};
+
+struct shtbl {
+	uint32_t nb_ent;  /* max number of entries in the table. */
+	rte_spinlock_t l; /* lock to protect the hash table */
+	struct rte_hash *t;
+	struct stbl_entry *ent;
+} __rte_cache_aligned;
+
+struct stbl {
+	struct shtbl ht[TLE_VNUM];
+};
+
+struct stbl4_key {
+	union l4_ports port;
+	union ipv4_addrs addr;
+} __attribute__((__packed__));
+
+struct stbl6_key {
+	union l4_ports port;
+	union ipv6_addrs addr;
+} __attribute__((__packed__));
+
+struct stbl_key {
+	union l4_ports port;
+	union {
+		union ipv4_addrs addr4;
+		union ipv6_addrs addr6;
+	};
+} __attribute__((__packed__));
+
+extern void stbl_fini(struct stbl *st);
+
+extern int stbl_init(struct stbl *st, uint32_t num, int32_t socket);
+
+static inline void
+stbl_pkt_fill_key(struct stbl_key *k, const union pkt_info *pi, uint32_t type)
+{
+	static const struct stbl_key zero = {
+		.port.raw = 0,
+	};
+
+	k->port = pi->port;
+	if (type == TLE_V4)
+		k->addr4 = pi->addr4;
+	else if (type == TLE_V6)
+		k->addr6 = *pi->addr6;
+	else
+		*k = zero;
+}
+
+static inline void
+stbl_lock(struct stbl *st, uint32_t type)
+{
+	rte_spinlock_lock(&st->ht[type].l);
+}
+
+static inline void
+stbl_unlock(struct stbl *st, uint32_t type)
+{
+	rte_spinlock_unlock(&st->ht[type].l);
+}
+
+static inline struct stbl_entry *
+stbl_add_entry(struct stbl *st, const union pkt_info *pi)
+{
+	int32_t rc;
+	uint32_t type;
+	struct shtbl *ht;
+	struct stbl_key k;
+
+	type = pi->tf.type;
+	stbl_pkt_fill_key(&k, pi, type);
+	ht = st->ht + type;
+
+	rc = rte_hash_add_key(ht->t, &k);
+	if ((uint32_t)rc >= ht->nb_ent)
+		return NULL;
+	return ht->ent + rc;
+}
+
+static inline struct stbl_entry *
+stbl_add_pkt(struct stbl *st, const union pkt_info *pi, const void *pkt)
+{
+	struct stbl_entry *se;
+
+	se = stbl_add_entry(st, pi);
+	if (se != NULL)
+		se->data = (void *)((uintptr_t)pkt | STE_PKT);
+	return se;
+}
+
+static inline struct stbl_entry *
+stbl_find_entry(struct stbl *st, const union pkt_info *pi)
+{
+	int32_t rc;
+	uint32_t type;
+	struct shtbl *ht;
+	struct stbl_key k;
+
+	type = pi->tf.type;
+	stbl_pkt_fill_key(&k, pi, type);
+	ht = st->ht + type;
+
+	rc = rte_hash_lookup(ht->t, &k);
+	if ((uint32_t)rc >= ht->nb_ent)
+		return NULL;
+	return ht->ent + rc;
+}
+
+static inline int
+stbl_data_pkt(const void *p)
+{
+	return ((uintptr_t)p & STE_PKT);
+}
+
+static inline void *
+stbl_get_pkt(const struct stbl_entry *se)
+{
+	return (void *)((uintptr_t)se->data ^ STE_PKT);
+}
+
+static inline void *
+stbl_find_data(struct stbl *st, const union pkt_info *pi)
+{
+	struct stbl_entry *ent;
+
+	ent = stbl_find_entry(st, pi);
+	return (ent == NULL) ? NULL : ent->data;
+}
+
+static inline void
+stbl_del_pkt(struct stbl *st, struct stbl_entry *se, const union pkt_info *pi)
+{
+	uint32_t type;
+	struct stbl_key k;
+
+	se->data = NULL;
+
+	type = pi->tf.type;
+	stbl_pkt_fill_key(&k, pi, type);
+	rte_hash_del_key(st->ht[type].t, &k);
+}
+
+static inline void
+stbl_del_pkt_lock(struct stbl *st, struct stbl_entry *se,
+	const union pkt_info *pi)
+{
+	uint32_t type;
+	struct stbl_key k;
+
+	se->data = NULL;
+
+	type = pi->tf.type;
+	stbl_pkt_fill_key(&k, pi, type);
+	stbl_lock(st, type);
+	rte_hash_del_key(st->ht[type].t, &k);
+	stbl_unlock(st, type);
+}
+
+#include "tcp_stream.h"
+
+static inline void
+stbl_stream_fill_key(struct stbl_key *k, const struct tle_stream *s,
+	uint32_t type)
+{
+	static const struct stbl_key zero = {
+		.port.raw = 0,
+	};
+
+	k->port = s->port;
+	if (type == TLE_V4)
+		k->addr4 = s->ipv4.addr;
+	else if (type == TLE_V6)
+		k->addr6 = s->ipv6.addr;
+	else
+		*k = zero;
+}
+
+static inline struct stbl_entry *
+stbl_add_stream_lock(struct stbl *st, const struct tle_tcp_stream *s)
+{
+	uint32_t type;
+	struct stbl_key k;
+	struct stbl_entry *se;
+	struct shtbl *ht;
+	int32_t rc;
+
+	type = s->s.type;
+	stbl_stream_fill_key(&k, &s->s, type);
+	ht = st->ht + type;
+
+	stbl_lock(st, type);
+	rc = rte_hash_add_key(ht->t, &k);
+	stbl_unlock(st, type);
+
+	if ((uint32_t)rc >= ht->nb_ent)
+		return NULL;
+
+	se = ht->ent + rc;
+	if (se != NULL)
+		se->data = (void *)(uintptr_t)s;
+
+	return se;
+}
+
+static inline void
+stbl_del_stream_lock(struct stbl *st, struct stbl_entry *se,
+	const struct tle_tcp_stream *s)
+{
+	uint32_t type;
+	struct stbl_key k;
+
+	if (se == NULL)
+		return;
+
+	se->data = NULL;
+
+	type = s->s.type;
+	stbl_stream_fill_key(&k, &s->s, type);
+	stbl_lock(st, type);
+	rte_hash_del_key(st->ht[type].t, &k);
+	stbl_unlock(st, type);
+}
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* _STREAM_TABLE_H_ */