drivers/net/netdevsim/dev.c

   1 /*
   2  * Copyright (c) 2018 Cumulus Networks. All rights reserved.
   3  * Copyright (c) 2018 David Ahern <dsa@cumulusnetworks.com>
   4  * Copyright (c) 2019 Mellanox Technologies. All rights reserved.
   5  *
   6  * This software is licensed under the GNU General License Version 2,
   7  * June 1991 as shown in the file COPYING in the top-level directory of this
   8  * source tree.
   9  *
  10  * THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS"
  11  * WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING,
  12  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  13  * FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE
  14  * OF THE PROGRAM IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME
  15  * THE COST OF ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
  16  */
  17
  18 #include <linux/debugfs.h>
  19 #include <linux/device.h>
  20 #include <linux/etherdevice.h>
  21 #include <linux/inet.h>
  22 #include <linux/jiffies.h>
  23 #include <linux/kernel.h>
  24 #include <linux/list.h>
  25 #include <linux/mutex.h>
  26 #include <linux/random.h>
  27 #include <linux/rtnetlink.h>
  28 #include <linux/workqueue.h>
  29 #include <net/devlink.h>
  30 #include <net/ip.h>
  31 #include <uapi/linux/devlink.h>
  32 #include <uapi/linux/ip.h>
  33 #include <uapi/linux/udp.h>
  34
  35 #include "netdevsim.h"
  36
  37 static struct dentry *nsim_dev_ddir;
  38
  39 #define NSIM_DEV_DUMMY_REGION_SIZE (1024 * 32)
  40
  41 static ssize_t nsim_dev_take_snapshot_write(struct file *file,
  42                                             const char __user *data,
  43                                             size_t count, loff_t *ppos)
  44 {
  45         struct nsim_dev *nsim_dev = file->private_data;
  46         void *dummy_data;
  47         int err;
  48         u32 id;
  49
  50         dummy_data = kmalloc(NSIM_DEV_DUMMY_REGION_SIZE, GFP_KERNEL);
  51         if (!dummy_data)
  52                 return -ENOMEM;
  53
  54         get_random_bytes(dummy_data, NSIM_DEV_DUMMY_REGION_SIZE);
  55
  56         id = devlink_region_shapshot_id_get(priv_to_devlink(nsim_dev));
  57         err = devlink_region_snapshot_create(nsim_dev->dummy_region,
  58                                              dummy_data, id, kfree);
  59         if (err) {
  60                 pr_err("Failed to create region snapshot\n");
  61                 kfree(dummy_data);
  62                 return err;
  63         }
  64
  65         return count;
  66 }
  67
  68 static const struct file_operations nsim_dev_take_snapshot_fops = {
  69         .open = simple_open,
  70         .write = nsim_dev_take_snapshot_write,
  71         .llseek = generic_file_llseek,
  72 };
  73
  74 static int nsim_dev_debugfs_init(struct nsim_dev *nsim_dev)
  75 {
  76         char dev_ddir_name[16];
  77
  78         sprintf(dev_ddir_name, DRV_NAME "%u", nsim_dev->nsim_bus_dev->dev.id);
  79         nsim_dev->ddir = debugfs_create_dir(dev_ddir_name, nsim_dev_ddir);
  80         if (IS_ERR_OR_NULL(nsim_dev->ddir))
  81                 return PTR_ERR_OR_ZERO(nsim_dev->ddir) ?: -EINVAL;
  82         nsim_dev->ports_ddir = debugfs_create_dir("ports", nsim_dev->ddir);
  83         if (IS_ERR_OR_NULL(nsim_dev->ports_ddir))
  84                 return PTR_ERR_OR_ZERO(nsim_dev->ports_ddir) ?: -EINVAL;
  85         debugfs_create_bool("fw_update_status", 0600, nsim_dev->ddir,
  86                             &nsim_dev->fw_update_status);
  87         debugfs_create_u32("max_macs", 0600, nsim_dev->ddir,
  88                            &nsim_dev->max_macs);
  89         debugfs_create_bool("test1", 0600, nsim_dev->ddir,
  90                             &nsim_dev->test1);
  91         debugfs_create_file("take_snapshot", 0200, nsim_dev->ddir, nsim_dev,
  92                             &nsim_dev_take_snapshot_fops);
  93         debugfs_create_bool("dont_allow_reload", 0600, nsim_dev->ddir,
  94                             &nsim_dev->dont_allow_reload);
  95         debugfs_create_bool("fail_reload", 0600, nsim_dev->ddir,
  96                             &nsim_dev->fail_reload);
  97         return 0;
  98 }
  99
 100 static void nsim_dev_debugfs_exit(struct nsim_dev *nsim_dev)
 101 {
 102         debugfs_remove_recursive(nsim_dev->ports_ddir);
 103         debugfs_remove_recursive(nsim_dev->ddir);
 104 }
 105
 106 static int nsim_dev_port_debugfs_init(struct nsim_dev *nsim_dev,
 107                                       struct nsim_dev_port *nsim_dev_port)
 108 {
 109         char port_ddir_name[16];
 110         char dev_link_name[32];
 111
 112         sprintf(port_ddir_name, "%u", nsim_dev_port->port_index);
 113         nsim_dev_port->ddir = debugfs_create_dir(port_ddir_name,
 114                                                  nsim_dev->ports_ddir);
 115         if (IS_ERR_OR_NULL(nsim_dev_port->ddir))
 116                 return -ENOMEM;
 117
 118         sprintf(dev_link_name, "../../../" DRV_NAME "%u",
 119                 nsim_dev->nsim_bus_dev->dev.id);
 120         debugfs_create_symlink("dev", nsim_dev_port->ddir, dev_link_name);
 121
 122         return 0;
 123 }
 124
 125 static void nsim_dev_port_debugfs_exit(struct nsim_dev_port *nsim_dev_port)
 126 {
 127         debugfs_remove_recursive(nsim_dev_port->ddir);
 128 }
 129
 130 static int nsim_dev_resources_register(struct devlink *devlink)
 131 {
 132         struct devlink_resource_size_params params = {
 133                 .size_max = (u64)-1,
 134                 .size_granularity = 1,
 135                 .unit = DEVLINK_RESOURCE_UNIT_ENTRY
 136         };
 137         int err;
 138
 139         /* Resources for IPv4 */
 140         err = devlink_resource_register(devlink, "IPv4", (u64)-1,
 141                                         NSIM_RESOURCE_IPV4,
 142                                         DEVLINK_RESOURCE_ID_PARENT_TOP,
 143                                         &params);
 144         if (err) {
 145                 pr_err("Failed to register IPv4 top resource\n");
 146                 goto out;
 147         }
 148
 149         err = devlink_resource_register(devlink, "fib", (u64)-1,
 150                                         NSIM_RESOURCE_IPV4_FIB,
 151                                         NSIM_RESOURCE_IPV4, &params);
 152         if (err) {
 153                 pr_err("Failed to register IPv4 FIB resource\n");
 154                 return err;
 155         }
 156
 157         err = devlink_resource_register(devlink, "fib-rules", (u64)-1,
 158                                         NSIM_RESOURCE_IPV4_FIB_RULES,
 159                                         NSIM_RESOURCE_IPV4, &params);
 160         if (err) {
 161                 pr_err("Failed to register IPv4 FIB rules resource\n");
 162                 return err;
 163         }
 164
 165         /* Resources for IPv6 */
 166         err = devlink_resource_register(devlink, "IPv6", (u64)-1,
 167                                         NSIM_RESOURCE_IPV6,
 168                                         DEVLINK_RESOURCE_ID_PARENT_TOP,
 169                                         &params);
 170         if (err) {
 171                 pr_err("Failed to register IPv6 top resource\n");
 172                 goto out;
 173         }
 174
 175         err = devlink_resource_register(devlink, "fib", (u64)-1,
 176                                         NSIM_RESOURCE_IPV6_FIB,
 177                                         NSIM_RESOURCE_IPV6, &params);
 178         if (err) {
 179                 pr_err("Failed to register IPv6 FIB resource\n");
 180                 return err;
 181         }
 182
 183         err = devlink_resource_register(devlink, "fib-rules", (u64)-1,
 184                                         NSIM_RESOURCE_IPV6_FIB_RULES,
 185                                         NSIM_RESOURCE_IPV6, &params);
 186         if (err) {
 187                 pr_err("Failed to register IPv6 FIB rules resource\n");
 188                 return err;
 189         }
 190
 191 out:
 192         return err;
 193 }
 194
 195 enum nsim_devlink_param_id {
 196         NSIM_DEVLINK_PARAM_ID_BASE = DEVLINK_PARAM_GENERIC_ID_MAX,
 197         NSIM_DEVLINK_PARAM_ID_TEST1,
 198 };
 199
 200 static const struct devlink_param nsim_devlink_params[] = {
 201         DEVLINK_PARAM_GENERIC(MAX_MACS,
 202                               BIT(DEVLINK_PARAM_CMODE_DRIVERINIT),
 203                               NULL, NULL, NULL),
 204         DEVLINK_PARAM_DRIVER(NSIM_DEVLINK_PARAM_ID_TEST1,
 205                              "test1", DEVLINK_PARAM_TYPE_BOOL,
 206                              BIT(DEVLINK_PARAM_CMODE_DRIVERINIT),
 207                              NULL, NULL, NULL),
 208 };
 209
 210 static void nsim_devlink_set_params_init_values(struct nsim_dev *nsim_dev,
 211                                                 struct devlink *devlink)
 212 {
 213         union devlink_param_value value;
 214
 215         value.vu32 = nsim_dev->max_macs;
 216         devlink_param_driverinit_value_set(devlink,
 217                                            DEVLINK_PARAM_GENERIC_ID_MAX_MACS,
 218                                            value);
 219         value.vbool = nsim_dev->test1;
 220         devlink_param_driverinit_value_set(devlink,
 221                                            NSIM_DEVLINK_PARAM_ID_TEST1,
 222                                            value);
 223 }
 224
 225 static void nsim_devlink_param_load_driverinit_values(struct devlink *devlink)
 226 {
 227         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 228         union devlink_param_value saved_value;
 229         int err;
 230
 231         err = devlink_param_driverinit_value_get(devlink,
 232                                                  DEVLINK_PARAM_GENERIC_ID_MAX_MACS,
 233                                                  &saved_value);
 234         if (!err)
 235                 nsim_dev->max_macs = saved_value.vu32;
 236         err = devlink_param_driverinit_value_get(devlink,
 237                                                  NSIM_DEVLINK_PARAM_ID_TEST1,
 238                                                  &saved_value);
 239         if (!err)
 240                 nsim_dev->test1 = saved_value.vbool;
 241 }
 242
 243 #define NSIM_DEV_DUMMY_REGION_SNAPSHOT_MAX 16
 244
 245 static int nsim_dev_dummy_region_init(struct nsim_dev *nsim_dev,
 246                                       struct devlink *devlink)
 247 {
 248         nsim_dev->dummy_region =
 249                 devlink_region_create(devlink, "dummy",
 250                                       NSIM_DEV_DUMMY_REGION_SNAPSHOT_MAX,
 251                                       NSIM_DEV_DUMMY_REGION_SIZE);
 252         return PTR_ERR_OR_ZERO(nsim_dev->dummy_region);
 253 }
 254
 255 static void nsim_dev_dummy_region_exit(struct nsim_dev *nsim_dev)
 256 {
 257         devlink_region_destroy(nsim_dev->dummy_region);
 258 }
 259
 260 struct nsim_trap_item {
 261         void *trap_ctx;
 262         enum devlink_trap_action action;
 263 };
 264
 265 struct nsim_trap_data {
 266         struct delayed_work trap_report_dw;
 267         struct nsim_trap_item *trap_items_arr;
 268         struct nsim_dev *nsim_dev;
 269         spinlock_t trap_lock;   /* Protects trap_items_arr */
 270 };
 271
 272 /* All driver-specific traps must be documented in
 273  * Documentation/networking/devlink-trap-netdevsim.rst
 274  */
 275 enum {
 276         NSIM_TRAP_ID_BASE = DEVLINK_TRAP_GENERIC_ID_MAX,
 277         NSIM_TRAP_ID_FID_MISS,
 278 };
 279
 280 #define NSIM_TRAP_NAME_FID_MISS "fid_miss"
 281
 282 #define NSIM_TRAP_METADATA DEVLINK_TRAP_METADATA_TYPE_F_IN_PORT
 283
 284 #define NSIM_TRAP_DROP(_id, _group_id)                                        \
 285         DEVLINK_TRAP_GENERIC(DROP, DROP, _id,                                 \
 286                              DEVLINK_TRAP_GROUP_GENERIC(_group_id),           \
 287                              NSIM_TRAP_METADATA)
 288 #define NSIM_TRAP_EXCEPTION(_id, _group_id)                                   \
 289         DEVLINK_TRAP_GENERIC(EXCEPTION, TRAP, _id,                            \
 290                              DEVLINK_TRAP_GROUP_GENERIC(_group_id),           \
 291                              NSIM_TRAP_METADATA)
 292 #define NSIM_TRAP_DRIVER_EXCEPTION(_id, _group_id)                            \
 293         DEVLINK_TRAP_DRIVER(EXCEPTION, TRAP, NSIM_TRAP_ID_##_id,              \
 294                             NSIM_TRAP_NAME_##_id,                             \
 295                             DEVLINK_TRAP_GROUP_GENERIC(_group_id),            \
 296                             NSIM_TRAP_METADATA)
 297
 298 static const struct devlink_trap nsim_traps_arr[] = {
 299         NSIM_TRAP_DROP(SMAC_MC, L2_DROPS),
 300         NSIM_TRAP_DROP(VLAN_TAG_MISMATCH, L2_DROPS),
 301         NSIM_TRAP_DROP(INGRESS_VLAN_FILTER, L2_DROPS),
 302         NSIM_TRAP_DROP(INGRESS_STP_FILTER, L2_DROPS),
 303         NSIM_TRAP_DROP(EMPTY_TX_LIST, L2_DROPS),
 304         NSIM_TRAP_DROP(PORT_LOOPBACK_FILTER, L2_DROPS),
 305         NSIM_TRAP_DRIVER_EXCEPTION(FID_MISS, L2_DROPS),
 306         NSIM_TRAP_DROP(BLACKHOLE_ROUTE, L3_DROPS),
 307         NSIM_TRAP_EXCEPTION(TTL_ERROR, L3_DROPS),
 308         NSIM_TRAP_DROP(TAIL_DROP, BUFFER_DROPS),
 309 };
 310
 311 #define NSIM_TRAP_L4_DATA_LEN 100
 312
 313 static struct sk_buff *nsim_dev_trap_skb_build(void)
 314 {
 315         int tot_len, data_len = NSIM_TRAP_L4_DATA_LEN;
 316         struct sk_buff *skb;
 317         struct udphdr *udph;
 318         struct ethhdr *eth;
 319         struct iphdr *iph;
 320
 321         skb = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
 322         if (!skb)
 323                 return NULL;
 324         tot_len = sizeof(struct iphdr) + sizeof(struct udphdr) + data_len;
 325
 326         skb_reset_mac_header(skb);
 327         eth = skb_put(skb, sizeof(struct ethhdr));
 328         eth_random_addr(eth->h_dest);
 329         eth_random_addr(eth->h_source);
 330         eth->h_proto = htons(ETH_P_IP);
 331         skb->protocol = htons(ETH_P_IP);
 332
 333         skb_set_network_header(skb, skb->len);
 334         iph = skb_put(skb, sizeof(struct iphdr));
 335         iph->protocol = IPPROTO_UDP;
 336         iph->saddr = in_aton("192.0.2.1");
 337         iph->daddr = in_aton("198.51.100.1");
 338         iph->version = 0x4;
 339         iph->frag_off = 0;
 340         iph->ihl = 0x5;
 341         iph->tot_len = htons(tot_len);
 342         iph->ttl = 100;
 343         iph->check = 0;
 344         iph->check = ip_fast_csum((unsigned char *)iph, iph->ihl);
 345
 346         skb_set_transport_header(skb, skb->len);
 347         udph = skb_put_zero(skb, sizeof(struct udphdr) + data_len);
 348         get_random_bytes(&udph->source, sizeof(u16));
 349         get_random_bytes(&udph->dest, sizeof(u16));
 350         udph->len = htons(sizeof(struct udphdr) + data_len);
 351
 352         return skb;
 353 }
 354
 355 static void nsim_dev_trap_report(struct nsim_dev_port *nsim_dev_port)
 356 {
 357         struct nsim_dev *nsim_dev = nsim_dev_port->ns->nsim_dev;
 358         struct devlink *devlink = priv_to_devlink(nsim_dev);
 359         struct nsim_trap_data *nsim_trap_data;
 360         int i;
 361
 362         nsim_trap_data = nsim_dev->trap_data;
 363
 364         spin_lock(&nsim_trap_data->trap_lock);
 365         for (i = 0; i < ARRAY_SIZE(nsim_traps_arr); i++) {
 366                 struct nsim_trap_item *nsim_trap_item;
 367                 struct sk_buff *skb;
 368
 369                 nsim_trap_item = &nsim_trap_data->trap_items_arr[i];
 370                 if (nsim_trap_item->action == DEVLINK_TRAP_ACTION_DROP)
 371                         continue;
 372
 373                 skb = nsim_dev_trap_skb_build();
 374                 if (!skb)
 375                         continue;
 376                 skb->dev = nsim_dev_port->ns->netdev;
 377
 378                 /* Trapped packets are usually passed to devlink in softIRQ,
 379                  * but in this case they are generated in a workqueue. Disable
 380                  * softIRQs to prevent lockdep from complaining about
 381                  * "incosistent lock state".
 382                  */
 383                 local_bh_disable();
 384                 devlink_trap_report(devlink, skb, nsim_trap_item->trap_ctx,
 385                                     &nsim_dev_port->devlink_port);
 386                 local_bh_enable();
 387                 consume_skb(skb);
 388         }
 389         spin_unlock(&nsim_trap_data->trap_lock);
 390 }
 391
 392 #define NSIM_TRAP_REPORT_INTERVAL_MS    100
 393
 394 static void nsim_dev_trap_report_work(struct work_struct *work)
 395 {
 396         struct nsim_trap_data *nsim_trap_data;
 397         struct nsim_dev_port *nsim_dev_port;
 398         struct nsim_dev *nsim_dev;
 399
 400         nsim_trap_data = container_of(work, struct nsim_trap_data,
 401                                       trap_report_dw.work);
 402         nsim_dev = nsim_trap_data->nsim_dev;
 403
 404         /* For each running port and enabled packet trap, generate a UDP
 405          * packet with a random 5-tuple and report it.
 406          */
 407         mutex_lock(&nsim_dev->port_list_lock);
 408         list_for_each_entry(nsim_dev_port, &nsim_dev->port_list, list) {
 409                 if (!netif_running(nsim_dev_port->ns->netdev))
 410                         continue;
 411
 412                 nsim_dev_trap_report(nsim_dev_port);
 413         }
 414         mutex_unlock(&nsim_dev->port_list_lock);
 415
 416         schedule_delayed_work(&nsim_dev->trap_data->trap_report_dw,
 417                               msecs_to_jiffies(NSIM_TRAP_REPORT_INTERVAL_MS));
 418 }
 419
 420 static int nsim_dev_traps_init(struct devlink *devlink)
 421 {
 422         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 423         struct nsim_trap_data *nsim_trap_data;
 424         int err;
 425
 426         nsim_trap_data = kzalloc(sizeof(*nsim_trap_data), GFP_KERNEL);
 427         if (!nsim_trap_data)
 428                 return -ENOMEM;
 429
 430         nsim_trap_data->trap_items_arr = kcalloc(ARRAY_SIZE(nsim_traps_arr),
 431                                                  sizeof(struct nsim_trap_item),
 432                                                  GFP_KERNEL);
 433         if (!nsim_trap_data->trap_items_arr) {
 434                 err = -ENOMEM;
 435                 goto err_trap_data_free;
 436         }
 437
 438         /* The lock is used to protect the action state of the registered
 439          * traps. The value is written by user and read in delayed work when
 440          * iterating over all the traps.
 441          */
 442         spin_lock_init(&nsim_trap_data->trap_lock);
 443         nsim_trap_data->nsim_dev = nsim_dev;
 444         nsim_dev->trap_data = nsim_trap_data;
 445
 446         err = devlink_traps_register(devlink, nsim_traps_arr,
 447                                      ARRAY_SIZE(nsim_traps_arr), NULL);
 448         if (err)
 449                 goto err_trap_items_free;
 450
 451         INIT_DELAYED_WORK(&nsim_dev->trap_data->trap_report_dw,
 452                           nsim_dev_trap_report_work);
 453         schedule_delayed_work(&nsim_dev->trap_data->trap_report_dw,
 454                               msecs_to_jiffies(NSIM_TRAP_REPORT_INTERVAL_MS));
 455
 456         return 0;
 457
 458 err_trap_items_free:
 459         kfree(nsim_trap_data->trap_items_arr);
 460 err_trap_data_free:
 461         kfree(nsim_trap_data);
 462         return err;
 463 }
 464
 465 static void nsim_dev_traps_exit(struct devlink *devlink)
 466 {
 467         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 468
 469         cancel_delayed_work_sync(&nsim_dev->trap_data->trap_report_dw);
 470         devlink_traps_unregister(devlink, nsim_traps_arr,
 471                                  ARRAY_SIZE(nsim_traps_arr));
 472         kfree(nsim_dev->trap_data->trap_items_arr);
 473         kfree(nsim_dev->trap_data);
 474 }
 475
 476 static int nsim_dev_reload_create(struct nsim_dev *nsim_dev,
 477                                   struct netlink_ext_ack *extack);
 478 static void nsim_dev_reload_destroy(struct nsim_dev *nsim_dev);
 479
 480 static int nsim_dev_reload_down(struct devlink *devlink, bool netns_change,
 481                                 struct netlink_ext_ack *extack)
 482 {
 483         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 484
 485         if (nsim_dev->dont_allow_reload) {
 486                 /* For testing purposes, user set debugfs dont_allow_reload
 487                  * value to true. So forbid it.
 488                  */
 489                 NL_SET_ERR_MSG_MOD(extack, "User forbid the reload for testing purposes");
 490                 return -EOPNOTSUPP;
 491         }
 492
 493         nsim_dev_reload_destroy(nsim_dev);
 494         return 0;
 495 }
 496
 497 static int nsim_dev_reload_up(struct devlink *devlink,
 498                               struct netlink_ext_ack *extack)
 499 {
 500         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 501
 502         if (nsim_dev->fail_reload) {
 503                 /* For testing purposes, user set debugfs fail_reload
 504                  * value to true. Fail right away.
 505                  */
 506                 NL_SET_ERR_MSG_MOD(extack, "User setup the reload to fail for testing purposes");
 507                 return -EINVAL;
 508         }
 509
 510         return nsim_dev_reload_create(nsim_dev, extack);
 511 }
 512
 513 static int nsim_dev_info_get(struct devlink *devlink,
 514                              struct devlink_info_req *req,
 515                              struct netlink_ext_ack *extack)
 516 {
 517         return devlink_info_driver_name_put(req, DRV_NAME);
 518 }
 519
 520 #define NSIM_DEV_FLASH_SIZE 500000
 521 #define NSIM_DEV_FLASH_CHUNK_SIZE 1000
 522 #define NSIM_DEV_FLASH_CHUNK_TIME_MS 10
 523
 524 static int nsim_dev_flash_update(struct devlink *devlink, const char *file_name,
 525                                  const char *component,
 526                                  struct netlink_ext_ack *extack)
 527 {
 528         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 529         int i;
 530
 531         if (nsim_dev->fw_update_status) {
 532                 devlink_flash_update_begin_notify(devlink);
 533                 devlink_flash_update_status_notify(devlink,
 534                                                    "Preparing to flash",
 535                                                    component, 0, 0);
 536         }
 537
 538         for (i = 0; i < NSIM_DEV_FLASH_SIZE / NSIM_DEV_FLASH_CHUNK_SIZE; i++) {
 539                 if (nsim_dev->fw_update_status)
 540                         devlink_flash_update_status_notify(devlink, "Flashing",
 541                                                            component,
 542                                                            i * NSIM_DEV_FLASH_CHUNK_SIZE,
 543                                                            NSIM_DEV_FLASH_SIZE);
 544                 msleep(NSIM_DEV_FLASH_CHUNK_TIME_MS);
 545         }
 546
 547         if (nsim_dev->fw_update_status) {
 548                 devlink_flash_update_status_notify(devlink, "Flashing",
 549                                                    component,
 550                                                    NSIM_DEV_FLASH_SIZE,
 551                                                    NSIM_DEV_FLASH_SIZE);
 552                 devlink_flash_update_status_notify(devlink, "Flashing done",
 553                                                    component, 0, 0);
 554                 devlink_flash_update_end_notify(devlink);
 555         }
 556
 557         return 0;
 558 }
 559
 560 static struct nsim_trap_item *
 561 nsim_dev_trap_item_lookup(struct nsim_dev *nsim_dev, u16 trap_id)
 562 {
 563         struct nsim_trap_data *nsim_trap_data = nsim_dev->trap_data;
 564         int i;
 565
 566         for (i = 0; i < ARRAY_SIZE(nsim_traps_arr); i++) {
 567                 if (nsim_traps_arr[i].id == trap_id)
 568                         return &nsim_trap_data->trap_items_arr[i];
 569         }
 570
 571         return NULL;
 572 }
 573
 574 static int nsim_dev_devlink_trap_init(struct devlink *devlink,
 575                                       const struct devlink_trap *trap,
 576                                       void *trap_ctx)
 577 {
 578         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 579         struct nsim_trap_item *nsim_trap_item;
 580
 581         nsim_trap_item = nsim_dev_trap_item_lookup(nsim_dev, trap->id);
 582         if (WARN_ON(!nsim_trap_item))
 583                 return -ENOENT;
 584
 585         nsim_trap_item->trap_ctx = trap_ctx;
 586         nsim_trap_item->action = trap->init_action;
 587
 588         return 0;
 589 }
 590
 591 static int
 592 nsim_dev_devlink_trap_action_set(struct devlink *devlink,
 593                                  const struct devlink_trap *trap,
 594                                  enum devlink_trap_action action)
 595 {
 596         struct nsim_dev *nsim_dev = devlink_priv(devlink);
 597         struct nsim_trap_item *nsim_trap_item;
 598
 599         nsim_trap_item = nsim_dev_trap_item_lookup(nsim_dev, trap->id);
 600         if (WARN_ON(!nsim_trap_item))
 601                 return -ENOENT;
 602
 603         spin_lock(&nsim_dev->trap_data->trap_lock);
 604         nsim_trap_item->action = action;
 605         spin_unlock(&nsim_dev->trap_data->trap_lock);
 606
 607         return 0;
 608 }
 609
 610 static const struct devlink_ops nsim_dev_devlink_ops = {
 611         .reload_down = nsim_dev_reload_down,
 612         .reload_up = nsim_dev_reload_up,
 613         .info_get = nsim_dev_info_get,
 614         .flash_update = nsim_dev_flash_update,
 615         .trap_init = nsim_dev_devlink_trap_init,
 616         .trap_action_set = nsim_dev_devlink_trap_action_set,
 617 };
 618
 619 #define NSIM_DEV_MAX_MACS_DEFAULT 32
 620 #define NSIM_DEV_TEST1_DEFAULT true
 621
 622 static int __nsim_dev_port_add(struct nsim_dev *nsim_dev,
 623                                unsigned int port_index)
 624 {
 625         struct nsim_dev_port *nsim_dev_port;
 626         struct devlink_port *devlink_port;
 627         int err;
 628
 629         nsim_dev_port = kzalloc(sizeof(*nsim_dev_port), GFP_KERNEL);
 630         if (!nsim_dev_port)
 631                 return -ENOMEM;
 632         nsim_dev_port->port_index = port_index;
 633
 634         devlink_port = &nsim_dev_port->devlink_port;
 635         devlink_port_attrs_set(devlink_port, DEVLINK_PORT_FLAVOUR_PHYSICAL,
 636                                port_index + 1, 0, 0,
 637                                nsim_dev->switch_id.id,
 638                                nsim_dev->switch_id.id_len);
 639         err = devlink_port_register(priv_to_devlink(nsim_dev), devlink_port,
 640                                     port_index);
 641         if (err)
 642                 goto err_port_free;
 643
 644         err = nsim_dev_port_debugfs_init(nsim_dev, nsim_dev_port);
 645         if (err)
 646                 goto err_dl_port_unregister;
 647
 648         nsim_dev_port->ns = nsim_create(nsim_dev, nsim_dev_port);
 649         if (IS_ERR(nsim_dev_port->ns)) {
 650                 err = PTR_ERR(nsim_dev_port->ns);
 651                 goto err_port_debugfs_exit;
 652         }
 653
 654         devlink_port_type_eth_set(devlink_port, nsim_dev_port->ns->netdev);
 655         list_add(&nsim_dev_port->list, &nsim_dev->port_list);
 656
 657         return 0;
 658
 659 err_port_debugfs_exit:
 660         nsim_dev_port_debugfs_exit(nsim_dev_port);
 661 err_dl_port_unregister:
 662         devlink_port_unregister(devlink_port);
 663 err_port_free:
 664         kfree(nsim_dev_port);
 665         return err;
 666 }
 667
 668 static void __nsim_dev_port_del(struct nsim_dev_port *nsim_dev_port)
 669 {
 670         struct devlink_port *devlink_port = &nsim_dev_port->devlink_port;
 671
 672         list_del(&nsim_dev_port->list);
 673         devlink_port_type_clear(devlink_port);
 674         nsim_destroy(nsim_dev_port->ns);
 675         nsim_dev_port_debugfs_exit(nsim_dev_port);
 676         devlink_port_unregister(devlink_port);
 677         kfree(nsim_dev_port);
 678 }
 679
 680 static void nsim_dev_port_del_all(struct nsim_dev *nsim_dev)
 681 {
 682         struct nsim_dev_port *nsim_dev_port, *tmp;
 683
 684         list_for_each_entry_safe(nsim_dev_port, tmp,
 685                                  &nsim_dev->port_list, list)
 686                 __nsim_dev_port_del(nsim_dev_port);
 687 }
 688
 689 static int nsim_dev_port_add_all(struct nsim_dev *nsim_dev,
 690                                  unsigned int port_count)
 691 {
 692         int i, err;
 693
 694         for (i = 0; i < port_count; i++) {
 695                 err = __nsim_dev_port_add(nsim_dev, i);
 696                 if (err)
 697                         goto err_port_del_all;
 698         }
 699         return 0;
 700
 701 err_port_del_all:
 702         nsim_dev_port_del_all(nsim_dev);
 703         return err;
 704 }
 705
 706 static int nsim_dev_reload_create(struct nsim_dev *nsim_dev,
 707                                   struct netlink_ext_ack *extack)
 708 {
 709         struct nsim_bus_dev *nsim_bus_dev = nsim_dev->nsim_bus_dev;
 710         struct devlink *devlink;
 711         int err;
 712
 713         devlink = priv_to_devlink(nsim_dev);
 714         nsim_dev = devlink_priv(devlink);
 715         INIT_LIST_HEAD(&nsim_dev->port_list);
 716         mutex_init(&nsim_dev->port_list_lock);
 717         nsim_dev->fw_update_status = true;
 718
 719         nsim_dev->fib_data = nsim_fib_create(devlink, extack);
 720         if (IS_ERR(nsim_dev->fib_data))
 721                 return PTR_ERR(nsim_dev->fib_data);
 722
 723         nsim_devlink_param_load_driverinit_values(devlink);
 724
 725         err = nsim_dev_dummy_region_init(nsim_dev, devlink);
 726         if (err)
 727                 goto err_fib_destroy;
 728
 729         err = nsim_dev_traps_init(devlink);
 730         if (err)
 731                 goto err_dummy_region_exit;
 732
 733         err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count);
 734         if (err)
 735                 goto err_traps_exit;
 736
 737         return 0;
 738
 739 err_traps_exit:
 740         nsim_dev_traps_exit(devlink);
 741 err_dummy_region_exit:
 742         nsim_dev_dummy_region_exit(nsim_dev);
 743 err_fib_destroy:
 744         nsim_fib_destroy(devlink, nsim_dev->fib_data);
 745         return err;
 746 }
 747
 748 static struct nsim_dev *nsim_dev_create(struct nsim_bus_dev *nsim_bus_dev)
 749 {
 750         struct nsim_dev *nsim_dev;
 751         struct devlink *devlink;
 752         int err;
 753
 754         devlink = devlink_alloc(&nsim_dev_devlink_ops, sizeof(*nsim_dev));
 755         if (!devlink)
 756                 return ERR_PTR(-ENOMEM);
 757         devlink_net_set(devlink, nsim_bus_dev->initial_net);
 758         nsim_dev = devlink_priv(devlink);
 759         nsim_dev->nsim_bus_dev = nsim_bus_dev;
 760         nsim_dev->switch_id.id_len = sizeof(nsim_dev->switch_id.id);
 761         get_random_bytes(nsim_dev->switch_id.id, nsim_dev->switch_id.id_len);
 762         INIT_LIST_HEAD(&nsim_dev->port_list);
 763         mutex_init(&nsim_dev->port_list_lock);
 764         nsim_dev->fw_update_status = true;
 765         nsim_dev->max_macs = NSIM_DEV_MAX_MACS_DEFAULT;
 766         nsim_dev->test1 = NSIM_DEV_TEST1_DEFAULT;
 767
 768         err = nsim_dev_resources_register(devlink);
 769         if (err)
 770                 goto err_devlink_free;
 771
 772         nsim_dev->fib_data = nsim_fib_create(devlink, NULL);
 773         if (IS_ERR(nsim_dev->fib_data)) {
 774                 err = PTR_ERR(nsim_dev->fib_data);
 775                 goto err_resources_unregister;
 776         }
 777
 778         err = devlink_register(devlink, &nsim_bus_dev->dev);
 779         if (err)
 780                 goto err_fib_destroy;
 781
 782         err = devlink_params_register(devlink, nsim_devlink_params,
 783                                       ARRAY_SIZE(nsim_devlink_params));
 784         if (err)
 785                 goto err_dl_unregister;
 786         nsim_devlink_set_params_init_values(nsim_dev, devlink);
 787
 788         err = nsim_dev_dummy_region_init(nsim_dev, devlink);
 789         if (err)
 790                 goto err_params_unregister;
 791
 792         err = nsim_dev_traps_init(devlink);
 793         if (err)
 794                 goto err_dummy_region_exit;
 795
 796         err = nsim_dev_debugfs_init(nsim_dev);
 797         if (err)
 798                 goto err_traps_exit;
 799
 800         err = nsim_bpf_dev_init(nsim_dev);
 801         if (err)
 802                 goto err_debugfs_exit;
 803
 804         err = nsim_dev_port_add_all(nsim_dev, nsim_bus_dev->port_count);
 805         if (err)
 806                 goto err_bpf_dev_exit;
 807
 808         devlink_params_publish(devlink);
 809         return nsim_dev;
 810
 811 err_bpf_dev_exit:
 812         nsim_bpf_dev_exit(nsim_dev);
 813 err_debugfs_exit:
 814         nsim_dev_debugfs_exit(nsim_dev);
 815 err_traps_exit:
 816         nsim_dev_traps_exit(devlink);
 817 err_dummy_region_exit:
 818         nsim_dev_dummy_region_exit(nsim_dev);
 819 err_params_unregister:
 820         devlink_params_unregister(devlink, nsim_devlink_params,
 821                                   ARRAY_SIZE(nsim_devlink_params));
 822 err_dl_unregister:
 823         devlink_unregister(devlink);
 824 err_fib_destroy:
 825         nsim_fib_destroy(devlink, nsim_dev->fib_data);
 826 err_resources_unregister:
 827         devlink_resources_unregister(devlink, NULL);
 828 err_devlink_free:
 829         devlink_free(devlink);
 830         return ERR_PTR(err);
 831 }
 832
 833 static void nsim_dev_reload_destroy(struct nsim_dev *nsim_dev)
 834 {
 835         struct devlink *devlink = priv_to_devlink(nsim_dev);
 836
 837         if (devlink_is_reload_failed(devlink))
 838                 return;
 839         nsim_dev_port_del_all(nsim_dev);
 840         nsim_dev_traps_exit(devlink);
 841         nsim_dev_dummy_region_exit(nsim_dev);
 842         mutex_destroy(&nsim_dev->port_list_lock);
 843         nsim_fib_destroy(devlink, nsim_dev->fib_data);
 844 }
 845
 846 static void nsim_dev_destroy(struct nsim_dev *nsim_dev)
 847 {
 848         struct devlink *devlink = priv_to_devlink(nsim_dev);
 849
 850         nsim_dev_reload_destroy(nsim_dev);
 851
 852         nsim_bpf_dev_exit(nsim_dev);
 853         nsim_dev_debugfs_exit(nsim_dev);
 854         devlink_params_unregister(devlink, nsim_devlink_params,
 855                                   ARRAY_SIZE(nsim_devlink_params));
 856         devlink_unregister(devlink);
 857         devlink_resources_unregister(devlink, NULL);
 858         devlink_free(devlink);
 859 }
 860
 861 int nsim_dev_probe(struct nsim_bus_dev *nsim_bus_dev)
 862 {
 863         struct nsim_dev *nsim_dev;
 864
 865         nsim_dev = nsim_dev_create(nsim_bus_dev);
 866         if (IS_ERR(nsim_dev))
 867                 return PTR_ERR(nsim_dev);
 868         dev_set_drvdata(&nsim_bus_dev->dev, nsim_dev);
 869
 870         return 0;
 871 }
 872
 873 void nsim_dev_remove(struct nsim_bus_dev *nsim_bus_dev)
 874 {
 875         struct nsim_dev *nsim_dev = dev_get_drvdata(&nsim_bus_dev->dev);
 876
 877         nsim_dev_destroy(nsim_dev);
 878 }
 879
 880 static struct nsim_dev_port *
 881 __nsim_dev_port_lookup(struct nsim_dev *nsim_dev, unsigned int port_index)
 882 {
 883         struct nsim_dev_port *nsim_dev_port;
 884
 885         list_for_each_entry(nsim_dev_port, &nsim_dev->port_list, list)
 886                 if (nsim_dev_port->port_index == port_index)
 887                         return nsim_dev_port;
 888         return NULL;
 889 }
 890
 891 int nsim_dev_port_add(struct nsim_bus_dev *nsim_bus_dev,
 892                       unsigned int port_index)
 893 {
 894         struct nsim_dev *nsim_dev = dev_get_drvdata(&nsim_bus_dev->dev);
 895         int err;
 896
 897         mutex_lock(&nsim_dev->port_list_lock);
 898         if (__nsim_dev_port_lookup(nsim_dev, port_index))
 899                 err = -EEXIST;
 900         else
 901                 err = __nsim_dev_port_add(nsim_dev, port_index);
 902         mutex_unlock(&nsim_dev->port_list_lock);
 903         return err;
 904 }
 905
 906 int nsim_dev_port_del(struct nsim_bus_dev *nsim_bus_dev,
 907                       unsigned int port_index)
 908 {
 909         struct nsim_dev *nsim_dev = dev_get_drvdata(&nsim_bus_dev->dev);
 910         struct nsim_dev_port *nsim_dev_port;
 911         int err = 0;
 912
 913         mutex_lock(&nsim_dev->port_list_lock);
 914         nsim_dev_port = __nsim_dev_port_lookup(nsim_dev, port_index);
 915         if (!nsim_dev_port)
 916                 err = -ENOENT;
 917         else
 918                 __nsim_dev_port_del(nsim_dev_port);
 919         mutex_unlock(&nsim_dev->port_list_lock);
 920         return err;
 921 }
 922
 923 int nsim_dev_init(void)
 924 {
 925         nsim_dev_ddir = debugfs_create_dir(DRV_NAME, NULL);
 926         if (IS_ERR_OR_NULL(nsim_dev_ddir))
 927                 return -ENOMEM;
 928         return 0;
 929 }
 930
 931 void nsim_dev_exit(void)
 932 {
 933         debugfs_remove_recursive(nsim_dev_ddir);
 934 }