Changes in protocols/jabber/s5bytestream.c [9216eff:5ebff60]

File:

• protocols/jabber/s5bytestream.c (modified) (23 diffs)

protocols/jabber/s5bytestream.c

@@ -23 +23 @@
 
 #include "jabber.h"
+#include "sha1.h"
 #include "lib/ftutil.h"
 #include <poll.h>
@@ -41 +42 @@
 
         /* SHA1( SID + Initiator JID + Target JID) */
-        char *pseudoaddr;
+        char *pseudoadr;
 
         gint connect_timeout;
@@ -129 +130 @@
         }
 
-        g_free(bt->pseudoaddr);
+        g_free(bt->pseudoadr);
 
         while (bt->streamhosts) {
@@ -253 +254 @@
 }
 
-void jabber_bs_remove_events(struct bs_transfer *bt)
-{
-        struct jabber_transfer *tf = bt->tf;
-
-        if (tf->watch_out) {
-                b_event_remove(tf->watch_out);
-                tf->watch_out = 0;
-        }
-
-        if (tf->watch_in) {
-                b_event_remove(tf->watch_in);
-                tf->watch_in = 0;
-        }
-
-        if (tf->fd != -1) {
-                closesocket(tf->fd);
-                tf->fd = -1;
-        }
-
-        if (bt->connect_timeout) {
-                b_event_remove(bt->connect_timeout);
-                bt->connect_timeout = 0;
-        }
-}
-
 /* Bad luck */
 void jabber_bs_canceled(file_transfer_t *ft, char *reason)
@@ -284 +260 @@
 
         imcb_log(tf->ic, "File transfer aborted: %s", reason);
-}
-
-static struct jabber_transfer *get_ft_by_sid(GSList *tflist, char *sid) {
-        GSList *l;
-        for (l = tflist; l; l = g_slist_next(l)) {
-                struct jabber_transfer *tft = l->data;
-                if ((strcmp(tft->sid, sid) == 0)) { 
-                        return tft;
-                }
-        }
-        return NULL;
-}
-
-/* SHA1( SID + Initiator JID + Target JID ) is given to the streamhost which it will match against the initiator's value
- * Returns a newly allocated string */
-static char *generate_pseudoaddr(char *sid, char *ini_jid, char *tgt_jid) {
-        char *contents = g_strconcat(sid, ini_jid, tgt_jid, NULL);
-        char *hash_hex = g_compute_checksum_for_string(G_CHECKSUM_SHA1, contents, -1);
-        g_free(contents);
-        return hash_hex;
-}
-
-static jabber_streamhost_t *jabber_streamhost_new(char *jid, char *host, int port)
-{
-        jabber_streamhost_t *sh = g_new0(jabber_streamhost_t, 1);
-        sh->jid = g_strdup(jid);
-        sh->host = g_strdup(host);
-        if (port) {
-                g_snprintf(sh->port, sizeof(sh->port), "%u", port);
-        }
-        return sh;
 }
 
@@ -325 +270 @@
         struct jabber_data *jd = ic->proto_data;
         struct jabber_transfer *tf = NULL;
+        GSList *tflist;
         struct bs_transfer *bt;
         GSList *shlist = NULL;
         struct xt_node *shnode;
+
+        sha1_state_t sha;
+        char hash_hex[41];
+        unsigned char hash[20];
+        int i;
 
         if (!(iq_id   = xt_find_attr(node, "id")) ||
@@ -352 +303 @@
                     (port_s = xt_find_attr(shnode, "port")) &&
                     (sscanf(port_s, "%d", &port) == 1)) {
-                        shlist = g_slist_append(shlist, jabber_streamhost_new(jid, host, port));
+                        jabber_streamhost_t *sh = g_new0(jabber_streamhost_t, 1);
+                        sh->jid = g_strdup(jid);
+                        sh->host = g_strdup(host);
+                        sprintf(sh->port, "%u", port);
+                        shlist = g_slist_append(shlist, sh);
                 }
                 shnode = shnode->next;
@@ -362 +317 @@
         }
 
-        if (!(tf = get_ft_by_sid(jd->filetransfers, sid))) {
+        /* Let's see if we can find out what this bytestream should be for... */
+
+        for (tflist = jd->filetransfers; tflist; tflist = g_slist_next(tflist)) {
+                struct jabber_transfer *tft = tflist->data;
+                if ((strcmp(tft->sid, sid) == 0) &&
+                    (strcmp(tft->ini_jid, ini_jid) == 0) &&
+                    (strcmp(tft->tgt_jid, tgt_jid) == 0)) {
+                        tf = tft;
+                        break;
+                }
+        }
+
+        if (!tf) {
                 imcb_log(ic, "WARNING: Received bytestream request from %s that doesn't match an SI request", ini_jid);
                 return XT_HANDLED;
@@ -372 +339 @@
 
         tf->ft->canceled = jabber_bs_canceled;
+
+        /* SHA1( SID + Initiator JID + Target JID ) is given to the streamhost which it will match against the initiator's value */
+        sha1_init(&sha);
+        sha1_append(&sha, (unsigned char *) sid, strlen(sid));
+        sha1_append(&sha, (unsigned char *) ini_jid, strlen(ini_jid));
+        sha1_append(&sha, (unsigned char *) tgt_jid, strlen(tgt_jid));
+        sha1_finish(&sha, hash);
+
+        for (i = 0; i < 20; i++) {
+                sprintf(hash_hex + i * 2, "%02x", hash[i]);
+        }
 
         bt = g_new0(struct bs_transfer, 1);
@@ -378 +356 @@
         bt->sh = shlist->data;
         bt->phase = BS_PHASE_CONNECT;
-        bt->pseudoaddr = generate_pseudoaddr(sid, ini_jid, tgt_jid);
+        bt->pseudoadr = g_strdup(hash_hex);
         tf->streamhandle = bt;
         tf->ft->free = jabber_bs_free_transfer;
@@ -473 +451 @@
                         .rsv = 0,
                         .atyp = 0x03,
-                        .addrlen = strlen(bt->pseudoaddr),
+                        .addrlen = strlen(bt->pseudoadr),
                         .port = 0
                 };
@@ -490 +468 @@
 
                 /* copy hash into connect message */
-                memcpy(socks5_connect.address, bt->pseudoaddr, socks5_connect.addrlen);
+                memcpy(socks5_connect.address, bt->pseudoadr, socks5_connect.addrlen);
 
                 ASSERTSOCKOP(send(fd, &socks5_connect, sizeof(struct socks5_message), 0), "Sending SOCKS5 Connect");
@@ -580 +558 @@
         if (shlist && shlist->next) {
                 bt->sh = shlist->next->data;
-                jabber_bs_remove_events(bt);
                 return jabber_bs_recv_handshake(bt, -1, 0);
         }
@@ -754 +731 @@
         struct jabber_data *jd = ic->proto_data;
         struct bs_transfer *bt;
+        GSList *tflist;
         struct xt_node *c;
         char *sid, *jid;
@@ -772 +750 @@
         /* Let's see if we can find out what this bytestream should be for... */
 
-        if (!(tf = get_ft_by_sid(jd->filetransfers, sid))) {
+        for (tflist = jd->filetransfers; tflist; tflist = g_slist_next(tflist)) {
+                struct jabber_transfer *tft = tflist->data;
+                if ((strcmp(tft->sid, sid) == 0)) {
+                        tf = tft;
+                        break;
+                }
+        }
+
+        if (!tf) {
                 imcb_log(ic, "WARNING: Received SOCKS5 bytestream reply to unknown request");
                 return XT_HANDLED;
@@ -790 +776 @@
                 /* using a proxy, abort listen */
 
-                jabber_bs_remove_events(bt);
+                if (tf->watch_in) {
+                        b_event_remove(tf->watch_in);
+                        tf->watch_in = 0;
+                }
+
+                if (tf->fd != -1) {
+                        closesocket(tf->fd);
+                        tf->fd = -1;
+                }
+
+                if (bt->connect_timeout) {
+                        b_event_remove(bt->connect_timeout);
+                        bt->connect_timeout = 0;
+                }
 
                 GSList *shlist;
@@ -839 +838 @@
 {
         char *sid;
+        GSList *tflist;
         struct jabber_transfer *tf = NULL;
         struct xt_node *query;
@@ -846 +846 @@
         sid = xt_find_attr(query, "sid");
 
-        if (!(tf = get_ft_by_sid(jd->filetransfers, sid))) {
+        for (tflist = jd->filetransfers; tflist; tflist = g_slist_next(tflist)) {
+                struct jabber_transfer *tft = tflist->data;
+                if ((strcmp(tft->sid, sid) == 0)) {
+                        tf = tft;
+                        break;
+                }
+        }
+
+        if (!tf) {
                 imcb_log(ic, "WARNING: Received SOCKS5 bytestream activation for unknown stream");
                 return XT_HANDLED;
@@ -875 +883 @@
         *port++ = '\0';
 
-        sh = jabber_streamhost_new(jid, host, 0);
-        strncpy(sh->port, port, sizeof(sh->port));
+        sh = g_new0(jabber_streamhost_t, 1);
+        sh->jid = g_strdup(jid);
+        sh->host = g_strdup(host);
+        g_snprintf(sh->port, sizeof(sh->port), "%s", port);
 
         return sh;
@@ -900 +910 @@
                 if (strcmp(proxy, "<local>") == 0) {
                         if ((tf->fd = ft_listen(&tf->saddr, host, port, jd->fd, FALSE, &errmsg)) != -1) {
-                                sh = jabber_streamhost_new(tf->ini_jid, host, 0);
-                                strncpy(sh->port, port, sizeof(sh->port));
+                                sh = g_new0(jabber_streamhost_t, 1);
+                                sh->jid = g_strdup(tf->ini_jid);
+                                sh->host = g_strdup(host);
+                                g_snprintf(sh->port, sizeof(sh->port), "%s", port);
                                 bt->streamhosts = g_slist_append(bt->streamhosts, sh);
 
@@ -936 +948 @@
 {
         struct bs_transfer *bt;
+        sha1_state_t sha;
+        char hash_hex[41];
+        unsigned char hash[20];
+        int i, ret;
+
+        /* SHA1( SID + Initiator JID + Target JID ) is given to the streamhost which it will match against the initiator's value */
+        sha1_init(&sha);
+        sha1_append(&sha, (unsigned char *) tf->sid, strlen(tf->sid));
+        sha1_append(&sha, (unsigned char *) tf->ini_jid, strlen(tf->ini_jid));
+        sha1_append(&sha, (unsigned char *) tf->tgt_jid, strlen(tf->tgt_jid));
+        sha1_finish(&sha, hash);
+
+        for (i = 0; i < 20; i++) {
+                sprintf(hash_hex + i * 2, "%02x", hash[i]);
+        }
 
         bt = g_new0(struct bs_transfer, 1);
         bt->tf = tf;
         bt->phase = BS_PHASE_CONNECT;
-        bt->pseudoaddr = generate_pseudoaddr(tf->sid, tf->ini_jid, tf->tgt_jid);
+        bt->pseudoadr = g_strdup(hash_hex);
         tf->streamhandle = bt;
         tf->ft->free = jabber_bs_free_transfer;
@@ -947 +974 @@
         jabber_si_set_proxies(bt);
 
-        return jabber_bs_send_request(tf, bt->streamhosts);
+        ret = jabber_bs_send_request(tf, bt->streamhosts);
+
+        return ret;
 }
 
@@ -1110 +1139 @@
                                                socks5_connect.atyp);
                 }
-                if (!(memcmp(socks5_connect.address, bt->pseudoaddr, 40) == 0)) {
+                if (!(memcmp(socks5_connect.address, bt->pseudoadr, 40) == 0)) {
                         return jabber_bs_abort(bt, "SOCKS5 Connect message contained wrong digest");
                 }