• source navigation  • diff markup  • identifier search  • freetext search  • 

Sources/firmware-utils/src/iptime-crc32.c

  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