66 lines
1.9 KiB
C
66 lines
1.9 KiB
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;
|
||
|
|
}
|