onvif_rtsp/udp_handler.cpp at master · littleBigMen/onvif_rtsp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "udp_handler.h"

udp_handler::udp_handler(QObject *parent) :
    QObject(parent)
{
    listener = new udp_listen(0,this);
    sender = new udp_send();

    connect(listener, SIGNAL(dataRecieved(QByteArray,char*,int)), this, SLOT(parse_msg(QByteArray,char*,int)));
    connect(this,SIGNAL(send_udp(QString,char*,int)), sender, SLOT(send_udp(QString,char*,int)));
}

void udp_handler::start()
{
    listener->start(QThread::LowPriority);
}

void udp_handler::parse_msg(QByteArray msg, char* src_addr, int src_port)
{
    QString msg_uid;
    bool isNVT = false;

    //convert byte array to dom doc
    QDomDocument doc;
    doc.setContent(msg);

    //get the root element
    QDomNode root = doc.firstChildElement();

    QDomNodeList children = root.childNodes();
    for(int i = 0; i < children.count(); i++)
    {
        QDomElement ele = children.at(i).toElement();

        if(ele.nodeName().contains("Header", Qt::CaseInsensitive)){
            msg_uid = parseHeader(ele);
        }
        else if(ele.nodeName().contains("Body", Qt::CaseInsensitive)){
            isNVT = parseBody(ele);
        }
        else{
            qDebug() << "Unknown";
        }
        if(isNVT && msg_uid.length()!=0){
            emit send_udp(msg_uid, src_addr, src_port);
            emit probe_received();
        }
    }
}


QString udp_handler::parseHeader(QDomElement hdr)
{
    QString ID;
    QDomNodeList list = hdr.childNodes();
    for(int i = 0; i < list.count(); i++)
    {
        QDomElement ele = list.at(i).toElement();
        if(ele.nodeName().contains("MessageID", Qt::CaseInsensitive)){
            QString HdrSrcID;
            QTextStream(&HdrSrcID) << ele.text();
            ID = HdrSrcID;
        }
    }
    return ID;
}

bool udp_handler::parseBody(QDomElement bdy)
{
    QDomNodeList list = bdy.childNodes();
    for(int i = 0; i < list.count(); i++)
    {
        QDomElement ele = list.at(i).toElement();
        if(ele.nodeName().contains("Probe", Qt::CaseInsensitive)){
            QString BdyType;
            QTextStream(&BdyType) << ele.text();
            if(BdyType.contains("NetworkVideoTransmitter", Qt::CaseInsensitive)){
                return true;
            }
        }
    }
    return false;
}