1 /* authd/providers/rdns.c - rDNS lookup provider for authd
2 * Copyright (c) 2016 Elizabeth Myers <elizabeth@interlinked.me>
4 * Permission to use, copy, modify, and/or distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice is present in all copies.
8 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
9 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
10 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
11 * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
12 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
13 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
14 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
15 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
16 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
17 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
18 * POSSIBILITY OF SUCH DAMAGE.
22 #include "rb_commio.h"
32 struct dns_query
*query
; /* Pending DNS query */
33 time_t timeout
; /* When the request times out */
36 /* Goinked from old s_auth.c --Elizabeth */
37 static const char *messages
[] =
39 "*** Looking up your hostname...",
40 "*** Found your hostname",
41 "*** Couldn't look up your hostname",
42 "*** Your hostname is too long, ignoring hostname",
53 static void client_fail(struct auth_client
*auth
, dns_message message
);
54 static void client_success(struct auth_client
*auth
);
55 static void get_dns_answer(const char *res
, bool status
, query_type type
, void *data
);
57 static struct ev_entry
*timeout_ev
;
58 static EVH timeout_dns_queries_event
;
59 static int rdns_timeout
= 30;
62 bool client_dns_init(void)
64 timeout_ev
= rb_event_addish("timeout_dns_queries_event", timeout_dns_queries_event
, NULL
, 5);
65 return (timeout_ev
!= NULL
);
68 void client_dns_destroy(void)
70 struct auth_client
*auth
;
71 struct DictionaryIter iter
;
73 DICTIONARY_FOREACH(auth
, &iter
, auth_clients
)
75 if(auth
->data
[PROVIDER_RDNS
] != NULL
)
76 client_fail(auth
, REPORT_FAIL
);
79 rb_event_delete(timeout_ev
);
82 bool client_dns_start(struct auth_client
*auth
)
84 struct user_query
*query
= rb_malloc(sizeof(struct user_query
));
86 query
->timeout
= rb_current_time() + rdns_timeout
;
88 auth
->data
[PROVIDER_RDNS
] = query
;
90 query
->query
= lookup_hostname(auth
->c_ip
, get_dns_answer
, auth
);
92 notice_client(auth
, messages
[REPORT_LOOKUP
]);
93 set_provider(auth
, PROVIDER_RDNS
);
97 void client_dns_cancel(struct auth_client
*auth
)
99 struct user_query
*query
= auth
->data
[PROVIDER_RDNS
];
102 client_fail(auth
, REPORT_FAIL
);
106 get_dns_answer(const char *res
, bool status
, query_type type
, void *data
)
108 struct auth_client
*auth
= data
;
109 struct user_query
*query
= auth
->data
[PROVIDER_RDNS
];
111 if(query
== NULL
|| res
== NULL
|| status
== false)
112 client_fail(auth
, REPORT_FAIL
);
113 else if(strlen(res
) > HOSTLEN
)
114 client_fail(auth
, REPORT_TOOLONG
);
117 rb_strlcpy(auth
->hostname
, res
, HOSTLEN
+ 1);
118 client_success(auth
);
122 /* Timeout outstanding queries */
123 static void timeout_dns_queries_event(void *notused
)
125 struct auth_client
*auth
;
126 struct DictionaryIter iter
;
128 DICTIONARY_FOREACH(auth
, &iter
, auth_clients
)
130 struct user_query
*query
= auth
->data
[PROVIDER_RDNS
];
132 if(query
!= NULL
&& query
->timeout
< rb_current_time())
134 client_fail(auth
, REPORT_FAIL
);
140 static void client_fail(struct auth_client
*auth
, dns_message report
)
142 struct user_query
*query
= auth
->data
[PROVIDER_RDNS
];
147 rb_strlcpy(auth
->hostname
, "*", sizeof(auth
->hostname
));
149 notice_client(auth
, messages
[report
]);
150 cancel_query(query
->query
);
153 auth
->data
[PROVIDER_RDNS
] = NULL
;
155 provider_done(auth
, PROVIDER_RDNS
);
158 static void client_success(struct auth_client
*auth
)
160 struct user_query
*query
= auth
->data
[PROVIDER_RDNS
];
162 notice_client(auth
, messages
[REPORT_FOUND
]);
163 cancel_query(query
->query
);
166 auth
->data
[PROVIDER_RDNS
] = NULL
;
168 provider_done(auth
, PROVIDER_RDNS
);
171 struct auth_provider rdns_provider
=
174 .init
= client_dns_init
,
175 .destroy
= client_dns_destroy
,
176 .start
= client_dns_start
,
177 .cancel
= client_dns_cancel
,