diff options
Diffstat (limited to 'drivers/net/netdevsim')
| -rw-r--r-- | drivers/net/netdevsim/Makefile | 4 | ||||
| -rw-r--r-- | drivers/net/netdevsim/dev.c | 17 | ||||
| -rw-r--r-- | drivers/net/netdevsim/ethtool.c | 36 | ||||
| -rw-r--r-- | drivers/net/netdevsim/fib.c | 147 | ||||
| -rw-r--r-- | drivers/net/netdevsim/health.c | 11 | ||||
| -rw-r--r-- | drivers/net/netdevsim/netdevsim.h | 18 | ||||
| -rw-r--r-- | drivers/net/netdevsim/psample.c | 265 | 
7 files changed, 476 insertions, 22 deletions
| diff --git a/drivers/net/netdevsim/Makefile b/drivers/net/netdevsim/Makefile index ade086eed955..a1cbfa44a1e1 100644 --- a/drivers/net/netdevsim/Makefile +++ b/drivers/net/netdevsim/Makefile @@ -13,3 +13,7 @@ endif  ifneq ($(CONFIG_XFRM_OFFLOAD),)  netdevsim-objs += ipsec.o  endif + +ifneq ($(CONFIG_PSAMPLE),) +netdevsim-objs += psample.o +endif diff --git a/drivers/net/netdevsim/dev.c b/drivers/net/netdevsim/dev.c index dbeb29fa16e8..6189a4c0d39e 100644 --- a/drivers/net/netdevsim/dev.c +++ b/drivers/net/netdevsim/dev.c @@ -1032,10 +1032,14 @@ static int nsim_dev_reload_create(struct nsim_dev *nsim_dev,  	if (err)  		goto err_fib_destroy; -	err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count); +	err = nsim_dev_psample_init(nsim_dev);  	if (err)  		goto err_health_exit; +	err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count); +	if (err) +		goto err_psample_exit; +  	nsim_dev->take_snapshot = debugfs_create_file("take_snapshot",  						      0200,  						      nsim_dev->ddir, @@ -1043,6 +1047,8 @@ static int nsim_dev_reload_create(struct nsim_dev *nsim_dev,  						&nsim_dev_take_snapshot_fops);  	return 0; +err_psample_exit: +	nsim_dev_psample_exit(nsim_dev);  err_health_exit:  	nsim_dev_health_exit(nsim_dev);  err_fib_destroy: @@ -1118,14 +1124,20 @@ int nsim_dev_probe(struct nsim_bus_dev *nsim_bus_dev)  	if (err)  		goto err_health_exit; -	err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count); +	err = nsim_dev_psample_init(nsim_dev);  	if (err)  		goto err_bpf_dev_exit; +	err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count); +	if (err) +		goto err_psample_exit; +  	devlink_params_publish(devlink);  	devlink_reload_enable(devlink);  	return 0; +err_psample_exit: +	nsim_dev_psample_exit(nsim_dev);  err_bpf_dev_exit:  	nsim_bpf_dev_exit(nsim_dev);  err_health_exit: @@ -1158,6 +1170,7 @@ static void nsim_dev_reload_destroy(struct nsim_dev *nsim_dev)  		return;  	debugfs_remove(nsim_dev->take_snapshot);  	nsim_dev_port_del_all(nsim_dev); +	nsim_dev_psample_exit(nsim_dev);  	nsim_dev_health_exit(nsim_dev);  	nsim_fib_destroy(devlink, nsim_dev->fib_data);  	nsim_dev_traps_exit(devlink); diff --git a/drivers/net/netdevsim/ethtool.c b/drivers/net/netdevsim/ethtool.c index 166f0d6cbcf7..c9ae52595a8f 100644 --- a/drivers/net/netdevsim/ethtool.c +++ b/drivers/net/netdevsim/ethtool.c @@ -77,6 +77,34 @@ static int nsim_set_ringparam(struct net_device *dev,  	return 0;  } +static int +nsim_get_fecparam(struct net_device *dev, struct ethtool_fecparam *fecparam) +{ +	struct netdevsim *ns = netdev_priv(dev); + +	if (ns->ethtool.get_err) +		return -ns->ethtool.get_err; +	memcpy(fecparam, &ns->ethtool.fec, sizeof(ns->ethtool.fec)); +	return 0; +} + +static int +nsim_set_fecparam(struct net_device *dev, struct ethtool_fecparam *fecparam) +{ +	struct netdevsim *ns = netdev_priv(dev); +	u32 fec; + +	if (ns->ethtool.set_err) +		return -ns->ethtool.set_err; +	memcpy(&ns->ethtool.fec, fecparam, sizeof(ns->ethtool.fec)); +	fec = fecparam->fec; +	if (fec == ETHTOOL_FEC_AUTO) +		fec |= ETHTOOL_FEC_OFF; +	fec |= ETHTOOL_FEC_NONE; +	ns->ethtool.fec.active_fec = 1 << (fls(fec) - 1); +	return 0; +} +  static const struct ethtool_ops nsim_ethtool_ops = {  	.supported_coalesce_params	= ETHTOOL_COALESCE_ALL_PARAMS,  	.get_pause_stats	        = nsim_get_pause_stats, @@ -86,6 +114,8 @@ static const struct ethtool_ops nsim_ethtool_ops = {  	.get_coalesce			= nsim_get_coalesce,  	.get_ringparam			= nsim_get_ringparam,  	.set_ringparam			= nsim_set_ringparam, +	.get_fecparam			= nsim_get_fecparam, +	.set_fecparam			= nsim_set_fecparam,  };  static void nsim_ethtool_ring_init(struct netdevsim *ns) @@ -104,8 +134,14 @@ void nsim_ethtool_init(struct netdevsim *ns)  	nsim_ethtool_ring_init(ns); +	ns->ethtool.fec.fec = ETHTOOL_FEC_NONE; +	ns->ethtool.fec.active_fec = ETHTOOL_FEC_NONE; +  	ethtool = debugfs_create_dir("ethtool", ns->nsim_dev_port->ddir); +	debugfs_create_u32("get_err", 0600, ethtool, &ns->ethtool.get_err); +	debugfs_create_u32("set_err", 0600, ethtool, &ns->ethtool.set_err); +  	dir = debugfs_create_dir("pause", ethtool);  	debugfs_create_bool("report_stats_rx", 0600, dir,  			    &ns->ethtool.pauseparam.report_stats_rx); diff --git a/drivers/net/netdevsim/fib.c b/drivers/net/netdevsim/fib.c index 46fb414f7ca6..213d3e5056c8 100644 --- a/drivers/net/netdevsim/fib.c +++ b/drivers/net/netdevsim/fib.c @@ -14,6 +14,7 @@   * THE COST OF ALL NECESSARY SERVICING, REPAIR OR CORRECTION.   */ +#include <linux/bitmap.h>  #include <linux/in6.h>  #include <linux/kernel.h>  #include <linux/list.h> @@ -47,15 +48,18 @@ struct nsim_fib_data {  	struct nsim_fib_entry nexthops;  	struct rhashtable fib_rt_ht;  	struct list_head fib_rt_list; -	struct mutex fib_lock; /* Protects hashtable and list */ +	struct mutex fib_lock; /* Protects FIB HT and list */  	struct notifier_block nexthop_nb;  	struct rhashtable nexthop_ht;  	struct devlink *devlink;  	struct work_struct fib_event_work;  	struct list_head fib_event_queue;  	spinlock_t fib_event_queue_lock; /* Protects fib event queue list */ +	struct mutex nh_lock; /* Protects NH HT */  	struct dentry *ddir;  	bool fail_route_offload; +	bool fail_res_nexthop_group_replace; +	bool fail_nexthop_bucket_replace;  };  struct nsim_fib_rt_key { @@ -116,6 +120,7 @@ struct nsim_nexthop {  	struct rhash_head ht_node;  	u64 occ;  	u32 id; +	bool is_resilient;  };  static const struct rhashtable_params nsim_nexthop_ht_params = { @@ -561,7 +566,7 @@ nsim_fib6_rt_create(struct nsim_fib_data *data,  err_fib6_rt_nh_del:  	for (i--; i >= 0; i--) {  		nsim_fib6_rt_nh_del(fib6_rt, rt_arr[i]); -	}; +	}  	nsim_fib_rt_fini(&fib6_rt->common);  	kfree(fib6_rt);  	return ERR_PTR(err); @@ -869,10 +874,8 @@ err_rt_offload_failed_flag_set:  	return err;  } -static int nsim_fib_event(struct nsim_fib_event *fib_event) +static void nsim_fib_event(struct nsim_fib_event *fib_event)  { -	int err = 0; -  	switch (fib_event->family) {  	case AF_INET:  		nsim_fib4_event(fib_event->data, &fib_event->fen_info, @@ -885,8 +888,6 @@ static int nsim_fib_event(struct nsim_fib_event *fib_event)  		nsim_fib6_event_fini(&fib_event->fib6_event);  		break;  	} - -	return err;  }  static int nsim_fib4_prepare_event(struct fib_notifier_info *info, @@ -1118,6 +1119,10 @@ static struct nsim_nexthop *nsim_nexthop_create(struct nsim_fib_data *data,  		for (i = 0; i < info->nh_grp->num_nh; i++)  			occ += info->nh_grp->nh_entries[i].weight;  		break; +	case NH_NOTIFIER_INFO_TYPE_RES_TABLE: +		occ = info->nh_res_table->num_nh_buckets; +		nexthop->is_resilient = true; +		break;  	default:  		NL_SET_ERR_MSG_MOD(info->extack, "Unsupported nexthop type");  		kfree(nexthop); @@ -1160,6 +1165,21 @@ err_num_decrease:  } +static void nsim_nexthop_hw_flags_set(struct net *net, +				      const struct nsim_nexthop *nexthop, +				      bool trap) +{ +	int i; + +	nexthop_set_hw_flags(net, nexthop->id, false, trap); + +	if (!nexthop->is_resilient) +		return; + +	for (i = 0; i < nexthop->occ; i++) +		nexthop_bucket_set_hw_flags(net, nexthop->id, i, false, trap); +} +  static int nsim_nexthop_add(struct nsim_fib_data *data,  			    struct nsim_nexthop *nexthop,  			    struct netlink_ext_ack *extack) @@ -1178,7 +1198,7 @@ static int nsim_nexthop_add(struct nsim_fib_data *data,  		goto err_nexthop_dismiss;  	} -	nexthop_set_hw_flags(net, nexthop->id, false, true); +	nsim_nexthop_hw_flags_set(net, nexthop, true);  	return 0; @@ -1207,7 +1227,7 @@ static int nsim_nexthop_replace(struct nsim_fib_data *data,  		goto err_nexthop_dismiss;  	} -	nexthop_set_hw_flags(net, nexthop->id, false, true); +	nsim_nexthop_hw_flags_set(net, nexthop, true);  	nsim_nexthop_account(data, nexthop_old->occ, false, extack);  	nsim_nexthop_destroy(nexthop_old); @@ -1258,6 +1278,32 @@ static void nsim_nexthop_remove(struct nsim_fib_data *data,  	nsim_nexthop_destroy(nexthop);  } +static int nsim_nexthop_res_table_pre_replace(struct nsim_fib_data *data, +					      struct nh_notifier_info *info) +{ +	if (data->fail_res_nexthop_group_replace) { +		NL_SET_ERR_MSG_MOD(info->extack, "Failed to replace a resilient nexthop group"); +		return -EINVAL; +	} + +	return 0; +} + +static int nsim_nexthop_bucket_replace(struct nsim_fib_data *data, +				       struct nh_notifier_info *info) +{ +	if (data->fail_nexthop_bucket_replace) { +		NL_SET_ERR_MSG_MOD(info->extack, "Failed to replace nexthop bucket"); +		return -EINVAL; +	} + +	nexthop_bucket_set_hw_flags(info->net, info->id, +				    info->nh_res_bucket->bucket_index, +				    false, true); + +	return 0; +} +  static int nsim_nexthop_event_nb(struct notifier_block *nb, unsigned long event,  				 void *ptr)  { @@ -1266,8 +1312,7 @@ static int nsim_nexthop_event_nb(struct notifier_block *nb, unsigned long event,  	struct nh_notifier_info *info = ptr;  	int err = 0; -	ASSERT_RTNL(); - +	mutex_lock(&data->nh_lock);  	switch (event) {  	case NEXTHOP_EVENT_REPLACE:  		err = nsim_nexthop_insert(data, info); @@ -1275,10 +1320,17 @@ static int nsim_nexthop_event_nb(struct notifier_block *nb, unsigned long event,  	case NEXTHOP_EVENT_DEL:  		nsim_nexthop_remove(data, info);  		break; +	case NEXTHOP_EVENT_RES_TABLE_PRE_REPLACE: +		err = nsim_nexthop_res_table_pre_replace(data, info); +		break; +	case NEXTHOP_EVENT_BUCKET_REPLACE: +		err = nsim_nexthop_bucket_replace(data, info); +		break;  	default:  		break;  	} +	mutex_unlock(&data->nh_lock);  	return notifier_from_errno(err);  } @@ -1289,11 +1341,68 @@ static void nsim_nexthop_free(void *ptr, void *arg)  	struct net *net;  	net = devlink_net(data->devlink); -	nexthop_set_hw_flags(net, nexthop->id, false, false); +	nsim_nexthop_hw_flags_set(net, nexthop, false);  	nsim_nexthop_account(data, nexthop->occ, false, NULL);  	nsim_nexthop_destroy(nexthop);  } +static ssize_t nsim_nexthop_bucket_activity_write(struct file *file, +						  const char __user *user_buf, +						  size_t size, loff_t *ppos) +{ +	struct nsim_fib_data *data = file->private_data; +	struct net *net = devlink_net(data->devlink); +	struct nsim_nexthop *nexthop; +	unsigned long *activity; +	loff_t pos = *ppos; +	u16 bucket_index; +	char buf[128]; +	int err = 0; +	u32 nhid; + +	if (pos != 0) +		return -EINVAL; +	if (size > sizeof(buf)) +		return -EINVAL; +	if (copy_from_user(buf, user_buf, size)) +		return -EFAULT; +	if (sscanf(buf, "%u %hu", &nhid, &bucket_index) != 2) +		return -EINVAL; + +	rtnl_lock(); + +	nexthop = rhashtable_lookup_fast(&data->nexthop_ht, &nhid, +					 nsim_nexthop_ht_params); +	if (!nexthop || !nexthop->is_resilient || +	    bucket_index >= nexthop->occ) { +		err = -EINVAL; +		goto out; +	} + +	activity = bitmap_zalloc(nexthop->occ, GFP_KERNEL); +	if (!activity) { +		err = -ENOMEM; +		goto out; +	} + +	bitmap_set(activity, bucket_index, 1); +	nexthop_res_grp_activity_update(net, nhid, nexthop->occ, activity); +	bitmap_free(activity); + +out: +	rtnl_unlock(); + +	*ppos = size; +	return err ?: size; +} + +static const struct file_operations nsim_nexthop_bucket_activity_fops = { +	.open = simple_open, +	.write = nsim_nexthop_bucket_activity_write, +	.llseek = no_llseek, +	.owner = THIS_MODULE, +}; +  static u64 nsim_fib_ipv4_resource_occ_get(void *priv)  {  	struct nsim_fib_data *data = priv; @@ -1383,6 +1492,17 @@ nsim_fib_debugfs_init(struct nsim_fib_data *data, struct nsim_dev *nsim_dev)  	data->fail_route_offload = false;  	debugfs_create_bool("fail_route_offload", 0600, data->ddir,  			    &data->fail_route_offload); + +	data->fail_res_nexthop_group_replace = false; +	debugfs_create_bool("fail_res_nexthop_group_replace", 0600, data->ddir, +			    &data->fail_res_nexthop_group_replace); + +	data->fail_nexthop_bucket_replace = false; +	debugfs_create_bool("fail_nexthop_bucket_replace", 0600, data->ddir, +			    &data->fail_nexthop_bucket_replace); + +	debugfs_create_file("nexthop_bucket_activity", 0200, data->ddir, +			    data, &nsim_nexthop_bucket_activity_fops);  	return 0;  } @@ -1408,6 +1528,7 @@ struct nsim_fib_data *nsim_fib_create(struct devlink *devlink,  	if (err)  		goto err_data_free; +	mutex_init(&data->nh_lock);  	err = rhashtable_init(&data->nexthop_ht, &nsim_nexthop_ht_params);  	if (err)  		goto err_debugfs_exit; @@ -1473,6 +1594,7 @@ err_rhashtable_nexthop_destroy:  				    data);  	mutex_destroy(&data->fib_lock);  err_debugfs_exit: +	mutex_destroy(&data->nh_lock);  	nsim_fib_debugfs_exit(data);  err_data_free:  	kfree(data); @@ -1501,6 +1623,7 @@ void nsim_fib_destroy(struct devlink *devlink, struct nsim_fib_data *data)  	WARN_ON_ONCE(!list_empty(&data->fib_event_queue));  	WARN_ON_ONCE(!list_empty(&data->fib_rt_list));  	mutex_destroy(&data->fib_lock); +	mutex_destroy(&data->nh_lock);  	nsim_fib_debugfs_exit(data);  	kfree(data);  } diff --git a/drivers/net/netdevsim/health.c b/drivers/net/netdevsim/health.c index 21e2974660e7..04aebdf85747 100644 --- a/drivers/net/netdevsim/health.c +++ b/drivers/net/netdevsim/health.c @@ -235,15 +235,10 @@ static ssize_t nsim_dev_health_break_write(struct file *file,  	char *break_msg;  	int err; -	break_msg = kmalloc(count + 1, GFP_KERNEL); -	if (!break_msg) -		return -ENOMEM; +	break_msg = memdup_user_nul(data, count); +	if (IS_ERR(break_msg)) +		return PTR_ERR(break_msg); -	if (copy_from_user(break_msg, data, count)) { -		err = -EFAULT; -		goto out; -	} -	break_msg[count] = '\0';  	if (break_msg[count - 1] == '\n')  		break_msg[count - 1] = '\0'; diff --git a/drivers/net/netdevsim/netdevsim.h b/drivers/net/netdevsim/netdevsim.h index 48163c5f2ec9..7ff24e03577b 100644 --- a/drivers/net/netdevsim/netdevsim.h +++ b/drivers/net/netdevsim/netdevsim.h @@ -60,9 +60,12 @@ struct nsim_ethtool_pauseparam {  };  struct nsim_ethtool { +	u32 get_err; +	u32 set_err;  	struct nsim_ethtool_pauseparam pauseparam;  	struct ethtool_coalesce coalesce;  	struct ethtool_ringparam ring; +	struct ethtool_fecparam fec;  };  struct netdevsim { @@ -180,6 +183,20 @@ struct nsim_dev_health {  int nsim_dev_health_init(struct nsim_dev *nsim_dev, struct devlink *devlink);  void nsim_dev_health_exit(struct nsim_dev *nsim_dev); +#if IS_ENABLED(CONFIG_PSAMPLE) +int nsim_dev_psample_init(struct nsim_dev *nsim_dev); +void nsim_dev_psample_exit(struct nsim_dev *nsim_dev); +#else +static inline int nsim_dev_psample_init(struct nsim_dev *nsim_dev) +{ +	return 0; +} + +static inline void nsim_dev_psample_exit(struct nsim_dev *nsim_dev) +{ +} +#endif +  struct nsim_dev_port {  	struct list_head list;  	struct devlink_port devlink_port; @@ -229,6 +246,7 @@ struct nsim_dev {  		bool static_iana_vxlan;  		u32 sleep;  	} udp_ports; +	struct nsim_dev_psample *psample;  };  static inline struct net *nsim_dev_net(struct nsim_dev *nsim_dev) diff --git a/drivers/net/netdevsim/psample.c b/drivers/net/netdevsim/psample.c new file mode 100644 index 000000000000..f0c6477dd0ae --- /dev/null +++ b/drivers/net/netdevsim/psample.c @@ -0,0 +1,265 @@ +// SPDX-License-Identifier: GPL-2.0 +/* Copyright (c) 2021 Mellanox Technologies. All rights reserved */ + +#include <linux/debugfs.h> +#include <linux/err.h> +#include <linux/etherdevice.h> +#include <linux/inet.h> +#include <linux/kernel.h> +#include <linux/random.h> +#include <linux/slab.h> +#include <net/devlink.h> +#include <net/ip.h> +#include <net/psample.h> +#include <uapi/linux/ip.h> +#include <uapi/linux/udp.h> + +#include "netdevsim.h" + +#define NSIM_PSAMPLE_REPORT_INTERVAL_MS	100 +#define NSIM_PSAMPLE_INVALID_TC		0xFFFF +#define NSIM_PSAMPLE_L4_DATA_LEN	100 + +struct nsim_dev_psample { +	struct delayed_work psample_dw; +	struct dentry *ddir; +	struct psample_group *group; +	u32 rate; +	u32 group_num; +	u32 trunc_size; +	int in_ifindex; +	int out_ifindex; +	u16 out_tc; +	u64 out_tc_occ_max; +	u64 latency_max; +	bool is_active; +}; + +static struct sk_buff *nsim_dev_psample_skb_build(void) +{ +	int tot_len, data_len = NSIM_PSAMPLE_L4_DATA_LEN; +	struct sk_buff *skb; +	struct udphdr *udph; +	struct ethhdr *eth; +	struct iphdr *iph; + +	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); +	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->id = 0; +	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_psample_md_prepare(const struct nsim_dev_psample *psample, +					struct psample_metadata *md, +					unsigned int len) +{ +	md->trunc_size = psample->trunc_size ? psample->trunc_size : len; +	md->in_ifindex = psample->in_ifindex; +	md->out_ifindex = psample->out_ifindex; + +	if (psample->out_tc != NSIM_PSAMPLE_INVALID_TC) { +		md->out_tc = psample->out_tc; +		md->out_tc_valid = 1; +	} + +	if (psample->out_tc_occ_max) { +		u64 out_tc_occ; + +		get_random_bytes(&out_tc_occ, sizeof(u64)); +		md->out_tc_occ = out_tc_occ & (psample->out_tc_occ_max - 1); +		md->out_tc_occ_valid = 1; +	} + +	if (psample->latency_max) { +		u64 latency; + +		get_random_bytes(&latency, sizeof(u64)); +		md->latency = latency & (psample->latency_max - 1); +		md->latency_valid = 1; +	} +} + +static void nsim_dev_psample_report_work(struct work_struct *work) +{ +	struct nsim_dev_psample *psample; +	struct psample_metadata md = {}; +	struct sk_buff *skb; +	unsigned long delay; + +	psample = container_of(work, struct nsim_dev_psample, psample_dw.work); + +	skb = nsim_dev_psample_skb_build(); +	if (!skb) +		goto out; + +	nsim_dev_psample_md_prepare(psample, &md, skb->len); +	psample_sample_packet(psample->group, skb, psample->rate, &md); +	consume_skb(skb); + +out: +	delay = msecs_to_jiffies(NSIM_PSAMPLE_REPORT_INTERVAL_MS); +	schedule_delayed_work(&psample->psample_dw, delay); +} + +static int nsim_dev_psample_enable(struct nsim_dev *nsim_dev) +{ +	struct nsim_dev_psample *psample = nsim_dev->psample; +	struct devlink *devlink; +	unsigned long delay; + +	if (psample->is_active) +		return -EBUSY; + +	devlink = priv_to_devlink(nsim_dev); +	psample->group = psample_group_get(devlink_net(devlink), +					   psample->group_num); +	if (!psample->group) +		return -EINVAL; + +	delay = msecs_to_jiffies(NSIM_PSAMPLE_REPORT_INTERVAL_MS); +	schedule_delayed_work(&psample->psample_dw, delay); + +	psample->is_active = true; + +	return 0; +} + +static int nsim_dev_psample_disable(struct nsim_dev *nsim_dev) +{ +	struct nsim_dev_psample *psample = nsim_dev->psample; + +	if (!psample->is_active) +		return -EINVAL; + +	psample->is_active = false; + +	cancel_delayed_work_sync(&psample->psample_dw); +	psample_group_put(psample->group); + +	return 0; +} + +static ssize_t nsim_dev_psample_enable_write(struct file *file, +					     const char __user *data, +					     size_t count, loff_t *ppos) +{ +	struct nsim_dev *nsim_dev = file->private_data; +	bool enable; +	int err; + +	err = kstrtobool_from_user(data, count, &enable); +	if (err) +		return err; + +	if (enable) +		err = nsim_dev_psample_enable(nsim_dev); +	else +		err = nsim_dev_psample_disable(nsim_dev); + +	return err ? err : count; +} + +static const struct file_operations nsim_psample_enable_fops = { +	.open = simple_open, +	.write = nsim_dev_psample_enable_write, +	.llseek = generic_file_llseek, +	.owner = THIS_MODULE, +}; + +int nsim_dev_psample_init(struct nsim_dev *nsim_dev) +{ +	struct nsim_dev_psample *psample; +	int err; + +	psample = kzalloc(sizeof(*psample), GFP_KERNEL); +	if (!psample) +		return -ENOMEM; +	nsim_dev->psample = psample; + +	INIT_DELAYED_WORK(&psample->psample_dw, nsim_dev_psample_report_work); + +	psample->ddir = debugfs_create_dir("psample", nsim_dev->ddir); +	if (IS_ERR(psample->ddir)) { +		err = PTR_ERR(psample->ddir); +		goto err_psample_free; +	} + +	/* Populate sampling parameters with sane defaults. */ +	psample->rate = 100; +	debugfs_create_u32("rate", 0600, psample->ddir, &psample->rate); + +	psample->group_num = 10; +	debugfs_create_u32("group_num", 0600, psample->ddir, +			   &psample->group_num); + +	psample->trunc_size = 0; +	debugfs_create_u32("trunc_size", 0600, psample->ddir, +			   &psample->trunc_size); + +	psample->in_ifindex = 1; +	debugfs_create_u32("in_ifindex", 0600, psample->ddir, +			   &psample->in_ifindex); + +	psample->out_ifindex = 2; +	debugfs_create_u32("out_ifindex", 0600, psample->ddir, +			   &psample->out_ifindex); + +	psample->out_tc = 0; +	debugfs_create_u16("out_tc", 0600, psample->ddir, &psample->out_tc); + +	psample->out_tc_occ_max = 10000; +	debugfs_create_u64("out_tc_occ_max", 0600, psample->ddir, +			   &psample->out_tc_occ_max); + +	psample->latency_max = 50; +	debugfs_create_u64("latency_max", 0600, psample->ddir, +			   &psample->latency_max); + +	debugfs_create_file("enable", 0200, psample->ddir, nsim_dev, +			    &nsim_psample_enable_fops); + +	return 0; + +err_psample_free: +	kfree(nsim_dev->psample); +	return err; +} + +void nsim_dev_psample_exit(struct nsim_dev *nsim_dev) +{ +	debugfs_remove_recursive(nsim_dev->psample->ddir); +	if (nsim_dev->psample->is_active) { +		cancel_delayed_work_sync(&nsim_dev->psample->psample_dw); +		psample_group_put(nsim_dev->psample->group); +	} +	kfree(nsim_dev->psample); +} | 
