]> git.proxmox.com Git - mirror_ubuntu-artful-kernel.git/blob - fs/pstore/platform.c
pstore: Use zlib_inflateInit2 instead of zlib_inflateInit
[mirror_ubuntu-artful-kernel.git] / fs / pstore / platform.c
1 /*
2 * Persistent Storage - platform driver interface parts.
3 *
4 * Copyright (C) 2007-2008 Google, Inc.
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
21 #include <linux/atomic.h>
22 #include <linux/types.h>
23 #include <linux/errno.h>
24 #include <linux/init.h>
25 #include <linux/kmsg_dump.h>
26 #include <linux/console.h>
27 #include <linux/module.h>
28 #include <linux/pstore.h>
29 #include <linux/zlib.h>
30 #include <linux/string.h>
31 #include <linux/timer.h>
32 #include <linux/slab.h>
33 #include <linux/uaccess.h>
34 #include <linux/hardirq.h>
35 #include <linux/jiffies.h>
36 #include <linux/workqueue.h>
37
38 #include "internal.h"
39
40 /*
41 * We defer making "oops" entries appear in pstore - see
42 * whether the system is actually still running well enough
43 * to let someone see the entry
44 */
45 static int pstore_update_ms = -1;
46 module_param_named(update_ms, pstore_update_ms, int, 0600);
47 MODULE_PARM_DESC(update_ms, "milliseconds before pstore updates its content "
48 "(default is -1, which means runtime updates are disabled; "
49 "enabling this option is not safe, it may lead to further "
50 "corruption on Oopses)");
51
52 static int pstore_new_entry;
53
54 static void pstore_timefunc(unsigned long);
55 static DEFINE_TIMER(pstore_timer, pstore_timefunc, 0, 0);
56
57 static void pstore_dowork(struct work_struct *);
58 static DECLARE_WORK(pstore_work, pstore_dowork);
59
60 /*
61 * pstore_lock just protects "psinfo" during
62 * calls to pstore_register()
63 */
64 static DEFINE_SPINLOCK(pstore_lock);
65 struct pstore_info *psinfo;
66
67 static char *backend;
68
69 /* Compression parameters */
70 #define COMPR_LEVEL 6
71 #define WINDOW_BITS 12
72 #define MEM_LEVEL 4
73 static struct z_stream_s stream;
74
75 static char *big_oops_buf;
76 static size_t big_oops_buf_sz;
77
78 /* How much of the console log to snapshot */
79 static unsigned long kmsg_bytes = 10240;
80
81 void pstore_set_kmsg_bytes(int bytes)
82 {
83 kmsg_bytes = bytes;
84 }
85
86 /* Tag each group of saved records with a sequence number */
87 static int oopscount;
88
89 static const char *get_reason_str(enum kmsg_dump_reason reason)
90 {
91 switch (reason) {
92 case KMSG_DUMP_PANIC:
93 return "Panic";
94 case KMSG_DUMP_OOPS:
95 return "Oops";
96 case KMSG_DUMP_EMERG:
97 return "Emergency";
98 case KMSG_DUMP_RESTART:
99 return "Restart";
100 case KMSG_DUMP_HALT:
101 return "Halt";
102 case KMSG_DUMP_POWEROFF:
103 return "Poweroff";
104 default:
105 return "Unknown";
106 }
107 }
108
109 bool pstore_cannot_block_path(enum kmsg_dump_reason reason)
110 {
111 /*
112 * In case of NMI path, pstore shouldn't be blocked
113 * regardless of reason.
114 */
115 if (in_nmi())
116 return true;
117
118 switch (reason) {
119 /* In panic case, other cpus are stopped by smp_send_stop(). */
120 case KMSG_DUMP_PANIC:
121 /* Emergency restart shouldn't be blocked by spin lock. */
122 case KMSG_DUMP_EMERG:
123 return true;
124 default:
125 return false;
126 }
127 }
128 EXPORT_SYMBOL_GPL(pstore_cannot_block_path);
129
130 /* Derived from logfs_compress() */
131 static int pstore_compress(const void *in, void *out, size_t inlen,
132 size_t outlen)
133 {
134 int err, ret;
135
136 ret = -EIO;
137 err = zlib_deflateInit2(&stream, COMPR_LEVEL, Z_DEFLATED, WINDOW_BITS,
138 MEM_LEVEL, Z_DEFAULT_STRATEGY);
139 if (err != Z_OK)
140 goto error;
141
142 stream.next_in = in;
143 stream.avail_in = inlen;
144 stream.total_in = 0;
145 stream.next_out = out;
146 stream.avail_out = outlen;
147 stream.total_out = 0;
148
149 err = zlib_deflate(&stream, Z_FINISH);
150 if (err != Z_STREAM_END)
151 goto error;
152
153 err = zlib_deflateEnd(&stream);
154 if (err != Z_OK)
155 goto error;
156
157 if (stream.total_out >= stream.total_in)
158 goto error;
159
160 ret = stream.total_out;
161 error:
162 return ret;
163 }
164
165 /* Derived from logfs_uncompress */
166 static int pstore_decompress(void *in, void *out, size_t inlen, size_t outlen)
167 {
168 int err, ret;
169
170 ret = -EIO;
171 err = zlib_inflateInit2(&stream, WINDOW_BITS);
172 if (err != Z_OK)
173 goto error;
174
175 stream.next_in = in;
176 stream.avail_in = inlen;
177 stream.total_in = 0;
178 stream.next_out = out;
179 stream.avail_out = outlen;
180 stream.total_out = 0;
181
182 err = zlib_inflate(&stream, Z_FINISH);
183 if (err != Z_STREAM_END)
184 goto error;
185
186 err = zlib_inflateEnd(&stream);
187 if (err != Z_OK)
188 goto error;
189
190 ret = stream.total_out;
191 error:
192 return ret;
193 }
194
195 static void allocate_buf_for_compression(void)
196 {
197 size_t size;
198 size_t cmpr;
199
200 switch (psinfo->bufsize) {
201 /* buffer range for efivars */
202 case 1000 ... 2000:
203 cmpr = 56;
204 break;
205 case 2001 ... 3000:
206 cmpr = 54;
207 break;
208 case 3001 ... 3999:
209 cmpr = 52;
210 break;
211 /* buffer range for nvram, erst */
212 case 4000 ... 10000:
213 cmpr = 45;
214 break;
215 default:
216 cmpr = 60;
217 break;
218 }
219
220 big_oops_buf_sz = (psinfo->bufsize * 100) / cmpr;
221 big_oops_buf = kmalloc(big_oops_buf_sz, GFP_KERNEL);
222 if (big_oops_buf) {
223 size = max(zlib_deflate_workspacesize(WINDOW_BITS, MEM_LEVEL),
224 zlib_inflate_workspacesize());
225 stream.workspace = kmalloc(size, GFP_KERNEL);
226 if (!stream.workspace) {
227 pr_err("pstore: No memory for compression workspace; "
228 "skipping compression\n");
229 kfree(big_oops_buf);
230 big_oops_buf = NULL;
231 }
232 } else {
233 pr_err("No memory for uncompressed data; "
234 "skipping compression\n");
235 stream.workspace = NULL;
236 }
237
238 }
239
240 /*
241 * Called when compression fails, since the printk buffer
242 * would be fetched for compression calling it again when
243 * compression fails would have moved the iterator of
244 * printk buffer which results in fetching old contents.
245 * Copy the recent messages from big_oops_buf to psinfo->buf
246 */
247 static size_t copy_kmsg_to_buffer(int hsize, size_t len)
248 {
249 size_t total_len;
250 size_t diff;
251
252 total_len = hsize + len;
253
254 if (total_len > psinfo->bufsize) {
255 diff = total_len - psinfo->bufsize + hsize;
256 memcpy(psinfo->buf, big_oops_buf, hsize);
257 memcpy(psinfo->buf + hsize, big_oops_buf + diff,
258 psinfo->bufsize - hsize);
259 total_len = psinfo->bufsize;
260 } else
261 memcpy(psinfo->buf, big_oops_buf, total_len);
262
263 return total_len;
264 }
265
266 /*
267 * callback from kmsg_dump. (s2,l2) has the most recently
268 * written bytes, older bytes are in (s1,l1). Save as much
269 * as we can from the end of the buffer.
270 */
271 static void pstore_dump(struct kmsg_dumper *dumper,
272 enum kmsg_dump_reason reason)
273 {
274 unsigned long total = 0;
275 const char *why;
276 u64 id;
277 unsigned int part = 1;
278 unsigned long flags = 0;
279 int is_locked = 0;
280 int ret;
281
282 why = get_reason_str(reason);
283
284 if (pstore_cannot_block_path(reason)) {
285 is_locked = spin_trylock_irqsave(&psinfo->buf_lock, flags);
286 if (!is_locked) {
287 pr_err("pstore dump routine blocked in %s path, may corrupt error record\n"
288 , in_nmi() ? "NMI" : why);
289 }
290 } else
291 spin_lock_irqsave(&psinfo->buf_lock, flags);
292 oopscount++;
293 while (total < kmsg_bytes) {
294 char *dst;
295 unsigned long size;
296 int hsize;
297 int zipped_len = -1;
298 size_t len;
299 bool compressed;
300 size_t total_len;
301
302 if (big_oops_buf) {
303 dst = big_oops_buf;
304 hsize = sprintf(dst, "%s#%d Part%d\n", why,
305 oopscount, part);
306 size = big_oops_buf_sz - hsize;
307
308 if (!kmsg_dump_get_buffer(dumper, true, dst + hsize,
309 size, &len))
310 break;
311
312 zipped_len = pstore_compress(dst, psinfo->buf,
313 hsize + len, psinfo->bufsize);
314
315 if (zipped_len > 0) {
316 compressed = true;
317 total_len = zipped_len;
318 } else {
319 pr_err("pstore: compression failed for Part %d"
320 " returned %d\n", part, zipped_len);
321 pr_err("pstore: Capture uncompressed"
322 " oops/panic report of Part %d\n", part);
323 compressed = false;
324 total_len = copy_kmsg_to_buffer(hsize, len);
325 }
326 } else {
327 dst = psinfo->buf;
328 hsize = sprintf(dst, "%s#%d Part%d\n", why, oopscount,
329 part);
330 size = psinfo->bufsize - hsize;
331 dst += hsize;
332
333 if (!kmsg_dump_get_buffer(dumper, true, dst,
334 size, &len))
335 break;
336
337 compressed = false;
338 total_len = hsize + len;
339 }
340
341 ret = psinfo->write(PSTORE_TYPE_DMESG, reason, &id, part,
342 oopscount, compressed, total_len, psinfo);
343 if (ret == 0 && reason == KMSG_DUMP_OOPS && pstore_is_mounted())
344 pstore_new_entry = 1;
345
346 total += total_len;
347 part++;
348 }
349 if (pstore_cannot_block_path(reason)) {
350 if (is_locked)
351 spin_unlock_irqrestore(&psinfo->buf_lock, flags);
352 } else
353 spin_unlock_irqrestore(&psinfo->buf_lock, flags);
354 }
355
356 static struct kmsg_dumper pstore_dumper = {
357 .dump = pstore_dump,
358 };
359
360 #ifdef CONFIG_PSTORE_CONSOLE
361 static void pstore_console_write(struct console *con, const char *s, unsigned c)
362 {
363 const char *e = s + c;
364
365 while (s < e) {
366 unsigned long flags;
367 u64 id;
368
369 if (c > psinfo->bufsize)
370 c = psinfo->bufsize;
371
372 if (oops_in_progress) {
373 if (!spin_trylock_irqsave(&psinfo->buf_lock, flags))
374 break;
375 } else {
376 spin_lock_irqsave(&psinfo->buf_lock, flags);
377 }
378 memcpy(psinfo->buf, s, c);
379 psinfo->write(PSTORE_TYPE_CONSOLE, 0, &id, 0, 0, 0, c, psinfo);
380 spin_unlock_irqrestore(&psinfo->buf_lock, flags);
381 s += c;
382 c = e - s;
383 }
384 }
385
386 static struct console pstore_console = {
387 .name = "pstore",
388 .write = pstore_console_write,
389 .flags = CON_PRINTBUFFER | CON_ENABLED | CON_ANYTIME,
390 .index = -1,
391 };
392
393 static void pstore_register_console(void)
394 {
395 register_console(&pstore_console);
396 }
397 #else
398 static void pstore_register_console(void) {}
399 #endif
400
401 static int pstore_write_compat(enum pstore_type_id type,
402 enum kmsg_dump_reason reason,
403 u64 *id, unsigned int part, int count,
404 bool compressed, size_t size,
405 struct pstore_info *psi)
406 {
407 return psi->write_buf(type, reason, id, part, psinfo->buf, compressed,
408 size, psi);
409 }
410
411 /*
412 * platform specific persistent storage driver registers with
413 * us here. If pstore is already mounted, call the platform
414 * read function right away to populate the file system. If not
415 * then the pstore mount code will call us later to fill out
416 * the file system.
417 *
418 * Register with kmsg_dump to save last part of console log on panic.
419 */
420 int pstore_register(struct pstore_info *psi)
421 {
422 struct module *owner = psi->owner;
423
424 if (backend && strcmp(backend, psi->name))
425 return -EPERM;
426
427 spin_lock(&pstore_lock);
428 if (psinfo) {
429 spin_unlock(&pstore_lock);
430 return -EBUSY;
431 }
432
433 if (!psi->write)
434 psi->write = pstore_write_compat;
435 psinfo = psi;
436 mutex_init(&psinfo->read_mutex);
437 spin_unlock(&pstore_lock);
438
439 if (owner && !try_module_get(owner)) {
440 psinfo = NULL;
441 return -EINVAL;
442 }
443
444 allocate_buf_for_compression();
445
446 if (pstore_is_mounted())
447 pstore_get_records(0);
448
449 kmsg_dump_register(&pstore_dumper);
450 pstore_register_console();
451 pstore_register_ftrace();
452
453 if (pstore_update_ms >= 0) {
454 pstore_timer.expires = jiffies +
455 msecs_to_jiffies(pstore_update_ms);
456 add_timer(&pstore_timer);
457 }
458
459 pr_info("pstore: Registered %s as persistent store backend\n",
460 psi->name);
461
462 return 0;
463 }
464 EXPORT_SYMBOL_GPL(pstore_register);
465
466 /*
467 * Read all the records from the persistent store. Create
468 * files in our filesystem. Don't warn about -EEXIST errors
469 * when we are re-scanning the backing store looking to add new
470 * error records.
471 */
472 void pstore_get_records(int quiet)
473 {
474 struct pstore_info *psi = psinfo;
475 char *buf = NULL;
476 ssize_t size;
477 u64 id;
478 int count;
479 enum pstore_type_id type;
480 struct timespec time;
481 int failed = 0, rc;
482 bool compressed;
483 int unzipped_len = -1;
484
485 if (!psi)
486 return;
487
488 mutex_lock(&psi->read_mutex);
489 if (psi->open && psi->open(psi))
490 goto out;
491
492 while ((size = psi->read(&id, &type, &count, &time, &buf, &compressed,
493 psi)) > 0) {
494 if (compressed && (type == PSTORE_TYPE_DMESG)) {
495 if (big_oops_buf)
496 unzipped_len = pstore_decompress(buf,
497 big_oops_buf, size,
498 big_oops_buf_sz);
499
500 if (unzipped_len > 0) {
501 buf = big_oops_buf;
502 size = unzipped_len;
503 compressed = false;
504 } else {
505 pr_err("pstore: decompression failed;"
506 "returned %d\n", unzipped_len);
507 compressed = true;
508 }
509 }
510 rc = pstore_mkfile(type, psi->name, id, count, buf,
511 compressed, (size_t)size, time, psi);
512 if (unzipped_len < 0) {
513 /* Free buffer other than big oops */
514 kfree(buf);
515 buf = NULL;
516 } else
517 unzipped_len = -1;
518 if (rc && (rc != -EEXIST || !quiet))
519 failed++;
520 }
521 if (psi->close)
522 psi->close(psi);
523 out:
524 mutex_unlock(&psi->read_mutex);
525
526 if (failed)
527 printk(KERN_WARNING "pstore: failed to load %d record(s) from '%s'\n",
528 failed, psi->name);
529 }
530
531 static void pstore_dowork(struct work_struct *work)
532 {
533 pstore_get_records(1);
534 }
535
536 static void pstore_timefunc(unsigned long dummy)
537 {
538 if (pstore_new_entry) {
539 pstore_new_entry = 0;
540 schedule_work(&pstore_work);
541 }
542
543 mod_timer(&pstore_timer, jiffies + msecs_to_jiffies(pstore_update_ms));
544 }
545
546 module_param(backend, charp, 0444);
547 MODULE_PARM_DESC(backend, "Pstore backend to use");