include cleanup: Update gfp.h and slab.h includes to prepare for breaking implicit...
[GitHub/mt8127/android_kernel_alcatel_ttab.git] / arch / powerpc / kernel / rtas_flash.c
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 <asm/delay.h>
21 #include <asm/uaccess.h>
22 #include <asm/rtas.h>
23 #include <asm/abs_addr.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 #define VALIDATE_TMP_UPDATE 0 /* Validate Return Status */
61 #define VALIDATE_FLASH_AUTH 1 /* Validate Return Status */
62 #define VALIDATE_INVALID_IMG 2 /* Validate Return Status */
63 #define VALIDATE_CUR_UNKNOWN 3 /* Validate Return Status */
64 #define VALIDATE_TMP_COMMIT_DL 4 /* Validate Return Status */
65 #define VALIDATE_TMP_COMMIT 5 /* Validate Return Status */
66 #define VALIDATE_TMP_UPDATE_DL 6 /* Validate Return Status */
67
68 /* ibm,manage-flash-image operation tokens */
69 #define RTAS_REJECT_TMP_IMG 0
70 #define RTAS_COMMIT_TMP_IMG 1
71
72 /* Array sizes */
73 #define VALIDATE_BUF_SIZE 4096
74 #define RTAS_MSG_MAXLEN 64
75
76 /* Quirk - RTAS requires 4k list length and block size */
77 #define RTAS_BLKLIST_LENGTH 4096
78 #define RTAS_BLK_SIZE 4096
79
80 struct flash_block {
81 char *data;
82 unsigned long length;
83 };
84
85 /* This struct is very similar but not identical to
86 * that needed by the rtas flash update.
87 * All we need to do for rtas is rewrite num_blocks
88 * into a version/length and translate the pointers
89 * to absolute.
90 */
91 #define FLASH_BLOCKS_PER_NODE ((RTAS_BLKLIST_LENGTH - 16) / sizeof(struct flash_block))
92 struct flash_block_list {
93 unsigned long num_blocks;
94 struct flash_block_list *next;
95 struct flash_block blocks[FLASH_BLOCKS_PER_NODE];
96 };
97 struct flash_block_list_header { /* just the header of flash_block_list */
98 unsigned long num_blocks;
99 struct flash_block_list *next;
100 };
101
102 static struct flash_block_list_header rtas_firmware_flash_list = {0, NULL};
103
104 /* Use slab cache to guarantee 4k alignment */
105 static struct kmem_cache *flash_block_cache = NULL;
106
107 #define FLASH_BLOCK_LIST_VERSION (1UL)
108
109 /* Local copy of the flash block list.
110 * We only allow one open of the flash proc file and create this
111 * list as we go. This list will be put in the
112 * rtas_firmware_flash_list var once it is fully read.
113 *
114 * For convenience as we build the list we use virtual addrs,
115 * we do not fill in the version number, and the length field
116 * is treated as the number of entries currently in the block
117 * (i.e. not a byte count). This is all fixed on release.
118 */
119
120 /* Status int must be first member of struct */
121 struct rtas_update_flash_t
122 {
123 int status; /* Flash update status */
124 struct flash_block_list *flist; /* Local copy of flash block list */
125 };
126
127 /* Status int must be first member of struct */
128 struct rtas_manage_flash_t
129 {
130 int status; /* Returned status */
131 unsigned int op; /* Reject or commit image */
132 };
133
134 /* Status int must be first member of struct */
135 struct rtas_validate_flash_t
136 {
137 int status; /* Returned status */
138 char buf[VALIDATE_BUF_SIZE]; /* Candidate image buffer */
139 unsigned int buf_size; /* Size of image buf */
140 unsigned int update_results; /* Update results token */
141 };
142
143 static DEFINE_SPINLOCK(flash_file_open_lock);
144 static struct proc_dir_entry *firmware_flash_pde;
145 static struct proc_dir_entry *firmware_update_pde;
146 static struct proc_dir_entry *validate_pde;
147 static struct proc_dir_entry *manage_pde;
148
149 /* Do simple sanity checks on the flash image. */
150 static int flash_list_valid(struct flash_block_list *flist)
151 {
152 struct flash_block_list *f;
153 int i;
154 unsigned long block_size, image_size;
155
156 /* Paranoid self test here. We also collect the image size. */
157 image_size = 0;
158 for (f = flist; f; f = f->next) {
159 for (i = 0; i < f->num_blocks; i++) {
160 if (f->blocks[i].data == NULL) {
161 return FLASH_IMG_NULL_DATA;
162 }
163 block_size = f->blocks[i].length;
164 if (block_size <= 0 || block_size > RTAS_BLK_SIZE) {
165 return FLASH_IMG_BAD_LEN;
166 }
167 image_size += block_size;
168 }
169 }
170
171 if (image_size < (256 << 10)) {
172 if (image_size < 2)
173 return FLASH_NO_OP;
174 }
175
176 printk(KERN_INFO "FLASH: flash image with %ld bytes stored for hardware flash on reboot\n", image_size);
177
178 return FLASH_IMG_READY;
179 }
180
181 static void free_flash_list(struct flash_block_list *f)
182 {
183 struct flash_block_list *next;
184 int i;
185
186 while (f) {
187 for (i = 0; i < f->num_blocks; i++)
188 kmem_cache_free(flash_block_cache, f->blocks[i].data);
189 next = f->next;
190 kmem_cache_free(flash_block_cache, f);
191 f = next;
192 }
193 }
194
195 static int rtas_flash_release(struct inode *inode, struct file *file)
196 {
197 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
198 struct rtas_update_flash_t *uf;
199
200 uf = (struct rtas_update_flash_t *) dp->data;
201 if (uf->flist) {
202 /* File was opened in write mode for a new flash attempt */
203 /* Clear saved list */
204 if (rtas_firmware_flash_list.next) {
205 free_flash_list(rtas_firmware_flash_list.next);
206 rtas_firmware_flash_list.next = NULL;
207 }
208
209 if (uf->status != FLASH_AUTH)
210 uf->status = flash_list_valid(uf->flist);
211
212 if (uf->status == FLASH_IMG_READY)
213 rtas_firmware_flash_list.next = uf->flist;
214 else
215 free_flash_list(uf->flist);
216
217 uf->flist = NULL;
218 }
219
220 atomic_dec(&dp->count);
221 return 0;
222 }
223
224 static void get_flash_status_msg(int status, char *buf)
225 {
226 char *msg;
227
228 switch (status) {
229 case FLASH_AUTH:
230 msg = "error: this partition does not have service authority\n";
231 break;
232 case FLASH_NO_OP:
233 msg = "info: no firmware image for flash\n";
234 break;
235 case FLASH_IMG_SHORT:
236 msg = "error: flash image short\n";
237 break;
238 case FLASH_IMG_BAD_LEN:
239 msg = "error: internal error bad length\n";
240 break;
241 case FLASH_IMG_NULL_DATA:
242 msg = "error: internal error null data\n";
243 break;
244 case FLASH_IMG_READY:
245 msg = "ready: firmware image ready for flash on reboot\n";
246 break;
247 default:
248 sprintf(buf, "error: unexpected status value %d\n", status);
249 return;
250 }
251
252 strcpy(buf, msg);
253 }
254
255 /* Reading the proc file will show status (not the firmware contents) */
256 static ssize_t rtas_flash_read(struct file *file, char __user *buf,
257 size_t count, loff_t *ppos)
258 {
259 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
260 struct rtas_update_flash_t *uf;
261 char msg[RTAS_MSG_MAXLEN];
262 int msglen;
263
264 uf = (struct rtas_update_flash_t *) dp->data;
265
266 if (!strcmp(dp->name, FIRMWARE_FLASH_NAME)) {
267 get_flash_status_msg(uf->status, msg);
268 } else { /* FIRMWARE_UPDATE_NAME */
269 sprintf(msg, "%d\n", uf->status);
270 }
271 msglen = strlen(msg);
272 if (msglen > count)
273 msglen = count;
274
275 if (ppos && *ppos != 0)
276 return 0; /* be cheap */
277
278 if (!access_ok(VERIFY_WRITE, buf, msglen))
279 return -EINVAL;
280
281 if (copy_to_user(buf, msg, msglen))
282 return -EFAULT;
283
284 if (ppos)
285 *ppos = msglen;
286 return msglen;
287 }
288
289 /* constructor for flash_block_cache */
290 void rtas_block_ctor(void *ptr)
291 {
292 memset(ptr, 0, RTAS_BLK_SIZE);
293 }
294
295 /* We could be much more efficient here. But to keep this function
296 * simple we allocate a page to the block list no matter how small the
297 * count is. If the system is low on memory it will be just as well
298 * that we fail....
299 */
300 static ssize_t rtas_flash_write(struct file *file, const char __user *buffer,
301 size_t count, loff_t *off)
302 {
303 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
304 struct rtas_update_flash_t *uf;
305 char *p;
306 int next_free;
307 struct flash_block_list *fl;
308
309 uf = (struct rtas_update_flash_t *) dp->data;
310
311 if (uf->status == FLASH_AUTH || count == 0)
312 return count; /* discard data */
313
314 /* In the case that the image is not ready for flashing, the memory
315 * allocated for the block list will be freed upon the release of the
316 * proc file
317 */
318 if (uf->flist == NULL) {
319 uf->flist = kmem_cache_alloc(flash_block_cache, GFP_KERNEL);
320 if (!uf->flist)
321 return -ENOMEM;
322 }
323
324 fl = uf->flist;
325 while (fl->next)
326 fl = fl->next; /* seek to last block_list for append */
327 next_free = fl->num_blocks;
328 if (next_free == FLASH_BLOCKS_PER_NODE) {
329 /* Need to allocate another block_list */
330 fl->next = kmem_cache_alloc(flash_block_cache, GFP_KERNEL);
331 if (!fl->next)
332 return -ENOMEM;
333 fl = fl->next;
334 next_free = 0;
335 }
336
337 if (count > RTAS_BLK_SIZE)
338 count = RTAS_BLK_SIZE;
339 p = kmem_cache_alloc(flash_block_cache, GFP_KERNEL);
340 if (!p)
341 return -ENOMEM;
342
343 if(copy_from_user(p, buffer, count)) {
344 kmem_cache_free(flash_block_cache, p);
345 return -EFAULT;
346 }
347 fl->blocks[next_free].data = p;
348 fl->blocks[next_free].length = count;
349 fl->num_blocks++;
350
351 return count;
352 }
353
354 static int rtas_excl_open(struct inode *inode, struct file *file)
355 {
356 struct proc_dir_entry *dp = PDE(inode);
357
358 /* Enforce exclusive open with use count of PDE */
359 spin_lock(&flash_file_open_lock);
360 if (atomic_read(&dp->count) > 2) {
361 spin_unlock(&flash_file_open_lock);
362 return -EBUSY;
363 }
364
365 atomic_inc(&dp->count);
366 spin_unlock(&flash_file_open_lock);
367
368 return 0;
369 }
370
371 static int rtas_excl_release(struct inode *inode, struct file *file)
372 {
373 struct proc_dir_entry *dp = PDE(inode);
374
375 atomic_dec(&dp->count);
376
377 return 0;
378 }
379
380 static void manage_flash(struct rtas_manage_flash_t *args_buf)
381 {
382 s32 rc;
383
384 do {
385 rc = rtas_call(rtas_token("ibm,manage-flash-image"), 1,
386 1, NULL, args_buf->op);
387 } while (rtas_busy_delay(rc));
388
389 args_buf->status = rc;
390 }
391
392 static ssize_t manage_flash_read(struct file *file, char __user *buf,
393 size_t count, loff_t *ppos)
394 {
395 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
396 struct rtas_manage_flash_t *args_buf;
397 char msg[RTAS_MSG_MAXLEN];
398 int msglen;
399
400 args_buf = (struct rtas_manage_flash_t *) dp->data;
401 if (args_buf == NULL)
402 return 0;
403
404 msglen = sprintf(msg, "%d\n", args_buf->status);
405 if (msglen > count)
406 msglen = count;
407
408 if (ppos && *ppos != 0)
409 return 0; /* be cheap */
410
411 if (!access_ok(VERIFY_WRITE, buf, msglen))
412 return -EINVAL;
413
414 if (copy_to_user(buf, msg, msglen))
415 return -EFAULT;
416
417 if (ppos)
418 *ppos = msglen;
419 return msglen;
420 }
421
422 static ssize_t manage_flash_write(struct file *file, const char __user *buf,
423 size_t count, loff_t *off)
424 {
425 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
426 struct rtas_manage_flash_t *args_buf;
427 const char reject_str[] = "0";
428 const char commit_str[] = "1";
429 char stkbuf[10];
430 int op;
431
432 args_buf = (struct rtas_manage_flash_t *) dp->data;
433 if ((args_buf->status == MANAGE_AUTH) || (count == 0))
434 return count;
435
436 op = -1;
437 if (buf) {
438 if (count > 9) count = 9;
439 if (copy_from_user (stkbuf, buf, count)) {
440 return -EFAULT;
441 }
442 if (strncmp(stkbuf, reject_str, strlen(reject_str)) == 0)
443 op = RTAS_REJECT_TMP_IMG;
444 else if (strncmp(stkbuf, commit_str, strlen(commit_str)) == 0)
445 op = RTAS_COMMIT_TMP_IMG;
446 }
447
448 if (op == -1) /* buf is empty, or contains invalid string */
449 return -EINVAL;
450
451 args_buf->op = op;
452 manage_flash(args_buf);
453
454 return count;
455 }
456
457 static void validate_flash(struct rtas_validate_flash_t *args_buf)
458 {
459 int token = rtas_token("ibm,validate-flash-image");
460 int update_results;
461 s32 rc;
462
463 rc = 0;
464 do {
465 spin_lock(&rtas_data_buf_lock);
466 memcpy(rtas_data_buf, args_buf->buf, VALIDATE_BUF_SIZE);
467 rc = rtas_call(token, 2, 2, &update_results,
468 (u32) __pa(rtas_data_buf), args_buf->buf_size);
469 memcpy(args_buf->buf, rtas_data_buf, VALIDATE_BUF_SIZE);
470 spin_unlock(&rtas_data_buf_lock);
471 } while (rtas_busy_delay(rc));
472
473 args_buf->status = rc;
474 args_buf->update_results = update_results;
475 }
476
477 static int get_validate_flash_msg(struct rtas_validate_flash_t *args_buf,
478 char *msg)
479 {
480 int n;
481
482 if (args_buf->status >= VALIDATE_TMP_UPDATE) {
483 n = sprintf(msg, "%d\n", args_buf->update_results);
484 if ((args_buf->update_results >= VALIDATE_CUR_UNKNOWN) ||
485 (args_buf->update_results == VALIDATE_TMP_UPDATE))
486 n += sprintf(msg + n, "%s\n", args_buf->buf);
487 } else {
488 n = sprintf(msg, "%d\n", args_buf->status);
489 }
490 return n;
491 }
492
493 static ssize_t validate_flash_read(struct file *file, char __user *buf,
494 size_t count, loff_t *ppos)
495 {
496 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
497 struct rtas_validate_flash_t *args_buf;
498 char msg[RTAS_MSG_MAXLEN];
499 int msglen;
500
501 args_buf = (struct rtas_validate_flash_t *) dp->data;
502
503 if (ppos && *ppos != 0)
504 return 0; /* be cheap */
505
506 msglen = get_validate_flash_msg(args_buf, msg);
507 if (msglen > count)
508 msglen = count;
509
510 if (!access_ok(VERIFY_WRITE, buf, msglen))
511 return -EINVAL;
512
513 if (copy_to_user(buf, msg, msglen))
514 return -EFAULT;
515
516 if (ppos)
517 *ppos = msglen;
518 return msglen;
519 }
520
521 static ssize_t validate_flash_write(struct file *file, const char __user *buf,
522 size_t count, loff_t *off)
523 {
524 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
525 struct rtas_validate_flash_t *args_buf;
526 int rc;
527
528 args_buf = (struct rtas_validate_flash_t *) dp->data;
529
530 if (dp->data == NULL) {
531 dp->data = kmalloc(sizeof(struct rtas_validate_flash_t),
532 GFP_KERNEL);
533 if (dp->data == NULL)
534 return -ENOMEM;
535 }
536
537 /* We are only interested in the first 4K of the
538 * candidate image */
539 if ((*off >= VALIDATE_BUF_SIZE) ||
540 (args_buf->status == VALIDATE_AUTH)) {
541 *off += count;
542 return count;
543 }
544
545 if (*off + count >= VALIDATE_BUF_SIZE) {
546 count = VALIDATE_BUF_SIZE - *off;
547 args_buf->status = VALIDATE_READY;
548 } else {
549 args_buf->status = VALIDATE_INCOMPLETE;
550 }
551
552 if (!access_ok(VERIFY_READ, buf, count)) {
553 rc = -EFAULT;
554 goto done;
555 }
556 if (copy_from_user(args_buf->buf + *off, buf, count)) {
557 rc = -EFAULT;
558 goto done;
559 }
560
561 *off += count;
562 rc = count;
563 done:
564 if (rc < 0) {
565 kfree(dp->data);
566 dp->data = NULL;
567 }
568 return rc;
569 }
570
571 static int validate_flash_release(struct inode *inode, struct file *file)
572 {
573 struct proc_dir_entry *dp = PDE(file->f_path.dentry->d_inode);
574 struct rtas_validate_flash_t *args_buf;
575
576 args_buf = (struct rtas_validate_flash_t *) dp->data;
577
578 if (args_buf->status == VALIDATE_READY) {
579 args_buf->buf_size = VALIDATE_BUF_SIZE;
580 validate_flash(args_buf);
581 }
582
583 /* The matching atomic_inc was in rtas_excl_open() */
584 atomic_dec(&dp->count);
585
586 return 0;
587 }
588
589 static void rtas_flash_firmware(int reboot_type)
590 {
591 unsigned long image_size;
592 struct flash_block_list *f, *next, *flist;
593 unsigned long rtas_block_list;
594 int i, status, update_token;
595
596 if (rtas_firmware_flash_list.next == NULL)
597 return; /* nothing to do */
598
599 if (reboot_type != SYS_RESTART) {
600 printk(KERN_ALERT "FLASH: firmware flash requires a reboot\n");
601 printk(KERN_ALERT "FLASH: the firmware image will NOT be flashed\n");
602 return;
603 }
604
605 update_token = rtas_token("ibm,update-flash-64-and-reboot");
606 if (update_token == RTAS_UNKNOWN_SERVICE) {
607 printk(KERN_ALERT "FLASH: ibm,update-flash-64-and-reboot "
608 "is not available -- not a service partition?\n");
609 printk(KERN_ALERT "FLASH: firmware will not be flashed\n");
610 return;
611 }
612
613 /* NOTE: the "first" block list is a global var with no data
614 * blocks in the kernel data segment. We do this because
615 * we want to ensure this block_list addr is under 4GB.
616 */
617 rtas_firmware_flash_list.num_blocks = 0;
618 flist = (struct flash_block_list *)&rtas_firmware_flash_list;
619 rtas_block_list = virt_to_abs(flist);
620 if (rtas_block_list >= 4UL*1024*1024*1024) {
621 printk(KERN_ALERT "FLASH: kernel bug...flash list header addr above 4GB\n");
622 return;
623 }
624
625 printk(KERN_ALERT "FLASH: preparing saved firmware image for flash\n");
626 /* Update the block_list in place. */
627 image_size = 0;
628 for (f = flist; f; f = next) {
629 /* Translate data addrs to absolute */
630 for (i = 0; i < f->num_blocks; i++) {
631 f->blocks[i].data = (char *)virt_to_abs(f->blocks[i].data);
632 image_size += f->blocks[i].length;
633 }
634 next = f->next;
635 /* Don't translate NULL pointer for last entry */
636 if (f->next)
637 f->next = (struct flash_block_list *)virt_to_abs(f->next);
638 else
639 f->next = NULL;
640 /* make num_blocks into the version/length field */
641 f->num_blocks = (FLASH_BLOCK_LIST_VERSION << 56) | ((f->num_blocks+1)*16);
642 }
643
644 printk(KERN_ALERT "FLASH: flash image is %ld bytes\n", image_size);
645 printk(KERN_ALERT "FLASH: performing flash and reboot\n");
646 rtas_progress("Flashing \n", 0x0);
647 rtas_progress("Please Wait... ", 0x0);
648 printk(KERN_ALERT "FLASH: this will take several minutes. Do not power off!\n");
649 status = rtas_call(update_token, 1, 1, NULL, rtas_block_list);
650 switch (status) { /* should only get "bad" status */
651 case 0:
652 printk(KERN_ALERT "FLASH: success\n");
653 break;
654 case -1:
655 printk(KERN_ALERT "FLASH: hardware error. Firmware may not be not flashed\n");
656 break;
657 case -3:
658 printk(KERN_ALERT "FLASH: image is corrupt or not correct for this platform. Firmware not flashed\n");
659 break;
660 case -4:
661 printk(KERN_ALERT "FLASH: flash failed when partially complete. System may not reboot\n");
662 break;
663 default:
664 printk(KERN_ALERT "FLASH: unknown flash return code %d\n", status);
665 break;
666 }
667 }
668
669 static void remove_flash_pde(struct proc_dir_entry *dp)
670 {
671 if (dp) {
672 kfree(dp->data);
673 remove_proc_entry(dp->name, dp->parent);
674 }
675 }
676
677 static int initialize_flash_pde_data(const char *rtas_call_name,
678 size_t buf_size,
679 struct proc_dir_entry *dp)
680 {
681 int *status;
682 int token;
683
684 dp->data = kzalloc(buf_size, GFP_KERNEL);
685 if (dp->data == NULL) {
686 remove_flash_pde(dp);
687 return -ENOMEM;
688 }
689
690 /*
691 * This code assumes that the status int is the first member of the
692 * struct
693 */
694 status = (int *) dp->data;
695 token = rtas_token(rtas_call_name);
696 if (token == RTAS_UNKNOWN_SERVICE)
697 *status = FLASH_AUTH;
698 else
699 *status = FLASH_NO_OP;
700
701 return 0;
702 }
703
704 static struct proc_dir_entry *create_flash_pde(const char *filename,
705 const struct file_operations *fops)
706 {
707 return proc_create(filename, S_IRUSR | S_IWUSR, NULL, fops);
708 }
709
710 static const struct file_operations rtas_flash_operations = {
711 .owner = THIS_MODULE,
712 .read = rtas_flash_read,
713 .write = rtas_flash_write,
714 .open = rtas_excl_open,
715 .release = rtas_flash_release,
716 };
717
718 static const struct file_operations manage_flash_operations = {
719 .owner = THIS_MODULE,
720 .read = manage_flash_read,
721 .write = manage_flash_write,
722 .open = rtas_excl_open,
723 .release = rtas_excl_release,
724 };
725
726 static const struct file_operations validate_flash_operations = {
727 .owner = THIS_MODULE,
728 .read = validate_flash_read,
729 .write = validate_flash_write,
730 .open = rtas_excl_open,
731 .release = validate_flash_release,
732 };
733
734 static int __init rtas_flash_init(void)
735 {
736 int rc;
737
738 if (rtas_token("ibm,update-flash-64-and-reboot") ==
739 RTAS_UNKNOWN_SERVICE) {
740 printk(KERN_ERR "rtas_flash: no firmware flash support\n");
741 return 1;
742 }
743
744 firmware_flash_pde = create_flash_pde("powerpc/rtas/"
745 FIRMWARE_FLASH_NAME,
746 &rtas_flash_operations);
747 if (firmware_flash_pde == NULL) {
748 rc = -ENOMEM;
749 goto cleanup;
750 }
751
752 rc = initialize_flash_pde_data("ibm,update-flash-64-and-reboot",
753 sizeof(struct rtas_update_flash_t),
754 firmware_flash_pde);
755 if (rc != 0)
756 goto cleanup;
757
758 firmware_update_pde = create_flash_pde("powerpc/rtas/"
759 FIRMWARE_UPDATE_NAME,
760 &rtas_flash_operations);
761 if (firmware_update_pde == NULL) {
762 rc = -ENOMEM;
763 goto cleanup;
764 }
765
766 rc = initialize_flash_pde_data("ibm,update-flash-64-and-reboot",
767 sizeof(struct rtas_update_flash_t),
768 firmware_update_pde);
769 if (rc != 0)
770 goto cleanup;
771
772 validate_pde = create_flash_pde("powerpc/rtas/" VALIDATE_FLASH_NAME,
773 &validate_flash_operations);
774 if (validate_pde == NULL) {
775 rc = -ENOMEM;
776 goto cleanup;
777 }
778
779 rc = initialize_flash_pde_data("ibm,validate-flash-image",
780 sizeof(struct rtas_validate_flash_t),
781 validate_pde);
782 if (rc != 0)
783 goto cleanup;
784
785 manage_pde = create_flash_pde("powerpc/rtas/" MANAGE_FLASH_NAME,
786 &manage_flash_operations);
787 if (manage_pde == NULL) {
788 rc = -ENOMEM;
789 goto cleanup;
790 }
791
792 rc = initialize_flash_pde_data("ibm,manage-flash-image",
793 sizeof(struct rtas_manage_flash_t),
794 manage_pde);
795 if (rc != 0)
796 goto cleanup;
797
798 rtas_flash_term_hook = rtas_flash_firmware;
799
800 flash_block_cache = kmem_cache_create("rtas_flash_cache",
801 RTAS_BLK_SIZE, RTAS_BLK_SIZE, 0,
802 rtas_block_ctor);
803 if (!flash_block_cache) {
804 printk(KERN_ERR "%s: failed to create block cache\n",
805 __func__);
806 rc = -ENOMEM;
807 goto cleanup;
808 }
809 return 0;
810
811 cleanup:
812 remove_flash_pde(firmware_flash_pde);
813 remove_flash_pde(firmware_update_pde);
814 remove_flash_pde(validate_pde);
815 remove_flash_pde(manage_pde);
816
817 return rc;
818 }
819
820 static void __exit rtas_flash_cleanup(void)
821 {
822 rtas_flash_term_hook = NULL;
823
824 if (flash_block_cache)
825 kmem_cache_destroy(flash_block_cache);
826
827 remove_flash_pde(firmware_flash_pde);
828 remove_flash_pde(firmware_update_pde);
829 remove_flash_pde(validate_pde);
830 remove_flash_pde(manage_pde);
831 }
832
833 module_init(rtas_flash_init);
834 module_exit(rtas_flash_cleanup);
835 MODULE_LICENSE("GPL");