diff options
Diffstat (limited to 'xmpp-vala/src/core/xmpp_stream.vala')
| -rw-r--r-- | xmpp-vala/src/core/xmpp_stream.vala | 147 |
1 files changed, 138 insertions, 9 deletions
diff --git a/xmpp-vala/src/core/xmpp_stream.vala b/xmpp-vala/src/core/xmpp_stream.vala index 0283920b..684c7ed0 100644 --- a/xmpp-vala/src/core/xmpp_stream.vala +++ b/xmpp-vala/src/core/xmpp_stream.vala @@ -10,11 +10,133 @@ public errordomain IOStreamError { } +public class XmppLog { + protected const string ANSI_COLOR_END = "\x1b[0m"; + protected const string ANSI_COLOR_WHITE = "\x1b[37;1m"; + + class NodeLogDesc { + public string? name; + private string? ns_uri; + private string? val; + private Map<string, string?> attrs = new HashMap<string, string?>(); + private NodeLogDesc? inner; + + public NodeLogDesc(string desc) { + string d = desc; + + if (d.contains("[")) { + int start = d.index_of("["); + int end = d.index_of("]"); + string attrs = d.substring(start + 1, end - start - 1); + d = d.substring(0, start) + d.substring(end + 1); + foreach (string attr in attrs.split(",")) { + if (attr.contains("=")) { + string key = attr.substring(0, attr.index_of("=")); + string val = attr.substring(attr.index_of("=") + 1); + this.attrs[key] = val; + } else { + this.attrs[attr] = null; + } + } + } + if (d.contains(":") && d.index_of("{") == 0 && d.index_of("}") != -1) { + int end = d.index_of("}"); + this.ns_uri = d.substring(1, end - 2); + d = d.substring(end + 2); + } + if (d.contains(".")) { + inner = new NodeLogDesc(d.substring(d.index_of(".") + 1)); + d = d.substring(0, d.index_of(".")); + } else if (d.contains("=")) { + this.val = d.substring(d.index_of("=")); + d = d.substring(0, d.index_of("=")); + } + + if (d != "") this.name = d; + } + + public bool matches(StanzaNode node) { + if (name != null && node.name != name) return false; + if (ns_uri != null && node.ns_uri != ns_uri) return false; + if (val != null && node.val != val) return false; + foreach (var pair in attrs.entries) { + if (pair.value == null && node.get_attribute(pair.key) == null) return false; + else if (pair.value != null && pair.value != node.get_attribute(pair.key)) return false; + } + if (inner == null) return true; + foreach (StanzaNode snode in node.get_all_subnodes()) { + if (inner.matches(snode)) return true; + } + return false; + } + } + + private bool use_ansi; + private bool hide_ns = true; + private string ident; + private string desc; + private ArrayList<NodeLogDesc> descs = new ArrayList<NodeLogDesc>(); + + public XmppLog(string? ident = null, string? desc = null) { + this.ident = ident; + this.desc = desc; + this.use_ansi = is_atty(stderr.fileno()); + while (this.desc != null && this.desc.contains(";")) { + string opt = this.desc.substring(0, this.desc.index_of(";")); + this.desc = this.desc.substring(opt.length + 1); + switch (opt) { + case "ansi": use_ansi = true; break; + case "no-ansi": use_ansi = false; break; + case "hide-ns": hide_ns = true; break; + case "show-ns": hide_ns = false; break; + } + } + if (desc != null) { + foreach (string d in this.desc.split("|")) { + descs.add(new NodeLogDesc(d)); + } + } + } + + public virtual bool should_log_node(StanzaNode node) { + if (ident == null || desc == null) return false; + if (desc == "all") return true; + foreach (var desc in descs) { + if (desc.matches(node)) return true; + } + return false; + } + + public virtual bool should_log_str(string str) { + if (ident == null || desc == null) return false; + if (desc == "all") return true; + foreach (var desc in descs) { + if (desc.name == "#text") return true; + } + return false; + } + + public void node(string what, StanzaNode node) { + if (should_log_node(node)) { + stderr.printf("%sXMPP %s [%s]%s\n%s\n", ANSI_COLOR_WHITE, what, ident, ANSI_COLOR_END, use_ansi ? node.to_ansi_string(hide_ns) : node.to_string()); + } + } + + public void str(string what, string str) { + if (should_log_str(str)) { + stderr.printf("%sXMPP %s [%s]%s\n%s\n", ANSI_COLOR_WHITE, what, ident, ANSI_COLOR_END, str); + } + } + + [CCode (cname = "isatty")] + private static extern bool is_atty(int fd); +} + public class XmppStream { private static string NS_URI = "http://etherx.jabber.org/streams"; public string remote_name; - public bool debug = false; + public XmppLog log = new XmppLog(); public StanzaNode? features { get; private set; default = new StanzaNode.build("features", NS_URI); } private IOStream? stream; @@ -51,7 +173,7 @@ public class XmppStream { public void disconnect() throws IOStreamError { if (writer == null) throw new IOStreamError.DISCONNECT("trying to disconnect, but no stream open"); - if (debug) stderr.puts("OUT\n</stream:stream>\n"); + log.str("OUT", "</stream:stream>"); writer.write.begin("</stream:stream>"); reader.cancel(); stream.close_async.begin(); @@ -75,8 +197,8 @@ public class XmppStream { public StanzaNode read() throws IOStreamError { if (reader == null) throw new IOStreamError.READ("trying to read, but no stream open"); try { - var node = reader.read_node(); - if (debug) stderr.printf("IN\n%s\n", node.to_string()); + StanzaNode node = reader.read_node(); + log.node("IN", node); return node; } catch (XmlError e) { throw new IOStreamError.READ(e.message); @@ -86,7 +208,7 @@ public class XmppStream { public void write(StanzaNode node) throws IOStreamError { if (writer == null) throw new IOStreamError.WRITE("trying to write, but no stream open"); try { - if (debug) stderr.printf("OUT\n%s\n", node.to_string()); + log.node("OUT", node); writer.write_node(node); } catch (XmlError e) { throw new IOStreamError.WRITE(e.message); @@ -138,18 +260,19 @@ public class XmppStream { } private void setup() throws IOStreamError { - var outs = new StanzaNode.build("stream", "http://etherx.jabber.org/streams") + StanzaNode outs = new StanzaNode.build("stream", "http://etherx.jabber.org/streams") .put_attribute("to", remote_name) .put_attribute("version", "1.0") .put_attribute("xmlns", "jabber:client") .put_attribute("stream", "http://etherx.jabber.org/streams", XMLNS_URI); outs.has_nodes = true; + log.node("OUT ROOT", outs); write(outs); received_root_node(this, read_root()); } private void loop() throws IOStreamError { - while(true) { + while (true) { if (setup_needed) { setup(); setup_needed = false; @@ -219,8 +342,8 @@ public class XmppStream { private StanzaNode read_root() throws IOStreamError { try { - var node = reader.read_root_node(); - if (debug) stderr.printf("IN\n%s\n", node.to_string()); + StanzaNode node = reader.read_root_node(); + log.node("IN ROOT", node); return node; } catch (XmlError e) { throw new IOStreamError.READ(e.message); @@ -248,6 +371,7 @@ public class FlagIdentity<T> : Object { public abstract class XmppStreamFlag : Object { public abstract string get_ns(); + public abstract string get_id(); } @@ -271,13 +395,18 @@ public class ModuleIdentity<T> : Object { public abstract class XmppStreamModule : Object { public abstract void attach(XmppStream stream); + public abstract void detach(XmppStream stream); + public abstract string get_ns(); + public abstract string get_id(); } public abstract class XmppStreamNegotiationModule : XmppStreamModule { public abstract bool mandatory_outstanding(XmppStream stream); + public abstract bool negotiation_active(XmppStream stream); } + } |