powerpc/rtas: Turn rtas lock into a raw spinlock
[safe/jmp/linux-2.6] / arch / powerpc / kernel / rtas.c
1 /*
2  *
3  * Procedures for interfacing to the RTAS on CHRP machines.
4  *
5  * Peter Bergner, IBM   March 2001.
6  * Copyright (C) 2001 IBM.
7  *
8  *      This program is free software; you can redistribute it and/or
9  *      modify it under the terms of the GNU General Public License
10  *      as published by the Free Software Foundation; either version
11  *      2 of the License, or (at your option) any later version.
12  */
13
14 #include <stdarg.h>
15 #include <linux/kernel.h>
16 #include <linux/types.h>
17 #include <linux/spinlock.h>
18 #include <linux/module.h>
19 #include <linux/init.h>
20 #include <linux/capability.h>
21 #include <linux/delay.h>
22 #include <linux/smp.h>
23 #include <linux/completion.h>
24 #include <linux/cpumask.h>
25 #include <linux/lmb.h>
26
27 #include <asm/prom.h>
28 #include <asm/rtas.h>
29 #include <asm/hvcall.h>
30 #include <asm/machdep.h>
31 #include <asm/firmware.h>
32 #include <asm/page.h>
33 #include <asm/param.h>
34 #include <asm/system.h>
35 #include <asm/delay.h>
36 #include <asm/uaccess.h>
37 #include <asm/udbg.h>
38 #include <asm/syscalls.h>
39 #include <asm/smp.h>
40 #include <asm/atomic.h>
41
42 struct rtas_t rtas = {
43         .lock = __RAW_SPIN_LOCK_UNLOCKED
44 };
45 EXPORT_SYMBOL(rtas);
46
47 struct rtas_suspend_me_data {
48         atomic_t working; /* number of cpus accessing this struct */
49         atomic_t done;
50         int token; /* ibm,suspend-me */
51         int error;
52         struct completion *complete; /* wait on this until working == 0 */
53 };
54
55 DEFINE_SPINLOCK(rtas_data_buf_lock);
56 EXPORT_SYMBOL(rtas_data_buf_lock);
57
58 char rtas_data_buf[RTAS_DATA_BUF_SIZE] __cacheline_aligned;
59 EXPORT_SYMBOL(rtas_data_buf);
60
61 unsigned long rtas_rmo_buf;
62
63 /*
64  * If non-NULL, this gets called when the kernel terminates.
65  * This is done like this so rtas_flash can be a module.
66  */
67 void (*rtas_flash_term_hook)(int);
68 EXPORT_SYMBOL(rtas_flash_term_hook);
69
70 /* RTAS use home made raw locking instead of spin_lock_irqsave
71  * because those can be called from within really nasty contexts
72  * such as having the timebase stopped which would lockup with
73  * normal locks and spinlock debugging enabled
74  */
75 static unsigned long lock_rtas(void)
76 {
77         unsigned long flags;
78
79         local_irq_save(flags);
80         preempt_disable();
81         __raw_spin_lock_flags(&rtas.lock, flags);
82         return flags;
83 }
84
85 static void unlock_rtas(unsigned long flags)
86 {
87         __raw_spin_unlock(&rtas.lock);
88         local_irq_restore(flags);
89         preempt_enable();
90 }
91
92 /*
93  * call_rtas_display_status and call_rtas_display_status_delay
94  * are designed only for very early low-level debugging, which
95  * is why the token is hard-coded to 10.
96  */
97 static void call_rtas_display_status(char c)
98 {
99         struct rtas_args *args = &rtas.args;
100         unsigned long s;
101
102         if (!rtas.base)
103                 return;
104         s = lock_rtas();
105
106         args->token = 10;
107         args->nargs = 1;
108         args->nret  = 1;
109         args->rets  = (rtas_arg_t *)&(args->args[1]);
110         args->args[0] = (unsigned char)c;
111
112         enter_rtas(__pa(args));
113
114         unlock_rtas(s);
115 }
116
117 static void call_rtas_display_status_delay(char c)
118 {
119         static int pending_newline = 0;  /* did last write end with unprinted newline? */
120         static int width = 16;
121
122         if (c == '\n') {        
123                 while (width-- > 0)
124                         call_rtas_display_status(' ');
125                 width = 16;
126                 mdelay(500);
127                 pending_newline = 1;
128         } else {
129                 if (pending_newline) {
130                         call_rtas_display_status('\r');
131                         call_rtas_display_status('\n');
132                 } 
133                 pending_newline = 0;
134                 if (width--) {
135                         call_rtas_display_status(c);
136                         udelay(10000);
137                 }
138         }
139 }
140
141 void __init udbg_init_rtas_panel(void)
142 {
143         udbg_putc = call_rtas_display_status_delay;
144 }
145
146 #ifdef CONFIG_UDBG_RTAS_CONSOLE
147
148 /* If you think you're dying before early_init_dt_scan_rtas() does its
149  * work, you can hard code the token values for your firmware here and
150  * hardcode rtas.base/entry etc.
151  */
152 static unsigned int rtas_putchar_token = RTAS_UNKNOWN_SERVICE;
153 static unsigned int rtas_getchar_token = RTAS_UNKNOWN_SERVICE;
154
155 static void udbg_rtascon_putc(char c)
156 {
157         int tries;
158
159         if (!rtas.base)
160                 return;
161
162         /* Add CRs before LFs */
163         if (c == '\n')
164                 udbg_rtascon_putc('\r');
165
166         /* if there is more than one character to be displayed, wait a bit */
167         for (tries = 0; tries < 16; tries++) {
168                 if (rtas_call(rtas_putchar_token, 1, 1, NULL, c) == 0)
169                         break;
170                 udelay(1000);
171         }
172 }
173
174 static int udbg_rtascon_getc_poll(void)
175 {
176         int c;
177
178         if (!rtas.base)
179                 return -1;
180
181         if (rtas_call(rtas_getchar_token, 0, 2, &c))
182                 return -1;
183
184         return c;
185 }
186
187 static int udbg_rtascon_getc(void)
188 {
189         int c;
190
191         while ((c = udbg_rtascon_getc_poll()) == -1)
192                 ;
193
194         return c;
195 }
196
197
198 void __init udbg_init_rtas_console(void)
199 {
200         udbg_putc = udbg_rtascon_putc;
201         udbg_getc = udbg_rtascon_getc;
202         udbg_getc_poll = udbg_rtascon_getc_poll;
203 }
204 #endif /* CONFIG_UDBG_RTAS_CONSOLE */
205
206 void rtas_progress(char *s, unsigned short hex)
207 {
208         struct device_node *root;
209         int width;
210         const int *p;
211         char *os;
212         static int display_character, set_indicator;
213         static int display_width, display_lines, form_feed;
214         static const int *row_width;
215         static DEFINE_SPINLOCK(progress_lock);
216         static int current_line;
217         static int pending_newline = 0;  /* did last write end with unprinted newline? */
218
219         if (!rtas.base)
220                 return;
221
222         if (display_width == 0) {
223                 display_width = 0x10;
224                 if ((root = of_find_node_by_path("/rtas"))) {
225                         if ((p = of_get_property(root,
226                                         "ibm,display-line-length", NULL)))
227                                 display_width = *p;
228                         if ((p = of_get_property(root,
229                                         "ibm,form-feed", NULL)))
230                                 form_feed = *p;
231                         if ((p = of_get_property(root,
232                                         "ibm,display-number-of-lines", NULL)))
233                                 display_lines = *p;
234                         row_width = of_get_property(root,
235                                         "ibm,display-truncation-length", NULL);
236                         of_node_put(root);
237                 }
238                 display_character = rtas_token("display-character");
239                 set_indicator = rtas_token("set-indicator");
240         }
241
242         if (display_character == RTAS_UNKNOWN_SERVICE) {
243                 /* use hex display if available */
244                 if (set_indicator != RTAS_UNKNOWN_SERVICE)
245                         rtas_call(set_indicator, 3, 1, NULL, 6, 0, hex);
246                 return;
247         }
248
249         spin_lock(&progress_lock);
250
251         /*
252          * Last write ended with newline, but we didn't print it since
253          * it would just clear the bottom line of output. Print it now
254          * instead.
255          *
256          * If no newline is pending and form feed is supported, clear the
257          * display with a form feed; otherwise, print a CR to start output
258          * at the beginning of the line.
259          */
260         if (pending_newline) {
261                 rtas_call(display_character, 1, 1, NULL, '\r');
262                 rtas_call(display_character, 1, 1, NULL, '\n');
263                 pending_newline = 0;
264         } else {
265                 current_line = 0;
266                 if (form_feed)
267                         rtas_call(display_character, 1, 1, NULL,
268                                   (char)form_feed);
269                 else
270                         rtas_call(display_character, 1, 1, NULL, '\r');
271         }
272  
273         if (row_width)
274                 width = row_width[current_line];
275         else
276                 width = display_width;
277         os = s;
278         while (*os) {
279                 if (*os == '\n' || *os == '\r') {
280                         /* If newline is the last character, save it
281                          * until next call to avoid bumping up the
282                          * display output.
283                          */
284                         if (*os == '\n' && !os[1]) {
285                                 pending_newline = 1;
286                                 current_line++;
287                                 if (current_line > display_lines-1)
288                                         current_line = display_lines-1;
289                                 spin_unlock(&progress_lock);
290                                 return;
291                         }
292  
293                         /* RTAS wants CR-LF, not just LF */
294  
295                         if (*os == '\n') {
296                                 rtas_call(display_character, 1, 1, NULL, '\r');
297                                 rtas_call(display_character, 1, 1, NULL, '\n');
298                         } else {
299                                 /* CR might be used to re-draw a line, so we'll
300                                  * leave it alone and not add LF.
301                                  */
302                                 rtas_call(display_character, 1, 1, NULL, *os);
303                         }
304  
305                         if (row_width)
306                                 width = row_width[current_line];
307                         else
308                                 width = display_width;
309                 } else {
310                         width--;
311                         rtas_call(display_character, 1, 1, NULL, *os);
312                 }
313  
314                 os++;
315  
316                 /* if we overwrite the screen length */
317                 if (width <= 0)
318                         while ((*os != 0) && (*os != '\n') && (*os != '\r'))
319                                 os++;
320         }
321  
322         spin_unlock(&progress_lock);
323 }
324 EXPORT_SYMBOL(rtas_progress);           /* needed by rtas_flash module */
325
326 int rtas_token(const char *service)
327 {
328         const int *tokp;
329         if (rtas.dev == NULL)
330                 return RTAS_UNKNOWN_SERVICE;
331         tokp = of_get_property(rtas.dev, service, NULL);
332         return tokp ? *tokp : RTAS_UNKNOWN_SERVICE;
333 }
334 EXPORT_SYMBOL(rtas_token);
335
336 int rtas_service_present(const char *service)
337 {
338         return rtas_token(service) != RTAS_UNKNOWN_SERVICE;
339 }
340 EXPORT_SYMBOL(rtas_service_present);
341
342 #ifdef CONFIG_RTAS_ERROR_LOGGING
343 /*
344  * Return the firmware-specified size of the error log buffer
345  *  for all rtas calls that require an error buffer argument.
346  *  This includes 'check-exception' and 'rtas-last-error'.
347  */
348 int rtas_get_error_log_max(void)
349 {
350         static int rtas_error_log_max;
351         if (rtas_error_log_max)
352                 return rtas_error_log_max;
353
354         rtas_error_log_max = rtas_token ("rtas-error-log-max");
355         if ((rtas_error_log_max == RTAS_UNKNOWN_SERVICE) ||
356             (rtas_error_log_max > RTAS_ERROR_LOG_MAX)) {
357                 printk (KERN_WARNING "RTAS: bad log buffer size %d\n",
358                         rtas_error_log_max);
359                 rtas_error_log_max = RTAS_ERROR_LOG_MAX;
360         }
361         return rtas_error_log_max;
362 }
363 EXPORT_SYMBOL(rtas_get_error_log_max);
364
365
366 static char rtas_err_buf[RTAS_ERROR_LOG_MAX];
367 static int rtas_last_error_token;
368
369 /** Return a copy of the detailed error text associated with the
370  *  most recent failed call to rtas.  Because the error text
371  *  might go stale if there are any other intervening rtas calls,
372  *  this routine must be called atomically with whatever produced
373  *  the error (i.e. with rtas.lock still held from the previous call).
374  */
375 static char *__fetch_rtas_last_error(char *altbuf)
376 {
377         struct rtas_args err_args, save_args;
378         u32 bufsz;
379         char *buf = NULL;
380
381         if (rtas_last_error_token == -1)
382                 return NULL;
383
384         bufsz = rtas_get_error_log_max();
385
386         err_args.token = rtas_last_error_token;
387         err_args.nargs = 2;
388         err_args.nret = 1;
389         err_args.args[0] = (rtas_arg_t)__pa(rtas_err_buf);
390         err_args.args[1] = bufsz;
391         err_args.args[2] = 0;
392
393         save_args = rtas.args;
394         rtas.args = err_args;
395
396         enter_rtas(__pa(&rtas.args));
397
398         err_args = rtas.args;
399         rtas.args = save_args;
400
401         /* Log the error in the unlikely case that there was one. */
402         if (unlikely(err_args.args[2] == 0)) {
403                 if (altbuf) {
404                         buf = altbuf;
405                 } else {
406                         buf = rtas_err_buf;
407                         if (mem_init_done)
408                                 buf = kmalloc(RTAS_ERROR_LOG_MAX, GFP_ATOMIC);
409                 }
410                 if (buf)
411                         memcpy(buf, rtas_err_buf, RTAS_ERROR_LOG_MAX);
412         }
413
414         return buf;
415 }
416
417 #define get_errorlog_buffer()   kmalloc(RTAS_ERROR_LOG_MAX, GFP_KERNEL)
418
419 #else /* CONFIG_RTAS_ERROR_LOGGING */
420 #define __fetch_rtas_last_error(x)      NULL
421 #define get_errorlog_buffer()           NULL
422 #endif
423
424 int rtas_call(int token, int nargs, int nret, int *outputs, ...)
425 {
426         va_list list;
427         int i;
428         unsigned long s;
429         struct rtas_args *rtas_args;
430         char *buff_copy = NULL;
431         int ret;
432
433         if (!rtas.entry || token == RTAS_UNKNOWN_SERVICE)
434                 return -1;
435
436         s = lock_rtas();
437         rtas_args = &rtas.args;
438
439         rtas_args->token = token;
440         rtas_args->nargs = nargs;
441         rtas_args->nret  = nret;
442         rtas_args->rets  = (rtas_arg_t *)&(rtas_args->args[nargs]);
443         va_start(list, outputs);
444         for (i = 0; i < nargs; ++i)
445                 rtas_args->args[i] = va_arg(list, rtas_arg_t);
446         va_end(list);
447
448         for (i = 0; i < nret; ++i)
449                 rtas_args->rets[i] = 0;
450
451         enter_rtas(__pa(rtas_args));
452
453         /* A -1 return code indicates that the last command couldn't
454            be completed due to a hardware error. */
455         if (rtas_args->rets[0] == -1)
456                 buff_copy = __fetch_rtas_last_error(NULL);
457
458         if (nret > 1 && outputs != NULL)
459                 for (i = 0; i < nret-1; ++i)
460                         outputs[i] = rtas_args->rets[i+1];
461         ret = (nret > 0)? rtas_args->rets[0]: 0;
462
463         unlock_rtas(s);
464
465         if (buff_copy) {
466                 log_error(buff_copy, ERR_TYPE_RTAS_LOG, 0);
467                 if (mem_init_done)
468                         kfree(buff_copy);
469         }
470         return ret;
471 }
472 EXPORT_SYMBOL(rtas_call);
473
474 /* For RTAS_BUSY (-2), delay for 1 millisecond.  For an extended busy status
475  * code of 990n, perform the hinted delay of 10^n (last digit) milliseconds.
476  */
477 unsigned int rtas_busy_delay_time(int status)
478 {
479         int order;
480         unsigned int ms = 0;
481
482         if (status == RTAS_BUSY) {
483                 ms = 1;
484         } else if (status >= 9900 && status <= 9905) {
485                 order = status - 9900;
486                 for (ms = 1; order > 0; order--)
487                         ms *= 10;
488         }
489
490         return ms;
491 }
492 EXPORT_SYMBOL(rtas_busy_delay_time);
493
494 /* For an RTAS busy status code, perform the hinted delay. */
495 unsigned int rtas_busy_delay(int status)
496 {
497         unsigned int ms;
498
499         might_sleep();
500         ms = rtas_busy_delay_time(status);
501         if (ms)
502                 msleep(ms);
503
504         return ms;
505 }
506 EXPORT_SYMBOL(rtas_busy_delay);
507
508 static int rtas_error_rc(int rtas_rc)
509 {
510         int rc;
511
512         switch (rtas_rc) {
513                 case -1:                /* Hardware Error */
514                         rc = -EIO;
515                         break;
516                 case -3:                /* Bad indicator/domain/etc */
517                         rc = -EINVAL;
518                         break;
519                 case -9000:             /* Isolation error */
520                         rc = -EFAULT;
521                         break;
522                 case -9001:             /* Outstanding TCE/PTE */
523                         rc = -EEXIST;
524                         break;
525                 case -9002:             /* No usable slot */
526                         rc = -ENODEV;
527                         break;
528                 default:
529                         printk(KERN_ERR "%s: unexpected RTAS error %d\n",
530                                         __func__, rtas_rc);
531                         rc = -ERANGE;
532                         break;
533         }
534         return rc;
535 }
536
537 int rtas_get_power_level(int powerdomain, int *level)
538 {
539         int token = rtas_token("get-power-level");
540         int rc;
541
542         if (token == RTAS_UNKNOWN_SERVICE)
543                 return -ENOENT;
544
545         while ((rc = rtas_call(token, 1, 2, level, powerdomain)) == RTAS_BUSY)
546                 udelay(1);
547
548         if (rc < 0)
549                 return rtas_error_rc(rc);
550         return rc;
551 }
552 EXPORT_SYMBOL(rtas_get_power_level);
553
554 int rtas_set_power_level(int powerdomain, int level, int *setlevel)
555 {
556         int token = rtas_token("set-power-level");
557         int rc;
558
559         if (token == RTAS_UNKNOWN_SERVICE)
560                 return -ENOENT;
561
562         do {
563                 rc = rtas_call(token, 2, 2, setlevel, powerdomain, level);
564         } while (rtas_busy_delay(rc));
565
566         if (rc < 0)
567                 return rtas_error_rc(rc);
568         return rc;
569 }
570 EXPORT_SYMBOL(rtas_set_power_level);
571
572 int rtas_get_sensor(int sensor, int index, int *state)
573 {
574         int token = rtas_token("get-sensor-state");
575         int rc;
576
577         if (token == RTAS_UNKNOWN_SERVICE)
578                 return -ENOENT;
579
580         do {
581                 rc = rtas_call(token, 2, 2, state, sensor, index);
582         } while (rtas_busy_delay(rc));
583
584         if (rc < 0)
585                 return rtas_error_rc(rc);
586         return rc;
587 }
588 EXPORT_SYMBOL(rtas_get_sensor);
589
590 bool rtas_indicator_present(int token, int *maxindex)
591 {
592         int proplen, count, i;
593         const struct indicator_elem {
594                 u32 token;
595                 u32 maxindex;
596         } *indicators;
597
598         indicators = of_get_property(rtas.dev, "rtas-indicators", &proplen);
599         if (!indicators)
600                 return false;
601
602         count = proplen / sizeof(struct indicator_elem);
603
604         for (i = 0; i < count; i++) {
605                 if (indicators[i].token != token)
606                         continue;
607                 if (maxindex)
608                         *maxindex = indicators[i].maxindex;
609                 return true;
610         }
611
612         return false;
613 }
614 EXPORT_SYMBOL(rtas_indicator_present);
615
616 int rtas_set_indicator(int indicator, int index, int new_value)
617 {
618         int token = rtas_token("set-indicator");
619         int rc;
620
621         if (token == RTAS_UNKNOWN_SERVICE)
622                 return -ENOENT;
623
624         do {
625                 rc = rtas_call(token, 3, 1, NULL, indicator, index, new_value);
626         } while (rtas_busy_delay(rc));
627
628         if (rc < 0)
629                 return rtas_error_rc(rc);
630         return rc;
631 }
632 EXPORT_SYMBOL(rtas_set_indicator);
633
634 /*
635  * Ignoring RTAS extended delay
636  */
637 int rtas_set_indicator_fast(int indicator, int index, int new_value)
638 {
639         int rc;
640         int token = rtas_token("set-indicator");
641
642         if (token == RTAS_UNKNOWN_SERVICE)
643                 return -ENOENT;
644
645         rc = rtas_call(token, 3, 1, NULL, indicator, index, new_value);
646
647         WARN_ON(rc == -2 || (rc >= 9900 && rc <= 9905));
648
649         if (rc < 0)
650                 return rtas_error_rc(rc);
651
652         return rc;
653 }
654
655 void rtas_restart(char *cmd)
656 {
657         if (rtas_flash_term_hook)
658                 rtas_flash_term_hook(SYS_RESTART);
659         printk("RTAS system-reboot returned %d\n",
660                rtas_call(rtas_token("system-reboot"), 0, 1, NULL));
661         for (;;);
662 }
663
664 void rtas_power_off(void)
665 {
666         if (rtas_flash_term_hook)
667                 rtas_flash_term_hook(SYS_POWER_OFF);
668         /* allow power on only with power button press */
669         printk("RTAS power-off returned %d\n",
670                rtas_call(rtas_token("power-off"), 2, 1, NULL, -1, -1));
671         for (;;);
672 }
673
674 void rtas_halt(void)
675 {
676         if (rtas_flash_term_hook)
677                 rtas_flash_term_hook(SYS_HALT);
678         /* allow power on only with power button press */
679         printk("RTAS power-off returned %d\n",
680                rtas_call(rtas_token("power-off"), 2, 1, NULL, -1, -1));
681         for (;;);
682 }
683
684 /* Must be in the RMO region, so we place it here */
685 static char rtas_os_term_buf[2048];
686
687 void rtas_os_term(char *str)
688 {
689         int status;
690
691         if (panic_timeout)
692                 return;
693
694         if (RTAS_UNKNOWN_SERVICE == rtas_token("ibm,os-term"))
695                 return;
696
697         snprintf(rtas_os_term_buf, 2048, "OS panic: %s", str);
698
699         do {
700                 status = rtas_call(rtas_token("ibm,os-term"), 1, 1, NULL,
701                                    __pa(rtas_os_term_buf));
702         } while (rtas_busy_delay(status));
703
704         if (status != 0)
705                 printk(KERN_EMERG "ibm,os-term call failed %d\n",
706                                status);
707 }
708
709 static int ibm_suspend_me_token = RTAS_UNKNOWN_SERVICE;
710 #ifdef CONFIG_PPC_PSERIES
711 static void rtas_percpu_suspend_me(void *info)
712 {
713         long rc = H_SUCCESS;
714         unsigned long msr_save;
715         int cpu;
716         struct rtas_suspend_me_data *data =
717                 (struct rtas_suspend_me_data *)info;
718
719         atomic_inc(&data->working);
720
721         /* really need to ensure MSR.EE is off for H_JOIN */
722         msr_save = mfmsr();
723         mtmsr(msr_save & ~(MSR_EE));
724
725         while (rc == H_SUCCESS && !atomic_read(&data->done))
726                 rc = plpar_hcall_norets(H_JOIN);
727
728         mtmsr(msr_save);
729
730         if (rc == H_SUCCESS) {
731                 /* This cpu was prodded and the suspend is complete. */
732                 goto out;
733         } else if (rc == H_CONTINUE) {
734                 /* All other cpus are in H_JOIN, this cpu does
735                  * the suspend.
736                  */
737                 printk(KERN_DEBUG "calling ibm,suspend-me on cpu %i\n",
738                        smp_processor_id());
739                 data->error = rtas_call(data->token, 0, 1, NULL);
740
741                 if (data->error)
742                         printk(KERN_DEBUG "ibm,suspend-me returned %d\n",
743                                data->error);
744         } else {
745                 printk(KERN_ERR "H_JOIN on cpu %i failed with rc = %ld\n",
746                        smp_processor_id(), rc);
747                 data->error = rc;
748         }
749
750         atomic_set(&data->done, 1);
751
752         /* This cpu did the suspend or got an error; in either case,
753          * we need to prod all other other cpus out of join state.
754          * Extra prods are harmless.
755          */
756         for_each_online_cpu(cpu)
757                 plpar_hcall_norets(H_PROD, get_hard_smp_processor_id(cpu));
758 out:
759         if (atomic_dec_return(&data->working) == 0)
760                 complete(data->complete);
761 }
762
763 static int rtas_ibm_suspend_me(struct rtas_args *args)
764 {
765         long state;
766         long rc;
767         unsigned long retbuf[PLPAR_HCALL_BUFSIZE];
768         struct rtas_suspend_me_data data;
769         DECLARE_COMPLETION_ONSTACK(done);
770
771         if (!rtas_service_present("ibm,suspend-me"))
772                 return -ENOSYS;
773
774         /* Make sure the state is valid */
775         rc = plpar_hcall(H_VASI_STATE, retbuf,
776                          ((u64)args->args[0] << 32) | args->args[1]);
777
778         state = retbuf[0];
779
780         if (rc) {
781                 printk(KERN_ERR "rtas_ibm_suspend_me: vasi_state returned %ld\n",rc);
782                 return rc;
783         } else if (state == H_VASI_ENABLED) {
784                 args->args[args->nargs] = RTAS_NOT_SUSPENDABLE;
785                 return 0;
786         } else if (state != H_VASI_SUSPENDING) {
787                 printk(KERN_ERR "rtas_ibm_suspend_me: vasi_state returned state %ld\n",
788                        state);
789                 args->args[args->nargs] = -1;
790                 return 0;
791         }
792
793         atomic_set(&data.working, 0);
794         atomic_set(&data.done, 0);
795         data.token = rtas_token("ibm,suspend-me");
796         data.error = 0;
797         data.complete = &done;
798
799         /* Call function on all CPUs.  One of us will make the
800          * rtas call
801          */
802         if (on_each_cpu(rtas_percpu_suspend_me, &data, 0))
803                 data.error = -EINVAL;
804
805         wait_for_completion(&done);
806
807         if (data.error != 0)
808                 printk(KERN_ERR "Error doing global join\n");
809
810         return data.error;
811 }
812 #else /* CONFIG_PPC_PSERIES */
813 static int rtas_ibm_suspend_me(struct rtas_args *args)
814 {
815         return -ENOSYS;
816 }
817 #endif
818
819 asmlinkage int ppc_rtas(struct rtas_args __user *uargs)
820 {
821         struct rtas_args args;
822         unsigned long flags;
823         char *buff_copy, *errbuf = NULL;
824         int nargs;
825         int rc;
826
827         if (!capable(CAP_SYS_ADMIN))
828                 return -EPERM;
829
830         if (copy_from_user(&args, uargs, 3 * sizeof(u32)) != 0)
831                 return -EFAULT;
832
833         nargs = args.nargs;
834         if (nargs > ARRAY_SIZE(args.args)
835             || args.nret > ARRAY_SIZE(args.args)
836             || nargs + args.nret > ARRAY_SIZE(args.args))
837                 return -EINVAL;
838
839         /* Copy in args. */
840         if (copy_from_user(args.args, uargs->args,
841                            nargs * sizeof(rtas_arg_t)) != 0)
842                 return -EFAULT;
843
844         if (args.token == RTAS_UNKNOWN_SERVICE)
845                 return -EINVAL;
846
847         args.rets = &args.args[nargs];
848         memset(args.rets, 0, args.nret * sizeof(rtas_arg_t));
849
850         /* Need to handle ibm,suspend_me call specially */
851         if (args.token == ibm_suspend_me_token) {
852                 rc = rtas_ibm_suspend_me(&args);
853                 if (rc)
854                         return rc;
855                 goto copy_return;
856         }
857
858         buff_copy = get_errorlog_buffer();
859
860         flags = lock_rtas();
861
862         rtas.args = args;
863         enter_rtas(__pa(&rtas.args));
864         args = rtas.args;
865
866         /* A -1 return code indicates that the last command couldn't
867            be completed due to a hardware error. */
868         if (args.rets[0] == -1)
869                 errbuf = __fetch_rtas_last_error(buff_copy);
870
871         unlock_rtas(flags);
872
873         if (buff_copy) {
874                 if (errbuf)
875                         log_error(errbuf, ERR_TYPE_RTAS_LOG, 0);
876                 kfree(buff_copy);
877         }
878
879  copy_return:
880         /* Copy out args. */
881         if (copy_to_user(uargs->args + nargs,
882                          args.args + nargs,
883                          args.nret * sizeof(rtas_arg_t)) != 0)
884                 return -EFAULT;
885
886         return 0;
887 }
888
889 /*
890  * Call early during boot, before mem init or bootmem, to retrieve the RTAS
891  * informations from the device-tree and allocate the RMO buffer for userland
892  * accesses.
893  */
894 void __init rtas_initialize(void)
895 {
896         unsigned long rtas_region = RTAS_INSTANTIATE_MAX;
897
898         /* Get RTAS dev node and fill up our "rtas" structure with infos
899          * about it.
900          */
901         rtas.dev = of_find_node_by_name(NULL, "rtas");
902         if (rtas.dev) {
903                 const u32 *basep, *entryp, *sizep;
904
905                 basep = of_get_property(rtas.dev, "linux,rtas-base", NULL);
906                 sizep = of_get_property(rtas.dev, "rtas-size", NULL);
907                 if (basep != NULL && sizep != NULL) {
908                         rtas.base = *basep;
909                         rtas.size = *sizep;
910                         entryp = of_get_property(rtas.dev,
911                                         "linux,rtas-entry", NULL);
912                         if (entryp == NULL) /* Ugh */
913                                 rtas.entry = rtas.base;
914                         else
915                                 rtas.entry = *entryp;
916                 } else
917                         rtas.dev = NULL;
918         }
919         if (!rtas.dev)
920                 return;
921
922         /* If RTAS was found, allocate the RMO buffer for it and look for
923          * the stop-self token if any
924          */
925 #ifdef CONFIG_PPC64
926         if (machine_is(pseries) && firmware_has_feature(FW_FEATURE_LPAR)) {
927                 rtas_region = min(lmb.rmo_size, RTAS_INSTANTIATE_MAX);
928                 ibm_suspend_me_token = rtas_token("ibm,suspend-me");
929         }
930 #endif
931         rtas_rmo_buf = lmb_alloc_base(RTAS_RMOBUF_MAX, PAGE_SIZE, rtas_region);
932
933 #ifdef CONFIG_RTAS_ERROR_LOGGING
934         rtas_last_error_token = rtas_token("rtas-last-error");
935 #endif
936 }
937
938 int __init early_init_dt_scan_rtas(unsigned long node,
939                 const char *uname, int depth, void *data)
940 {
941         u32 *basep, *entryp, *sizep;
942
943         if (depth != 1 || strcmp(uname, "rtas") != 0)
944                 return 0;
945
946         basep  = of_get_flat_dt_prop(node, "linux,rtas-base", NULL);
947         entryp = of_get_flat_dt_prop(node, "linux,rtas-entry", NULL);
948         sizep  = of_get_flat_dt_prop(node, "rtas-size", NULL);
949
950         if (basep && entryp && sizep) {
951                 rtas.base = *basep;
952                 rtas.entry = *entryp;
953                 rtas.size = *sizep;
954         }
955
956 #ifdef CONFIG_UDBG_RTAS_CONSOLE
957         basep = of_get_flat_dt_prop(node, "put-term-char", NULL);
958         if (basep)
959                 rtas_putchar_token = *basep;
960
961         basep = of_get_flat_dt_prop(node, "get-term-char", NULL);
962         if (basep)
963                 rtas_getchar_token = *basep;
964
965         if (rtas_putchar_token != RTAS_UNKNOWN_SERVICE &&
966             rtas_getchar_token != RTAS_UNKNOWN_SERVICE)
967                 udbg_init_rtas_console();
968
969 #endif
970
971         /* break now */
972         return 1;
973 }