135749060SStanislav Fomichev // SPDX-License-Identifier: GPL-2.0 235749060SStanislav Fomichev 335749060SStanislav Fomichev #include <linux/stddef.h> 435749060SStanislav Fomichev #include <linux/bpf.h> 535749060SStanislav Fomichev #include <linux/in.h> 635749060SStanislav Fomichev #include <sys/socket.h> 735749060SStanislav Fomichev 835749060SStanislav Fomichev #include <bpf/bpf_helpers.h> 935749060SStanislav Fomichev #include <bpf/bpf_endian.h> 1035749060SStanislav Fomichev 11*4c3384d7SStanislav Fomichev #include <bpf_sockopt_helpers.h> 12*4c3384d7SStanislav Fomichev 1335749060SStanislav Fomichev #define SERV4_IP 0xc0a801feU /* 192.168.1.254 */ 1435749060SStanislav Fomichev #define SERV4_PORT 4040 1535749060SStanislav Fomichev 1635749060SStanislav Fomichev SEC("cgroup/recvmsg4") recvmsg4_prog(struct bpf_sock_addr * ctx)1735749060SStanislav Fomichevint recvmsg4_prog(struct bpf_sock_addr *ctx) 1835749060SStanislav Fomichev { 1935749060SStanislav Fomichev struct bpf_sock *sk; 2035749060SStanislav Fomichev 2135749060SStanislav Fomichev sk = ctx->sk; 2235749060SStanislav Fomichev if (!sk) 2335749060SStanislav Fomichev return 1; 2435749060SStanislav Fomichev 2535749060SStanislav Fomichev if (sk->family != AF_INET) 2635749060SStanislav Fomichev return 1; 2735749060SStanislav Fomichev 2835749060SStanislav Fomichev if (ctx->type != SOCK_STREAM && ctx->type != SOCK_DGRAM) 2935749060SStanislav Fomichev return 1; 3035749060SStanislav Fomichev 31*4c3384d7SStanislav Fomichev if (!get_set_sk_priority(ctx)) 32*4c3384d7SStanislav Fomichev return 1; 33*4c3384d7SStanislav Fomichev 3435749060SStanislav Fomichev ctx->user_ip4 = bpf_htonl(SERV4_IP); 3535749060SStanislav Fomichev ctx->user_port = bpf_htons(SERV4_PORT); 3635749060SStanislav Fomichev 3735749060SStanislav Fomichev return 1; 3835749060SStanislav Fomichev } 3935749060SStanislav Fomichev 4035749060SStanislav Fomichev char _license[] SEC("license") = "GPL"; 41