113 lines
3.2 KiB
C
113 lines
3.2 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2019-2021, OARC, Inc.
|
||
|
|
* Copyright (c) 2019, DENIC eG
|
||
|
|
* All rights reserved.
|
||
|
|
*
|
||
|
|
* This file is part of dnsmeter.
|
||
|
|
*
|
||
|
|
* dnsmeter 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.
|
||
|
|
*
|
||
|
|
* dnsmeter 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 dnsmeter. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "dns_receiver_thread.h"
|
||
|
|
#include "payload_file.h"
|
||
|
|
#include "system_stat.h"
|
||
|
|
|
||
|
|
#include <ppl7.h>
|
||
|
|
|
||
|
|
#ifndef __dnsmeter_dns_sender_h
|
||
|
|
#define __dnsmeter_dns_sender_h
|
||
|
|
|
||
|
|
class DNSSender {
|
||
|
|
private:
|
||
|
|
#if defined(__GXX_EXPERIMENTAL_CXX0X__) || __cplusplus >= 201103L
|
||
|
|
DNSSender& operator=(const DNSSender& other);
|
||
|
|
DNSSender(DNSSender &&other) noexcept;
|
||
|
|
DNSSender const & operator=(DNSSender &&other);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
public:
|
||
|
|
class Results {
|
||
|
|
public:
|
||
|
|
int queryrate;
|
||
|
|
ppluint64 counter_send;
|
||
|
|
ppluint64 counter_received;
|
||
|
|
ppluint64 bytes_send;
|
||
|
|
ppluint64 bytes_received;
|
||
|
|
ppluint64 counter_errors;
|
||
|
|
ppluint64 packages_lost;
|
||
|
|
ppluint64 counter_0bytes;
|
||
|
|
ppluint64 counter_errorcodes[255];
|
||
|
|
ppluint64 rcodes[16];
|
||
|
|
ppluint64 truncated;
|
||
|
|
double rtt_total;
|
||
|
|
double rtt_avg;
|
||
|
|
double rtt_min;
|
||
|
|
double rtt_max;
|
||
|
|
Results();
|
||
|
|
void clear();
|
||
|
|
};
|
||
|
|
|
||
|
|
private:
|
||
|
|
ppl7::ThreadPool threadpool;
|
||
|
|
ppl7::IPAddress TargetIP;
|
||
|
|
ppl7::IPAddress SourceIP;
|
||
|
|
ppl7::IPNetwork SourceNet;
|
||
|
|
ppl7::String CSVFileName;
|
||
|
|
ppl7::String QueryFilename;
|
||
|
|
ppl7::File CSVFile;
|
||
|
|
ppl7::Array rates;
|
||
|
|
ppl7::String InterfaceName;
|
||
|
|
PayloadFile payload;
|
||
|
|
DNSReceiverThread* Receiver;
|
||
|
|
DNSSender::Results vis_prev_results;
|
||
|
|
SystemStat sys1, sys2;
|
||
|
|
|
||
|
|
int TargetPort;
|
||
|
|
int Runtime;
|
||
|
|
int Timeout;
|
||
|
|
int ThreadCount;
|
||
|
|
int DnssecRate;
|
||
|
|
float Timeslices;
|
||
|
|
bool ignoreResponses;
|
||
|
|
bool spoofingEnabled;
|
||
|
|
bool spoofFromPcap;
|
||
|
|
|
||
|
|
void openCSVFile(const ppl7::String& Filename);
|
||
|
|
void run(int queryrate);
|
||
|
|
void presentResults(const DNSSender::Results& result);
|
||
|
|
void saveResultsToCsv(const DNSSender::Results& result);
|
||
|
|
void prepareThreads();
|
||
|
|
void getResults(DNSSender::Results& result);
|
||
|
|
ppl7::Array getQueryRates(const ppl7::String& QueryRates);
|
||
|
|
void readSourceIPList(const ppl7::String& filename);
|
||
|
|
|
||
|
|
void getTarget(int argc, char** argv);
|
||
|
|
void getSource(int argc, char** argv);
|
||
|
|
int getParameter(int argc, char** argv);
|
||
|
|
int openFiles();
|
||
|
|
void calcTimeslice(int queryrate);
|
||
|
|
|
||
|
|
void showCurrentStats(ppl7::ppl_time_t start_time);
|
||
|
|
|
||
|
|
public:
|
||
|
|
DNSSender();
|
||
|
|
~DNSSender();
|
||
|
|
void help();
|
||
|
|
int main(int argc, char** argv);
|
||
|
|
};
|
||
|
|
|
||
|
|
DNSSender::Results operator-(const DNSSender::Results& first, const DNSSender::Results& second);
|
||
|
|
|
||
|
|
#endif
|