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