1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 2008, 2009 Yahoo!, Inc.
5 * All rights reserved.
6 *
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 * 3. The names of the authors may not be used to endorse or promote
16 * products derived from this software without specific prior written
17 * permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #include <sys/param.h>
33 #include <sys/errno.h>
34 #include <sys/stat.h>
35 #include <err.h>
36 #include <fcntl.h>
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <string.h>
40 #include <unistd.h>
41 #include "mfiutil.h"
42
43 #define FLASH_BUF_SIZE (64 * 1024)
44
45 static int
display_pending_firmware(int fd)46 display_pending_firmware(int fd)
47 {
48 struct mfi_ctrl_info info;
49 struct mfi_info_component header;
50 int error;
51 u_int i;
52
53 if (mfi_ctrl_get_info(fd, &info, NULL) < 0) {
54 error = errno;
55 warn("Failed to get controller info");
56 return (error);
57 }
58
59 printf("%s Pending Firmware Images:\n", mfi_device);
60 strcpy(header.name, "Name");
61 strcpy(header.version, "Version");
62 strcpy(header.build_date, "Date");
63 strcpy(header.build_time, "Time");
64 scan_firmware(&header);
65 if (info.pending_image_component_count > 8)
66 info.pending_image_component_count = 8;
67 for (i = 0; i < info.pending_image_component_count; i++)
68 scan_firmware(&info.pending_image_component[i]);
69 display_firmware(&header, "");
70 for (i = 0; i < info.pending_image_component_count; i++)
71 display_firmware(&info.pending_image_component[i], "");
72
73 return (0);
74 }
75
76 static void
mbox_store_word(uint8_t * mbox,uint32_t val)77 mbox_store_word(uint8_t *mbox, uint32_t val)
78 {
79
80 mbox[0] = val & 0xff;
81 mbox[1] = val >> 8 & 0xff;
82 mbox[2] = val >> 16 & 0xff;
83 mbox[3] = val >> 24;
84 }
85
86 static int
flash_adapter(int ac,char ** av)87 flash_adapter(int ac, char **av)
88 {
89 struct mfi_progress dummy;
90 off_t offset;
91 size_t nread;
92 char *buf;
93 struct stat sb;
94 int error, fd, flash;
95 uint8_t mbox[4], status;
96
97 if (ac != 2) {
98 warnx("flash: Firmware file required");
99 return (EINVAL);
100 }
101
102 flash = open(av[1], O_RDONLY);
103 if (flash < 0) {
104 error = errno;
105 warn("flash: Failed to open %s", av[1]);
106 return (error);
107 }
108
109 buf = NULL;
110 fd = -1;
111
112 if (fstat(flash, &sb) < 0) {
113 error = errno;
114 warn("fstat(%s)", av[1]);
115 goto error;
116 }
117 if (sb.st_size % 1024 != 0 || sb.st_size > 0x7fffffff) {
118 warnx("Invalid flash file size");
119 error = EINVAL;
120 goto error;
121 }
122
123 fd = mfi_open(mfi_device, O_RDWR);
124 if (fd < 0) {
125 error = errno;
126 warn("mfi_open");
127 goto error;
128 }
129
130 /* First, ask the firmware to allocate space for the flash file. */
131 mbox_store_word(mbox, sb.st_size);
132 if (mfi_dcmd_command(fd, MFI_DCMD_FLASH_FW_OPEN, NULL, 0, mbox, 4,
133 &status) < 0) {
134 error = errno;
135 warn("Failed to allocate flash memory");
136 goto error;
137 }
138 if (status != MFI_STAT_OK) {
139 warnx("Failed to allocate flash memory: %s",
140 mfi_status(status));
141 error = EIO;
142 goto error;
143 }
144
145 /* Upload the file 64k at a time. */
146 buf = malloc(FLASH_BUF_SIZE);
147 if (buf == NULL) {
148 warnx("malloc failed");
149 error = ENOMEM;
150 goto error;
151 }
152 offset = 0;
153 while (sb.st_size > 0) {
154 nread = read(flash, buf, FLASH_BUF_SIZE);
155 if (nread <= 0 || nread % 1024 != 0) {
156 warnx("Bad read from flash file");
157 if (mfi_dcmd_command(fd, MFI_DCMD_FLASH_FW_CLOSE,
158 NULL, 0, NULL, 0, NULL) < 0) {
159 warn("Failed to discard flash memory");
160 }
161 error = ENXIO;
162 goto error;
163 }
164
165 mbox_store_word(mbox, offset);
166 if (mfi_dcmd_command(fd, MFI_DCMD_FLASH_FW_DOWNLOAD, buf, nread,
167 mbox, 4, &status) < 0) {
168 error = errno;
169 warn("Failed to download firmware");
170 goto error;
171 }
172 if (status != MFI_STAT_OK) {
173 warnx("Failed to download firmware: %s",
174 mfi_status(status));
175 mfi_dcmd_command(fd, MFI_DCMD_FLASH_FW_CLOSE, NULL,
176 0, NULL, 0, NULL);
177 error = ENXIO;
178 goto error;
179 }
180 sb.st_size -= nread;
181 offset += nread;
182 }
183
184 /* Kick off the flash. */
185 printf("WARNING: Firmware flash in progress, do not reboot machine... ");
186 fflush(stdout);
187 if (mfi_dcmd_command(fd, MFI_DCMD_FLASH_FW_FLASH, &dummy, sizeof(dummy),
188 NULL, 0, &status) < 0) {
189 error = errno;
190 printf("failed:\n\t%s\n", strerror(error));
191 goto error;
192 }
193 if (status != MFI_STAT_OK) {
194 printf("failed:\n\t%s\n", mfi_status(status));
195 error = ENXIO;
196 goto error;
197 }
198 printf("finished\n");
199 error = display_pending_firmware(fd);
200
201 error:
202 free(buf);
203 if (fd >= 0)
204 close(fd);
205 close(flash);
206
207 return (error);
208 }
209 MFI_COMMAND(top, flash, flash_adapter);
210