libhv/examples/protorpc/protorpc.c

#include "protorpc.h"

#include <string.h> // import memcpy

int protorpc_pack(const protorpc_message* msg, void* buf, int len) {
    if (!msg || !buf || !len) return -1;
    const protorpc_head* head = &(msg->head);
    unsigned int packlen = protorpc_package_length(head);
    // Check is buffer enough
    if (len < packlen) {
        return -2;
    }
    unsigned char* p = (unsigned char*)buf;
    *p++ = head->protocol[0];
    *p++ = head->protocol[1];
    *p++ = head->protocol[2];
    *p++ = head->protocol[3];
    *p++ = head->version;
    *p++ = head->flags;
    *p++ = head->reserved[0];
    *p++ = head->reserved[1];
    // hton length
    unsigned int length = head->length;
    *p++ = (length >> 24) & 0xFF;
    *p++ = (length >> 16) & 0xFF;
    *p++ = (length >>  8) & 0xFF;
    *p++ =  length        & 0xFF;
    // memcpy body
    if (msg->body && head->length) {
        memcpy(p, msg->body, head->length);
    }

    return packlen;
}

int protorpc_unpack(protorpc_message* msg, const void* buf, int len) {
    if (!msg || !buf || !len) return -1;
    if (len < PROTORPC_HEAD_LENGTH) return -2;
    protorpc_head* head = &(msg->head);
    const unsigned char* p = (const unsigned char*)buf;
    head->protocol[0] = *p++;
    head->protocol[1] = *p++;
    head->protocol[2] = *p++;
    head->protocol[3] = *p++;
    head->version = *p++;
    head->flags = *p++;
    head->reserved[0] = *p++;
    head->reserved[1] = *p++;
    // ntoh length
    head->length  = ((unsigned int)*p++) << 24;
    head->length |= ((unsigned int)*p++) << 16;
    head->length |= ((unsigned int)*p++) << 8;
    head->length |= *p++;
    // Check is buffer enough
    unsigned int packlen = protorpc_package_length(head);
    if (len < packlen) {
        return -3;
    }
    // NOTE: just shadow copy
    if (len > PROTORPC_HEAD_LENGTH) {
        msg->body = (const char*)buf + PROTORPC_HEAD_LENGTH;
    }

    return packlen;
}