mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-04-23 06:50:02 +02:00
114 lines
3.9 KiB
C++
114 lines
3.9 KiB
C++
// SPDX-License-Identifier: LGPL-3.0-or-other
|
|
// Copyright (C) 2021 Contributors to the SLS Detector Package
|
|
#include "sls/ClientSocket.h"
|
|
#include "sls/logger.h"
|
|
#include "sls/sls_detector_defs.h"
|
|
#include "sls/sls_detector_exceptions.h"
|
|
#include <arpa/inet.h>
|
|
#include <cassert>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include <unistd.h>
|
|
namespace sls {
|
|
|
|
ClientSocket::ClientSocket(std::string stype, const std::string &host,
|
|
uint16_t port)
|
|
: DataSocket(socket(AF_INET, SOCK_STREAM, 0)), socketType(stype) {
|
|
|
|
struct addrinfo hints, *result;
|
|
memset(&hints, 0, sizeof(hints));
|
|
hints.ai_family = AF_INET;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_flags |= AI_CANONNAME;
|
|
|
|
if (getaddrinfo(host.c_str(), nullptr, &hints, &result) != 0) {
|
|
std::string msg = "ClientSocket cannot decode host:" + host +
|
|
" on port " + std::to_string(port) + "\n";
|
|
throw SocketError(msg);
|
|
}
|
|
|
|
// TODO! Erik, results could have multiple entries do we need to loop
|
|
// through them? struct sockaddr_in serverAddr {};
|
|
serverAddr.sin_family = AF_INET;
|
|
serverAddr.sin_port = htons(port);
|
|
memcpy((char *)&serverAddr.sin_addr.s_addr,
|
|
&((struct sockaddr_in *)result->ai_addr)->sin_addr,
|
|
sizeof(in_addr_t));
|
|
|
|
if (::connect(getSocketId(), (struct sockaddr *)&serverAddr,
|
|
sizeof(serverAddr)) != 0) {
|
|
freeaddrinfo(result);
|
|
std::string msg = "ClientSocket: Cannot connect to " + socketType +
|
|
":" + host + " on port " + std::to_string(port) +
|
|
"\n";
|
|
throw SocketError(msg);
|
|
}
|
|
freeaddrinfo(result);
|
|
}
|
|
|
|
ClientSocket::ClientSocket(std::string sType, struct sockaddr_in addr)
|
|
: DataSocket(socket(AF_INET, SOCK_STREAM, 0)), socketType(sType) {
|
|
|
|
if (::connect(getSocketId(), (struct sockaddr *)&addr, sizeof(addr)) != 0) {
|
|
char address[INET_ADDRSTRLEN];
|
|
inet_ntop(AF_INET, &addr.sin_addr, address, INET_ADDRSTRLEN);
|
|
std::string msg = "ClientSocket: Cannot connect to " + socketType +
|
|
":" + address + " on port " +
|
|
std::to_string(addr.sin_port) + "\n";
|
|
throw SocketError(msg);
|
|
}
|
|
}
|
|
|
|
int ClientSocket::sendCommandThenRead(int fnum, const void *args,
|
|
size_t args_size, void *retval,
|
|
size_t retval_size) {
|
|
int ret = slsDetectorDefs::FAIL;
|
|
Send(&fnum, sizeof(fnum));
|
|
setFnum(fnum);
|
|
Send(args, args_size);
|
|
readReply(ret, retval, retval_size);
|
|
return ret;
|
|
}
|
|
|
|
void ClientSocket::readReply(int &ret, void *retval, size_t retval_size) {
|
|
|
|
try {
|
|
Receive(&ret, sizeof(ret));
|
|
if (ret == slsDetectorDefs::FAIL) {
|
|
char mess[MAX_STR_LENGTH]{};
|
|
// get error message
|
|
Receive(mess, sizeof(mess));
|
|
// Do we need to know hostname here?
|
|
// In that case save it???
|
|
if (socketType == "Receiver") {
|
|
throw ReceiverError("Receiver returned: " + std::string(mess));
|
|
} else if (socketType == "Detector") {
|
|
throw DetectorError("Detector returned: " + std::string(mess));
|
|
} else {
|
|
throw GuiError(mess);
|
|
}
|
|
}
|
|
// get retval
|
|
Receive(retval, retval_size);
|
|
}
|
|
// debugging
|
|
catch (sls::SocketError &e) {
|
|
if (socketType == "Receiver") {
|
|
throw ReceiverError("Receiver returned: " + std::string(e.what()));
|
|
} else if (socketType == "Detector") {
|
|
throw DetectorError("Detector returned: " + std::string(e.what()));
|
|
} else {
|
|
throw GuiError(e.what());
|
|
}
|
|
}
|
|
}
|
|
|
|
std::string ClientSocket::readErrorMessage() {
|
|
std::string error_msg(MAX_STR_LENGTH, '\0');
|
|
Receive(&error_msg[0], error_msg.size());
|
|
return error_msg;
|
|
}
|
|
|
|
}; // namespace sls
|