Filter TLS support from server

[cascardo/rnetproxy.git] / jabber_server.c

diff --git a/jabber_server.c b/jabber_server.c
index b4fe6c3..728aab4 100644 (file)
--- a/jabber_server.c
+++ b/jabber_server.c
@@ -20,6 +20,7 @@
 #include <gnet.h>
 #include <glib.h>
 #include <iksemel.h>
+#include "iksemel_extra.h"
 #include "jabber.h"
 
 static void jabber_server_connect (net_hook_t* hook)
@@ -29,18 +30,65 @@ static void jabber_server_connect (net_hook_t* hook)
 
 static void jabber_server_close (net_hook_t* hook)
 {
+  if (hook->peer)
+    {
+      hook->peer->peer = NULL;
+      gnet_conn_disconnect (hook->peer->conn);
+    }
+  gnet_conn_delete (hook->conn);
+  iks_parser_delete (hook->data);
+  g_slice_free (net_hook_t, hook);
   g_message ("Server disconnected.");
 }
 
 static void jabber_server_write (net_hook_t* hook)
 {
-  g_message ("Have written bytes to server.");
 }
 
 static void jabber_server_read (net_hook_t* hook, gchar* buffer, size_t len)
 {
-  g_message ("Received response from server.");
-  gnet_conn_write (hook->peer, buffer, len);
+  iks_parse (hook->data, buffer, len, FALSE);
+}
+
+void jabber_server_tls_filter (iks* node)
+{
+  iks* tls;
+  if (g_str_equal (iks_name (node), "stream:features"))
+    {
+      tls = iks_find (node, "starttls");
+      if (tls)
+       iks_hide (tls);
+    }
+}
+
+int jabber_server_parser (gpointer data, int type, iks* node)
+{
+  net_hook_t* hook;
+  GString* buffer;
+  hook = (net_hook_t*) data;
+  switch (type)
+    {
+    case IKS_NODE_START:
+      buffer = jabber_new_start (node);
+      gnet_conn_write (hook->peer->conn, buffer->str, buffer->len);
+      g_string_free (buffer, TRUE);
+      iks_delete (node);
+      break;
+    case IKS_NODE_NORMAL:
+      jabber_server_tls_filter (node);
+      buffer = g_string_new (iks_string (iks_stack (node), node));
+      gnet_conn_write (hook->peer->conn, buffer->str, buffer->len);
+      g_string_free (buffer, TRUE);
+      iks_delete (node);
+      break;
+    case IKS_NODE_STOP:
+      gnet_conn_write (hook->peer->conn, "</stream:stream>", 16);
+      break;
+    case IKS_NODE_ERROR:
+      g_debug ("Parse error!!");
+      break;
+    }
+  return IKS_OK;
 }
 
 net_hook_t* jabber_server_hook_new (net_hook_t* client_hook, char* server)
@@ -48,12 +96,12 @@ net_hook_t* jabber_server_hook_new (net_hook_t* client_hook, char* server)
   net_hook_t* hook;
   hook = g_slice_new (net_hook_t);
   hook->conn = gnet_conn_new (server, 5222, nethook_event, hook);
-  hook->peer = client_hook->conn;
+  hook->peer = client_hook;
   hook->server = TRUE;
   hook->connect = jabber_server_connect;
   hook->close = jabber_server_close;
   hook->write = jabber_server_write;
   hook->read = jabber_server_read;
-  hook->data = NULL; /*iks_stream_new ("jabber:client", hook, jabber_server_parser);*/
+  hook->data = iks_extra_stream_new (hook, jabber_server_parser);
   return hook;
 }