1 /* 2 * (C) Copyright 2011 3 * Stefano Babic, DENX Software Engineering, sbabic@denx.de. 4 * 5 * See file CREDITS for list of people who contributed to this 6 * project. 7 * 8 * This program is free software; you can redistribute it and/or 9 * modify it under the terms of the GNU General Public License as 10 * published by the Free Software Foundation; either version 2 of 11 * the License, or (at your option) any later version. 12 * 13 * This program is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU General Public License 19 * along with this program; if not, write to the Free Software 20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 21 * MA 02111-1307 USA 22 */ 23 24 /* Required to obtain the getline prototype from stdio.h */ 25 #define _GNU_SOURCE 26 27 #include "mkimage.h" 28 #include "aisimage.h" 29 #include <image.h> 30 31 #define IS_FNC_EXEC(c) (cmd_table[c].AIS_cmd == AIS_CMD_FNLOAD) 32 #define WORD_ALIGN0 4 33 #define WORD_ALIGN(len) (((len)+WORD_ALIGN0-1) & ~(WORD_ALIGN0-1)) 34 #define MAX_CMD_BUFFER 4096 35 #define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0])) 36 37 static uint32_t ais_img_size; 38 39 /* 40 * Supported commands for configuration file 41 */ 42 static table_entry_t aisimage_cmds[] = { 43 {CMD_DATA, "DATA", "Reg Write Data"}, 44 {CMD_FILL, "FILL", "Fill range with pattern"}, 45 {CMD_CRCON, "CRCON", "CRC Enable"}, 46 {CMD_CRCOFF, "CRCOFF", "CRC Disable"}, 47 {CMD_CRCCHECK, "CRCCHECK", "CRC Validate"}, 48 {CMD_JMPCLOSE, "JMPCLOSE", "Jump & Close"}, 49 {CMD_JMP, "JMP", "Jump"}, 50 {CMD_SEQREAD, "SEQREAD", "Sequential read"}, 51 {CMD_PLL0, "PLL0", "PLL0"}, 52 {CMD_PLL1, "PLL1", "PLL1"}, 53 {CMD_CLK, "CLK", "Clock configuration"}, 54 {CMD_DDR2, "DDR2", "DDR2 Configuration"}, 55 {CMD_EMIFA, "EMIFA", "EMIFA"}, 56 {CMD_EMIFA_ASYNC, "EMIFA_ASYNC", "EMIFA Async"}, 57 {CMD_PLL, "PLL", "PLL & Clock configuration"}, 58 {CMD_PSC, "PSC", "PSC setup"}, 59 {CMD_PINMUX, "PINMUX", "Pinmux setup"}, 60 {CMD_BOOTTABLE, "BOOT_TABLE", "Boot table command"}, 61 {-1, "", ""}, 62 }; 63 64 static struct ais_func_exec { 65 uint32_t index; 66 uint32_t argcnt; 67 } ais_func_table[] = { 68 [CMD_PLL0] = {0, 2}, 69 [CMD_PLL1] = {1, 2}, 70 [CMD_CLK] = {2, 1}, 71 [CMD_DDR2] = {3, 8}, 72 [CMD_EMIFA] = {4, 5}, 73 [CMD_EMIFA_ASYNC] = {5, 5}, 74 [CMD_PLL] = {6, 3}, 75 [CMD_PSC] = {7, 1}, 76 [CMD_PINMUX] = {8, 3} 77 }; 78 79 static struct cmd_table_t { 80 uint32_t nargs; 81 uint32_t AIS_cmd; 82 } cmd_table[] = { 83 [CMD_FILL] = { 4, AIS_CMD_FILL}, 84 [CMD_CRCON] = { 0, AIS_CMD_ENCRC}, 85 [CMD_CRCOFF] = { 0, AIS_CMD_DISCRC}, 86 [CMD_CRCCHECK] = { 2, AIS_CMD_ENCRC}, 87 [CMD_JMPCLOSE] = { 1, AIS_CMD_JMPCLOSE}, 88 [CMD_JMP] = { 1, AIS_CMD_JMP}, 89 [CMD_SEQREAD] = { 0, AIS_CMD_SEQREAD}, 90 [CMD_PLL0] = { 2, AIS_CMD_FNLOAD}, 91 [CMD_PLL1] = { 2, AIS_CMD_FNLOAD}, 92 [CMD_CLK] = { 1, AIS_CMD_FNLOAD}, 93 [CMD_DDR2] = { 8, AIS_CMD_FNLOAD}, 94 [CMD_EMIFA] = { 5, AIS_CMD_FNLOAD}, 95 [CMD_EMIFA_ASYNC] = { 5, AIS_CMD_FNLOAD}, 96 [CMD_PLL] = { 3, AIS_CMD_FNLOAD}, 97 [CMD_PSC] = { 1, AIS_CMD_FNLOAD}, 98 [CMD_PINMUX] = { 3, AIS_CMD_FNLOAD}, 99 [CMD_BOOTTABLE] = { 4, AIS_CMD_BOOTTBL}, 100 }; 101 102 static uint32_t get_cfg_value(char *token, char *name, int linenr) 103 { 104 char *endptr; 105 uint32_t value; 106 107 errno = 0; 108 value = strtoul(token, &endptr, 16); 109 if (errno || (token == endptr)) { 110 fprintf(stderr, "Error: %s[%d] - Invalid hex data(%s)\n", 111 name, linenr, token); 112 exit(EXIT_FAILURE); 113 } 114 return value; 115 } 116 117 static int get_ais_table_id(uint32_t *ptr) 118 { 119 120 int i; 121 int func_no; 122 123 for (i = 0; i < ARRAY_SIZE(cmd_table); i++) { 124 if (*ptr == cmd_table[i].AIS_cmd) { 125 if (cmd_table[i].AIS_cmd != AIS_CMD_FNLOAD) 126 return i; 127 128 func_no = ((struct ais_cmd_func *)ptr)->func_args 129 & 0xFFFF; 130 if (func_no == ais_func_table[i].index) 131 return i; 132 } 133 } 134 135 return -1; 136 } 137 138 static void aisimage_print_header(const void *hdr) 139 { 140 struct ais_header *ais_hdr = (struct ais_header *)hdr; 141 uint32_t *ptr; 142 struct ais_cmd_load *ais_load; 143 int id; 144 145 if (ais_hdr->magic != AIS_MAGIC_WORD) { 146 fprintf(stderr, "Error: - AIS Magic Number not found\n"); 147 return; 148 } 149 fprintf(stdout, "Image Type: TI Davinci AIS Boot Image\n"); 150 fprintf(stdout, "AIS magic : %08x\n", ais_hdr->magic); 151 ptr = (uint32_t *)&ais_hdr->magic; 152 ptr++; 153 154 while (*ptr != AIS_CMD_JMPCLOSE) { 155 /* Check if we find the image */ 156 if (*ptr == AIS_CMD_LOAD) { 157 ais_load = (struct ais_cmd_load *)ptr; 158 fprintf(stdout, "Image at : 0x%08x size 0x%08x\n", 159 ais_load->addr, 160 ais_load->size); 161 ptr = ais_load->data + ais_load->size / sizeof(*ptr); 162 continue; 163 } 164 165 id = get_ais_table_id(ptr); 166 if (id < 0) { 167 fprintf(stderr, "Error: - AIS Image corrupted\n"); 168 return; 169 } 170 fprintf(stdout, "AIS cmd : %s\n", 171 get_table_entry_name(aisimage_cmds, NULL, id)); 172 ptr += cmd_table[id].nargs + IS_FNC_EXEC(id) + 1; 173 if (((void *)ptr - hdr) > ais_img_size) { 174 fprintf(stderr, 175 "AIS Image not terminated by JMPCLOSE\n"); 176 return; 177 } 178 } 179 } 180 181 static uint32_t *ais_insert_cmd_header(uint32_t cmd, uint32_t nargs, 182 uint32_t *parms, struct image_type_params *tparams, 183 uint32_t *ptr) 184 { 185 int i; 186 187 *ptr++ = cmd_table[cmd].AIS_cmd; 188 if (IS_FNC_EXEC(cmd)) 189 *ptr++ = ((nargs & 0xFFFF) << 16) + ais_func_table[cmd].index; 190 191 /* Copy parameters */ 192 for (i = 0; i < nargs; i++) 193 *ptr++ = cpu_to_le32(parms[i]); 194 195 return ptr; 196 197 } 198 199 static uint32_t *ais_alloc_buffer(struct mkimage_params *params) 200 { 201 int dfd; 202 struct stat sbuf; 203 char *datafile = params->datafile; 204 uint32_t *ptr; 205 206 dfd = open(datafile, O_RDONLY|O_BINARY); 207 if (dfd < 0) { 208 fprintf(stderr, "%s: Can't open %s: %s\n", 209 params->cmdname, datafile, strerror(errno)); 210 exit(EXIT_FAILURE); 211 } 212 213 if (fstat(dfd, &sbuf) < 0) { 214 fprintf(stderr, "%s: Can't stat %s: %s\n", 215 params->cmdname, datafile, strerror(errno)); 216 exit(EXIT_FAILURE); 217 } 218 219 /* 220 * Place for header is allocated. The size is taken from 221 * the size of the datafile, that the ais_image_generate() 222 * will copy into the header. Copying the datafile 223 * is not left to the main program, because after the datafile 224 * the header must be terminated with the Jump & Close command. 225 */ 226 ais_img_size = WORD_ALIGN(sbuf.st_size) + MAX_CMD_BUFFER; 227 ptr = (uint32_t *)malloc(WORD_ALIGN(sbuf.st_size) + MAX_CMD_BUFFER); 228 if (!ptr) { 229 fprintf(stderr, "%s: malloc return failure: %s\n", 230 params->cmdname, strerror(errno)); 231 exit(EXIT_FAILURE); 232 } 233 234 close(dfd); 235 236 return ptr; 237 } 238 239 static uint32_t *ais_copy_image(struct mkimage_params *params, 240 uint32_t *aisptr) 241 242 { 243 int dfd; 244 struct stat sbuf; 245 char *datafile = params->datafile; 246 void *ptr; 247 248 dfd = open(datafile, O_RDONLY|O_BINARY); 249 if (dfd < 0) { 250 fprintf(stderr, "%s: Can't open %s: %s\n", 251 params->cmdname, datafile, strerror(errno)); 252 exit(EXIT_FAILURE); 253 } 254 255 if (fstat(dfd, &sbuf) < 0) { 256 fprintf(stderr, "%s: Can't stat %s: %s\n", 257 params->cmdname, datafile, strerror(errno)); 258 exit(EXIT_FAILURE); 259 } 260 261 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, dfd, 0); 262 *aisptr++ = AIS_CMD_LOAD; 263 *aisptr++ = params->ep; 264 *aisptr++ = sbuf.st_size; 265 memcpy((void *)aisptr, ptr, sbuf.st_size); 266 aisptr += WORD_ALIGN(sbuf.st_size) / sizeof(uint32_t); 267 268 (void) munmap((void *)ptr, sbuf.st_size); 269 (void) close(dfd); 270 271 return aisptr; 272 273 } 274 275 static int aisimage_generate(struct mkimage_params *params, 276 struct image_type_params *tparams) 277 { 278 FILE *fd = NULL; 279 char *line = NULL; 280 char *token, *saveptr1, *saveptr2; 281 int lineno = 0; 282 int fld; 283 size_t len; 284 int32_t cmd; 285 uint32_t nargs, cmd_parms[10]; 286 uint32_t value, size; 287 char *name = params->imagename; 288 uint32_t *aishdr; 289 290 fd = fopen(name, "r"); 291 if (fd == 0) { 292 fprintf(stderr, 293 "Error: %s - Can't open AIS configuration\n", name); 294 exit(EXIT_FAILURE); 295 } 296 297 /* 298 * the size of the header is variable and is computed 299 * scanning the configuration file. 300 */ 301 tparams->header_size = 0; 302 303 /* 304 * Start allocating a buffer suitable for most command 305 * The buffer is then reallocated if it is too small 306 */ 307 aishdr = ais_alloc_buffer(params); 308 tparams->hdr = aishdr; 309 *aishdr++ = AIS_MAGIC_WORD; 310 311 /* Very simple parsing, line starting with # are comments 312 * and are dropped 313 */ 314 while ((getline(&line, &len, fd)) > 0) { 315 lineno++; 316 317 token = strtok_r(line, "\r\n", &saveptr1); 318 if (token == NULL) 319 continue; 320 321 /* Check inside the single line */ 322 line = token; 323 fld = CFG_COMMAND; 324 cmd = CMD_INVALID; 325 nargs = 0; 326 while (token != NULL) { 327 token = strtok_r(line, " \t", &saveptr2); 328 if (token == NULL) 329 break; 330 331 /* Drop all text starting with '#' as comments */ 332 if (token[0] == '#') 333 break; 334 335 switch (fld) { 336 case CFG_COMMAND: 337 cmd = get_table_entry_id(aisimage_cmds, 338 "aisimage commands", token); 339 if (cmd < 0) { 340 fprintf(stderr, 341 "Error: %s[%d] - Invalid command" 342 "(%s)\n", name, lineno, token); 343 344 exit(EXIT_FAILURE); 345 } 346 break; 347 case CFG_VALUE: 348 value = get_cfg_value(token, name, lineno); 349 cmd_parms[nargs++] = value; 350 if (nargs > cmd_table[cmd].nargs) { 351 fprintf(stderr, 352 "Error: %s[%d] - too much arguments:" 353 "(%s) for command %s\n", name, 354 lineno, token, 355 aisimage_cmds[cmd].sname); 356 exit(EXIT_FAILURE); 357 } 358 break; 359 } 360 line = NULL; 361 fld = CFG_VALUE; 362 } 363 if (cmd != CMD_INVALID) { 364 /* Now insert the command into the header */ 365 aishdr = ais_insert_cmd_header(cmd, nargs, cmd_parms, 366 tparams, aishdr); 367 } 368 369 } 370 fclose(fd); 371 372 aishdr = ais_copy_image(params, aishdr); 373 374 /* Add Jmp & Close */ 375 *aishdr++ = AIS_CMD_JMPCLOSE; 376 *aishdr++ = params->ep; 377 378 size = (aishdr - (uint32_t *)tparams->hdr) * sizeof(uint32_t); 379 tparams->header_size = size; 380 381 return 0; 382 } 383 384 static int aisimage_check_image_types(uint8_t type) 385 { 386 if (type == IH_TYPE_AISIMAGE) 387 return EXIT_SUCCESS; 388 else 389 return EXIT_FAILURE; 390 } 391 392 static int aisimage_verify_header(unsigned char *ptr, int image_size, 393 struct mkimage_params *params) 394 { 395 struct ais_header *ais_hdr = (struct ais_header *)ptr; 396 397 if (ais_hdr->magic != AIS_MAGIC_WORD) 398 return -FDT_ERR_BADSTRUCTURE; 399 400 /* Store the total size to remember in print_hdr */ 401 ais_img_size = image_size; 402 403 return 0; 404 } 405 406 static void aisimage_set_header(void *ptr, struct stat *sbuf, int ifd, 407 struct mkimage_params *params) 408 { 409 } 410 411 int aisimage_check_params(struct mkimage_params *params) 412 { 413 if (!params) 414 return CFG_INVALID; 415 if (!strlen(params->imagename)) { 416 fprintf(stderr, "Error: %s - Configuration file not specified, " 417 "it is needed for aisimage generation\n", 418 params->cmdname); 419 return CFG_INVALID; 420 } 421 /* 422 * Check parameters: 423 * XIP is not allowed and verify that incompatible 424 * parameters are not sent at the same time 425 * For example, if list is required a data image must not be provided 426 */ 427 return (params->dflag && (params->fflag || params->lflag)) || 428 (params->fflag && (params->dflag || params->lflag)) || 429 (params->lflag && (params->dflag || params->fflag)) || 430 (params->xflag) || !(strlen(params->imagename)); 431 } 432 433 /* 434 * aisimage parameters 435 */ 436 static struct image_type_params aisimage_params = { 437 .name = "TI Davinci AIS Boot Image support", 438 .header_size = 0, 439 .hdr = NULL, 440 .check_image_type = aisimage_check_image_types, 441 .verify_header = aisimage_verify_header, 442 .print_header = aisimage_print_header, 443 .set_header = aisimage_set_header, 444 .check_params = aisimage_check_params, 445 .vrec_header = aisimage_generate, 446 }; 447 448 void init_ais_image_type(void) 449 { 450 mkimage_register(&aisimage_params); 451 } 452