On Mon, Aug 2, 2021 at 9:30 AM Leon Romanovsky <leon@xxxxxxxxxx> wrote:
>
> On Fri, Jul 30, 2021 at 03:18:31PM +0200, Jack Wang wrote:
> > From: Md Haris Iqbal <haris.iqbal@xxxxxxxxxxxxxxx>
> >
> > This commit adds support to reject connection on a specific IB port which
> > can be specified in the added sysfs entry for the rtrs-server module.
> >
> > Example,
> >
> > $ echo "mlx4_0 1" > /sys/class/rtrs-server/ctl/disable_port
> >
> > When a connection request is received on the above IB port, rtrs_srv
> > rejects the connection and notifies the client to disable reconnection
> > attempts. A manual reconnect has to be triggerred in such a case.
> >
> > A manual reconnect can be triggered by doing the following,
> >
> > echo 1 > /sys/class/rtrs-client/blya/paths/<select-path>/reconnect
> >
> > Signed-off-by: Md Haris Iqbal <haris.iqbal@xxxxxxxxxxxxxxx>
> > Reviewed-by: Gioh Kim <gi-oh.kim@xxxxxxxxx>
> > Signed-off-by: Jack Wang <jinpu.wang@xxxxxxxxx>
> > ---
> > drivers/infiniband/ulp/rtrs/rtrs-clt.c | 10 ++++
> > drivers/infiniband/ulp/rtrs/rtrs-srv.c | 82 +++++++++++++++++++++++++-
> > 2 files changed, 90 insertions(+), 2 deletions(-)
> >
> > diff --git a/drivers/infiniband/ulp/rtrs/rtrs-clt.c b/drivers/infiniband/ulp/rtrs/rtrs-clt.c
> > index 5cce727abca0..21001818e607 100644
> > --- a/drivers/infiniband/ulp/rtrs/rtrs-clt.c
> > +++ b/drivers/infiniband/ulp/rtrs/rtrs-clt.c
> > @@ -1898,6 +1898,7 @@ static int rtrs_rdma_conn_rejected(struct rtrs_clt_con *con,
> > struct rdma_cm_event *ev)
> > {
> > struct rtrs_sess *s = con->c.sess;
> > + struct rtrs_clt_sess *sess = to_clt_sess(s);
> > const struct rtrs_msg_conn_rsp *msg;
> > const char *rej_msg;
> > int status, errno;
> > @@ -1916,6 +1917,15 @@ static int rtrs_rdma_conn_rejected(struct rtrs_clt_con *con,
> > rtrs_err(s,
> > "Connect rejected: status %d (%s), rtrs errno %d\n",
> > status, rej_msg, errno);
> > +
> > + if (errno == -ENETDOWN) {
> > + /*
> > + * Stop reconnection attempts
> > + */
> > + sess->reconnect_attempts = -1;
> > + rtrs_err(s,
> > + "Disabling auto-reconnect. Trigger a manual reconnect after issue is resolved\n");
> > + }
> > } else {
> > rtrs_err(s,
> > "Connect rejected but with malformed message: status %d (%s)\n",
> > diff --git a/drivers/infiniband/ulp/rtrs/rtrs-srv.c b/drivers/infiniband/ulp/rtrs/rtrs-srv.c
> > index cc65cffdc65a..90d833041ccf 100644
> > --- a/drivers/infiniband/ulp/rtrs/rtrs-srv.c
> > +++ b/drivers/infiniband/ulp/rtrs/rtrs-srv.c
> > @@ -32,7 +32,9 @@ MODULE_LICENSE("GPL");
> > static struct rtrs_rdma_dev_pd dev_pd;
> > static mempool_t *chunk_pool;
> > struct class *rtrs_dev_class;
> > +static struct device *rtrs_dev;
> > static struct rtrs_srv_ib_ctx ib_ctx;
> > +static char disabled_port[NAME_MAX];
> >
> > static int __read_mostly max_chunk_size = DEFAULT_MAX_CHUNK_SIZE;
> > static int __read_mostly sess_queue_depth = DEFAULT_SESS_QUEUE_DEPTH;
> > @@ -1826,6 +1828,20 @@ static int rtrs_rdma_connect(struct rdma_cm_id *cm_id,
> > u16 recon_cnt;
> > int err = -ECONNRESET;
> >
> > + if (strnlen(disabled_port, NAME_MAX) > 0) {
> > + char ib_device[NAME_MAX];
> > +
> > + snprintf(ib_device, NAME_MAX, "%s %u", cm_id->device->name, cm_id->port_num);
> > + if (strncmp(disabled_port, ib_device, NAME_MAX) == 0) {
> > + /*
> > + * Reject connection attempt on disabled port
> > + */
> > + pr_err("Error: Connection request on a disabled port");
> > + err = -ENETDOWN;
> > + goto reject_w_err;
> > + }
> > + }
> > +
> > if (len < sizeof(*msg)) {
> > pr_err("Invalid RTRS connection request\n");
> > goto reject_w_err;
> > @@ -2242,6 +2258,56 @@ static int check_module_params(void)
> > return 0;
> > }
> >
> > +static ssize_t disable_port_show(struct kobject *kobj,
> > + struct kobj_attribute *attr,
> > + char *page)
> > +{
> > + return sysfs_emit(page, "%s\n", disabled_port);
> > +}
> > +
> > +static ssize_t disable_port_store(struct kobject *kobj,
> > + struct kobj_attribute *attr,
> > + const char *buf, size_t count)
> > +{
> > + int ret, len;
> > +
> > + if (count > 1 && strnlen(disabled_port, NAME_MAX) > 0) {
> > + pr_err("A disabled port already exists.\n");
> > + return -EINVAL;
> > + }
> > +
> > + ret = strscpy(disabled_port, buf, NAME_MAX);
> > + if (ret == -E2BIG) {
> > + pr_err("String too big.\n");
> > + disabled_port[0] = '\0';
> > + return ret;
> > + }
> > +
> > + len = strlen(disabled_port);
> > + if (len > 0 && disabled_port[len-1] == '\n')
>
> All the "\n" checks in rtrs sysfs looks like cargo cult. You don't need
> them.
Thanks. Will change and resend.
>
> And maybe Jason thinks differently, but I don't feel comfortable with
> such new sysfs file at all.
>
> Thanks
