forked from torvalds/linux
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Cong Wang <cong.wang@bytedance.com>
- Loading branch information
Cong Wang
committed
Jan 21, 2021
1 parent
86f84ea
commit 1c9cd5c
Showing
4 changed files
with
72 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,50 @@ | ||
// SPDX-License-Identifier: GPL-2.0 | ||
/* Copyright (c) 2021 Bytedance Inc. */ | ||
|
||
#include <linux/skmsg.h> | ||
#include <net/sock.h> | ||
#include <net/af_unix.h> | ||
|
||
static struct proto *unix_prot_saved __read_mostly; | ||
static DEFINE_SPINLOCK(unix_prot_lock); | ||
static struct proto unix_bpf_prot; | ||
|
||
static void unix_bpf_rebuild_protos(struct proto *prot, const struct proto *base) | ||
{ | ||
*prot = *base; | ||
prot->close = sock_map_close; | ||
} | ||
|
||
static void unix_bpf_check_needs_rebuild(struct proto *ops) | ||
{ | ||
if (unlikely(ops != smp_load_acquire(&unix_prot_saved))) { | ||
spin_lock_bh(&unix_prot_lock); | ||
if (likely(ops != unix_prot_saved)) { | ||
unix_bpf_rebuild_protos(&unix_bpf_prot, ops); | ||
smp_store_release(&unix_prot_saved, ops); | ||
} | ||
spin_unlock_bh(&unix_prot_lock); | ||
} | ||
} | ||
|
||
int unix_bpf_update_proto(struct sock *sk, bool restore) | ||
{ | ||
struct sk_psock *psock = sk_psock(sk); | ||
|
||
if (restore) { | ||
sk->sk_write_space = psock->saved_write_space; | ||
/* Pairs with lockless read in sk_clone_lock() */ | ||
WRITE_ONCE(sk->sk_prot, psock->sk_proto); | ||
return 0; | ||
} | ||
|
||
unix_bpf_check_needs_rebuild(psock->sk_proto); | ||
/* Pairs with lockless read in sk_clone_lock() */ | ||
WRITE_ONCE(sk->sk_prot, &unix_bpf_prot); | ||
return 0; | ||
} | ||
|
||
void __init unix_bpf_build_proto(void) | ||
{ | ||
unix_bpf_rebuild_protos(&unix_bpf_prot, &unix_proto); | ||
} |