efi_loader: LoadImage: always allocate new pages
[platform/kernel/u-boot.git] / lib / efi_loader / efi_boottime.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  *  EFI application boot time services
4  *
5  *  Copyright (c) 2016 Alexander Graf
6  */
7
8 #include <common.h>
9 #include <div64.h>
10 #include <efi_loader.h>
11 #include <environment.h>
12 #include <malloc.h>
13 #include <linux/libfdt_env.h>
14 #include <u-boot/crc.h>
15 #include <bootm.h>
16 #include <watchdog.h>
17
18 DECLARE_GLOBAL_DATA_PTR;
19
20 /* Task priority level */
21 static efi_uintn_t efi_tpl = TPL_APPLICATION;
22
23 /* This list contains all the EFI objects our payload has access to */
24 LIST_HEAD(efi_obj_list);
25
26 /* List of all events */
27 LIST_HEAD(efi_events);
28
29 /*
30  * If we're running on nasty systems (32bit ARM booting into non-EFI Linux)
31  * we need to do trickery with caches. Since we don't want to break the EFI
32  * aware boot path, only apply hacks when loading exiting directly (breaking
33  * direct Linux EFI booting along the way - oh well).
34  */
35 static bool efi_is_direct_boot = true;
36
37 #ifdef CONFIG_ARM
38 /*
39  * The "gd" pointer lives in a register on ARM and AArch64 that we declare
40  * fixed when compiling U-Boot. However, the payload does not know about that
41  * restriction so we need to manually swap its and our view of that register on
42  * EFI callback entry/exit.
43  */
44 static volatile void *efi_gd, *app_gd;
45 #endif
46
47 static int entry_count;
48 static int nesting_level;
49 /* GUID of the device tree table */
50 const efi_guid_t efi_guid_fdt = EFI_FDT_GUID;
51 /* GUID of the EFI_DRIVER_BINDING_PROTOCOL */
52 const efi_guid_t efi_guid_driver_binding_protocol =
53                         EFI_DRIVER_BINDING_PROTOCOL_GUID;
54
55 /* event group ExitBootServices() invoked */
56 const efi_guid_t efi_guid_event_group_exit_boot_services =
57                         EFI_EVENT_GROUP_EXIT_BOOT_SERVICES;
58 /* event group SetVirtualAddressMap() invoked */
59 const efi_guid_t efi_guid_event_group_virtual_address_change =
60                         EFI_EVENT_GROUP_VIRTUAL_ADDRESS_CHANGE;
61 /* event group memory map changed */
62 const efi_guid_t efi_guid_event_group_memory_map_change =
63                         EFI_EVENT_GROUP_MEMORY_MAP_CHANGE;
64 /* event group boot manager about to boot */
65 const efi_guid_t efi_guid_event_group_ready_to_boot =
66                         EFI_EVENT_GROUP_READY_TO_BOOT;
67 /* event group ResetSystem() invoked (before ExitBootServices) */
68 const efi_guid_t efi_guid_event_group_reset_system =
69                         EFI_EVENT_GROUP_RESET_SYSTEM;
70
71 static efi_status_t EFIAPI efi_disconnect_controller(
72                                         efi_handle_t controller_handle,
73                                         efi_handle_t driver_image_handle,
74                                         efi_handle_t child_handle);
75
76 /* Called on every callback entry */
77 int __efi_entry_check(void)
78 {
79         int ret = entry_count++ == 0;
80 #ifdef CONFIG_ARM
81         assert(efi_gd);
82         app_gd = gd;
83         gd = efi_gd;
84 #endif
85         return ret;
86 }
87
88 /* Called on every callback exit */
89 int __efi_exit_check(void)
90 {
91         int ret = --entry_count == 0;
92 #ifdef CONFIG_ARM
93         gd = app_gd;
94 #endif
95         return ret;
96 }
97
98 /* Called from do_bootefi_exec() */
99 void efi_save_gd(void)
100 {
101 #ifdef CONFIG_ARM
102         efi_gd = gd;
103 #endif
104 }
105
106 /*
107  * Special case handler for error/abort that just forces things back to u-boot
108  * world so we can dump out an abort message, without any care about returning
109  * back to UEFI world.
110  */
111 void efi_restore_gd(void)
112 {
113 #ifdef CONFIG_ARM
114         /* Only restore if we're already in EFI context */
115         if (!efi_gd)
116                 return;
117         gd = efi_gd;
118 #endif
119 }
120
121 /**
122  * indent_string() - returns a string for indenting with two spaces per level
123  * @level: indent level
124  *
125  * A maximum of ten indent levels is supported. Higher indent levels will be
126  * truncated.
127  *
128  * Return: A string for indenting with two spaces per level is
129  *         returned.
130  */
131 static const char *indent_string(int level)
132 {
133         const char *indent = "                    ";
134         const int max = strlen(indent);
135
136         level = min(max, level * 2);
137         return &indent[max - level];
138 }
139
140 const char *__efi_nesting(void)
141 {
142         return indent_string(nesting_level);
143 }
144
145 const char *__efi_nesting_inc(void)
146 {
147         return indent_string(nesting_level++);
148 }
149
150 const char *__efi_nesting_dec(void)
151 {
152         return indent_string(--nesting_level);
153 }
154
155 /**
156  * efi_queue_event() - queue an EFI event
157  * @event:     event to signal
158  * @check_tpl: check the TPL level
159  *
160  * This function queues the notification function of the event for future
161  * execution.
162  *
163  * The notification function is called if the task priority level of the event
164  * is higher than the current task priority level.
165  *
166  * For the SignalEvent service see efi_signal_event_ext.
167  *
168  */
169 static void efi_queue_event(struct efi_event *event, bool check_tpl)
170 {
171         if (event->notify_function) {
172                 event->is_queued = true;
173                 /* Check TPL */
174                 if (check_tpl && efi_tpl >= event->notify_tpl)
175                         return;
176                 EFI_CALL_VOID(event->notify_function(event,
177                                                      event->notify_context));
178         }
179         event->is_queued = false;
180 }
181
182 /**
183  * is_valid_tpl() - check if the task priority level is valid
184  *
185  * @tpl:                TPL level to check
186  * Return:              status code
187  */
188 efi_status_t is_valid_tpl(efi_uintn_t tpl)
189 {
190         switch (tpl) {
191         case TPL_APPLICATION:
192         case TPL_CALLBACK:
193         case TPL_NOTIFY:
194         case TPL_HIGH_LEVEL:
195                 return EFI_SUCCESS;
196         default:
197                 return EFI_INVALID_PARAMETER;
198         }
199 }
200
201 /**
202  * efi_signal_event() - signal an EFI event
203  * @event:     event to signal
204  * @check_tpl: check the TPL level
205  *
206  * This function signals an event. If the event belongs to an event group all
207  * events of the group are signaled. If they are of type EVT_NOTIFY_SIGNAL
208  * their notification function is queued.
209  *
210  * For the SignalEvent service see efi_signal_event_ext.
211  */
212 void efi_signal_event(struct efi_event *event, bool check_tpl)
213 {
214         if (event->group) {
215                 struct efi_event *evt;
216
217                 /*
218                  * The signaled state has to set before executing any
219                  * notification function
220                  */
221                 list_for_each_entry(evt, &efi_events, link) {
222                         if (!evt->group || guidcmp(evt->group, event->group))
223                                 continue;
224                         if (evt->is_signaled)
225                                 continue;
226                         evt->is_signaled = true;
227                         if (evt->type & EVT_NOTIFY_SIGNAL &&
228                             evt->notify_function)
229                                 evt->is_queued = true;
230                 }
231                 list_for_each_entry(evt, &efi_events, link) {
232                         if (!evt->group || guidcmp(evt->group, event->group))
233                                 continue;
234                         if (evt->is_queued)
235                                 efi_queue_event(evt, check_tpl);
236                 }
237         } else if (!event->is_signaled) {
238                 event->is_signaled = true;
239                 if (event->type & EVT_NOTIFY_SIGNAL)
240                         efi_queue_event(event, check_tpl);
241         }
242 }
243
244 /**
245  * efi_raise_tpl() - raise the task priority level
246  * @new_tpl: new value of the task priority level
247  *
248  * This function implements the RaiseTpl service.
249  *
250  * See the Unified Extensible Firmware Interface (UEFI) specification for
251  * details.
252  *
253  * Return: old value of the task priority level
254  */
255 static unsigned long EFIAPI efi_raise_tpl(efi_uintn_t new_tpl)
256 {
257         efi_uintn_t old_tpl = efi_tpl;
258
259         EFI_ENTRY("0x%zx", new_tpl);
260
261         if (new_tpl < efi_tpl)
262                 debug("WARNING: new_tpl < current_tpl in %s\n", __func__);
263         efi_tpl = new_tpl;
264         if (efi_tpl > TPL_HIGH_LEVEL)
265                 efi_tpl = TPL_HIGH_LEVEL;
266
267         EFI_EXIT(EFI_SUCCESS);
268         return old_tpl;
269 }
270
271 /**
272  * efi_restore_tpl() - lower the task priority level
273  * @old_tpl: value of the task priority level to be restored
274  *
275  * This function implements the RestoreTpl service.
276  *
277  * See the Unified Extensible Firmware Interface (UEFI) specification for
278  * details.
279  */
280 static void EFIAPI efi_restore_tpl(efi_uintn_t old_tpl)
281 {
282         EFI_ENTRY("0x%zx", old_tpl);
283
284         if (old_tpl > efi_tpl)
285                 debug("WARNING: old_tpl > current_tpl in %s\n", __func__);
286         efi_tpl = old_tpl;
287         if (efi_tpl > TPL_HIGH_LEVEL)
288                 efi_tpl = TPL_HIGH_LEVEL;
289
290         /*
291          * Lowering the TPL may have made queued events eligible for execution.
292          */
293         efi_timer_check();
294
295         EFI_EXIT(EFI_SUCCESS);
296 }
297
298 /**
299  * efi_allocate_pages_ext() - allocate memory pages
300  * @type:        type of allocation to be performed
301  * @memory_type: usage type of the allocated memory
302  * @pages:       number of pages to be allocated
303  * @memory:      allocated memory
304  *
305  * This function implements the AllocatePages service.
306  *
307  * See the Unified Extensible Firmware Interface (UEFI) specification for
308  * details.
309  *
310  * Return: status code
311  */
312 static efi_status_t EFIAPI efi_allocate_pages_ext(int type, int memory_type,
313                                                   efi_uintn_t pages,
314                                                   uint64_t *memory)
315 {
316         efi_status_t r;
317
318         EFI_ENTRY("%d, %d, 0x%zx, %p", type, memory_type, pages, memory);
319         r = efi_allocate_pages(type, memory_type, pages, memory);
320         return EFI_EXIT(r);
321 }
322
323 /**
324  * efi_free_pages_ext() - Free memory pages.
325  * @memory: start of the memory area to be freed
326  * @pages:  number of pages to be freed
327  *
328  * This function implements the FreePages service.
329  *
330  * See the Unified Extensible Firmware Interface (UEFI) specification for
331  * details.
332  *
333  * Return: status code
334  */
335 static efi_status_t EFIAPI efi_free_pages_ext(uint64_t memory,
336                                               efi_uintn_t pages)
337 {
338         efi_status_t r;
339
340         EFI_ENTRY("%llx, 0x%zx", memory, pages);
341         r = efi_free_pages(memory, pages);
342         return EFI_EXIT(r);
343 }
344
345 /**
346  * efi_get_memory_map_ext() - get map describing memory usage
347  * @memory_map_size:    on entry the size, in bytes, of the memory map buffer,
348  *                      on exit the size of the copied memory map
349  * @memory_map:         buffer to which the memory map is written
350  * @map_key:            key for the memory map
351  * @descriptor_size:    size of an individual memory descriptor
352  * @descriptor_version: version number of the memory descriptor structure
353  *
354  * This function implements the GetMemoryMap service.
355  *
356  * See the Unified Extensible Firmware Interface (UEFI) specification for
357  * details.
358  *
359  * Return: status code
360  */
361 static efi_status_t EFIAPI efi_get_memory_map_ext(
362                                         efi_uintn_t *memory_map_size,
363                                         struct efi_mem_desc *memory_map,
364                                         efi_uintn_t *map_key,
365                                         efi_uintn_t *descriptor_size,
366                                         uint32_t *descriptor_version)
367 {
368         efi_status_t r;
369
370         EFI_ENTRY("%p, %p, %p, %p, %p", memory_map_size, memory_map,
371                   map_key, descriptor_size, descriptor_version);
372         r = efi_get_memory_map(memory_map_size, memory_map, map_key,
373                                descriptor_size, descriptor_version);
374         return EFI_EXIT(r);
375 }
376
377 /**
378  * efi_allocate_pool_ext() - allocate memory from pool
379  * @pool_type: type of the pool from which memory is to be allocated
380  * @size:      number of bytes to be allocated
381  * @buffer:    allocated memory
382  *
383  * This function implements the AllocatePool service.
384  *
385  * See the Unified Extensible Firmware Interface (UEFI) specification for
386  * details.
387  *
388  * Return: status code
389  */
390 static efi_status_t EFIAPI efi_allocate_pool_ext(int pool_type,
391                                                  efi_uintn_t size,
392                                                  void **buffer)
393 {
394         efi_status_t r;
395
396         EFI_ENTRY("%d, %zd, %p", pool_type, size, buffer);
397         r = efi_allocate_pool(pool_type, size, buffer);
398         return EFI_EXIT(r);
399 }
400
401 /**
402  * efi_free_pool_ext() - free memory from pool
403  * @buffer: start of memory to be freed
404  *
405  * This function implements the FreePool service.
406  *
407  * See the Unified Extensible Firmware Interface (UEFI) specification for
408  * details.
409  *
410  * Return: status code
411  */
412 static efi_status_t EFIAPI efi_free_pool_ext(void *buffer)
413 {
414         efi_status_t r;
415
416         EFI_ENTRY("%p", buffer);
417         r = efi_free_pool(buffer);
418         return EFI_EXIT(r);
419 }
420
421 /**
422  * efi_add_handle() - add a new object to the object list
423  * @obj: object to be added
424  *
425  * The protocols list is initialized. The object handle is set.
426  */
427 void efi_add_handle(efi_handle_t handle)
428 {
429         if (!handle)
430                 return;
431         INIT_LIST_HEAD(&handle->protocols);
432         list_add_tail(&handle->link, &efi_obj_list);
433 }
434
435 /**
436  * efi_create_handle() - create handle
437  * @handle: new handle
438  *
439  * Return: status code
440  */
441 efi_status_t efi_create_handle(efi_handle_t *handle)
442 {
443         struct efi_object *obj;
444
445         obj = calloc(1, sizeof(struct efi_object));
446         if (!obj)
447                 return EFI_OUT_OF_RESOURCES;
448
449         efi_add_handle(obj);
450         *handle = obj;
451
452         return EFI_SUCCESS;
453 }
454
455 /**
456  * efi_search_protocol() - find a protocol on a handle.
457  * @handle:        handle
458  * @protocol_guid: GUID of the protocol
459  * @handler:       reference to the protocol
460  *
461  * Return: status code
462  */
463 efi_status_t efi_search_protocol(const efi_handle_t handle,
464                                  const efi_guid_t *protocol_guid,
465                                  struct efi_handler **handler)
466 {
467         struct efi_object *efiobj;
468         struct list_head *lhandle;
469
470         if (!handle || !protocol_guid)
471                 return EFI_INVALID_PARAMETER;
472         efiobj = efi_search_obj(handle);
473         if (!efiobj)
474                 return EFI_INVALID_PARAMETER;
475         list_for_each(lhandle, &efiobj->protocols) {
476                 struct efi_handler *protocol;
477
478                 protocol = list_entry(lhandle, struct efi_handler, link);
479                 if (!guidcmp(protocol->guid, protocol_guid)) {
480                         if (handler)
481                                 *handler = protocol;
482                         return EFI_SUCCESS;
483                 }
484         }
485         return EFI_NOT_FOUND;
486 }
487
488 /**
489  * efi_remove_protocol() - delete protocol from a handle
490  * @handle:             handle from which the protocol shall be deleted
491  * @protocol:           GUID of the protocol to be deleted
492  * @protocol_interface: interface of the protocol implementation
493  *
494  * Return: status code
495  */
496 efi_status_t efi_remove_protocol(const efi_handle_t handle,
497                                  const efi_guid_t *protocol,
498                                  void *protocol_interface)
499 {
500         struct efi_handler *handler;
501         efi_status_t ret;
502
503         ret = efi_search_protocol(handle, protocol, &handler);
504         if (ret != EFI_SUCCESS)
505                 return ret;
506         if (guidcmp(handler->guid, protocol))
507                 return EFI_INVALID_PARAMETER;
508         if (handler->protocol_interface != protocol_interface)
509                 return EFI_INVALID_PARAMETER;
510         list_del(&handler->link);
511         free(handler);
512         return EFI_SUCCESS;
513 }
514
515 /**
516  * efi_remove_all_protocols() - delete all protocols from a handle
517  * @handle: handle from which the protocols shall be deleted
518  *
519  * Return: status code
520  */
521 efi_status_t efi_remove_all_protocols(const efi_handle_t handle)
522 {
523         struct efi_object *efiobj;
524         struct efi_handler *protocol;
525         struct efi_handler *pos;
526
527         efiobj = efi_search_obj(handle);
528         if (!efiobj)
529                 return EFI_INVALID_PARAMETER;
530         list_for_each_entry_safe(protocol, pos, &efiobj->protocols, link) {
531                 efi_status_t ret;
532
533                 ret = efi_remove_protocol(handle, protocol->guid,
534                                           protocol->protocol_interface);
535                 if (ret != EFI_SUCCESS)
536                         return ret;
537         }
538         return EFI_SUCCESS;
539 }
540
541 /**
542  * efi_delete_handle() - delete handle
543  *
544  * @obj: handle to delete
545  */
546 void efi_delete_handle(efi_handle_t handle)
547 {
548         if (!handle)
549                 return;
550         efi_remove_all_protocols(handle);
551         list_del(&handle->link);
552         free(handle);
553 }
554
555 /**
556  * efi_is_event() - check if a pointer is a valid event
557  * @event: pointer to check
558  *
559  * Return: status code
560  */
561 static efi_status_t efi_is_event(const struct efi_event *event)
562 {
563         const struct efi_event *evt;
564
565         if (!event)
566                 return EFI_INVALID_PARAMETER;
567         list_for_each_entry(evt, &efi_events, link) {
568                 if (evt == event)
569                         return EFI_SUCCESS;
570         }
571         return EFI_INVALID_PARAMETER;
572 }
573
574 /**
575  * efi_create_event() - create an event
576  * @type:            type of the event to create
577  * @notify_tpl:      task priority level of the event
578  * @notify_function: notification function of the event
579  * @notify_context:  pointer passed to the notification function
580  * @group:           event group
581  * @event:           created event
582  *
583  * This function is used inside U-Boot code to create an event.
584  *
585  * For the API function implementing the CreateEvent service see
586  * efi_create_event_ext.
587  *
588  * Return: status code
589  */
590 efi_status_t efi_create_event(uint32_t type, efi_uintn_t notify_tpl,
591                               void (EFIAPI *notify_function) (
592                                         struct efi_event *event,
593                                         void *context),
594                               void *notify_context, efi_guid_t *group,
595                               struct efi_event **event)
596 {
597         struct efi_event *evt;
598
599         if (event == NULL)
600                 return EFI_INVALID_PARAMETER;
601
602         switch (type) {
603         case 0:
604         case EVT_TIMER:
605         case EVT_NOTIFY_SIGNAL:
606         case EVT_TIMER | EVT_NOTIFY_SIGNAL:
607         case EVT_NOTIFY_WAIT:
608         case EVT_TIMER | EVT_NOTIFY_WAIT:
609         case EVT_SIGNAL_EXIT_BOOT_SERVICES:
610         case EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE:
611                 break;
612         default:
613                 return EFI_INVALID_PARAMETER;
614         }
615
616         if ((type & (EVT_NOTIFY_WAIT | EVT_NOTIFY_SIGNAL)) &&
617             (is_valid_tpl(notify_tpl) != EFI_SUCCESS))
618                 return EFI_INVALID_PARAMETER;
619
620         evt = calloc(1, sizeof(struct efi_event));
621         if (!evt)
622                 return EFI_OUT_OF_RESOURCES;
623         evt->type = type;
624         evt->notify_tpl = notify_tpl;
625         evt->notify_function = notify_function;
626         evt->notify_context = notify_context;
627         evt->group = group;
628         /* Disable timers on boot up */
629         evt->trigger_next = -1ULL;
630         evt->is_queued = false;
631         evt->is_signaled = false;
632         list_add_tail(&evt->link, &efi_events);
633         *event = evt;
634         return EFI_SUCCESS;
635 }
636
637 /*
638  * efi_create_event_ex() - create an event in a group
639  * @type:            type of the event to create
640  * @notify_tpl:      task priority level of the event
641  * @notify_function: notification function of the event
642  * @notify_context:  pointer passed to the notification function
643  * @event:           created event
644  * @event_group:     event group
645  *
646  * This function implements the CreateEventEx service.
647  *
648  * See the Unified Extensible Firmware Interface (UEFI) specification for
649  * details.
650  *
651  * Return: status code
652  */
653 efi_status_t EFIAPI efi_create_event_ex(uint32_t type, efi_uintn_t notify_tpl,
654                                         void (EFIAPI *notify_function) (
655                                                         struct efi_event *event,
656                                                         void *context),
657                                         void *notify_context,
658                                         efi_guid_t *event_group,
659                                         struct efi_event **event)
660 {
661         EFI_ENTRY("%d, 0x%zx, %p, %p, %pUl", type, notify_tpl, notify_function,
662                   notify_context, event_group);
663         return EFI_EXIT(efi_create_event(type, notify_tpl, notify_function,
664                                          notify_context, event_group, event));
665 }
666
667 /**
668  * efi_create_event_ext() - create an event
669  * @type:            type of the event to create
670  * @notify_tpl:      task priority level of the event
671  * @notify_function: notification function of the event
672  * @notify_context:  pointer passed to the notification function
673  * @event:           created event
674  *
675  * This function implements the CreateEvent service.
676  *
677  * See the Unified Extensible Firmware Interface (UEFI) specification for
678  * details.
679  *
680  * Return: status code
681  */
682 static efi_status_t EFIAPI efi_create_event_ext(
683                         uint32_t type, efi_uintn_t notify_tpl,
684                         void (EFIAPI *notify_function) (
685                                         struct efi_event *event,
686                                         void *context),
687                         void *notify_context, struct efi_event **event)
688 {
689         EFI_ENTRY("%d, 0x%zx, %p, %p", type, notify_tpl, notify_function,
690                   notify_context);
691         return EFI_EXIT(efi_create_event(type, notify_tpl, notify_function,
692                                          notify_context, NULL, event));
693 }
694
695 /**
696  * efi_timer_check() - check if a timer event has occurred
697  *
698  * Check if a timer event has occurred or a queued notification function should
699  * be called.
700  *
701  * Our timers have to work without interrupts, so we check whenever keyboard
702  * input or disk accesses happen if enough time elapsed for them to fire.
703  */
704 void efi_timer_check(void)
705 {
706         struct efi_event *evt;
707         u64 now = timer_get_us();
708
709         list_for_each_entry(evt, &efi_events, link) {
710                 if (evt->is_queued)
711                         efi_queue_event(evt, true);
712                 if (!(evt->type & EVT_TIMER) || now < evt->trigger_next)
713                         continue;
714                 switch (evt->trigger_type) {
715                 case EFI_TIMER_RELATIVE:
716                         evt->trigger_type = EFI_TIMER_STOP;
717                         break;
718                 case EFI_TIMER_PERIODIC:
719                         evt->trigger_next += evt->trigger_time;
720                         break;
721                 default:
722                         continue;
723                 }
724                 evt->is_signaled = false;
725                 efi_signal_event(evt, true);
726         }
727         WATCHDOG_RESET();
728 }
729
730 /**
731  * efi_set_timer() - set the trigger time for a timer event or stop the event
732  * @event:        event for which the timer is set
733  * @type:         type of the timer
734  * @trigger_time: trigger period in multiples of 100 ns
735  *
736  * This is the function for internal usage in U-Boot. For the API function
737  * implementing the SetTimer service see efi_set_timer_ext.
738  *
739  * Return: status code
740  */
741 efi_status_t efi_set_timer(struct efi_event *event, enum efi_timer_delay type,
742                            uint64_t trigger_time)
743 {
744         /* Check that the event is valid */
745         if (efi_is_event(event) != EFI_SUCCESS || !(event->type & EVT_TIMER))
746                 return EFI_INVALID_PARAMETER;
747
748         /*
749          * The parameter defines a multiple of 100 ns.
750          * We use multiples of 1000 ns. So divide by 10.
751          */
752         do_div(trigger_time, 10);
753
754         switch (type) {
755         case EFI_TIMER_STOP:
756                 event->trigger_next = -1ULL;
757                 break;
758         case EFI_TIMER_PERIODIC:
759         case EFI_TIMER_RELATIVE:
760                 event->trigger_next = timer_get_us() + trigger_time;
761                 break;
762         default:
763                 return EFI_INVALID_PARAMETER;
764         }
765         event->trigger_type = type;
766         event->trigger_time = trigger_time;
767         event->is_signaled = false;
768         return EFI_SUCCESS;
769 }
770
771 /**
772  * efi_set_timer_ext() - Set the trigger time for a timer event or stop the
773  *                       event
774  * @event:        event for which the timer is set
775  * @type:         type of the timer
776  * @trigger_time: trigger period in multiples of 100 ns
777  *
778  * This function implements the SetTimer service.
779  *
780  * See the Unified Extensible Firmware Interface (UEFI) specification for
781  * details.
782  *
783  *
784  * Return: status code
785  */
786 static efi_status_t EFIAPI efi_set_timer_ext(struct efi_event *event,
787                                              enum efi_timer_delay type,
788                                              uint64_t trigger_time)
789 {
790         EFI_ENTRY("%p, %d, %llx", event, type, trigger_time);
791         return EFI_EXIT(efi_set_timer(event, type, trigger_time));
792 }
793
794 /**
795  * efi_wait_for_event() - wait for events to be signaled
796  * @num_events: number of events to be waited for
797  * @event:      events to be waited for
798  * @index:      index of the event that was signaled
799  *
800  * This function implements the WaitForEvent service.
801  *
802  * See the Unified Extensible Firmware Interface (UEFI) specification for
803  * details.
804  *
805  * Return: status code
806  */
807 static efi_status_t EFIAPI efi_wait_for_event(efi_uintn_t num_events,
808                                               struct efi_event **event,
809                                               efi_uintn_t *index)
810 {
811         int i;
812
813         EFI_ENTRY("%zd, %p, %p", num_events, event, index);
814
815         /* Check parameters */
816         if (!num_events || !event)
817                 return EFI_EXIT(EFI_INVALID_PARAMETER);
818         /* Check TPL */
819         if (efi_tpl != TPL_APPLICATION)
820                 return EFI_EXIT(EFI_UNSUPPORTED);
821         for (i = 0; i < num_events; ++i) {
822                 if (efi_is_event(event[i]) != EFI_SUCCESS)
823                         return EFI_EXIT(EFI_INVALID_PARAMETER);
824                 if (!event[i]->type || event[i]->type & EVT_NOTIFY_SIGNAL)
825                         return EFI_EXIT(EFI_INVALID_PARAMETER);
826                 if (!event[i]->is_signaled)
827                         efi_queue_event(event[i], true);
828         }
829
830         /* Wait for signal */
831         for (;;) {
832                 for (i = 0; i < num_events; ++i) {
833                         if (event[i]->is_signaled)
834                                 goto out;
835                 }
836                 /* Allow events to occur. */
837                 efi_timer_check();
838         }
839
840 out:
841         /*
842          * Reset the signal which is passed to the caller to allow periodic
843          * events to occur.
844          */
845         event[i]->is_signaled = false;
846         if (index)
847                 *index = i;
848
849         return EFI_EXIT(EFI_SUCCESS);
850 }
851
852 /**
853  * efi_signal_event_ext() - signal an EFI event
854  * @event: event to signal
855  *
856  * This function implements the SignalEvent service.
857  *
858  * See the Unified Extensible Firmware Interface (UEFI) specification for
859  * details.
860  *
861  * This functions sets the signaled state of the event and queues the
862  * notification function for execution.
863  *
864  * Return: status code
865  */
866 static efi_status_t EFIAPI efi_signal_event_ext(struct efi_event *event)
867 {
868         EFI_ENTRY("%p", event);
869         if (efi_is_event(event) != EFI_SUCCESS)
870                 return EFI_EXIT(EFI_INVALID_PARAMETER);
871         efi_signal_event(event, true);
872         return EFI_EXIT(EFI_SUCCESS);
873 }
874
875 /**
876  * efi_close_event() - close an EFI event
877  * @event: event to close
878  *
879  * This function implements the CloseEvent service.
880  *
881  * See the Unified Extensible Firmware Interface (UEFI) specification for
882  * details.
883  *
884  * Return: status code
885  */
886 static efi_status_t EFIAPI efi_close_event(struct efi_event *event)
887 {
888         EFI_ENTRY("%p", event);
889         if (efi_is_event(event) != EFI_SUCCESS)
890                 return EFI_EXIT(EFI_INVALID_PARAMETER);
891         list_del(&event->link);
892         free(event);
893         return EFI_EXIT(EFI_SUCCESS);
894 }
895
896 /**
897  * efi_check_event() - check if an event is signaled
898  * @event: event to check
899  *
900  * This function implements the CheckEvent service.
901  *
902  * See the Unified Extensible Firmware Interface (UEFI) specification for
903  * details.
904  *
905  * If an event is not signaled yet, the notification function is queued. The
906  * signaled state is cleared.
907  *
908  * Return: status code
909  */
910 static efi_status_t EFIAPI efi_check_event(struct efi_event *event)
911 {
912         EFI_ENTRY("%p", event);
913         efi_timer_check();
914         if (efi_is_event(event) != EFI_SUCCESS ||
915             event->type & EVT_NOTIFY_SIGNAL)
916                 return EFI_EXIT(EFI_INVALID_PARAMETER);
917         if (!event->is_signaled)
918                 efi_queue_event(event, true);
919         if (event->is_signaled) {
920                 event->is_signaled = false;
921                 return EFI_EXIT(EFI_SUCCESS);
922         }
923         return EFI_EXIT(EFI_NOT_READY);
924 }
925
926 /**
927  * efi_search_obj() - find the internal EFI object for a handle
928  * @handle: handle to find
929  *
930  * Return: EFI object
931  */
932 struct efi_object *efi_search_obj(const efi_handle_t handle)
933 {
934         struct efi_object *efiobj;
935
936         list_for_each_entry(efiobj, &efi_obj_list, link) {
937                 if (efiobj == handle)
938                         return efiobj;
939         }
940
941         return NULL;
942 }
943
944 /**
945  * efi_open_protocol_info_entry() - create open protocol info entry and add it
946  *                                  to a protocol
947  * @handler: handler of a protocol
948  *
949  * Return: open protocol info entry
950  */
951 static struct efi_open_protocol_info_entry *efi_create_open_info(
952                         struct efi_handler *handler)
953 {
954         struct efi_open_protocol_info_item *item;
955
956         item = calloc(1, sizeof(struct efi_open_protocol_info_item));
957         if (!item)
958                 return NULL;
959         /* Append the item to the open protocol info list. */
960         list_add_tail(&item->link, &handler->open_infos);
961
962         return &item->info;
963 }
964
965 /**
966  * efi_delete_open_info() - remove an open protocol info entry from a protocol
967  * @item: open protocol info entry to delete
968  *
969  * Return: status code
970  */
971 static efi_status_t efi_delete_open_info(
972                         struct efi_open_protocol_info_item *item)
973 {
974         list_del(&item->link);
975         free(item);
976         return EFI_SUCCESS;
977 }
978
979 /**
980  * efi_add_protocol() - install new protocol on a handle
981  * @handle:             handle on which the protocol shall be installed
982  * @protocol:           GUID of the protocol to be installed
983  * @protocol_interface: interface of the protocol implementation
984  *
985  * Return: status code
986  */
987 efi_status_t efi_add_protocol(const efi_handle_t handle,
988                               const efi_guid_t *protocol,
989                               void *protocol_interface)
990 {
991         struct efi_object *efiobj;
992         struct efi_handler *handler;
993         efi_status_t ret;
994
995         efiobj = efi_search_obj(handle);
996         if (!efiobj)
997                 return EFI_INVALID_PARAMETER;
998         ret = efi_search_protocol(handle, protocol, NULL);
999         if (ret != EFI_NOT_FOUND)
1000                 return EFI_INVALID_PARAMETER;
1001         handler = calloc(1, sizeof(struct efi_handler));
1002         if (!handler)
1003                 return EFI_OUT_OF_RESOURCES;
1004         handler->guid = protocol;
1005         handler->protocol_interface = protocol_interface;
1006         INIT_LIST_HEAD(&handler->open_infos);
1007         list_add_tail(&handler->link, &efiobj->protocols);
1008         if (!guidcmp(&efi_guid_device_path, protocol))
1009                 EFI_PRINT("installed device path '%pD'\n", protocol_interface);
1010         return EFI_SUCCESS;
1011 }
1012
1013 /**
1014  * efi_install_protocol_interface() - install protocol interface
1015  * @handle:                  handle on which the protocol shall be installed
1016  * @protocol:                GUID of the protocol to be installed
1017  * @protocol_interface_type: type of the interface to be installed,
1018  *                           always EFI_NATIVE_INTERFACE
1019  * @protocol_interface:      interface of the protocol implementation
1020  *
1021  * This function implements the InstallProtocolInterface service.
1022  *
1023  * See the Unified Extensible Firmware Interface (UEFI) specification for
1024  * details.
1025  *
1026  * Return: status code
1027  */
1028 static efi_status_t EFIAPI efi_install_protocol_interface(
1029                         efi_handle_t *handle, const efi_guid_t *protocol,
1030                         int protocol_interface_type, void *protocol_interface)
1031 {
1032         efi_status_t r;
1033
1034         EFI_ENTRY("%p, %pUl, %d, %p", handle, protocol, protocol_interface_type,
1035                   protocol_interface);
1036
1037         if (!handle || !protocol ||
1038             protocol_interface_type != EFI_NATIVE_INTERFACE) {
1039                 r = EFI_INVALID_PARAMETER;
1040                 goto out;
1041         }
1042
1043         /* Create new handle if requested. */
1044         if (!*handle) {
1045                 r = efi_create_handle(handle);
1046                 if (r != EFI_SUCCESS)
1047                         goto out;
1048                 debug("%sEFI: new handle %p\n", indent_string(nesting_level),
1049                       *handle);
1050         } else {
1051                 debug("%sEFI: handle %p\n", indent_string(nesting_level),
1052                       *handle);
1053         }
1054         /* Add new protocol */
1055         r = efi_add_protocol(*handle, protocol, protocol_interface);
1056 out:
1057         return EFI_EXIT(r);
1058 }
1059
1060 /**
1061  * efi_get_drivers() - get all drivers associated to a controller
1062  * @handle:               handle of the controller
1063  * @protocol:             protocol GUID (optional)
1064  * @number_of_drivers:    number of child controllers
1065  * @driver_handle_buffer: handles of the the drivers
1066  *
1067  * The allocated buffer has to be freed with free().
1068  *
1069  * Return: status code
1070  */
1071 static efi_status_t efi_get_drivers(efi_handle_t handle,
1072                                     const efi_guid_t *protocol,
1073                                     efi_uintn_t *number_of_drivers,
1074                                     efi_handle_t **driver_handle_buffer)
1075 {
1076         struct efi_handler *handler;
1077         struct efi_open_protocol_info_item *item;
1078         efi_uintn_t count = 0, i;
1079         bool duplicate;
1080
1081         /* Count all driver associations */
1082         list_for_each_entry(handler, &handle->protocols, link) {
1083                 if (protocol && guidcmp(handler->guid, protocol))
1084                         continue;
1085                 list_for_each_entry(item, &handler->open_infos, link) {
1086                         if (item->info.attributes &
1087                             EFI_OPEN_PROTOCOL_BY_DRIVER)
1088                                 ++count;
1089                 }
1090         }
1091         /*
1092          * Create buffer. In case of duplicate driver assignments the buffer
1093          * will be too large. But that does not harm.
1094          */
1095         *number_of_drivers = 0;
1096         *driver_handle_buffer = calloc(count, sizeof(efi_handle_t));
1097         if (!*driver_handle_buffer)
1098                 return EFI_OUT_OF_RESOURCES;
1099         /* Collect unique driver handles */
1100         list_for_each_entry(handler, &handle->protocols, link) {
1101                 if (protocol && guidcmp(handler->guid, protocol))
1102                         continue;
1103                 list_for_each_entry(item, &handler->open_infos, link) {
1104                         if (item->info.attributes &
1105                             EFI_OPEN_PROTOCOL_BY_DRIVER) {
1106                                 /* Check this is a new driver */
1107                                 duplicate = false;
1108                                 for (i = 0; i < *number_of_drivers; ++i) {
1109                                         if ((*driver_handle_buffer)[i] ==
1110                                             item->info.agent_handle)
1111                                                 duplicate = true;
1112                                 }
1113                                 /* Copy handle to buffer */
1114                                 if (!duplicate) {
1115                                         i = (*number_of_drivers)++;
1116                                         (*driver_handle_buffer)[i] =
1117                                                 item->info.agent_handle;
1118                                 }
1119                         }
1120                 }
1121         }
1122         return EFI_SUCCESS;
1123 }
1124
1125 /**
1126  * efi_disconnect_all_drivers() - disconnect all drivers from a controller
1127  * @handle:       handle of the controller
1128  * @protocol:     protocol GUID (optional)
1129  * @child_handle: handle of the child to destroy
1130  *
1131  * This function implements the DisconnectController service.
1132  *
1133  * See the Unified Extensible Firmware Interface (UEFI) specification for
1134  * details.
1135  *
1136  * Return: status code
1137  */
1138 static efi_status_t efi_disconnect_all_drivers
1139                                 (efi_handle_t handle,
1140                                  const efi_guid_t *protocol,
1141                                  efi_handle_t child_handle)
1142 {
1143         efi_uintn_t number_of_drivers;
1144         efi_handle_t *driver_handle_buffer;
1145         efi_status_t r, ret;
1146
1147         ret = efi_get_drivers(handle, protocol, &number_of_drivers,
1148                               &driver_handle_buffer);
1149         if (ret != EFI_SUCCESS)
1150                 return ret;
1151
1152         ret = EFI_NOT_FOUND;
1153         while (number_of_drivers) {
1154                 r = EFI_CALL(efi_disconnect_controller(
1155                                 handle,
1156                                 driver_handle_buffer[--number_of_drivers],
1157                                 child_handle));
1158                 if (r == EFI_SUCCESS)
1159                         ret = r;
1160         }
1161         free(driver_handle_buffer);
1162         return ret;
1163 }
1164
1165 /**
1166  * efi_uninstall_protocol() - uninstall protocol interface
1167  *
1168  * @handle:             handle from which the protocol shall be removed
1169  * @protocol:           GUID of the protocol to be removed
1170  * @protocol_interface: interface to be removed
1171  *
1172  * This function DOES NOT delete a handle without installed protocol.
1173  *
1174  * Return: status code
1175  */
1176 static efi_status_t efi_uninstall_protocol
1177                         (efi_handle_t handle, const efi_guid_t *protocol,
1178                          void *protocol_interface)
1179 {
1180         struct efi_object *efiobj;
1181         struct efi_handler *handler;
1182         struct efi_open_protocol_info_item *item;
1183         struct efi_open_protocol_info_item *pos;
1184         efi_status_t r;
1185
1186         /* Check handle */
1187         efiobj = efi_search_obj(handle);
1188         if (!efiobj) {
1189                 r = EFI_INVALID_PARAMETER;
1190                 goto out;
1191         }
1192         /* Find the protocol on the handle */
1193         r = efi_search_protocol(handle, protocol, &handler);
1194         if (r != EFI_SUCCESS)
1195                 goto out;
1196         /* Disconnect controllers */
1197         efi_disconnect_all_drivers(efiobj, protocol, NULL);
1198         if (!list_empty(&handler->open_infos)) {
1199                 r =  EFI_ACCESS_DENIED;
1200                 goto out;
1201         }
1202         /* Close protocol */
1203         list_for_each_entry_safe(item, pos, &handler->open_infos, link) {
1204                 if (item->info.attributes ==
1205                         EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL ||
1206                     item->info.attributes == EFI_OPEN_PROTOCOL_GET_PROTOCOL ||
1207                     item->info.attributes == EFI_OPEN_PROTOCOL_TEST_PROTOCOL)
1208                         list_del(&item->link);
1209         }
1210         if (!list_empty(&handler->open_infos)) {
1211                 r =  EFI_ACCESS_DENIED;
1212                 goto out;
1213         }
1214         r = efi_remove_protocol(handle, protocol, protocol_interface);
1215 out:
1216         return r;
1217 }
1218
1219 /**
1220  * efi_uninstall_protocol_interface() - uninstall protocol interface
1221  * @handle:             handle from which the protocol shall be removed
1222  * @protocol:           GUID of the protocol to be removed
1223  * @protocol_interface: interface to be removed
1224  *
1225  * This function implements the UninstallProtocolInterface service.
1226  *
1227  * See the Unified Extensible Firmware Interface (UEFI) specification for
1228  * details.
1229  *
1230  * Return: status code
1231  */
1232 static efi_status_t EFIAPI efi_uninstall_protocol_interface
1233                         (efi_handle_t handle, const efi_guid_t *protocol,
1234                          void *protocol_interface)
1235 {
1236         efi_status_t ret;
1237
1238         EFI_ENTRY("%p, %pUl, %p", handle, protocol, protocol_interface);
1239
1240         ret = efi_uninstall_protocol(handle, protocol, protocol_interface);
1241         if (ret != EFI_SUCCESS)
1242                 goto out;
1243
1244         /* If the last protocol has been removed, delete the handle. */
1245         if (list_empty(&handle->protocols)) {
1246                 list_del(&handle->link);
1247                 free(handle);
1248         }
1249 out:
1250         return EFI_EXIT(ret);
1251 }
1252
1253 /**
1254  * efi_register_protocol_notify() - register an event for notification when a
1255  *                                  protocol is installed.
1256  * @protocol:     GUID of the protocol whose installation shall be notified
1257  * @event:        event to be signaled upon installation of the protocol
1258  * @registration: key for retrieving the registration information
1259  *
1260  * This function implements the RegisterProtocolNotify service.
1261  * See the Unified Extensible Firmware Interface (UEFI) specification
1262  * for details.
1263  *
1264  * Return: status code
1265  */
1266 static efi_status_t EFIAPI efi_register_protocol_notify(
1267                                                 const efi_guid_t *protocol,
1268                                                 struct efi_event *event,
1269                                                 void **registration)
1270 {
1271         EFI_ENTRY("%pUl, %p, %p", protocol, event, registration);
1272         return EFI_EXIT(EFI_OUT_OF_RESOURCES);
1273 }
1274
1275 /**
1276  * efi_search() - determine if an EFI handle implements a protocol
1277  * @search_type: selection criterion
1278  * @protocol:    GUID of the protocol
1279  * @search_key:  registration key
1280  * @handle:      handle
1281  *
1282  * See the documentation of the LocateHandle service in the UEFI specification.
1283  *
1284  * Return: 0 if the handle implements the protocol
1285  */
1286 static int efi_search(enum efi_locate_search_type search_type,
1287                       const efi_guid_t *protocol, void *search_key,
1288                       efi_handle_t handle)
1289 {
1290         efi_status_t ret;
1291
1292         switch (search_type) {
1293         case ALL_HANDLES:
1294                 return 0;
1295         case BY_REGISTER_NOTIFY:
1296                 /* TODO: RegisterProtocolNotify is not implemented yet */
1297                 return -1;
1298         case BY_PROTOCOL:
1299                 ret = efi_search_protocol(handle, protocol, NULL);
1300                 return (ret != EFI_SUCCESS);
1301         default:
1302                 /* Invalid search type */
1303                 return -1;
1304         }
1305 }
1306
1307 /**
1308  * efi_locate_handle() - locate handles implementing a protocol
1309  * @search_type: selection criterion
1310  * @protocol:    GUID of the protocol
1311  * @search_key: registration key
1312  * @buffer_size: size of the buffer to receive the handles in bytes
1313  * @buffer:      buffer to receive the relevant handles
1314  *
1315  * This function is meant for U-Boot internal calls. For the API implementation
1316  * of the LocateHandle service see efi_locate_handle_ext.
1317  *
1318  * Return: status code
1319  */
1320 static efi_status_t efi_locate_handle(
1321                         enum efi_locate_search_type search_type,
1322                         const efi_guid_t *protocol, void *search_key,
1323                         efi_uintn_t *buffer_size, efi_handle_t *buffer)
1324 {
1325         struct efi_object *efiobj;
1326         efi_uintn_t size = 0;
1327
1328         /* Check parameters */
1329         switch (search_type) {
1330         case ALL_HANDLES:
1331                 break;
1332         case BY_REGISTER_NOTIFY:
1333                 if (!search_key)
1334                         return EFI_INVALID_PARAMETER;
1335                 /* RegisterProtocolNotify is not implemented yet */
1336                 return EFI_UNSUPPORTED;
1337         case BY_PROTOCOL:
1338                 if (!protocol)
1339                         return EFI_INVALID_PARAMETER;
1340                 break;
1341         default:
1342                 return EFI_INVALID_PARAMETER;
1343         }
1344
1345         /*
1346          * efi_locate_handle_buffer uses this function for
1347          * the calculation of the necessary buffer size.
1348          * So do not require a buffer for buffersize == 0.
1349          */
1350         if (!buffer_size || (*buffer_size && !buffer))
1351                 return EFI_INVALID_PARAMETER;
1352
1353         /* Count how much space we need */
1354         list_for_each_entry(efiobj, &efi_obj_list, link) {
1355                 if (!efi_search(search_type, protocol, search_key, efiobj))
1356                         size += sizeof(void *);
1357         }
1358
1359         if (*buffer_size < size) {
1360                 *buffer_size = size;
1361                 return EFI_BUFFER_TOO_SMALL;
1362         }
1363
1364         *buffer_size = size;
1365         if (size == 0)
1366                 return EFI_NOT_FOUND;
1367
1368         /* Then fill the array */
1369         list_for_each_entry(efiobj, &efi_obj_list, link) {
1370                 if (!efi_search(search_type, protocol, search_key, efiobj))
1371                         *buffer++ = efiobj;
1372         }
1373
1374         return EFI_SUCCESS;
1375 }
1376
1377 /**
1378  * efi_locate_handle_ext() - locate handles implementing a protocol.
1379  * @search_type: selection criterion
1380  * @protocol:    GUID of the protocol
1381  * @search_key:  registration key
1382  * @buffer_size: size of the buffer to receive the handles in bytes
1383  * @buffer:      buffer to receive the relevant handles
1384  *
1385  * This function implements the LocateHandle service.
1386  *
1387  * See the Unified Extensible Firmware Interface (UEFI) specification for
1388  * details.
1389  *
1390  * Return: 0 if the handle implements the protocol
1391  */
1392 static efi_status_t EFIAPI efi_locate_handle_ext(
1393                         enum efi_locate_search_type search_type,
1394                         const efi_guid_t *protocol, void *search_key,
1395                         efi_uintn_t *buffer_size, efi_handle_t *buffer)
1396 {
1397         EFI_ENTRY("%d, %pUl, %p, %p, %p", search_type, protocol, search_key,
1398                   buffer_size, buffer);
1399
1400         return EFI_EXIT(efi_locate_handle(search_type, protocol, search_key,
1401                         buffer_size, buffer));
1402 }
1403
1404 /**
1405  * efi_remove_configuration_table() - collapses configuration table entries,
1406  *                                    removing index i
1407  *
1408  * @i: index of the table entry to be removed
1409  */
1410 static void efi_remove_configuration_table(int i)
1411 {
1412         struct efi_configuration_table *this = &systab.tables[i];
1413         struct efi_configuration_table *next = &systab.tables[i + 1];
1414         struct efi_configuration_table *end = &systab.tables[systab.nr_tables];
1415
1416         memmove(this, next, (ulong)end - (ulong)next);
1417         systab.nr_tables--;
1418 }
1419
1420 /**
1421  * efi_install_configuration_table() - adds, updates, or removes a
1422  *                                     configuration table
1423  * @guid:  GUID of the installed table
1424  * @table: table to be installed
1425  *
1426  * This function is used for internal calls. For the API implementation of the
1427  * InstallConfigurationTable service see efi_install_configuration_table_ext.
1428  *
1429  * Return: status code
1430  */
1431 efi_status_t efi_install_configuration_table(const efi_guid_t *guid,
1432                                              void *table)
1433 {
1434         struct efi_event *evt;
1435         int i;
1436
1437         if (!guid)
1438                 return EFI_INVALID_PARAMETER;
1439
1440         /* Check for GUID override */
1441         for (i = 0; i < systab.nr_tables; i++) {
1442                 if (!guidcmp(guid, &systab.tables[i].guid)) {
1443                         if (table)
1444                                 systab.tables[i].table = table;
1445                         else
1446                                 efi_remove_configuration_table(i);
1447                         goto out;
1448                 }
1449         }
1450
1451         if (!table)
1452                 return EFI_NOT_FOUND;
1453
1454         /* No override, check for overflow */
1455         if (i >= EFI_MAX_CONFIGURATION_TABLES)
1456                 return EFI_OUT_OF_RESOURCES;
1457
1458         /* Add a new entry */
1459         memcpy(&systab.tables[i].guid, guid, sizeof(*guid));
1460         systab.tables[i].table = table;
1461         systab.nr_tables = i + 1;
1462
1463 out:
1464         /* systab.nr_tables may have changed. So we need to update the CRC32 */
1465         efi_update_table_header_crc32(&systab.hdr);
1466
1467         /* Notify that the configuration table was changed */
1468         list_for_each_entry(evt, &efi_events, link) {
1469                 if (evt->group && !guidcmp(evt->group, guid)) {
1470                         efi_signal_event(evt, false);
1471                         break;
1472                 }
1473         }
1474
1475         return EFI_SUCCESS;
1476 }
1477
1478 /**
1479  * efi_install_configuration_table_ex() - Adds, updates, or removes a
1480  *                                        configuration table.
1481  * @guid:  GUID of the installed table
1482  * @table: table to be installed
1483  *
1484  * This function implements the InstallConfigurationTable service.
1485  *
1486  * See the Unified Extensible Firmware Interface (UEFI) specification for
1487  * details.
1488  *
1489  * Return: status code
1490  */
1491 static efi_status_t EFIAPI efi_install_configuration_table_ext(efi_guid_t *guid,
1492                                                                void *table)
1493 {
1494         EFI_ENTRY("%pUl, %p", guid, table);
1495         return EFI_EXIT(efi_install_configuration_table(guid, table));
1496 }
1497
1498 /**
1499  * efi_setup_loaded_image() - initialize a loaded image
1500  *
1501  * Initialize a loaded_image_info and loaded_image_info object with correct
1502  * protocols, boot-device, etc.
1503  *
1504  * In case of an error *handle_ptr and *info_ptr are set to NULL and an error
1505  * code is returned.
1506  *
1507  * @device_path:        device path of the loaded image
1508  * @file_path:          file path of the loaded image
1509  * @handle_ptr:         handle of the loaded image
1510  * @info_ptr:           loaded image protocol
1511  * Return:              status code
1512  */
1513 efi_status_t efi_setup_loaded_image(struct efi_device_path *device_path,
1514                                     struct efi_device_path *file_path,
1515                                     struct efi_loaded_image_obj **handle_ptr,
1516                                     struct efi_loaded_image **info_ptr)
1517 {
1518         efi_status_t ret;
1519         struct efi_loaded_image *info = NULL;
1520         struct efi_loaded_image_obj *obj = NULL;
1521
1522         /* In case of EFI_OUT_OF_RESOURCES avoid illegal free by caller. */
1523         *handle_ptr = NULL;
1524         *info_ptr = NULL;
1525
1526         info = calloc(1, sizeof(*info));
1527         if (!info)
1528                 return EFI_OUT_OF_RESOURCES;
1529         obj = calloc(1, sizeof(*obj));
1530         if (!obj) {
1531                 free(info);
1532                 return EFI_OUT_OF_RESOURCES;
1533         }
1534
1535         /* Add internal object to object list */
1536         efi_add_handle(&obj->header);
1537
1538         info->revision =  EFI_LOADED_IMAGE_PROTOCOL_REVISION;
1539         info->file_path = file_path;
1540         info->system_table = &systab;
1541
1542         if (device_path) {
1543                 info->device_handle = efi_dp_find_obj(device_path, NULL);
1544                 /*
1545                  * When asking for the device path interface, return
1546                  * bootefi_device_path
1547                  */
1548                 ret = efi_add_protocol(&obj->header,
1549                                        &efi_guid_device_path, device_path);
1550                 if (ret != EFI_SUCCESS)
1551                         goto failure;
1552         }
1553
1554         /*
1555          * When asking for the loaded_image interface, just
1556          * return handle which points to loaded_image_info
1557          */
1558         ret = efi_add_protocol(&obj->header,
1559                                &efi_guid_loaded_image, info);
1560         if (ret != EFI_SUCCESS)
1561                 goto failure;
1562
1563 #if CONFIG_IS_ENABLED(EFI_LOADER_HII)
1564         ret = efi_add_protocol(&obj->header,
1565                                &efi_guid_hii_string_protocol,
1566                                (void *)&efi_hii_string);
1567         if (ret != EFI_SUCCESS)
1568                 goto failure;
1569
1570         ret = efi_add_protocol(&obj->header,
1571                                &efi_guid_hii_database_protocol,
1572                                (void *)&efi_hii_database);
1573         if (ret != EFI_SUCCESS)
1574                 goto failure;
1575
1576         ret = efi_add_protocol(&obj->header,
1577                                &efi_guid_hii_config_routing_protocol,
1578                                (void *)&efi_hii_config_routing);
1579         if (ret != EFI_SUCCESS)
1580                 goto failure;
1581 #endif
1582
1583         if (info_ptr)
1584                 *info_ptr = info;
1585         if (handle_ptr)
1586                 *handle_ptr = obj;
1587
1588         return ret;
1589 failure:
1590         printf("ERROR: Failure to install protocols for loaded image\n");
1591         efi_delete_handle(&obj->header);
1592         free(info);
1593         return ret;
1594 }
1595
1596 /**
1597  * efi_load_image_from_path() - load an image using a file path
1598  *
1599  * Read a file into a buffer allocated as EFI_BOOT_SERVICES_DATA. It is the
1600  * callers obligation to update the memory type as needed.
1601  *
1602  * @file_path:  the path of the image to load
1603  * @buffer:     buffer containing the loaded image
1604  * @size:       size of the loaded image
1605  * Return:      status code
1606  */
1607 efi_status_t efi_load_image_from_path(struct efi_device_path *file_path,
1608                                       void **buffer, efi_uintn_t *size)
1609 {
1610         struct efi_file_info *info = NULL;
1611         struct efi_file_handle *f;
1612         static efi_status_t ret;
1613         u64 addr;
1614         efi_uintn_t bs;
1615
1616         /* In case of failure nothing is returned */
1617         *buffer = NULL;
1618         *size = 0;
1619
1620         /* Open file */
1621         f = efi_file_from_path(file_path);
1622         if (!f)
1623                 return EFI_DEVICE_ERROR;
1624
1625         /* Get file size */
1626         bs = 0;
1627         EFI_CALL(ret = f->getinfo(f, (efi_guid_t *)&efi_file_info_guid,
1628                                   &bs, info));
1629         if (ret != EFI_BUFFER_TOO_SMALL) {
1630                 ret =  EFI_DEVICE_ERROR;
1631                 goto error;
1632         }
1633
1634         info = malloc(bs);
1635         EFI_CALL(ret = f->getinfo(f, (efi_guid_t *)&efi_file_info_guid, &bs,
1636                                   info));
1637         if (ret != EFI_SUCCESS)
1638                 goto error;
1639
1640         /*
1641          * When reading the file we do not yet know if it contains an
1642          * application, a boottime driver, or a runtime driver. So here we
1643          * allocate a buffer as EFI_BOOT_SERVICES_DATA. The caller has to
1644          * update the reservation according to the image type.
1645          */
1646         bs = info->file_size;
1647         ret = efi_allocate_pages(EFI_ALLOCATE_ANY_PAGES,
1648                                  EFI_BOOT_SERVICES_DATA,
1649                                  efi_size_in_pages(bs), &addr);
1650         if (ret != EFI_SUCCESS) {
1651                 ret = EFI_OUT_OF_RESOURCES;
1652                 goto error;
1653         }
1654
1655         /* Read file */
1656         EFI_CALL(ret = f->read(f, &bs, (void *)(uintptr_t)addr));
1657         if (ret != EFI_SUCCESS)
1658                 efi_free_pages(addr, efi_size_in_pages(bs));
1659         *buffer = (void *)(uintptr_t)addr;
1660         *size = bs;
1661 error:
1662         EFI_CALL(f->close(f));
1663         free(info);
1664         return ret;
1665 }
1666
1667 /**
1668  * efi_load_image() - load an EFI image into memory
1669  * @boot_policy:   true for request originating from the boot manager
1670  * @parent_image:  the caller's image handle
1671  * @file_path:     the path of the image to load
1672  * @source_buffer: memory location from which the image is installed
1673  * @source_size:   size of the memory area from which the image is installed
1674  * @image_handle:  handle for the newly installed image
1675  *
1676  * This function implements the LoadImage service.
1677  *
1678  * See the Unified Extensible Firmware Interface (UEFI) specification
1679  * for details.
1680  *
1681  * Return: status code
1682  */
1683 static efi_status_t EFIAPI efi_load_image(bool boot_policy,
1684                                           efi_handle_t parent_image,
1685                                           struct efi_device_path *file_path,
1686                                           void *source_buffer,
1687                                           efi_uintn_t source_size,
1688                                           efi_handle_t *image_handle)
1689 {
1690         struct efi_device_path *dp, *fp;
1691         struct efi_loaded_image *info = NULL;
1692         struct efi_loaded_image_obj **image_obj =
1693                 (struct efi_loaded_image_obj **)image_handle;
1694         efi_status_t ret;
1695
1696         EFI_ENTRY("%d, %p, %pD, %p, %zd, %p", boot_policy, parent_image,
1697                   file_path, source_buffer, source_size, image_handle);
1698
1699         if (!image_handle || !parent_image) {
1700                 ret = EFI_INVALID_PARAMETER;
1701                 goto error;
1702         }
1703
1704         if (!source_buffer && !file_path) {
1705                 ret = EFI_NOT_FOUND;
1706                 goto error;
1707         }
1708
1709         if (!source_buffer) {
1710                 ret = efi_load_image_from_path(file_path, &source_buffer,
1711                                                &source_size);
1712                 if (ret != EFI_SUCCESS)
1713                         goto error;
1714                 /*
1715                  * split file_path which contains both the device and
1716                  * file parts:
1717                  */
1718                 efi_dp_split_file_path(file_path, &dp, &fp);
1719         } else {
1720                 /* In this case, file_path is the "device" path, i.e.
1721                  * something like a HARDWARE_DEVICE:MEMORY_MAPPED
1722                  */
1723                 u64 addr;
1724                 void *dest_buffer;
1725
1726                 ret = efi_allocate_pages(EFI_ALLOCATE_ANY_PAGES,
1727                                          EFI_RUNTIME_SERVICES_CODE,
1728                                          efi_size_in_pages(source_size), &addr);
1729                 if (ret != EFI_SUCCESS)
1730                         goto error;
1731                 dest_buffer = (void *)(uintptr_t)addr;
1732                 memcpy(dest_buffer, source_buffer, source_size);
1733                 source_buffer = dest_buffer;
1734
1735                 dp = file_path;
1736                 fp = NULL;
1737         }
1738         ret = efi_setup_loaded_image(dp, fp, image_obj, &info);
1739         if (ret != EFI_SUCCESS)
1740                 goto error_invalid_image;
1741         (*image_obj)->entry = efi_load_pe(*image_obj, source_buffer, info);
1742         if (!(*image_obj)->entry) {
1743                 ret = EFI_UNSUPPORTED;
1744                 goto error_invalid_image;
1745         }
1746         /* Update the type of the allocated memory */
1747         efi_add_memory_map((uintptr_t)source_buffer,
1748                            efi_size_in_pages(source_size),
1749                            info->image_code_type, false);
1750         info->system_table = &systab;
1751         info->parent_handle = parent_image;
1752         return EFI_EXIT(EFI_SUCCESS);
1753 error_invalid_image:
1754         /* The image is invalid. Release all associated resources. */
1755         efi_free_pages((uintptr_t)source_buffer,
1756                        efi_size_in_pages(source_size));
1757         efi_delete_handle(*image_handle);
1758         *image_handle = NULL;
1759         free(info);
1760 error:
1761         return EFI_EXIT(ret);
1762 }
1763
1764 /**
1765  * efi_start_image() - call the entry point of an image
1766  * @image_handle:   handle of the image
1767  * @exit_data_size: size of the buffer
1768  * @exit_data:      buffer to receive the exit data of the called image
1769  *
1770  * This function implements the StartImage service.
1771  *
1772  * See the Unified Extensible Firmware Interface (UEFI) specification for
1773  * details.
1774  *
1775  * Return: status code
1776  */
1777 static efi_status_t EFIAPI efi_start_image(efi_handle_t image_handle,
1778                                            efi_uintn_t *exit_data_size,
1779                                            u16 **exit_data)
1780 {
1781         struct efi_loaded_image_obj *image_obj =
1782                 (struct efi_loaded_image_obj *)image_handle;
1783         efi_status_t ret;
1784
1785         EFI_ENTRY("%p, %p, %p", image_handle, exit_data_size, exit_data);
1786
1787         efi_is_direct_boot = false;
1788
1789         /* call the image! */
1790         if (setjmp(&image_obj->exit_jmp)) {
1791                 /*
1792                  * We called the entry point of the child image with EFI_CALL
1793                  * in the lines below. The child image called the Exit() boot
1794                  * service efi_exit() which executed the long jump that brought
1795                  * us to the current line. This implies that the second half
1796                  * of the EFI_CALL macro has not been executed.
1797                  */
1798 #ifdef CONFIG_ARM
1799                 /*
1800                  * efi_exit() called efi_restore_gd(). We have to undo this
1801                  * otherwise __efi_entry_check() will put the wrong value into
1802                  * app_gd.
1803                  */
1804                 gd = app_gd;
1805 #endif
1806                 /*
1807                  * To get ready to call EFI_EXIT below we have to execute the
1808                  * missed out steps of EFI_CALL.
1809                  */
1810                 assert(__efi_entry_check());
1811                 debug("%sEFI: %lu returned by started image\n",
1812                       __efi_nesting_dec(),
1813                       (unsigned long)((uintptr_t)image_obj->exit_status &
1814                                       ~EFI_ERROR_MASK));
1815                 return EFI_EXIT(image_obj->exit_status);
1816         }
1817
1818         ret = EFI_CALL(image_obj->entry(image_handle, &systab));
1819
1820         /*
1821          * Usually UEFI applications call Exit() instead of returning.
1822          * But because the world doesn't consist of ponies and unicorns,
1823          * we're happy to emulate that behavior on behalf of a payload
1824          * that forgot.
1825          */
1826         return EFI_CALL(systab.boottime->exit(image_handle, ret, 0, NULL));
1827 }
1828
1829 /**
1830  * efi_exit() - leave an EFI application or driver
1831  * @image_handle:   handle of the application or driver that is exiting
1832  * @exit_status:    status code
1833  * @exit_data_size: size of the buffer in bytes
1834  * @exit_data:      buffer with data describing an error
1835  *
1836  * This function implements the Exit service.
1837  *
1838  * See the Unified Extensible Firmware Interface (UEFI) specification for
1839  * details.
1840  *
1841  * Return: status code
1842  */
1843 static efi_status_t EFIAPI efi_exit(efi_handle_t image_handle,
1844                                     efi_status_t exit_status,
1845                                     efi_uintn_t exit_data_size,
1846                                     u16 *exit_data)
1847 {
1848         /*
1849          * TODO: We should call the unload procedure of the loaded
1850          *       image protocol.
1851          */
1852         struct efi_loaded_image_obj *image_obj =
1853                 (struct efi_loaded_image_obj *)image_handle;
1854
1855         EFI_ENTRY("%p, %ld, %zu, %p", image_handle, exit_status,
1856                   exit_data_size, exit_data);
1857
1858         /* Make sure entry/exit counts for EFI world cross-overs match */
1859         EFI_EXIT(exit_status);
1860
1861         /*
1862          * But longjmp out with the U-Boot gd, not the application's, as
1863          * the other end is a setjmp call inside EFI context.
1864          */
1865         efi_restore_gd();
1866
1867         image_obj->exit_status = exit_status;
1868         longjmp(&image_obj->exit_jmp, 1);
1869
1870         panic("EFI application exited");
1871 }
1872
1873 /**
1874  * efi_unload_image() - unload an EFI image
1875  * @image_handle: handle of the image to be unloaded
1876  *
1877  * This function implements the UnloadImage service.
1878  *
1879  * See the Unified Extensible Firmware Interface (UEFI) specification for
1880  * details.
1881  *
1882  * Return: status code
1883  */
1884 static efi_status_t EFIAPI efi_unload_image(efi_handle_t image_handle)
1885 {
1886         struct efi_object *efiobj;
1887
1888         EFI_ENTRY("%p", image_handle);
1889         efiobj = efi_search_obj(image_handle);
1890         if (efiobj)
1891                 list_del(&efiobj->link);
1892
1893         return EFI_EXIT(EFI_SUCCESS);
1894 }
1895
1896 /**
1897  * efi_exit_caches() - fix up caches for EFI payloads if necessary
1898  */
1899 static void efi_exit_caches(void)
1900 {
1901 #if defined(CONFIG_ARM) && !defined(CONFIG_ARM64)
1902         /*
1903          * Grub on 32bit ARM needs to have caches disabled before jumping into
1904          * a zImage, but does not know of all cache layers. Give it a hand.
1905          */
1906         if (efi_is_direct_boot)
1907                 cleanup_before_linux();
1908 #endif
1909 }
1910
1911 /**
1912  * efi_exit_boot_services() - stop all boot services
1913  * @image_handle: handle of the loaded image
1914  * @map_key:      key of the memory map
1915  *
1916  * This function implements the ExitBootServices service.
1917  *
1918  * See the Unified Extensible Firmware Interface (UEFI) specification
1919  * for details.
1920  *
1921  * All timer events are disabled. For exit boot services events the
1922  * notification function is called. The boot services are disabled in the
1923  * system table.
1924  *
1925  * Return: status code
1926  */
1927 static efi_status_t EFIAPI efi_exit_boot_services(efi_handle_t image_handle,
1928                                                   unsigned long map_key)
1929 {
1930         struct efi_event *evt;
1931
1932         EFI_ENTRY("%p, %ld", image_handle, map_key);
1933
1934         /* Check that the caller has read the current memory map */
1935         if (map_key != efi_memory_map_key)
1936                 return EFI_INVALID_PARAMETER;
1937
1938         /* Make sure that notification functions are not called anymore */
1939         efi_tpl = TPL_HIGH_LEVEL;
1940
1941         /* Check if ExitBootServices has already been called */
1942         if (!systab.boottime)
1943                 return EFI_EXIT(EFI_SUCCESS);
1944
1945         /* Add related events to the event group */
1946         list_for_each_entry(evt, &efi_events, link) {
1947                 if (evt->type == EVT_SIGNAL_EXIT_BOOT_SERVICES)
1948                         evt->group = &efi_guid_event_group_exit_boot_services;
1949         }
1950         /* Notify that ExitBootServices is invoked. */
1951         list_for_each_entry(evt, &efi_events, link) {
1952                 if (evt->group &&
1953                     !guidcmp(evt->group,
1954                              &efi_guid_event_group_exit_boot_services)) {
1955                         efi_signal_event(evt, false);
1956                         break;
1957                 }
1958         }
1959
1960         /* TODO: Should persist EFI variables here */
1961
1962         board_quiesce_devices();
1963
1964         /* Fix up caches for EFI payloads if necessary */
1965         efi_exit_caches();
1966
1967         /* This stops all lingering devices */
1968         bootm_disable_interrupts();
1969
1970         /* Disable boot time services */
1971         systab.con_in_handle = NULL;
1972         systab.con_in = NULL;
1973         systab.con_out_handle = NULL;
1974         systab.con_out = NULL;
1975         systab.stderr_handle = NULL;
1976         systab.std_err = NULL;
1977         systab.boottime = NULL;
1978
1979         /* Recalculate CRC32 */
1980         efi_update_table_header_crc32(&systab.hdr);
1981
1982         /* Give the payload some time to boot */
1983         efi_set_watchdog(0);
1984         WATCHDOG_RESET();
1985
1986         return EFI_EXIT(EFI_SUCCESS);
1987 }
1988
1989 /**
1990  * efi_get_next_monotonic_count() - get next value of the counter
1991  * @count: returned value of the counter
1992  *
1993  * This function implements the NextMonotonicCount service.
1994  *
1995  * See the Unified Extensible Firmware Interface (UEFI) specification for
1996  * details.
1997  *
1998  * Return: status code
1999  */
2000 static efi_status_t EFIAPI efi_get_next_monotonic_count(uint64_t *count)
2001 {
2002         static uint64_t mono;
2003
2004         EFI_ENTRY("%p", count);
2005         *count = mono++;
2006         return EFI_EXIT(EFI_SUCCESS);
2007 }
2008
2009 /**
2010  * efi_stall() - sleep
2011  * @microseconds: period to sleep in microseconds
2012  *
2013  * This function implements the Stall service.
2014  *
2015  * See the Unified Extensible Firmware Interface (UEFI) specification for
2016  * details.
2017  *
2018  * Return:  status code
2019  */
2020 static efi_status_t EFIAPI efi_stall(unsigned long microseconds)
2021 {
2022         EFI_ENTRY("%ld", microseconds);
2023         udelay(microseconds);
2024         return EFI_EXIT(EFI_SUCCESS);
2025 }
2026
2027 /**
2028  * efi_set_watchdog_timer() - reset the watchdog timer
2029  * @timeout:       seconds before reset by watchdog
2030  * @watchdog_code: code to be logged when resetting
2031  * @data_size:     size of buffer in bytes
2032  * @watchdog_data: buffer with data describing the reset reason
2033  *
2034  * This function implements the SetWatchdogTimer service.
2035  *
2036  * See the Unified Extensible Firmware Interface (UEFI) specification for
2037  * details.
2038  *
2039  * Return: status code
2040  */
2041 static efi_status_t EFIAPI efi_set_watchdog_timer(unsigned long timeout,
2042                                                   uint64_t watchdog_code,
2043                                                   unsigned long data_size,
2044                                                   uint16_t *watchdog_data)
2045 {
2046         EFI_ENTRY("%ld, 0x%llx, %ld, %p", timeout, watchdog_code,
2047                   data_size, watchdog_data);
2048         return EFI_EXIT(efi_set_watchdog(timeout));
2049 }
2050
2051 /**
2052  * efi_close_protocol() - close a protocol
2053  * @handle:            handle on which the protocol shall be closed
2054  * @protocol:          GUID of the protocol to close
2055  * @agent_handle:      handle of the driver
2056  * @controller_handle: handle of the controller
2057  *
2058  * This function implements the CloseProtocol service.
2059  *
2060  * See the Unified Extensible Firmware Interface (UEFI) specification for
2061  * details.
2062  *
2063  * Return: status code
2064  */
2065 static efi_status_t EFIAPI efi_close_protocol(efi_handle_t handle,
2066                                               const efi_guid_t *protocol,
2067                                               efi_handle_t agent_handle,
2068                                               efi_handle_t controller_handle)
2069 {
2070         struct efi_handler *handler;
2071         struct efi_open_protocol_info_item *item;
2072         struct efi_open_protocol_info_item *pos;
2073         efi_status_t r;
2074
2075         EFI_ENTRY("%p, %pUl, %p, %p", handle, protocol, agent_handle,
2076                   controller_handle);
2077
2078         if (!agent_handle) {
2079                 r = EFI_INVALID_PARAMETER;
2080                 goto out;
2081         }
2082         r = efi_search_protocol(handle, protocol, &handler);
2083         if (r != EFI_SUCCESS)
2084                 goto out;
2085
2086         r = EFI_NOT_FOUND;
2087         list_for_each_entry_safe(item, pos, &handler->open_infos, link) {
2088                 if (item->info.agent_handle == agent_handle &&
2089                     item->info.controller_handle == controller_handle) {
2090                         efi_delete_open_info(item);
2091                         r = EFI_SUCCESS;
2092                         break;
2093                 }
2094         }
2095 out:
2096         return EFI_EXIT(r);
2097 }
2098
2099 /**
2100  * efi_open_protocol_information() - provide information about then open status
2101  *                                   of a protocol on a handle
2102  * @handle:       handle for which the information shall be retrieved
2103  * @protocol:     GUID of the protocol
2104  * @entry_buffer: buffer to receive the open protocol information
2105  * @entry_count:  number of entries available in the buffer
2106  *
2107  * This function implements the OpenProtocolInformation service.
2108  *
2109  * See the Unified Extensible Firmware Interface (UEFI) specification for
2110  * details.
2111  *
2112  * Return: status code
2113  */
2114 static efi_status_t EFIAPI efi_open_protocol_information(
2115                         efi_handle_t handle, const efi_guid_t *protocol,
2116                         struct efi_open_protocol_info_entry **entry_buffer,
2117                         efi_uintn_t *entry_count)
2118 {
2119         unsigned long buffer_size;
2120         unsigned long count;
2121         struct efi_handler *handler;
2122         struct efi_open_protocol_info_item *item;
2123         efi_status_t r;
2124
2125         EFI_ENTRY("%p, %pUl, %p, %p", handle, protocol, entry_buffer,
2126                   entry_count);
2127
2128         /* Check parameters */
2129         if (!entry_buffer) {
2130                 r = EFI_INVALID_PARAMETER;
2131                 goto out;
2132         }
2133         r = efi_search_protocol(handle, protocol, &handler);
2134         if (r != EFI_SUCCESS)
2135                 goto out;
2136
2137         /* Count entries */
2138         count = 0;
2139         list_for_each_entry(item, &handler->open_infos, link) {
2140                 if (item->info.open_count)
2141                         ++count;
2142         }
2143         *entry_count = count;
2144         *entry_buffer = NULL;
2145         if (!count) {
2146                 r = EFI_SUCCESS;
2147                 goto out;
2148         }
2149
2150         /* Copy entries */
2151         buffer_size = count * sizeof(struct efi_open_protocol_info_entry);
2152         r = efi_allocate_pool(EFI_BOOT_SERVICES_DATA, buffer_size,
2153                               (void **)entry_buffer);
2154         if (r != EFI_SUCCESS)
2155                 goto out;
2156         list_for_each_entry_reverse(item, &handler->open_infos, link) {
2157                 if (item->info.open_count)
2158                         (*entry_buffer)[--count] = item->info;
2159         }
2160 out:
2161         return EFI_EXIT(r);
2162 }
2163
2164 /**
2165  * efi_protocols_per_handle() - get protocols installed on a handle
2166  * @handle:                handle for which the information is retrieved
2167  * @protocol_buffer:       buffer with protocol GUIDs
2168  * @protocol_buffer_count: number of entries in the buffer
2169  *
2170  * This function implements the ProtocolsPerHandleService.
2171  *
2172  * See the Unified Extensible Firmware Interface (UEFI) specification for
2173  * details.
2174  *
2175  * Return: status code
2176  */
2177 static efi_status_t EFIAPI efi_protocols_per_handle(
2178                         efi_handle_t handle, efi_guid_t ***protocol_buffer,
2179                         efi_uintn_t *protocol_buffer_count)
2180 {
2181         unsigned long buffer_size;
2182         struct efi_object *efiobj;
2183         struct list_head *protocol_handle;
2184         efi_status_t r;
2185
2186         EFI_ENTRY("%p, %p, %p", handle, protocol_buffer,
2187                   protocol_buffer_count);
2188
2189         if (!handle || !protocol_buffer || !protocol_buffer_count)
2190                 return EFI_EXIT(EFI_INVALID_PARAMETER);
2191
2192         *protocol_buffer = NULL;
2193         *protocol_buffer_count = 0;
2194
2195         efiobj = efi_search_obj(handle);
2196         if (!efiobj)
2197                 return EFI_EXIT(EFI_INVALID_PARAMETER);
2198
2199         /* Count protocols */
2200         list_for_each(protocol_handle, &efiobj->protocols) {
2201                 ++*protocol_buffer_count;
2202         }
2203
2204         /* Copy GUIDs */
2205         if (*protocol_buffer_count) {
2206                 size_t j = 0;
2207
2208                 buffer_size = sizeof(efi_guid_t *) * *protocol_buffer_count;
2209                 r = efi_allocate_pool(EFI_BOOT_SERVICES_DATA, buffer_size,
2210                                       (void **)protocol_buffer);
2211                 if (r != EFI_SUCCESS)
2212                         return EFI_EXIT(r);
2213                 list_for_each(protocol_handle, &efiobj->protocols) {
2214                         struct efi_handler *protocol;
2215
2216                         protocol = list_entry(protocol_handle,
2217                                               struct efi_handler, link);
2218                         (*protocol_buffer)[j] = (void *)protocol->guid;
2219                         ++j;
2220                 }
2221         }
2222
2223         return EFI_EXIT(EFI_SUCCESS);
2224 }
2225
2226 /**
2227  * efi_locate_handle_buffer() - locate handles implementing a protocol
2228  * @search_type: selection criterion
2229  * @protocol:    GUID of the protocol
2230  * @search_key:  registration key
2231  * @no_handles:  number of returned handles
2232  * @buffer:      buffer with the returned handles
2233  *
2234  * This function implements the LocateHandleBuffer service.
2235  *
2236  * See the Unified Extensible Firmware Interface (UEFI) specification for
2237  * details.
2238  *
2239  * Return: status code
2240  */
2241 static efi_status_t EFIAPI efi_locate_handle_buffer(
2242                         enum efi_locate_search_type search_type,
2243                         const efi_guid_t *protocol, void *search_key,
2244                         efi_uintn_t *no_handles, efi_handle_t **buffer)
2245 {
2246         efi_status_t r;
2247         efi_uintn_t buffer_size = 0;
2248
2249         EFI_ENTRY("%d, %pUl, %p, %p, %p", search_type, protocol, search_key,
2250                   no_handles, buffer);
2251
2252         if (!no_handles || !buffer) {
2253                 r = EFI_INVALID_PARAMETER;
2254                 goto out;
2255         }
2256         *no_handles = 0;
2257         *buffer = NULL;
2258         r = efi_locate_handle(search_type, protocol, search_key, &buffer_size,
2259                               *buffer);
2260         if (r != EFI_BUFFER_TOO_SMALL)
2261                 goto out;
2262         r = efi_allocate_pool(EFI_BOOT_SERVICES_DATA, buffer_size,
2263                               (void **)buffer);
2264         if (r != EFI_SUCCESS)
2265                 goto out;
2266         r = efi_locate_handle(search_type, protocol, search_key, &buffer_size,
2267                               *buffer);
2268         if (r == EFI_SUCCESS)
2269                 *no_handles = buffer_size / sizeof(efi_handle_t);
2270 out:
2271         return EFI_EXIT(r);
2272 }
2273
2274 /**
2275  * efi_locate_protocol() - find an interface implementing a protocol
2276  * @protocol:           GUID of the protocol
2277  * @registration:       registration key passed to the notification function
2278  * @protocol_interface: interface implementing the protocol
2279  *
2280  * This function implements the LocateProtocol service.
2281  *
2282  * See the Unified Extensible Firmware Interface (UEFI) specification for
2283  * details.
2284  *
2285  * Return: status code
2286  */
2287 static efi_status_t EFIAPI efi_locate_protocol(const efi_guid_t *protocol,
2288                                                void *registration,
2289                                                void **protocol_interface)
2290 {
2291         struct list_head *lhandle;
2292         efi_status_t ret;
2293
2294         EFI_ENTRY("%pUl, %p, %p", protocol, registration, protocol_interface);
2295
2296         if (!protocol || !protocol_interface)
2297                 return EFI_EXIT(EFI_INVALID_PARAMETER);
2298
2299         list_for_each(lhandle, &efi_obj_list) {
2300                 struct efi_object *efiobj;
2301                 struct efi_handler *handler;
2302
2303                 efiobj = list_entry(lhandle, struct efi_object, link);
2304
2305                 ret = efi_search_protocol(efiobj, protocol, &handler);
2306                 if (ret == EFI_SUCCESS) {
2307                         *protocol_interface = handler->protocol_interface;
2308                         return EFI_EXIT(EFI_SUCCESS);
2309                 }
2310         }
2311         *protocol_interface = NULL;
2312
2313         return EFI_EXIT(EFI_NOT_FOUND);
2314 }
2315
2316 /**
2317  * efi_locate_device_path() - Get the device path and handle of an device
2318  *                            implementing a protocol
2319  * @protocol:    GUID of the protocol
2320  * @device_path: device path
2321  * @device:      handle of the device
2322  *
2323  * This function implements the LocateDevicePath service.
2324  *
2325  * See the Unified Extensible Firmware Interface (UEFI) specification for
2326  * details.
2327  *
2328  * Return: status code
2329  */
2330 static efi_status_t EFIAPI efi_locate_device_path(
2331                         const efi_guid_t *protocol,
2332                         struct efi_device_path **device_path,
2333                         efi_handle_t *device)
2334 {
2335         struct efi_device_path *dp;
2336         size_t i;
2337         struct efi_handler *handler;
2338         efi_handle_t *handles;
2339         size_t len, len_dp;
2340         size_t len_best = 0;
2341         efi_uintn_t no_handles;
2342         u8 *remainder;
2343         efi_status_t ret;
2344
2345         EFI_ENTRY("%pUl, %p, %p", protocol, device_path, device);
2346
2347         if (!protocol || !device_path || !*device_path || !device) {
2348                 ret = EFI_INVALID_PARAMETER;
2349                 goto out;
2350         }
2351
2352         /* Find end of device path */
2353         len = efi_dp_instance_size(*device_path);
2354
2355         /* Get all handles implementing the protocol */
2356         ret = EFI_CALL(efi_locate_handle_buffer(BY_PROTOCOL, protocol, NULL,
2357                                                 &no_handles, &handles));
2358         if (ret != EFI_SUCCESS)
2359                 goto out;
2360
2361         for (i = 0; i < no_handles; ++i) {
2362                 /* Find the device path protocol */
2363                 ret = efi_search_protocol(handles[i], &efi_guid_device_path,
2364                                           &handler);
2365                 if (ret != EFI_SUCCESS)
2366                         continue;
2367                 dp = (struct efi_device_path *)handler->protocol_interface;
2368                 len_dp = efi_dp_instance_size(dp);
2369                 /*
2370                  * This handle can only be a better fit
2371                  * if its device path length is longer than the best fit and
2372                  * if its device path length is shorter of equal the searched
2373                  * device path.
2374                  */
2375                 if (len_dp <= len_best || len_dp > len)
2376                         continue;
2377                 /* Check if dp is a subpath of device_path */
2378                 if (memcmp(*device_path, dp, len_dp))
2379                         continue;
2380                 *device = handles[i];
2381                 len_best = len_dp;
2382         }
2383         if (len_best) {
2384                 remainder = (u8 *)*device_path + len_best;
2385                 *device_path = (struct efi_device_path *)remainder;
2386                 ret = EFI_SUCCESS;
2387         } else {
2388                 ret = EFI_NOT_FOUND;
2389         }
2390 out:
2391         return EFI_EXIT(ret);
2392 }
2393
2394 /**
2395  * efi_install_multiple_protocol_interfaces() - Install multiple protocol
2396  *                                              interfaces
2397  * @handle: handle on which the protocol interfaces shall be installed
2398  * @...:    NULL terminated argument list with pairs of protocol GUIDS and
2399  *          interfaces
2400  *
2401  * This function implements the MultipleProtocolInterfaces service.
2402  *
2403  * See the Unified Extensible Firmware Interface (UEFI) specification for
2404  * details.
2405  *
2406  * Return: status code
2407  */
2408 static efi_status_t EFIAPI efi_install_multiple_protocol_interfaces
2409                                 (efi_handle_t *handle, ...)
2410 {
2411         EFI_ENTRY("%p", handle);
2412
2413         efi_va_list argptr;
2414         const efi_guid_t *protocol;
2415         void *protocol_interface;
2416         efi_status_t r = EFI_SUCCESS;
2417         int i = 0;
2418
2419         if (!handle)
2420                 return EFI_EXIT(EFI_INVALID_PARAMETER);
2421
2422         efi_va_start(argptr, handle);
2423         for (;;) {
2424                 protocol = efi_va_arg(argptr, efi_guid_t*);
2425                 if (!protocol)
2426                         break;
2427                 protocol_interface = efi_va_arg(argptr, void*);
2428                 r = EFI_CALL(efi_install_protocol_interface(
2429                                                 handle, protocol,
2430                                                 EFI_NATIVE_INTERFACE,
2431                                                 protocol_interface));
2432                 if (r != EFI_SUCCESS)
2433                         break;
2434                 i++;
2435         }
2436         efi_va_end(argptr);
2437         if (r == EFI_SUCCESS)
2438                 return EFI_EXIT(r);
2439
2440         /* If an error occurred undo all changes. */
2441         efi_va_start(argptr, handle);
2442         for (; i; --i) {
2443                 protocol = efi_va_arg(argptr, efi_guid_t*);
2444                 protocol_interface = efi_va_arg(argptr, void*);
2445                 EFI_CALL(efi_uninstall_protocol_interface(*handle, protocol,
2446                                                           protocol_interface));
2447         }
2448         efi_va_end(argptr);
2449
2450         return EFI_EXIT(r);
2451 }
2452
2453 /**
2454  * efi_uninstall_multiple_protocol_interfaces() - uninstall multiple protocol
2455  *                                                interfaces
2456  * @handle: handle from which the protocol interfaces shall be removed
2457  * @...:    NULL terminated argument list with pairs of protocol GUIDS and
2458  *          interfaces
2459  *
2460  * This function implements the UninstallMultipleProtocolInterfaces service.
2461  *
2462  * See the Unified Extensible Firmware Interface (UEFI) specification for
2463  * details.
2464  *
2465  * Return: status code
2466  */
2467 static efi_status_t EFIAPI efi_uninstall_multiple_protocol_interfaces(
2468                         efi_handle_t handle, ...)
2469 {
2470         EFI_ENTRY("%p", handle);
2471
2472         efi_va_list argptr;
2473         const efi_guid_t *protocol;
2474         void *protocol_interface;
2475         efi_status_t r = EFI_SUCCESS;
2476         size_t i = 0;
2477
2478         if (!handle)
2479                 return EFI_EXIT(EFI_INVALID_PARAMETER);
2480
2481         efi_va_start(argptr, handle);
2482         for (;;) {
2483                 protocol = efi_va_arg(argptr, efi_guid_t*);
2484                 if (!protocol)
2485                         break;
2486                 protocol_interface = efi_va_arg(argptr, void*);
2487                 r = efi_uninstall_protocol(handle, protocol,
2488                                            protocol_interface);
2489                 if (r != EFI_SUCCESS)
2490                         break;
2491                 i++;
2492         }
2493         efi_va_end(argptr);
2494         if (r == EFI_SUCCESS) {
2495                 /* If the last protocol has been removed, delete the handle. */
2496                 if (list_empty(&handle->protocols)) {
2497                         list_del(&handle->link);
2498                         free(handle);
2499                 }
2500                 return EFI_EXIT(r);
2501         }
2502
2503         /* If an error occurred undo all changes. */
2504         efi_va_start(argptr, handle);
2505         for (; i; --i) {
2506                 protocol = efi_va_arg(argptr, efi_guid_t*);
2507                 protocol_interface = efi_va_arg(argptr, void*);
2508                 EFI_CALL(efi_install_protocol_interface(&handle, protocol,
2509                                                         EFI_NATIVE_INTERFACE,
2510                                                         protocol_interface));
2511         }
2512         efi_va_end(argptr);
2513
2514         /* In case of an error always return EFI_INVALID_PARAMETER */
2515         return EFI_EXIT(EFI_INVALID_PARAMETER);
2516 }
2517
2518 /**
2519  * efi_calculate_crc32() - calculate cyclic redundancy code
2520  * @data:      buffer with data
2521  * @data_size: size of buffer in bytes
2522  * @crc32_p:   cyclic redundancy code
2523  *
2524  * This function implements the CalculateCrc32 service.
2525  *
2526  * See the Unified Extensible Firmware Interface (UEFI) specification for
2527  * details.
2528  *
2529  * Return: status code
2530  */
2531 static efi_status_t EFIAPI efi_calculate_crc32(const void *data,
2532                                                efi_uintn_t data_size,
2533                                                u32 *crc32_p)
2534 {
2535         EFI_ENTRY("%p, %zu", data, data_size);
2536         *crc32_p = crc32(0, data, data_size);
2537         return EFI_EXIT(EFI_SUCCESS);
2538 }
2539
2540 /**
2541  * efi_copy_mem() - copy memory
2542  * @destination: destination of the copy operation
2543  * @source:      source of the copy operation
2544  * @length:      number of bytes to copy
2545  *
2546  * This function implements the CopyMem service.
2547  *
2548  * See the Unified Extensible Firmware Interface (UEFI) specification for
2549  * details.
2550  */
2551 static void EFIAPI efi_copy_mem(void *destination, const void *source,
2552                                 size_t length)
2553 {
2554         EFI_ENTRY("%p, %p, %ld", destination, source, (unsigned long)length);
2555         memmove(destination, source, length);
2556         EFI_EXIT(EFI_SUCCESS);
2557 }
2558
2559 /**
2560  * efi_set_mem() - Fill memory with a byte value.
2561  * @buffer: buffer to fill
2562  * @size:   size of buffer in bytes
2563  * @value:  byte to copy to the buffer
2564  *
2565  * This function implements the SetMem service.
2566  *
2567  * See the Unified Extensible Firmware Interface (UEFI) specification for
2568  * details.
2569  */
2570 static void EFIAPI efi_set_mem(void *buffer, size_t size, uint8_t value)
2571 {
2572         EFI_ENTRY("%p, %ld, 0x%x", buffer, (unsigned long)size, value);
2573         memset(buffer, value, size);
2574         EFI_EXIT(EFI_SUCCESS);
2575 }
2576
2577 /**
2578  * efi_protocol_open() - open protocol interface on a handle
2579  * @handler:            handler of a protocol
2580  * @protocol_interface: interface implementing the protocol
2581  * @agent_handle:       handle of the driver
2582  * @controller_handle:  handle of the controller
2583  * @attributes:         attributes indicating how to open the protocol
2584  *
2585  * Return: status code
2586  */
2587 static efi_status_t efi_protocol_open(
2588                         struct efi_handler *handler,
2589                         void **protocol_interface, void *agent_handle,
2590                         void *controller_handle, uint32_t attributes)
2591 {
2592         struct efi_open_protocol_info_item *item;
2593         struct efi_open_protocol_info_entry *match = NULL;
2594         bool opened_by_driver = false;
2595         bool opened_exclusive = false;
2596
2597         /* If there is no agent, only return the interface */
2598         if (!agent_handle)
2599                 goto out;
2600
2601         /* For TEST_PROTOCOL ignore interface attribute */
2602         if (attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL)
2603                 *protocol_interface = NULL;
2604
2605         /*
2606          * Check if the protocol is already opened by a driver with the same
2607          * attributes or opened exclusively
2608          */
2609         list_for_each_entry(item, &handler->open_infos, link) {
2610                 if (item->info.agent_handle == agent_handle) {
2611                         if ((attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) &&
2612                             (item->info.attributes == attributes))
2613                                 return EFI_ALREADY_STARTED;
2614                 }
2615                 if (item->info.attributes & EFI_OPEN_PROTOCOL_EXCLUSIVE)
2616                         opened_exclusive = true;
2617         }
2618
2619         /* Only one controller can open the protocol exclusively */
2620         if (opened_exclusive && attributes &
2621             (EFI_OPEN_PROTOCOL_EXCLUSIVE | EFI_OPEN_PROTOCOL_BY_DRIVER))
2622                 return EFI_ACCESS_DENIED;
2623
2624         /* Prepare exclusive opening */
2625         if (attributes & EFI_OPEN_PROTOCOL_EXCLUSIVE) {
2626                 /* Try to disconnect controllers */
2627                 list_for_each_entry(item, &handler->open_infos, link) {
2628                         if (item->info.attributes ==
2629                                         EFI_OPEN_PROTOCOL_BY_DRIVER)
2630                                 EFI_CALL(efi_disconnect_controller(
2631                                                 item->info.controller_handle,
2632                                                 item->info.agent_handle,
2633                                                 NULL));
2634                 }
2635                 opened_by_driver = false;
2636                 /* Check if all controllers are disconnected */
2637                 list_for_each_entry(item, &handler->open_infos, link) {
2638                         if (item->info.attributes & EFI_OPEN_PROTOCOL_BY_DRIVER)
2639                                 opened_by_driver = true;
2640                 }
2641                 /* Only one controller can be connected */
2642                 if (opened_by_driver)
2643                         return EFI_ACCESS_DENIED;
2644         }
2645
2646         /* Find existing entry */
2647         list_for_each_entry(item, &handler->open_infos, link) {
2648                 if (item->info.agent_handle == agent_handle &&
2649                     item->info.controller_handle == controller_handle)
2650                         match = &item->info;
2651         }
2652         /* None found, create one */
2653         if (!match) {
2654                 match = efi_create_open_info(handler);
2655                 if (!match)
2656                         return EFI_OUT_OF_RESOURCES;
2657         }
2658
2659         match->agent_handle = agent_handle;
2660         match->controller_handle = controller_handle;
2661         match->attributes = attributes;
2662         match->open_count++;
2663
2664 out:
2665         /* For TEST_PROTOCOL ignore interface attribute. */
2666         if (attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL)
2667                 *protocol_interface = handler->protocol_interface;
2668
2669         return EFI_SUCCESS;
2670 }
2671
2672 /**
2673  * efi_open_protocol() - open protocol interface on a handle
2674  * @handle:             handle on which the protocol shall be opened
2675  * @protocol:           GUID of the protocol
2676  * @protocol_interface: interface implementing the protocol
2677  * @agent_handle:       handle of the driver
2678  * @controller_handle:  handle of the controller
2679  * @attributes:         attributes indicating how to open the protocol
2680  *
2681  * This function implements the OpenProtocol interface.
2682  *
2683  * See the Unified Extensible Firmware Interface (UEFI) specification for
2684  * details.
2685  *
2686  * Return: status code
2687  */
2688 static efi_status_t EFIAPI efi_open_protocol
2689                         (efi_handle_t handle, const efi_guid_t *protocol,
2690                          void **protocol_interface, efi_handle_t agent_handle,
2691                          efi_handle_t controller_handle, uint32_t attributes)
2692 {
2693         struct efi_handler *handler;
2694         efi_status_t r = EFI_INVALID_PARAMETER;
2695
2696         EFI_ENTRY("%p, %pUl, %p, %p, %p, 0x%x", handle, protocol,
2697                   protocol_interface, agent_handle, controller_handle,
2698                   attributes);
2699
2700         if (!handle || !protocol ||
2701             (!protocol_interface && attributes !=
2702              EFI_OPEN_PROTOCOL_TEST_PROTOCOL)) {
2703                 goto out;
2704         }
2705
2706         switch (attributes) {
2707         case EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL:
2708         case EFI_OPEN_PROTOCOL_GET_PROTOCOL:
2709         case EFI_OPEN_PROTOCOL_TEST_PROTOCOL:
2710                 break;
2711         case EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER:
2712                 if (controller_handle == handle)
2713                         goto out;
2714                 /* fall-through */
2715         case EFI_OPEN_PROTOCOL_BY_DRIVER:
2716         case EFI_OPEN_PROTOCOL_BY_DRIVER | EFI_OPEN_PROTOCOL_EXCLUSIVE:
2717                 /* Check that the controller handle is valid */
2718                 if (!efi_search_obj(controller_handle))
2719                         goto out;
2720                 /* fall-through */
2721         case EFI_OPEN_PROTOCOL_EXCLUSIVE:
2722                 /* Check that the agent handle is valid */
2723                 if (!efi_search_obj(agent_handle))
2724                         goto out;
2725                 break;
2726         default:
2727                 goto out;
2728         }
2729
2730         r = efi_search_protocol(handle, protocol, &handler);
2731         if (r != EFI_SUCCESS)
2732                 goto out;
2733
2734         r = efi_protocol_open(handler, protocol_interface, agent_handle,
2735                               controller_handle, attributes);
2736 out:
2737         return EFI_EXIT(r);
2738 }
2739
2740 /**
2741  * efi_handle_protocol() - get interface of a protocol on a handle
2742  * @handle:             handle on which the protocol shall be opened
2743  * @protocol:           GUID of the protocol
2744  * @protocol_interface: interface implementing the protocol
2745  *
2746  * This function implements the HandleProtocol service.
2747  *
2748  * See the Unified Extensible Firmware Interface (UEFI) specification for
2749  * details.
2750  *
2751  * Return: status code
2752  */
2753 static efi_status_t EFIAPI efi_handle_protocol(efi_handle_t handle,
2754                                                const efi_guid_t *protocol,
2755                                                void **protocol_interface)
2756 {
2757         return efi_open_protocol(handle, protocol, protocol_interface, NULL,
2758                                  NULL, EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL);
2759 }
2760
2761 /**
2762  * efi_bind_controller() - bind a single driver to a controller
2763  * @controller_handle:   controller handle
2764  * @driver_image_handle: driver handle
2765  * @remain_device_path:  remaining path
2766  *
2767  * Return: status code
2768  */
2769 static efi_status_t efi_bind_controller(
2770                         efi_handle_t controller_handle,
2771                         efi_handle_t driver_image_handle,
2772                         struct efi_device_path *remain_device_path)
2773 {
2774         struct efi_driver_binding_protocol *binding_protocol;
2775         efi_status_t r;
2776
2777         r = EFI_CALL(efi_open_protocol(driver_image_handle,
2778                                        &efi_guid_driver_binding_protocol,
2779                                        (void **)&binding_protocol,
2780                                        driver_image_handle, NULL,
2781                                        EFI_OPEN_PROTOCOL_GET_PROTOCOL));
2782         if (r != EFI_SUCCESS)
2783                 return r;
2784         r = EFI_CALL(binding_protocol->supported(binding_protocol,
2785                                                  controller_handle,
2786                                                  remain_device_path));
2787         if (r == EFI_SUCCESS)
2788                 r = EFI_CALL(binding_protocol->start(binding_protocol,
2789                                                      controller_handle,
2790                                                      remain_device_path));
2791         EFI_CALL(efi_close_protocol(driver_image_handle,
2792                                     &efi_guid_driver_binding_protocol,
2793                                     driver_image_handle, NULL));
2794         return r;
2795 }
2796
2797 /**
2798  * efi_connect_single_controller() - connect a single driver to a controller
2799  * @controller_handle:   controller
2800  * @driver_image_handle: driver
2801  * @remain_device_path:  remaining path
2802  *
2803  * Return: status code
2804  */
2805 static efi_status_t efi_connect_single_controller(
2806                         efi_handle_t controller_handle,
2807                         efi_handle_t *driver_image_handle,
2808                         struct efi_device_path *remain_device_path)
2809 {
2810         efi_handle_t *buffer;
2811         size_t count;
2812         size_t i;
2813         efi_status_t r;
2814         size_t connected = 0;
2815
2816         /* Get buffer with all handles with driver binding protocol */
2817         r = EFI_CALL(efi_locate_handle_buffer(BY_PROTOCOL,
2818                                               &efi_guid_driver_binding_protocol,
2819                                               NULL, &count, &buffer));
2820         if (r != EFI_SUCCESS)
2821                 return r;
2822
2823         /*  Context Override */
2824         if (driver_image_handle) {
2825                 for (; *driver_image_handle; ++driver_image_handle) {
2826                         for (i = 0; i < count; ++i) {
2827                                 if (buffer[i] == *driver_image_handle) {
2828                                         buffer[i] = NULL;
2829                                         r = efi_bind_controller(
2830                                                         controller_handle,
2831                                                         *driver_image_handle,
2832                                                         remain_device_path);
2833                                         /*
2834                                          * For drivers that do not support the
2835                                          * controller or are already connected
2836                                          * we receive an error code here.
2837                                          */
2838                                         if (r == EFI_SUCCESS)
2839                                                 ++connected;
2840                                 }
2841                         }
2842                 }
2843         }
2844
2845         /*
2846          * TODO: Some overrides are not yet implemented:
2847          * - Platform Driver Override
2848          * - Driver Family Override Search
2849          * - Bus Specific Driver Override
2850          */
2851
2852         /* Driver Binding Search */
2853         for (i = 0; i < count; ++i) {
2854                 if (buffer[i]) {
2855                         r = efi_bind_controller(controller_handle,
2856                                                 buffer[i],
2857                                                 remain_device_path);
2858                         if (r == EFI_SUCCESS)
2859                                 ++connected;
2860                 }
2861         }
2862
2863         efi_free_pool(buffer);
2864         if (!connected)
2865                 return EFI_NOT_FOUND;
2866         return EFI_SUCCESS;
2867 }
2868
2869 /**
2870  * efi_connect_controller() - connect a controller to a driver
2871  * @controller_handle:   handle of the controller
2872  * @driver_image_handle: handle of the driver
2873  * @remain_device_path:  device path of a child controller
2874  * @recursive:           true to connect all child controllers
2875  *
2876  * This function implements the ConnectController service.
2877  *
2878  * See the Unified Extensible Firmware Interface (UEFI) specification for
2879  * details.
2880  *
2881  * First all driver binding protocol handles are tried for binding drivers.
2882  * Afterwards all handles that have opened a protocol of the controller
2883  * with EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER are connected to drivers.
2884  *
2885  * Return: status code
2886  */
2887 static efi_status_t EFIAPI efi_connect_controller(
2888                         efi_handle_t controller_handle,
2889                         efi_handle_t *driver_image_handle,
2890                         struct efi_device_path *remain_device_path,
2891                         bool recursive)
2892 {
2893         efi_status_t r;
2894         efi_status_t ret = EFI_NOT_FOUND;
2895         struct efi_object *efiobj;
2896
2897         EFI_ENTRY("%p, %p, %pD, %d", controller_handle, driver_image_handle,
2898                   remain_device_path, recursive);
2899
2900         efiobj = efi_search_obj(controller_handle);
2901         if (!efiobj) {
2902                 ret = EFI_INVALID_PARAMETER;
2903                 goto out;
2904         }
2905
2906         r = efi_connect_single_controller(controller_handle,
2907                                           driver_image_handle,
2908                                           remain_device_path);
2909         if (r == EFI_SUCCESS)
2910                 ret = EFI_SUCCESS;
2911         if (recursive) {
2912                 struct efi_handler *handler;
2913                 struct efi_open_protocol_info_item *item;
2914
2915                 list_for_each_entry(handler, &efiobj->protocols, link) {
2916                         list_for_each_entry(item, &handler->open_infos, link) {
2917                                 if (item->info.attributes &
2918                                     EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) {
2919                                         r = EFI_CALL(efi_connect_controller(
2920                                                 item->info.controller_handle,
2921                                                 driver_image_handle,
2922                                                 remain_device_path,
2923                                                 recursive));
2924                                         if (r == EFI_SUCCESS)
2925                                                 ret = EFI_SUCCESS;
2926                                 }
2927                         }
2928                 }
2929         }
2930         /*  Check for child controller specified by end node */
2931         if (ret != EFI_SUCCESS && remain_device_path &&
2932             remain_device_path->type == DEVICE_PATH_TYPE_END)
2933                 ret = EFI_SUCCESS;
2934 out:
2935         return EFI_EXIT(ret);
2936 }
2937
2938 /**
2939  * efi_reinstall_protocol_interface() - reinstall protocol interface
2940  * @handle:        handle on which the protocol shall be reinstalled
2941  * @protocol:      GUID of the protocol to be installed
2942  * @old_interface: interface to be removed
2943  * @new_interface: interface to be installed
2944  *
2945  * This function implements the ReinstallProtocolInterface service.
2946  *
2947  * See the Unified Extensible Firmware Interface (UEFI) specification for
2948  * details.
2949  *
2950  * The old interface is uninstalled. The new interface is installed.
2951  * Drivers are connected.
2952  *
2953  * Return: status code
2954  */
2955 static efi_status_t EFIAPI efi_reinstall_protocol_interface(
2956                         efi_handle_t handle, const efi_guid_t *protocol,
2957                         void *old_interface, void *new_interface)
2958 {
2959         efi_status_t ret;
2960
2961         EFI_ENTRY("%p, %pUl, %p, %p", handle, protocol, old_interface,
2962                   new_interface);
2963
2964         /* Uninstall protocol but do not delete handle */
2965         ret = efi_uninstall_protocol(handle, protocol, old_interface);
2966         if (ret != EFI_SUCCESS)
2967                 goto out;
2968
2969         /* Install the new protocol */
2970         ret = efi_add_protocol(handle, protocol, new_interface);
2971         /*
2972          * The UEFI spec does not specify what should happen to the handle
2973          * if in case of an error no protocol interface remains on the handle.
2974          * So let's do nothing here.
2975          */
2976         if (ret != EFI_SUCCESS)
2977                 goto out;
2978         /*
2979          * The returned status code has to be ignored.
2980          * Do not create an error if no suitable driver for the handle exists.
2981          */
2982         EFI_CALL(efi_connect_controller(handle, NULL, NULL, true));
2983 out:
2984         return EFI_EXIT(ret);
2985 }
2986
2987 /**
2988  * efi_get_child_controllers() - get all child controllers associated to a driver
2989  * @efiobj:              handle of the controller
2990  * @driver_handle:       handle of the driver
2991  * @number_of_children:  number of child controllers
2992  * @child_handle_buffer: handles of the the child controllers
2993  *
2994  * The allocated buffer has to be freed with free().
2995  *
2996  * Return: status code
2997  */
2998 static efi_status_t efi_get_child_controllers(
2999                                 struct efi_object *efiobj,
3000                                 efi_handle_t driver_handle,
3001                                 efi_uintn_t *number_of_children,
3002                                 efi_handle_t **child_handle_buffer)
3003 {
3004         struct efi_handler *handler;
3005         struct efi_open_protocol_info_item *item;
3006         efi_uintn_t count = 0, i;
3007         bool duplicate;
3008
3009         /* Count all child controller associations */
3010         list_for_each_entry(handler, &efiobj->protocols, link) {
3011                 list_for_each_entry(item, &handler->open_infos, link) {
3012                         if (item->info.agent_handle == driver_handle &&
3013                             item->info.attributes &
3014                             EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER)
3015                                 ++count;
3016                 }
3017         }
3018         /*
3019          * Create buffer. In case of duplicate child controller assignments
3020          * the buffer will be too large. But that does not harm.
3021          */
3022         *number_of_children = 0;
3023         *child_handle_buffer = calloc(count, sizeof(efi_handle_t));
3024         if (!*child_handle_buffer)
3025                 return EFI_OUT_OF_RESOURCES;
3026         /* Copy unique child handles */
3027         list_for_each_entry(handler, &efiobj->protocols, link) {
3028                 list_for_each_entry(item, &handler->open_infos, link) {
3029                         if (item->info.agent_handle == driver_handle &&
3030                             item->info.attributes &
3031                             EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER) {
3032                                 /* Check this is a new child controller */
3033                                 duplicate = false;
3034                                 for (i = 0; i < *number_of_children; ++i) {
3035                                         if ((*child_handle_buffer)[i] ==
3036                                             item->info.controller_handle)
3037                                                 duplicate = true;
3038                                 }
3039                                 /* Copy handle to buffer */
3040                                 if (!duplicate) {
3041                                         i = (*number_of_children)++;
3042                                         (*child_handle_buffer)[i] =
3043                                                 item->info.controller_handle;
3044                                 }
3045                         }
3046                 }
3047         }
3048         return EFI_SUCCESS;
3049 }
3050
3051 /**
3052  * efi_disconnect_controller() - disconnect a controller from a driver
3053  * @controller_handle:   handle of the controller
3054  * @driver_image_handle: handle of the driver
3055  * @child_handle:        handle of the child to destroy
3056  *
3057  * This function implements the DisconnectController service.
3058  *
3059  * See the Unified Extensible Firmware Interface (UEFI) specification for
3060  * details.
3061  *
3062  * Return: status code
3063  */
3064 static efi_status_t EFIAPI efi_disconnect_controller(
3065                                 efi_handle_t controller_handle,
3066                                 efi_handle_t driver_image_handle,
3067                                 efi_handle_t child_handle)
3068 {
3069         struct efi_driver_binding_protocol *binding_protocol;
3070         efi_handle_t *child_handle_buffer = NULL;
3071         size_t number_of_children = 0;
3072         efi_status_t r;
3073         size_t stop_count = 0;
3074         struct efi_object *efiobj;
3075
3076         EFI_ENTRY("%p, %p, %p", controller_handle, driver_image_handle,
3077                   child_handle);
3078
3079         efiobj = efi_search_obj(controller_handle);
3080         if (!efiobj) {
3081                 r = EFI_INVALID_PARAMETER;
3082                 goto out;
3083         }
3084
3085         if (child_handle && !efi_search_obj(child_handle)) {
3086                 r = EFI_INVALID_PARAMETER;
3087                 goto out;
3088         }
3089
3090         /* If no driver handle is supplied, disconnect all drivers */
3091         if (!driver_image_handle) {
3092                 r = efi_disconnect_all_drivers(efiobj, NULL, child_handle);
3093                 goto out;
3094         }
3095
3096         /* Create list of child handles */
3097         if (child_handle) {
3098                 number_of_children = 1;
3099                 child_handle_buffer = &child_handle;
3100         } else {
3101                 efi_get_child_controllers(efiobj,
3102                                           driver_image_handle,
3103                                           &number_of_children,
3104                                           &child_handle_buffer);
3105         }
3106
3107         /* Get the driver binding protocol */
3108         r = EFI_CALL(efi_open_protocol(driver_image_handle,
3109                                        &efi_guid_driver_binding_protocol,
3110                                        (void **)&binding_protocol,
3111                                        driver_image_handle, NULL,
3112                                        EFI_OPEN_PROTOCOL_GET_PROTOCOL));
3113         if (r != EFI_SUCCESS)
3114                 goto out;
3115         /* Remove the children */
3116         if (number_of_children) {
3117                 r = EFI_CALL(binding_protocol->stop(binding_protocol,
3118                                                     controller_handle,
3119                                                     number_of_children,
3120                                                     child_handle_buffer));
3121                 if (r == EFI_SUCCESS)
3122                         ++stop_count;
3123         }
3124         /* Remove the driver */
3125         if (!child_handle)
3126                 r = EFI_CALL(binding_protocol->stop(binding_protocol,
3127                                                     controller_handle,
3128                                                     0, NULL));
3129         if (r == EFI_SUCCESS)
3130                 ++stop_count;
3131         EFI_CALL(efi_close_protocol(driver_image_handle,
3132                                     &efi_guid_driver_binding_protocol,
3133                                     driver_image_handle, NULL));
3134
3135         if (stop_count)
3136                 r = EFI_SUCCESS;
3137         else
3138                 r = EFI_NOT_FOUND;
3139 out:
3140         if (!child_handle)
3141                 free(child_handle_buffer);
3142         return EFI_EXIT(r);
3143 }
3144
3145 static struct efi_boot_services efi_boot_services = {
3146         .hdr = {
3147                 .signature = EFI_BOOT_SERVICES_SIGNATURE,
3148                 .revision = EFI_SPECIFICATION_VERSION,
3149                 .headersize = sizeof(struct efi_boot_services),
3150         },
3151         .raise_tpl = efi_raise_tpl,
3152         .restore_tpl = efi_restore_tpl,
3153         .allocate_pages = efi_allocate_pages_ext,
3154         .free_pages = efi_free_pages_ext,
3155         .get_memory_map = efi_get_memory_map_ext,
3156         .allocate_pool = efi_allocate_pool_ext,
3157         .free_pool = efi_free_pool_ext,
3158         .create_event = efi_create_event_ext,
3159         .set_timer = efi_set_timer_ext,
3160         .wait_for_event = efi_wait_for_event,
3161         .signal_event = efi_signal_event_ext,
3162         .close_event = efi_close_event,
3163         .check_event = efi_check_event,
3164         .install_protocol_interface = efi_install_protocol_interface,
3165         .reinstall_protocol_interface = efi_reinstall_protocol_interface,
3166         .uninstall_protocol_interface = efi_uninstall_protocol_interface,
3167         .handle_protocol = efi_handle_protocol,
3168         .reserved = NULL,
3169         .register_protocol_notify = efi_register_protocol_notify,
3170         .locate_handle = efi_locate_handle_ext,
3171         .locate_device_path = efi_locate_device_path,
3172         .install_configuration_table = efi_install_configuration_table_ext,
3173         .load_image = efi_load_image,
3174         .start_image = efi_start_image,
3175         .exit = efi_exit,
3176         .unload_image = efi_unload_image,
3177         .exit_boot_services = efi_exit_boot_services,
3178         .get_next_monotonic_count = efi_get_next_monotonic_count,
3179         .stall = efi_stall,
3180         .set_watchdog_timer = efi_set_watchdog_timer,
3181         .connect_controller = efi_connect_controller,
3182         .disconnect_controller = efi_disconnect_controller,
3183         .open_protocol = efi_open_protocol,
3184         .close_protocol = efi_close_protocol,
3185         .open_protocol_information = efi_open_protocol_information,
3186         .protocols_per_handle = efi_protocols_per_handle,
3187         .locate_handle_buffer = efi_locate_handle_buffer,
3188         .locate_protocol = efi_locate_protocol,
3189         .install_multiple_protocol_interfaces =
3190                         efi_install_multiple_protocol_interfaces,
3191         .uninstall_multiple_protocol_interfaces =
3192                         efi_uninstall_multiple_protocol_interfaces,
3193         .calculate_crc32 = efi_calculate_crc32,
3194         .copy_mem = efi_copy_mem,
3195         .set_mem = efi_set_mem,
3196         .create_event_ex = efi_create_event_ex,
3197 };
3198
3199 static u16 __efi_runtime_data firmware_vendor[] = L"Das U-Boot";
3200
3201 struct efi_system_table __efi_runtime_data systab = {
3202         .hdr = {
3203                 .signature = EFI_SYSTEM_TABLE_SIGNATURE,
3204                 .revision = EFI_SPECIFICATION_VERSION,
3205                 .headersize = sizeof(struct efi_system_table),
3206         },
3207         .fw_vendor = firmware_vendor,
3208         .fw_revision = FW_VERSION << 16 | FW_PATCHLEVEL << 8,
3209         .con_in = (void *)&efi_con_in,
3210         .con_out = (void *)&efi_con_out,
3211         .std_err = (void *)&efi_con_out,
3212         .runtime = (void *)&efi_runtime_services,
3213         .boottime = (void *)&efi_boot_services,
3214         .nr_tables = 0,
3215         .tables = NULL,
3216 };
3217
3218 /**
3219  * efi_initialize_system_table() - Initialize system table
3220  *
3221  * Return:      status code
3222  */
3223 efi_status_t efi_initialize_system_table(void)
3224 {
3225         efi_status_t ret;
3226
3227         /* Allocate configuration table array */
3228         ret = efi_allocate_pool(EFI_RUNTIME_SERVICES_DATA,
3229                                 EFI_MAX_CONFIGURATION_TABLES *
3230                                 sizeof(struct efi_configuration_table),
3231                                 (void **)&systab.tables);
3232
3233         /* Set CRC32 field in table headers */
3234         efi_update_table_header_crc32(&systab.hdr);
3235         efi_update_table_header_crc32(&efi_runtime_services.hdr);
3236         efi_update_table_header_crc32(&efi_boot_services.hdr);
3237
3238         return ret;
3239 }