You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
sems/core/AmSipMsg.cpp

302 lines
6.6 KiB

#include <string.h>
#include <stdlib.h>
#include "AmUtils.h"
#include "AmSipMsg.h"
#include "AmSipHeaders.h"
#include "sip/sip_trans.h"
#include "sip/sip_parser.h"
#include "sip/msg_logger.h"
AmSipRequest::AmSipRequest()
: _AmSipMsgInDlg(),
max_forwards(-1)
{
}
string getHeader(const string& hdrs,const string& hdr_name, bool single)
{
if (hdr_name.empty())
return "";
size_t pos1;
size_t pos2;
size_t pos_s;
size_t skip = 0;
string ret = "";
while(findHeader(hdrs, hdr_name, skip, pos1, pos2, pos_s))
{
if(skip)
ret.append(", ");
else
if(single) return hdrs.substr(pos1,pos2-pos1);
ret.append(hdrs.substr(pos1,pos2-pos1));
skip = pos2+1;
}
return ret;
}
string getHeader(const string& hdrs,const string& hdr_name,
const string& compact_hdr_name, bool single)
{
string res = getHeader(hdrs, hdr_name, single);
if (!res.length())
return getHeader(hdrs, compact_hdr_name, single);
return res;
}
bool hasHeader(const string& hdrs,const string& hdr_name) {
size_t skip = 0, pos1 = 0, pos2 = 0, hdr_start = 0;
return findHeader(hdrs, hdr_name, skip, pos1, pos2, hdr_start);
}
#include "log.h"
bool findHeader(const string& hdrs,const string& hdr_name, const size_t skip,
size_t& pos1, size_t& pos2, size_t& hdr_start)
{
unsigned int p;
if(skip >= hdrs.length()) return false;
char* hdr = strdup(hdr_name.c_str());
const char* hdrs_c = hdrs.c_str() + skip;
char* hdr_c = hdr;
const char* hdrs_end = hdrs.c_str() + hdrs.length();
const char* hdr_end = hdr_c + hdr_name.length();
while(hdr_c != hdr_end){
if('A' <= *hdr_c && *hdr_c <= 'Z')
*hdr_c -= 'A' - 'a';
hdr_c++;
}
while(hdrs_c != hdrs_end){
hdr_c = hdr;
while((hdrs_c != hdrs_end) && (hdr_c != hdr_end)){
char c = *hdrs_c;
if('A' <= *hdrs_c && *hdrs_c <= 'Z')
c -= 'A' - 'a';
if(c != *hdr_c)
break;
hdr_c++;
hdrs_c++;
}
if(hdr_c == hdr_end) {
// matched whole of needle.
// ...all of current header?
const char* srccol = hdrs_c;
while (*srccol==' ' || *srccol=='\t')
srccol++;
if (*srccol == ':')
break; // was end of current hdr
// current hdr just starts with hdr, continue search
}
while((hdrs_c != hdrs_end) && (*hdrs_c != '\n'))
hdrs_c++;
if(hdrs_c != hdrs_end)
hdrs_c++;
}
if(hdr_c == hdr_end){
hdr_start = hdrs_c - hdrs.c_str();;
while((hdrs_c != hdrs_end) && (*hdrs_c == ' '))
hdrs_c++;
if((hdrs_c != hdrs_end) && (*hdrs_c == ':')){
hdrs_c++;
while((hdrs_c != hdrs_end) && (*hdrs_c == ' '))
hdrs_c++;
p = hdrs_c - hdrs.c_str();
string::size_type p_end = p;
while (p_end < hdrs.size() &&
hdrs[p_end] != '\r' &&
hdrs[p_end] != '\n')
p_end++;
free(hdr);
// return hdrs.substr(p,p_end-p);
pos1 = p;
pos2 = p_end;
return true;
}
}
free(hdr);
// return "";
return false;
}
bool removeHeader(string& hdrs, const string& hdr_name) {
size_t pos1, pos2, hdr_start;
bool found = false;
while (findHeader(hdrs, hdr_name, 0, pos1, pos2, hdr_start)) {
while (pos2 < hdrs.length() &&
(hdrs[pos2]=='\r' || hdrs[pos2]=='\n'))
pos2++;
hdr_start -= hdr_name.length();
hdrs.erase(hdr_start, pos2 - hdr_start);
found = true;
}
return found;
}
void addOptionTag(string& hdrs, const string& hdr_name, const string& tag) {
// see if option tag already exists
string options = getHeader(hdrs, hdr_name);
if (options.size()) {
std::vector<string> option_entries = explode(options, ",");
for (std::vector<string>::iterator it=option_entries.begin();
it != option_entries.end(); it++) {
if (trim(*it," ") == tag)
// found - no need to add again
return;
}
// tag not found - add our tag to the (first) hdr_name header
size_t pos1; size_t pos2; size_t hdr_start;
if (!findHeader(hdrs, hdr_name, 0, pos1, pos2, hdr_start)) {
ERROR("internal error: header '%s' disappeared in-between (hdrs = '%s'!\n",
hdr_name.c_str(), hdrs.c_str());
hdrs += hdr_name + COLSP + tag + CRLF;
return;
}
hdrs.insert(pos1, tag+", ");
} else {
// hdr does not exist - add it
hdrs += hdr_name + COLSP + tag + CRLF;
}
}
void removeOptionTag(string& hdrs, const string& hdr_name, const string& tag) {
string options = getHeader(hdrs, hdr_name);
// does hdr hdr_name exist?
if (options.empty())
return;
// todo: optimize by doing inplace
std::vector<string> options_v = explode(options, ",");
string o_hdr;
bool found = false;
for (std::vector<string>::iterator it=options_v.begin();
it != options_v.end(); it++) {
if (trim(*it, " ")==tag) {
found = true;
continue;
}
if (it != options_v.begin())
o_hdr = ", ";
o_hdr+=*it;
}
if (!found)
return;
removeHeader(hdrs, hdr_name);
if (o_hdr.empty())
return;
hdrs += hdr_name + COLSP + o_hdr + CRLF;
}
/* Print Member */
#define _PM(member, name) \
do { \
if (! member.empty()) \
buf += string(name) + ":" + member + ";"; \
} while (0)
/* Print Member in Brackets */
#define _PMB(member, name) \
do { \
if (! member.empty()) \
buf += string(name) + ":" + "[" + member + "]" + ";"; \
} while (0)
string AmSipRequest::print() const
{
string buf;
_PM(r_uri, "r-uri");
_PM(callid, "i");
_PM(int2str(cseq), "cseq");
_PM(from_tag, "l-tag");
_PM(to_tag, "r-tag");
_PMB(route, "rtset");
_PM(contact, "m");
_PMB(hdrs, "hdr");
//TODO: find some good debug info to print here
//_PM(content_type, "c");
//_PMB(body, "body");
_PM(user, "user");
_PM(domain, "domain");
_PM(from_uri, "f-uri");
_PM(from, "from");
_PM(to, "to");
buf = method + " [" + buf + "]";
return buf;
}
void AmSipRequest::log(msg_logger *logger) const
{
tt.lock_bucket();
const sip_trans* t = tt.get_trans();
if (t) {
sip_msg* msg = t->msg;
logger->log(msg->buf,msg->len,&msg->remote_ip,
&msg->local_ip,msg->u.request->method_str);
}
tt.unlock_bucket();
}
string AmSipReply::print() const
{
string buf;
_PM(int2str(code), "code");
_PMB(reason, "phrase");
_PM(callid, "i");
_PM(int2str(cseq), "cseq");
//_PM(method, "cseq meth");
_PM(from_tag, "from-tag");
_PM(to_tag, "to-tag");
//_PM(next_hop, "nhop");
_PMB(route, "rtset");
_PM(contact, "m");
_PMB(hdrs, "hdr");
//TODO: find some good debug info to print here
//_PM(content_type, "c");
//_PMB(body, "body");
_PM(contact, "contact");
buf = /*method +*/ " [" + buf + "]";
return buf;
}
#undef _PM
#undef _PMB
/** EMACS **
* Local variables:
* mode: c++
* c-basic-offset: 2
* End:
*/