145 lines
4.9 KiB
C++
145 lines
4.9 KiB
C++
|
|
extern "C" {
|
|
#include <sys/socket.h>
|
|
}
|
|
|
|
#include <expected>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <thread>
|
|
#include <chrono>
|
|
#include <cstring>
|
|
#include <semaphore>
|
|
#include <format>
|
|
|
|
#include <msgheader.hpp>
|
|
#include <sockhand.hpp>
|
|
#include <uxlogger.hpp>
|
|
|
|
|
|
#include <uxcontrol.pb.h>
|
|
|
|
const std::string internetPkgMsg = "internetPkg";
|
|
const std::string tunAddressMsg = "tunAddress";
|
|
const std::string Msg = "localRoute";
|
|
|
|
|
|
using namespace std::chrono_literals;
|
|
|
|
|
|
void SocketHandler::Handle(int newsock, std::string laddr, std::string raddr, int prefix) {
|
|
sock = newsock;
|
|
auto interfaceName = std::format("uxsrv{}", sock);
|
|
auto createRes = interface.Create(interfaceName);
|
|
if (!createRes) {
|
|
uxlogger.Error(createRes.error());
|
|
return;
|
|
}
|
|
uxlogger.Debug(std::format("Set local ip address {}/{} for {}", laddr, prefix, interfaceName));
|
|
auto setAddrRes = interface.SetIP4Address(laddr);
|
|
if (!setAddrRes) {
|
|
uxlogger.Error(setAddrRes.error());
|
|
return;
|
|
}
|
|
auto setPrefixRes = interface.SetIP4Netmask(prefix);
|
|
if (!setPrefixRes) {
|
|
uxlogger.Error(setPrefixRes.error());
|
|
return;
|
|
}
|
|
auto upRes = interface.Up();
|
|
if (!createRes) {
|
|
uxlogger.Error(createRes.error());
|
|
return;
|
|
}
|
|
|
|
std::thread recvThr(&SocketHandler::RecvMessages, this);
|
|
recvThr.detach();
|
|
|
|
#if 1
|
|
uxcontrol::AddressMessage addrMsg;
|
|
auto meta = addrMsg.mutable_meta();
|
|
meta->set_kind(tunAddressMsg);
|
|
addrMsg.set_address(raddr);
|
|
addrMsg.set_prefix(prefix);
|
|
std::string rawMessage;
|
|
addrMsg.SerializeToString(&rawMessage);
|
|
|
|
MessageHeader header(rawMessage.size());
|
|
auto rawHeader = header.Encode();
|
|
std::string rawPacket;
|
|
rawPacket.append(rawHeader);
|
|
rawPacket.append(rawMessage);
|
|
|
|
int wsize;
|
|
if ((wsize = write(sock, rawPacket.data(), rawPacket.size())) < 0) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
uxlogger.Log(std::format("Write message error: {}", error));
|
|
return;
|
|
}
|
|
#endif
|
|
std::thread sendThr(&SocketHandler::SendMessages, this);
|
|
sendThr.detach();
|
|
|
|
done.acquire();
|
|
uxlogger.Log("Handler done");
|
|
}
|
|
|
|
|
|
void SocketHandler::SendMessages(void) {
|
|
while (true) {
|
|
MessageHeader header(0);
|
|
auto rawHeader = header.Encode();
|
|
int wsize;
|
|
{
|
|
if ((wsize = write(sock, rawHeader.data(), rawHeader.size())) < 0) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
uxlogger.Log(std::format("Write message error: {}", error));
|
|
break;
|
|
}
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::seconds(30));
|
|
}
|
|
done.release();
|
|
uxlogger.Log("Send messages done");
|
|
}
|
|
|
|
void SocketHandler::RecvMessages(void) {
|
|
while (true) {
|
|
int rsize = 0;
|
|
std::string rawHeader(msgHeaderSize, 0);
|
|
if ((rsize = recv(sock, rawHeader.data(), rawHeader.size(), MSG_WAITALL)) < 0) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
uxlogger.Log(std::format("Read header error: {}", error));
|
|
break;
|
|
}
|
|
if (rsize < msgHeaderSize) {
|
|
uxlogger.Log(std::format("Read only {} from {} bytes", rsize, msgHeaderSize));
|
|
break;
|
|
}
|
|
MessageHeader header;
|
|
auto decodeRes = header.Decode(rawHeader);
|
|
if (!decodeRes) {
|
|
uxlogger.Log(std::format("Decode header error: {}", decodeRes.error()));
|
|
break;
|
|
}
|
|
auto pSize = header.PayloadSize();
|
|
if (pSize > 0) {
|
|
std::string rawMessage(pSize, 0);
|
|
if ((rsize = recv(sock, rawMessage.data(), rawMessage.size(), MSG_WAITALL)) < 0) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
uxlogger.Log(std::format("Read message error: {}", error));
|
|
break;
|
|
}
|
|
uxcontrol::SimpleMessage msg;
|
|
msg.ParseFromString(rawMessage);
|
|
uxlogger.Log(std::format("Receive message {}", msg.meta().kind()));
|
|
}
|
|
}
|
|
done.release();
|
|
uxlogger.Log("Recv messages done");
|
|
}
|