Added simple TCP socket server to handle requests and TCP socket client to test and measure.

This commit is contained in:
ftc
2023-02-14 22:48:18 +03:00
commit 09457cb6be
9 changed files with 535 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
build/
+23
View File
@@ -0,0 +1,23 @@
cmake_minimum_required(VERSION 3.0)
project(ipc_test_server)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/../cmake")
## SOURCES ####################################################################
set(SOURCE_FILES
ipc_server.cpp
task.h
task.cpp
SynchronousSocket.cpp
SynchronousSocket.h
)
##################################################################################
link_libraries(stdc++)
add_executable(ipc_test_server ${SOURCE_FILES} ./ipc_server.cpp)
+161
View File
@@ -0,0 +1,161 @@
#include "SynchronousSocket.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <cstring>
#include <unistd.h>
#include <errno.h>
SynchronousSocket::SynchronousSocket()
{
socket = -1;
}
void SynchronousSocket::Init(int s)
{
socket = s;
}
bool SynchronousSocket::IsConnected()
{
return (socket != -1);
}
void SynchronousSocket::Connect(string host, int port)
{
// уже подключены?
if (socket != -1) return;
struct sockaddr_in remoteaddr;
struct addrinfo *addresses;
struct addrinfo hints;
memset(&hints, 0, sizeof(hints));
hints.ai_socktype = SOCK_STREAM;
if (getaddrinfo(host.c_str(), NULL, NULL, &addresses) != 0)
{
throw (string)("Unable to getaddrinfo");
return;
}
// копипастим себе самый первый адрес (в принципе, нам любой норм)
memcpy(&remoteaddr, addresses[0].ai_addr, sizeof(remoteaddr));
remoteaddr.sin_port = htons(port);
freeaddrinfo(addresses);
// создаем сокет
if ((socket = ::socket(PF_INET, SOCK_STREAM, 0)) == -1)
{
throw (string)("Unable to create socket");
return;
}
if (connect(socket, (struct sockaddr *)(&remoteaddr), sizeof(struct sockaddr_in)) == -1)
{
if (errno != EINPROGRESS)
{
socket = -1;
throw (string)("Unable to connect");
return;
}
}
}
void SynchronousSocket::Send(const string & data)
{
int pos = 0;
time_t start = time(NULL);
while (pos < (int)data.length())
{
int res = send(socket, &((data.data())[pos]), data.length() - pos, 0);
if (res == -1)
{
if (errno == EAGAIN || errno == EWOULDBLOCK)
{
if (time(NULL) - start > TIMEOUT)
{
throw (string)("Timeout");
}
continue;
}
else
{
// сокет поломался - закрываем его
throw (string)("Socket closed");
}
}
else
{
pos += res;
}
}
}
string SynchronousSocket::Receive(int amount)
{
string r = "";
time_t start = time(NULL);
while ((int)r.length() < amount)
{
char buf[1024];
int to_recv = min((int)sizeof(buf), amount - (int)r.length());
int res = recv(socket, buf, to_recv, 0);
if (res == 0)
{
// бросаем специфический exception, означающий, что сокет закрыт
throw (string)("Socket closed");
}
else if (res == -1)
{
if (errno == EAGAIN || errno == EWOULDBLOCK)
{
if (time(NULL) - start > TIMEOUT)
{
throw (string)("Timeout");
}
continue;
}
else
{
// что-то сломалось, надо закрыть сокет на всякий случай
// бросаем специфический exception, означающий, что сокет закрыт
throw (string)("Socket closed");
}
}
else
{
for (int i = 0; i < res; i++) r += buf[i];
}
}
return r;
}
void SynchronousSocket::Disconnect()
{
if (socket == -1) return;
close(socket);
socket = -1;
}
string SynchronousSocket::ReceiveMessage()
{
string len_s = Receive(4);
int len = 0;
for (size_t i = 0; i < len_s.length(); i++)
{
len |= ((unsigned char)len_s[i]) << (8 * i);
}
return Receive(len);
}
void SynchronousSocket::SendMessage(const string & m)
{
int len = m.length();
string len_s = "";
for (int i = 0; i < 4; i++)
{
len_s += (char)((len >> (8 * i)) & 0xFF);
}
Send(len_s);
Send(m);
}
+29
View File
@@ -0,0 +1,29 @@
#ifndef SYNCHRONOUS_SOCKET_HEADER_FILE_INCLUDED
#define SYNCHRONOUS_SOCKET_HEADER_FILE_INCLUDED
#include <string>
using namespace std;
class SynchronousSocket
{
private:
const int TIMEOUT = 10; // in seconds
protected:
// собственно сокет
int socket;
public:
SynchronousSocket();
// Используется для проброса уже готового сокета и работы с ним
void Init(int s);
bool IsConnected();
virtual void Connect(string host, int port);
virtual void Send(const string &data);
virtual string Receive(int amount);
void Disconnect();
string ReceiveMessage();
void SendMessage(const string &m);
};
#endif
+88
View File
@@ -0,0 +1,88 @@
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string>
#include <cstring>
#include <unistd.h>
#include <sys/file.h>
#include <errno.h>
#include <iostream>
#include "SynchronousSocket.h"
#include "task.h"
using namespace std;
const int LISTEN_PORT = 19876;
const int LISTEN_BACKLOG = 10;
const int EXIT_FAIL = 1;
void ProcessRequests(int socket_fd)
{
SynchronousSocket s;
s.Init(socket_fd);
try {
while (true)
{
string message = s.ReceiveMessage();
s.SendMessage(Task(message));
}
}
catch (string s)
{
cerr << "Error in socket " << s << "\n";
}
catch (...)
{
cerr << "Unknown error in socket\n";
}
s.Disconnect();
}
int main()
{
int listensock;
if ((listensock = socket(PF_INET, SOCK_STREAM, 0)) == -1)
{
cerr << "Could not create socket for listening...\n";
return EXIT_FAIL;
}
struct sockaddr_in listenaddr;
memset(&listenaddr, 0, sizeof(listenaddr));
listenaddr.sin_family = AF_INET;
listenaddr.sin_addr.s_addr = htonl(INADDR_ANY);
listenaddr.sin_port = htons(LISTEN_PORT);
if (bind(listensock, (struct sockaddr *)(&listenaddr), sizeof(struct sockaddr_in)) == -1)
{
cerr << "Could not bind listening socket...\n";
return EXIT_FAIL;
}
if (listen(listensock, LISTEN_BACKLOG) == -1)
{
cerr << "Could not set socket into listening mode...\n";
return EXIT_FAIL;
}
struct sockaddr_in peeraddr;
socklen_t paddrlen;
paddrlen = sizeof(peeraddr);
int recsock;
cout << "Press Ctrl-C for exit\n";
cout << "Accepting connections on port " << LISTEN_PORT << "\n";
while (true)
{
(recsock = accept(listensock, (struct sockaddr *)(&peeraddr), &paddrlen));
if (recsock == -1)
{
cerr << "Error accepting connection, errno = " << errno << "\n";
continue;
}
cout << "Accepted connection from " << inet_ntoa(peeraddr.sin_addr) << ":" << ntohs(peeraddr.sin_port) << "\n";
ProcessRequests(recsock);
}
cerr << "Accept error...\n";
if (close(listensock) == -1)
{
cerr << "Could not close listening socket...\n";
exit(EXIT_FAIL);
}
return 0;
}
+23
View File
@@ -0,0 +1,23 @@
#include "task.h"
#include <cstring>
#include <sstream>
using namespace std;
int cnt[256];
string Task(const string &message)
{
memset(cnt, 0, sizeof(cnt));
for (int i = 0; i < (int)message.length(); i++)
{
cnt[(int)(unsigned char)message[i]]++;
}
ostringstream result;
for (int i = 0; i < 256; i++)
{
if (cnt[i] == 0) continue;
result << i << ":" << cnt[i] << ";";
}
return result.str();
}
+8
View File
@@ -0,0 +1,8 @@
#ifndef TASK_HEADER_FILE_INCLUDED
#define TASK_HEADER_FILE_INCLUDED
#include <string>
std::string Task(const std::string &message);
#endif