907 lines · 24.7 KB · C
RawDownload
/* * Copyright (c) 2018 Cumulus Networks. All rights reserved. * Copyright (c) 2018 David Ahern <dsa@cumulusnetworks.com> * Copyright (c) 2019 Mellanox Technologies. All rights reserved. * * This software is licensed under the GNU General License Version 2, * June 1991 as shown in the file COPYING in the top-level directory of this * source tree. * * THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" * WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE * OF THE PROGRAM IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME * THE COST OF ALL NECESSARY SERVICING, REPAIR OR CORRECTION. */ #include <linux/debugfs.h>#include <linux/device.h>#include <linux/etherdevice.h>#include <linux/inet.h>#include <linux/jiffies.h>#include <linux/kernel.h>#include <linux/list.h>#include <linux/mutex.h>#include <linux/random.h>#include <linux/rtnetlink.h>#include <linux/workqueue.h>#include <net/devlink.h>#include <net/ip.h>#include <uapi/linux/devlink.h>#include <uapi/linux/ip.h>#include <uapi/linux/udp.h> #include "netdevsim.h" static struct dentry *nsim_dev_ddir; #define NSIM_DEV_DUMMY_REGION_SIZE (1024 * 32) static ssize_t nsim_dev_take_snapshot_write(struct file *file,					    const char __user *data,					    size_t count, loff_t *ppos){	struct nsim_dev *nsim_dev = file->private_data;	void *dummy_data;	int err;	u32 id; 	dummy_data = kmalloc(NSIM_DEV_DUMMY_REGION_SIZE, GFP_KERNEL);	if (!dummy_data)		return -ENOMEM; 	get_random_bytes(dummy_data, NSIM_DEV_DUMMY_REGION_SIZE); 	id = devlink_region_shapshot_id_get(priv_to_devlink(nsim_dev));	err = devlink_region_snapshot_create(nsim_dev->dummy_region,					     dummy_data, id, kfree);	if (err) {		pr_err("Failed to create region snapshot\n");		kfree(dummy_data);		return err;	} 	return count;} static const struct file_operations nsim_dev_take_snapshot_fops = {	.open = simple_open,	.write = nsim_dev_take_snapshot_write,	.llseek = generic_file_llseek,}; static int nsim_dev_debugfs_init(struct nsim_dev *nsim_dev){	char dev_ddir_name[16]; 	sprintf(dev_ddir_name, DRV_NAME "%u", nsim_dev->nsim_bus_dev->dev.id);	nsim_dev->ddir = debugfs_create_dir(dev_ddir_name, nsim_dev_ddir);	if (IS_ERR_OR_NULL(nsim_dev->ddir))		return PTR_ERR_OR_ZERO(nsim_dev->ddir) ?: -EINVAL;	nsim_dev->ports_ddir = debugfs_create_dir("ports", nsim_dev->ddir);	if (IS_ERR_OR_NULL(nsim_dev->ports_ddir))		return PTR_ERR_OR_ZERO(nsim_dev->ports_ddir) ?: -EINVAL;	debugfs_create_bool("fw_update_status", 0600, nsim_dev->ddir,			    &nsim_dev->fw_update_status);	debugfs_create_u32("max_macs", 0600, nsim_dev->ddir,			   &nsim_dev->max_macs);	debugfs_create_bool("test1", 0600, nsim_dev->ddir,			    &nsim_dev->test1);	debugfs_create_file("take_snapshot", 0200, nsim_dev->ddir, nsim_dev,			    &nsim_dev_take_snapshot_fops);	return 0;} static void nsim_dev_debugfs_exit(struct nsim_dev *nsim_dev){	debugfs_remove_recursive(nsim_dev->ports_ddir);	debugfs_remove_recursive(nsim_dev->ddir);} static int nsim_dev_port_debugfs_init(struct nsim_dev *nsim_dev,				      struct nsim_dev_port *nsim_dev_port){	char port_ddir_name[16];	char dev_link_name[32]; 	sprintf(port_ddir_name, "%u", nsim_dev_port->port_index);	nsim_dev_port->ddir = debugfs_create_dir(port_ddir_name,						 nsim_dev->ports_ddir);	if (IS_ERR_OR_NULL(nsim_dev_port->ddir))		return -ENOMEM; 	sprintf(dev_link_name, "../../../" DRV_NAME "%u",		nsim_dev->nsim_bus_dev->dev.id);	debugfs_create_symlink("dev", nsim_dev_port->ddir, dev_link_name); 	return 0;} static void nsim_dev_port_debugfs_exit(struct nsim_dev_port *nsim_dev_port){	debugfs_remove_recursive(nsim_dev_port->ddir);} static struct net *nsim_devlink_net(struct devlink *devlink){	return &init_net;} static u64 nsim_dev_ipv4_fib_resource_occ_get(void *priv){	struct net *net = priv; 	return nsim_fib_get_val(net, NSIM_RESOURCE_IPV4_FIB, false);} static u64 nsim_dev_ipv4_fib_rules_res_occ_get(void *priv){	struct net *net = priv; 	return nsim_fib_get_val(net, NSIM_RESOURCE_IPV4_FIB_RULES, false);} static u64 nsim_dev_ipv6_fib_resource_occ_get(void *priv){	struct net *net = priv; 	return nsim_fib_get_val(net, NSIM_RESOURCE_IPV6_FIB, false);} static u64 nsim_dev_ipv6_fib_rules_res_occ_get(void *priv){	struct net *net = priv; 	return nsim_fib_get_val(net, NSIM_RESOURCE_IPV6_FIB_RULES, false);} static int nsim_dev_resources_register(struct devlink *devlink){	struct devlink_resource_size_params params = {		.size_max = (u64)-1,		.size_granularity = 1,		.unit = DEVLINK_RESOURCE_UNIT_ENTRY	};	struct net *net = nsim_devlink_net(devlink);	int err;	u64 n; 	/* Resources for IPv4 */	err = devlink_resource_register(devlink, "IPv4", (u64)-1,					NSIM_RESOURCE_IPV4,					DEVLINK_RESOURCE_ID_PARENT_TOP,					&params);	if (err) {		pr_err("Failed to register IPv4 top resource\n");		goto out;	} 	n = nsim_fib_get_val(net, NSIM_RESOURCE_IPV4_FIB, true);	err = devlink_resource_register(devlink, "fib", n,					NSIM_RESOURCE_IPV4_FIB,					NSIM_RESOURCE_IPV4, &params);	if (err) {		pr_err("Failed to register IPv4 FIB resource\n");		return err;	} 	n = nsim_fib_get_val(net, NSIM_RESOURCE_IPV4_FIB_RULES, true);	err = devlink_resource_register(devlink, "fib-rules", n,					NSIM_RESOURCE_IPV4_FIB_RULES,					NSIM_RESOURCE_IPV4, &params);	if (err) {		pr_err("Failed to register IPv4 FIB rules resource\n");		return err;	} 	/* Resources for IPv6 */	err = devlink_resource_register(devlink, "IPv6", (u64)-1,					NSIM_RESOURCE_IPV6,					DEVLINK_RESOURCE_ID_PARENT_TOP,					&params);	if (err) {		pr_err("Failed to register IPv6 top resource\n");		goto out;	} 	n = nsim_fib_get_val(net, NSIM_RESOURCE_IPV6_FIB, true);	err = devlink_resource_register(devlink, "fib", n,					NSIM_RESOURCE_IPV6_FIB,					NSIM_RESOURCE_IPV6, &params);	if (err) {		pr_err("Failed to register IPv6 FIB resource\n");		return err;	} 	n = nsim_fib_get_val(net, NSIM_RESOURCE_IPV6_FIB_RULES, true);	err = devlink_resource_register(devlink, "fib-rules", n,					NSIM_RESOURCE_IPV6_FIB_RULES,					NSIM_RESOURCE_IPV6, &params);	if (err) {		pr_err("Failed to register IPv6 FIB rules resource\n");		return err;	} 	devlink_resource_occ_get_register(devlink,					  NSIM_RESOURCE_IPV4_FIB,					  nsim_dev_ipv4_fib_resource_occ_get,					  net);	devlink_resource_occ_get_register(devlink,					  NSIM_RESOURCE_IPV4_FIB_RULES,					  nsim_dev_ipv4_fib_rules_res_occ_get,					  net);	devlink_resource_occ_get_register(devlink,					  NSIM_RESOURCE_IPV6_FIB,					  nsim_dev_ipv6_fib_resource_occ_get,					  net);	devlink_resource_occ_get_register(devlink,					  NSIM_RESOURCE_IPV6_FIB_RULES,					  nsim_dev_ipv6_fib_rules_res_occ_get,					  net);out:	return err;} enum nsim_devlink_param_id {	NSIM_DEVLINK_PARAM_ID_BASE = DEVLINK_PARAM_GENERIC_ID_MAX,	NSIM_DEVLINK_PARAM_ID_TEST1,}; static const struct devlink_param nsim_devlink_params[] = {	DEVLINK_PARAM_GENERIC(MAX_MACS,			      BIT(DEVLINK_PARAM_CMODE_DRIVERINIT),			      NULL, NULL, NULL),	DEVLINK_PARAM_DRIVER(NSIM_DEVLINK_PARAM_ID_TEST1,			     "test1", DEVLINK_PARAM_TYPE_BOOL,			     BIT(DEVLINK_PARAM_CMODE_DRIVERINIT),			     NULL, NULL, NULL),}; static void nsim_devlink_set_params_init_values(struct nsim_dev *nsim_dev,						struct devlink *devlink){	union devlink_param_value value; 	value.vu32 = nsim_dev->max_macs;	devlink_param_driverinit_value_set(devlink,					   DEVLINK_PARAM_GENERIC_ID_MAX_MACS,					   value);	value.vbool = nsim_dev->test1;	devlink_param_driverinit_value_set(devlink,					   NSIM_DEVLINK_PARAM_ID_TEST1,					   value);} static void nsim_devlink_param_load_driverinit_values(struct devlink *devlink){	struct nsim_dev *nsim_dev = devlink_priv(devlink);	union devlink_param_value saved_value;	int err; 	err = devlink_param_driverinit_value_get(devlink,						 DEVLINK_PARAM_GENERIC_ID_MAX_MACS,						 &saved_value);	if (!err)		nsim_dev->max_macs = saved_value.vu32;	err = devlink_param_driverinit_value_get(devlink,						 NSIM_DEVLINK_PARAM_ID_TEST1,						 &saved_value);	if (!err)		nsim_dev->test1 = saved_value.vbool;} #define NSIM_DEV_DUMMY_REGION_SNAPSHOT_MAX 16 static int nsim_dev_dummy_region_init(struct nsim_dev *nsim_dev,				      struct devlink *devlink){	nsim_dev->dummy_region =		devlink_region_create(devlink, "dummy",				      NSIM_DEV_DUMMY_REGION_SNAPSHOT_MAX,				      NSIM_DEV_DUMMY_REGION_SIZE);	return PTR_ERR_OR_ZERO(nsim_dev->dummy_region);} static void nsim_dev_dummy_region_exit(struct nsim_dev *nsim_dev){	devlink_region_destroy(nsim_dev->dummy_region);} struct nsim_trap_item {	void *trap_ctx;	enum devlink_trap_action action;}; struct nsim_trap_data {	struct delayed_work trap_report_dw;	struct nsim_trap_item *trap_items_arr;	struct nsim_dev *nsim_dev;	spinlock_t trap_lock;	/* Protects trap_items_arr */}; /* All driver-specific traps must be documented in * Documentation/networking/devlink-trap-netdevsim.rst */enum {	NSIM_TRAP_ID_BASE = DEVLINK_TRAP_GENERIC_ID_MAX,	NSIM_TRAP_ID_FID_MISS,}; #define NSIM_TRAP_NAME_FID_MISS "fid_miss" #define NSIM_TRAP_METADATA DEVLINK_TRAP_METADATA_TYPE_F_IN_PORT #define NSIM_TRAP_DROP(_id, _group_id)					      \	DEVLINK_TRAP_GENERIC(DROP, DROP, _id,				      \			     DEVLINK_TRAP_GROUP_GENERIC(_group_id),	      \			     NSIM_TRAP_METADATA)#define NSIM_TRAP_EXCEPTION(_id, _group_id)				      \	DEVLINK_TRAP_GENERIC(EXCEPTION, TRAP, _id,			      \			     DEVLINK_TRAP_GROUP_GENERIC(_group_id),	      \			     NSIM_TRAP_METADATA)#define NSIM_TRAP_DRIVER_EXCEPTION(_id, _group_id)			      \	DEVLINK_TRAP_DRIVER(EXCEPTION, TRAP, NSIM_TRAP_ID_##_id,	      \			    NSIM_TRAP_NAME_##_id,			      \			    DEVLINK_TRAP_GROUP_GENERIC(_group_id),	      \			    NSIM_TRAP_METADATA) static const struct devlink_trap nsim_traps_arr[] = {	NSIM_TRAP_DROP(SMAC_MC, L2_DROPS),	NSIM_TRAP_DROP(VLAN_TAG_MISMATCH, L2_DROPS),	NSIM_TRAP_DROP(INGRESS_VLAN_FILTER, L2_DROPS),	NSIM_TRAP_DROP(INGRESS_STP_FILTER, L2_DROPS),	NSIM_TRAP_DROP(EMPTY_TX_LIST, L2_DROPS),	NSIM_TRAP_DROP(PORT_LOOPBACK_FILTER, L2_DROPS),	NSIM_TRAP_DRIVER_EXCEPTION(FID_MISS, L2_DROPS),	NSIM_TRAP_DROP(BLACKHOLE_ROUTE, L3_DROPS),	NSIM_TRAP_EXCEPTION(TTL_ERROR, L3_DROPS),	NSIM_TRAP_DROP(TAIL_DROP, BUFFER_DROPS),}; #define NSIM_TRAP_L4_DATA_LEN 100 static struct sk_buff *nsim_dev_trap_skb_build(void){	int tot_len, data_len = NSIM_TRAP_L4_DATA_LEN;	struct sk_buff *skb;	struct udphdr *udph;	struct ethhdr *eth;	struct iphdr *iph; 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);	if (!skb)		return NULL;	tot_len = sizeof(struct iphdr) + sizeof(struct udphdr) + data_len; 	skb_reset_mac_header(skb);	eth = skb_put(skb, sizeof(struct ethhdr));	eth_random_addr(eth->h_dest);	eth_random_addr(eth->h_source);	eth->h_proto = htons(ETH_P_IP);	skb->protocol = htons(ETH_P_IP); 	skb_set_network_header(skb, skb->len);	iph = skb_put(skb, sizeof(struct iphdr));	iph->protocol = IPPROTO_UDP;	iph->saddr = in_aton("192.0.2.1");	iph->daddr = in_aton("198.51.100.1");	iph->version = 0x4;	iph->frag_off = 0;	iph->ihl = 0x5;	iph->tot_len = htons(tot_len);	iph->ttl = 100;	iph->check = 0;	iph->check = ip_fast_csum((unsigned char *)iph, iph->ihl); 	skb_set_transport_header(skb, skb->len);	udph = skb_put_zero(skb, sizeof(struct udphdr) + data_len);	get_random_bytes(&udph->source, sizeof(u16));	get_random_bytes(&udph->dest, sizeof(u16));	udph->len = htons(sizeof(struct udphdr) + data_len); 	return skb;} static void nsim_dev_trap_report(struct nsim_dev_port *nsim_dev_port){	struct nsim_dev *nsim_dev = nsim_dev_port->ns->nsim_dev;	struct devlink *devlink = priv_to_devlink(nsim_dev);	struct nsim_trap_data *nsim_trap_data;	int i; 	nsim_trap_data = nsim_dev->trap_data; 	spin_lock(&nsim_trap_data->trap_lock);	for (i = 0; i < ARRAY_SIZE(nsim_traps_arr); i++) {		struct nsim_trap_item *nsim_trap_item;		struct sk_buff *skb; 		nsim_trap_item = &nsim_trap_data->trap_items_arr[i];		if (nsim_trap_item->action == DEVLINK_TRAP_ACTION_DROP)			continue; 		skb = nsim_dev_trap_skb_build();		if (!skb)			continue;		skb->dev = nsim_dev_port->ns->netdev; 		/* Trapped packets are usually passed to devlink in softIRQ,		 * but in this case they are generated in a workqueue. Disable		 * softIRQs to prevent lockdep from complaining about		 * "incosistent lock state".		 */		local_bh_disable();		devlink_trap_report(devlink, skb, nsim_trap_item->trap_ctx,				    &nsim_dev_port->devlink_port);		local_bh_enable();		consume_skb(skb);	}	spin_unlock(&nsim_trap_data->trap_lock);} #define NSIM_TRAP_REPORT_INTERVAL_MS	100 static void nsim_dev_trap_report_work(struct work_struct *work){	struct nsim_trap_data *nsim_trap_data;	struct nsim_dev_port *nsim_dev_port;	struct nsim_dev *nsim_dev; 	nsim_trap_data = container_of(work, struct nsim_trap_data,				      trap_report_dw.work);	nsim_dev = nsim_trap_data->nsim_dev; 	/* For each running port and enabled packet trap, generate a UDP	 * packet with a random 5-tuple and report it.	 */	mutex_lock(&nsim_dev->port_list_lock);	list_for_each_entry(nsim_dev_port, &nsim_dev->port_list, list) {		if (!netif_running(nsim_dev_port->ns->netdev))			continue; 		nsim_dev_trap_report(nsim_dev_port);	}	mutex_unlock(&nsim_dev->port_list_lock); 	schedule_delayed_work(&nsim_dev->trap_data->trap_report_dw,			      msecs_to_jiffies(NSIM_TRAP_REPORT_INTERVAL_MS));} static int nsim_dev_traps_init(struct devlink *devlink){	struct nsim_dev *nsim_dev = devlink_priv(devlink);	struct nsim_trap_data *nsim_trap_data;	int err; 	nsim_trap_data = kzalloc(sizeof(*nsim_trap_data), GFP_KERNEL);	if (!nsim_trap_data)		return -ENOMEM; 	nsim_trap_data->trap_items_arr = kcalloc(ARRAY_SIZE(nsim_traps_arr),						 sizeof(struct nsim_trap_item),						 GFP_KERNEL);	if (!nsim_trap_data->trap_items_arr) {		err = -ENOMEM;		goto err_trap_data_free;	} 	/* The lock is used to protect the action state of the registered	 * traps. The value is written by user and read in delayed work when	 * iterating over all the traps.	 */	spin_lock_init(&nsim_trap_data->trap_lock);	nsim_trap_data->nsim_dev = nsim_dev;	nsim_dev->trap_data = nsim_trap_data; 	err = devlink_traps_register(devlink, nsim_traps_arr,				     ARRAY_SIZE(nsim_traps_arr), NULL);	if (err)		goto err_trap_items_free;

Lines 1–500 of 907.