ramips: add missing information to dlink headers

[openwrt/openwrt.git] / tools / firmware-utils / src / uimage_sgehdr.c

diff --git a/tools/firmware-utils/src/uimage_sgehdr.c b/tools/firmware-utils/src/uimage_sgehdr.c
new file mode 100644 (file)
index 0000000..28143a8
--- /dev/null
+++ b/tools/firmware-utils/src/uimage_sgehdr.c
@@ -0,0 +1,167 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * uimage_sgehdr.c : add 96 bytes of extra header information after the normal tail of uimage header
+ * this is an edited version of uimage_padhdr.c
+ *
+ * Copyright (C) 2019 NOGUCHI Hiroshi <drvlabo@gmail.com>
+ */
+
+#include <stdio.h>
+#include <errno.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <arpa/inet.h>
+#include <zlib.h>
+
+
+/* from u-boot/include/image.h */
+#define IH_NMLEN               32      /* Image Name Length            */
+#define SGE_PRODUCTLEN         64      /* sge_Product Length           */
+#define SGE_VERSIONLEN         16      /* sge Version Length           */
+#define OrignalHL              64      /* Original Header Length       */
+
+/*
+ * SGE format image header,
+ * all data in network byte order (aka natural aka bigendian).
+ */
+struct image_header {
+       uint32_t        ih_magic;       /* Image Header Magic Number    */
+       uint32_t        ih_hcrc;        /* Image Header CRC Checksum    */
+       uint32_t        ih_time;        /* Image Creation Timestamp     */
+       uint32_t        ih_size;        /* Image Data Size              */
+       uint32_t        ih_load;        /* Data  Load  Address          */
+       uint32_t        ih_ep;          /* Entry Point Address          */
+       uint32_t        ih_dcrc;        /* Image Data CRC Checksum      */
+       uint8_t         ih_os;          /* Operating System             */
+       uint8_t         ih_arch;        /* CPU architecture             */
+       uint8_t         ih_type;        /* Image Type                   */
+       uint8_t         ih_comp;        /* Compression Type             */
+       uint8_t         ih_name[IH_NMLEN];      /* Image Name           */
+       uint8_t         sgeih_p[SGE_PRODUCTLEN];        /* sge_Product          */
+       uint8_t         sgeih_sv[SGE_VERSIONLEN];       /* sge Software Version         */
+       uint8_t         sgeih_hv[SGE_VERSIONLEN];       /* sge Hardware Version         */
+};
+
+
+/* default padding size */
+#define        IH_PAD_BYTES            (96)
+
+
+static void usage(char *prog)
+{
+       fprintf(stderr,
+               "%s -i <input_uimage_file> -o <output_file> -m <model> -h <hardware version> -s <software version>\n",
+               prog);
+}
+
+int main(int argc, char *argv[])
+{
+       struct stat statbuf;
+       u_int8_t *filebuf;
+       int ifd;
+       int ofd;
+       ssize_t rsz;
+       u_int32_t crc_recalc;
+       struct image_header *imgh;
+       int opt;
+       char *infname = NULL;
+       char *outfname = NULL;
+       char *model = NULL;
+       char *hversion = NULL;
+       char *sversion = NULL;
+       int padsz = IH_PAD_BYTES;
+       int ltmp;
+
+       while ((opt = getopt(argc, argv, "i:o:m:h:s:")) != -1) {
+               switch (opt) {
+               case 'i':
+                       infname = optarg;
+                       break;
+               case 'o':
+                       outfname = optarg;
+                       break;
+               case 'm':
+                       model = optarg;
+                       break;
+               case 'h':
+                       hversion = optarg;
+                       break;
+               case 's':
+                       sversion = optarg;
+                       break;
+               default:
+                       break;
+               }
+       }
+
+       if (!infname || !outfname) {
+               usage(argv[0]);
+               exit(1);
+       }
+
+       ifd = open(infname, O_RDONLY);
+       if (ifd < 0) {
+               fprintf(stderr,
+                       "could not open input file. (errno = %d)\n", errno);
+               exit(1);
+       }
+
+       ofd = open(outfname, O_WRONLY | O_CREAT, 0644);
+       if (ofd < 0) {
+               fprintf(stderr,
+                       "could not open output file. (errno = %d)\n", errno);
+               exit(1);
+       }
+
+       if (fstat(ifd, &statbuf) < 0) {
+               fprintf(stderr,
+                       "could not fstat input file. (errno = %d)\n", errno);
+               exit(1);
+       }
+
+       filebuf = malloc(statbuf.st_size + padsz);
+       if (!filebuf) {
+               fprintf(stderr, "buffer allocation failed\n");
+               exit(1);
+       }
+
+       rsz = read(ifd, filebuf, OrignalHL);
+       if (rsz != OrignalHL) {
+               fprintf(stderr,
+                       "could not read input file (errno = %d).\n", errno);
+               exit(1);
+       }
+
+       memset(&(filebuf[OrignalHL]), 0, padsz);
+
+       rsz = read(ifd, &(filebuf[sizeof(*imgh)]),
+                               statbuf.st_size - OrignalHL);
+       if (rsz != (int32_t)(statbuf.st_size - OrignalHL)) {
+               fprintf(stderr,
+                       "could not read input file (errno = %d).\n", errno);
+               exit(1);
+       }
+
+       imgh = (struct image_header *)filebuf;
+
+       imgh->ih_hcrc = 0;
+
+       strncpy(imgh->sgeih_p, model, sizeof(imgh->sgeih_p));
+       strncpy(imgh->sgeih_sv, sversion, sizeof(imgh->sgeih_sv));
+       strncpy(imgh->sgeih_hv, hversion, sizeof(imgh->sgeih_hv));
+
+       crc_recalc = crc32(0, filebuf, sizeof(*imgh));
+       imgh->ih_hcrc = htonl(crc_recalc);
+
+       rsz = write(ofd, filebuf, statbuf.st_size + padsz);
+       if (rsz != (int32_t)statbuf.st_size + padsz) {
+               fprintf(stderr,
+                       "could not write output file (errnor = %d).\n", errno);
+               exit(1);
+       }
+
+       return 0;
+}