From 308910a8fd2dd03732bd2c7020fbbc84aef271a0 Mon Sep 17 00:00:00 2001
From: Kurochan <kuro@kurochan.org>
Date: Tue, 15 Oct 2024 00:28:48 +0900
Subject: [PATCH] Check if the frame is writable before copyPlanes (#84)

* check if the frame is writable before copyPlanes

* apply review

* make f2 writable
---
 frame_data.go      | 11 ++++++++---
 frame_data_test.go |  9 +++++++++
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/frame_data.go b/frame_data.go
index 30e795e..273e714 100644
--- a/frame_data.go
+++ b/frame_data.go
@@ -280,7 +280,12 @@ func (f *frameDataFrame) bytes(align int) ([]byte, error) {
 	return nil, errors.New("astiav: frame type not implemented")
 }
 
-func (f *frameDataFrame) copyPlanes(ps []frameDataPlane) (err error) {
+func (f *frameDataFrame) copyPlanes(ps []frameDataPlane) error {
+	// Check writability
+	if !f.f.IsWritable() {
+		return errors.New("astiav: frame is not writable")
+	}
+
 	switch {
 	// Video
 	case f.height() > 0 && f.width() > 0:
@@ -300,9 +305,9 @@ func (f *frameDataFrame) copyPlanes(ps []frameDataPlane) (err error) {
 
 		// Copy image
 		C.av_image_copy(&f.f.c.data[0], &f.f.c.linesize[0], &cdata[0], &clinesizes[0], (C.enum_AVPixelFormat)(f.f.c.format), f.f.c.width, f.f.c.height)
-		return
+		return nil
 	}
-	return
+	return nil
 }
 
 func (f *frameDataFrame) height() int {
diff --git a/frame_data_test.go b/frame_data_test.go
index db0f30a..968528c 100644
--- a/frame_data_test.go
+++ b/frame_data_test.go
@@ -518,5 +518,14 @@ func TestFrameData(t *testing.T) {
 		require.NoError(t, err)
 		b9 := []byte(fmt.Sprintf("%+v", b8))
 		require.Equal(t, b3, b9)
+
+		f3 := AllocFrame()
+		defer f3.Free()
+		require.NoError(t, f3.Ref(f2))
+		require.Error(t, fd2.FromImage(i1))
+		require.Error(t, fd2.SetBytes(b1, align))
+		f2.MakeWritable()
+		require.NoError(t, fd2.FromImage(i1))
+		require.NoError(t, fd2.SetBytes(b1, align))
 	}
 }