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