OpenWrt – Blame information for rev 4

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