/*
 * Copyright (c) 2016-2017  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

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_stream(struct stbl *st, const union pkt_info *pi, const void *s)
{
	struct stbl_entry *se;

	se = stbl_add_entry(st, pi);
	if (se != NULL)
		se->data = (void *)(uintptr_t)s;
	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 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;
}

#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(struct stbl *st, struct stbl_entry *se,
	const struct tle_tcp_stream *s, uint32_t lock)
{
	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);
	if (lock != 0)
		stbl_lock(st, type);
	rte_hash_del_key(st->ht[type].t, &k);
	if (lock != 0)
		stbl_unlock(st, type);
}

#ifdef __cplusplus
}
#endif

#endif /* _STREAM_TABLE_H_ */