зеркало из https://github.com/microsoft/Incast.git
153 строки
3.4 KiB
C++
153 строки
3.4 KiB
C++
// Incast
|
|
//
|
|
// Copyright (c) Microsoft Corporation
|
|
//
|
|
// All rights reserved.
|
|
//
|
|
// MIT License
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the "Software"),
|
|
// to deal in the Software without restriction, including without limitation
|
|
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
// and/or sell copies of the Software, and to permit persons to whom the
|
|
// Software is furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
// DEALINGS IN THE SOFTWARE.
|
|
|
|
#ifndef _INCAST_H
|
|
#define _INCAST_H
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#define NOMINMAX
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#include <windows.h>
|
|
#include <iphlpapi.h>
|
|
#include <process.h>
|
|
#include <conio.h>
|
|
#include <sys/stat.h>
|
|
#include <vector>
|
|
#include <algorithm>
|
|
#include <map>
|
|
#include <limits>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <fstream>
|
|
|
|
#ifndef _M_ARM
|
|
#include <mmsystem.h>
|
|
#endif
|
|
|
|
#include "barrier.h"
|
|
|
|
const unsigned PORT = 27779;
|
|
const int DEFAULT_ITERS = 10000;
|
|
const int WARMUP_ITERS = 10;
|
|
const int DEFAULT_FO_MSG_SIZE = 256;
|
|
const int DEFAULT_FI_MSG_SIZE = 4096;
|
|
|
|
enum DelayMethod
|
|
{
|
|
NONE,
|
|
RANDOM_JITTER,
|
|
UNIFORM_SCHED
|
|
};
|
|
|
|
struct GlobalTestParameters
|
|
{
|
|
int clients;
|
|
int iters;
|
|
bool rate_limited;
|
|
int target_rate;
|
|
int fo_msg_size;
|
|
int fi_msg_size;
|
|
bool clients_limited;
|
|
int client_limit;
|
|
int delay;
|
|
DelayMethod delay_method;
|
|
|
|
// ISSUE-REVIEW
|
|
// Should these be broken down into distinct
|
|
// client and server options?
|
|
bool nagle;
|
|
int send_buffer;
|
|
int recv_buffer;
|
|
|
|
bool histogram;
|
|
|
|
GlobalTestParameters()
|
|
: clients(0)
|
|
, iters(DEFAULT_ITERS)
|
|
, rate_limited(false)
|
|
, target_rate(0)
|
|
, fo_msg_size(DEFAULT_FO_MSG_SIZE)
|
|
, fi_msg_size(DEFAULT_FI_MSG_SIZE)
|
|
, clients_limited(false)
|
|
, client_limit(0)
|
|
, delay(0)
|
|
, delay_method(NONE)
|
|
, nagle(true)
|
|
, send_buffer(-1)
|
|
, recv_buffer(-1)
|
|
, histogram(false)
|
|
{};
|
|
} gtp;
|
|
|
|
struct ClientSpecificTestParameters
|
|
{
|
|
int client_num;
|
|
ClientSpecificTestParameters()
|
|
: client_num(-1)
|
|
{};
|
|
};
|
|
|
|
struct ClientResultData
|
|
{
|
|
int retransmits;
|
|
|
|
ClientResultData()
|
|
: retransmits(0)
|
|
{};
|
|
};
|
|
|
|
struct Measurement
|
|
{
|
|
__int64 actual_delay;
|
|
__int64 start;
|
|
__int64 stop;
|
|
};
|
|
|
|
typedef std::vector<Measurement> Measurements;
|
|
|
|
struct TestResult
|
|
{
|
|
ClientResultData crd;
|
|
Measurements measurements;
|
|
};
|
|
|
|
std::vector<TestResult> clientResults;
|
|
std::vector<HANDLE> clientThreads;
|
|
std::vector<SOCKET> clientSockets;
|
|
|
|
barrier *pb;
|
|
|
|
std::ofstream histfile;
|
|
|
|
MIB_TCPSTATS tcpStatsBefore, tcpStatsAfter;
|
|
|
|
typedef std::map<std::string,std::vector<int>> ClientAddressMap;
|
|
ClientAddressMap clientAddressMap;
|
|
|
|
#endif // _INCAST_H
|