ziggi/stvpn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

working commit

Browse Source
This commit is contained in:
Олег Бородин
2026-05-17 11:31:28 +02:00
parent 886684e224
commit cfb8329741
5 changed files with 107 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
uxclient.cpp
+50 -34
View File
@@ -56,11 +56,10 @@ std::expected<void, std::string> UxClient::Connect(const std::string naddress, c
if (inet_pton(AF_INET, addr, &serv_addr.sin_addr) <= 0) {
return std::unexpected("Invalid server IP address");
}
#ifdef CLI_TIMEOUT
struct timeval timeout;
timeout.tv_sec = 5;
timeout.tv_usec = 0;
if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof timeout) < 0) {
int errnoCopy = errno;
std::string error = std::strerror(errnoCopy);
@@ -71,6 +70,7 @@ std::expected<void, std::string> UxClient::Connect(const std::string naddress, c
std::string error = std::strerror(errnoCopy);
return std::unexpected(std::format("Set send timeout error: {}", error));
}
#endif
if (connect(sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0) {
int errnoCopy = errno;
std::string error = std::strerror(errnoCopy);
@@ -101,17 +101,41 @@ void UxClient::RecvMessages(void) {
uxlogger.Log(std::format("Decode header error: {}", decodeRes.error()));
}
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;
}
uxlogger.Log(std::format("Receive message with size {}", pSize));
if (pSize == 0) {
continue;
}
uxlogger.Log(std::format("Receive message with size {}", pSize));
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 meta;
meta.ParseFromString(rawMessage);
auto metaPart = meta.meta();
auto kind = metaPart.kind();
uxlogger.Debug(std::format("Receive message with call {}", kind));
if (kind == tunAddressMsg) {
uxcontrol::AddressMessage addrMessage;
addrMessage.ParseFromString(rawMessage);
uxlogger.Debug(std::format("Receive address {}", addrMessage.address()));
uxlogger.Debug(std::format("Receive prefix {}", addrMessage.prefix()));
auto setAddrRes = interface.SetIP4Address(addrMessage.address());
if (!setAddrRes) {
uxlogger.Error(setAddrRes.error());
}
auto setMaskRes = interface.SetIP4Netmask(addrMessage.prefix());
if (!setMaskRes) {
uxlogger.Error(setMaskRes.error());
}
} else {
uxlogger.Warning("Received unknown kind of message");
}
}
done.release();
return;
@@ -130,7 +154,7 @@ void UxClient::SendMessages(void) {
uxcontrol::PacketMessage intPacket;
auto meta = intPacket.mutable_meta();
meta->set_kind("internetPacket");
meta->set_kind(internetPkgMsg);
intPacket.set_payload(packet);
std::string rawMessage;
@@ -139,17 +163,15 @@ void UxClient::SendMessages(void) {
MessageHeader header(rawMessage.size());
auto rawHeader = header.Encode();
std::string rawPacket;
rawPacket.append(rawHeader);
rawPacket.append(rawMessage);
int wsize = 0;
if ((wsize = send(sock, rawHeader.data(), rawHeader.size(), 0)) < 0) {
if ((wsize = send(sock, rawPacket.data(), rawPacket.size(), 0)) < 0) {
int errnoCopy = errno;
std::string error = std::strerror(errnoCopy);
uxlogger.Log(std::format("Write header error: {}", error));
break;
}
if ((wsize = send(sock, rawMessage.data(), rawMessage.size(), 0)) < 0) {
int errnoCopy = errno;
std::string error = std::strerror(errnoCopy);
uxlogger.Log(std::format("Write message error: {}", error));
uxlogger.Log(std::format("Write packet error: {}", error));
break;
}
uxlogger.Log("Send message");
@@ -165,14 +187,7 @@ std::expected<void, std::string> UxClient::Run(void) {
if (!createRes) {
return std::unexpected(createRes.error());
}
auto setAddrRes = interface.SetIP4Address("10.1.1.2");
if (!setAddrRes) {
return std::unexpected(setAddrRes.error());
}
auto setMaskRes = interface.SetIP4Netmask(30);
if (!setMaskRes) {
return std::unexpected(setMaskRes.error());
}
auto setMTURes = interface.SetMTU(1280);
if (!setMTURes) {
return std::unexpected(setMTURes.error());
@@ -182,12 +197,13 @@ std::expected<void, std::string> UxClient::Run(void) {
return std::unexpected(createRes.error());
}
Router router;
auto addRouteRes = router.AddRoute("1.1.1.2", 32, "10.1.1.2");
if (!addRouteRes) {
return std::unexpected(addRouteRes.error());
}
/*
Router router;
auto addRouteRes = router.AddRoute("1.1.1.2", 32, "10.1.1.2");
if (!addRouteRes) {
return std::unexpected(addRouteRes.error());
}
*/
std::thread sendThr(&UxClient::SendMessages, this);
sendThr.detach();