addresschunk.c

#include "addresschunk.h"

int AddressChunk_Init(AddressChunk *ac)
{
	if( AddressList_Init(&(ac -> TCPAddresses)) != 0 )
	{
		return 1;
	}

	if( AddressList_Init(&(ac -> UDPAddresses)) != 0 )
	{
		return 2;
	}

	if( StringChunk_Init(&(ac -> Dedicated), NULL) != 0 )
	{
		return 3;
	}

	return 0;
}

int AddressChunk_AddATCPAddress_FromString(AddressChunk *ac, const char *Addr_Port)
{
	return AddressList_Add_From_String(&(ac -> TCPAddresses), Addr_Port, 53);
}

int AddressChunk_AddAUDPAddress_FromString(AddressChunk *ac, const char *Addr_Port)
{
	return AddressList_Add_From_String(&(ac -> UDPAddresses), Addr_Port, 53);
}

int AddressChunk_AddADedicatedAddress_FromString(AddressChunk *ac, const char *Domain, const char *Addr_Port)
{
	Address_Type	Tmp;

	if( AddressList_ConvertToAddressFromString(&Tmp, Addr_Port, 53) == AF_UNSPEC )
	{
		return -1;
	}

	if( StringChunk_Add(&(ac -> Dedicated), Domain, (const char *)&Tmp, sizeof(Address_Type)) != 0 )
	{
		return -2;
	}

	return 0;

}

struct sockaddr *AddressChunk_GetDedicated(AddressChunk *ac, sa_family_t *family, const char *RequestingDomain, int *HashValue)
{
	Address_Type *Result;

	if( StringChunk_Match(&(ac -> Dedicated), RequestingDomain, HashValue, (char **)&Result) == TRUE )
	{
		if( Result -> family == AF_INET )
		{
			*family = Result -> family;
			return (struct sockaddr *)&(Result -> Addr.Addr4);
		} else {
			*family = Result -> family;
			return (struct sockaddr *)&(Result -> Addr.Addr6);
		}

	} else {
		return NULL;
	}
}

struct sockaddr *AddressChunk_GetOne(AddressChunk *ac, sa_family_t *family, DNSQuaryProtocol Protocol)
{
	if( Protocol == DNS_QUARY_PROTOCOL_UDP )
	{
		return AddressList_GetOne(&(ac -> UDPAddresses), family);
	} else {
		return AddressList_GetOne(&(ac -> TCPAddresses), family);
	}
}

struct sockaddr *AddressChunk_GetOneUDPBySubscript(AddressChunk *ac, sa_family_t *family, int Subscript)
{
	return AddressList_GetOneBySubscript(&(ac -> UDPAddresses), family, Subscript);
}

int AddressChunk_Advance(AddressChunk *ac, DNSQuaryProtocol Protocol)
{
	if( Protocol == DNS_QUARY_PROTOCOL_UDP )
	{
		AddressList_Advance(&(ac -> UDPAddresses));
	} else {
		AddressList_Advance(&(ac -> TCPAddresses));
	}

	return 0;
}