1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * Copyright (c) 2021 Sungbo Eo <mans0n@gorani.run> 4 * 5 * This code is based on mkdhpimg.c and mkzcfw.c 6 * Copyright (C) 2010 Gabor Juhos <juhosg@openwrt.org> 7 * Copyright (c) 2016 FUKAUMI Naoki <naobsd@gmail.com> 8 * 9 * Checksum algorithm is derived from add_iptime_fw_header.c 10 * Copyright (C) 2020 Jaehoon You <teslamint@gmail.com> 11 */ 12 13 #include <byteswap.h> 14 #include <endian.h> 15 #include <err.h> 16 #include <errno.h> 17 #include <fcntl.h> 18 #include <stdint.h> 19 #include <stdio.h> 20 #include <stdlib.h> 21 #include <string.h> 22 #include <sys/stat.h> 23 #include <unistd.h> 24 25 #include "cyg_crc.h" 26 27 #if !defined(__BYTE_ORDER) 28 #error "Unknown byte order" 29 #endif 30 31 #if (__BYTE_ORDER == __BIG_ENDIAN) 32 #define HOST_TO_LE32(x) bswap_32(x) 33 #elif (__BYTE_ORDER == __LITTLE_ENDIAN) 34 #define HOST_TO_LE32(x) (x) 35 #else 36 #error "Unsupported endianness" 37 #endif 38 39 #define FW_VERSION "00_000" 40 41 struct fw_header { 42 uint8_t model[8]; 43 uint8_t version[8]; 44 uint8_t reserved[32]; 45 uint32_t size; 46 uint32_t checksum; 47 } __attribute__ ((packed)); 48 49 struct board_info { 50 const char *model; 51 size_t payload_offset; 52 }; 53 54 struct board_info boards[] = { 55 { .model = "a6004mx", .payload_offset = 0x800 }, 56 { .model = "ax2004m", .payload_offset = 0x38 }, 57 { .model = "ax3000m", .payload_offset = 0x38 }, 58 { .model = "ax8004m", .payload_offset = 0x38 }, 59 { /* sentinel */ } 60 }; 61 62 struct board_info *find_board(const char *model) 63 { 64 struct board_info *ret = NULL; 65 struct board_info *board; 66 67 for (board = boards; board->model != NULL; board++) { 68 if (strcmp(model, board->model) == 0) { 69 ret = board; 70 break; 71 } 72 } 73 74 return ret; 75 } 76 77 uint32_t make_checksum(struct fw_header *header, uint8_t *payload, int size) 78 { 79 cyg_uint32 checksum; 80 81 /* get CRC of header */ 82 checksum = cyg_crc32_accumulate(~0L, header, sizeof(*header)); 83 84 /* get CRC of payload buffer with header CRC as initial value */ 85 return (uint32_t)cyg_crc32_accumulate(checksum, payload, size); 86 } 87 88 void make_header(struct board_info *board, uint8_t *buffer, size_t img_size) 89 { 90 struct fw_header *header = (struct fw_header *)buffer; 91 uint32_t checksum; 92 93 strncpy((char *)header->model, board->model, sizeof(header->model)-1); 94 strncpy((char *)header->version, FW_VERSION, sizeof(header->version)-1); 95 header->size = HOST_TO_LE32(img_size); 96 checksum = make_checksum(header, buffer + board->payload_offset, img_size); 97 header->checksum = HOST_TO_LE32(checksum); 98 } 99 100 int main(int argc, const char *argv[]) 101 { 102 const char *model_name, *img_in, *img_out; 103 struct board_info *board; 104 int file_in, file_out; 105 struct stat stat_in; 106 size_t size_in, size_out; 107 uint8_t *buffer; 108 109 if (argc != 4) { 110 fprintf(stderr, "Usage: %s <model> <input> <output>\n", argv[0]); 111 return EXIT_FAILURE; 112 } 113 model_name = argv[1]; 114 img_in = argv[2]; 115 img_out = argv[3]; 116 117 board = find_board(model_name); 118 if (board == NULL) { 119 fprintf(stderr, "%s: Not supported model\n", model_name); 120 return EXIT_FAILURE; 121 } 122 123 if ((file_in = open(img_in, O_RDONLY)) == -1) 124 err(EXIT_FAILURE, "%s", img_in); 125 126 if (fstat(file_in, &stat_in) == -1) 127 err(EXIT_FAILURE, "%s", img_in); 128 129 size_in = stat_in.st_size; 130 size_out = board->payload_offset + size_in; 131 132 if ((buffer = malloc(size_out)) == NULL) 133 err(EXIT_FAILURE, "malloc"); 134 135 read(file_in, buffer + board->payload_offset, size_in); 136 close(file_in); 137 138 memset(buffer, 0, board->payload_offset); 139 140 make_header(board, buffer, size_in); 141 142 if ((file_out = creat(img_out, 0644)) == -1) 143 err(EXIT_FAILURE, "%s", img_out); 144 write(file_out, buffer, size_out); 145 close(file_out); 146 147 free(buffer); 148 149 return EXIT_SUCCESS; 150 } 151
This page was automatically generated by LXR 0.3.1. • OpenWrt