OpenCloudOS-Kernel/kernel/tkernel/netbind.c

90 lines
1.7 KiB
C

#include <linux/tkernel.h>
#include <linux/seq_file.h>
#include <linux/proc_fs.h>
#include <linux/init.h>
#include <net/sock.h>
bool nonpriv_prot_sock_flag[PROT_SOCK];
EXPORT_SYMBOL(nonpriv_prot_sock_flag);
static int netbind_proc_show(struct seq_file *m, void *v)
{
int i;
for (i = 1; i < PROT_SOCK; i++) {
if (nonpriv_prot_sock_flag[i])
seq_printf(m, "%d\n", i);
}
return 0;
}
static int netbind_proc_open(struct inode *inode, struct file *file)
{
return single_open(file, netbind_proc_show, NULL);
}
static ssize_t netbind_proc_write(struct file *file, const char __user *buf,
size_t length, loff_t *ppos)
{
int port, en;
char *buffer, *p;
int err;
if (!buf || length > PAGE_SIZE)
return -EINVAL;
buffer = (char *)__get_free_page(GFP_KERNEL);
if (!buffer)
return -ENOMEM;
err = -EFAULT;
if (copy_from_user(buffer, buf, length))
goto out;
err = -EINVAL;
if (length < PAGE_SIZE)
buffer[length] = '\0';
else if (buffer[PAGE_SIZE-1])
goto out;
en = 1;
p = buffer;
if (*p == '+') {
p++;
} else if (*p == '-') {
en = 0;
p++;
}
if (*p < '0' || *p > '9')
goto out;
port = simple_strtoul(p, &p, 0);
if (*p != '\n' && *p != '\r' && *p != '\0')
goto out;
if (port <= 0 || port >= PROT_SOCK)
goto out;
nonpriv_prot_sock_flag[port] = en;
err = length;
out:
free_page((unsigned long)buffer);
return err;
}
static const struct proc_ops netbind_proc_ops = {
.proc_open = netbind_proc_open,
.proc_read = seq_read,
.proc_write = netbind_proc_write,
.proc_lseek = seq_lseek,
.proc_release = single_release,
};
static int __init nonpriv_netbind_init(void)
{
proc_create("nonpriv_netbind", 0, proc_tkernel, &netbind_proc_ops);
return 0;
}
late_initcall(nonpriv_netbind_init);