Олег Бородин
94 lines
3.1 KiB
C++
94 lines
3.1 KiB
C++
|
|
extern "C" {
|
|
#include <arpa/inet.h>
|
|
#include <linux/rtnetlink.h>
|
|
#include <net/route.h>
|
|
#include <netinet/in.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
}
|
|
|
|
#define BUFSIZE 8192
|
|
|
|
#include <expected>
|
|
#include <vector>
|
|
#include <format>
|
|
#include <cstring>
|
|
|
|
#include <iprouter.hpp>
|
|
|
|
std::expected<void, std::string> Router::AddRoute(std::string address, uint64_t prefix, std::string gateway) {
|
|
struct sockaddr_in sa;
|
|
if (inet_pton(AF_INET, address.data(), &(sa.sin_addr)) == 1) {
|
|
} else {
|
|
return std::unexpected(std::format("Incorrect address {}", address));
|
|
}
|
|
|
|
struct rtentry rt;
|
|
memset(&rt, 0, sizeof(rt));
|
|
|
|
struct sockaddr_in *sin;
|
|
sin = (struct sockaddr_in*)&rt.rt_dst;
|
|
sin->sin_family = AF_INET;
|
|
if (inet_pton(AF_INET, address.data(), &sin->sin_addr) != 1) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Incorrect address {}, error: {}", address, error));
|
|
};
|
|
|
|
sin = (struct sockaddr_in*) &rt.rt_gateway;
|
|
sin->sin_family = AF_INET;
|
|
if (inet_pton(AF_INET, gateway.data(), &sin->sin_addr) != 1) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Incorrect address {}, error: {}", gateway, error));
|
|
};
|
|
|
|
char buffer[INET_ADDRSTRLEN];
|
|
uint32_t mask = (prefix == 0) ? 0 : htonl(~((1U << (32 - prefix)) - 1));
|
|
//uint32_t mask = (prefix == 0) ? 0 : (~0U << (32 - prefix));
|
|
|
|
struct in_addr addr;
|
|
addr.s_addr = mask;
|
|
|
|
const char *dottedmask;
|
|
if ((dottedmask = inet_ntop(AF_INET, &addr, buffer, INET_ADDRSTRLEN)) == NULL) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Incorrect prefix {}, error: {}", prefix, error));
|
|
};
|
|
|
|
sin = (struct sockaddr_in*) &rt.rt_genmask;
|
|
sin->sin_family = AF_INET;
|
|
if (inet_pton(AF_INET, dottedmask, &sin->sin_addr) != 1) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Incorrect address {}, error: {}", gateway, error));
|
|
};
|
|
|
|
rt.rt_flags = RTF_UP | RTF_GATEWAY;
|
|
//rt.rt_dev = std::string("eth0").data();
|
|
|
|
int sockfd = 0;
|
|
if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Error: {}", error));
|
|
}
|
|
|
|
if (ioctl(sockfd, SIOCADDRT, &rt) < 0) {
|
|
close(sockfd);
|
|
int errnoCopy = errno;
|
|
std::string error = std::strerror(errnoCopy);
|
|
return std::unexpected(std::format("Cannot set route, error: {}", error));
|
|
}
|
|
close(sockfd);
|
|
return {};
|
|
}
|
|
|
|
|