net: bpfilter: use cleanup callback to release umh_info
[platform/kernel/linux-starfive.git] / net / ipv4 / bpfilter / sockopt.c
1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/init.h>
3 #include <linux/module.h>
4 #include <linux/uaccess.h>
5 #include <linux/bpfilter.h>
6 #include <uapi/linux/bpf.h>
7 #include <linux/wait.h>
8 #include <linux/kmod.h>
9 #include <linux/fs.h>
10 #include <linux/file.h>
11
12 struct bpfilter_umh_ops bpfilter_ops;
13 EXPORT_SYMBOL_GPL(bpfilter_ops);
14
15 static void bpfilter_umh_cleanup(struct umh_info *info)
16 {
17         fput(info->pipe_to_umh);
18         fput(info->pipe_from_umh);
19         info->pid = 0;
20 }
21
22 static int bpfilter_mbox_request(struct sock *sk, int optname,
23                                  char __user *optval,
24                                  unsigned int optlen, bool is_set)
25 {
26         if (!bpfilter_ops.sockopt) {
27                 int err = request_module("bpfilter");
28
29                 if (err)
30                         return err;
31                 if (!bpfilter_ops.sockopt)
32                         return -ECHILD;
33         }
34         return bpfilter_ops.sockopt(sk, optname, optval, optlen, is_set);
35 }
36
37 int bpfilter_ip_set_sockopt(struct sock *sk, int optname, char __user *optval,
38                             unsigned int optlen)
39 {
40         return bpfilter_mbox_request(sk, optname, optval, optlen, true);
41 }
42
43 int bpfilter_ip_get_sockopt(struct sock *sk, int optname, char __user *optval,
44                             int __user *optlen)
45 {
46         int len;
47
48         if (get_user(len, optlen))
49                 return -EFAULT;
50
51         return bpfilter_mbox_request(sk, optname, optval, len, false);
52 }
53
54 static int __init bpfilter_sockopt_init(void)
55 {
56         bpfilter_ops.info.cmdline = "bpfilter_umh";
57         bpfilter_ops.info.cleanup = &bpfilter_umh_cleanup;
58
59         return 0;
60 }
61
62 module_init(bpfilter_sockopt_init);