]> Pileus Git - ~andy/linux/blob - drivers/staging/lustre/lustre/libcfs/tracefile.c
Merge branch 'timers/core' of git://git.kernel.org/pub/scm/linux/kernel/git/tip/tip
[~andy/linux] / drivers / staging / lustre / lustre / libcfs / tracefile.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 only,
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License version 2 for more details (a copy is included
14  * in the LICENSE file that accompanied this code).
15  *
16  * You should have received a copy of the GNU General Public License
17  * version 2 along with this program; If not, see
18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
19  *
20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
21  * CA 95054 USA or visit www.sun.com if you need additional information or
22  * have any questions.
23  *
24  * GPL HEADER END
25  */
26 /*
27  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  *
30  * Copyright (c) 2012, Intel Corporation.
31  */
32 /*
33  * This file is part of Lustre, http://www.lustre.org/
34  * Lustre is a trademark of Sun Microsystems, Inc.
35  *
36  * libcfs/libcfs/tracefile.c
37  *
38  * Author: Zach Brown <zab@clusterfs.com>
39  * Author: Phil Schwan <phil@clusterfs.com>
40  */
41
42
43 #define DEBUG_SUBSYSTEM S_LNET
44 #define LUSTRE_TRACEFILE_PRIVATE
45 #include "tracefile.h"
46
47 #include <linux/libcfs/libcfs.h>
48
49 /* XXX move things up to the top, comment */
50 union cfs_trace_data_union (*cfs_trace_data[TCD_MAX_TYPES])[NR_CPUS] __cacheline_aligned;
51
52 char cfs_tracefile[TRACEFILE_NAME_SIZE];
53 long long cfs_tracefile_size = CFS_TRACEFILE_SIZE;
54 static struct tracefiled_ctl trace_tctl;
55 struct mutex cfs_trace_thread_mutex;
56 static int thread_running = 0;
57
58 atomic_t cfs_tage_allocated = ATOMIC_INIT(0);
59
60 static void put_pages_on_tcd_daemon_list(struct page_collection *pc,
61                                          struct cfs_trace_cpu_data *tcd);
62
63 static inline struct cfs_trace_page *
64 cfs_tage_from_list(struct list_head *list)
65 {
66         return list_entry(list, struct cfs_trace_page, linkage);
67 }
68
69 static struct cfs_trace_page *cfs_tage_alloc(int gfp)
70 {
71         struct page         *page;
72         struct cfs_trace_page *tage;
73
74         /* My caller is trying to free memory */
75         if (!in_interrupt() && memory_pressure_get())
76                 return NULL;
77
78         /*
79          * Don't spam console with allocation failures: they will be reported
80          * by upper layer anyway.
81          */
82         gfp |= __GFP_NOWARN;
83         page = alloc_page(gfp);
84         if (page == NULL)
85                 return NULL;
86
87         tage = kmalloc(sizeof(*tage), gfp);
88         if (tage == NULL) {
89                 __free_page(page);
90                 return NULL;
91         }
92
93         tage->page = page;
94         atomic_inc(&cfs_tage_allocated);
95         return tage;
96 }
97
98 static void cfs_tage_free(struct cfs_trace_page *tage)
99 {
100         __LASSERT(tage != NULL);
101         __LASSERT(tage->page != NULL);
102
103         __free_page(tage->page);
104         kfree(tage);
105         atomic_dec(&cfs_tage_allocated);
106 }
107
108 static void cfs_tage_to_tail(struct cfs_trace_page *tage,
109                              struct list_head *queue)
110 {
111         __LASSERT(tage != NULL);
112         __LASSERT(queue != NULL);
113
114         list_move_tail(&tage->linkage, queue);
115 }
116
117 int cfs_trace_refill_stock(struct cfs_trace_cpu_data *tcd, int gfp,
118                            struct list_head *stock)
119 {
120         int i;
121
122         /*
123          * XXX nikita: do NOT call portals_debug_msg() (CDEBUG/ENTRY/EXIT)
124          * from here: this will lead to infinite recursion.
125          */
126
127         for (i = 0; i + tcd->tcd_cur_stock_pages < TCD_STOCK_PAGES ; ++ i) {
128                 struct cfs_trace_page *tage;
129
130                 tage = cfs_tage_alloc(gfp);
131                 if (tage == NULL)
132                         break;
133                 list_add_tail(&tage->linkage, stock);
134         }
135         return i;
136 }
137
138 /* return a page that has 'len' bytes left at the end */
139 static struct cfs_trace_page *
140 cfs_trace_get_tage_try(struct cfs_trace_cpu_data *tcd, unsigned long len)
141 {
142         struct cfs_trace_page *tage;
143
144         if (tcd->tcd_cur_pages > 0) {
145                 __LASSERT(!list_empty(&tcd->tcd_pages));
146                 tage = cfs_tage_from_list(tcd->tcd_pages.prev);
147                 if (tage->used + len <= PAGE_CACHE_SIZE)
148                         return tage;
149         }
150
151         if (tcd->tcd_cur_pages < tcd->tcd_max_pages) {
152                 if (tcd->tcd_cur_stock_pages > 0) {
153                         tage = cfs_tage_from_list(tcd->tcd_stock_pages.prev);
154                         --tcd->tcd_cur_stock_pages;
155                         list_del_init(&tage->linkage);
156                 } else {
157                         tage = cfs_tage_alloc(GFP_ATOMIC);
158                         if (unlikely(tage == NULL)) {
159                                 if ((!memory_pressure_get() ||
160                                      in_interrupt()) && printk_ratelimit())
161                                         printk(KERN_WARNING
162                                                "cannot allocate a tage (%ld)\n",
163                                                tcd->tcd_cur_pages);
164                                 return NULL;
165                         }
166                 }
167
168                 tage->used = 0;
169                 tage->cpu = smp_processor_id();
170                 tage->type = tcd->tcd_type;
171                 list_add_tail(&tage->linkage, &tcd->tcd_pages);
172                 tcd->tcd_cur_pages++;
173
174                 if (tcd->tcd_cur_pages > 8 && thread_running) {
175                         struct tracefiled_ctl *tctl = &trace_tctl;
176                         /*
177                          * wake up tracefiled to process some pages.
178                          */
179                         wake_up(&tctl->tctl_waitq);
180                 }
181                 return tage;
182         }
183         return NULL;
184 }
185
186 static void cfs_tcd_shrink(struct cfs_trace_cpu_data *tcd)
187 {
188         int pgcount = tcd->tcd_cur_pages / 10;
189         struct page_collection pc;
190         struct cfs_trace_page *tage;
191         struct cfs_trace_page *tmp;
192
193         /*
194          * XXX nikita: do NOT call portals_debug_msg() (CDEBUG/ENTRY/EXIT)
195          * from here: this will lead to infinite recursion.
196          */
197
198         if (printk_ratelimit())
199                 printk(KERN_WARNING "debug daemon buffer overflowed; "
200                        "discarding 10%% of pages (%d of %ld)\n",
201                        pgcount + 1, tcd->tcd_cur_pages);
202
203         INIT_LIST_HEAD(&pc.pc_pages);
204         spin_lock_init(&pc.pc_lock);
205
206         list_for_each_entry_safe(tage, tmp, &tcd->tcd_pages, linkage) {
207                 if (pgcount-- == 0)
208                         break;
209
210                 list_move_tail(&tage->linkage, &pc.pc_pages);
211                 tcd->tcd_cur_pages--;
212         }
213         put_pages_on_tcd_daemon_list(&pc, tcd);
214 }
215
216 /* return a page that has 'len' bytes left at the end */
217 static struct cfs_trace_page *cfs_trace_get_tage(struct cfs_trace_cpu_data *tcd,
218                                                  unsigned long len)
219 {
220         struct cfs_trace_page *tage;
221
222         /*
223          * XXX nikita: do NOT call portals_debug_msg() (CDEBUG/ENTRY/EXIT)
224          * from here: this will lead to infinite recursion.
225          */
226
227         if (len > PAGE_CACHE_SIZE) {
228                 printk(KERN_ERR
229                        "cowardly refusing to write %lu bytes in a page\n", len);
230                 return NULL;
231         }
232
233         tage = cfs_trace_get_tage_try(tcd, len);
234         if (tage != NULL)
235                 return tage;
236         if (thread_running)
237                 cfs_tcd_shrink(tcd);
238         if (tcd->tcd_cur_pages > 0) {
239                 tage = cfs_tage_from_list(tcd->tcd_pages.next);
240                 tage->used = 0;
241                 cfs_tage_to_tail(tage, &tcd->tcd_pages);
242         }
243         return tage;
244 }
245
246 int libcfs_debug_msg(struct libcfs_debug_msg_data *msgdata,
247                      const char *format, ...)
248 {
249         va_list args;
250         int     rc;
251
252         va_start(args, format);
253         rc = libcfs_debug_vmsg2(msgdata, format, args, NULL);
254         va_end(args);
255
256         return rc;
257 }
258 EXPORT_SYMBOL(libcfs_debug_msg);
259
260 int libcfs_debug_vmsg2(struct libcfs_debug_msg_data *msgdata,
261                        const char *format1, va_list args,
262                        const char *format2, ...)
263 {
264         struct cfs_trace_cpu_data *tcd = NULL;
265         struct ptldebug_header     header = {0};
266         struct cfs_trace_page     *tage;
267         /* string_buf is used only if tcd != NULL, and is always set then */
268         char                  *string_buf = NULL;
269         char                  *debug_buf;
270         int                     known_size;
271         int                     needed = 85; /* average message length */
272         int                     max_nob;
273         va_list             ap;
274         int                     depth;
275         int                     i;
276         int                     remain;
277         int                     mask = msgdata->msg_mask;
278         const char              *file = kbasename(msgdata->msg_file);
279         cfs_debug_limit_state_t   *cdls = msgdata->msg_cdls;
280
281         tcd = cfs_trace_get_tcd();
282
283         /* cfs_trace_get_tcd() grabs a lock, which disables preemption and
284          * pins us to a particular CPU.  This avoids an smp_processor_id()
285          * warning on Linux when debugging is enabled. */
286         cfs_set_ptldebug_header(&header, msgdata, CDEBUG_STACK());
287
288         if (tcd == NULL)                /* arch may not log in IRQ context */
289                 goto console;
290
291         if (tcd->tcd_cur_pages == 0)
292                 header.ph_flags |= PH_FLAG_FIRST_RECORD;
293
294         if (tcd->tcd_shutting_down) {
295                 cfs_trace_put_tcd(tcd);
296                 tcd = NULL;
297                 goto console;
298         }
299
300         depth = __current_nesting_level();
301         known_size = strlen(file) + 1 + depth;
302         if (msgdata->msg_fn)
303                 known_size += strlen(msgdata->msg_fn) + 1;
304
305         if (libcfs_debug_binary)
306                 known_size += sizeof(header);
307
308         /*/
309          * '2' used because vsnprintf return real size required for output
310          * _without_ terminating NULL.
311          * if needed is to small for this format.
312          */
313         for (i = 0; i < 2; i++) {
314                 tage = cfs_trace_get_tage(tcd, needed + known_size + 1);
315                 if (tage == NULL) {
316                         if (needed + known_size > PAGE_CACHE_SIZE)
317                                 mask |= D_ERROR;
318
319                         cfs_trace_put_tcd(tcd);
320                         tcd = NULL;
321                         goto console;
322                 }
323
324                 string_buf = (char *)page_address(tage->page) +
325                                         tage->used + known_size;
326
327                 max_nob = PAGE_CACHE_SIZE - tage->used - known_size;
328                 if (max_nob <= 0) {
329                         printk(KERN_EMERG "negative max_nob: %d\n",
330                                max_nob);
331                         mask |= D_ERROR;
332                         cfs_trace_put_tcd(tcd);
333                         tcd = NULL;
334                         goto console;
335                 }
336
337                 needed = 0;
338                 if (format1) {
339                         va_copy(ap, args);
340                         needed = vsnprintf(string_buf, max_nob, format1, ap);
341                         va_end(ap);
342                 }
343
344                 if (format2) {
345                         remain = max_nob - needed;
346                         if (remain < 0)
347                                 remain = 0;
348
349                         va_start(ap, format2);
350                         needed += vsnprintf(string_buf + needed, remain,
351                                             format2, ap);
352                         va_end(ap);
353                 }
354
355                 if (needed < max_nob) /* well. printing ok.. */
356                         break;
357         }
358
359         if (*(string_buf+needed-1) != '\n')
360                 printk(KERN_INFO "format at %s:%d:%s doesn't end in "
361                        "newline\n", file, msgdata->msg_line, msgdata->msg_fn);
362
363         header.ph_len = known_size + needed;
364         debug_buf = (char *)page_address(tage->page) + tage->used;
365
366         if (libcfs_debug_binary) {
367                 memcpy(debug_buf, &header, sizeof(header));
368                 tage->used += sizeof(header);
369                 debug_buf += sizeof(header);
370         }
371
372         /* indent message according to the nesting level */
373         while (depth-- > 0) {
374                 *(debug_buf++) = '.';
375                 ++ tage->used;
376         }
377
378         strcpy(debug_buf, file);
379         tage->used += strlen(file) + 1;
380         debug_buf += strlen(file) + 1;
381
382         if (msgdata->msg_fn) {
383                 strcpy(debug_buf, msgdata->msg_fn);
384                 tage->used += strlen(msgdata->msg_fn) + 1;
385                 debug_buf += strlen(msgdata->msg_fn) + 1;
386         }
387
388         __LASSERT(debug_buf == string_buf);
389
390         tage->used += needed;
391         __LASSERT (tage->used <= PAGE_CACHE_SIZE);
392
393 console:
394         if ((mask & libcfs_printk) == 0) {
395                 /* no console output requested */
396                 if (tcd != NULL)
397                         cfs_trace_put_tcd(tcd);
398                 return 1;
399         }
400
401         if (cdls != NULL) {
402                 if (libcfs_console_ratelimit &&
403                     cdls->cdls_next != 0 &&     /* not first time ever */
404                     !cfs_time_after(cfs_time_current(), cdls->cdls_next)) {
405                         /* skipping a console message */
406                         cdls->cdls_count++;
407                         if (tcd != NULL)
408                                 cfs_trace_put_tcd(tcd);
409                         return 1;
410                 }
411
412                 if (cfs_time_after(cfs_time_current(), cdls->cdls_next +
413                                                        libcfs_console_max_delay
414                                                        + cfs_time_seconds(10))) {
415                         /* last timeout was a long time ago */
416                         cdls->cdls_delay /= libcfs_console_backoff * 4;
417                 } else {
418                         cdls->cdls_delay *= libcfs_console_backoff;
419
420                         if (cdls->cdls_delay < libcfs_console_min_delay)
421                                 cdls->cdls_delay = libcfs_console_min_delay;
422                         else if (cdls->cdls_delay > libcfs_console_max_delay)
423                                 cdls->cdls_delay = libcfs_console_max_delay;
424                 }
425
426                 /* ensure cdls_next is never zero after it's been seen */
427                 cdls->cdls_next = (cfs_time_current() + cdls->cdls_delay) | 1;
428         }
429
430         if (tcd != NULL) {
431                 cfs_print_to_console(&header, mask, string_buf, needed, file,
432                                      msgdata->msg_fn);
433                 cfs_trace_put_tcd(tcd);
434         } else {
435                 string_buf = cfs_trace_get_console_buffer();
436
437                 needed = 0;
438                 if (format1 != NULL) {
439                         va_copy(ap, args);
440                         needed = vsnprintf(string_buf,
441                                            CFS_TRACE_CONSOLE_BUFFER_SIZE,
442                                            format1, ap);
443                         va_end(ap);
444                 }
445                 if (format2 != NULL) {
446                         remain = CFS_TRACE_CONSOLE_BUFFER_SIZE - needed;
447                         if (remain > 0) {
448                                 va_start(ap, format2);
449                                 needed += vsnprintf(string_buf+needed, remain,
450                                                     format2, ap);
451                                 va_end(ap);
452                         }
453                 }
454                 cfs_print_to_console(&header, mask,
455                                      string_buf, needed, file, msgdata->msg_fn);
456
457                 cfs_trace_put_console_buffer(string_buf);
458         }
459
460         if (cdls != NULL && cdls->cdls_count != 0) {
461                 string_buf = cfs_trace_get_console_buffer();
462
463                 needed = snprintf(string_buf, CFS_TRACE_CONSOLE_BUFFER_SIZE,
464                                   "Skipped %d previous similar message%s\n",
465                                   cdls->cdls_count,
466                                   (cdls->cdls_count > 1) ? "s" : "");
467
468                 cfs_print_to_console(&header, mask,
469                                      string_buf, needed, file, msgdata->msg_fn);
470
471                 cfs_trace_put_console_buffer(string_buf);
472                 cdls->cdls_count = 0;
473         }
474
475         return 0;
476 }
477 EXPORT_SYMBOL(libcfs_debug_vmsg2);
478
479 void
480 cfs_trace_assertion_failed(const char *str,
481                            struct libcfs_debug_msg_data *msgdata)
482 {
483         struct ptldebug_header hdr;
484
485         libcfs_panic_in_progress = 1;
486         libcfs_catastrophe = 1;
487         mb();
488
489         cfs_set_ptldebug_header(&hdr, msgdata, CDEBUG_STACK());
490
491         cfs_print_to_console(&hdr, D_EMERG, str, strlen(str),
492                              msgdata->msg_file, msgdata->msg_fn);
493
494         panic("Lustre debug assertion failure\n");
495
496         /* not reached */
497 }
498
499 static void
500 panic_collect_pages(struct page_collection *pc)
501 {
502         /* Do the collect_pages job on a single CPU: assumes that all other
503          * CPUs have been stopped during a panic.  If this isn't true for some
504          * arch, this will have to be implemented separately in each arch.  */
505         int                     i;
506         int                     j;
507         struct cfs_trace_cpu_data *tcd;
508
509         INIT_LIST_HEAD(&pc->pc_pages);
510
511         cfs_tcd_for_each(tcd, i, j) {
512                 list_splice_init(&tcd->tcd_pages, &pc->pc_pages);
513                 tcd->tcd_cur_pages = 0;
514
515                 if (pc->pc_want_daemon_pages) {
516                         list_splice_init(&tcd->tcd_daemon_pages,
517                                              &pc->pc_pages);
518                         tcd->tcd_cur_daemon_pages = 0;
519                 }
520         }
521 }
522
523 static void collect_pages_on_all_cpus(struct page_collection *pc)
524 {
525         struct cfs_trace_cpu_data *tcd;
526         int i, cpu;
527
528         spin_lock(&pc->pc_lock);
529         for_each_possible_cpu(cpu) {
530                 cfs_tcd_for_each_type_lock(tcd, i, cpu) {
531                         list_splice_init(&tcd->tcd_pages, &pc->pc_pages);
532                         tcd->tcd_cur_pages = 0;
533                         if (pc->pc_want_daemon_pages) {
534                                 list_splice_init(&tcd->tcd_daemon_pages,
535                                                      &pc->pc_pages);
536                                 tcd->tcd_cur_daemon_pages = 0;
537                         }
538                 }
539         }
540         spin_unlock(&pc->pc_lock);
541 }
542
543 static void collect_pages(struct page_collection *pc)
544 {
545         INIT_LIST_HEAD(&pc->pc_pages);
546
547         if (libcfs_panic_in_progress)
548                 panic_collect_pages(pc);
549         else
550                 collect_pages_on_all_cpus(pc);
551 }
552
553 static void put_pages_back_on_all_cpus(struct page_collection *pc)
554 {
555         struct cfs_trace_cpu_data *tcd;
556         struct list_head *cur_head;
557         struct cfs_trace_page *tage;
558         struct cfs_trace_page *tmp;
559         int i, cpu;
560
561         spin_lock(&pc->pc_lock);
562         for_each_possible_cpu(cpu) {
563                 cfs_tcd_for_each_type_lock(tcd, i, cpu) {
564                         cur_head = tcd->tcd_pages.next;
565
566                         list_for_each_entry_safe(tage, tmp, &pc->pc_pages,
567                                                  linkage) {
568
569                                 __LASSERT_TAGE_INVARIANT(tage);
570
571                                 if (tage->cpu != cpu || tage->type != i)
572                                         continue;
573
574                                 cfs_tage_to_tail(tage, cur_head);
575                                 tcd->tcd_cur_pages++;
576                         }
577                 }
578         }
579         spin_unlock(&pc->pc_lock);
580 }
581
582 static void put_pages_back(struct page_collection *pc)
583 {
584         if (!libcfs_panic_in_progress)
585                 put_pages_back_on_all_cpus(pc);
586 }
587
588 /* Add pages to a per-cpu debug daemon ringbuffer.  This buffer makes sure that
589  * we have a good amount of data at all times for dumping during an LBUG, even
590  * if we have been steadily writing (and otherwise discarding) pages via the
591  * debug daemon. */
592 static void put_pages_on_tcd_daemon_list(struct page_collection *pc,
593                                          struct cfs_trace_cpu_data *tcd)
594 {
595         struct cfs_trace_page *tage;
596         struct cfs_trace_page *tmp;
597
598         spin_lock(&pc->pc_lock);
599         list_for_each_entry_safe(tage, tmp, &pc->pc_pages, linkage) {
600
601                 __LASSERT_TAGE_INVARIANT(tage);
602
603                 if (tage->cpu != tcd->tcd_cpu || tage->type != tcd->tcd_type)
604                         continue;
605
606                 cfs_tage_to_tail(tage, &tcd->tcd_daemon_pages);
607                 tcd->tcd_cur_daemon_pages++;
608
609                 if (tcd->tcd_cur_daemon_pages > tcd->tcd_max_pages) {
610                         struct cfs_trace_page *victim;
611
612                         __LASSERT(!list_empty(&tcd->tcd_daemon_pages));
613                         victim = cfs_tage_from_list(tcd->tcd_daemon_pages.next);
614
615                         __LASSERT_TAGE_INVARIANT(victim);
616
617                         list_del(&victim->linkage);
618                         cfs_tage_free(victim);
619                         tcd->tcd_cur_daemon_pages--;
620                 }
621         }
622         spin_unlock(&pc->pc_lock);
623 }
624
625 static void put_pages_on_daemon_list(struct page_collection *pc)
626 {
627         struct cfs_trace_cpu_data *tcd;
628         int i, cpu;
629
630         for_each_possible_cpu(cpu) {
631                 cfs_tcd_for_each_type_lock(tcd, i, cpu)
632                         put_pages_on_tcd_daemon_list(pc, tcd);
633         }
634 }
635
636 void cfs_trace_debug_print(void)
637 {
638         struct page_collection pc;
639         struct cfs_trace_page *tage;
640         struct cfs_trace_page *tmp;
641
642         spin_lock_init(&pc.pc_lock);
643
644         pc.pc_want_daemon_pages = 1;
645         collect_pages(&pc);
646         list_for_each_entry_safe(tage, tmp, &pc.pc_pages, linkage) {
647                 char *p, *file, *fn;
648                 struct page *page;
649
650                 __LASSERT_TAGE_INVARIANT(tage);
651
652                 page = tage->page;
653                 p = page_address(page);
654                 while (p < ((char *)page_address(page) + tage->used)) {
655                         struct ptldebug_header *hdr;
656                         int len;
657                         hdr = (void *)p;
658                         p += sizeof(*hdr);
659                         file = p;
660                         p += strlen(file) + 1;
661                         fn = p;
662                         p += strlen(fn) + 1;
663                         len = hdr->ph_len - (int)(p - (char *)hdr);
664
665                         cfs_print_to_console(hdr, D_EMERG, p, len, file, fn);
666
667                         p += len;
668                 }
669
670                 list_del(&tage->linkage);
671                 cfs_tage_free(tage);
672         }
673 }
674
675 int cfs_tracefile_dump_all_pages(char *filename)
676 {
677         struct page_collection  pc;
678         struct file             *filp;
679         struct cfs_trace_page   *tage;
680         struct cfs_trace_page   *tmp;
681         int rc;
682
683         DECL_MMSPACE;
684
685         cfs_tracefile_write_lock();
686
687         filp = filp_open(filename, O_CREAT|O_EXCL|O_WRONLY|O_LARGEFILE, 0600);
688         if (IS_ERR(filp)) {
689                 rc = PTR_ERR(filp);
690                 filp = NULL;
691                 printk(KERN_ERR "LustreError: can't open %s for dump: rc %d\n",
692                       filename, rc);
693                 goto out;
694         }
695
696         spin_lock_init(&pc.pc_lock);
697         pc.pc_want_daemon_pages = 1;
698         collect_pages(&pc);
699         if (list_empty(&pc.pc_pages)) {
700                 rc = 0;
701                 goto close;
702         }
703
704         /* ok, for now, just write the pages.  in the future we'll be building
705          * iobufs with the pages and calling generic_direct_IO */
706         MMSPACE_OPEN;
707         list_for_each_entry_safe(tage, tmp, &pc.pc_pages, linkage) {
708
709                 __LASSERT_TAGE_INVARIANT(tage);
710
711                 rc = filp_write(filp, page_address(tage->page),
712                                 tage->used, filp_poff(filp));
713                 if (rc != (int)tage->used) {
714                         printk(KERN_WARNING "wanted to write %u but wrote "
715                                "%d\n", tage->used, rc);
716                         put_pages_back(&pc);
717                         __LASSERT(list_empty(&pc.pc_pages));
718                         break;
719                 }
720                 list_del(&tage->linkage);
721                 cfs_tage_free(tage);
722         }
723         MMSPACE_CLOSE;
724         rc = filp_fsync(filp);
725         if (rc)
726                 printk(KERN_ERR "sync returns %d\n", rc);
727 close:
728         filp_close(filp, NULL);
729 out:
730         cfs_tracefile_write_unlock();
731         return rc;
732 }
733
734 void cfs_trace_flush_pages(void)
735 {
736         struct page_collection pc;
737         struct cfs_trace_page *tage;
738         struct cfs_trace_page *tmp;
739
740         spin_lock_init(&pc.pc_lock);
741
742         pc.pc_want_daemon_pages = 1;
743         collect_pages(&pc);
744         list_for_each_entry_safe(tage, tmp, &pc.pc_pages, linkage) {
745
746                 __LASSERT_TAGE_INVARIANT(tage);
747
748                 list_del(&tage->linkage);
749                 cfs_tage_free(tage);
750         }
751 }
752
753 int cfs_trace_copyin_string(char *knl_buffer, int knl_buffer_nob,
754                             const char *usr_buffer, int usr_buffer_nob)
755 {
756         int    nob;
757
758         if (usr_buffer_nob > knl_buffer_nob)
759                 return -EOVERFLOW;
760
761         if (copy_from_user((void *)knl_buffer,
762                            (void *)usr_buffer, usr_buffer_nob))
763                 return -EFAULT;
764
765         nob = strnlen(knl_buffer, usr_buffer_nob);
766         while (nob-- >= 0)                    /* strip trailing whitespace */
767                 if (!isspace(knl_buffer[nob]))
768                         break;
769
770         if (nob < 0)                        /* empty string */
771                 return -EINVAL;
772
773         if (nob == knl_buffer_nob)            /* no space to terminate */
774                 return -EOVERFLOW;
775
776         knl_buffer[nob + 1] = 0;                /* terminate */
777         return 0;
778 }
779 EXPORT_SYMBOL(cfs_trace_copyin_string);
780
781 int cfs_trace_copyout_string(char *usr_buffer, int usr_buffer_nob,
782                              const char *knl_buffer, char *append)
783 {
784         /* NB if 'append' != NULL, it's a single character to append to the
785          * copied out string - usually "\n", for /proc entries and "" (i.e. a
786          * terminating zero byte) for sysctl entries */
787         int   nob = strlen(knl_buffer);
788
789         if (nob > usr_buffer_nob)
790                 nob = usr_buffer_nob;
791
792         if (copy_to_user(usr_buffer, knl_buffer, nob))
793                 return -EFAULT;
794
795         if (append != NULL && nob < usr_buffer_nob) {
796                 if (copy_to_user(usr_buffer + nob, append, 1))
797                         return -EFAULT;
798
799                 nob++;
800         }
801
802         return nob;
803 }
804 EXPORT_SYMBOL(cfs_trace_copyout_string);
805
806 int cfs_trace_allocate_string_buffer(char **str, int nob)
807 {
808         if (nob > 2 * PAGE_CACHE_SIZE)      /* string must be "sensible" */
809                 return -EINVAL;
810
811         *str = kmalloc(nob, GFP_IOFS | __GFP_ZERO);
812         if (*str == NULL)
813                 return -ENOMEM;
814
815         return 0;
816 }
817
818 void cfs_trace_free_string_buffer(char *str, int nob)
819 {
820         kfree(str);
821 }
822
823 int cfs_trace_dump_debug_buffer_usrstr(void *usr_str, int usr_str_nob)
824 {
825         char     *str;
826         int        rc;
827
828         rc = cfs_trace_allocate_string_buffer(&str, usr_str_nob + 1);
829         if (rc != 0)
830                 return rc;
831
832         rc = cfs_trace_copyin_string(str, usr_str_nob + 1,
833                                      usr_str, usr_str_nob);
834         if (rc != 0)
835                 goto out;
836
837         if (str[0] != '/') {
838                 rc = -EINVAL;
839                 goto out;
840         }
841         rc = cfs_tracefile_dump_all_pages(str);
842 out:
843         cfs_trace_free_string_buffer(str, usr_str_nob + 1);
844         return rc;
845 }
846
847 int cfs_trace_daemon_command(char *str)
848 {
849         int       rc = 0;
850
851         cfs_tracefile_write_lock();
852
853         if (strcmp(str, "stop") == 0) {
854                 cfs_tracefile_write_unlock();
855                 cfs_trace_stop_thread();
856                 cfs_tracefile_write_lock();
857                 memset(cfs_tracefile, 0, sizeof(cfs_tracefile));
858
859         } else if (strncmp(str, "size=", 5) == 0) {
860                 cfs_tracefile_size = simple_strtoul(str + 5, NULL, 0);
861                 if (cfs_tracefile_size < 10 || cfs_tracefile_size > 20480)
862                         cfs_tracefile_size = CFS_TRACEFILE_SIZE;
863                 else
864                         cfs_tracefile_size <<= 20;
865
866         } else if (strlen(str) >= sizeof(cfs_tracefile)) {
867                 rc = -ENAMETOOLONG;
868         } else if (str[0] != '/') {
869                 rc = -EINVAL;
870         } else {
871                 strcpy(cfs_tracefile, str);
872
873                 printk(KERN_INFO
874                        "Lustre: debug daemon will attempt to start writing "
875                        "to %s (%lukB max)\n", cfs_tracefile,
876                        (long)(cfs_tracefile_size >> 10));
877
878                 cfs_trace_start_thread();
879         }
880
881         cfs_tracefile_write_unlock();
882         return rc;
883 }
884
885 int cfs_trace_daemon_command_usrstr(void *usr_str, int usr_str_nob)
886 {
887         char *str;
888         int   rc;
889
890         rc = cfs_trace_allocate_string_buffer(&str, usr_str_nob + 1);
891         if (rc != 0)
892                 return rc;
893
894         rc = cfs_trace_copyin_string(str, usr_str_nob + 1,
895                                  usr_str, usr_str_nob);
896         if (rc == 0)
897                 rc = cfs_trace_daemon_command(str);
898
899         cfs_trace_free_string_buffer(str, usr_str_nob + 1);
900         return rc;
901 }
902
903 int cfs_trace_set_debug_mb(int mb)
904 {
905         int i;
906         int j;
907         int pages;
908         int limit = cfs_trace_max_debug_mb();
909         struct cfs_trace_cpu_data *tcd;
910
911         if (mb < num_possible_cpus()) {
912                 printk(KERN_WARNING
913                        "Lustre: %d MB is too small for debug buffer size, "
914                        "setting it to %d MB.\n", mb, num_possible_cpus());
915                 mb = num_possible_cpus();
916         }
917
918         if (mb > limit) {
919                 printk(KERN_WARNING
920                        "Lustre: %d MB is too large for debug buffer size, "
921                        "setting it to %d MB.\n", mb, limit);
922                 mb = limit;
923         }
924
925         mb /= num_possible_cpus();
926         pages = mb << (20 - PAGE_CACHE_SHIFT);
927
928         cfs_tracefile_write_lock();
929
930         cfs_tcd_for_each(tcd, i, j)
931                 tcd->tcd_max_pages = (pages * tcd->tcd_pages_factor) / 100;
932
933         cfs_tracefile_write_unlock();
934
935         return 0;
936 }
937
938 int cfs_trace_set_debug_mb_usrstr(void *usr_str, int usr_str_nob)
939 {
940         char     str[32];
941         int      rc;
942
943         rc = cfs_trace_copyin_string(str, sizeof(str), usr_str, usr_str_nob);
944         if (rc < 0)
945                 return rc;
946
947         return cfs_trace_set_debug_mb(simple_strtoul(str, NULL, 0));
948 }
949
950 int cfs_trace_get_debug_mb(void)
951 {
952         int i;
953         int j;
954         struct cfs_trace_cpu_data *tcd;
955         int total_pages = 0;
956
957         cfs_tracefile_read_lock();
958
959         cfs_tcd_for_each(tcd, i, j)
960                 total_pages += tcd->tcd_max_pages;
961
962         cfs_tracefile_read_unlock();
963
964         return (total_pages >> (20 - PAGE_CACHE_SHIFT)) + 1;
965 }
966
967 static int tracefiled(void *arg)
968 {
969         struct page_collection pc;
970         struct tracefiled_ctl *tctl = arg;
971         struct cfs_trace_page *tage;
972         struct cfs_trace_page *tmp;
973         struct file *filp;
974         int last_loop = 0;
975         int rc;
976
977         DECL_MMSPACE;
978
979         /* we're started late enough that we pick up init's fs context */
980         /* this is so broken in uml?  what on earth is going on? */
981
982         spin_lock_init(&pc.pc_lock);
983         complete(&tctl->tctl_start);
984
985         while (1) {
986                 wait_queue_t __wait;
987
988                 pc.pc_want_daemon_pages = 0;
989                 collect_pages(&pc);
990                 if (list_empty(&pc.pc_pages))
991                         goto end_loop;
992
993                 filp = NULL;
994                 cfs_tracefile_read_lock();
995                 if (cfs_tracefile[0] != 0) {
996                         filp = filp_open(cfs_tracefile,
997                                          O_CREAT | O_RDWR | O_LARGEFILE,
998                                          0600);
999                         if (IS_ERR(filp)) {
1000                                 rc = PTR_ERR(filp);
1001                                 filp = NULL;
1002                                 printk(KERN_WARNING "couldn't open %s: "
1003                                        "%d\n", cfs_tracefile, rc);
1004                         }
1005                 }
1006                 cfs_tracefile_read_unlock();
1007                 if (filp == NULL) {
1008                         put_pages_on_daemon_list(&pc);
1009                         __LASSERT(list_empty(&pc.pc_pages));
1010                         goto end_loop;
1011                 }
1012
1013                 MMSPACE_OPEN;
1014
1015                 list_for_each_entry_safe(tage, tmp, &pc.pc_pages,
1016                                                    linkage) {
1017                         static loff_t f_pos;
1018
1019                         __LASSERT_TAGE_INVARIANT(tage);
1020
1021                         if (f_pos >= (off_t)cfs_tracefile_size)
1022                                 f_pos = 0;
1023                         else if (f_pos > (off_t)filp_size(filp))
1024                                 f_pos = filp_size(filp);
1025
1026                         rc = filp_write(filp, page_address(tage->page),
1027                                         tage->used, &f_pos);
1028                         if (rc != (int)tage->used) {
1029                                 printk(KERN_WARNING "wanted to write %u "
1030                                        "but wrote %d\n", tage->used, rc);
1031                                 put_pages_back(&pc);
1032                                 __LASSERT(list_empty(&pc.pc_pages));
1033                         }
1034                 }
1035                 MMSPACE_CLOSE;
1036
1037                 filp_close(filp, NULL);
1038                 put_pages_on_daemon_list(&pc);
1039                 if (!list_empty(&pc.pc_pages)) {
1040                         int i;
1041
1042                         printk(KERN_ALERT "Lustre: trace pages aren't "
1043                                " empty\n");
1044                         printk(KERN_ERR "total cpus(%d): ",
1045                                num_possible_cpus());
1046                         for (i = 0; i < num_possible_cpus(); i++)
1047                                 if (cpu_online(i))
1048                                         printk(KERN_ERR "%d(on) ", i);
1049                                 else
1050                                         printk(KERN_ERR "%d(off) ", i);
1051                         printk(KERN_ERR "\n");
1052
1053                         i = 0;
1054                         list_for_each_entry_safe(tage, tmp, &pc.pc_pages,
1055                                                      linkage)
1056                                 printk(KERN_ERR "page %d belongs to cpu "
1057                                        "%d\n", ++i, tage->cpu);
1058                         printk(KERN_ERR "There are %d pages unwritten\n",
1059                                i);
1060                 }
1061                 __LASSERT(list_empty(&pc.pc_pages));
1062 end_loop:
1063                 if (atomic_read(&tctl->tctl_shutdown)) {
1064                         if (last_loop == 0) {
1065                                 last_loop = 1;
1066                                 continue;
1067                         } else {
1068                                 break;
1069                         }
1070                 }
1071                 init_waitqueue_entry_current(&__wait);
1072                 add_wait_queue(&tctl->tctl_waitq, &__wait);
1073                 set_current_state(TASK_INTERRUPTIBLE);
1074                 waitq_timedwait(&__wait, TASK_INTERRUPTIBLE,
1075                                     cfs_time_seconds(1));
1076                 remove_wait_queue(&tctl->tctl_waitq, &__wait);
1077         }
1078         complete(&tctl->tctl_stop);
1079         return 0;
1080 }
1081
1082 int cfs_trace_start_thread(void)
1083 {
1084         struct tracefiled_ctl *tctl = &trace_tctl;
1085         int rc = 0;
1086
1087         mutex_lock(&cfs_trace_thread_mutex);
1088         if (thread_running)
1089                 goto out;
1090
1091         init_completion(&tctl->tctl_start);
1092         init_completion(&tctl->tctl_stop);
1093         init_waitqueue_head(&tctl->tctl_waitq);
1094         atomic_set(&tctl->tctl_shutdown, 0);
1095
1096         if (IS_ERR(kthread_run(tracefiled, tctl, "ktracefiled"))) {
1097                 rc = -ECHILD;
1098                 goto out;
1099         }
1100
1101         wait_for_completion(&tctl->tctl_start);
1102         thread_running = 1;
1103 out:
1104         mutex_unlock(&cfs_trace_thread_mutex);
1105         return rc;
1106 }
1107
1108 void cfs_trace_stop_thread(void)
1109 {
1110         struct tracefiled_ctl *tctl = &trace_tctl;
1111
1112         mutex_lock(&cfs_trace_thread_mutex);
1113         if (thread_running) {
1114                 printk(KERN_INFO
1115                        "Lustre: shutting down debug daemon thread...\n");
1116                 atomic_set(&tctl->tctl_shutdown, 1);
1117                 wait_for_completion(&tctl->tctl_stop);
1118                 thread_running = 0;
1119         }
1120         mutex_unlock(&cfs_trace_thread_mutex);
1121 }
1122
1123 int cfs_tracefile_init(int max_pages)
1124 {
1125         struct cfs_trace_cpu_data *tcd;
1126         int                 i;
1127         int                 j;
1128         int                 rc;
1129         int                 factor;
1130
1131         rc = cfs_tracefile_init_arch();
1132         if (rc != 0)
1133                 return rc;
1134
1135         cfs_tcd_for_each(tcd, i, j) {
1136                 /* tcd_pages_factor is initialized int tracefile_init_arch. */
1137                 factor = tcd->tcd_pages_factor;
1138                 INIT_LIST_HEAD(&tcd->tcd_pages);
1139                 INIT_LIST_HEAD(&tcd->tcd_stock_pages);
1140                 INIT_LIST_HEAD(&tcd->tcd_daemon_pages);
1141                 tcd->tcd_cur_pages = 0;
1142                 tcd->tcd_cur_stock_pages = 0;
1143                 tcd->tcd_cur_daemon_pages = 0;
1144                 tcd->tcd_max_pages = (max_pages * factor) / 100;
1145                 LASSERT(tcd->tcd_max_pages > 0);
1146                 tcd->tcd_shutting_down = 0;
1147         }
1148
1149         return 0;
1150 }
1151
1152 static void trace_cleanup_on_all_cpus(void)
1153 {
1154         struct cfs_trace_cpu_data *tcd;
1155         struct cfs_trace_page *tage;
1156         struct cfs_trace_page *tmp;
1157         int i, cpu;
1158
1159         for_each_possible_cpu(cpu) {
1160                 cfs_tcd_for_each_type_lock(tcd, i, cpu) {
1161                         tcd->tcd_shutting_down = 1;
1162
1163                         list_for_each_entry_safe(tage, tmp, &tcd->tcd_pages,
1164                                                            linkage) {
1165                                 __LASSERT_TAGE_INVARIANT(tage);
1166
1167                                 list_del(&tage->linkage);
1168                                 cfs_tage_free(tage);
1169                         }
1170
1171                         tcd->tcd_cur_pages = 0;
1172                 }
1173         }
1174 }
1175
1176 static void cfs_trace_cleanup(void)
1177 {
1178         struct page_collection pc;
1179
1180         INIT_LIST_HEAD(&pc.pc_pages);
1181         spin_lock_init(&pc.pc_lock);
1182
1183         trace_cleanup_on_all_cpus();
1184
1185         cfs_tracefile_fini_arch();
1186 }
1187
1188 void cfs_tracefile_exit(void)
1189 {
1190         cfs_trace_stop_thread();
1191         cfs_trace_cleanup();
1192 }