OpenWrt – Blame information for rev 2
?pathlinks?
Rev | Author | Line No. | Line |
---|---|---|---|
1 | office | 1 | #include <stdio.h> |
2 | #include <stdlib.h> |
||
3 | #include <string.h> |
||
4 | #include <libgen.h> |
||
5 | #include "mktitanimg.h" |
||
6 | |||
7 | |||
8 | struct checksumrecord |
||
9 | { |
||
10 | unsigned int magic; |
||
11 | unsigned int chksum; /* The checksum for the complete header. |
||
12 | Excepting the |
||
13 | checksum block */ |
||
14 | }; |
||
15 | /*************************************************************************** |
||
16 | * void print_help(void) |
||
17 | ***************************************************************************/ |
||
18 | void print_help(void) |
||
19 | { |
||
20 | static char* help_page[]= |
||
21 | { |
||
22 | "mknspimg version 1.0, Texas Instruments, 2004", |
||
23 | "Syntax:", |
||
24 | " mknspimg -o outfile -i image1 image2 -a align1 align2 [-v] [-b] [-p prod_id] [-r rel_id] [-s rel_name] [-f flags]", |
||
25 | "Example:", |
||
26 | " mknspimg -o nsp_image.bin -i kernel.bin files.img -a 0 4096", |
||
27 | "This generates 'nsp_image.bin' from two input files aligning first to 0 and second to 4096 bytes." |
||
28 | }; |
||
29 | |||
30 | int num_lines = sizeof(help_page)/sizeof(char*); |
||
31 | int i; |
||
32 | for(i=0; i < num_lines; i++) { |
||
33 | printf("%s\n", help_page[i]); |
||
34 | } |
||
35 | } |
||
36 | |||
37 | /*************************************************************************** |
||
38 | * void mknspimg_print_hdr(NSP_IMG_HDR* p_img_hdr) |
||
39 | ***************************************************************************/ |
||
40 | void mknspimg_print_hdr(struct nsp_img_hdr *hdr) |
||
41 | { |
||
42 | struct nsp_img_hdr_chksum *chksum; |
||
43 | struct nsp_img_hdr_section_info *sect_info; |
||
44 | struct nsp_img_hdr_sections *section; |
||
45 | int i; |
||
46 | |||
47 | printf("****************** NSP Image Summary ******************\n"); |
||
48 | printf("Magic: 0x%x\n", hdr->head.magic); |
||
49 | printf("Image Header Size: 0x%x bytes\n", hdr->head.hdr_size); |
||
50 | printf("Total Image Size: %d bytes\n", hdr->head.image_size); |
||
51 | printf("Product ID: 0x%x\n", hdr->head.prod_id); |
||
52 | printf("Release ID: 0x%x\n", hdr->head.rel_id); |
||
53 | printf("Version ID: 0x%x\n", hdr->head.version); |
||
54 | |||
55 | printf("Offset Info: 0x%x\n", hdr->head.info_offset); |
||
56 | printf("Offset Sect info: 0x%x\n", hdr->head.sect_info_offset); |
||
57 | printf("Offset Sections: 0x%x\n", hdr->sect_info.sections_offset); |
||
58 | |||
59 | chksum=(struct nsp_img_hdr_chksum *)(hdr+hdr->head.chksum_offset); |
||
60 | printf("Header Checksum: 0x%x\n", chksum->hdr_chksum); |
||
61 | |||
62 | printf("+++ Section Information +++\n"); |
||
63 | printf("# of sections: %u\n", hdr->sect_info.num_sects); |
||
64 | section=&(hdr->sections); |
||
65 | for(i = 0; i < hdr->sect_info.num_sects; i++, section++) { |
||
66 | printf("+++++ Section %d +++++\n", i); |
||
67 | printf("Total size: %u bytes\n", section->total_size); |
||
68 | printf("Raw Size: %u bytes\n", section->raw_size); |
||
69 | printf("Offset: 0x%x\n", section->offset); |
||
70 | printf("Type: 0x%x\n", section->type); |
||
71 | printf("Name: %s\n", section->name); |
||
72 | } |
||
73 | printf("*******************************************************\n"); |
||
74 | } |
||
75 | |||
76 | CMDLINE_CFG cmd_line_cfg = |
||
77 | { |
||
78 | { |
||
79 | /* MIN MAX FLAGS OPTION */ |
||
80 | { 2, 2, (CMDLINE_OPTFLAG_ALLOW | CMDLINE_OPTFLAG_MANDAT) }, /* '-a' align1 align2 */ |
||
81 | { 0, 0, CMDLINE_OPTFLAG_ALLOW }, /* '-b' bootstrap */ |
||
82 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-c' */ |
||
83 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-d' */ |
||
84 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-e' */ |
||
85 | { 1, 1, CMDLINE_OPTFLAG_ALLOW }, /* '-f' flags */ |
||
86 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-g' */ |
||
87 | { 1, 1, CMDLINE_OPTFLAG_ALLOW }, /* '-h' */ |
||
88 | { 2, 2, (CMDLINE_OPTFLAG_ALLOW | CMDLINE_OPTFLAG_MANDAT) }, /* '-i arg1 arg2 ' */ |
||
89 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-j' */ |
||
90 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-k' */ |
||
91 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-l' */ |
||
92 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-m' */ |
||
93 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-n' */ |
||
94 | { 1, 1, (CMDLINE_OPTFLAG_ALLOW | CMDLINE_OPTFLAG_MANDAT) }, /* '-o arg' */ |
||
95 | { 1, 1, CMDLINE_OPTFLAG_ALLOW }, /* '-p' PROD_ID */ |
||
96 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-q' */ |
||
97 | { 1, 1, CMDLINE_OPTFLAG_ALLOW }, /* '-r' REL_ID */ |
||
98 | { 1, 1, CMDLINE_OPTFLAG_ALLOW }, /* '-s' "Release XXX.XXX" */ |
||
99 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-t' */ |
||
100 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-u' */ |
||
101 | { 0, 0, CMDLINE_OPTFLAG_ALLOW }, /* '-v' control VERBOSE/NON-VERBOSE mode */ |
||
102 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-w' */ |
||
103 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-x' */ |
||
104 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* '-y' */ |
||
105 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW } /* '-z' */ |
||
106 | }, |
||
107 | { 0, 0, !CMDLINE_OPTFLAG_ALLOW }, /* global arguments */ |
||
108 | }; |
||
109 | |||
110 | /*************************************************************************** |
||
111 | * int nsp_img_write(void* image, char* file, int padding) |
||
112 | * Write out the image. |
||
113 | ***************************************************************************/ |
||
114 | int main(int argc, char* argv[], char* env[]) |
||
115 | { |
||
116 | FILE* nsp_image = NULL; |
||
117 | int header_version=1; |
||
118 | int cmdline_err; |
||
119 | char* cmdline_error_msg; |
||
120 | |||
121 | char* filen_kernel; |
||
122 | char* filen_files; |
||
123 | char* filen_out; |
||
124 | |||
125 | int i,count; /* loop variables */ |
||
126 | int num_sects = 2; /* We require exactly two image with -i option |
||
127 | (see CMDLINE_CFG structure above) */ |
||
128 | int desc_count=0; |
||
129 | int total = 0; |
||
130 | |||
131 | int header_size=0; |
||
132 | struct nsp_img_hdr_head *img_hdr_head; /* Start of image header */ |
||
133 | struct nsp_img_hdr_info *img_hdr_info; |
||
134 | struct nsp_img_hdr_section_info *img_hdr_section_info ; |
||
135 | struct nsp_img_hdr_sections *img_hdr_sections, *section; /* Section pointers */ |
||
136 | |||
137 | |||
138 | /* Configure the command line. */ |
||
139 | cmdline_configure(&cmd_line_cfg); |
||
140 | |||
141 | /* Read and parse the command line. */ |
||
142 | cmdline_err = cmdline_read(argc, argv); |
||
143 | |||
144 | /* Check for parsing errors. */ |
||
145 | if(cmdline_err != 0) { |
||
146 | /* Get the parse error message */ |
||
147 | cmdline_error_msg = cmdline_error(cmdline_err); |
||
148 | |||
149 | /* Print it out */ |
||
150 | printf("%s\n", cmdline_error_msg); |
||
151 | |||
152 | /* Print our help too */ |
||
153 | print_help(); |
||
154 | return -1; |
||
155 | } |
||
156 | if(cmdline_getopt_count('h') > 0) |
||
157 | { |
||
158 | header_version=atoi(argv[cmdline_getarg(cmdline_getarg_list('h'),0)]); |
||
159 | } |
||
160 | /* Set up arguments */ |
||
161 | filen_kernel = argv[cmdline_getarg(cmdline_getarg_list('i'),0)]; |
||
162 | filen_files = argv[cmdline_getarg(cmdline_getarg_list('i'),1)]; |
||
163 | filen_out = argv[cmdline_getarg(cmdline_getarg_list('o'),0)]; |
||
164 | /* Command line arguments have been parsed. Start doing our work. */ |
||
165 | |||
166 | /* Caculate the header size, and allocate the memory, and assign the sub pointers */ |
||
167 | header_size = sizeof(struct nsp_img_hdr_head) + /* This has a single section |
||
168 | desc block already */ |
||
169 | (header_version==1?0:4) + |
||
170 | sizeof(struct nsp_img_hdr_info) + |
||
171 | sizeof(struct nsp_img_hdr_section_info) + |
||
172 | sizeof(struct nsp_img_hdr_sections) * num_sects ; |
||
173 | |||
174 | img_hdr_head = (struct nsp_img_hdr_head *)malloc(header_size); |
||
175 | memset(img_hdr_head, 0x0, header_size); |
||
176 | img_hdr_info = (struct nsp_img_hdr_info*)((char *)img_hdr_head + sizeof(struct nsp_img_hdr_head) + (header_version==1?0:4)); |
||
177 | img_hdr_section_info = (struct nsp_img_hdr_section_info*)((char *)img_hdr_info + sizeof(struct nsp_img_hdr_info)); |
||
178 | img_hdr_sections = (struct nsp_img_hdr_sections*)((char *)img_hdr_section_info + sizeof(struct nsp_img_hdr_section_info)); |
||
179 | section = img_hdr_sections; |
||
180 | memset(img_hdr_head, 0xff, (void*)img_hdr_info - (void*)img_hdr_head); |
||
181 | |||
182 | img_hdr_head->hdr_version = header_version; |
||
183 | img_hdr_head->hdr_size = header_size; |
||
184 | img_hdr_head->info_offset = (void*)img_hdr_info - (void*)img_hdr_head; |
||
185 | img_hdr_head->sect_info_offset = (void*)img_hdr_section_info - (void*)img_hdr_head; |
||
186 | |||
187 | img_hdr_section_info->num_sects = num_sects; |
||
188 | img_hdr_section_info->sect_size = sizeof(struct nsp_img_hdr_sections); |
||
189 | img_hdr_section_info->sections_offset = (void*)img_hdr_sections - (void*)img_hdr_head; |
||
190 | |||
191 | /* chksum = (struct nsp_img_hdr_chksum *) |
||
192 | ((unsigned int)image_hdr + header_size - sizeof(struct nsp_img_hdr_chksum));*/ |
||
193 | |||
194 | /* Open the out file */ |
||
195 | nsp_image = fopen(filen_out,"wb+"); |
||
196 | if(nsp_image==NULL) { |
||
197 | printf("ERROR: can't open %s for writing.\n", filen_out); |
||
198 | return -1; |
||
199 | } |
||
200 | |||
201 | /* Skip image header. We'll come back to it after we've written out the images. */ |
||
202 | fseek(nsp_image,header_size,SEEK_SET); |
||
203 | total = ftell(nsp_image); |
||
204 | total = header_size; |
||
205 | printf("total=%x\n",total); |
||
206 | { |
||
207 | int align; |
||
208 | int padding; |
||
209 | char * buf; |
||
210 | align = (header_version==1?0x10000:0x4000); |
||
211 | if(align==0) { |
||
212 | /* The user indicated no padding */ |
||
213 | padding = 0; |
||
214 | } else { |
||
215 | /* Calculate number padding bytes */ |
||
216 | if((total %align) ==0) |
||
217 | padding=0; |
||
218 | else |
||
219 | padding = align - (total % align); |
||
220 | } |
||
221 | if(padding>0) |
||
222 | { |
||
223 | buf=malloc(padding); |
||
224 | memset(buf, 0xff, padding); |
||
225 | if(fwrite((void*)buf,1,padding,nsp_image)!=padding) { |
||
226 | printf("ERROR: can't write to %s.\n", filen_out); |
||
227 | free(buf); |
||
228 | return -1; |
||
229 | } |
||
230 | free(buf); |
||
231 | |||
232 | } |
||
233 | total+=padding; |
||
234 | |||
235 | |||
236 | } |
||
237 | /* Write out all specified images (with -i option) */ |
||
238 | for(i=0; i < num_sects; i++) { |
||
239 | char* file_name; /* input file name */ |
||
240 | FILE* filep; /* input file pointer */ |
||
241 | int padding; /* number of padding bytes to prepend */ |
||
242 | int align; /* align factor from command line */ |
||
243 | int result; /* intermediate result */ |
||
244 | char * buf; |
||
245 | |||
246 | /* Open the specified image for reading */ |
||
247 | file_name = argv[cmdline_getarg(cmdline_getarg_list('i'),i)]; |
||
248 | filep = fopen(file_name, "rb"); |
||
249 | if(filep==NULL) { |
||
250 | printf("ERROR: can't open file %s for reading.\n", file_name); |
||
251 | return -1; |
||
252 | } |
||
253 | section->flags = ~0x00; |
||
254 | /* Determine file size */ |
||
255 | fseek(filep,0,SEEK_END); |
||
256 | section->raw_size=ftell(filep); |
||
257 | fseek(filep,0,SEEK_SET); |
||
258 | cs_calc_sum(filep,(unsigned long *)§ion->chksum,0); |
||
259 | fseek(filep,0,SEEK_SET); |
||
260 | |||
261 | /* Retrieve the alignment constant */ |
||
262 | /* Set image offset from the beginning of the out file */ |
||
263 | section->offset=total;// + padding; |
||
264 | |||
265 | //total += padding; |
||
266 | |||
267 | /* Copy the image file into nsp_image */ |
||
268 | count = section->raw_size; |
||
269 | buf=malloc(count); |
||
270 | result=fread(buf, 1, count, filep); |
||
271 | fwrite(buf, 1, result, nsp_image); |
||
272 | free(buf); |
||
273 | |||
274 | /* HACK: This is a hack to get the names and types to the files. |
||
275 | TODO: Fix this to be a real method */ |
||
276 | if(i==0){ |
||
277 | section->type=NSP_IMG_SECTION_TYPE_KERNEL; |
||
278 | strncpy(section->name, "kernel", 16); |
||
279 | } else if(i==1){ |
||
280 | section->type=NSP_IMG_SECTION_TYPE_FILESYSTEM_ROOT; |
||
281 | strncpy(section->name, "root", 16); |
||
282 | } |
||
283 | |||
284 | /* Account for the total */ |
||
285 | align = strtoul(argv[cmdline_getarg(cmdline_getarg_list('a'),i)],NULL,0); |
||
286 | if(i==0){ |
||
287 | if(align==0 || (((section->raw_size+ section->offset)%align)==0)) |
||
288 | padding=0; |
||
289 | else |
||
290 | padding = align - ((section->raw_size+ section->offset) % align); |
||
291 | |||
292 | section->total_size=section->raw_size + padding; |
||
293 | } |
||
294 | else{ |
||
295 | #define EXTRA_BLOCK 0x10000 |
||
296 | unsigned int squash_padding; |
||
297 | squash_padding = EXTRA_BLOCK - section->raw_size % EXTRA_BLOCK; |
||
298 | buf=malloc(EXTRA_BLOCK + 4); |
||
299 | memset(buf, 0, squash_padding); |
||
300 | fwrite(buf, 1, squash_padding, nsp_image); |
||
301 | memset(buf, 0, EXTRA_BLOCK + 4); |
||
302 | *((unsigned int *)buf)=0xdec0adde; |
||
303 | *((unsigned int *)(buf+EXTRA_BLOCK))=0xdec0adde; |
||
304 | fwrite(buf, 1, EXTRA_BLOCK+4, nsp_image); |
||
305 | free(buf); |
||
306 | |||
307 | if(align==0 || (((section->raw_size + (EXTRA_BLOCK + 4 + squash_padding)) %align)==0)) |
||
308 | padding=0; |
||
309 | else |
||
310 | padding = align - ((section->raw_size + (EXTRA_BLOCK + 4 + squash_padding)) % align); |
||
311 | section->total_size=section->raw_size + (EXTRA_BLOCK + 4 + squash_padding) + padding; |
||
312 | } |
||
313 | if(padding>0){ |
||
314 | buf=malloc(padding); |
||
315 | memset(buf, 0xff, padding); |
||
316 | fwrite(buf, 1, padding, nsp_image); |
||
317 | free(buf); |
||
318 | } |
||
319 | printf("*****padding is %d\ttotal_size=%d\traw_size=%d\n",padding, section->total_size, section->raw_size); |
||
320 | |||
321 | //total += section->raw_size; |
||
322 | total = section->total_size + section->offset; |
||
323 | printf("total=0x%x\n",total); |
||
324 | /* Close the input file */ |
||
325 | fclose(filep); |
||
326 | |||
327 | /* Move the section pointer to the next slot */ |
||
328 | section++; |
||
329 | } |
||
330 | |||
331 | /* Take care of the NSP image header fields */ |
||
332 | |||
333 | /* head fields */ |
||
334 | img_hdr_head->magic = NSP_IMG_MAGIC_NUMBER; |
||
335 | img_hdr_head->boot_offset = img_hdr_sections->offset; |
||
336 | img_hdr_head->flags = ~0x00; /* Set to all 1's */ |
||
337 | |||
338 | if(cmdline_getopt_count('b')) |
||
339 | img_hdr_head->flags &= ~(NSP_IMG_FLAG_FAILBACK_5 | NSP_IMG_FLAG_FAILBACK_1); |
||
340 | |||
341 | if(cmdline_getopt_count('f')) |
||
342 | img_hdr_head->flags = strtoul(argv[cmdline_getarg(cmdline_getarg_list('f'),0)], 0, 16); |
||
343 | |||
344 | #if 0 |
||
345 | img_hdr_head->hdr_version = 2; |
||
346 | img_hdr_head->hdr_size = header_size; |
||
347 | #endif |
||
348 | |||
349 | if(cmdline_getopt_count('p')) |
||
350 | img_hdr_head->prod_id = strtoul(argv[cmdline_getarg(cmdline_getarg_list('p'),0)], 0, 16); |
||
351 | else |
||
352 | img_hdr_head->prod_id = 0x4C575943; |
||
353 | |||
354 | if(cmdline_getopt_count('r')) |
||
355 | img_hdr_head->rel_id = strtoul(argv[cmdline_getarg(cmdline_getarg_list('r'),0)], 0, 0); |
||
356 | else |
||
357 | img_hdr_head->rel_id = 0x10203040; |
||
358 | |||
359 | if(cmdline_getopt_count('s')) |
||
360 | img_hdr_head->version = strtoul(argv[cmdline_getarg(cmdline_getarg_list('s'),0)], 0, 0); |
||
361 | else |
||
362 | img_hdr_head->version = 0x0b040000; |
||
363 | img_hdr_head->image_size = total; |
||
364 | #if 0 |
||
365 | img_hdr_head->info_offset = (unsigned int)(&(image_hdr->info)) - |
||
366 | (unsigned int)image_hdr; |
||
367 | img_hdr_head->sect_info_offset= (unsigned int)(&(image_hdr->sect_info)) - |
||
368 | (unsigned int)image_hdr; |
||
369 | #endif |
||
370 | // image_hdr->head.chksum_offset = (unsigned int)chksum - (unsigned int)image_hdr; |
||
371 | img_hdr_head->chksum_offset = 0xffffffff; |
||
372 | // image_hdr->head.pad1 = 0xffffffff; |
||
373 | /* info fields */ |
||
374 | /* TODO: Fix. Do nothing yet */ |
||
375 | // strncpy(nsp_img_hdr.id.prod_info,NSP_PRODINFO_STRING,sizeof(NSP_PRODINFO_STRING)); |
||
376 | strcpy(img_hdr_info->image_filename, (const char *)basename(filen_out)); |
||
377 | /* section fields */ |
||
378 | #if 0 |
||
379 | img_hdr_section_info->num_sects= num_sects; |
||
380 | img_hdr_section_info->sect_size= sizeof(struct nsp_img_hdr_sections); |
||
381 | img_hdr_section_info->sections_offset= (unsigned int)(&(image_hdr->sections)) - |
||
382 | (unsigned int)image_hdr; |
||
383 | #endif |
||
384 | |||
385 | /* Calculate checksum(s) */ |
||
386 | #if 0 |
||
387 | chksum->hdr_chksum = cs_calc_buf_sum((char*)image_hdr, |
||
388 | header_size - sizeof(struct nsp_img_hdr_chksum)); |
||
389 | #endif |
||
390 | /* Write out the NSP header. */ |
||
391 | fseek(nsp_image,0,SEEK_SET); |
||
392 | count = fwrite((void*)img_hdr_head, header_size, 1, nsp_image); |
||
393 | if(count!=1) { |
||
394 | printf("ERROR: can't write to %s.\n", filen_out); |
||
395 | return -1; |
||
396 | } |
||
397 | |||
398 | /* Check if -v option was specified (no arg needed) */ |
||
399 | if(cmdline_getopt_count('v') > 0) |
||
400 | { |
||
401 | struct nsp_img_hdr_head head; |
||
402 | struct nsp_img_hdr *hdr; |
||
403 | |||
404 | /* Rewind the file back to the beginning */ |
||
405 | fseek(nsp_image,0,SEEK_SET); |
||
406 | |||
407 | /* Read header from the file */ |
||
408 | fread((void*)&head, sizeof(struct nsp_img_hdr_head), |
||
409 | 1, nsp_image); |
||
410 | |||
411 | /* Get memory to store the complete header */ |
||
412 | hdr = (struct nsp_img_hdr *)malloc(head.hdr_size); |
||
413 | |||
414 | /* Read header from the file */ |
||
415 | fseek(nsp_image,0,SEEK_SET); |
||
416 | fread((void*)hdr, head.hdr_size, 1, nsp_image); |
||
417 | |||
418 | /* Print it out */ |
||
419 | mknspimg_print_hdr(hdr); |
||
420 | printf("Generated total %d bytes\n",total); |
||
421 | free(hdr); |
||
422 | } |
||
423 | |||
424 | free(img_hdr_head); |
||
425 | |||
426 | { |
||
427 | struct checksumrecord cr; |
||
428 | cr.magic=CKSUM_MAGIC_NUMBER; |
||
429 | cs_calc_sum(nsp_image, (unsigned long *)&cr.chksum, 0); |
||
430 | fseek(nsp_image,0, SEEK_END); |
||
431 | fwrite(&cr, 1, sizeof(cr), nsp_image); |
||
432 | } |
||
433 | { |
||
434 | FILE * non_web; |
||
435 | char fname[256]; |
||
436 | char * img_buf; |
||
437 | unsigned int len; |
||
438 | strcpy(fname, filen_out); |
||
439 | strcat(fname, ".non_web"); |
||
440 | non_web = fopen(fname,"wb+"); |
||
441 | fseek(nsp_image, 0, SEEK_END); |
||
442 | len = ftell(nsp_image); |
||
443 | img_buf=malloc(len); |
||
444 | fseek(nsp_image, 0, SEEK_SET); |
||
445 | fread(img_buf, 1, len, nsp_image); |
||
446 | img_buf[0xb] = 0x17; |
||
447 | fwrite(img_buf, 1, len-sizeof(struct checksumrecord), non_web); |
||
448 | fclose(non_web); |
||
449 | free(img_buf); |
||
450 | } |
||
451 | /* Close NSP image file */ |
||
452 | fclose(nsp_image); |
||
453 | |||
454 | /* return result */ |
||
455 | return(0); |
||
456 | } |
||
457 | |||
458 | #ifdef DMALLOC |
||
459 | #include <dmalloc.h> |
||
460 | #endif /* DMALLOC */ |
||
461 | |||
462 | #define BUFLEN (1 << 16) |
||
463 | |||
464 | static unsigned long crctab[256] = |
||
465 | { |
||
466 | 0x0, |
||
467 | 0x04C11DB7, 0x09823B6E, 0x0D4326D9, 0x130476DC, 0x17C56B6B, |
||
468 | 0x1A864DB2, 0x1E475005, 0x2608EDB8, 0x22C9F00F, 0x2F8AD6D6, |
||
469 | 0x2B4BCB61, 0x350C9B64, 0x31CD86D3, 0x3C8EA00A, 0x384FBDBD, |
||
470 | 0x4C11DB70, 0x48D0C6C7, 0x4593E01E, 0x4152FDA9, 0x5F15ADAC, |
||
471 | 0x5BD4B01B, 0x569796C2, 0x52568B75, 0x6A1936C8, 0x6ED82B7F, |
||
472 | 0x639B0DA6, 0x675A1011, 0x791D4014, 0x7DDC5DA3, 0x709F7B7A, |
||
473 | 0x745E66CD, 0x9823B6E0, 0x9CE2AB57, 0x91A18D8E, 0x95609039, |
||
474 | 0x8B27C03C, 0x8FE6DD8B, 0x82A5FB52, 0x8664E6E5, 0xBE2B5B58, |
||
475 | 0xBAEA46EF, 0xB7A96036, 0xB3687D81, 0xAD2F2D84, 0xA9EE3033, |
||
476 | 0xA4AD16EA, 0xA06C0B5D, 0xD4326D90, 0xD0F37027, 0xDDB056FE, |
||
477 | 0xD9714B49, 0xC7361B4C, 0xC3F706FB, 0xCEB42022, 0xCA753D95, |
||
478 | 0xF23A8028, 0xF6FB9D9F, 0xFBB8BB46, 0xFF79A6F1, 0xE13EF6F4, |
||
479 | 0xE5FFEB43, 0xE8BCCD9A, 0xEC7DD02D, 0x34867077, 0x30476DC0, |
||
480 | 0x3D044B19, 0x39C556AE, 0x278206AB, 0x23431B1C, 0x2E003DC5, |
||
481 | 0x2AC12072, 0x128E9DCF, 0x164F8078, 0x1B0CA6A1, 0x1FCDBB16, |
||
482 | 0x018AEB13, 0x054BF6A4, 0x0808D07D, 0x0CC9CDCA, 0x7897AB07, |
||
483 | 0x7C56B6B0, 0x71159069, 0x75D48DDE, 0x6B93DDDB, 0x6F52C06C, |
||
484 | 0x6211E6B5, 0x66D0FB02, 0x5E9F46BF, 0x5A5E5B08, 0x571D7DD1, |
||
485 | 0x53DC6066, 0x4D9B3063, 0x495A2DD4, 0x44190B0D, 0x40D816BA, |
||
486 | 0xACA5C697, 0xA864DB20, 0xA527FDF9, 0xA1E6E04E, 0xBFA1B04B, |
||
487 | 0xBB60ADFC, 0xB6238B25, 0xB2E29692, 0x8AAD2B2F, 0x8E6C3698, |
||
488 | 0x832F1041, 0x87EE0DF6, 0x99A95DF3, 0x9D684044, 0x902B669D, |
||
489 | 0x94EA7B2A, 0xE0B41DE7, 0xE4750050, 0xE9362689, 0xEDF73B3E, |
||
490 | 0xF3B06B3B, 0xF771768C, 0xFA325055, 0xFEF34DE2, 0xC6BCF05F, |
||
491 | 0xC27DEDE8, 0xCF3ECB31, 0xCBFFD686, 0xD5B88683, 0xD1799B34, |
||
492 | 0xDC3ABDED, 0xD8FBA05A, 0x690CE0EE, 0x6DCDFD59, 0x608EDB80, |
||
493 | 0x644FC637, 0x7A089632, 0x7EC98B85, 0x738AAD5C, 0x774BB0EB, |
||
494 | 0x4F040D56, 0x4BC510E1, 0x46863638, 0x42472B8F, 0x5C007B8A, |
||
495 | 0x58C1663D, 0x558240E4, 0x51435D53, 0x251D3B9E, 0x21DC2629, |
||
496 | 0x2C9F00F0, 0x285E1D47, 0x36194D42, 0x32D850F5, 0x3F9B762C, |
||
497 | 0x3B5A6B9B, 0x0315D626, 0x07D4CB91, 0x0A97ED48, 0x0E56F0FF, |
||
498 | 0x1011A0FA, 0x14D0BD4D, 0x19939B94, 0x1D528623, 0xF12F560E, |
||
499 | 0xF5EE4BB9, 0xF8AD6D60, 0xFC6C70D7, 0xE22B20D2, 0xE6EA3D65, |
||
500 | 0xEBA91BBC, 0xEF68060B, 0xD727BBB6, 0xD3E6A601, 0xDEA580D8, |
||
501 | 0xDA649D6F, 0xC423CD6A, 0xC0E2D0DD, 0xCDA1F604, 0xC960EBB3, |
||
502 | 0xBD3E8D7E, 0xB9FF90C9, 0xB4BCB610, 0xB07DABA7, 0xAE3AFBA2, |
||
503 | 0xAAFBE615, 0xA7B8C0CC, 0xA379DD7B, 0x9B3660C6, 0x9FF77D71, |
||
504 | 0x92B45BA8, 0x9675461F, 0x8832161A, 0x8CF30BAD, 0x81B02D74, |
||
505 | 0x857130C3, 0x5D8A9099, 0x594B8D2E, 0x5408ABF7, 0x50C9B640, |
||
506 | 0x4E8EE645, 0x4A4FFBF2, 0x470CDD2B, 0x43CDC09C, 0x7B827D21, |
||
507 | 0x7F436096, 0x7200464F, 0x76C15BF8, 0x68860BFD, 0x6C47164A, |
||
508 | 0x61043093, 0x65C52D24, 0x119B4BE9, 0x155A565E, 0x18197087, |
||
509 | 0x1CD86D30, 0x029F3D35, 0x065E2082, 0x0B1D065B, 0x0FDC1BEC, |
||
510 | 0x3793A651, 0x3352BBE6, 0x3E119D3F, 0x3AD08088, 0x2497D08D, |
||
511 | 0x2056CD3A, 0x2D15EBE3, 0x29D4F654, 0xC5A92679, 0xC1683BCE, |
||
512 | 0xCC2B1D17, 0xC8EA00A0, 0xD6AD50A5, 0xD26C4D12, 0xDF2F6BCB, |
||
513 | 0xDBEE767C, 0xE3A1CBC1, 0xE760D676, 0xEA23F0AF, 0xEEE2ED18, |
||
514 | 0xF0A5BD1D, 0xF464A0AA, 0xF9278673, 0xFDE69BC4, 0x89B8FD09, |
||
515 | 0x8D79E0BE, 0x803AC667, 0x84FBDBD0, 0x9ABC8BD5, 0x9E7D9662, |
||
516 | 0x933EB0BB, 0x97FFAD0C, 0xAFB010B1, 0xAB710D06, 0xA6322BDF, |
||
517 | 0xA2F33668, 0xBCB4666D, 0xB8757BDA, 0xB5365D03, 0xB1F740B4 |
||
518 | }; |
||
519 | |||
520 | int cs_is_tagged(FILE *fp) |
||
521 | { |
||
522 | char buf[8]; |
||
523 | |||
524 | fseek(fp, -8, SEEK_END); |
||
525 | fread(buf, 8, 1, fp); |
||
526 | if(*(unsigned long*)buf == CKSUM_MAGIC_NUMBER) |
||
527 | return 1; |
||
528 | return 0; |
||
529 | } |
||
530 | |||
531 | unsigned long cs_read_sum(FILE *fp) |
||
532 | { |
||
533 | char buf[8]; |
||
534 | |||
535 | fseek(fp, -8, SEEK_END); |
||
536 | fread(buf, 8, 1, fp); |
||
537 | return *((unsigned long*)&buf[4]); |
||
538 | } |
||
539 | |||
540 | int cs_calc_sum(FILE *fp, unsigned long *res, int tagged) |
||
541 | { |
||
542 | unsigned char buf[BUFLEN]; |
||
543 | unsigned long crc = 0; |
||
544 | uintmax_t length = 0; |
||
545 | size_t bytes_read; |
||
546 | |||
547 | fseek(fp, 0, SEEK_SET); |
||
548 | |||
549 | while((bytes_read = fread(buf, 1, BUFLEN, fp)) > 0) |
||
550 | { |
||
551 | unsigned char *cp = buf; |
||
552 | |||
553 | if(length + bytes_read < length) |
||
554 | return 0; |
||
555 | |||
556 | if(bytes_read != BUFLEN && tagged) |
||
557 | bytes_read -= 8; |
||
558 | |||
559 | length += bytes_read; |
||
560 | while(bytes_read--) |
||
561 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ *cp++) & 0xFF]; |
||
562 | } |
||
563 | |||
564 | if(ferror(fp)) |
||
565 | return 0; |
||
566 | |||
567 | for(; length; length >>= 8) |
||
568 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ length) & 0xFF]; |
||
569 | |||
570 | crc = ~crc & 0xFFFFFFFF; |
||
571 | |||
572 | *res = crc; |
||
573 | |||
574 | return 1; |
||
575 | } |
||
576 | |||
577 | unsigned long cs_calc_buf_sum(char *buf, int size) |
||
578 | { |
||
579 | unsigned long crc = 0; |
||
580 | char *cp = buf; |
||
581 | unsigned long length = size; |
||
582 | |||
583 | while(size--) |
||
584 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ *cp++) & 0xFF]; |
||
585 | |||
586 | for(; length; length >>= 8) |
||
587 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ length) & 0xFF]; |
||
588 | |||
589 | crc = ~crc & 0xFFFFFFFF; |
||
590 | |||
591 | return crc; |
||
592 | } |
||
593 | |||
594 | unsigned long cs_calc_buf_sum_ds(char *buf, int buf_size, char *sign, int sign_len) |
||
595 | { |
||
596 | unsigned long crc = 0; |
||
597 | char *cp = buf; |
||
598 | unsigned long length = buf_size+sign_len; |
||
599 | |||
600 | while(buf_size--) |
||
601 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ *cp++) & 0xFF]; |
||
602 | |||
603 | cp = sign; |
||
604 | while(sign_len--) |
||
605 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ *cp++) & 0xFF]; |
||
606 | |||
607 | |||
608 | for(; length; length >>= 8) |
||
609 | crc =(crc << 8) ^ crctab[((crc >> 24) ^ length) & 0xFF]; |
||
610 | |||
611 | crc = ~crc & 0xFFFFFFFF; |
||
612 | |||
613 | return crc; |
||
614 | } |
||
615 | |||
616 | int cs_set_sum(FILE *fp, unsigned long sum, int tagged) |
||
617 | { |
||
618 | unsigned long magic = CKSUM_MAGIC_NUMBER; |
||
619 | |||
620 | if(tagged) |
||
621 | fseek(fp, -8, SEEK_END); |
||
622 | else |
||
623 | fseek(fp, 0, SEEK_END); |
||
624 | |||
625 | if(fwrite(&magic, 1, 4, fp) < 4) |
||
626 | return 0; |
||
627 | if(fwrite(&sum, 1, 4, fp) < 4) |
||
628 | return 0; |
||
629 | |||
630 | return 1; |
||
631 | } |
||
632 | |||
633 | void cs_get_sum(FILE *fp, unsigned long *sum) |
||
634 | { |
||
635 | unsigned long magic = 0; |
||
636 | |||
637 | fseek(fp, -8, SEEK_END); |
||
638 | |||
639 | fread(&magic, 4, 1, fp); |
||
640 | fread(sum, 4, 1, fp); |
||
641 | } |
||
642 | |||
643 | int cs_validate_file(char *filename) |
||
644 | { |
||
645 | FILE *pFile = NULL; |
||
646 | unsigned long sum = 0, res = 0; |
||
647 | |||
648 | if((pFile = fopen(filename, "r")) == NULL) |
||
649 | return 0; |
||
650 | |||
651 | if(!cs_is_tagged(pFile)) |
||
652 | { |
||
653 | fclose(pFile); |
||
654 | return 0; |
||
655 | } |
||
656 | if(!cs_calc_sum(pFile, &sum, 1)) |
||
657 | { |
||
658 | fclose(pFile); |
||
659 | return 0; |
||
660 | } |
||
661 | cs_get_sum(pFile, &res); |
||
662 | fclose(pFile); |
||
663 | |||
664 | if(sum != res) |
||
665 | return 0; |
||
666 | return 1; |
||
667 | } |
||
668 | |||
669 | /* ********* Library internal data ********* */ |
||
670 | #define CMDLINE_TRUE 1 |
||
671 | #define CMDLINE_FALSE 0 |
||
672 | |||
673 | typedef enum CMDLINE_ERR |
||
674 | { |
||
675 | CMDLINE_ERR_OK = 0, /* No Error (OK) */ |
||
676 | CMDLINE_ERR_ERROR = -1, /* Unspecified error */ |
||
677 | CMDLINE_ERR_INVKEY = -3, /* Invalid option key */ |
||
678 | CMDLINE_ERR_MANYARG = -4, /* Too many arguments */ |
||
679 | CMDLINE_ERR_FEWARG = -5, /* Too few arguments */ |
||
680 | CMDLINE_ERR_ILLOPT = -6, /* Option not allowed (illegal option) */ |
||
681 | CMDLINE_ERR_NOMEM = -7, /* No memory */ |
||
682 | CMDLINE_ERR_OPTMIS = -8 /* A mandatory option is missing */ |
||
683 | } CMDLINE_ERR; |
||
684 | |||
685 | /* Argument list */ |
||
686 | typedef struct CMDLINE_ARG |
||
687 | { |
||
688 | int index; /* Index of the argument in the command line */ |
||
689 | struct CMDLINE_ARG* p_next; /* Next node in the linked list */ |
||
690 | } CMDLINE_ARG; |
||
691 | |||
692 | /* Master control block for an option */ |
||
693 | typedef struct CMDLINE_ARGS |
||
694 | { |
||
695 | int argc; /* Total count of arguments found */ |
||
696 | int optc; /* Total count of options found */ |
||
697 | CMDLINE_ARG* list; /* Argument list */ |
||
698 | } CMDLINE_ARGS; |
||
699 | |||
700 | /* Master control block for all found arguments */ |
||
701 | typedef struct CMDLINE_DATA |
||
702 | { |
||
703 | CMDLINE_ARGS opt_args[26]; /* Array of MCBs for each option ('a' through 'z') */ |
||
704 | CMDLINE_ARGS glb_args; /* Global arguments */ |
||
705 | int parsed; /* Internal flag to prevent client calls if library is not initialized */ |
||
706 | } CMDLINE_DATA; |
||
707 | |||
708 | /* ********* Local Data ********* */ |
||
709 | static CMDLINE_CFG cmdline_cfg; |
||
710 | static CMDLINE_DATA cmdline_data; |
||
711 | |||
712 | char* cmdline_errmsg = "CMDLINE ERROR"; |
||
713 | |||
714 | /* *************************************************************** |
||
715 | * Print all found command line options and their arguments |
||
716 | ****************************************************************** */ |
||
717 | void* cmdline_getarg_list(char opt) |
||
718 | { |
||
719 | int index = (opt - 'a'); |
||
720 | |||
721 | /* Check the validity of the index */ |
||
722 | if((index < 0) || (index > 25)) |
||
723 | { |
||
724 | /* ERROR: Wrong option */ |
||
725 | return NULL; |
||
726 | } |
||
727 | |||
728 | /* Return a pointer to the ARGS control structure */ |
||
729 | return((void*)(&cmdline_data.opt_args[index])); |
||
730 | } |
||
731 | |||
732 | /* *************************************************************** |
||
733 | * Print all found command line options and their arguments |
||
734 | ****************************************************************** */ |
||
735 | int cmdline_getarg_count(void* list) |
||
736 | { |
||
737 | CMDLINE_ARGS* p_args = (CMDLINE_ARGS*)list; |
||
738 | |||
739 | /* Return number of arguments for this option */ |
||
740 | return(p_args->argc); |
||
741 | } |
||
742 | |||
743 | /* *************************************************************** |
||
744 | * Print all found command line options and their arguments |
||
745 | ****************************************************************** */ |
||
746 | int cmdline_getopt_count(char opt) |
||
747 | { |
||
748 | int index; |
||
749 | |||
750 | /* Calculate index value */ |
||
751 | index = opt - 'a'; |
||
752 | if(index < 0 || index > 25) return -1; |
||
753 | |||
754 | /* Return number of arguments for this option */ |
||
755 | return(cmdline_data.opt_args[index].optc); |
||
756 | } |
||
757 | |||
758 | /* *************************************************************** |
||
759 | * Print all found command line options and their arguments |
||
760 | ****************************************************************** */ |
||
761 | int cmdline_getarg(void* list, int num) |
||
762 | { |
||
763 | int i; |
||
764 | CMDLINE_ARGS* p_args = (CMDLINE_ARGS*)list; |
||
765 | CMDLINE_ARG* p_arg; |
||
766 | |||
767 | /* Search the 'num' argument in the list for this option */ |
||
768 | for(i=0,p_arg=p_args->list; (p_arg!=NULL) && (i<p_args->argc); i++, p_arg=p_arg->p_next) |
||
769 | { |
||
770 | /* if num matches i, we found it */ |
||
771 | if(i==num) return(p_arg->index); |
||
772 | } |
||
773 | /* We did not find the specified argument or the list was empty */ |
||
774 | return -1; |
||
775 | } |
||
776 | |||
777 | /* *************************************************************** |
||
778 | * Print all found command line options and their arguments |
||
779 | ****************************************************************** */ |
||
780 | int cmdline_configure(CMDLINE_CFG* p_cfg) |
||
781 | { |
||
782 | /* reset global data */ |
||
783 | memset(&cmdline_cfg,0,sizeof(cmdline_cfg)); |
||
784 | memset(&cmdline_data,0,sizeof(cmdline_data)); |
||
785 | |||
786 | /* Copy the user's config structure */ |
||
787 | cmdline_cfg = *p_cfg; |
||
788 | return 0; |
||
789 | } |
||
790 | |||
791 | /* *************************************************************** |
||
792 | * Print all found command line options and their arguments |
||
793 | ****************************************************************** */ |
||
794 | char* cmdline_error(int err) |
||
795 | { |
||
796 | /* TODO: implement a table of error messages */ |
||
797 | return(cmdline_errmsg); |
||
798 | } |
||
799 | |||
800 | /* *************************************************************** |
||
801 | * Print all found command line options and their arguments |
||
802 | ****************************************************************** */ |
||
803 | static void cmdline_print_args(CMDLINE_ARGS* p_arglist, char* argv[]) |
||
804 | { |
||
805 | CMDLINE_ARG* p_arg; |
||
806 | |||
807 | printf(" Number of times option was specified: %d\n", p_arglist->optc); |
||
808 | printf(" Number of Arguments: %d\n", p_arglist->argc); |
||
809 | |||
810 | if(p_arglist->argc > 0) |
||
811 | { |
||
812 | printf(" Argument List: "); |
||
813 | |||
814 | for(p_arg=p_arglist->list; p_arg != NULL; p_arg=p_arg->p_next) |
||
815 | printf("%s ", argv[p_arg->index]); |
||
816 | } |
||
817 | |||
818 | printf("\n"); |
||
819 | } |
||
820 | |||
821 | /* *************************************************************** |
||
822 | * Print all found command line options and their arguments |
||
823 | ****************************************************************** */ |
||
824 | void cmdline_print(char* argv[]) |
||
825 | { |
||
826 | int i; |
||
827 | |||
828 | /* Check if the command line was parsed */ |
||
829 | if(cmdline_data.parsed != CMDLINE_TRUE) |
||
830 | { |
||
831 | printf("The command line has not been parsed yet.\n"); |
||
832 | return; |
||
833 | } |
||
834 | |||
835 | /* Print out option arguments */ |
||
836 | for( i = 0; i < 26; i++ ) |
||
837 | { |
||
838 | /* Check if the option was specified */ |
||
839 | if(cmdline_data.opt_args[i].optc !=0 ) |
||
840 | { |
||
841 | /* Print out option name and arguments */ |
||
842 | printf("Option: -%c\n", (char)('a'+i)); |
||
843 | cmdline_print_args(&(cmdline_data.opt_args[i]), argv); |
||
844 | } |
||
845 | } |
||
846 | |||
847 | /* Print out global arguments */ |
||
848 | printf("Global arguments:\n"); |
||
849 | cmdline_print_args(&(cmdline_data.glb_args), argv); |
||
850 | } |
||
851 | |||
852 | /* *************************************************************** |
||
853 | * Print configuration |
||
854 | ****************************************************************** */ |
||
855 | void cmdline_print_cfg(void) |
||
856 | { |
||
857 | |||
858 | } |
||
859 | |||
860 | static void cmdline_argadd(CMDLINE_ARGS* p_arglist, CMDLINE_ARG* p_arg) |
||
861 | { |
||
862 | CMDLINE_ARG* p_list; |
||
863 | CMDLINE_ARG* p_prev=NULL; |
||
864 | |||
865 | /* See if we had anything in the list */ |
||
866 | if(p_arglist->argc == 0) |
||
867 | { |
||
868 | /* Link the argument in */ |
||
869 | p_arglist->list = p_arg; |
||
870 | } |
||
871 | else |
||
872 | { |
||
873 | /* Find the tail of the list */ |
||
874 | for(p_list=p_arglist->list; p_list != NULL; p_list=p_list->p_next) |
||
875 | p_prev = p_list; |
||
876 | |||
877 | /* Link the argument in */ |
||
878 | p_prev->p_next=p_arg; |
||
879 | } |
||
880 | |||
881 | /* Keep track of arg number */ |
||
882 | p_arglist->argc++; |
||
883 | } |
||
884 | |||
885 | /* *************************************************************** |
||
886 | * cmdline_read() |
||
887 | * Read and parse command line arguments |
||
888 | ****************************************************************** */ |
||
889 | int cmdline_read(int argc, char* argv[]) |
||
890 | { |
||
891 | int i, option=0; |
||
892 | |||
893 | /* Process every command line argument in argv[] array */ |
||
894 | for( i = 1; i < argc; i++ ) |
||
895 | { |
||
896 | /* Does the argument start with a dash? */ |
||
897 | if( *argv[i] == '-' ) |
||
898 | { |
||
899 | /* The argument must be two characters: a dash, and a letter */ |
||
900 | if( strlen(argv[i]) != 2 ) |
||
901 | { |
||
902 | /* ERROR: option syntax (needs to be a dash and one letter) */ |
||
903 | return(CMDLINE_ERR_ERROR); |
||
904 | } |
||
905 | |||
906 | /* Check validity of the option key ('a' through 'z') */ |
||
907 | if( ((*(argv[i] + 1)) < 'a') || ((*(argv[i] + 1)) > 'z') ) |
||
908 | { |
||
909 | /* ERROR: option sysntax (invalid option key) */ |
||
910 | return(CMDLINE_ERR_INVKEY); |
||
911 | } |
||
912 | |||
913 | /* Calculate the option index */ |
||
914 | option = (*(argv[i] + 1)) - 'a'; |
||
915 | if((option < 0) || (option > 25)) return(CMDLINE_ERR_INVKEY); |
||
916 | |||
917 | /* Check to see if the option is allowed */ |
||
918 | if( cmdline_cfg.opts[option].flags & CMDLINE_OPTFLAG_ALLOW ) |
||
919 | { |
||
920 | /* Option allowed. */ |
||
921 | cmdline_data.opt_args[option].optc++; |
||
922 | continue; |
||
923 | } |
||
924 | else |
||
925 | { |
||
926 | /* ERROR: Option is not allowed */ |
||
927 | return(CMDLINE_ERR_ILLOPT); |
||
928 | } |
||
929 | } |
||
930 | else |
||
931 | { |
||
932 | /* Read the arguments for the option */ |
||
933 | CMDLINE_ARG* p_arg; |
||
934 | |||
935 | /* Allocate space for the argument node */ |
||
936 | p_arg = (CMDLINE_ARG*)calloc(1,sizeof(CMDLINE_ARG)); |
||
937 | if( p_arg== NULL ) |
||
938 | { |
||
939 | /* ERROR: Can't allocate memory for the argument index */ |
||
940 | return(CMDLINE_ERR_NOMEM); |
||
941 | } |
||
942 | |||
943 | /* Initialize the argument */ |
||
944 | p_arg->index = i; |
||
945 | p_arg->p_next = NULL; |
||
946 | |||
947 | /* Check if we can add to the list of arguments for this option */ |
||
948 | if( (option < 0) /* Do we have to add to the global list? */ |
||
949 | || (cmdline_data.opt_args[option].argc == cmdline_cfg.opts[option].max) /* Did we reach MAX arguments? */ |
||
950 | ) |
||
951 | { |
||
952 | /* This option does not require arguments. Keep the argument in the global list. */ |
||
953 | cmdline_argadd(&(cmdline_data.glb_args), p_arg); |
||
954 | continue; |
||
955 | } |
||
956 | else |
||
957 | { |
||
958 | /* See if the current count has reached max for this option */ |
||
959 | if( cmdline_data.opt_args[option].argc == cmdline_cfg.opts[option].max ) |
||
960 | { |
||
961 | /* ERROR: too many arguments for an option */ |
||
962 | return(CMDLINE_ERR_MANYARG); |
||
963 | } |
||
964 | else |
||
965 | { |
||
966 | /* Link the argument to the arg list of the option */ |
||
967 | cmdline_argadd(&(cmdline_data.opt_args[option]), p_arg); |
||
968 | continue; |
||
969 | } |
||
970 | } |
||
971 | } |
||
972 | } |
||
973 | |||
974 | /* ****** We read the complete command line. See if what we collected matches the configuration ******* */ |
||
975 | |||
976 | /* Check every collected option against its configuration */ |
||
977 | for( i=0; i < 26; i++ ) |
||
978 | { |
||
979 | /* Check if this option was allowed */ |
||
980 | if(cmdline_cfg.opts[i].flags & CMDLINE_OPTFLAG_ALLOW) |
||
981 | { |
||
982 | /* See if it was mandatory */ |
||
983 | if(cmdline_cfg.opts[i].flags & CMDLINE_OPTFLAG_MANDAT) |
||
984 | { |
||
985 | /* Check if we really collected this option on the command line. */ |
||
986 | if(cmdline_data.opt_args[i].optc == 0) |
||
987 | { |
||
988 | /* ERROR: a missing mandatory option */ |
||
989 | return(CMDLINE_ERR_OPTMIS); |
||
990 | } |
||
991 | else |
||
992 | { |
||
993 | /* Option was there. Check how many args we got for it. */ |
||
994 | if(cmdline_data.opt_args[i].argc < cmdline_cfg.opts[i].min) |
||
995 | { |
||
996 | /* ERROR: too few arguments for an option */ |
||
997 | return(CMDLINE_ERR_FEWARG); |
||
998 | } |
||
999 | else |
||
1000 | { |
||
1001 | /* This mandatory option was proper. */ |
||
1002 | continue; |
||
1003 | } |
||
1004 | } |
||
1005 | } |
||
1006 | else /* This is non-mandatory option: */ |
||
1007 | { |
||
1008 | /* Check if the option was specified on the command line */ |
||
1009 | if(cmdline_data.opt_args[i].optc == 0) |
||
1010 | { |
||
1011 | /* option wasn't specified, go to the next */ |
||
1012 | continue; |
||
1013 | } |
||
1014 | else |
||
1015 | { |
||
1016 | /* Option was there. Check how many args we collected for it. */ |
||
1017 | if(cmdline_data.opt_args[i].argc < cmdline_cfg.opts[i].min) |
||
1018 | { |
||
1019 | /* ERROR: too few arguments for a non-mandatory option */ |
||
1020 | return(CMDLINE_ERR_FEWARG); |
||
1021 | } |
||
1022 | else |
||
1023 | { |
||
1024 | /* This non-mandatory option was proper. */ |
||
1025 | continue; |
||
1026 | } |
||
1027 | } |
||
1028 | } |
||
1029 | } |
||
1030 | else /* Option was not allowed. */ |
||
1031 | { |
||
1032 | /* We should not get here as the non-allowed options should have been |
||
1033 | trapped eariler. */ |
||
1034 | } |
||
1035 | } |
||
1036 | |||
1037 | /* Command line was proper as far as the number of options and their arguments */ |
||
1038 | cmdline_data.parsed = CMDLINE_TRUE; |
||
1039 | return(CMDLINE_ERR_OK); |
||
1040 | } |