mirror of
https://github.com/matrix-construct/construct
synced 2024-09-29 12:18:54 +02:00
modules: Add DNS module with Node.js compatible lookup().
This commit is contained in:
parent
4fa9985399
commit
7b4c06575f
4 changed files with 178 additions and 0 deletions
|
@ -36,6 +36,10 @@ conf_module_LTLIBRARIES = \
|
||||||
conf/conf_serverinfo.la \
|
conf/conf_serverinfo.la \
|
||||||
conf/conf_listen.la
|
conf/conf_listen.la
|
||||||
|
|
||||||
|
dns_moduledir=@moduledir@
|
||||||
|
dns_module_LTLIBRARIES = dns/dns.la
|
||||||
|
dns_dns_la_SOURCES = dns/dns.cc \
|
||||||
|
dns/lookup.cc
|
||||||
|
|
||||||
moduledir=@moduledir@
|
moduledir=@moduledir@
|
||||||
future_la_SOURCES = future.cc
|
future_la_SOURCES = future.cc
|
||||||
|
|
52
modules/dns/dns.cc
Normal file
52
modules/dns/dns.cc
Normal file
|
@ -0,0 +1,52 @@
|
||||||
|
/*
|
||||||
|
* Copyright (C) 2016 Charybdis Development Team
|
||||||
|
* Copyright (C) 2016 Jason Volk <jason@zemos.net>
|
||||||
|
*
|
||||||
|
* Permission to use, copy, modify, and/or distribute this software for any
|
||||||
|
* purpose with or without fee is hereby granted, provided that the above
|
||||||
|
* copyright notice and this permission notice is present in all copies.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||||||
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
|
||||||
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||||
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
||||||
|
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
|
||||||
|
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||||
|
* POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "dns.h"
|
||||||
|
|
||||||
|
namespace ircd {
|
||||||
|
namespace js {
|
||||||
|
|
||||||
|
struct dns
|
||||||
|
:trap
|
||||||
|
{
|
||||||
|
using trap::trap;
|
||||||
|
}
|
||||||
|
dns{"dns"};
|
||||||
|
|
||||||
|
} // namespace js
|
||||||
|
} // namespace ircd
|
||||||
|
|
||||||
|
ip::tcp::resolver tcp_resolver
|
||||||
|
{
|
||||||
|
*ircd::ios
|
||||||
|
};
|
||||||
|
|
||||||
|
ircd::mapi::fini dtors{[]
|
||||||
|
{
|
||||||
|
tcp_resolver.cancel();
|
||||||
|
}};
|
||||||
|
|
||||||
|
ircd::mapi::header IRCD_MODULE
|
||||||
|
{
|
||||||
|
"Client interface to the Domain Name Service",
|
||||||
|
ircd::mapi::NO_FLAGS,
|
||||||
|
&dtors
|
||||||
|
};
|
27
modules/dns/dns.h
Normal file
27
modules/dns/dns.h
Normal file
|
@ -0,0 +1,27 @@
|
||||||
|
/*
|
||||||
|
* Copyright (C) 2016 Charybdis Development Team
|
||||||
|
* Copyright (C) 2016 Jason Volk <jason@zemos.net>
|
||||||
|
*
|
||||||
|
* Permission to use, copy, modify, and/or distribute this software for any
|
||||||
|
* purpose with or without fee is hereby granted, provided that the above
|
||||||
|
* copyright notice and this permission notice is present in all copies.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||||||
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
|
||||||
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||||
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
||||||
|
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
|
||||||
|
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||||
|
* POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <ircd/js/js.h>
|
||||||
|
#include <boost/asio.hpp>
|
||||||
|
|
||||||
|
namespace ip = boost::asio::ip;
|
||||||
|
|
||||||
|
extern ip::tcp::resolver tcp_resolver; // dns.cc
|
95
modules/dns/lookup.cc
Normal file
95
modules/dns/lookup.cc
Normal file
|
@ -0,0 +1,95 @@
|
||||||
|
/*
|
||||||
|
* Copyright (C) 2016 Charybdis Development Team
|
||||||
|
* Copyright (C) 2016 Jason Volk <jason@zemos.net>
|
||||||
|
*
|
||||||
|
* Permission to use, copy, modify, and/or distribute this software for any
|
||||||
|
* purpose with or without fee is hereby granted, provided that the above
|
||||||
|
* copyright notice and this permission notice is present in all copies.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||||||
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
|
||||||
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||||
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
||||||
|
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
|
||||||
|
* IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||||
|
* POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "dns.h"
|
||||||
|
|
||||||
|
namespace ircd {
|
||||||
|
namespace js {
|
||||||
|
|
||||||
|
extern trap dns;
|
||||||
|
|
||||||
|
struct lookup
|
||||||
|
:trap::function
|
||||||
|
{
|
||||||
|
trap &future{trap::find("future")};
|
||||||
|
|
||||||
|
value handle_result(const boost::system::error_code &, ip::tcp::resolver::iterator &);
|
||||||
|
value on_call(object::handle obj, value::handle, const args &args) override;
|
||||||
|
using trap::function::function;
|
||||||
|
}
|
||||||
|
static lookup
|
||||||
|
{
|
||||||
|
dns, "lookup"
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
} // namespace js
|
||||||
|
} // namespace ircd
|
||||||
|
|
||||||
|
// system-level hostname resolution
|
||||||
|
//
|
||||||
|
// args[0]: hostname - string
|
||||||
|
// args[1]: [options] - object or integer; integer = 4 or 6
|
||||||
|
// args[2]: [callback] - function; undefined = yield
|
||||||
|
//
|
||||||
|
// https://nodejs.org/api/dns.html#dns_dns_lookup_hostname_options_callback
|
||||||
|
//
|
||||||
|
ircd::js::value
|
||||||
|
ircd::js::lookup::on_call(object::handle obj,
|
||||||
|
value::handle that,
|
||||||
|
const args &args)
|
||||||
|
{
|
||||||
|
const string hostname{args[0]};
|
||||||
|
const object options{args.has(1)? args[1] : value{}}; //TODO: Node.js options
|
||||||
|
const value callback{args.has(2)? args[2] : value{}};
|
||||||
|
const ip::tcp::resolver::query query
|
||||||
|
{
|
||||||
|
std::string(hostname),
|
||||||
|
std::string{}
|
||||||
|
};
|
||||||
|
|
||||||
|
auto &task(task::get());
|
||||||
|
contract result(ctor(future, vector<value>{{callback}}));
|
||||||
|
tcp_resolver.async_resolve(query, [this, result]
|
||||||
|
(const boost::system::error_code &ec, ip::tcp::resolver::iterator it)
|
||||||
|
mutable
|
||||||
|
{
|
||||||
|
result(std::bind(&lookup::handle_result, this, std::ref(ec), std::ref(it)));
|
||||||
|
});
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
ircd::js::value
|
||||||
|
ircd::js::lookup::handle_result(const boost::system::error_code &ec,
|
||||||
|
ip::tcp::resolver::iterator &it)
|
||||||
|
{
|
||||||
|
static const ip::tcp::resolver::iterator end;
|
||||||
|
|
||||||
|
if(ec)
|
||||||
|
throw jserror("%s", ec.message().c_str());
|
||||||
|
|
||||||
|
vector<value> addrs;
|
||||||
|
for(; it != end; ++it)
|
||||||
|
addrs.append(string(it->endpoint().address().to_string()));
|
||||||
|
|
||||||
|
return object(addrs);
|
||||||
|
}
|
Loading…
Reference in a new issue