On Thu, Jun 06, 2019 at 01:53:33PM +0300, Leon Romanovsky wrote: > From: Mark Zhang <markz@xxxxxxxxxxxx> > > Add an API to support set/clear per-port auto mode. > > Signed-off-by: Mark Zhang <markz@xxxxxxxxxxxx> > Reviewed-by: Majd Dibbiny <majd@xxxxxxxxxxxx> > Signed-off-by: Leon Romanovsky <leonro@xxxxxxxxxxxx> > --- > drivers/infiniband/core/Makefile | 2 +- > drivers/infiniband/core/counters.c | 74 ++++++++++++++++++++++++++++++ > drivers/infiniband/core/device.c | 7 ++- > include/rdma/ib_verbs.h | 2 + > include/rdma/rdma_counter.h | 24 ++++++++++ > include/uapi/rdma/rdma_netlink.h | 26 +++++++++++ > 6 files changed, 133 insertions(+), 2 deletions(-) > create mode 100644 drivers/infiniband/core/counters.c > > diff --git a/drivers/infiniband/core/Makefile b/drivers/infiniband/core/Makefile > index 313f2349b518..cddf748c15c9 100644 > --- a/drivers/infiniband/core/Makefile > +++ b/drivers/infiniband/core/Makefile > @@ -12,7 +12,7 @@ ib_core-y := packer.o ud_header.o verbs.o cq.o rw.o sysfs.o \ > device.o fmr_pool.o cache.o netlink.o \ > roce_gid_mgmt.o mr_pool.o addr.o sa_query.o \ > multicast.o mad.o smi.o agent.o mad_rmpp.o \ > - nldev.o restrack.o > + nldev.o restrack.o counters.o > > ib_core-$(CONFIG_SECURITY_INFINIBAND) += security.o > ib_core-$(CONFIG_CGROUP_RDMA) += cgroup.o > diff --git a/drivers/infiniband/core/counters.c b/drivers/infiniband/core/counters.c > new file mode 100644 > index 000000000000..6cae8bf14997 > --- /dev/null > +++ b/drivers/infiniband/core/counters.c > @@ -0,0 +1,74 @@ > +// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB > +/* > + * Copyright (c) 2019 Mellanox Technologies. All rights reserved. > + */ > +#include <rdma/ib_verbs.h> > +#include <rdma/rdma_counter.h> > + > +#include "core_priv.h" > +#include "restrack.h" > + > +#define ALL_AUTO_MODE_MASKS (RDMA_COUNTER_MASK_QP_TYPE) > + > +static int __counter_set_mode(struct rdma_counter_mode *curr, > + enum rdma_nl_counter_mode new_mode, > + enum rdma_nl_counter_mask new_mask) > +{ > + if ((new_mode == RDMA_COUNTER_MODE_AUTO) && > + ((new_mask & (~ALL_AUTO_MODE_MASKS)) || > + (curr->mode != RDMA_COUNTER_MODE_NONE))) > + return -EINVAL; > + > + curr->mode = new_mode; > + curr->mask = new_mask; > + return 0; > +} > + > +/** > + * rdma_counter_set_auto_mode() - Turn on/off per-port auto mode > + * > + * When @on is true, the @mask must be set > + */ > +int rdma_counter_set_auto_mode(struct ib_device *dev, u8 port, > + bool on, enum rdma_nl_counter_mask mask) > +{ > + struct rdma_port_counter *port_counter; > + int ret; > + > + port_counter = &dev->port_data[port].port_counter; > + mutex_lock(&port_counter->lock); > + if (on) { > + ret = __counter_set_mode(&port_counter->mode, > + RDMA_COUNTER_MODE_AUTO, mask); > + } else { > + if (port_counter->mode.mode != RDMA_COUNTER_MODE_AUTO) { > + ret = -EINVAL; > + goto out; > + } > + ret = __counter_set_mode(&port_counter->mode, > + RDMA_COUNTER_MODE_NONE, 0); > + } > + > +out: > + mutex_unlock(&port_counter->lock); > + return ret; > +} > + > +void rdma_counter_init(struct ib_device *dev) > +{ > + struct rdma_port_counter *port_counter; > + u32 port; > + > + if (!dev->port_data) It should be "if (!dev->ops.alloc_hw_stats || !dev->port_data)", otherwise mlx4 in SR-IOV crashes while loading mlx4_ib. Thanks