summaryrefslogtreecommitdiff
path: root/src/user_resolver.cpp
blob: f18fc9a0385a29604ec5586bd3d276e7a8de2d8f (plain)
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
/*
 * InspIRCd -- Internet Relay Chat Daemon
 *
 *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
 *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
 *
 * This file is part of InspIRCd.  InspIRCd is free software: you can
 * redistribute it and/or modify it under the terms of the GNU General Public
 * License as published by the Free Software Foundation, version 2.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


#include "inspircd.h"
UserResolver::UserResolver(LocalUser* user, std::string to_resolve, QueryType qt, bool &cache) :
	Resolver(to_resolve, qt, cache, NULL), uuid(user->uuid)
{
	this->fwd = (qt == DNS_QUERY_A || qt == DNS_QUERY_AAAA);
}

void UserResolver::OnLookupComplete(const std::string &result, unsigned int ttl, bool cached)
{
	UserResolver *res_forward; // for forward-resolution
	LocalUser* bound_user = (LocalUser*)ServerInstance->FindUUID(uuid);
	if (!bound_user)
	{
		ServerInstance->Logs->Log("RESOLVER", DEBUG, "Resolution finished for user '%s' who is gone", uuid.c_str());
		return;
	}

	ServerInstance->Logs->Log("RESOLVER", DEBUG, "DNS result for %s: '%s' -> '%s'", uuid.c_str(), input.c_str(), result.c_str());

	if (!fwd)
	{
		// first half of resolution is done. We now need to verify that the host matches.
		bound_user->stored_host = result;
		try
		{
			/* Check we didnt time out */
			if (bound_user->registered != REG_ALL)
			{
				bool lcached = false;
				if (bound_user->client_sa.sa.sa_family == AF_INET6)
				{
					/* IPV6 forward lookup */
					res_forward = new UserResolver(bound_user, result, DNS_QUERY_AAAA, lcached);
				}
				else
				{
					/* IPV4 lookup */
					res_forward = new UserResolver(bound_user, result, DNS_QUERY_A, lcached);
				}
				ServerInstance->AddResolver(res_forward, lcached);
			}
		}
		catch (CoreException& e)
		{
			ServerInstance->Logs->Log("RESOLVER", DEBUG,"Error in resolver: %s",e.GetReason());
		}
	}
	else
	{
		/* Both lookups completed */

		irc::sockets::sockaddrs* user_ip = &bound_user->client_sa;
		bool rev_match = false;
		if (user_ip->sa.sa_family == AF_INET6)
		{
			struct in6_addr res_bin;
			if (inet_pton(AF_INET6, result.c_str(), &res_bin))
			{
				rev_match = !memcmp(&user_ip->in6.sin6_addr, &res_bin, sizeof(res_bin));
			}
		}
		else
		{
			struct in_addr res_bin;
			if (inet_pton(AF_INET, result.c_str(), &res_bin))
			{
				rev_match = !memcmp(&user_ip->in4.sin_addr, &res_bin, sizeof(res_bin));
			}
		}
		
		if (rev_match)
		{
			std::string hostname = bound_user->stored_host;
			if (hostname.length() < 65)
			{
				/* Check we didnt time out */
				if ((bound_user->registered != REG_ALL) && (!bound_user->dns_done))
				{
					/* Hostnames starting with : are not a good thing (tm) */
					if (hostname[0] == ':')
						hostname.insert(0, "0");

					bound_user->WriteServ("NOTICE Auth :*** Found your hostname (%s)%s", hostname.c_str(), (cached ? " -- cached" : ""));
					bound_user->dns_done = true;
					bound_user->dhost.assign(hostname, 0, 64);
					bound_user->host.assign(hostname, 0, 64);
					/* Invalidate cache */
					bound_user->InvalidateCache();
				}
			}
			else
			{
				if (!bound_user->dns_done)
				{
					bound_user->WriteServ("NOTICE Auth :*** Your hostname is longer than the maximum of 64 characters, using your IP address (%s) instead.", bound_user->GetIPString());
					bound_user->dns_done = true;
				}
			}
		}
		else
		{
			if (!bound_user->dns_done)
			{
				bound_user->WriteServ("NOTICE Auth :*** Your hostname does not match up with your IP address. Sorry, using your IP address (%s) instead.", bound_user->GetIPString());
				bound_user->dns_done = true;
			}
		}

		// Save some memory by freeing this up; it's never used again in the user's lifetime.
		bound_user->stored_host.resize(0);
	}
}

void UserResolver::OnError(ResolverError e, const std::string &errormessage)
{
	LocalUser* bound_user = (LocalUser*)ServerInstance->FindUUID(uuid);
	if (bound_user)
	{
		bound_user->WriteServ("NOTICE Auth :*** Could not resolve your hostname: %s; using your IP address (%s) instead.", errormessage.c_str(), bound_user->GetIPString());
		bound_user->dns_done = true;
		bound_user->stored_host.resize(0);
		ServerInstance->stats->statsDnsBad++;
	}
}