]> git.proxmox.com Git - mirror_ubuntu-bionic-kernel.git/blame - fs/pstore/platform.c
PCI: PM: Skip devices in D0 for suspend-to-idle
[mirror_ubuntu-bionic-kernel.git] / fs / pstore / platform.c
CommitLineData
ca01d6dd
TL
1/*
2 * Persistent Storage - platform driver interface parts.
3 *
f29e5956 4 * Copyright (C) 2007-2008 Google, Inc.
ca01d6dd
TL
5 * Copyright (C) 2010 Intel Corporation <tony.luck@intel.com>
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License version 2 as
9 * published by the Free Software Foundation.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
ef748853
FF
21#define pr_fmt(fmt) "pstore: " fmt
22
ca01d6dd
TL
23#include <linux/atomic.h>
24#include <linux/types.h>
25#include <linux/errno.h>
26#include <linux/init.h>
27#include <linux/kmsg_dump.h>
f29e5956 28#include <linux/console.h>
ca01d6dd
TL
29#include <linux/module.h>
30#include <linux/pstore.h>
8cfc8ddc 31#ifdef CONFIG_PSTORE_ZLIB_COMPRESS
b0aad7a9 32#include <linux/zlib.h>
8cfc8ddc
GT
33#endif
34#ifdef CONFIG_PSTORE_LZO_COMPRESS
35#include <linux/lzo.h>
36#endif
37#ifdef CONFIG_PSTORE_LZ4_COMPRESS
38#include <linux/lz4.h>
39#endif
ca01d6dd 40#include <linux/string.h>
6dda9266 41#include <linux/timer.h>
ca01d6dd
TL
42#include <linux/slab.h>
43#include <linux/uaccess.h>
abd4d558 44#include <linux/hardirq.h>
a3f5f075 45#include <linux/jiffies.h>
6dda9266 46#include <linux/workqueue.h>
ca01d6dd
TL
47
48#include "internal.h"
49
6dda9266
TL
50/*
51 * We defer making "oops" entries appear in pstore - see
52 * whether the system is actually still running well enough
53 * to let someone see the entry
54 */
521f7288 55static int pstore_update_ms = -1;
a3f5f075
AV
56module_param_named(update_ms, pstore_update_ms, int, 0600);
57MODULE_PARM_DESC(update_ms, "milliseconds before pstore updates its content "
521f7288
AV
58 "(default is -1, which means runtime updates are disabled; "
59 "enabling this option is not safe, it may lead to further "
60 "corruption on Oopses)");
6dda9266
TL
61
62static int pstore_new_entry;
63
24ed960a 64static void pstore_timefunc(struct timer_list *);
1d27e3e2 65static DEFINE_TIMER(pstore_timer, pstore_timefunc);
6dda9266
TL
66
67static void pstore_dowork(struct work_struct *);
68static DECLARE_WORK(pstore_work, pstore_dowork);
69
ca01d6dd
TL
70/*
71 * pstore_lock just protects "psinfo" during
72 * calls to pstore_register()
73 */
74static DEFINE_SPINLOCK(pstore_lock);
060287b8 75struct pstore_info *psinfo;
ca01d6dd 76
dee28e72
MG
77static char *backend;
78
b0aad7a9 79/* Compression parameters */
8cfc8ddc 80#ifdef CONFIG_PSTORE_ZLIB_COMPRESS
b0aad7a9
AB
81#define COMPR_LEVEL 6
82#define WINDOW_BITS 12
83#define MEM_LEVEL 4
84static struct z_stream_s stream;
8cfc8ddc
GT
85#else
86static unsigned char *workspace;
87#endif
88
89struct pstore_zbackend {
90 int (*compress)(const void *in, void *out, size_t inlen, size_t outlen);
91 int (*decompress)(void *in, void *out, size_t inlen, size_t outlen);
92 void (*allocate)(void);
93 void (*free)(void);
94
95 const char *name;
96};
b0aad7a9
AB
97
98static char *big_oops_buf;
99static size_t big_oops_buf_sz;
100
366f7e7a 101/* How much of the console log to snapshot */
349d7438 102unsigned long kmsg_bytes = PSTORE_DEFAULT_KMSG_BYTES;
ca01d6dd 103
366f7e7a 104void pstore_set_kmsg_bytes(int bytes)
ca01d6dd 105{
366f7e7a 106 kmsg_bytes = bytes;
ca01d6dd
TL
107}
108
ca01d6dd
TL
109/* Tag each group of saved records with a sequence number */
110static int oopscount;
111
381b872c
SA
112static const char *get_reason_str(enum kmsg_dump_reason reason)
113{
114 switch (reason) {
115 case KMSG_DUMP_PANIC:
116 return "Panic";
117 case KMSG_DUMP_OOPS:
118 return "Oops";
119 case KMSG_DUMP_EMERG:
120 return "Emergency";
121 case KMSG_DUMP_RESTART:
122 return "Restart";
123 case KMSG_DUMP_HALT:
124 return "Halt";
125 case KMSG_DUMP_POWEROFF:
126 return "Poweroff";
127 default:
128 return "Unknown";
129 }
130}
9f6af27f 131
9f244e9c
SA
132bool pstore_cannot_block_path(enum kmsg_dump_reason reason)
133{
134 /*
135 * In case of NMI path, pstore shouldn't be blocked
136 * regardless of reason.
137 */
138 if (in_nmi())
139 return true;
140
141 switch (reason) {
142 /* In panic case, other cpus are stopped by smp_send_stop(). */
143 case KMSG_DUMP_PANIC:
144 /* Emergency restart shouldn't be blocked by spin lock. */
145 case KMSG_DUMP_EMERG:
146 return true;
147 default:
148 return false;
149 }
150}
151EXPORT_SYMBOL_GPL(pstore_cannot_block_path);
152
8cfc8ddc 153#ifdef CONFIG_PSTORE_ZLIB_COMPRESS
b0aad7a9 154/* Derived from logfs_compress() */
8cfc8ddc 155static int compress_zlib(const void *in, void *out, size_t inlen, size_t outlen)
b0aad7a9
AB
156{
157 int err, ret;
158
159 ret = -EIO;
160 err = zlib_deflateInit2(&stream, COMPR_LEVEL, Z_DEFLATED, WINDOW_BITS,
161 MEM_LEVEL, Z_DEFAULT_STRATEGY);
162 if (err != Z_OK)
163 goto error;
164
165 stream.next_in = in;
166 stream.avail_in = inlen;
167 stream.total_in = 0;
168 stream.next_out = out;
169 stream.avail_out = outlen;
170 stream.total_out = 0;
171
172 err = zlib_deflate(&stream, Z_FINISH);
173 if (err != Z_STREAM_END)
174 goto error;
175
176 err = zlib_deflateEnd(&stream);
177 if (err != Z_OK)
178 goto error;
179
180 if (stream.total_out >= stream.total_in)
181 goto error;
182
183 ret = stream.total_out;
184error:
185 return ret;
186}
187
adb42f5e 188/* Derived from logfs_uncompress */
8cfc8ddc 189static int decompress_zlib(void *in, void *out, size_t inlen, size_t outlen)
adb42f5e
AB
190{
191 int err, ret;
192
193 ret = -EIO;
b61edf8e 194 err = zlib_inflateInit2(&stream, WINDOW_BITS);
adb42f5e
AB
195 if (err != Z_OK)
196 goto error;
197
198 stream.next_in = in;
199 stream.avail_in = inlen;
200 stream.total_in = 0;
201 stream.next_out = out;
202 stream.avail_out = outlen;
203 stream.total_out = 0;
204
205 err = zlib_inflate(&stream, Z_FINISH);
206 if (err != Z_STREAM_END)
207 goto error;
208
209 err = zlib_inflateEnd(&stream);
210 if (err != Z_OK)
211 goto error;
212
213 ret = stream.total_out;
214error:
215 return ret;
216}
217
8cfc8ddc 218static void allocate_zlib(void)
b0aad7a9
AB
219{
220 size_t size;
7de8fe2f
AB
221 size_t cmpr;
222
223 switch (psinfo->bufsize) {
224 /* buffer range for efivars */
225 case 1000 ... 2000:
226 cmpr = 56;
227 break;
228 case 2001 ... 3000:
229 cmpr = 54;
230 break;
231 case 3001 ... 3999:
232 cmpr = 52;
233 break;
234 /* buffer range for nvram, erst */
235 case 4000 ... 10000:
236 cmpr = 45;
237 break;
238 default:
239 cmpr = 60;
240 break;
241 }
b0aad7a9 242
7de8fe2f 243 big_oops_buf_sz = (psinfo->bufsize * 100) / cmpr;
b0aad7a9
AB
244 big_oops_buf = kmalloc(big_oops_buf_sz, GFP_KERNEL);
245 if (big_oops_buf) {
246 size = max(zlib_deflate_workspacesize(WINDOW_BITS, MEM_LEVEL),
247 zlib_inflate_workspacesize());
248 stream.workspace = kmalloc(size, GFP_KERNEL);
249 if (!stream.workspace) {
ef748853 250 pr_err("No memory for compression workspace; skipping compression\n");
b0aad7a9
AB
251 kfree(big_oops_buf);
252 big_oops_buf = NULL;
253 }
254 } else {
ef748853 255 pr_err("No memory for uncompressed data; skipping compression\n");
b0aad7a9
AB
256 stream.workspace = NULL;
257 }
258
259}
260
8cfc8ddc 261static void free_zlib(void)
ee1d2674
GT
262{
263 kfree(stream.workspace);
264 stream.workspace = NULL;
265 kfree(big_oops_buf);
266 big_oops_buf = NULL;
8cfc8ddc
GT
267 big_oops_buf_sz = 0;
268}
269
3faf9354 270static const struct pstore_zbackend backend_zlib = {
8cfc8ddc
GT
271 .compress = compress_zlib,
272 .decompress = decompress_zlib,
273 .allocate = allocate_zlib,
274 .free = free_zlib,
275 .name = "zlib",
276};
277#endif
278
279#ifdef CONFIG_PSTORE_LZO_COMPRESS
280static int compress_lzo(const void *in, void *out, size_t inlen, size_t outlen)
281{
282 int ret;
283
284 ret = lzo1x_1_compress(in, inlen, out, &outlen, workspace);
285 if (ret != LZO_E_OK) {
286 pr_err("lzo_compress error, ret = %d!\n", ret);
287 return -EIO;
288 }
289
290 return outlen;
291}
292
293static int decompress_lzo(void *in, void *out, size_t inlen, size_t outlen)
294{
295 int ret;
296
297 ret = lzo1x_decompress_safe(in, inlen, out, &outlen);
298 if (ret != LZO_E_OK) {
299 pr_err("lzo_decompress error, ret = %d!\n", ret);
300 return -EIO;
301 }
302
303 return outlen;
304}
305
306static void allocate_lzo(void)
307{
308 big_oops_buf_sz = lzo1x_worst_compress(psinfo->bufsize);
309 big_oops_buf = kmalloc(big_oops_buf_sz, GFP_KERNEL);
310 if (big_oops_buf) {
311 workspace = kmalloc(LZO1X_MEM_COMPRESS, GFP_KERNEL);
312 if (!workspace) {
313 pr_err("No memory for compression workspace; skipping compression\n");
314 kfree(big_oops_buf);
315 big_oops_buf = NULL;
316 }
317 } else {
318 pr_err("No memory for uncompressed data; skipping compression\n");
319 workspace = NULL;
320 }
321}
322
323static void free_lzo(void)
324{
325 kfree(workspace);
326 kfree(big_oops_buf);
327 big_oops_buf = NULL;
328 big_oops_buf_sz = 0;
329}
330
3faf9354 331static const struct pstore_zbackend backend_lzo = {
8cfc8ddc
GT
332 .compress = compress_lzo,
333 .decompress = decompress_lzo,
334 .allocate = allocate_lzo,
335 .free = free_lzo,
336 .name = "lzo",
337};
338#endif
339
340#ifdef CONFIG_PSTORE_LZ4_COMPRESS
341static int compress_lz4(const void *in, void *out, size_t inlen, size_t outlen)
342{
343 int ret;
344
d21b5ff1
SS
345 ret = LZ4_compress_default(in, out, inlen, outlen, workspace);
346 if (!ret) {
347 pr_err("LZ4_compress_default error; compression failed!\n");
8cfc8ddc
GT
348 return -EIO;
349 }
350
d21b5ff1 351 return ret;
8cfc8ddc
GT
352}
353
354static int decompress_lz4(void *in, void *out, size_t inlen, size_t outlen)
355{
356 int ret;
357
d21b5ff1
SS
358 ret = LZ4_decompress_safe(in, out, inlen, outlen);
359 if (ret < 0) {
360 /*
361 * LZ4_decompress_safe will return an error code
362 * (< 0) if decompression failed
363 */
364 pr_err("LZ4_decompress_safe error, ret = %d!\n", ret);
8cfc8ddc
GT
365 return -EIO;
366 }
367
d21b5ff1 368 return ret;
8cfc8ddc
GT
369}
370
371static void allocate_lz4(void)
372{
d21b5ff1 373 big_oops_buf_sz = LZ4_compressBound(psinfo->bufsize);
8cfc8ddc
GT
374 big_oops_buf = kmalloc(big_oops_buf_sz, GFP_KERNEL);
375 if (big_oops_buf) {
376 workspace = kmalloc(LZ4_MEM_COMPRESS, GFP_KERNEL);
377 if (!workspace) {
378 pr_err("No memory for compression workspace; skipping compression\n");
379 kfree(big_oops_buf);
380 big_oops_buf = NULL;
381 }
382 } else {
383 pr_err("No memory for uncompressed data; skipping compression\n");
384 workspace = NULL;
385 }
386}
387
388static void free_lz4(void)
389{
390 kfree(workspace);
391 kfree(big_oops_buf);
392 big_oops_buf = NULL;
393 big_oops_buf_sz = 0;
394}
395
3faf9354 396static const struct pstore_zbackend backend_lz4 = {
8cfc8ddc
GT
397 .compress = compress_lz4,
398 .decompress = decompress_lz4,
399 .allocate = allocate_lz4,
400 .free = free_lz4,
401 .name = "lz4",
402};
403#endif
404
3faf9354 405static const struct pstore_zbackend *zbackend =
8cfc8ddc
GT
406#if defined(CONFIG_PSTORE_ZLIB_COMPRESS)
407 &backend_zlib;
408#elif defined(CONFIG_PSTORE_LZO_COMPRESS)
409 &backend_lzo;
410#elif defined(CONFIG_PSTORE_LZ4_COMPRESS)
411 &backend_lz4;
412#else
413 NULL;
414#endif
415
416static int pstore_compress(const void *in, void *out,
417 size_t inlen, size_t outlen)
418{
419 if (zbackend)
420 return zbackend->compress(in, out, inlen, outlen);
421 else
422 return -EIO;
423}
424
425static int pstore_decompress(void *in, void *out, size_t inlen, size_t outlen)
426{
427 if (zbackend)
428 return zbackend->decompress(in, out, inlen, outlen);
429 else
430 return -EIO;
431}
432
433static void allocate_buf_for_compression(void)
434{
435 if (zbackend) {
436 pr_info("using %s compression\n", zbackend->name);
437 zbackend->allocate();
438 } else {
439 pr_err("allocate compression buffer error!\n");
440 }
441}
442
443static void free_buf_for_compression(void)
444{
445 if (zbackend)
446 zbackend->free();
447 else
448 pr_err("free compression buffer error!\n");
ee1d2674
GT
449}
450
b0aad7a9
AB
451/*
452 * Called when compression fails, since the printk buffer
453 * would be fetched for compression calling it again when
454 * compression fails would have moved the iterator of
455 * printk buffer which results in fetching old contents.
456 * Copy the recent messages from big_oops_buf to psinfo->buf
457 */
458static size_t copy_kmsg_to_buffer(int hsize, size_t len)
459{
460 size_t total_len;
461 size_t diff;
462
463 total_len = hsize + len;
464
465 if (total_len > psinfo->bufsize) {
466 diff = total_len - psinfo->bufsize + hsize;
467 memcpy(psinfo->buf, big_oops_buf, hsize);
468 memcpy(psinfo->buf + hsize, big_oops_buf + diff,
469 psinfo->bufsize - hsize);
470 total_len = psinfo->bufsize;
471 } else
472 memcpy(psinfo->buf, big_oops_buf, total_len);
473
474 return total_len;
475}
476
e581ca81
KC
477void pstore_record_init(struct pstore_record *record,
478 struct pstore_info *psinfo)
479{
480 memset(record, 0, sizeof(*record));
481
482 record->psi = psinfo;
c7f3c595
KC
483
484 /* Report zeroed timestamp if called before timekeeping has resumed. */
df27067e 485 record->time = ns_to_timespec(ktime_get_real_fast_ns());
e581ca81
KC
486}
487
ca01d6dd
TL
488/*
489 * callback from kmsg_dump. (s2,l2) has the most recently
490 * written bytes, older bytes are in (s1,l1). Save as much
491 * as we can from the end of the buffer.
492 */
493static void pstore_dump(struct kmsg_dumper *dumper,
e2ae715d 494 enum kmsg_dump_reason reason)
ca01d6dd 495{
e2ae715d 496 unsigned long total = 0;
381b872c 497 const char *why;
b94fdd07 498 unsigned int part = 1;
abd4d558 499 unsigned long flags = 0;
98e44fda 500 int is_locked;
e2ae715d 501 int ret;
ca01d6dd 502
381b872c 503 why = get_reason_str(reason);
9f6af27f 504
9f244e9c
SA
505 if (pstore_cannot_block_path(reason)) {
506 is_locked = spin_trylock_irqsave(&psinfo->buf_lock, flags);
507 if (!is_locked) {
508 pr_err("pstore dump routine blocked in %s path, may corrupt error record\n"
509 , in_nmi() ? "NMI" : why);
959217c8 510 return;
9f244e9c 511 }
98e44fda 512 } else {
abd4d558 513 spin_lock_irqsave(&psinfo->buf_lock, flags);
98e44fda
NK
514 is_locked = 1;
515 }
ca01d6dd
TL
516 oopscount++;
517 while (total < kmsg_bytes) {
e2ae715d 518 char *dst;
76cc9580
KC
519 size_t dst_size;
520 int header_size;
b0aad7a9 521 int zipped_len = -1;
76cc9580 522 size_t dump_size;
e581ca81
KC
523 struct pstore_record record;
524
525 pstore_record_init(&record, psinfo);
526 record.type = PSTORE_TYPE_DMESG;
527 record.count = oopscount;
528 record.reason = reason;
529 record.part = part;
530 record.buf = psinfo->buf;
e2ae715d 531
f0e2efcf 532 if (big_oops_buf && is_locked) {
b0aad7a9 533 dst = big_oops_buf;
76cc9580 534 dst_size = big_oops_buf_sz;
235f6d15
NK
535 } else {
536 dst = psinfo->buf;
76cc9580 537 dst_size = psinfo->bufsize;
235f6d15
NK
538 }
539
76cc9580
KC
540 /* Write dump header. */
541 header_size = snprintf(dst, dst_size, "%s#%d Part%u\n", why,
542 oopscount, part);
543 dst_size -= header_size;
ca01d6dd 544
76cc9580
KC
545 /* Write dump contents. */
546 if (!kmsg_dump_get_buffer(dumper, true, dst + header_size,
547 dst_size, &dump_size))
235f6d15 548 break;
b0aad7a9 549
235f6d15 550 if (big_oops_buf && is_locked) {
b0aad7a9 551 zipped_len = pstore_compress(dst, psinfo->buf,
76cc9580
KC
552 header_size + dump_size,
553 psinfo->bufsize);
b0aad7a9
AB
554
555 if (zipped_len > 0) {
76cc9580
KC
556 record.compressed = true;
557 record.size = zipped_len;
b0aad7a9 558 } else {
76cc9580
KC
559 record.size = copy_kmsg_to_buffer(header_size,
560 dump_size);
b0aad7a9
AB
561 }
562 } else {
76cc9580 563 record.size = header_size + dump_size;
b0aad7a9 564 }
ca01d6dd 565
76cc9580 566 ret = psinfo->write(&record);
b238b8fa 567 if (ret == 0 && reason == KMSG_DUMP_OOPS && pstore_is_mounted())
6dda9266 568 pstore_new_entry = 1;
e2ae715d 569
76cc9580 570 total += record.size;
56280682 571 part++;
ca01d6dd 572 }
98e44fda 573 if (is_locked)
abd4d558 574 spin_unlock_irqrestore(&psinfo->buf_lock, flags);
ca01d6dd
TL
575}
576
577static struct kmsg_dumper pstore_dumper = {
578 .dump = pstore_dump,
579};
580
306e5c2a
GT
581/*
582 * Register with kmsg_dump to save last part of console log on panic.
583 */
18730411
GT
584static void pstore_register_kmsg(void)
585{
586 kmsg_dump_register(&pstore_dumper);
587}
588
ee1d2674
GT
589static void pstore_unregister_kmsg(void)
590{
591 kmsg_dump_unregister(&pstore_dumper);
592}
593
f29e5956
AV
594#ifdef CONFIG_PSTORE_CONSOLE
595static void pstore_console_write(struct console *con, const char *s, unsigned c)
596{
597 const char *e = s + c;
598
599 while (s < e) {
e581ca81 600 struct pstore_record record;
f29e5956
AV
601 unsigned long flags;
602
e581ca81
KC
603 pstore_record_init(&record, psinfo);
604 record.type = PSTORE_TYPE_CONSOLE;
605
f29e5956
AV
606 if (c > psinfo->bufsize)
607 c = psinfo->bufsize;
80c9d03c
CL
608
609 if (oops_in_progress) {
610 if (!spin_trylock_irqsave(&psinfo->buf_lock, flags))
611 break;
612 } else {
613 spin_lock_irqsave(&psinfo->buf_lock, flags);
614 }
b10b4711
KC
615 record.buf = (char *)s;
616 record.size = c;
4c9ec219 617 psinfo->write(&record);
f29e5956
AV
618 spin_unlock_irqrestore(&psinfo->buf_lock, flags);
619 s += c;
620 c = e - s;
621 }
622}
623
624static struct console pstore_console = {
625 .name = "pstore",
626 .write = pstore_console_write,
627 .flags = CON_PRINTBUFFER | CON_ENABLED | CON_ANYTIME,
628 .index = -1,
629};
630
631static void pstore_register_console(void)
632{
633 register_console(&pstore_console);
634}
ee1d2674
GT
635
636static void pstore_unregister_console(void)
637{
638 unregister_console(&pstore_console);
639}
f29e5956
AV
640#else
641static void pstore_register_console(void) {}
ee1d2674 642static void pstore_unregister_console(void) {}
f29e5956
AV
643#endif
644
4c9ec219
KC
645static int pstore_write_user_compat(struct pstore_record *record,
646 const char __user *buf)
5bf6d1b9 647{
30800d99
KC
648 int ret = 0;
649
650 if (record->buf)
651 return -EINVAL;
652
077090af 653 record->buf = memdup_user(buf, record->size);
dfd6fa39 654 if (IS_ERR(record->buf)) {
077090af 655 ret = PTR_ERR(record->buf);
30800d99 656 goto out;
5bf6d1b9 657 }
30800d99
KC
658
659 ret = record->psi->write(record);
660
30800d99 661 kfree(record->buf);
077090af 662out:
30800d99
KC
663 record->buf = NULL;
664
665 return unlikely(ret < 0) ? ret : record->size;
5bf6d1b9
MS
666}
667
ca01d6dd
TL
668/*
669 * platform specific persistent storage driver registers with
670 * us here. If pstore is already mounted, call the platform
671 * read function right away to populate the file system. If not
672 * then the pstore mount code will call us later to fill out
673 * the file system.
ca01d6dd
TL
674 */
675int pstore_register(struct pstore_info *psi)
676{
677 struct module *owner = psi->owner;
678
0d7cd09a
KC
679 if (backend && strcmp(backend, psi->name)) {
680 pr_warn("ignoring unexpected backend '%s'\n", psi->name);
8e48b1a8 681 return -EPERM;
0d7cd09a 682 }
8e48b1a8 683
4c9ec219
KC
684 /* Sanity check flags. */
685 if (!psi->flags) {
686 pr_warn("backend '%s' must support at least one frontend\n",
687 psi->name);
688 return -EINVAL;
689 }
690
691 /* Check for required functions. */
692 if (!psi->read || !psi->write) {
693 pr_warn("backend '%s' must implement read() and write()\n",
694 psi->name);
695 return -EINVAL;
696 }
697
ca01d6dd
TL
698 spin_lock(&pstore_lock);
699 if (psinfo) {
0d7cd09a
KC
700 pr_warn("backend '%s' already loaded: ignoring '%s'\n",
701 psinfo->name, psi->name);
ca01d6dd
TL
702 spin_unlock(&pstore_lock);
703 return -EBUSY;
704 }
dee28e72 705
4c9ec219
KC
706 if (!psi->write_user)
707 psi->write_user = pstore_write_user_compat;
ca01d6dd 708 psinfo = psi;
f6f82851 709 mutex_init(&psinfo->read_mutex);
ca01d6dd
TL
710 spin_unlock(&pstore_lock);
711
712 if (owner && !try_module_get(owner)) {
713 psinfo = NULL;
714 return -EINVAL;
715 }
716
b0aad7a9
AB
717 allocate_buf_for_compression();
718
ca01d6dd 719 if (pstore_is_mounted())
6dda9266 720 pstore_get_records(0);
ca01d6dd 721
c950fd6f
NK
722 if (psi->flags & PSTORE_FLAGS_DMESG)
723 pstore_register_kmsg();
724 if (psi->flags & PSTORE_FLAGS_CONSOLE)
df36ac1b 725 pstore_register_console();
c950fd6f 726 if (psi->flags & PSTORE_FLAGS_FTRACE)
df36ac1b 727 pstore_register_ftrace();
c950fd6f 728 if (psi->flags & PSTORE_FLAGS_PMSG)
9d5438f4 729 pstore_register_pmsg();
ca01d6dd 730
6330d553 731 /* Start watching for new records, if desired. */
a3f5f075
AV
732 if (pstore_update_ms >= 0) {
733 pstore_timer.expires = jiffies +
734 msecs_to_jiffies(pstore_update_ms);
735 add_timer(&pstore_timer);
736 }
6dda9266 737
42222c2a
WL
738 /*
739 * Update the module parameter backend, so it is visible
740 * through /sys/module/pstore/parameters/backend
741 */
742 backend = psi->name;
743
ef748853 744 pr_info("Registered %s as persistent store backend\n", psi->name);
8e48b1a8 745
1344dd86
KC
746 module_put(owner);
747
ca01d6dd
TL
748 return 0;
749}
750EXPORT_SYMBOL_GPL(pstore_register);
751
ee1d2674
GT
752void pstore_unregister(struct pstore_info *psi)
753{
6330d553
KC
754 /* Stop timer and make sure all work has finished. */
755 pstore_update_ms = -1;
756 del_timer_sync(&pstore_timer);
757 flush_work(&pstore_work);
758
c950fd6f 759 if (psi->flags & PSTORE_FLAGS_PMSG)
a1db8060 760 pstore_unregister_pmsg();
c950fd6f 761 if (psi->flags & PSTORE_FLAGS_FTRACE)
a1db8060 762 pstore_unregister_ftrace();
c950fd6f 763 if (psi->flags & PSTORE_FLAGS_CONSOLE)
a1db8060 764 pstore_unregister_console();
c950fd6f
NK
765 if (psi->flags & PSTORE_FLAGS_DMESG)
766 pstore_unregister_kmsg();
ee1d2674
GT
767
768 free_buf_for_compression();
769
770 psinfo = NULL;
771 backend = NULL;
772}
773EXPORT_SYMBOL_GPL(pstore_unregister);
774
634f8f51
KC
775static void decompress_record(struct pstore_record *record)
776{
777 int unzipped_len;
7e8cc8dc 778 char *decompressed;
634f8f51 779
4a16d1cb
AK
780 if (!record->compressed)
781 return;
782
634f8f51 783 /* Only PSTORE_TYPE_DMESG support compression. */
4a16d1cb 784 if (record->type != PSTORE_TYPE_DMESG) {
634f8f51
KC
785 pr_warn("ignored compressed record type %d\n", record->type);
786 return;
787 }
788
789 /* No compression method has created the common buffer. */
790 if (!big_oops_buf) {
791 pr_warn("no decompression buffer allocated\n");
792 return;
793 }
794
795 unzipped_len = pstore_decompress(record->buf, big_oops_buf,
796 record->size, big_oops_buf_sz);
7e8cc8dc 797 if (unzipped_len <= 0) {
634f8f51 798 pr_err("decompression failed: %d\n", unzipped_len);
7e8cc8dc
KC
799 return;
800 }
801
802 /* Build new buffer for decompressed contents. */
803 decompressed = kmalloc(unzipped_len + record->ecc_notice_size,
804 GFP_KERNEL);
805 if (!decompressed) {
806 pr_err("decompression ran out of memory\n");
807 return;
808 }
809 memcpy(decompressed, big_oops_buf, unzipped_len);
810
811 /* Append ECC notice to decompressed buffer. */
812 memcpy(decompressed + unzipped_len, record->buf + record->size,
813 record->ecc_notice_size);
814
815 /* Swap out compresed contents with decompressed contents. */
816 kfree(record->buf);
817 record->buf = decompressed;
818 record->size = unzipped_len;
819 record->compressed = false;
634f8f51
KC
820}
821
ca01d6dd 822/*
3a7d2fd1 823 * Read all the records from one persistent store backend. Create
6dda9266
TL
824 * files in our filesystem. Don't warn about -EEXIST errors
825 * when we are re-scanning the backing store looking to add new
826 * error records.
ca01d6dd 827 */
3a7d2fd1
KC
828void pstore_get_backend_records(struct pstore_info *psi,
829 struct dentry *root, int quiet)
ca01d6dd 830{
2a2b0acf 831 int failed = 0;
656de42e 832 unsigned int stop_loop = 65536;
ca01d6dd 833
3a7d2fd1 834 if (!psi || !root)
ca01d6dd
TL
835 return;
836
f6f82851 837 mutex_lock(&psi->read_mutex);
2174f6df 838 if (psi->open && psi->open(psi))
06cf91b4
CG
839 goto out;
840
1dfff7dd
KC
841 /*
842 * Backend callback read() allocates record.buf. decompress_record()
843 * may reallocate record.buf. On success, pstore_mkfile() will keep
844 * the record.buf, so free it only on failure.
845 */
656de42e 846 for (; stop_loop; stop_loop--) {
2a2b0acf
KC
847 struct pstore_record *record;
848 int rc;
849
850 record = kzalloc(sizeof(*record), GFP_KERNEL);
851 if (!record) {
852 pr_err("out of memory creating record\n");
853 break;
854 }
e581ca81 855 pstore_record_init(record, psi);
2a2b0acf
KC
856
857 record->size = psi->read(record);
858
859 /* No more records left in backend? */
f6525b96
DA
860 if (record->size <= 0) {
861 kfree(record);
2a2b0acf 862 break;
f6525b96 863 }
2a2b0acf
KC
864
865 decompress_record(record);
3a7d2fd1 866 rc = pstore_mkfile(root, record);
1dfff7dd 867 if (rc) {
83f70f07 868 /* pstore_mkfile() did not take record, so free it. */
2a2b0acf 869 kfree(record->buf);
83f70f07 870 kfree(record);
1dfff7dd
KC
871 if (rc != -EEXIST || !quiet)
872 failed++;
873 }
ca01d6dd 874 }
2174f6df
KC
875 if (psi->close)
876 psi->close(psi);
06cf91b4 877out:
f6f82851 878 mutex_unlock(&psi->read_mutex);
ca01d6dd
TL
879
880 if (failed)
656de42e 881 pr_warn("failed to create %d record(s) from '%s'\n",
ef748853 882 failed, psi->name);
656de42e
KC
883 if (!stop_loop)
884 pr_err("looping? Too many records seen from '%s'\n",
885 psi->name);
ca01d6dd
TL
886}
887
6dda9266
TL
888static void pstore_dowork(struct work_struct *work)
889{
890 pstore_get_records(1);
891}
892
24ed960a 893static void pstore_timefunc(struct timer_list *unused)
6dda9266
TL
894{
895 if (pstore_new_entry) {
896 pstore_new_entry = 0;
897 schedule_work(&pstore_work);
898 }
899
6330d553
KC
900 if (pstore_update_ms >= 0)
901 mod_timer(&pstore_timer,
902 jiffies + msecs_to_jiffies(pstore_update_ms));
6dda9266
TL
903}
904
dee28e72
MG
905module_param(backend, charp, 0444);
906MODULE_PARM_DESC(backend, "Pstore backend to use");