ostinato/server/packetsequence.h
Srivats P 35c56b9181 sign: Calculate stream stats only if configured
Stream Stats Rx and Tx both are calculated only if 'Track Stream Stats'
is configured on the port
2016-12-06 21:41:47 +05:30

89 lines
2.6 KiB
C++

/*
Copyright (C) 2010-2016 Srivats P.
This file is part of "Ostinato"
This is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
#ifndef _PACKET_SEQUENCE_H
#define _PACKET_SEQUENCE_H
#include "pcapextra.h"
#include "../common/sign.h"
#include "streamstats.h"
class PacketSequence
{
public:
PacketSequence(bool trackGuidStats) {
trackGuidStats_ = trackGuidStats;
sendQueue_ = pcap_sendqueue_alloc(1*1024*1024);
lastPacket_ = NULL;
packets_ = 0;
bytes_ = 0;
usecDuration_ = 0;
repeatCount_ = 1;
repeatSize_ = 1;
usecDelay_ = 0;
}
~PacketSequence() {
pcap_sendqueue_destroy(sendQueue_);
}
bool hasFreeSpace(int size) {
if ((sendQueue_->len + size) <= sendQueue_->maxlen)
return true;
else
return false;
}
int appendPacket(const struct pcap_pkthdr *pktHeader,
const uchar *pktData) {
int ret;
if (lastPacket_)
{
usecDuration_ += (pktHeader->ts.tv_sec
- lastPacket_->ts.tv_sec) * long(1e6);
usecDuration_ += (pktHeader->ts.tv_usec
- lastPacket_->ts.tv_usec);
}
packets_++;
bytes_ += pktHeader->caplen;
lastPacket_ = (struct pcap_pkthdr *)
(sendQueue_->buffer + sendQueue_->len);
ret = pcap_sendqueue_queue(sendQueue_, pktHeader, pktData);
if (trackGuidStats_ && (ret >= 0)) {
uint guid;
if (SignProtocol::packetGuid(pktData, pktHeader->caplen, &guid)) {
streamStatsMeta_[guid].tx_pkts++;
streamStatsMeta_[guid].tx_bytes += pktHeader->caplen;
}
}
return ret;
}
pcap_send_queue *sendQueue_;
struct pcap_pkthdr *lastPacket_;
long packets_;
long bytes_;
ulong usecDuration_;
int repeatCount_;
int repeatSize_;
long usecDelay_;
StreamStats streamStatsMeta_;
private:
bool trackGuidStats_;
};
#endif