drop-tld.diff revision 356345
1diff --git a/daemon/worker.c b/daemon/worker.c
2index 263fcdd..f787b70 100644
3--- a/daemon/worker.c
4+++ b/daemon/worker.c
5@@ -1213,6 +1213,15 @@ worker_handle_request(struct comm_point* c, void* arg, int error,
6 		addr_to_str(&repinfo->addr, repinfo->addrlen, ip, sizeof(ip));
7 		log_query_in(ip, qinfo.qname, qinfo.qtype, qinfo.qclass);
8 	}
9+
10+	if(worker->env.cfg->drop_tld) {
11+		int lab = dname_count_labels(qinfo.qname);
12+		if (lab == 2) {
13+			comm_point_drop_reply(repinfo);
14+			verbose(VERB_ALGO, "Dropping one label query.");
15+			return 0;
16+		}
17+	}
18 	if(qinfo.qtype == LDNS_RR_TYPE_AXFR || 
19 		qinfo.qtype == LDNS_RR_TYPE_IXFR) {
20 		verbose(VERB_ALGO, "worker request: refused zone transfer.");
21diff --git a/util/config_file.h b/util/config_file.h
22index b3ef930..2791541 100644
23--- a/util/config_file.h
24+++ b/util/config_file.h
25@@ -274,6 +274,8 @@ struct config_file {
26 	int prefetch_key;
27 	/** deny queries of type ANY with an empty answer */
28 	int deny_any;
29+	/** Drop TLD queries from clients **/
30+	int drop_tld;
31 
32 	/** chrootdir, if not "" or chroot will be done */
33 	char* chrootdir;
34diff --git a/util/configlexer.lex b/util/configlexer.lex
35index a86ddf5..9bbedbb 100644
36--- a/util/configlexer.lex
37+++ b/util/configlexer.lex
38@@ -299,6 +299,7 @@ private-domain{COLON}		{ YDVAR(1, VAR_PRIVATE_DOMAIN) }
39 prefetch-key{COLON}		{ YDVAR(1, VAR_PREFETCH_KEY) }
40 prefetch{COLON}			{ YDVAR(1, VAR_PREFETCH) }
41 deny-any{COLON}			{ YDVAR(1, VAR_DENY_ANY) }
42+drop-tld{COLON}			{ YDVAR(1, VAR_DROP_TLD) }
43 stub-zone{COLON}		{ YDVAR(0, VAR_STUB_ZONE) }
44 name{COLON}			{ YDVAR(1, VAR_NAME) }
45 stub-addr{COLON}		{ YDVAR(1, VAR_STUB_ADDR) }
46diff --git a/util/configparser.y b/util/configparser.y
47index 10227a2..567d68e 100644
48--- a/util/configparser.y
49+++ b/util/configparser.y
50@@ -164,6 +164,7 @@ extern struct config_parser_state* cfg_parser;
51 %token VAR_FAST_SERVER_PERMIL VAR_FAST_SERVER_NUM
52 %token VAR_ALLOW_NOTIFY VAR_TLS_WIN_CERT VAR_TCP_CONNECTION_LIMIT
53 %token VAR_FORWARD_NO_CACHE VAR_STUB_NO_CACHE VAR_LOG_SERVFAIL VAR_DENY_ANY
54+%token VAR_DROP_TLD
55 %token VAR_UNKNOWN_SERVER_TIME_LIMIT VAR_LOG_TAG_QUERYREPLY
56 %token VAR_STREAM_WAIT_SIZE VAR_TLS_CIPHERS VAR_TLS_CIPHERSUITES
57 %token VAR_TLS_SESSION_TICKET_KEYS
58@@ -266,6 +267,7 @@ content_server: server_num_threads | server_verbosity | server_port |
59 	server_tls_cert_bundle | server_tls_additional_port | server_low_rtt |
60 	server_fast_server_permil | server_fast_server_num  | server_tls_win_cert |
61 	server_tcp_connection_limit | server_log_servfail | server_deny_any |
62+	server_drop_tld |
63 	server_unknown_server_time_limit | server_log_tag_queryreply |
64 	server_stream_wait_size | server_tls_ciphers |
65 	server_tls_ciphersuites | server_tls_session_ticket_keys
66@@ -1466,6 +1468,16 @@ server_deny_any: VAR_DENY_ANY STRING_ARG
67 		free($2);
68 	}
69 	;
70+
71+server_drop_tld: VAR_DROP_TLD STRING_ARG
72+	{
73+		OUTYY(("P(server_drop_tld:%s)\n", $2));
74+		if(strcmp($2, "yes") != 0 && strcmp($2, "no") != 0)
75+			yyerror("expected yes or no.");
76+		else cfg_parser->cfg->drop_tld = (strcmp($2, "yes")==0);
77+		free($2);
78+	}
79+	;
80 server_unwanted_reply_threshold: VAR_UNWANTED_REPLY_THRESHOLD STRING_ARG
81 	{
82 		OUTYY(("P(server_unwanted_reply_threshold:%s)\n", $2));
83