File: resolver.hpp

package info (click to toggle)
smtpping 1.1.3-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid
  • size: 148 kB
  • sloc: cpp: 862; makefile: 3
file content (76 lines) | stat: -rw-r--r-- 1,869 bytes parent folder | download | duplicates (2)
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
/*
	SMTP PING
	Copyright (C) 2011 Halon Security <support@halon.se>

	This program 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; either version 2
	of the License, or (at your option) any later version.

	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, write to the Free Software
	Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

*/

#ifndef _RESOLVER_HPP_
#define _RESOLVER_HPP_

#include <string>
#include <vector>
#include <stdexcept>

#if defined(__APPLE__) or defined(__FreeBSD__) or defined(__linux)
#include <netinet/in.h>
#include <arpa/nameser.h>
#endif
#ifdef __WIN32__
#include <winsock2.h>
#include <windns.h>

typedef VOID (WINAPI DNSRECORDLISTFREE)(PDNS_RECORD, DNS_FREE_TYPE);
typedef DNSRECORDLISTFREE* LPDNSRECORDLISTFREE;
typedef DNS_STATUS (WINAPI DNSQUERY)(LPCTSTR, WORD, DWORD, PIP4_ARRAY, PDNS_RECORD*, PVOID*);
typedef DNSQUERY* LPDNSQUERY;
#else
#include <resolv.h>
#endif

#ifndef MSG_NOSIGNAL
#define MSG_NOSIGNAL 0
#endif

class Resolver
{
	public:
		typedef enum {
			RR_MX,
			RR_A,
			RR_AAAA,
		} RecordType;

		Resolver();
		~Resolver();

		bool Lookup(const std::string& domain, RecordType recordType, std::vector<std::string>& result);
		int GetLastError() const {
#ifndef __WIN32__
			return m_res.res_h_errno;
#endif
		}
	private:
#ifdef __WIN32__
		HINSTANCE m_hDnsInst;
		LPDNSRECORDLISTFREE m_lpfnDnsRecordListFree;
		LPDNSQUERY m_lpfnDnsQuery;
#else
		struct __res_state m_res;
#endif
};

#endif