fix bug: #151 out of range

caoyaping · caoyaping · commit 7dde65f704ee · 2024-09-05T01:00:46.000+08:00
diff --git a/go-codec/h265.go b/go-codec/h265.go
@@ -96,9 +96,11 @@ func (vps *VPS) Decode(nalu []byte) {
     }
     vps.Vps_max_layer_id = bs.Uint8(6)
     vps.Vps_num_layer_sets_minus1 = bs.ReadUE()
-    vps.Layer_id_included_flag = make([][]uint8, vps.Vps_num_layer_sets_minus1)
+    //  0 <=  Vps_num_layer_sets_minus1  <= 1023
+    //  0 <=  Vps_max_layer_id <= 63
+    vps.Layer_id_included_flag = make([][]uint8, vps.Vps_num_layer_sets_minus1 + 1)
     for i := 1; i <= int(vps.Vps_num_layer_sets_minus1); i++ {
-        vps.Layer_id_included_flag[i] = make([]uint8, vps.Vps_max_layer_id)
+        vps.Layer_id_included_flag[i] = make([]uint8, vps.Vps_max_layer_id + 1)
         for j := 0; j <= int(vps.Vps_max_layer_id); j++ {
             vps.Layer_id_included_flag[i][j] = bs.Uint8(1)
         }

Original file line number	Diff line number	Diff line change
`@@ -96,9 +96,11 @@ func (vps *VPS) Decode(nalu []byte) {`
`96`	`96`	`}`
`97`	`97`	`vps.Vps_max_layer_id = bs.Uint8(6)`
`98`	`98`	`vps.Vps_num_layer_sets_minus1 = bs.ReadUE()`
`99`		`- vps.Layer_id_included_flag = make([][]uint8, vps.Vps_num_layer_sets_minus1)`
	`99`	`+ // 0 <= Vps_num_layer_sets_minus1 <= 1023`
	`100`	`+ // 0 <= Vps_max_layer_id <= 63`
	`101`	`+ vps.Layer_id_included_flag = make([][]uint8, vps.Vps_num_layer_sets_minus1 + 1)`
`100`	`102`	`for i := 1; i <= int(vps.Vps_num_layer_sets_minus1); i++ {`
`101`		`- vps.Layer_id_included_flag[i] = make([]uint8, vps.Vps_max_layer_id)`
	`103`	`+ vps.Layer_id_included_flag[i] = make([]uint8, vps.Vps_max_layer_id + 1)`
`102`	`104`	`for j := 0; j <= int(vps.Vps_max_layer_id); j++ {`
`103`	`105`	`vps.Layer_id_included_flag[i][j] = bs.Uint8(1)`
`104`	`106`	`}`