libipc-old/core/msg.c

#include "msg.h"
#include "error.h"
#include "usocket.h"

#include <assert.h>

void ipc_message_print (const struct ipc_message *m)
{
    assert (m != NULL);
    printf ("msg: type %d len %d\n", m->type, m->length);
}

int ipc_message_format_read (struct ipc_message *m, const char *buf, ssize_t msize)
{
    assert (m != NULL);
    assert (buf != NULL);
    assert (msize <= BUFSIZ - 3);

    if (m == NULL)
        return -1;

    m->type = buf[0];
    memcpy (&m->length, buf+1, 2);

    assert (m->length <= BUFSIZ - 3);
    printf ("type %d : msize = %ld, length = %d\n", m->type, msize, m->length);
    assert (m->length == msize - 3 || m->length == 0);

    if (m->payload != NULL)
        free (m->payload), m->payload = NULL;

    if (m->payload == NULL && m->length > 0) {
        m->payload = malloc (m->length);
        memcpy (m->payload, buf+3, m->length);
    }

    return 0;
}

int ipc_message_format_write (const struct ipc_message *m, char **buf, ssize_t *msize)
{
    assert (m != NULL);
    assert (buf != NULL);
    assert (msize != NULL);
    assert (m->length <= BUFSIZ -3);

    if (m == NULL)
        return -1;

    if (buf == NULL)
        return -2;

    if (msize == NULL)
        return -3;

    if (*buf == NULL) {
        *buf = malloc (3 + m->length);
		memset (*buf, 0, 3 + m->length);
    }

    char *buffer = *buf;

    buffer[0] = m->type;
    memcpy (buffer + 1, &m->length, 2);
	if (m->payload != NULL) {
		memcpy (buffer + 3, m->payload, m->length);
	}

    *msize = 3 + m->length;

	printf ("sending msg: type %u, size %d, msize %ld\n", m->type, m->length, *msize);

    return 0;
}

// 1 on a recipient socket close
int ipc_message_read (int fd, struct ipc_message *m)
{
    assert (m != NULL);

    char *buf = NULL;
    ssize_t msize = BUFSIZ;

    int ret = usock_recv (fd, &buf, &msize);
    if (ret < 0) {
		// on error, buffer already freed
        handle_err ("msg_read", "usock_recv");
        return -1;
    }
	
	// closed recipient, buffer already freed
	if (ret == 1) {
		return 1;
	}

    if (ipc_message_format_read (m, buf, msize) < 0) {
        return -1;
    }
    free (buf);

    return 0;
}

int ipc_message_write (int fd, const struct ipc_message *m)
{
    assert (m != NULL);

    char *buf = NULL;
    ssize_t msize = 0;
    ipc_message_format_write (m, &buf, &msize);

	ssize_t nbytes_sent = 0;
    int ret = usock_send (fd, buf, msize, &nbytes_sent);
    if (ret < 0) {
		if (buf != NULL)
			free (buf);
        handle_err ("msg_write", "usock_send ret < 0");
        return -1;
    }

	// what was sent != what should have been sent
	if (nbytes_sent != msize) {
		if (buf != NULL)
			free (buf);
        handle_err ("msg_write", "usock_send did not send enough data");
        return -1;
	}

    if (buf != NULL)
        free (buf);

    return 0;
}

// MSG FORMAT

int ipc_message_format (struct ipc_message *m, char type, const char *payload, ssize_t length)
{
    assert (m != NULL);
    assert (length + 3 <= BUFSIZ);
    assert ((length == 0 && payload == NULL) || (length > 0 && payload != NULL));

    if (length + 3 > BUFSIZ) {
        handle_err ("msg_format_con", "msgsize > BUFSIZ");
        return -1;
    }

    m->type = type;
    m->length = (short) length;

    if (payload != NULL) {
		if (m->payload != NULL) {
			free (m->payload);
		}

        m->payload = malloc (length);
		if (m->payload == NULL) {
			return IPC_ERROR_NOT_ENOUGH_MEMORY;
		}
		memset (m->payload, 0, length);
	}

	if (payload != NULL) {
		memcpy (m->payload, payload, length);
	}
    return 0;
}

int ipc_message_format_data (struct ipc_message *m, const char *payload, ssize_t length)
{
    return ipc_message_format (m, MSG_TYPE_DATA, payload, length);
}

int ipc_message_format_server_close (struct ipc_message *m)
{
    return ipc_message_format (m, MSG_TYPE_SERVER_CLOSE, NULL, 0);
}

int ipc_message_empty (struct ipc_message *m)
{
    assert (m != NULL);

    if (m == NULL)
        return -1;

    if (m->payload != NULL)
        free (m->payload), m->payload = NULL;

    m->length = 0;

    return 0;
}