example/diameter/launcher/Procedure.cpp

   1 // ANNA - Anna is Not Nothingness Anymore                                                         //
   2 //                                                                                                //
   3 // (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo                         //
   4 //                                                                                                //
   5 // See project site at http://redmine.teslayout.com/projects/anna-suite                           //
   6 // See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
   7
   8 // Standard
   9 #include <iostream>
  10 #include <string>
  11 //#include <cstdlib>
  12
  13 // Project
  14 #include <anna/core/util/Tokenizer.hpp>
  15 #include <anna/core/Exception.hpp>
  16 #include <TestManager.hpp>
  17 #include <anna/diameter/codec/Message.hpp>
  18 #include <anna/diameter/codec/Avp.hpp>
  19 #include <anna/diameter/helpers/dcca/functions.hpp>
  20 #include <anna/diameter/helpers/dcca/functions.hpp>
  21
  22 // Local
  23 #include "Procedure.hpp"
  24 #include "Launcher.hpp"
  25
  26 namespace {
  27
  28    void usage (std::string &response) {
  29      response += "\n\nInvalid arguments. Provide these ones:";
  30      response += "\n";
  31      response += "\n<initial sequence>|<final sequence>|<digits>|<CCR-I xml file>[|CCR-T xml file]";
  32      response += "\n";
  33      response += "\nSequences are parsed when needed, over AVPs or internal values:";
  34      response += "\n";
  35      response += "\nSession-Id: <DiameterIdentity>;<high 32 bits>;<low 32 bits>[;<optional value>]";
  36      response += "\n            \\_fixed_/\\_digits_/                            \\_fixed_/\\_digits_/";
  37      response += "\n";
  38      response += "\nFor example, imagine a 13-digits diameter identity, and a 10-digits optional part.";
  39      response += "\nThis procedure will sequence the range for 8 digits in this way:";
  40      response += "\n";
  41      response += "\n111111<7-digit-sequence>;BAT004;esmdx0900.gxrel10plusrealm.com;901<7-digit-sequence>";
  42      response += "\n";
  43      response += "\nThe same will be done in MSISDN and IMSI (Subscription-Data AVPs).";
  44      response += "\nFramed-IP-Address will be sequenced with a direct correspondence to hex value.";
  45      response += "\n";
  46      response += "\nThen, you could provide these arguments: \"2000000|9000000|7|CCR-I.xml|CCR-T.xml\"";
  47      response += "\n";
  48    }
  49 }
  50
  51 void Procedure::execute(const std::string &args, std::string &response)  throw(anna::RuntimeException) {
  52
  53   response = "ERROR : ";
  54
  55   anna::Tokenizer targs;
  56   targs.apply(args, "|");
  57
  58   if (targs.size() < 4) {
  59     usage(response);
  60     return;
  61   }
  62
  63   anna::Tokenizer::const_iterator tok_it = targs.begin();
  64
  65   std::string seq_i = anna::Tokenizer::data(tok_it); tok_it++;
  66   std::string seq_f = anna::Tokenizer::data(tok_it); tok_it++;
  67   std::string digits = anna::Tokenizer::data(tok_it); tok_it++;
  68   std::string ccr_i = anna::Tokenizer::data(tok_it); tok_it++;
  69   std::string ccr_t = ((tok_it != targs.end()) ? anna::Tokenizer::data(tok_it):"");
  70   bool haveTermination = (ccr_t != "");
  71
  72   // Test cases cycles:
  73   unsigned int ll_seq_i = std::atol(seq_i.c_str());
  74   unsigned int ll_seq_f = std::atol(seq_f.c_str());
  75   unsigned int ll_seq_size = ll_seq_f - ll_seq_i + 1;
  76   int i_digits = std::atoi(digits.c_str());
  77
  78   if (ll_seq_i > ll_seq_f) {
  79     response += "<final sequence> must be greater or equal than <initial sequence>";
  80     return;
  81   }
  82
  83   if (seq_f.size() > i_digits) {
  84     response += "<final sequence> must be lesser than number of <digits>";
  85     return;
  86   }
  87
  88   unsigned int ll_seq, ll_index;
  89   anna::Millisecond timeout(5000);
  90
  91   // Load xml messages:
  92   anna::diameter::codec::Message ccri, ccrt;
  93   anna::diameter::codec::Avp *ccri_sessionId, *ccrt_sessionId, *ccri_framedIPAddress, *ccrt_framedIPAddress, *ccri_msisdn, *ccri_imsi;
  94   anna::diameter::codec::Avp *si1, *si2, *sidata1, *sidata2, *sitype1, *sitype2;
  95
  96   try {
  97
  98     ///////// CCR-Initial:
  99     ccri.loadXML(ccr_i);
 100
 101     // Session-Id & Framed-Ip-Address AVPs
 102     ccri_sessionId = ccri.getAvp("Session-Id");
 103     ccri_framedIPAddress = ccri.getAvp("Framed-IP-Address");
 104
 105     // Subscription-Id AVPs
 106     if (ccri.countAvp("Subscription-Id") != 2) {
 107       response += "Both Subscription-Id MSISDN & IMSI Avps must be present in the CCR-Initial provided !";
 108       return;
 109     }
 110
 111     si1 = ccri.getAvp("Subscription-Id", 1, anna::Exception::Mode::Ignore);
 112     si2 = ccri.getAvp("Subscription-Id", 2, anna::Exception::Mode::Ignore);
 113
 114     if (!si1 || !si2) {
 115       response += "Cannot found Subscription-Id MSISDN & IMSI Avps !" ;
 116       return;
 117     }
 118
 119     sidata1 = si1->getAvp("Subscription-Id-Data");
 120     sidata2 = si2->getAvp("Subscription-Id-Data");
 121     sitype1 = si1->getAvp("Subscription-Id-Type");
 122     //sitype2 = si2->getAvp("Subscription-Id-Type");
 123
 124     if (sitype1->getEnumerated()->getValue() == anna::diameter::helpers::dcca::AVPVALUES__Subscription_Id_Type::END_USER_E164) {
 125       ccri_msisdn = sidata1;
 126       ccri_imsi = sidata2;
 127     }
 128     else {
 129       ccri_msisdn = sidata2;
 130       ccri_imsi = sidata1;
 131     }
 132
 133     ///////// CCR-Termination:
 134     if (haveTermination) {
 135       ccrt.loadXML(ccr_t);
 136
 137       // Session-Id & Framed-Ip-Address AVPs
 138       ccrt_sessionId = ccrt.getAvp("Session-Id");
 139       ccrt_framedIPAddress = ccrt.getAvp("Framed-IP-Address");
 140     }
 141   }
 142   catch(anna::RuntimeException &ex) {
 143     ex.trace();
 144   }
 145
 146   // Prepare session-id string:
 147   std::string sessionId = ccri_sessionId->getUTF8String()->getValue();
 148   std::size_t last_semicolon = sessionId.rfind(";");
 149   anna::Tokenizer tsessionid;
 150   tsessionid.apply(sessionId, ";");
 151
 152   if (tsessionid.size() < 4) {
 153     response += "Session-Id must be in form '<a>;<b>;<c>;<d>'.\n\n";
 154     usage(response);
 155     return;
 156   }
 157
 158   tok_it = tsessionid.begin();
 159   std::string d_identity = anna::Tokenizer::data(tok_it); tok_it++; tok_it++; tok_it++;
 160   std::string o_part = anna::Tokenizer::data(tok_it);
 161   int d_identity_len = d_identity.size();
 162   int left_di = d_identity_len - i_digits;
 163   if (left_di < 0) {
 164     response += "Session-Id diameter identity length is lower than selected number of digits ";
 165     response += anna::functions::asString("(%d < %d).\n\n", d_identity_len, i_digits);
 166     usage(response);
 167     return;
 168   }
 169   int o_part_len = o_part.size();
 170   int left_op = o_part_len - i_digits;
 171   if (left_op < 0) {
 172     response += "Session-Id optional part length is lower than selected number of digits ";
 173     response += anna::functions::asString("(%d < %d).\n\n", o_part_len, i_digits);
 174     usage(response);
 175     return;
 176   }
 177
 178   // Idem for MSISDN & IMSI Subscription-Data:
 179   std::string msisdn = ccri_msisdn->getUTF8String()->getValue();
 180   int msisdn_len = msisdn.size();
 181   int left_msisdn = msisdn_len - i_digits;
 182   if (left_msisdn < 0) {
 183     response += "MSISDN Subscription-Data length is lower than selected number of digits ";
 184     response += anna::functions::asString("(%d < %d).\n\n", left_msisdn, i_digits);
 185     usage(response);
 186     return;
 187   }
 188   std::string imsi = ccri_imsi->getUTF8String()->getValue();
 189   int imsi_len = imsi.size();
 190   int left_imsi = imsi_len - i_digits;
 191   if (left_imsi < 0) {
 192     response += "IMSI Subscription-Data length is lower than selected number of digits ";
 193     response += anna::functions::asString("(%d < %d).\n\n", left_imsi, i_digits);
 194     usage(response);
 195     return;
 196   }
 197
 198
 199   // Format strings:
 200   char ndigit_format[32];
 201   sprintf(ndigit_format, "%s%d%s", "%0", i_digits, "u");
 202
 203   //response += "\nSession-Id original: "; response += sessionId;
 204   //response += "\nMSISDN original:     "; response += msisdn;
 205   //response += "\nIMSI original:       "; response += imsi;
 206
 207   // TestManager:
 208   TestManager &testManager = TestManager::instantiate();
 209   char cad_aux[16];
 210   char cad_framed[16];
 211   TestCase *tc;
 212   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
 213   OriginHost *oh = my_app.getWorkingNode();
 214
 215   for (ll_index = 0; ll_index < ll_seq_size; ll_index++) {
 216
 217     // Calculate next values ////////////////////////////////////////////////////////////
 218     ll_seq = ll_seq_i + ll_index;
 219     sprintf(cad_aux, ndigit_format, ll_seq);
 220     sprintf(cad_framed, "%08x", ll_seq);
 221
 222     // Activity indicator:
 223     if (ll_seq % 10000 == 0) std::cout << ".";
 224
 225     sessionId.replace(left_di, i_digits, cad_aux);
 226     sessionId.replace(last_semicolon + left_op + 1, i_digits, cad_aux);
 227
 228     msisdn.replace(left_msisdn, i_digits, cad_aux);
 229
 230     imsi.replace(left_imsi, i_digits, cad_aux);
 231
 232     //response += "\nSession-Id modified: "; response += sessionId;
 233     //response += "\nMSISDN modified:     "; response += msisdn;
 234     //response += "\nIMSI modified:       "; response += imsi;
 235     //response += "\nFramedIP modified:   "; response += cad_framed;
 236
 237     // Update diameter messages /////////////////////////////////////////////////////////
 238     ccri_sessionId->getUTF8String()->setValue(sessionId);
 239     ccri_framedIPAddress->getOctetString()->fromHexString(cad_framed);
 240     ccri_msisdn->getUTF8String()->setValue(msisdn);
 241     ccri_imsi->getUTF8String()->setValue(imsi);
 242
 243     if (haveTermination) {
 244       ccrt_sessionId->getUTF8String()->setValue(sessionId);
 245       ccrt_framedIPAddress->getOctetString()->fromHexString(cad_framed);
 246     }
 247
 248     // Create testcase:
 249     tc = testManager.getTestCase(ll_index+1);
 250     //    test|__TESTID__|timeout|6000
 251     //    test|__TESTID__|sendxml2e|ccr-i.xml
 252     //    test|__TESTID__|waitfe|272|0|||11111__SEQ8__;BAT004;esmdx0900.gxrel10plusrealm.com;90__SEQ8__|2001
 253     //    test|__TESTID__|sendxml2e|ccr-t.xml
 254     //    test|__TESTID__|waitfe|272|0|||11111__SEQ8__;BAT004;esmdx0900.gxrel10plusrealm.com;90__SEQ8__|2001
 255
 256     // Step 1: timeout 5000 ms:
 257     tc->addTimeout(timeout);
 258
 259     // Step 2: sendxml2e: CCR-Initial
 260     tc->addSendxml2e(ccri.code(), oh, -1 /* 'wait for request' step number for answers */);
 261
 262     // Step 3: waitfe: CCA with same session id
 263     // PARAM: 1     2            3      4          5           6             7           8          9       10         11
 264     //             wait<fe/fc>|[code]|[bitR]|[hopByHop]|[applicationId]|[sessionId]|[resultCode]|[msisdn]|[imsi]|[serviceContextId]
 265     tc->addWait(true /* from entity */, "272", "0", "", "", sessionId, "2001", "", "", "");
 266
 267     if (haveTermination) {
 268       // Step 4: sendxml2e: CCR-Termination
 269       tc->addSendxml2e(ccrt.code(), oh, -1 /* 'wait for request' step number for answers */);
 270
 271       // Step 5: waitfe: CCA with same session id
 272       tc->addWait(true /* from entity */, "272", "0", "", "", sessionId, "2001", "", "", "");
 273     }
 274   } // loop
 275
 276   response = "Completed provision for pid "; response += anna::functions::asString(a_app->getPid()); response += "; range [";
 277   response += seq_i; response += ", "; response += seq_f; response += "]; scenary: ";
 278   response += "CCR-Initial"; if (haveTermination) response += " + CCR-Termination";
 279 }
 280