1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (c) Copyright 2016 by VRT Technology
4  *
5  * Author:
6  *  Stuart Longland <stuartl@vrt.com.au>
7  *
8  * Based on FAT environment driver
9  * (c) Copyright 2011 by Tigris Elektronik GmbH
10  *
11  * Author:
12  *  Maximilian Schwerin <mvs@tigris.de>
13  *
14  * and EXT4 filesystem implementation
15  * (C) Copyright 2011 - 2012 Samsung Electronics
16  * EXT4 filesystem implementation in Uboot by
17  * Uma Shankar <uma.shankar@samsung.com>
18  * Manjunatha C Achar <a.manjunatha@samsung.com>
19  */
20 
21 #include <common.h>
22 #include <part.h>
23 
24 #include <command.h>
25 #include <env.h>
26 #include <env_internal.h>
27 #include <linux/stddef.h>
28 #include <malloc.h>
29 #include <memalign.h>
30 #include <search.h>
31 #include <errno.h>
32 #include <ext4fs.h>
33 #include <mmc.h>
34 #include <scsi.h>
35 #include <asm/global_data.h>
36 
37 DECLARE_GLOBAL_DATA_PTR;
38 
env_ext4_get_intf(void)39 __weak const char *env_ext4_get_intf(void)
40 {
41 	return (const char *)CONFIG_ENV_EXT4_INTERFACE;
42 }
43 
env_ext4_get_dev_part(void)44 __weak const char *env_ext4_get_dev_part(void)
45 {
46 #ifdef CONFIG_MMC
47 	static char *part_str;
48 
49 	if (!part_str) {
50 		part_str = CONFIG_ENV_EXT4_DEVICE_AND_PART;
51 		if (!strcmp(CONFIG_ENV_EXT4_INTERFACE, "mmc") && part_str[0] == ':') {
52 			part_str = "0" CONFIG_ENV_EXT4_DEVICE_AND_PART;
53 			part_str[0] += mmc_get_env_dev();
54 		}
55 	}
56 
57 	return part_str;
58 #else
59 	return (const char *)CONFIG_ENV_EXT4_DEVICE_AND_PART;
60 #endif
61 }
62 
env_ext4_save_buffer(env_t * env_new)63 static int env_ext4_save_buffer(env_t *env_new)
64 {
65 	struct blk_desc *dev_desc = NULL;
66 	struct disk_partition info;
67 	int dev, part;
68 	int err;
69 	const char *ifname = env_ext4_get_intf();
70 	const char *dev_and_part = env_ext4_get_dev_part();
71 
72 	part = blk_get_device_part_str(ifname, dev_and_part,
73 				       &dev_desc, &info, 1);
74 	if (part < 0)
75 		return 1;
76 
77 	dev = dev_desc->devnum;
78 	ext4fs_set_blk_dev(dev_desc, &info);
79 
80 	if (!ext4fs_mount(info.size)) {
81 		printf("\n** Unable to use %s %s for saveenv **\n",
82 		       ifname, dev_and_part);
83 		return 1;
84 	}
85 
86 	err = ext4fs_write(CONFIG_ENV_EXT4_FILE, (void *)env_new,
87 			   sizeof(env_t), FILETYPE_REG);
88 	ext4fs_close();
89 
90 	if (err == -1) {
91 		printf("\n** Unable to write \"%s\" from %s%d:%d **\n",
92 			CONFIG_ENV_EXT4_FILE, ifname, dev, part);
93 		return 1;
94 	}
95 
96 	return 0;
97 }
98 
env_ext4_save(void)99 static int env_ext4_save(void)
100 {
101 	env_t env_new;
102 	int err;
103 
104 	err = env_export(&env_new);
105 	if (err)
106 		return err;
107 
108 	err = env_ext4_save_buffer(&env_new);
109 	if (err)
110 		return err;
111 
112 	gd->env_valid = ENV_VALID;
113 	puts("done\n");
114 
115 	return 0;
116 }
117 
env_ext4_erase(void)118 static int env_ext4_erase(void)
119 {
120 	env_t env_new;
121 	int err;
122 
123 	memset(&env_new, 0, sizeof(env_t));
124 
125 	err = env_ext4_save_buffer(&env_new);
126 	if (err)
127 		return err;
128 
129 	gd->env_valid = ENV_INVALID;
130 	puts("done\n");
131 
132 	return 0;
133 }
134 
env_ext4_load(void)135 static int env_ext4_load(void)
136 {
137 	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
138 	struct blk_desc *dev_desc = NULL;
139 	struct disk_partition info;
140 	int dev, part;
141 	int err;
142 	loff_t off;
143 	const char *ifname = env_ext4_get_intf();
144 	const char *dev_and_part = env_ext4_get_dev_part();
145 
146 #ifdef CONFIG_MMC
147 	if (!strcmp(ifname, "mmc"))
148 		mmc_initialize(NULL);
149 #endif
150 #if defined(CONFIG_AHCI) || defined(CONFIG_SCSI)
151 	if (!strcmp(ifname, "scsi"))
152 		scsi_scan(true);
153 #endif
154 
155 	part = blk_get_device_part_str(ifname, dev_and_part,
156 				       &dev_desc, &info, 1);
157 	if (part < 0)
158 		goto err_env_relocate;
159 
160 	dev = dev_desc->devnum;
161 	ext4fs_set_blk_dev(dev_desc, &info);
162 
163 	if (!ext4fs_mount(info.size)) {
164 		printf("\n** Unable to use %s %s for loading the env **\n",
165 		       ifname, dev_and_part);
166 		goto err_env_relocate;
167 	}
168 
169 	err = ext4_read_file(CONFIG_ENV_EXT4_FILE, buf, 0, CONFIG_ENV_SIZE,
170 			     &off);
171 	ext4fs_close();
172 
173 	if (err == -1) {
174 		printf("\n** Unable to read \"%s\" from %s%d:%d **\n",
175 			CONFIG_ENV_EXT4_FILE, ifname, dev, part);
176 		goto err_env_relocate;
177 	}
178 
179 	err = env_import(buf, 1, H_EXTERNAL);
180 	if (!err)
181 		gd->env_valid = ENV_VALID;
182 
183 	return err;
184 
185 err_env_relocate:
186 	env_set_default(NULL, 0);
187 
188 	return -EIO;
189 }
190 
191 U_BOOT_ENV_LOCATION(ext4) = {
192 	.location	= ENVL_EXT4,
193 	ENV_NAME("EXT4")
194 	.load		= env_ext4_load,
195 	.save		= ENV_SAVE_PTR(env_ext4_save),
196 	.erase		= ENV_ERASE_PTR(env_ext4_erase),
197 };
198