moxdns/packages/moxdns_linux/linux/moxdns_linux_plugin.cc

115 lines
4.0 KiB
C++
Raw Normal View History

2022-04-29 13:48:36 +00:00
#include "include/moxdns_linux/moxdns_linux_plugin.h"
#include <flutter_linux/flutter_linux.h>
#include <gtk/gtk.h>
#include <netinet/in.h>
#include <resolv.h>
#include <arpa/nameser.h>
#include <cstring>
#include <iostream>
#define MOXDNS_LINUX_PLUGIN(obj) \
(G_TYPE_CHECK_INSTANCE_CAST((obj), moxdns_linux_plugin_get_type(), \
MoxdnsLinuxPlugin))
struct _MoxdnsLinuxPlugin {
GObject parent_instance;
};
G_DEFINE_TYPE(MoxdnsLinuxPlugin, moxdns_linux_plugin, g_object_get_type())
// Convert the data field at offset [offset] of the RR [record] into an integer
// [FlValue].
FlValue *rr_data_field_to_value(ns_rr &record, int offset) {
auto data = ntohs(*((unsigned short*)ns_rr_rdata(record) + offset));
return fl_value_new_int(data);
}
// Called when a method call is received from Flutter.
static void moxdns_linux_plugin_handle_method_call(
MoxdnsLinuxPlugin* self,
FlMethodCall* method_call) {
g_autoptr(FlMethodResponse) response = nullptr;
const gchar* method = fl_method_call_get_name(method_call);
if (strcmp(method, "srvQuery") == 0) {
FlValue* args = fl_method_call_get_args(method_call);
gchar *domain = fl_value_to_string(fl_value_lookup_string(args, "domain"));
//bool dnssec = fl_value_get_bool(fl_value_lookup_string(args, "dnssec"));
res_init();
// This is based on this gist: https://gist.githubusercontent.com/ajdavis/e5f5ddbf50b5aecdc5e1d686d72a8a7a/raw/31ec3c642d5ca4a3d0688dde8cc4a6ef8bd97146/dnssd.cpp
unsigned char query_buffer[1024];
int result = res_query(domain, C_IN, ns_t_srv, query_buffer, sizeof(query_buffer));
ns_msg records;
ns_initparse(query_buffer, result, &records);
FlValue *results = fl_value_new_list();
for (int i = 0; i < ns_msg_count(records, ns_s_an); i++) {
ns_rr record;
ns_parserr(&records, ns_s_an, i, &record);
char target[1024];
dn_expand(ns_msg_base(records),
ns_msg_end(records),
ns_rr_rdata(record) + 6,
target, sizeof(target));
FlValue *map = fl_value_new_map();
fl_value_set(map,
fl_value_new_string("target"),
fl_value_new_string(target));
fl_value_set(map,
fl_value_new_string("port"),
rr_data_field_to_value(record, 2));
fl_value_set(map,
fl_value_new_string("priority"),
rr_data_field_to_value(record, 0));
fl_value_set(map,
fl_value_new_string("weight"),
rr_data_field_to_value(record, 1));
fl_value_append(results, map);
}
response = FL_METHOD_RESPONSE(fl_method_success_response_new(results));
} else {
response = FL_METHOD_RESPONSE(fl_method_not_implemented_response_new());
}
fl_method_call_respond(method_call, response, nullptr);
}
static void moxdns_linux_plugin_dispose(GObject* object) {
G_OBJECT_CLASS(moxdns_linux_plugin_parent_class)->dispose(object);
}
static void moxdns_linux_plugin_class_init(MoxdnsLinuxPluginClass* klass) {
G_OBJECT_CLASS(klass)->dispose = moxdns_linux_plugin_dispose;
}
static void moxdns_linux_plugin_init(MoxdnsLinuxPlugin* self) {}
static void method_call_cb(FlMethodChannel* channel, FlMethodCall* method_call,
gpointer user_data) {
MoxdnsLinuxPlugin* plugin = MOXDNS_LINUX_PLUGIN(user_data);
moxdns_linux_plugin_handle_method_call(plugin, method_call);
}
void moxdns_linux_plugin_register_with_registrar(FlPluginRegistrar* registrar) {
MoxdnsLinuxPlugin* plugin = MOXDNS_LINUX_PLUGIN(
g_object_new(moxdns_linux_plugin_get_type(), nullptr));
g_autoptr(FlStandardMethodCodec) codec = fl_standard_method_codec_new();
g_autoptr(FlMethodChannel) channel =
fl_method_channel_new(fl_plugin_registrar_get_messenger(registrar),
"me.polynom.moxdns_linux",
FL_METHOD_CODEC(codec));
fl_method_channel_set_method_call_handler(channel, method_call_cb,
g_object_ref(plugin),
g_object_unref);
g_object_unref(plugin);
}