103 	struct bpf_sock *sk;  in redir_port()  local
109 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in redir_port()
110 	if (!sk)  in redir_port()
113 	err = bpf_sk_assign(ctx, sk, 0);  in redir_port()
114 	bpf_sk_release(sk);  in redir_port()
122 	struct bpf_sock *sk;  in redir_ip4()  local
132 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in redir_ip4()
133 	if (!sk)  in redir_ip4()
136 	err = bpf_sk_assign(ctx, sk, 0);  in redir_ip4()
137 	bpf_sk_release(sk);  in redir_ip4()
145 	struct bpf_sock *sk;  in redir_ip6()  local
158 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in redir_ip6()
159 	if (!sk)  in redir_ip6()
162 	err = bpf_sk_assign(ctx, sk, 0);  in redir_ip6()
163 	bpf_sk_release(sk);  in redir_ip6()
170 	struct bpf_sock *sk;  in select_sock_a()  local
173 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in select_sock_a()
174 	if (!sk)  in select_sock_a()
177 	err = bpf_sk_assign(ctx, sk, 0);  in select_sock_a()
178 	bpf_sk_release(sk);  in select_sock_a()
185 	struct bpf_sock *sk;  in select_sock_a_no_reuseport()  local
188 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in select_sock_a_no_reuseport()
189 	if (!sk)  in select_sock_a_no_reuseport()
192 	err = bpf_sk_assign(ctx, sk, BPF_SK_LOOKUP_F_NO_REUSEPORT);  in select_sock_a_no_reuseport()
193 	bpf_sk_release(sk);  in select_sock_a_no_reuseport()
211 	struct bpf_sock *sk;  in sk_assign_eexist()  local
215 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_B);  in sk_assign_eexist()
216 	if (!sk)  in sk_assign_eexist()
218 	err = bpf_sk_assign(ctx, sk, 0);  in sk_assign_eexist()
221 	bpf_sk_release(sk);  in sk_assign_eexist()
223 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in sk_assign_eexist()
224 	if (!sk)  in sk_assign_eexist()
226 	err = bpf_sk_assign(ctx, sk, 0);  in sk_assign_eexist()
235 	if (sk)  in sk_assign_eexist()
236 		bpf_sk_release(sk);  in sk_assign_eexist()
244 	struct bpf_sock *sk;  in sk_assign_replace_flag()  local
248 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in sk_assign_replace_flag()
249 	if (!sk)  in sk_assign_replace_flag()
251 	err = bpf_sk_assign(ctx, sk, 0);  in sk_assign_replace_flag()
254 	bpf_sk_release(sk);  in sk_assign_replace_flag()
256 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_B);  in sk_assign_replace_flag()
257 	if (!sk)  in sk_assign_replace_flag()
259 	err = bpf_sk_assign(ctx, sk, BPF_SK_LOOKUP_F_REPLACE);  in sk_assign_replace_flag()
267 	if (sk)  in sk_assign_replace_flag()
268 		bpf_sk_release(sk);  in sk_assign_replace_flag()
276 	struct bpf_sock *sk = NULL;  in sk_assign_null()  local
287 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_B);  in sk_assign_null()
288 	if (!sk)  in sk_assign_null()
290 	err = bpf_sk_assign(ctx, sk, BPF_SK_LOOKUP_F_REPLACE);  in sk_assign_null()
296 	if (ctx->sk != sk)  in sk_assign_null()
304 	err = bpf_sk_assign(ctx, sk, BPF_SK_LOOKUP_F_REPLACE);  in sk_assign_null()
310 	if (sk)  in sk_assign_null()
311 		bpf_sk_release(sk);  in sk_assign_null()
325 	if (ctx->sk && ctx->sk->family != AF_INET)  in access_ctx_sk()
335 	if (ctx->sk != sk1)  in access_ctx_sk()
339 	if (ctx->sk->family != AF_INET ||  in access_ctx_sk()
340 	    ctx->sk->type != SOCK_STREAM ||  in access_ctx_sk()
341 	    ctx->sk->state != BPF_TCP_LISTEN)  in access_ctx_sk()
348 	if (ctx->sk)  in access_ctx_sk()
358 	if (ctx->sk != sk2)  in access_ctx_sk()
362 	if (ctx->sk->family != AF_INET ||  in access_ctx_sk()
363 	    ctx->sk->type != SOCK_STREAM ||  in access_ctx_sk()
364 	    ctx->sk->state != BPF_TCP_LISTEN)  in access_ctx_sk()
385 	struct bpf_sock *sk;  in ctx_narrow_access()  local
546 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_B);  in ctx_narrow_access()
547 	if (sk) {  in ctx_narrow_access()
548 		bpf_sk_assign(ctx, sk, 0);  in ctx_narrow_access()
549 		bpf_sk_release(sk);  in ctx_narrow_access()
558 	struct bpf_sock *sk;  in sk_assign_esocknosupport()  local
562 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in sk_assign_esocknosupport()
563 	if (!sk)  in sk_assign_esocknosupport()
566 	err = bpf_sk_assign(ctx, sk, 0);  in sk_assign_esocknosupport()
575 	if (sk)  in sk_assign_esocknosupport()
576 		bpf_sk_release(sk);  in sk_assign_esocknosupport()
610 	struct bpf_sock *sk;  in select_server_a()  local
613 	sk = bpf_map_lookup_elem(&redir_map, &KEY_SERVER_A);  in select_server_a()
614 	if (!sk)  in select_server_a()
617 	err = bpf_sk_assign(ctx, sk, 0);  in select_server_a()
618 	bpf_sk_release(sk);  in select_server_a()