introduced SocketAddress wrappers and use them in TCP.

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@20818 a95241bf-73f2-0310-859d-f6bbb57e9c96
2007-04-25 17:41:01 +00:00
parent c36e4d4c9f
commit 585195c28d
5 changed files with 254 additions and 67 deletions
@@ -0,0 +1,191 @@
+/*
+ * Copyright 2007, Haiku, Inc. All Rights Reserved.
+ * Distributed under the terms of the MIT License.
+ *
+ * Authors:
+ *      Hugo Santos, [email protected]
+ */
+
+#ifndef ADDRESS_UTILITIES_H
+#define ADDRESS_UTILITIES_H
+
+#include <net_datalink.h>
+
+#include <sys/socket.h>
+
+class SocketAddress {
+public:
+	SocketAddress(net_address_module_info *module, sockaddr *address)
+		: fModule(module), fAddress(NULL)
+	{
+		SetAddressTo(address);
+	}
+
+	SocketAddress(net_address_module_info *module, sockaddr_storage *address)
+		: fModule(module), fAddress(NULL)
+	{
+		SetAddressTo((sockaddr *)address);
+	}
+
+	SocketAddress(const SocketAddress &address)
+		: fModule(address.fModule), fAddress(address.fAddress)
+	{}
+
+	void SetAddressTo(sockaddr *address)
+	{
+		fAddress = address;
+	}
+
+	bool IsEmpty(bool checkPort) const
+	{
+		return fModule->is_empty_address(fAddress, checkPort);
+	}
+
+	uint32 HashPair(const sockaddr *second) const
+	{
+		return fModule->hash_address_pair(fAddress, second);
+	}
+
+	bool EqualTo(const sockaddr *address, bool checkPort = false) const
+	{
+		if (checkPort)
+			return fModule->equal_addresses_and_ports(fAddress, address);
+		else
+			return fModule->equal_addresses(fAddress, address);
+	}
+
+	bool EqualTo(const SocketAddress &second, bool checkPort = false) const
+	{
+		return EqualTo(second.fAddress, checkPort);
+	}
+
+	uint16 GetPort() const
+	{
+		return fModule->get_port(fAddress);
+	}
+
+	status_t SetTo(const sockaddr *from)
+	{
+		return fModule->set_to(fAddress, from);
+	}
+
+	status_t SetTo(const sockaddr_storage *from)
+	{
+		return SetTo((sockaddr *)from);
+	}
+
+	status_t CopyTo(sockaddr *to) const
+	{
+		return fModule->set_to(to, fAddress);
+	}
+
+	status_t CopyTo(sockaddr_storage *to) const
+	{
+		return CopyTo((sockaddr *)to);
+	}
+
+	status_t SetPort(uint16 port)
+	{
+		return fModule->set_port(fAddress, port);
+	}
+
+	status_t SetToEmpty()
+	{
+		return fModule->set_to_empty_address(fAddress);
+	}
+
+	status_t UpdateTo(const sockaddr *from)
+	{
+		return fModule->update_to(fAddress, from);
+	}
+
+	const sockaddr *operator *() const { return fAddress; }
+	sockaddr *operator *() { return fAddress; }
+
+	net_address_module_info *Module() const { return fModule; }
+
+private:
+	net_address_module_info *fModule;
+	sockaddr *fAddress;
+};
+
+class ConstSocketAddress {
+public:
+	ConstSocketAddress(net_address_module_info *module,
+		const sockaddr *address)
+		: fModule(module), fAddress(NULL)
+	{
+		SetAddressTo(address);
+	}
+
+	ConstSocketAddress(net_address_module_info *module,
+		const sockaddr_storage *address)
+		: fModule(module), fAddress(NULL)
+	{
+		SetAddressTo((sockaddr *)address);
+	}
+
+	ConstSocketAddress(const ConstSocketAddress &address)
+		: fModule(address.fModule), fAddress(address.fAddress)
+	{}
+
+	ConstSocketAddress(const SocketAddress &address)
+		: fModule(address.Module()), fAddress(*address)
+	{}
+
+	void SetAddressTo(const sockaddr *address)
+	{
+		fAddress = address;
+	}
+
+	bool IsEmpty(bool checkPort) const
+	{
+		return fModule->is_empty_address(fAddress, checkPort);
+	}
+
+	uint32 HashPair(const sockaddr *second) const
+	{
+		return fModule->hash_address_pair(fAddress, second);
+	}
+
+	bool EqualTo(const sockaddr *address, bool checkPort = false) const
+	{
+		if (checkPort)
+			return fModule->equal_addresses_and_ports(fAddress, address);
+		else
+			return fModule->equal_addresses(fAddress, address);
+	}
+
+	uint16 GetPort() const
+	{
+		return fModule->get_port(fAddress);
+	}
+
+	status_t CopyTo(sockaddr *to) const
+	{
+		return fModule->set_to(to, fAddress);
+	}
+
+	status_t CopyTo(sockaddr_storage *to) const
+	{
+		return CopyTo((sockaddr *)to);
+	}
+
+	const sockaddr *operator *() const { return fAddress; }
+
+private:
+	net_address_module_info *fModule;
+	const sockaddr *fAddress;
+};
+
+class SocketAddressStorage : public SocketAddress {
+public:
+	SocketAddressStorage(net_address_module_info *module)
+		: SocketAddress(module, &fAddressStorage)
+	{}
+
+private:
+	sockaddr_storage fAddressStorage;
+};
+
+#endif