2 * InspIRCd -- Internet Relay Chat Daemon
4 * Copyright (C) 2013 Adam <Adam@anope.org>
5 * Copyright (C) 2003-2013 Anope Team <team@anope.org>
7 * This file is part of InspIRCd. InspIRCd is free software: you can
8 * redistribute it and/or modify it under the terms of the GNU General Public
9 * License as published by the Free Software Foundation, version 2.
11 * This program is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
13 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
30 /* A simple A lookup */
34 /* Reverse DNS lookup */
36 /* IPv6 AAAA lookup */
40 /** Flags that can be AND'd into DNSPacket::flags to receive certain values
44 QUERYFLAGS_QR = 0x8000,
45 QUERYFLAGS_OPCODE = 0x7800,
46 QUERYFLAGS_AA = 0x400,
47 QUERYFLAGS_TC = 0x200,
48 QUERYFLAGS_RD = 0x100,
51 QUERYFLAGS_RCODE = 0xF
61 ERROR_NONSTANDARD_QUERY,
64 ERROR_DOMAIN_NOT_FOUND,
65 ERROR_NOT_IMPLEMENTED,
71 typedef uint16_t RequestId;
75 class Exception : public ModuleException
78 Exception(const std::string& message) : ModuleException(message) { }
86 Question() : type(QUERY_NONE) { }
87 Question(const std::string& n, QueryType t) : name(n), type(t) { }
88 bool operator==(const Question& other) const { return ((name == other.name) && (type == other.type)); }
89 bool operator!=(const Question& other) const { return (!(*this == other)); }
93 size_t operator()(const Question& question) const
95 return irc::insensitive()(question.name);
100 struct ResourceRecord : Question
106 ResourceRecord(const std::string& n, QueryType t) : Question(n, t), ttl(0), created(ServerInstance->Time()) { }
107 ResourceRecord(const Question& question) : Question(question), ttl(0), created(ServerInstance->Time()) { }
113 std::vector<ResourceRecord> answers;
117 Query() : error(ERROR_NONE), cached(false) { }
118 Query(const Question& q) : question(q), error(ERROR_NONE), cached(false) { }
126 class Manager : public DataProvider
129 Manager(Module* mod) : DataProvider(mod, "DNS") { }
131 virtual void Process(Request* req) = 0;
132 virtual void RemoveRequest(Request* req) = 0;
133 virtual std::string GetErrorStr(Error) = 0;
138 class Request : public Timer, public Question
141 Manager* const manager;
143 /* Use result cache if available */
147 /* Creator of this request */
148 Module* const creator;
150 Request(Manager* mgr, Module* mod, const std::string& addr, QueryType qt, bool usecache = true)
151 : Timer((ServerInstance->Config->dns_timeout ? ServerInstance->Config->dns_timeout : 5))
154 , use_cache(usecache)
158 ServerInstance->Timers.AddTimer(this);
163 manager->RemoveRequest(this);
166 /** Called when this request succeeds
167 * @param r The query sent back from the nameserver
169 virtual void OnLookupComplete(const Query* req) = 0;
171 /** Called when this request fails or times out.
172 * @param r The query sent back from the nameserver, check the error code.
174 virtual void OnError(const Query* req) { }
176 /** Used to time out the query, calls OnError and asks the TimerManager
177 * to delete this request
179 bool Tick(time_t now)
182 rr.error = ERROR_TIMEDOUT;