1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * Copyright (C) 2012 Google, Inc. 4 */ 5 6 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 7 8 #include <linux/device.h> 9 #include <linux/err.h> 10 #include <linux/errno.h> 11 #include <linux/init.h> 12 #include <linux/io.h> 13 #include <linux/kernel.h> 14 #include <linux/list.h> 15 #include <linux/memblock.h> 16 #include <linux/pstore_ram.h> 17 #include <linux/rslib.h> 18 #include <linux/slab.h> 19 #include <linux/uaccess.h> 20 #include <linux/vmalloc.h> 21 #include <asm/page.h> 22 23 /** 24 * struct persistent_ram_buffer - persistent circular RAM buffer 25 * 26 * @sig: 27 * signature to indicate header (PERSISTENT_RAM_SIG xor PRZ-type value) 28 * @start: 29 * offset into @data where the beginning of the stored bytes begin 30 * @size: 31 * number of valid bytes stored in @data 32 */ 33 struct persistent_ram_buffer { 34 uint32_t sig; 35 atomic_t start; 36 atomic_t size; 37 uint8_t data[]; 38 }; 39 40 #define PERSISTENT_RAM_SIG (0x43474244) /* DBGC */ 41 42 static inline size_t buffer_size(struct persistent_ram_zone *prz) 43 { 44 return atomic_read(&prz->buffer->size); 45 } 46 47 static inline size_t buffer_start(struct persistent_ram_zone *prz) 48 { 49 return atomic_read(&prz->buffer->start); 50 } 51 52 /* increase and wrap the start pointer, returning the old value */ 53 static size_t buffer_start_add(struct persistent_ram_zone *prz, size_t a) 54 { 55 int old; 56 int new; 57 unsigned long flags = 0; 58 59 if (!(prz->flags & PRZ_FLAG_NO_LOCK)) 60 raw_spin_lock_irqsave(&prz->buffer_lock, flags); 61 62 old = atomic_read(&prz->buffer->start); 63 new = old + a; 64 while (unlikely(new >= prz->buffer_size)) 65 new -= prz->buffer_size; 66 atomic_set(&prz->buffer->start, new); 67 68 if (!(prz->flags & PRZ_FLAG_NO_LOCK)) 69 raw_spin_unlock_irqrestore(&prz->buffer_lock, flags); 70 71 return old; 72 } 73 74 /* increase the size counter until it hits the max size */ 75 static void buffer_size_add(struct persistent_ram_zone *prz, size_t a) 76 { 77 size_t old; 78 size_t new; 79 unsigned long flags = 0; 80 81 if (!(prz->flags & PRZ_FLAG_NO_LOCK)) 82 raw_spin_lock_irqsave(&prz->buffer_lock, flags); 83 84 old = atomic_read(&prz->buffer->size); 85 if (old == prz->buffer_size) 86 goto exit; 87 88 new = old + a; 89 if (new > prz->buffer_size) 90 new = prz->buffer_size; 91 atomic_set(&prz->buffer->size, new); 92 93 exit: 94 if (!(prz->flags & PRZ_FLAG_NO_LOCK)) 95 raw_spin_unlock_irqrestore(&prz->buffer_lock, flags); 96 } 97 98 static void notrace persistent_ram_encode_rs8(struct persistent_ram_zone *prz, 99 uint8_t *data, size_t len, uint8_t *ecc) 100 { 101 int i; 102 103 /* Initialize the parity buffer */ 104 memset(prz->ecc_info.par, 0, 105 prz->ecc_info.ecc_size * sizeof(prz->ecc_info.par[0])); 106 encode_rs8(prz->rs_decoder, data, len, prz->ecc_info.par, 0); 107 for (i = 0; i < prz->ecc_info.ecc_size; i++) 108 ecc[i] = prz->ecc_info.par[i]; 109 } 110 111 static int persistent_ram_decode_rs8(struct persistent_ram_zone *prz, 112 void *data, size_t len, uint8_t *ecc) 113 { 114 int i; 115 116 for (i = 0; i < prz->ecc_info.ecc_size; i++) 117 prz->ecc_info.par[i] = ecc[i]; 118 return decode_rs8(prz->rs_decoder, data, prz->ecc_info.par, len, 119 NULL, 0, NULL, 0, NULL); 120 } 121 122 static void notrace persistent_ram_update_ecc(struct persistent_ram_zone *prz, 123 unsigned int start, unsigned int count) 124 { 125 struct persistent_ram_buffer *buffer = prz->buffer; 126 uint8_t *buffer_end = buffer->data + prz->buffer_size; 127 uint8_t *block; 128 uint8_t *par; 129 int ecc_block_size = prz->ecc_info.block_size; 130 int ecc_size = prz->ecc_info.ecc_size; 131 int size = ecc_block_size; 132 133 if (!ecc_size) 134 return; 135 136 block = buffer->data + (start & ~(ecc_block_size - 1)); 137 par = prz->par_buffer + (start / ecc_block_size) * ecc_size; 138 139 do { 140 if (block + ecc_block_size > buffer_end) 141 size = buffer_end - block; 142 persistent_ram_encode_rs8(prz, block, size, par); 143 block += ecc_block_size; 144 par += ecc_size; 145 } while (block < buffer->data + start + count); 146 } 147 148 static void persistent_ram_update_header_ecc(struct persistent_ram_zone *prz) 149 { 150 struct persistent_ram_buffer *buffer = prz->buffer; 151 152 if (!prz->ecc_info.ecc_size) 153 return; 154 155 persistent_ram_encode_rs8(prz, (uint8_t *)buffer, sizeof(*buffer), 156 prz->par_header); 157 } 158 159 static void persistent_ram_ecc_old(struct persistent_ram_zone *prz) 160 { 161 struct persistent_ram_buffer *buffer = prz->buffer; 162 uint8_t *block; 163 uint8_t *par; 164 165 if (!prz->ecc_info.ecc_size) 166 return; 167 168 block = buffer->data; 169 par = prz->par_buffer; 170 while (block < buffer->data + buffer_size(prz)) { 171 int numerr; 172 int size = prz->ecc_info.block_size; 173 if (block + size > buffer->data + prz->buffer_size) 174 size = buffer->data + prz->buffer_size - block; 175 numerr = persistent_ram_decode_rs8(prz, block, size, par); 176 if (numerr > 0) { 177 pr_devel("error in block %p, %d\n", block, numerr); 178 prz->corrected_bytes += numerr; 179 } else if (numerr < 0) { 180 pr_devel("uncorrectable error in block %p\n", block); 181 prz->bad_blocks++; 182 } 183 block += prz->ecc_info.block_size; 184 par += prz->ecc_info.ecc_size; 185 } 186 } 187 188 static int persistent_ram_init_ecc(struct persistent_ram_zone *prz, 189 struct persistent_ram_ecc_info *ecc_info) 190 { 191 int numerr; 192 struct persistent_ram_buffer *buffer = prz->buffer; 193 int ecc_blocks; 194 size_t ecc_total; 195 196 if (!ecc_info || !ecc_info->ecc_size) 197 return 0; 198 199 prz->ecc_info.block_size = ecc_info->block_size ?: 128; 200 prz->ecc_info.ecc_size = ecc_info->ecc_size ?: 16; 201 prz->ecc_info.symsize = ecc_info->symsize ?: 8; 202 prz->ecc_info.poly = ecc_info->poly ?: 0x11d; 203 204 ecc_blocks = DIV_ROUND_UP(prz->buffer_size - prz->ecc_info.ecc_size, 205 prz->ecc_info.block_size + 206 prz->ecc_info.ecc_size); 207 ecc_total = (ecc_blocks + 1) * prz->ecc_info.ecc_size; 208 if (ecc_total >= prz->buffer_size) { 209 pr_err("%s: invalid ecc_size %u (total %zu, buffer size %zu)\n", 210 __func__, prz->ecc_info.ecc_size, 211 ecc_total, prz->buffer_size); 212 return -EINVAL; 213 } 214 215 prz->buffer_size -= ecc_total; 216 prz->par_buffer = buffer->data + prz->buffer_size; 217 prz->par_header = prz->par_buffer + 218 ecc_blocks * prz->ecc_info.ecc_size; 219 220 /* 221 * first consecutive root is 0 222 * primitive element to generate roots = 1 223 */ 224 prz->rs_decoder = init_rs(prz->ecc_info.symsize, prz->ecc_info.poly, 225 0, 1, prz->ecc_info.ecc_size); 226 if (prz->rs_decoder == NULL) { 227 pr_info("init_rs failed\n"); 228 return -EINVAL; 229 } 230 231 /* allocate workspace instead of using stack VLA */ 232 prz->ecc_info.par = kmalloc_array(prz->ecc_info.ecc_size, 233 sizeof(*prz->ecc_info.par), 234 GFP_KERNEL); 235 if (!prz->ecc_info.par) { 236 pr_err("cannot allocate ECC parity workspace\n"); 237 return -ENOMEM; 238 } 239 240 prz->corrected_bytes = 0; 241 prz->bad_blocks = 0; 242 243 numerr = persistent_ram_decode_rs8(prz, buffer, sizeof(*buffer), 244 prz->par_header); 245 if (numerr > 0) { 246 pr_info("error in header, %d\n", numerr); 247 prz->corrected_bytes += numerr; 248 } else if (numerr < 0) { 249 pr_info_ratelimited("uncorrectable error in header\n"); 250 prz->bad_blocks++; 251 } 252 253 return 0; 254 } 255 256 ssize_t persistent_ram_ecc_string(struct persistent_ram_zone *prz, 257 char *str, size_t len) 258 { 259 ssize_t ret; 260 261 if (!prz->ecc_info.ecc_size) 262 return 0; 263 264 if (prz->corrected_bytes || prz->bad_blocks) 265 ret = snprintf(str, len, "" 266 "\nECC: %d Corrected bytes, %d unrecoverable blocks\n", 267 prz->corrected_bytes, prz->bad_blocks); 268 else 269 ret = snprintf(str, len, "\nECC: No errors detected\n"); 270 271 return ret; 272 } 273 274 static void notrace persistent_ram_update(struct persistent_ram_zone *prz, 275 const void *s, unsigned int start, unsigned int count) 276 { 277 struct persistent_ram_buffer *buffer = prz->buffer; 278 memcpy_toio(buffer->data + start, s, count); 279 persistent_ram_update_ecc(prz, start, count); 280 } 281 282 static int notrace persistent_ram_update_user(struct persistent_ram_zone *prz, 283 const void __user *s, unsigned int start, unsigned int count) 284 { 285 struct persistent_ram_buffer *buffer = prz->buffer; 286 int ret = unlikely(copy_from_user(buffer->data + start, s, count)) ? 287 -EFAULT : 0; 288 persistent_ram_update_ecc(prz, start, count); 289 return ret; 290 } 291 292 void persistent_ram_save_old(struct persistent_ram_zone *prz) 293 { 294 struct persistent_ram_buffer *buffer = prz->buffer; 295 size_t size = buffer_size(prz); 296 size_t start = buffer_start(prz); 297 298 if (!size) 299 return; 300 301 if (!prz->old_log) { 302 persistent_ram_ecc_old(prz); 303 prz->old_log = kmalloc(size, GFP_KERNEL); 304 } 305 if (!prz->old_log) { 306 pr_err("failed to allocate buffer\n"); 307 return; 308 } 309 310 prz->old_log_size = size; 311 memcpy_fromio(prz->old_log, &buffer->data[start], size - start); 312 memcpy_fromio(prz->old_log + size - start, &buffer->data[0], start); 313 } 314 315 int notrace persistent_ram_write(struct persistent_ram_zone *prz, 316 const void *s, unsigned int count) 317 { 318 int rem; 319 int c = count; 320 size_t start; 321 322 if (unlikely(c > prz->buffer_size)) { 323 s += c - prz->buffer_size; 324 c = prz->buffer_size; 325 } 326 327 buffer_size_add(prz, c); 328 329 start = buffer_start_add(prz, c); 330 331 rem = prz->buffer_size - start; 332 if (unlikely(rem < c)) { 333 persistent_ram_update(prz, s, start, rem); 334 s += rem; 335 c -= rem; 336 start = 0; 337 } 338 persistent_ram_update(prz, s, start, c); 339 340 persistent_ram_update_header_ecc(prz); 341 342 return count; 343 } 344 345 int notrace persistent_ram_write_user(struct persistent_ram_zone *prz, 346 const void __user *s, unsigned int count) 347 { 348 int rem, ret = 0, c = count; 349 size_t start; 350 351 if (unlikely(c > prz->buffer_size)) { 352 s += c - prz->buffer_size; 353 c = prz->buffer_size; 354 } 355 356 buffer_size_add(prz, c); 357 358 start = buffer_start_add(prz, c); 359 360 rem = prz->buffer_size - start; 361 if (unlikely(rem < c)) { 362 ret = persistent_ram_update_user(prz, s, start, rem); 363 s += rem; 364 c -= rem; 365 start = 0; 366 } 367 if (likely(!ret)) 368 ret = persistent_ram_update_user(prz, s, start, c); 369 370 persistent_ram_update_header_ecc(prz); 371 372 return unlikely(ret) ? ret : count; 373 } 374 375 size_t persistent_ram_old_size(struct persistent_ram_zone *prz) 376 { 377 return prz->old_log_size; 378 } 379 380 void *persistent_ram_old(struct persistent_ram_zone *prz) 381 { 382 return prz->old_log; 383 } 384 385 void persistent_ram_free_old(struct persistent_ram_zone *prz) 386 { 387 kfree(prz->old_log); 388 prz->old_log = NULL; 389 prz->old_log_size = 0; 390 } 391 392 void persistent_ram_zap(struct persistent_ram_zone *prz) 393 { 394 atomic_set(&prz->buffer->start, 0); 395 atomic_set(&prz->buffer->size, 0); 396 persistent_ram_update_header_ecc(prz); 397 } 398 399 #define MEM_TYPE_WCOMBINE 0 400 #define MEM_TYPE_NONCACHED 1 401 #define MEM_TYPE_NORMAL 2 402 403 static void *persistent_ram_vmap(phys_addr_t start, size_t size, 404 unsigned int memtype) 405 { 406 struct page **pages; 407 phys_addr_t page_start; 408 unsigned int page_count; 409 pgprot_t prot; 410 unsigned int i; 411 void *vaddr; 412 413 page_start = start - offset_in_page(start); 414 page_count = DIV_ROUND_UP(size + offset_in_page(start), PAGE_SIZE); 415 416 switch (memtype) { 417 case MEM_TYPE_NORMAL: 418 prot = PAGE_KERNEL; 419 break; 420 case MEM_TYPE_NONCACHED: 421 prot = pgprot_noncached(PAGE_KERNEL); 422 break; 423 case MEM_TYPE_WCOMBINE: 424 prot = pgprot_writecombine(PAGE_KERNEL); 425 break; 426 default: 427 pr_err("invalid mem_type=%d\n", memtype); 428 return NULL; 429 } 430 431 pages = kmalloc_array(page_count, sizeof(struct page *), GFP_KERNEL); 432 if (!pages) { 433 pr_err("%s: Failed to allocate array for %u pages\n", 434 __func__, page_count); 435 return NULL; 436 } 437 438 for (i = 0; i < page_count; i++) { 439 phys_addr_t addr = page_start + i * PAGE_SIZE; 440 pages[i] = pfn_to_page(addr >> PAGE_SHIFT); 441 } 442 vaddr = vmap(pages, page_count, VM_MAP, prot); 443 kfree(pages); 444 445 /* 446 * Since vmap() uses page granularity, we must add the offset 447 * into the page here, to get the byte granularity address 448 * into the mapping to represent the actual "start" location. 449 */ 450 return vaddr + offset_in_page(start); 451 } 452 453 static void *persistent_ram_iomap(phys_addr_t start, size_t size, 454 unsigned int memtype, char *label) 455 { 456 void *va; 457 458 if (!request_mem_region(start, size, label ?: "ramoops")) { 459 pr_err("request mem region (%s 0x%llx@0x%llx) failed\n", 460 label ?: "ramoops", 461 (unsigned long long)size, (unsigned long long)start); 462 return NULL; 463 } 464 465 if (memtype) 466 va = ioremap(start, size); 467 else 468 va = ioremap_wc(start, size); 469 470 /* 471 * Since request_mem_region() and ioremap() are byte-granularity 472 * there is no need handle anything special like we do when the 473 * vmap() case in persistent_ram_vmap() above. 474 */ 475 return va; 476 } 477 478 static int persistent_ram_buffer_map(phys_addr_t start, phys_addr_t size, 479 struct persistent_ram_zone *prz, int memtype) 480 { 481 prz->paddr = start; 482 prz->size = size; 483 484 if (pfn_valid(start >> PAGE_SHIFT)) 485 prz->vaddr = persistent_ram_vmap(start, size, memtype); 486 else 487 prz->vaddr = persistent_ram_iomap(start, size, memtype, 488 prz->label); 489 490 if (!prz->vaddr) { 491 pr_err("%s: Failed to map 0x%llx pages at 0x%llx\n", __func__, 492 (unsigned long long)size, (unsigned long long)start); 493 return -ENOMEM; 494 } 495 496 prz->buffer = prz->vaddr; 497 prz->buffer_size = size - sizeof(struct persistent_ram_buffer); 498 499 return 0; 500 } 501 502 static int persistent_ram_post_init(struct persistent_ram_zone *prz, u32 sig, 503 struct persistent_ram_ecc_info *ecc_info) 504 { 505 int ret; 506 bool zap = !!(prz->flags & PRZ_FLAG_ZAP_OLD); 507 508 ret = persistent_ram_init_ecc(prz, ecc_info); 509 if (ret) { 510 pr_warn("ECC failed %s\n", prz->label); 511 return ret; 512 } 513 514 sig ^= PERSISTENT_RAM_SIG; 515 516 if (prz->buffer->sig == sig) { 517 if (buffer_size(prz) == 0) { 518 pr_debug("found existing empty buffer\n"); 519 return 0; 520 } 521 522 if (buffer_size(prz) > prz->buffer_size || 523 buffer_start(prz) > buffer_size(prz)) { 524 pr_info("found existing invalid buffer, size %zu, start %zu\n", 525 buffer_size(prz), buffer_start(prz)); 526 zap = true; 527 } else { 528 pr_debug("found existing buffer, size %zu, start %zu\n", 529 buffer_size(prz), buffer_start(prz)); 530 persistent_ram_save_old(prz); 531 } 532 } else { 533 pr_debug("no valid data in buffer (sig = 0x%08x)\n", 534 prz->buffer->sig); 535 prz->buffer->sig = sig; 536 zap = true; 537 } 538 539 /* Reset missing, invalid, or single-use memory area. */ 540 if (zap) 541 persistent_ram_zap(prz); 542 543 return 0; 544 } 545 546 void persistent_ram_free(struct persistent_ram_zone *prz) 547 { 548 if (!prz) 549 return; 550 551 if (prz->vaddr) { 552 if (pfn_valid(prz->paddr >> PAGE_SHIFT)) { 553 /* We must vunmap() at page-granularity. */ 554 vunmap(prz->vaddr - offset_in_page(prz->paddr)); 555 } else { 556 iounmap(prz->vaddr); 557 release_mem_region(prz->paddr, prz->size); 558 } 559 prz->vaddr = NULL; 560 } 561 if (prz->rs_decoder) { 562 free_rs(prz->rs_decoder); 563 prz->rs_decoder = NULL; 564 } 565 kfree(prz->ecc_info.par); 566 prz->ecc_info.par = NULL; 567 568 persistent_ram_free_old(prz); 569 kfree(prz->label); 570 kfree(prz); 571 } 572 573 struct persistent_ram_zone *persistent_ram_new(phys_addr_t start, size_t size, 574 u32 sig, struct persistent_ram_ecc_info *ecc_info, 575 unsigned int memtype, u32 flags, char *label) 576 { 577 struct persistent_ram_zone *prz; 578 int ret = -ENOMEM; 579 580 prz = kzalloc(sizeof(struct persistent_ram_zone), GFP_KERNEL); 581 if (!prz) { 582 pr_err("failed to allocate persistent ram zone\n"); 583 goto err; 584 } 585 586 /* Initialize general buffer state. */ 587 raw_spin_lock_init(&prz->buffer_lock); 588 prz->flags = flags; 589 prz->label = kstrdup(label, GFP_KERNEL); 590 591 ret = persistent_ram_buffer_map(start, size, prz, memtype); 592 if (ret) 593 goto err; 594 595 ret = persistent_ram_post_init(prz, sig, ecc_info); 596 if (ret) 597 goto err; 598 599 pr_debug("attached %s 0x%zx@0x%llx: %zu header, %zu data, %zu ecc (%d/%d)\n", 600 prz->label, prz->size, (unsigned long long)prz->paddr, 601 sizeof(*prz->buffer), prz->buffer_size, 602 prz->size - sizeof(*prz->buffer) - prz->buffer_size, 603 prz->ecc_info.ecc_size, prz->ecc_info.block_size); 604 605 return prz; 606 err: 607 persistent_ram_free(prz); 608 return ERR_PTR(ret); 609 } 610