Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
       3             :  */
       4             : 
       5             : #include "xmpp/xmpp_proto.h"
       6             : #include <iostream>
       7             : #include <string>
       8             : #include <boost/algorithm/string/replace.hpp>
       9             : #include <boost/algorithm/string/predicate.hpp>
      10             : 
      11             : #include "xmpp/xmpp_connection.h"
      12             : #include "xmpp/xmpp_log.h"
      13             : #include "xmpp/xmpp_session.h"
      14             : #include "xmpp/xmpp_str.h"
      15             : 
      16             : #include "sandesh/sandesh_trace.h"
      17             : #include "sandesh/common/vns_types.h"
      18             : #include "sandesh/common/vns_constants.h"
      19             : #include "sandesh/xmpp_message_sandesh_types.h"
      20             : #include "sandesh/xmpp_trace_sandesh_types.h"
      21             : 
      22             : using namespace std;
      23             : 
      24             : unique_ptr<XmlBase> XmppProto::open_doc_(AllocXmppXmlImpl(sXMPP_STREAM_OPEN));
      25             : 
      26           0 : XmppStanza::XmppStanza() {
      27           0 : }
      28             : 
      29           0 : XmppProto::XmppProto() {
      30           0 : }
      31             : 
      32           0 : XmppProto::~XmppProto() {
      33           0 : }
      34             : 
      35       20300 : int XmppProto::EncodeStream(const XmppStreamMessage &str, string &to,
      36             :                             string &from, const string &xmlns, uint8_t *buf,
      37             :                             size_t size) {
      38       20300 :     int len = 0;
      39             : 
      40       20300 :     switch (str.strmtype) {
      41        8041 :         case (XmppStanza::XmppStreamMessage::INIT_STREAM_HEADER):
      42        8041 :             len = EncodeOpen(buf, to, from, xmlns, size);
      43        8041 :             break;
      44        7817 :         case (XmppStanza::XmppStreamMessage::INIT_STREAM_HEADER_RESP):
      45        7817 :             len = EncodeOpenResp(buf, to, from, size);
      46        7818 :             break;
      47        4443 :         case (XmppStanza::XmppStreamMessage::FEATURE_TLS):
      48        4443 :             switch (str.strmtlstype) {
      49        1510 :                 case (XmppStanza::XmppStreamMessage::TLS_FEATURE_REQUEST):
      50        1510 :                     len = EncodeFeatureTlsRequest(buf);
      51        1510 :                     break;
      52        1469 :                 case (XmppStanza::XmppStreamMessage::TLS_START):
      53        1469 :                     len = EncodeFeatureTlsStart(buf);
      54        1469 :                     break;
      55        1464 :                 case (XmppStanza::XmppStreamMessage::TLS_PROCEED):
      56        1464 :                     len = EncodeFeatureTlsProceed(buf);
      57        1464 :                     break;
      58             :             }
      59        4443 :             break;
      60           0 :         default:
      61           0 :             break;
      62             :     }
      63             : 
      64       20301 :     return len;
      65             : }
      66             : 
      67        6440 : int XmppProto::EncodeStream(const XmppStanza::XmppMessage &str, uint8_t *buf,
      68             :                             size_t size) {
      69        6440 :     int ret = 0;
      70             : 
      71        6440 :     if (str.type == XmppStanza::WHITESPACE_MESSAGE_STANZA) {
      72        6440 :         return EncodeWhitespace(buf);
      73             :     }
      74             : 
      75           0 :     return ret;
      76             : }
      77             : 
      78           0 : int XmppProto::EncodeMessage(XmlBase *dom, uint8_t *buf, size_t size) {
      79           0 :     int len = dom->WriteDoc(buf);
      80             : 
      81           0 :     return len;
      82             : }
      83             : 
      84           0 : int XmppProto::EncodePresence(uint8_t *buf, size_t size) {
      85           0 :     return 0;
      86             : }
      87             : 
      88           0 : int XmppProto::EncodeIq(const XmppStanza::XmppMessageIq *iq,
      89             :                         XmlBase *doc, uint8_t *buf, size_t size) {
      90           0 :     unique_ptr<XmlBase> send_doc_(AllocXmppXmlImpl());
      91             : 
      92             :     // create
      93           0 :     send_doc_->LoadDoc("");
      94           0 :     send_doc_->AddNode("iq", "");
      95             : 
      96           0 :     switch(iq->stype) {
      97           0 :         case XmppStanza::XmppMessageIq::GET:
      98           0 :             send_doc_->AddAttribute("type", "get");
      99           0 :             break;
     100           0 :         case XmppStanza::XmppMessageIq::SET:
     101           0 :             send_doc_->AddAttribute("type", "set");
     102           0 :             break;
     103           0 :         case XmppStanza::XmppMessageIq::RESULT:
     104           0 :             send_doc_->AddAttribute("type", "result");
     105           0 :             break;
     106           0 :         case XmppStanza::XmppMessageIq::ERROR:
     107           0 :             send_doc_->AddAttribute("type", "error");
     108           0 :             break;
     109           0 :         default:
     110           0 :             break;
     111             :     }
     112           0 :     send_doc_->AddAttribute("from", iq->from);
     113           0 :     send_doc_->AddAttribute("to", iq->to);
     114           0 :     send_doc_->AddAttribute("id", "id1");
     115             : 
     116           0 :     send_doc_->AddChildNode("pubsub", "");
     117           0 :     send_doc_->AddAttribute("xmlns", "http://jabber.org/protocol/pubsub");
     118             : 
     119           0 :     send_doc_->AppendDoc("pubsub", doc);
     120             : 
     121             :     //Returns byte encoded in the doc
     122           0 :     int len = send_doc_->WriteDoc(buf);
     123             : 
     124           0 :     return len;
     125           0 : }
     126             : 
     127        6440 : int XmppProto::EncodeWhitespace(uint8_t *buf) {
     128        6440 :     string str(sXMPP_WHITESPACE);
     129             : 
     130        6440 :     int len = str.size();
     131        6440 :     if (len > 0) {
     132        6440 :         memcpy(buf, str.data(), len);
     133             :     }
     134             : 
     135        6440 :     return len;
     136        6440 : }
     137             : 
     138        7817 : int XmppProto::EncodeOpenResp(uint8_t *buf, string &to, string &from,
     139             :                               size_t max_size) {
     140             : 
     141        7817 :     unique_ptr<XmlBase> resp_doc(XmppStanza::AllocXmppXmlImpl(sXMPP_STREAM_RESP));
     142             : 
     143        7818 :     if (resp_doc.get() == NULL) {
     144           0 :         return 0;
     145             :     }
     146             : 
     147        7818 :     SetTo(to, resp_doc.get());
     148        7816 :     SetFrom(from, resp_doc.get());
     149             : 
     150        7816 :     std::stringstream ss;
     151        7818 :     resp_doc->PrintDoc(ss);
     152        7818 :     std::string msg;
     153        7818 :     msg = ss.str();
     154        7817 :     size_t len = msg.size();
     155        7817 :     if (len > max_size) {
     156           0 :         LOG(ERROR, "\n (Open Confirm) size greater than max buffer size \n");
     157           0 :         return 0;
     158             :     } else {
     159        7817 :         boost::algorithm::ireplace_last(msg, "/", " ");
     160        7818 :         memcpy(buf, msg.c_str(), len);
     161        7818 :         return len;
     162             :     }
     163        7818 : }
     164             : 
     165        8041 : int XmppProto::EncodeOpen(uint8_t *buf, string &to, string &from,
     166             :                           const string &xmlns, size_t max_size) {
     167             : 
     168        8041 :     if (open_doc_.get() ==  NULL) {
     169           0 :         return 0;
     170             :     }
     171             : 
     172        8041 :     SetTo(to, open_doc_.get());
     173        8041 :     SetFrom(from, open_doc_.get());
     174        8041 :     SetXmlns(xmlns, open_doc_.get());
     175             : 
     176             :     //Returns byte encoded in the doc
     177        8041 :     std::stringstream ss;
     178        8041 :     open_doc_->PrintDoc(ss);
     179        8041 :     std::string msg;
     180        8041 :     msg = ss.str();
     181        8041 :     size_t len = msg.size();
     182        8041 :     if (len > max_size) {
     183           0 :         LOG(ERROR, "\n (Open Message) size greater than max buffer size \n");
     184           0 :         return 0;
     185             :     } else {
     186        8041 :         boost::algorithm::ireplace_last(msg, "/", " ");
     187        8041 :         memcpy(buf, msg.c_str(), len);
     188        8041 :         return len;
     189             :     }
     190        8041 : }
     191             : 
     192        1510 : int XmppProto::EncodeFeatureTlsRequest(uint8_t *buf) {
     193        1510 :     unique_ptr<XmlBase> resp_doc(XmppStanza::AllocXmppXmlImpl(sXMPP_STREAM_FEATURE_TLS));
     194             :     //Returns byte encoded in the doc
     195        1510 :     int len = resp_doc->WriteDoc(buf);
     196        1510 :     return len;
     197        1510 : }
     198             : 
     199        1469 : int XmppProto::EncodeFeatureTlsStart(uint8_t *buf) {
     200        1469 :     unique_ptr<XmlBase> resp_doc(XmppStanza::AllocXmppXmlImpl(sXMPP_STREAM_START_TLS));
     201             :     //Returns byte encoded in the doc
     202        1469 :     int len = resp_doc->WriteDoc(buf);
     203        1469 :     return len;
     204        1469 : }
     205             : 
     206        1464 : int XmppProto::EncodeFeatureTlsProceed(uint8_t *buf) {
     207        1464 :     unique_ptr<XmlBase> resp_doc(XmppStanza::AllocXmppXmlImpl(sXMPP_STREAM_PROCEED_TLS));
     208             :     //Returns byte encoded in the doc
     209        1464 :     int len = resp_doc->WriteDoc(buf);
     210        1464 :     return len;
     211        1464 : }
     212             : 
     213     4139865 : XmppStanza::XmppMessage *XmppProto::Decode(const XmppConnection *connection,
     214             :                                            const string &ts) {
     215     4139865 :     XmlBase *impl = XmppStanza::AllocXmppXmlImpl();
     216     4139817 :     if (impl == nullptr) {
     217           0 :         return nullptr;
     218             :     }
     219             : 
     220     4139817 :     XmppStanza::XmppMessage *msg = DecodeInternal(connection, ts, impl);
     221     4139886 :     if (!msg) {
     222           0 :         return nullptr;
     223             :     }
     224             : 
     225             :     // transfer ownership of the dom implementation
     226     4139886 :     msg->dom.reset(impl);
     227             : 
     228     4139844 :     return msg;
     229             : }
     230             : 
     231     4139818 : XmppStanza::XmppMessage *XmppProto::DecodeInternal(
     232             :         const XmppConnection *connection, const string &ts, XmlBase *impl) {
     233     4139818 :     XmppStanza::XmppMessage *ret = nullptr;
     234             : 
     235     4139818 :     string ns(sXMPP_STREAM_O);
     236     4139818 :     string ws(sXMPP_WHITESPACE);
     237     4139807 :     string iq(sXMPP_IQ_KEY);
     238             : 
     239     4139813 :     if (ts.find(sXMPP_IQ) != string::npos) {
     240       98552 :         string ts_tmp = ts;
     241             : 
     242       98545 :         if (impl->LoadDoc(ts) == -1) {
     243           0 :             XMPP_WARNING(XmppIqMessageParseFail, connection->ToUVEKey(),
     244             :                          XMPP_PEER_DIR_IN);
     245           0 :             assert(false);
     246             :             goto done;
     247             :         }
     248             : 
     249       98551 :         XmppStanza::XmppMessageIq *msg = new XmppStanza::XmppMessageIq;
     250       98549 :         impl->ReadNode(iq);
     251       98549 :         msg->to = XmppProto::GetTo(impl);
     252       98549 :         msg->from = XmppProto::GetFrom(impl);
     253       98550 :         msg->id = XmppProto::GetId(impl);
     254       98551 :         msg->iq_type = XmppProto::GetType(impl);
     255             :         // action is subscribe,publish,collection
     256       98548 :         const char *action = XmppProto::GetAction(impl, msg->iq_type);
     257       98549 :         if (action) {
     258       98534 :             msg->action = action;
     259             :         }
     260       98549 :         if (XmppProto::GetNode(impl, msg->action)) {
     261       98536 :             msg->node = XmppProto::GetNode(impl, msg->action);
     262             :         }
     263             :         //associate or dissociate collection node
     264       98548 :         if (msg->action.compare("collection") == 0) {
     265       41661 :             if (XmppProto::GetAsNode(impl)) {
     266       27295 :                 msg->as_node = XmppProto::GetAsNode(impl);
     267       27295 :                 msg->is_as_node = true;
     268       14366 :             } else if (XmppProto::GetDsNode(impl)) {
     269       14337 :                 msg->as_node = XmppProto::GetDsNode(impl);
     270       14338 :                 msg->is_as_node = false;
     271             :             }
     272             :         }
     273             : 
     274             :         //msg->dom.reset(impl);
     275             : 
     276       98549 :         ret = msg;
     277             : 
     278       98549 :         XMPP_UTDEBUG(XmppIqMessageProcess, connection->ToUVEKey(),
     279             :                      XMPP_PEER_DIR_IN, msg->node, msg->action, msg->from,
     280             :                      msg->to, msg->id, msg->iq_type);
     281       98550 :         goto done;
     282             : 
     283     4139887 :     } else if (ts.find(sXMPP_MESSAGE) != string::npos) {
     284             : 
     285     1549872 :         if (impl->LoadDoc(ts) == -1) {
     286           0 :             XMPP_WARNING(XmppChatMessageParseFail, connection->ToUVEKey(),
     287             :                          XMPP_PEER_DIR_IN);
     288           0 :             goto done;
     289             :         }
     290             :         XmppStanza::XmppMessage *msg = new XmppStanza::XmppChatMessage(
     291     1549872 :                                            STATE_NONE);
     292     1549872 :         impl->ReadNode(sXMPP_MESSAGE_KEY);
     293             : 
     294     1549872 :         msg->to = XmppProto::GetTo(impl);
     295     1549872 :         msg->from = XmppProto::GetFrom(impl);
     296     1549872 :         ret = msg;
     297             : 
     298     1549872 :         XMPP_UTDEBUG(XmppChatMessageProcess, connection->ToUVEKey(),
     299             :                      XMPP_PEER_DIR_IN, msg->type, msg->from, msg->to);
     300     1549872 :         goto done;
     301             : 
     302     2491465 :     } else if (ts.find(sXMPP_STREAM_O) != string::npos) {
     303             : 
     304             :         // ensusre stream open is at the beginning of the message
     305       15704 :         string ts_tmp = ts;
     306       15705 :         ts_tmp.erase(std::remove(ts_tmp.begin(), ts_tmp.end(), '\n'), ts_tmp.end());
     307             : 
     308       15703 :         if ((ts_tmp.compare(0, strlen(sXMPP_STREAM_START),
     309       31406 :              sXMPP_STREAM_START) != 0) &&
     310       15703 :             (ts_tmp.compare(0, strlen(sXMPP_STREAM_START_S),
     311             :              sXMPP_STREAM_START_S) != 0)) {
     312           0 :             XMPP_WARNING(XmppBadMessage, connection->ToUVEKey(),
     313             :                          XMPP_PEER_DIR_IN,
     314             :                          "Open message not at the beginning.", ts);
     315           0 :             goto done;
     316             :         }
     317             : 
     318             :         // check if the buf is xmpp open or response message
     319             :         // As end tag will be missing we need to modify the
     320             :         // string for stream open, else dom decoder will fail
     321       15703 :         boost::algorithm::replace_last(ts_tmp, ">", "/>");
     322       15703 :         if (impl->LoadDoc(ts_tmp) == -1) {
     323           0 :             XMPP_WARNING(XmppBadMessage, connection->ToUVEKey(),
     324             :                          XMPP_PEER_DIR_IN, "Open message parse failed.", ts);
     325           0 :             goto done;
     326             :         }
     327             : 
     328             :         XmppStanza::XmppStreamMessage *strm =
     329       15706 :             new XmppStanza::XmppStreamMessage();
     330       15704 :         strm->strmtype = XmppStanza::XmppStreamMessage::INIT_STREAM_HEADER;
     331       15704 :         impl->ReadNode(ns);
     332       15703 :         strm->to = XmppProto::GetTo(impl);
     333       15703 :         strm->from = XmppProto::GetFrom(impl);
     334       15704 :         strm->xmlns = XmppProto::GetXmlns(impl);
     335             : 
     336       15701 :         ret = strm;
     337             : 
     338       15701 :         XMPP_UTDEBUG(XmppRxOpenMessage, connection->ToUVEKey(),
     339             :                      XMPP_PEER_DIR_IN, strm->from, strm->to);
     340             : 
     341     2491465 :     } else if (ts.find(sXMPP_STREAM_NS_TLS) != string::npos) {
     342             : 
     343        4365 :         if (impl->LoadDoc(ts) == -1) {
     344           0 :             XMPP_WARNING(XmppBadMessage, connection->ToUVEKey(),
     345             :                          XMPP_PEER_DIR_IN, "Stream TLS parse failed.", ts);
     346           0 :             goto done;
     347             :         }
     348             : 
     349             :         // find stream:features tls required
     350        5820 :         if ((ts.find(sXMPP_STREAM_FEATURES_O) != string::npos) &&
     351        5820 :             (ts.find(sXMPP_STREAM_STARTTLS_O) != string::npos) &&
     352        1455 :             (ts.find(sXMPP_REQUIRED_O) != string::npos)) {
     353             : 
     354             :             XmppStanza::XmppStreamMessage *strm =
     355        1455 :                 new XmppStanza::XmppStreamMessage();
     356        1455 :             strm->strmtype = XmppStanza::XmppStreamMessage::FEATURE_TLS;
     357        1455 :             strm->strmtlstype = XmppStanza::XmppStreamMessage::TLS_FEATURE_REQUEST;
     358             : 
     359        1455 :             ret = strm;
     360             : 
     361        1455 :             XMPP_UTDEBUG(XmppRxStreamTlsRequired, connection->ToUVEKey(),
     362             :                          XMPP_PEER_DIR_IN);
     363             : 
     364        2910 :         } else if (ts.find(sXMPP_STREAM_STARTTLS_O) != string::npos) {
     365             :             XmppStanza::XmppStreamMessage *strm =
     366        1455 :                 new XmppStanza::XmppStreamMessage();
     367        1455 :             strm->strmtype = XmppStanza::XmppStreamMessage::FEATURE_TLS;
     368        1455 :             strm->strmtlstype = XmppStanza::XmppStreamMessage::TLS_START;
     369        1455 :             ret = strm;
     370             : 
     371        1455 :             XMPP_UTDEBUG(XmppRxStreamStartTls, connection->ToUVEKey(),
     372             :                          XMPP_PEER_DIR_IN);
     373             : 
     374        1455 :         } else if (ts.find(sXMPP_STREAM_PROCEED_O) != string::npos) {
     375             :             XmppStanza::XmppStreamMessage *strm =
     376        1455 :                 new XmppStanza::XmppStreamMessage();
     377        1455 :             strm->strmtype = XmppStanza::XmppStreamMessage::FEATURE_TLS;
     378        1455 :             strm->strmtlstype = XmppStanza::XmppStreamMessage::TLS_PROCEED;
     379             : 
     380        1455 :             ret = strm;
     381             : 
     382        1455 :             XMPP_UTDEBUG(XmppRxStreamProceed, connection->ToUVEKey(),
     383             :                          XMPP_PEER_DIR_IN);
     384             :         }
     385        4365 :         goto done;
     386             : 
     387     2471400 :     } else if (ts.find_first_of(sXMPP_VALIDWS) != string::npos) {
     388             : 
     389             :         XmppStanza::XmppMessage *msg =
     390     2471400 :             new XmppStanza::XmppMessage(WHITESPACE_MESSAGE_STANZA);
     391     2471401 :         return msg;
     392             :     } else {
     393           1 :         XMPP_WARNING(XmppBadMessage, connection->ToUVEKey(),
     394             :                      XMPP_PEER_DIR_IN, "Message not supported", ts);
     395             :     }
     396             : 
     397     1668488 : done:
     398             : 
     399     1668488 :     return ret;
     400     4139889 : }
     401             : 
     402       15859 : int XmppProto::SetTo(string &to, XmlBase *doc) {
     403       15859 :     if (!doc) return -1;
     404             : 
     405       15859 :     string ns(sXMPP_STREAM_O);
     406       15859 :     doc->ReadNode(ns);
     407       15859 :     doc->ModifyAttribute("to", to);
     408             : 
     409       15857 :     return 0;
     410       15857 : }
     411             : 
     412       15857 : int XmppProto::SetFrom(string &from, XmlBase *doc) {
     413       15857 :     if (!doc) return -1;
     414             : 
     415       15857 :     string ns(sXMPP_STREAM_O);
     416       15857 :     doc->ReadNode(ns);
     417       15857 :     return doc->ModifyAttribute("from", from);
     418       15857 : }
     419             : 
     420        8041 : int XmppProto::SetXmlns(const string &xmlns, XmlBase *doc) {
     421        8041 :     if (!doc)
     422           0 :         return -1;
     423             : 
     424        8041 :     string ns(sXMPP_STREAM_O);
     425        8041 :     doc->ReadNode(ns);
     426        8041 :     return doc->ModifyAttribute("xmlns", xmlns);
     427        8041 : }
     428             : 
     429     1664099 : const char *XmppProto::GetTo(XmlBase *doc) {
     430     1664099 :     if (!doc) return NULL;
     431             : 
     432     1664099 :     string tmp("to");
     433     1664094 :     return doc->ReadAttrib(tmp);
     434     1664098 : }
     435             : 
     436     1664093 : const char *XmppProto::GetFrom(XmlBase *doc) {
     437     1664093 :     if (!doc) return NULL;
     438             : 
     439     1664093 :     string tmp("from");
     440     1664104 :     return doc->ReadAttrib(tmp);
     441     1664101 : }
     442             : 
     443       15703 : const char *XmppProto::GetXmlns(XmlBase *doc) {
     444       15703 :     if (!doc)
     445           0 :         return NULL;
     446             : 
     447       15703 :     string tmp("xmlns");
     448       15703 :     return doc->ReadAttrib(tmp);
     449       15701 : }
     450             : 
     451       98550 : const char *XmppProto::GetId(XmlBase *doc) {
     452       98550 :     if (!doc) return NULL;
     453             : 
     454       98550 :     string tmp("id");
     455       98550 :     return doc->ReadAttrib(tmp);
     456       98551 : }
     457             : 
     458       98550 : const char *XmppProto::GetType(XmlBase *doc) {
     459       98550 :     if (!doc) return NULL;
     460             : 
     461       98550 :     string tmp("type");
     462       98549 :     return doc->ReadAttrib(tmp);
     463       98548 : }
     464             : 
     465       98548 : const char *XmppProto::GetAction(XmlBase *doc, const string &str) {
     466       98548 :     if (!doc) return NULL;
     467             : 
     468       98548 :     if (str.compare("set") == 0) {
     469       98538 :         doc->ReadNode("pubsub");
     470       98535 :         return(doc->ReadChildNodeName());
     471          15 :     } else if (str.compare("get") == 0) {
     472             :     }
     473             : 
     474          15 :     return(NULL);
     475             : }
     476             : 
     477      197084 : const char *XmppProto::GetNode(XmlBase *doc, const string &str) {
     478      197084 :     if (!doc) return NULL;
     479             : 
     480      197084 :     if (!str.empty()) {
     481      197069 :         return(doc->ReadAttrib("node"));
     482             :     }
     483             : 
     484          15 :     return(NULL);
     485             : }
     486             : 
     487       68956 : const char *XmppProto::GetAsNode(XmlBase *doc) {
     488       68956 :     if (!doc) return NULL;
     489             : 
     490       68956 :     const char *node = doc->ReadNode("associate");
     491       68954 :     if (node != NULL) {
     492       54588 :         return(doc->ReadAttrib("node"));
     493             :     }
     494             : 
     495       14366 :     return(NULL);
     496             : }
     497             : 
     498       28703 : const char *XmppProto::GetDsNode(XmlBase *doc) {
     499       28703 :     if (!doc) return NULL;
     500             : 
     501       28703 :     const char *node = doc->ReadNode("dissociate");
     502       28702 :     if (node != NULL) {
     503       28674 :         return(doc->ReadAttrib("node"));
     504             :     }
     505             : 
     506          28 :     return(NULL);
     507             : }