zyimage.c

  1 // SPDX-License-Identifier: GPL-2.0-only
  2 /*
  3  * Copyright (C) 2014 Soul Trace <S-trace@list.ru>
  4  */
  5 
  6 #include <stdio.h>
  7 #include <stdlib.h>
  8 #include <sys/types.h>
  9 #include <string.h>
 10 #include <unistd.h>
 11 
 12 #define szbuf 32768
 13 
 14 u_int32_t crc_tab[256];
 15 
 16 u_int32_t chksum_crc32 (FILE *f)
 17 {
 18   register unsigned long crc;
 19   unsigned long i, j;
 20   char *buffer = malloc(szbuf);
 21   char *buf;
 22 
 23   crc = 0xFFFFFFFF;
 24   while (!feof(f))
 25   {
 26     j = fread(buffer, 1, szbuf, f);
 27     buf = buffer;
 28     for (i = 0; i < j; i++)
 29       crc = ((crc >> 8) & 0x00FFFFFF) ^ crc_tab[(crc ^ *buf++) & 0xFF];
 30   }
 31   free(buffer);
 32   return crc;
 33 }
 34 
 35 void chksum_crc32gentab ()
 36 {
 37   unsigned long crc, poly;
 38   int i, j;
 39 
 40   poly = 0xEDB88320L;
 41   for (i = 0; i < 256; i++)
 42   {
 43     crc = i;
 44     for (j = 8; j > 0; j--)
 45     {
 46       if (crc & 1)
 47         crc = (crc >> 1) ^ poly;
 48       else
 49         crc >>= 1;
 50     }
 51     crc_tab[i] = crc;
 52   }
 53 }
 54 
 55 void usage(char *progname)
 56 {
 57   printf("Usage: %s [ -v Version ] [ -d Device_ID ] <input file>\n", progname);
 58   exit(1);
 59 }
 60 
 61 int main(int argc, char *argv[]) {
 62   struct signature
 63   {
 64     const char magic[4];
 65     unsigned int device_id;
 66     char firmware_version[48];
 67     unsigned int crc32;
 68   }
 69   sign =
 70   {
 71     { 'Z', 'N', 'B', 'G' },
 72     1,
 73     { "V.1.0.0(1.0.0)" },
 74     0
 75   };
 76   FILE *f;
 77   struct signature oldsign;
 78   char *filename;
 79   static const char *optString;
 80   int opt;
 81 
 82   if (argc < 1)
 83     usage(argv[0]);
 84 
 85   optString = "v:d:h";
 86   opt = getopt( argc, argv, optString );
 87   while( opt != -1 ) {
 88     switch( opt ) {
 89       case 'v':
 90         if (optarg == NULL)
 91           usage(argv[0]);
 92         strncpy(sign.firmware_version, optarg, sizeof(sign.firmware_version)-1);
 93        sign.firmware_version[sizeof(sign.firmware_version)-1]='\0'; /* Make sure that string is terminated correctly */
 94         break;
 95 
 96       case 'd':
 97         sign.device_id = atoi(optarg);
 98         if (sign.device_id == 0)
 99           sign.device_id = (int)strtol(optarg, NULL, 16);
100         break;
101 
102       case '?':
103       case 'h':
104         usage(argv[0]);
105         break;
106 
107       default:
108         break;
109     }
110 
111     opt = getopt( argc, argv, optString );
112   }
113 
114   chksum_crc32gentab();
115 
116   filename=argv[optind];
117   if (access(filename, W_OK) || access(filename, R_OK))
118   {
119     printf("Not open input file %s\n", filename);
120     exit(1);
121   }
122   f = fopen(argv[optind], "r+");
123   if (f != NULL)
124   {
125     fseek(f, sizeof(sign)*-1, SEEK_END);
126     fread(&oldsign, sizeof(oldsign), 1, f);
127 
128     if (strncmp(oldsign.magic,"ZNBG", sizeof(oldsign.magic)) == 0 )
129     {
130       printf("Image is already signed as:\nDevice ID: 0x%08x\nFirmware version: %s\nImage CRC32: 0x%x\n", oldsign.device_id, oldsign.firmware_version, oldsign.crc32);
131       exit(0);
132     }
133 
134     fseek(f, 0, SEEK_SET);
135     sign.crc32 = chksum_crc32(f);
136     fwrite(&sign, sizeof(sign), 1, f);
137     fclose(f);
138 
139     printf("Image signed as:\nDevice ID: 0x%08x\nFirmware version: %s\nImage CRC32: 0x%x\n", sign.device_id, sign.firmware_version, sign.crc32);
140   }
141   return 0;
142 }
143
This page was automatically generated by LXR 0.3.1. • OpenWrt
OpenWrt.org Cross Reference

Sources/firmware-utils/src/zyimage.c
