net/core/devlink.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * net/core/devlink.c - Network physical/parent device Netlink interface
   4  *
   5  * Heavily inspired by net/wireless/
   6  * Copyright (c) 2016 Mellanox Technologies. All rights reserved.
   7  * Copyright (c) 2016 Jiri Pirko <jiri@mellanox.com>
   8  */
   9
  10 #include <linux/kernel.h>
  11 #include <linux/module.h>
  12 #include <linux/types.h>
  13 #include <linux/slab.h>
  14 #include <linux/gfp.h>
  15 #include <linux/device.h>
  16 #include <linux/list.h>
  17 #include <linux/netdevice.h>
  18 #include <linux/spinlock.h>
  19 #include <linux/refcount.h>
  20 #include <linux/workqueue.h>
  21 #include <linux/u64_stats_sync.h>
  22 #include <linux/timekeeping.h>
  23 #include <rdma/ib_verbs.h>
  24 #include <net/netlink.h>
  25 #include <net/genetlink.h>
  26 #include <net/rtnetlink.h>
  27 #include <net/net_namespace.h>
  28 #include <net/sock.h>
  29 #include <net/devlink.h>
  30 #define CREATE_TRACE_POINTS
  31 #include <trace/events/devlink.h>
  32
  33 static struct devlink_dpipe_field devlink_dpipe_fields_ethernet[] = {
  34         {
  35                 .name = "destination mac",
  36                 .id = DEVLINK_DPIPE_FIELD_ETHERNET_DST_MAC,
  37                 .bitwidth = 48,
  38         },
  39 };
  40
  41 struct devlink_dpipe_header devlink_dpipe_header_ethernet = {
  42         .name = "ethernet",
  43         .id = DEVLINK_DPIPE_HEADER_ETHERNET,
  44         .fields = devlink_dpipe_fields_ethernet,
  45         .fields_count = ARRAY_SIZE(devlink_dpipe_fields_ethernet),
  46         .global = true,
  47 };
  48 EXPORT_SYMBOL(devlink_dpipe_header_ethernet);
  49
  50 static struct devlink_dpipe_field devlink_dpipe_fields_ipv4[] = {
  51         {
  52                 .name = "destination ip",
  53                 .id = DEVLINK_DPIPE_FIELD_IPV4_DST_IP,
  54                 .bitwidth = 32,
  55         },
  56 };
  57
  58 struct devlink_dpipe_header devlink_dpipe_header_ipv4 = {
  59         .name = "ipv4",
  60         .id = DEVLINK_DPIPE_HEADER_IPV4,
  61         .fields = devlink_dpipe_fields_ipv4,
  62         .fields_count = ARRAY_SIZE(devlink_dpipe_fields_ipv4),
  63         .global = true,
  64 };
  65 EXPORT_SYMBOL(devlink_dpipe_header_ipv4);
  66
  67 static struct devlink_dpipe_field devlink_dpipe_fields_ipv6[] = {
  68         {
  69                 .name = "destination ip",
  70                 .id = DEVLINK_DPIPE_FIELD_IPV6_DST_IP,
  71                 .bitwidth = 128,
  72         },
  73 };
  74
  75 struct devlink_dpipe_header devlink_dpipe_header_ipv6 = {
  76         .name = "ipv6",
  77         .id = DEVLINK_DPIPE_HEADER_IPV6,
  78         .fields = devlink_dpipe_fields_ipv6,
  79         .fields_count = ARRAY_SIZE(devlink_dpipe_fields_ipv6),
  80         .global = true,
  81 };
  82 EXPORT_SYMBOL(devlink_dpipe_header_ipv6);
  83
  84 EXPORT_TRACEPOINT_SYMBOL_GPL(devlink_hwmsg);
  85 EXPORT_TRACEPOINT_SYMBOL_GPL(devlink_hwerr);
  86 EXPORT_TRACEPOINT_SYMBOL_GPL(devlink_trap_report);
  87
  88 static const struct nla_policy devlink_function_nl_policy[DEVLINK_PORT_FUNCTION_ATTR_MAX + 1] = {
  89         [DEVLINK_PORT_FUNCTION_ATTR_HW_ADDR] = { .type = NLA_BINARY },
  90 };
  91
  92 static LIST_HEAD(devlink_list);
  93
  94 /* devlink_mutex
  95  *
  96  * An overall lock guarding every operation coming from userspace.
  97  * It also guards devlink devices list and it is taken when
  98  * driver registers/unregisters it.
  99  */
 100 static DEFINE_MUTEX(devlink_mutex);
 101
 102 struct net *devlink_net(const struct devlink *devlink)
 103 {
 104         return read_pnet(&devlink->_net);
 105 }
 106 EXPORT_SYMBOL_GPL(devlink_net);
 107
 108 static void __devlink_net_set(struct devlink *devlink, struct net *net)
 109 {
 110         write_pnet(&devlink->_net, net);
 111 }
 112
 113 void devlink_net_set(struct devlink *devlink, struct net *net)
 114 {
 115         if (WARN_ON(devlink->registered))
 116                 return;
 117         __devlink_net_set(devlink, net);
 118 }
 119 EXPORT_SYMBOL_GPL(devlink_net_set);
 120
 121 static struct devlink *devlink_get_from_attrs(struct net *net,
 122                                               struct nlattr **attrs)
 123 {
 124         struct devlink *devlink;
 125         char *busname;
 126         char *devname;
 127
 128         if (!attrs[DEVLINK_ATTR_BUS_NAME] || !attrs[DEVLINK_ATTR_DEV_NAME])
 129                 return ERR_PTR(-EINVAL);
 130
 131         busname = nla_data(attrs[DEVLINK_ATTR_BUS_NAME]);
 132         devname = nla_data(attrs[DEVLINK_ATTR_DEV_NAME]);
 133
 134         lockdep_assert_held(&devlink_mutex);
 135
 136         list_for_each_entry(devlink, &devlink_list, list) {
 137                 if (strcmp(devlink->dev->bus->name, busname) == 0 &&
 138                     strcmp(dev_name(devlink->dev), devname) == 0 &&
 139                     net_eq(devlink_net(devlink), net))
 140                         return devlink;
 141         }
 142
 143         return ERR_PTR(-ENODEV);
 144 }
 145
 146 static struct devlink *devlink_get_from_info(struct genl_info *info)
 147 {
 148         return devlink_get_from_attrs(genl_info_net(info), info->attrs);
 149 }
 150
 151 static struct devlink_port *devlink_port_get_by_index(struct devlink *devlink,
 152                                                       unsigned int port_index)
 153 {
 154         struct devlink_port *devlink_port;
 155
 156         list_for_each_entry(devlink_port, &devlink->port_list, list) {
 157                 if (devlink_port->index == port_index)
 158                         return devlink_port;
 159         }
 160         return NULL;
 161 }
 162
 163 static bool devlink_port_index_exists(struct devlink *devlink,
 164                                       unsigned int port_index)
 165 {
 166         return devlink_port_get_by_index(devlink, port_index);
 167 }
 168
 169 static struct devlink_port *devlink_port_get_from_attrs(struct devlink *devlink,
 170                                                         struct nlattr **attrs)
 171 {
 172         if (attrs[DEVLINK_ATTR_PORT_INDEX]) {
 173                 u32 port_index = nla_get_u32(attrs[DEVLINK_ATTR_PORT_INDEX]);
 174                 struct devlink_port *devlink_port;
 175
 176                 devlink_port = devlink_port_get_by_index(devlink, port_index);
 177                 if (!devlink_port)
 178                         return ERR_PTR(-ENODEV);
 179                 return devlink_port;
 180         }
 181         return ERR_PTR(-EINVAL);
 182 }
 183
 184 static struct devlink_port *devlink_port_get_from_info(struct devlink *devlink,
 185                                                        struct genl_info *info)
 186 {
 187         return devlink_port_get_from_attrs(devlink, info->attrs);
 188 }
 189
 190 struct devlink_sb {
 191         struct list_head list;
 192         unsigned int index;
 193         u32 size;
 194         u16 ingress_pools_count;
 195         u16 egress_pools_count;
 196         u16 ingress_tc_count;
 197         u16 egress_tc_count;
 198 };
 199
 200 static u16 devlink_sb_pool_count(struct devlink_sb *devlink_sb)
 201 {
 202         return devlink_sb->ingress_pools_count + devlink_sb->egress_pools_count;
 203 }
 204
 205 static struct devlink_sb *devlink_sb_get_by_index(struct devlink *devlink,
 206                                                   unsigned int sb_index)
 207 {
 208         struct devlink_sb *devlink_sb;
 209
 210         list_for_each_entry(devlink_sb, &devlink->sb_list, list) {
 211                 if (devlink_sb->index == sb_index)
 212                         return devlink_sb;
 213         }
 214         return NULL;
 215 }
 216
 217 static bool devlink_sb_index_exists(struct devlink *devlink,
 218                                     unsigned int sb_index)
 219 {
 220         return devlink_sb_get_by_index(devlink, sb_index);
 221 }
 222
 223 static struct devlink_sb *devlink_sb_get_from_attrs(struct devlink *devlink,
 224                                                     struct nlattr **attrs)
 225 {
 226         if (attrs[DEVLINK_ATTR_SB_INDEX]) {
 227                 u32 sb_index = nla_get_u32(attrs[DEVLINK_ATTR_SB_INDEX]);
 228                 struct devlink_sb *devlink_sb;
 229
 230                 devlink_sb = devlink_sb_get_by_index(devlink, sb_index);
 231                 if (!devlink_sb)
 232                         return ERR_PTR(-ENODEV);
 233                 return devlink_sb;
 234         }
 235         return ERR_PTR(-EINVAL);
 236 }
 237
 238 static struct devlink_sb *devlink_sb_get_from_info(struct devlink *devlink,
 239                                                    struct genl_info *info)
 240 {
 241         return devlink_sb_get_from_attrs(devlink, info->attrs);
 242 }
 243
 244 static int devlink_sb_pool_index_get_from_attrs(struct devlink_sb *devlink_sb,
 245                                                 struct nlattr **attrs,
 246                                                 u16 *p_pool_index)
 247 {
 248         u16 val;
 249
 250         if (!attrs[DEVLINK_ATTR_SB_POOL_INDEX])
 251                 return -EINVAL;
 252
 253         val = nla_get_u16(attrs[DEVLINK_ATTR_SB_POOL_INDEX]);
 254         if (val >= devlink_sb_pool_count(devlink_sb))
 255                 return -EINVAL;
 256         *p_pool_index = val;
 257         return 0;
 258 }
 259
 260 static int devlink_sb_pool_index_get_from_info(struct devlink_sb *devlink_sb,
 261                                                struct genl_info *info,
 262                                                u16 *p_pool_index)
 263 {
 264         return devlink_sb_pool_index_get_from_attrs(devlink_sb, info->attrs,
 265                                                     p_pool_index);
 266 }
 267
 268 static int
 269 devlink_sb_pool_type_get_from_attrs(struct nlattr **attrs,
 270                                     enum devlink_sb_pool_type *p_pool_type)
 271 {
 272         u8 val;
 273
 274         if (!attrs[DEVLINK_ATTR_SB_POOL_TYPE])
 275                 return -EINVAL;
 276
 277         val = nla_get_u8(attrs[DEVLINK_ATTR_SB_POOL_TYPE]);
 278         if (val != DEVLINK_SB_POOL_TYPE_INGRESS &&
 279             val != DEVLINK_SB_POOL_TYPE_EGRESS)
 280                 return -EINVAL;
 281         *p_pool_type = val;
 282         return 0;
 283 }
 284
 285 static int
 286 devlink_sb_pool_type_get_from_info(struct genl_info *info,
 287                                    enum devlink_sb_pool_type *p_pool_type)
 288 {
 289         return devlink_sb_pool_type_get_from_attrs(info->attrs, p_pool_type);
 290 }
 291
 292 static int
 293 devlink_sb_th_type_get_from_attrs(struct nlattr **attrs,
 294                                   enum devlink_sb_threshold_type *p_th_type)
 295 {
 296         u8 val;
 297
 298         if (!attrs[DEVLINK_ATTR_SB_POOL_THRESHOLD_TYPE])
 299                 return -EINVAL;
 300
 301         val = nla_get_u8(attrs[DEVLINK_ATTR_SB_POOL_THRESHOLD_TYPE]);
 302         if (val != DEVLINK_SB_THRESHOLD_TYPE_STATIC &&
 303             val != DEVLINK_SB_THRESHOLD_TYPE_DYNAMIC)
 304                 return -EINVAL;
 305         *p_th_type = val;
 306         return 0;
 307 }
 308
 309 static int
 310 devlink_sb_th_type_get_from_info(struct genl_info *info,
 311                                  enum devlink_sb_threshold_type *p_th_type)
 312 {
 313         return devlink_sb_th_type_get_from_attrs(info->attrs, p_th_type);
 314 }
 315
 316 static int
 317 devlink_sb_tc_index_get_from_attrs(struct devlink_sb *devlink_sb,
 318                                    struct nlattr **attrs,
 319                                    enum devlink_sb_pool_type pool_type,
 320                                    u16 *p_tc_index)
 321 {
 322         u16 val;
 323
 324         if (!attrs[DEVLINK_ATTR_SB_TC_INDEX])
 325                 return -EINVAL;
 326
 327         val = nla_get_u16(attrs[DEVLINK_ATTR_SB_TC_INDEX]);
 328         if (pool_type == DEVLINK_SB_POOL_TYPE_INGRESS &&
 329             val >= devlink_sb->ingress_tc_count)
 330                 return -EINVAL;
 331         if (pool_type == DEVLINK_SB_POOL_TYPE_EGRESS &&
 332             val >= devlink_sb->egress_tc_count)
 333                 return -EINVAL;
 334         *p_tc_index = val;
 335         return 0;
 336 }
 337
 338 static int
 339 devlink_sb_tc_index_get_from_info(struct devlink_sb *devlink_sb,
 340                                   struct genl_info *info,
 341                                   enum devlink_sb_pool_type pool_type,
 342                                   u16 *p_tc_index)
 343 {
 344         return devlink_sb_tc_index_get_from_attrs(devlink_sb, info->attrs,
 345                                                   pool_type, p_tc_index);
 346 }
 347
 348 struct devlink_region {
 349         struct devlink *devlink;
 350         struct list_head list;
 351         const struct devlink_region_ops *ops;
 352         struct list_head snapshot_list;
 353         u32 max_snapshots;
 354         u32 cur_snapshots;
 355         u64 size;
 356 };
 357
 358 struct devlink_snapshot {
 359         struct list_head list;
 360         struct devlink_region *region;
 361         u8 *data;
 362         u32 id;
 363 };
 364
 365 static struct devlink_region *
 366 devlink_region_get_by_name(struct devlink *devlink, const char *region_name)
 367 {
 368         struct devlink_region *region;
 369
 370         list_for_each_entry(region, &devlink->region_list, list)
 371                 if (!strcmp(region->ops->name, region_name))
 372                         return region;
 373
 374         return NULL;
 375 }
 376
 377 static struct devlink_snapshot *
 378 devlink_region_snapshot_get_by_id(struct devlink_region *region, u32 id)
 379 {
 380         struct devlink_snapshot *snapshot;
 381
 382         list_for_each_entry(snapshot, &region->snapshot_list, list)
 383                 if (snapshot->id == id)
 384                         return snapshot;
 385
 386         return NULL;
 387 }
 388
 389 #define DEVLINK_NL_FLAG_NEED_PORT               BIT(0)
 390 #define DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT    BIT(1)
 391
 392 /* The per devlink instance lock is taken by default in the pre-doit
 393  * operation, yet several commands do not require this. The global
 394  * devlink lock is taken and protects from disruption by user-calls.
 395  */
 396 #define DEVLINK_NL_FLAG_NO_LOCK                 BIT(2)
 397
 398 static int devlink_nl_pre_doit(const struct genl_ops *ops,
 399                                struct sk_buff *skb, struct genl_info *info)
 400 {
 401         struct devlink_port *devlink_port;
 402         struct devlink *devlink;
 403         int err;
 404
 405         mutex_lock(&devlink_mutex);
 406         devlink = devlink_get_from_info(info);
 407         if (IS_ERR(devlink)) {
 408                 mutex_unlock(&devlink_mutex);
 409                 return PTR_ERR(devlink);
 410         }
 411         if (~ops->internal_flags & DEVLINK_NL_FLAG_NO_LOCK)
 412                 mutex_lock(&devlink->lock);
 413         info->user_ptr[0] = devlink;
 414         if (ops->internal_flags & DEVLINK_NL_FLAG_NEED_PORT) {
 415                 devlink_port = devlink_port_get_from_info(devlink, info);
 416                 if (IS_ERR(devlink_port)) {
 417                         err = PTR_ERR(devlink_port);
 418                         goto unlock;
 419                 }
 420                 info->user_ptr[1] = devlink_port;
 421         } else if (ops->internal_flags & DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT) {
 422                 devlink_port = devlink_port_get_from_info(devlink, info);
 423                 if (!IS_ERR(devlink_port))
 424                         info->user_ptr[1] = devlink_port;
 425         }
 426         return 0;
 427
 428 unlock:
 429         if (~ops->internal_flags & DEVLINK_NL_FLAG_NO_LOCK)
 430                 mutex_unlock(&devlink->lock);
 431         mutex_unlock(&devlink_mutex);
 432         return err;
 433 }
 434
 435 static void devlink_nl_post_doit(const struct genl_ops *ops,
 436                                  struct sk_buff *skb, struct genl_info *info)
 437 {
 438         struct devlink *devlink;
 439
 440         devlink = info->user_ptr[0];
 441         if (~ops->internal_flags & DEVLINK_NL_FLAG_NO_LOCK)
 442                 mutex_unlock(&devlink->lock);
 443         mutex_unlock(&devlink_mutex);
 444 }
 445
 446 static struct genl_family devlink_nl_family;
 447
 448 enum devlink_multicast_groups {
 449         DEVLINK_MCGRP_CONFIG,
 450 };
 451
 452 static const struct genl_multicast_group devlink_nl_mcgrps[] = {
 453         [DEVLINK_MCGRP_CONFIG] = { .name = DEVLINK_GENL_MCGRP_CONFIG_NAME },
 454 };
 455
 456 static int devlink_nl_put_handle(struct sk_buff *msg, struct devlink *devlink)
 457 {
 458         if (nla_put_string(msg, DEVLINK_ATTR_BUS_NAME, devlink->dev->bus->name))
 459                 return -EMSGSIZE;
 460         if (nla_put_string(msg, DEVLINK_ATTR_DEV_NAME, dev_name(devlink->dev)))
 461                 return -EMSGSIZE;
 462         return 0;
 463 }
 464
 465 static int devlink_nl_fill(struct sk_buff *msg, struct devlink *devlink,
 466                            enum devlink_command cmd, u32 portid,
 467                            u32 seq, int flags)
 468 {
 469         void *hdr;
 470
 471         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
 472         if (!hdr)
 473                 return -EMSGSIZE;
 474
 475         if (devlink_nl_put_handle(msg, devlink))
 476                 goto nla_put_failure;
 477         if (nla_put_u8(msg, DEVLINK_ATTR_RELOAD_FAILED, devlink->reload_failed))
 478                 goto nla_put_failure;
 479
 480         genlmsg_end(msg, hdr);
 481         return 0;
 482
 483 nla_put_failure:
 484         genlmsg_cancel(msg, hdr);
 485         return -EMSGSIZE;
 486 }
 487
 488 static void devlink_notify(struct devlink *devlink, enum devlink_command cmd)
 489 {
 490         struct sk_buff *msg;
 491         int err;
 492
 493         WARN_ON(cmd != DEVLINK_CMD_NEW && cmd != DEVLINK_CMD_DEL);
 494
 495         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 496         if (!msg)
 497                 return;
 498
 499         err = devlink_nl_fill(msg, devlink, cmd, 0, 0, 0);
 500         if (err) {
 501                 nlmsg_free(msg);
 502                 return;
 503         }
 504
 505         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
 506                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
 507 }
 508
 509 static int devlink_nl_port_attrs_put(struct sk_buff *msg,
 510                                      struct devlink_port *devlink_port)
 511 {
 512         struct devlink_port_attrs *attrs = &devlink_port->attrs;
 513
 514         if (!devlink_port->attrs_set)
 515                 return 0;
 516         if (attrs->lanes) {
 517                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_LANES, attrs->lanes))
 518                         return -EMSGSIZE;
 519         }
 520         if (nla_put_u8(msg, DEVLINK_ATTR_PORT_SPLITTABLE, attrs->splittable))
 521                 return -EMSGSIZE;
 522         if (nla_put_u16(msg, DEVLINK_ATTR_PORT_FLAVOUR, attrs->flavour))
 523                 return -EMSGSIZE;
 524         switch (devlink_port->attrs.flavour) {
 525         case DEVLINK_PORT_FLAVOUR_PCI_PF:
 526                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_CONTROLLER_NUMBER,
 527                                 attrs->pci_pf.controller) ||
 528                     nla_put_u16(msg, DEVLINK_ATTR_PORT_PCI_PF_NUMBER, attrs->pci_pf.pf))
 529                         return -EMSGSIZE;
 530                 if (nla_put_u8(msg, DEVLINK_ATTR_PORT_EXTERNAL, attrs->pci_pf.external))
 531                         return -EMSGSIZE;
 532                 break;
 533         case DEVLINK_PORT_FLAVOUR_PCI_VF:
 534                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_CONTROLLER_NUMBER,
 535                                 attrs->pci_vf.controller) ||
 536                     nla_put_u16(msg, DEVLINK_ATTR_PORT_PCI_PF_NUMBER, attrs->pci_vf.pf) ||
 537                     nla_put_u16(msg, DEVLINK_ATTR_PORT_PCI_VF_NUMBER, attrs->pci_vf.vf))
 538                         return -EMSGSIZE;
 539                 if (nla_put_u8(msg, DEVLINK_ATTR_PORT_EXTERNAL, attrs->pci_vf.external))
 540                         return -EMSGSIZE;
 541                 break;
 542         case DEVLINK_PORT_FLAVOUR_PHYSICAL:
 543         case DEVLINK_PORT_FLAVOUR_CPU:
 544         case DEVLINK_PORT_FLAVOUR_DSA:
 545         case DEVLINK_PORT_FLAVOUR_VIRTUAL:
 546                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_NUMBER,
 547                                 attrs->phys.port_number))
 548                         return -EMSGSIZE;
 549                 if (!attrs->split)
 550                         return 0;
 551                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_SPLIT_GROUP,
 552                                 attrs->phys.port_number))
 553                         return -EMSGSIZE;
 554                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_SPLIT_SUBPORT_NUMBER,
 555                                 attrs->phys.split_subport_number))
 556                         return -EMSGSIZE;
 557                 break;
 558         default:
 559                 break;
 560         }
 561         return 0;
 562 }
 563
 564 static int
 565 devlink_nl_port_function_attrs_put(struct sk_buff *msg, struct devlink_port *port,
 566                                    struct netlink_ext_ack *extack)
 567 {
 568         struct devlink *devlink = port->devlink;
 569         const struct devlink_ops *ops;
 570         struct nlattr *function_attr;
 571         bool empty_nest = true;
 572         int err = 0;
 573
 574         function_attr = nla_nest_start_noflag(msg, DEVLINK_ATTR_PORT_FUNCTION);
 575         if (!function_attr)
 576                 return -EMSGSIZE;
 577
 578         ops = devlink->ops;
 579         if (ops->port_function_hw_addr_get) {
 580                 int hw_addr_len;
 581                 u8 hw_addr[MAX_ADDR_LEN];
 582
 583                 err = ops->port_function_hw_addr_get(devlink, port, hw_addr, &hw_addr_len, extack);
 584                 if (err == -EOPNOTSUPP) {
 585                         /* Port function attributes are optional for a port. If port doesn't
 586                          * support function attribute, returning -EOPNOTSUPP is not an error.
 587                          */
 588                         err = 0;
 589                         goto out;
 590                 } else if (err) {
 591                         goto out;
 592                 }
 593                 err = nla_put(msg, DEVLINK_PORT_FUNCTION_ATTR_HW_ADDR, hw_addr_len, hw_addr);
 594                 if (err)
 595                         goto out;
 596                 empty_nest = false;
 597         }
 598
 599 out:
 600         if (err || empty_nest)
 601                 nla_nest_cancel(msg, function_attr);
 602         else
 603                 nla_nest_end(msg, function_attr);
 604         return err;
 605 }
 606
 607 static int devlink_nl_port_fill(struct sk_buff *msg, struct devlink *devlink,
 608                                 struct devlink_port *devlink_port,
 609                                 enum devlink_command cmd, u32 portid,
 610                                 u32 seq, int flags,
 611                                 struct netlink_ext_ack *extack)
 612 {
 613         void *hdr;
 614
 615         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
 616         if (!hdr)
 617                 return -EMSGSIZE;
 618
 619         if (devlink_nl_put_handle(msg, devlink))
 620                 goto nla_put_failure;
 621         if (nla_put_u32(msg, DEVLINK_ATTR_PORT_INDEX, devlink_port->index))
 622                 goto nla_put_failure;
 623
 624         spin_lock_bh(&devlink_port->type_lock);
 625         if (nla_put_u16(msg, DEVLINK_ATTR_PORT_TYPE, devlink_port->type))
 626                 goto nla_put_failure_type_locked;
 627         if (devlink_port->desired_type != DEVLINK_PORT_TYPE_NOTSET &&
 628             nla_put_u16(msg, DEVLINK_ATTR_PORT_DESIRED_TYPE,
 629                         devlink_port->desired_type))
 630                 goto nla_put_failure_type_locked;
 631         if (devlink_port->type == DEVLINK_PORT_TYPE_ETH) {
 632                 struct net_device *netdev = devlink_port->type_dev;
 633
 634                 if (netdev &&
 635                     (nla_put_u32(msg, DEVLINK_ATTR_PORT_NETDEV_IFINDEX,
 636                                  netdev->ifindex) ||
 637                      nla_put_string(msg, DEVLINK_ATTR_PORT_NETDEV_NAME,
 638                                     netdev->name)))
 639                         goto nla_put_failure_type_locked;
 640         }
 641         if (devlink_port->type == DEVLINK_PORT_TYPE_IB) {
 642                 struct ib_device *ibdev = devlink_port->type_dev;
 643
 644                 if (ibdev &&
 645                     nla_put_string(msg, DEVLINK_ATTR_PORT_IBDEV_NAME,
 646                                    ibdev->name))
 647                         goto nla_put_failure_type_locked;
 648         }
 649         spin_unlock_bh(&devlink_port->type_lock);
 650         if (devlink_nl_port_attrs_put(msg, devlink_port))
 651                 goto nla_put_failure;
 652         if (devlink_nl_port_function_attrs_put(msg, devlink_port, extack))
 653                 goto nla_put_failure;
 654
 655         genlmsg_end(msg, hdr);
 656         return 0;
 657
 658 nla_put_failure_type_locked:
 659         spin_unlock_bh(&devlink_port->type_lock);
 660 nla_put_failure:
 661         genlmsg_cancel(msg, hdr);
 662         return -EMSGSIZE;
 663 }
 664
 665 static void devlink_port_notify(struct devlink_port *devlink_port,
 666                                 enum devlink_command cmd)
 667 {
 668         struct devlink *devlink = devlink_port->devlink;
 669         struct sk_buff *msg;
 670         int err;
 671
 672         if (!devlink_port->registered)
 673                 return;
 674
 675         WARN_ON(cmd != DEVLINK_CMD_PORT_NEW && cmd != DEVLINK_CMD_PORT_DEL);
 676
 677         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 678         if (!msg)
 679                 return;
 680
 681         err = devlink_nl_port_fill(msg, devlink, devlink_port, cmd, 0, 0, 0,
 682                                    NULL);
 683         if (err) {
 684                 nlmsg_free(msg);
 685                 return;
 686         }
 687
 688         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
 689                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
 690 }
 691
 692 static int devlink_nl_cmd_get_doit(struct sk_buff *skb, struct genl_info *info)
 693 {
 694         struct devlink *devlink = info->user_ptr[0];
 695         struct sk_buff *msg;
 696         int err;
 697
 698         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 699         if (!msg)
 700                 return -ENOMEM;
 701
 702         err = devlink_nl_fill(msg, devlink, DEVLINK_CMD_NEW,
 703                               info->snd_portid, info->snd_seq, 0);
 704         if (err) {
 705                 nlmsg_free(msg);
 706                 return err;
 707         }
 708
 709         return genlmsg_reply(msg, info);
 710 }
 711
 712 static int devlink_nl_cmd_get_dumpit(struct sk_buff *msg,
 713                                      struct netlink_callback *cb)
 714 {
 715         struct devlink *devlink;
 716         int start = cb->args[0];
 717         int idx = 0;
 718         int err;
 719
 720         mutex_lock(&devlink_mutex);
 721         list_for_each_entry(devlink, &devlink_list, list) {
 722                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
 723                         continue;
 724                 if (idx < start) {
 725                         idx++;
 726                         continue;
 727                 }
 728                 err = devlink_nl_fill(msg, devlink, DEVLINK_CMD_NEW,
 729                                       NETLINK_CB(cb->skb).portid,
 730                                       cb->nlh->nlmsg_seq, NLM_F_MULTI);
 731                 if (err)
 732                         goto out;
 733                 idx++;
 734         }
 735 out:
 736         mutex_unlock(&devlink_mutex);
 737
 738         cb->args[0] = idx;
 739         return msg->len;
 740 }
 741
 742 static int devlink_nl_cmd_port_get_doit(struct sk_buff *skb,
 743                                         struct genl_info *info)
 744 {
 745         struct devlink_port *devlink_port = info->user_ptr[1];
 746         struct devlink *devlink = devlink_port->devlink;
 747         struct sk_buff *msg;
 748         int err;
 749
 750         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 751         if (!msg)
 752                 return -ENOMEM;
 753
 754         err = devlink_nl_port_fill(msg, devlink, devlink_port,
 755                                    DEVLINK_CMD_PORT_NEW,
 756                                    info->snd_portid, info->snd_seq, 0,
 757                                    info->extack);
 758         if (err) {
 759                 nlmsg_free(msg);
 760                 return err;
 761         }
 762
 763         return genlmsg_reply(msg, info);
 764 }
 765
 766 static int devlink_nl_cmd_port_get_dumpit(struct sk_buff *msg,
 767                                           struct netlink_callback *cb)
 768 {
 769         struct devlink *devlink;
 770         struct devlink_port *devlink_port;
 771         int start = cb->args[0];
 772         int idx = 0;
 773         int err;
 774
 775         mutex_lock(&devlink_mutex);
 776         list_for_each_entry(devlink, &devlink_list, list) {
 777                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
 778                         continue;
 779                 mutex_lock(&devlink->lock);
 780                 list_for_each_entry(devlink_port, &devlink->port_list, list) {
 781                         if (idx < start) {
 782                                 idx++;
 783                                 continue;
 784                         }
 785                         err = devlink_nl_port_fill(msg, devlink, devlink_port,
 786                                                    DEVLINK_CMD_NEW,
 787                                                    NETLINK_CB(cb->skb).portid,
 788                                                    cb->nlh->nlmsg_seq,
 789                                                    NLM_F_MULTI,
 790                                                    cb->extack);
 791                         if (err) {
 792                                 mutex_unlock(&devlink->lock);
 793                                 goto out;
 794                         }
 795                         idx++;
 796                 }
 797                 mutex_unlock(&devlink->lock);
 798         }
 799 out:
 800         mutex_unlock(&devlink_mutex);
 801
 802         cb->args[0] = idx;
 803         return msg->len;
 804 }
 805
 806 static int devlink_port_type_set(struct devlink *devlink,
 807                                  struct devlink_port *devlink_port,
 808                                  enum devlink_port_type port_type)
 809
 810 {
 811         int err;
 812
 813         if (devlink->ops->port_type_set) {
 814                 if (port_type == devlink_port->type)
 815                         return 0;
 816                 err = devlink->ops->port_type_set(devlink_port, port_type);
 817                 if (err)
 818                         return err;
 819                 devlink_port->desired_type = port_type;
 820                 devlink_port_notify(devlink_port, DEVLINK_CMD_PORT_NEW);
 821                 return 0;
 822         }
 823         return -EOPNOTSUPP;
 824 }
 825
 826 static int
 827 devlink_port_function_hw_addr_set(struct devlink *devlink, struct devlink_port *port,
 828                                   const struct nlattr *attr, struct netlink_ext_ack *extack)
 829 {
 830         const struct devlink_ops *ops;
 831         const u8 *hw_addr;
 832         int hw_addr_len;
 833         int err;
 834
 835         hw_addr = nla_data(attr);
 836         hw_addr_len = nla_len(attr);
 837         if (hw_addr_len > MAX_ADDR_LEN) {
 838                 NL_SET_ERR_MSG_MOD(extack, "Port function hardware address too long");
 839                 return -EINVAL;
 840         }
 841         if (port->type == DEVLINK_PORT_TYPE_ETH) {
 842                 if (hw_addr_len != ETH_ALEN) {
 843                         NL_SET_ERR_MSG_MOD(extack, "Address must be 6 bytes for Ethernet device");
 844                         return -EINVAL;
 845                 }
 846                 if (!is_unicast_ether_addr(hw_addr)) {
 847                         NL_SET_ERR_MSG_MOD(extack, "Non-unicast hardware address unsupported");
 848                         return -EINVAL;
 849                 }
 850         }
 851
 852         ops = devlink->ops;
 853         if (!ops->port_function_hw_addr_set) {
 854                 NL_SET_ERR_MSG_MOD(extack, "Port doesn't support function attributes");
 855                 return -EOPNOTSUPP;
 856         }
 857
 858         err = ops->port_function_hw_addr_set(devlink, port, hw_addr, hw_addr_len, extack);
 859         if (err)
 860                 return err;
 861
 862         devlink_port_notify(port, DEVLINK_CMD_PORT_NEW);
 863         return 0;
 864 }
 865
 866 static int
 867 devlink_port_function_set(struct devlink *devlink, struct devlink_port *port,
 868                           const struct nlattr *attr, struct netlink_ext_ack *extack)
 869 {
 870         struct nlattr *tb[DEVLINK_PORT_FUNCTION_ATTR_MAX + 1];
 871         int err;
 872
 873         err = nla_parse_nested(tb, DEVLINK_PORT_FUNCTION_ATTR_MAX, attr,
 874                                devlink_function_nl_policy, extack);
 875         if (err < 0) {
 876                 NL_SET_ERR_MSG_MOD(extack, "Fail to parse port function attributes");
 877                 return err;
 878         }
 879
 880         attr = tb[DEVLINK_PORT_FUNCTION_ATTR_HW_ADDR];
 881         if (attr)
 882                 err = devlink_port_function_hw_addr_set(devlink, port, attr, extack);
 883
 884         return err;
 885 }
 886
 887 static int devlink_nl_cmd_port_set_doit(struct sk_buff *skb,
 888                                         struct genl_info *info)
 889 {
 890         struct devlink_port *devlink_port = info->user_ptr[1];
 891         struct devlink *devlink = devlink_port->devlink;
 892         int err;
 893
 894         if (info->attrs[DEVLINK_ATTR_PORT_TYPE]) {
 895                 enum devlink_port_type port_type;
 896
 897                 port_type = nla_get_u16(info->attrs[DEVLINK_ATTR_PORT_TYPE]);
 898                 err = devlink_port_type_set(devlink, devlink_port, port_type);
 899                 if (err)
 900                         return err;
 901         }
 902
 903         if (info->attrs[DEVLINK_ATTR_PORT_FUNCTION]) {
 904                 struct nlattr *attr = info->attrs[DEVLINK_ATTR_PORT_FUNCTION];
 905                 struct netlink_ext_ack *extack = info->extack;
 906
 907                 err = devlink_port_function_set(devlink, devlink_port, attr, extack);
 908                 if (err)
 909                         return err;
 910         }
 911
 912         return 0;
 913 }
 914
 915 static int devlink_port_split(struct devlink *devlink, u32 port_index,
 916                               u32 count, struct netlink_ext_ack *extack)
 917
 918 {
 919         if (devlink->ops->port_split)
 920                 return devlink->ops->port_split(devlink, port_index, count,
 921                                                 extack);
 922         return -EOPNOTSUPP;
 923 }
 924
 925 static int devlink_nl_cmd_port_split_doit(struct sk_buff *skb,
 926                                           struct genl_info *info)
 927 {
 928         struct devlink *devlink = info->user_ptr[0];
 929         struct devlink_port *devlink_port;
 930         u32 port_index;
 931         u32 count;
 932
 933         if (!info->attrs[DEVLINK_ATTR_PORT_INDEX] ||
 934             !info->attrs[DEVLINK_ATTR_PORT_SPLIT_COUNT])
 935                 return -EINVAL;
 936
 937         devlink_port = devlink_port_get_from_info(devlink, info);
 938         port_index = nla_get_u32(info->attrs[DEVLINK_ATTR_PORT_INDEX]);
 939         count = nla_get_u32(info->attrs[DEVLINK_ATTR_PORT_SPLIT_COUNT]);
 940
 941         if (IS_ERR(devlink_port))
 942                 return -EINVAL;
 943
 944         if (!devlink_port->attrs.splittable) {
 945                 /* Split ports cannot be split. */
 946                 if (devlink_port->attrs.split)
 947                         NL_SET_ERR_MSG_MOD(info->extack, "Port cannot be split further");
 948                 else
 949                         NL_SET_ERR_MSG_MOD(info->extack, "Port cannot be split");
 950                 return -EINVAL;
 951         }
 952
 953         if (count < 2 || !is_power_of_2(count) || count > devlink_port->attrs.lanes) {
 954                 NL_SET_ERR_MSG_MOD(info->extack, "Invalid split count");
 955                 return -EINVAL;
 956         }
 957
 958         return devlink_port_split(devlink, port_index, count, info->extack);
 959 }
 960
 961 static int devlink_port_unsplit(struct devlink *devlink, u32 port_index,
 962                                 struct netlink_ext_ack *extack)
 963
 964 {
 965         if (devlink->ops->port_unsplit)
 966                 return devlink->ops->port_unsplit(devlink, port_index, extack);
 967         return -EOPNOTSUPP;
 968 }
 969
 970 static int devlink_nl_cmd_port_unsplit_doit(struct sk_buff *skb,
 971                                             struct genl_info *info)
 972 {
 973         struct devlink *devlink = info->user_ptr[0];
 974         u32 port_index;
 975
 976         if (!info->attrs[DEVLINK_ATTR_PORT_INDEX])
 977                 return -EINVAL;
 978
 979         port_index = nla_get_u32(info->attrs[DEVLINK_ATTR_PORT_INDEX]);
 980         return devlink_port_unsplit(devlink, port_index, info->extack);
 981 }
 982
 983 static int devlink_nl_sb_fill(struct sk_buff *msg, struct devlink *devlink,
 984                               struct devlink_sb *devlink_sb,
 985                               enum devlink_command cmd, u32 portid,
 986                               u32 seq, int flags)
 987 {
 988         void *hdr;
 989
 990         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
 991         if (!hdr)
 992                 return -EMSGSIZE;
 993
 994         if (devlink_nl_put_handle(msg, devlink))
 995                 goto nla_put_failure;
 996         if (nla_put_u32(msg, DEVLINK_ATTR_SB_INDEX, devlink_sb->index))
 997                 goto nla_put_failure;
 998         if (nla_put_u32(msg, DEVLINK_ATTR_SB_SIZE, devlink_sb->size))
 999                 goto nla_put_failure;
1000         if (nla_put_u16(msg, DEVLINK_ATTR_SB_INGRESS_POOL_COUNT,
1001                         devlink_sb->ingress_pools_count))
1002                 goto nla_put_failure;
1003         if (nla_put_u16(msg, DEVLINK_ATTR_SB_EGRESS_POOL_COUNT,
1004                         devlink_sb->egress_pools_count))
1005                 goto nla_put_failure;
1006         if (nla_put_u16(msg, DEVLINK_ATTR_SB_INGRESS_TC_COUNT,
1007                         devlink_sb->ingress_tc_count))
1008                 goto nla_put_failure;
1009         if (nla_put_u16(msg, DEVLINK_ATTR_SB_EGRESS_TC_COUNT,
1010                         devlink_sb->egress_tc_count))
1011                 goto nla_put_failure;
1012
1013         genlmsg_end(msg, hdr);
1014         return 0;
1015
1016 nla_put_failure:
1017         genlmsg_cancel(msg, hdr);
1018         return -EMSGSIZE;
1019 }
1020
1021 static int devlink_nl_cmd_sb_get_doit(struct sk_buff *skb,
1022                                       struct genl_info *info)
1023 {
1024         struct devlink *devlink = info->user_ptr[0];
1025         struct devlink_sb *devlink_sb;
1026         struct sk_buff *msg;
1027         int err;
1028
1029         devlink_sb = devlink_sb_get_from_info(devlink, info);
1030         if (IS_ERR(devlink_sb))
1031                 return PTR_ERR(devlink_sb);
1032
1033         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1034         if (!msg)
1035                 return -ENOMEM;
1036
1037         err = devlink_nl_sb_fill(msg, devlink, devlink_sb,
1038                                  DEVLINK_CMD_SB_NEW,
1039                                  info->snd_portid, info->snd_seq, 0);
1040         if (err) {
1041                 nlmsg_free(msg);
1042                 return err;
1043         }
1044
1045         return genlmsg_reply(msg, info);
1046 }
1047
1048 static int devlink_nl_cmd_sb_get_dumpit(struct sk_buff *msg,
1049                                         struct netlink_callback *cb)
1050 {
1051         struct devlink *devlink;
1052         struct devlink_sb *devlink_sb;
1053         int start = cb->args[0];
1054         int idx = 0;
1055         int err;
1056
1057         mutex_lock(&devlink_mutex);
1058         list_for_each_entry(devlink, &devlink_list, list) {
1059                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
1060                         continue;
1061                 mutex_lock(&devlink->lock);
1062                 list_for_each_entry(devlink_sb, &devlink->sb_list, list) {
1063                         if (idx < start) {
1064                                 idx++;
1065                                 continue;
1066                         }
1067                         err = devlink_nl_sb_fill(msg, devlink, devlink_sb,
1068                                                  DEVLINK_CMD_SB_NEW,
1069                                                  NETLINK_CB(cb->skb).portid,
1070                                                  cb->nlh->nlmsg_seq,
1071                                                  NLM_F_MULTI);
1072                         if (err) {
1073                                 mutex_unlock(&devlink->lock);
1074                                 goto out;
1075                         }
1076                         idx++;
1077                 }
1078                 mutex_unlock(&devlink->lock);
1079         }
1080 out:
1081         mutex_unlock(&devlink_mutex);
1082
1083         cb->args[0] = idx;
1084         return msg->len;
1085 }
1086
1087 static int devlink_nl_sb_pool_fill(struct sk_buff *msg, struct devlink *devlink,
1088                                    struct devlink_sb *devlink_sb,
1089                                    u16 pool_index, enum devlink_command cmd,
1090                                    u32 portid, u32 seq, int flags)
1091 {
1092         struct devlink_sb_pool_info pool_info;
1093         void *hdr;
1094         int err;
1095
1096         err = devlink->ops->sb_pool_get(devlink, devlink_sb->index,
1097                                         pool_index, &pool_info);
1098         if (err)
1099                 return err;
1100
1101         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
1102         if (!hdr)
1103                 return -EMSGSIZE;
1104
1105         if (devlink_nl_put_handle(msg, devlink))
1106                 goto nla_put_failure;
1107         if (nla_put_u32(msg, DEVLINK_ATTR_SB_INDEX, devlink_sb->index))
1108                 goto nla_put_failure;
1109         if (nla_put_u16(msg, DEVLINK_ATTR_SB_POOL_INDEX, pool_index))
1110                 goto nla_put_failure;
1111         if (nla_put_u8(msg, DEVLINK_ATTR_SB_POOL_TYPE, pool_info.pool_type))
1112                 goto nla_put_failure;
1113         if (nla_put_u32(msg, DEVLINK_ATTR_SB_POOL_SIZE, pool_info.size))
1114                 goto nla_put_failure;
1115         if (nla_put_u8(msg, DEVLINK_ATTR_SB_POOL_THRESHOLD_TYPE,
1116                        pool_info.threshold_type))
1117                 goto nla_put_failure;
1118         if (nla_put_u32(msg, DEVLINK_ATTR_SB_POOL_CELL_SIZE,
1119                         pool_info.cell_size))
1120                 goto nla_put_failure;
1121
1122         genlmsg_end(msg, hdr);
1123         return 0;
1124
1125 nla_put_failure:
1126         genlmsg_cancel(msg, hdr);
1127         return -EMSGSIZE;
1128 }
1129
1130 static int devlink_nl_cmd_sb_pool_get_doit(struct sk_buff *skb,
1131                                            struct genl_info *info)
1132 {
1133         struct devlink *devlink = info->user_ptr[0];
1134         struct devlink_sb *devlink_sb;
1135         struct sk_buff *msg;
1136         u16 pool_index;
1137         int err;
1138
1139         devlink_sb = devlink_sb_get_from_info(devlink, info);
1140         if (IS_ERR(devlink_sb))
1141                 return PTR_ERR(devlink_sb);
1142
1143         err = devlink_sb_pool_index_get_from_info(devlink_sb, info,
1144                                                   &pool_index);
1145         if (err)
1146                 return err;
1147
1148         if (!devlink->ops->sb_pool_get)
1149                 return -EOPNOTSUPP;
1150
1151         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1152         if (!msg)
1153                 return -ENOMEM;
1154
1155         err = devlink_nl_sb_pool_fill(msg, devlink, devlink_sb, pool_index,
1156                                       DEVLINK_CMD_SB_POOL_NEW,
1157                                       info->snd_portid, info->snd_seq, 0);
1158         if (err) {
1159                 nlmsg_free(msg);
1160                 return err;
1161         }
1162
1163         return genlmsg_reply(msg, info);
1164 }
1165
1166 static int __sb_pool_get_dumpit(struct sk_buff *msg, int start, int *p_idx,
1167                                 struct devlink *devlink,
1168                                 struct devlink_sb *devlink_sb,
1169                                 u32 portid, u32 seq)
1170 {
1171         u16 pool_count = devlink_sb_pool_count(devlink_sb);
1172         u16 pool_index;
1173         int err;
1174
1175         for (pool_index = 0; pool_index < pool_count; pool_index++) {
1176                 if (*p_idx < start) {
1177                         (*p_idx)++;
1178                         continue;
1179                 }
1180                 err = devlink_nl_sb_pool_fill(msg, devlink,
1181                                               devlink_sb,
1182                                               pool_index,
1183                                               DEVLINK_CMD_SB_POOL_NEW,
1184                                               portid, seq, NLM_F_MULTI);
1185                 if (err)
1186                         return err;
1187                 (*p_idx)++;
1188         }
1189         return 0;
1190 }
1191
1192 static int devlink_nl_cmd_sb_pool_get_dumpit(struct sk_buff *msg,
1193                                              struct netlink_callback *cb)
1194 {
1195         struct devlink *devlink;
1196         struct devlink_sb *devlink_sb;
1197         int start = cb->args[0];
1198         int idx = 0;
1199         int err = 0;
1200
1201         mutex_lock(&devlink_mutex);
1202         list_for_each_entry(devlink, &devlink_list, list) {
1203                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)) ||
1204                     !devlink->ops->sb_pool_get)
1205                         continue;
1206                 mutex_lock(&devlink->lock);
1207                 list_for_each_entry(devlink_sb, &devlink->sb_list, list) {
1208                         err = __sb_pool_get_dumpit(msg, start, &idx, devlink,
1209                                                    devlink_sb,
1210                                                    NETLINK_CB(cb->skb).portid,
1211                                                    cb->nlh->nlmsg_seq);
1212                         if (err == -EOPNOTSUPP) {
1213                                 err = 0;
1214                         } else if (err) {
1215                                 mutex_unlock(&devlink->lock);
1216                                 goto out;
1217                         }
1218                 }
1219                 mutex_unlock(&devlink->lock);
1220         }
1221 out:
1222         mutex_unlock(&devlink_mutex);
1223
1224         if (err != -EMSGSIZE)
1225                 return err;
1226
1227         cb->args[0] = idx;
1228         return msg->len;
1229 }
1230
1231 static int devlink_sb_pool_set(struct devlink *devlink, unsigned int sb_index,
1232                                u16 pool_index, u32 size,
1233                                enum devlink_sb_threshold_type threshold_type,
1234                                struct netlink_ext_ack *extack)
1235
1236 {
1237         const struct devlink_ops *ops = devlink->ops;
1238
1239         if (ops->sb_pool_set)
1240                 return ops->sb_pool_set(devlink, sb_index, pool_index,
1241                                         size, threshold_type, extack);
1242         return -EOPNOTSUPP;
1243 }
1244
1245 static int devlink_nl_cmd_sb_pool_set_doit(struct sk_buff *skb,
1246                                            struct genl_info *info)
1247 {
1248         struct devlink *devlink = info->user_ptr[0];
1249         enum devlink_sb_threshold_type threshold_type;
1250         struct devlink_sb *devlink_sb;
1251         u16 pool_index;
1252         u32 size;
1253         int err;
1254
1255         devlink_sb = devlink_sb_get_from_info(devlink, info);
1256         if (IS_ERR(devlink_sb))
1257                 return PTR_ERR(devlink_sb);
1258
1259         err = devlink_sb_pool_index_get_from_info(devlink_sb, info,
1260                                                   &pool_index);
1261         if (err)
1262                 return err;
1263
1264         err = devlink_sb_th_type_get_from_info(info, &threshold_type);
1265         if (err)
1266                 return err;
1267
1268         if (!info->attrs[DEVLINK_ATTR_SB_POOL_SIZE])
1269                 return -EINVAL;
1270
1271         size = nla_get_u32(info->attrs[DEVLINK_ATTR_SB_POOL_SIZE]);
1272         return devlink_sb_pool_set(devlink, devlink_sb->index,
1273                                    pool_index, size, threshold_type,
1274                                    info->extack);
1275 }
1276
1277 static int devlink_nl_sb_port_pool_fill(struct sk_buff *msg,
1278                                         struct devlink *devlink,
1279                                         struct devlink_port *devlink_port,
1280                                         struct devlink_sb *devlink_sb,
1281                                         u16 pool_index,
1282                                         enum devlink_command cmd,
1283                                         u32 portid, u32 seq, int flags)
1284 {
1285         const struct devlink_ops *ops = devlink->ops;
1286         u32 threshold;
1287         void *hdr;
1288         int err;
1289
1290         err = ops->sb_port_pool_get(devlink_port, devlink_sb->index,
1291                                     pool_index, &threshold);
1292         if (err)
1293                 return err;
1294
1295         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
1296         if (!hdr)
1297                 return -EMSGSIZE;
1298
1299         if (devlink_nl_put_handle(msg, devlink))
1300                 goto nla_put_failure;
1301         if (nla_put_u32(msg, DEVLINK_ATTR_PORT_INDEX, devlink_port->index))
1302                 goto nla_put_failure;
1303         if (nla_put_u32(msg, DEVLINK_ATTR_SB_INDEX, devlink_sb->index))
1304                 goto nla_put_failure;
1305         if (nla_put_u16(msg, DEVLINK_ATTR_SB_POOL_INDEX, pool_index))
1306                 goto nla_put_failure;
1307         if (nla_put_u32(msg, DEVLINK_ATTR_SB_THRESHOLD, threshold))
1308                 goto nla_put_failure;
1309
1310         if (ops->sb_occ_port_pool_get) {
1311                 u32 cur;
1312                 u32 max;
1313
1314                 err = ops->sb_occ_port_pool_get(devlink_port, devlink_sb->index,
1315                                                 pool_index, &cur, &max);
1316                 if (err && err != -EOPNOTSUPP)
1317                         return err;
1318                 if (!err) {
1319                         if (nla_put_u32(msg, DEVLINK_ATTR_SB_OCC_CUR, cur))
1320                                 goto nla_put_failure;
1321                         if (nla_put_u32(msg, DEVLINK_ATTR_SB_OCC_MAX, max))
1322                                 goto nla_put_failure;
1323                 }
1324         }
1325
1326         genlmsg_end(msg, hdr);
1327         return 0;
1328
1329 nla_put_failure:
1330         genlmsg_cancel(msg, hdr);
1331         return -EMSGSIZE;
1332 }
1333
1334 static int devlink_nl_cmd_sb_port_pool_get_doit(struct sk_buff *skb,
1335                                                 struct genl_info *info)
1336 {
1337         struct devlink_port *devlink_port = info->user_ptr[1];
1338         struct devlink *devlink = devlink_port->devlink;
1339         struct devlink_sb *devlink_sb;
1340         struct sk_buff *msg;
1341         u16 pool_index;
1342         int err;
1343
1344         devlink_sb = devlink_sb_get_from_info(devlink, info);
1345         if (IS_ERR(devlink_sb))
1346                 return PTR_ERR(devlink_sb);
1347
1348         err = devlink_sb_pool_index_get_from_info(devlink_sb, info,
1349                                                   &pool_index);
1350         if (err)
1351                 return err;
1352
1353         if (!devlink->ops->sb_port_pool_get)
1354                 return -EOPNOTSUPP;
1355
1356         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1357         if (!msg)
1358                 return -ENOMEM;
1359
1360         err = devlink_nl_sb_port_pool_fill(msg, devlink, devlink_port,
1361                                            devlink_sb, pool_index,
1362                                            DEVLINK_CMD_SB_PORT_POOL_NEW,
1363                                            info->snd_portid, info->snd_seq, 0);
1364         if (err) {
1365                 nlmsg_free(msg);
1366                 return err;
1367         }
1368
1369         return genlmsg_reply(msg, info);
1370 }
1371
1372 static int __sb_port_pool_get_dumpit(struct sk_buff *msg, int start, int *p_idx,
1373                                      struct devlink *devlink,
1374                                      struct devlink_sb *devlink_sb,
1375                                      u32 portid, u32 seq)
1376 {
1377         struct devlink_port *devlink_port;
1378         u16 pool_count = devlink_sb_pool_count(devlink_sb);
1379         u16 pool_index;
1380         int err;
1381
1382         list_for_each_entry(devlink_port, &devlink->port_list, list) {
1383                 for (pool_index = 0; pool_index < pool_count; pool_index++) {
1384                         if (*p_idx < start) {
1385                                 (*p_idx)++;
1386                                 continue;
1387                         }
1388                         err = devlink_nl_sb_port_pool_fill(msg, devlink,
1389                                                            devlink_port,
1390                                                            devlink_sb,
1391                                                            pool_index,
1392                                                            DEVLINK_CMD_SB_PORT_POOL_NEW,
1393                                                            portid, seq,
1394                                                            NLM_F_MULTI);
1395                         if (err)
1396                                 return err;
1397                         (*p_idx)++;
1398                 }
1399         }
1400         return 0;
1401 }
1402
1403 static int devlink_nl_cmd_sb_port_pool_get_dumpit(struct sk_buff *msg,
1404                                                   struct netlink_callback *cb)
1405 {
1406         struct devlink *devlink;
1407         struct devlink_sb *devlink_sb;
1408         int start = cb->args[0];
1409         int idx = 0;
1410         int err = 0;
1411
1412         mutex_lock(&devlink_mutex);
1413         list_for_each_entry(devlink, &devlink_list, list) {
1414                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)) ||
1415                     !devlink->ops->sb_port_pool_get)
1416                         continue;
1417                 mutex_lock(&devlink->lock);
1418                 list_for_each_entry(devlink_sb, &devlink->sb_list, list) {
1419                         err = __sb_port_pool_get_dumpit(msg, start, &idx,
1420                                                         devlink, devlink_sb,
1421                                                         NETLINK_CB(cb->skb).portid,
1422                                                         cb->nlh->nlmsg_seq);
1423                         if (err == -EOPNOTSUPP) {
1424                                 err = 0;
1425                         } else if (err) {
1426                                 mutex_unlock(&devlink->lock);
1427                                 goto out;
1428                         }
1429                 }
1430                 mutex_unlock(&devlink->lock);
1431         }
1432 out:
1433         mutex_unlock(&devlink_mutex);
1434
1435         if (err != -EMSGSIZE)
1436                 return err;
1437
1438         cb->args[0] = idx;
1439         return msg->len;
1440 }
1441
1442 static int devlink_sb_port_pool_set(struct devlink_port *devlink_port,
1443                                     unsigned int sb_index, u16 pool_index,
1444                                     u32 threshold,
1445                                     struct netlink_ext_ack *extack)
1446
1447 {
1448         const struct devlink_ops *ops = devlink_port->devlink->ops;
1449
1450         if (ops->sb_port_pool_set)
1451                 return ops->sb_port_pool_set(devlink_port, sb_index,
1452                                              pool_index, threshold, extack);
1453         return -EOPNOTSUPP;
1454 }
1455
1456 static int devlink_nl_cmd_sb_port_pool_set_doit(struct sk_buff *skb,
1457                                                 struct genl_info *info)
1458 {
1459         struct devlink_port *devlink_port = info->user_ptr[1];
1460         struct devlink *devlink = info->user_ptr[0];
1461         struct devlink_sb *devlink_sb;
1462         u16 pool_index;
1463         u32 threshold;
1464         int err;
1465
1466         devlink_sb = devlink_sb_get_from_info(devlink, info);
1467         if (IS_ERR(devlink_sb))
1468                 return PTR_ERR(devlink_sb);
1469
1470         err = devlink_sb_pool_index_get_from_info(devlink_sb, info,
1471                                                   &pool_index);
1472         if (err)
1473                 return err;
1474
1475         if (!info->attrs[DEVLINK_ATTR_SB_THRESHOLD])
1476                 return -EINVAL;
1477
1478         threshold = nla_get_u32(info->attrs[DEVLINK_ATTR_SB_THRESHOLD]);
1479         return devlink_sb_port_pool_set(devlink_port, devlink_sb->index,
1480                                         pool_index, threshold, info->extack);
1481 }
1482
1483 static int
1484 devlink_nl_sb_tc_pool_bind_fill(struct sk_buff *msg, struct devlink *devlink,
1485                                 struct devlink_port *devlink_port,
1486                                 struct devlink_sb *devlink_sb, u16 tc_index,
1487                                 enum devlink_sb_pool_type pool_type,
1488                                 enum devlink_command cmd,
1489                                 u32 portid, u32 seq, int flags)
1490 {
1491         const struct devlink_ops *ops = devlink->ops;
1492         u16 pool_index;
1493         u32 threshold;
1494         void *hdr;
1495         int err;
1496
1497         err = ops->sb_tc_pool_bind_get(devlink_port, devlink_sb->index,
1498                                        tc_index, pool_type,
1499                                        &pool_index, &threshold);
1500         if (err)
1501                 return err;
1502
1503         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
1504         if (!hdr)
1505                 return -EMSGSIZE;
1506
1507         if (devlink_nl_put_handle(msg, devlink))
1508                 goto nla_put_failure;
1509         if (nla_put_u32(msg, DEVLINK_ATTR_PORT_INDEX, devlink_port->index))
1510                 goto nla_put_failure;
1511         if (nla_put_u32(msg, DEVLINK_ATTR_SB_INDEX, devlink_sb->index))
1512                 goto nla_put_failure;
1513         if (nla_put_u16(msg, DEVLINK_ATTR_SB_TC_INDEX, tc_index))
1514                 goto nla_put_failure;
1515         if (nla_put_u8(msg, DEVLINK_ATTR_SB_POOL_TYPE, pool_type))
1516                 goto nla_put_failure;
1517         if (nla_put_u16(msg, DEVLINK_ATTR_SB_POOL_INDEX, pool_index))
1518                 goto nla_put_failure;
1519         if (nla_put_u32(msg, DEVLINK_ATTR_SB_THRESHOLD, threshold))
1520                 goto nla_put_failure;
1521
1522         if (ops->sb_occ_tc_port_bind_get) {
1523                 u32 cur;
1524                 u32 max;
1525
1526                 err = ops->sb_occ_tc_port_bind_get(devlink_port,
1527                                                    devlink_sb->index,
1528                                                    tc_index, pool_type,
1529                                                    &cur, &max);
1530                 if (err && err != -EOPNOTSUPP)
1531                         return err;
1532                 if (!err) {
1533                         if (nla_put_u32(msg, DEVLINK_ATTR_SB_OCC_CUR, cur))
1534                                 goto nla_put_failure;
1535                         if (nla_put_u32(msg, DEVLINK_ATTR_SB_OCC_MAX, max))
1536                                 goto nla_put_failure;
1537                 }
1538         }
1539
1540         genlmsg_end(msg, hdr);
1541         return 0;
1542
1543 nla_put_failure:
1544         genlmsg_cancel(msg, hdr);
1545         return -EMSGSIZE;
1546 }
1547
1548 static int devlink_nl_cmd_sb_tc_pool_bind_get_doit(struct sk_buff *skb,
1549                                                    struct genl_info *info)
1550 {
1551         struct devlink_port *devlink_port = info->user_ptr[1];
1552         struct devlink *devlink = devlink_port->devlink;
1553         struct devlink_sb *devlink_sb;
1554         struct sk_buff *msg;
1555         enum devlink_sb_pool_type pool_type;
1556         u16 tc_index;
1557         int err;
1558
1559         devlink_sb = devlink_sb_get_from_info(devlink, info);
1560         if (IS_ERR(devlink_sb))
1561                 return PTR_ERR(devlink_sb);
1562
1563         err = devlink_sb_pool_type_get_from_info(info, &pool_type);
1564         if (err)
1565                 return err;
1566
1567         err = devlink_sb_tc_index_get_from_info(devlink_sb, info,
1568                                                 pool_type, &tc_index);
1569         if (err)
1570                 return err;
1571
1572         if (!devlink->ops->sb_tc_pool_bind_get)
1573                 return -EOPNOTSUPP;
1574
1575         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1576         if (!msg)
1577                 return -ENOMEM;
1578
1579         err = devlink_nl_sb_tc_pool_bind_fill(msg, devlink, devlink_port,
1580                                               devlink_sb, tc_index, pool_type,
1581                                               DEVLINK_CMD_SB_TC_POOL_BIND_NEW,
1582                                               info->snd_portid,
1583                                               info->snd_seq, 0);
1584         if (err) {
1585                 nlmsg_free(msg);
1586                 return err;
1587         }
1588
1589         return genlmsg_reply(msg, info);
1590 }
1591
1592 static int __sb_tc_pool_bind_get_dumpit(struct sk_buff *msg,
1593                                         int start, int *p_idx,
1594                                         struct devlink *devlink,
1595                                         struct devlink_sb *devlink_sb,
1596                                         u32 portid, u32 seq)
1597 {
1598         struct devlink_port *devlink_port;
1599         u16 tc_index;
1600         int err;
1601
1602         list_for_each_entry(devlink_port, &devlink->port_list, list) {
1603                 for (tc_index = 0;
1604                      tc_index < devlink_sb->ingress_tc_count; tc_index++) {
1605                         if (*p_idx < start) {
1606                                 (*p_idx)++;
1607                                 continue;
1608                         }
1609                         err = devlink_nl_sb_tc_pool_bind_fill(msg, devlink,
1610                                                               devlink_port,
1611                                                               devlink_sb,
1612                                                               tc_index,
1613                                                               DEVLINK_SB_POOL_TYPE_INGRESS,
1614                                                               DEVLINK_CMD_SB_TC_POOL_BIND_NEW,
1615                                                               portid, seq,
1616                                                               NLM_F_MULTI);
1617                         if (err)
1618                                 return err;
1619                         (*p_idx)++;
1620                 }
1621                 for (tc_index = 0;
1622                      tc_index < devlink_sb->egress_tc_count; tc_index++) {
1623                         if (*p_idx < start) {
1624                                 (*p_idx)++;
1625                                 continue;
1626                         }
1627                         err = devlink_nl_sb_tc_pool_bind_fill(msg, devlink,
1628                                                               devlink_port,
1629                                                               devlink_sb,
1630                                                               tc_index,
1631                                                               DEVLINK_SB_POOL_TYPE_EGRESS,
1632                                                               DEVLINK_CMD_SB_TC_POOL_BIND_NEW,
1633                                                               portid, seq,
1634                                                               NLM_F_MULTI);
1635                         if (err)
1636                                 return err;
1637                         (*p_idx)++;
1638                 }
1639         }
1640         return 0;
1641 }
1642
1643 static int
1644 devlink_nl_cmd_sb_tc_pool_bind_get_dumpit(struct sk_buff *msg,
1645                                           struct netlink_callback *cb)
1646 {
1647         struct devlink *devlink;
1648         struct devlink_sb *devlink_sb;
1649         int start = cb->args[0];
1650         int idx = 0;
1651         int err = 0;
1652
1653         mutex_lock(&devlink_mutex);
1654         list_for_each_entry(devlink, &devlink_list, list) {
1655                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)) ||
1656                     !devlink->ops->sb_tc_pool_bind_get)
1657                         continue;
1658
1659                 mutex_lock(&devlink->lock);
1660                 list_for_each_entry(devlink_sb, &devlink->sb_list, list) {
1661                         err = __sb_tc_pool_bind_get_dumpit(msg, start, &idx,
1662                                                            devlink,
1663                                                            devlink_sb,
1664                                                            NETLINK_CB(cb->skb).portid,
1665                                                            cb->nlh->nlmsg_seq);
1666                         if (err == -EOPNOTSUPP) {
1667                                 err = 0;
1668                         } else if (err) {
1669                                 mutex_unlock(&devlink->lock);
1670                                 goto out;
1671                         }
1672                 }
1673                 mutex_unlock(&devlink->lock);
1674         }
1675 out:
1676         mutex_unlock(&devlink_mutex);
1677
1678         if (err != -EMSGSIZE)
1679                 return err;
1680
1681         cb->args[0] = idx;
1682         return msg->len;
1683 }
1684
1685 static int devlink_sb_tc_pool_bind_set(struct devlink_port *devlink_port,
1686                                        unsigned int sb_index, u16 tc_index,
1687                                        enum devlink_sb_pool_type pool_type,
1688                                        u16 pool_index, u32 threshold,
1689                                        struct netlink_ext_ack *extack)
1690
1691 {
1692         const struct devlink_ops *ops = devlink_port->devlink->ops;
1693
1694         if (ops->sb_tc_pool_bind_set)
1695                 return ops->sb_tc_pool_bind_set(devlink_port, sb_index,
1696                                                 tc_index, pool_type,
1697                                                 pool_index, threshold, extack);
1698         return -EOPNOTSUPP;
1699 }
1700
1701 static int devlink_nl_cmd_sb_tc_pool_bind_set_doit(struct sk_buff *skb,
1702                                                    struct genl_info *info)
1703 {
1704         struct devlink_port *devlink_port = info->user_ptr[1];
1705         struct devlink *devlink = info->user_ptr[0];
1706         enum devlink_sb_pool_type pool_type;
1707         struct devlink_sb *devlink_sb;
1708         u16 tc_index;
1709         u16 pool_index;
1710         u32 threshold;
1711         int err;
1712
1713         devlink_sb = devlink_sb_get_from_info(devlink, info);
1714         if (IS_ERR(devlink_sb))
1715                 return PTR_ERR(devlink_sb);
1716
1717         err = devlink_sb_pool_type_get_from_info(info, &pool_type);
1718         if (err)
1719                 return err;
1720
1721         err = devlink_sb_tc_index_get_from_info(devlink_sb, info,
1722                                                 pool_type, &tc_index);
1723         if (err)
1724                 return err;
1725
1726         err = devlink_sb_pool_index_get_from_info(devlink_sb, info,
1727                                                   &pool_index);
1728         if (err)
1729                 return err;
1730
1731         if (!info->attrs[DEVLINK_ATTR_SB_THRESHOLD])
1732                 return -EINVAL;
1733
1734         threshold = nla_get_u32(info->attrs[DEVLINK_ATTR_SB_THRESHOLD]);
1735         return devlink_sb_tc_pool_bind_set(devlink_port, devlink_sb->index,
1736                                            tc_index, pool_type,
1737                                            pool_index, threshold, info->extack);
1738 }
1739
1740 static int devlink_nl_cmd_sb_occ_snapshot_doit(struct sk_buff *skb,
1741                                                struct genl_info *info)
1742 {
1743         struct devlink *devlink = info->user_ptr[0];
1744         const struct devlink_ops *ops = devlink->ops;
1745         struct devlink_sb *devlink_sb;
1746
1747         devlink_sb = devlink_sb_get_from_info(devlink, info);
1748         if (IS_ERR(devlink_sb))
1749                 return PTR_ERR(devlink_sb);
1750
1751         if (ops->sb_occ_snapshot)
1752                 return ops->sb_occ_snapshot(devlink, devlink_sb->index);
1753         return -EOPNOTSUPP;
1754 }
1755
1756 static int devlink_nl_cmd_sb_occ_max_clear_doit(struct sk_buff *skb,
1757                                                 struct genl_info *info)
1758 {
1759         struct devlink *devlink = info->user_ptr[0];
1760         const struct devlink_ops *ops = devlink->ops;
1761         struct devlink_sb *devlink_sb;
1762
1763         devlink_sb = devlink_sb_get_from_info(devlink, info);
1764         if (IS_ERR(devlink_sb))
1765                 return PTR_ERR(devlink_sb);
1766
1767         if (ops->sb_occ_max_clear)
1768                 return ops->sb_occ_max_clear(devlink, devlink_sb->index);
1769         return -EOPNOTSUPP;
1770 }
1771
1772 static int devlink_nl_eswitch_fill(struct sk_buff *msg, struct devlink *devlink,
1773                                    enum devlink_command cmd, u32 portid,
1774                                    u32 seq, int flags)
1775 {
1776         const struct devlink_ops *ops = devlink->ops;
1777         enum devlink_eswitch_encap_mode encap_mode;
1778         u8 inline_mode;
1779         void *hdr;
1780         int err = 0;
1781         u16 mode;
1782
1783         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
1784         if (!hdr)
1785                 return -EMSGSIZE;
1786
1787         err = devlink_nl_put_handle(msg, devlink);
1788         if (err)
1789                 goto nla_put_failure;
1790
1791         if (ops->eswitch_mode_get) {
1792                 err = ops->eswitch_mode_get(devlink, &mode);
1793                 if (err)
1794                         goto nla_put_failure;
1795                 err = nla_put_u16(msg, DEVLINK_ATTR_ESWITCH_MODE, mode);
1796                 if (err)
1797                         goto nla_put_failure;
1798         }
1799
1800         if (ops->eswitch_inline_mode_get) {
1801                 err = ops->eswitch_inline_mode_get(devlink, &inline_mode);
1802                 if (err)
1803                         goto nla_put_failure;
1804                 err = nla_put_u8(msg, DEVLINK_ATTR_ESWITCH_INLINE_MODE,
1805                                  inline_mode);
1806                 if (err)
1807                         goto nla_put_failure;
1808         }
1809
1810         if (ops->eswitch_encap_mode_get) {
1811                 err = ops->eswitch_encap_mode_get(devlink, &encap_mode);
1812                 if (err)
1813                         goto nla_put_failure;
1814                 err = nla_put_u8(msg, DEVLINK_ATTR_ESWITCH_ENCAP_MODE, encap_mode);
1815                 if (err)
1816                         goto nla_put_failure;
1817         }
1818
1819         genlmsg_end(msg, hdr);
1820         return 0;
1821
1822 nla_put_failure:
1823         genlmsg_cancel(msg, hdr);
1824         return err;
1825 }
1826
1827 static int devlink_nl_cmd_eswitch_get_doit(struct sk_buff *skb,
1828                                            struct genl_info *info)
1829 {
1830         struct devlink *devlink = info->user_ptr[0];
1831         struct sk_buff *msg;
1832         int err;
1833
1834         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1835         if (!msg)
1836                 return -ENOMEM;
1837
1838         err = devlink_nl_eswitch_fill(msg, devlink, DEVLINK_CMD_ESWITCH_GET,
1839                                       info->snd_portid, info->snd_seq, 0);
1840
1841         if (err) {
1842                 nlmsg_free(msg);
1843                 return err;
1844         }
1845
1846         return genlmsg_reply(msg, info);
1847 }
1848
1849 static int devlink_nl_cmd_eswitch_set_doit(struct sk_buff *skb,
1850                                            struct genl_info *info)
1851 {
1852         struct devlink *devlink = info->user_ptr[0];
1853         const struct devlink_ops *ops = devlink->ops;
1854         enum devlink_eswitch_encap_mode encap_mode;
1855         u8 inline_mode;
1856         int err = 0;
1857         u16 mode;
1858
1859         if (info->attrs[DEVLINK_ATTR_ESWITCH_MODE]) {
1860                 if (!ops->eswitch_mode_set)
1861                         return -EOPNOTSUPP;
1862                 mode = nla_get_u16(info->attrs[DEVLINK_ATTR_ESWITCH_MODE]);
1863                 err = ops->eswitch_mode_set(devlink, mode, info->extack);
1864                 if (err)
1865                         return err;
1866         }
1867
1868         if (info->attrs[DEVLINK_ATTR_ESWITCH_INLINE_MODE]) {
1869                 if (!ops->eswitch_inline_mode_set)
1870                         return -EOPNOTSUPP;
1871                 inline_mode = nla_get_u8(
1872                                 info->attrs[DEVLINK_ATTR_ESWITCH_INLINE_MODE]);
1873                 err = ops->eswitch_inline_mode_set(devlink, inline_mode,
1874                                                    info->extack);
1875                 if (err)
1876                         return err;
1877         }
1878
1879         if (info->attrs[DEVLINK_ATTR_ESWITCH_ENCAP_MODE]) {
1880                 if (!ops->eswitch_encap_mode_set)
1881                         return -EOPNOTSUPP;
1882                 encap_mode = nla_get_u8(info->attrs[DEVLINK_ATTR_ESWITCH_ENCAP_MODE]);
1883                 err = ops->eswitch_encap_mode_set(devlink, encap_mode,
1884                                                   info->extack);
1885                 if (err)
1886                         return err;
1887         }
1888
1889         return 0;
1890 }
1891
1892 int devlink_dpipe_match_put(struct sk_buff *skb,
1893                             struct devlink_dpipe_match *match)
1894 {
1895         struct devlink_dpipe_header *header = match->header;
1896         struct devlink_dpipe_field *field = &header->fields[match->field_id];
1897         struct nlattr *match_attr;
1898
1899         match_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_MATCH);
1900         if (!match_attr)
1901                 return -EMSGSIZE;
1902
1903         if (nla_put_u32(skb, DEVLINK_ATTR_DPIPE_MATCH_TYPE, match->type) ||
1904             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_HEADER_INDEX, match->header_index) ||
1905             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_HEADER_ID, header->id) ||
1906             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_FIELD_ID, field->id) ||
1907             nla_put_u8(skb, DEVLINK_ATTR_DPIPE_HEADER_GLOBAL, header->global))
1908                 goto nla_put_failure;
1909
1910         nla_nest_end(skb, match_attr);
1911         return 0;
1912
1913 nla_put_failure:
1914         nla_nest_cancel(skb, match_attr);
1915         return -EMSGSIZE;
1916 }
1917 EXPORT_SYMBOL_GPL(devlink_dpipe_match_put);
1918
1919 static int devlink_dpipe_matches_put(struct devlink_dpipe_table *table,
1920                                      struct sk_buff *skb)
1921 {
1922         struct nlattr *matches_attr;
1923
1924         matches_attr = nla_nest_start_noflag(skb,
1925                                              DEVLINK_ATTR_DPIPE_TABLE_MATCHES);
1926         if (!matches_attr)
1927                 return -EMSGSIZE;
1928
1929         if (table->table_ops->matches_dump(table->priv, skb))
1930                 goto nla_put_failure;
1931
1932         nla_nest_end(skb, matches_attr);
1933         return 0;
1934
1935 nla_put_failure:
1936         nla_nest_cancel(skb, matches_attr);
1937         return -EMSGSIZE;
1938 }
1939
1940 int devlink_dpipe_action_put(struct sk_buff *skb,
1941                              struct devlink_dpipe_action *action)
1942 {
1943         struct devlink_dpipe_header *header = action->header;
1944         struct devlink_dpipe_field *field = &header->fields[action->field_id];
1945         struct nlattr *action_attr;
1946
1947         action_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_ACTION);
1948         if (!action_attr)
1949                 return -EMSGSIZE;
1950
1951         if (nla_put_u32(skb, DEVLINK_ATTR_DPIPE_ACTION_TYPE, action->type) ||
1952             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_HEADER_INDEX, action->header_index) ||
1953             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_HEADER_ID, header->id) ||
1954             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_FIELD_ID, field->id) ||
1955             nla_put_u8(skb, DEVLINK_ATTR_DPIPE_HEADER_GLOBAL, header->global))
1956                 goto nla_put_failure;
1957
1958         nla_nest_end(skb, action_attr);
1959         return 0;
1960
1961 nla_put_failure:
1962         nla_nest_cancel(skb, action_attr);
1963         return -EMSGSIZE;
1964 }
1965 EXPORT_SYMBOL_GPL(devlink_dpipe_action_put);
1966
1967 static int devlink_dpipe_actions_put(struct devlink_dpipe_table *table,
1968                                      struct sk_buff *skb)
1969 {
1970         struct nlattr *actions_attr;
1971
1972         actions_attr = nla_nest_start_noflag(skb,
1973                                              DEVLINK_ATTR_DPIPE_TABLE_ACTIONS);
1974         if (!actions_attr)
1975                 return -EMSGSIZE;
1976
1977         if (table->table_ops->actions_dump(table->priv, skb))
1978                 goto nla_put_failure;
1979
1980         nla_nest_end(skb, actions_attr);
1981         return 0;
1982
1983 nla_put_failure:
1984         nla_nest_cancel(skb, actions_attr);
1985         return -EMSGSIZE;
1986 }
1987
1988 static int devlink_dpipe_table_put(struct sk_buff *skb,
1989                                    struct devlink_dpipe_table *table)
1990 {
1991         struct nlattr *table_attr;
1992         u64 table_size;
1993
1994         table_size = table->table_ops->size_get(table->priv);
1995         table_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_TABLE);
1996         if (!table_attr)
1997                 return -EMSGSIZE;
1998
1999         if (nla_put_string(skb, DEVLINK_ATTR_DPIPE_TABLE_NAME, table->name) ||
2000             nla_put_u64_64bit(skb, DEVLINK_ATTR_DPIPE_TABLE_SIZE, table_size,
2001                               DEVLINK_ATTR_PAD))
2002                 goto nla_put_failure;
2003         if (nla_put_u8(skb, DEVLINK_ATTR_DPIPE_TABLE_COUNTERS_ENABLED,
2004                        table->counters_enabled))
2005                 goto nla_put_failure;
2006
2007         if (table->resource_valid) {
2008                 if (nla_put_u64_64bit(skb, DEVLINK_ATTR_DPIPE_TABLE_RESOURCE_ID,
2009                                       table->resource_id, DEVLINK_ATTR_PAD) ||
2010                     nla_put_u64_64bit(skb, DEVLINK_ATTR_DPIPE_TABLE_RESOURCE_UNITS,
2011                                       table->resource_units, DEVLINK_ATTR_PAD))
2012                         goto nla_put_failure;
2013         }
2014         if (devlink_dpipe_matches_put(table, skb))
2015                 goto nla_put_failure;
2016
2017         if (devlink_dpipe_actions_put(table, skb))
2018                 goto nla_put_failure;
2019
2020         nla_nest_end(skb, table_attr);
2021         return 0;
2022
2023 nla_put_failure:
2024         nla_nest_cancel(skb, table_attr);
2025         return -EMSGSIZE;
2026 }
2027
2028 static int devlink_dpipe_send_and_alloc_skb(struct sk_buff **pskb,
2029                                             struct genl_info *info)
2030 {
2031         int err;
2032
2033         if (*pskb) {
2034                 err = genlmsg_reply(*pskb, info);
2035                 if (err)
2036                         return err;
2037         }
2038         *pskb = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
2039         if (!*pskb)
2040                 return -ENOMEM;
2041         return 0;
2042 }
2043
2044 static int devlink_dpipe_tables_fill(struct genl_info *info,
2045                                      enum devlink_command cmd, int flags,
2046                                      struct list_head *dpipe_tables,
2047                                      const char *table_name)
2048 {
2049         struct devlink *devlink = info->user_ptr[0];
2050         struct devlink_dpipe_table *table;
2051         struct nlattr *tables_attr;
2052         struct sk_buff *skb = NULL;
2053         struct nlmsghdr *nlh;
2054         bool incomplete;
2055         void *hdr;
2056         int i;
2057         int err;
2058
2059         table = list_first_entry(dpipe_tables,
2060                                  struct devlink_dpipe_table, list);
2061 start_again:
2062         err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2063         if (err)
2064                 return err;
2065
2066         hdr = genlmsg_put(skb, info->snd_portid, info->snd_seq,
2067                           &devlink_nl_family, NLM_F_MULTI, cmd);
2068         if (!hdr) {
2069                 nlmsg_free(skb);
2070                 return -EMSGSIZE;
2071         }
2072
2073         if (devlink_nl_put_handle(skb, devlink))
2074                 goto nla_put_failure;
2075         tables_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_TABLES);
2076         if (!tables_attr)
2077                 goto nla_put_failure;
2078
2079         i = 0;
2080         incomplete = false;
2081         list_for_each_entry_from(table, dpipe_tables, list) {
2082                 if (!table_name) {
2083                         err = devlink_dpipe_table_put(skb, table);
2084                         if (err) {
2085                                 if (!i)
2086                                         goto err_table_put;
2087                                 incomplete = true;
2088                                 break;
2089                         }
2090                 } else {
2091                         if (!strcmp(table->name, table_name)) {
2092                                 err = devlink_dpipe_table_put(skb, table);
2093                                 if (err)
2094                                         break;
2095                         }
2096                 }
2097                 i++;
2098         }
2099
2100         nla_nest_end(skb, tables_attr);
2101         genlmsg_end(skb, hdr);
2102         if (incomplete)
2103                 goto start_again;
2104
2105 send_done:
2106         nlh = nlmsg_put(skb, info->snd_portid, info->snd_seq,
2107                         NLMSG_DONE, 0, flags | NLM_F_MULTI);
2108         if (!nlh) {
2109                 err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2110                 if (err)
2111                         return err;
2112                 goto send_done;
2113         }
2114
2115         return genlmsg_reply(skb, info);
2116
2117 nla_put_failure:
2118         err = -EMSGSIZE;
2119 err_table_put:
2120         nlmsg_free(skb);
2121         return err;
2122 }
2123
2124 static int devlink_nl_cmd_dpipe_table_get(struct sk_buff *skb,
2125                                           struct genl_info *info)
2126 {
2127         struct devlink *devlink = info->user_ptr[0];
2128         const char *table_name =  NULL;
2129
2130         if (info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME])
2131                 table_name = nla_data(info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME]);
2132
2133         return devlink_dpipe_tables_fill(info, DEVLINK_CMD_DPIPE_TABLE_GET, 0,
2134                                          &devlink->dpipe_table_list,
2135                                          table_name);
2136 }
2137
2138 static int devlink_dpipe_value_put(struct sk_buff *skb,
2139                                    struct devlink_dpipe_value *value)
2140 {
2141         if (nla_put(skb, DEVLINK_ATTR_DPIPE_VALUE,
2142                     value->value_size, value->value))
2143                 return -EMSGSIZE;
2144         if (value->mask)
2145                 if (nla_put(skb, DEVLINK_ATTR_DPIPE_VALUE_MASK,
2146                             value->value_size, value->mask))
2147                         return -EMSGSIZE;
2148         if (value->mapping_valid)
2149                 if (nla_put_u32(skb, DEVLINK_ATTR_DPIPE_VALUE_MAPPING,
2150                                 value->mapping_value))
2151                         return -EMSGSIZE;
2152         return 0;
2153 }
2154
2155 static int devlink_dpipe_action_value_put(struct sk_buff *skb,
2156                                           struct devlink_dpipe_value *value)
2157 {
2158         if (!value->action)
2159                 return -EINVAL;
2160         if (devlink_dpipe_action_put(skb, value->action))
2161                 return -EMSGSIZE;
2162         if (devlink_dpipe_value_put(skb, value))
2163                 return -EMSGSIZE;
2164         return 0;
2165 }
2166
2167 static int devlink_dpipe_action_values_put(struct sk_buff *skb,
2168                                            struct devlink_dpipe_value *values,
2169                                            unsigned int values_count)
2170 {
2171         struct nlattr *action_attr;
2172         int i;
2173         int err;
2174
2175         for (i = 0; i < values_count; i++) {
2176                 action_attr = nla_nest_start_noflag(skb,
2177                                                     DEVLINK_ATTR_DPIPE_ACTION_VALUE);
2178                 if (!action_attr)
2179                         return -EMSGSIZE;
2180                 err = devlink_dpipe_action_value_put(skb, &values[i]);
2181                 if (err)
2182                         goto err_action_value_put;
2183                 nla_nest_end(skb, action_attr);
2184         }
2185         return 0;
2186
2187 err_action_value_put:
2188         nla_nest_cancel(skb, action_attr);
2189         return err;
2190 }
2191
2192 static int devlink_dpipe_match_value_put(struct sk_buff *skb,
2193                                          struct devlink_dpipe_value *value)
2194 {
2195         if (!value->match)
2196                 return -EINVAL;
2197         if (devlink_dpipe_match_put(skb, value->match))
2198                 return -EMSGSIZE;
2199         if (devlink_dpipe_value_put(skb, value))
2200                 return -EMSGSIZE;
2201         return 0;
2202 }
2203
2204 static int devlink_dpipe_match_values_put(struct sk_buff *skb,
2205                                           struct devlink_dpipe_value *values,
2206                                           unsigned int values_count)
2207 {
2208         struct nlattr *match_attr;
2209         int i;
2210         int err;
2211
2212         for (i = 0; i < values_count; i++) {
2213                 match_attr = nla_nest_start_noflag(skb,
2214                                                    DEVLINK_ATTR_DPIPE_MATCH_VALUE);
2215                 if (!match_attr)
2216                         return -EMSGSIZE;
2217                 err = devlink_dpipe_match_value_put(skb, &values[i]);
2218                 if (err)
2219                         goto err_match_value_put;
2220                 nla_nest_end(skb, match_attr);
2221         }
2222         return 0;
2223
2224 err_match_value_put:
2225         nla_nest_cancel(skb, match_attr);
2226         return err;
2227 }
2228
2229 static int devlink_dpipe_entry_put(struct sk_buff *skb,
2230                                    struct devlink_dpipe_entry *entry)
2231 {
2232         struct nlattr *entry_attr, *matches_attr, *actions_attr;
2233         int err;
2234
2235         entry_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_ENTRY);
2236         if (!entry_attr)
2237                 return  -EMSGSIZE;
2238
2239         if (nla_put_u64_64bit(skb, DEVLINK_ATTR_DPIPE_ENTRY_INDEX, entry->index,
2240                               DEVLINK_ATTR_PAD))
2241                 goto nla_put_failure;
2242         if (entry->counter_valid)
2243                 if (nla_put_u64_64bit(skb, DEVLINK_ATTR_DPIPE_ENTRY_COUNTER,
2244                                       entry->counter, DEVLINK_ATTR_PAD))
2245                         goto nla_put_failure;
2246
2247         matches_attr = nla_nest_start_noflag(skb,
2248                                              DEVLINK_ATTR_DPIPE_ENTRY_MATCH_VALUES);
2249         if (!matches_attr)
2250                 goto nla_put_failure;
2251
2252         err = devlink_dpipe_match_values_put(skb, entry->match_values,
2253                                              entry->match_values_count);
2254         if (err) {
2255                 nla_nest_cancel(skb, matches_attr);
2256                 goto err_match_values_put;
2257         }
2258         nla_nest_end(skb, matches_attr);
2259
2260         actions_attr = nla_nest_start_noflag(skb,
2261                                              DEVLINK_ATTR_DPIPE_ENTRY_ACTION_VALUES);
2262         if (!actions_attr)
2263                 goto nla_put_failure;
2264
2265         err = devlink_dpipe_action_values_put(skb, entry->action_values,
2266                                               entry->action_values_count);
2267         if (err) {
2268                 nla_nest_cancel(skb, actions_attr);
2269                 goto err_action_values_put;
2270         }
2271         nla_nest_end(skb, actions_attr);
2272
2273         nla_nest_end(skb, entry_attr);
2274         return 0;
2275
2276 nla_put_failure:
2277         err = -EMSGSIZE;
2278 err_match_values_put:
2279 err_action_values_put:
2280         nla_nest_cancel(skb, entry_attr);
2281         return err;
2282 }
2283
2284 static struct devlink_dpipe_table *
2285 devlink_dpipe_table_find(struct list_head *dpipe_tables,
2286                          const char *table_name, struct devlink *devlink)
2287 {
2288         struct devlink_dpipe_table *table;
2289         list_for_each_entry_rcu(table, dpipe_tables, list,
2290                                 lockdep_is_held(&devlink->lock)) {
2291                 if (!strcmp(table->name, table_name))
2292                         return table;
2293         }
2294         return NULL;
2295 }
2296
2297 int devlink_dpipe_entry_ctx_prepare(struct devlink_dpipe_dump_ctx *dump_ctx)
2298 {
2299         struct devlink *devlink;
2300         int err;
2301
2302         err = devlink_dpipe_send_and_alloc_skb(&dump_ctx->skb,
2303                                                dump_ctx->info);
2304         if (err)
2305                 return err;
2306
2307         dump_ctx->hdr = genlmsg_put(dump_ctx->skb,
2308                                     dump_ctx->info->snd_portid,
2309                                     dump_ctx->info->snd_seq,
2310                                     &devlink_nl_family, NLM_F_MULTI,
2311                                     dump_ctx->cmd);
2312         if (!dump_ctx->hdr)
2313                 goto nla_put_failure;
2314
2315         devlink = dump_ctx->info->user_ptr[0];
2316         if (devlink_nl_put_handle(dump_ctx->skb, devlink))
2317                 goto nla_put_failure;
2318         dump_ctx->nest = nla_nest_start_noflag(dump_ctx->skb,
2319                                                DEVLINK_ATTR_DPIPE_ENTRIES);
2320         if (!dump_ctx->nest)
2321                 goto nla_put_failure;
2322         return 0;
2323
2324 nla_put_failure:
2325         nlmsg_free(dump_ctx->skb);
2326         return -EMSGSIZE;
2327 }
2328 EXPORT_SYMBOL_GPL(devlink_dpipe_entry_ctx_prepare);
2329
2330 int devlink_dpipe_entry_ctx_append(struct devlink_dpipe_dump_ctx *dump_ctx,
2331                                    struct devlink_dpipe_entry *entry)
2332 {
2333         return devlink_dpipe_entry_put(dump_ctx->skb, entry);
2334 }
2335 EXPORT_SYMBOL_GPL(devlink_dpipe_entry_ctx_append);
2336
2337 int devlink_dpipe_entry_ctx_close(struct devlink_dpipe_dump_ctx *dump_ctx)
2338 {
2339         nla_nest_end(dump_ctx->skb, dump_ctx->nest);
2340         genlmsg_end(dump_ctx->skb, dump_ctx->hdr);
2341         return 0;
2342 }
2343 EXPORT_SYMBOL_GPL(devlink_dpipe_entry_ctx_close);
2344
2345 void devlink_dpipe_entry_clear(struct devlink_dpipe_entry *entry)
2346
2347 {
2348         unsigned int value_count, value_index;
2349         struct devlink_dpipe_value *value;
2350
2351         value = entry->action_values;
2352         value_count = entry->action_values_count;
2353         for (value_index = 0; value_index < value_count; value_index++) {
2354                 kfree(value[value_index].value);
2355                 kfree(value[value_index].mask);
2356         }
2357
2358         value = entry->match_values;
2359         value_count = entry->match_values_count;
2360         for (value_index = 0; value_index < value_count; value_index++) {
2361                 kfree(value[value_index].value);
2362                 kfree(value[value_index].mask);
2363         }
2364 }
2365 EXPORT_SYMBOL(devlink_dpipe_entry_clear);
2366
2367 static int devlink_dpipe_entries_fill(struct genl_info *info,
2368                                       enum devlink_command cmd, int flags,
2369                                       struct devlink_dpipe_table *table)
2370 {
2371         struct devlink_dpipe_dump_ctx dump_ctx;
2372         struct nlmsghdr *nlh;
2373         int err;
2374
2375         dump_ctx.skb = NULL;
2376         dump_ctx.cmd = cmd;
2377         dump_ctx.info = info;
2378
2379         err = table->table_ops->entries_dump(table->priv,
2380                                              table->counters_enabled,
2381                                              &dump_ctx);
2382         if (err)
2383                 return err;
2384
2385 send_done:
2386         nlh = nlmsg_put(dump_ctx.skb, info->snd_portid, info->snd_seq,
2387                         NLMSG_DONE, 0, flags | NLM_F_MULTI);
2388         if (!nlh) {
2389                 err = devlink_dpipe_send_and_alloc_skb(&dump_ctx.skb, info);
2390                 if (err)
2391                         return err;
2392                 goto send_done;
2393         }
2394         return genlmsg_reply(dump_ctx.skb, info);
2395 }
2396
2397 static int devlink_nl_cmd_dpipe_entries_get(struct sk_buff *skb,
2398                                             struct genl_info *info)
2399 {
2400         struct devlink *devlink = info->user_ptr[0];
2401         struct devlink_dpipe_table *table;
2402         const char *table_name;
2403
2404         if (!info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME])
2405                 return -EINVAL;
2406
2407         table_name = nla_data(info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME]);
2408         table = devlink_dpipe_table_find(&devlink->dpipe_table_list,
2409                                          table_name, devlink);
2410         if (!table)
2411                 return -EINVAL;
2412
2413         if (!table->table_ops->entries_dump)
2414                 return -EINVAL;
2415
2416         return devlink_dpipe_entries_fill(info, DEVLINK_CMD_DPIPE_ENTRIES_GET,
2417                                           0, table);
2418 }
2419
2420 static int devlink_dpipe_fields_put(struct sk_buff *skb,
2421                                     const struct devlink_dpipe_header *header)
2422 {
2423         struct devlink_dpipe_field *field;
2424         struct nlattr *field_attr;
2425         int i;
2426
2427         for (i = 0; i < header->fields_count; i++) {
2428                 field = &header->fields[i];
2429                 field_attr = nla_nest_start_noflag(skb,
2430                                                    DEVLINK_ATTR_DPIPE_FIELD);
2431                 if (!field_attr)
2432                         return -EMSGSIZE;
2433                 if (nla_put_string(skb, DEVLINK_ATTR_DPIPE_FIELD_NAME, field->name) ||
2434                     nla_put_u32(skb, DEVLINK_ATTR_DPIPE_FIELD_ID, field->id) ||
2435                     nla_put_u32(skb, DEVLINK_ATTR_DPIPE_FIELD_BITWIDTH, field->bitwidth) ||
2436                     nla_put_u32(skb, DEVLINK_ATTR_DPIPE_FIELD_MAPPING_TYPE, field->mapping_type))
2437                         goto nla_put_failure;
2438                 nla_nest_end(skb, field_attr);
2439         }
2440         return 0;
2441
2442 nla_put_failure:
2443         nla_nest_cancel(skb, field_attr);
2444         return -EMSGSIZE;
2445 }
2446
2447 static int devlink_dpipe_header_put(struct sk_buff *skb,
2448                                     struct devlink_dpipe_header *header)
2449 {
2450         struct nlattr *fields_attr, *header_attr;
2451         int err;
2452
2453         header_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_HEADER);
2454         if (!header_attr)
2455                 return -EMSGSIZE;
2456
2457         if (nla_put_string(skb, DEVLINK_ATTR_DPIPE_HEADER_NAME, header->name) ||
2458             nla_put_u32(skb, DEVLINK_ATTR_DPIPE_HEADER_ID, header->id) ||
2459             nla_put_u8(skb, DEVLINK_ATTR_DPIPE_HEADER_GLOBAL, header->global))
2460                 goto nla_put_failure;
2461
2462         fields_attr = nla_nest_start_noflag(skb,
2463                                             DEVLINK_ATTR_DPIPE_HEADER_FIELDS);
2464         if (!fields_attr)
2465                 goto nla_put_failure;
2466
2467         err = devlink_dpipe_fields_put(skb, header);
2468         if (err) {
2469                 nla_nest_cancel(skb, fields_attr);
2470                 goto nla_put_failure;
2471         }
2472         nla_nest_end(skb, fields_attr);
2473         nla_nest_end(skb, header_attr);
2474         return 0;
2475
2476 nla_put_failure:
2477         err = -EMSGSIZE;
2478         nla_nest_cancel(skb, header_attr);
2479         return err;
2480 }
2481
2482 static int devlink_dpipe_headers_fill(struct genl_info *info,
2483                                       enum devlink_command cmd, int flags,
2484                                       struct devlink_dpipe_headers *
2485                                       dpipe_headers)
2486 {
2487         struct devlink *devlink = info->user_ptr[0];
2488         struct nlattr *headers_attr;
2489         struct sk_buff *skb = NULL;
2490         struct nlmsghdr *nlh;
2491         void *hdr;
2492         int i, j;
2493         int err;
2494
2495         i = 0;
2496 start_again:
2497         err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2498         if (err)
2499                 return err;
2500
2501         hdr = genlmsg_put(skb, info->snd_portid, info->snd_seq,
2502                           &devlink_nl_family, NLM_F_MULTI, cmd);
2503         if (!hdr) {
2504                 nlmsg_free(skb);
2505                 return -EMSGSIZE;
2506         }
2507
2508         if (devlink_nl_put_handle(skb, devlink))
2509                 goto nla_put_failure;
2510         headers_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_DPIPE_HEADERS);
2511         if (!headers_attr)
2512                 goto nla_put_failure;
2513
2514         j = 0;
2515         for (; i < dpipe_headers->headers_count; i++) {
2516                 err = devlink_dpipe_header_put(skb, dpipe_headers->headers[i]);
2517                 if (err) {
2518                         if (!j)
2519                                 goto err_table_put;
2520                         break;
2521                 }
2522                 j++;
2523         }
2524         nla_nest_end(skb, headers_attr);
2525         genlmsg_end(skb, hdr);
2526         if (i != dpipe_headers->headers_count)
2527                 goto start_again;
2528
2529 send_done:
2530         nlh = nlmsg_put(skb, info->snd_portid, info->snd_seq,
2531                         NLMSG_DONE, 0, flags | NLM_F_MULTI);
2532         if (!nlh) {
2533                 err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2534                 if (err)
2535                         return err;
2536                 goto send_done;
2537         }
2538         return genlmsg_reply(skb, info);
2539
2540 nla_put_failure:
2541         err = -EMSGSIZE;
2542 err_table_put:
2543         nlmsg_free(skb);
2544         return err;
2545 }
2546
2547 static int devlink_nl_cmd_dpipe_headers_get(struct sk_buff *skb,
2548                                             struct genl_info *info)
2549 {
2550         struct devlink *devlink = info->user_ptr[0];
2551
2552         if (!devlink->dpipe_headers)
2553                 return -EOPNOTSUPP;
2554         return devlink_dpipe_headers_fill(info, DEVLINK_CMD_DPIPE_HEADERS_GET,
2555                                           0, devlink->dpipe_headers);
2556 }
2557
2558 static int devlink_dpipe_table_counters_set(struct devlink *devlink,
2559                                             const char *table_name,
2560                                             bool enable)
2561 {
2562         struct devlink_dpipe_table *table;
2563
2564         table = devlink_dpipe_table_find(&devlink->dpipe_table_list,
2565                                          table_name, devlink);
2566         if (!table)
2567                 return -EINVAL;
2568
2569         if (table->counter_control_extern)
2570                 return -EOPNOTSUPP;
2571
2572         if (!(table->counters_enabled ^ enable))
2573                 return 0;
2574
2575         table->counters_enabled = enable;
2576         if (table->table_ops->counters_set_update)
2577                 table->table_ops->counters_set_update(table->priv, enable);
2578         return 0;
2579 }
2580
2581 static int devlink_nl_cmd_dpipe_table_counters_set(struct sk_buff *skb,
2582                                                    struct genl_info *info)
2583 {
2584         struct devlink *devlink = info->user_ptr[0];
2585         const char *table_name;
2586         bool counters_enable;
2587
2588         if (!info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME] ||
2589             !info->attrs[DEVLINK_ATTR_DPIPE_TABLE_COUNTERS_ENABLED])
2590                 return -EINVAL;
2591
2592         table_name = nla_data(info->attrs[DEVLINK_ATTR_DPIPE_TABLE_NAME]);
2593         counters_enable = !!nla_get_u8(info->attrs[DEVLINK_ATTR_DPIPE_TABLE_COUNTERS_ENABLED]);
2594
2595         return devlink_dpipe_table_counters_set(devlink, table_name,
2596                                                 counters_enable);
2597 }
2598
2599 static struct devlink_resource *
2600 devlink_resource_find(struct devlink *devlink,
2601                       struct devlink_resource *resource, u64 resource_id)
2602 {
2603         struct list_head *resource_list;
2604
2605         if (resource)
2606                 resource_list = &resource->resource_list;
2607         else
2608                 resource_list = &devlink->resource_list;
2609
2610         list_for_each_entry(resource, resource_list, list) {
2611                 struct devlink_resource *child_resource;
2612
2613                 if (resource->id == resource_id)
2614                         return resource;
2615
2616                 child_resource = devlink_resource_find(devlink, resource,
2617                                                        resource_id);
2618                 if (child_resource)
2619                         return child_resource;
2620         }
2621         return NULL;
2622 }
2623
2624 static void
2625 devlink_resource_validate_children(struct devlink_resource *resource)
2626 {
2627         struct devlink_resource *child_resource;
2628         bool size_valid = true;
2629         u64 parts_size = 0;
2630
2631         if (list_empty(&resource->resource_list))
2632                 goto out;
2633
2634         list_for_each_entry(child_resource, &resource->resource_list, list)
2635                 parts_size += child_resource->size_new;
2636
2637         if (parts_size > resource->size_new)
2638                 size_valid = false;
2639 out:
2640         resource->size_valid = size_valid;
2641 }
2642
2643 static int
2644 devlink_resource_validate_size(struct devlink_resource *resource, u64 size,
2645                                struct netlink_ext_ack *extack)
2646 {
2647         u64 reminder;
2648         int err = 0;
2649
2650         if (size > resource->size_params.size_max) {
2651                 NL_SET_ERR_MSG_MOD(extack, "Size larger than maximum");
2652                 err = -EINVAL;
2653         }
2654
2655         if (size < resource->size_params.size_min) {
2656                 NL_SET_ERR_MSG_MOD(extack, "Size smaller than minimum");
2657                 err = -EINVAL;
2658         }
2659
2660         div64_u64_rem(size, resource->size_params.size_granularity, &reminder);
2661         if (reminder) {
2662                 NL_SET_ERR_MSG_MOD(extack, "Wrong granularity");
2663                 err = -EINVAL;
2664         }
2665
2666         return err;
2667 }
2668
2669 static int devlink_nl_cmd_resource_set(struct sk_buff *skb,
2670                                        struct genl_info *info)
2671 {
2672         struct devlink *devlink = info->user_ptr[0];
2673         struct devlink_resource *resource;
2674         u64 resource_id;
2675         u64 size;
2676         int err;
2677
2678         if (!info->attrs[DEVLINK_ATTR_RESOURCE_ID] ||
2679             !info->attrs[DEVLINK_ATTR_RESOURCE_SIZE])
2680                 return -EINVAL;
2681         resource_id = nla_get_u64(info->attrs[DEVLINK_ATTR_RESOURCE_ID]);
2682
2683         resource = devlink_resource_find(devlink, NULL, resource_id);
2684         if (!resource)
2685                 return -EINVAL;
2686
2687         size = nla_get_u64(info->attrs[DEVLINK_ATTR_RESOURCE_SIZE]);
2688         err = devlink_resource_validate_size(resource, size, info->extack);
2689         if (err)
2690                 return err;
2691
2692         resource->size_new = size;
2693         devlink_resource_validate_children(resource);
2694         if (resource->parent)
2695                 devlink_resource_validate_children(resource->parent);
2696         return 0;
2697 }
2698
2699 static int
2700 devlink_resource_size_params_put(struct devlink_resource *resource,
2701                                  struct sk_buff *skb)
2702 {
2703         struct devlink_resource_size_params *size_params;
2704
2705         size_params = &resource->size_params;
2706         if (nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_SIZE_GRAN,
2707                               size_params->size_granularity, DEVLINK_ATTR_PAD) ||
2708             nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_SIZE_MAX,
2709                               size_params->size_max, DEVLINK_ATTR_PAD) ||
2710             nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_SIZE_MIN,
2711                               size_params->size_min, DEVLINK_ATTR_PAD) ||
2712             nla_put_u8(skb, DEVLINK_ATTR_RESOURCE_UNIT, size_params->unit))
2713                 return -EMSGSIZE;
2714         return 0;
2715 }
2716
2717 static int devlink_resource_occ_put(struct devlink_resource *resource,
2718                                     struct sk_buff *skb)
2719 {
2720         if (!resource->occ_get)
2721                 return 0;
2722         return nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_OCC,
2723                                  resource->occ_get(resource->occ_get_priv),
2724                                  DEVLINK_ATTR_PAD);
2725 }
2726
2727 static int devlink_resource_put(struct devlink *devlink, struct sk_buff *skb,
2728                                 struct devlink_resource *resource)
2729 {
2730         struct devlink_resource *child_resource;
2731         struct nlattr *child_resource_attr;
2732         struct nlattr *resource_attr;
2733
2734         resource_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_RESOURCE);
2735         if (!resource_attr)
2736                 return -EMSGSIZE;
2737
2738         if (nla_put_string(skb, DEVLINK_ATTR_RESOURCE_NAME, resource->name) ||
2739             nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_SIZE, resource->size,
2740                               DEVLINK_ATTR_PAD) ||
2741             nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_ID, resource->id,
2742                               DEVLINK_ATTR_PAD))
2743                 goto nla_put_failure;
2744         if (resource->size != resource->size_new)
2745                 nla_put_u64_64bit(skb, DEVLINK_ATTR_RESOURCE_SIZE_NEW,
2746                                   resource->size_new, DEVLINK_ATTR_PAD);
2747         if (devlink_resource_occ_put(resource, skb))
2748                 goto nla_put_failure;
2749         if (devlink_resource_size_params_put(resource, skb))
2750                 goto nla_put_failure;
2751         if (list_empty(&resource->resource_list))
2752                 goto out;
2753
2754         if (nla_put_u8(skb, DEVLINK_ATTR_RESOURCE_SIZE_VALID,
2755                        resource->size_valid))
2756                 goto nla_put_failure;
2757
2758         child_resource_attr = nla_nest_start_noflag(skb,
2759                                                     DEVLINK_ATTR_RESOURCE_LIST);
2760         if (!child_resource_attr)
2761                 goto nla_put_failure;
2762
2763         list_for_each_entry(child_resource, &resource->resource_list, list) {
2764                 if (devlink_resource_put(devlink, skb, child_resource))
2765                         goto resource_put_failure;
2766         }
2767
2768         nla_nest_end(skb, child_resource_attr);
2769 out:
2770         nla_nest_end(skb, resource_attr);
2771         return 0;
2772
2773 resource_put_failure:
2774         nla_nest_cancel(skb, child_resource_attr);
2775 nla_put_failure:
2776         nla_nest_cancel(skb, resource_attr);
2777         return -EMSGSIZE;
2778 }
2779
2780 static int devlink_resource_fill(struct genl_info *info,
2781                                  enum devlink_command cmd, int flags)
2782 {
2783         struct devlink *devlink = info->user_ptr[0];
2784         struct devlink_resource *resource;
2785         struct nlattr *resources_attr;
2786         struct sk_buff *skb = NULL;
2787         struct nlmsghdr *nlh;
2788         bool incomplete;
2789         void *hdr;
2790         int i;
2791         int err;
2792
2793         resource = list_first_entry(&devlink->resource_list,
2794                                     struct devlink_resource, list);
2795 start_again:
2796         err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2797         if (err)
2798                 return err;
2799
2800         hdr = genlmsg_put(skb, info->snd_portid, info->snd_seq,
2801                           &devlink_nl_family, NLM_F_MULTI, cmd);
2802         if (!hdr) {
2803                 nlmsg_free(skb);
2804                 return -EMSGSIZE;
2805         }
2806
2807         if (devlink_nl_put_handle(skb, devlink))
2808                 goto nla_put_failure;
2809
2810         resources_attr = nla_nest_start_noflag(skb,
2811                                                DEVLINK_ATTR_RESOURCE_LIST);
2812         if (!resources_attr)
2813                 goto nla_put_failure;
2814
2815         incomplete = false;
2816         i = 0;
2817         list_for_each_entry_from(resource, &devlink->resource_list, list) {
2818                 err = devlink_resource_put(devlink, skb, resource);
2819                 if (err) {
2820                         if (!i)
2821                                 goto err_resource_put;
2822                         incomplete = true;
2823                         break;
2824                 }
2825                 i++;
2826         }
2827         nla_nest_end(skb, resources_attr);
2828         genlmsg_end(skb, hdr);
2829         if (incomplete)
2830                 goto start_again;
2831 send_done:
2832         nlh = nlmsg_put(skb, info->snd_portid, info->snd_seq,
2833                         NLMSG_DONE, 0, flags | NLM_F_MULTI);
2834         if (!nlh) {
2835                 err = devlink_dpipe_send_and_alloc_skb(&skb, info);
2836                 if (err)
2837                         return err;
2838                 goto send_done;
2839         }
2840         return genlmsg_reply(skb, info);
2841
2842 nla_put_failure:
2843         err = -EMSGSIZE;
2844 err_resource_put:
2845         nlmsg_free(skb);
2846         return err;
2847 }
2848
2849 static int devlink_nl_cmd_resource_dump(struct sk_buff *skb,
2850                                         struct genl_info *info)
2851 {
2852         struct devlink *devlink = info->user_ptr[0];
2853
2854         if (list_empty(&devlink->resource_list))
2855                 return -EOPNOTSUPP;
2856
2857         return devlink_resource_fill(info, DEVLINK_CMD_RESOURCE_DUMP, 0);
2858 }
2859
2860 static int
2861 devlink_resources_validate(struct devlink *devlink,
2862                            struct devlink_resource *resource,
2863                            struct genl_info *info)
2864 {
2865         struct list_head *resource_list;
2866         int err = 0;
2867
2868         if (resource)
2869                 resource_list = &resource->resource_list;
2870         else
2871                 resource_list = &devlink->resource_list;
2872
2873         list_for_each_entry(resource, resource_list, list) {
2874                 if (!resource->size_valid)
2875                         return -EINVAL;
2876                 err = devlink_resources_validate(devlink, resource, info);
2877                 if (err)
2878                         return err;
2879         }
2880         return err;
2881 }
2882
2883 static struct net *devlink_netns_get(struct sk_buff *skb,
2884                                      struct genl_info *info)
2885 {
2886         struct nlattr *netns_pid_attr = info->attrs[DEVLINK_ATTR_NETNS_PID];
2887         struct nlattr *netns_fd_attr = info->attrs[DEVLINK_ATTR_NETNS_FD];
2888         struct nlattr *netns_id_attr = info->attrs[DEVLINK_ATTR_NETNS_ID];
2889         struct net *net;
2890
2891         if (!!netns_pid_attr + !!netns_fd_attr + !!netns_id_attr > 1) {
2892                 NL_SET_ERR_MSG_MOD(info->extack, "multiple netns identifying attributes specified");
2893                 return ERR_PTR(-EINVAL);
2894         }
2895
2896         if (netns_pid_attr) {
2897                 net = get_net_ns_by_pid(nla_get_u32(netns_pid_attr));
2898         } else if (netns_fd_attr) {
2899                 net = get_net_ns_by_fd(nla_get_u32(netns_fd_attr));
2900         } else if (netns_id_attr) {
2901                 net = get_net_ns_by_id(sock_net(skb->sk),
2902                                        nla_get_u32(netns_id_attr));
2903                 if (!net)
2904                         net = ERR_PTR(-EINVAL);
2905         } else {
2906                 WARN_ON(1);
2907                 net = ERR_PTR(-EINVAL);
2908         }
2909         if (IS_ERR(net)) {
2910                 NL_SET_ERR_MSG_MOD(info->extack, "Unknown network namespace");
2911                 return ERR_PTR(-EINVAL);
2912         }
2913         if (!netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN)) {
2914                 put_net(net);
2915                 return ERR_PTR(-EPERM);
2916         }
2917         return net;
2918 }
2919
2920 static void devlink_param_notify(struct devlink *devlink,
2921                                  unsigned int port_index,
2922                                  struct devlink_param_item *param_item,
2923                                  enum devlink_command cmd);
2924
2925 static void devlink_reload_netns_change(struct devlink *devlink,
2926                                         struct net *dest_net)
2927 {
2928         struct devlink_param_item *param_item;
2929
2930         /* Userspace needs to be notified about devlink objects
2931          * removed from original and entering new network namespace.
2932          * The rest of the devlink objects are re-created during
2933          * reload process so the notifications are generated separatelly.
2934          */
2935
2936         list_for_each_entry(param_item, &devlink->param_list, list)
2937                 devlink_param_notify(devlink, 0, param_item,
2938                                      DEVLINK_CMD_PARAM_DEL);
2939         devlink_notify(devlink, DEVLINK_CMD_DEL);
2940
2941         __devlink_net_set(devlink, dest_net);
2942
2943         devlink_notify(devlink, DEVLINK_CMD_NEW);
2944         list_for_each_entry(param_item, &devlink->param_list, list)
2945                 devlink_param_notify(devlink, 0, param_item,
2946                                      DEVLINK_CMD_PARAM_NEW);
2947 }
2948
2949 static bool devlink_reload_supported(const struct devlink *devlink)
2950 {
2951         return devlink->ops->reload_down && devlink->ops->reload_up;
2952 }
2953
2954 static void devlink_reload_failed_set(struct devlink *devlink,
2955                                       bool reload_failed)
2956 {
2957         if (devlink->reload_failed == reload_failed)
2958                 return;
2959         devlink->reload_failed = reload_failed;
2960         devlink_notify(devlink, DEVLINK_CMD_NEW);
2961 }
2962
2963 bool devlink_is_reload_failed(const struct devlink *devlink)
2964 {
2965         return devlink->reload_failed;
2966 }
2967 EXPORT_SYMBOL_GPL(devlink_is_reload_failed);
2968
2969 static int devlink_reload(struct devlink *devlink, struct net *dest_net,
2970                           struct netlink_ext_ack *extack)
2971 {
2972         int err;
2973
2974         if (!devlink->reload_enabled)
2975                 return -EOPNOTSUPP;
2976
2977         err = devlink->ops->reload_down(devlink, !!dest_net, extack);
2978         if (err)
2979                 return err;
2980
2981         if (dest_net && !net_eq(dest_net, devlink_net(devlink)))
2982                 devlink_reload_netns_change(devlink, dest_net);
2983
2984         err = devlink->ops->reload_up(devlink, extack);
2985         devlink_reload_failed_set(devlink, !!err);
2986         return err;
2987 }
2988
2989 static int devlink_nl_cmd_reload(struct sk_buff *skb, struct genl_info *info)
2990 {
2991         struct devlink *devlink = info->user_ptr[0];
2992         struct net *dest_net = NULL;
2993         int err;
2994
2995         if (!devlink_reload_supported(devlink))
2996                 return -EOPNOTSUPP;
2997
2998         err = devlink_resources_validate(devlink, NULL, info);
2999         if (err) {
3000                 NL_SET_ERR_MSG_MOD(info->extack, "resources size validation failed");
3001                 return err;
3002         }
3003
3004         if (info->attrs[DEVLINK_ATTR_NETNS_PID] ||
3005             info->attrs[DEVLINK_ATTR_NETNS_FD] ||
3006             info->attrs[DEVLINK_ATTR_NETNS_ID]) {
3007                 dest_net = devlink_netns_get(skb, info);
3008                 if (IS_ERR(dest_net))
3009                         return PTR_ERR(dest_net);
3010         }
3011
3012         err = devlink_reload(devlink, dest_net, info->extack);
3013
3014         if (dest_net)
3015                 put_net(dest_net);
3016
3017         return err;
3018 }
3019
3020 static int devlink_nl_flash_update_fill(struct sk_buff *msg,
3021                                         struct devlink *devlink,
3022                                         enum devlink_command cmd,
3023                                         struct devlink_flash_notify *params)
3024 {
3025         void *hdr;
3026
3027         hdr = genlmsg_put(msg, 0, 0, &devlink_nl_family, 0, cmd);
3028         if (!hdr)
3029                 return -EMSGSIZE;
3030
3031         if (devlink_nl_put_handle(msg, devlink))
3032                 goto nla_put_failure;
3033
3034         if (cmd != DEVLINK_CMD_FLASH_UPDATE_STATUS)
3035                 goto out;
3036
3037         if (params->status_msg &&
3038             nla_put_string(msg, DEVLINK_ATTR_FLASH_UPDATE_STATUS_MSG,
3039                            params->status_msg))
3040                 goto nla_put_failure;
3041         if (params->component &&
3042             nla_put_string(msg, DEVLINK_ATTR_FLASH_UPDATE_COMPONENT,
3043                            params->component))
3044                 goto nla_put_failure;
3045         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_FLASH_UPDATE_STATUS_DONE,
3046                               params->done, DEVLINK_ATTR_PAD))
3047                 goto nla_put_failure;
3048         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_FLASH_UPDATE_STATUS_TOTAL,
3049                               params->total, DEVLINK_ATTR_PAD))
3050                 goto nla_put_failure;
3051         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_FLASH_UPDATE_STATUS_TIMEOUT,
3052                               params->timeout, DEVLINK_ATTR_PAD))
3053                 goto nla_put_failure;
3054
3055 out:
3056         genlmsg_end(msg, hdr);
3057         return 0;
3058
3059 nla_put_failure:
3060         genlmsg_cancel(msg, hdr);
3061         return -EMSGSIZE;
3062 }
3063
3064 static void __devlink_flash_update_notify(struct devlink *devlink,
3065                                           enum devlink_command cmd,
3066                                           struct devlink_flash_notify *params)
3067 {
3068         struct sk_buff *msg;
3069         int err;
3070
3071         WARN_ON(cmd != DEVLINK_CMD_FLASH_UPDATE &&
3072                 cmd != DEVLINK_CMD_FLASH_UPDATE_END &&
3073                 cmd != DEVLINK_CMD_FLASH_UPDATE_STATUS);
3074
3075         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3076         if (!msg)
3077                 return;
3078
3079         err = devlink_nl_flash_update_fill(msg, devlink, cmd, params);
3080         if (err)
3081                 goto out_free_msg;
3082
3083         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
3084                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
3085         return;
3086
3087 out_free_msg:
3088         nlmsg_free(msg);
3089 }
3090
3091 void devlink_flash_update_begin_notify(struct devlink *devlink)
3092 {
3093         struct devlink_flash_notify params = { 0 };
3094
3095         __devlink_flash_update_notify(devlink,
3096                                       DEVLINK_CMD_FLASH_UPDATE,
3097                                       &params);
3098 }
3099 EXPORT_SYMBOL_GPL(devlink_flash_update_begin_notify);
3100
3101 void devlink_flash_update_end_notify(struct devlink *devlink)
3102 {
3103         struct devlink_flash_notify params = { 0 };
3104
3105         __devlink_flash_update_notify(devlink,
3106                                       DEVLINK_CMD_FLASH_UPDATE_END,
3107                                       &params);
3108 }
3109 EXPORT_SYMBOL_GPL(devlink_flash_update_end_notify);
3110
3111 void devlink_flash_update_status_notify(struct devlink *devlink,
3112                                         const char *status_msg,
3113                                         const char *component,
3114                                         unsigned long done,
3115                                         unsigned long total)
3116 {
3117         struct devlink_flash_notify params = {
3118                 .status_msg = status_msg,
3119                 .component = component,
3120                 .done = done,
3121                 .total = total,
3122         };
3123
3124         __devlink_flash_update_notify(devlink,
3125                                       DEVLINK_CMD_FLASH_UPDATE_STATUS,
3126                                       &params);
3127 }
3128 EXPORT_SYMBOL_GPL(devlink_flash_update_status_notify);
3129
3130 void devlink_flash_update_timeout_notify(struct devlink *devlink,
3131                                          const char *status_msg,
3132                                          const char *component,
3133                                          unsigned long timeout)
3134 {
3135         struct devlink_flash_notify params = {
3136                 .status_msg = status_msg,
3137                 .component = component,
3138                 .timeout = timeout,
3139         };
3140
3141         __devlink_flash_update_notify(devlink,
3142                                       DEVLINK_CMD_FLASH_UPDATE_STATUS,
3143                                       &params);
3144 }
3145 EXPORT_SYMBOL_GPL(devlink_flash_update_timeout_notify);
3146
3147 static int devlink_nl_cmd_flash_update(struct sk_buff *skb,
3148                                        struct genl_info *info)
3149 {
3150         struct nlattr *nla_component, *nla_overwrite_mask;
3151         struct devlink_flash_update_params params = {};
3152         struct devlink *devlink = info->user_ptr[0];
3153         u32 supported_params;
3154
3155         if (!devlink->ops->flash_update)
3156                 return -EOPNOTSUPP;
3157
3158         if (!info->attrs[DEVLINK_ATTR_FLASH_UPDATE_FILE_NAME])
3159                 return -EINVAL;
3160
3161         supported_params = devlink->ops->supported_flash_update_params;
3162
3163         params.file_name = nla_data(info->attrs[DEVLINK_ATTR_FLASH_UPDATE_FILE_NAME]);
3164
3165         nla_component = info->attrs[DEVLINK_ATTR_FLASH_UPDATE_COMPONENT];
3166         if (nla_component) {
3167                 if (!(supported_params & DEVLINK_SUPPORT_FLASH_UPDATE_COMPONENT)) {
3168                         NL_SET_ERR_MSG_ATTR(info->extack, nla_component,
3169                                             "component update is not supported by this device");
3170                         return -EOPNOTSUPP;
3171                 }
3172                 params.component = nla_data(nla_component);
3173         }
3174
3175         nla_overwrite_mask = info->attrs[DEVLINK_ATTR_FLASH_UPDATE_OVERWRITE_MASK];
3176         if (nla_overwrite_mask) {
3177                 struct nla_bitfield32 sections;
3178
3179                 if (!(supported_params & DEVLINK_SUPPORT_FLASH_UPDATE_OVERWRITE_MASK)) {
3180                         NL_SET_ERR_MSG_ATTR(info->extack, nla_overwrite_mask,
3181                                             "overwrite settings are not supported by this device");
3182                         return -EOPNOTSUPP;
3183                 }
3184                 sections = nla_get_bitfield32(nla_overwrite_mask);
3185                 params.overwrite_mask = sections.value & sections.selector;
3186         }
3187
3188         return devlink->ops->flash_update(devlink, &params, info->extack);
3189 }
3190
3191 static const struct devlink_param devlink_param_generic[] = {
3192         {
3193                 .id = DEVLINK_PARAM_GENERIC_ID_INT_ERR_RESET,
3194                 .name = DEVLINK_PARAM_GENERIC_INT_ERR_RESET_NAME,
3195                 .type = DEVLINK_PARAM_GENERIC_INT_ERR_RESET_TYPE,
3196         },
3197         {
3198                 .id = DEVLINK_PARAM_GENERIC_ID_MAX_MACS,
3199                 .name = DEVLINK_PARAM_GENERIC_MAX_MACS_NAME,
3200                 .type = DEVLINK_PARAM_GENERIC_MAX_MACS_TYPE,
3201         },
3202         {
3203                 .id = DEVLINK_PARAM_GENERIC_ID_ENABLE_SRIOV,
3204                 .name = DEVLINK_PARAM_GENERIC_ENABLE_SRIOV_NAME,
3205                 .type = DEVLINK_PARAM_GENERIC_ENABLE_SRIOV_TYPE,
3206         },
3207         {
3208                 .id = DEVLINK_PARAM_GENERIC_ID_REGION_SNAPSHOT,
3209                 .name = DEVLINK_PARAM_GENERIC_REGION_SNAPSHOT_NAME,
3210                 .type = DEVLINK_PARAM_GENERIC_REGION_SNAPSHOT_TYPE,
3211         },
3212         {
3213                 .id = DEVLINK_PARAM_GENERIC_ID_IGNORE_ARI,
3214                 .name = DEVLINK_PARAM_GENERIC_IGNORE_ARI_NAME,
3215                 .type = DEVLINK_PARAM_GENERIC_IGNORE_ARI_TYPE,
3216         },
3217         {
3218                 .id = DEVLINK_PARAM_GENERIC_ID_MSIX_VEC_PER_PF_MAX,
3219                 .name = DEVLINK_PARAM_GENERIC_MSIX_VEC_PER_PF_MAX_NAME,
3220                 .type = DEVLINK_PARAM_GENERIC_MSIX_VEC_PER_PF_MAX_TYPE,
3221         },
3222         {
3223                 .id = DEVLINK_PARAM_GENERIC_ID_MSIX_VEC_PER_PF_MIN,
3224                 .name = DEVLINK_PARAM_GENERIC_MSIX_VEC_PER_PF_MIN_NAME,
3225                 .type = DEVLINK_PARAM_GENERIC_MSIX_VEC_PER_PF_MIN_TYPE,
3226         },
3227         {
3228                 .id = DEVLINK_PARAM_GENERIC_ID_FW_LOAD_POLICY,
3229                 .name = DEVLINK_PARAM_GENERIC_FW_LOAD_POLICY_NAME,
3230                 .type = DEVLINK_PARAM_GENERIC_FW_LOAD_POLICY_TYPE,
3231         },
3232         {
3233                 .id = DEVLINK_PARAM_GENERIC_ID_RESET_DEV_ON_DRV_PROBE,
3234                 .name = DEVLINK_PARAM_GENERIC_RESET_DEV_ON_DRV_PROBE_NAME,
3235                 .type = DEVLINK_PARAM_GENERIC_RESET_DEV_ON_DRV_PROBE_TYPE,
3236         },
3237         {
3238                 .id = DEVLINK_PARAM_GENERIC_ID_ENABLE_ROCE,
3239                 .name = DEVLINK_PARAM_GENERIC_ENABLE_ROCE_NAME,
3240                 .type = DEVLINK_PARAM_GENERIC_ENABLE_ROCE_TYPE,
3241         },
3242 };
3243
3244 static int devlink_param_generic_verify(const struct devlink_param *param)
3245 {
3246         /* verify it match generic parameter by id and name */
3247         if (param->id > DEVLINK_PARAM_GENERIC_ID_MAX)
3248                 return -EINVAL;
3249         if (strcmp(param->name, devlink_param_generic[param->id].name))
3250                 return -ENOENT;
3251
3252         WARN_ON(param->type != devlink_param_generic[param->id].type);
3253
3254         return 0;
3255 }
3256
3257 static int devlink_param_driver_verify(const struct devlink_param *param)
3258 {
3259         int i;
3260
3261         if (param->id <= DEVLINK_PARAM_GENERIC_ID_MAX)
3262                 return -EINVAL;
3263         /* verify no such name in generic params */
3264         for (i = 0; i <= DEVLINK_PARAM_GENERIC_ID_MAX; i++)
3265                 if (!strcmp(param->name, devlink_param_generic[i].name))
3266                         return -EEXIST;
3267
3268         return 0;
3269 }
3270
3271 static struct devlink_param_item *
3272 devlink_param_find_by_name(struct list_head *param_list,
3273                            const char *param_name)
3274 {
3275         struct devlink_param_item *param_item;
3276
3277         list_for_each_entry(param_item, param_list, list)
3278                 if (!strcmp(param_item->param->name, param_name))
3279                         return param_item;
3280         return NULL;
3281 }
3282
3283 static struct devlink_param_item *
3284 devlink_param_find_by_id(struct list_head *param_list, u32 param_id)
3285 {
3286         struct devlink_param_item *param_item;
3287
3288         list_for_each_entry(param_item, param_list, list)
3289                 if (param_item->param->id == param_id)
3290                         return param_item;
3291         return NULL;
3292 }
3293
3294 static bool
3295 devlink_param_cmode_is_supported(const struct devlink_param *param,
3296                                  enum devlink_param_cmode cmode)
3297 {
3298         return test_bit(cmode, &param->supported_cmodes);
3299 }
3300
3301 static int devlink_param_get(struct devlink *devlink,
3302                              const struct devlink_param *param,
3303                              struct devlink_param_gset_ctx *ctx)
3304 {
3305         if (!param->get)
3306                 return -EOPNOTSUPP;
3307         return param->get(devlink, param->id, ctx);
3308 }
3309
3310 static int devlink_param_set(struct devlink *devlink,
3311                              const struct devlink_param *param,
3312                              struct devlink_param_gset_ctx *ctx)
3313 {
3314         if (!param->set)
3315                 return -EOPNOTSUPP;
3316         return param->set(devlink, param->id, ctx);
3317 }
3318
3319 static int
3320 devlink_param_type_to_nla_type(enum devlink_param_type param_type)
3321 {
3322         switch (param_type) {
3323         case DEVLINK_PARAM_TYPE_U8:
3324                 return NLA_U8;
3325         case DEVLINK_PARAM_TYPE_U16:
3326                 return NLA_U16;
3327         case DEVLINK_PARAM_TYPE_U32:
3328                 return NLA_U32;
3329         case DEVLINK_PARAM_TYPE_STRING:
3330                 return NLA_STRING;
3331         case DEVLINK_PARAM_TYPE_BOOL:
3332                 return NLA_FLAG;
3333         default:
3334                 return -EINVAL;
3335         }
3336 }
3337
3338 static int
3339 devlink_nl_param_value_fill_one(struct sk_buff *msg,
3340                                 enum devlink_param_type type,
3341                                 enum devlink_param_cmode cmode,
3342                                 union devlink_param_value val)
3343 {
3344         struct nlattr *param_value_attr;
3345
3346         param_value_attr = nla_nest_start_noflag(msg,
3347                                                  DEVLINK_ATTR_PARAM_VALUE);
3348         if (!param_value_attr)
3349                 goto nla_put_failure;
3350
3351         if (nla_put_u8(msg, DEVLINK_ATTR_PARAM_VALUE_CMODE, cmode))
3352                 goto value_nest_cancel;
3353
3354         switch (type) {
3355         case DEVLINK_PARAM_TYPE_U8:
3356                 if (nla_put_u8(msg, DEVLINK_ATTR_PARAM_VALUE_DATA, val.vu8))
3357                         goto value_nest_cancel;
3358                 break;
3359         case DEVLINK_PARAM_TYPE_U16:
3360                 if (nla_put_u16(msg, DEVLINK_ATTR_PARAM_VALUE_DATA, val.vu16))
3361                         goto value_nest_cancel;
3362                 break;
3363         case DEVLINK_PARAM_TYPE_U32:
3364                 if (nla_put_u32(msg, DEVLINK_ATTR_PARAM_VALUE_DATA, val.vu32))
3365                         goto value_nest_cancel;
3366                 break;
3367         case DEVLINK_PARAM_TYPE_STRING:
3368                 if (nla_put_string(msg, DEVLINK_ATTR_PARAM_VALUE_DATA,
3369                                    val.vstr))
3370                         goto value_nest_cancel;
3371                 break;
3372         case DEVLINK_PARAM_TYPE_BOOL:
3373                 if (val.vbool &&
3374                     nla_put_flag(msg, DEVLINK_ATTR_PARAM_VALUE_DATA))
3375                         goto value_nest_cancel;
3376                 break;
3377         }
3378
3379         nla_nest_end(msg, param_value_attr);
3380         return 0;
3381
3382 value_nest_cancel:
3383         nla_nest_cancel(msg, param_value_attr);
3384 nla_put_failure:
3385         return -EMSGSIZE;
3386 }
3387
3388 static int devlink_nl_param_fill(struct sk_buff *msg, struct devlink *devlink,
3389                                  unsigned int port_index,
3390                                  struct devlink_param_item *param_item,
3391                                  enum devlink_command cmd,
3392                                  u32 portid, u32 seq, int flags)
3393 {
3394         union devlink_param_value param_value[DEVLINK_PARAM_CMODE_MAX + 1];
3395         bool param_value_set[DEVLINK_PARAM_CMODE_MAX + 1] = {};
3396         const struct devlink_param *param = param_item->param;
3397         struct devlink_param_gset_ctx ctx;
3398         struct nlattr *param_values_list;
3399         struct nlattr *param_attr;
3400         int nla_type;
3401         void *hdr;
3402         int err;
3403         int i;
3404
3405         /* Get value from driver part to driverinit configuration mode */
3406         for (i = 0; i <= DEVLINK_PARAM_CMODE_MAX; i++) {
3407                 if (!devlink_param_cmode_is_supported(param, i))
3408                         continue;
3409                 if (i == DEVLINK_PARAM_CMODE_DRIVERINIT) {
3410                         if (!param_item->driverinit_value_valid)
3411                                 return -EOPNOTSUPP;
3412                         param_value[i] = param_item->driverinit_value;
3413                 } else {
3414                         if (!param_item->published)
3415                                 continue;
3416                         ctx.cmode = i;
3417                         err = devlink_param_get(devlink, param, &ctx);
3418                         if (err)
3419                                 return err;
3420                         param_value[i] = ctx.val;
3421                 }
3422                 param_value_set[i] = true;
3423         }
3424
3425         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
3426         if (!hdr)
3427                 return -EMSGSIZE;
3428
3429         if (devlink_nl_put_handle(msg, devlink))
3430                 goto genlmsg_cancel;
3431
3432         if (cmd == DEVLINK_CMD_PORT_PARAM_GET ||
3433             cmd == DEVLINK_CMD_PORT_PARAM_NEW ||
3434             cmd == DEVLINK_CMD_PORT_PARAM_DEL)
3435                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_INDEX, port_index))
3436                         goto genlmsg_cancel;
3437
3438         param_attr = nla_nest_start_noflag(msg, DEVLINK_ATTR_PARAM);
3439         if (!param_attr)
3440                 goto genlmsg_cancel;
3441         if (nla_put_string(msg, DEVLINK_ATTR_PARAM_NAME, param->name))
3442                 goto param_nest_cancel;
3443         if (param->generic && nla_put_flag(msg, DEVLINK_ATTR_PARAM_GENERIC))
3444                 goto param_nest_cancel;
3445
3446         nla_type = devlink_param_type_to_nla_type(param->type);
3447         if (nla_type < 0)
3448                 goto param_nest_cancel;
3449         if (nla_put_u8(msg, DEVLINK_ATTR_PARAM_TYPE, nla_type))
3450                 goto param_nest_cancel;
3451
3452         param_values_list = nla_nest_start_noflag(msg,
3453                                                   DEVLINK_ATTR_PARAM_VALUES_LIST);
3454         if (!param_values_list)
3455                 goto param_nest_cancel;
3456
3457         for (i = 0; i <= DEVLINK_PARAM_CMODE_MAX; i++) {
3458                 if (!param_value_set[i])
3459                         continue;
3460                 err = devlink_nl_param_value_fill_one(msg, param->type,
3461                                                       i, param_value[i]);
3462                 if (err)
3463                         goto values_list_nest_cancel;
3464         }
3465
3466         nla_nest_end(msg, param_values_list);
3467         nla_nest_end(msg, param_attr);
3468         genlmsg_end(msg, hdr);
3469         return 0;
3470
3471 values_list_nest_cancel:
3472         nla_nest_end(msg, param_values_list);
3473 param_nest_cancel:
3474         nla_nest_cancel(msg, param_attr);
3475 genlmsg_cancel:
3476         genlmsg_cancel(msg, hdr);
3477         return -EMSGSIZE;
3478 }
3479
3480 static void devlink_param_notify(struct devlink *devlink,
3481                                  unsigned int port_index,
3482                                  struct devlink_param_item *param_item,
3483                                  enum devlink_command cmd)
3484 {
3485         struct sk_buff *msg;
3486         int err;
3487
3488         WARN_ON(cmd != DEVLINK_CMD_PARAM_NEW && cmd != DEVLINK_CMD_PARAM_DEL &&
3489                 cmd != DEVLINK_CMD_PORT_PARAM_NEW &&
3490                 cmd != DEVLINK_CMD_PORT_PARAM_DEL);
3491
3492         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3493         if (!msg)
3494                 return;
3495         err = devlink_nl_param_fill(msg, devlink, port_index, param_item, cmd,
3496                                     0, 0, 0);
3497         if (err) {
3498                 nlmsg_free(msg);
3499                 return;
3500         }
3501
3502         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
3503                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
3504 }
3505
3506 static int devlink_nl_cmd_param_get_dumpit(struct sk_buff *msg,
3507                                            struct netlink_callback *cb)
3508 {
3509         struct devlink_param_item *param_item;
3510         struct devlink *devlink;
3511         int start = cb->args[0];
3512         int idx = 0;
3513         int err = 0;
3514
3515         mutex_lock(&devlink_mutex);
3516         list_for_each_entry(devlink, &devlink_list, list) {
3517                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
3518                         continue;
3519                 mutex_lock(&devlink->lock);
3520                 list_for_each_entry(param_item, &devlink->param_list, list) {
3521                         if (idx < start) {
3522                                 idx++;
3523                                 continue;
3524                         }
3525                         err = devlink_nl_param_fill(msg, devlink, 0, param_item,
3526                                                     DEVLINK_CMD_PARAM_GET,
3527                                                     NETLINK_CB(cb->skb).portid,
3528                                                     cb->nlh->nlmsg_seq,
3529                                                     NLM_F_MULTI);
3530                         if (err == -EOPNOTSUPP) {
3531                                 err = 0;
3532                         } else if (err) {
3533                                 mutex_unlock(&devlink->lock);
3534                                 goto out;
3535                         }
3536                         idx++;
3537                 }
3538                 mutex_unlock(&devlink->lock);
3539         }
3540 out:
3541         mutex_unlock(&devlink_mutex);
3542
3543         if (err != -EMSGSIZE)
3544                 return err;
3545
3546         cb->args[0] = idx;
3547         return msg->len;
3548 }
3549
3550 static int
3551 devlink_param_type_get_from_info(struct genl_info *info,
3552                                  enum devlink_param_type *param_type)
3553 {
3554         if (!info->attrs[DEVLINK_ATTR_PARAM_TYPE])
3555                 return -EINVAL;
3556
3557         switch (nla_get_u8(info->attrs[DEVLINK_ATTR_PARAM_TYPE])) {
3558         case NLA_U8:
3559                 *param_type = DEVLINK_PARAM_TYPE_U8;
3560                 break;
3561         case NLA_U16:
3562                 *param_type = DEVLINK_PARAM_TYPE_U16;
3563                 break;
3564         case NLA_U32:
3565                 *param_type = DEVLINK_PARAM_TYPE_U32;
3566                 break;
3567         case NLA_STRING:
3568                 *param_type = DEVLINK_PARAM_TYPE_STRING;
3569                 break;
3570         case NLA_FLAG:
3571                 *param_type = DEVLINK_PARAM_TYPE_BOOL;
3572                 break;
3573         default:
3574                 return -EINVAL;
3575         }
3576
3577         return 0;
3578 }
3579
3580 static int
3581 devlink_param_value_get_from_info(const struct devlink_param *param,
3582                                   struct genl_info *info,
3583                                   union devlink_param_value *value)
3584 {
3585         struct nlattr *param_data;
3586         int len;
3587
3588         param_data = info->attrs[DEVLINK_ATTR_PARAM_VALUE_DATA];
3589
3590         if (param->type != DEVLINK_PARAM_TYPE_BOOL && !param_data)
3591                 return -EINVAL;
3592
3593         switch (param->type) {
3594         case DEVLINK_PARAM_TYPE_U8:
3595                 if (nla_len(param_data) != sizeof(u8))
3596                         return -EINVAL;
3597                 value->vu8 = nla_get_u8(param_data);
3598                 break;
3599         case DEVLINK_PARAM_TYPE_U16:
3600                 if (nla_len(param_data) != sizeof(u16))
3601                         return -EINVAL;
3602                 value->vu16 = nla_get_u16(param_data);
3603                 break;
3604         case DEVLINK_PARAM_TYPE_U32:
3605                 if (nla_len(param_data) != sizeof(u32))
3606                         return -EINVAL;
3607                 value->vu32 = nla_get_u32(param_data);
3608                 break;
3609         case DEVLINK_PARAM_TYPE_STRING:
3610                 len = strnlen(nla_data(param_data), nla_len(param_data));
3611                 if (len == nla_len(param_data) ||
3612                     len >= __DEVLINK_PARAM_MAX_STRING_VALUE)
3613                         return -EINVAL;
3614                 strcpy(value->vstr, nla_data(param_data));
3615                 break;
3616         case DEVLINK_PARAM_TYPE_BOOL:
3617                 if (param_data && nla_len(param_data))
3618                         return -EINVAL;
3619                 value->vbool = nla_get_flag(param_data);
3620                 break;
3621         }
3622         return 0;
3623 }
3624
3625 static struct devlink_param_item *
3626 devlink_param_get_from_info(struct list_head *param_list,
3627                             struct genl_info *info)
3628 {
3629         char *param_name;
3630
3631         if (!info->attrs[DEVLINK_ATTR_PARAM_NAME])
3632                 return NULL;
3633
3634         param_name = nla_data(info->attrs[DEVLINK_ATTR_PARAM_NAME]);
3635         return devlink_param_find_by_name(param_list, param_name);
3636 }
3637
3638 static int devlink_nl_cmd_param_get_doit(struct sk_buff *skb,
3639                                          struct genl_info *info)
3640 {
3641         struct devlink *devlink = info->user_ptr[0];
3642         struct devlink_param_item *param_item;
3643         struct sk_buff *msg;
3644         int err;
3645
3646         param_item = devlink_param_get_from_info(&devlink->param_list, info);
3647         if (!param_item)
3648                 return -EINVAL;
3649
3650         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3651         if (!msg)
3652                 return -ENOMEM;
3653
3654         err = devlink_nl_param_fill(msg, devlink, 0, param_item,
3655                                     DEVLINK_CMD_PARAM_GET,
3656                                     info->snd_portid, info->snd_seq, 0);
3657         if (err) {
3658                 nlmsg_free(msg);
3659                 return err;
3660         }
3661
3662         return genlmsg_reply(msg, info);
3663 }
3664
3665 static int __devlink_nl_cmd_param_set_doit(struct devlink *devlink,
3666                                            unsigned int port_index,
3667                                            struct list_head *param_list,
3668                                            struct genl_info *info,
3669                                            enum devlink_command cmd)
3670 {
3671         enum devlink_param_type param_type;
3672         struct devlink_param_gset_ctx ctx;
3673         enum devlink_param_cmode cmode;
3674         struct devlink_param_item *param_item;
3675         const struct devlink_param *param;
3676         union devlink_param_value value;
3677         int err = 0;
3678
3679         param_item = devlink_param_get_from_info(param_list, info);
3680         if (!param_item)
3681                 return -EINVAL;
3682         param = param_item->param;
3683         err = devlink_param_type_get_from_info(info, &param_type);
3684         if (err)
3685                 return err;
3686         if (param_type != param->type)
3687                 return -EINVAL;
3688         err = devlink_param_value_get_from_info(param, info, &value);
3689         if (err)
3690                 return err;
3691         if (param->validate) {
3692                 err = param->validate(devlink, param->id, value, info->extack);
3693                 if (err)
3694                         return err;
3695         }
3696
3697         if (!info->attrs[DEVLINK_ATTR_PARAM_VALUE_CMODE])
3698                 return -EINVAL;
3699         cmode = nla_get_u8(info->attrs[DEVLINK_ATTR_PARAM_VALUE_CMODE]);
3700         if (!devlink_param_cmode_is_supported(param, cmode))
3701                 return -EOPNOTSUPP;
3702
3703         if (cmode == DEVLINK_PARAM_CMODE_DRIVERINIT) {
3704                 if (param->type == DEVLINK_PARAM_TYPE_STRING)
3705                         strcpy(param_item->driverinit_value.vstr, value.vstr);
3706                 else
3707                         param_item->driverinit_value = value;
3708                 param_item->driverinit_value_valid = true;
3709         } else {
3710                 if (!param->set)
3711                         return -EOPNOTSUPP;
3712                 ctx.val = value;
3713                 ctx.cmode = cmode;
3714                 err = devlink_param_set(devlink, param, &ctx);
3715                 if (err)
3716                         return err;
3717         }
3718
3719         devlink_param_notify(devlink, port_index, param_item, cmd);
3720         return 0;
3721 }
3722
3723 static int devlink_nl_cmd_param_set_doit(struct sk_buff *skb,
3724                                          struct genl_info *info)
3725 {
3726         struct devlink *devlink = info->user_ptr[0];
3727
3728         return __devlink_nl_cmd_param_set_doit(devlink, 0, &devlink->param_list,
3729                                                info, DEVLINK_CMD_PARAM_NEW);
3730 }
3731
3732 static int devlink_param_register_one(struct devlink *devlink,
3733                                       unsigned int port_index,
3734                                       struct list_head *param_list,
3735                                       const struct devlink_param *param,
3736                                       enum devlink_command cmd)
3737 {
3738         struct devlink_param_item *param_item;
3739
3740         if (devlink_param_find_by_name(param_list, param->name))
3741                 return -EEXIST;
3742
3743         if (param->supported_cmodes == BIT(DEVLINK_PARAM_CMODE_DRIVERINIT))
3744                 WARN_ON(param->get || param->set);
3745         else
3746                 WARN_ON(!param->get || !param->set);
3747
3748         param_item = kzalloc(sizeof(*param_item), GFP_KERNEL);
3749         if (!param_item)
3750                 return -ENOMEM;
3751         param_item->param = param;
3752
3753         list_add_tail(&param_item->list, param_list);
3754         devlink_param_notify(devlink, port_index, param_item, cmd);
3755         return 0;
3756 }
3757
3758 static void devlink_param_unregister_one(struct devlink *devlink,
3759                                          unsigned int port_index,
3760                                          struct list_head *param_list,
3761                                          const struct devlink_param *param,
3762                                          enum devlink_command cmd)
3763 {
3764         struct devlink_param_item *param_item;
3765
3766         param_item = devlink_param_find_by_name(param_list, param->name);
3767         WARN_ON(!param_item);
3768         devlink_param_notify(devlink, port_index, param_item, cmd);
3769         list_del(&param_item->list);
3770         kfree(param_item);
3771 }
3772
3773 static int devlink_nl_cmd_port_param_get_dumpit(struct sk_buff *msg,
3774                                                 struct netlink_callback *cb)
3775 {
3776         struct devlink_param_item *param_item;
3777         struct devlink_port *devlink_port;
3778         struct devlink *devlink;
3779         int start = cb->args[0];
3780         int idx = 0;
3781         int err = 0;
3782
3783         mutex_lock(&devlink_mutex);
3784         list_for_each_entry(devlink, &devlink_list, list) {
3785                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
3786                         continue;
3787                 mutex_lock(&devlink->lock);
3788                 list_for_each_entry(devlink_port, &devlink->port_list, list) {
3789                         list_for_each_entry(param_item,
3790                                             &devlink_port->param_list, list) {
3791                                 if (idx < start) {
3792                                         idx++;
3793                                         continue;
3794                                 }
3795                                 err = devlink_nl_param_fill(msg,
3796                                                 devlink_port->devlink,
3797                                                 devlink_port->index, param_item,
3798                                                 DEVLINK_CMD_PORT_PARAM_GET,
3799                                                 NETLINK_CB(cb->skb).portid,
3800                                                 cb->nlh->nlmsg_seq,
3801                                                 NLM_F_MULTI);
3802                                 if (err == -EOPNOTSUPP) {
3803                                         err = 0;
3804                                 } else if (err) {
3805                                         mutex_unlock(&devlink->lock);
3806                                         goto out;
3807                                 }
3808                                 idx++;
3809                         }
3810                 }
3811                 mutex_unlock(&devlink->lock);
3812         }
3813 out:
3814         mutex_unlock(&devlink_mutex);
3815
3816         if (err != -EMSGSIZE)
3817                 return err;
3818
3819         cb->args[0] = idx;
3820         return msg->len;
3821 }
3822
3823 static int devlink_nl_cmd_port_param_get_doit(struct sk_buff *skb,
3824                                               struct genl_info *info)
3825 {
3826         struct devlink_port *devlink_port = info->user_ptr[0];
3827         struct devlink_param_item *param_item;
3828         struct sk_buff *msg;
3829         int err;
3830
3831         param_item = devlink_param_get_from_info(&devlink_port->param_list,
3832                                                  info);
3833         if (!param_item)
3834                 return -EINVAL;
3835
3836         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3837         if (!msg)
3838                 return -ENOMEM;
3839
3840         err = devlink_nl_param_fill(msg, devlink_port->devlink,
3841                                     devlink_port->index, param_item,
3842                                     DEVLINK_CMD_PORT_PARAM_GET,
3843                                     info->snd_portid, info->snd_seq, 0);
3844         if (err) {
3845                 nlmsg_free(msg);
3846                 return err;
3847         }
3848
3849         return genlmsg_reply(msg, info);
3850 }
3851
3852 static int devlink_nl_cmd_port_param_set_doit(struct sk_buff *skb,
3853                                               struct genl_info *info)
3854 {
3855         struct devlink_port *devlink_port = info->user_ptr[0];
3856
3857         return __devlink_nl_cmd_param_set_doit(devlink_port->devlink,
3858                                                devlink_port->index,
3859                                                &devlink_port->param_list, info,
3860                                                DEVLINK_CMD_PORT_PARAM_NEW);
3861 }
3862
3863 static int devlink_nl_region_snapshot_id_put(struct sk_buff *msg,
3864                                              struct devlink *devlink,
3865                                              struct devlink_snapshot *snapshot)
3866 {
3867         struct nlattr *snap_attr;
3868         int err;
3869
3870         snap_attr = nla_nest_start_noflag(msg, DEVLINK_ATTR_REGION_SNAPSHOT);
3871         if (!snap_attr)
3872                 return -EINVAL;
3873
3874         err = nla_put_u32(msg, DEVLINK_ATTR_REGION_SNAPSHOT_ID, snapshot->id);
3875         if (err)
3876                 goto nla_put_failure;
3877
3878         nla_nest_end(msg, snap_attr);
3879         return 0;
3880
3881 nla_put_failure:
3882         nla_nest_cancel(msg, snap_attr);
3883         return err;
3884 }
3885
3886 static int devlink_nl_region_snapshots_id_put(struct sk_buff *msg,
3887                                               struct devlink *devlink,
3888                                               struct devlink_region *region)
3889 {
3890         struct devlink_snapshot *snapshot;
3891         struct nlattr *snapshots_attr;
3892         int err;
3893
3894         snapshots_attr = nla_nest_start_noflag(msg,
3895                                                DEVLINK_ATTR_REGION_SNAPSHOTS);
3896         if (!snapshots_attr)
3897                 return -EINVAL;
3898
3899         list_for_each_entry(snapshot, &region->snapshot_list, list) {
3900                 err = devlink_nl_region_snapshot_id_put(msg, devlink, snapshot);
3901                 if (err)
3902                         goto nla_put_failure;
3903         }
3904
3905         nla_nest_end(msg, snapshots_attr);
3906         return 0;
3907
3908 nla_put_failure:
3909         nla_nest_cancel(msg, snapshots_attr);
3910         return err;
3911 }
3912
3913 static int devlink_nl_region_fill(struct sk_buff *msg, struct devlink *devlink,
3914                                   enum devlink_command cmd, u32 portid,
3915                                   u32 seq, int flags,
3916                                   struct devlink_region *region)
3917 {
3918         void *hdr;
3919         int err;
3920
3921         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
3922         if (!hdr)
3923                 return -EMSGSIZE;
3924
3925         err = devlink_nl_put_handle(msg, devlink);
3926         if (err)
3927                 goto nla_put_failure;
3928
3929         err = nla_put_string(msg, DEVLINK_ATTR_REGION_NAME, region->ops->name);
3930         if (err)
3931                 goto nla_put_failure;
3932
3933         err = nla_put_u64_64bit(msg, DEVLINK_ATTR_REGION_SIZE,
3934                                 region->size,
3935                                 DEVLINK_ATTR_PAD);
3936         if (err)
3937                 goto nla_put_failure;
3938
3939         err = devlink_nl_region_snapshots_id_put(msg, devlink, region);
3940         if (err)
3941                 goto nla_put_failure;
3942
3943         genlmsg_end(msg, hdr);
3944         return 0;
3945
3946 nla_put_failure:
3947         genlmsg_cancel(msg, hdr);
3948         return err;
3949 }
3950
3951 static struct sk_buff *
3952 devlink_nl_region_notify_build(struct devlink_region *region,
3953                                struct devlink_snapshot *snapshot,
3954                                enum devlink_command cmd, u32 portid, u32 seq)
3955 {
3956         struct devlink *devlink = region->devlink;
3957         struct sk_buff *msg;
3958         void *hdr;
3959         int err;
3960
3961
3962         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3963         if (!msg)
3964                 return ERR_PTR(-ENOMEM);
3965
3966         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, 0, cmd);
3967         if (!hdr) {
3968                 err = -EMSGSIZE;
3969                 goto out_free_msg;
3970         }
3971
3972         err = devlink_nl_put_handle(msg, devlink);
3973         if (err)
3974                 goto out_cancel_msg;
3975
3976         err = nla_put_string(msg, DEVLINK_ATTR_REGION_NAME,
3977                              region->ops->name);
3978         if (err)
3979                 goto out_cancel_msg;
3980
3981         if (snapshot) {
3982                 err = nla_put_u32(msg, DEVLINK_ATTR_REGION_SNAPSHOT_ID,
3983                                   snapshot->id);
3984                 if (err)
3985                         goto out_cancel_msg;
3986         } else {
3987                 err = nla_put_u64_64bit(msg, DEVLINK_ATTR_REGION_SIZE,
3988                                         region->size, DEVLINK_ATTR_PAD);
3989                 if (err)
3990                         goto out_cancel_msg;
3991         }
3992         genlmsg_end(msg, hdr);
3993
3994         return msg;
3995
3996 out_cancel_msg:
3997         genlmsg_cancel(msg, hdr);
3998 out_free_msg:
3999         nlmsg_free(msg);
4000         return ERR_PTR(err);
4001 }
4002
4003 static void devlink_nl_region_notify(struct devlink_region *region,
4004                                      struct devlink_snapshot *snapshot,
4005                                      enum devlink_command cmd)
4006 {
4007         struct devlink *devlink = region->devlink;
4008         struct sk_buff *msg;
4009
4010         WARN_ON(cmd != DEVLINK_CMD_REGION_NEW && cmd != DEVLINK_CMD_REGION_DEL);
4011
4012         msg = devlink_nl_region_notify_build(region, snapshot, cmd, 0, 0);
4013         if (IS_ERR(msg))
4014                 return;
4015
4016         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
4017                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
4018 }
4019
4020 /**
4021  * __devlink_snapshot_id_increment - Increment number of snapshots using an id
4022  *      @devlink: devlink instance
4023  *      @id: the snapshot id
4024  *
4025  *      Track when a new snapshot begins using an id. Load the count for the
4026  *      given id from the snapshot xarray, increment it, and store it back.
4027  *
4028  *      Called when a new snapshot is created with the given id.
4029  *
4030  *      The id *must* have been previously allocated by
4031  *      devlink_region_snapshot_id_get().
4032  *
4033  *      Returns 0 on success, or an error on failure.
4034  */
4035 static int __devlink_snapshot_id_increment(struct devlink *devlink, u32 id)
4036 {
4037         unsigned long count;
4038         void *p;
4039
4040         lockdep_assert_held(&devlink->lock);
4041
4042         p = xa_load(&devlink->snapshot_ids, id);
4043         if (WARN_ON(!p))
4044                 return -EINVAL;
4045
4046         if (WARN_ON(!xa_is_value(p)))
4047                 return -EINVAL;
4048
4049         count = xa_to_value(p);
4050         count++;
4051
4052         return xa_err(xa_store(&devlink->snapshot_ids, id, xa_mk_value(count),
4053                                GFP_KERNEL));
4054 }
4055
4056 /**
4057  * __devlink_snapshot_id_decrement - Decrease number of snapshots using an id
4058  *      @devlink: devlink instance
4059  *      @id: the snapshot id
4060  *
4061  *      Track when a snapshot is deleted and stops using an id. Load the count
4062  *      for the given id from the snapshot xarray, decrement it, and store it
4063  *      back.
4064  *
4065  *      If the count reaches zero, erase this id from the xarray, freeing it
4066  *      up for future re-use by devlink_region_snapshot_id_get().
4067  *
4068  *      Called when a snapshot using the given id is deleted, and when the
4069  *      initial allocator of the id is finished using it.
4070  */
4071 static void __devlink_snapshot_id_decrement(struct devlink *devlink, u32 id)
4072 {
4073         unsigned long count;
4074         void *p;
4075
4076         lockdep_assert_held(&devlink->lock);
4077
4078         p = xa_load(&devlink->snapshot_ids, id);
4079         if (WARN_ON(!p))
4080                 return;
4081
4082         if (WARN_ON(!xa_is_value(p)))
4083                 return;
4084
4085         count = xa_to_value(p);
4086
4087         if (count > 1) {
4088                 count--;
4089                 xa_store(&devlink->snapshot_ids, id, xa_mk_value(count),
4090                          GFP_KERNEL);
4091         } else {
4092                 /* If this was the last user, we can erase this id */
4093                 xa_erase(&devlink->snapshot_ids, id);
4094         }
4095 }
4096
4097 /**
4098  *      __devlink_snapshot_id_insert - Insert a specific snapshot ID
4099  *      @devlink: devlink instance
4100  *      @id: the snapshot id
4101  *
4102  *      Mark the given snapshot id as used by inserting a zero value into the
4103  *      snapshot xarray.
4104  *
4105  *      This must be called while holding the devlink instance lock. Unlike
4106  *      devlink_snapshot_id_get, the initial reference count is zero, not one.
4107  *      It is expected that the id will immediately be used before
4108  *      releasing the devlink instance lock.
4109  *
4110  *      Returns zero on success, or an error code if the snapshot id could not
4111  *      be inserted.
4112  */
4113 static int __devlink_snapshot_id_insert(struct devlink *devlink, u32 id)
4114 {
4115         lockdep_assert_held(&devlink->lock);
4116
4117         if (xa_load(&devlink->snapshot_ids, id))
4118                 return -EEXIST;
4119
4120         return xa_err(xa_store(&devlink->snapshot_ids, id, xa_mk_value(0),
4121                                GFP_KERNEL));
4122 }
4123
4124 /**
4125  *      __devlink_region_snapshot_id_get - get snapshot ID
4126  *      @devlink: devlink instance
4127  *      @id: storage to return snapshot id
4128  *
4129  *      Allocates a new snapshot id. Returns zero on success, or a negative
4130  *      error on failure. Must be called while holding the devlink instance
4131  *      lock.
4132  *
4133  *      Snapshot IDs are tracked using an xarray which stores the number of
4134  *      users of the snapshot id.
4135  *
4136  *      Note that the caller of this function counts as a 'user', in order to
4137  *      avoid race conditions. The caller must release its hold on the
4138  *      snapshot by using devlink_region_snapshot_id_put.
4139  */
4140 static int __devlink_region_snapshot_id_get(struct devlink *devlink, u32 *id)
4141 {
4142         lockdep_assert_held(&devlink->lock);
4143
4144         return xa_alloc(&devlink->snapshot_ids, id, xa_mk_value(1),
4145                         xa_limit_32b, GFP_KERNEL);
4146 }
4147
4148 /**
4149  *      __devlink_region_snapshot_create - create a new snapshot
4150  *      This will add a new snapshot of a region. The snapshot
4151  *      will be stored on the region struct and can be accessed
4152  *      from devlink. This is useful for future analyses of snapshots.
4153  *      Multiple snapshots can be created on a region.
4154  *      The @snapshot_id should be obtained using the getter function.
4155  *
4156  *      Must be called only while holding the devlink instance lock.
4157  *
4158  *      @region: devlink region of the snapshot
4159  *      @data: snapshot data
4160  *      @snapshot_id: snapshot id to be created
4161  */
4162 static int
4163 __devlink_region_snapshot_create(struct devlink_region *region,
4164                                  u8 *data, u32 snapshot_id)
4165 {
4166         struct devlink *devlink = region->devlink;
4167         struct devlink_snapshot *snapshot;
4168         int err;
4169
4170         lockdep_assert_held(&devlink->lock);
4171
4172         /* check if region can hold one more snapshot */
4173         if (region->cur_snapshots == region->max_snapshots)
4174                 return -ENOSPC;
4175
4176         if (devlink_region_snapshot_get_by_id(region, snapshot_id))
4177                 return -EEXIST;
4178
4179         snapshot = kzalloc(sizeof(*snapshot), GFP_KERNEL);
4180         if (!snapshot)
4181                 return -ENOMEM;
4182
4183         err = __devlink_snapshot_id_increment(devlink, snapshot_id);
4184         if (err)
4185                 goto err_snapshot_id_increment;
4186
4187         snapshot->id = snapshot_id;
4188         snapshot->region = region;
4189         snapshot->data = data;
4190
4191         list_add_tail(&snapshot->list, &region->snapshot_list);
4192
4193         region->cur_snapshots++;
4194
4195         devlink_nl_region_notify(region, snapshot, DEVLINK_CMD_REGION_NEW);
4196         return 0;
4197
4198 err_snapshot_id_increment:
4199         kfree(snapshot);
4200         return err;
4201 }
4202
4203 static void devlink_region_snapshot_del(struct devlink_region *region,
4204                                         struct devlink_snapshot *snapshot)
4205 {
4206         struct devlink *devlink = region->devlink;
4207
4208         lockdep_assert_held(&devlink->lock);
4209
4210         devlink_nl_region_notify(region, snapshot, DEVLINK_CMD_REGION_DEL);
4211         region->cur_snapshots--;
4212         list_del(&snapshot->list);
4213         region->ops->destructor(snapshot->data);
4214         __devlink_snapshot_id_decrement(devlink, snapshot->id);
4215         kfree(snapshot);
4216 }
4217
4218 static int devlink_nl_cmd_region_get_doit(struct sk_buff *skb,
4219                                           struct genl_info *info)
4220 {
4221         struct devlink *devlink = info->user_ptr[0];
4222         struct devlink_region *region;
4223         const char *region_name;
4224         struct sk_buff *msg;
4225         int err;
4226
4227         if (!info->attrs[DEVLINK_ATTR_REGION_NAME])
4228                 return -EINVAL;
4229
4230         region_name = nla_data(info->attrs[DEVLINK_ATTR_REGION_NAME]);
4231         region = devlink_region_get_by_name(devlink, region_name);
4232         if (!region)
4233                 return -EINVAL;
4234
4235         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4236         if (!msg)
4237                 return -ENOMEM;
4238
4239         err = devlink_nl_region_fill(msg, devlink, DEVLINK_CMD_REGION_GET,
4240                                      info->snd_portid, info->snd_seq, 0,
4241                                      region);
4242         if (err) {
4243                 nlmsg_free(msg);
4244                 return err;
4245         }
4246
4247         return genlmsg_reply(msg, info);
4248 }
4249
4250 static int devlink_nl_cmd_region_get_dumpit(struct sk_buff *msg,
4251                                             struct netlink_callback *cb)
4252 {
4253         struct devlink_region *region;
4254         struct devlink *devlink;
4255         int start = cb->args[0];
4256         int idx = 0;
4257         int err;
4258
4259         mutex_lock(&devlink_mutex);
4260         list_for_each_entry(devlink, &devlink_list, list) {
4261                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
4262                         continue;
4263
4264                 mutex_lock(&devlink->lock);
4265                 list_for_each_entry(region, &devlink->region_list, list) {
4266                         if (idx < start) {
4267                                 idx++;
4268                                 continue;
4269                         }
4270                         err = devlink_nl_region_fill(msg, devlink,
4271                                                      DEVLINK_CMD_REGION_GET,
4272                                                      NETLINK_CB(cb->skb).portid,
4273                                                      cb->nlh->nlmsg_seq,
4274                                                      NLM_F_MULTI, region);
4275                         if (err) {
4276                                 mutex_unlock(&devlink->lock);
4277                                 goto out;
4278                         }
4279                         idx++;
4280                 }
4281                 mutex_unlock(&devlink->lock);
4282         }
4283 out:
4284         mutex_unlock(&devlink_mutex);
4285         cb->args[0] = idx;
4286         return msg->len;
4287 }
4288
4289 static int devlink_nl_cmd_region_del(struct sk_buff *skb,
4290                                      struct genl_info *info)
4291 {
4292         struct devlink *devlink = info->user_ptr[0];
4293         struct devlink_snapshot *snapshot;
4294         struct devlink_region *region;
4295         const char *region_name;
4296         u32 snapshot_id;
4297
4298         if (!info->attrs[DEVLINK_ATTR_REGION_NAME] ||
4299             !info->attrs[DEVLINK_ATTR_REGION_SNAPSHOT_ID])
4300                 return -EINVAL;
4301
4302         region_name = nla_data(info->attrs[DEVLINK_ATTR_REGION_NAME]);
4303         snapshot_id = nla_get_u32(info->attrs[DEVLINK_ATTR_REGION_SNAPSHOT_ID]);
4304
4305         region = devlink_region_get_by_name(devlink, region_name);
4306         if (!region)
4307                 return -EINVAL;
4308
4309         snapshot = devlink_region_snapshot_get_by_id(region, snapshot_id);
4310         if (!snapshot)
4311                 return -EINVAL;
4312
4313         devlink_region_snapshot_del(region, snapshot);
4314         return 0;
4315 }
4316
4317 static int
4318 devlink_nl_cmd_region_new(struct sk_buff *skb, struct genl_info *info)
4319 {
4320         struct devlink *devlink = info->user_ptr[0];
4321         struct devlink_snapshot *snapshot;
4322         struct nlattr *snapshot_id_attr;
4323         struct devlink_region *region;
4324         const char *region_name;
4325         u32 snapshot_id;
4326         u8 *data;
4327         int err;
4328
4329         if (!info->attrs[DEVLINK_ATTR_REGION_NAME]) {
4330                 NL_SET_ERR_MSG_MOD(info->extack, "No region name provided");
4331                 return -EINVAL;
4332         }
4333
4334         region_name = nla_data(info->attrs[DEVLINK_ATTR_REGION_NAME]);
4335         region = devlink_region_get_by_name(devlink, region_name);
4336         if (!region) {
4337                 NL_SET_ERR_MSG_MOD(info->extack, "The requested region does not exist");
4338                 return -EINVAL;
4339         }
4340
4341         if (!region->ops->snapshot) {
4342                 NL_SET_ERR_MSG_MOD(info->extack, "The requested region does not support taking an immediate snapshot");
4343                 return -EOPNOTSUPP;
4344         }
4345
4346         if (region->cur_snapshots == region->max_snapshots) {
4347                 NL_SET_ERR_MSG_MOD(info->extack, "The region has reached the maximum number of stored snapshots");
4348                 return -ENOSPC;
4349         }
4350
4351         snapshot_id_attr = info->attrs[DEVLINK_ATTR_REGION_SNAPSHOT_ID];
4352         if (snapshot_id_attr) {
4353                 snapshot_id = nla_get_u32(snapshot_id_attr);
4354
4355                 if (devlink_region_snapshot_get_by_id(region, snapshot_id)) {
4356                         NL_SET_ERR_MSG_MOD(info->extack, "The requested snapshot id is already in use");
4357                         return -EEXIST;
4358                 }
4359
4360                 err = __devlink_snapshot_id_insert(devlink, snapshot_id);
4361                 if (err)
4362                         return err;
4363         } else {
4364                 err = __devlink_region_snapshot_id_get(devlink, &snapshot_id);
4365                 if (err) {
4366                         NL_SET_ERR_MSG_MOD(info->extack, "Failed to allocate a new snapshot id");
4367                         return err;
4368                 }
4369         }
4370
4371         err = region->ops->snapshot(devlink, region->ops, info->extack, &data);
4372         if (err)
4373                 goto err_snapshot_capture;
4374
4375         err = __devlink_region_snapshot_create(region, data, snapshot_id);
4376         if (err)
4377                 goto err_snapshot_create;
4378
4379         if (!snapshot_id_attr) {
4380                 struct sk_buff *msg;
4381
4382                 snapshot = devlink_region_snapshot_get_by_id(region,
4383                                                              snapshot_id);
4384                 if (WARN_ON(!snapshot))
4385                         return -EINVAL;
4386
4387                 msg = devlink_nl_region_notify_build(region, snapshot,
4388                                                      DEVLINK_CMD_REGION_NEW,
4389                                                      info->snd_portid,
4390                                                      info->snd_seq);
4391                 err = PTR_ERR_OR_ZERO(msg);
4392                 if (err)
4393                         goto err_notify;
4394
4395                 err = genlmsg_reply(msg, info);
4396                 if (err)
4397                         goto err_notify;
4398         }
4399
4400         return 0;
4401
4402 err_snapshot_create:
4403         region->ops->destructor(data);
4404 err_snapshot_capture:
4405         __devlink_snapshot_id_decrement(devlink, snapshot_id);
4406         return err;
4407
4408 err_notify:
4409         devlink_region_snapshot_del(region, snapshot);
4410         return err;
4411 }
4412
4413 static int devlink_nl_cmd_region_read_chunk_fill(struct sk_buff *msg,
4414                                                  struct devlink *devlink,
4415                                                  u8 *chunk, u32 chunk_size,
4416                                                  u64 addr)
4417 {
4418         struct nlattr *chunk_attr;
4419         int err;
4420
4421         chunk_attr = nla_nest_start_noflag(msg, DEVLINK_ATTR_REGION_CHUNK);
4422         if (!chunk_attr)
4423                 return -EINVAL;
4424
4425         err = nla_put(msg, DEVLINK_ATTR_REGION_CHUNK_DATA, chunk_size, chunk);
4426         if (err)
4427                 goto nla_put_failure;
4428
4429         err = nla_put_u64_64bit(msg, DEVLINK_ATTR_REGION_CHUNK_ADDR, addr,
4430                                 DEVLINK_ATTR_PAD);
4431         if (err)
4432                 goto nla_put_failure;
4433
4434         nla_nest_end(msg, chunk_attr);
4435         return 0;
4436
4437 nla_put_failure:
4438         nla_nest_cancel(msg, chunk_attr);
4439         return err;
4440 }
4441
4442 #define DEVLINK_REGION_READ_CHUNK_SIZE 256
4443
4444 static int devlink_nl_region_read_snapshot_fill(struct sk_buff *skb,
4445                                                 struct devlink *devlink,
4446                                                 struct devlink_region *region,
4447                                                 struct nlattr **attrs,
4448                                                 u64 start_offset,
4449                                                 u64 end_offset,
4450                                                 u64 *new_offset)
4451 {
4452         struct devlink_snapshot *snapshot;
4453         u64 curr_offset = start_offset;
4454         u32 snapshot_id;
4455         int err = 0;
4456
4457         *new_offset = start_offset;
4458
4459         snapshot_id = nla_get_u32(attrs[DEVLINK_ATTR_REGION_SNAPSHOT_ID]);
4460         snapshot = devlink_region_snapshot_get_by_id(region, snapshot_id);
4461         if (!snapshot)
4462                 return -EINVAL;
4463
4464         while (curr_offset < end_offset) {
4465                 u32 data_size;
4466                 u8 *data;
4467
4468                 if (end_offset - curr_offset < DEVLINK_REGION_READ_CHUNK_SIZE)
4469                         data_size = end_offset - curr_offset;
4470                 else
4471                         data_size = DEVLINK_REGION_READ_CHUNK_SIZE;
4472
4473                 data = &snapshot->data[curr_offset];
4474                 err = devlink_nl_cmd_region_read_chunk_fill(skb, devlink,
4475                                                             data, data_size,
4476                                                             curr_offset);
4477                 if (err)
4478                         break;
4479
4480                 curr_offset += data_size;
4481         }
4482         *new_offset = curr_offset;
4483
4484         return err;
4485 }
4486
4487 static int devlink_nl_cmd_region_read_dumpit(struct sk_buff *skb,
4488                                              struct netlink_callback *cb)
4489 {
4490         const struct genl_dumpit_info *info = genl_dumpit_info(cb);
4491         u64 ret_offset, start_offset, end_offset = U64_MAX;
4492         struct nlattr **attrs = info->attrs;
4493         struct devlink_region *region;
4494         struct nlattr *chunks_attr;
4495         const char *region_name;
4496         struct devlink *devlink;
4497         void *hdr;
4498         int err;
4499
4500         start_offset = *((u64 *)&cb->args[0]);
4501
4502         mutex_lock(&devlink_mutex);
4503         devlink = devlink_get_from_attrs(sock_net(cb->skb->sk), attrs);
4504         if (IS_ERR(devlink)) {
4505                 err = PTR_ERR(devlink);
4506                 goto out_dev;
4507         }
4508
4509         mutex_lock(&devlink->lock);
4510
4511         if (!attrs[DEVLINK_ATTR_REGION_NAME] ||
4512             !attrs[DEVLINK_ATTR_REGION_SNAPSHOT_ID]) {
4513                 err = -EINVAL;
4514                 goto out_unlock;
4515         }
4516
4517         region_name = nla_data(attrs[DEVLINK_ATTR_REGION_NAME]);
4518         region = devlink_region_get_by_name(devlink, region_name);
4519         if (!region) {
4520                 err = -EINVAL;
4521                 goto out_unlock;
4522         }
4523
4524         if (attrs[DEVLINK_ATTR_REGION_CHUNK_ADDR] &&
4525             attrs[DEVLINK_ATTR_REGION_CHUNK_LEN]) {
4526                 if (!start_offset)
4527                         start_offset =
4528                                 nla_get_u64(attrs[DEVLINK_ATTR_REGION_CHUNK_ADDR]);
4529
4530                 end_offset = nla_get_u64(attrs[DEVLINK_ATTR_REGION_CHUNK_ADDR]);
4531                 end_offset += nla_get_u64(attrs[DEVLINK_ATTR_REGION_CHUNK_LEN]);
4532         }
4533
4534         if (end_offset > region->size)
4535                 end_offset = region->size;
4536
4537         /* return 0 if there is no further data to read */
4538         if (start_offset == end_offset) {
4539                 err = 0;
4540                 goto out_unlock;
4541         }
4542
4543         hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
4544                           &devlink_nl_family, NLM_F_ACK | NLM_F_MULTI,
4545                           DEVLINK_CMD_REGION_READ);
4546         if (!hdr) {
4547                 err = -EMSGSIZE;
4548                 goto out_unlock;
4549         }
4550
4551         err = devlink_nl_put_handle(skb, devlink);
4552         if (err)
4553                 goto nla_put_failure;
4554
4555         err = nla_put_string(skb, DEVLINK_ATTR_REGION_NAME, region_name);
4556         if (err)
4557                 goto nla_put_failure;
4558
4559         chunks_attr = nla_nest_start_noflag(skb, DEVLINK_ATTR_REGION_CHUNKS);
4560         if (!chunks_attr) {
4561                 err = -EMSGSIZE;
4562                 goto nla_put_failure;
4563         }
4564
4565         err = devlink_nl_region_read_snapshot_fill(skb, devlink,
4566                                                    region, attrs,
4567                                                    start_offset,
4568                                                    end_offset, &ret_offset);
4569
4570         if (err && err != -EMSGSIZE)
4571                 goto nla_put_failure;
4572
4573         /* Check if there was any progress done to prevent infinite loop */
4574         if (ret_offset == start_offset) {
4575                 err = -EINVAL;
4576                 goto nla_put_failure;
4577         }
4578
4579         *((u64 *)&cb->args[0]) = ret_offset;
4580
4581         nla_nest_end(skb, chunks_attr);
4582         genlmsg_end(skb, hdr);
4583         mutex_unlock(&devlink->lock);
4584         mutex_unlock(&devlink_mutex);
4585
4586         return skb->len;
4587
4588 nla_put_failure:
4589         genlmsg_cancel(skb, hdr);
4590 out_unlock:
4591         mutex_unlock(&devlink->lock);
4592 out_dev:
4593         mutex_unlock(&devlink_mutex);
4594         return err;
4595 }
4596
4597 struct devlink_info_req {
4598         struct sk_buff *msg;
4599 };
4600
4601 int devlink_info_driver_name_put(struct devlink_info_req *req, const char *name)
4602 {
4603         return nla_put_string(req->msg, DEVLINK_ATTR_INFO_DRIVER_NAME, name);
4604 }
4605 EXPORT_SYMBOL_GPL(devlink_info_driver_name_put);
4606
4607 int devlink_info_serial_number_put(struct devlink_info_req *req, const char *sn)
4608 {
4609         return nla_put_string(req->msg, DEVLINK_ATTR_INFO_SERIAL_NUMBER, sn);
4610 }
4611 EXPORT_SYMBOL_GPL(devlink_info_serial_number_put);
4612
4613 int devlink_info_board_serial_number_put(struct devlink_info_req *req,
4614                                          const char *bsn)
4615 {
4616         return nla_put_string(req->msg, DEVLINK_ATTR_INFO_BOARD_SERIAL_NUMBER,
4617                               bsn);
4618 }
4619 EXPORT_SYMBOL_GPL(devlink_info_board_serial_number_put);
4620
4621 static int devlink_info_version_put(struct devlink_info_req *req, int attr,
4622                                     const char *version_name,
4623                                     const char *version_value)
4624 {
4625         struct nlattr *nest;
4626         int err;
4627
4628         nest = nla_nest_start_noflag(req->msg, attr);
4629         if (!nest)
4630                 return -EMSGSIZE;
4631
4632         err = nla_put_string(req->msg, DEVLINK_ATTR_INFO_VERSION_NAME,
4633                              version_name);
4634         if (err)
4635                 goto nla_put_failure;
4636
4637         err = nla_put_string(req->msg, DEVLINK_ATTR_INFO_VERSION_VALUE,
4638                              version_value);
4639         if (err)
4640                 goto nla_put_failure;
4641
4642         nla_nest_end(req->msg, nest);
4643
4644         return 0;
4645
4646 nla_put_failure:
4647         nla_nest_cancel(req->msg, nest);
4648         return err;
4649 }
4650
4651 int devlink_info_version_fixed_put(struct devlink_info_req *req,
4652                                    const char *version_name,
4653                                    const char *version_value)
4654 {
4655         return devlink_info_version_put(req, DEVLINK_ATTR_INFO_VERSION_FIXED,
4656                                         version_name, version_value);
4657 }
4658 EXPORT_SYMBOL_GPL(devlink_info_version_fixed_put);
4659
4660 int devlink_info_version_stored_put(struct devlink_info_req *req,
4661                                     const char *version_name,
4662                                     const char *version_value)
4663 {
4664         return devlink_info_version_put(req, DEVLINK_ATTR_INFO_VERSION_STORED,
4665                                         version_name, version_value);
4666 }
4667 EXPORT_SYMBOL_GPL(devlink_info_version_stored_put);
4668
4669 int devlink_info_version_running_put(struct devlink_info_req *req,
4670                                      const char *version_name,
4671                                      const char *version_value)
4672 {
4673         return devlink_info_version_put(req, DEVLINK_ATTR_INFO_VERSION_RUNNING,
4674                                         version_name, version_value);
4675 }
4676 EXPORT_SYMBOL_GPL(devlink_info_version_running_put);
4677
4678 static int
4679 devlink_nl_info_fill(struct sk_buff *msg, struct devlink *devlink,
4680                      enum devlink_command cmd, u32 portid,
4681                      u32 seq, int flags, struct netlink_ext_ack *extack)
4682 {
4683         struct devlink_info_req req;
4684         void *hdr;
4685         int err;
4686
4687         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
4688         if (!hdr)
4689                 return -EMSGSIZE;
4690
4691         err = -EMSGSIZE;
4692         if (devlink_nl_put_handle(msg, devlink))
4693                 goto err_cancel_msg;
4694
4695         req.msg = msg;
4696         err = devlink->ops->info_get(devlink, &req, extack);
4697         if (err)
4698                 goto err_cancel_msg;
4699
4700         genlmsg_end(msg, hdr);
4701         return 0;
4702
4703 err_cancel_msg:
4704         genlmsg_cancel(msg, hdr);
4705         return err;
4706 }
4707
4708 static int devlink_nl_cmd_info_get_doit(struct sk_buff *skb,
4709                                         struct genl_info *info)
4710 {
4711         struct devlink *devlink = info->user_ptr[0];
4712         struct sk_buff *msg;
4713         int err;
4714
4715         if (!devlink->ops->info_get)
4716                 return -EOPNOTSUPP;
4717
4718         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4719         if (!msg)
4720                 return -ENOMEM;
4721
4722         err = devlink_nl_info_fill(msg, devlink, DEVLINK_CMD_INFO_GET,
4723                                    info->snd_portid, info->snd_seq, 0,
4724                                    info->extack);
4725         if (err) {
4726                 nlmsg_free(msg);
4727                 return err;
4728         }
4729
4730         return genlmsg_reply(msg, info);
4731 }
4732
4733 static int devlink_nl_cmd_info_get_dumpit(struct sk_buff *msg,
4734                                           struct netlink_callback *cb)
4735 {
4736         struct devlink *devlink;
4737         int start = cb->args[0];
4738         int idx = 0;
4739         int err = 0;
4740
4741         mutex_lock(&devlink_mutex);
4742         list_for_each_entry(devlink, &devlink_list, list) {
4743                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
4744                         continue;
4745                 if (idx < start) {
4746                         idx++;
4747                         continue;
4748                 }
4749
4750                 if (!devlink->ops->info_get) {
4751                         idx++;
4752                         continue;
4753                 }
4754
4755                 mutex_lock(&devlink->lock);
4756                 err = devlink_nl_info_fill(msg, devlink, DEVLINK_CMD_INFO_GET,
4757                                            NETLINK_CB(cb->skb).portid,
4758                                            cb->nlh->nlmsg_seq, NLM_F_MULTI,
4759                                            cb->extack);
4760                 mutex_unlock(&devlink->lock);
4761                 if (err == -EOPNOTSUPP)
4762                         err = 0;
4763                 else if (err)
4764                         break;
4765                 idx++;
4766         }
4767         mutex_unlock(&devlink_mutex);
4768
4769         if (err != -EMSGSIZE)
4770                 return err;
4771
4772         cb->args[0] = idx;
4773         return msg->len;
4774 }
4775
4776 struct devlink_fmsg_item {
4777         struct list_head list;
4778         int attrtype;
4779         u8 nla_type;
4780         u16 len;
4781         int value[];
4782 };
4783
4784 struct devlink_fmsg {
4785         struct list_head item_list;
4786         bool putting_binary; /* This flag forces enclosing of binary data
4787                               * in an array brackets. It forces using
4788                               * of designated API:
4789                               * devlink_fmsg_binary_pair_nest_start()
4790                               * devlink_fmsg_binary_pair_nest_end()
4791                               */
4792 };
4793
4794 static struct devlink_fmsg *devlink_fmsg_alloc(void)
4795 {
4796         struct devlink_fmsg *fmsg;
4797
4798         fmsg = kzalloc(sizeof(*fmsg), GFP_KERNEL);
4799         if (!fmsg)
4800                 return NULL;
4801
4802         INIT_LIST_HEAD(&fmsg->item_list);
4803
4804         return fmsg;
4805 }
4806
4807 static void devlink_fmsg_free(struct devlink_fmsg *fmsg)
4808 {
4809         struct devlink_fmsg_item *item, *tmp;
4810
4811         list_for_each_entry_safe(item, tmp, &fmsg->item_list, list) {
4812                 list_del(&item->list);
4813                 kfree(item);
4814         }
4815         kfree(fmsg);
4816 }
4817
4818 static int devlink_fmsg_nest_common(struct devlink_fmsg *fmsg,
4819                                     int attrtype)
4820 {
4821         struct devlink_fmsg_item *item;
4822
4823         item = kzalloc(sizeof(*item), GFP_KERNEL);
4824         if (!item)
4825                 return -ENOMEM;
4826
4827         item->attrtype = attrtype;
4828         list_add_tail(&item->list, &fmsg->item_list);
4829
4830         return 0;
4831 }
4832
4833 int devlink_fmsg_obj_nest_start(struct devlink_fmsg *fmsg)
4834 {
4835         if (fmsg->putting_binary)
4836                 return -EINVAL;
4837
4838         return devlink_fmsg_nest_common(fmsg, DEVLINK_ATTR_FMSG_OBJ_NEST_START);
4839 }
4840 EXPORT_SYMBOL_GPL(devlink_fmsg_obj_nest_start);
4841
4842 static int devlink_fmsg_nest_end(struct devlink_fmsg *fmsg)
4843 {
4844         if (fmsg->putting_binary)
4845                 return -EINVAL;
4846
4847         return devlink_fmsg_nest_common(fmsg, DEVLINK_ATTR_FMSG_NEST_END);
4848 }
4849
4850 int devlink_fmsg_obj_nest_end(struct devlink_fmsg *fmsg)
4851 {
4852         if (fmsg->putting_binary)
4853                 return -EINVAL;
4854
4855         return devlink_fmsg_nest_end(fmsg);
4856 }
4857 EXPORT_SYMBOL_GPL(devlink_fmsg_obj_nest_end);
4858
4859 #define DEVLINK_FMSG_MAX_SIZE (GENLMSG_DEFAULT_SIZE - GENL_HDRLEN - NLA_HDRLEN)
4860
4861 static int devlink_fmsg_put_name(struct devlink_fmsg *fmsg, const char *name)
4862 {
4863         struct devlink_fmsg_item *item;
4864
4865         if (fmsg->putting_binary)
4866                 return -EINVAL;
4867
4868         if (strlen(name) + 1 > DEVLINK_FMSG_MAX_SIZE)
4869                 return -EMSGSIZE;
4870
4871         item = kzalloc(sizeof(*item) + strlen(name) + 1, GFP_KERNEL);
4872         if (!item)
4873                 return -ENOMEM;
4874
4875         item->nla_type = NLA_NUL_STRING;
4876         item->len = strlen(name) + 1;
4877         item->attrtype = DEVLINK_ATTR_FMSG_OBJ_NAME;
4878         memcpy(&item->value, name, item->len);
4879         list_add_tail(&item->list, &fmsg->item_list);
4880
4881         return 0;
4882 }
4883
4884 int devlink_fmsg_pair_nest_start(struct devlink_fmsg *fmsg, const char *name)
4885 {
4886         int err;
4887
4888         if (fmsg->putting_binary)
4889                 return -EINVAL;
4890
4891         err = devlink_fmsg_nest_common(fmsg, DEVLINK_ATTR_FMSG_PAIR_NEST_START);
4892         if (err)
4893                 return err;
4894
4895         err = devlink_fmsg_put_name(fmsg, name);
4896         if (err)
4897                 return err;
4898
4899         return 0;
4900 }
4901 EXPORT_SYMBOL_GPL(devlink_fmsg_pair_nest_start);
4902
4903 int devlink_fmsg_pair_nest_end(struct devlink_fmsg *fmsg)
4904 {
4905         if (fmsg->putting_binary)
4906                 return -EINVAL;
4907
4908         return devlink_fmsg_nest_end(fmsg);
4909 }
4910 EXPORT_SYMBOL_GPL(devlink_fmsg_pair_nest_end);
4911
4912 int devlink_fmsg_arr_pair_nest_start(struct devlink_fmsg *fmsg,
4913                                      const char *name)
4914 {
4915         int err;
4916
4917         if (fmsg->putting_binary)
4918                 return -EINVAL;
4919
4920         err = devlink_fmsg_pair_nest_start(fmsg, name);
4921         if (err)
4922                 return err;
4923
4924         err = devlink_fmsg_nest_common(fmsg, DEVLINK_ATTR_FMSG_ARR_NEST_START);
4925         if (err)
4926                 return err;
4927
4928         return 0;
4929 }
4930 EXPORT_SYMBOL_GPL(devlink_fmsg_arr_pair_nest_start);
4931
4932 int devlink_fmsg_arr_pair_nest_end(struct devlink_fmsg *fmsg)
4933 {
4934         int err;
4935
4936         if (fmsg->putting_binary)
4937                 return -EINVAL;
4938
4939         err = devlink_fmsg_nest_end(fmsg);
4940         if (err)
4941                 return err;
4942
4943         err = devlink_fmsg_nest_end(fmsg);
4944         if (err)
4945                 return err;
4946
4947         return 0;
4948 }
4949 EXPORT_SYMBOL_GPL(devlink_fmsg_arr_pair_nest_end);
4950
4951 int devlink_fmsg_binary_pair_nest_start(struct devlink_fmsg *fmsg,
4952                                         const char *name)
4953 {
4954         int err;
4955
4956         err = devlink_fmsg_arr_pair_nest_start(fmsg, name);
4957         if (err)
4958                 return err;
4959
4960         fmsg->putting_binary = true;
4961         return err;
4962 }
4963 EXPORT_SYMBOL_GPL(devlink_fmsg_binary_pair_nest_start);
4964
4965 int devlink_fmsg_binary_pair_nest_end(struct devlink_fmsg *fmsg)
4966 {
4967         if (!fmsg->putting_binary)
4968                 return -EINVAL;
4969
4970         fmsg->putting_binary = false;
4971         return devlink_fmsg_arr_pair_nest_end(fmsg);
4972 }
4973 EXPORT_SYMBOL_GPL(devlink_fmsg_binary_pair_nest_end);
4974
4975 static int devlink_fmsg_put_value(struct devlink_fmsg *fmsg,
4976                                   const void *value, u16 value_len,
4977                                   u8 value_nla_type)
4978 {
4979         struct devlink_fmsg_item *item;
4980
4981         if (value_len > DEVLINK_FMSG_MAX_SIZE)
4982                 return -EMSGSIZE;
4983
4984         item = kzalloc(sizeof(*item) + value_len, GFP_KERNEL);
4985         if (!item)
4986                 return -ENOMEM;
4987
4988         item->nla_type = value_nla_type;
4989         item->len = value_len;
4990         item->attrtype = DEVLINK_ATTR_FMSG_OBJ_VALUE_DATA;
4991         memcpy(&item->value, value, item->len);
4992         list_add_tail(&item->list, &fmsg->item_list);
4993
4994         return 0;
4995 }
4996
4997 int devlink_fmsg_bool_put(struct devlink_fmsg *fmsg, bool value)
4998 {
4999         if (fmsg->putting_binary)
5000                 return -EINVAL;
5001
5002         return devlink_fmsg_put_value(fmsg, &value, sizeof(value), NLA_FLAG);
5003 }
5004 EXPORT_SYMBOL_GPL(devlink_fmsg_bool_put);
5005
5006 int devlink_fmsg_u8_put(struct devlink_fmsg *fmsg, u8 value)
5007 {
5008         if (fmsg->putting_binary)
5009                 return -EINVAL;
5010
5011         return devlink_fmsg_put_value(fmsg, &value, sizeof(value), NLA_U8);
5012 }
5013 EXPORT_SYMBOL_GPL(devlink_fmsg_u8_put);
5014
5015 int devlink_fmsg_u32_put(struct devlink_fmsg *fmsg, u32 value)
5016 {
5017         if (fmsg->putting_binary)
5018                 return -EINVAL;
5019
5020         return devlink_fmsg_put_value(fmsg, &value, sizeof(value), NLA_U32);
5021 }
5022 EXPORT_SYMBOL_GPL(devlink_fmsg_u32_put);
5023
5024 int devlink_fmsg_u64_put(struct devlink_fmsg *fmsg, u64 value)
5025 {
5026         if (fmsg->putting_binary)
5027                 return -EINVAL;
5028
5029         return devlink_fmsg_put_value(fmsg, &value, sizeof(value), NLA_U64);
5030 }
5031 EXPORT_SYMBOL_GPL(devlink_fmsg_u64_put);
5032
5033 int devlink_fmsg_string_put(struct devlink_fmsg *fmsg, const char *value)
5034 {
5035         if (fmsg->putting_binary)
5036                 return -EINVAL;
5037
5038         return devlink_fmsg_put_value(fmsg, value, strlen(value) + 1,
5039                                       NLA_NUL_STRING);
5040 }
5041 EXPORT_SYMBOL_GPL(devlink_fmsg_string_put);
5042
5043 int devlink_fmsg_binary_put(struct devlink_fmsg *fmsg, const void *value,
5044                             u16 value_len)
5045 {
5046         if (!fmsg->putting_binary)
5047                 return -EINVAL;
5048
5049         return devlink_fmsg_put_value(fmsg, value, value_len, NLA_BINARY);
5050 }
5051 EXPORT_SYMBOL_GPL(devlink_fmsg_binary_put);
5052
5053 int devlink_fmsg_bool_pair_put(struct devlink_fmsg *fmsg, const char *name,
5054                                bool value)
5055 {
5056         int err;
5057
5058         err = devlink_fmsg_pair_nest_start(fmsg, name);
5059         if (err)
5060                 return err;
5061
5062         err = devlink_fmsg_bool_put(fmsg, value);
5063         if (err)
5064                 return err;
5065
5066         err = devlink_fmsg_pair_nest_end(fmsg);
5067         if (err)
5068                 return err;
5069
5070         return 0;
5071 }
5072 EXPORT_SYMBOL_GPL(devlink_fmsg_bool_pair_put);
5073
5074 int devlink_fmsg_u8_pair_put(struct devlink_fmsg *fmsg, const char *name,
5075                              u8 value)
5076 {
5077         int err;
5078
5079         err = devlink_fmsg_pair_nest_start(fmsg, name);
5080         if (err)
5081                 return err;
5082
5083         err = devlink_fmsg_u8_put(fmsg, value);
5084         if (err)
5085                 return err;
5086
5087         err = devlink_fmsg_pair_nest_end(fmsg);
5088         if (err)
5089                 return err;
5090
5091         return 0;
5092 }
5093 EXPORT_SYMBOL_GPL(devlink_fmsg_u8_pair_put);
5094
5095 int devlink_fmsg_u32_pair_put(struct devlink_fmsg *fmsg, const char *name,
5096                               u32 value)
5097 {
5098         int err;
5099
5100         err = devlink_fmsg_pair_nest_start(fmsg, name);
5101         if (err)
5102                 return err;
5103
5104         err = devlink_fmsg_u32_put(fmsg, value);
5105         if (err)
5106                 return err;
5107
5108         err = devlink_fmsg_pair_nest_end(fmsg);
5109         if (err)
5110                 return err;
5111
5112         return 0;
5113 }
5114 EXPORT_SYMBOL_GPL(devlink_fmsg_u32_pair_put);
5115
5116 int devlink_fmsg_u64_pair_put(struct devlink_fmsg *fmsg, const char *name,
5117                               u64 value)
5118 {
5119         int err;
5120
5121         err = devlink_fmsg_pair_nest_start(fmsg, name);
5122         if (err)
5123                 return err;
5124
5125         err = devlink_fmsg_u64_put(fmsg, value);
5126         if (err)
5127                 return err;
5128
5129         err = devlink_fmsg_pair_nest_end(fmsg);
5130         if (err)
5131                 return err;
5132
5133         return 0;
5134 }
5135 EXPORT_SYMBOL_GPL(devlink_fmsg_u64_pair_put);
5136
5137 int devlink_fmsg_string_pair_put(struct devlink_fmsg *fmsg, const char *name,
5138                                  const char *value)
5139 {
5140         int err;
5141
5142         err = devlink_fmsg_pair_nest_start(fmsg, name);
5143         if (err)
5144                 return err;
5145
5146         err = devlink_fmsg_string_put(fmsg, value);
5147         if (err)
5148                 return err;
5149
5150         err = devlink_fmsg_pair_nest_end(fmsg);
5151         if (err)
5152                 return err;
5153
5154         return 0;
5155 }
5156 EXPORT_SYMBOL_GPL(devlink_fmsg_string_pair_put);
5157
5158 int devlink_fmsg_binary_pair_put(struct devlink_fmsg *fmsg, const char *name,
5159                                  const void *value, u32 value_len)
5160 {
5161         u32 data_size;
5162         int end_err;
5163         u32 offset;
5164         int err;
5165
5166         err = devlink_fmsg_binary_pair_nest_start(fmsg, name);
5167         if (err)
5168                 return err;
5169
5170         for (offset = 0; offset < value_len; offset += data_size) {
5171                 data_size = value_len - offset;
5172                 if (data_size > DEVLINK_FMSG_MAX_SIZE)
5173                         data_size = DEVLINK_FMSG_MAX_SIZE;
5174                 err = devlink_fmsg_binary_put(fmsg, value + offset, data_size);
5175                 if (err)
5176                         break;
5177                 /* Exit from loop with a break (instead of
5178                  * return) to make sure putting_binary is turned off in
5179                  * devlink_fmsg_binary_pair_nest_end
5180                  */
5181         }
5182
5183         end_err = devlink_fmsg_binary_pair_nest_end(fmsg);
5184         if (end_err)
5185                 err = end_err;
5186
5187         return err;
5188 }
5189 EXPORT_SYMBOL_GPL(devlink_fmsg_binary_pair_put);
5190
5191 static int
5192 devlink_fmsg_item_fill_type(struct devlink_fmsg_item *msg, struct sk_buff *skb)
5193 {
5194         switch (msg->nla_type) {
5195         case NLA_FLAG:
5196         case NLA_U8:
5197         case NLA_U32:
5198         case NLA_U64:
5199         case NLA_NUL_STRING:
5200         case NLA_BINARY:
5201                 return nla_put_u8(skb, DEVLINK_ATTR_FMSG_OBJ_VALUE_TYPE,
5202                                   msg->nla_type);
5203         default:
5204                 return -EINVAL;
5205         }
5206 }
5207
5208 static int
5209 devlink_fmsg_item_fill_data(struct devlink_fmsg_item *msg, struct sk_buff *skb)
5210 {
5211         int attrtype = DEVLINK_ATTR_FMSG_OBJ_VALUE_DATA;
5212         u8 tmp;
5213
5214         switch (msg->nla_type) {
5215         case NLA_FLAG:
5216                 /* Always provide flag data, regardless of its value */
5217                 tmp = *(bool *) msg->value;
5218
5219                 return nla_put_u8(skb, attrtype, tmp);
5220         case NLA_U8:
5221                 return nla_put_u8(skb, attrtype, *(u8 *) msg->value);
5222         case NLA_U32:
5223                 return nla_put_u32(skb, attrtype, *(u32 *) msg->value);
5224         case NLA_U64:
5225                 return nla_put_u64_64bit(skb, attrtype, *(u64 *) msg->value,
5226                                          DEVLINK_ATTR_PAD);
5227         case NLA_NUL_STRING:
5228                 return nla_put_string(skb, attrtype, (char *) &msg->value);
5229         case NLA_BINARY:
5230                 return nla_put(skb, attrtype, msg->len, (void *) &msg->value);
5231         default:
5232                 return -EINVAL;
5233         }
5234 }
5235
5236 static int
5237 devlink_fmsg_prepare_skb(struct devlink_fmsg *fmsg, struct sk_buff *skb,
5238                          int *start)
5239 {
5240         struct devlink_fmsg_item *item;
5241         struct nlattr *fmsg_nlattr;
5242         int i = 0;
5243         int err;
5244
5245         fmsg_nlattr = nla_nest_start_noflag(skb, DEVLINK_ATTR_FMSG);
5246         if (!fmsg_nlattr)
5247                 return -EMSGSIZE;
5248
5249         list_for_each_entry(item, &fmsg->item_list, list) {
5250                 if (i < *start) {
5251                         i++;
5252                         continue;
5253                 }
5254
5255                 switch (item->attrtype) {
5256                 case DEVLINK_ATTR_FMSG_OBJ_NEST_START:
5257                 case DEVLINK_ATTR_FMSG_PAIR_NEST_START:
5258                 case DEVLINK_ATTR_FMSG_ARR_NEST_START:
5259                 case DEVLINK_ATTR_FMSG_NEST_END:
5260                         err = nla_put_flag(skb, item->attrtype);
5261                         break;
5262                 case DEVLINK_ATTR_FMSG_OBJ_VALUE_DATA:
5263                         err = devlink_fmsg_item_fill_type(item, skb);
5264                         if (err)
5265                                 break;
5266                         err = devlink_fmsg_item_fill_data(item, skb);
5267                         break;
5268                 case DEVLINK_ATTR_FMSG_OBJ_NAME:
5269                         err = nla_put_string(skb, item->attrtype,
5270                                              (char *) &item->value);
5271                         break;
5272                 default:
5273                         err = -EINVAL;
5274                         break;
5275                 }
5276                 if (!err)
5277                         *start = ++i;
5278                 else
5279                         break;
5280         }
5281
5282         nla_nest_end(skb, fmsg_nlattr);
5283         return err;
5284 }
5285
5286 static int devlink_fmsg_snd(struct devlink_fmsg *fmsg,
5287                             struct genl_info *info,
5288                             enum devlink_command cmd, int flags)
5289 {
5290         struct nlmsghdr *nlh;
5291         struct sk_buff *skb;
5292         bool last = false;
5293         int index = 0;
5294         void *hdr;
5295         int err;
5296
5297         while (!last) {
5298                 int tmp_index = index;
5299
5300                 skb = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
5301                 if (!skb)
5302                         return -ENOMEM;
5303
5304                 hdr = genlmsg_put(skb, info->snd_portid, info->snd_seq,
5305                                   &devlink_nl_family, flags | NLM_F_MULTI, cmd);
5306                 if (!hdr) {
5307                         err = -EMSGSIZE;
5308                         goto nla_put_failure;
5309                 }
5310
5311                 err = devlink_fmsg_prepare_skb(fmsg, skb, &index);
5312                 if (!err)
5313                         last = true;
5314                 else if (err != -EMSGSIZE || tmp_index == index)
5315                         goto nla_put_failure;
5316
5317                 genlmsg_end(skb, hdr);
5318                 err = genlmsg_reply(skb, info);
5319                 if (err)
5320                         return err;
5321         }
5322
5323         skb = genlmsg_new(GENLMSG_DEFAULT_SIZE, GFP_KERNEL);
5324         if (!skb)
5325                 return -ENOMEM;
5326         nlh = nlmsg_put(skb, info->snd_portid, info->snd_seq,
5327                         NLMSG_DONE, 0, flags | NLM_F_MULTI);
5328         if (!nlh) {
5329                 err = -EMSGSIZE;
5330                 goto nla_put_failure;
5331         }
5332
5333         return genlmsg_reply(skb, info);
5334
5335 nla_put_failure:
5336         nlmsg_free(skb);
5337         return err;
5338 }
5339
5340 static int devlink_fmsg_dumpit(struct devlink_fmsg *fmsg, struct sk_buff *skb,
5341                                struct netlink_callback *cb,
5342                                enum devlink_command cmd)
5343 {
5344         int index = cb->args[0];
5345         int tmp_index = index;
5346         void *hdr;
5347         int err;
5348
5349         hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
5350                           &devlink_nl_family, NLM_F_ACK | NLM_F_MULTI, cmd);
5351         if (!hdr) {
5352                 err = -EMSGSIZE;
5353                 goto nla_put_failure;
5354         }
5355
5356         err = devlink_fmsg_prepare_skb(fmsg, skb, &index);
5357         if ((err && err != -EMSGSIZE) || tmp_index == index)
5358                 goto nla_put_failure;
5359
5360         cb->args[0] = index;
5361         genlmsg_end(skb, hdr);
5362         return skb->len;
5363
5364 nla_put_failure:
5365         genlmsg_cancel(skb, hdr);
5366         return err;
5367 }
5368
5369 struct devlink_health_reporter {
5370         struct list_head list;
5371         void *priv;
5372         const struct devlink_health_reporter_ops *ops;
5373         struct devlink *devlink;
5374         struct devlink_port *devlink_port;
5375         struct devlink_fmsg *dump_fmsg;
5376         struct mutex dump_lock; /* lock parallel read/write from dump buffers */
5377         u64 graceful_period;
5378         bool auto_recover;
5379         bool auto_dump;
5380         u8 health_state;
5381         u64 dump_ts;
5382         u64 dump_real_ts;
5383         u64 error_count;
5384         u64 recovery_count;
5385         u64 last_recovery_ts;
5386         refcount_t refcount;
5387 };
5388
5389 void *
5390 devlink_health_reporter_priv(struct devlink_health_reporter *reporter)
5391 {
5392         return reporter->priv;
5393 }
5394 EXPORT_SYMBOL_GPL(devlink_health_reporter_priv);
5395
5396 static struct devlink_health_reporter *
5397 __devlink_health_reporter_find_by_name(struct list_head *reporter_list,
5398                                        struct mutex *list_lock,
5399                                        const char *reporter_name)
5400 {
5401         struct devlink_health_reporter *reporter;
5402
5403         lockdep_assert_held(list_lock);
5404         list_for_each_entry(reporter, reporter_list, list)
5405                 if (!strcmp(reporter->ops->name, reporter_name))
5406                         return reporter;
5407         return NULL;
5408 }
5409
5410 static struct devlink_health_reporter *
5411 devlink_health_reporter_find_by_name(struct devlink *devlink,
5412                                      const char *reporter_name)
5413 {
5414         return __devlink_health_reporter_find_by_name(&devlink->reporter_list,
5415                                                       &devlink->reporters_lock,
5416                                                       reporter_name);
5417 }
5418
5419 static struct devlink_health_reporter *
5420 devlink_port_health_reporter_find_by_name(struct devlink_port *devlink_port,
5421                                           const char *reporter_name)
5422 {
5423         return __devlink_health_reporter_find_by_name(&devlink_port->reporter_list,
5424                                                       &devlink_port->reporters_lock,
5425                                                       reporter_name);
5426 }
5427
5428 static struct devlink_health_reporter *
5429 __devlink_health_reporter_create(struct devlink *devlink,
5430                                  const struct devlink_health_reporter_ops *ops,
5431                                  u64 graceful_period, void *priv)
5432 {
5433         struct devlink_health_reporter *reporter;
5434
5435         if (WARN_ON(graceful_period && !ops->recover))
5436                 return ERR_PTR(-EINVAL);
5437
5438         reporter = kzalloc(sizeof(*reporter), GFP_KERNEL);
5439         if (!reporter)
5440                 return ERR_PTR(-ENOMEM);
5441
5442         reporter->priv = priv;
5443         reporter->ops = ops;
5444         reporter->devlink = devlink;
5445         reporter->graceful_period = graceful_period;
5446         reporter->auto_recover = !!ops->recover;
5447         reporter->auto_dump = !!ops->dump;
5448         mutex_init(&reporter->dump_lock);
5449         refcount_set(&reporter->refcount, 1);
5450         return reporter;
5451 }
5452
5453 /**
5454  *      devlink_port_health_reporter_create - create devlink health reporter for
5455  *                                            specified port instance
5456  *
5457  *      @port: devlink_port which should contain the new reporter
5458  *      @ops: ops
5459  *      @graceful_period: to avoid recovery loops, in msecs
5460  *      @priv: priv
5461  */
5462 struct devlink_health_reporter *
5463 devlink_port_health_reporter_create(struct devlink_port *port,
5464                                     const struct devlink_health_reporter_ops *ops,
5465                                     u64 graceful_period, void *priv)
5466 {
5467         struct devlink_health_reporter *reporter;
5468
5469         mutex_lock(&port->reporters_lock);
5470         if (__devlink_health_reporter_find_by_name(&port->reporter_list,
5471                                                    &port->reporters_lock, ops->name)) {
5472                 reporter = ERR_PTR(-EEXIST);
5473                 goto unlock;
5474         }
5475
5476         reporter = __devlink_health_reporter_create(port->devlink, ops,
5477                                                     graceful_period, priv);
5478         if (IS_ERR(reporter))
5479                 goto unlock;
5480
5481         reporter->devlink_port = port;
5482         list_add_tail(&reporter->list, &port->reporter_list);
5483 unlock:
5484         mutex_unlock(&port->reporters_lock);
5485         return reporter;
5486 }
5487 EXPORT_SYMBOL_GPL(devlink_port_health_reporter_create);
5488
5489 /**
5490  *      devlink_health_reporter_create - create devlink health reporter
5491  *
5492  *      @devlink: devlink
5493  *      @ops: ops
5494  *      @graceful_period: to avoid recovery loops, in msecs
5495  *      @priv: priv
5496  */
5497 struct devlink_health_reporter *
5498 devlink_health_reporter_create(struct devlink *devlink,
5499                                const struct devlink_health_reporter_ops *ops,
5500                                u64 graceful_period, void *priv)
5501 {
5502         struct devlink_health_reporter *reporter;
5503
5504         mutex_lock(&devlink->reporters_lock);
5505         if (devlink_health_reporter_find_by_name(devlink, ops->name)) {
5506                 reporter = ERR_PTR(-EEXIST);
5507                 goto unlock;
5508         }
5509
5510         reporter = __devlink_health_reporter_create(devlink, ops,
5511                                                     graceful_period, priv);
5512         if (IS_ERR(reporter))
5513                 goto unlock;
5514
5515         list_add_tail(&reporter->list, &devlink->reporter_list);
5516 unlock:
5517         mutex_unlock(&devlink->reporters_lock);
5518         return reporter;
5519 }
5520 EXPORT_SYMBOL_GPL(devlink_health_reporter_create);
5521
5522 static void
5523 devlink_health_reporter_free(struct devlink_health_reporter *reporter)
5524 {
5525         mutex_destroy(&reporter->dump_lock);
5526         if (reporter->dump_fmsg)
5527                 devlink_fmsg_free(reporter->dump_fmsg);
5528         kfree(reporter);
5529 }
5530
5531 static void
5532 devlink_health_reporter_put(struct devlink_health_reporter *reporter)
5533 {
5534         if (refcount_dec_and_test(&reporter->refcount))
5535                 devlink_health_reporter_free(reporter);
5536 }
5537
5538 static void
5539 __devlink_health_reporter_destroy(struct devlink_health_reporter *reporter)
5540 {
5541         list_del(&reporter->list);
5542         devlink_health_reporter_put(reporter);
5543 }
5544
5545 /**
5546  *      devlink_health_reporter_destroy - destroy devlink health reporter
5547  *
5548  *      @reporter: devlink health reporter to destroy
5549  */
5550 void
5551 devlink_health_reporter_destroy(struct devlink_health_reporter *reporter)
5552 {
5553         struct mutex *lock = &reporter->devlink->reporters_lock;
5554
5555         mutex_lock(lock);
5556         __devlink_health_reporter_destroy(reporter);
5557         mutex_unlock(lock);
5558 }
5559 EXPORT_SYMBOL_GPL(devlink_health_reporter_destroy);
5560
5561 /**
5562  *      devlink_port_health_reporter_destroy - destroy devlink port health reporter
5563  *
5564  *      @reporter: devlink health reporter to destroy
5565  */
5566 void
5567 devlink_port_health_reporter_destroy(struct devlink_health_reporter *reporter)
5568 {
5569         struct mutex *lock = &reporter->devlink_port->reporters_lock;
5570
5571         mutex_lock(lock);
5572         __devlink_health_reporter_destroy(reporter);
5573         mutex_unlock(lock);
5574 }
5575 EXPORT_SYMBOL_GPL(devlink_port_health_reporter_destroy);
5576
5577 static int
5578 devlink_nl_health_reporter_fill(struct sk_buff *msg,
5579                                 struct devlink *devlink,
5580                                 struct devlink_health_reporter *reporter,
5581                                 enum devlink_command cmd, u32 portid,
5582                                 u32 seq, int flags)
5583 {
5584         struct nlattr *reporter_attr;
5585         void *hdr;
5586
5587         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
5588         if (!hdr)
5589                 return -EMSGSIZE;
5590
5591         if (devlink_nl_put_handle(msg, devlink))
5592                 goto genlmsg_cancel;
5593
5594         if (reporter->devlink_port) {
5595                 if (nla_put_u32(msg, DEVLINK_ATTR_PORT_INDEX, reporter->devlink_port->index))
5596                         goto genlmsg_cancel;
5597         }
5598         reporter_attr = nla_nest_start_noflag(msg,
5599                                               DEVLINK_ATTR_HEALTH_REPORTER);
5600         if (!reporter_attr)
5601                 goto genlmsg_cancel;
5602         if (nla_put_string(msg, DEVLINK_ATTR_HEALTH_REPORTER_NAME,
5603                            reporter->ops->name))
5604                 goto reporter_nest_cancel;
5605         if (nla_put_u8(msg, DEVLINK_ATTR_HEALTH_REPORTER_STATE,
5606                        reporter->health_state))
5607                 goto reporter_nest_cancel;
5608         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_HEALTH_REPORTER_ERR_COUNT,
5609                               reporter->error_count, DEVLINK_ATTR_PAD))
5610                 goto reporter_nest_cancel;
5611         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_HEALTH_REPORTER_RECOVER_COUNT,
5612                               reporter->recovery_count, DEVLINK_ATTR_PAD))
5613                 goto reporter_nest_cancel;
5614         if (reporter->ops->recover &&
5615             nla_put_u64_64bit(msg, DEVLINK_ATTR_HEALTH_REPORTER_GRACEFUL_PERIOD,
5616                               reporter->graceful_period,
5617                               DEVLINK_ATTR_PAD))
5618                 goto reporter_nest_cancel;
5619         if (reporter->ops->recover &&
5620             nla_put_u8(msg, DEVLINK_ATTR_HEALTH_REPORTER_AUTO_RECOVER,
5621                        reporter->auto_recover))
5622                 goto reporter_nest_cancel;
5623         if (reporter->dump_fmsg &&
5624             nla_put_u64_64bit(msg, DEVLINK_ATTR_HEALTH_REPORTER_DUMP_TS,
5625                               jiffies_to_msecs(reporter->dump_ts),
5626                               DEVLINK_ATTR_PAD))
5627                 goto reporter_nest_cancel;
5628         if (reporter->dump_fmsg &&
5629             nla_put_u64_64bit(msg, DEVLINK_ATTR_HEALTH_REPORTER_DUMP_TS_NS,
5630                               reporter->dump_real_ts, DEVLINK_ATTR_PAD))
5631                 goto reporter_nest_cancel;
5632         if (reporter->ops->dump &&
5633             nla_put_u8(msg, DEVLINK_ATTR_HEALTH_REPORTER_AUTO_DUMP,
5634                        reporter->auto_dump))
5635                 goto reporter_nest_cancel;
5636
5637         nla_nest_end(msg, reporter_attr);
5638         genlmsg_end(msg, hdr);
5639         return 0;
5640
5641 reporter_nest_cancel:
5642         nla_nest_end(msg, reporter_attr);
5643 genlmsg_cancel:
5644         genlmsg_cancel(msg, hdr);
5645         return -EMSGSIZE;
5646 }
5647
5648 static void devlink_recover_notify(struct devlink_health_reporter *reporter,
5649                                    enum devlink_command cmd)
5650 {
5651         struct sk_buff *msg;
5652         int err;
5653
5654         WARN_ON(cmd != DEVLINK_CMD_HEALTH_REPORTER_RECOVER);
5655
5656         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
5657         if (!msg)
5658                 return;
5659
5660         err = devlink_nl_health_reporter_fill(msg, reporter->devlink,
5661                                               reporter, cmd, 0, 0, 0);
5662         if (err) {
5663                 nlmsg_free(msg);
5664                 return;
5665         }
5666
5667         genlmsg_multicast_netns(&devlink_nl_family,
5668                                 devlink_net(reporter->devlink),
5669                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
5670 }
5671
5672 void
5673 devlink_health_reporter_recovery_done(struct devlink_health_reporter *reporter)
5674 {
5675         reporter->recovery_count++;
5676         reporter->last_recovery_ts = jiffies;
5677 }
5678 EXPORT_SYMBOL_GPL(devlink_health_reporter_recovery_done);
5679
5680 static int
5681 devlink_health_reporter_recover(struct devlink_health_reporter *reporter,
5682                                 void *priv_ctx, struct netlink_ext_ack *extack)
5683 {
5684         int err;
5685
5686         if (reporter->health_state == DEVLINK_HEALTH_REPORTER_STATE_HEALTHY)
5687                 return 0;
5688
5689         if (!reporter->ops->recover)
5690                 return -EOPNOTSUPP;
5691
5692         err = reporter->ops->recover(reporter, priv_ctx, extack);
5693         if (err)
5694                 return err;
5695
5696         devlink_health_reporter_recovery_done(reporter);
5697         reporter->health_state = DEVLINK_HEALTH_REPORTER_STATE_HEALTHY;
5698         devlink_recover_notify(reporter, DEVLINK_CMD_HEALTH_REPORTER_RECOVER);
5699
5700         return 0;
5701 }
5702
5703 static void
5704 devlink_health_dump_clear(struct devlink_health_reporter *reporter)
5705 {
5706         if (!reporter->dump_fmsg)
5707                 return;
5708         devlink_fmsg_free(reporter->dump_fmsg);
5709         reporter->dump_fmsg = NULL;
5710 }
5711
5712 static int devlink_health_do_dump(struct devlink_health_reporter *reporter,
5713                                   void *priv_ctx,
5714                                   struct netlink_ext_ack *extack)
5715 {
5716         int err;
5717
5718         if (!reporter->ops->dump)
5719                 return 0;
5720
5721         if (reporter->dump_fmsg)
5722                 return 0;
5723
5724         reporter->dump_fmsg = devlink_fmsg_alloc();
5725         if (!reporter->dump_fmsg) {
5726                 err = -ENOMEM;
5727                 return err;
5728         }
5729
5730         err = devlink_fmsg_obj_nest_start(reporter->dump_fmsg);
5731         if (err)
5732                 goto dump_err;
5733
5734         err = reporter->ops->dump(reporter, reporter->dump_fmsg,
5735                                   priv_ctx, extack);
5736         if (err)
5737                 goto dump_err;
5738
5739         err = devlink_fmsg_obj_nest_end(reporter->dump_fmsg);
5740         if (err)
5741                 goto dump_err;
5742
5743         reporter->dump_ts = jiffies;
5744         reporter->dump_real_ts = ktime_get_real_ns();
5745
5746         return 0;
5747
5748 dump_err:
5749         devlink_health_dump_clear(reporter);
5750         return err;
5751 }
5752
5753 int devlink_health_report(struct devlink_health_reporter *reporter,
5754                           const char *msg, void *priv_ctx)
5755 {
5756         enum devlink_health_reporter_state prev_health_state;
5757         struct devlink *devlink = reporter->devlink;
5758         unsigned long recover_ts_threshold;
5759
5760         /* write a log message of the current error */
5761         WARN_ON(!msg);
5762         trace_devlink_health_report(devlink, reporter->ops->name, msg);
5763         reporter->error_count++;
5764         prev_health_state = reporter->health_state;
5765         reporter->health_state = DEVLINK_HEALTH_REPORTER_STATE_ERROR;
5766         devlink_recover_notify(reporter, DEVLINK_CMD_HEALTH_REPORTER_RECOVER);
5767
5768         /* abort if the previous error wasn't recovered */
5769         recover_ts_threshold = reporter->last_recovery_ts +
5770                                msecs_to_jiffies(reporter->graceful_period);
5771         if (reporter->auto_recover &&
5772             (prev_health_state != DEVLINK_HEALTH_REPORTER_STATE_HEALTHY ||
5773              (reporter->last_recovery_ts && reporter->recovery_count &&
5774               time_is_after_jiffies(recover_ts_threshold)))) {
5775                 trace_devlink_health_recover_aborted(devlink,
5776                                                      reporter->ops->name,
5777                                                      reporter->health_state,
5778                                                      jiffies -
5779                                                      reporter->last_recovery_ts);
5780                 return -ECANCELED;
5781         }
5782
5783         reporter->health_state = DEVLINK_HEALTH_REPORTER_STATE_ERROR;
5784
5785         if (reporter->auto_dump) {
5786                 mutex_lock(&reporter->dump_lock);
5787                 /* store current dump of current error, for later analysis */
5788                 devlink_health_do_dump(reporter, priv_ctx, NULL);
5789                 mutex_unlock(&reporter->dump_lock);
5790         }
5791
5792         if (reporter->auto_recover)
5793                 return devlink_health_reporter_recover(reporter,
5794                                                        priv_ctx, NULL);
5795
5796         return 0;
5797 }
5798 EXPORT_SYMBOL_GPL(devlink_health_report);
5799
5800 static struct devlink_health_reporter *
5801 devlink_health_reporter_get_from_attrs(struct devlink *devlink,
5802                                        struct nlattr **attrs)
5803 {
5804         struct devlink_health_reporter *reporter;
5805         struct devlink_port *devlink_port;
5806         char *reporter_name;
5807
5808         if (!attrs[DEVLINK_ATTR_HEALTH_REPORTER_NAME])
5809                 return NULL;
5810
5811         reporter_name = nla_data(attrs[DEVLINK_ATTR_HEALTH_REPORTER_NAME]);
5812         devlink_port = devlink_port_get_from_attrs(devlink, attrs);
5813         if (IS_ERR(devlink_port)) {
5814                 mutex_lock(&devlink->reporters_lock);
5815                 reporter = devlink_health_reporter_find_by_name(devlink, reporter_name);
5816                 if (reporter)
5817                         refcount_inc(&reporter->refcount);
5818                 mutex_unlock(&devlink->reporters_lock);
5819         } else {
5820                 mutex_lock(&devlink_port->reporters_lock);
5821                 reporter = devlink_port_health_reporter_find_by_name(devlink_port, reporter_name);
5822                 if (reporter)
5823                         refcount_inc(&reporter->refcount);
5824                 mutex_unlock(&devlink_port->reporters_lock);
5825         }
5826
5827         return reporter;
5828 }
5829
5830 static struct devlink_health_reporter *
5831 devlink_health_reporter_get_from_info(struct devlink *devlink,
5832                                       struct genl_info *info)
5833 {
5834         return devlink_health_reporter_get_from_attrs(devlink, info->attrs);
5835 }
5836
5837 static struct devlink_health_reporter *
5838 devlink_health_reporter_get_from_cb(struct netlink_callback *cb)
5839 {
5840         const struct genl_dumpit_info *info = genl_dumpit_info(cb);
5841         struct devlink_health_reporter *reporter;
5842         struct nlattr **attrs = info->attrs;
5843         struct devlink *devlink;
5844
5845         mutex_lock(&devlink_mutex);
5846         devlink = devlink_get_from_attrs(sock_net(cb->skb->sk), attrs);
5847         if (IS_ERR(devlink))
5848                 goto unlock;
5849
5850         reporter = devlink_health_reporter_get_from_attrs(devlink, attrs);
5851         mutex_unlock(&devlink_mutex);
5852         return reporter;
5853 unlock:
5854         mutex_unlock(&devlink_mutex);
5855         return NULL;
5856 }
5857
5858 void
5859 devlink_health_reporter_state_update(struct devlink_health_reporter *reporter,
5860                                      enum devlink_health_reporter_state state)
5861 {
5862         if (WARN_ON(state != DEVLINK_HEALTH_REPORTER_STATE_HEALTHY &&
5863                     state != DEVLINK_HEALTH_REPORTER_STATE_ERROR))
5864                 return;
5865
5866         if (reporter->health_state == state)
5867                 return;
5868
5869         reporter->health_state = state;
5870         trace_devlink_health_reporter_state_update(reporter->devlink,
5871                                                    reporter->ops->name, state);
5872         devlink_recover_notify(reporter, DEVLINK_CMD_HEALTH_REPORTER_RECOVER);
5873 }
5874 EXPORT_SYMBOL_GPL(devlink_health_reporter_state_update);
5875
5876 static int devlink_nl_cmd_health_reporter_get_doit(struct sk_buff *skb,
5877                                                    struct genl_info *info)
5878 {
5879         struct devlink *devlink = info->user_ptr[0];
5880         struct devlink_health_reporter *reporter;
5881         struct sk_buff *msg;
5882         int err;
5883
5884         reporter = devlink_health_reporter_get_from_info(devlink, info);
5885         if (!reporter)
5886                 return -EINVAL;
5887
5888         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
5889         if (!msg) {
5890                 err = -ENOMEM;
5891                 goto out;
5892         }
5893
5894         err = devlink_nl_health_reporter_fill(msg, devlink, reporter,
5895                                               DEVLINK_CMD_HEALTH_REPORTER_GET,
5896                                               info->snd_portid, info->snd_seq,
5897                                               0);
5898         if (err) {
5899                 nlmsg_free(msg);
5900                 goto out;
5901         }
5902
5903         err = genlmsg_reply(msg, info);
5904 out:
5905         devlink_health_reporter_put(reporter);
5906         return err;
5907 }
5908
5909 static int
5910 devlink_nl_cmd_health_reporter_get_dumpit(struct sk_buff *msg,
5911                                           struct netlink_callback *cb)
5912 {
5913         struct devlink_health_reporter *reporter;
5914         struct devlink_port *port;
5915         struct devlink *devlink;
5916         int start = cb->args[0];
5917         int idx = 0;
5918         int err;
5919
5920         mutex_lock(&devlink_mutex);
5921         list_for_each_entry(devlink, &devlink_list, list) {
5922                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
5923                         continue;
5924                 mutex_lock(&devlink->reporters_lock);
5925                 list_for_each_entry(reporter, &devlink->reporter_list,
5926                                     list) {
5927                         if (idx < start) {
5928                                 idx++;
5929                                 continue;
5930                         }
5931                         err = devlink_nl_health_reporter_fill(msg, devlink,
5932                                                               reporter,
5933                                                               DEVLINK_CMD_HEALTH_REPORTER_GET,
5934                                                               NETLINK_CB(cb->skb).portid,
5935                                                               cb->nlh->nlmsg_seq,
5936                                                               NLM_F_MULTI);
5937                         if (err) {
5938                                 mutex_unlock(&devlink->reporters_lock);
5939                                 goto out;
5940                         }
5941                         idx++;
5942                 }
5943                 mutex_unlock(&devlink->reporters_lock);
5944         }
5945
5946         list_for_each_entry(devlink, &devlink_list, list) {
5947                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
5948                         continue;
5949                 mutex_lock(&devlink->lock);
5950                 list_for_each_entry(port, &devlink->port_list, list) {
5951                         mutex_lock(&port->reporters_lock);
5952                         list_for_each_entry(reporter, &port->reporter_list, list) {
5953                                 if (idx < start) {
5954                                         idx++;
5955                                         continue;
5956                                 }
5957                                 err = devlink_nl_health_reporter_fill(msg, devlink, reporter,
5958                                                                       DEVLINK_CMD_HEALTH_REPORTER_GET,
5959                                                                       NETLINK_CB(cb->skb).portid,
5960                                                                       cb->nlh->nlmsg_seq,
5961                                                                       NLM_F_MULTI);
5962                                 if (err) {
5963                                         mutex_unlock(&port->reporters_lock);
5964                                         mutex_unlock(&devlink->lock);
5965                                         goto out;
5966                                 }
5967                                 idx++;
5968                         }
5969                         mutex_unlock(&port->reporters_lock);
5970                 }
5971                 mutex_unlock(&devlink->lock);
5972         }
5973 out:
5974         mutex_unlock(&devlink_mutex);
5975
5976         cb->args[0] = idx;
5977         return msg->len;
5978 }
5979
5980 static int
5981 devlink_nl_cmd_health_reporter_set_doit(struct sk_buff *skb,
5982                                         struct genl_info *info)
5983 {
5984         struct devlink *devlink = info->user_ptr[0];
5985         struct devlink_health_reporter *reporter;
5986         int err;
5987
5988         reporter = devlink_health_reporter_get_from_info(devlink, info);
5989         if (!reporter)
5990                 return -EINVAL;
5991
5992         if (!reporter->ops->recover &&
5993             (info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_GRACEFUL_PERIOD] ||
5994              info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_RECOVER])) {
5995                 err = -EOPNOTSUPP;
5996                 goto out;
5997         }
5998         if (!reporter->ops->dump &&
5999             info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_DUMP]) {
6000                 err = -EOPNOTSUPP;
6001                 goto out;
6002         }
6003
6004         if (info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_GRACEFUL_PERIOD])
6005                 reporter->graceful_period =
6006                         nla_get_u64(info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_GRACEFUL_PERIOD]);
6007
6008         if (info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_RECOVER])
6009                 reporter->auto_recover =
6010                         nla_get_u8(info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_RECOVER]);
6011
6012         if (info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_DUMP])
6013                 reporter->auto_dump =
6014                 nla_get_u8(info->attrs[DEVLINK_ATTR_HEALTH_REPORTER_AUTO_DUMP]);
6015
6016         devlink_health_reporter_put(reporter);
6017         return 0;
6018 out:
6019         devlink_health_reporter_put(reporter);
6020         return err;
6021 }
6022
6023 static int devlink_nl_cmd_health_reporter_recover_doit(struct sk_buff *skb,
6024                                                        struct genl_info *info)
6025 {
6026         struct devlink *devlink = info->user_ptr[0];
6027         struct devlink_health_reporter *reporter;
6028         int err;
6029
6030         reporter = devlink_health_reporter_get_from_info(devlink, info);
6031         if (!reporter)
6032                 return -EINVAL;
6033
6034         err = devlink_health_reporter_recover(reporter, NULL, info->extack);
6035
6036         devlink_health_reporter_put(reporter);
6037         return err;
6038 }
6039
6040 static int devlink_nl_cmd_health_reporter_diagnose_doit(struct sk_buff *skb,
6041                                                         struct genl_info *info)
6042 {
6043         struct devlink *devlink = info->user_ptr[0];
6044         struct devlink_health_reporter *reporter;
6045         struct devlink_fmsg *fmsg;
6046         int err;
6047
6048         reporter = devlink_health_reporter_get_from_info(devlink, info);
6049         if (!reporter)
6050                 return -EINVAL;
6051
6052         if (!reporter->ops->diagnose) {
6053                 devlink_health_reporter_put(reporter);
6054                 return -EOPNOTSUPP;
6055         }
6056
6057         fmsg = devlink_fmsg_alloc();
6058         if (!fmsg) {
6059                 devlink_health_reporter_put(reporter);
6060                 return -ENOMEM;
6061         }
6062
6063         err = devlink_fmsg_obj_nest_start(fmsg);
6064         if (err)
6065                 goto out;
6066
6067         err = reporter->ops->diagnose(reporter, fmsg, info->extack);
6068         if (err)
6069                 goto out;
6070
6071         err = devlink_fmsg_obj_nest_end(fmsg);
6072         if (err)
6073                 goto out;
6074
6075         err = devlink_fmsg_snd(fmsg, info,
6076                                DEVLINK_CMD_HEALTH_REPORTER_DIAGNOSE, 0);
6077
6078 out:
6079         devlink_fmsg_free(fmsg);
6080         devlink_health_reporter_put(reporter);
6081         return err;
6082 }
6083
6084 static int
6085 devlink_nl_cmd_health_reporter_dump_get_dumpit(struct sk_buff *skb,
6086                                                struct netlink_callback *cb)
6087 {
6088         struct devlink_health_reporter *reporter;
6089         u64 start = cb->args[0];
6090         int err;
6091
6092         reporter = devlink_health_reporter_get_from_cb(cb);
6093         if (!reporter)
6094                 return -EINVAL;
6095
6096         if (!reporter->ops->dump) {
6097                 err = -EOPNOTSUPP;
6098                 goto out;
6099         }
6100         mutex_lock(&reporter->dump_lock);
6101         if (!start) {
6102                 err = devlink_health_do_dump(reporter, NULL, cb->extack);
6103                 if (err)
6104                         goto unlock;
6105                 cb->args[1] = reporter->dump_ts;
6106         }
6107         if (!reporter->dump_fmsg || cb->args[1] != reporter->dump_ts) {
6108                 NL_SET_ERR_MSG_MOD(cb->extack, "Dump trampled, please retry");
6109                 err = -EAGAIN;
6110                 goto unlock;
6111         }
6112
6113         err = devlink_fmsg_dumpit(reporter->dump_fmsg, skb, cb,
6114                                   DEVLINK_CMD_HEALTH_REPORTER_DUMP_GET);
6115 unlock:
6116         mutex_unlock(&reporter->dump_lock);
6117 out:
6118         devlink_health_reporter_put(reporter);
6119         return err;
6120 }
6121
6122 static int
6123 devlink_nl_cmd_health_reporter_dump_clear_doit(struct sk_buff *skb,
6124                                                struct genl_info *info)
6125 {
6126         struct devlink *devlink = info->user_ptr[0];
6127         struct devlink_health_reporter *reporter;
6128
6129         reporter = devlink_health_reporter_get_from_info(devlink, info);
6130         if (!reporter)
6131                 return -EINVAL;
6132
6133         if (!reporter->ops->dump) {
6134                 devlink_health_reporter_put(reporter);
6135                 return -EOPNOTSUPP;
6136         }
6137
6138         mutex_lock(&reporter->dump_lock);
6139         devlink_health_dump_clear(reporter);
6140         mutex_unlock(&reporter->dump_lock);
6141         devlink_health_reporter_put(reporter);
6142         return 0;
6143 }
6144
6145 static int devlink_nl_cmd_health_reporter_test_doit(struct sk_buff *skb,
6146                                                     struct genl_info *info)
6147 {
6148         struct devlink *devlink = info->user_ptr[0];
6149         struct devlink_health_reporter *reporter;
6150         int err;
6151
6152         reporter = devlink_health_reporter_get_from_info(devlink, info);
6153         if (!reporter)
6154                 return -EINVAL;
6155
6156         if (!reporter->ops->test) {
6157                 devlink_health_reporter_put(reporter);
6158                 return -EOPNOTSUPP;
6159         }
6160
6161         err = reporter->ops->test(reporter, info->extack);
6162
6163         devlink_health_reporter_put(reporter);
6164         return err;
6165 }
6166
6167 struct devlink_stats {
6168         u64 rx_bytes;
6169         u64 rx_packets;
6170         struct u64_stats_sync syncp;
6171 };
6172
6173 /**
6174  * struct devlink_trap_policer_item - Packet trap policer attributes.
6175  * @policer: Immutable packet trap policer attributes.
6176  * @rate: Rate in packets / sec.
6177  * @burst: Burst size in packets.
6178  * @list: trap_policer_list member.
6179  *
6180  * Describes packet trap policer attributes. Created by devlink during trap
6181  * policer registration.
6182  */
6183 struct devlink_trap_policer_item {
6184         const struct devlink_trap_policer *policer;
6185         u64 rate;
6186         u64 burst;
6187         struct list_head list;
6188 };
6189
6190 /**
6191  * struct devlink_trap_group_item - Packet trap group attributes.
6192  * @group: Immutable packet trap group attributes.
6193  * @policer_item: Associated policer item. Can be NULL.
6194  * @list: trap_group_list member.
6195  * @stats: Trap group statistics.
6196  *
6197  * Describes packet trap group attributes. Created by devlink during trap
6198  * group registration.
6199  */
6200 struct devlink_trap_group_item {
6201         const struct devlink_trap_group *group;
6202         struct devlink_trap_policer_item *policer_item;
6203         struct list_head list;
6204         struct devlink_stats __percpu *stats;
6205 };
6206
6207 /**
6208  * struct devlink_trap_item - Packet trap attributes.
6209  * @trap: Immutable packet trap attributes.
6210  * @group_item: Associated group item.
6211  * @list: trap_list member.
6212  * @action: Trap action.
6213  * @stats: Trap statistics.
6214  * @priv: Driver private information.
6215  *
6216  * Describes both mutable and immutable packet trap attributes. Created by
6217  * devlink during trap registration and used for all trap related operations.
6218  */
6219 struct devlink_trap_item {
6220         const struct devlink_trap *trap;
6221         struct devlink_trap_group_item *group_item;
6222         struct list_head list;
6223         enum devlink_trap_action action;
6224         struct devlink_stats __percpu *stats;
6225         void *priv;
6226 };
6227
6228 static struct devlink_trap_policer_item *
6229 devlink_trap_policer_item_lookup(struct devlink *devlink, u32 id)
6230 {
6231         struct devlink_trap_policer_item *policer_item;
6232
6233         list_for_each_entry(policer_item, &devlink->trap_policer_list, list) {
6234                 if (policer_item->policer->id == id)
6235                         return policer_item;
6236         }
6237
6238         return NULL;
6239 }
6240
6241 static struct devlink_trap_item *
6242 devlink_trap_item_lookup(struct devlink *devlink, const char *name)
6243 {
6244         struct devlink_trap_item *trap_item;
6245
6246         list_for_each_entry(trap_item, &devlink->trap_list, list) {
6247                 if (!strcmp(trap_item->trap->name, name))
6248                         return trap_item;
6249         }
6250
6251         return NULL;
6252 }
6253
6254 static struct devlink_trap_item *
6255 devlink_trap_item_get_from_info(struct devlink *devlink,
6256                                 struct genl_info *info)
6257 {
6258         struct nlattr *attr;
6259
6260         if (!info->attrs[DEVLINK_ATTR_TRAP_NAME])
6261                 return NULL;
6262         attr = info->attrs[DEVLINK_ATTR_TRAP_NAME];
6263
6264         return devlink_trap_item_lookup(devlink, nla_data(attr));
6265 }
6266
6267 static int
6268 devlink_trap_action_get_from_info(struct genl_info *info,
6269                                   enum devlink_trap_action *p_trap_action)
6270 {
6271         u8 val;
6272
6273         val = nla_get_u8(info->attrs[DEVLINK_ATTR_TRAP_ACTION]);
6274         switch (val) {
6275         case DEVLINK_TRAP_ACTION_DROP:
6276         case DEVLINK_TRAP_ACTION_TRAP:
6277         case DEVLINK_TRAP_ACTION_MIRROR:
6278                 *p_trap_action = val;
6279                 break;
6280         default:
6281                 return -EINVAL;
6282         }
6283
6284         return 0;
6285 }
6286
6287 static int devlink_trap_metadata_put(struct sk_buff *msg,
6288                                      const struct devlink_trap *trap)
6289 {
6290         struct nlattr *attr;
6291
6292         attr = nla_nest_start(msg, DEVLINK_ATTR_TRAP_METADATA);
6293         if (!attr)
6294                 return -EMSGSIZE;
6295
6296         if ((trap->metadata_cap & DEVLINK_TRAP_METADATA_TYPE_F_IN_PORT) &&
6297             nla_put_flag(msg, DEVLINK_ATTR_TRAP_METADATA_TYPE_IN_PORT))
6298                 goto nla_put_failure;
6299         if ((trap->metadata_cap & DEVLINK_TRAP_METADATA_TYPE_F_FA_COOKIE) &&
6300             nla_put_flag(msg, DEVLINK_ATTR_TRAP_METADATA_TYPE_FA_COOKIE))
6301                 goto nla_put_failure;
6302
6303         nla_nest_end(msg, attr);
6304
6305         return 0;
6306
6307 nla_put_failure:
6308         nla_nest_cancel(msg, attr);
6309         return -EMSGSIZE;
6310 }
6311
6312 static void devlink_trap_stats_read(struct devlink_stats __percpu *trap_stats,
6313                                     struct devlink_stats *stats)
6314 {
6315         int i;
6316
6317         memset(stats, 0, sizeof(*stats));
6318         for_each_possible_cpu(i) {
6319                 struct devlink_stats *cpu_stats;
6320                 u64 rx_packets, rx_bytes;
6321                 unsigned int start;
6322
6323                 cpu_stats = per_cpu_ptr(trap_stats, i);
6324                 do {
6325                         start = u64_stats_fetch_begin_irq(&cpu_stats->syncp);
6326                         rx_packets = cpu_stats->rx_packets;
6327                         rx_bytes = cpu_stats->rx_bytes;
6328                 } while (u64_stats_fetch_retry_irq(&cpu_stats->syncp, start));
6329
6330                 stats->rx_packets += rx_packets;
6331                 stats->rx_bytes += rx_bytes;
6332         }
6333 }
6334
6335 static int devlink_trap_stats_put(struct sk_buff *msg,
6336                                   struct devlink_stats __percpu *trap_stats)
6337 {
6338         struct devlink_stats stats;
6339         struct nlattr *attr;
6340
6341         devlink_trap_stats_read(trap_stats, &stats);
6342
6343         attr = nla_nest_start(msg, DEVLINK_ATTR_STATS);
6344         if (!attr)
6345                 return -EMSGSIZE;
6346
6347         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_STATS_RX_PACKETS,
6348                               stats.rx_packets, DEVLINK_ATTR_PAD))
6349                 goto nla_put_failure;
6350
6351         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_STATS_RX_BYTES,
6352                               stats.rx_bytes, DEVLINK_ATTR_PAD))
6353                 goto nla_put_failure;
6354
6355         nla_nest_end(msg, attr);
6356
6357         return 0;
6358
6359 nla_put_failure:
6360         nla_nest_cancel(msg, attr);
6361         return -EMSGSIZE;
6362 }
6363
6364 static int devlink_nl_trap_fill(struct sk_buff *msg, struct devlink *devlink,
6365                                 const struct devlink_trap_item *trap_item,
6366                                 enum devlink_command cmd, u32 portid, u32 seq,
6367                                 int flags)
6368 {
6369         struct devlink_trap_group_item *group_item = trap_item->group_item;
6370         void *hdr;
6371         int err;
6372
6373         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
6374         if (!hdr)
6375                 return -EMSGSIZE;
6376
6377         if (devlink_nl_put_handle(msg, devlink))
6378                 goto nla_put_failure;
6379
6380         if (nla_put_string(msg, DEVLINK_ATTR_TRAP_GROUP_NAME,
6381                            group_item->group->name))
6382                 goto nla_put_failure;
6383
6384         if (nla_put_string(msg, DEVLINK_ATTR_TRAP_NAME, trap_item->trap->name))
6385                 goto nla_put_failure;
6386
6387         if (nla_put_u8(msg, DEVLINK_ATTR_TRAP_TYPE, trap_item->trap->type))
6388                 goto nla_put_failure;
6389
6390         if (trap_item->trap->generic &&
6391             nla_put_flag(msg, DEVLINK_ATTR_TRAP_GENERIC))
6392                 goto nla_put_failure;
6393
6394         if (nla_put_u8(msg, DEVLINK_ATTR_TRAP_ACTION, trap_item->action))
6395                 goto nla_put_failure;
6396
6397         err = devlink_trap_metadata_put(msg, trap_item->trap);
6398         if (err)
6399                 goto nla_put_failure;
6400
6401         err = devlink_trap_stats_put(msg, trap_item->stats);
6402         if (err)
6403                 goto nla_put_failure;
6404
6405         genlmsg_end(msg, hdr);
6406
6407         return 0;
6408
6409 nla_put_failure:
6410         genlmsg_cancel(msg, hdr);
6411         return -EMSGSIZE;
6412 }
6413
6414 static int devlink_nl_cmd_trap_get_doit(struct sk_buff *skb,
6415                                         struct genl_info *info)
6416 {
6417         struct netlink_ext_ack *extack = info->extack;
6418         struct devlink *devlink = info->user_ptr[0];
6419         struct devlink_trap_item *trap_item;
6420         struct sk_buff *msg;
6421         int err;
6422
6423         if (list_empty(&devlink->trap_list))
6424                 return -EOPNOTSUPP;
6425
6426         trap_item = devlink_trap_item_get_from_info(devlink, info);
6427         if (!trap_item) {
6428                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap");
6429                 return -ENOENT;
6430         }
6431
6432         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6433         if (!msg)
6434                 return -ENOMEM;
6435
6436         err = devlink_nl_trap_fill(msg, devlink, trap_item,
6437                                    DEVLINK_CMD_TRAP_NEW, info->snd_portid,
6438                                    info->snd_seq, 0);
6439         if (err)
6440                 goto err_trap_fill;
6441
6442         return genlmsg_reply(msg, info);
6443
6444 err_trap_fill:
6445         nlmsg_free(msg);
6446         return err;
6447 }
6448
6449 static int devlink_nl_cmd_trap_get_dumpit(struct sk_buff *msg,
6450                                           struct netlink_callback *cb)
6451 {
6452         struct devlink_trap_item *trap_item;
6453         struct devlink *devlink;
6454         int start = cb->args[0];
6455         int idx = 0;
6456         int err;
6457
6458         mutex_lock(&devlink_mutex);
6459         list_for_each_entry(devlink, &devlink_list, list) {
6460                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
6461                         continue;
6462                 mutex_lock(&devlink->lock);
6463                 list_for_each_entry(trap_item, &devlink->trap_list, list) {
6464                         if (idx < start) {
6465                                 idx++;
6466                                 continue;
6467                         }
6468                         err = devlink_nl_trap_fill(msg, devlink, trap_item,
6469                                                    DEVLINK_CMD_TRAP_NEW,
6470                                                    NETLINK_CB(cb->skb).portid,
6471                                                    cb->nlh->nlmsg_seq,
6472                                                    NLM_F_MULTI);
6473                         if (err) {
6474                                 mutex_unlock(&devlink->lock);
6475                                 goto out;
6476                         }
6477                         idx++;
6478                 }
6479                 mutex_unlock(&devlink->lock);
6480         }
6481 out:
6482         mutex_unlock(&devlink_mutex);
6483
6484         cb->args[0] = idx;
6485         return msg->len;
6486 }
6487
6488 static int __devlink_trap_action_set(struct devlink *devlink,
6489                                      struct devlink_trap_item *trap_item,
6490                                      enum devlink_trap_action trap_action,
6491                                      struct netlink_ext_ack *extack)
6492 {
6493         int err;
6494
6495         if (trap_item->action != trap_action &&
6496             trap_item->trap->type != DEVLINK_TRAP_TYPE_DROP) {
6497                 NL_SET_ERR_MSG_MOD(extack, "Cannot change action of non-drop traps. Skipping");
6498                 return 0;
6499         }
6500
6501         err = devlink->ops->trap_action_set(devlink, trap_item->trap,
6502                                             trap_action, extack);
6503         if (err)
6504                 return err;
6505
6506         trap_item->action = trap_action;
6507
6508         return 0;
6509 }
6510
6511 static int devlink_trap_action_set(struct devlink *devlink,
6512                                    struct devlink_trap_item *trap_item,
6513                                    struct genl_info *info)
6514 {
6515         enum devlink_trap_action trap_action;
6516         int err;
6517
6518         if (!info->attrs[DEVLINK_ATTR_TRAP_ACTION])
6519                 return 0;
6520
6521         err = devlink_trap_action_get_from_info(info, &trap_action);
6522         if (err) {
6523                 NL_SET_ERR_MSG_MOD(info->extack, "Invalid trap action");
6524                 return -EINVAL;
6525         }
6526
6527         return __devlink_trap_action_set(devlink, trap_item, trap_action,
6528                                          info->extack);
6529 }
6530
6531 static int devlink_nl_cmd_trap_set_doit(struct sk_buff *skb,
6532                                         struct genl_info *info)
6533 {
6534         struct netlink_ext_ack *extack = info->extack;
6535         struct devlink *devlink = info->user_ptr[0];
6536         struct devlink_trap_item *trap_item;
6537         int err;
6538
6539         if (list_empty(&devlink->trap_list))
6540                 return -EOPNOTSUPP;
6541
6542         trap_item = devlink_trap_item_get_from_info(devlink, info);
6543         if (!trap_item) {
6544                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap");
6545                 return -ENOENT;
6546         }
6547
6548         err = devlink_trap_action_set(devlink, trap_item, info);
6549         if (err)
6550                 return err;
6551
6552         return 0;
6553 }
6554
6555 static struct devlink_trap_group_item *
6556 devlink_trap_group_item_lookup(struct devlink *devlink, const char *name)
6557 {
6558         struct devlink_trap_group_item *group_item;
6559
6560         list_for_each_entry(group_item, &devlink->trap_group_list, list) {
6561                 if (!strcmp(group_item->group->name, name))
6562                         return group_item;
6563         }
6564
6565         return NULL;
6566 }
6567
6568 static struct devlink_trap_group_item *
6569 devlink_trap_group_item_lookup_by_id(struct devlink *devlink, u16 id)
6570 {
6571         struct devlink_trap_group_item *group_item;
6572
6573         list_for_each_entry(group_item, &devlink->trap_group_list, list) {
6574                 if (group_item->group->id == id)
6575                         return group_item;
6576         }
6577
6578         return NULL;
6579 }
6580
6581 static struct devlink_trap_group_item *
6582 devlink_trap_group_item_get_from_info(struct devlink *devlink,
6583                                       struct genl_info *info)
6584 {
6585         char *name;
6586
6587         if (!info->attrs[DEVLINK_ATTR_TRAP_GROUP_NAME])
6588                 return NULL;
6589         name = nla_data(info->attrs[DEVLINK_ATTR_TRAP_GROUP_NAME]);
6590
6591         return devlink_trap_group_item_lookup(devlink, name);
6592 }
6593
6594 static int
6595 devlink_nl_trap_group_fill(struct sk_buff *msg, struct devlink *devlink,
6596                            const struct devlink_trap_group_item *group_item,
6597                            enum devlink_command cmd, u32 portid, u32 seq,
6598                            int flags)
6599 {
6600         void *hdr;
6601         int err;
6602
6603         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
6604         if (!hdr)
6605                 return -EMSGSIZE;
6606
6607         if (devlink_nl_put_handle(msg, devlink))
6608                 goto nla_put_failure;
6609
6610         if (nla_put_string(msg, DEVLINK_ATTR_TRAP_GROUP_NAME,
6611                            group_item->group->name))
6612                 goto nla_put_failure;
6613
6614         if (group_item->group->generic &&
6615             nla_put_flag(msg, DEVLINK_ATTR_TRAP_GENERIC))
6616                 goto nla_put_failure;
6617
6618         if (group_item->policer_item &&
6619             nla_put_u32(msg, DEVLINK_ATTR_TRAP_POLICER_ID,
6620                         group_item->policer_item->policer->id))
6621                 goto nla_put_failure;
6622
6623         err = devlink_trap_stats_put(msg, group_item->stats);
6624         if (err)
6625                 goto nla_put_failure;
6626
6627         genlmsg_end(msg, hdr);
6628
6629         return 0;
6630
6631 nla_put_failure:
6632         genlmsg_cancel(msg, hdr);
6633         return -EMSGSIZE;
6634 }
6635
6636 static int devlink_nl_cmd_trap_group_get_doit(struct sk_buff *skb,
6637                                               struct genl_info *info)
6638 {
6639         struct netlink_ext_ack *extack = info->extack;
6640         struct devlink *devlink = info->user_ptr[0];
6641         struct devlink_trap_group_item *group_item;
6642         struct sk_buff *msg;
6643         int err;
6644
6645         if (list_empty(&devlink->trap_group_list))
6646                 return -EOPNOTSUPP;
6647
6648         group_item = devlink_trap_group_item_get_from_info(devlink, info);
6649         if (!group_item) {
6650                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap group");
6651                 return -ENOENT;
6652         }
6653
6654         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6655         if (!msg)
6656                 return -ENOMEM;
6657
6658         err = devlink_nl_trap_group_fill(msg, devlink, group_item,
6659                                          DEVLINK_CMD_TRAP_GROUP_NEW,
6660                                          info->snd_portid, info->snd_seq, 0);
6661         if (err)
6662                 goto err_trap_group_fill;
6663
6664         return genlmsg_reply(msg, info);
6665
6666 err_trap_group_fill:
6667         nlmsg_free(msg);
6668         return err;
6669 }
6670
6671 static int devlink_nl_cmd_trap_group_get_dumpit(struct sk_buff *msg,
6672                                                 struct netlink_callback *cb)
6673 {
6674         enum devlink_command cmd = DEVLINK_CMD_TRAP_GROUP_NEW;
6675         struct devlink_trap_group_item *group_item;
6676         u32 portid = NETLINK_CB(cb->skb).portid;
6677         struct devlink *devlink;
6678         int start = cb->args[0];
6679         int idx = 0;
6680         int err;
6681
6682         mutex_lock(&devlink_mutex);
6683         list_for_each_entry(devlink, &devlink_list, list) {
6684                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
6685                         continue;
6686                 mutex_lock(&devlink->lock);
6687                 list_for_each_entry(group_item, &devlink->trap_group_list,
6688                                     list) {
6689                         if (idx < start) {
6690                                 idx++;
6691                                 continue;
6692                         }
6693                         err = devlink_nl_trap_group_fill(msg, devlink,
6694                                                          group_item, cmd,
6695                                                          portid,
6696                                                          cb->nlh->nlmsg_seq,
6697                                                          NLM_F_MULTI);
6698                         if (err) {
6699                                 mutex_unlock(&devlink->lock);
6700                                 goto out;
6701                         }
6702                         idx++;
6703                 }
6704                 mutex_unlock(&devlink->lock);
6705         }
6706 out:
6707         mutex_unlock(&devlink_mutex);
6708
6709         cb->args[0] = idx;
6710         return msg->len;
6711 }
6712
6713 static int
6714 __devlink_trap_group_action_set(struct devlink *devlink,
6715                                 struct devlink_trap_group_item *group_item,
6716                                 enum devlink_trap_action trap_action,
6717                                 struct netlink_ext_ack *extack)
6718 {
6719         const char *group_name = group_item->group->name;
6720         struct devlink_trap_item *trap_item;
6721         int err;
6722
6723         list_for_each_entry(trap_item, &devlink->trap_list, list) {
6724                 if (strcmp(trap_item->group_item->group->name, group_name))
6725                         continue;
6726                 err = __devlink_trap_action_set(devlink, trap_item,
6727                                                 trap_action, extack);
6728                 if (err)
6729                         return err;
6730         }
6731
6732         return 0;
6733 }
6734
6735 static int
6736 devlink_trap_group_action_set(struct devlink *devlink,
6737                               struct devlink_trap_group_item *group_item,
6738                               struct genl_info *info, bool *p_modified)
6739 {
6740         enum devlink_trap_action trap_action;
6741         int err;
6742
6743         if (!info->attrs[DEVLINK_ATTR_TRAP_ACTION])
6744                 return 0;
6745
6746         err = devlink_trap_action_get_from_info(info, &trap_action);
6747         if (err) {
6748                 NL_SET_ERR_MSG_MOD(info->extack, "Invalid trap action");
6749                 return -EINVAL;
6750         }
6751
6752         err = __devlink_trap_group_action_set(devlink, group_item, trap_action,
6753                                               info->extack);
6754         if (err)
6755                 return err;
6756
6757         *p_modified = true;
6758
6759         return 0;
6760 }
6761
6762 static int devlink_trap_group_set(struct devlink *devlink,
6763                                   struct devlink_trap_group_item *group_item,
6764                                   struct genl_info *info)
6765 {
6766         struct devlink_trap_policer_item *policer_item;
6767         struct netlink_ext_ack *extack = info->extack;
6768         const struct devlink_trap_policer *policer;
6769         struct nlattr **attrs = info->attrs;
6770         int err;
6771
6772         if (!attrs[DEVLINK_ATTR_TRAP_POLICER_ID])
6773                 return 0;
6774
6775         if (!devlink->ops->trap_group_set)
6776                 return -EOPNOTSUPP;
6777
6778         policer_item = group_item->policer_item;
6779         if (attrs[DEVLINK_ATTR_TRAP_POLICER_ID]) {
6780                 u32 policer_id;
6781
6782                 policer_id = nla_get_u32(attrs[DEVLINK_ATTR_TRAP_POLICER_ID]);
6783                 policer_item = devlink_trap_policer_item_lookup(devlink,
6784                                                                 policer_id);
6785                 if (policer_id && !policer_item) {
6786                         NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap policer");
6787                         return -ENOENT;
6788                 }
6789         }
6790         policer = policer_item ? policer_item->policer : NULL;
6791
6792         err = devlink->ops->trap_group_set(devlink, group_item->group, policer,
6793                                            extack);
6794         if (err)
6795                 return err;
6796
6797         group_item->policer_item = policer_item;
6798
6799         return 0;
6800 }
6801
6802 static int devlink_nl_cmd_trap_group_set_doit(struct sk_buff *skb,
6803                                               struct genl_info *info)
6804 {
6805         struct netlink_ext_ack *extack = info->extack;
6806         struct devlink *devlink = info->user_ptr[0];
6807         struct devlink_trap_group_item *group_item;
6808         bool modified = false;
6809         int err;
6810
6811         if (list_empty(&devlink->trap_group_list))
6812                 return -EOPNOTSUPP;
6813
6814         group_item = devlink_trap_group_item_get_from_info(devlink, info);
6815         if (!group_item) {
6816                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap group");
6817                 return -ENOENT;
6818         }
6819
6820         err = devlink_trap_group_action_set(devlink, group_item, info,
6821                                             &modified);
6822         if (err)
6823                 return err;
6824
6825         err = devlink_trap_group_set(devlink, group_item, info);
6826         if (err)
6827                 goto err_trap_group_set;
6828
6829         return 0;
6830
6831 err_trap_group_set:
6832         if (modified)
6833                 NL_SET_ERR_MSG_MOD(extack, "Trap group set failed, but some changes were committed already");
6834         return err;
6835 }
6836
6837 static struct devlink_trap_policer_item *
6838 devlink_trap_policer_item_get_from_info(struct devlink *devlink,
6839                                         struct genl_info *info)
6840 {
6841         u32 id;
6842
6843         if (!info->attrs[DEVLINK_ATTR_TRAP_POLICER_ID])
6844                 return NULL;
6845         id = nla_get_u32(info->attrs[DEVLINK_ATTR_TRAP_POLICER_ID]);
6846
6847         return devlink_trap_policer_item_lookup(devlink, id);
6848 }
6849
6850 static int
6851 devlink_trap_policer_stats_put(struct sk_buff *msg, struct devlink *devlink,
6852                                const struct devlink_trap_policer *policer)
6853 {
6854         struct nlattr *attr;
6855         u64 drops;
6856         int err;
6857
6858         if (!devlink->ops->trap_policer_counter_get)
6859                 return 0;
6860
6861         err = devlink->ops->trap_policer_counter_get(devlink, policer, &drops);
6862         if (err)
6863                 return err;
6864
6865         attr = nla_nest_start(msg, DEVLINK_ATTR_STATS);
6866         if (!attr)
6867                 return -EMSGSIZE;
6868
6869         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_STATS_RX_DROPPED, drops,
6870                               DEVLINK_ATTR_PAD))
6871                 goto nla_put_failure;
6872
6873         nla_nest_end(msg, attr);
6874
6875         return 0;
6876
6877 nla_put_failure:
6878         nla_nest_cancel(msg, attr);
6879         return -EMSGSIZE;
6880 }
6881
6882 static int
6883 devlink_nl_trap_policer_fill(struct sk_buff *msg, struct devlink *devlink,
6884                              const struct devlink_trap_policer_item *policer_item,
6885                              enum devlink_command cmd, u32 portid, u32 seq,
6886                              int flags)
6887 {
6888         void *hdr;
6889         int err;
6890
6891         hdr = genlmsg_put(msg, portid, seq, &devlink_nl_family, flags, cmd);
6892         if (!hdr)
6893                 return -EMSGSIZE;
6894
6895         if (devlink_nl_put_handle(msg, devlink))
6896                 goto nla_put_failure;
6897
6898         if (nla_put_u32(msg, DEVLINK_ATTR_TRAP_POLICER_ID,
6899                         policer_item->policer->id))
6900                 goto nla_put_failure;
6901
6902         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_TRAP_POLICER_RATE,
6903                               policer_item->rate, DEVLINK_ATTR_PAD))
6904                 goto nla_put_failure;
6905
6906         if (nla_put_u64_64bit(msg, DEVLINK_ATTR_TRAP_POLICER_BURST,
6907                               policer_item->burst, DEVLINK_ATTR_PAD))
6908                 goto nla_put_failure;
6909
6910         err = devlink_trap_policer_stats_put(msg, devlink,
6911                                              policer_item->policer);
6912         if (err)
6913                 goto nla_put_failure;
6914
6915         genlmsg_end(msg, hdr);
6916
6917         return 0;
6918
6919 nla_put_failure:
6920         genlmsg_cancel(msg, hdr);
6921         return -EMSGSIZE;
6922 }
6923
6924 static int devlink_nl_cmd_trap_policer_get_doit(struct sk_buff *skb,
6925                                                 struct genl_info *info)
6926 {
6927         struct devlink_trap_policer_item *policer_item;
6928         struct netlink_ext_ack *extack = info->extack;
6929         struct devlink *devlink = info->user_ptr[0];
6930         struct sk_buff *msg;
6931         int err;
6932
6933         if (list_empty(&devlink->trap_policer_list))
6934                 return -EOPNOTSUPP;
6935
6936         policer_item = devlink_trap_policer_item_get_from_info(devlink, info);
6937         if (!policer_item) {
6938                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap policer");
6939                 return -ENOENT;
6940         }
6941
6942         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
6943         if (!msg)
6944                 return -ENOMEM;
6945
6946         err = devlink_nl_trap_policer_fill(msg, devlink, policer_item,
6947                                            DEVLINK_CMD_TRAP_POLICER_NEW,
6948                                            info->snd_portid, info->snd_seq, 0);
6949         if (err)
6950                 goto err_trap_policer_fill;
6951
6952         return genlmsg_reply(msg, info);
6953
6954 err_trap_policer_fill:
6955         nlmsg_free(msg);
6956         return err;
6957 }
6958
6959 static int devlink_nl_cmd_trap_policer_get_dumpit(struct sk_buff *msg,
6960                                                   struct netlink_callback *cb)
6961 {
6962         enum devlink_command cmd = DEVLINK_CMD_TRAP_POLICER_NEW;
6963         struct devlink_trap_policer_item *policer_item;
6964         u32 portid = NETLINK_CB(cb->skb).portid;
6965         struct devlink *devlink;
6966         int start = cb->args[0];
6967         int idx = 0;
6968         int err;
6969
6970         mutex_lock(&devlink_mutex);
6971         list_for_each_entry(devlink, &devlink_list, list) {
6972                 if (!net_eq(devlink_net(devlink), sock_net(msg->sk)))
6973                         continue;
6974                 mutex_lock(&devlink->lock);
6975                 list_for_each_entry(policer_item, &devlink->trap_policer_list,
6976                                     list) {
6977                         if (idx < start) {
6978                                 idx++;
6979                                 continue;
6980                         }
6981                         err = devlink_nl_trap_policer_fill(msg, devlink,
6982                                                            policer_item, cmd,
6983                                                            portid,
6984                                                            cb->nlh->nlmsg_seq,
6985                                                            NLM_F_MULTI);
6986                         if (err) {
6987                                 mutex_unlock(&devlink->lock);
6988                                 goto out;
6989                         }
6990                         idx++;
6991                 }
6992                 mutex_unlock(&devlink->lock);
6993         }
6994 out:
6995         mutex_unlock(&devlink_mutex);
6996
6997         cb->args[0] = idx;
6998         return msg->len;
6999 }
7000
7001 static int
7002 devlink_trap_policer_set(struct devlink *devlink,
7003                          struct devlink_trap_policer_item *policer_item,
7004                          struct genl_info *info)
7005 {
7006         struct netlink_ext_ack *extack = info->extack;
7007         struct nlattr **attrs = info->attrs;
7008         u64 rate, burst;
7009         int err;
7010
7011         rate = policer_item->rate;
7012         burst = policer_item->burst;
7013
7014         if (attrs[DEVLINK_ATTR_TRAP_POLICER_RATE])
7015                 rate = nla_get_u64(attrs[DEVLINK_ATTR_TRAP_POLICER_RATE]);
7016
7017         if (attrs[DEVLINK_ATTR_TRAP_POLICER_BURST])
7018                 burst = nla_get_u64(attrs[DEVLINK_ATTR_TRAP_POLICER_BURST]);
7019
7020         if (rate < policer_item->policer->min_rate) {
7021                 NL_SET_ERR_MSG_MOD(extack, "Policer rate lower than limit");
7022                 return -EINVAL;
7023         }
7024
7025         if (rate > policer_item->policer->max_rate) {
7026                 NL_SET_ERR_MSG_MOD(extack, "Policer rate higher than limit");
7027                 return -EINVAL;
7028         }
7029
7030         if (burst < policer_item->policer->min_burst) {
7031                 NL_SET_ERR_MSG_MOD(extack, "Policer burst size lower than limit");
7032                 return -EINVAL;
7033         }
7034
7035         if (burst > policer_item->policer->max_burst) {
7036                 NL_SET_ERR_MSG_MOD(extack, "Policer burst size higher than limit");
7037                 return -EINVAL;
7038         }
7039
7040         err = devlink->ops->trap_policer_set(devlink, policer_item->policer,
7041                                              rate, burst, info->extack);
7042         if (err)
7043                 return err;
7044
7045         policer_item->rate = rate;
7046         policer_item->burst = burst;
7047
7048         return 0;
7049 }
7050
7051 static int devlink_nl_cmd_trap_policer_set_doit(struct sk_buff *skb,
7052                                                 struct genl_info *info)
7053 {
7054         struct devlink_trap_policer_item *policer_item;
7055         struct netlink_ext_ack *extack = info->extack;
7056         struct devlink *devlink = info->user_ptr[0];
7057
7058         if (list_empty(&devlink->trap_policer_list))
7059                 return -EOPNOTSUPP;
7060
7061         if (!devlink->ops->trap_policer_set)
7062                 return -EOPNOTSUPP;
7063
7064         policer_item = devlink_trap_policer_item_get_from_info(devlink, info);
7065         if (!policer_item) {
7066                 NL_SET_ERR_MSG_MOD(extack, "Device did not register this trap policer");
7067                 return -ENOENT;
7068         }
7069
7070         return devlink_trap_policer_set(devlink, policer_item, info);
7071 }
7072
7073 static const struct nla_policy devlink_nl_policy[DEVLINK_ATTR_MAX + 1] = {
7074         [DEVLINK_ATTR_UNSPEC] = { .strict_start_type =
7075                 DEVLINK_ATTR_TRAP_POLICER_ID },
7076         [DEVLINK_ATTR_BUS_NAME] = { .type = NLA_NUL_STRING },
7077         [DEVLINK_ATTR_DEV_NAME] = { .type = NLA_NUL_STRING },
7078         [DEVLINK_ATTR_PORT_INDEX] = { .type = NLA_U32 },
7079         [DEVLINK_ATTR_PORT_TYPE] = NLA_POLICY_RANGE(NLA_U16, DEVLINK_PORT_TYPE_AUTO,
7080                                                     DEVLINK_PORT_TYPE_IB),
7081         [DEVLINK_ATTR_PORT_SPLIT_COUNT] = { .type = NLA_U32 },
7082         [DEVLINK_ATTR_SB_INDEX] = { .type = NLA_U32 },
7083         [DEVLINK_ATTR_SB_POOL_INDEX] = { .type = NLA_U16 },
7084         [DEVLINK_ATTR_SB_POOL_TYPE] = { .type = NLA_U8 },
7085         [DEVLINK_ATTR_SB_POOL_SIZE] = { .type = NLA_U32 },
7086         [DEVLINK_ATTR_SB_POOL_THRESHOLD_TYPE] = { .type = NLA_U8 },
7087         [DEVLINK_ATTR_SB_THRESHOLD] = { .type = NLA_U32 },
7088         [DEVLINK_ATTR_SB_TC_INDEX] = { .type = NLA_U16 },
7089         [DEVLINK_ATTR_ESWITCH_MODE] = NLA_POLICY_RANGE(NLA_U16, DEVLINK_ESWITCH_MODE_LEGACY,
7090                                                        DEVLINK_ESWITCH_MODE_SWITCHDEV),
7091         [DEVLINK_ATTR_ESWITCH_INLINE_MODE] = { .type = NLA_U8 },
7092         [DEVLINK_ATTR_ESWITCH_ENCAP_MODE] = { .type = NLA_U8 },
7093         [DEVLINK_ATTR_DPIPE_TABLE_NAME] = { .type = NLA_NUL_STRING },
7094         [DEVLINK_ATTR_DPIPE_TABLE_COUNTERS_ENABLED] = { .type = NLA_U8 },
7095         [DEVLINK_ATTR_RESOURCE_ID] = { .type = NLA_U64},
7096         [DEVLINK_ATTR_RESOURCE_SIZE] = { .type = NLA_U64},
7097         [DEVLINK_ATTR_PARAM_NAME] = { .type = NLA_NUL_STRING },
7098         [DEVLINK_ATTR_PARAM_TYPE] = { .type = NLA_U8 },
7099         [DEVLINK_ATTR_PARAM_VALUE_CMODE] = { .type = NLA_U8 },
7100         [DEVLINK_ATTR_REGION_NAME] = { .type = NLA_NUL_STRING },
7101         [DEVLINK_ATTR_REGION_SNAPSHOT_ID] = { .type = NLA_U32 },
7102         [DEVLINK_ATTR_REGION_CHUNK_ADDR] = { .type = NLA_U64 },
7103         [DEVLINK_ATTR_REGION_CHUNK_LEN] = { .type = NLA_U64 },
7104         [DEVLINK_ATTR_HEALTH_REPORTER_NAME] = { .type = NLA_NUL_STRING },
7105         [DEVLINK_ATTR_HEALTH_REPORTER_GRACEFUL_PERIOD] = { .type = NLA_U64 },
7106         [DEVLINK_ATTR_HEALTH_REPORTER_AUTO_RECOVER] = { .type = NLA_U8 },
7107         [DEVLINK_ATTR_FLASH_UPDATE_FILE_NAME] = { .type = NLA_NUL_STRING },
7108         [DEVLINK_ATTR_FLASH_UPDATE_COMPONENT] = { .type = NLA_NUL_STRING },
7109         [DEVLINK_ATTR_FLASH_UPDATE_OVERWRITE_MASK] =
7110                 NLA_POLICY_BITFIELD32(DEVLINK_SUPPORTED_FLASH_OVERWRITE_SECTIONS),
7111         [DEVLINK_ATTR_TRAP_NAME] = { .type = NLA_NUL_STRING },
7112         [DEVLINK_ATTR_TRAP_ACTION] = { .type = NLA_U8 },
7113         [DEVLINK_ATTR_TRAP_GROUP_NAME] = { .type = NLA_NUL_STRING },
7114         [DEVLINK_ATTR_NETNS_PID] = { .type = NLA_U32 },
7115         [DEVLINK_ATTR_NETNS_FD] = { .type = NLA_U32 },
7116         [DEVLINK_ATTR_NETNS_ID] = { .type = NLA_U32 },
7117         [DEVLINK_ATTR_HEALTH_REPORTER_AUTO_DUMP] = { .type = NLA_U8 },
7118         [DEVLINK_ATTR_TRAP_POLICER_ID] = { .type = NLA_U32 },
7119         [DEVLINK_ATTR_TRAP_POLICER_RATE] = { .type = NLA_U64 },
7120         [DEVLINK_ATTR_TRAP_POLICER_BURST] = { .type = NLA_U64 },
7121         [DEVLINK_ATTR_PORT_FUNCTION] = { .type = NLA_NESTED },
7122 };
7123
7124 static const struct genl_ops devlink_nl_ops[] = {
7125         {
7126                 .cmd = DEVLINK_CMD_GET,
7127                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7128                 .doit = devlink_nl_cmd_get_doit,
7129                 .dumpit = devlink_nl_cmd_get_dumpit,
7130                 /* can be retrieved by unprivileged users */
7131         },
7132         {
7133                 .cmd = DEVLINK_CMD_PORT_GET,
7134                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7135                 .doit = devlink_nl_cmd_port_get_doit,
7136                 .dumpit = devlink_nl_cmd_port_get_dumpit,
7137                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7138                 /* can be retrieved by unprivileged users */
7139         },
7140         {
7141                 .cmd = DEVLINK_CMD_PORT_SET,
7142                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7143                 .doit = devlink_nl_cmd_port_set_doit,
7144                 .flags = GENL_ADMIN_PERM,
7145                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7146         },
7147         {
7148                 .cmd = DEVLINK_CMD_PORT_SPLIT,
7149                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7150                 .doit = devlink_nl_cmd_port_split_doit,
7151                 .flags = GENL_ADMIN_PERM,
7152                 .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
7153         },
7154         {
7155                 .cmd = DEVLINK_CMD_PORT_UNSPLIT,
7156                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7157                 .doit = devlink_nl_cmd_port_unsplit_doit,
7158                 .flags = GENL_ADMIN_PERM,
7159                 .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
7160         },
7161         {
7162                 .cmd = DEVLINK_CMD_SB_GET,
7163                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7164                 .doit = devlink_nl_cmd_sb_get_doit,
7165                 .dumpit = devlink_nl_cmd_sb_get_dumpit,
7166                 /* can be retrieved by unprivileged users */
7167         },
7168         {
7169                 .cmd = DEVLINK_CMD_SB_POOL_GET,
7170                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7171                 .doit = devlink_nl_cmd_sb_pool_get_doit,
7172                 .dumpit = devlink_nl_cmd_sb_pool_get_dumpit,
7173                 /* can be retrieved by unprivileged users */
7174         },
7175         {
7176                 .cmd = DEVLINK_CMD_SB_POOL_SET,
7177                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7178                 .doit = devlink_nl_cmd_sb_pool_set_doit,
7179                 .flags = GENL_ADMIN_PERM,
7180         },
7181         {
7182                 .cmd = DEVLINK_CMD_SB_PORT_POOL_GET,
7183                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7184                 .doit = devlink_nl_cmd_sb_port_pool_get_doit,
7185                 .dumpit = devlink_nl_cmd_sb_port_pool_get_dumpit,
7186                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7187                 /* can be retrieved by unprivileged users */
7188         },
7189         {
7190                 .cmd = DEVLINK_CMD_SB_PORT_POOL_SET,
7191                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7192                 .doit = devlink_nl_cmd_sb_port_pool_set_doit,
7193                 .flags = GENL_ADMIN_PERM,
7194                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7195         },
7196         {
7197                 .cmd = DEVLINK_CMD_SB_TC_POOL_BIND_GET,
7198                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7199                 .doit = devlink_nl_cmd_sb_tc_pool_bind_get_doit,
7200                 .dumpit = devlink_nl_cmd_sb_tc_pool_bind_get_dumpit,
7201                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7202                 /* can be retrieved by unprivileged users */
7203         },
7204         {
7205                 .cmd = DEVLINK_CMD_SB_TC_POOL_BIND_SET,
7206                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7207                 .doit = devlink_nl_cmd_sb_tc_pool_bind_set_doit,
7208                 .flags = GENL_ADMIN_PERM,
7209                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7210         },
7211         {
7212                 .cmd = DEVLINK_CMD_SB_OCC_SNAPSHOT,
7213                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7214                 .doit = devlink_nl_cmd_sb_occ_snapshot_doit,
7215                 .flags = GENL_ADMIN_PERM,
7216         },
7217         {
7218                 .cmd = DEVLINK_CMD_SB_OCC_MAX_CLEAR,
7219                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7220                 .doit = devlink_nl_cmd_sb_occ_max_clear_doit,
7221                 .flags = GENL_ADMIN_PERM,
7222         },
7223         {
7224                 .cmd = DEVLINK_CMD_ESWITCH_GET,
7225                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7226                 .doit = devlink_nl_cmd_eswitch_get_doit,
7227                 .flags = GENL_ADMIN_PERM,
7228                 .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
7229         },
7230         {
7231                 .cmd = DEVLINK_CMD_ESWITCH_SET,
7232                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7233                 .doit = devlink_nl_cmd_eswitch_set_doit,
7234                 .flags = GENL_ADMIN_PERM,
7235                 .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
7236         },
7237         {
7238                 .cmd = DEVLINK_CMD_DPIPE_TABLE_GET,
7239                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7240                 .doit = devlink_nl_cmd_dpipe_table_get,
7241                 /* can be retrieved by unprivileged users */
7242         },
7243         {
7244                 .cmd = DEVLINK_CMD_DPIPE_ENTRIES_GET,
7245                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7246                 .doit = devlink_nl_cmd_dpipe_entries_get,
7247                 /* can be retrieved by unprivileged users */
7248         },
7249         {
7250                 .cmd = DEVLINK_CMD_DPIPE_HEADERS_GET,
7251                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7252                 .doit = devlink_nl_cmd_dpipe_headers_get,
7253                 /* can be retrieved by unprivileged users */
7254         },
7255         {
7256                 .cmd = DEVLINK_CMD_DPIPE_TABLE_COUNTERS_SET,
7257                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7258                 .doit = devlink_nl_cmd_dpipe_table_counters_set,
7259                 .flags = GENL_ADMIN_PERM,
7260         },
7261         {
7262                 .cmd = DEVLINK_CMD_RESOURCE_SET,
7263                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7264                 .doit = devlink_nl_cmd_resource_set,
7265                 .flags = GENL_ADMIN_PERM,
7266         },
7267         {
7268                 .cmd = DEVLINK_CMD_RESOURCE_DUMP,
7269                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7270                 .doit = devlink_nl_cmd_resource_dump,
7271                 /* can be retrieved by unprivileged users */
7272         },
7273         {
7274                 .cmd = DEVLINK_CMD_RELOAD,
7275                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7276                 .doit = devlink_nl_cmd_reload,
7277                 .flags = GENL_ADMIN_PERM,
7278                 .internal_flags = DEVLINK_NL_FLAG_NO_LOCK,
7279         },
7280         {
7281                 .cmd = DEVLINK_CMD_PARAM_GET,
7282                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7283                 .doit = devlink_nl_cmd_param_get_doit,
7284                 .dumpit = devlink_nl_cmd_param_get_dumpit,
7285                 /* can be retrieved by unprivileged users */
7286         },
7287         {
7288                 .cmd = DEVLINK_CMD_PARAM_SET,
7289                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7290                 .doit = devlink_nl_cmd_param_set_doit,
7291                 .flags = GENL_ADMIN_PERM,
7292         },
7293         {
7294                 .cmd = DEVLINK_CMD_PORT_PARAM_GET,
7295                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7296                 .doit = devlink_nl_cmd_port_param_get_doit,
7297                 .dumpit = devlink_nl_cmd_port_param_get_dumpit,
7298                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7299                 /* can be retrieved by unprivileged users */
7300         },
7301         {
7302                 .cmd = DEVLINK_CMD_PORT_PARAM_SET,
7303                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7304                 .doit = devlink_nl_cmd_port_param_set_doit,
7305                 .flags = GENL_ADMIN_PERM,
7306                 .internal_flags = DEVLINK_NL_FLAG_NEED_PORT,
7307         },
7308         {
7309                 .cmd = DEVLINK_CMD_REGION_GET,
7310                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7311                 .doit = devlink_nl_cmd_region_get_doit,
7312                 .dumpit = devlink_nl_cmd_region_get_dumpit,
7313                 .flags = GENL_ADMIN_PERM,
7314         },
7315         {
7316                 .cmd = DEVLINK_CMD_REGION_NEW,
7317                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7318                 .doit = devlink_nl_cmd_region_new,
7319                 .flags = GENL_ADMIN_PERM,
7320         },
7321         {
7322                 .cmd = DEVLINK_CMD_REGION_DEL,
7323                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7324                 .doit = devlink_nl_cmd_region_del,
7325                 .flags = GENL_ADMIN_PERM,
7326         },
7327         {
7328                 .cmd = DEVLINK_CMD_REGION_READ,
7329                 .validate = GENL_DONT_VALIDATE_STRICT |
7330                             GENL_DONT_VALIDATE_DUMP_STRICT,
7331                 .dumpit = devlink_nl_cmd_region_read_dumpit,
7332                 .flags = GENL_ADMIN_PERM,
7333         },
7334         {
7335                 .cmd = DEVLINK_CMD_INFO_GET,
7336                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7337                 .doit = devlink_nl_cmd_info_get_doit,
7338                 .dumpit = devlink_nl_cmd_info_get_dumpit,
7339                 /* can be retrieved by unprivileged users */
7340         },
7341         {
7342                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_GET,
7343                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7344                 .doit = devlink_nl_cmd_health_reporter_get_doit,
7345                 .dumpit = devlink_nl_cmd_health_reporter_get_dumpit,
7346                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7347                                   DEVLINK_NL_FLAG_NO_LOCK,
7348                 /* can be retrieved by unprivileged users */
7349         },
7350         {
7351                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_SET,
7352                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7353                 .doit = devlink_nl_cmd_health_reporter_set_doit,
7354                 .flags = GENL_ADMIN_PERM,
7355                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7356                                   DEVLINK_NL_FLAG_NO_LOCK,
7357         },
7358         {
7359                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_RECOVER,
7360                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7361                 .doit = devlink_nl_cmd_health_reporter_recover_doit,
7362                 .flags = GENL_ADMIN_PERM,
7363                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7364                                   DEVLINK_NL_FLAG_NO_LOCK,
7365         },
7366         {
7367                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_DIAGNOSE,
7368                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7369                 .doit = devlink_nl_cmd_health_reporter_diagnose_doit,
7370                 .flags = GENL_ADMIN_PERM,
7371                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7372                                   DEVLINK_NL_FLAG_NO_LOCK,
7373         },
7374         {
7375                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_DUMP_GET,
7376                 .validate = GENL_DONT_VALIDATE_STRICT |
7377                             GENL_DONT_VALIDATE_DUMP_STRICT,
7378                 .dumpit = devlink_nl_cmd_health_reporter_dump_get_dumpit,
7379                 .flags = GENL_ADMIN_PERM,
7380                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7381                                   DEVLINK_NL_FLAG_NO_LOCK,
7382         },
7383         {
7384                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_DUMP_CLEAR,
7385                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7386                 .doit = devlink_nl_cmd_health_reporter_dump_clear_doit,
7387                 .flags = GENL_ADMIN_PERM,
7388                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7389                                   DEVLINK_NL_FLAG_NO_LOCK,
7390         },
7391         {
7392                 .cmd = DEVLINK_CMD_HEALTH_REPORTER_TEST,
7393                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7394                 .doit = devlink_nl_cmd_health_reporter_test_doit,
7395                 .flags = GENL_ADMIN_PERM,
7396                 .internal_flags = DEVLINK_NL_FLAG_NEED_DEVLINK_OR_PORT |
7397                                   DEVLINK_NL_FLAG_NO_LOCK,
7398         },
7399         {
7400                 .cmd = DEVLINK_CMD_FLASH_UPDATE,
7401                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
7402                 .doit = devlink_nl_cmd_flash_update,
7403                 .flags = GENL_ADMIN_PERM,
7404         },
7405         {
7406                 .cmd = DEVLINK_CMD_TRAP_GET,
7407                 .doit = devlink_nl_cmd_trap_get_doit,
7408                 .dumpit = devlink_nl_cmd_trap_get_dumpit,
7409                 /* can be retrieved by unprivileged users */
7410         },
7411         {
7412                 .cmd = DEVLINK_CMD_TRAP_SET,
7413                 .doit = devlink_nl_cmd_trap_set_doit,
7414                 .flags = GENL_ADMIN_PERM,
7415         },
7416         {
7417                 .cmd = DEVLINK_CMD_TRAP_GROUP_GET,
7418                 .doit = devlink_nl_cmd_trap_group_get_doit,
7419                 .dumpit = devlink_nl_cmd_trap_group_get_dumpit,
7420                 /* can be retrieved by unprivileged users */
7421         },
7422         {
7423                 .cmd = DEVLINK_CMD_TRAP_GROUP_SET,
7424                 .doit = devlink_nl_cmd_trap_group_set_doit,
7425                 .flags = GENL_ADMIN_PERM,
7426         },
7427         {
7428                 .cmd = DEVLINK_CMD_TRAP_POLICER_GET,
7429                 .doit = devlink_nl_cmd_trap_policer_get_doit,
7430                 .dumpit = devlink_nl_cmd_trap_policer_get_dumpit,
7431                 /* can be retrieved by unprivileged users */
7432         },
7433         {
7434                 .cmd = DEVLINK_CMD_TRAP_POLICER_SET,
7435                 .doit = devlink_nl_cmd_trap_policer_set_doit,
7436                 .flags = GENL_ADMIN_PERM,
7437         },
7438 };
7439
7440 static struct genl_family devlink_nl_family __ro_after_init = {
7441         .name           = DEVLINK_GENL_NAME,
7442         .version        = DEVLINK_GENL_VERSION,
7443         .maxattr        = DEVLINK_ATTR_MAX,
7444         .policy = devlink_nl_policy,
7445         .netnsok        = true,
7446         .pre_doit       = devlink_nl_pre_doit,
7447         .post_doit      = devlink_nl_post_doit,
7448         .module         = THIS_MODULE,
7449         .ops            = devlink_nl_ops,
7450         .n_ops          = ARRAY_SIZE(devlink_nl_ops),
7451         .mcgrps         = devlink_nl_mcgrps,
7452         .n_mcgrps       = ARRAY_SIZE(devlink_nl_mcgrps),
7453 };
7454
7455 /**
7456  *      devlink_alloc - Allocate new devlink instance resources
7457  *
7458  *      @ops: ops
7459  *      @priv_size: size of user private data
7460  *
7461  *      Allocate new devlink instance resources, including devlink index
7462  *      and name.
7463  */
7464 struct devlink *devlink_alloc(const struct devlink_ops *ops, size_t priv_size)
7465 {
7466         struct devlink *devlink;
7467
7468         if (WARN_ON(!ops))
7469                 return NULL;
7470
7471         devlink = kzalloc(sizeof(*devlink) + priv_size, GFP_KERNEL);
7472         if (!devlink)
7473                 return NULL;
7474         devlink->ops = ops;
7475         xa_init_flags(&devlink->snapshot_ids, XA_FLAGS_ALLOC);
7476         __devlink_net_set(devlink, &init_net);
7477         INIT_LIST_HEAD(&devlink->port_list);
7478         INIT_LIST_HEAD(&devlink->sb_list);
7479         INIT_LIST_HEAD_RCU(&devlink->dpipe_table_list);
7480         INIT_LIST_HEAD(&devlink->resource_list);
7481         INIT_LIST_HEAD(&devlink->param_list);
7482         INIT_LIST_HEAD(&devlink->region_list);
7483         INIT_LIST_HEAD(&devlink->reporter_list);
7484         INIT_LIST_HEAD(&devlink->trap_list);
7485         INIT_LIST_HEAD(&devlink->trap_group_list);
7486         INIT_LIST_HEAD(&devlink->trap_policer_list);
7487         mutex_init(&devlink->lock);
7488         mutex_init(&devlink->reporters_lock);
7489         return devlink;
7490 }
7491 EXPORT_SYMBOL_GPL(devlink_alloc);
7492
7493 /**
7494  *      devlink_register - Register devlink instance
7495  *
7496  *      @devlink: devlink
7497  *      @dev: parent device
7498  */
7499 int devlink_register(struct devlink *devlink, struct device *dev)
7500 {
7501         devlink->dev = dev;
7502         devlink->registered = true;
7503         mutex_lock(&devlink_mutex);
7504         list_add_tail(&devlink->list, &devlink_list);
7505         devlink_notify(devlink, DEVLINK_CMD_NEW);
7506         mutex_unlock(&devlink_mutex);
7507         return 0;
7508 }
7509 EXPORT_SYMBOL_GPL(devlink_register);
7510
7511 /**
7512  *      devlink_unregister - Unregister devlink instance
7513  *
7514  *      @devlink: devlink
7515  */
7516 void devlink_unregister(struct devlink *devlink)
7517 {
7518         mutex_lock(&devlink_mutex);
7519         WARN_ON(devlink_reload_supported(devlink) &&
7520                 devlink->reload_enabled);
7521         devlink_notify(devlink, DEVLINK_CMD_DEL);
7522         list_del(&devlink->list);
7523         mutex_unlock(&devlink_mutex);
7524 }
7525 EXPORT_SYMBOL_GPL(devlink_unregister);
7526
7527 /**
7528  *      devlink_reload_enable - Enable reload of devlink instance
7529  *
7530  *      @devlink: devlink
7531  *
7532  *      Should be called at end of device initialization
7533  *      process when reload operation is supported.
7534  */
7535 void devlink_reload_enable(struct devlink *devlink)
7536 {
7537         mutex_lock(&devlink_mutex);
7538         devlink->reload_enabled = true;
7539         mutex_unlock(&devlink_mutex);
7540 }
7541 EXPORT_SYMBOL_GPL(devlink_reload_enable);
7542
7543 /**
7544  *      devlink_reload_disable - Disable reload of devlink instance
7545  *
7546  *      @devlink: devlink
7547  *
7548  *      Should be called at the beginning of device cleanup
7549  *      process when reload operation is supported.
7550  */
7551 void devlink_reload_disable(struct devlink *devlink)
7552 {
7553         mutex_lock(&devlink_mutex);
7554         /* Mutex is taken which ensures that no reload operation is in
7555          * progress while setting up forbidded flag.
7556          */
7557         devlink->reload_enabled = false;
7558         mutex_unlock(&devlink_mutex);
7559 }
7560 EXPORT_SYMBOL_GPL(devlink_reload_disable);
7561
7562 /**
7563  *      devlink_free - Free devlink instance resources
7564  *
7565  *      @devlink: devlink
7566  */
7567 void devlink_free(struct devlink *devlink)
7568 {
7569         mutex_destroy(&devlink->reporters_lock);
7570         mutex_destroy(&devlink->lock);
7571         WARN_ON(!list_empty(&devlink->trap_policer_list));
7572         WARN_ON(!list_empty(&devlink->trap_group_list));
7573         WARN_ON(!list_empty(&devlink->trap_list));
7574         WARN_ON(!list_empty(&devlink->reporter_list));
7575         WARN_ON(!list_empty(&devlink->region_list));
7576         WARN_ON(!list_empty(&devlink->param_list));
7577         WARN_ON(!list_empty(&devlink->resource_list));
7578         WARN_ON(!list_empty(&devlink->dpipe_table_list));
7579         WARN_ON(!list_empty(&devlink->sb_list));
7580         WARN_ON(!list_empty(&devlink->port_list));
7581
7582         xa_destroy(&devlink->snapshot_ids);
7583
7584         kfree(devlink);
7585 }
7586 EXPORT_SYMBOL_GPL(devlink_free);
7587
7588 static void devlink_port_type_warn(struct work_struct *work)
7589 {
7590         WARN(true, "Type was not set for devlink port.");
7591 }
7592
7593 static bool devlink_port_type_should_warn(struct devlink_port *devlink_port)
7594 {
7595         /* Ignore CPU and DSA flavours. */
7596         return devlink_port->attrs.flavour != DEVLINK_PORT_FLAVOUR_CPU &&
7597                devlink_port->attrs.flavour != DEVLINK_PORT_FLAVOUR_DSA;
7598 }
7599
7600 #define DEVLINK_PORT_TYPE_WARN_TIMEOUT (HZ * 3600)
7601
7602 static void devlink_port_type_warn_schedule(struct devlink_port *devlink_port)
7603 {
7604         if (!devlink_port_type_should_warn(devlink_port))
7605                 return;
7606         /* Schedule a work to WARN in case driver does not set port
7607          * type within timeout.
7608          */
7609         schedule_delayed_work(&devlink_port->type_warn_dw,
7610                               DEVLINK_PORT_TYPE_WARN_TIMEOUT);
7611 }
7612
7613 static void devlink_port_type_warn_cancel(struct devlink_port *devlink_port)
7614 {
7615         if (!devlink_port_type_should_warn(devlink_port))
7616                 return;
7617         cancel_delayed_work_sync(&devlink_port->type_warn_dw);
7618 }
7619
7620 /**
7621  *      devlink_port_register - Register devlink port
7622  *
7623  *      @devlink: devlink
7624  *      @devlink_port: devlink port
7625  *      @port_index: driver-specific numerical identifier of the port
7626  *
7627  *      Register devlink port with provided port index. User can use
7628  *      any indexing, even hw-related one. devlink_port structure
7629  *      is convenient to be embedded inside user driver private structure.
7630  *      Note that the caller should take care of zeroing the devlink_port
7631  *      structure.
7632  */
7633 int devlink_port_register(struct devlink *devlink,
7634                           struct devlink_port *devlink_port,
7635                           unsigned int port_index)
7636 {
7637         mutex_lock(&devlink->lock);
7638         if (devlink_port_index_exists(devlink, port_index)) {
7639                 mutex_unlock(&devlink->lock);
7640                 return -EEXIST;
7641         }
7642         devlink_port->devlink = devlink;
7643         devlink_port->index = port_index;
7644         devlink_port->registered = true;
7645         spin_lock_init(&devlink_port->type_lock);
7646         INIT_LIST_HEAD(&devlink_port->reporter_list);
7647         mutex_init(&devlink_port->reporters_lock);
7648         list_add_tail(&devlink_port->list, &devlink->port_list);
7649         INIT_LIST_HEAD(&devlink_port->param_list);
7650         mutex_unlock(&devlink->lock);
7651         INIT_DELAYED_WORK(&devlink_port->type_warn_dw, &devlink_port_type_warn);
7652         devlink_port_type_warn_schedule(devlink_port);
7653         devlink_port_notify(devlink_port, DEVLINK_CMD_PORT_NEW);
7654         return 0;
7655 }
7656 EXPORT_SYMBOL_GPL(devlink_port_register);
7657
7658 /**
7659  *      devlink_port_unregister - Unregister devlink port
7660  *
7661  *      @devlink_port: devlink port
7662  */
7663 void devlink_port_unregister(struct devlink_port *devlink_port)
7664 {
7665         struct devlink *devlink = devlink_port->devlink;
7666
7667         devlink_port_type_warn_cancel(devlink_port);
7668         devlink_port_notify(devlink_port, DEVLINK_CMD_PORT_DEL);
7669         mutex_lock(&devlink->lock);
7670         list_del(&devlink_port->list);
7671         mutex_unlock(&devlink->lock);
7672         WARN_ON(!list_empty(&devlink_port->reporter_list));
7673         mutex_destroy(&devlink_port->reporters_lock);
7674 }
7675 EXPORT_SYMBOL_GPL(devlink_port_unregister);
7676
7677 static void __devlink_port_type_set(struct devlink_port *devlink_port,
7678                                     enum devlink_port_type type,
7679                                     void *type_dev)
7680 {
7681         if (WARN_ON(!devlink_port->registered))
7682                 return;
7683         devlink_port_type_warn_cancel(devlink_port);
7684         spin_lock_bh(&devlink_port->type_lock);
7685         devlink_port->type = type;
7686         devlink_port->type_dev = type_dev;
7687         spin_unlock_bh(&devlink_port->type_lock);
7688         devlink_port_notify(devlink_port, DEVLINK_CMD_PORT_NEW);
7689 }
7690
7691 static void devlink_port_type_netdev_checks(struct devlink_port *devlink_port,
7692                                             struct net_device *netdev)
7693 {
7694         const struct net_device_ops *ops = netdev->netdev_ops;
7695
7696         /* If driver registers devlink port, it should set devlink port
7697          * attributes accordingly so the compat functions are called
7698          * and the original ops are not used.
7699          */
7700         if (ops->ndo_get_phys_port_name) {
7701                 /* Some drivers use the same set of ndos for netdevs
7702                  * that have devlink_port registered and also for
7703                  * those who don't. Make sure that ndo_get_phys_port_name
7704                  * returns -EOPNOTSUPP here in case it is defined.
7705                  * Warn if not.
7706                  */
7707                 char name[IFNAMSIZ];
7708                 int err;
7709
7710                 err = ops->ndo_get_phys_port_name(netdev, name, sizeof(name));
7711                 WARN_ON(err != -EOPNOTSUPP);
7712         }
7713         if (ops->ndo_get_port_parent_id) {
7714                 /* Some drivers use the same set of ndos for netdevs
7715                  * that have devlink_port registered and also for
7716                  * those who don't. Make sure that ndo_get_port_parent_id
7717                  * returns -EOPNOTSUPP here in case it is defined.
7718                  * Warn if not.
7719                  */
7720                 struct netdev_phys_item_id ppid;
7721                 int err;
7722
7723                 err = ops->ndo_get_port_parent_id(netdev, &ppid);
7724                 WARN_ON(err != -EOPNOTSUPP);
7725         }
7726 }
7727
7728 /**
7729  *      devlink_port_type_eth_set - Set port type to Ethernet
7730  *
7731  *      @devlink_port: devlink port
7732  *      @netdev: related netdevice
7733  */
7734 void devlink_port_type_eth_set(struct devlink_port *devlink_port,
7735                                struct net_device *netdev)
7736 {
7737         if (netdev)
7738                 devlink_port_type_netdev_checks(devlink_port, netdev);
7739         else
7740                 dev_warn(devlink_port->devlink->dev,
7741                          "devlink port type for port %d set to Ethernet without a software interface reference, device type not supported by the kernel?\n",
7742                          devlink_port->index);
7743
7744         __devlink_port_type_set(devlink_port, DEVLINK_PORT_TYPE_ETH, netdev);
7745 }
7746 EXPORT_SYMBOL_GPL(devlink_port_type_eth_set);
7747
7748 /**
7749  *      devlink_port_type_ib_set - Set port type to InfiniBand
7750  *
7751  *      @devlink_port: devlink port
7752  *      @ibdev: related IB device
7753  */
7754 void devlink_port_type_ib_set(struct devlink_port *devlink_port,
7755                               struct ib_device *ibdev)
7756 {
7757         __devlink_port_type_set(devlink_port, DEVLINK_PORT_TYPE_IB, ibdev);
7758 }
7759 EXPORT_SYMBOL_GPL(devlink_port_type_ib_set);
7760
7761 /**
7762  *      devlink_port_type_clear - Clear port type
7763  *
7764  *      @devlink_port: devlink port
7765  */
7766 void devlink_port_type_clear(struct devlink_port *devlink_port)
7767 {
7768         __devlink_port_type_set(devlink_port, DEVLINK_PORT_TYPE_NOTSET, NULL);
7769         devlink_port_type_warn_schedule(devlink_port);
7770 }
7771 EXPORT_SYMBOL_GPL(devlink_port_type_clear);
7772
7773 static int __devlink_port_attrs_set(struct devlink_port *devlink_port,
7774                                     enum devlink_port_flavour flavour)
7775 {
7776         struct devlink_port_attrs *attrs = &devlink_port->attrs;
7777
7778         if (WARN_ON(devlink_port->registered))
7779                 return -EEXIST;
7780         devlink_port->attrs_set = true;
7781         attrs->flavour = flavour;
7782         if (attrs->switch_id.id_len) {
7783                 devlink_port->switch_port = true;
7784                 if (WARN_ON(attrs->switch_id.id_len > MAX_PHYS_ITEM_ID_LEN))
7785                         attrs->switch_id.id_len = MAX_PHYS_ITEM_ID_LEN;
7786         } else {
7787                 devlink_port->switch_port = false;
7788         }
7789         return 0;
7790 }
7791
7792 /**
7793  *      devlink_port_attrs_set - Set port attributes
7794  *
7795  *      @devlink_port: devlink port
7796  *      @attrs: devlink port attrs
7797  */
7798 void devlink_port_attrs_set(struct devlink_port *devlink_port,
7799                             struct devlink_port_attrs *attrs)
7800 {
7801         int ret;
7802
7803         devlink_port->attrs = *attrs;
7804         ret = __devlink_port_attrs_set(devlink_port, attrs->flavour);
7805         if (ret)
7806                 return;
7807         WARN_ON(attrs->splittable && attrs->split);
7808 }
7809 EXPORT_SYMBOL_GPL(devlink_port_attrs_set);
7810
7811 /**
7812  *      devlink_port_attrs_pci_pf_set - Set PCI PF port attributes
7813  *
7814  *      @devlink_port: devlink port
7815  *      @controller: associated controller number for the devlink port instance
7816  *      @pf: associated PF for the devlink port instance
7817  *      @external: indicates if the port is for an external controller
7818  */
7819 void devlink_port_attrs_pci_pf_set(struct devlink_port *devlink_port, u32 controller,
7820                                    u16 pf, bool external)
7821 {
7822         struct devlink_port_attrs *attrs = &devlink_port->attrs;
7823         int ret;
7824
7825         ret = __devlink_port_attrs_set(devlink_port,
7826                                        DEVLINK_PORT_FLAVOUR_PCI_PF);
7827         if (ret)
7828                 return;
7829         attrs->pci_pf.controller = controller;
7830         attrs->pci_pf.pf = pf;
7831         attrs->pci_pf.external = external;
7832 }
7833 EXPORT_SYMBOL_GPL(devlink_port_attrs_pci_pf_set);
7834
7835 /**
7836  *      devlink_port_attrs_pci_vf_set - Set PCI VF port attributes
7837  *
7838  *      @devlink_port: devlink port
7839  *      @controller: associated controller number for the devlink port instance
7840  *      @pf: associated PF for the devlink port instance
7841  *      @vf: associated VF of a PF for the devlink port instance
7842  *      @external: indicates if the port is for an external controller
7843  */
7844 void devlink_port_attrs_pci_vf_set(struct devlink_port *devlink_port, u32 controller,
7845                                    u16 pf, u16 vf, bool external)
7846 {
7847         struct devlink_port_attrs *attrs = &devlink_port->attrs;
7848         int ret;
7849
7850         ret = __devlink_port_attrs_set(devlink_port,
7851                                        DEVLINK_PORT_FLAVOUR_PCI_VF);
7852         if (ret)
7853                 return;
7854         attrs->pci_vf.controller = controller;
7855         attrs->pci_vf.pf = pf;
7856         attrs->pci_vf.vf = vf;
7857         attrs->pci_vf.external = external;
7858 }
7859 EXPORT_SYMBOL_GPL(devlink_port_attrs_pci_vf_set);
7860
7861 static int __devlink_port_phys_port_name_get(struct devlink_port *devlink_port,
7862                                              char *name, size_t len)
7863 {
7864         struct devlink_port_attrs *attrs = &devlink_port->attrs;
7865         int n = 0;
7866
7867         if (!devlink_port->attrs_set)
7868                 return -EOPNOTSUPP;
7869
7870         switch (attrs->flavour) {
7871         case DEVLINK_PORT_FLAVOUR_PHYSICAL:
7872         case DEVLINK_PORT_FLAVOUR_VIRTUAL:
7873                 if (!attrs->split)
7874                         n = snprintf(name, len, "p%u", attrs->phys.port_number);
7875                 else
7876                         n = snprintf(name, len, "p%us%u",
7877                                      attrs->phys.port_number,
7878                                      attrs->phys.split_subport_number);
7879                 break;
7880         case DEVLINK_PORT_FLAVOUR_CPU:
7881         case DEVLINK_PORT_FLAVOUR_DSA:
7882                 /* As CPU and DSA ports do not have a netdevice associated
7883                  * case should not ever happen.
7884                  */
7885                 WARN_ON(1);
7886                 return -EINVAL;
7887         case DEVLINK_PORT_FLAVOUR_PCI_PF:
7888                 if (attrs->pci_pf.external) {
7889                         n = snprintf(name, len, "c%u", attrs->pci_pf.controller);
7890                         if (n >= len)
7891                                 return -EINVAL;
7892                         len -= n;
7893                         name += n;
7894                 }
7895                 n = snprintf(name, len, "pf%u", attrs->pci_pf.pf);
7896                 break;
7897         case DEVLINK_PORT_FLAVOUR_PCI_VF:
7898                 if (attrs->pci_vf.external) {
7899                         n = snprintf(name, len, "c%u", attrs->pci_vf.controller);
7900                         if (n >= len)
7901                                 return -EINVAL;
7902                         len -= n;
7903                         name += n;
7904                 }
7905                 n = snprintf(name, len, "pf%uvf%u",
7906                              attrs->pci_vf.pf, attrs->pci_vf.vf);
7907                 break;
7908         }
7909
7910         if (n >= len)
7911                 return -EINVAL;
7912
7913         return 0;
7914 }
7915
7916 int devlink_sb_register(struct devlink *devlink, unsigned int sb_index,
7917                         u32 size, u16 ingress_pools_count,
7918                         u16 egress_pools_count, u16 ingress_tc_count,
7919                         u16 egress_tc_count)
7920 {
7921         struct devlink_sb *devlink_sb;
7922         int err = 0;
7923
7924         mutex_lock(&devlink->lock);
7925         if (devlink_sb_index_exists(devlink, sb_index)) {
7926                 err = -EEXIST;
7927                 goto unlock;
7928         }
7929
7930         devlink_sb = kzalloc(sizeof(*devlink_sb), GFP_KERNEL);
7931         if (!devlink_sb) {
7932                 err = -ENOMEM;
7933                 goto unlock;
7934         }
7935         devlink_sb->index = sb_index;
7936         devlink_sb->size = size;
7937         devlink_sb->ingress_pools_count = ingress_pools_count;
7938         devlink_sb->egress_pools_count = egress_pools_count;
7939         devlink_sb->ingress_tc_count = ingress_tc_count;
7940         devlink_sb->egress_tc_count = egress_tc_count;
7941         list_add_tail(&devlink_sb->list, &devlink->sb_list);
7942 unlock:
7943         mutex_unlock(&devlink->lock);
7944         return err;
7945 }
7946 EXPORT_SYMBOL_GPL(devlink_sb_register);
7947
7948 void devlink_sb_unregister(struct devlink *devlink, unsigned int sb_index)
7949 {
7950         struct devlink_sb *devlink_sb;
7951
7952         mutex_lock(&devlink->lock);
7953         devlink_sb = devlink_sb_get_by_index(devlink, sb_index);
7954         WARN_ON(!devlink_sb);
7955         list_del(&devlink_sb->list);
7956         mutex_unlock(&devlink->lock);
7957         kfree(devlink_sb);
7958 }
7959 EXPORT_SYMBOL_GPL(devlink_sb_unregister);
7960
7961 /**
7962  *      devlink_dpipe_headers_register - register dpipe headers
7963  *
7964  *      @devlink: devlink
7965  *      @dpipe_headers: dpipe header array
7966  *
7967  *      Register the headers supported by hardware.
7968  */
7969 int devlink_dpipe_headers_register(struct devlink *devlink,
7970                                    struct devlink_dpipe_headers *dpipe_headers)
7971 {
7972         mutex_lock(&devlink->lock);
7973         devlink->dpipe_headers = dpipe_headers;
7974         mutex_unlock(&devlink->lock);
7975         return 0;
7976 }
7977 EXPORT_SYMBOL_GPL(devlink_dpipe_headers_register);
7978
7979 /**
7980  *      devlink_dpipe_headers_unregister - unregister dpipe headers
7981  *
7982  *      @devlink: devlink
7983  *
7984  *      Unregister the headers supported by hardware.
7985  */
7986 void devlink_dpipe_headers_unregister(struct devlink *devlink)
7987 {
7988         mutex_lock(&devlink->lock);
7989         devlink->dpipe_headers = NULL;
7990         mutex_unlock(&devlink->lock);
7991 }
7992 EXPORT_SYMBOL_GPL(devlink_dpipe_headers_unregister);
7993
7994 /**
7995  *      devlink_dpipe_table_counter_enabled - check if counter allocation
7996  *                                            required
7997  *      @devlink: devlink
7998  *      @table_name: tables name
7999  *
8000  *      Used by driver to check if counter allocation is required.
8001  *      After counter allocation is turned on the table entries
8002  *      are updated to include counter statistics.
8003  *
8004  *      After that point on the driver must respect the counter
8005  *      state so that each entry added to the table is added
8006  *      with a counter.
8007  */
8008 bool devlink_dpipe_table_counter_enabled(struct devlink *devlink,
8009                                          const char *table_name)
8010 {
8011         struct devlink_dpipe_table *table;
8012         bool enabled;
8013
8014         rcu_read_lock();
8015         table = devlink_dpipe_table_find(&devlink->dpipe_table_list,
8016                                          table_name, devlink);
8017         enabled = false;
8018         if (table)
8019                 enabled = table->counters_enabled;
8020         rcu_read_unlock();
8021         return enabled;
8022 }
8023 EXPORT_SYMBOL_GPL(devlink_dpipe_table_counter_enabled);
8024
8025 /**
8026  *      devlink_dpipe_table_register - register dpipe table
8027  *
8028  *      @devlink: devlink
8029  *      @table_name: table name
8030  *      @table_ops: table ops
8031  *      @priv: priv
8032  *      @counter_control_extern: external control for counters
8033  */
8034 int devlink_dpipe_table_register(struct devlink *devlink,
8035                                  const char *table_name,
8036                                  struct devlink_dpipe_table_ops *table_ops,
8037                                  void *priv, bool counter_control_extern)
8038 {
8039         struct devlink_dpipe_table *table;
8040         int err = 0;
8041
8042         if (WARN_ON(!table_ops->size_get))
8043                 return -EINVAL;
8044
8045         mutex_lock(&devlink->lock);
8046
8047         if (devlink_dpipe_table_find(&devlink->dpipe_table_list, table_name,
8048                                      devlink)) {
8049                 err = -EEXIST;
8050                 goto unlock;
8051         }
8052
8053         table = kzalloc(sizeof(*table), GFP_KERNEL);
8054         if (!table) {
8055                 err = -ENOMEM;
8056                 goto unlock;
8057         }
8058
8059         table->name = table_name;
8060         table->table_ops = table_ops;
8061         table->priv = priv;
8062         table->counter_control_extern = counter_control_extern;
8063
8064         list_add_tail_rcu(&table->list, &devlink->dpipe_table_list);
8065 unlock:
8066         mutex_unlock(&devlink->lock);
8067         return err;
8068 }
8069 EXPORT_SYMBOL_GPL(devlink_dpipe_table_register);
8070
8071 /**
8072  *      devlink_dpipe_table_unregister - unregister dpipe table
8073  *
8074  *      @devlink: devlink
8075  *      @table_name: table name
8076  */
8077 void devlink_dpipe_table_unregister(struct devlink *devlink,
8078                                     const char *table_name)
8079 {
8080         struct devlink_dpipe_table *table;
8081
8082         mutex_lock(&devlink->lock);
8083         table = devlink_dpipe_table_find(&devlink->dpipe_table_list,
8084                                          table_name, devlink);
8085         if (!table)
8086                 goto unlock;
8087         list_del_rcu(&table->list);
8088         mutex_unlock(&devlink->lock);
8089         kfree_rcu(table, rcu);
8090         return;
8091 unlock:
8092         mutex_unlock(&devlink->lock);
8093 }
8094 EXPORT_SYMBOL_GPL(devlink_dpipe_table_unregister);
8095
8096 /**
8097  *      devlink_resource_register - devlink resource register
8098  *
8099  *      @devlink: devlink
8100  *      @resource_name: resource's name
8101  *      @resource_size: resource's size
8102  *      @resource_id: resource's id
8103  *      @parent_resource_id: resource's parent id
8104  *      @size_params: size parameters
8105  */
8106 int devlink_resource_register(struct devlink *devlink,
8107                               const char *resource_name,
8108                               u64 resource_size,
8109                               u64 resource_id,
8110                               u64 parent_resource_id,
8111                               const struct devlink_resource_size_params *size_params)
8112 {
8113         struct devlink_resource *resource;
8114         struct list_head *resource_list;
8115         bool top_hierarchy;
8116         int err = 0;
8117
8118         top_hierarchy = parent_resource_id == DEVLINK_RESOURCE_ID_PARENT_TOP;
8119
8120         mutex_lock(&devlink->lock);
8121         resource = devlink_resource_find(devlink, NULL, resource_id);
8122         if (resource) {
8123                 err = -EINVAL;
8124                 goto out;
8125         }
8126
8127         resource = kzalloc(sizeof(*resource), GFP_KERNEL);
8128         if (!resource) {
8129                 err = -ENOMEM;
8130                 goto out;
8131         }
8132
8133         if (top_hierarchy) {
8134                 resource_list = &devlink->resource_list;
8135         } else {
8136                 struct devlink_resource *parent_resource;
8137
8138                 parent_resource = devlink_resource_find(devlink, NULL,
8139                                                         parent_resource_id);
8140                 if (parent_resource) {
8141                         resource_list = &parent_resource->resource_list;
8142                         resource->parent = parent_resource;
8143                 } else {
8144                         kfree(resource);
8145                         err = -EINVAL;
8146                         goto out;
8147                 }
8148         }
8149
8150         resource->name = resource_name;
8151         resource->size = resource_size;
8152         resource->size_new = resource_size;
8153         resource->id = resource_id;
8154         resource->size_valid = true;
8155         memcpy(&resource->size_params, size_params,
8156                sizeof(resource->size_params));
8157         INIT_LIST_HEAD(&resource->resource_list);
8158         list_add_tail(&resource->list, resource_list);
8159 out:
8160         mutex_unlock(&devlink->lock);
8161         return err;
8162 }
8163 EXPORT_SYMBOL_GPL(devlink_resource_register);
8164
8165 /**
8166  *      devlink_resources_unregister - free all resources
8167  *
8168  *      @devlink: devlink
8169  *      @resource: resource
8170  */
8171 void devlink_resources_unregister(struct devlink *devlink,
8172                                   struct devlink_resource *resource)
8173 {
8174         struct devlink_resource *tmp, *child_resource;
8175         struct list_head *resource_list;
8176
8177         if (resource)
8178                 resource_list = &resource->resource_list;
8179         else
8180                 resource_list = &devlink->resource_list;
8181
8182         if (!resource)
8183                 mutex_lock(&devlink->lock);
8184
8185         list_for_each_entry_safe(child_resource, tmp, resource_list, list) {
8186                 devlink_resources_unregister(devlink, child_resource);
8187                 list_del(&child_resource->list);
8188                 kfree(child_resource);
8189         }
8190
8191         if (!resource)
8192                 mutex_unlock(&devlink->lock);
8193 }
8194 EXPORT_SYMBOL_GPL(devlink_resources_unregister);
8195
8196 /**
8197  *      devlink_resource_size_get - get and update size
8198  *
8199  *      @devlink: devlink
8200  *      @resource_id: the requested resource id
8201  *      @p_resource_size: ptr to update
8202  */
8203 int devlink_resource_size_get(struct devlink *devlink,
8204                               u64 resource_id,
8205                               u64 *p_resource_size)
8206 {
8207         struct devlink_resource *resource;
8208         int err = 0;
8209
8210         mutex_lock(&devlink->lock);
8211         resource = devlink_resource_find(devlink, NULL, resource_id);
8212         if (!resource) {
8213                 err = -EINVAL;
8214                 goto out;
8215         }
8216         *p_resource_size = resource->size_new;
8217         resource->size = resource->size_new;
8218 out:
8219         mutex_unlock(&devlink->lock);
8220         return err;
8221 }
8222 EXPORT_SYMBOL_GPL(devlink_resource_size_get);
8223
8224 /**
8225  *      devlink_dpipe_table_resource_set - set the resource id
8226  *
8227  *      @devlink: devlink
8228  *      @table_name: table name
8229  *      @resource_id: resource id
8230  *      @resource_units: number of resource's units consumed per table's entry
8231  */
8232 int devlink_dpipe_table_resource_set(struct devlink *devlink,
8233                                      const char *table_name, u64 resource_id,
8234                                      u64 resource_units)
8235 {
8236         struct devlink_dpipe_table *table;
8237         int err = 0;
8238
8239         mutex_lock(&devlink->lock);
8240         table = devlink_dpipe_table_find(&devlink->dpipe_table_list,
8241                                          table_name, devlink);
8242         if (!table) {
8243                 err = -EINVAL;
8244                 goto out;
8245         }
8246         table->resource_id = resource_id;
8247         table->resource_units = resource_units;
8248         table->resource_valid = true;
8249 out:
8250         mutex_unlock(&devlink->lock);
8251         return err;
8252 }
8253 EXPORT_SYMBOL_GPL(devlink_dpipe_table_resource_set);
8254
8255 /**
8256  *      devlink_resource_occ_get_register - register occupancy getter
8257  *
8258  *      @devlink: devlink
8259  *      @resource_id: resource id
8260  *      @occ_get: occupancy getter callback
8261  *      @occ_get_priv: occupancy getter callback priv
8262  */
8263 void devlink_resource_occ_get_register(struct devlink *devlink,
8264                                        u64 resource_id,
8265                                        devlink_resource_occ_get_t *occ_get,
8266                                        void *occ_get_priv)
8267 {
8268         struct devlink_resource *resource;
8269
8270         mutex_lock(&devlink->lock);
8271         resource = devlink_resource_find(devlink, NULL, resource_id);
8272         if (WARN_ON(!resource))
8273                 goto out;
8274         WARN_ON(resource->occ_get);
8275
8276         resource->occ_get = occ_get;
8277         resource->occ_get_priv = occ_get_priv;
8278 out:
8279         mutex_unlock(&devlink->lock);
8280 }
8281 EXPORT_SYMBOL_GPL(devlink_resource_occ_get_register);
8282
8283 /**
8284  *      devlink_resource_occ_get_unregister - unregister occupancy getter
8285  *
8286  *      @devlink: devlink
8287  *      @resource_id: resource id
8288  */
8289 void devlink_resource_occ_get_unregister(struct devlink *devlink,
8290                                          u64 resource_id)
8291 {
8292         struct devlink_resource *resource;
8293
8294         mutex_lock(&devlink->lock);
8295         resource = devlink_resource_find(devlink, NULL, resource_id);
8296         if (WARN_ON(!resource))
8297                 goto out;
8298         WARN_ON(!resource->occ_get);
8299
8300         resource->occ_get = NULL;
8301         resource->occ_get_priv = NULL;
8302 out:
8303         mutex_unlock(&devlink->lock);
8304 }
8305 EXPORT_SYMBOL_GPL(devlink_resource_occ_get_unregister);
8306
8307 static int devlink_param_verify(const struct devlink_param *param)
8308 {
8309         if (!param || !param->name || !param->supported_cmodes)
8310                 return -EINVAL;
8311         if (param->generic)
8312                 return devlink_param_generic_verify(param);
8313         else
8314                 return devlink_param_driver_verify(param);
8315 }
8316
8317 static int __devlink_params_register(struct devlink *devlink,
8318                                      unsigned int port_index,
8319                                      struct list_head *param_list,
8320                                      const struct devlink_param *params,
8321                                      size_t params_count,
8322                                      enum devlink_command reg_cmd,
8323                                      enum devlink_command unreg_cmd)
8324 {
8325         const struct devlink_param *param = params;
8326         int i;
8327         int err;
8328
8329         mutex_lock(&devlink->lock);
8330         for (i = 0; i < params_count; i++, param++) {
8331                 err = devlink_param_verify(param);
8332                 if (err)
8333                         goto rollback;
8334
8335                 err = devlink_param_register_one(devlink, port_index,
8336                                                  param_list, param, reg_cmd);
8337                 if (err)
8338                         goto rollback;
8339         }
8340
8341         mutex_unlock(&devlink->lock);
8342         return 0;
8343
8344 rollback:
8345         if (!i)
8346                 goto unlock;
8347         for (param--; i > 0; i--, param--)
8348                 devlink_param_unregister_one(devlink, port_index, param_list,
8349                                              param, unreg_cmd);
8350 unlock:
8351         mutex_unlock(&devlink->lock);
8352         return err;
8353 }
8354
8355 static void __devlink_params_unregister(struct devlink *devlink,
8356                                         unsigned int port_index,
8357                                         struct list_head *param_list,
8358                                         const struct devlink_param *params,
8359                                         size_t params_count,
8360                                         enum devlink_command cmd)
8361 {
8362         const struct devlink_param *param = params;
8363         int i;
8364
8365         mutex_lock(&devlink->lock);
8366         for (i = 0; i < params_count; i++, param++)
8367                 devlink_param_unregister_one(devlink, 0, param_list, param,
8368                                              cmd);
8369         mutex_unlock(&devlink->lock);
8370 }
8371
8372 /**
8373  *      devlink_params_register - register configuration parameters
8374  *
8375  *      @devlink: devlink
8376  *      @params: configuration parameters array
8377  *      @params_count: number of parameters provided
8378  *
8379  *      Register the configuration parameters supported by the driver.
8380  */
8381 int devlink_params_register(struct devlink *devlink,
8382                             const struct devlink_param *params,
8383                             size_t params_count)
8384 {
8385         return __devlink_params_register(devlink, 0, &devlink->param_list,
8386                                          params, params_count,
8387                                          DEVLINK_CMD_PARAM_NEW,
8388                                          DEVLINK_CMD_PARAM_DEL);
8389 }
8390 EXPORT_SYMBOL_GPL(devlink_params_register);
8391
8392 /**
8393  *      devlink_params_unregister - unregister configuration parameters
8394  *      @devlink: devlink
8395  *      @params: configuration parameters to unregister
8396  *      @params_count: number of parameters provided
8397  */
8398 void devlink_params_unregister(struct devlink *devlink,
8399                                const struct devlink_param *params,
8400                                size_t params_count)
8401 {
8402         return __devlink_params_unregister(devlink, 0, &devlink->param_list,
8403                                            params, params_count,
8404                                            DEVLINK_CMD_PARAM_DEL);
8405 }
8406 EXPORT_SYMBOL_GPL(devlink_params_unregister);
8407
8408 /**
8409  *      devlink_params_publish - publish configuration parameters
8410  *
8411  *      @devlink: devlink
8412  *
8413  *      Publish previously registered configuration parameters.
8414  */
8415 void devlink_params_publish(struct devlink *devlink)
8416 {
8417         struct devlink_param_item *param_item;
8418
8419         list_for_each_entry(param_item, &devlink->param_list, list) {
8420                 if (param_item->published)
8421                         continue;
8422                 param_item->published = true;
8423                 devlink_param_notify(devlink, 0, param_item,
8424                                      DEVLINK_CMD_PARAM_NEW);
8425         }
8426 }
8427 EXPORT_SYMBOL_GPL(devlink_params_publish);
8428
8429 /**
8430  *      devlink_params_unpublish - unpublish configuration parameters
8431  *
8432  *      @devlink: devlink
8433  *
8434  *      Unpublish previously registered configuration parameters.
8435  */
8436 void devlink_params_unpublish(struct devlink *devlink)
8437 {
8438         struct devlink_param_item *param_item;
8439
8440         list_for_each_entry(param_item, &devlink->param_list, list) {
8441                 if (!param_item->published)
8442                         continue;
8443                 param_item->published = false;
8444                 devlink_param_notify(devlink, 0, param_item,
8445                                      DEVLINK_CMD_PARAM_DEL);
8446         }
8447 }
8448 EXPORT_SYMBOL_GPL(devlink_params_unpublish);
8449
8450 /**
8451  *      devlink_port_params_register - register port configuration parameters
8452  *
8453  *      @devlink_port: devlink port
8454  *      @params: configuration parameters array
8455  *      @params_count: number of parameters provided
8456  *
8457  *      Register the configuration parameters supported by the port.
8458  */
8459 int devlink_port_params_register(struct devlink_port *devlink_port,
8460                                  const struct devlink_param *params,
8461                                  size_t params_count)
8462 {
8463         return __devlink_params_register(devlink_port->devlink,
8464                                          devlink_port->index,
8465                                          &devlink_port->param_list, params,
8466                                          params_count,
8467                                          DEVLINK_CMD_PORT_PARAM_NEW,
8468                                          DEVLINK_CMD_PORT_PARAM_DEL);
8469 }
8470 EXPORT_SYMBOL_GPL(devlink_port_params_register);
8471
8472 /**
8473  *      devlink_port_params_unregister - unregister port configuration
8474  *      parameters
8475  *
8476  *      @devlink_port: devlink port
8477  *      @params: configuration parameters array
8478  *      @params_count: number of parameters provided
8479  */
8480 void devlink_port_params_unregister(struct devlink_port *devlink_port,
8481                                     const struct devlink_param *params,
8482                                     size_t params_count)
8483 {
8484         return __devlink_params_unregister(devlink_port->devlink,
8485                                            devlink_port->index,
8486                                            &devlink_port->param_list,
8487                                            params, params_count,
8488                                            DEVLINK_CMD_PORT_PARAM_DEL);
8489 }
8490 EXPORT_SYMBOL_GPL(devlink_port_params_unregister);
8491
8492 static int
8493 __devlink_param_driverinit_value_get(struct list_head *param_list, u32 param_id,
8494                                      union devlink_param_value *init_val)
8495 {
8496         struct devlink_param_item *param_item;
8497
8498         param_item = devlink_param_find_by_id(param_list, param_id);
8499         if (!param_item)
8500                 return -EINVAL;
8501
8502         if (!param_item->driverinit_value_valid ||
8503             !devlink_param_cmode_is_supported(param_item->param,
8504                                               DEVLINK_PARAM_CMODE_DRIVERINIT))
8505                 return -EOPNOTSUPP;
8506
8507         if (param_item->param->type == DEVLINK_PARAM_TYPE_STRING)
8508                 strcpy(init_val->vstr, param_item->driverinit_value.vstr);
8509         else
8510                 *init_val = param_item->driverinit_value;
8511
8512         return 0;
8513 }
8514
8515 static int
8516 __devlink_param_driverinit_value_set(struct devlink *devlink,
8517                                      unsigned int port_index,
8518                                      struct list_head *param_list, u32 param_id,
8519                                      union devlink_param_value init_val,
8520                                      enum devlink_command cmd)
8521 {
8522         struct devlink_param_item *param_item;
8523
8524         param_item = devlink_param_find_by_id(param_list, param_id);
8525         if (!param_item)
8526                 return -EINVAL;
8527
8528         if (!devlink_param_cmode_is_supported(param_item->param,
8529                                               DEVLINK_PARAM_CMODE_DRIVERINIT))
8530                 return -EOPNOTSUPP;
8531
8532         if (param_item->param->type == DEVLINK_PARAM_TYPE_STRING)
8533                 strcpy(param_item->driverinit_value.vstr, init_val.vstr);
8534         else
8535                 param_item->driverinit_value = init_val;
8536         param_item->driverinit_value_valid = true;
8537
8538         devlink_param_notify(devlink, port_index, param_item, cmd);
8539         return 0;
8540 }
8541
8542 /**
8543  *      devlink_param_driverinit_value_get - get configuration parameter
8544  *                                           value for driver initializing
8545  *
8546  *      @devlink: devlink
8547  *      @param_id: parameter ID
8548  *      @init_val: value of parameter in driverinit configuration mode
8549  *
8550  *      This function should be used by the driver to get driverinit
8551  *      configuration for initialization after reload command.
8552  */
8553 int devlink_param_driverinit_value_get(struct devlink *devlink, u32 param_id,
8554                                        union devlink_param_value *init_val)
8555 {
8556         if (!devlink_reload_supported(devlink))
8557                 return -EOPNOTSUPP;
8558
8559         return __devlink_param_driverinit_value_get(&devlink->param_list,
8560                                                     param_id, init_val);
8561 }
8562 EXPORT_SYMBOL_GPL(devlink_param_driverinit_value_get);
8563
8564 /**
8565  *      devlink_param_driverinit_value_set - set value of configuration
8566  *                                           parameter for driverinit
8567  *                                           configuration mode
8568  *
8569  *      @devlink: devlink
8570  *      @param_id: parameter ID
8571  *      @init_val: value of parameter to set for driverinit configuration mode
8572  *
8573  *      This function should be used by the driver to set driverinit
8574  *      configuration mode default value.
8575  */
8576 int devlink_param_driverinit_value_set(struct devlink *devlink, u32 param_id,
8577                                        union devlink_param_value init_val)
8578 {
8579         return __devlink_param_driverinit_value_set(devlink, 0,
8580                                                     &devlink->param_list,
8581                                                     param_id, init_val,
8582                                                     DEVLINK_CMD_PARAM_NEW);
8583 }
8584 EXPORT_SYMBOL_GPL(devlink_param_driverinit_value_set);
8585
8586 /**
8587  *      devlink_port_param_driverinit_value_get - get configuration parameter
8588  *                                              value for driver initializing
8589  *
8590  *      @devlink_port: devlink_port
8591  *      @param_id: parameter ID
8592  *      @init_val: value of parameter in driverinit configuration mode
8593  *
8594  *      This function should be used by the driver to get driverinit
8595  *      configuration for initialization after reload command.
8596  */
8597 int devlink_port_param_driverinit_value_get(struct devlink_port *devlink_port,
8598                                             u32 param_id,
8599                                             union devlink_param_value *init_val)
8600 {
8601         struct devlink *devlink = devlink_port->devlink;
8602
8603         if (!devlink_reload_supported(devlink))
8604                 return -EOPNOTSUPP;
8605
8606         return __devlink_param_driverinit_value_get(&devlink_port->param_list,
8607                                                     param_id, init_val);
8608 }
8609 EXPORT_SYMBOL_GPL(devlink_port_param_driverinit_value_get);
8610
8611 /**
8612  *     devlink_port_param_driverinit_value_set - set value of configuration
8613  *                                               parameter for driverinit
8614  *                                               configuration mode
8615  *
8616  *     @devlink_port: devlink_port
8617  *     @param_id: parameter ID
8618  *     @init_val: value of parameter to set for driverinit configuration mode
8619  *
8620  *     This function should be used by the driver to set driverinit
8621  *     configuration mode default value.
8622  */
8623 int devlink_port_param_driverinit_value_set(struct devlink_port *devlink_port,
8624                                             u32 param_id,
8625                                             union devlink_param_value init_val)
8626 {
8627         return __devlink_param_driverinit_value_set(devlink_port->devlink,
8628                                                     devlink_port->index,
8629                                                     &devlink_port->param_list,
8630                                                     param_id, init_val,
8631                                                     DEVLINK_CMD_PORT_PARAM_NEW);
8632 }
8633 EXPORT_SYMBOL_GPL(devlink_port_param_driverinit_value_set);
8634
8635 /**
8636  *      devlink_param_value_changed - notify devlink on a parameter's value
8637  *                                    change. Should be called by the driver
8638  *                                    right after the change.
8639  *
8640  *      @devlink: devlink
8641  *      @param_id: parameter ID
8642  *
8643  *      This function should be used by the driver to notify devlink on value
8644  *      change, excluding driverinit configuration mode.
8645  *      For driverinit configuration mode driver should use the function
8646  */
8647 void devlink_param_value_changed(struct devlink *devlink, u32 param_id)
8648 {
8649         struct devlink_param_item *param_item;
8650
8651         param_item = devlink_param_find_by_id(&devlink->param_list, param_id);
8652         WARN_ON(!param_item);
8653
8654         devlink_param_notify(devlink, 0, param_item, DEVLINK_CMD_PARAM_NEW);
8655 }
8656 EXPORT_SYMBOL_GPL(devlink_param_value_changed);
8657
8658 /**
8659  *     devlink_port_param_value_changed - notify devlink on a parameter's value
8660  *                                      change. Should be called by the driver
8661  *                                      right after the change.
8662  *
8663  *     @devlink_port: devlink_port
8664  *     @param_id: parameter ID
8665  *
8666  *     This function should be used by the driver to notify devlink on value
8667  *     change, excluding driverinit configuration mode.
8668  *     For driverinit configuration mode driver should use the function
8669  *     devlink_port_param_driverinit_value_set() instead.
8670  */
8671 void devlink_port_param_value_changed(struct devlink_port *devlink_port,
8672                                       u32 param_id)
8673 {
8674         struct devlink_param_item *param_item;
8675
8676         param_item = devlink_param_find_by_id(&devlink_port->param_list,
8677                                               param_id);
8678         WARN_ON(!param_item);
8679
8680         devlink_param_notify(devlink_port->devlink, devlink_port->index,
8681                              param_item, DEVLINK_CMD_PORT_PARAM_NEW);
8682 }
8683 EXPORT_SYMBOL_GPL(devlink_port_param_value_changed);
8684
8685 /**
8686  *      devlink_param_value_str_fill - Safely fill-up the string preventing
8687  *                                     from overflow of the preallocated buffer
8688  *
8689  *      @dst_val: destination devlink_param_value
8690  *      @src: source buffer
8691  */
8692 void devlink_param_value_str_fill(union devlink_param_value *dst_val,
8693                                   const char *src)
8694 {
8695         size_t len;
8696
8697         len = strlcpy(dst_val->vstr, src, __DEVLINK_PARAM_MAX_STRING_VALUE);
8698         WARN_ON(len >= __DEVLINK_PARAM_MAX_STRING_VALUE);
8699 }
8700 EXPORT_SYMBOL_GPL(devlink_param_value_str_fill);
8701
8702 /**
8703  *      devlink_region_create - create a new address region
8704  *
8705  *      @devlink: devlink
8706  *      @ops: region operations and name
8707  *      @region_max_snapshots: Maximum supported number of snapshots for region
8708  *      @region_size: size of region
8709  */
8710 struct devlink_region *
8711 devlink_region_create(struct devlink *devlink,
8712                       const struct devlink_region_ops *ops,
8713                       u32 region_max_snapshots, u64 region_size)
8714 {
8715         struct devlink_region *region;
8716         int err = 0;
8717
8718         if (WARN_ON(!ops) || WARN_ON(!ops->destructor))
8719                 return ERR_PTR(-EINVAL);
8720
8721         mutex_lock(&devlink->lock);
8722
8723         if (devlink_region_get_by_name(devlink, ops->name)) {
8724                 err = -EEXIST;
8725                 goto unlock;
8726         }
8727
8728         region = kzalloc(sizeof(*region), GFP_KERNEL);
8729         if (!region) {
8730                 err = -ENOMEM;
8731                 goto unlock;
8732         }
8733
8734         region->devlink = devlink;
8735         region->max_snapshots = region_max_snapshots;
8736         region->ops = ops;
8737         region->size = region_size;
8738         INIT_LIST_HEAD(&region->snapshot_list);
8739         list_add_tail(&region->list, &devlink->region_list);
8740         devlink_nl_region_notify(region, NULL, DEVLINK_CMD_REGION_NEW);
8741
8742         mutex_unlock(&devlink->lock);
8743         return region;
8744
8745 unlock:
8746         mutex_unlock(&devlink->lock);
8747         return ERR_PTR(err);
8748 }
8749 EXPORT_SYMBOL_GPL(devlink_region_create);
8750
8751 /**
8752  *      devlink_region_destroy - destroy address region
8753  *
8754  *      @region: devlink region to destroy
8755  */
8756 void devlink_region_destroy(struct devlink_region *region)
8757 {
8758         struct devlink *devlink = region->devlink;
8759         struct devlink_snapshot *snapshot, *ts;
8760
8761         mutex_lock(&devlink->lock);
8762
8763         /* Free all snapshots of region */
8764         list_for_each_entry_safe(snapshot, ts, &region->snapshot_list, list)
8765                 devlink_region_snapshot_del(region, snapshot);
8766
8767         list_del(&region->list);
8768
8769         devlink_nl_region_notify(region, NULL, DEVLINK_CMD_REGION_DEL);
8770         mutex_unlock(&devlink->lock);
8771         kfree(region);
8772 }
8773 EXPORT_SYMBOL_GPL(devlink_region_destroy);
8774
8775 /**
8776  *      devlink_region_snapshot_id_get - get snapshot ID
8777  *
8778  *      This callback should be called when adding a new snapshot,
8779  *      Driver should use the same id for multiple snapshots taken
8780  *      on multiple regions at the same time/by the same trigger.
8781  *
8782  *      The caller of this function must use devlink_region_snapshot_id_put
8783  *      when finished creating regions using this id.
8784  *
8785  *      Returns zero on success, or a negative error code on failure.
8786  *
8787  *      @devlink: devlink
8788  *      @id: storage to return id
8789  */
8790 int devlink_region_snapshot_id_get(struct devlink *devlink, u32 *id)
8791 {
8792         int err;
8793
8794         mutex_lock(&devlink->lock);
8795         err = __devlink_region_snapshot_id_get(devlink, id);
8796         mutex_unlock(&devlink->lock);
8797
8798         return err;
8799 }
8800 EXPORT_SYMBOL_GPL(devlink_region_snapshot_id_get);
8801
8802 /**
8803  *      devlink_region_snapshot_id_put - put snapshot ID reference
8804  *
8805  *      This should be called by a driver after finishing creating snapshots
8806  *      with an id. Doing so ensures that the ID can later be released in the
8807  *      event that all snapshots using it have been destroyed.
8808  *
8809  *      @devlink: devlink
8810  *      @id: id to release reference on
8811  */
8812 void devlink_region_snapshot_id_put(struct devlink *devlink, u32 id)
8813 {
8814         mutex_lock(&devlink->lock);
8815         __devlink_snapshot_id_decrement(devlink, id);
8816         mutex_unlock(&devlink->lock);
8817 }
8818 EXPORT_SYMBOL_GPL(devlink_region_snapshot_id_put);
8819
8820 /**
8821  *      devlink_region_snapshot_create - create a new snapshot
8822  *      This will add a new snapshot of a region. The snapshot
8823  *      will be stored on the region struct and can be accessed
8824  *      from devlink. This is useful for future analyses of snapshots.
8825  *      Multiple snapshots can be created on a region.
8826  *      The @snapshot_id should be obtained using the getter function.
8827  *
8828  *      @region: devlink region of the snapshot
8829  *      @data: snapshot data
8830  *      @snapshot_id: snapshot id to be created
8831  */
8832 int devlink_region_snapshot_create(struct devlink_region *region,
8833                                    u8 *data, u32 snapshot_id)
8834 {
8835         struct devlink *devlink = region->devlink;
8836         int err;
8837
8838         mutex_lock(&devlink->lock);
8839         err = __devlink_region_snapshot_create(region, data, snapshot_id);
8840         mutex_unlock(&devlink->lock);
8841
8842         return err;
8843 }
8844 EXPORT_SYMBOL_GPL(devlink_region_snapshot_create);
8845
8846 #define DEVLINK_TRAP(_id, _type)                                              \
8847         {                                                                     \
8848                 .type = DEVLINK_TRAP_TYPE_##_type,                            \
8849                 .id = DEVLINK_TRAP_GENERIC_ID_##_id,                          \
8850                 .name = DEVLINK_TRAP_GENERIC_NAME_##_id,                      \
8851         }
8852
8853 static const struct devlink_trap devlink_trap_generic[] = {
8854         DEVLINK_TRAP(SMAC_MC, DROP),
8855         DEVLINK_TRAP(VLAN_TAG_MISMATCH, DROP),
8856         DEVLINK_TRAP(INGRESS_VLAN_FILTER, DROP),
8857         DEVLINK_TRAP(INGRESS_STP_FILTER, DROP),
8858         DEVLINK_TRAP(EMPTY_TX_LIST, DROP),
8859         DEVLINK_TRAP(PORT_LOOPBACK_FILTER, DROP),
8860         DEVLINK_TRAP(BLACKHOLE_ROUTE, DROP),
8861         DEVLINK_TRAP(TTL_ERROR, EXCEPTION),
8862         DEVLINK_TRAP(TAIL_DROP, DROP),
8863         DEVLINK_TRAP(NON_IP_PACKET, DROP),
8864         DEVLINK_TRAP(UC_DIP_MC_DMAC, DROP),
8865         DEVLINK_TRAP(DIP_LB, DROP),
8866         DEVLINK_TRAP(SIP_MC, DROP),
8867         DEVLINK_TRAP(SIP_LB, DROP),
8868         DEVLINK_TRAP(CORRUPTED_IP_HDR, DROP),
8869         DEVLINK_TRAP(IPV4_SIP_BC, DROP),
8870         DEVLINK_TRAP(IPV6_MC_DIP_RESERVED_SCOPE, DROP),
8871         DEVLINK_TRAP(IPV6_MC_DIP_INTERFACE_LOCAL_SCOPE, DROP),
8872         DEVLINK_TRAP(MTU_ERROR, EXCEPTION),
8873         DEVLINK_TRAP(UNRESOLVED_NEIGH, EXCEPTION),
8874         DEVLINK_TRAP(RPF, EXCEPTION),
8875         DEVLINK_TRAP(REJECT_ROUTE, EXCEPTION),
8876         DEVLINK_TRAP(IPV4_LPM_UNICAST_MISS, EXCEPTION),
8877         DEVLINK_TRAP(IPV6_LPM_UNICAST_MISS, EXCEPTION),
8878         DEVLINK_TRAP(NON_ROUTABLE, DROP),
8879         DEVLINK_TRAP(DECAP_ERROR, EXCEPTION),
8880         DEVLINK_TRAP(OVERLAY_SMAC_MC, DROP),
8881         DEVLINK_TRAP(INGRESS_FLOW_ACTION_DROP, DROP),
8882         DEVLINK_TRAP(EGRESS_FLOW_ACTION_DROP, DROP),
8883         DEVLINK_TRAP(STP, CONTROL),
8884         DEVLINK_TRAP(LACP, CONTROL),
8885         DEVLINK_TRAP(LLDP, CONTROL),
8886         DEVLINK_TRAP(IGMP_QUERY, CONTROL),
8887         DEVLINK_TRAP(IGMP_V1_REPORT, CONTROL),
8888         DEVLINK_TRAP(IGMP_V2_REPORT, CONTROL),
8889         DEVLINK_TRAP(IGMP_V3_REPORT, CONTROL),
8890         DEVLINK_TRAP(IGMP_V2_LEAVE, CONTROL),
8891         DEVLINK_TRAP(MLD_QUERY, CONTROL),
8892         DEVLINK_TRAP(MLD_V1_REPORT, CONTROL),
8893         DEVLINK_TRAP(MLD_V2_REPORT, CONTROL),
8894         DEVLINK_TRAP(MLD_V1_DONE, CONTROL),
8895         DEVLINK_TRAP(IPV4_DHCP, CONTROL),
8896         DEVLINK_TRAP(IPV6_DHCP, CONTROL),
8897         DEVLINK_TRAP(ARP_REQUEST, CONTROL),
8898         DEVLINK_TRAP(ARP_RESPONSE, CONTROL),
8899         DEVLINK_TRAP(ARP_OVERLAY, CONTROL),
8900         DEVLINK_TRAP(IPV6_NEIGH_SOLICIT, CONTROL),
8901         DEVLINK_TRAP(IPV6_NEIGH_ADVERT, CONTROL),
8902         DEVLINK_TRAP(IPV4_BFD, CONTROL),
8903         DEVLINK_TRAP(IPV6_BFD, CONTROL),
8904         DEVLINK_TRAP(IPV4_OSPF, CONTROL),
8905         DEVLINK_TRAP(IPV6_OSPF, CONTROL),
8906         DEVLINK_TRAP(IPV4_BGP, CONTROL),
8907         DEVLINK_TRAP(IPV6_BGP, CONTROL),
8908         DEVLINK_TRAP(IPV4_VRRP, CONTROL),
8909         DEVLINK_TRAP(IPV6_VRRP, CONTROL),
8910         DEVLINK_TRAP(IPV4_PIM, CONTROL),
8911         DEVLINK_TRAP(IPV6_PIM, CONTROL),
8912         DEVLINK_TRAP(UC_LB, CONTROL),
8913         DEVLINK_TRAP(LOCAL_ROUTE, CONTROL),
8914         DEVLINK_TRAP(EXTERNAL_ROUTE, CONTROL),
8915         DEVLINK_TRAP(IPV6_UC_DIP_LINK_LOCAL_SCOPE, CONTROL),
8916         DEVLINK_TRAP(IPV6_DIP_ALL_NODES, CONTROL),
8917         DEVLINK_TRAP(IPV6_DIP_ALL_ROUTERS, CONTROL),
8918         DEVLINK_TRAP(IPV6_ROUTER_SOLICIT, CONTROL),
8919         DEVLINK_TRAP(IPV6_ROUTER_ADVERT, CONTROL),
8920         DEVLINK_TRAP(IPV6_REDIRECT, CONTROL),
8921         DEVLINK_TRAP(IPV4_ROUTER_ALERT, CONTROL),
8922         DEVLINK_TRAP(IPV6_ROUTER_ALERT, CONTROL),
8923         DEVLINK_TRAP(PTP_EVENT, CONTROL),
8924         DEVLINK_TRAP(PTP_GENERAL, CONTROL),
8925         DEVLINK_TRAP(FLOW_ACTION_SAMPLE, CONTROL),
8926         DEVLINK_TRAP(FLOW_ACTION_TRAP, CONTROL),
8927         DEVLINK_TRAP(EARLY_DROP, DROP),
8928 };
8929
8930 #define DEVLINK_TRAP_GROUP(_id)                                               \
8931         {                                                                     \
8932                 .id = DEVLINK_TRAP_GROUP_GENERIC_ID_##_id,                    \
8933                 .name = DEVLINK_TRAP_GROUP_GENERIC_NAME_##_id,                \
8934         }
8935
8936 static const struct devlink_trap_group devlink_trap_group_generic[] = {
8937         DEVLINK_TRAP_GROUP(L2_DROPS),
8938         DEVLINK_TRAP_GROUP(L3_DROPS),
8939         DEVLINK_TRAP_GROUP(L3_EXCEPTIONS),
8940         DEVLINK_TRAP_GROUP(BUFFER_DROPS),
8941         DEVLINK_TRAP_GROUP(TUNNEL_DROPS),
8942         DEVLINK_TRAP_GROUP(ACL_DROPS),
8943         DEVLINK_TRAP_GROUP(STP),
8944         DEVLINK_TRAP_GROUP(LACP),
8945         DEVLINK_TRAP_GROUP(LLDP),
8946         DEVLINK_TRAP_GROUP(MC_SNOOPING),
8947         DEVLINK_TRAP_GROUP(DHCP),
8948         DEVLINK_TRAP_GROUP(NEIGH_DISCOVERY),
8949         DEVLINK_TRAP_GROUP(BFD),
8950         DEVLINK_TRAP_GROUP(OSPF),
8951         DEVLINK_TRAP_GROUP(BGP),
8952         DEVLINK_TRAP_GROUP(VRRP),
8953         DEVLINK_TRAP_GROUP(PIM),
8954         DEVLINK_TRAP_GROUP(UC_LB),
8955         DEVLINK_TRAP_GROUP(LOCAL_DELIVERY),
8956         DEVLINK_TRAP_GROUP(EXTERNAL_DELIVERY),
8957         DEVLINK_TRAP_GROUP(IPV6),
8958         DEVLINK_TRAP_GROUP(PTP_EVENT),
8959         DEVLINK_TRAP_GROUP(PTP_GENERAL),
8960         DEVLINK_TRAP_GROUP(ACL_SAMPLE),
8961         DEVLINK_TRAP_GROUP(ACL_TRAP),
8962 };
8963
8964 static int devlink_trap_generic_verify(const struct devlink_trap *trap)
8965 {
8966         if (trap->id > DEVLINK_TRAP_GENERIC_ID_MAX)
8967                 return -EINVAL;
8968
8969         if (strcmp(trap->name, devlink_trap_generic[trap->id].name))
8970                 return -EINVAL;
8971
8972         if (trap->type != devlink_trap_generic[trap->id].type)
8973                 return -EINVAL;
8974
8975         return 0;
8976 }
8977
8978 static int devlink_trap_driver_verify(const struct devlink_trap *trap)
8979 {
8980         int i;
8981
8982         if (trap->id <= DEVLINK_TRAP_GENERIC_ID_MAX)
8983                 return -EINVAL;
8984
8985         for (i = 0; i < ARRAY_SIZE(devlink_trap_generic); i++) {
8986                 if (!strcmp(trap->name, devlink_trap_generic[i].name))
8987                         return -EEXIST;
8988         }
8989
8990         return 0;
8991 }
8992
8993 static int devlink_trap_verify(const struct devlink_trap *trap)
8994 {
8995         if (!trap || !trap->name)
8996                 return -EINVAL;
8997
8998         if (trap->generic)
8999                 return devlink_trap_generic_verify(trap);
9000         else
9001                 return devlink_trap_driver_verify(trap);
9002 }
9003
9004 static int
9005 devlink_trap_group_generic_verify(const struct devlink_trap_group *group)
9006 {
9007         if (group->id > DEVLINK_TRAP_GROUP_GENERIC_ID_MAX)
9008                 return -EINVAL;
9009
9010         if (strcmp(group->name, devlink_trap_group_generic[group->id].name))
9011                 return -EINVAL;
9012
9013         return 0;
9014 }
9015
9016 static int
9017 devlink_trap_group_driver_verify(const struct devlink_trap_group *group)
9018 {
9019         int i;
9020
9021         if (group->id <= DEVLINK_TRAP_GROUP_GENERIC_ID_MAX)
9022                 return -EINVAL;
9023
9024         for (i = 0; i < ARRAY_SIZE(devlink_trap_group_generic); i++) {
9025                 if (!strcmp(group->name, devlink_trap_group_generic[i].name))
9026                         return -EEXIST;
9027         }
9028
9029         return 0;
9030 }
9031
9032 static int devlink_trap_group_verify(const struct devlink_trap_group *group)
9033 {
9034         if (group->generic)
9035                 return devlink_trap_group_generic_verify(group);
9036         else
9037                 return devlink_trap_group_driver_verify(group);
9038 }
9039
9040 static void
9041 devlink_trap_group_notify(struct devlink *devlink,
9042                           const struct devlink_trap_group_item *group_item,
9043                           enum devlink_command cmd)
9044 {
9045         struct sk_buff *msg;
9046         int err;
9047
9048         WARN_ON_ONCE(cmd != DEVLINK_CMD_TRAP_GROUP_NEW &&
9049                      cmd != DEVLINK_CMD_TRAP_GROUP_DEL);
9050
9051         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9052         if (!msg)
9053                 return;
9054
9055         err = devlink_nl_trap_group_fill(msg, devlink, group_item, cmd, 0, 0,
9056                                          0);
9057         if (err) {
9058                 nlmsg_free(msg);
9059                 return;
9060         }
9061
9062         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
9063                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
9064 }
9065
9066 static int
9067 devlink_trap_item_group_link(struct devlink *devlink,
9068                              struct devlink_trap_item *trap_item)
9069 {
9070         u16 group_id = trap_item->trap->init_group_id;
9071         struct devlink_trap_group_item *group_item;
9072
9073         group_item = devlink_trap_group_item_lookup_by_id(devlink, group_id);
9074         if (WARN_ON_ONCE(!group_item))
9075                 return -EINVAL;
9076
9077         trap_item->group_item = group_item;
9078
9079         return 0;
9080 }
9081
9082 static void devlink_trap_notify(struct devlink *devlink,
9083                                 const struct devlink_trap_item *trap_item,
9084                                 enum devlink_command cmd)
9085 {
9086         struct sk_buff *msg;
9087         int err;
9088
9089         WARN_ON_ONCE(cmd != DEVLINK_CMD_TRAP_NEW &&
9090                      cmd != DEVLINK_CMD_TRAP_DEL);
9091
9092         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9093         if (!msg)
9094                 return;
9095
9096         err = devlink_nl_trap_fill(msg, devlink, trap_item, cmd, 0, 0, 0);
9097         if (err) {
9098                 nlmsg_free(msg);
9099                 return;
9100         }
9101
9102         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
9103                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
9104 }
9105
9106 static int
9107 devlink_trap_register(struct devlink *devlink,
9108                       const struct devlink_trap *trap, void *priv)
9109 {
9110         struct devlink_trap_item *trap_item;
9111         int err;
9112
9113         if (devlink_trap_item_lookup(devlink, trap->name))
9114                 return -EEXIST;
9115
9116         trap_item = kzalloc(sizeof(*trap_item), GFP_KERNEL);
9117         if (!trap_item)
9118                 return -ENOMEM;
9119
9120         trap_item->stats = netdev_alloc_pcpu_stats(struct devlink_stats);
9121         if (!trap_item->stats) {
9122                 err = -ENOMEM;
9123                 goto err_stats_alloc;
9124         }
9125
9126         trap_item->trap = trap;
9127         trap_item->action = trap->init_action;
9128         trap_item->priv = priv;
9129
9130         err = devlink_trap_item_group_link(devlink, trap_item);
9131         if (err)
9132                 goto err_group_link;
9133
9134         err = devlink->ops->trap_init(devlink, trap, trap_item);
9135         if (err)
9136                 goto err_trap_init;
9137
9138         list_add_tail(&trap_item->list, &devlink->trap_list);
9139         devlink_trap_notify(devlink, trap_item, DEVLINK_CMD_TRAP_NEW);
9140
9141         return 0;
9142
9143 err_trap_init:
9144 err_group_link:
9145         free_percpu(trap_item->stats);
9146 err_stats_alloc:
9147         kfree(trap_item);
9148         return err;
9149 }
9150
9151 static void devlink_trap_unregister(struct devlink *devlink,
9152                                     const struct devlink_trap *trap)
9153 {
9154         struct devlink_trap_item *trap_item;
9155
9156         trap_item = devlink_trap_item_lookup(devlink, trap->name);
9157         if (WARN_ON_ONCE(!trap_item))
9158                 return;
9159
9160         devlink_trap_notify(devlink, trap_item, DEVLINK_CMD_TRAP_DEL);
9161         list_del(&trap_item->list);
9162         if (devlink->ops->trap_fini)
9163                 devlink->ops->trap_fini(devlink, trap, trap_item);
9164         free_percpu(trap_item->stats);
9165         kfree(trap_item);
9166 }
9167
9168 static void devlink_trap_disable(struct devlink *devlink,
9169                                  const struct devlink_trap *trap)
9170 {
9171         struct devlink_trap_item *trap_item;
9172
9173         trap_item = devlink_trap_item_lookup(devlink, trap->name);
9174         if (WARN_ON_ONCE(!trap_item))
9175                 return;
9176
9177         devlink->ops->trap_action_set(devlink, trap, DEVLINK_TRAP_ACTION_DROP,
9178                                       NULL);
9179         trap_item->action = DEVLINK_TRAP_ACTION_DROP;
9180 }
9181
9182 /**
9183  * devlink_traps_register - Register packet traps with devlink.
9184  * @devlink: devlink.
9185  * @traps: Packet traps.
9186  * @traps_count: Count of provided packet traps.
9187  * @priv: Driver private information.
9188  *
9189  * Return: Non-zero value on failure.
9190  */
9191 int devlink_traps_register(struct devlink *devlink,
9192                            const struct devlink_trap *traps,
9193                            size_t traps_count, void *priv)
9194 {
9195         int i, err;
9196
9197         if (!devlink->ops->trap_init || !devlink->ops->trap_action_set)
9198                 return -EINVAL;
9199
9200         mutex_lock(&devlink->lock);
9201         for (i = 0; i < traps_count; i++) {
9202                 const struct devlink_trap *trap = &traps[i];
9203
9204                 err = devlink_trap_verify(trap);
9205                 if (err)
9206                         goto err_trap_verify;
9207
9208                 err = devlink_trap_register(devlink, trap, priv);
9209                 if (err)
9210                         goto err_trap_register;
9211         }
9212         mutex_unlock(&devlink->lock);
9213
9214         return 0;
9215
9216 err_trap_register:
9217 err_trap_verify:
9218         for (i--; i >= 0; i--)
9219                 devlink_trap_unregister(devlink, &traps[i]);
9220         mutex_unlock(&devlink->lock);
9221         return err;
9222 }
9223 EXPORT_SYMBOL_GPL(devlink_traps_register);
9224
9225 /**
9226  * devlink_traps_unregister - Unregister packet traps from devlink.
9227  * @devlink: devlink.
9228  * @traps: Packet traps.
9229  * @traps_count: Count of provided packet traps.
9230  */
9231 void devlink_traps_unregister(struct devlink *devlink,
9232                               const struct devlink_trap *traps,
9233                               size_t traps_count)
9234 {
9235         int i;
9236
9237         mutex_lock(&devlink->lock);
9238         /* Make sure we do not have any packets in-flight while unregistering
9239          * traps by disabling all of them and waiting for a grace period.
9240          */
9241         for (i = traps_count - 1; i >= 0; i--)
9242                 devlink_trap_disable(devlink, &traps[i]);
9243         synchronize_rcu();
9244         for (i = traps_count - 1; i >= 0; i--)
9245                 devlink_trap_unregister(devlink, &traps[i]);
9246         mutex_unlock(&devlink->lock);
9247 }
9248 EXPORT_SYMBOL_GPL(devlink_traps_unregister);
9249
9250 static void
9251 devlink_trap_stats_update(struct devlink_stats __percpu *trap_stats,
9252                           size_t skb_len)
9253 {
9254         struct devlink_stats *stats;
9255
9256         stats = this_cpu_ptr(trap_stats);
9257         u64_stats_update_begin(&stats->syncp);
9258         stats->rx_bytes += skb_len;
9259         stats->rx_packets++;
9260         u64_stats_update_end(&stats->syncp);
9261 }
9262
9263 static void
9264 devlink_trap_report_metadata_set(struct devlink_trap_metadata *metadata,
9265                                  const struct devlink_trap_item *trap_item,
9266                                  struct devlink_port *in_devlink_port,
9267                                  const struct flow_action_cookie *fa_cookie)
9268 {
9269         metadata->trap_name = trap_item->trap->name;
9270         metadata->trap_group_name = trap_item->group_item->group->name;
9271         metadata->fa_cookie = fa_cookie;
9272
9273         spin_lock(&in_devlink_port->type_lock);
9274         if (in_devlink_port->type == DEVLINK_PORT_TYPE_ETH)
9275                 metadata->input_dev = in_devlink_port->type_dev;
9276         spin_unlock(&in_devlink_port->type_lock);
9277 }
9278
9279 /**
9280  * devlink_trap_report - Report trapped packet to drop monitor.
9281  * @devlink: devlink.
9282  * @skb: Trapped packet.
9283  * @trap_ctx: Trap context.
9284  * @in_devlink_port: Input devlink port.
9285  * @fa_cookie: Flow action cookie. Could be NULL.
9286  */
9287 void devlink_trap_report(struct devlink *devlink, struct sk_buff *skb,
9288                          void *trap_ctx, struct devlink_port *in_devlink_port,
9289                          const struct flow_action_cookie *fa_cookie)
9290
9291 {
9292         struct devlink_trap_item *trap_item = trap_ctx;
9293
9294         devlink_trap_stats_update(trap_item->stats, skb->len);
9295         devlink_trap_stats_update(trap_item->group_item->stats, skb->len);
9296
9297         /* Control packets were not dropped by the device or encountered an
9298          * exception during forwarding and therefore should not be reported to
9299          * the kernel's drop monitor.
9300          */
9301         if (trap_item->trap->type == DEVLINK_TRAP_TYPE_CONTROL)
9302                 return;
9303
9304         if (trace_devlink_trap_report_enabled()) {
9305                 struct devlink_trap_metadata metadata = {};
9306
9307                 devlink_trap_report_metadata_set(&metadata, trap_item,
9308                                                  in_devlink_port, fa_cookie);
9309                 trace_devlink_trap_report(devlink, skb, &metadata);
9310         }
9311 }
9312 EXPORT_SYMBOL_GPL(devlink_trap_report);
9313
9314 /**
9315  * devlink_trap_ctx_priv - Trap context to driver private information.
9316  * @trap_ctx: Trap context.
9317  *
9318  * Return: Driver private information passed during registration.
9319  */
9320 void *devlink_trap_ctx_priv(void *trap_ctx)
9321 {
9322         struct devlink_trap_item *trap_item = trap_ctx;
9323
9324         return trap_item->priv;
9325 }
9326 EXPORT_SYMBOL_GPL(devlink_trap_ctx_priv);
9327
9328 static int
9329 devlink_trap_group_item_policer_link(struct devlink *devlink,
9330                                      struct devlink_trap_group_item *group_item)
9331 {
9332         u32 policer_id = group_item->group->init_policer_id;
9333         struct devlink_trap_policer_item *policer_item;
9334
9335         if (policer_id == 0)
9336                 return 0;
9337
9338         policer_item = devlink_trap_policer_item_lookup(devlink, policer_id);
9339         if (WARN_ON_ONCE(!policer_item))
9340                 return -EINVAL;
9341
9342         group_item->policer_item = policer_item;
9343
9344         return 0;
9345 }
9346
9347 static int
9348 devlink_trap_group_register(struct devlink *devlink,
9349                             const struct devlink_trap_group *group)
9350 {
9351         struct devlink_trap_group_item *group_item;
9352         int err;
9353
9354         if (devlink_trap_group_item_lookup(devlink, group->name))
9355                 return -EEXIST;
9356
9357         group_item = kzalloc(sizeof(*group_item), GFP_KERNEL);
9358         if (!group_item)
9359                 return -ENOMEM;
9360
9361         group_item->stats = netdev_alloc_pcpu_stats(struct devlink_stats);
9362         if (!group_item->stats) {
9363                 err = -ENOMEM;
9364                 goto err_stats_alloc;
9365         }
9366
9367         group_item->group = group;
9368
9369         err = devlink_trap_group_item_policer_link(devlink, group_item);
9370         if (err)
9371                 goto err_policer_link;
9372
9373         if (devlink->ops->trap_group_init) {
9374                 err = devlink->ops->trap_group_init(devlink, group);
9375                 if (err)
9376                         goto err_group_init;
9377         }
9378
9379         list_add_tail(&group_item->list, &devlink->trap_group_list);
9380         devlink_trap_group_notify(devlink, group_item,
9381                                   DEVLINK_CMD_TRAP_GROUP_NEW);
9382
9383         return 0;
9384
9385 err_group_init:
9386 err_policer_link:
9387         free_percpu(group_item->stats);
9388 err_stats_alloc:
9389         kfree(group_item);
9390         return err;
9391 }
9392
9393 static void
9394 devlink_trap_group_unregister(struct devlink *devlink,
9395                               const struct devlink_trap_group *group)
9396 {
9397         struct devlink_trap_group_item *group_item;
9398
9399         group_item = devlink_trap_group_item_lookup(devlink, group->name);
9400         if (WARN_ON_ONCE(!group_item))
9401                 return;
9402
9403         devlink_trap_group_notify(devlink, group_item,
9404                                   DEVLINK_CMD_TRAP_GROUP_DEL);
9405         list_del(&group_item->list);
9406         free_percpu(group_item->stats);
9407         kfree(group_item);
9408 }
9409
9410 /**
9411  * devlink_trap_groups_register - Register packet trap groups with devlink.
9412  * @devlink: devlink.
9413  * @groups: Packet trap groups.
9414  * @groups_count: Count of provided packet trap groups.
9415  *
9416  * Return: Non-zero value on failure.
9417  */
9418 int devlink_trap_groups_register(struct devlink *devlink,
9419                                  const struct devlink_trap_group *groups,
9420                                  size_t groups_count)
9421 {
9422         int i, err;
9423
9424         mutex_lock(&devlink->lock);
9425         for (i = 0; i < groups_count; i++) {
9426                 const struct devlink_trap_group *group = &groups[i];
9427
9428                 err = devlink_trap_group_verify(group);
9429                 if (err)
9430                         goto err_trap_group_verify;
9431
9432                 err = devlink_trap_group_register(devlink, group);
9433                 if (err)
9434                         goto err_trap_group_register;
9435         }
9436         mutex_unlock(&devlink->lock);
9437
9438         return 0;
9439
9440 err_trap_group_register:
9441 err_trap_group_verify:
9442         for (i--; i >= 0; i--)
9443                 devlink_trap_group_unregister(devlink, &groups[i]);
9444         mutex_unlock(&devlink->lock);
9445         return err;
9446 }
9447 EXPORT_SYMBOL_GPL(devlink_trap_groups_register);
9448
9449 /**
9450  * devlink_trap_groups_unregister - Unregister packet trap groups from devlink.
9451  * @devlink: devlink.
9452  * @groups: Packet trap groups.
9453  * @groups_count: Count of provided packet trap groups.
9454  */
9455 void devlink_trap_groups_unregister(struct devlink *devlink,
9456                                     const struct devlink_trap_group *groups,
9457                                     size_t groups_count)
9458 {
9459         int i;
9460
9461         mutex_lock(&devlink->lock);
9462         for (i = groups_count - 1; i >= 0; i--)
9463                 devlink_trap_group_unregister(devlink, &groups[i]);
9464         mutex_unlock(&devlink->lock);
9465 }
9466 EXPORT_SYMBOL_GPL(devlink_trap_groups_unregister);
9467
9468 static void
9469 devlink_trap_policer_notify(struct devlink *devlink,
9470                             const struct devlink_trap_policer_item *policer_item,
9471                             enum devlink_command cmd)
9472 {
9473         struct sk_buff *msg;
9474         int err;
9475
9476         WARN_ON_ONCE(cmd != DEVLINK_CMD_TRAP_POLICER_NEW &&
9477                      cmd != DEVLINK_CMD_TRAP_POLICER_DEL);
9478
9479         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9480         if (!msg)
9481                 return;
9482
9483         err = devlink_nl_trap_policer_fill(msg, devlink, policer_item, cmd, 0,
9484                                            0, 0);
9485         if (err) {
9486                 nlmsg_free(msg);
9487                 return;
9488         }
9489
9490         genlmsg_multicast_netns(&devlink_nl_family, devlink_net(devlink),
9491                                 msg, 0, DEVLINK_MCGRP_CONFIG, GFP_KERNEL);
9492 }
9493
9494 static int
9495 devlink_trap_policer_register(struct devlink *devlink,
9496                               const struct devlink_trap_policer *policer)
9497 {
9498         struct devlink_trap_policer_item *policer_item;
9499         int err;
9500
9501         if (devlink_trap_policer_item_lookup(devlink, policer->id))
9502                 return -EEXIST;
9503
9504         policer_item = kzalloc(sizeof(*policer_item), GFP_KERNEL);
9505         if (!policer_item)
9506                 return -ENOMEM;
9507
9508         policer_item->policer = policer;
9509         policer_item->rate = policer->init_rate;
9510         policer_item->burst = policer->init_burst;
9511
9512         if (devlink->ops->trap_policer_init) {
9513                 err = devlink->ops->trap_policer_init(devlink, policer);
9514                 if (err)
9515                         goto err_policer_init;
9516         }
9517
9518         list_add_tail(&policer_item->list, &devlink->trap_policer_list);
9519         devlink_trap_policer_notify(devlink, policer_item,
9520                                     DEVLINK_CMD_TRAP_POLICER_NEW);
9521
9522         return 0;
9523
9524 err_policer_init:
9525         kfree(policer_item);
9526         return err;
9527 }
9528
9529 static void
9530 devlink_trap_policer_unregister(struct devlink *devlink,
9531                                 const struct devlink_trap_policer *policer)
9532 {
9533         struct devlink_trap_policer_item *policer_item;
9534
9535         policer_item = devlink_trap_policer_item_lookup(devlink, policer->id);
9536         if (WARN_ON_ONCE(!policer_item))
9537                 return;
9538
9539         devlink_trap_policer_notify(devlink, policer_item,
9540                                     DEVLINK_CMD_TRAP_POLICER_DEL);
9541         list_del(&policer_item->list);
9542         if (devlink->ops->trap_policer_fini)
9543                 devlink->ops->trap_policer_fini(devlink, policer);
9544         kfree(policer_item);
9545 }
9546
9547 /**
9548  * devlink_trap_policers_register - Register packet trap policers with devlink.
9549  * @devlink: devlink.
9550  * @policers: Packet trap policers.
9551  * @policers_count: Count of provided packet trap policers.
9552  *
9553  * Return: Non-zero value on failure.
9554  */
9555 int
9556 devlink_trap_policers_register(struct devlink *devlink,
9557                                const struct devlink_trap_policer *policers,
9558                                size_t policers_count)
9559 {
9560         int i, err;
9561
9562         mutex_lock(&devlink->lock);
9563         for (i = 0; i < policers_count; i++) {
9564                 const struct devlink_trap_policer *policer = &policers[i];
9565
9566                 if (WARN_ON(policer->id == 0 ||
9567                             policer->max_rate < policer->min_rate ||
9568                             policer->max_burst < policer->min_burst)) {
9569                         err = -EINVAL;
9570                         goto err_trap_policer_verify;
9571                 }
9572
9573                 err = devlink_trap_policer_register(devlink, policer);
9574                 if (err)
9575                         goto err_trap_policer_register;
9576         }
9577         mutex_unlock(&devlink->lock);
9578
9579         return 0;
9580
9581 err_trap_policer_register:
9582 err_trap_policer_verify:
9583         for (i--; i >= 0; i--)
9584                 devlink_trap_policer_unregister(devlink, &policers[i]);
9585         mutex_unlock(&devlink->lock);
9586         return err;
9587 }
9588 EXPORT_SYMBOL_GPL(devlink_trap_policers_register);
9589
9590 /**
9591  * devlink_trap_policers_unregister - Unregister packet trap policers from devlink.
9592  * @devlink: devlink.
9593  * @policers: Packet trap policers.
9594  * @policers_count: Count of provided packet trap policers.
9595  */
9596 void
9597 devlink_trap_policers_unregister(struct devlink *devlink,
9598                                  const struct devlink_trap_policer *policers,
9599                                  size_t policers_count)
9600 {
9601         int i;
9602
9603         mutex_lock(&devlink->lock);
9604         for (i = policers_count - 1; i >= 0; i--)
9605                 devlink_trap_policer_unregister(devlink, &policers[i]);
9606         mutex_unlock(&devlink->lock);
9607 }
9608 EXPORT_SYMBOL_GPL(devlink_trap_policers_unregister);
9609
9610 static void __devlink_compat_running_version(struct devlink *devlink,
9611                                              char *buf, size_t len)
9612 {
9613         const struct nlattr *nlattr;
9614         struct devlink_info_req req;
9615         struct sk_buff *msg;
9616         int rem, err;
9617
9618         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
9619         if (!msg)
9620                 return;
9621
9622         req.msg = msg;
9623         err = devlink->ops->info_get(devlink, &req, NULL);
9624         if (err)
9625                 goto free_msg;
9626
9627         nla_for_each_attr(nlattr, (void *)msg->data, msg->len, rem) {
9628                 const struct nlattr *kv;
9629                 int rem_kv;
9630
9631                 if (nla_type(nlattr) != DEVLINK_ATTR_INFO_VERSION_RUNNING)
9632                         continue;
9633
9634                 nla_for_each_nested(kv, nlattr, rem_kv) {
9635                         if (nla_type(kv) != DEVLINK_ATTR_INFO_VERSION_VALUE)
9636                                 continue;
9637
9638                         strlcat(buf, nla_data(kv), len);
9639                         strlcat(buf, " ", len);
9640                 }
9641         }
9642 free_msg:
9643         nlmsg_free(msg);
9644 }
9645
9646 void devlink_compat_running_version(struct net_device *dev,
9647                                     char *buf, size_t len)
9648 {
9649         struct devlink *devlink;
9650
9651         dev_hold(dev);
9652         rtnl_unlock();
9653
9654         devlink = netdev_to_devlink(dev);
9655         if (!devlink || !devlink->ops->info_get)
9656                 goto out;
9657
9658         mutex_lock(&devlink->lock);
9659         __devlink_compat_running_version(devlink, buf, len);
9660         mutex_unlock(&devlink->lock);
9661
9662 out:
9663         rtnl_lock();
9664         dev_put(dev);
9665 }
9666
9667 int devlink_compat_flash_update(struct net_device *dev, const char *file_name)
9668 {
9669         struct devlink_flash_update_params params = {};
9670         struct devlink *devlink;
9671         int ret;
9672
9673         dev_hold(dev);
9674         rtnl_unlock();
9675
9676         devlink = netdev_to_devlink(dev);
9677         if (!devlink || !devlink->ops->flash_update) {
9678                 ret = -EOPNOTSUPP;
9679                 goto out;
9680         }
9681
9682         params.file_name = file_name;
9683
9684         mutex_lock(&devlink->lock);
9685         ret = devlink->ops->flash_update(devlink, &params, NULL);
9686         mutex_unlock(&devlink->lock);
9687
9688 out:
9689         rtnl_lock();
9690         dev_put(dev);
9691
9692         return ret;
9693 }
9694
9695 int devlink_compat_phys_port_name_get(struct net_device *dev,
9696                                       char *name, size_t len)
9697 {
9698         struct devlink_port *devlink_port;
9699
9700         /* RTNL mutex is held here which ensures that devlink_port
9701          * instance cannot disappear in the middle. No need to take
9702          * any devlink lock as only permanent values are accessed.
9703          */
9704         ASSERT_RTNL();
9705
9706         devlink_port = netdev_to_devlink_port(dev);
9707         if (!devlink_port)
9708                 return -EOPNOTSUPP;
9709
9710         return __devlink_port_phys_port_name_get(devlink_port, name, len);
9711 }
9712
9713 int devlink_compat_switch_id_get(struct net_device *dev,
9714                                  struct netdev_phys_item_id *ppid)
9715 {
9716         struct devlink_port *devlink_port;
9717
9718         /* Caller must hold RTNL mutex or reference to dev, which ensures that
9719          * devlink_port instance cannot disappear in the middle. No need to take
9720          * any devlink lock as only permanent values are accessed.
9721          */
9722         devlink_port = netdev_to_devlink_port(dev);
9723         if (!devlink_port || !devlink_port->switch_port)
9724                 return -EOPNOTSUPP;
9725
9726         memcpy(ppid, &devlink_port->attrs.switch_id, sizeof(*ppid));
9727
9728         return 0;
9729 }
9730
9731 static void __net_exit devlink_pernet_pre_exit(struct net *net)
9732 {
9733         struct devlink *devlink;
9734         int err;
9735
9736         /* In case network namespace is getting destroyed, reload
9737          * all devlink instances from this namespace into init_net.
9738          */
9739         mutex_lock(&devlink_mutex);
9740         list_for_each_entry(devlink, &devlink_list, list) {
9741                 if (net_eq(devlink_net(devlink), net)) {
9742                         if (WARN_ON(!devlink_reload_supported(devlink)))
9743                                 continue;
9744                         err = devlink_reload(devlink, &init_net, NULL);
9745                         if (err && err != -EOPNOTSUPP)
9746                                 pr_warn("Failed to reload devlink instance into init_net\n");
9747                 }
9748         }
9749         mutex_unlock(&devlink_mutex);
9750 }
9751
9752 static struct pernet_operations devlink_pernet_ops __net_initdata = {
9753         .pre_exit = devlink_pernet_pre_exit,
9754 };
9755
9756 static int __init devlink_init(void)
9757 {
9758         int err;
9759
9760         err = genl_register_family(&devlink_nl_family);
9761         if (err)
9762                 goto out;
9763         err = register_pernet_subsys(&devlink_pernet_ops);
9764
9765 out:
9766         WARN_ON(err);
9767         return err;
9768 }
9769
9770 subsys_initcall(devlink_init);