@@ -23,86 +23,17 @@ def FullOTA_InstallEnd(info):
2323def IncrementalOTA_InstallEnd (info ):
2424 OTA_InstallEnd (info , True )
2525
26- def AddImageOnly (info , basename , incremental , firmware ):
26+ def AddImage (info , basename , dest , incremental ):
27+ name = basename
2728 if incremental :
2829 input_zip = info .source_zip
2930 else :
3031 input_zip = info .input_zip
31- if firmware :
32- data = input_zip .read ("RADIO/" + basename )
33- else :
34- data = input_zip .read ("IMAGES/" + basename )
35- common .ZipWriteStr (info .output_zip , basename , data )
36-
37- def AddImage (info , basename , dest , incremental ):
38- AddImageOnly (info , basename , incremental , False )
39- info .script .AppendExtra ('package_extract_file("%s", "%s");' % (basename , dest ))
32+ data = input_zip .read ("IMAGES/" + basename )
33+ common .ZipWriteStr (info .output_zip , name , data )
34+ info .script .AppendExtra ('package_extract_file("%s", "%s");' % (name , dest ))
4035
4136def OTA_InstallEnd (info , incremental ):
4237 info .script .Print ("Patching vbmeta and dtbo images..." )
4338 AddImage (info , "vbmeta.img" , "/dev/block/by-name/vbmeta" , incremental )
4439 AddImage (info , "dtbo.img" , "/dev/block/by-name/dtbo" , incremental )
45- Firmware_Images (info , incremental )
46-
47- def Firmware_Images (info , incremental ):
48- bin_map = {
49- 'logo' : ['logo' ]
50- }
51-
52- img_map = {
53- 'audio_dsp' : ['audio_dsp' ],
54- 'cam_vpu1' : ['cam_vpu1' ],
55- 'cam_vpu2' : ['cam_vpu2' ],
56- 'cam_vpu3' : ['cam_vpu3' ],
57- 'gz' : ['gz1' , 'gz2' ],
58- 'lk' : ['lk' , 'lk2' ],
59- 'md1img' : ['md1img' ],
60- 'scp' : ['scp1' , 'scp2' ],
61- 'spmfw' : ['spmfw' ],
62- 'sspm' : ['sspm_1' , 'sspm_2' ],
63- 'tee' : ['tee1' , 'tee2' ]
64- }
65-
66- pl = 'preloader_ufs'
67- pl_part = ['sda' , 'sdb' ]
68-
69- fw_cmd = 'ifelse(getprop("ro.boot.hwc") == "India",\n (\n '
70- fw_cmd += 'ui_print("Flashing begoniain (Indian) firmware...");\n '
71-
72- # Flash Indian Firmware
73- AddImageOnly (info , "{}_in.img" .format (pl ), incremental , True )
74- for part in pl_part :
75- fw_cmd += 'package_extract_file("{}_in.img", "/dev/block/{}");\n ' .format (pl , part )
76-
77- for img in img_map .keys ():
78- AddImageOnly (info , '{}_in.img' .format (img ), incremental , True )
79- for part in img_map [img ]:
80- fw_cmd += 'package_extract_file("{}_in.img", "/dev/block/bootdevice/by-name/{}");\n ' .format (img , part )
81-
82- for _bin in bin_map .keys ():
83- AddImageOnly (info , '{}_in.bin' .format (_bin ), incremental , True )
84- for part in bin_map [_bin ]:
85- fw_cmd += 'package_extract_file("{}_in.bin", "/dev/block/bootdevice/by-name/{}");\n ' .format (_bin , part )
86- # END Flash Indian Firmware
87-
88- fw_cmd += '),\n (\n '
89- fw_cmd += 'ui_print("Flashing begonia (Global) firmware...");\n '
90-
91- # Flash Global Firmware
92- AddImageOnly (info , "{}.img" .format (pl ), incremental , True )
93- for part in pl_part :
94- fw_cmd += 'package_extract_file("{}.img", "/dev/block/{}");\n ' .format (pl , part )
95-
96- for img in img_map .keys ():
97- AddImageOnly (info , '{}.img' .format (img ), incremental , True )
98- for part in img_map [img ]:
99- fw_cmd += 'package_extract_file("{}.img", "/dev/block/bootdevice/by-name/{}");\n ' .format (img , part )
100-
101- for _bin in bin_map .keys ():
102- AddImageOnly (info , '{}.bin' .format (_bin ), incremental , True )
103- for part in bin_map [_bin ]:
104- fw_cmd += 'package_extract_file("{}.bin", "/dev/block/bootdevice/by-name/{}");\n ' .format (_bin , part )
105- # END Flash Global Firmware
106-
107- fw_cmd += ')\n );'
108- info .script .AppendExtra (fw_cmd )
0 commit comments