File tree 1 file changed +11
-5
lines changed
1 file changed +11
-5
lines changed Original file line number Diff line number Diff line change @@ -418,19 +418,25 @@ def main():
418
418
data = f .read ()
419
419
f .close ()
420
420
421
+ firmware = None
422
+
421
423
if sys .argv [2 ][- 4 :] == '.dfu' :
422
424
suf_len , vendor , product = dfu_suffix .check_suffix (data )
423
425
dfu = init_dfu ()
424
426
firmware = data [:- suf_len ]
425
- elif sys .argv [2 ][- 4 :] == '.rdt' and len (data ) == 262709 and data [0 :5 ] == 'DfuSe' :
426
- dfu = init_dfu ()
427
- firmware = data [549 :len (data ) - 16 ]
427
+ elif sys .argv [2 ][- 4 :] == '.rdt' :
428
+ if len (data ) == 262709 and data [0 :5 ] == 'DfuSe' :
429
+ dfu = init_dfu ()
430
+ firmware = data [549 :len (data ) - 16 ]
431
+ else :
432
+ print ('%s not a valid codeplug (wrong size, or wrong magic).' % sys .argv [2 ])
428
433
else :
429
434
dfu = init_dfu ()
430
435
firmware = data
431
436
432
- download_codeplug (dfu , firmware )
433
- print ('Write complete' )
437
+ if firmware is not None :
438
+ download_codeplug (dfu , firmware )
439
+ print ('Write complete' )
434
440
435
441
elif sys .argv [1 ] == 'sign' :
436
442
filename = sys .argv [2 ]
You can’t perform that action at this time.
0 commit comments