| // SPDX-License-Identifier: GPL-2.0+ |
| /* |
| * (c) Copyright 2019 by Google, Inc |
| * |
| * Author: |
| * David Anderson <dvander@google.com> |
| */ |
| |
| #include <common.h> |
| |
| #include <command.h> |
| #include <dm/device.h> |
| #include <environment.h> |
| #include <linux/stddef.h> |
| #include <malloc.h> |
| #include <memalign.h> |
| #include <search.h> |
| #include <errno.h> |
| #include <mmc.h> |
| #include <virtio_types.h> |
| #include <virtio.h> |
| |
| #ifdef CONFIG_CMD_SAVEENV |
| static int env_raw_disk_save(void) |
| { |
| ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1); |
| struct blk_desc *dev_desc = NULL; |
| disk_partition_t info; |
| uint blk_cnt, n; |
| int part, err; |
| |
| err = env_export(env_new); |
| if (err) |
| return err; |
| |
| part = blk_get_device_part_str(CONFIG_ENV_RAW_DISK_INTERFACE, |
| CONFIG_ENV_RAW_DISK_DEVICE_AND_PART, |
| &dev_desc, &info, 1); |
| if (part < 0) |
| return 1; |
| |
| printf("Writing to disk..."); |
| |
| blk_cnt = ALIGN(CONFIG_ENV_SIZE, info.blksz) / info.blksz; |
| n = blk_dwrite(dev_desc, info.start, blk_cnt, (u_char *)env_new); |
| if (n != blk_cnt) { |
| puts("failed\n"); |
| return 1; |
| } |
| |
| puts("done\n"); |
| return 0; |
| } |
| #endif /* CONFIG_CMD_SAVEENV */ |
| |
| static int env_raw_disk_load(void) |
| { |
| ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE); |
| struct blk_desc *dev_desc = NULL; |
| disk_partition_t info; |
| uint blk_cnt, n; |
| int part; |
| |
| #ifdef CONFIG_MMC |
| if (!strcmp(CONFIG_ENV_RAW_DISK_INTERFACE, "mmc")) |
| mmc_initialize(NULL); |
| #endif |
| #ifdef CONFIG_VIRTIO |
| if (!strcmp(CONFIG_ENV_RAW_DISK_INTERFACE, "virtio")) |
| virtio_init(); |
| #endif |
| |
| part = blk_get_device_part_str(CONFIG_ENV_RAW_DISK_INTERFACE, |
| CONFIG_ENV_RAW_DISK_DEVICE_AND_PART, |
| &dev_desc, &info, 1); |
| if (part < 0) { |
| set_default_env(NULL, 0); |
| return -EINVAL; |
| } |
| |
| blk_cnt = ALIGN(CONFIG_ENV_SIZE, info.blksz) / info.blksz; |
| n = blk_dread(dev_desc, info.start, blk_cnt, buf); |
| if (n != blk_cnt) { |
| set_default_env(NULL, 0); |
| return -EIO; |
| } |
| |
| return env_import(buf, 1); |
| } |
| |
| U_BOOT_ENV_LOCATION(raw_disk) = { |
| .location = ENVL_RAW_DISK, |
| ENV_NAME("Raw Disk") |
| .load = env_raw_disk_load, |
| #ifdef CONFIG_CMD_SAVEENV |
| .save = env_save_ptr(env_raw_disk_save), |
| #endif |
| }; |
| |