LCOV - code coverage report
Current view: top level - kernel/trace - trace_probe.c (source / functions) Hit Total Coverage
Test: combined.info Lines: 0 480 0.0 %
Date: 2022-03-28 13:20:08 Functions: 0 42 0.0 %
Branches: 0 335 0.0 %

           Branch data     Line data    Source code
       1                 :            : // SPDX-License-Identifier: GPL-2.0
       2                 :            : /*
       3                 :            :  * Common code for probe-based Dynamic events.
       4                 :            :  *
       5                 :            :  * This code was copied from kernel/trace/trace_kprobe.c written by
       6                 :            :  * Masami Hiramatsu <masami.hiramatsu.pt@hitachi.com>
       7                 :            :  *
       8                 :            :  * Updates to make this generic:
       9                 :            :  * Copyright (C) IBM Corporation, 2010-2011
      10                 :            :  * Author:     Srikar Dronamraju
      11                 :            :  */
      12                 :            : #define pr_fmt(fmt)     "trace_probe: " fmt
      13                 :            : 
      14                 :            : #include "trace_probe.h"
      15                 :            : 
      16                 :            : #undef C
      17                 :            : #define C(a, b)         b
      18                 :            : 
      19                 :            : static const char *trace_probe_err_text[] = { ERRORS };
      20                 :            : 
      21                 :            : static const char *reserved_field_names[] = {
      22                 :            :         "common_type",
      23                 :            :         "common_flags",
      24                 :            :         "common_preempt_count",
      25                 :            :         "common_pid",
      26                 :            :         "common_tgid",
      27                 :            :         FIELD_STRING_IP,
      28                 :            :         FIELD_STRING_RETIP,
      29                 :            :         FIELD_STRING_FUNC,
      30                 :            : };
      31                 :            : 
      32                 :            : /* Printing  in basic type function template */
      33                 :            : #define DEFINE_BASIC_PRINT_TYPE_FUNC(tname, type, fmt)                  \
      34                 :            : int PRINT_TYPE_FUNC_NAME(tname)(struct trace_seq *s, void *data, void *ent)\
      35                 :            : {                                                                       \
      36                 :            :         trace_seq_printf(s, fmt, *(type *)data);                        \
      37                 :            :         return !trace_seq_has_overflowed(s);                            \
      38                 :            : }                                                                       \
      39                 :            : const char PRINT_TYPE_FMT_NAME(tname)[] = fmt;
      40                 :            : 
      41         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(u8,  u8,  "%u")
      42         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(u16, u16, "%u")
      43         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(u32, u32, "%u")
      44         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(u64, u64, "%Lu")
      45         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(s8,  s8,  "%d")
      46         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(s16, s16, "%d")
      47         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(s32, s32, "%d")
      48         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(s64, s64, "%Ld")
      49         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(x8,  u8,  "0x%x")
      50         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(x16, u16, "0x%x")
      51         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(x32, u32, "0x%x")
      52         [ #  # ]:          0 : DEFINE_BASIC_PRINT_TYPE_FUNC(x64, u64, "0x%Lx")
      53                 :            : 
      54                 :          0 : int PRINT_TYPE_FUNC_NAME(symbol)(struct trace_seq *s, void *data, void *ent)
      55                 :            : {
      56                 :          0 :         trace_seq_printf(s, "%pS", (void *)*(unsigned long *)data);
      57         [ #  # ]:          0 :         return !trace_seq_has_overflowed(s);
      58                 :            : }
      59                 :            : const char PRINT_TYPE_FMT_NAME(symbol)[] = "%pS";
      60                 :            : 
      61                 :            : /* Print type function for string type */
      62                 :          0 : int PRINT_TYPE_FUNC_NAME(string)(struct trace_seq *s, void *data, void *ent)
      63                 :            : {
      64                 :          0 :         int len = *(u32 *)data >> 16;
      65                 :            : 
      66         [ #  # ]:          0 :         if (!len)
      67                 :          0 :                 trace_seq_puts(s, "(fault)");
      68                 :            :         else
      69                 :          0 :                 trace_seq_printf(s, "\"%s\"",
      70                 :            :                                  (const char *)get_loc_data(data, ent));
      71         [ #  # ]:          0 :         return !trace_seq_has_overflowed(s);
      72                 :            : }
      73                 :            : 
      74                 :            : const char PRINT_TYPE_FMT_NAME(string)[] = "\\\"%s\\\"";
      75                 :            : 
      76                 :            : /* Fetch type information table */
      77                 :            : static const struct fetch_type probe_fetch_types[] = {
      78                 :            :         /* Special types */
      79                 :            :         __ASSIGN_FETCH_TYPE("string", string, string, sizeof(u32), 1,
      80                 :            :                             "__data_loc char[]"),
      81                 :            :         __ASSIGN_FETCH_TYPE("ustring", string, string, sizeof(u32), 1,
      82                 :            :                             "__data_loc char[]"),
      83                 :            :         /* Basic types */
      84                 :            :         ASSIGN_FETCH_TYPE(u8,  u8,  0),
      85                 :            :         ASSIGN_FETCH_TYPE(u16, u16, 0),
      86                 :            :         ASSIGN_FETCH_TYPE(u32, u32, 0),
      87                 :            :         ASSIGN_FETCH_TYPE(u64, u64, 0),
      88                 :            :         ASSIGN_FETCH_TYPE(s8,  u8,  1),
      89                 :            :         ASSIGN_FETCH_TYPE(s16, u16, 1),
      90                 :            :         ASSIGN_FETCH_TYPE(s32, u32, 1),
      91                 :            :         ASSIGN_FETCH_TYPE(s64, u64, 1),
      92                 :            :         ASSIGN_FETCH_TYPE_ALIAS(x8,  u8,  u8,  0),
      93                 :            :         ASSIGN_FETCH_TYPE_ALIAS(x16, u16, u16, 0),
      94                 :            :         ASSIGN_FETCH_TYPE_ALIAS(x32, u32, u32, 0),
      95                 :            :         ASSIGN_FETCH_TYPE_ALIAS(x64, u64, u64, 0),
      96                 :            :         ASSIGN_FETCH_TYPE_ALIAS(symbol, ADDR_FETCH_TYPE, ADDR_FETCH_TYPE, 0),
      97                 :            : 
      98                 :            :         ASSIGN_FETCH_TYPE_END
      99                 :            : };
     100                 :            : 
     101                 :          0 : static const struct fetch_type *find_fetch_type(const char *type)
     102                 :            : {
     103                 :          0 :         int i;
     104                 :            : 
     105         [ #  # ]:          0 :         if (!type)
     106                 :          0 :                 type = DEFAULT_FETCH_TYPE_STR;
     107                 :            : 
     108                 :            :         /* Special case: bitfield */
     109         [ #  # ]:          0 :         if (*type == 'b') {
     110                 :          0 :                 unsigned long bs;
     111                 :            : 
     112                 :          0 :                 type = strchr(type, '/');
     113         [ #  # ]:          0 :                 if (!type)
     114                 :          0 :                         goto fail;
     115                 :            : 
     116                 :          0 :                 type++;
     117         [ #  # ]:          0 :                 if (kstrtoul(type, 0, &bs))
     118                 :          0 :                         goto fail;
     119                 :            : 
     120   [ #  #  #  #  :          0 :                 switch (bs) {
                      # ]
     121                 :          0 :                 case 8:
     122                 :          0 :                         return find_fetch_type("u8");
     123                 :          0 :                 case 16:
     124                 :          0 :                         return find_fetch_type("u16");
     125                 :          0 :                 case 32:
     126                 :          0 :                         return find_fetch_type("u32");
     127                 :          0 :                 case 64:
     128                 :          0 :                         return find_fetch_type("u64");
     129                 :          0 :                 default:
     130                 :          0 :                         goto fail;
     131                 :            :                 }
     132                 :            :         }
     133                 :            : 
     134         [ #  # ]:          0 :         for (i = 0; probe_fetch_types[i].name; i++) {
     135         [ #  # ]:          0 :                 if (strcmp(type, probe_fetch_types[i].name) == 0)
     136                 :          0 :                         return &probe_fetch_types[i];
     137                 :            :         }
     138                 :            : 
     139                 :          0 : fail:
     140                 :            :         return NULL;
     141                 :            : }
     142                 :            : 
     143                 :            : static struct trace_probe_log trace_probe_log;
     144                 :            : 
     145                 :          0 : void trace_probe_log_init(const char *subsystem, int argc, const char **argv)
     146                 :            : {
     147                 :          0 :         trace_probe_log.subsystem = subsystem;
     148                 :          0 :         trace_probe_log.argc = argc;
     149                 :          0 :         trace_probe_log.argv = argv;
     150                 :          0 :         trace_probe_log.index = 0;
     151                 :          0 : }
     152                 :            : 
     153                 :          0 : void trace_probe_log_clear(void)
     154                 :            : {
     155                 :          0 :         memset(&trace_probe_log, 0, sizeof(trace_probe_log));
     156                 :          0 : }
     157                 :            : 
     158                 :          0 : void trace_probe_log_set_index(int index)
     159                 :            : {
     160                 :          0 :         trace_probe_log.index = index;
     161                 :          0 : }
     162                 :            : 
     163                 :          0 : void __trace_probe_log_err(int offset, int err_type)
     164                 :            : {
     165                 :          0 :         char *command, *p;
     166                 :          0 :         int i, len = 0, pos = 0;
     167                 :            : 
     168         [ #  # ]:          0 :         if (!trace_probe_log.argv)
     169                 :            :                 return;
     170                 :            : 
     171                 :            :         /* Recalcurate the length and allocate buffer */
     172         [ #  # ]:          0 :         for (i = 0; i < trace_probe_log.argc; i++) {
     173         [ #  # ]:          0 :                 if (i == trace_probe_log.index)
     174                 :          0 :                         pos = len;
     175                 :          0 :                 len += strlen(trace_probe_log.argv[i]) + 1;
     176                 :            :         }
     177                 :          0 :         command = kzalloc(len, GFP_KERNEL);
     178         [ #  # ]:          0 :         if (!command)
     179                 :            :                 return;
     180                 :            : 
     181         [ #  # ]:          0 :         if (trace_probe_log.index >= trace_probe_log.argc) {
     182                 :            :                 /**
     183                 :            :                  * Set the error position is next to the last arg + space.
     184                 :            :                  * Note that len includes the terminal null and the cursor
     185                 :            :                  * appaers at pos + 1.
     186                 :            :                  */
     187                 :          0 :                 pos = len;
     188                 :          0 :                 offset = 0;
     189                 :            :         }
     190                 :            : 
     191                 :            :         /* And make a command string from argv array */
     192                 :          0 :         p = command;
     193         [ #  # ]:          0 :         for (i = 0; i < trace_probe_log.argc; i++) {
     194                 :          0 :                 len = strlen(trace_probe_log.argv[i]);
     195                 :          0 :                 strcpy(p, trace_probe_log.argv[i]);
     196                 :          0 :                 p[len] = ' ';
     197                 :          0 :                 p += len + 1;
     198                 :            :         }
     199                 :          0 :         *(p - 1) = '\0';
     200                 :            : 
     201                 :          0 :         tracing_log_err(NULL, trace_probe_log.subsystem, command,
     202                 :          0 :                         trace_probe_err_text, err_type, pos + offset);
     203                 :            : 
     204                 :          0 :         kfree(command);
     205                 :            : }
     206                 :            : 
     207                 :            : /* Split symbol and offset. */
     208                 :          0 : int traceprobe_split_symbol_offset(char *symbol, long *offset)
     209                 :            : {
     210                 :          0 :         char *tmp;
     211                 :          0 :         int ret;
     212                 :            : 
     213         [ #  # ]:          0 :         if (!offset)
     214                 :            :                 return -EINVAL;
     215                 :            : 
     216                 :          0 :         tmp = strpbrk(symbol, "+-");
     217         [ #  # ]:          0 :         if (tmp) {
     218                 :          0 :                 ret = kstrtol(tmp, 0, offset);
     219         [ #  # ]:          0 :                 if (ret)
     220                 :            :                         return ret;
     221                 :          0 :                 *tmp = '\0';
     222                 :            :         } else
     223                 :          0 :                 *offset = 0;
     224                 :            : 
     225                 :            :         return 0;
     226                 :            : }
     227                 :            : 
     228                 :            : /* @buf must has MAX_EVENT_NAME_LEN size */
     229                 :          0 : int traceprobe_parse_event_name(const char **pevent, const char **pgroup,
     230                 :            :                                 char *buf, int offset)
     231                 :            : {
     232                 :          0 :         const char *slash, *event = *pevent;
     233                 :          0 :         int len;
     234                 :            : 
     235                 :          0 :         slash = strchr(event, '/');
     236         [ #  # ]:          0 :         if (slash) {
     237         [ #  # ]:          0 :                 if (slash == event) {
     238                 :          0 :                         trace_probe_log_err(offset, NO_GROUP_NAME);
     239                 :          0 :                         return -EINVAL;
     240                 :            :                 }
     241         [ #  # ]:          0 :                 if (slash - event + 1 > MAX_EVENT_NAME_LEN) {
     242                 :          0 :                         trace_probe_log_err(offset, GROUP_TOO_LONG);
     243                 :          0 :                         return -EINVAL;
     244                 :            :                 }
     245                 :          0 :                 strlcpy(buf, event, slash - event + 1);
     246         [ #  # ]:          0 :                 if (!is_good_name(buf)) {
     247                 :          0 :                         trace_probe_log_err(offset, BAD_GROUP_NAME);
     248                 :          0 :                         return -EINVAL;
     249                 :            :                 }
     250                 :          0 :                 *pgroup = buf;
     251                 :          0 :                 *pevent = slash + 1;
     252                 :          0 :                 offset += slash - event + 1;
     253                 :          0 :                 event = *pevent;
     254                 :            :         }
     255                 :          0 :         len = strlen(event);
     256         [ #  # ]:          0 :         if (len == 0) {
     257                 :          0 :                 trace_probe_log_err(offset, NO_EVENT_NAME);
     258                 :          0 :                 return -EINVAL;
     259         [ #  # ]:          0 :         } else if (len > MAX_EVENT_NAME_LEN) {
     260                 :          0 :                 trace_probe_log_err(offset, EVENT_TOO_LONG);
     261                 :          0 :                 return -EINVAL;
     262                 :            :         }
     263         [ #  # ]:          0 :         if (!is_good_name(event)) {
     264                 :          0 :                 trace_probe_log_err(offset, BAD_EVENT_NAME);
     265                 :          0 :                 return -EINVAL;
     266                 :            :         }
     267                 :            :         return 0;
     268                 :            : }
     269                 :            : 
     270                 :            : #define PARAM_MAX_STACK (THREAD_SIZE / sizeof(unsigned long))
     271                 :            : 
     272                 :            : static int parse_probe_vars(char *arg, const struct fetch_type *t,
     273                 :            :                         struct fetch_insn *code, unsigned int flags, int offs)
     274                 :            : {
     275                 :            :         unsigned long param;
     276                 :            :         int ret = 0;
     277                 :            :         int len;
     278                 :            : 
     279                 :            :         if (strcmp(arg, "retval") == 0) {
     280                 :            :                 if (flags & TPARG_FL_RETURN) {
     281                 :            :                         code->op = FETCH_OP_RETVAL;
     282                 :            :                 } else {
     283                 :            :                         trace_probe_log_err(offs, RETVAL_ON_PROBE);
     284                 :            :                         ret = -EINVAL;
     285                 :            :                 }
     286                 :            :         } else if ((len = str_has_prefix(arg, "stack"))) {
     287                 :            :                 if (arg[len] == '\0') {
     288                 :            :                         code->op = FETCH_OP_STACKP;
     289                 :            :                 } else if (isdigit(arg[len])) {
     290                 :            :                         ret = kstrtoul(arg + len, 10, &param);
     291                 :            :                         if (ret) {
     292                 :            :                                 goto inval_var;
     293                 :            :                         } else if ((flags & TPARG_FL_KERNEL) &&
     294                 :            :                                     param > PARAM_MAX_STACK) {
     295                 :            :                                 trace_probe_log_err(offs, BAD_STACK_NUM);
     296                 :            :                                 ret = -EINVAL;
     297                 :            :                         } else {
     298                 :            :                                 code->op = FETCH_OP_STACK;
     299                 :            :                                 code->param = (unsigned int)param;
     300                 :            :                         }
     301                 :            :                 } else
     302                 :            :                         goto inval_var;
     303                 :            :         } else if (strcmp(arg, "comm") == 0) {
     304                 :            :                 code->op = FETCH_OP_COMM;
     305                 :            : #ifdef CONFIG_HAVE_FUNCTION_ARG_ACCESS_API
     306                 :            :         } else if (((flags & TPARG_FL_MASK) ==
     307                 :            :                     (TPARG_FL_KERNEL | TPARG_FL_FENTRY)) &&
     308                 :            :                    (len = str_has_prefix(arg, "arg"))) {
     309                 :            :                 ret = kstrtoul(arg + len, 10, &param);
     310                 :            :                 if (ret) {
     311                 :            :                         goto inval_var;
     312                 :            :                 } else if (!param || param > PARAM_MAX_STACK) {
     313                 :            :                         trace_probe_log_err(offs, BAD_ARG_NUM);
     314                 :            :                         return -EINVAL;
     315                 :            :                 }
     316                 :            :                 code->op = FETCH_OP_ARG;
     317                 :            :                 code->param = (unsigned int)param - 1;
     318                 :            : #endif
     319                 :            :         } else
     320                 :            :                 goto inval_var;
     321                 :            : 
     322                 :            :         return ret;
     323                 :            : 
     324                 :            : inval_var:
     325                 :            :         trace_probe_log_err(offs, BAD_VAR);
     326                 :            :         return -EINVAL;
     327                 :            : }
     328                 :            : 
     329                 :          0 : static int str_to_immediate(char *str, unsigned long *imm)
     330                 :            : {
     331         [ #  # ]:          0 :         if (isdigit(str[0]))
     332                 :          0 :                 return kstrtoul(str, 0, imm);
     333         [ #  # ]:          0 :         else if (str[0] == '-')
     334                 :          0 :                 return kstrtol(str, 0, (long *)imm);
     335         [ #  # ]:          0 :         else if (str[0] == '+')
     336                 :          0 :                 return kstrtol(str + 1, 0, (long *)imm);
     337                 :            :         return -EINVAL;
     338                 :            : }
     339                 :            : 
     340                 :          0 : static int __parse_imm_string(char *str, char **pbuf, int offs)
     341                 :            : {
     342                 :          0 :         size_t len = strlen(str);
     343                 :            : 
     344         [ #  # ]:          0 :         if (str[len - 1] != '"') {
     345                 :          0 :                 trace_probe_log_err(offs + len, IMMSTR_NO_CLOSE);
     346                 :          0 :                 return -EINVAL;
     347                 :            :         }
     348                 :          0 :         *pbuf = kstrndup(str, len - 1, GFP_KERNEL);
     349                 :          0 :         return 0;
     350                 :            : }
     351                 :            : 
     352                 :            : /* Recursive argument parser */
     353                 :            : static int
     354                 :            : parse_probe_arg(char *arg, const struct fetch_type *type,
     355                 :            :                 struct fetch_insn **pcode, struct fetch_insn *end,
     356                 :            :                 unsigned int flags, int offs)
     357                 :            : {
     358                 :            :         struct fetch_insn *code = *pcode;
     359                 :            :         unsigned long param;
     360                 :            :         int deref = FETCH_OP_DEREF;
     361                 :            :         long offset = 0;
     362                 :            :         char *tmp;
     363                 :            :         int ret = 0;
     364                 :            : 
     365                 :            :         switch (arg[0]) {
     366                 :            :         case '$':
     367                 :            :                 ret = parse_probe_vars(arg + 1, type, code, flags, offs);
     368                 :            :                 break;
     369                 :            : 
     370                 :            :         case '%':       /* named register */
     371                 :            :                 ret = regs_query_register_offset(arg + 1);
     372                 :            :                 if (ret >= 0) {
     373                 :            :                         code->op = FETCH_OP_REG;
     374                 :            :                         code->param = (unsigned int)ret;
     375                 :            :                         ret = 0;
     376                 :            :                 } else
     377                 :            :                         trace_probe_log_err(offs, BAD_REG_NAME);
     378                 :            :                 break;
     379                 :            : 
     380                 :            :         case '@':       /* memory, file-offset or symbol */
     381                 :            :                 if (isdigit(arg[1])) {
     382                 :            :                         ret = kstrtoul(arg + 1, 0, &param);
     383                 :            :                         if (ret) {
     384                 :            :                                 trace_probe_log_err(offs, BAD_MEM_ADDR);
     385                 :            :                                 break;
     386                 :            :                         }
     387                 :            :                         /* load address */
     388                 :            :                         code->op = FETCH_OP_IMM;
     389                 :            :                         code->immediate = param;
     390                 :            :                 } else if (arg[1] == '+') {
     391                 :            :                         /* kprobes don't support file offsets */
     392                 :            :                         if (flags & TPARG_FL_KERNEL) {
     393                 :            :                                 trace_probe_log_err(offs, FILE_ON_KPROBE);
     394                 :            :                                 return -EINVAL;
     395                 :            :                         }
     396                 :            :                         ret = kstrtol(arg + 2, 0, &offset);
     397                 :            :                         if (ret) {
     398                 :            :                                 trace_probe_log_err(offs, BAD_FILE_OFFS);
     399                 :            :                                 break;
     400                 :            :                         }
     401                 :            : 
     402                 :            :                         code->op = FETCH_OP_FOFFS;
     403                 :            :                         code->immediate = (unsigned long)offset;  // imm64?
     404                 :            :                 } else {
     405                 :            :                         /* uprobes don't support symbols */
     406                 :            :                         if (!(flags & TPARG_FL_KERNEL)) {
     407                 :            :                                 trace_probe_log_err(offs, SYM_ON_UPROBE);
     408                 :            :                                 return -EINVAL;
     409                 :            :                         }
     410                 :            :                         /* Preserve symbol for updating */
     411                 :            :                         code->op = FETCH_NOP_SYMBOL;
     412                 :            :                         code->data = kstrdup(arg + 1, GFP_KERNEL);
     413                 :            :                         if (!code->data)
     414                 :            :                                 return -ENOMEM;
     415                 :            :                         if (++code == end) {
     416                 :            :                                 trace_probe_log_err(offs, TOO_MANY_OPS);
     417                 :            :                                 return -EINVAL;
     418                 :            :                         }
     419                 :            :                         code->op = FETCH_OP_IMM;
     420                 :            :                         code->immediate = 0;
     421                 :            :                 }
     422                 :            :                 /* These are fetching from memory */
     423                 :            :                 if (++code == end) {
     424                 :            :                         trace_probe_log_err(offs, TOO_MANY_OPS);
     425                 :            :                         return -EINVAL;
     426                 :            :                 }
     427                 :            :                 *pcode = code;
     428                 :            :                 code->op = FETCH_OP_DEREF;
     429                 :            :                 code->offset = offset;
     430                 :            :                 break;
     431                 :            : 
     432                 :            :         case '+':       /* deref memory */
     433                 :            :         case '-':
     434                 :            :                 if (arg[1] == 'u') {
     435                 :            :                         deref = FETCH_OP_UDEREF;
     436                 :            :                         arg[1] = arg[0];
     437                 :            :                         arg++;
     438                 :            :                 }
     439                 :            :                 if (arg[0] == '+')
     440                 :            :                         arg++;  /* Skip '+', because kstrtol() rejects it. */
     441                 :            :                 tmp = strchr(arg, '(');
     442                 :            :                 if (!tmp) {
     443                 :            :                         trace_probe_log_err(offs, DEREF_NEED_BRACE);
     444                 :            :                         return -EINVAL;
     445                 :            :                 }
     446                 :            :                 *tmp = '\0';
     447                 :            :                 ret = kstrtol(arg, 0, &offset);
     448                 :            :                 if (ret) {
     449                 :            :                         trace_probe_log_err(offs, BAD_DEREF_OFFS);
     450                 :            :                         break;
     451                 :            :                 }
     452                 :            :                 offs += (tmp + 1 - arg) + (arg[0] != '-' ? 1 : 0);
     453                 :            :                 arg = tmp + 1;
     454                 :            :                 tmp = strrchr(arg, ')');
     455                 :            :                 if (!tmp) {
     456                 :            :                         trace_probe_log_err(offs + strlen(arg),
     457                 :            :                                             DEREF_OPEN_BRACE);
     458                 :            :                         return -EINVAL;
     459                 :            :                 } else {
     460                 :            :                         const struct fetch_type *t2 = find_fetch_type(NULL);
     461                 :            : 
     462                 :            :                         *tmp = '\0';
     463                 :            :                         ret = parse_probe_arg(arg, t2, &code, end, flags, offs);
     464                 :            :                         if (ret)
     465                 :            :                                 break;
     466                 :            :                         if (code->op == FETCH_OP_COMM ||
     467                 :            :                             code->op == FETCH_OP_DATA) {
     468                 :            :                                 trace_probe_log_err(offs, COMM_CANT_DEREF);
     469                 :            :                                 return -EINVAL;
     470                 :            :                         }
     471                 :            :                         if (++code == end) {
     472                 :            :                                 trace_probe_log_err(offs, TOO_MANY_OPS);
     473                 :            :                                 return -EINVAL;
     474                 :            :                         }
     475                 :            :                         *pcode = code;
     476                 :            : 
     477                 :            :                         code->op = deref;
     478                 :            :                         code->offset = offset;
     479                 :            :                 }
     480                 :            :                 break;
     481                 :            :         case '\\':      /* Immediate value */
     482                 :            :                 if (arg[1] == '"') {       /* Immediate string */
     483                 :            :                         ret = __parse_imm_string(arg + 2, &tmp, offs + 2);
     484                 :            :                         if (ret)
     485                 :            :                                 break;
     486                 :            :                         code->op = FETCH_OP_DATA;
     487                 :            :                         code->data = tmp;
     488                 :            :                 } else {
     489                 :            :                         ret = str_to_immediate(arg + 1, &code->immediate);
     490                 :            :                         if (ret)
     491                 :            :                                 trace_probe_log_err(offs + 1, BAD_IMM);
     492                 :            :                         else
     493                 :            :                                 code->op = FETCH_OP_IMM;
     494                 :            :                 }
     495                 :            :                 break;
     496                 :            :         }
     497                 :            :         if (!ret && code->op == FETCH_OP_NOP) {
     498                 :            :                 /* Parsed, but do not find fetch method */
     499                 :            :                 trace_probe_log_err(offs, BAD_FETCH_ARG);
     500                 :            :                 ret = -EINVAL;
     501                 :            :         }
     502                 :            :         return ret;
     503                 :            : }
     504                 :            : 
     505                 :            : #define BYTES_TO_BITS(nb)       ((BITS_PER_LONG * (nb)) / sizeof(long))
     506                 :            : 
     507                 :            : /* Bitfield type needs to be parsed into a fetch function */
     508                 :            : static int __parse_bitfield_probe_arg(const char *bf,
     509                 :            :                                       const struct fetch_type *t,
     510                 :            :                                       struct fetch_insn **pcode)
     511                 :            : {
     512                 :            :         struct fetch_insn *code = *pcode;
     513                 :            :         unsigned long bw, bo;
     514                 :            :         char *tail;
     515                 :            : 
     516                 :            :         if (*bf != 'b')
     517                 :            :                 return 0;
     518                 :            : 
     519                 :            :         bw = simple_strtoul(bf + 1, &tail, 0);      /* Use simple one */
     520                 :            : 
     521                 :            :         if (bw == 0 || *tail != '@')
     522                 :            :                 return -EINVAL;
     523                 :            : 
     524                 :            :         bf = tail + 1;
     525                 :            :         bo = simple_strtoul(bf, &tail, 0);
     526                 :            : 
     527                 :            :         if (tail == bf || *tail != '/')
     528                 :            :                 return -EINVAL;
     529                 :            :         code++;
     530                 :            :         if (code->op != FETCH_OP_NOP)
     531                 :            :                 return -EINVAL;
     532                 :            :         *pcode = code;
     533                 :            : 
     534                 :            :         code->op = FETCH_OP_MOD_BF;
     535                 :            :         code->lshift = BYTES_TO_BITS(t->size) - (bw + bo);
     536                 :            :         code->rshift = BYTES_TO_BITS(t->size) - bw;
     537                 :            :         code->basesize = t->size;
     538                 :            : 
     539                 :            :         return (BYTES_TO_BITS(t->size) < (bw + bo)) ? -EINVAL : 0;
     540                 :            : }
     541                 :            : 
     542                 :            : /* String length checking wrapper */
     543                 :          0 : static int traceprobe_parse_probe_arg_body(char *arg, ssize_t *size,
     544                 :            :                 struct probe_arg *parg, unsigned int flags, int offset)
     545                 :            : {
     546                 :          0 :         struct fetch_insn *code, *scode, *tmp = NULL;
     547                 :          0 :         char *t, *t2, *t3;
     548                 :          0 :         int ret, len;
     549                 :            : 
     550                 :          0 :         len = strlen(arg);
     551         [ #  # ]:          0 :         if (len > MAX_ARGSTR_LEN) {
     552                 :          0 :                 trace_probe_log_err(offset, ARG_TOO_LONG);
     553                 :          0 :                 return -EINVAL;
     554         [ #  # ]:          0 :         } else if (len == 0) {
     555                 :          0 :                 trace_probe_log_err(offset, NO_ARG_BODY);
     556                 :          0 :                 return -EINVAL;
     557                 :            :         }
     558                 :            : 
     559                 :          0 :         parg->comm = kstrdup(arg, GFP_KERNEL);
     560         [ #  # ]:          0 :         if (!parg->comm)
     561                 :            :                 return -ENOMEM;
     562                 :            : 
     563                 :          0 :         t = strchr(arg, ':');
     564         [ #  # ]:          0 :         if (t) {
     565                 :          0 :                 *t = '\0';
     566                 :          0 :                 t2 = strchr(++t, '[');
     567         [ #  # ]:          0 :                 if (t2) {
     568                 :          0 :                         *t2++ = '\0';
     569                 :          0 :                         t3 = strchr(t2, ']');
     570         [ #  # ]:          0 :                         if (!t3) {
     571                 :          0 :                                 offset += t2 + strlen(t2) - arg;
     572                 :          0 :                                 trace_probe_log_err(offset,
     573                 :            :                                                     ARRAY_NO_CLOSE);
     574                 :          0 :                                 return -EINVAL;
     575         [ #  # ]:          0 :                         } else if (t3[1] != '\0') {
     576                 :          0 :                                 trace_probe_log_err(offset + t3 + 1 - arg,
     577                 :            :                                                     BAD_ARRAY_SUFFIX);
     578                 :          0 :                                 return -EINVAL;
     579                 :            :                         }
     580                 :          0 :                         *t3 = '\0';
     581   [ #  #  #  # ]:          0 :                         if (kstrtouint(t2, 0, &parg->count) || !parg->count) {
     582                 :          0 :                                 trace_probe_log_err(offset + t2 - arg,
     583                 :            :                                                     BAD_ARRAY_NUM);
     584                 :          0 :                                 return -EINVAL;
     585                 :            :                         }
     586         [ #  # ]:          0 :                         if (parg->count > MAX_ARRAY_LEN) {
     587                 :          0 :                                 trace_probe_log_err(offset + t2 - arg,
     588                 :            :                                                     ARRAY_TOO_BIG);
     589                 :          0 :                                 return -EINVAL;
     590                 :            :                         }
     591                 :            :                 }
     592                 :            :         }
     593                 :            : 
     594                 :            :         /*
     595                 :            :          * Since $comm and immediate string can not be dereferred,
     596                 :            :          * we can find those by strcmp.
     597                 :            :          */
     598   [ #  #  #  # ]:          0 :         if (strcmp(arg, "$comm") == 0 || strncmp(arg, "\\\"", 2) == 0) {
     599                 :            :                 /* The type of $comm must be "string", and not an array. */
     600   [ #  #  #  #  :          0 :                 if (parg->count || (t && strcmp(t, "string")))
                   #  # ]
     601                 :            :                         return -EINVAL;
     602                 :          0 :                 parg->type = find_fetch_type("string");
     603                 :            :         } else
     604                 :          0 :                 parg->type = find_fetch_type(t);
     605         [ #  # ]:          0 :         if (!parg->type) {
     606         [ #  # ]:          0 :                 trace_probe_log_err(offset + (t ? (t - arg) : 0), BAD_TYPE);
     607                 :          0 :                 return -EINVAL;
     608                 :            :         }
     609                 :          0 :         parg->offset = *size;
     610                 :          0 :         *size += parg->type->size * (parg->count ?: 1);
     611                 :            : 
     612         [ #  # ]:          0 :         if (parg->count) {
     613                 :          0 :                 len = strlen(parg->type->fmttype) + 6;
     614         [ #  # ]:          0 :                 parg->fmt = kmalloc(len, GFP_KERNEL);
     615         [ #  # ]:          0 :                 if (!parg->fmt)
     616                 :            :                         return -ENOMEM;
     617                 :          0 :                 snprintf(parg->fmt, len, "%s[%d]", parg->type->fmttype,
     618                 :            :                          parg->count);
     619                 :            :         }
     620                 :            : 
     621                 :          0 :         code = tmp = kcalloc(FETCH_INSN_MAX, sizeof(*code), GFP_KERNEL);
     622         [ #  # ]:          0 :         if (!code)
     623                 :            :                 return -ENOMEM;
     624                 :          0 :         code[FETCH_INSN_MAX - 1].op = FETCH_OP_END;
     625                 :            : 
     626                 :          0 :         ret = parse_probe_arg(arg, parg->type, &code, &code[FETCH_INSN_MAX - 1],
     627                 :            :                               flags, offset);
     628         [ #  # ]:          0 :         if (ret)
     629                 :          0 :                 goto fail;
     630                 :            : 
     631                 :            :         /* Store operation */
     632         [ #  # ]:          0 :         if (!strcmp(parg->type->name, "string") ||
     633         [ #  # ]:          0 :             !strcmp(parg->type->name, "ustring")) {
     634         [ #  # ]:          0 :                 if (code->op != FETCH_OP_DEREF && code->op != FETCH_OP_UDEREF &&
     635   [ #  #  #  # ]:          0 :                     code->op != FETCH_OP_IMM && code->op != FETCH_OP_COMM &&
     636                 :            :                     code->op != FETCH_OP_DATA) {
     637         [ #  # ]:          0 :                         trace_probe_log_err(offset + (t ? (t - arg) : 0),
     638                 :            :                                             BAD_STRING);
     639                 :          0 :                         ret = -EINVAL;
     640                 :          0 :                         goto fail;
     641                 :            :                 }
     642         [ #  # ]:          0 :                 if ((code->op == FETCH_OP_IMM || code->op == FETCH_OP_COMM) ||
     643         [ #  # ]:          0 :                      parg->count) {
     644                 :            :                         /*
     645                 :            :                          * IMM, DATA and COMM is pointing actual address, those
     646                 :            :                          * must be kept, and if parg->count != 0, this is an
     647                 :            :                          * array of string pointers instead of string address
     648                 :            :                          * itself.
     649                 :            :                          */
     650                 :          0 :                         code++;
     651         [ #  # ]:          0 :                         if (code->op != FETCH_OP_NOP) {
     652                 :          0 :                                 trace_probe_log_err(offset, TOO_MANY_OPS);
     653                 :          0 :                                 ret = -EINVAL;
     654                 :          0 :                                 goto fail;
     655                 :            :                         }
     656                 :            :                 }
     657                 :            :                 /* If op == DEREF, replace it with STRING */
     658         [ #  # ]:          0 :                 if (!strcmp(parg->type->name, "ustring") ||
     659         [ #  # ]:          0 :                     code->op == FETCH_OP_UDEREF)
     660                 :          0 :                         code->op = FETCH_OP_ST_USTRING;
     661                 :            :                 else
     662                 :          0 :                         code->op = FETCH_OP_ST_STRING;
     663                 :          0 :                 code->size = parg->type->size;
     664                 :          0 :                 parg->dynamic = true;
     665         [ #  # ]:          0 :         } else if (code->op == FETCH_OP_DEREF) {
     666                 :          0 :                 code->op = FETCH_OP_ST_MEM;
     667                 :          0 :                 code->size = parg->type->size;
     668         [ #  # ]:          0 :         } else if (code->op == FETCH_OP_UDEREF) {
     669                 :          0 :                 code->op = FETCH_OP_ST_UMEM;
     670                 :          0 :                 code->size = parg->type->size;
     671                 :            :         } else {
     672                 :          0 :                 code++;
     673         [ #  # ]:          0 :                 if (code->op != FETCH_OP_NOP) {
     674                 :          0 :                         trace_probe_log_err(offset, TOO_MANY_OPS);
     675                 :          0 :                         ret = -EINVAL;
     676                 :          0 :                         goto fail;
     677                 :            :                 }
     678                 :          0 :                 code->op = FETCH_OP_ST_RAW;
     679                 :          0 :                 code->size = parg->type->size;
     680                 :            :         }
     681                 :          0 :         scode = code;
     682                 :            :         /* Modify operation */
     683         [ #  # ]:          0 :         if (t != NULL) {
     684                 :          0 :                 ret = __parse_bitfield_probe_arg(t, parg->type, &code);
     685         [ #  # ]:          0 :                 if (ret) {
     686                 :          0 :                         trace_probe_log_err(offset + t - arg, BAD_BITFIELD);
     687                 :          0 :                         goto fail;
     688                 :            :                 }
     689                 :            :         }
     690                 :            :         /* Loop(Array) operation */
     691         [ #  # ]:          0 :         if (parg->count) {
     692         [ #  # ]:          0 :                 if (scode->op != FETCH_OP_ST_MEM &&
     693         [ #  # ]:          0 :                     scode->op != FETCH_OP_ST_STRING &&
     694                 :            :                     scode->op != FETCH_OP_ST_USTRING) {
     695         [ #  # ]:          0 :                         trace_probe_log_err(offset + (t ? (t - arg) : 0),
     696                 :            :                                             BAD_STRING);
     697                 :          0 :                         ret = -EINVAL;
     698                 :          0 :                         goto fail;
     699                 :            :                 }
     700                 :          0 :                 code++;
     701         [ #  # ]:          0 :                 if (code->op != FETCH_OP_NOP) {
     702                 :          0 :                         trace_probe_log_err(offset, TOO_MANY_OPS);
     703                 :          0 :                         ret = -EINVAL;
     704                 :          0 :                         goto fail;
     705                 :            :                 }
     706                 :          0 :                 code->op = FETCH_OP_LP_ARRAY;
     707                 :          0 :                 code->param = parg->count;
     708                 :            :         }
     709                 :          0 :         code++;
     710                 :          0 :         code->op = FETCH_OP_END;
     711                 :            : 
     712                 :            :         /* Shrink down the code buffer */
     713                 :          0 :         parg->code = kcalloc(code - tmp + 1, sizeof(*code), GFP_KERNEL);
     714         [ #  # ]:          0 :         if (!parg->code)
     715                 :            :                 ret = -ENOMEM;
     716                 :            :         else
     717                 :          0 :                 memcpy(parg->code, tmp, sizeof(*code) * (code - tmp + 1));
     718                 :            : 
     719                 :          0 : fail:
     720         [ #  # ]:          0 :         if (ret) {
     721         [ #  # ]:          0 :                 for (code = tmp; code < tmp + FETCH_INSN_MAX; code++)
     722         [ #  # ]:          0 :                         if (code->op == FETCH_NOP_SYMBOL ||
     723                 :            :                             code->op == FETCH_OP_DATA)
     724                 :          0 :                                 kfree(code->data);
     725                 :            :         }
     726                 :          0 :         kfree(tmp);
     727                 :            : 
     728                 :          0 :         return ret;
     729                 :            : }
     730                 :            : 
     731                 :            : /* Return 1 if name is reserved or already used by another argument */
     732                 :          0 : static int traceprobe_conflict_field_name(const char *name,
     733                 :            :                                           struct probe_arg *args, int narg)
     734                 :            : {
     735                 :          0 :         int i;
     736                 :            : 
     737         [ #  # ]:          0 :         for (i = 0; i < ARRAY_SIZE(reserved_field_names); i++)
     738         [ #  # ]:          0 :                 if (strcmp(reserved_field_names[i], name) == 0)
     739                 :            :                         return 1;
     740                 :            : 
     741         [ #  # ]:          0 :         for (i = 0; i < narg; i++)
     742         [ #  # ]:          0 :                 if (strcmp(args[i].name, name) == 0)
     743                 :            :                         return 1;
     744                 :            : 
     745                 :            :         return 0;
     746                 :            : }
     747                 :            : 
     748                 :          0 : int traceprobe_parse_probe_arg(struct trace_probe *tp, int i, char *arg,
     749                 :            :                                 unsigned int flags)
     750                 :            : {
     751                 :          0 :         struct probe_arg *parg = &tp->args[i];
     752                 :          0 :         char *body;
     753                 :            : 
     754                 :            :         /* Increment count for freeing args in error case */
     755                 :          0 :         tp->nr_args++;
     756                 :            : 
     757                 :          0 :         body = strchr(arg, '=');
     758         [ #  # ]:          0 :         if (body) {
     759         [ #  # ]:          0 :                 if (body - arg > MAX_ARG_NAME_LEN) {
     760                 :          0 :                         trace_probe_log_err(0, ARG_NAME_TOO_LONG);
     761                 :          0 :                         return -EINVAL;
     762         [ #  # ]:          0 :                 } else if (body == arg) {
     763                 :          0 :                         trace_probe_log_err(0, NO_ARG_NAME);
     764                 :          0 :                         return -EINVAL;
     765                 :            :                 }
     766                 :          0 :                 parg->name = kmemdup_nul(arg, body - arg, GFP_KERNEL);
     767                 :          0 :                 body++;
     768                 :            :         } else {
     769                 :            :                 /* If argument name is omitted, set "argN" */
     770                 :          0 :                 parg->name = kasprintf(GFP_KERNEL, "arg%d", i + 1);
     771                 :          0 :                 body = arg;
     772                 :            :         }
     773         [ #  # ]:          0 :         if (!parg->name)
     774                 :            :                 return -ENOMEM;
     775                 :            : 
     776         [ #  # ]:          0 :         if (!is_good_name(parg->name)) {
     777                 :          0 :                 trace_probe_log_err(0, BAD_ARG_NAME);
     778                 :          0 :                 return -EINVAL;
     779                 :            :         }
     780         [ #  # ]:          0 :         if (traceprobe_conflict_field_name(parg->name, tp->args, i)) {
     781                 :          0 :                 trace_probe_log_err(0, USED_ARG_NAME);
     782                 :          0 :                 return -EINVAL;
     783                 :            :         }
     784                 :            :         /* Parse fetch argument */
     785                 :          0 :         return traceprobe_parse_probe_arg_body(body, &tp->size, parg, flags,
     786                 :          0 :                                                body - arg);
     787                 :            : }
     788                 :            : 
     789                 :          0 : void traceprobe_free_probe_arg(struct probe_arg *arg)
     790                 :            : {
     791                 :          0 :         struct fetch_insn *code = arg->code;
     792                 :            : 
     793   [ #  #  #  # ]:          0 :         while (code && code->op != FETCH_OP_END) {
     794         [ #  # ]:          0 :                 if (code->op == FETCH_NOP_SYMBOL ||
     795                 :            :                     code->op == FETCH_OP_DATA)
     796                 :          0 :                         kfree(code->data);
     797                 :          0 :                 code++;
     798                 :            :         }
     799                 :          0 :         kfree(arg->code);
     800                 :          0 :         kfree(arg->name);
     801                 :          0 :         kfree(arg->comm);
     802                 :          0 :         kfree(arg->fmt);
     803                 :          0 : }
     804                 :            : 
     805                 :          0 : int traceprobe_update_arg(struct probe_arg *arg)
     806                 :            : {
     807                 :          0 :         struct fetch_insn *code = arg->code;
     808                 :          0 :         long offset;
     809                 :          0 :         char *tmp;
     810                 :          0 :         char c;
     811                 :          0 :         int ret = 0;
     812                 :            : 
     813   [ #  #  #  # ]:          0 :         while (code && code->op != FETCH_OP_END) {
     814         [ #  # ]:          0 :                 if (code->op == FETCH_NOP_SYMBOL) {
     815         [ #  # ]:          0 :                         if (code[1].op != FETCH_OP_IMM)
     816                 :            :                                 return -EINVAL;
     817                 :            : 
     818                 :          0 :                         tmp = strpbrk(code->data, "+-");
     819         [ #  # ]:          0 :                         if (tmp)
     820                 :          0 :                                 c = *tmp;
     821                 :          0 :                         ret = traceprobe_split_symbol_offset(code->data,
     822                 :            :                                                              &offset);
     823         [ #  # ]:          0 :                         if (ret)
     824                 :          0 :                                 return ret;
     825                 :            : 
     826                 :          0 :                         code[1].immediate =
     827                 :          0 :                                 (unsigned long)kallsyms_lookup_name(code->data);
     828         [ #  # ]:          0 :                         if (tmp)
     829                 :          0 :                                 *tmp = c;
     830         [ #  # ]:          0 :                         if (!code[1].immediate)
     831                 :            :                                 return -ENOENT;
     832                 :          0 :                         code[1].immediate += offset;
     833                 :            :                 }
     834                 :          0 :                 code++;
     835                 :            :         }
     836                 :            :         return 0;
     837                 :            : }
     838                 :            : 
     839                 :            : /* When len=0, we just calculate the needed length */
     840                 :            : #define LEN_OR_ZERO (len ? len - pos : 0)
     841                 :          0 : static int __set_print_fmt(struct trace_probe *tp, char *buf, int len,
     842                 :            :                            bool is_return)
     843                 :            : {
     844                 :          0 :         struct probe_arg *parg;
     845                 :          0 :         int i, j;
     846                 :          0 :         int pos = 0;
     847                 :          0 :         const char *fmt, *arg;
     848                 :            : 
     849         [ #  # ]:          0 :         if (!is_return) {
     850                 :            :                 fmt = "(%lx)";
     851                 :            :                 arg = "REC->" FIELD_STRING_IP;
     852                 :            :         } else {
     853                 :          0 :                 fmt = "(%lx <- %lx)";
     854                 :          0 :                 arg = "REC->" FIELD_STRING_FUNC ", REC->" FIELD_STRING_RETIP;
     855                 :            :         }
     856                 :            : 
     857         [ #  # ]:          0 :         pos += snprintf(buf + pos, LEN_OR_ZERO, "\"%s", fmt);
     858                 :            : 
     859         [ #  # ]:          0 :         for (i = 0; i < tp->nr_args; i++) {
     860                 :          0 :                 parg = tp->args + i;
     861         [ #  # ]:          0 :                 pos += snprintf(buf + pos, LEN_OR_ZERO, " %s=", parg->name);
     862         [ #  # ]:          0 :                 if (parg->count) {
     863                 :          0 :                         pos += snprintf(buf + pos, LEN_OR_ZERO, "{%s",
     864         [ #  # ]:          0 :                                         parg->type->fmt);
     865         [ #  # ]:          0 :                         for (j = 1; j < parg->count; j++)
     866                 :          0 :                                 pos += snprintf(buf + pos, LEN_OR_ZERO, ",%s",
     867         [ #  # ]:          0 :                                                 parg->type->fmt);
     868         [ #  # ]:          0 :                         pos += snprintf(buf + pos, LEN_OR_ZERO, "}");
     869                 :            :                 } else
     870                 :          0 :                         pos += snprintf(buf + pos, LEN_OR_ZERO, "%s",
     871         [ #  # ]:          0 :                                         parg->type->fmt);
     872                 :            :         }
     873                 :            : 
     874         [ #  # ]:          0 :         pos += snprintf(buf + pos, LEN_OR_ZERO, "\", %s", arg);
     875                 :            : 
     876         [ #  # ]:          0 :         for (i = 0; i < tp->nr_args; i++) {
     877                 :          0 :                 parg = tp->args + i;
     878         [ #  # ]:          0 :                 if (parg->count) {
     879         [ #  # ]:          0 :                         if ((strcmp(parg->type->name, "string") == 0) ||
     880         [ #  # ]:          0 :                             (strcmp(parg->type->name, "ustring") == 0))
     881                 :            :                                 fmt = ", __get_str(%s[%d])";
     882                 :            :                         else
     883                 :          0 :                                 fmt = ", REC->%s[%d]";
     884         [ #  # ]:          0 :                         for (j = 0; j < parg->count; j++)
     885         [ #  # ]:          0 :                                 pos += snprintf(buf + pos, LEN_OR_ZERO,
     886                 :            :                                                 fmt, parg->name, j);
     887                 :            :                 } else {
     888         [ #  # ]:          0 :                         if ((strcmp(parg->type->name, "string") == 0) ||
     889         [ #  # ]:          0 :                             (strcmp(parg->type->name, "ustring") == 0))
     890                 :            :                                 fmt = ", __get_str(%s)";
     891                 :            :                         else
     892                 :          0 :                                 fmt = ", REC->%s";
     893         [ #  # ]:          0 :                         pos += snprintf(buf + pos, LEN_OR_ZERO,
     894                 :            :                                         fmt, parg->name);
     895                 :            :                 }
     896                 :            :         }
     897                 :            : 
     898                 :            :         /* return the length of print_fmt */
     899                 :          0 :         return pos;
     900                 :            : }
     901                 :            : #undef LEN_OR_ZERO
     902                 :            : 
     903                 :          0 : int traceprobe_set_print_fmt(struct trace_probe *tp, bool is_return)
     904                 :            : {
     905                 :          0 :         struct trace_event_call *call = trace_probe_event_call(tp);
     906                 :          0 :         int len;
     907                 :          0 :         char *print_fmt;
     908                 :            : 
     909                 :            :         /* First: called with 0 length to calculate the needed length */
     910                 :          0 :         len = __set_print_fmt(tp, NULL, 0, is_return);
     911         [ #  # ]:          0 :         print_fmt = kmalloc(len + 1, GFP_KERNEL);
     912         [ #  # ]:          0 :         if (!print_fmt)
     913                 :            :                 return -ENOMEM;
     914                 :            : 
     915                 :            :         /* Second: actually write the @print_fmt */
     916                 :          0 :         __set_print_fmt(tp, print_fmt, len + 1, is_return);
     917                 :          0 :         call->print_fmt = print_fmt;
     918                 :            : 
     919                 :          0 :         return 0;
     920                 :            : }
     921                 :            : 
     922                 :          0 : int traceprobe_define_arg_fields(struct trace_event_call *event_call,
     923                 :            :                                  size_t offset, struct trace_probe *tp)
     924                 :            : {
     925                 :          0 :         int ret, i;
     926                 :            : 
     927                 :            :         /* Set argument names as fields */
     928         [ #  # ]:          0 :         for (i = 0; i < tp->nr_args; i++) {
     929                 :          0 :                 struct probe_arg *parg = &tp->args[i];
     930                 :          0 :                 const char *fmt = parg->type->fmttype;
     931                 :          0 :                 int size = parg->type->size;
     932                 :            : 
     933         [ #  # ]:          0 :                 if (parg->fmt)
     934                 :          0 :                         fmt = parg->fmt;
     935         [ #  # ]:          0 :                 if (parg->count)
     936                 :          0 :                         size *= parg->count;
     937                 :          0 :                 ret = trace_define_field(event_call, fmt, parg->name,
     938                 :          0 :                                          offset + parg->offset, size,
     939                 :            :                                          parg->type->is_signed,
     940                 :            :                                          FILTER_OTHER);
     941         [ #  # ]:          0 :                 if (ret)
     942                 :          0 :                         return ret;
     943                 :            :         }
     944                 :            :         return 0;
     945                 :            : }
     946                 :            : 
     947                 :          0 : static void trace_probe_event_free(struct trace_probe_event *tpe)
     948                 :            : {
     949                 :          0 :         kfree(tpe->class.system);
     950                 :          0 :         kfree(tpe->call.name);
     951                 :          0 :         kfree(tpe->call.print_fmt);
     952                 :          0 :         kfree(tpe);
     953                 :          0 : }
     954                 :            : 
     955                 :          0 : int trace_probe_append(struct trace_probe *tp, struct trace_probe *to)
     956                 :            : {
     957   [ #  #  #  # ]:          0 :         if (trace_probe_has_sibling(tp))
     958                 :            :                 return -EBUSY;
     959                 :            : 
     960                 :          0 :         list_del_init(&tp->list);
     961                 :          0 :         trace_probe_event_free(tp->event);
     962                 :            : 
     963                 :          0 :         tp->event = to->event;
     964                 :          0 :         list_add_tail(&tp->list, trace_probe_probe_list(to));
     965                 :            : 
     966                 :          0 :         return 0;
     967                 :            : }
     968                 :            : 
     969                 :          0 : void trace_probe_unlink(struct trace_probe *tp)
     970                 :            : {
     971         [ #  # ]:          0 :         list_del_init(&tp->list);
     972         [ #  # ]:          0 :         if (list_empty(trace_probe_probe_list(tp)))
     973                 :          0 :                 trace_probe_event_free(tp->event);
     974                 :          0 :         tp->event = NULL;
     975                 :          0 : }
     976                 :            : 
     977                 :          0 : void trace_probe_cleanup(struct trace_probe *tp)
     978                 :            : {
     979                 :          0 :         int i;
     980                 :            : 
     981         [ #  # ]:          0 :         for (i = 0; i < tp->nr_args; i++)
     982                 :          0 :                 traceprobe_free_probe_arg(&tp->args[i]);
     983                 :            : 
     984         [ #  # ]:          0 :         if (tp->event)
     985                 :          0 :                 trace_probe_unlink(tp);
     986                 :          0 : }
     987                 :            : 
     988                 :          0 : int trace_probe_init(struct trace_probe *tp, const char *event,
     989                 :            :                      const char *group, bool alloc_filter)
     990                 :            : {
     991                 :          0 :         struct trace_event_call *call;
     992                 :          0 :         size_t size = sizeof(struct trace_probe_event);
     993                 :          0 :         int ret = 0;
     994                 :            : 
     995         [ #  # ]:          0 :         if (!event || !group)
     996                 :            :                 return -EINVAL;
     997                 :            : 
     998         [ #  # ]:          0 :         if (alloc_filter)
     999                 :          0 :                 size += sizeof(struct trace_uprobe_filter);
    1000                 :            : 
    1001                 :          0 :         tp->event = kzalloc(size, GFP_KERNEL);
    1002         [ #  # ]:          0 :         if (!tp->event)
    1003                 :            :                 return -ENOMEM;
    1004                 :            : 
    1005                 :          0 :         INIT_LIST_HEAD(&tp->event->files);
    1006                 :          0 :         INIT_LIST_HEAD(&tp->event->class.fields);
    1007                 :          0 :         INIT_LIST_HEAD(&tp->event->probes);
    1008                 :          0 :         INIT_LIST_HEAD(&tp->list);
    1009                 :          0 :         list_add(&tp->event->probes, &tp->list);
    1010                 :            : 
    1011                 :          0 :         call = trace_probe_event_call(tp);
    1012                 :          0 :         call->class = &tp->event->class;
    1013                 :          0 :         call->name = kstrdup(event, GFP_KERNEL);
    1014         [ #  # ]:          0 :         if (!call->name) {
    1015                 :          0 :                 ret = -ENOMEM;
    1016                 :          0 :                 goto error;
    1017                 :            :         }
    1018                 :            : 
    1019                 :          0 :         tp->event->class.system = kstrdup(group, GFP_KERNEL);
    1020         [ #  # ]:          0 :         if (!tp->event->class.system) {
    1021                 :          0 :                 ret = -ENOMEM;
    1022                 :          0 :                 goto error;
    1023                 :            :         }
    1024                 :            : 
    1025                 :            :         return 0;
    1026                 :            : 
    1027                 :          0 : error:
    1028                 :          0 :         trace_probe_cleanup(tp);
    1029                 :          0 :         return ret;
    1030                 :            : }
    1031                 :            : 
    1032                 :          0 : int trace_probe_register_event_call(struct trace_probe *tp)
    1033                 :            : {
    1034                 :          0 :         struct trace_event_call *call = trace_probe_event_call(tp);
    1035                 :          0 :         int ret;
    1036                 :            : 
    1037                 :          0 :         ret = register_trace_event(&call->event);
    1038         [ #  # ]:          0 :         if (!ret)
    1039                 :            :                 return -ENODEV;
    1040                 :            : 
    1041                 :          0 :         ret = trace_add_event_call(call);
    1042         [ #  # ]:          0 :         if (ret)
    1043                 :          0 :                 unregister_trace_event(&call->event);
    1044                 :            : 
    1045                 :            :         return ret;
    1046                 :            : }
    1047                 :            : 
    1048                 :          0 : int trace_probe_add_file(struct trace_probe *tp, struct trace_event_file *file)
    1049                 :            : {
    1050                 :          0 :         struct event_file_link *link;
    1051                 :            : 
    1052                 :          0 :         link = kmalloc(sizeof(*link), GFP_KERNEL);
    1053         [ #  # ]:          0 :         if (!link)
    1054                 :            :                 return -ENOMEM;
    1055                 :            : 
    1056                 :          0 :         link->file = file;
    1057                 :          0 :         INIT_LIST_HEAD(&link->list);
    1058                 :          0 :         list_add_tail_rcu(&link->list, &tp->event->files);
    1059                 :          0 :         trace_probe_set_flag(tp, TP_FLAG_TRACE);
    1060                 :          0 :         return 0;
    1061                 :            : }
    1062                 :            : 
    1063                 :          0 : struct event_file_link *trace_probe_get_file_link(struct trace_probe *tp,
    1064                 :            :                                                   struct trace_event_file *file)
    1065                 :            : {
    1066                 :          0 :         struct event_file_link *link;
    1067                 :            : 
    1068   [ #  #  #  # ]:          0 :         trace_probe_for_each_link(link, tp) {
    1069   [ #  #  #  # ]:          0 :                 if (link->file == file)
    1070                 :          0 :                         return link;
    1071                 :            :         }
    1072                 :            : 
    1073                 :            :         return NULL;
    1074                 :            : }
    1075                 :            : 
    1076                 :          0 : int trace_probe_remove_file(struct trace_probe *tp,
    1077                 :            :                             struct trace_event_file *file)
    1078                 :            : {
    1079                 :          0 :         struct event_file_link *link;
    1080                 :            : 
    1081                 :          0 :         link = trace_probe_get_file_link(tp, file);
    1082         [ #  # ]:          0 :         if (!link)
    1083                 :            :                 return -ENOENT;
    1084                 :            : 
    1085                 :          0 :         list_del_rcu(&link->list);
    1086                 :          0 :         synchronize_rcu();
    1087                 :          0 :         kfree(link);
    1088                 :            : 
    1089         [ #  # ]:          0 :         if (list_empty(&tp->event->files))
    1090                 :          0 :                 trace_probe_clear_flag(tp, TP_FLAG_TRACE);
    1091                 :            : 
    1092                 :            :         return 0;
    1093                 :            : }
    1094                 :            : 
    1095                 :            : /*
    1096                 :            :  * Return the smallest index of different type argument (start from 1).
    1097                 :            :  * If all argument types and name are same, return 0.
    1098                 :            :  */
    1099                 :          0 : int trace_probe_compare_arg_type(struct trace_probe *a, struct trace_probe *b)
    1100                 :            : {
    1101                 :          0 :         int i;
    1102                 :            : 
    1103                 :            :         /* In case of more arguments */
    1104         [ #  # ]:          0 :         if (a->nr_args < b->nr_args)
    1105                 :          0 :                 return a->nr_args + 1;
    1106         [ #  # ]:          0 :         if (a->nr_args > b->nr_args)
    1107                 :          0 :                 return b->nr_args + 1;
    1108                 :            : 
    1109         [ #  # ]:          0 :         for (i = 0; i < a->nr_args; i++) {
    1110         [ #  # ]:          0 :                 if ((b->nr_args <= i) ||
    1111         [ #  # ]:          0 :                     ((a->args[i].type != b->args[i].type) ||
    1112         [ #  # ]:          0 :                      (a->args[i].count != b->args[i].count) ||
    1113         [ #  # ]:          0 :                      strcmp(a->args[i].name, b->args[i].name)))
    1114                 :          0 :                         return i + 1;
    1115                 :            :         }
    1116                 :            : 
    1117                 :            :         return 0;
    1118                 :            : }
    1119                 :            : 
    1120                 :          0 : bool trace_probe_match_command_args(struct trace_probe *tp,
    1121                 :            :                                     int argc, const char **argv)
    1122                 :            : {
    1123                 :          0 :         char buf[MAX_ARGSTR_LEN + 1];
    1124                 :          0 :         int i;
    1125                 :            : 
    1126         [ #  # ]:          0 :         if (tp->nr_args < argc)
    1127                 :            :                 return false;
    1128                 :            : 
    1129         [ #  # ]:          0 :         for (i = 0; i < argc; i++) {
    1130                 :          0 :                 snprintf(buf, sizeof(buf), "%s=%s",
    1131                 :            :                          tp->args[i].name, tp->args[i].comm);
    1132         [ #  # ]:          0 :                 if (strcmp(buf, argv[i]))
    1133                 :            :                         return false;
    1134                 :            :         }
    1135                 :            :         return true;
    1136                 :            : }

Generated by: LCOV version 1.14