lib/dynamic_debug.c

   1 /*
   2  * lib/dynamic_debug.c
   3  *
   4  * make pr_debug()/dev_dbg() calls runtime configurable based upon their
   5  * source module.
   6  *
   7  * Copyright (C) 2008 Jason Baron <jbaron@redhat.com>
   8  * By Greg Banks <gnb@melbourne.sgi.com>
   9  * Copyright (c) 2008 Silicon Graphics Inc.  All Rights Reserved.
  10  * Copyright (C) 2011 Bart Van Assche.  All Rights Reserved.
  11  */
  12
  13 #define pr_fmt(fmt) KBUILD_MODNAME ":%s: " fmt, __func__
  14
  15 #include <linux/kernel.h>
  16 #include <linux/module.h>
  17 #include <linux/moduleparam.h>
  18 #include <linux/kallsyms.h>
  19 #include <linux/types.h>
  20 #include <linux/mutex.h>
  21 #include <linux/proc_fs.h>
  22 #include <linux/seq_file.h>
  23 #include <linux/list.h>
  24 #include <linux/sysctl.h>
  25 #include <linux/ctype.h>
  26 #include <linux/string.h>
  27 #include <linux/uaccess.h>
  28 #include <linux/dynamic_debug.h>
  29 #include <linux/debugfs.h>
  30 #include <linux/slab.h>
  31 #include <linux/jump_label.h>
  32 #include <linux/hardirq.h>
  33 #include <linux/sched.h>
  34 #include <linux/device.h>
  35 #include <linux/netdevice.h>
  36
  37 extern struct _ddebug __start___verbose[];
  38 extern struct _ddebug __stop___verbose[];
  39
  40 struct ddebug_table {
  41         struct list_head link;
  42         char *mod_name;
  43         unsigned int num_ddebugs;
  44         unsigned int num_enabled;
  45         struct _ddebug *ddebugs;
  46 };
  47
  48 struct ddebug_query {
  49         const char *filename;
  50         const char *module;
  51         const char *function;
  52         const char *format;
  53         unsigned int first_lineno, last_lineno;
  54 };
  55
  56 struct ddebug_iter {
  57         struct ddebug_table *table;
  58         unsigned int idx;
  59 };
  60
  61 static DEFINE_MUTEX(ddebug_lock);
  62 static LIST_HEAD(ddebug_tables);
  63 static int verbose = 0;
  64
  65 /* Return the last part of a pathname */
  66 static inline const char *basename(const char *path)
  67 {
  68         const char *tail = strrchr(path, '/');
  69         return tail ? tail+1 : path;
  70 }
  71
  72 static struct { unsigned flag:8; char opt_char; } opt_array[] = {
  73         { _DPRINTK_FLAGS_PRINT, 'p' },
  74         { _DPRINTK_FLAGS_INCL_MODNAME, 'm' },
  75         { _DPRINTK_FLAGS_INCL_FUNCNAME, 'f' },
  76         { _DPRINTK_FLAGS_INCL_LINENO, 'l' },
  77         { _DPRINTK_FLAGS_INCL_TID, 't' },
  78 };
  79
  80 /* format a string into buf[] which describes the _ddebug's flags */
  81 static char *ddebug_describe_flags(struct _ddebug *dp, char *buf,
  82                                     size_t maxlen)
  83 {
  84         char *p = buf;
  85         int i;
  86
  87         BUG_ON(maxlen < 4);
  88         for (i = 0; i < ARRAY_SIZE(opt_array); ++i)
  89                 if (dp->flags & opt_array[i].flag)
  90                         *p++ = opt_array[i].opt_char;
  91         if (p == buf)
  92                 *p++ = '-';
  93         *p = '\0';
  94
  95         return buf;
  96 }
  97
  98 /*
  99  * Search the tables for _ddebug's which match the given
 100  * `query' and apply the `flags' and `mask' to them.  Tells
 101  * the user which ddebug's were changed, or whether none
 102  * were matched.
 103  */
 104 static void ddebug_change(const struct ddebug_query *query,
 105                            unsigned int flags, unsigned int mask)
 106 {
 107         int i;
 108         struct ddebug_table *dt;
 109         unsigned int newflags;
 110         unsigned int nfound = 0;
 111         char flagbuf[8];
 112
 113         /* search for matching ddebugs */
 114         mutex_lock(&ddebug_lock);
 115         list_for_each_entry(dt, &ddebug_tables, link) {
 116
 117                 /* match against the module name */
 118                 if (query->module != NULL &&
 119                     strcmp(query->module, dt->mod_name))
 120                         continue;
 121
 122                 for (i = 0 ; i < dt->num_ddebugs ; i++) {
 123                         struct _ddebug *dp = &dt->ddebugs[i];
 124
 125                         /* match against the source filename */
 126                         if (query->filename != NULL &&
 127                             strcmp(query->filename, dp->filename) &&
 128                             strcmp(query->filename, basename(dp->filename)))
 129                                 continue;
 130
 131                         /* match against the function */
 132                         if (query->function != NULL &&
 133                             strcmp(query->function, dp->function))
 134                                 continue;
 135
 136                         /* match against the format */
 137                         if (query->format != NULL &&
 138                             strstr(dp->format, query->format) == NULL)
 139                                 continue;
 140
 141                         /* match against the line number range */
 142                         if (query->first_lineno &&
 143                             dp->lineno < query->first_lineno)
 144                                 continue;
 145                         if (query->last_lineno &&
 146                             dp->lineno > query->last_lineno)
 147                                 continue;
 148
 149                         nfound++;
 150
 151                         newflags = (dp->flags & mask) | flags;
 152                         if (newflags == dp->flags)
 153                                 continue;
 154
 155                         if (!newflags)
 156                                 dt->num_enabled--;
 157                         else if (!dp->flags)
 158                                 dt->num_enabled++;
 159                         dp->flags = newflags;
 160                         if (newflags)
 161                                 dp->enabled = 1;
 162                         else
 163                                 dp->enabled = 0;
 164                         if (verbose)
 165                                 pr_info("changed %s:%d [%s]%s %s\n",
 166                                         dp->filename, dp->lineno,
 167                                         dt->mod_name, dp->function,
 168                                         ddebug_describe_flags(dp, flagbuf,
 169                                                         sizeof(flagbuf)));
 170                 }
 171         }
 172         mutex_unlock(&ddebug_lock);
 173
 174         if (!nfound && verbose)
 175                 pr_info("no matches for query\n");
 176 }
 177
 178 /*
 179  * Split the buffer `buf' into space-separated words.
 180  * Handles simple " and ' quoting, i.e. without nested,
 181  * embedded or escaped \".  Return the number of words
 182  * or <0 on error.
 183  */
 184 static int ddebug_tokenize(char *buf, char *words[], int maxwords)
 185 {
 186         int nwords = 0;
 187
 188         while (*buf) {
 189                 char *end;
 190
 191                 /* Skip leading whitespace */
 192                 buf = skip_spaces(buf);
 193                 if (!*buf)
 194                         break;  /* oh, it was trailing whitespace */
 195
 196                 /* Run `end' over a word, either whitespace separated or quoted */
 197                 if (*buf == '"' || *buf == '\'') {
 198                         int quote = *buf++;
 199                         for (end = buf ; *end && *end != quote ; end++)
 200                                 ;
 201                         if (!*end)
 202                                 return -EINVAL; /* unclosed quote */
 203                 } else {
 204                         for (end = buf ; *end && !isspace(*end) ; end++)
 205                                 ;
 206                         BUG_ON(end == buf);
 207                 }
 208                 /* Here `buf' is the start of the word, `end' is one past the end */
 209
 210                 if (nwords == maxwords)
 211                         return -EINVAL; /* ran out of words[] before bytes */
 212                 if (*end)
 213                         *end++ = '\0';  /* terminate the word */
 214                 words[nwords++] = buf;
 215                 buf = end;
 216         }
 217
 218         if (verbose) {
 219                 int i;
 220                 pr_info("split into words:");
 221                 for (i = 0 ; i < nwords ; i++)
 222                         pr_cont(" \"%s\"", words[i]);
 223                 pr_cont("\n");
 224         }
 225
 226         return nwords;
 227 }
 228
 229 /*
 230  * Parse a single line number.  Note that the empty string ""
 231  * is treated as a special case and converted to zero, which
 232  * is later treated as a "don't care" value.
 233  */
 234 static inline int parse_lineno(const char *str, unsigned int *val)
 235 {
 236         char *end = NULL;
 237         BUG_ON(str == NULL);
 238         if (*str == '\0') {
 239                 *val = 0;
 240                 return 0;
 241         }
 242         *val = simple_strtoul(str, &end, 10);
 243         return end == NULL || end == str || *end != '\0' ? -EINVAL : 0;
 244 }
 245
 246 /*
 247  * Undo octal escaping in a string, inplace.  This is useful to
 248  * allow the user to express a query which matches a format
 249  * containing embedded spaces.
 250  */
 251 #define isodigit(c)             ((c) >= '0' && (c) <= '7')
 252 static char *unescape(char *str)
 253 {
 254         char *in = str;
 255         char *out = str;
 256
 257         while (*in) {
 258                 if (*in == '\\') {
 259                         if (in[1] == '\\') {
 260                                 *out++ = '\\';
 261                                 in += 2;
 262                                 continue;
 263                         } else if (in[1] == 't') {
 264                                 *out++ = '\t';
 265                                 in += 2;
 266                                 continue;
 267                         } else if (in[1] == 'n') {
 268                                 *out++ = '\n';
 269                                 in += 2;
 270                                 continue;
 271                         } else if (isodigit(in[1]) &&
 272                                  isodigit(in[2]) &&
 273                                  isodigit(in[3])) {
 274                                 *out++ = ((in[1] - '0')<<6) |
 275                                           ((in[2] - '0')<<3) |
 276                                           (in[3] - '0');
 277                                 in += 4;
 278                                 continue;
 279                         }
 280                 }
 281                 *out++ = *in++;
 282         }
 283         *out = '\0';
 284
 285         return str;
 286 }
 287
 288 /*
 289  * Parse words[] as a ddebug query specification, which is a series
 290  * of (keyword, value) pairs chosen from these possibilities:
 291  *
 292  * func <function-name>
 293  * file <full-pathname>
 294  * file <base-filename>
 295  * module <module-name>
 296  * format <escaped-string-to-find-in-format>
 297  * line <lineno>
 298  * line <first-lineno>-<last-lineno> // where either may be empty
 299  */
 300 static int ddebug_parse_query(char *words[], int nwords,
 301                                struct ddebug_query *query)
 302 {
 303         unsigned int i;
 304
 305         /* check we have an even number of words */
 306         if (nwords % 2 != 0)
 307                 return -EINVAL;
 308         memset(query, 0, sizeof(*query));
 309
 310         for (i = 0 ; i < nwords ; i += 2) {
 311                 if (!strcmp(words[i], "func"))
 312                         query->function = words[i+1];
 313                 else if (!strcmp(words[i], "file"))
 314                         query->filename = words[i+1];
 315                 else if (!strcmp(words[i], "module"))
 316                         query->module = words[i+1];
 317                 else if (!strcmp(words[i], "format"))
 318                         query->format = unescape(words[i+1]);
 319                 else if (!strcmp(words[i], "line")) {
 320                         char *first = words[i+1];
 321                         char *last = strchr(first, '-');
 322                         if (last)
 323                                 *last++ = '\0';
 324                         if (parse_lineno(first, &query->first_lineno) < 0)
 325                                 return -EINVAL;
 326                         if (last != NULL) {
 327                                 /* range <first>-<last> */
 328                                 if (parse_lineno(last, &query->last_lineno) < 0)
 329                                         return -EINVAL;
 330                         } else {
 331                                 query->last_lineno = query->first_lineno;
 332                         }
 333                 } else {
 334                         if (verbose)
 335                                 pr_err("unknown keyword \"%s\"\n", words[i]);
 336                         return -EINVAL;
 337                 }
 338         }
 339
 340         if (verbose)
 341                 pr_info("q->function=\"%s\" q->filename=\"%s\" "
 342                         "q->module=\"%s\" q->format=\"%s\" q->lineno=%u-%u\n",
 343                         query->function, query->filename,
 344                         query->module, query->format, query->first_lineno,
 345                         query->last_lineno);
 346
 347         return 0;
 348 }
 349
 350 /*
 351  * Parse `str' as a flags specification, format [-+=][p]+.
 352  * Sets up *maskp and *flagsp to be used when changing the
 353  * flags fields of matched _ddebug's.  Returns 0 on success
 354  * or <0 on error.
 355  */
 356 static int ddebug_parse_flags(const char *str, unsigned int *flagsp,
 357                                unsigned int *maskp)
 358 {
 359         unsigned flags = 0;
 360         int op = '=', i;
 361
 362         switch (*str) {
 363         case '+':
 364         case '-':
 365         case '=':
 366                 op = *str++;
 367                 break;
 368         default:
 369                 return -EINVAL;
 370         }
 371         if (verbose)
 372                 pr_info("op='%c'\n", op);
 373
 374         for ( ; *str ; ++str) {
 375                 for (i = ARRAY_SIZE(opt_array) - 1; i >= 0; i--) {
 376                         if (*str == opt_array[i].opt_char) {
 377                                 flags |= opt_array[i].flag;
 378                                 break;
 379                         }
 380                 }
 381                 if (i < 0)
 382                         return -EINVAL;
 383         }
 384         if (flags == 0)
 385                 return -EINVAL;
 386         if (verbose)
 387                 pr_info("flags=0x%x\n", flags);
 388
 389         /* calculate final *flagsp, *maskp according to mask and op */
 390         switch (op) {
 391         case '=':
 392                 *maskp = 0;
 393                 *flagsp = flags;
 394                 break;
 395         case '+':
 396                 *maskp = ~0U;
 397                 *flagsp = flags;
 398                 break;
 399         case '-':
 400                 *maskp = ~flags;
 401                 *flagsp = 0;
 402                 break;
 403         }
 404         if (verbose)
 405                 pr_info("*flagsp=0x%x *maskp=0x%x\n", *flagsp, *maskp);
 406         return 0;
 407 }
 408
 409 static int ddebug_exec_query(char *query_string)
 410 {
 411         unsigned int flags = 0, mask = 0;
 412         struct ddebug_query query;
 413 #define MAXWORDS 9
 414         int nwords;
 415         char *words[MAXWORDS];
 416
 417         nwords = ddebug_tokenize(query_string, words, MAXWORDS);
 418         if (nwords <= 0)
 419                 return -EINVAL;
 420         if (ddebug_parse_query(words, nwords-1, &query))
 421                 return -EINVAL;
 422         if (ddebug_parse_flags(words[nwords-1], &flags, &mask))
 423                 return -EINVAL;
 424
 425         /* actually go and implement the change */
 426         ddebug_change(&query, flags, mask);
 427         return 0;
 428 }
 429
 430 static int dynamic_emit_prefix(const struct _ddebug *descriptor)
 431 {
 432         char tid[sizeof(int) + sizeof(int)/2 + 4];
 433         char lineno[sizeof(int) + sizeof(int)/2];
 434
 435         if (descriptor->flags & _DPRINTK_FLAGS_INCL_TID) {
 436                 if (in_interrupt())
 437                         snprintf(tid, sizeof(tid), "%s", "<intr> ");
 438                 else
 439                         snprintf(tid, sizeof(tid), "[%d] ",
 440                                  task_pid_vnr(current));
 441         } else {
 442                 tid[0] = 0;
 443         }
 444
 445         if (descriptor->flags & _DPRINTK_FLAGS_INCL_LINENO)
 446                 snprintf(lineno, sizeof(lineno), "%d", descriptor->lineno);
 447         else
 448                 lineno[0] = 0;
 449
 450         return printk(KERN_DEBUG "%s%s%s%s%s%s",
 451                       tid,
 452                       (descriptor->flags & _DPRINTK_FLAGS_INCL_MODNAME) ?
 453                       descriptor->modname : "",
 454                       (descriptor->flags & _DPRINTK_FLAGS_INCL_MODNAME) ?
 455                       ":" : "",
 456                       (descriptor->flags & _DPRINTK_FLAGS_INCL_FUNCNAME) ?
 457                       descriptor->function : "",
 458                       (descriptor->flags & _DPRINTK_FLAGS_INCL_FUNCNAME) ?
 459                       ":" : "",
 460                       lineno);
 461 }
 462
 463 int __dynamic_pr_debug(struct _ddebug *descriptor, const char *fmt, ...)
 464 {
 465         va_list args;
 466         int res;
 467
 468         BUG_ON(!descriptor);
 469         BUG_ON(!fmt);
 470
 471         va_start(args, fmt);
 472
 473         res = dynamic_emit_prefix(descriptor);
 474         res += vprintk(fmt, args);
 475
 476         va_end(args);
 477
 478         return res;
 479 }
 480 EXPORT_SYMBOL(__dynamic_pr_debug);
 481
 482 int __dynamic_dev_dbg(struct _ddebug *descriptor,
 483                       const struct device *dev, const char *fmt, ...)
 484 {
 485         struct va_format vaf;
 486         va_list args;
 487         int res;
 488
 489         BUG_ON(!descriptor);
 490         BUG_ON(!fmt);
 491
 492         va_start(args, fmt);
 493
 494         vaf.fmt = fmt;
 495         vaf.va = &args;
 496
 497         res = dynamic_emit_prefix(descriptor);
 498         res += __dev_printk(KERN_CONT, dev, &vaf);
 499
 500         va_end(args);
 501
 502         return res;
 503 }
 504 EXPORT_SYMBOL(__dynamic_dev_dbg);
 505
 506 int __dynamic_netdev_dbg(struct _ddebug *descriptor,
 507                       const struct net_device *dev, const char *fmt, ...)
 508 {
 509         struct va_format vaf;
 510         va_list args;
 511         int res;
 512
 513         BUG_ON(!descriptor);
 514         BUG_ON(!fmt);
 515
 516         va_start(args, fmt);
 517
 518         vaf.fmt = fmt;
 519         vaf.va = &args;
 520
 521         res = dynamic_emit_prefix(descriptor);
 522         res += __netdev_printk(KERN_CONT, dev, &vaf);
 523
 524         va_end(args);
 525
 526         return res;
 527 }
 528 EXPORT_SYMBOL(__dynamic_netdev_dbg);
 529
 530 static __initdata char ddebug_setup_string[1024];
 531 static __init int ddebug_setup_query(char *str)
 532 {
 533         if (strlen(str) >= 1024) {
 534                 pr_warn("ddebug boot param string too large\n");
 535                 return 0;
 536         }
 537         strcpy(ddebug_setup_string, str);
 538         return 1;
 539 }
 540
 541 __setup("ddebug_query=", ddebug_setup_query);
 542
 543 /*
 544  * File_ops->write method for <debugfs>/dynamic_debug/conrol.  Gathers the
 545  * command text from userspace, parses and executes it.
 546  */
 547 static ssize_t ddebug_proc_write(struct file *file, const char __user *ubuf,
 548                                   size_t len, loff_t *offp)
 549 {
 550         char tmpbuf[256];
 551         int ret;
 552
 553         if (len == 0)
 554                 return 0;
 555         /* we don't check *offp -- multiple writes() are allowed */
 556         if (len > sizeof(tmpbuf)-1)
 557                 return -E2BIG;
 558         if (copy_from_user(tmpbuf, ubuf, len))
 559                 return -EFAULT;
 560         tmpbuf[len] = '\0';
 561         if (verbose)
 562                 pr_info("read %d bytes from userspace\n", (int)len);
 563
 564         ret = ddebug_exec_query(tmpbuf);
 565         if (ret)
 566                 return ret;
 567
 568         *offp += len;
 569         return len;
 570 }
 571
 572 /*
 573  * Set the iterator to point to the first _ddebug object
 574  * and return a pointer to that first object.  Returns
 575  * NULL if there are no _ddebugs at all.
 576  */
 577 static struct _ddebug *ddebug_iter_first(struct ddebug_iter *iter)
 578 {
 579         if (list_empty(&ddebug_tables)) {
 580                 iter->table = NULL;
 581                 iter->idx = 0;
 582                 return NULL;
 583         }
 584         iter->table = list_entry(ddebug_tables.next,
 585                                  struct ddebug_table, link);
 586         iter->idx = 0;
 587         return &iter->table->ddebugs[iter->idx];
 588 }
 589
 590 /*
 591  * Advance the iterator to point to the next _ddebug
 592  * object from the one the iterator currently points at,
 593  * and returns a pointer to the new _ddebug.  Returns
 594  * NULL if the iterator has seen all the _ddebugs.
 595  */
 596 static struct _ddebug *ddebug_iter_next(struct ddebug_iter *iter)
 597 {
 598         if (iter->table == NULL)
 599                 return NULL;
 600         if (++iter->idx == iter->table->num_ddebugs) {
 601                 /* iterate to next table */
 602                 iter->idx = 0;
 603                 if (list_is_last(&iter->table->link, &ddebug_tables)) {
 604                         iter->table = NULL;
 605                         return NULL;
 606                 }
 607                 iter->table = list_entry(iter->table->link.next,
 608                                          struct ddebug_table, link);
 609         }
 610         return &iter->table->ddebugs[iter->idx];
 611 }
 612
 613 /*
 614  * Seq_ops start method.  Called at the start of every
 615  * read() call from userspace.  Takes the ddebug_lock and
 616  * seeks the seq_file's iterator to the given position.
 617  */
 618 static void *ddebug_proc_start(struct seq_file *m, loff_t *pos)
 619 {
 620         struct ddebug_iter *iter = m->private;
 621         struct _ddebug *dp;
 622         int n = *pos;
 623
 624         if (verbose)
 625                 pr_info("called m=%p *pos=%lld\n", m, (unsigned long long)*pos);
 626
 627         mutex_lock(&ddebug_lock);
 628
 629         if (!n)
 630                 return SEQ_START_TOKEN;
 631         if (n < 0)
 632                 return NULL;
 633         dp = ddebug_iter_first(iter);
 634         while (dp != NULL && --n > 0)
 635                 dp = ddebug_iter_next(iter);
 636         return dp;
 637 }
 638
 639 /*
 640  * Seq_ops next method.  Called several times within a read()
 641  * call from userspace, with ddebug_lock held.  Walks to the
 642  * next _ddebug object with a special case for the header line.
 643  */
 644 static void *ddebug_proc_next(struct seq_file *m, void *p, loff_t *pos)
 645 {
 646         struct ddebug_iter *iter = m->private;
 647         struct _ddebug *dp;
 648
 649         if (verbose)
 650                 pr_info("called m=%p p=%p *pos=%lld\n",
 651                         m, p, (unsigned long long)*pos);
 652
 653         if (p == SEQ_START_TOKEN)
 654                 dp = ddebug_iter_first(iter);
 655         else
 656                 dp = ddebug_iter_next(iter);
 657         ++*pos;
 658         return dp;
 659 }
 660
 661 /*
 662  * Seq_ops show method.  Called several times within a read()
 663  * call from userspace, with ddebug_lock held.  Formats the
 664  * current _ddebug as a single human-readable line, with a
 665  * special case for the header line.
 666  */
 667 static int ddebug_proc_show(struct seq_file *m, void *p)
 668 {
 669         struct ddebug_iter *iter = m->private;
 670         struct _ddebug *dp = p;
 671         char flagsbuf[8];
 672
 673         if (verbose)
 674                 pr_info("called m=%p p=%p\n", m, p);
 675
 676         if (p == SEQ_START_TOKEN) {
 677                 seq_puts(m,
 678                         "# filename:lineno [module]function flags format\n");
 679                 return 0;
 680         }
 681
 682         seq_printf(m, "%s:%u [%s]%s %s \"",
 683                    dp->filename, dp->lineno,
 684                    iter->table->mod_name, dp->function,
 685                    ddebug_describe_flags(dp, flagsbuf, sizeof(flagsbuf)));
 686         seq_escape(m, dp->format, "\t\r\n\"");
 687         seq_puts(m, "\"\n");
 688
 689         return 0;
 690 }
 691
 692 /*
 693  * Seq_ops stop method.  Called at the end of each read()
 694  * call from userspace.  Drops ddebug_lock.
 695  */
 696 static void ddebug_proc_stop(struct seq_file *m, void *p)
 697 {
 698         if (verbose)
 699                 pr_info("called m=%p p=%p\n", m, p);
 700         mutex_unlock(&ddebug_lock);
 701 }
 702
 703 static const struct seq_operations ddebug_proc_seqops = {
 704         .start = ddebug_proc_start,
 705         .next = ddebug_proc_next,
 706         .show = ddebug_proc_show,
 707         .stop = ddebug_proc_stop
 708 };
 709
 710 /*
 711  * File_ops->open method for <debugfs>/dynamic_debug/control.  Does the seq_file
 712  * setup dance, and also creates an iterator to walk the _ddebugs.
 713  * Note that we create a seq_file always, even for O_WRONLY files
 714  * where it's not needed, as doing so simplifies the ->release method.
 715  */
 716 static int ddebug_proc_open(struct inode *inode, struct file *file)
 717 {
 718         struct ddebug_iter *iter;
 719         int err;
 720
 721         if (verbose)
 722                 pr_info("called\n");
 723
 724         iter = kzalloc(sizeof(*iter), GFP_KERNEL);
 725         if (iter == NULL)
 726                 return -ENOMEM;
 727
 728         err = seq_open(file, &ddebug_proc_seqops);
 729         if (err) {
 730                 kfree(iter);
 731                 return err;
 732         }
 733         ((struct seq_file *) file->private_data)->private = iter;
 734         return 0;
 735 }
 736
 737 static const struct file_operations ddebug_proc_fops = {
 738         .owner = THIS_MODULE,
 739         .open = ddebug_proc_open,
 740         .read = seq_read,
 741         .llseek = seq_lseek,
 742         .release = seq_release_private,
 743         .write = ddebug_proc_write
 744 };
 745
 746 /*
 747  * Allocate a new ddebug_table for the given module
 748  * and add it to the global list.
 749  */
 750 int ddebug_add_module(struct _ddebug *tab, unsigned int n,
 751                              const char *name)
 752 {
 753         struct ddebug_table *dt;
 754         char *new_name;
 755
 756         dt = kzalloc(sizeof(*dt), GFP_KERNEL);
 757         if (dt == NULL)
 758                 return -ENOMEM;
 759         new_name = kstrdup(name, GFP_KERNEL);
 760         if (new_name == NULL) {
 761                 kfree(dt);
 762                 return -ENOMEM;
 763         }
 764         dt->mod_name = new_name;
 765         dt->num_ddebugs = n;
 766         dt->num_enabled = 0;
 767         dt->ddebugs = tab;
 768
 769         mutex_lock(&ddebug_lock);
 770         list_add_tail(&dt->link, &ddebug_tables);
 771         mutex_unlock(&ddebug_lock);
 772
 773         if (verbose)
 774                 pr_info("%u debug prints in module %s\n", n, dt->mod_name);
 775         return 0;
 776 }
 777 EXPORT_SYMBOL_GPL(ddebug_add_module);
 778
 779 static void ddebug_table_free(struct ddebug_table *dt)
 780 {
 781         list_del_init(&dt->link);
 782         kfree(dt->mod_name);
 783         kfree(dt);
 784 }
 785
 786 /*
 787  * Called in response to a module being unloaded.  Removes
 788  * any ddebug_table's which point at the module.
 789  */
 790 int ddebug_remove_module(const char *mod_name)
 791 {
 792         struct ddebug_table *dt, *nextdt;
 793         int ret = -ENOENT;
 794
 795         if (verbose)
 796                 pr_info("removing module \"%s\"\n", mod_name);
 797
 798         mutex_lock(&ddebug_lock);
 799         list_for_each_entry_safe(dt, nextdt, &ddebug_tables, link) {
 800                 if (!strcmp(dt->mod_name, mod_name)) {
 801                         ddebug_table_free(dt);
 802                         ret = 0;
 803                 }
 804         }
 805         mutex_unlock(&ddebug_lock);
 806         return ret;
 807 }
 808 EXPORT_SYMBOL_GPL(ddebug_remove_module);
 809
 810 static void ddebug_remove_all_tables(void)
 811 {
 812         mutex_lock(&ddebug_lock);
 813         while (!list_empty(&ddebug_tables)) {
 814                 struct ddebug_table *dt = list_entry(ddebug_tables.next,
 815                                                       struct ddebug_table,
 816                                                       link);
 817                 ddebug_table_free(dt);
 818         }
 819         mutex_unlock(&ddebug_lock);
 820 }
 821
 822 static __initdata int ddebug_init_success;
 823
 824 static int __init dynamic_debug_init_debugfs(void)
 825 {
 826         struct dentry *dir, *file;
 827
 828         if (!ddebug_init_success)
 829                 return -ENODEV;
 830
 831         dir = debugfs_create_dir("dynamic_debug", NULL);
 832         if (!dir)
 833                 return -ENOMEM;
 834         file = debugfs_create_file("control", 0644, dir, NULL,
 835                                         &ddebug_proc_fops);
 836         if (!file) {
 837                 debugfs_remove(dir);
 838                 return -ENOMEM;
 839         }
 840         return 0;
 841 }
 842
 843 static int __init dynamic_debug_init(void)
 844 {
 845         struct _ddebug *iter, *iter_start;
 846         const char *modname = NULL;
 847         int ret = 0;
 848         int n = 0;
 849
 850         if (__start___verbose != __stop___verbose) {
 851                 iter = __start___verbose;
 852                 modname = iter->modname;
 853                 iter_start = iter;
 854                 for (; iter < __stop___verbose; iter++) {
 855                         if (strcmp(modname, iter->modname)) {
 856                                 ret = ddebug_add_module(iter_start, n, modname);
 857                                 if (ret)
 858                                         goto out_free;
 859                                 n = 0;
 860                                 modname = iter->modname;
 861                                 iter_start = iter;
 862                         }
 863                         n++;
 864                 }
 865                 ret = ddebug_add_module(iter_start, n, modname);
 866         }
 867
 868         /* ddebug_query boot param got passed -> set it up */
 869         if (ddebug_setup_string[0] != '\0') {
 870                 ret = ddebug_exec_query(ddebug_setup_string);
 871                 if (ret)
 872                         pr_warn("Invalid ddebug boot param %s",
 873                                 ddebug_setup_string);
 874                 else
 875                         pr_info("ddebug initialized with string %s",
 876                                 ddebug_setup_string);
 877         }
 878
 879 out_free:
 880         if (ret)
 881                 ddebug_remove_all_tables();
 882         else
 883                 ddebug_init_success = 1;
 884         return 0;
 885 }
 886 /* Allow early initialization for boot messages via boot param */
 887 arch_initcall(dynamic_debug_init);
 888 /* Debugfs setup must be done later */
 889 module_init(dynamic_debug_init_debugfs);