|
@@ -1149,7 +1149,8 @@ void bpf_prog_destroy(struct bpf_prog *fp)
|
|
}
|
|
}
|
|
EXPORT_SYMBOL_GPL(bpf_prog_destroy);
|
|
EXPORT_SYMBOL_GPL(bpf_prog_destroy);
|
|
|
|
|
|
-static int __sk_attach_prog(struct bpf_prog *prog, struct sock *sk)
|
|
|
|
|
|
+static int __sk_attach_prog(struct bpf_prog *prog, struct sock *sk,
|
|
|
|
+ bool locked)
|
|
{
|
|
{
|
|
struct sk_filter *fp, *old_fp;
|
|
struct sk_filter *fp, *old_fp;
|
|
|
|
|
|
@@ -1165,10 +1166,8 @@ static int __sk_attach_prog(struct bpf_prog *prog, struct sock *sk)
|
|
return -ENOMEM;
|
|
return -ENOMEM;
|
|
}
|
|
}
|
|
|
|
|
|
- old_fp = rcu_dereference_protected(sk->sk_filter,
|
|
|
|
- sock_owned_by_user(sk));
|
|
|
|
|
|
+ old_fp = rcu_dereference_protected(sk->sk_filter, locked);
|
|
rcu_assign_pointer(sk->sk_filter, fp);
|
|
rcu_assign_pointer(sk->sk_filter, fp);
|
|
-
|
|
|
|
if (old_fp)
|
|
if (old_fp)
|
|
sk_filter_uncharge(sk, old_fp);
|
|
sk_filter_uncharge(sk, old_fp);
|
|
|
|
|
|
@@ -1247,7 +1246,8 @@ struct bpf_prog *__get_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
* occurs or there is insufficient memory for the filter a negative
|
|
* occurs or there is insufficient memory for the filter a negative
|
|
* errno code is returned. On success the return is zero.
|
|
* errno code is returned. On success the return is zero.
|
|
*/
|
|
*/
|
|
-int sk_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
|
|
|
|
+int __sk_attach_filter(struct sock_fprog *fprog, struct sock *sk,
|
|
|
|
+ bool locked)
|
|
{
|
|
{
|
|
struct bpf_prog *prog = __get_filter(fprog, sk);
|
|
struct bpf_prog *prog = __get_filter(fprog, sk);
|
|
int err;
|
|
int err;
|
|
@@ -1255,7 +1255,7 @@ int sk_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
if (IS_ERR(prog))
|
|
if (IS_ERR(prog))
|
|
return PTR_ERR(prog);
|
|
return PTR_ERR(prog);
|
|
|
|
|
|
- err = __sk_attach_prog(prog, sk);
|
|
|
|
|
|
+ err = __sk_attach_prog(prog, sk, locked);
|
|
if (err < 0) {
|
|
if (err < 0) {
|
|
__bpf_prog_release(prog);
|
|
__bpf_prog_release(prog);
|
|
return err;
|
|
return err;
|
|
@@ -1263,7 +1263,12 @@ int sk_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
|
|
|
|
return 0;
|
|
return 0;
|
|
}
|
|
}
|
|
-EXPORT_SYMBOL_GPL(sk_attach_filter);
|
|
|
|
|
|
+EXPORT_SYMBOL_GPL(__sk_attach_filter);
|
|
|
|
+
|
|
|
|
+int sk_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
|
|
+{
|
|
|
|
+ return __sk_attach_filter(fprog, sk, sock_owned_by_user(sk));
|
|
|
|
+}
|
|
|
|
|
|
int sk_reuseport_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
int sk_reuseport_attach_filter(struct sock_fprog *fprog, struct sock *sk)
|
|
{
|
|
{
|
|
@@ -1309,7 +1314,7 @@ int sk_attach_bpf(u32 ufd, struct sock *sk)
|
|
if (IS_ERR(prog))
|
|
if (IS_ERR(prog))
|
|
return PTR_ERR(prog);
|
|
return PTR_ERR(prog);
|
|
|
|
|
|
- err = __sk_attach_prog(prog, sk);
|
|
|
|
|
|
+ err = __sk_attach_prog(prog, sk, sock_owned_by_user(sk));
|
|
if (err < 0) {
|
|
if (err < 0) {
|
|
bpf_prog_put(prog);
|
|
bpf_prog_put(prog);
|
|
return err;
|
|
return err;
|
|
@@ -2250,7 +2255,7 @@ static int __init register_sk_filter_ops(void)
|
|
}
|
|
}
|
|
late_initcall(register_sk_filter_ops);
|
|
late_initcall(register_sk_filter_ops);
|
|
|
|
|
|
-int sk_detach_filter(struct sock *sk)
|
|
|
|
|
|
+int __sk_detach_filter(struct sock *sk, bool locked)
|
|
{
|
|
{
|
|
int ret = -ENOENT;
|
|
int ret = -ENOENT;
|
|
struct sk_filter *filter;
|
|
struct sk_filter *filter;
|
|
@@ -2258,8 +2263,7 @@ int sk_detach_filter(struct sock *sk)
|
|
if (sock_flag(sk, SOCK_FILTER_LOCKED))
|
|
if (sock_flag(sk, SOCK_FILTER_LOCKED))
|
|
return -EPERM;
|
|
return -EPERM;
|
|
|
|
|
|
- filter = rcu_dereference_protected(sk->sk_filter,
|
|
|
|
- sock_owned_by_user(sk));
|
|
|
|
|
|
+ filter = rcu_dereference_protected(sk->sk_filter, locked);
|
|
if (filter) {
|
|
if (filter) {
|
|
RCU_INIT_POINTER(sk->sk_filter, NULL);
|
|
RCU_INIT_POINTER(sk->sk_filter, NULL);
|
|
sk_filter_uncharge(sk, filter);
|
|
sk_filter_uncharge(sk, filter);
|
|
@@ -2268,7 +2272,12 @@ int sk_detach_filter(struct sock *sk)
|
|
|
|
|
|
return ret;
|
|
return ret;
|
|
}
|
|
}
|
|
-EXPORT_SYMBOL_GPL(sk_detach_filter);
|
|
|
|
|
|
+EXPORT_SYMBOL_GPL(__sk_detach_filter);
|
|
|
|
+
|
|
|
|
+int sk_detach_filter(struct sock *sk)
|
|
|
|
+{
|
|
|
|
+ return __sk_detach_filter(sk, sock_owned_by_user(sk));
|
|
|
|
+}
|
|
|
|
|
|
int sk_get_filter(struct sock *sk, struct sock_filter __user *ubuf,
|
|
int sk_get_filter(struct sock *sk, struct sock_filter __user *ubuf,
|
|
unsigned int len)
|
|
unsigned int len)
|