Fixed PES data parsing

asticode · Sep 17, 2019 · e47b3a7 · e47b3a7
1 parent af4a358
commit e47b3a7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/data_pes.go b/data_pes.go
@@ -123,7 +123,7 @@ func parsePESData(i *astibyte.Iterator) (d *PESData, err error) {
 
 	// Parse data
 	if d.Header != nil && d.Header.PacketLength > 0 {
-		if d.Data, err = i.NextBytes(int(d.Header.PacketLength)); err != nil {
+		if d.Data, err = i.NextBytes(int(d.Header.PacketLength)-dataStart); err != nil {
 			err = errors.Wrap(err, "astits: getting next bytes failed")
 			return
 		}

diff --git a/data_pes_test.go b/data_pes_test.go
@@ -222,10 +222,10 @@ func TestParsePESSection(t *testing.T) {
 	// No optional header and specific packet length
 	d, err := parsePESData(astibyte.NewIterator(pesWithoutHeaderBytes()))
 	assert.NoError(t, err)
-	assert.Equal(t, d, pesWithoutHeader)
+	assert.Equal(t, pesWithoutHeader, d)
 
 	// Optional header and no specific header length
 	d, err = parsePESData(astibyte.NewIterator(pesWithHeaderBytes()))
 	assert.NoError(t, err)
-	assert.Equal(t, d, pesWithHeader)
+	assert.Equal(t, pesWithHeader, d)
 }