summaryrefslogtreecommitdiffstats
path: root/apps/hiperf/src/common.h
blob: e6ba526f933af399f0e182d616fb4e36bdb4e326 (plain)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
/*
 * Copyright (c) 2021 Cisco and/or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <hicn/transport/auth/identity.h>
#include <hicn/transport/auth/signer.h>
#include <hicn/transport/config.h>
#include <hicn/transport/core/content_object.h>
#include <hicn/transport/core/interest.h>
#include <hicn/transport/interfaces/global_conf_interface.h>
#include <hicn/transport/interfaces/p2psecure_socket_consumer.h>
#include <hicn/transport/interfaces/p2psecure_socket_producer.h>
#include <hicn/transport/interfaces/socket_consumer.h>
#include <hicn/transport/interfaces/socket_producer.h>
#include <hicn/transport/utils/chrono_typedefs.h>
#include <hicn/transport/utils/literals.h>

#ifndef _WIN32
#include <hicn/transport/utils/daemonizator.h>
#endif

#include <asio.hpp>
#include <cmath>
#include <fstream>
#include <iomanip>
#include <sstream>
#include <string>
#include <unordered_set>

#ifndef ERROR_SUCCESS
#define ERROR_SUCCESS 0
#endif
#define ERROR_SETUP -5
#define MIN_PROBE_SEQ 0xefffffff

using namespace transport::interface;
using namespace transport::auth;
using namespace transport::core;

static inline uint64_t _ntohll(const uint64_t *input) {
  uint64_t return_val;
  uint8_t *tmp = (uint8_t *)&return_val;

  tmp[0] = *input >> 56;
  tmp[1] = *input >> 48;
  tmp[2] = *input >> 40;
  tmp[3] = *input >> 32;
  tmp[4] = *input >> 24;
  tmp[5] = *input >> 16;
  tmp[6] = *input >> 8;
  tmp[7] = *input >> 0;

  return return_val;
}

static inline uint64_t _htonll(const uint64_t *input) {
  return (_ntohll(input));
}

namespace hiperf {

/**
 * Class for handling the production rate for the RTC producer.
 */
class Rate {
 public:
  Rate() : rate_kbps_(0) {}

  Rate(const std::string &rate) {
    std::size_t found = rate.find("kbps");
    if (found != std::string::npos) {
      rate_kbps_ = std::stof(rate.substr(0, found));
    } else {
      throw std::runtime_error("Format " + rate + " not correct");
    }
  }

  Rate(const Rate &other) : rate_kbps_(other.rate_kbps_) {}

  Rate &operator=(const std::string &rate) {
    std::size_t found = rate.find("kbps");
    if (found != std::string::npos) {
      rate_kbps_ = std::stof(rate.substr(0, found));
    } else {
      throw std::runtime_error("Format " + rate + " not correct");
    }

    return *this;
  }

  std::chrono::microseconds getMicrosecondsForPacket(std::size_t packet_size) {
    return std::chrono::microseconds(
        (uint32_t)std::round(packet_size * 1000.0 * 8.0 / (double)rate_kbps_));
  }

 private:
  float rate_kbps_;
};

struct packet_t {
  uint64_t timestamp;
  uint32_t size;
};

/**
 * Container for command line configuration for hiperf client.
 */
struct ClientConfiguration {
  ClientConfiguration()
      : name("b001::abcd", 0),
        beta(-1.f),
        drop_factor(-1.f),
        window(-1),
        producer_certificate(""),
        passphrase(""),
        receive_buffer(nullptr),
        receive_buffer_size_(128 * 1024),
        download_size(0),
        report_interval_milliseconds_(1000),
        transport_protocol_(CBR),
        rtc_(false),
        test_mode_(false),
        relay_(false),
        secure_(false),
        producer_prefix_(),
        interest_lifetime_(500),
        relay_name_("c001::abcd/64"),
        output_stream_mode_(false),
        port_(0) {}

  Name name;
  double beta;
  double drop_factor;
  double window;
  std::string producer_certificate;
  std::string passphrase;
  std::shared_ptr<utils::MemBuf> receive_buffer;
  std::size_t receive_buffer_size_;
  std::size_t download_size;
  std::uint32_t report_interval_milliseconds_;
  TransportProtocolAlgorithms transport_protocol_;
  bool rtc_;
  bool test_mode_;
  bool relay_;
  bool secure_;
  Prefix producer_prefix_;
  uint32_t interest_lifetime_;
  Prefix relay_name_;
  bool output_stream_mode_;
  uint16_t port_;
};

/**
 * Container for command line configuration for hiperf server.
 */
struct ServerConfiguration {
  ServerConfiguration()
      : name("b001::abcd/64"),
        virtual_producer(true),
        manifest(false),
        live_production(false),
        content_lifetime(600000000_U32),
        download_size(20 * 1024 * 1024),
        hash_algorithm(CryptoHashType::SHA256),
        keystore_name(""),
        passphrase(""),
        keystore_password("cisco"),
        multiphase_produce_(false),
        rtc_(false),
        interactive_(false),
        trace_based_(false),
        trace_index_(0),
        trace_file_(nullptr),
        production_rate_(std::string("2048kbps")),
        payload_size_(1400),
        secure_(false),
        input_stream_mode_(false),
        port_(0) {}

  Prefix name;
  bool virtual_producer;
  bool manifest;
  bool live_production;
  std::uint32_t content_lifetime;
  std::uint32_t download_size;
  CryptoHashType hash_algorithm;
  std::string keystore_name;
  std::string passphrase;
  std::string keystore_password;
  bool multiphase_produce_;
  bool rtc_;
  bool interactive_;
  bool trace_based_;
  std::uint32_t trace_index_;
  char *trace_file_;
  Rate production_rate_;
  std::size_t payload_size_;
  bool secure_;
  bool input_stream_mode_;
  uint16_t port_;
  std::vector<struct packet_t> trace_;
};

}  // namespace hiperf