this->sucks

This commit is contained in:
Theis Pieter Hollebeek 2024-11-20 13:04:55 +01:00
parent f96d4849f6
commit d26da549cc
4 changed files with 126 additions and 163 deletions

View File

@ -1,15 +1,77 @@
#include "rpc_server.hpp" #include "rpc_server.hpp"
#include "socket.hpp" #include <netinet/in.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <unistd.h> #include <unistd.h>
#include <utility>
auto slige_rpc::RpcServer::bind( auto create_address(uint16_t port) -> sockaddr_in
uint16_t port) -> std::variant<slige_rpc::RpcServer, slige_socket::Ewwow>
{ {
std::variant<slige_socket::ServerSocket, slige_socket::Ewwow> socket_result return {
= slige_socket::Socket::bind(port); .sin_family = AF_INET,
if (std::holds_alternative<slige_socket::Ewwow>(socket_result)) { .sin_port = htons(port),
return std::get<slige_socket::Ewwow>(socket_result); .sin_addr = { .s_addr = inet_addr("127.0.0.1") },
} .sin_zero = { 0 },
return RpcServer(std::get<slige_socket::ServerSocket>(socket_result)); };
}
template <typename Functor>
auto slige_rpc::RpcServer<Functor>::listen() -> Res<Unit>
{
int socket_fd = ::socket(AF_INET, SOCK_STREAM, 0);
if (socket_fd < 0) {
return Err { "could not get socket" };
};
sockaddr_in address = create_address(13370);
unsigned int address_size = sizeof(this->address);
if (::bind(socket_fd, (struct sockaddr*)&address, sizeof(address)) < 0) {
return Err { "could not bind" };
};
if (::listen(socket_fd, 0) < 0) {
return Err { "could not listen" };
}
while (true) {
int client = ::accept(
this->fd, (struct sockaddr*)&this->address, &address_size);
if (client < 0) {
return Err { "could not accept" };
}
uint8_t buffer[1024] = {};
}
std::unreachable();
}
auto slige_rpc::BufferedWriter::write(std::string message) -> Res<Unit>
{
for (size_t i = 0; i < message.length(); ++i) {
auto res = this->write((uint8_t)message[i]);
if (!res.is_ok()) {
return res.err();
}
}
return Unit {};
}
auto slige_rpc::BufferedWriter::write(uint8_t byte) -> Res<Unit>
{
if (this->occupied >= length) {
auto res = this->flush();
if (!res.is_ok()) {
return res.err();
}
}
this->buffer[this->occupied] = byte;
return Unit {};
}
auto slige_rpc::BufferedWriter::flush() -> Res<size_t>
{
auto result = ::write(this->fd, this->buffer, this->occupied);
if (result < 0) {
return { { "unable to write" } };
}
this->occupied = 0;
return (size_t)result;
} }

View File

@ -1,15 +1,60 @@
#pragma once #pragma once
#include "socket.hpp"
#include <arpa/inet.h> #include <arpa/inet.h>
#include <cstdint>
#include <netdb.h> #include <netdb.h>
#include <stdlib.h> #include <stdlib.h>
#include <string>
#include <unistd.h> #include <unistd.h>
#include <variant>
namespace slige_rpc { namespace slige_rpc {
struct Err {
std::string msg;
};
template <typename T> class Res {
public:
Res(T value)
{
this->value = value;
this->holds_value = true;
}
Res(Err error)
{
this->error = error;
this->holds_value = false;
}
auto is_ok() -> bool { return this->holds_value; }
auto ok() -> T { return this->value; }
auto err() -> Err { return this->error; }
private:
bool holds_value;
T value;
Err error;
};
struct Unit { };
struct Req { };
class BufferedWriter {
BufferedWriter(int fd)
: fd(fd)
{
}
~BufferedWriter() { close(fd); }
auto flush() -> Res<size_t>;
auto write(uint8_t byte) -> Res<Unit>;
auto write(std::string message) -> Res<Unit>;
private:
static const size_t length = 1024;
size_t occupied = 0;
uint8_t buffer[length];
int fd;
};
/// - load code /// - load code
/// - program input /// - program input
/// - run /// - run
@ -22,16 +67,15 @@ namespace slige_rpc {
/// - json string /// - json string
/// - fetch stack /// - fetch stack
/// - json string /// - json string
class RpcServer { template <typename Functor> class RpcServer {
public: public:
auto static bind(uint16_t port) RpcServer(Functor functor)
-> std::variant<slige_rpc::RpcServer, slige_socket::Ewwow>; : functor(functor) {};
auto listen() -> Res<Unit>;
private: private:
RpcServer(slige_socket::ServerSocket socket) Functor functor;
: socket(socket)
{
}
slige_socket::ServerSocket socket;
}; };
}; };

View File

@ -1,88 +0,0 @@
#include "socket.hpp"
#include <sys/socket.h>
#include <unistd.h>
auto slige_socket::ClientSocket::read(
uint8_t* buffer, size_t length) -> std::variant<size_t, slige_socket::Ewwow>
{
ssize_t bytes_read = recv(this->fd, buffer, length, 0);
if (bytes_read < 0) {
return Ewwow { .message = "unable to read" };
}
return (size_t)bytes_read;
};
auto slige_socket::ClientSocket::write(
uint8_t* buffer, size_t length) -> std::variant<size_t, slige_socket::Ewwow>
{
ssize_t bytes_written = send(this->fd, buffer, length, 0);
if (bytes_written < 0) {
return Ewwow { .message = "unable to write" };
}
return (size_t)bytes_written;
};
auto slige_socket::Socket::connect(
uint16_t port) -> std::variant<ClientSocket, slige_socket::Ewwow>
{
auto address = Socket::create_address(port);
int socket_fd;
if ((socket_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
return { Ewwow { .message = "could not get socket" } };
}
// TODO:
// 1) does address (presumably 'socket)
// live as long as child socket ('client)?
// - immediate answer is no
// 2) does it need to?
if (::connect(socket_fd, (struct sockaddr*)&address, sizeof(address)) < 0) {
return { Ewwow {
.message = "could not connect, is the server running?" } };
}
return { ClientSocket(socket_fd) };
}
auto slige_socket::ServerSocket::accept()
-> std::variant<ClientSocket, slige_socket::Ewwow>
{
unsigned int address_size = sizeof(this->address);
// TODO:
// 1) does address (presumably 'server)
// live as long as child socket ('socket)?
// - immediate answer is no
// 2) does it need to?
int client
= ::accept(this->fd, (struct sockaddr*)&this->address, &address_size);
if (client < 0) {
return Ewwow { .message = "could not accept" };
}
return { ClientSocket(client) };
}
auto slige_socket::Socket::bind(
uint16_t port) -> std::variant<ServerSocket, slige_socket::Ewwow>
{
sockaddr_in address = Socket::create_address(port);
int socket_fd;
if ((socket_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
return { Ewwow { .message = "could not get socket" } };
}
// TODO:
// 1) does address (presumably 'socket)
// live as long as child socket ('server)?
// - immediate answer is no
// 2) does it need to?
if (::bind(socket_fd, (struct sockaddr*)&address, sizeof(address)) < 0) {
return { Ewwow { .message = "could not bind" } };
}
if (::listen(socket_fd, 0) < 0) {
return { Ewwow { .message = "could not listen" } };
}
// TODO:
// 1) does this address get moved, copied or cloned?
// it should be represented as a u32,
// so copying should be fine - but does it?
return { ServerSocket(socket_fd, address) };
}

View File

@ -1,55 +0,0 @@
#pragma once
#include <arpa/inet.h>
#include <cstdint>
#include <netdb.h>
#include <stdlib.h>
#include <string>
#include <unistd.h>
#include <variant>
namespace slige_socket {
struct Ewwow {
std::string message;
};
class ClientSocket {
public:
ClientSocket(int fd)
: fd(fd)
{
}
~ClientSocket() { close(fd); };
auto read(uint8_t* buffer, size_t length) -> std::variant<size_t, Ewwow>;
auto write(uint8_t* buffer, size_t length) -> std::variant<size_t, Ewwow>;
private:
int fd;
};
class ServerSocket {
public:
ServerSocket(int fd, sockaddr_in address)
: fd(fd)
, address(address)
{
}
~ServerSocket() { close(fd); };
auto accept() -> std::variant<ClientSocket, Ewwow>;
private:
int fd;
sockaddr_in address;
};
class Socket {
public:
auto static connect(uint16_t port) -> std::variant<ClientSocket, Ewwow>;
auto static bind(uint16_t port) -> std::variant<ServerSocket, Ewwow>;
private:
auto static create_address(uint16_t port) -> sockaddr_in;
};
};