copy.c revision 293301 
1/*-
2 * Copyright (c) 2013 The FreeBSD Foundation
3 * All rights reserved.
4 *
5 * This software was developed by Benno Rice under sponsorship from
6 * the FreeBSD Foundation.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30__FBSDID("$FreeBSD: stable/10/sys/boot/efi/loader/copy.c 293301 2016-01-07 02:42:56Z emaste $");
31
32#include <sys/param.h>
33
34#include <stand.h>
35#include <bootstrap.h>
36
37#include <efi.h>
38#include <efilib.h>
39
40#ifndef EFI_STAGING_SIZE
41#define	EFI_STAGING_SIZE	48
42#endif
43
44#define	STAGE_PAGES	EFI_SIZE_TO_PAGES((EFI_STAGING_SIZE) * 1024 * 1024)
45
46EFI_PHYSICAL_ADDRESS	staging, staging_end;
47int			stage_offset_set = 0;
48ssize_t			stage_offset;
49
50int
51efi_copy_init(void)
52{
53	EFI_STATUS	status;
54
55	status = BS->AllocatePages(AllocateAnyPages, EfiLoaderData,
56	    STAGE_PAGES, &staging);
57	if (EFI_ERROR(status)) {
58		printf("failed to allocate staging area: %lu\n",
59		    (unsigned long)(status & EFI_ERROR_MASK));
60		return (status);
61	}
62	staging_end = staging + STAGE_PAGES * EFI_PAGE_SIZE;
63
64	return (0);
65}
66
67ssize_t
68efi_copyin(const void *src, vm_offset_t dest, const size_t len)
69{
70
71	if (!stage_offset_set) {
72		stage_offset = (vm_offset_t)staging - dest;
73		stage_offset_set = 1;
74	}
75
76	/* XXX: Callers do not check for failure. */
77	if (dest + stage_offset + len > staging_end) {
78		errno = ENOMEM;
79		return (-1);
80	}
81	bcopy(src, (void *)(dest + stage_offset), len);
82	return (len);
83}
84
85ssize_t
86efi_copyout(const vm_offset_t src, void *dest, const size_t len)
87{
88
89	/* XXX: Callers do not check for failure. */
90	if (src + stage_offset + len > staging_end) {
91		errno = ENOMEM;
92		return (-1);
93	}
94	bcopy((void *)(src + stage_offset), dest, len);
95	return (len);
96}
97
98
99ssize_t
100efi_readin(const int fd, vm_offset_t dest, const size_t len)
101{
102
103	if (dest + stage_offset + len > staging_end) {
104		errno = ENOMEM;
105		return (-1);
106	}
107	return (read(fd, (void *)(dest + stage_offset), len));
108}
109
110void
111efi_copy_finish(void)
112{
113	uint64_t	*src, *dst, *last;
114
115	src = (uint64_t *)staging;
116	dst = (uint64_t *)(staging - stage_offset);
117	last = (uint64_t *)staging_end;
118
119	while (src < last)
120		*dst++ = *src++;
121}
122