Skip to content
Snippets Groups Projects
efi_boottime.c 63.4 KiB
Newer Older
  • Learn to ignore specific revisions
  • 		*device = handles[i];
    		len_best = len_dp;
    	}
    	if (len_best) {
    		remainder = (u8 *)*device_path + len_best;
    		*device_path = (struct efi_device_path *)remainder;
    		ret = EFI_SUCCESS;
    	} else {
    		ret = EFI_NOT_FOUND;
    	}
    out:
    	return EFI_EXIT(ret);
    }
    
    
    /*
     * Install multiple protocol interfaces.
     *
     * This function implements the MultipleProtocolInterfaces service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @handle	handle on which the protocol interfaces shall be installed
     * @...		NULL terminated argument list with pairs of protocol GUIDS and
     *		interfaces
     * @return	status code
     */
    
    static efi_status_t EFIAPI efi_install_multiple_protocol_interfaces(
    			void **handle, ...)
    {
    	EFI_ENTRY("%p", handle);
    
    	const efi_guid_t *protocol;
    
    	void *protocol_interface;
    	efi_status_t r = EFI_SUCCESS;
    	int i = 0;
    
    	if (!handle)
    		return EFI_EXIT(EFI_INVALID_PARAMETER);
    
    	va_start(argptr, handle);
    	for (;;) {
    		protocol = va_arg(argptr, efi_guid_t*);
    		if (!protocol)
    			break;
    		protocol_interface = va_arg(argptr, void*);
    
    		r = EFI_CALL(efi_install_protocol_interface(
    						handle, protocol,
    						EFI_NATIVE_INTERFACE,
    						protocol_interface));
    
    		if (r != EFI_SUCCESS)
    			break;
    		i++;
    	}
    	va_end(argptr);
    	if (r == EFI_SUCCESS)
    		return EFI_EXIT(r);
    
    
    	/* If an error occurred undo all changes. */
    
    	va_start(argptr, handle);
    	for (; i; --i) {
    		protocol = va_arg(argptr, efi_guid_t*);
    		protocol_interface = va_arg(argptr, void*);
    
    		EFI_CALL(efi_uninstall_protocol_interface(handle, protocol,
    							  protocol_interface));
    
    /*
     * Uninstall multiple protocol interfaces.
     *
     * This function implements the UninstallMultipleProtocolInterfaces service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @handle	handle from which the protocol interfaces shall be removed
     * @...		NULL terminated argument list with pairs of protocol GUIDS and
     *		interfaces
     * @return	status code
     */
    
    static efi_status_t EFIAPI efi_uninstall_multiple_protocol_interfaces(
    			void *handle, ...)
    {
    	EFI_ENTRY("%p", handle);
    
    
    	va_list argptr;
    	const efi_guid_t *protocol;
    	void *protocol_interface;
    	efi_status_t r = EFI_SUCCESS;
    	size_t i = 0;
    
    	if (!handle)
    		return EFI_EXIT(EFI_INVALID_PARAMETER);
    
    	va_start(argptr, handle);
    	for (;;) {
    		protocol = va_arg(argptr, efi_guid_t*);
    		if (!protocol)
    			break;
    		protocol_interface = va_arg(argptr, void*);
    		r = EFI_CALL(efi_uninstall_protocol_interface(
    						handle, protocol,
    						protocol_interface));
    		if (r != EFI_SUCCESS)
    			break;
    		i++;
    	}
    	va_end(argptr);
    	if (r == EFI_SUCCESS)
    		return EFI_EXIT(r);
    
    	/* If an error occurred undo all changes. */
    	va_start(argptr, handle);
    	for (; i; --i) {
    		protocol = va_arg(argptr, efi_guid_t*);
    		protocol_interface = va_arg(argptr, void*);
    		EFI_CALL(efi_install_protocol_interface(&handle, protocol,
    							EFI_NATIVE_INTERFACE,
    							protocol_interface));
    	}
    	va_end(argptr);
    
    	return EFI_EXIT(r);
    
    /*
     * Calculate cyclic redundancy code.
     *
     * This function implements the CalculateCrc32 service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @data	buffer with data
     * @data_size	size of buffer in bytes
     * @crc32_p	cyclic redundancy code
     * @return	status code
     */
    
    static efi_status_t EFIAPI efi_calculate_crc32(void *data,
    					       unsigned long data_size,
    					       uint32_t *crc32_p)
    {
    	EFI_ENTRY("%p, %ld", data, data_size);
    	*crc32_p = crc32(0, data, data_size);
    	return EFI_EXIT(EFI_SUCCESS);
    }
    
    
    /*
     * Copy memory.
     *
     * This function implements the CopyMem service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @destination		destination of the copy operation
     * @source		source of the copy operation
     * @length		number of bytes to copy
     */
    
    static void EFIAPI efi_copy_mem(void *destination, const void *source,
    				size_t length)
    
    	EFI_ENTRY("%p, %p, %ld", destination, source, (unsigned long)length);
    
    	memcpy(destination, source, length);
    
    /*
     * Fill memory with a byte value.
     *
     * This function implements the SetMem service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @buffer		buffer to fill
     * @size		size of buffer in bytes
     * @value		byte to copy to the buffer
     */
    
    static void EFIAPI efi_set_mem(void *buffer, size_t size, uint8_t value)
    
    	EFI_ENTRY("%p, %ld, 0x%x", buffer, (unsigned long)size, value);
    
    	memset(buffer, value, size);
    
    /*
     * Open protocol interface on a handle.
     *
     * @handler		handler of a protocol
     * @protocol_interface	interface implementing the protocol
     * @agent_handle	handle of the driver
     * @controller_handle	handle of the controller
     * @attributes		attributes indicating how to open the protocol
     * @return		status code
     */
    static efi_status_t efi_protocol_open(
    			struct efi_handler *handler,
    			void **protocol_interface, void *agent_handle,
    			void *controller_handle, uint32_t attributes)
    {
    	struct efi_open_protocol_info_item *item;
    	struct efi_open_protocol_info_entry *match = NULL;
    	bool opened_by_driver = false;
    	bool opened_exclusive = false;
    
    	/* If there is no agent, only return the interface */
    	if (!agent_handle)
    		goto out;
    
    	/* For TEST_PROTOCOL ignore interface attribute */
    	if (attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL)
    		*protocol_interface = NULL;
    
    	/*
    	 * Check if the protocol is already opened by a driver with the same
    	 * attributes or opened exclusively
    	 */
    	list_for_each_entry(item, &handler->open_infos, link) {
    		if (item->info.agent_handle == agent_handle) {
    			if ((attributes & EFI_OPEN_PROTOCOL_BY_DRIVER) &&
    			    (item->info.attributes == attributes))
    				return EFI_ALREADY_STARTED;
    		}
    		if (item->info.attributes & EFI_OPEN_PROTOCOL_EXCLUSIVE)
    			opened_exclusive = true;
    	}
    
    	/* Only one controller can open the protocol exclusively */
    	if (opened_exclusive && attributes &
    	    (EFI_OPEN_PROTOCOL_EXCLUSIVE | EFI_OPEN_PROTOCOL_BY_DRIVER))
    		return EFI_ACCESS_DENIED;
    
    	/* Prepare exclusive opening */
    	if (attributes & EFI_OPEN_PROTOCOL_EXCLUSIVE) {
    		/* Try to disconnect controllers */
    		list_for_each_entry(item, &handler->open_infos, link) {
    			if (item->info.attributes ==
    					EFI_OPEN_PROTOCOL_BY_DRIVER)
    				EFI_CALL(efi_disconnect_controller(
    						item->info.controller_handle,
    						item->info.agent_handle,
    						NULL));
    		}
    		opened_by_driver = false;
    		/* Check if all controllers are disconnected */
    		list_for_each_entry(item, &handler->open_infos, link) {
    			if (item->info.attributes & EFI_OPEN_PROTOCOL_BY_DRIVER)
    				opened_by_driver = true;
    		}
    		/* Only one controller can be conncected */
    		if (opened_by_driver)
    			return EFI_ACCESS_DENIED;
    	}
    
    	/* Find existing entry */
    	list_for_each_entry(item, &handler->open_infos, link) {
    		if (item->info.agent_handle == agent_handle &&
    		    item->info.controller_handle == controller_handle)
    			match = &item->info;
    	}
    	/* None found, create one */
    	if (!match) {
    		match = efi_create_open_info(handler);
    		if (!match)
    			return EFI_OUT_OF_RESOURCES;
    	}
    
    	match->agent_handle = agent_handle;
    	match->controller_handle = controller_handle;
    	match->attributes = attributes;
    	match->open_count++;
    
    out:
    	/* For TEST_PROTOCOL ignore interface attribute. */
    	if (attributes != EFI_OPEN_PROTOCOL_TEST_PROTOCOL)
    		*protocol_interface = handler->protocol_interface;
    
    	return EFI_SUCCESS;
    }
    
    
    /*
     * Open protocol interface on a handle.
     *
     * This function implements the OpenProtocol interface.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @handle		handle on which the protocol shall be opened
     * @protocol		GUID of the protocol
     * @protocol_interface	interface implementing the protocol
     * @agent_handle	handle of the driver
     * @controller_handle	handle of the controller
     * @attributes		attributes indicating how to open the protocol
     * @return		status code
     */
    
    static efi_status_t EFIAPI efi_open_protocol(
    
    			void *handle, const efi_guid_t *protocol,
    
    			void **protocol_interface, void *agent_handle,
    			void *controller_handle, uint32_t attributes)
    {
    
    	struct efi_handler *handler;
    
    	efi_status_t r = EFI_INVALID_PARAMETER;
    
    Rob Clark's avatar
    Rob Clark committed
    	EFI_ENTRY("%p, %pUl, %p, %p, %p, 0x%x", handle, protocol,
    
    		  protocol_interface, agent_handle, controller_handle,
    		  attributes);
    
    	if (!handle || !protocol ||
    	    (!protocol_interface && attributes !=
    	     EFI_OPEN_PROTOCOL_TEST_PROTOCOL)) {
    		goto out;
    	}
    
    	switch (attributes) {
    	case EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL:
    	case EFI_OPEN_PROTOCOL_GET_PROTOCOL:
    	case EFI_OPEN_PROTOCOL_TEST_PROTOCOL:
    		break;
    	case EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER:
    		if (controller_handle == handle)
    			goto out;
    
    		/* fall-through */
    
    	case EFI_OPEN_PROTOCOL_BY_DRIVER:
    	case EFI_OPEN_PROTOCOL_BY_DRIVER | EFI_OPEN_PROTOCOL_EXCLUSIVE:
    
    		/* Check that the controller handle is valid */
    		if (!efi_search_obj(controller_handle))
    
    		/* fall-through */
    
    	case EFI_OPEN_PROTOCOL_EXCLUSIVE:
    
    		/* Check that the agent handle is valid */
    		if (!efi_search_obj(agent_handle))
    
    	r = efi_search_protocol(handle, protocol, &handler);
    	if (r != EFI_SUCCESS)
    		goto out;
    
    	r = efi_protocol_open(handler, protocol_interface, agent_handle,
    			      controller_handle, attributes);
    
    out:
    	return EFI_EXIT(r);
    }
    
    
    /*
     * Get interface of a protocol on a handle.
     *
     * This function implements the HandleProtocol service.
     * See the Unified Extensible Firmware Interface (UEFI) specification
     * for details.
     *
     * @handle		handle on which the protocol shall be opened
     * @protocol		GUID of the protocol
     * @protocol_interface  interface implementing the protocol
     * @return		status code
     */
    
    static efi_status_t EFIAPI efi_handle_protocol(void *handle,
    
    					       const efi_guid_t *protocol,
    
    					       void **protocol_interface)
    {
    
    	return efi_open_protocol(handle, protocol, protocol_interface, NULL,
    				 NULL, EFI_OPEN_PROTOCOL_BY_HANDLE_PROTOCOL);
    
    }
    
    static const struct efi_boot_services efi_boot_services = {
    	.hdr = {
    		.headersize = sizeof(struct efi_table_hdr),
    	},
    	.raise_tpl = efi_raise_tpl,
    	.restore_tpl = efi_restore_tpl,
    	.allocate_pages = efi_allocate_pages_ext,
    	.free_pages = efi_free_pages_ext,
    	.get_memory_map = efi_get_memory_map_ext,
    
    	.allocate_pool = efi_allocate_pool_ext,
    
    	.free_pool = efi_free_pool_ext,
    
    	.create_event = efi_create_event_ext,
    
    	.set_timer = efi_set_timer_ext,
    
    	.wait_for_event = efi_wait_for_event,
    
    	.signal_event = efi_signal_event_ext,
    
    	.close_event = efi_close_event,
    	.check_event = efi_check_event,
    
    	.install_protocol_interface = efi_install_protocol_interface,
    
    	.reinstall_protocol_interface = efi_reinstall_protocol_interface,
    
    	.uninstall_protocol_interface = efi_uninstall_protocol_interface,
    
    	.handle_protocol = efi_handle_protocol,
    	.reserved = NULL,
    	.register_protocol_notify = efi_register_protocol_notify,
    
    	.locate_handle = efi_locate_handle_ext,
    
    	.locate_device_path = efi_locate_device_path,
    
    	.install_configuration_table = efi_install_configuration_table_ext,
    
    	.load_image = efi_load_image,
    	.start_image = efi_start_image,
    
    	.exit = efi_exit,
    
    	.unload_image = efi_unload_image,
    	.exit_boot_services = efi_exit_boot_services,
    	.get_next_monotonic_count = efi_get_next_monotonic_count,
    	.stall = efi_stall,
    	.set_watchdog_timer = efi_set_watchdog_timer,
    	.connect_controller = efi_connect_controller,
    	.disconnect_controller = efi_disconnect_controller,
    	.open_protocol = efi_open_protocol,
    	.close_protocol = efi_close_protocol,
    	.open_protocol_information = efi_open_protocol_information,
    	.protocols_per_handle = efi_protocols_per_handle,
    	.locate_handle_buffer = efi_locate_handle_buffer,
    	.locate_protocol = efi_locate_protocol,
    	.install_multiple_protocol_interfaces = efi_install_multiple_protocol_interfaces,
    	.uninstall_multiple_protocol_interfaces = efi_uninstall_multiple_protocol_interfaces,
    	.calculate_crc32 = efi_calculate_crc32,
    	.copy_mem = efi_copy_mem,
    	.set_mem = efi_set_mem,
    };
    
    
    
    static uint16_t __efi_runtime_data firmware_vendor[] = L"Das U-Boot";
    
    struct efi_system_table __efi_runtime_data systab = {
    
    	.hdr = {
    		.signature = EFI_SYSTEM_TABLE_SIGNATURE,
    		.revision = 0x20005, /* 2.5 */
    		.headersize = sizeof(struct efi_table_hdr),
    	},
    	.fw_vendor = (long)firmware_vendor,
    	.con_in = (void*)&efi_con_in,
    	.con_out = (void*)&efi_con_out,
    	.std_err = (void*)&efi_con_out,
    	.runtime = (void*)&efi_runtime_services,
    	.boottime = (void*)&efi_boot_services,
    	.nr_tables = 0,
    	.tables = (void*)efi_conf_table,
    };