WARN_ON_ONCE(cmd != DEVLINK_CMD_TRAP_GROUP_NEW &&
                     cmd != DEVLINK_CMD_TRAP_GROUP_DEL);
-       if (!xa_get_mark(&devlinks, devlink->index, DEVLINK_REGISTERED))
-               return;
+       ASSERT_DEVLINK_REGISTERED(devlink);
 
        msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
        if (!msg)
        }
 
        list_add_tail(&group_item->list, &devlink->trap_group_list);
-       devlink_trap_group_notify(devlink, group_item,
-                                 DEVLINK_CMD_TRAP_GROUP_NEW);
-
        return 0;
 
 err_group_init:
        if (WARN_ON_ONCE(!group_item))
                return;
 
-       devlink_trap_group_notify(devlink, group_item,
-                                 DEVLINK_CMD_TRAP_GROUP_DEL);
        list_del(&group_item->list);
        free_percpu(group_item->stats);
        kfree(group_item);
 {
        int i, err;
 
+       ASSERT_DEVLINK_NOT_REGISTERED(devlink);
+
        mutex_lock(&devlink->lock);
        for (i = 0; i < groups_count; i++) {
                const struct devlink_trap_group *group = &groups[i];
 {
        int i;
 
+       ASSERT_DEVLINK_NOT_REGISTERED(devlink);
+
        mutex_lock(&devlink->lock);
        for (i = groups_count - 1; i >= 0; i--)
                devlink_trap_group_unregister(devlink, &groups[i]);