xref: /openbmc/linux/arch/powerpc/kernel/rtas_flash.c (revision 6d99a79c)
1 /*
2  *  c 2001 PPC 64 Team, IBM Corp
3  *
4  *      This program is free software; you can redistribute it and/or
5  *      modify it under the terms of the GNU General Public License
6  *      as published by the Free Software Foundation; either version
7  *      2 of the License, or (at your option) any later version.
8  *
9  * /proc/powerpc/rtas/firmware_flash interface
10  *
11  * This file implements a firmware_flash interface to pump a firmware
12  * image into the kernel.  At reboot time rtas_restart() will see the
13  * firmware image and flash it as it reboots (see rtas.c).
14  */
15 
16 #include <linux/module.h>
17 #include <linux/init.h>
18 #include <linux/slab.h>
19 #include <linux/proc_fs.h>
20 #include <linux/reboot.h>
21 #include <asm/delay.h>
22 #include <linux/uaccess.h>
23 #include <asm/rtas.h>
24 
25 #define MODULE_VERS "1.0"
26 #define MODULE_NAME "rtas_flash"
27 
28 #define FIRMWARE_FLASH_NAME "firmware_flash"
29 #define FIRMWARE_UPDATE_NAME "firmware_update"
30 #define MANAGE_FLASH_NAME "manage_flash"
31 #define VALIDATE_FLASH_NAME "validate_flash"
32 
33 /* General RTAS Status Codes */
34 #define RTAS_RC_SUCCESS  0
35 #define RTAS_RC_HW_ERR	-1
36 #define RTAS_RC_BUSY	-2
37 
38 /* Flash image status values */
39 #define FLASH_AUTH           -9002 /* RTAS Not Service Authority Partition */
40 #define FLASH_NO_OP          -1099 /* No operation initiated by user */
41 #define FLASH_IMG_SHORT	     -1005 /* Flash image shorter than expected */
42 #define FLASH_IMG_BAD_LEN    -1004 /* Bad length value in flash list block */
43 #define FLASH_IMG_NULL_DATA  -1003 /* Bad data value in flash list block */
44 #define FLASH_IMG_READY      0     /* Firmware img ready for flash on reboot */
45 
46 /* Manage image status values */
47 #define MANAGE_AUTH          -9002 /* RTAS Not Service Authority Partition */
48 #define MANAGE_ACTIVE_ERR    -9001 /* RTAS Cannot Overwrite Active Img */
49 #define MANAGE_NO_OP         -1099 /* No operation initiated by user */
50 #define MANAGE_PARAM_ERR     -3    /* RTAS Parameter Error */
51 #define MANAGE_HW_ERR        -1    /* RTAS Hardware Error */
52 
53 /* Validate image status values */
54 #define VALIDATE_AUTH          -9002 /* RTAS Not Service Authority Partition */
55 #define VALIDATE_NO_OP         -1099 /* No operation initiated by the user */
56 #define VALIDATE_INCOMPLETE    -1002 /* User copied < VALIDATE_BUF_SIZE */
57 #define VALIDATE_READY	       -1001 /* Firmware image ready for validation */
58 #define VALIDATE_PARAM_ERR     -3    /* RTAS Parameter Error */
59 #define VALIDATE_HW_ERR        -1    /* RTAS Hardware Error */
60 
61 /* ibm,validate-flash-image update result tokens */
62 #define VALIDATE_TMP_UPDATE    0     /* T side will be updated */
63 #define VALIDATE_FLASH_AUTH    1     /* Partition does not have authority */
64 #define VALIDATE_INVALID_IMG   2     /* Candidate image is not valid */
65 #define VALIDATE_CUR_UNKNOWN   3     /* Current fixpack level is unknown */
66 /*
67  * Current T side will be committed to P side before being replace with new
68  * image, and the new image is downlevel from current image
69  */
70 #define VALIDATE_TMP_COMMIT_DL 4
71 /*
72  * Current T side will be committed to P side before being replaced with new
73  * image
74  */
75 #define VALIDATE_TMP_COMMIT    5
76 /*
77  * T side will be updated with a downlevel image
78  */
79 #define VALIDATE_TMP_UPDATE_DL 6
80 /*
81  * The candidate image's release date is later than the system's firmware
82  * service entitlement date - service warranty period has expired
83  */
84 #define VALIDATE_OUT_OF_WRNTY  7
85 
86 /* ibm,manage-flash-image operation tokens */
87 #define RTAS_REJECT_TMP_IMG   0
88 #define RTAS_COMMIT_TMP_IMG   1
89 
90 /* Array sizes */
91 #define VALIDATE_BUF_SIZE 4096
92 #define VALIDATE_MSG_LEN  256
93 #define RTAS_MSG_MAXLEN   64
94 
95 /* Quirk - RTAS requires 4k list length and block size */
96 #define RTAS_BLKLIST_LENGTH 4096
97 #define RTAS_BLK_SIZE 4096
98 
99 struct flash_block {
100 	char *data;
101 	unsigned long length;
102 };
103 
104 /* This struct is very similar but not identical to
105  * that needed by the rtas flash update.
106  * All we need to do for rtas is rewrite num_blocks
107  * into a version/length and translate the pointers
108  * to absolute.
109  */
110 #define FLASH_BLOCKS_PER_NODE ((RTAS_BLKLIST_LENGTH - 16) / sizeof(struct flash_block))
111 struct flash_block_list {
112 	unsigned long num_blocks;
113 	struct flash_block_list *next;
114 	struct flash_block blocks[FLASH_BLOCKS_PER_NODE];
115 };
116 
117 static struct flash_block_list *rtas_firmware_flash_list;
118 
119 /* Use slab cache to guarantee 4k alignment */
120 static struct kmem_cache *flash_block_cache = NULL;
121 
122 #define FLASH_BLOCK_LIST_VERSION (1UL)
123 
124 /*
125  * Local copy of the flash block list.
126  *
127  * The rtas_firmware_flash_list varable will be
128  * set once the data is fully read.
129  *
130  * For convenience as we build the list we use virtual addrs,
131  * we do not fill in the version number, and the length field
132  * is treated as the number of entries currently in the block
133  * (i.e. not a byte count).  This is all fixed when calling
134  * the flash routine.
135  */
136 
137 /* Status int must be first member of struct */
138 struct rtas_update_flash_t
139 {
140 	int status;			/* Flash update status */
141 	struct flash_block_list *flist; /* Local copy of flash block list */
142 };
143 
144 /* Status int must be first member of struct */
145 struct rtas_manage_flash_t
146 {
147 	int status;			/* Returned status */
148 };
149 
150 /* Status int must be first member of struct */
151 struct rtas_validate_flash_t
152 {
153 	int status;		 	/* Returned status */
154 	char *buf;			/* Candidate image buffer */
155 	unsigned int buf_size;		/* Size of image buf */
156 	unsigned int update_results;	/* Update results token */
157 };
158 
159 static struct rtas_update_flash_t rtas_update_flash_data;
160 static struct rtas_manage_flash_t rtas_manage_flash_data;
161 static struct rtas_validate_flash_t rtas_validate_flash_data;
162 static DEFINE_MUTEX(rtas_update_flash_mutex);
163 static DEFINE_MUTEX(rtas_manage_flash_mutex);
164 static DEFINE_MUTEX(rtas_validate_flash_mutex);
165 
166 /* Do simple sanity checks on the flash image. */
167 static int flash_list_valid(struct flash_block_list *flist)
168 {
169 	struct flash_block_list *f;
170 	int i;
171 	unsigned long block_size, image_size;
172 
173 	/* Paranoid self test here.  We also collect the image size. */
174 	image_size = 0;
175 	for (f = flist; f; f = f->next) {
176 		for (i = 0; i < f->num_blocks; i++) {
177 			if (f->blocks[i].data == NULL) {
178 				return FLASH_IMG_NULL_DATA;
179 			}
180 			block_size = f->blocks[i].length;
181 			if (block_size <= 0 || block_size > RTAS_BLK_SIZE) {
182 				return FLASH_IMG_BAD_LEN;
183 			}
184 			image_size += block_size;
185 		}
186 	}
187 
188 	if (image_size < (256 << 10)) {
189 		if (image_size < 2)
190 			return FLASH_NO_OP;
191 	}
192 
193 	printk(KERN_INFO "FLASH: flash image with %ld bytes stored for hardware flash on reboot\n", image_size);
194 
195 	return FLASH_IMG_READY;
196 }
197 
198 static void free_flash_list(struct flash_block_list *f)
199 {
200 	struct flash_block_list *next;
201 	int i;
202 
203 	while (f) {
204 		for (i = 0; i < f->num_blocks; i++)
205 			kmem_cache_free(flash_block_cache, f->blocks[i].data);
206 		next = f->next;
207 		kmem_cache_free(flash_block_cache, f);
208 		f = next;
209 	}
210 }
211 
212 static int rtas_flash_release(struct inode *inode, struct file *file)
213 {
214 	struct rtas_update_flash_t *const uf = &rtas_update_flash_data;
215 
216 	mutex_lock(&rtas_update_flash_mutex);
217 
218 	if (uf->flist) {
219 		/* File was opened in write mode for a new flash attempt */
220 		/* Clear saved list */
221 		if (rtas_firmware_flash_list) {
222 			free_flash_list(rtas_firmware_flash_list);
223 			rtas_firmware_flash_list = NULL;
224 		}
225 
226 		if (uf->status != FLASH_AUTH)
227 			uf->status = flash_list_valid(uf->flist);
228 
229 		if (uf->status == FLASH_IMG_READY)
230 			rtas_firmware_flash_list = uf->flist;
231 		else
232 			free_flash_list(uf->flist);
233 
234 		uf->flist = NULL;
235 	}
236 
237 	mutex_unlock(&rtas_update_flash_mutex);
238 	return 0;
239 }
240 
241 static size_t get_flash_status_msg(int status, char *buf)
242 {
243 	const char *msg;
244 	size_t len;
245 
246 	switch (status) {
247 	case FLASH_AUTH:
248 		msg = "error: this partition does not have service authority\n";
249 		break;
250 	case FLASH_NO_OP:
251 		msg = "info: no firmware image for flash\n";
252 		break;
253 	case FLASH_IMG_SHORT:
254 		msg = "error: flash image short\n";
255 		break;
256 	case FLASH_IMG_BAD_LEN:
257 		msg = "error: internal error bad length\n";
258 		break;
259 	case FLASH_IMG_NULL_DATA:
260 		msg = "error: internal error null data\n";
261 		break;
262 	case FLASH_IMG_READY:
263 		msg = "ready: firmware image ready for flash on reboot\n";
264 		break;
265 	default:
266 		return sprintf(buf, "error: unexpected status value %d\n",
267 			       status);
268 	}
269 
270 	len = strlen(msg);
271 	memcpy(buf, msg, len + 1);
272 	return len;
273 }
274 
275 /* Reading the proc file will show status (not the firmware contents) */
276 static ssize_t rtas_flash_read_msg(struct file *file, char __user *buf,
277 				   size_t count, loff_t *ppos)
278 {
279 	struct rtas_update_flash_t *const uf = &rtas_update_flash_data;
280 	char msg[RTAS_MSG_MAXLEN];
281 	size_t len;
282 	int status;
283 
284 	mutex_lock(&rtas_update_flash_mutex);
285 	status = uf->status;
286 	mutex_unlock(&rtas_update_flash_mutex);
287 
288 	/* Read as text message */
289 	len = get_flash_status_msg(status, msg);
290 	return simple_read_from_buffer(buf, count, ppos, msg, len);
291 }
292 
293 static ssize_t rtas_flash_read_num(struct file *file, char __user *buf,
294 				   size_t count, loff_t *ppos)
295 {
296 	struct rtas_update_flash_t *const uf = &rtas_update_flash_data;
297 	char msg[RTAS_MSG_MAXLEN];
298 	int status;
299 
300 	mutex_lock(&rtas_update_flash_mutex);
301 	status = uf->status;
302 	mutex_unlock(&rtas_update_flash_mutex);
303 
304 	/* Read as number */
305 	sprintf(msg, "%d\n", status);
306 	return simple_read_from_buffer(buf, count, ppos, msg, strlen(msg));
307 }
308 
309 /* We could be much more efficient here.  But to keep this function
310  * simple we allocate a page to the block list no matter how small the
311  * count is.  If the system is low on memory it will be just as well
312  * that we fail....
313  */
314 static ssize_t rtas_flash_write(struct file *file, const char __user *buffer,
315 				size_t count, loff_t *off)
316 {
317 	struct rtas_update_flash_t *const uf = &rtas_update_flash_data;
318 	char *p;
319 	int next_free, rc;
320 	struct flash_block_list *fl;
321 
322 	mutex_lock(&rtas_update_flash_mutex);
323 
324 	if (uf->status == FLASH_AUTH || count == 0)
325 		goto out;	/* discard data */
326 
327 	/* In the case that the image is not ready for flashing, the memory
328 	 * allocated for the block list will be freed upon the release of the
329 	 * proc file
330 	 */
331 	if (uf->flist == NULL) {
332 		uf->flist = kmem_cache_zalloc(flash_block_cache, GFP_KERNEL);
333 		if (!uf->flist)
334 			goto nomem;
335 	}
336 
337 	fl = uf->flist;
338 	while (fl->next)
339 		fl = fl->next; /* seek to last block_list for append */
340 	next_free = fl->num_blocks;
341 	if (next_free == FLASH_BLOCKS_PER_NODE) {
342 		/* Need to allocate another block_list */
343 		fl->next = kmem_cache_zalloc(flash_block_cache, GFP_KERNEL);
344 		if (!fl->next)
345 			goto nomem;
346 		fl = fl->next;
347 		next_free = 0;
348 	}
349 
350 	if (count > RTAS_BLK_SIZE)
351 		count = RTAS_BLK_SIZE;
352 	p = kmem_cache_zalloc(flash_block_cache, GFP_KERNEL);
353 	if (!p)
354 		goto nomem;
355 
356 	if(copy_from_user(p, buffer, count)) {
357 		kmem_cache_free(flash_block_cache, p);
358 		rc = -EFAULT;
359 		goto error;
360 	}
361 	fl->blocks[next_free].data = p;
362 	fl->blocks[next_free].length = count;
363 	fl->num_blocks++;
364 out:
365 	mutex_unlock(&rtas_update_flash_mutex);
366 	return count;
367 
368 nomem:
369 	rc = -ENOMEM;
370 error:
371 	mutex_unlock(&rtas_update_flash_mutex);
372 	return rc;
373 }
374 
375 /*
376  * Flash management routines.
377  */
378 static void manage_flash(struct rtas_manage_flash_t *args_buf, unsigned int op)
379 {
380 	s32 rc;
381 
382 	do {
383 		rc = rtas_call(rtas_token("ibm,manage-flash-image"), 1, 1,
384 			       NULL, op);
385 	} while (rtas_busy_delay(rc));
386 
387 	args_buf->status = rc;
388 }
389 
390 static ssize_t manage_flash_read(struct file *file, char __user *buf,
391 			       size_t count, loff_t *ppos)
392 {
393 	struct rtas_manage_flash_t *const args_buf = &rtas_manage_flash_data;
394 	char msg[RTAS_MSG_MAXLEN];
395 	int msglen, status;
396 
397 	mutex_lock(&rtas_manage_flash_mutex);
398 	status = args_buf->status;
399 	mutex_unlock(&rtas_manage_flash_mutex);
400 
401 	msglen = sprintf(msg, "%d\n", status);
402 	return simple_read_from_buffer(buf, count, ppos, msg, msglen);
403 }
404 
405 static ssize_t manage_flash_write(struct file *file, const char __user *buf,
406 				size_t count, loff_t *off)
407 {
408 	struct rtas_manage_flash_t *const args_buf = &rtas_manage_flash_data;
409 	static const char reject_str[] = "0";
410 	static const char commit_str[] = "1";
411 	char stkbuf[10];
412 	int op, rc;
413 
414 	mutex_lock(&rtas_manage_flash_mutex);
415 
416 	if ((args_buf->status == MANAGE_AUTH) || (count == 0))
417 		goto out;
418 
419 	op = -1;
420 	if (buf) {
421 		if (count > 9) count = 9;
422 		rc = -EFAULT;
423 		if (copy_from_user (stkbuf, buf, count))
424 			goto error;
425 		if (strncmp(stkbuf, reject_str, strlen(reject_str)) == 0)
426 			op = RTAS_REJECT_TMP_IMG;
427 		else if (strncmp(stkbuf, commit_str, strlen(commit_str)) == 0)
428 			op = RTAS_COMMIT_TMP_IMG;
429 	}
430 
431 	if (op == -1) {   /* buf is empty, or contains invalid string */
432 		rc = -EINVAL;
433 		goto error;
434 	}
435 
436 	manage_flash(args_buf, op);
437 out:
438 	mutex_unlock(&rtas_manage_flash_mutex);
439 	return count;
440 
441 error:
442 	mutex_unlock(&rtas_manage_flash_mutex);
443 	return rc;
444 }
445 
446 /*
447  * Validation routines.
448  */
449 static void validate_flash(struct rtas_validate_flash_t *args_buf)
450 {
451 	int token = rtas_token("ibm,validate-flash-image");
452 	int update_results;
453 	s32 rc;
454 
455 	rc = 0;
456 	do {
457 		spin_lock(&rtas_data_buf_lock);
458 		memcpy(rtas_data_buf, args_buf->buf, VALIDATE_BUF_SIZE);
459 		rc = rtas_call(token, 2, 2, &update_results,
460 			       (u32) __pa(rtas_data_buf), args_buf->buf_size);
461 		memcpy(args_buf->buf, rtas_data_buf, VALIDATE_BUF_SIZE);
462 		spin_unlock(&rtas_data_buf_lock);
463 	} while (rtas_busy_delay(rc));
464 
465 	args_buf->status = rc;
466 	args_buf->update_results = update_results;
467 }
468 
469 static int get_validate_flash_msg(struct rtas_validate_flash_t *args_buf,
470 		                   char *msg, int msglen)
471 {
472 	int n;
473 
474 	if (args_buf->status >= VALIDATE_TMP_UPDATE) {
475 		n = sprintf(msg, "%d\n", args_buf->update_results);
476 		if ((args_buf->update_results >= VALIDATE_CUR_UNKNOWN) ||
477 		    (args_buf->update_results == VALIDATE_TMP_UPDATE))
478 			n += snprintf(msg + n, msglen - n, "%s\n",
479 					args_buf->buf);
480 	} else {
481 		n = sprintf(msg, "%d\n", args_buf->status);
482 	}
483 	return n;
484 }
485 
486 static ssize_t validate_flash_read(struct file *file, char __user *buf,
487 			       size_t count, loff_t *ppos)
488 {
489 	struct rtas_validate_flash_t *const args_buf =
490 		&rtas_validate_flash_data;
491 	char msg[VALIDATE_MSG_LEN];
492 	int msglen;
493 
494 	mutex_lock(&rtas_validate_flash_mutex);
495 	msglen = get_validate_flash_msg(args_buf, msg, VALIDATE_MSG_LEN);
496 	mutex_unlock(&rtas_validate_flash_mutex);
497 
498 	return simple_read_from_buffer(buf, count, ppos, msg, msglen);
499 }
500 
501 static ssize_t validate_flash_write(struct file *file, const char __user *buf,
502 				    size_t count, loff_t *off)
503 {
504 	struct rtas_validate_flash_t *const args_buf =
505 		&rtas_validate_flash_data;
506 	int rc;
507 
508 	mutex_lock(&rtas_validate_flash_mutex);
509 
510 	/* We are only interested in the first 4K of the
511 	 * candidate image */
512 	if ((*off >= VALIDATE_BUF_SIZE) ||
513 		(args_buf->status == VALIDATE_AUTH)) {
514 		*off += count;
515 		mutex_unlock(&rtas_validate_flash_mutex);
516 		return count;
517 	}
518 
519 	if (*off + count >= VALIDATE_BUF_SIZE)  {
520 		count = VALIDATE_BUF_SIZE - *off;
521 		args_buf->status = VALIDATE_READY;
522 	} else {
523 		args_buf->status = VALIDATE_INCOMPLETE;
524 	}
525 
526 	if (!access_ok(VERIFY_READ, buf, count)) {
527 		rc = -EFAULT;
528 		goto done;
529 	}
530 	if (copy_from_user(args_buf->buf + *off, buf, count)) {
531 		rc = -EFAULT;
532 		goto done;
533 	}
534 
535 	*off += count;
536 	rc = count;
537 done:
538 	mutex_unlock(&rtas_validate_flash_mutex);
539 	return rc;
540 }
541 
542 static int validate_flash_release(struct inode *inode, struct file *file)
543 {
544 	struct rtas_validate_flash_t *const args_buf =
545 		&rtas_validate_flash_data;
546 
547 	mutex_lock(&rtas_validate_flash_mutex);
548 
549 	if (args_buf->status == VALIDATE_READY) {
550 		args_buf->buf_size = VALIDATE_BUF_SIZE;
551 		validate_flash(args_buf);
552 	}
553 
554 	mutex_unlock(&rtas_validate_flash_mutex);
555 	return 0;
556 }
557 
558 /*
559  * On-reboot flash update applicator.
560  */
561 static void rtas_flash_firmware(int reboot_type)
562 {
563 	unsigned long image_size;
564 	struct flash_block_list *f, *next, *flist;
565 	unsigned long rtas_block_list;
566 	int i, status, update_token;
567 
568 	if (rtas_firmware_flash_list == NULL)
569 		return;		/* nothing to do */
570 
571 	if (reboot_type != SYS_RESTART) {
572 		printk(KERN_ALERT "FLASH: firmware flash requires a reboot\n");
573 		printk(KERN_ALERT "FLASH: the firmware image will NOT be flashed\n");
574 		return;
575 	}
576 
577 	update_token = rtas_token("ibm,update-flash-64-and-reboot");
578 	if (update_token == RTAS_UNKNOWN_SERVICE) {
579 		printk(KERN_ALERT "FLASH: ibm,update-flash-64-and-reboot "
580 		       "is not available -- not a service partition?\n");
581 		printk(KERN_ALERT "FLASH: firmware will not be flashed\n");
582 		return;
583 	}
584 
585 	/*
586 	 * Just before starting the firmware flash, cancel the event scan work
587 	 * to avoid any soft lockup issues.
588 	 */
589 	rtas_cancel_event_scan();
590 
591 	/*
592 	 * NOTE: the "first" block must be under 4GB, so we create
593 	 * an entry with no data blocks in the reserved buffer in
594 	 * the kernel data segment.
595 	 */
596 	spin_lock(&rtas_data_buf_lock);
597 	flist = (struct flash_block_list *)&rtas_data_buf[0];
598 	flist->num_blocks = 0;
599 	flist->next = rtas_firmware_flash_list;
600 	rtas_block_list = __pa(flist);
601 	if (rtas_block_list >= 4UL*1024*1024*1024) {
602 		printk(KERN_ALERT "FLASH: kernel bug...flash list header addr above 4GB\n");
603 		spin_unlock(&rtas_data_buf_lock);
604 		return;
605 	}
606 
607 	printk(KERN_ALERT "FLASH: preparing saved firmware image for flash\n");
608 	/* Update the block_list in place. */
609 	rtas_firmware_flash_list = NULL; /* too hard to backout on error */
610 	image_size = 0;
611 	for (f = flist; f; f = next) {
612 		/* Translate data addrs to absolute */
613 		for (i = 0; i < f->num_blocks; i++) {
614 			f->blocks[i].data = (char *)cpu_to_be64(__pa(f->blocks[i].data));
615 			image_size += f->blocks[i].length;
616 			f->blocks[i].length = cpu_to_be64(f->blocks[i].length);
617 		}
618 		next = f->next;
619 		/* Don't translate NULL pointer for last entry */
620 		if (f->next)
621 			f->next = (struct flash_block_list *)cpu_to_be64(__pa(f->next));
622 		else
623 			f->next = NULL;
624 		/* make num_blocks into the version/length field */
625 		f->num_blocks = (FLASH_BLOCK_LIST_VERSION << 56) | ((f->num_blocks+1)*16);
626 		f->num_blocks = cpu_to_be64(f->num_blocks);
627 	}
628 
629 	printk(KERN_ALERT "FLASH: flash image is %ld bytes\n", image_size);
630 	printk(KERN_ALERT "FLASH: performing flash and reboot\n");
631 	rtas_progress("Flashing        \n", 0x0);
632 	rtas_progress("Please Wait...  ", 0x0);
633 	printk(KERN_ALERT "FLASH: this will take several minutes.  Do not power off!\n");
634 	status = rtas_call(update_token, 1, 1, NULL, rtas_block_list);
635 	switch (status) {	/* should only get "bad" status */
636 	    case 0:
637 		printk(KERN_ALERT "FLASH: success\n");
638 		break;
639 	    case -1:
640 		printk(KERN_ALERT "FLASH: hardware error.  Firmware may not be not flashed\n");
641 		break;
642 	    case -3:
643 		printk(KERN_ALERT "FLASH: image is corrupt or not correct for this platform.  Firmware not flashed\n");
644 		break;
645 	    case -4:
646 		printk(KERN_ALERT "FLASH: flash failed when partially complete.  System may not reboot\n");
647 		break;
648 	    default:
649 		printk(KERN_ALERT "FLASH: unknown flash return code %d\n", status);
650 		break;
651 	}
652 	spin_unlock(&rtas_data_buf_lock);
653 }
654 
655 /*
656  * Manifest of proc files to create
657  */
658 struct rtas_flash_file {
659 	const char *filename;
660 	const char *rtas_call_name;
661 	int *status;
662 	const struct file_operations fops;
663 };
664 
665 static const struct rtas_flash_file rtas_flash_files[] = {
666 	{
667 		.filename	= "powerpc/rtas/" FIRMWARE_FLASH_NAME,
668 		.rtas_call_name	= "ibm,update-flash-64-and-reboot",
669 		.status		= &rtas_update_flash_data.status,
670 		.fops.read	= rtas_flash_read_msg,
671 		.fops.write	= rtas_flash_write,
672 		.fops.release	= rtas_flash_release,
673 		.fops.llseek	= default_llseek,
674 	},
675 	{
676 		.filename	= "powerpc/rtas/" FIRMWARE_UPDATE_NAME,
677 		.rtas_call_name	= "ibm,update-flash-64-and-reboot",
678 		.status		= &rtas_update_flash_data.status,
679 		.fops.read	= rtas_flash_read_num,
680 		.fops.write	= rtas_flash_write,
681 		.fops.release	= rtas_flash_release,
682 		.fops.llseek	= default_llseek,
683 	},
684 	{
685 		.filename	= "powerpc/rtas/" VALIDATE_FLASH_NAME,
686 		.rtas_call_name	= "ibm,validate-flash-image",
687 		.status		= &rtas_validate_flash_data.status,
688 		.fops.read	= validate_flash_read,
689 		.fops.write	= validate_flash_write,
690 		.fops.release	= validate_flash_release,
691 		.fops.llseek	= default_llseek,
692 	},
693 	{
694 		.filename	= "powerpc/rtas/" MANAGE_FLASH_NAME,
695 		.rtas_call_name	= "ibm,manage-flash-image",
696 		.status		= &rtas_manage_flash_data.status,
697 		.fops.read	= manage_flash_read,
698 		.fops.write	= manage_flash_write,
699 		.fops.llseek	= default_llseek,
700 	}
701 };
702 
703 static int __init rtas_flash_init(void)
704 {
705 	int i;
706 
707 	if (rtas_token("ibm,update-flash-64-and-reboot") ==
708 		       RTAS_UNKNOWN_SERVICE) {
709 		pr_info("rtas_flash: no firmware flash support\n");
710 		return -EINVAL;
711 	}
712 
713 	rtas_validate_flash_data.buf = kzalloc(VALIDATE_BUF_SIZE, GFP_KERNEL);
714 	if (!rtas_validate_flash_data.buf)
715 		return -ENOMEM;
716 
717 	flash_block_cache = kmem_cache_create("rtas_flash_cache",
718 					      RTAS_BLK_SIZE, RTAS_BLK_SIZE, 0,
719 					      NULL);
720 	if (!flash_block_cache) {
721 		printk(KERN_ERR "%s: failed to create block cache\n",
722 				__func__);
723 		goto enomem_buf;
724 	}
725 
726 	for (i = 0; i < ARRAY_SIZE(rtas_flash_files); i++) {
727 		const struct rtas_flash_file *f = &rtas_flash_files[i];
728 		int token;
729 
730 		if (!proc_create(f->filename, 0600, NULL, &f->fops))
731 			goto enomem;
732 
733 		/*
734 		 * This code assumes that the status int is the first member of the
735 		 * struct
736 		 */
737 		token = rtas_token(f->rtas_call_name);
738 		if (token == RTAS_UNKNOWN_SERVICE)
739 			*f->status = FLASH_AUTH;
740 		else
741 			*f->status = FLASH_NO_OP;
742 	}
743 
744 	rtas_flash_term_hook = rtas_flash_firmware;
745 	return 0;
746 
747 enomem:
748 	while (--i >= 0) {
749 		const struct rtas_flash_file *f = &rtas_flash_files[i];
750 		remove_proc_entry(f->filename, NULL);
751 	}
752 
753 	kmem_cache_destroy(flash_block_cache);
754 enomem_buf:
755 	kfree(rtas_validate_flash_data.buf);
756 	return -ENOMEM;
757 }
758 
759 static void __exit rtas_flash_cleanup(void)
760 {
761 	int i;
762 
763 	rtas_flash_term_hook = NULL;
764 
765 	if (rtas_firmware_flash_list) {
766 		free_flash_list(rtas_firmware_flash_list);
767 		rtas_firmware_flash_list = NULL;
768 	}
769 
770 	for (i = 0; i < ARRAY_SIZE(rtas_flash_files); i++) {
771 		const struct rtas_flash_file *f = &rtas_flash_files[i];
772 		remove_proc_entry(f->filename, NULL);
773 	}
774 
775 	kmem_cache_destroy(flash_block_cache);
776 	kfree(rtas_validate_flash_data.buf);
777 }
778 
779 module_init(rtas_flash_init);
780 module_exit(rtas_flash_cleanup);
781 MODULE_LICENSE("GPL");
782