On Wed, 14 Feb 2024 17:50:44 +0000
Beau Belgrave <be...@linux.microsoft.com> wrote:

> +static char *user_event_group_system_multi_name(void)
> +{
> +     char *system_name;
> +     int len = sizeof(USER_EVENTS_MULTI_SYSTEM) + 1;

FYI, the sizeof() will include the "\0" so no need for "+ 1", but I don't
think this matters as for what I mention below.

> +
> +     system_name = kmalloc(len, GFP_KERNEL);
> +
> +     if (!system_name)
> +             return NULL;
> +
> +     snprintf(system_name, len, "%s", USER_EVENTS_MULTI_SYSTEM);
> +
> +     return system_name;

Hmm, the above looks like an open coded version of:

        system_name = kstrdup(USER_EVENTS_MULTI_SYSTEM, GFP_KERNEL);


> +}
> +
>  static struct user_event_group *current_user_event_group(void)
>  {
>       return init_group;
> @@ -367,6 +390,11 @@ static struct user_event_group 
> *user_event_group_create(void)
>       if (!group->system_name)
>               goto error;
>  
> +     group->system_multi_name = user_event_group_system_multi_name();
> +
> +     if (!group->system_multi_name)
> +             goto error;
> +
>       mutex_init(&group->reg_mutex);
>       hash_init(group->register_table);
>  
> @@ -1482,6 +1510,11 @@ static int destroy_user_event(struct user_event *user)
>       hash_del(&user->node);
>  
>       user_event_destroy_validators(user);
> +
> +     /* If we have different names, both must be freed */
> +     if (EVENT_NAME(user) != EVENT_TP_NAME(user))
> +             kfree(EVENT_TP_NAME(user));
> +
>       kfree(user->call.print_fmt);
>       kfree(EVENT_NAME(user));
>       kfree(user);
> @@ -1504,12 +1537,24 @@ static struct user_event *find_user_event(struct 
> user_event_group *group,
>       *outkey = key;
>  
>       hash_for_each_possible(group->register_table, user, node, key) {
> +             /*
> +              * Single-format events shouldn't return multi-format
> +              * events. Callers expect the underlying tracepoint to match
> +              * the name exactly in these cases. Only check like-formats.
> +              */
> +             if (EVENT_MULTI_FORMAT(flags) != 
> EVENT_MULTI_FORMAT(user->reg_flags))
> +                     continue;
> +
>               if (strcmp(EVENT_NAME(user), name))
>                       continue;
>  
>               if (user_fields_match(user, argc, argv))
>                       return user_event_get(user);
>  
> +             /* Scan others if this is a multi-format event */
> +             if (EVENT_MULTI_FORMAT(flags))
> +                     continue;
> +
>               return ERR_PTR(-EADDRINUSE);
>       }
>  
> @@ -1889,8 +1934,12 @@ static bool user_event_match(const char *system, const 
> char *event,
>       struct user_event *user = container_of(ev, struct user_event, devent);
>       bool match;
>  
> -     match = strcmp(EVENT_NAME(user), event) == 0 &&
> -             (!system || strcmp(system, USER_EVENTS_SYSTEM) == 0);
> +     match = strcmp(EVENT_NAME(user), event) == 0;
> +
> +     if (match && system) {
> +             match = strcmp(system, user->group->system_name) == 0 ||
> +                     strcmp(system, user->group->system_multi_name) == 0;
> +     }
>  
>       if (match)
>               match = user_fields_match(user, argc, argv);
> @@ -1923,6 +1972,39 @@ static int user_event_trace_register(struct user_event 
> *user)
>       return ret;
>  }
>  
> +static int user_event_set_tp_name(struct user_event *user)
> +{
> +     lockdep_assert_held(&user->group->reg_mutex);
> +
> +     if (EVENT_MULTI_FORMAT(user->reg_flags)) {
> +             char *multi_name;
> +             int len;
> +
> +             len = snprintf(NULL, 0, "%s.%llx", user->reg_name,
> +                            user->group->multi_id) + 1;
> +
> +             multi_name = kzalloc(len, GFP_KERNEL_ACCOUNT);
> +
> +             if (!multi_name)
> +                     return -ENOMEM;
> +
> +             snprintf(multi_name, len, "%s.%llx", user->reg_name,
> +                      user->group->multi_id);

I believe the above can be replaced with:

                multi_name = kasprintf(GFP_KERNEL_ACCOUNT, "%s.%llx", 
user->reg_name,
                                       user->group->multi_id);
                if (!multi_name)
                        return -ENOMEM;

-- Steve

> +
> +             user->call.name = multi_name;
> +             user->tracepoint.name = multi_name;
> +
> +             /* Inc to ensure unique multi-event name next time */
> +             user->group->multi_id++;
> +     } else {
> +             /* Non Multi-format uses register name */
> +             user->call.name = user->reg_name;
> +             user->tracepoint.name = user->reg_name;
> +     }
> +
> +     return 0;
> +}

Reply via email to