source: protocols/msn/msn_util.c @ ef9ee0e

/********************************************************************\
  * BitlBee -- An IRC to other IM-networks gateway                     *
  *                                                                    *
  * Copyright 2002-2012 Wilmer van der Gaast and others                *
  \********************************************************************/

/* MSN module - Miscellaneous utilities                                 */

/*
  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License with
  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
  if not, write to the Free Software Foundation, Inc., 51 Franklin St.,
  Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "nogaim.h"
#include "msn.h"
#include "md5.h"
#include "soap.h"
#include <ctype.h>

static char *adlrml_entry(const char *handle_, msn_buddy_flags_t list)
{
        char *domain, handle[strlen(handle_) + 1];

        strcpy(handle, handle_);
        if ((domain = strchr(handle, '@'))) {
                *(domain++) = '\0';
        } else {
                return NULL;
        }

        return g_markup_printf_escaped("<ml><d n=\"%s\"><c n=\"%s\" l=\"%d\" t=\"1\"/></d></ml>",
                                       domain, handle, list);
}

int msn_buddy_list_add(struct im_connection *ic, msn_buddy_flags_t list, const char *who, const char *realname,
                       const char *group)
{
        struct msn_data *md = ic->proto_data;
        char groupid[8];
        bee_user_t *bu;
        struct msn_buddy_data *bd;
        char *adl;

        *groupid = '\0';

        if (!((bu = bee_user_by_handle(ic->bee, ic, who)) ||
              (bu = bee_user_new(ic->bee, ic, who, 0))) ||
            !(bd = bu->data) || bd->flags & list) {
                return 1;
        }

        bd->flags |= list;

        if (list == MSN_BUDDY_FL) {
                msn_soap_ab_contact_add(ic, bu);
        } else {
                msn_soap_memlist_edit(ic, who, TRUE, list);
        }

        if ((adl = adlrml_entry(who, list))) {
                int st = msn_ns_write(ic, "ADL %d %zd\r\n%s",
                                      ++md->trId, strlen(adl), adl);
                g_free(adl);

                return st;
        }

        return 1;
}

int msn_buddy_list_remove(struct im_connection *ic, msn_buddy_flags_t list, const char *who, const char *group)
{
        struct msn_data *md = ic->proto_data;
        char groupid[8];
        bee_user_t *bu;
        struct msn_buddy_data *bd;
        char *adl;

        *groupid = '\0';

        if (!(bu = bee_user_by_handle(ic->bee, ic, who)) ||
            !(bd = bu->data) || !(bd->flags & list)) {
                return 1;
        }

        bd->flags &= ~list;

        if (list == MSN_BUDDY_FL) {
                msn_soap_ab_contact_del(ic, bu);
        } else {
                msn_soap_memlist_edit(ic, who, FALSE, list);
        }

        if ((adl = adlrml_entry(who, list))) {
                int st = msn_ns_write(ic, "RML %d %zd\r\n%s",
                                      ++md->trId, strlen(adl), adl);
                g_free(adl);

                return st;
        }

        return 1;
}

struct msn_buddy_ask_data {
        struct im_connection *ic;
        char *handle;
        char *realname;
};

static void msn_buddy_ask_free(void *data)
{
        struct msn_buddy_ask_data *bla = data;

        g_free(bla->handle);
        g_free(bla->realname);
        g_free(bla);
}

static void msn_buddy_ask_yes(void *data)
{
        struct msn_buddy_ask_data *bla = data;

        msn_buddy_list_add(bla->ic, MSN_BUDDY_AL, bla->handle, bla->realname, NULL);

        imcb_ask_add(bla->ic, bla->handle, NULL);

        msn_buddy_ask_free(bla);
}

static void msn_buddy_ask_no(void *data)
{
        struct msn_buddy_ask_data *bla = data;

        msn_buddy_list_add(bla->ic, MSN_BUDDY_BL, bla->handle, bla->realname, NULL);

        msn_buddy_ask_free(bla);
}

void msn_buddy_ask(bee_user_t *bu)
{
        struct msn_buddy_ask_data *bla;
        struct msn_buddy_data *bd = bu->data;
        char buf[1024];

        if (!(bd->flags & MSN_BUDDY_PL)) {
                return;
        }

        bla = g_new0(struct msn_buddy_ask_data, 1);
        bla->ic = bu->ic;
        bla->handle = g_strdup(bu->handle);
        bla->realname = g_strdup(bu->fullname);

        g_snprintf(buf, sizeof(buf),
                   "The user %s (%s) wants to add you to his/her buddy list.",
                   bu->handle, bu->fullname);

        imcb_ask_with_free(bu->ic, buf, bla, msn_buddy_ask_yes, msn_buddy_ask_no, msn_buddy_ask_free);
}

void msn_queue_feed(struct msn_data *h, char *bytes, int st)
{
        h->rxq = g_renew(char, h->rxq, h->rxlen + st);
        memcpy(h->rxq + h->rxlen, bytes, st);
        h->rxlen += st;

        if (getenv("BITLBEE_DEBUG")) {
                fprintf(stderr, "\n\x1b[92m<<< ");
                write(2, bytes , st);
                fprintf(stderr, "\x1b[97m");
        }
}


static int msn_handle_command(struct msn_data *h) {
        char *msg, **cmd;
        int count, st;

        msg = g_strndup(h->rxq, h->msglen);

        cmd = g_strsplit_set(h->cmd_text, " ", -1);
        count = g_strv_length(cmd);

        st = msn_ns_command(h, cmd, count, msg, h->msglen);

        g_strfreev(cmd);
        g_free(msg);

        g_free(h->cmd_text);
        h->cmd_text = NULL;

        return st;
}

/* This one handles input from a MSN Messenger server. Both the NS and SB servers usually give
   commands, but sometimes they give additional data (payload). This function tries to handle
   this all in a nice way and send all data to the right places. */

/* Return values: -1: Read error, abort connection.
                   0: Command reported error; Abort *immediately*. (The connection does not exist anymore)
                   1: OK */

int msn_handler(struct msn_data *h)
{
        int st = 1;

        while (st) {
                int i;

                if (h->msglen == 0) {
                        for (i = 0; i < h->rxlen; i++) {
                                if (h->rxq[i] == '\r' || h->rxq[i] == '\n') {
                                        char *cmd_text, *last_param;
                                        guint64 parsed_len;

                                        cmd_text = g_strndup(h->rxq, i);

                                        /* find the position of the last parameter from the end */
                                        last_param = cmd_text + i;
                                        while (last_param != cmd_text && last_param[-1] != ' ') {
                                                last_param--;
                                        }

                                        /* that parameter is the payload size */
                                        if (!parse_int64(last_param, 10, &parsed_len)) {
                                                return -1;
                                        }

                                        h->msglen = (int) parsed_len;
                                        h->cmd_text = cmd_text;

                                        /* if there's no payload, handle it right away */
                                        if (parsed_len == 0 && !msn_handle_command(h)) {
                                                return 0;
                                        }

                                        /* Skip the \r\n */
                                        i += 2;

                                        break;
                                }
                        }

                        /* If we reached the end of the buffer, there's still an incomplete command there.
                           Return and wait for more data. */
                        if (i && i == h->rxlen && h->rxq[i - 1] != '\r' && h->rxq[i - 1] != '\n') {
                                break;
                        }
                } else {

                        /* Do we have the complete message already? */
                        if (h->msglen > h->rxlen) {
                                break;
                        } else if (!msn_handle_command(h)) {
                                return 0;
                        }

                        i = h->msglen;
                        h->msglen = 0;
                }

                /* More data after this block? */
                if (i < h->rxlen) {
                        char *tmp;

                        tmp = g_memdup(h->rxq + i, h->rxlen - i);
                        g_free(h->rxq);
                        h->rxq = tmp;
                        h->rxlen -= i;
                        i = 0;
                } else {
                        /* If not, reset the rx queue and get lost. */
                        g_free(h->rxq);
                        h->rxq = g_new0(char, 1);
                        h->rxlen = 0;
                        return(1);
                }
        }

        return(1);
}

/* Copied and heavily modified from http://tmsnc.sourceforge.net/chl.c */
char *msn_p11_challenge(char *challenge)
{
        char *output, buf[256];
        md5_state_t md5c;
        unsigned char md5Hash[16], *newHash;
        unsigned int *md5Parts, *chlStringParts, newHashParts[5];
        long long nHigh = 0, nLow = 0;
        int i, n;

        /* Create the MD5 hash */
        md5_init(&md5c);
        md5_append(&md5c, (unsigned char *) challenge, strlen(challenge));
        md5_append(&md5c, (unsigned char *) MSNP11_PROD_KEY, strlen(MSNP11_PROD_KEY));
        md5_finish(&md5c, md5Hash);

        /* Split it into four integers */
        md5Parts = (unsigned int *) md5Hash;
        for (i = 0; i < 4; i++) {
                md5Parts[i] = GUINT32_TO_LE(md5Parts[i]);

                /* & each integer with 0x7FFFFFFF */
                /* and save one unmodified array for later */
                newHashParts[i] = md5Parts[i];
                md5Parts[i] &= 0x7FFFFFFF;
        }

        /* make a new string and pad with '0' */
        n = g_snprintf(buf, sizeof(buf) - 5, "%s%s00000000", challenge, MSNP11_PROD_ID);
        /* truncate at an 8-byte boundary */
        buf[n &= ~7] = '\0';

        /* split into integers */
        chlStringParts = (unsigned int *) buf;

        /* this is magic */
        for (i = 0; i < (n / 4) - 1; i += 2) {
                long long temp;

                chlStringParts[i]   = GUINT32_TO_LE(chlStringParts[i]);
                chlStringParts[i + 1] = GUINT32_TO_LE(chlStringParts[i + 1]);

                temp  =
                        (md5Parts[0] *
                         (((0x0E79A9C1 *
                            (long long) chlStringParts[i]) % 0x7FFFFFFF) + nHigh) + md5Parts[1]) % 0x7FFFFFFF;
                nHigh =
                        (md5Parts[2] *
                         (((long long) chlStringParts[i + 1] + temp) % 0x7FFFFFFF) + md5Parts[3]) % 0x7FFFFFFF;
                nLow  = nLow + nHigh + temp;
        }
        nHigh = (nHigh + md5Parts[1]) % 0x7FFFFFFF;
        nLow = (nLow + md5Parts[3]) % 0x7FFFFFFF;

        newHashParts[0] ^= nHigh;
        newHashParts[1] ^= nLow;
        newHashParts[2] ^= nHigh;
        newHashParts[3] ^= nLow;

        /* swap more bytes if big endian */
        for (i = 0; i < 4; i++) {
                newHashParts[i] = GUINT32_TO_LE(newHashParts[i]);
        }

        /* make a string of the parts */
        newHash = (unsigned char *) newHashParts;

        /* convert to hexadecimal */
        output = g_new(char, 33);
        for (i = 0; i < 16; i++) {
                sprintf(output + i * 2, "%02x", newHash[i]);
        }

        return output;
}

gint msn_domaintree_cmp(gconstpointer a_, gconstpointer b_)
{
        const char *a = a_, *b = b_;
        gint ret;

        if (!(a = strchr(a, '@')) || !(b = strchr(b, '@')) ||
            (ret = strcmp(a, b)) == 0) {
                ret = strcmp(a_, b_);
        }

        return ret;
}

struct msn_group *msn_group_by_name(struct im_connection *ic, const char *name)
{
        struct msn_data *md = ic->proto_data;
        GSList *l;

        for (l = md->groups; l; l = l->next) {
                struct msn_group *mg = l->data;

                if (g_strcasecmp(mg->name, name) == 0) {
                        return mg;
                }
        }

        return NULL;
}

struct msn_group *msn_group_by_id(struct im_connection *ic, const char *id)
{
        struct msn_data *md = ic->proto_data;
        GSList *l;

        for (l = md->groups; l; l = l->next) {
                struct msn_group *mg = l->data;

                if (g_strcasecmp(mg->id, id) == 0) {
                        return mg;
                }
        }

        return NULL;
}

int msn_ns_set_display_name(struct im_connection *ic, const char *value)
{
        // TODO, implement this through msn_set_away's method
        return 1;
}

const char *msn_normalize_handle(const char *handle)
{
        if (strncmp(handle, "1:", 2) == 0) {
                return handle + 2;
        } else {
                return handle;
        }
}