simplify unpacking process
parent
5b2db7cd78
commit
a4ce84c5f9
|
@ -42,13 +42,13 @@ int32_t pkt_header_decode(pkt_header *table, void *data, size_t datalen) {
|
||||||
|
|
||||||
int32_t pkt_unpack_struct(cw_unpack_context *uc, pkt_desc *desc, void *raw_blob, uint32_t blob_size) {
|
int32_t pkt_unpack_struct(cw_unpack_context *uc, pkt_desc *desc, void *raw_blob, uint32_t blob_size) {
|
||||||
uint8_t *blob = (uint8_t*)raw_blob;
|
uint8_t *blob = (uint8_t*)raw_blob;
|
||||||
for (pkt_desc *field = desc; field->type != 0; ++field) {
|
for (pkt_desc *field = desc; field->type != CWP_NOT_AN_ITEM; ++field) {
|
||||||
cw_unpack_next(uc);
|
cw_unpack_next(uc);
|
||||||
if (uc->item.type != field->type) return -1; // unexpected field
|
if (uc->item.type != field->type) return -1; // unexpected field
|
||||||
if (blob + field->offset + field->size >= blob + blob_size) return -1; // field does not fit
|
if (blob + field->offset + field->size >= blob + blob_size) return -1; // field does not fit
|
||||||
switch (field->type) {
|
switch (field->type) {
|
||||||
case CWP_ITEM_POSITIVE_INTEGER: {
|
case CWP_ITEM_POSITIVE_INTEGER: {
|
||||||
zpl_memcopy(blob + field->offset, (uint8_t*)uc->item.as.u64, field->size);
|
zpl_memcopy(blob + field->offset, (uint8_t*)&uc->item.as.u64, field->size);
|
||||||
}break;
|
}break;
|
||||||
default: {
|
default: {
|
||||||
zpl_printf("[WARN] unsupported pkt field type %lld !\n", field->type);
|
zpl_printf("[WARN] unsupported pkt field type %lld !\n", field->type);
|
||||||
|
|
|
@ -25,14 +25,9 @@ typedef struct {
|
||||||
pkt_handler_proc *handler;
|
pkt_handler_proc *handler;
|
||||||
} pkt_handler;
|
} pkt_handler;
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
uint32_t chunk_size;
|
|
||||||
uint32_t chunk_amount;
|
|
||||||
} pkt_01_welcome;
|
|
||||||
|
|
||||||
int32_t pkt_header_decode(pkt_header *table, void *data, size_t datalen);
|
int32_t pkt_header_decode(pkt_header *table, void *data, size_t datalen);
|
||||||
|
|
||||||
PKT_HANDLER_PROC(pkt_01_welcome_handler);
|
|
||||||
|
|
||||||
extern pkt_handler pkt_handlers[];
|
extern pkt_handler pkt_handlers[];
|
||||||
extern uint8_t pkt_buffer[];
|
extern uint8_t pkt_buffer[];
|
||||||
|
|
||||||
|
#include "packet_list.h"
|
|
@ -0,0 +1,13 @@
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
// NOTE(zaklaus): pkt data
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
uint32_t chunk_size;
|
||||||
|
uint32_t chunk_amount;
|
||||||
|
} pkt_01_welcome;
|
||||||
|
|
||||||
|
// NOTE(zaklaus): pkt handlers
|
||||||
|
|
||||||
|
PKT_HANDLER_PROC(pkt_01_welcome_handler);
|
||||||
|
|
|
@ -31,6 +31,10 @@ inline int32_t pkt_validate_eof_msg(cw_unpack_context *uc) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
inline size_t pkt_pack_msg_size(cw_pack_context *pc) {
|
||||||
|
return pc->current - pc->start; // NOTE(zaklaus): length
|
||||||
|
}
|
||||||
|
|
||||||
#ifndef PKT_OFFSETOF
|
#ifndef PKT_OFFSETOF
|
||||||
#if defined(_MSC_VER) || defined(ZPL_COMPILER_TINYC)
|
#if defined(_MSC_VER) || defined(ZPL_COMPILER_TINYC)
|
||||||
# define PKT_OFFSETOF(Type, element) ((size_t) & (((Type *)0)->element))
|
# define PKT_OFFSETOF(Type, element) ((size_t) & (((Type *)0)->element))
|
||||||
|
@ -63,3 +67,11 @@ typedef struct pkt_desc {
|
||||||
} pkt_desc;
|
} pkt_desc;
|
||||||
|
|
||||||
int32_t pkt_unpack_struct(cw_unpack_context *uc, pkt_desc *desc, void *raw_blob, uint32_t blob_size);
|
int32_t pkt_unpack_struct(cw_unpack_context *uc, pkt_desc *desc, void *raw_blob, uint32_t blob_size);
|
||||||
|
|
||||||
|
inline int32_t pkt_msg_decode(pkt_header *header, pkt_desc* desc, uint32_t args, void *raw_blob, uint32_t blob_size) {
|
||||||
|
cw_unpack_context uc = {0};
|
||||||
|
pkt_unpack_msg(&uc, header, args);
|
||||||
|
pkt_unpack_struct(&uc, desc, header, raw_blob, blob_size);
|
||||||
|
|
||||||
|
return pkt_validate_eof_msg(&uc);
|
||||||
|
}
|
|
@ -1,9 +1,6 @@
|
||||||
#include "packet.h"
|
#include "packet.h"
|
||||||
#include "cwpack/cwpack.h"
|
|
||||||
#include "packet_utils.h"
|
#include "packet_utils.h"
|
||||||
|
|
||||||
#define PKT_01_WELCOME_ARGS 2
|
|
||||||
|
|
||||||
#ifdef SERVER
|
#ifdef SERVER
|
||||||
size_t pkt_01_welcome_encode(pkt_01_welcome *table) {
|
size_t pkt_01_welcome_encode(pkt_01_welcome *table) {
|
||||||
cw_pack_context pc = {0};
|
cw_pack_context pc = {0};
|
||||||
|
@ -12,7 +9,7 @@ size_t pkt_01_welcome_encode(pkt_01_welcome *table) {
|
||||||
cw_pack_unsigned(&pc, world_chunk_size());
|
cw_pack_unsigned(&pc, world_chunk_size());
|
||||||
cw_pack_unsigned(&pc, world_chunk_amount());
|
cw_pack_unsigned(&pc, world_chunk_amount());
|
||||||
|
|
||||||
return pc.current - pc.start; /* length */
|
return pkt_pack_msg_size(&pc);
|
||||||
}
|
}
|
||||||
#else
|
#else
|
||||||
pkt_desc pkt_01_welcome_desc[3] = {
|
pkt_desc pkt_01_welcome_desc[3] = {
|
||||||
|
@ -20,17 +17,10 @@ pkt_desc pkt_01_welcome_desc[3] = {
|
||||||
{ PKT_FIELD(CWP_ITEM_POSITIVE_INTEGER, pkt_01_welcome, chunk_amount) },
|
{ PKT_FIELD(CWP_ITEM_POSITIVE_INTEGER, pkt_01_welcome, chunk_amount) },
|
||||||
{ PKT_END },
|
{ PKT_END },
|
||||||
};
|
};
|
||||||
int32_t pkt_01_welcome_decode(pkt_01_welcome *table, pkt_header *header) {
|
|
||||||
cw_unpack_context uc = {0};
|
|
||||||
pkt_unpack_msg(&uc, header, PKT_01_WELCOME_ARGS);
|
|
||||||
pkt_unpack_struct(&uc, header, PKT_STRUCT_PTR(table));
|
|
||||||
|
|
||||||
return pkt_validate_eof_msg(&uc);
|
|
||||||
}
|
|
||||||
|
|
||||||
int32_t pkt_01_welcome_handler(pkt_header *header) {
|
int32_t pkt_01_welcome_handler(pkt_header *header) {
|
||||||
pkt_01_welcome table;
|
pkt_01_welcome table;
|
||||||
PKT_IF(pkt_01_welcome_decode(&table, header));
|
PKT_IF(pkt_msg_decode(header, pkt_01_welcome_desc, 2, PKT_STRUCT_PTR(&table)));
|
||||||
|
|
||||||
zpl_printf("we received: chunk_size: %d and chunk_amount: %d\n", table.chunk_size, table.chunk_amount);
|
zpl_printf("we received: chunk_size: %d and chunk_amount: %d\n", table.chunk_size, table.chunk_amount);
|
||||||
return 0;
|
return 0;
|
||||||
|
|
Loading…
Reference in New Issue