pluto-src 0.1.1+0.10.4

Sources of Pluto (Lua 5.4 dialect) and logic to build it.
Documentation 
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

#include "dnsSmartResolver.hpp"
#if !SOUP_WASM

#include "dnsHttpResolver.hpp"
#include "dnsLookupTask.hpp"
#include "dnsUdpResolver.hpp"
#include "ObfusString.hpp"
#include "WeakRef.hpp"

NAMESPACE_SOUP
{
	struct dnsSmartLookupTask : public dnsLookupTask
	{
		WeakRef<const dnsSmartResolver> resolv_wr;
		bool retry = false;
		dnsType qtype;
		std::string name;
		UniquePtr<dnsLookupTask> subtask;
		UniquePtr<dnsHttpResolver> http_resolver;

		dnsSmartLookupTask(const dnsSmartResolver& resolv, dnsType qtype, const std::string& name)
			: resolv_wr(&resolv), qtype(qtype), name(name), subtask(resolv.subresolver->makeLookupTask(qtype, name))
		{
		}

		void onTick() final
		{
			if (subtask->tickUntilDone())
			{
				if (auto resolv = resolv_wr.getPointer())
				{
					if (subtask->result.has_value())
					{
						if (retry)
						{
							resolv->subresolver = std::move(http_resolver);
							resolv->switched_to_http = true;
						}
						else if (!resolv->switched_to_http)
						{
							static_cast<dnsUdpResolver*>(resolv->subresolver.get())->num_retries = 1; // UDP resolver may now retry once to account for packet loss.
						}
						fulfil(std::move(subtask->result));
					}
					else
					{
						// Lookup failed
						if (!resolv->switched_to_http && !retry)
						{
							// Try switching to using HTTP transport
							retry = true;
							std::string server = resolv->server.toString();
							http_resolver = soup::make_unique<dnsHttpResolver>();
							static_cast<dnsHttpResolver*>(http_resolver.get())->server = std::move(server);
							subtask = http_resolver->makeLookupTask(qtype, name);
						}
						else
						{
							// Out of options, have to report failure.
							fulfil(std::move(subtask->result));
						}
					}
				}
				else
				{
					fulfil(std::move(subtask->result));
				}
			}
		}

		[[nodiscard]] std::string toString() const SOUP_EXCAL final
		{
			std::string str = ObfusString("dnsSmartLookupTask");
			if (subtask)
			{
				str.append(": ");
				str.push_back('[');
				str.append(subtask->toString());
				str.push_back(']');
			}
			return str;
		}
	};

	UniquePtr<dnsLookupTask> dnsSmartResolver::makeLookupTask(dnsType qtype, const std::string& name) const
	{
		if (!subresolver)
		{
			subresolver = soup::make_unique<dnsUdpResolver>();
			static_cast<dnsUdpResolver*>(subresolver.get())->server.ip = this->server;
			static_cast<dnsUdpResolver*>(subresolver.get())->num_retries = 0; // No retries during testing phase.
		}
		return soup::make_unique<dnsSmartLookupTask>(*this, qtype, name);
	}
}

#endif