162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0 262306a36Sopenharmony_ci 362306a36Sopenharmony_ci#include <linux/stddef.h> 462306a36Sopenharmony_ci#include <linux/bpf.h> 562306a36Sopenharmony_ci#include <linux/in.h> 662306a36Sopenharmony_ci#include <sys/socket.h> 762306a36Sopenharmony_ci 862306a36Sopenharmony_ci#include <bpf/bpf_helpers.h> 962306a36Sopenharmony_ci#include <bpf/bpf_endian.h> 1062306a36Sopenharmony_ci 1162306a36Sopenharmony_ci#include <bpf_sockopt_helpers.h> 1262306a36Sopenharmony_ci 1362306a36Sopenharmony_ci#define SERV4_IP 0xc0a801feU /* 192.168.1.254 */ 1462306a36Sopenharmony_ci#define SERV4_PORT 4040 1562306a36Sopenharmony_ci 1662306a36Sopenharmony_ciSEC("cgroup/recvmsg4") 1762306a36Sopenharmony_ciint recvmsg4_prog(struct bpf_sock_addr *ctx) 1862306a36Sopenharmony_ci{ 1962306a36Sopenharmony_ci struct bpf_sock *sk; 2062306a36Sopenharmony_ci 2162306a36Sopenharmony_ci sk = ctx->sk; 2262306a36Sopenharmony_ci if (!sk) 2362306a36Sopenharmony_ci return 1; 2462306a36Sopenharmony_ci 2562306a36Sopenharmony_ci if (sk->family != AF_INET) 2662306a36Sopenharmony_ci return 1; 2762306a36Sopenharmony_ci 2862306a36Sopenharmony_ci if (ctx->type != SOCK_STREAM && ctx->type != SOCK_DGRAM) 2962306a36Sopenharmony_ci return 1; 3062306a36Sopenharmony_ci 3162306a36Sopenharmony_ci if (!get_set_sk_priority(ctx)) 3262306a36Sopenharmony_ci return 1; 3362306a36Sopenharmony_ci 3462306a36Sopenharmony_ci ctx->user_ip4 = bpf_htonl(SERV4_IP); 3562306a36Sopenharmony_ci ctx->user_port = bpf_htons(SERV4_PORT); 3662306a36Sopenharmony_ci 3762306a36Sopenharmony_ci return 1; 3862306a36Sopenharmony_ci} 3962306a36Sopenharmony_ci 4062306a36Sopenharmony_cichar _license[] SEC("license") = "GPL"; 41