Skip to content

Commit

Permalink
socket-benchmark.
Browse files Browse the repository at this point in the history
  • Loading branch information
@b4rtaz
b4rtaz committed Nov 10, 2024
1 parent 293e68b commit bfd3967
Show file tree
Hide file tree
Showing 3 changed files with 153 additions and 51 deletions.
2 changes: 1 addition & 1 deletion .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
__pycache__

*-test
./*-benchmark
./socket-benchmark
main
run*.sh
server
Expand Down
Binary file removed socket-benchmark
View file
Binary file not shown.
202 changes: 152 additions & 50 deletions src/apps/socket-benchmark/socket-benchmark.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@
#include <chrono>
#include <cstring>
#include <cstdio>
#include <sys/socket.h>
#include <arpa/inet.h>

using namespace std::chrono;

unsigned int packageSizes[] = { 128, 256, 512, 768, 1024, 1280, 1518, 2048, 4096, 8192, 16384, 32768, 65536 };
Expand All @@ -12,69 +15,168 @@ int port = 7721;

void server() {
printf("nAttempts: %d\n", nAttempts);
SocketServer server(port);
Socket socket = server.accept();
char buffer[maPackageSize];

for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]
for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();
socket.read(buffer, currentPackageSize);
auto t1 = high_resolution_clock::now();
socket.write(buffer, currentPackageSize);
auto t2 = high_resolution_clock::now();

totalReadTime += duration_cast<microseconds>(t1 - t0).count();
totalWriteTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
printf("TCP test\n");

{
SocketServer server(port);
Socket socket = server.accept();
for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]
for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();
socket.read(buffer, currentPackageSize);
auto t1 = high_resolution_clock::now();
socket.write(buffer, currentPackageSize);
auto t2 = high_resolution_clock::now();

totalReadTime += duration_cast<microseconds>(t1 - t0).count();
totalWriteTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
}

double nPingPongs = (1.0 / (totalTime / 1000000.0)) * (double)nAttempts;
printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us, nPingPongs: %.2f\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime, nPingPongs);
}
}

printf("UDP test\n");

{
int serverSocket = ::socket(AF_INET, SOCK_DGRAM, 0);
struct sockaddr_in serverAddr;
struct sockaddr_in clientAddr;
socklen_t clientAddrLen = sizeof(clientAddr);
memset(&serverAddr, 0, sizeof(serverAddr));
memset(&clientAddr, 0, sizeof(clientAddr));
serverAddr.sin_family = AF_INET;
serverAddr.sin_addr.s_addr = INADDR_ANY;
serverAddr.sin_port = htons(port);

if (bind(serverSocket, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
throw std::runtime_error("Cannot bind socket");

double nPingPongs = (1.0 / (totalTime / 1000000.0)) * (double)nAttempts;
printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us, nPingPongs: %.2f\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime, nPingPongs);
for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]

for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();

ssize_t s0 = recvfrom(serverSocket, (char*)buffer, currentPackageSize, MSG_WAITALL, (struct sockaddr*)&clientAddr, &clientAddrLen);
if (s0 != currentPackageSize)
throw std::runtime_error("Cannot read from socket");

auto t1 = high_resolution_clock::now();

ssize_t s1 = sendto(serverSocket, (const char*)buffer, currentPackageSize, MSG_DONTWAIT, (const struct sockaddr*)&clientAddr, clientAddrLen);
if (s1 != currentPackageSize)
throw std::runtime_error("Cannot write to socket");

auto t2 = high_resolution_clock::now();

totalReadTime += duration_cast<microseconds>(t1 - t0).count();
totalWriteTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
}

double nPingPongs = (1.0 / (totalTime / 1000000.0)) * (double)nAttempts;
printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us, nPingPongs: %.2f\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime, nPingPongs);
}
}
}

void client(char* host) {
char** hosts = new char*[1];
hosts[0] = host;
int* ports = new int[1];
ports[0] = port;

SocketPool* pool = SocketPool::connect(1, hosts, ports);
char buffer[maPackageSize];
pool->setTurbo(true);

for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]
for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();
pool->write(0, buffer, currentPackageSize);
auto t1 = high_resolution_clock::now();
pool->read(0, buffer, currentPackageSize);
auto t2 = high_resolution_clock::now();

totalWriteTime += duration_cast<microseconds>(t1 - t0).count();
totalReadTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
printf("TCP test\n");

{
char** hosts = new char*[1];
hosts[0] = host;
int* ports = new int[1];
ports[0] = port;

SocketPool* pool = SocketPool::connect(1, hosts, ports);
pool->setTurbo(true);

for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]
for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();
pool->write(0, buffer, currentPackageSize);
auto t1 = high_resolution_clock::now();
pool->read(0, buffer, currentPackageSize);
auto t2 = high_resolution_clock::now();

totalWriteTime += duration_cast<microseconds>(t1 - t0).count();
totalReadTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
}

printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime);
}

printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime);
delete pool;
delete[] hosts;
delete[] ports;
}

delete pool;
delete[] hosts;
delete[] ports;
printf("UDP test\n");

{
int clientSocket = ::socket(AF_INET, SOCK_DGRAM, 0);
struct sockaddr_in serverAddr;
socklen_t serverAddrLen = sizeof(serverAddr);
memset(&serverAddr, 0, sizeof(serverAddr));
serverAddr.sin_family = AF_INET;
serverAddr.sin_port = htons(port);
serverAddr.sin_addr.s_addr = inet_addr(host);

for (long i = 0; i < packageSizesCount; i++) {
unsigned int currentPackageSize = packageSizes[i];

long long totalReadTime = 0;
long long totalWriteTime = 0;
long long totalTime = 0; // [us]
for (long a = 0; a < nAttempts; a++) {
auto t0 = high_resolution_clock::now();

ssize_t s0 = sendto(clientSocket, (const char *)buffer, currentPackageSize, 0, (const struct sockaddr*)&serverAddr, sizeof(serverAddr));
if (s0 != currentPackageSize)
throw std::runtime_error("Cannot write to socket");

auto t1 = high_resolution_clock::now();

ssize_t s1 = recvfrom(clientSocket, (char*)buffer, currentPackageSize, MSG_WAITALL, (struct sockaddr*)&serverAddr, &serverAddrLen);
if (s1 != currentPackageSize)
throw std::runtime_error("Cannot read from socket");

auto t2 = high_resolution_clock::now();

totalWriteTime += duration_cast<microseconds>(t1 - t0).count();
totalReadTime += duration_cast<microseconds>(t2 - t1).count();
totalTime += duration_cast<microseconds>(t2 - t0).count();
}

printf("[%6d bytes] write: %5lld us, read: %5lld us, total: %5lld us\n",
currentPackageSize, totalWriteTime, totalReadTime, totalTime);
}
}
}

int main(int argc, char *argv[]) {
Expand Down

0 comments on commit bfd3967

Please sign in to comment.