dino/xmpp-vala/src/module/xep/0065_socks5_bytestreams.vala

97 lines
3.5 KiB
Vala

using Gee;
using Xmpp;
using Xmpp.Xep;
namespace Xmpp.Xep.Socks5Bytestreams {
internal const string NS_URI = "http://jabber.org/protocol/bytestreams";
public class Proxy : Object {
public string host { get; private set; }
public Jid jid { get; private set; }
public int port { get; private set; }
public Proxy(string host, Jid jid, int port) {
this.host = host;
this.jid = jid;
this.port = port;
}
}
public delegate Gee.List<string> GetLocalIpAddresses();
public class Module : XmppStreamModule, Iq.Handler {
public static Xmpp.ModuleIdentity<Module> IDENTITY = new Xmpp.ModuleIdentity<Module>(NS_URI, "0065_socks5_bytestreams");
private GetLocalIpAddresses? get_local_ip_addresses_impl = null;
public override void attach(XmppStream stream) {
stream.add_flag(new Flag());
query_availability.begin(stream);
}
public override void detach(XmppStream stream) { }
public Gee.List<Proxy> get_proxies(XmppStream stream) {
return stream.get_flag(Flag.IDENTITY).proxies;
}
public void set_local_ip_address_handler(owned GetLocalIpAddresses get_local_ip_addresses) {
get_local_ip_addresses_impl = (owned)get_local_ip_addresses;
}
public Gee.List<string> get_local_ip_addresses() {
if (get_local_ip_addresses_impl == null) {
return Gee.List.empty();
}
return get_local_ip_addresses_impl();
}
private async void query_availability(XmppStream stream) {
ServiceDiscovery.ItemsResult? items_result = yield stream.get_module(ServiceDiscovery.Module.IDENTITY).request_items(stream, stream.remote_name);
if (items_result == null) return;
foreach (Xep.ServiceDiscovery.Item item in items_result.items) {
bool has_feature = yield stream.get_module(ServiceDiscovery.Module.IDENTITY).has_entity_feature(stream, item.jid, NS_URI);
if (!has_feature) continue;
StanzaNode query_node = new StanzaNode.build("query", NS_URI).add_self_xmlns();
Iq.Stanza iq = new Iq.Stanza.get(query_node) { to=item.jid };
Iq.Stanza iq_result = yield stream.get_module(Iq.Module.IDENTITY).send_iq_async(stream, iq);
if (iq_result.is_error()) continue;
StanzaNode? query_result_node = iq_result.stanza.get_subnode("query", NS_URI);
StanzaNode? stream_host = query_result_node != null ? query_result_node.get_subnode("streamhost", NS_URI) : null;
if (query_result_node == null || stream_host == null) {
return;
}
string? host = stream_host.get_attribute("host");
string? jid_str = stream_host.get_attribute("jid");
Jid? jid = null;
try {
jid = jid_str != null ? new Jid(jid_str) : null;
} catch (InvalidJidError ignored) { }
int port = stream_host.get_attribute_int("port");
if (host == null || jid == null || port <= 0 || port > 65535) {
continue;
}
stream.get_flag(Flag.IDENTITY).proxies.add(new Proxy(host, jid, port));
}
}
public override string get_ns() { return NS_URI; }
public override string get_id() { return IDENTITY.id; }
}
public class Flag : XmppStreamFlag {
public static FlagIdentity<Flag> IDENTITY = new FlagIdentity<Flag>(NS_URI, "socks5_bytestreams");
public Gee.List<Proxy> proxies = new ArrayList<Proxy>();
public override string get_ns() { return NS_URI; }
public override string get_id() { return IDENTITY.id; }
}
}