IntenseWebs/LibreQoS
3
0
Fork 0
mirror of https://github.com/LibreQoE/LibreQoS.git synced 2025-02-25 18:55:32 -06:00
Code Issues Packages Projects Releases Wiki Activity

WIP

Browse Source
This commit is contained in:
Herbert Wolverson
2024-02-08 08:32:58 -06:00
parent f7c9e82173
commit d412851560
3 changed files with 11 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/rust/lqos_sys/src/bpf/common/lpm.h
View File

@@ -48,7 +48,7 @@ struct {
} map_ip_to_cpu_and_tc_recip SEC(".maps");
// Determine the effective direction of a packet
static __always_inline int determine_effective_direction(int direction, __be16 internet_vlan, struct dissector_t * dissector) {
static __always_inline u_int8_t determine_effective_direction(int direction, __be16 internet_vlan, struct dissector_t * dissector) {
if (direction < 3) {
return direction;
} else {
@@ -64,7 +64,7 @@ static __always_inline int determine_effective_direction(int direction, __be16 i
// into account redirection and "on a stick" setup.
static __always_inline struct ip_hash_info * setup_lookup_key_and_tc_cpu(
// This must have been pre-calculated by `determine_effective_direction`.
int direction,
u_int8_t direction,
// Pointer to the "lookup key", which should contain the IP address
// to search for. Prefix length will be set for you.
struct ip_hash_key * lookup_key,

52
src/rust/lqos_sys/src/bpf/lqos_kern.c
View File

@@ -18,6 +18,7 @@
#include "common/tcp_rtt.h"
#include "common/bifrost.h"
#include "common/heimdall.h"
#include "common/flows.h"
//#define VERBOSE 1
@@ -98,56 +99,22 @@ int xdp_prog(struct xdp_md *ctx)
// is requested.
if (!dissector_find_l3_offset(&dissector, vlan_redirect)) return XDP_PASS;
if (!dissector_find_ip_header(&dissector)) return XDP_PASS;
int effective_direction = determine_effective_direction(
u_int8_t effective_direction = determine_effective_direction(
direction,
internet_vlan,
&dissector
);
// Per-Flow RTT Tracking
//bpf_debug("Checking for TCP");
struct tcphdr * tcp = get_tcp_header(&dissector);
if (tcp != NULL) {
if (tcp + 1 < dissector.end)
{
//bpf_debug("TCP found");
if (tcp->syn) {
// We've found a SYN packet, so the connection is just starting.
if (effective_direction == 1) {
bpf_debug("SYN->WAN, %d <-> %d", tcp->seq, tcp->ack_seq);
} else {
bpf_debug("SYN->LAN, %d <-> %d", tcp->seq, tcp->ack_seq);
}
} else if (tcp->fin) {
// We've found a FIN packet, so the connection is expecting to end.
bpf_debug("FIN packet, %d", effective_direction);
} else if (tcp->rst) {
// We've found a RST packet, so the connection is being reset.
bpf_debug("RST packet, %d", effective_direction);
} else if (tcp->ack) {
// We've found an ACK packet, so the connection is established.
void *nh_pos = (tcp + 1) + (tcp->doff << 2);
bool is_valid = nh_pos - dissector.start < ctx->data_end - ctx->data;
if (is_valid) {
//bpf_debug("ACK packet");
if (effective_direction == 1) {
// To the internet
bpf_debug("ACK->WAN, %d <-> %d", tcp->seq, tcp->ack_seq);
} else {
// To the LAN
bpf_debug("ACK->LAN, %d <-> %d", tcp->seq, tcp->ack_seq);
}
}
}
}
}
#ifdef VERBOSE
bpf_debug("(XDP) Effective direction: %d", effective_direction);
#endif
#ifdef VERBOSE
bpf_debug("(XDP) Spotted VLAN: %u", dissector.current_vlan);
#endif
// Per-Flow RTT Tracking
track_flows(&dissector, effective_direction);
// Determine the lookup key by direction
struct ip_hash_key lookup_key;
struct ip_hash_info * ip_info = setup_lookup_key_and_tc_cpu(
@@ -155,9 +122,6 @@ int xdp_prog(struct xdp_md *ctx)
&lookup_key,
&dissector
);
#ifdef VERBOSE
bpf_debug("(XDP) Effective direction: %d", effective_direction);
#endif
// Find the desired TC handle and CPU target
__u32 tc_handle = 0;

1
src/rust/remove_pinned_maps.sh
View File

@@ -13,3 +13,4 @@ rm -v /sys/fs/bpf/bifrost_vlan_map
rm -v /sys/fs/bpf/heimdall
rm -v /sys/fs/bpf/heimdall_config
rm -v /sys/fs/bpf/heimdall_watching
rm -v /sys/fs/bpf/flowbee