source: protocols/msn/gw.c @ dd43c62

Last change on this file since dd43c62 was 088b070, checked in by dequis <dx@…>, at 2015-05-08T02:02:14Z

msn: fix use-after-free when the server sends OUT

Also fix a leak in msn_ns_callback while i'm at it.

Also fix a potential null deref when req->reply_body is null and
BITLBEE_DEBUG is enabled, but i don't even know if this one can happen.

  • Property mode set to 100644
File size: 4.3 KB
Line 
1#include "bitlbee.h"
2#include "lib/http_client.h"
3#include "msn.h"
4
5#define GATEWAY_HOST "geo.gateway.messenger.live.com"
6#define GATEWAY_PORT 443
7
8#define REQUEST_TEMPLATE \
9        "POST /gateway/gateway.dll?SessionID=%s&%s HTTP/1.1\r\n" \
10        "Host: %s\r\n" \
11        "Content-Length: %zd\r\n" \
12        "\r\n" \
13        "%s"
14
15static gboolean msn_gw_poll_timeout(gpointer data, gint source, b_input_condition cond);
16
17struct msn_gw *msn_gw_new(struct im_connection *ic)
18{
19        struct msn_gw *gw = g_new0(struct msn_gw, 1);
20        gw->last_host = g_strdup(GATEWAY_HOST);
21        gw->port = GATEWAY_PORT;
22        gw->ssl = (GATEWAY_PORT == 443);
23        gw->poll_timeout = -1;
24        gw->ic = ic;
25        gw->md = ic->proto_data;
26        gw->in = g_byte_array_new();
27        gw->out = g_byte_array_new();
28        return gw;
29}
30
31void msn_gw_free(struct msn_gw *gw)
32{
33        if (gw->poll_timeout != -1) {
34                b_event_remove(gw->poll_timeout);
35        }
36        g_byte_array_free(gw->in, TRUE);
37        g_byte_array_free(gw->out, TRUE);
38        g_free(gw->session_id);
39        g_free(gw->last_host);
40        g_free(gw);
41}
42
43static struct msn_gw *msn_gw_from_ic(struct im_connection *ic)
44{
45        if (g_slist_find(msn_connections, ic) == NULL) {
46                return NULL;
47        } else {
48                struct msn_data *md = ic->proto_data;
49                return md->gw;
50        }
51}
52
53static gboolean msn_gw_parse_session_header(struct msn_gw *gw, char *value)
54{
55        int i;
56        char **subvalues;
57        gboolean closed = FALSE;
58
59        subvalues = g_strsplit(value, "; ", 0);
60
61        for (i = 0; subvalues[i]; i++) {
62                if (strcmp(subvalues[i], "Session=close") == 0) {
63                        /* gateway closed, signal the death of the socket */
64                        closed = TRUE;
65                } else if (g_str_has_prefix(subvalues[i], "SessionID=")) {
66                        /* copy the part after the = to session_id*/
67                        g_free(gw->session_id);
68                        gw->session_id = g_strdup(subvalues[i] + 10);
69                }
70        }
71
72        g_strfreev(subvalues);
73
74        return !closed;
75}
76
77void msn_gw_callback(struct http_request *req)
78{
79        struct msn_gw *gw;
80        char *value;
81
82        if (!(gw = msn_gw_from_ic(req->data))) {
83                return;
84        }
85
86        gw->waiting = FALSE;
87        gw->polling = FALSE;
88
89        if (req->status_code != 200 || !req->reply_body) {
90                gw->callback(gw->md, -1, B_EV_IO_READ);
91                return;
92        }
93
94        if (getenv("BITLBEE_DEBUG")) {
95                fprintf(stderr, "\n\x1b[90mHTTP:%s\n", req->reply_body);
96                fprintf(stderr, "\n\x1b[97m\n");
97        }
98
99        if ((value = get_rfc822_header(req->reply_headers, "X-MSN-Messenger", 0))) {
100                if (!msn_gw_parse_session_header(gw, value)) {
101                        gw->callback(gw->md, -1, B_EV_IO_READ);
102                        g_free(value);
103                        return;
104                }
105                g_free(value);
106        }
107       
108        if ((value = get_rfc822_header(req->reply_headers, "X-MSN-Host", 0))) {
109                g_free(gw->last_host);
110                gw->last_host = value; /* transfer */
111        }
112
113        if (req->body_size) {
114                g_byte_array_append(gw->in, (const guint8 *) req->reply_body, req->body_size);
115
116                if (!gw->callback(gw->md, -1, B_EV_IO_READ)) {
117                        return;
118                }
119        }
120
121        if (gw->poll_timeout != -1) {
122                b_event_remove(gw->poll_timeout);
123        }
124        gw->poll_timeout = b_timeout_add(500, msn_gw_poll_timeout, gw->ic);
125
126}
127
128void msn_gw_dorequest(struct msn_gw *gw, char *args)
129{
130        char *request = NULL;
131        char *body = NULL;
132        size_t bodylen = 0;
133
134        if (gw->out) {
135                bodylen = gw->out->len;
136                g_byte_array_append(gw->out, (guint8 *) "", 1); /* nullnullnull */
137                body = (char *) g_byte_array_free(gw->out, FALSE);
138                gw->out = g_byte_array_new();
139        }
140
141        if (!bodylen && !args) {
142                args = "Action=poll&Lifespan=60";
143                gw->polling = TRUE;
144        }
145
146        request = g_strdup_printf(REQUEST_TEMPLATE,
147                gw->session_id ? : "", args ? : "", gw->last_host, bodylen, body ? : "");
148
149        http_dorequest(gw->last_host, gw->port, gw->ssl, request, msn_gw_callback, gw->ic);
150        gw->open = TRUE;
151        gw->waiting = TRUE;
152
153        g_free(body);
154        g_free(request);
155}
156
157void msn_gw_open(struct msn_gw *gw)
158{
159        msn_gw_dorequest(gw, "Action=open&Server=NS");
160}
161
162static gboolean msn_gw_poll_timeout(gpointer data, gint source, b_input_condition cond)
163{
164        struct msn_gw *gw;
165
166        if (!(gw = msn_gw_from_ic(data))) {
167                return FALSE;
168        }
169
170        gw->poll_timeout = -1;
171        if (!gw->waiting) {
172                msn_gw_dorequest(gw, NULL);
173        }
174        return FALSE;
175}
176
177ssize_t msn_gw_read(struct msn_gw *gw, char **buf)
178{
179        size_t bodylen;
180        if (!gw->open) {
181                return 0;
182        }
183
184        bodylen = gw->in->len;
185        g_byte_array_append(gw->in, (guint8 *) "", 1); /* nullnullnull */
186        *buf = (char *) g_byte_array_free(gw->in, FALSE);
187        gw->in = g_byte_array_new();
188        return bodylen;
189}
190
191void msn_gw_write(struct msn_gw *gw, char *buf, size_t len)
192{
193        g_byte_array_append(gw->out, (const guint8 *) buf, len);
194        if (!gw->open) {
195                msn_gw_open(gw);
196        } else if (gw->polling || !gw->waiting) {
197                msn_gw_dorequest(gw, NULL);
198        }
199}
Note: See TracBrowser for help on using the repository browser.