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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/* main.c
* (c) Jonas Gunz, 2020
* License: MIT
* */
#include "server.h"
void run_dns_server ( server_config_t* _config )
{
int ret;
struct sockaddr_in sock_server_addr;
char recv_buffer[ UDP_BUFFER_LEN ];
signal ( SIGTERM, signal_term );
signal ( SIGINT, signal_term );
//Avoid zombie processes
//TODO currently useless, since no forking is done.
signal (SIGCHLD, SIG_IGN);
log_init_stdout ( _LOG_DEBUG );
LOGPRINTF(_LOG_NOTE, "Initializing DNS Server on %s:%i", _config->bind_ip, _config->bind_port);
sock_server = socket ( AF_INET, SOCK_DGRAM, 0 );
if ( sock_server == -1 ) {
LOGPRINTF(_LOG_ERROR, "socket() failed");
exit ( errno );
}
memset( &sock_server_addr, '\0', sizeof(struct sockaddr_in) );
sock_server_addr.sin_family = AF_INET;
sock_server_addr.sin_port = htons( _config->bind_port );
ret = inet_aton ( _config->bind_ip, & sock_server_addr.sin_addr );
if( ret == 0 ) { //Error on 0, no errno!
LOGPRINTF(_LOG_NOTE, "inet_aton(): Invalid bind IP\n" );
exit ( 1 );
}
ret = bind ( sock_server,
(struct sockaddr*) &sock_server_addr,
sizeof(struct sockaddr_in) );
if ( ret == -1 ) {
LOGPRINTF(_LOG_ERROR, "bind() failed");
exit ( errno );
}
LOGPRINTF(_LOG_NOTE, "Done!");
while( 1 )
{
struct sockaddr_in sock_client_addr;
socklen_t sock_client_addr_len;
sock_client_addr_len = sizeof ( struct sockaddr_in );
memset ( &sock_client_addr, '\0', sock_client_addr_len );
ret = recvfrom (sock_server,
recv_buffer,
UDP_BUFFER_LEN,
0,
(struct sockaddr*) &sock_client_addr,
&sock_client_addr_len );
if ( ret == -1 ) {
LOGPRINTF( _LOG_ERROR, "recvfrom()");
exit ( errno );
}
LOGPRINTF(_LOG_DEBUG, "UDP Packet size %i", ret);
handle_connection ( sock_server,
&sock_client_addr,
sock_client_addr_len,
recv_buffer,
ret );
}
close( sock_server );
}
int handle_connection ( int _socket,
struct sockaddr_in *sockaddr_client,
socklen_t sockaddr_client_len,
char* _buffer,
int _bufflen )
{
struct dns_message msg;
if (dns_parse_packet (_buffer, _bufflen, &msg) ) {
LOGPRINTF (_LOG_DEBUG, "Malformed packet recieved. parsing failed");
return 1;
}
if(msg.question_count > 0) {
char out[128];
qname_to_fqdn( (char*) msg.question[0].qname, 100, out, 128);
printf("%s %i\n", out, msg.question[0].qtype);
}
//Always return NXDOMAIN
struct dns_question quest = & msg.question[0];
struct dns_header head = {msg.header.id,1,OP_Q,0,0,0,0,0,dns_responsecode.NOERR,0,0,0,0};
struct dns_answer answ = {quest->qname, quest->qname_len, }
char ret[512];
int hlen = dns_construct_header ( ret, 512, &head );
int alen = dns_construct_answer ( ret + hlen, 512-hlen, &answer );
sendto (_socket, ret, hlen + alen, 0, (struct sockaddr*) sockaddr_client, sockaddr_client_len);
dns_destroy_struct ( &msg );
return 0;
}
void signal_term ( ) {
printf( "Recieved Signal. Terminating active connections and closing socket\n" );
//terminate all children >:)
kill ( 0, SIGTERM );
shutdown ( sock_server, SHUT_RDWR );
close ( sock_server );
printf( "done\n" );
exit( 0 );
}
|