Add packet IO module
* support marsio
* support dumpfile ( 1 thread read dumpfile & N thread handle packet)
This commit is contained in:
202
src/packet/packet_utils.cpp
Normal file
202
src/packet/packet_utils.cpp
Normal file
@@ -0,0 +1,202 @@
|
||||
#include "packet_utils.h"
|
||||
|
||||
void packet_set_io_ctx(struct packet *pkt, void *ctx)
|
||||
{
|
||||
pkt->io_ctx = ctx;
|
||||
}
|
||||
|
||||
void *packet_get_io_ctx(const struct packet *pkt)
|
||||
{
|
||||
return pkt->io_ctx;
|
||||
}
|
||||
|
||||
// return 0 success; return -1 failed
|
||||
int packet_set_sid(struct packet *pkt, uint16_t *sid, int num)
|
||||
{
|
||||
if (num > MAX_SID_NUM)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
pkt->sid_used = num;
|
||||
memcpy(pkt->sid_list, sid, num * sizeof(uint16_t));
|
||||
return 0;
|
||||
}
|
||||
|
||||
// return number of sid
|
||||
int packet_get_sid(const struct packet *pkt, uint16_t *sid, int size)
|
||||
{
|
||||
if (size < pkt->sid_used)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
memcpy(sid, pkt->sid_list, pkt->sid_used * sizeof(uint16_t));
|
||||
return pkt->sid_used;
|
||||
}
|
||||
|
||||
// return 0 success; return -1 failed
|
||||
int packet_prepend_sid(struct packet *pkt, uint16_t sid)
|
||||
{
|
||||
if (pkt->sid_used >= MAX_SID_NUM)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
memmove(pkt->sid_list + 1, pkt->sid_list, pkt->sid_used * sizeof(uint16_t));
|
||||
pkt->sid_list[0] = sid;
|
||||
pkt->sid_used++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// return 0 success; return -1 failed
|
||||
int packet_append_sid(struct packet *pkt, uint16_t sid)
|
||||
{
|
||||
if (pkt->sid_used >= MAX_SID_NUM)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
pkt->sid_list[pkt->sid_used] = sid;
|
||||
pkt->sid_used++;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// return 0 success; return -1 failed
|
||||
int packet_set_route_ctx(struct packet *pkt, const char *route, int len)
|
||||
{
|
||||
if (len > MAX_ROUTE_LEN)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
memcpy(pkt->route_ctx, route, len);
|
||||
pkt->route_len = len;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// return len of route ctx
|
||||
int packet_get_route_ctx(const struct packet *pkt, char *buff, int size)
|
||||
{
|
||||
if (pkt->route_len > size)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
memcpy(buff, pkt->route_ctx, pkt->route_len);
|
||||
return pkt->route_len;
|
||||
}
|
||||
|
||||
void packet_set_session_id(struct packet *pkt, uint64_t session_id)
|
||||
{
|
||||
pkt->session_id = session_id;
|
||||
}
|
||||
|
||||
uint64_t packet_get_session_id(const struct packet *pkt)
|
||||
{
|
||||
return pkt->session_id;
|
||||
}
|
||||
|
||||
void packet_set_direction(struct packet *pkt, uint8_t direction)
|
||||
{
|
||||
pkt->direction = direction;
|
||||
}
|
||||
|
||||
uint8_t packet_get_direction(const struct packet *pkt)
|
||||
{
|
||||
return pkt->direction;
|
||||
}
|
||||
|
||||
void packet_set_type(struct packet *pkt, uint8_t type)
|
||||
{
|
||||
pkt->type = type;
|
||||
}
|
||||
|
||||
uint8_t packet_get_type(const struct packet *pkt)
|
||||
{
|
||||
return pkt->type;
|
||||
}
|
||||
|
||||
void packet_set_action(struct packet *pkt, uint8_t action)
|
||||
{
|
||||
pkt->action = action;
|
||||
}
|
||||
|
||||
uint8_t packet_get_action(const struct packet *pkt)
|
||||
{
|
||||
return pkt->action;
|
||||
}
|
||||
|
||||
struct packet *packet_new(uint16_t len)
|
||||
{
|
||||
struct packet *pkt = (struct packet *)calloc(1, sizeof(struct packet) + len);
|
||||
if (pkt == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
pkt->data_len = len;
|
||||
pkt->data_ptr = (const char *)pkt + sizeof(struct packet);
|
||||
pkt->need_free = true;
|
||||
|
||||
return pkt;
|
||||
}
|
||||
|
||||
void packet_free(struct packet *pkt)
|
||||
{
|
||||
if (pkt && pkt->need_free)
|
||||
{
|
||||
free(pkt);
|
||||
pkt = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
struct packet *packet_dup(const struct packet *pkt)
|
||||
{
|
||||
if (pkt == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
struct packet *pkt_dup = packet_new(pkt->data_len);
|
||||
if (pkt_dup == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
memcpy(pkt_dup, pkt, sizeof(struct packet));
|
||||
memcpy((char *)pkt_dup->data_ptr, pkt->data_ptr, pkt->data_len);
|
||||
pkt_dup->need_free = true;
|
||||
|
||||
// update layers
|
||||
for (int8_t i = 0; i < pkt->layers_used; i++)
|
||||
{
|
||||
pkt_dup->layers[i].hdr_ptr = pkt_dup->data_ptr + pkt->layers[i].hdr_offset;
|
||||
pkt_dup->layers[i].pld_ptr = pkt_dup->data_ptr + pkt->layers[i].hdr_offset + pkt->layers[i].hdr_len;
|
||||
}
|
||||
|
||||
// update frag_layer
|
||||
if (pkt->frag_layer)
|
||||
{
|
||||
pkt_dup->frag_layer = &pkt_dup->layers[pkt->frag_layer - pkt->layers];
|
||||
}
|
||||
|
||||
return pkt_dup;
|
||||
}
|
||||
|
||||
const char *packet_get_data(const struct packet *pkt)
|
||||
{
|
||||
return pkt->data_ptr;
|
||||
}
|
||||
|
||||
uint16_t packet_get_len(const struct packet *pkt)
|
||||
{
|
||||
return pkt->data_len;
|
||||
}
|
||||
|
||||
bool paket_is_fragment(const struct packet *pkt)
|
||||
{
|
||||
if (pkt->frag_layer)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user