operations.cpp

#include "utils.h"

int send_all(int socket, void *buffer, size_t length) {
    char *buf = (char *) buffer;
    size_t bytes_sent = 0;

    while (bytes_sent < length) {
        ssize_t result = send(socket, buf + bytes_sent, length - bytes_sent, 0);
        if (result < 0) {
            perror("Error at send");
            return -1;
        }
        bytes_sent = bytes_sent + result;
    }
    return bytes_sent;
}

int recv_all(int socket, void *buffer, size_t length) {
    char *buf = (char *) buffer;
    size_t bytes_received = 0;
    while (bytes_received < length) {
        size_t result = recv(socket, buf + bytes_received, length - bytes_received, 0);
        if (result < 0) {
            perror("Error at recv");
            return result;
        }
        if (result == 0) {
            break;
        }
        bytes_received = bytes_received + result;
    }
    return bytes_received;
}

void send_message(int socket, void *buffer, size_t length) {
    send_all(socket, &length, sizeof(length));
    send_all(socket, buffer, length);
}