proxy advertises only SASL PLAIN mechanism if it is supported by server
[cascardo/rnetproxy.git] / jabber_server.c
1 /*
2 ** Copyright (C) 2006 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
3 **  
4 ** This program is free software; you can redistribute it and/or modify
5 ** it under the terms of the GNU General Public License as published by
6 ** the Free Software Foundation; either version 2 of the License, or
7 ** (at your option) any later version.
8 **  
9 ** This program is distributed in the hope that it will be useful,
10 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
11 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 ** GNU General Public License for more details.
13 **  
14 ** You should have received a copy of the GNU General Public License
15 ** along with this program; if not, write to the Free Software
16 ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17 **  
18 */
19
20 #include <gnet.h>
21 #include <glib.h>
22 #include <iksemel.h>
23 #include "iksemel_extra.h"
24 #include "jabber.h"
25
26 static void jabber_server_connect (net_hook_t* hook)
27 {
28   g_message ("Connected to jabber server.");
29 }
30
31 static void jabber_server_close (net_hook_t* hook)
32 {
33   if (hook->peer)
34     {
35       hook->peer->peer = NULL;
36       gnet_conn_disconnect (hook->peer->conn);
37     }
38   gnet_conn_delete (hook->conn);
39   iks_parser_delete (hook->data);
40   g_slice_free (net_hook_t, hook);
41   g_message ("Server disconnected.");
42 }
43
44 static void jabber_server_write (net_hook_t* hook)
45 {
46 }
47
48 static void jabber_server_read (net_hook_t* hook, gchar* buffer, size_t len)
49 {
50   iks_parse (hook->data, buffer, len, FALSE);
51 }
52
53 void jabber_server_tls_filter (iks* node)
54 {
55   iks* tls;
56   if (g_str_equal (iks_name (node), "stream:features"))
57     {
58       tls = iks_find (node, "starttls");
59       if (tls)
60         iks_hide (tls);
61     }
62 }
63
64 void jabber_server_sasl_filter (iks* node)
65 {
66   iks* sasl;
67   GString* mechs = NULL;
68   gboolean plain = FALSE;
69   if (g_str_equal (iks_name (node), "stream:features"))
70     {
71       sasl = iks_find (node, "mechanisms");
72       if (sasl && g_str_equal (iks_find_attrib (sasl, "xmlns"),
73                                "urn:ietf:params:xml:ns:xmpp-sasl"))
74         {
75           mechs = g_string_sized_new (256);
76           iks* mech;
77           for (mech = iks_child (sasl); mech != NULL; mech = iks_next (mech))
78             {
79               char* mech_name;
80               mech_name = iks_cdata (iks_child (mech));
81               if (g_str_equal (mech_name, "PLAIN"))
82                 plain = TRUE;
83               mechs = g_string_append (mechs, mech_name);
84               mechs = g_string_append_c (mechs, ' ');
85             }
86           g_debug ("Mechanisms supported by server: %s", mechs->str);
87           g_string_free (mechs, TRUE);
88           if (plain)
89             {
90               for (mech = iks_child (sasl); mech != NULL; mech = iks_next (mech))
91                 {
92                   char* mech_name;
93                   mech_name = iks_cdata (iks_child (mech));
94                   if (!g_str_equal (mech_name, "PLAIN"))
95                     iks_hide (mech);
96                 }
97             }
98         }
99     }
100 }
101
102 int jabber_server_parser (gpointer data, int type, iks* node)
103 {
104   net_hook_t* hook;
105   GString* buffer;
106   hook = (net_hook_t*) data;
107   switch (type)
108     {
109     case IKS_NODE_START:
110       buffer = jabber_new_start (node);
111       gnet_conn_write (hook->peer->conn, buffer->str, buffer->len);
112       g_string_free (buffer, TRUE);
113       iks_delete (node);
114       break;
115     case IKS_NODE_NORMAL:
116       jabber_server_tls_filter (node);
117       jabber_server_sasl_filter (node);
118       buffer = g_string_new (iks_string (iks_stack (node), node));
119       gnet_conn_write (hook->peer->conn, buffer->str, buffer->len);
120       g_string_free (buffer, TRUE);
121       iks_delete (node);
122       break;
123     case IKS_NODE_STOP:
124       gnet_conn_write (hook->peer->conn, "</stream:stream>", 16);
125       break;
126     case IKS_NODE_ERROR:
127       g_debug ("Parse error!!");
128       break;
129     }
130   return IKS_OK;
131 }
132
133 net_hook_t* jabber_server_hook_new (net_hook_t* client_hook, char* server)
134 {
135   net_hook_t* hook;
136   hook = g_slice_new (net_hook_t);
137   hook->conn = gnet_conn_new (server, 5222, nethook_event, hook);
138   hook->peer = client_hook;
139   hook->server = TRUE;
140   hook->connect = jabber_server_connect;
141   hook->close = jabber_server_close;
142   hook->write = jabber_server_write;
143   hook->read = jabber_server_read;
144   hook->data = iks_extra_stream_new (hook, jabber_server_parser);
145   return hook;
146 }