/*
 * Copyright 2006-2007, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Axel Dörfler, axeld@pinc-software.de
 *		Hugo Santos, hugosantos@gmail.com
 */
#ifndef ENDPOINT_MANAGER_H
#define ENDPOINT_MANAGER_H


#include "tcp.h"

#include <AddressUtilities.h>

#include <lock.h>
#include <util/DoublyLinkedList.h>
#include <util/MultiHashTable.h>
#include <util/OpenHashTable.h>

#include <utility>


struct net_address_module_info;
struct net_domain;
class EndpointManager;
class TCPEndpoint;

struct ConnectionHashDefinition {
public:
	typedef std::pair<const sockaddr *, const sockaddr *> KeyType;
	typedef TCPEndpoint ValueType;

	ConnectionHashDefinition(EndpointManager *manager);

	ConnectionHashDefinition(const ConnectionHashDefinition& definition)
		: fManager(definition.fManager)
	{
	}

	size_t HashKey(const KeyType &key) const;
	size_t Hash(TCPEndpoint *endpoint) const;
	bool Compare(const KeyType &key, TCPEndpoint *endpoint) const;
	HashTableLink<TCPEndpoint> *GetLink(TCPEndpoint *endpoint) const;

private:
	EndpointManager *fManager;
};


class EndpointHashDefinition {
public:
	typedef uint16 KeyType;
	typedef TCPEndpoint ValueType;

	size_t HashKey(uint16 port) const;
	size_t Hash(TCPEndpoint *endpoint) const;
	bool Compare(uint16 port, TCPEndpoint *endpoint) const;
	bool CompareValues(TCPEndpoint *first, TCPEndpoint *second) const;
	HashTableLink<TCPEndpoint> *GetLink(TCPEndpoint *endpoint) const;
};


class EndpointManager : public DoublyLinkedListLinkImpl<EndpointManager> {
	public:
		EndpointManager(net_domain *domain);
		~EndpointManager();

		status_t InitCheck() const;

		TCPEndpoint *FindConnection(sockaddr *local, sockaddr *peer);

		status_t SetConnection(TCPEndpoint *endpoint, const sockaddr *local,
			const sockaddr *peer, const sockaddr *interfaceLocal);
		status_t SetPassive(TCPEndpoint *endpoint);

		status_t Bind(TCPEndpoint *endpoint, const sockaddr *address);
		status_t BindChild(TCPEndpoint *endpoint);
		status_t Unbind(TCPEndpoint *endpoint);

		status_t ReplyWithReset(tcp_segment_header &segment,
			net_buffer *buffer);

		void DumpEndpoints() const;

		net_domain *Domain() const { return fDomain; }
		net_address_module_info *AddressModule() const
			{ return Domain()->address_module; }

	private:
		TCPEndpoint *_LookupConnection(const sockaddr *local,
			const sockaddr *peer);
		status_t _Bind(TCPEndpoint *endpoint, const sockaddr *address);
		status_t _BindToAddress(TCPEndpoint *endpoint, const sockaddr *address);
		status_t _BindToEphemeral(TCPEndpoint *endpoint,
			const sockaddr *address);

		net_domain *fDomain;

		typedef OpenHashTable<ConnectionHashDefinition> ConnectionTable;
		typedef MultiHashTable<EndpointHashDefinition> EndpointTable;

		ConnectionTable fConnectionHash;
		EndpointTable fEndpointHash;
		benaphore fLock;
};

#endif	// ENDPOINT_MANAGER_H