xref: /linux/fs/pstore/ram_core.c (revision 5148fa52a12fa1b97c730b2fe321f2aad7ea041c)
1 /*
2  * Copyright (C) 2012 Google, Inc.
3  *
4  * This software is licensed under the terms of the GNU General Public
5  * License version 2, as published by the Free Software Foundation, and
6  * may be copied, distributed, and modified under those terms.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11  * GNU General Public License for more details.
12  *
13  */
14 
15 #include <linux/device.h>
16 #include <linux/err.h>
17 #include <linux/errno.h>
18 #include <linux/kernel.h>
19 #include <linux/init.h>
20 #include <linux/io.h>
21 #include <linux/list.h>
22 #include <linux/memblock.h>
23 #include <linux/rslib.h>
24 #include <linux/slab.h>
25 #include <linux/vmalloc.h>
26 #include <linux/pstore_ram.h>
27 #include <asm/page.h>
28 
29 struct persistent_ram_buffer {
30 	uint32_t    sig;
31 	atomic_t    start;
32 	atomic_t    size;
33 	uint8_t     data[0];
34 };
35 
36 #define PERSISTENT_RAM_SIG (0x43474244) /* DBGC */
37 
38 static __initdata LIST_HEAD(persistent_ram_list);
39 
40 static inline size_t buffer_size(struct persistent_ram_zone *prz)
41 {
42 	return atomic_read(&prz->buffer->size);
43 }
44 
45 static inline size_t buffer_start(struct persistent_ram_zone *prz)
46 {
47 	return atomic_read(&prz->buffer->start);
48 }
49 
50 /* increase and wrap the start pointer, returning the old value */
51 static inline size_t buffer_start_add(struct persistent_ram_zone *prz, size_t a)
52 {
53 	int old;
54 	int new;
55 
56 	do {
57 		old = atomic_read(&prz->buffer->start);
58 		new = old + a;
59 		while (unlikely(new > prz->buffer_size))
60 			new -= prz->buffer_size;
61 	} while (atomic_cmpxchg(&prz->buffer->start, old, new) != old);
62 
63 	return old;
64 }
65 
66 /* increase the size counter until it hits the max size */
67 static inline void buffer_size_add(struct persistent_ram_zone *prz, size_t a)
68 {
69 	size_t old;
70 	size_t new;
71 
72 	if (atomic_read(&prz->buffer->size) == prz->buffer_size)
73 		return;
74 
75 	do {
76 		old = atomic_read(&prz->buffer->size);
77 		new = old + a;
78 		if (new > prz->buffer_size)
79 			new = prz->buffer_size;
80 	} while (atomic_cmpxchg(&prz->buffer->size, old, new) != old);
81 }
82 
83 static void notrace persistent_ram_encode_rs8(struct persistent_ram_zone *prz,
84 	uint8_t *data, size_t len, uint8_t *ecc)
85 {
86 	int i;
87 	uint16_t par[prz->ecc_size];
88 
89 	/* Initialize the parity buffer */
90 	memset(par, 0, sizeof(par));
91 	encode_rs8(prz->rs_decoder, data, len, par, 0);
92 	for (i = 0; i < prz->ecc_size; i++)
93 		ecc[i] = par[i];
94 }
95 
96 static int persistent_ram_decode_rs8(struct persistent_ram_zone *prz,
97 	void *data, size_t len, uint8_t *ecc)
98 {
99 	int i;
100 	uint16_t par[prz->ecc_size];
101 
102 	for (i = 0; i < prz->ecc_size; i++)
103 		par[i] = ecc[i];
104 	return decode_rs8(prz->rs_decoder, data, par, len,
105 				NULL, 0, NULL, 0, NULL);
106 }
107 
108 static void notrace persistent_ram_update_ecc(struct persistent_ram_zone *prz,
109 	unsigned int start, unsigned int count)
110 {
111 	struct persistent_ram_buffer *buffer = prz->buffer;
112 	uint8_t *buffer_end = buffer->data + prz->buffer_size;
113 	uint8_t *block;
114 	uint8_t *par;
115 	int ecc_block_size = prz->ecc_block_size;
116 	int ecc_size = prz->ecc_size;
117 	int size = prz->ecc_block_size;
118 
119 	if (!prz->ecc)
120 		return;
121 
122 	block = buffer->data + (start & ~(ecc_block_size - 1));
123 	par = prz->par_buffer + (start / ecc_block_size) * prz->ecc_size;
124 
125 	do {
126 		if (block + ecc_block_size > buffer_end)
127 			size = buffer_end - block;
128 		persistent_ram_encode_rs8(prz, block, size, par);
129 		block += ecc_block_size;
130 		par += ecc_size;
131 	} while (block < buffer->data + start + count);
132 }
133 
134 static void persistent_ram_update_header_ecc(struct persistent_ram_zone *prz)
135 {
136 	struct persistent_ram_buffer *buffer = prz->buffer;
137 
138 	if (!prz->ecc)
139 		return;
140 
141 	persistent_ram_encode_rs8(prz, (uint8_t *)buffer, sizeof(*buffer),
142 				  prz->par_header);
143 }
144 
145 static void persistent_ram_ecc_old(struct persistent_ram_zone *prz)
146 {
147 	struct persistent_ram_buffer *buffer = prz->buffer;
148 	uint8_t *block;
149 	uint8_t *par;
150 
151 	if (!prz->ecc)
152 		return;
153 
154 	block = buffer->data;
155 	par = prz->par_buffer;
156 	while (block < buffer->data + buffer_size(prz)) {
157 		int numerr;
158 		int size = prz->ecc_block_size;
159 		if (block + size > buffer->data + prz->buffer_size)
160 			size = buffer->data + prz->buffer_size - block;
161 		numerr = persistent_ram_decode_rs8(prz, block, size, par);
162 		if (numerr > 0) {
163 			pr_devel("persistent_ram: error in block %p, %d\n",
164 			       block, numerr);
165 			prz->corrected_bytes += numerr;
166 		} else if (numerr < 0) {
167 			pr_devel("persistent_ram: uncorrectable error in block %p\n",
168 				block);
169 			prz->bad_blocks++;
170 		}
171 		block += prz->ecc_block_size;
172 		par += prz->ecc_size;
173 	}
174 }
175 
176 static int persistent_ram_init_ecc(struct persistent_ram_zone *prz,
177 	size_t buffer_size)
178 {
179 	int numerr;
180 	struct persistent_ram_buffer *buffer = prz->buffer;
181 	int ecc_blocks;
182 
183 	if (!prz->ecc)
184 		return 0;
185 
186 	prz->ecc_block_size = 128;
187 	prz->ecc_size = 16;
188 	prz->ecc_symsize = 8;
189 	prz->ecc_poly = 0x11d;
190 
191 	ecc_blocks = DIV_ROUND_UP(prz->buffer_size, prz->ecc_block_size);
192 	prz->buffer_size -= (ecc_blocks + 1) * prz->ecc_size;
193 
194 	if (prz->buffer_size > buffer_size) {
195 		pr_err("persistent_ram: invalid size %zu, non-ecc datasize %zu\n",
196 		       buffer_size, prz->buffer_size);
197 		return -EINVAL;
198 	}
199 
200 	prz->par_buffer = buffer->data + prz->buffer_size;
201 	prz->par_header = prz->par_buffer + ecc_blocks * prz->ecc_size;
202 
203 	/*
204 	 * first consecutive root is 0
205 	 * primitive element to generate roots = 1
206 	 */
207 	prz->rs_decoder = init_rs(prz->ecc_symsize, prz->ecc_poly, 0, 1,
208 				  prz->ecc_size);
209 	if (prz->rs_decoder == NULL) {
210 		pr_info("persistent_ram: init_rs failed\n");
211 		return -EINVAL;
212 	}
213 
214 	prz->corrected_bytes = 0;
215 	prz->bad_blocks = 0;
216 
217 	numerr = persistent_ram_decode_rs8(prz, buffer, sizeof(*buffer),
218 					   prz->par_header);
219 	if (numerr > 0) {
220 		pr_info("persistent_ram: error in header, %d\n", numerr);
221 		prz->corrected_bytes += numerr;
222 	} else if (numerr < 0) {
223 		pr_info("persistent_ram: uncorrectable error in header\n");
224 		prz->bad_blocks++;
225 	}
226 
227 	return 0;
228 }
229 
230 ssize_t persistent_ram_ecc_string(struct persistent_ram_zone *prz,
231 	char *str, size_t len)
232 {
233 	ssize_t ret;
234 
235 	if (prz->corrected_bytes || prz->bad_blocks)
236 		ret = snprintf(str, len, ""
237 			"\n%d Corrected bytes, %d unrecoverable blocks\n",
238 			prz->corrected_bytes, prz->bad_blocks);
239 	else
240 		ret = snprintf(str, len, "\nNo errors detected\n");
241 
242 	return ret;
243 }
244 
245 static void notrace persistent_ram_update(struct persistent_ram_zone *prz,
246 	const void *s, unsigned int start, unsigned int count)
247 {
248 	struct persistent_ram_buffer *buffer = prz->buffer;
249 	memcpy(buffer->data + start, s, count);
250 	persistent_ram_update_ecc(prz, start, count);
251 }
252 
253 static void __init
254 persistent_ram_save_old(struct persistent_ram_zone *prz)
255 {
256 	struct persistent_ram_buffer *buffer = prz->buffer;
257 	size_t size = buffer_size(prz);
258 	size_t start = buffer_start(prz);
259 	char *dest;
260 
261 	persistent_ram_ecc_old(prz);
262 
263 	dest = kmalloc(size, GFP_KERNEL);
264 	if (dest == NULL) {
265 		pr_err("persistent_ram: failed to allocate buffer\n");
266 		return;
267 	}
268 
269 	prz->old_log = dest;
270 	prz->old_log_size = size;
271 	memcpy(prz->old_log, &buffer->data[start], size - start);
272 	memcpy(prz->old_log + size - start, &buffer->data[0], start);
273 }
274 
275 int notrace persistent_ram_write(struct persistent_ram_zone *prz,
276 	const void *s, unsigned int count)
277 {
278 	int rem;
279 	int c = count;
280 	size_t start;
281 
282 	if (unlikely(c > prz->buffer_size)) {
283 		s += c - prz->buffer_size;
284 		c = prz->buffer_size;
285 	}
286 
287 	buffer_size_add(prz, c);
288 
289 	start = buffer_start_add(prz, c);
290 
291 	rem = prz->buffer_size - start;
292 	if (unlikely(rem < c)) {
293 		persistent_ram_update(prz, s, start, rem);
294 		s += rem;
295 		c -= rem;
296 		start = 0;
297 	}
298 	persistent_ram_update(prz, s, start, c);
299 
300 	persistent_ram_update_header_ecc(prz);
301 
302 	return count;
303 }
304 
305 size_t persistent_ram_old_size(struct persistent_ram_zone *prz)
306 {
307 	return prz->old_log_size;
308 }
309 
310 void *persistent_ram_old(struct persistent_ram_zone *prz)
311 {
312 	return prz->old_log;
313 }
314 
315 void persistent_ram_free_old(struct persistent_ram_zone *prz)
316 {
317 	kfree(prz->old_log);
318 	prz->old_log = NULL;
319 	prz->old_log_size = 0;
320 }
321 
322 static void *persistent_ram_vmap(phys_addr_t start, size_t size)
323 {
324 	struct page **pages;
325 	phys_addr_t page_start;
326 	unsigned int page_count;
327 	pgprot_t prot;
328 	unsigned int i;
329 	void *vaddr;
330 
331 	page_start = start - offset_in_page(start);
332 	page_count = DIV_ROUND_UP(size + offset_in_page(start), PAGE_SIZE);
333 
334 	prot = pgprot_noncached(PAGE_KERNEL);
335 
336 	pages = kmalloc(sizeof(struct page *) * page_count, GFP_KERNEL);
337 	if (!pages) {
338 		pr_err("%s: Failed to allocate array for %u pages\n", __func__,
339 			page_count);
340 		return NULL;
341 	}
342 
343 	for (i = 0; i < page_count; i++) {
344 		phys_addr_t addr = page_start + i * PAGE_SIZE;
345 		pages[i] = pfn_to_page(addr >> PAGE_SHIFT);
346 	}
347 	vaddr = vmap(pages, page_count, VM_MAP, prot);
348 	kfree(pages);
349 
350 	return vaddr;
351 }
352 
353 static void *persistent_ram_iomap(phys_addr_t start, size_t size)
354 {
355 	if (!request_mem_region(start, size, "persistent_ram")) {
356 		pr_err("request mem region (0x%llx@0x%llx) failed\n",
357 			(unsigned long long)size, (unsigned long long)start);
358 		return NULL;
359 	}
360 
361 	return ioremap(start, size);
362 }
363 
364 static int persistent_ram_buffer_map(phys_addr_t start, phys_addr_t size,
365 		struct persistent_ram_zone *prz)
366 {
367 	prz->paddr = start;
368 	prz->size = size;
369 
370 	if (pfn_valid(start >> PAGE_SHIFT))
371 		prz->vaddr = persistent_ram_vmap(start, size);
372 	else
373 		prz->vaddr = persistent_ram_iomap(start, size);
374 
375 	if (!prz->vaddr) {
376 		pr_err("%s: Failed to map 0x%llx pages at 0x%llx\n", __func__,
377 			(unsigned long long)size, (unsigned long long)start);
378 		return -ENOMEM;
379 	}
380 
381 	prz->buffer = prz->vaddr + offset_in_page(start);
382 	prz->buffer_size = size - sizeof(struct persistent_ram_buffer);
383 
384 	return 0;
385 }
386 
387 static int __init persistent_ram_post_init(struct persistent_ram_zone *prz, bool ecc)
388 {
389 	int ret;
390 
391 	prz->ecc = ecc;
392 
393 	ret = persistent_ram_init_ecc(prz, prz->buffer_size);
394 	if (ret)
395 		return ret;
396 
397 	if (prz->buffer->sig == PERSISTENT_RAM_SIG) {
398 		if (buffer_size(prz) > prz->buffer_size ||
399 		    buffer_start(prz) > buffer_size(prz))
400 			pr_info("persistent_ram: found existing invalid buffer,"
401 				" size %zu, start %zu\n",
402 			       buffer_size(prz), buffer_start(prz));
403 		else {
404 			pr_info("persistent_ram: found existing buffer,"
405 				" size %zu, start %zu\n",
406 			       buffer_size(prz), buffer_start(prz));
407 			persistent_ram_save_old(prz);
408 		}
409 	} else {
410 		pr_info("persistent_ram: no valid data in buffer"
411 			" (sig = 0x%08x)\n", prz->buffer->sig);
412 	}
413 
414 	prz->buffer->sig = PERSISTENT_RAM_SIG;
415 	atomic_set(&prz->buffer->start, 0);
416 	atomic_set(&prz->buffer->size, 0);
417 
418 	return 0;
419 }
420 
421 void persistent_ram_free(struct persistent_ram_zone *prz)
422 {
423 	if (pfn_valid(prz->paddr >> PAGE_SHIFT)) {
424 		vunmap(prz->vaddr);
425 	} else {
426 		iounmap(prz->vaddr);
427 		release_mem_region(prz->paddr, prz->size);
428 	}
429 	persistent_ram_free_old(prz);
430 	kfree(prz);
431 }
432 
433 struct persistent_ram_zone * __init persistent_ram_new(phys_addr_t start,
434 						       size_t size,
435 						       bool ecc)
436 {
437 	struct persistent_ram_zone *prz;
438 	int ret = -ENOMEM;
439 
440 	prz = kzalloc(sizeof(struct persistent_ram_zone), GFP_KERNEL);
441 	if (!prz) {
442 		pr_err("persistent_ram: failed to allocate persistent ram zone\n");
443 		goto err;
444 	}
445 
446 	ret = persistent_ram_buffer_map(start, size, prz);
447 	if (ret)
448 		goto err;
449 
450 	persistent_ram_post_init(prz, ecc);
451 	persistent_ram_update_header_ecc(prz);
452 
453 	return prz;
454 err:
455 	kfree(prz);
456 	return ERR_PTR(ret);
457 }
458 
459 #ifndef MODULE
460 static int __init persistent_ram_buffer_init(const char *name,
461 		struct persistent_ram_zone *prz)
462 {
463 	int i;
464 	struct persistent_ram *ram;
465 	struct persistent_ram_descriptor *desc;
466 	phys_addr_t start;
467 
468 	list_for_each_entry(ram, &persistent_ram_list, node) {
469 		start = ram->start;
470 		for (i = 0; i < ram->num_descs; i++) {
471 			desc = &ram->descs[i];
472 			if (!strcmp(desc->name, name))
473 				return persistent_ram_buffer_map(start,
474 						desc->size, prz);
475 			start += desc->size;
476 		}
477 	}
478 
479 	return -EINVAL;
480 }
481 
482 static  __init
483 struct persistent_ram_zone *__persistent_ram_init(struct device *dev, bool ecc)
484 {
485 	struct persistent_ram_zone *prz;
486 	int ret = -ENOMEM;
487 
488 	prz = kzalloc(sizeof(struct persistent_ram_zone), GFP_KERNEL);
489 	if (!prz) {
490 		pr_err("persistent_ram: failed to allocate persistent ram zone\n");
491 		goto err;
492 	}
493 
494 	ret = persistent_ram_buffer_init(dev_name(dev), prz);
495 	if (ret) {
496 		pr_err("persistent_ram: failed to initialize buffer\n");
497 		goto err;
498 	}
499 
500 	persistent_ram_post_init(prz, ecc);
501 
502 	return prz;
503 err:
504 	kfree(prz);
505 	return ERR_PTR(ret);
506 }
507 
508 struct persistent_ram_zone * __init
509 persistent_ram_init_ringbuffer(struct device *dev, bool ecc)
510 {
511 	return __persistent_ram_init(dev, ecc);
512 }
513 
514 int __init persistent_ram_early_init(struct persistent_ram *ram)
515 {
516 	int ret;
517 
518 	ret = memblock_reserve(ram->start, ram->size);
519 	if (ret) {
520 		pr_err("Failed to reserve persistent memory from %08lx-%08lx\n",
521 			(long)ram->start, (long)(ram->start + ram->size - 1));
522 		return ret;
523 	}
524 
525 	list_add_tail(&ram->node, &persistent_ram_list);
526 
527 	pr_info("Initialized persistent memory from %08lx-%08lx\n",
528 		(long)ram->start, (long)(ram->start + ram->size - 1));
529 
530 	return 0;
531 }
532 #endif
533