Changeset 8d74291 for protocols/jabber/io.c

Timestamp:

2006-09-22T16:56:58Z (18 years ago)

Author:

Wilmer van der Gaast <wilmer@…>

Branches:

master

Children:

fe7a554

Parents:

5997488

Message:

Fixed return value on incomplete write()s in write handler, protection
against write()ing to sockets that are closed already, hopefully sane
detection for SASL support, and only sending type=unavailable presence
tag to logged in sessions.

File:

• protocols/jabber/io.c (modified) (11 diffs)

protocols/jabber/io.c

@@ -73 +73 @@
         int st;
         
+        if( jd->fd == -1 )
+                return FALSE;
+        
         st = write( jd->fd, jd->txq, jd->tx_len );
         
@@ -86 +89 @@
         else if( st == 0 || ( st < 0 && !sockerr_again() ) )
         {
+                /* Set fd to -1 to make sure we won't write to it anymore. */
+                closesocket( jd->fd );  /* Shouldn't be necessary after errors? */
+                jd->fd = -1;
+                
                 hide_login_progress_error( gc, "Short write() to server" );
                 signoff( gc );
@@ -99 +106 @@
                 jd->txq = s;
                 
-                return FALSE;
+                return TRUE;
         }
         else
@@ -115 +122 @@
         char buf[512];
         int st;
+        
+        if( jd->fd == -1 )
+                return FALSE;
         
         st = read( fd, buf, sizeof( buf ) );
@@ -139 +149 @@
                 {
                         jd->flags &= ~JFLAG_STREAM_RESTART;
-                        xt_reset( jd->xt );
                         jabber_start_stream( gc );
                 }
@@ -156 +165 @@
                         {
                                 jd->flags |= JFLAG_STREAM_STARTED;
-                                return jabber_start_auth( gc );
+                                
+                                /* If there's no version attribute, assume
+                                   this is an old server that can't do SASL
+                                   authentication. */
+                                if( !sasl_supported( gc ) )
+                                        return jabber_start_iq_auth( gc );
                         }
                         else
@@ -168 +182 @@
         else if( st == 0 || ( st < 0 && !sockerr_again() ) )
         {
+                closesocket( jd->fd );
+                jd->fd = -1;
+                
                 hide_login_progress_error( gc, "Error while reading from server" );
                 signoff( gc );
@@ -198 +215 @@
 }
 
+static xt_status jabber_pkt_features( struct xt_node *node, gpointer data )
+{
+        struct gaim_connection *gc = data;
+        struct jabber_data *jd = gc->proto_data;
+        struct xt_node *c;
+        
+        c = xt_find_node( node->children, "starttls" );
+        if( c )
+        {
+                /*
+                jd->flags |= JFLAG_SUPPORTS_TLS;
+                if( xt_find_node( c->children, "required" ) )
+                        jd->flags |= JFLAG_REQUIRES_TLS;
+                */
+        }
+        
+        /* This flag is already set if we authenticated via SASL, so now
+           we can resume the session in the new stream. */
+        if( jd->flags & JFLAG_AUTHENTICATED )
+        {
+                if( !jabber_get_roster( gc ) )
+                        return XT_ABORT;
+        }
+        
+        return XT_HANDLED;
+}
+
 static xt_status jabber_pkt_misc( struct xt_node *node, gpointer data )
 {
@@ -208 +252 @@
 static const struct xt_handler_entry jabber_handlers[] = {
         { "stream:stream",      "<root>",               jabber_end_of_stream },
-        { "iq",                 "stream:stream",        jabber_pkt_iq },
         { "message",            "stream:stream",        jabber_pkt_message },
         { "presence",           "stream:stream",        jabber_pkt_presence },
+        { "iq",                 "stream:stream",        jabber_pkt_iq },
+        { "stream:features",    "stream:stream",        jabber_pkt_features },
         { "mechanisms",         "stream:features",      sasl_pkt_mechanisms },
         { "challenge",          "stream:stream",        sasl_pkt_challenge },
@@ -231 +276 @@
         jd->xt->handlers = (struct xt_handler_entry*) jabber_handlers;
         
-        jd->r_inpa = b_input_add( jd->fd, GAIM_INPUT_READ, jabber_read_callback, gc );
+        if( jd->r_inpa <= 0 )
+                jd->r_inpa = b_input_add( jd->fd, GAIM_INPUT_READ, jabber_read_callback, gc );
         
         greet = g_strdup_printf( "<?xml version='1.0' ?>"
@@ -254 +300 @@
                 char eos[] = "</stream:stream>";
                 struct xt_node *node;
-                int st;
-                
-                node = jabber_make_packet( "presence", "unavailable", NULL, NULL );
-                st = jabber_write_packet( gc, node );
-                xt_free_node( node );
+                int st = 1;
+                
+                if( gc->flags & OPT_LOGGED_IN )
+                {
+                        node = jabber_make_packet( "presence", "unavailable", NULL, NULL );
+                        st = jabber_write_packet( gc, node );
+                        xt_free_node( node );
+                }
                 
                 if( st )