modernize backfire 10.03 so it can be operational again
[openwrt-10.03/.git] / package / mtd / src / fis.c
1 /*
2  * FIS table updating code for mtd
3  *
4  * Copyright (C) 2009 Felix Fietkau <nbd@openwrt.org>
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License v2
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  */
15 #include <sys/mman.h>
16 #include <stdint.h>
17 #include <stdlib.h>
18 #include <string.h>
19 #include <unistd.h>
20 #include <stdio.h>
21 #include "crc32.h"
22 #include "mtd.h"
23 #include "fis.h"
24
25 struct fis_image_hdr {
26         unsigned char name[16];
27         uint32_t flash_base;
28         uint32_t mem_base;
29         uint32_t size;
30         uint32_t entry_point;
31         uint32_t data_length;
32 } __attribute__((packed));
33
34 struct fis_image_crc {
35         uint32_t desc;
36         uint32_t file;
37 } __attribute__((packed));
38
39 struct fis_image_desc {
40         struct fis_image_hdr hdr;
41         char _pad[256 - sizeof(struct fis_image_hdr) - sizeof(struct fis_image_crc)];
42         struct fis_image_crc crc;
43 } __attribute__((packed));
44
45 static int fis_fd = -1;
46 static struct fis_image_desc *fis_desc;
47 static int fis_erasesize = 0;
48
49 static void
50 fis_close(void)
51 {
52         if (fis_desc)
53                 munmap(fis_desc, fis_erasesize);
54
55         if (fis_fd >= 0)
56                 close(fis_fd);
57
58         fis_fd = -1;
59         fis_desc = NULL;
60 }
61
62 static struct fis_image_desc *
63 fis_open(void)
64 {
65         struct fis_image_desc *desc;
66
67         if (fis_fd >= 0)
68                 fis_close();
69
70         fis_fd = mtd_check_open("FIS directory");
71         if (fis_fd < 0)
72                 goto error;
73
74         close(fis_fd);
75         fis_fd = mtd_open("FIS directory", true);
76         if (fis_fd < 0)
77                 goto error;
78
79         fis_erasesize = erasesize;
80         desc = mmap(NULL, erasesize, PROT_READ|PROT_WRITE, MAP_SHARED|MAP_LOCKED, fis_fd, 0);
81         if (desc == MAP_FAILED)
82                 goto error;
83
84         fis_desc = desc;
85         return desc;
86
87 error:
88         fis_close();
89         return NULL;
90 }
91
92 int
93 fis_validate(struct fis_part *old, int n_old, struct fis_part *new, int n_new)
94 {
95         struct fis_image_desc *desc;
96         void *end;
97         int found = 0;
98         int i;
99
100         desc = fis_open();
101         if (!desc)
102                 return -1;
103
104         for (i = 0; i < n_new - 1; i++) {
105                 if (!new[i].size) {
106                         fprintf(stderr, "FIS error: only the last partition can detect the size automatically\n");
107                         i = -1;
108                         goto done;
109                 }
110         }
111
112         end = desc;
113         end = (char *) end + fis_erasesize;
114         while ((void *) desc < end) {
115                 if (!desc->hdr.name[0] || (desc->hdr.name[0] == 0xff))
116                         break;
117
118                 for (i = 0; i < n_old; i++) {
119                         if (!strncmp((char *) desc->hdr.name, (char *) old[i].name, sizeof(desc->hdr.name))) {
120                                 found++;
121                                 goto next;
122                         }
123                 }
124 next:
125                 desc++;
126                 continue;
127         }
128
129         if (found == n_old)
130                 i = 1;
131         else
132                 i = -1;
133
134 done:
135         fis_close();
136         return i;
137 }
138
139 int
140 fis_remap(struct fis_part *old, int n_old, struct fis_part *new, int n_new)
141 {
142         struct fis_image_desc *fisdir = NULL;
143         struct fis_image_desc *redboot = NULL;
144         struct fis_image_desc *first = NULL;
145         struct fis_image_desc *last = NULL;
146         struct fis_image_desc *first_fb = NULL;
147         struct fis_image_desc *last_fb = NULL;
148         struct fis_image_desc *desc;
149         struct fis_part *part;
150         uint32_t offset = 0, size = 0;
151         char *start, *end, *tmp;
152         int i;
153
154         desc = fis_open();
155         if (!desc)
156                 return -1;
157
158         if (!quiet)
159                 fprintf(stderr, "Updating FIS table... \n");
160
161         start = (char *) desc;
162         end = (char *) desc + fis_erasesize;
163         while ((char *) desc < end) {
164                 if (!desc->hdr.name[0] || (desc->hdr.name[0] == 0xff))
165                         break;
166
167                 if (!strcmp((char *) desc->hdr.name, "FIS directory"))
168                         fisdir = desc;
169
170                 if (!strcmp((char *) desc->hdr.name, "RedBoot"))
171                         redboot = desc;
172
173                 /* update max offset */
174                 if (offset < desc->hdr.flash_base)
175                         offset = desc->hdr.flash_base;
176
177                 for (i = 0; i < n_old; i++) {
178                         if (!strncmp((char *) desc->hdr.name, (char *) old[i].name, sizeof(desc->hdr.name))) {
179                                 last = desc;
180                                 if (!first)
181                                         first = desc;
182                                 break;
183                         }
184                 }
185                 desc++;
186         }
187         desc--;
188
189         first_fb = first;
190         last_fb = last;
191
192         if (first_fb->hdr.flash_base > last_fb->hdr.flash_base) {
193                 first_fb = last;
194                 last_fb = first;
195         }
196
197         /* determine size of available space */
198         desc = (struct fis_image_desc *) start;
199         while ((char *) desc < end) {
200                 if (!desc->hdr.name[0] || (desc->hdr.name[0] == 0xff))
201                         break;
202
203                 if (desc->hdr.flash_base > last_fb->hdr.flash_base &&
204                     desc->hdr.flash_base < offset)
205                         offset = desc->hdr.flash_base;
206
207                 desc++;
208         }
209         desc--;
210
211         size = offset - first_fb->hdr.flash_base;
212
213 #ifdef notyet
214         desc = first - 1;
215         if (redboot && (desc >= redboot)) {
216                 if (first->hdr.flash_base - desc->hdr.size > desc->hdr.flash_base) {
217                         int delta = first->hdr.flash_base - desc->hdr.size - desc->hdr.flash_base;
218
219                         offset -= delta;
220                         size += delta;
221                 }
222         }
223 #endif
224
225         last++;
226         desc = first + n_new;
227         offset = first_fb->hdr.flash_base;
228
229         if (desc != last) {
230                 if (desc > last)
231                         tmp = (char *) desc;
232                 else
233                         tmp = (char *) last;
234
235                 memmove(desc, last, end - tmp);
236                 if (desc < last) {
237                         tmp = end - (last - desc) * sizeof(struct fis_image_desc);
238                         memset(tmp, 0xff, tmp - end);
239                 }
240         }
241
242         for (part = new, desc = first; desc < first + n_new; desc++, part++) {
243                 memset(desc, 0, sizeof(struct fis_image_desc));
244                 memcpy(desc->hdr.name, part->name, sizeof(desc->hdr.name));
245                 desc->crc.desc = 0;
246                 desc->crc.file = 0;
247
248                 desc->hdr.flash_base = offset;
249                 desc->hdr.mem_base = part->loadaddr;
250                 desc->hdr.entry_point = part->loadaddr;
251                 desc->hdr.size = (part->size > 0) ? part->size : size;
252                 desc->hdr.data_length = desc->hdr.size;
253
254                 offset += desc->hdr.size;
255                 size -= desc->hdr.size;
256         }
257
258         msync(fis_desc, fis_erasesize, MS_SYNC|MS_INVALIDATE);
259         fis_close();
260
261         return 0;
262 }