This repository has been archived on 2025-09-14. You can view files and clone it, but cannot push or open issues or pull requests.
Files
stellar-stellar/infra/packet_io/packet_io.h
2024-11-01 15:34:07 +08:00

90 lines
2.8 KiB
C

#pragma once
#ifdef __cplusplus
extern "C"
{
#endif
#include <stdint.h>
#include "log_internal.h"
#include "stellar/packet.h"
#define PACKET_IO_LOG_INFO(format, ...) STELLAR_LOG_INFO(__thread_local_logger, "packet IO", format, ##__VA_ARGS__)
#define PACKET_IO_LOG_ERROR(format, ...) STELLAR_LOG_ERROR(__thread_local_logger, "packet IO", format, ##__VA_ARGS__)
#define PACKET_IO_LOG_FATAL(format, ...) STELLAR_LOG_FATAL(__thread_local_logger, "packet IO", format, ##__VA_ARGS__)
struct packet_io_stat
{
// device packet
uint64_t pkts_rx;
uint64_t bytes_rx;
uint64_t pkts_tx;
uint64_t bytes_tx;
// keep-alive packet
uint64_t keep_alive_pkts;
uint64_t keep_alive_bytes;
// drop packet
uint64_t pkts_dropped;
uint64_t bytes_dropped;
// inject packet
uint64_t pkts_injected;
uint64_t bytes_injected;
// user freed
uint64_t pkts_user_freed;
uint64_t bytes_user_freed;
} __attribute__((aligned(64)));
#define PKT_IO_STAT_MAP(XX) \
XX(PKT_IO_STAT_PKTS_RX, pkts_rx) \
XX(PKT_IO_STAT_BYTES_RX, bytes_rx) \
XX(PKT_IO_STAT_PKTS_TX, pkts_tx) \
XX(PKT_IO_STAT_BYTES_TX, bytes_tx) \
XX(PKT_IO_STAT_KEEP_ALIVE_PKTS, keep_alive_pkts) \
XX(PKT_IO_STAT_KEEP_ALIVE_BYTES, keep_alive_bytes) \
XX(PKT_IO_STAT_PKTS_DROPPED, pkts_dropped) \
XX(PKT_IO_STAT_BYTES_DROPPED, bytes_dropped) \
XX(PKT_IO_STAT_PKTS_INJECTED, pkts_injected) \
XX(PKT_IO_STAT_BYTES_INJECTED, bytes_injected) \
XX(PKT_IO_STAT_PKTS_USER_FREED, pkts_user_freed) \
XX(PKT_IO_STAT_BYTES_USER_FREED, bytes_user_freed)
enum pkt_io_stat_type
{
#define XX(type, name) type,
PKT_IO_STAT_MAP(XX)
#undef XX
PKT_IO_STAT_MAX
};
__attribute__((unused)) static const char pkt_io_stat_str[PKT_IO_STAT_MAX][64] =
{
#define XX(type, name) #name,
PKT_IO_STAT_MAP(XX)
#undef XX
};
struct packet_io;
struct packet_io *packet_io_new(const char *toml_file);
void packet_io_free(struct packet_io *pkt_io);
// only used in pcap mode, to check if all pcap has been processed
int packet_io_is_done(struct packet_io *pkt_io);
int packet_io_init(struct packet_io *pkt_io, uint16_t thr_idx);
int packet_io_recv(struct packet_io *pkt_io, uint16_t thr_idx, struct packet *pkts[], int nr_pkts);
void packet_io_send(struct packet_io *pkt_io, uint16_t thr_idx, struct packet *pkts[], int nr_pkts);
void packet_io_drop(struct packet_io *pkt_io, uint16_t thr_idx, struct packet *pkts[], int nr_pkts);
void packet_io_yield(struct packet_io *pkt_io, uint16_t thr_idx);
void packet_io_clean(struct packet_io *pkt_io, uint16_t thr_idx);
uint64_t packet_io_stat_get(struct packet_io_stat *stat, enum pkt_io_stat_type type);
#ifdef __cplusplus
}
#endif