apps/zjab/xmpp.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2009  Tomas 'ZeXx86' Jedrzejek (zexx86@zexos.org)
   4  *
   5  *  This program is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include "net.h"
  24 #include "xml.h"
  25 #include "xmpp.h"
  26 #include "cmds.h"
  27 #include "utils.h"
  28
  29 static xmpp_t xmpp;
  30
  31 int xmpp_session ()
  32 {
  33         char buf[256];
  34
  35         int r = sprintf (buf, "<iq xmlns='jabber:client' type='set' id='%s_1'><session xmlns='urn:ietf:params:xml:ns:xmpp-session'/></iq>", xmpp.session);
  36
  37         net_send (buf, r);
  38
  39         return 0;
  40 }
  41
  42 int xmpp_bind ()
  43 {
  44         char buf[256];
  45
  46         int r = sprintf (buf, "<iq type='set' id='%s'><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'><resource>%s</resource></bind></iq>", xmpp.session, xmpp.resource);
  47
  48         net_send (buf, r);
  49
  50         return 0;
  51 }
  52
  53 int xmpp_presence (char *show, char *status)
  54 {
  55         char buf[256];
  56
  57         int r = sprintf (buf, "<presence type='available'><show>%s</show><status>%s</status></presence>", show, status);
  58
  59         net_send (buf, r);
  60
  61         return 0;
  62 }
  63
  64 int xmpp_message (char *to, char *message, unsigned len)
  65 {
  66         if (len > 480)
  67                 return -1;
  68
  69         char buf[512];
  70
  71         int r = sprintf (buf, "<message from='%s@%s' to='%s' type='chat'><body>%s</body></message>", xmpp.user, xmpp.server, to, message);
  72
  73         net_send (buf, r);
  74
  75         return 0;
  76 }
  77
  78 void xmpp_setup (char *user, char *password, char *resource)
  79 {
  80         xmpp.user = user;
  81         xmpp.password = password;
  82         xmpp.resource = resource;
  83 }
  84
  85 int xmpp_connect (char *server, int port)
  86 {
  87         if (net_connect (server, port) == -1)
  88                 return -1;
  89
  90         xmpp.session = 0;
  91         xmpp.server = server;
  92         xmpp.port = port;
  93         xmpp.mechanism = 0;
  94
  95         char buf[512];
  96
  97         int r = sprintf (buf, "<?xml version='1.0'?><stream:stream to='%s' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' version='1.0'>", server);
  98
  99         net_send (buf, r);
 100
 101         int len = net_recv (buf, 511);
 102
 103         if (len > 0) {
 104                 buf[len] = '\0';
 105                 xml_handler (buf, len);
 106         }
 107
 108         if (xmpp.mechanism)
 109                 goto success;
 110
 111         len = net_recv (buf, 511);
 112
 113         if (len > 0) {
 114                 buf[len] = '\0';
 115                 xml_handler (buf, len);
 116         }
 117 success:
 118         r = sprintf (buf, "<auth xmlns='urn:ietf:params:xml:ns:xmpp-sasl' mechanism='%s'/>", xmpp.mechanism);
 119
 120         net_send (buf, r);
 121
 122         len = net_recv (buf, 511);
 123
 124         if (len > 0) {
 125                 buf[len] = '\0';
 126                 xml_handler (buf, len);
 127         }
 128
 129         unsigned user_len = strlen (xmpp.user);
 130         unsigned pwd_len = strlen (xmpp.password);
 131
 132         buf[0] = '\0';
 133         strcpy (buf+1, xmpp.user);
 134         buf[1+user_len] = '\0';
 135         strcpy (buf+2+user_len, xmpp.password);
 136
 137         r = sprintf (buf, "<response xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>%s</response>", base64 (buf, 2+user_len+pwd_len));
 138
 139         net_send (buf, r);
 140
 141         len = net_recv (buf, 511);
 142
 143         if (len > 0) {
 144                 buf[len] = '\0';
 145                 xml_handler (buf, len);
 146         }
 147
 148         r = sprintf (buf, "<?xml version='1.0'?><stream:stream to='%s' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' version='1.0'>", server);
 149
 150         net_send (buf, r);
 151
 152         len = net_recv (buf, 511);
 153
 154         if (len > 0) {
 155                 buf[len] = '\0';
 156                 xml_handler (buf, len);
 157         }
 158
 159         len = net_recv (buf, 511);
 160
 161         if (len > 0) {
 162                 buf[len] = '\0';
 163                 xml_handler (buf, len);
 164         }
 165
 166         xmpp_bind ();
 167
 168         len = net_recv (buf, 511);
 169
 170         if (len > 0) {
 171                 buf[len] = '\0';
 172                 xml_handler (buf, len);
 173         }
 174
 175         xmpp_session ();
 176
 177         len = net_recv (buf, 511);
 178
 179         if (len > 0) {
 180                 buf[len] = '\0';
 181                 xml_handler (buf, len);
 182         }
 183
 184         xmpp_presence ("online", "ZeXOS");
 185
 186         return 0;
 187 }
 188
 189 int xmpp_loop ()
 190 {
 191         if (net_nonblock () == -1)
 192                 return -1;
 193
 194         for (;;) {
 195                 char buf[1024];
 196
 197                 int len = net_recv (buf, 1023);
 198
 199                 if (len > 0) {
 200                         buf[len] = '\0';
 201                         xml_handler (buf, len);
 202                 }
 203
 204                 if (cmds_get () == -1)
 205                         break;
 206 #ifndef LINUX
 207                 schedule ();
 208 #endif
 209         }
 210
 211         return 0;
 212 }
 213
 214 int xmpp_close ()
 215 {
 216         net_send ("</stream>", 9);
 217
 218         printf ("> Disconnected from server\n");
 219
 220         if (xmpp.mechanism)
 221                 free (xmpp.mechanism);
 222
 223         return net_close ();
 224 }
 225
 226 int xmpp_session_set (char *id, unsigned len)
 227 {
 228         if (!id || !len)
 229                 return -1;
 230
 231         if (xmpp.session)
 232                 free (xmpp.session);
 233
 234         xmpp.session = strndup (id, len);
 235
 236         if (!xmpp.session)
 237                 return -1;
 238
 239 //      printf ("XMPP -> session id: '%s'\n", xmpp.session);
 240
 241         return 0;
 242 }
 243
 244 int xmpp_mechanism_set (char *mechanism, unsigned len)
 245 {
 246         if (!mechanism || !len)
 247                 return -1;
 248
 249         if (!strncmp (mechanism, "PLAIN", 5))
 250                 xmpp.mechanism = strndup (mechanism, 5);
 251         if (!strncmp (mechanism, "DIGEST-MD5", 9))
 252                 return 0; /*TODO */
 253
 254         if (!xmpp.mechanism)
 255                 return -1;
 256
 257 //      printf ("XMPP -> mechanism: '%s'\n", xmpp.mechanism);
 258
 259         return 0;
 260 }
 261
 262 int xmpp_message_from (char *from, unsigned from_len, char *msg, unsigned msg_len)
 263 {
 264         setcolor (15, 0);
 265
 266         unsigned i;
 267         for (i = 0; i < from_len; i ++)
 268                 putchar (from[i]);
 269
 270         printf (": ");
 271
 272         setcolor (14, 0);
 273
 274         for (i = 0; i < msg_len; i ++)
 275                 putchar (msg[i]);
 276
 277         putchar ('\n');
 278
 279         setcolor (7, 0);
 280
 281         return 0;
 282 }