From e448a762ca584cea678b98a35acc8318546d308e Mon Sep 17 00:00:00 2001
From: Tomasz Andrzejak <andreiltd@gmail.com>
Date: Wed, 15 Dec 2021 21:06:45 +0000
Subject: [PATCH] gstreamer-rtp: Add payload_mut function to RTPBuffer

---
 gstreamer-rtp/src/rtp_buffer.rs | 42 +++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/gstreamer-rtp/src/rtp_buffer.rs b/gstreamer-rtp/src/rtp_buffer.rs
index 6595c0c89..091600aab 100644
--- a/gstreamer-rtp/src/rtp_buffer.rs
+++ b/gstreamer-rtp/src/rtp_buffer.rs
@@ -278,6 +278,23 @@ impl<'a, T> RTPBuffer<'a, T> {
         }
     }
 
+    #[doc(alias = "get_payload")]
+    #[doc(alias = "gst_rtp_buffer_get_payload")]
+    pub fn payload_mut(&mut self) -> Result<&mut [u8], glib::error::BoolError> {
+        let size = self.payload_size();
+        if size == 0 {
+            return Ok(&mut []);
+        }
+        unsafe {
+            let pointer = ffi::gst_rtp_buffer_get_payload(&mut self.rtp_buffer);
+            if pointer.is_null() {
+                Err(glib::bool_error!("Failed to get payload data"))
+            } else {
+                Ok(slice::from_raw_parts_mut(pointer as *mut u8, size as usize))
+            }
+        }
+    }
+
     #[doc(alias = "get_extension")]
     pub fn is_extension(&self) -> bool {
         unsafe {
@@ -467,6 +484,31 @@ mod tests {
         assert_eq!(payload.unwrap().len(), payload_size as usize);
     }
 
+    #[test]
+    fn test_mut_payload() {
+        gst::init().unwrap();
+
+        let csrc_count = 2;
+        let payload_size = 8;
+        let mut buffer = gst::Buffer::new_rtp_with_sizes(payload_size, 4, csrc_count).unwrap();
+        {
+            let buffer = buffer.get_mut().unwrap();
+            let mut rtp_buffer = RTPBuffer::from_buffer_writable(buffer).unwrap();
+
+            let payload = rtp_buffer.payload_mut();
+            assert!(payload.is_ok());
+
+            let payload = payload.unwrap();
+            payload[3] = 42;
+        }
+
+        let rtp_buffer = RTPBuffer::from_buffer_readable(&buffer).unwrap();
+        let payload = rtp_buffer.payload();
+
+        assert!(payload.is_ok());
+        assert_eq!(payload.unwrap()[3], 42);
+    }
+
     #[test]
     fn test_extension_header_onebyte() {
         gst::init().unwrap();