cmd: fpga: Use CMD_RET_FAILURE instead of simple 1
[oweals/u-boot.git] / cmd / fpga.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000, 2001
4  * Rich Ireland, Enterasys Networks, rireland@enterasys.com.
5  */
6
7 /*
8  *  FPGA support
9  */
10 #include <common.h>
11 #include <command.h>
12 #include <fpga.h>
13 #include <fs.h>
14 #include <malloc.h>
15
16 static long do_fpga_get_device(char *arg)
17 {
18         long dev = FPGA_INVALID_DEVICE;
19         char *devstr = env_get("fpga");
20
21         if (devstr)
22                 /* Should be strtol to handle -1 cases */
23                 dev = simple_strtol(devstr, NULL, 16);
24
25         if (dev == FPGA_INVALID_DEVICE && arg)
26                 dev = simple_strtol(arg, NULL, 16);
27
28         debug("%s: device = %ld\n", __func__, dev);
29
30         return dev;
31 }
32
33 static int do_fpga_check_params(long *dev, long *fpga_data, size_t *data_size,
34                                 cmd_tbl_t *cmdtp, int argc, char *const argv[])
35 {
36         size_t local_data_size;
37         long local_fpga_data;
38
39         debug("%s %d, %d\n", __func__, argc, cmdtp->maxargs);
40
41         if (argc != cmdtp->maxargs) {
42                 debug("fpga: incorrect parameters passed\n");
43                 return CMD_RET_USAGE;
44         }
45
46         *dev = do_fpga_get_device(argv[0]);
47
48         local_fpga_data = simple_strtol(argv[1], NULL, 16);
49         if (!local_fpga_data) {
50                 debug("fpga: zero fpga_data address\n");
51                 return CMD_RET_USAGE;
52         }
53         *fpga_data = local_fpga_data;
54
55         local_data_size = simple_strtoul(argv[2], NULL, 16);
56         if (!local_data_size) {
57                 debug("fpga: zero size\n");
58                 return CMD_RET_USAGE;
59         }
60         *data_size = local_data_size;
61
62         return 0;
63 }
64
65 /* Local defines */
66 enum {
67         FPGA_NONE = -1,
68         FPGA_LOADS,
69 };
70
71 /*
72  * Map op to supported operations.  We don't use a table since we
73  * would just have to relocate it from flash anyway.
74  */
75 static int fpga_get_op(char *opstr)
76 {
77         int op = FPGA_NONE;
78
79 #if defined(CONFIG_CMD_FPGA_LOAD_SECURE)
80         if (!strcmp("loads", opstr))
81                 op = FPGA_LOADS;
82 #endif
83
84         return op;
85 }
86
87 /* ------------------------------------------------------------------------- */
88 /* command form:
89  *   fpga <op> <device number> <data addr> <datasize>
90  * where op is 'load', 'dump', or 'info'
91  * If there is no device number field, the fpga environment variable is used.
92  * If there is no data addr field, the fpgadata environment variable is used.
93  * The info command requires no data address field.
94  */
95 int do_fpga(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
96 {
97         int op, dev = FPGA_INVALID_DEVICE;
98         size_t data_size = 0;
99         void *fpga_data = NULL;
100         int rc = FPGA_FAIL;
101 #if defined(CONFIG_CMD_FPGA_LOAD_SECURE)
102         struct fpga_secure_info fpga_sec_info;
103
104         memset(&fpga_sec_info, 0, sizeof(fpga_sec_info));
105 #endif
106
107         if (argc > 9 || argc < 2) {
108                 debug("%s: Too many or too few args (%d)\n", __func__, argc);
109                 return CMD_RET_USAGE;
110         }
111
112         op = fpga_get_op(argv[1]);
113
114         switch (op) {
115         case FPGA_NONE:
116                 printf("Unknown fpga operation \"%s\"\n", argv[1]);
117                 return CMD_RET_USAGE;
118 #if defined(CONFIG_CMD_FPGA_LOAD_SECURE)
119         case FPGA_LOADS:
120                 if (argc < 7)
121                         return CMD_RET_USAGE;
122                 if (argc == 8)
123                         fpga_sec_info.userkey_addr = (u8 *)(uintptr_t)
124                                                      simple_strtoull(argv[7],
125                                                                      NULL, 16);
126                 fpga_sec_info.encflag = (u8)simple_strtoul(argv[6], NULL, 16);
127                 fpga_sec_info.authflag = (u8)simple_strtoul(argv[5], NULL, 16);
128
129                 if (fpga_sec_info.authflag >= FPGA_NO_ENC_OR_NO_AUTH &&
130                     fpga_sec_info.encflag >= FPGA_NO_ENC_OR_NO_AUTH) {
131                         puts("ERR: Use <fpga load> for NonSecure bitstream\n");
132                         return CMD_RET_USAGE;
133                 }
134
135                 if (fpga_sec_info.encflag == FPGA_ENC_USR_KEY &&
136                     !fpga_sec_info.userkey_addr) {
137                         puts("ERR: User key not provided\n");
138                         return CMD_RET_USAGE;
139                 }
140
141                 argc = 5;
142                 break;
143 #endif
144         default:
145                 break;
146         }
147
148         switch (argc) {
149         case 5:         /* fpga <op> <dev> <data> <datasize> */
150                 data_size = simple_strtoul(argv[4], NULL, 16);
151                 if (!data_size) {
152                         puts("Zero data_size\n");
153                         return CMD_RET_USAGE;
154                 }
155         case 4:         /* fpga <op> <dev> <data> */
156                 {
157                         fpga_data = (void *)simple_strtoul(argv[3], NULL, 16);
158                         debug("*  fpga: cmdline image address = 0x%08lx\n",
159                               (ulong)fpga_data);
160                 }
161                 debug("%s: fpga_data = 0x%lx\n", __func__, (ulong)fpga_data);
162                 if (!fpga_data) {
163                         puts("Zero fpga_data address\n");
164                         return CMD_RET_USAGE;
165                 }
166         case 3:         /* fpga <op> <dev | data addr> */
167                 dev = (int)simple_strtoul(argv[2], NULL, 16);
168                 debug("%s: device = %d\n", __func__, dev);
169         }
170
171         if (dev == FPGA_INVALID_DEVICE) {
172                 puts("FPGA device not specified\n");
173                 return CMD_RET_USAGE;
174         }
175
176         switch (op) {
177 #if defined(CONFIG_CMD_FPGA_LOAD_SECURE)
178         case FPGA_LOADS:
179                 rc = fpga_loads(dev, fpga_data, data_size, &fpga_sec_info);
180                 break;
181 #endif
182
183         default:
184                 printf("Unknown operation\n");
185                 return CMD_RET_USAGE;
186         }
187         return rc;
188 }
189
190 #if defined(CONFIG_CMD_FPGA_LOADFS)
191 static int do_fpga_loadfs(cmd_tbl_t *cmdtp, int flag, int argc,
192                           char *const argv[])
193 {
194         size_t data_size = 0;
195         long fpga_data, dev;
196         int ret;
197         fpga_fs_info fpga_fsinfo;
198
199         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
200                                    cmdtp, argc, argv);
201         if (ret)
202                 return ret;
203
204         fpga_fsinfo.fstype = FS_TYPE_ANY;
205         fpga_fsinfo.blocksize = (unsigned int)simple_strtoul(argv[3], NULL, 16);
206         fpga_fsinfo.interface = argv[4];
207         fpga_fsinfo.dev_part = argv[5];
208         fpga_fsinfo.filename = argv[6];
209
210         return fpga_fsload(dev, (void *)fpga_data, data_size, &fpga_fsinfo);
211 }
212 #endif
213
214 static int do_fpga_info(cmd_tbl_t *cmdtp, int flag, int argc,
215                         char * const argv[])
216 {
217         long dev = do_fpga_get_device(argv[0]);
218
219         return fpga_info(dev);
220 }
221
222 static int do_fpga_dump(cmd_tbl_t *cmdtp, int flag, int argc,
223                         char * const argv[])
224 {
225         size_t data_size = 0;
226         long fpga_data, dev;
227         int ret;
228
229         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
230                                    cmdtp, argc, argv);
231         if (ret)
232                 return ret;
233
234         return fpga_dump(dev, (void *)fpga_data, data_size);
235 }
236
237 static int do_fpga_load(cmd_tbl_t *cmdtp, int flag, int argc,
238                         char * const argv[])
239 {
240         size_t data_size = 0;
241         long fpga_data, dev;
242         int ret;
243
244         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
245                                    cmdtp, argc, argv);
246         if (ret)
247                 return ret;
248
249         return fpga_load(dev, (void *)fpga_data, data_size, BIT_FULL);
250 }
251
252 static int do_fpga_loadb(cmd_tbl_t *cmdtp, int flag, int argc,
253                          char * const argv[])
254 {
255         size_t data_size = 0;
256         long fpga_data, dev;
257         int ret;
258
259         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
260                                    cmdtp, argc, argv);
261         if (ret)
262                 return ret;
263
264         return fpga_loadbitstream(dev, (void *)fpga_data, data_size, BIT_FULL);
265 }
266
267 #if defined(CONFIG_CMD_FPGA_LOADP)
268 static int do_fpga_loadp(cmd_tbl_t *cmdtp, int flag, int argc,
269                          char * const argv[])
270 {
271         size_t data_size = 0;
272         long fpga_data, dev;
273         int ret;
274
275         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
276                                    cmdtp, argc, argv);
277         if (ret)
278                 return ret;
279
280         return fpga_load(dev, (void *)fpga_data, data_size, BIT_PARTIAL);
281 }
282 #endif
283
284 #if defined(CONFIG_CMD_FPGA_LOADBP)
285 static int do_fpga_loadbp(cmd_tbl_t *cmdtp, int flag, int argc,
286                           char * const argv[])
287 {
288         size_t data_size = 0;
289         long fpga_data, dev;
290         int ret;
291
292         ret = do_fpga_check_params(&dev, &fpga_data, &data_size,
293                                    cmdtp, argc, argv);
294         if (ret)
295                 return ret;
296
297         return fpga_loadbitstream(dev, (void *)fpga_data, data_size,
298                                   BIT_PARTIAL);
299 }
300 #endif
301
302 #if defined(CONFIG_CMD_FPGA_LOADMK)
303 static int do_fpga_loadmk(cmd_tbl_t *cmdtp, int flag, int argc,
304                           char * const argv[])
305 {
306         size_t data_size = 0;
307         void *fpga_data = NULL;
308 #if defined(CONFIG_FIT)
309         const char *fit_uname = NULL;
310         ulong fit_addr;
311 #endif
312         ulong dev = do_fpga_get_device(argv[0]);
313         char *datastr = env_get("fpgadata");
314
315         debug("fpga: argc %x, dev %lx, datastr %s\n", argc, dev, datastr);
316
317         if (dev == FPGA_INVALID_DEVICE) {
318                 debug("fpga: Invalid fpga device\n");
319                 return CMD_RET_USAGE;
320         }
321
322         if (argc == 0 && !datastr) {
323                 debug("fpga: No datastr passed\n");
324                 return CMD_RET_USAGE;
325         }
326
327         if (argc == 2) {
328                 datastr = argv[1];
329                 debug("fpga: Full command with two args\n");
330         } else if (argc == 1 && !datastr) {
331                 debug("fpga: Dev is setup - fpgadata passed\n");
332                 datastr = argv[0];
333         }
334
335 #if defined(CONFIG_FIT)
336         if (fit_parse_subimage(datastr, (ulong)fpga_data,
337                                &fit_addr, &fit_uname)) {
338                 fpga_data = (void *)fit_addr;
339                 debug("*  fpga: subimage '%s' from FIT image ",
340                       fit_uname);
341                 debug("at 0x%08lx\n", fit_addr);
342         } else
343 #endif
344         {
345                 fpga_data = (void *)simple_strtoul(datastr, NULL, 16);
346                 debug("*  fpga: cmdline image address = 0x%08lx\n",
347                       (ulong)fpga_data);
348         }
349         debug("%s: fpga_data = 0x%lx\n", __func__, (ulong)fpga_data);
350         if (!fpga_data) {
351                 puts("Zero fpga_data address\n");
352                 return CMD_RET_USAGE;
353         }
354
355         switch (genimg_get_format(fpga_data)) {
356 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
357         case IMAGE_FORMAT_LEGACY:
358         {
359                 image_header_t *hdr = (image_header_t *)fpga_data;
360                 ulong data;
361                 u8 comp;
362
363                 comp = image_get_comp(hdr);
364                 if (comp == IH_COMP_GZIP) {
365 #if defined(CONFIG_GZIP)
366                         ulong image_buf = image_get_data(hdr);
367                         ulong image_size = ~0UL;
368
369                         data = image_get_load(hdr);
370
371                         if (gunzip((void *)data, ~0UL, (void *)image_buf,
372                                    &image_size) != 0) {
373                                 puts("GUNZIP: error\n");
374                                 return CMD_RET_FAILURE;
375                         }
376                         data_size = image_size;
377 #else
378                         puts("Gunzip image is not supported\n");
379                         return 1;
380 #endif
381                 } else {
382                         data = (ulong)image_get_data(hdr);
383                         data_size = image_get_data_size(hdr);
384                 }
385                 return fpga_load(dev, (void *)data, data_size,
386                                   BIT_FULL);
387         }
388 #endif
389 #if defined(CONFIG_FIT)
390         case IMAGE_FORMAT_FIT:
391         {
392                 const void *fit_hdr = (const void *)fpga_data;
393                 int noffset;
394                 const void *fit_data;
395
396                 if (!fit_uname) {
397                         puts("No FIT subimage unit name\n");
398                         return CMD_RET_FAILURE;
399                 }
400
401                 if (!fit_check_format(fit_hdr)) {
402                         puts("Bad FIT image format\n");
403                         return CMD_RET_FAILURE;
404                 }
405
406                 /* get fpga component image node offset */
407                 noffset = fit_image_get_node(fit_hdr, fit_uname);
408                 if (noffset < 0) {
409                         printf("Can't find '%s' FIT subimage\n", fit_uname);
410                         return CMD_RET_FAILURE;
411                 }
412
413                 /* verify integrity */
414                 if (!fit_image_verify(fit_hdr, noffset)) {
415                         puts("Bad Data Hash\n");
416                         return CMD_RET_FAILURE;
417                 }
418
419                 /* get fpga subimage data address and length */
420                 if (fit_image_get_data(fit_hdr, noffset, &fit_data,
421                                        &data_size)) {
422                         puts("Fpga subimage data not found\n");
423                         return CMD_RET_FAILURE;
424                 }
425
426                 return fpga_load(dev, fit_data, data_size, BIT_FULL);
427         }
428 #endif
429         default:
430                 puts("** Unknown image type\n");
431                 return CMD_RET_FAILURE;
432         }
433 }
434 #endif
435
436 static cmd_tbl_t fpga_commands[] = {
437         U_BOOT_CMD_MKENT(info, 1, 1, do_fpga_info, "", ""),
438         U_BOOT_CMD_MKENT(dump, 3, 1, do_fpga_dump, "", ""),
439         U_BOOT_CMD_MKENT(load, 3, 1, do_fpga_load, "", ""),
440         U_BOOT_CMD_MKENT(loadb, 3, 1, do_fpga_loadb, "", ""),
441 #if defined(CONFIG_CMD_FPGA_LOADP)
442         U_BOOT_CMD_MKENT(loadp, 3, 1, do_fpga_loadp, "", ""),
443 #endif
444 #if defined(CONFIG_CMD_FPGA_LOADBP)
445         U_BOOT_CMD_MKENT(loadbp, 3, 1, do_fpga_loadbp, "", ""),
446 #endif
447 #if defined(CONFIG_CMD_FPGA_LOADFS)
448         U_BOOT_CMD_MKENT(loadfs, 7, 1, do_fpga_loadfs, "", ""),
449 #endif
450 #if defined(CONFIG_CMD_FPGA_LOADMK)
451         U_BOOT_CMD_MKENT(loadmk, 2, 1, do_fpga_loadmk, "", ""),
452 #endif
453 };
454
455 static int do_fpga_wrapper(cmd_tbl_t *cmdtp, int flag, int argc,
456                            char *const argv[])
457 {
458         cmd_tbl_t *fpga_cmd;
459         int ret;
460
461         if (argc < 2)
462                 return CMD_RET_USAGE;
463
464         fpga_cmd = find_cmd_tbl(argv[1], fpga_commands,
465                                 ARRAY_SIZE(fpga_commands));
466
467         /* This should be removed when all functions are converted */
468         if (!fpga_cmd)
469                 return do_fpga(cmdtp, flag, argc, argv);
470
471         /* FIXME This can't be reached till all functions are converted */
472         if (!fpga_cmd) {
473                 debug("fpga: non existing command\n");
474                 return CMD_RET_USAGE;
475         }
476
477         argc -= 2;
478         argv += 2;
479
480         if (argc > fpga_cmd->maxargs) {
481                 debug("fpga: more parameters passed\n");
482                 return CMD_RET_USAGE;
483         }
484
485         ret = fpga_cmd->cmd(fpga_cmd, flag, argc, argv);
486
487         return cmd_process_error(fpga_cmd, ret);
488 }
489
490 #if defined(CONFIG_CMD_FPGA_LOADFS) || defined(CONFIG_CMD_FPGA_LOAD_SECURE)
491 U_BOOT_CMD(fpga, 9, 1, do_fpga_wrapper,
492 #else
493 U_BOOT_CMD(fpga, 6, 1, do_fpga_wrapper,
494 #endif
495            "loadable FPGA image support",
496            "[operation type] [device number] [image address] [image size]\n"
497            "fpga operations:\n"
498            "  dump\t[dev] [address] [size]\tLoad device to memory buffer\n"
499            "  info\t[dev]\t\t\tlist known device information\n"
500            "  load\t[dev] [address] [size]\tLoad device from memory buffer\n"
501 #if defined(CONFIG_CMD_FPGA_LOADP)
502            "  loadp\t[dev] [address] [size]\t"
503            "Load device from memory buffer with partial bitstream\n"
504 #endif
505            "  loadb\t[dev] [address] [size]\t"
506            "Load device from bitstream buffer (Xilinx only)\n"
507 #if defined(CONFIG_CMD_FPGA_LOADBP)
508            "  loadbp\t[dev] [address] [size]\t"
509            "Load device from bitstream buffer with partial bitstream"
510            "(Xilinx only)\n"
511 #endif
512 #if defined(CONFIG_CMD_FPGA_LOADFS)
513            "Load device from filesystem (FAT by default) (Xilinx only)\n"
514            "  loadfs [dev] [address] [image size] [blocksize] <interface>\n"
515            "        [<dev[:part]>] <filename>\n"
516 #endif
517 #if defined(CONFIG_CMD_FPGA_LOADMK)
518            "  loadmk [dev] [address]\tLoad device generated with mkimage"
519 #if defined(CONFIG_FIT)
520            "\n"
521            "\tFor loadmk operating on FIT format uImage address must include\n"
522            "\tsubimage unit name in the form of addr:<subimg_uname>"
523 #endif
524 #endif
525 #if defined(CONFIG_CMD_FPGA_LOAD_SECURE)
526            "Load encrypted bitstream (Xilinx only)\n"
527            "  loads [dev] [address] [size] [auth-OCM-0/DDR-1/noauth-2]\n"
528            "        [enc-devkey(0)/userkey(1)/nenc(2) [Userkey address]\n"
529            "Loads the secure bistreams(authenticated/encrypted/both\n"
530            "authenticated and encrypted) of [size] from [address].\n"
531            "The auth-OCM/DDR flag specifies to perform authentication\n"
532            "in OCM or in DDR. 0 for OCM, 1 for DDR, 2 for no authentication.\n"
533            "The enc flag specifies which key to be used for decryption\n"
534            "0-device key, 1-user key, 2-no encryption.\n"
535            "The optional Userkey address specifies from which address key\n"
536            "has to be used for decryption if user key is selected.\n"
537            "NOTE: the sceure bitstream has to be created using xilinx\n"
538            "bootgen tool only.\n"
539 #endif
540 );