1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
|
// SPDX-License-Identifier: GPL-2.0-only
#include <linux/debugfs.h>
#include <linux/fault-inject.h>
#include <linux/netdevice.h>
#include <linux/skbuff.h>
static struct {
struct fault_attr attr;
char devname[IFNAMSIZ];
bool filtered;
} skb_realloc = {
.attr = FAULT_ATTR_INITIALIZER,
.filtered = false,
};
static bool should_fail_net_realloc_skb(struct sk_buff *skb)
{
struct net_device *net = skb->dev;
if (skb_realloc.filtered &&
strncmp(net->name, skb_realloc.devname, IFNAMSIZ))
/* device name filter set, but names do not match */
return false;
if (!should_fail(&skb_realloc.attr, 1))
return false;
return true;
}
ALLOW_ERROR_INJECTION(should_fail_net_realloc_skb, TRUE);
void skb_might_realloc(struct sk_buff *skb)
{
if (!should_fail_net_realloc_skb(skb))
return;
pskb_expand_head(skb, 0, 0, GFP_ATOMIC);
}
EXPORT_SYMBOL(skb_might_realloc);
static int __init fail_skb_realloc_setup(char *str)
{
return setup_fault_attr(&skb_realloc.attr, str);
}
__setup("fail_skb_realloc=", fail_skb_realloc_setup);
static void reset_settings(void)
{
skb_realloc.filtered = false;
memset(&skb_realloc.devname, 0, IFNAMSIZ);
}
static ssize_t devname_write(struct file *file, const char __user *buffer,
size_t count, loff_t *ppos)
{
ssize_t ret;
reset_settings();
ret = simple_write_to_buffer(&skb_realloc.devname, IFNAMSIZ,
ppos, buffer, count);
if (ret < 0)
return ret;
skb_realloc.devname[IFNAMSIZ - 1] = '\0';
/* Remove a possible \n at the end of devname */
strim(skb_realloc.devname);
if (strnlen(skb_realloc.devname, IFNAMSIZ))
skb_realloc.filtered = true;
return count;
}
static ssize_t devname_read(struct file *file,
char __user *buffer,
size_t size, loff_t *ppos)
{
if (!skb_realloc.filtered)
return 0;
return simple_read_from_buffer(buffer, size, ppos, &skb_realloc.devname,
strlen(skb_realloc.devname));
}
static const struct file_operations devname_ops = {
.write = devname_write,
.read = devname_read,
};
static int __init fail_skb_realloc_debugfs(void)
{
umode_t mode = S_IFREG | 0600;
struct dentry *dir;
dir = fault_create_debugfs_attr("fail_skb_realloc", NULL,
&skb_realloc.attr);
if (IS_ERR(dir))
return PTR_ERR(dir);
debugfs_create_file("devname", mode, dir, NULL, &devname_ops);
return 0;
}
late_initcall(fail_skb_realloc_debugfs);
|