Enhancement: Improve error handling in stellar's packet API by checking return values of metadata functions and adding descriptive error logging

This commit is contained in:
luwenpeng
2024-04-22 14:23:45 +08:00
parent bb7d7410c9
commit dd32f0f231
3 changed files with 61 additions and 14 deletions

View File

@@ -26,6 +26,11 @@
#ifndef TIMEOUT_H #ifndef TIMEOUT_H
#define TIMEOUT_H #define TIMEOUT_H
#ifdef __cplusplus
extern "C"
{
#endif
#include <stdbool.h> /* bool */ #include <stdbool.h> /* bool */
#include <stdio.h> /* FILE */ #include <stdio.h> /* FILE */
@@ -259,4 +264,8 @@ TIMEOUT_PUBLIC struct timeout *timeouts_next(struct timeouts *, struct timeouts_
#define timeouts_addf(T, to, timeout) \ #define timeouts_addf(T, to, timeout) \
timeouts_add((T), (to), timeouts_f2i((T), (timeout))) timeouts_add((T), (to), timeouts_f2i((T), (timeout)))
#ifdef __cplusplus
}
#endif
#endif /* TIMEOUT_H */ #endif /* TIMEOUT_H */

View File

@@ -1,13 +1,13 @@
#ifndef _GTEST_UTILS_H #ifndef _GTEST_UTILS_H
#define _GTEST_UTILS_H #define _GTEST_UTILS_H
#include <gtest/gtest.h>
#ifdef __cplusplus #ifdef __cplusplus
extern "C" extern "C"
{ {
#endif #endif
#include <gtest/gtest.h>
#include "udp_utils.h" #include "udp_utils.h"
#include "tcp_utils.h" #include "tcp_utils.h"
#include "ipv4_utils.h" #include "ipv4_utils.h"

View File

@@ -176,7 +176,10 @@ void packet_set_direction(struct packet *pkt, int dir)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_set_metadata(mbuff, MR_BUFF_DIR, &dir, sizeof(dir)); if (marsio_buff_set_metadata(mbuff, MR_BUFF_DIR, &dir, sizeof(dir)) != 0)
{
PACKET_LOG_ERROR("failed to set direction");
}
} }
} }
@@ -188,7 +191,10 @@ int packet_get_direction(const struct packet *pkt)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_get_metadata(mbuff, MR_BUFF_DIR, &direction, sizeof(direction)); if (marsio_buff_get_metadata(mbuff, MR_BUFF_DIR, &direction, sizeof(direction)) <= 0)
{
PACKET_LOG_ERROR("failed to get direction");
}
} }
return direction; return direction;
@@ -200,7 +206,10 @@ void packet_set_session_id(struct packet *pkt, uint64_t sess_id)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_set_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)); if (marsio_buff_set_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)) != 0)
{
PACKET_LOG_ERROR("failed to set session id");
}
} }
} }
@@ -211,7 +220,10 @@ uint64_t packet_get_session_id(const struct packet *pkt)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_get_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)); if (marsio_buff_get_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)) <= 0)
{
PACKET_LOG_ERROR("failed to get session id");
}
} }
return sess_id; return sess_id;
@@ -223,8 +235,13 @@ void packet_set_domain(struct packet *pkt, uint64_t domain)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
// TODO // TODO
// marsio_buff_set_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)); #if 0
if (marsio_buff_set_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)) != 0)
{
PACKET_LOG_ERROR("failed to set domain");
}
#endif
} }
} }
@@ -235,8 +252,13 @@ uint64_t packet_get_domain(const struct packet *pkt)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
// TODO // TODO
// marsio_buff_get_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)); #if 0
if (marsio_buff_get_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)) <= 0)
{
PACKET_LOG_ERROR("failed to get domain");
}
#endif
} }
return domain; return domain;
@@ -248,7 +270,10 @@ void packet_set_route_ctx(struct packet *pkt, const char *route, int len)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_set_metadata(mbuff, MR_BUFF_ROUTE_CTX, (void *)route, len); if (marsio_buff_set_metadata(mbuff, MR_BUFF_ROUTE_CTX, (void *)route, len) != 0)
{
PACKET_LOG_ERROR("failed to set route ctx");
}
} }
} }
@@ -261,6 +286,10 @@ int packet_get_route_ctx(const struct packet *pkt, char *buff, int size)
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
len = marsio_buff_get_metadata(mbuff, MR_BUFF_ROUTE_CTX, buff, size); len = marsio_buff_get_metadata(mbuff, MR_BUFF_ROUTE_CTX, buff, size);
if (len <= 0)
{
PACKET_LOG_ERROR("failed to get route ctx");
}
} }
return len; return len;
@@ -272,7 +301,10 @@ void packet_set_sid_list(struct packet *pkt, uint16_t *sid, int num)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_set_sid_list(mbuff, sid, num); if (marsio_buff_set_sid_list(mbuff, sid, num) != 0)
{
PACKET_LOG_ERROR("failed to set sid list");
}
} }
} }
@@ -295,7 +327,10 @@ void packet_prepend_sid_list(struct packet *pkt, uint16_t *sid, int num)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_prepend_sid_list(mbuff, sid, num); if (marsio_buff_prepend_sid_list(mbuff, sid, num) != 0)
{
PACKET_LOG_ERROR("failed to prepend sid list");
}
} }
} }
@@ -305,6 +340,9 @@ void packet_append_sid_list(struct packet *pkt, uint16_t *sid, int num)
{ {
marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt); marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
assert(mbuff != NULL); assert(mbuff != NULL);
marsio_buff_append_sid_list(mbuff, sid, num); if (marsio_buff_append_sid_list(mbuff, sid, num) != 0)
{
PACKET_LOG_ERROR("failed to append sid list");
}
} }
} }