forked from yaosj2k/dnsforwarder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
addresschunk.c
executable file
·96 lines (78 loc) · 2.18 KB
/
addresschunk.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#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_Domain(&(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_Domain_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;
}