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