mirror of
https://github.com/Fishwaldo/Star64_linux.git
synced 2025-06-25 16:11:45 +00:00
bpf: sockmap: Only check ULP for TCP sockets
The sock map code checks that a socket does not have an active upper layer protocol before inserting it into the map. This requires casting via inet_csk, which isn't valid for UDP sockets. Guard checks for ULP by checking inet_sk(sk)->is_icsk first. Signed-off-by: Lorenz Bauer <lmb@cloudflare.com> Signed-off-by: Daniel Borkmann <daniel@iogearbox.net> Reviewed-by: Jakub Sitnicki <jakub@cloudflare.com> Acked-by: John Fastabend <john.fastabend@gmail.com> Link: https://lore.kernel.org/bpf/20200309111243.6982-2-lmb@cloudflare.com
This commit is contained in:
parent
3e7c67d90e
commit
7b70973d7e
4 changed files with 15 additions and 12 deletions
|
@ -384,7 +384,6 @@ static int sock_map_update_common(struct bpf_map *map, u32 idx,
|
|||
struct sock *sk, u64 flags)
|
||||
{
|
||||
struct bpf_stab *stab = container_of(map, struct bpf_stab, map);
|
||||
struct inet_connection_sock *icsk = inet_csk(sk);
|
||||
struct sk_psock_link *link;
|
||||
struct sk_psock *psock;
|
||||
struct sock *osk;
|
||||
|
@ -395,7 +394,7 @@ static int sock_map_update_common(struct bpf_map *map, u32 idx,
|
|||
return -EINVAL;
|
||||
if (unlikely(idx >= map->max_entries))
|
||||
return -E2BIG;
|
||||
if (unlikely(rcu_access_pointer(icsk->icsk_ulp_data)))
|
||||
if (inet_csk_has_ulp(sk))
|
||||
return -EINVAL;
|
||||
|
||||
link = sk_psock_init_link();
|
||||
|
@ -738,7 +737,6 @@ static int sock_hash_update_common(struct bpf_map *map, void *key,
|
|||
struct sock *sk, u64 flags)
|
||||
{
|
||||
struct bpf_htab *htab = container_of(map, struct bpf_htab, map);
|
||||
struct inet_connection_sock *icsk = inet_csk(sk);
|
||||
u32 key_size = map->key_size, hash;
|
||||
struct bpf_htab_elem *elem, *elem_new;
|
||||
struct bpf_htab_bucket *bucket;
|
||||
|
@ -749,7 +747,7 @@ static int sock_hash_update_common(struct bpf_map *map, void *key,
|
|||
WARN_ON_ONCE(!rcu_read_lock_held());
|
||||
if (unlikely(flags > BPF_EXIST))
|
||||
return -EINVAL;
|
||||
if (unlikely(icsk->icsk_ulp_data))
|
||||
if (inet_csk_has_ulp(sk))
|
||||
return -EINVAL;
|
||||
|
||||
link = sk_psock_init_link();
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue