rtsp_server: Add support for subclassing RTSPServer and RTSPClient

gstreamer-rtsp-server/src/subclass/mod.rs

@@ -7,10 +7,14 @@
 // except according to those terms.
 #![allow(clippy::cast_ptr_alignment)]
 
+pub mod rtsp_client;
 pub mod rtsp_media;
 pub mod rtsp_media_factory;
+pub mod rtsp_server;
 
 pub mod prelude {
+    pub use super::rtsp_client::{RTSPClientImpl, RTSPClientImplExt};
     pub use super::rtsp_media::{RTSPMediaImpl, RTSPMediaImplExt};
     pub use super::rtsp_media_factory::{RTSPMediaFactoryImpl, RTSPMediaFactoryImplExt};
+    pub use super::rtsp_server::{RTSPServerImpl, RTSPServerImplExt};
 }

gstreamer-rtsp-server/src/subclass/rtsp_media_factory.rs

@@ -284,8 +284,7 @@ where
 
     let pipeline: *mut gst_sys::GstPipeline = imp
         .create_pipeline(&wrap, &from_glib_borrow(media))
-        .to_glib_none()
-        .0;
+        .to_glib_full();
 
     // FIXME We somehow need to ensure the pipeline actually stays alive...
     gobject_sys::g_object_set_qdata_full(

gstreamer-rtsp-server/src/subclass/rtsp_server.rs

@@ -0,0 +1,92 @@
+// Copyright (C) 2020 Sebastian Dröge <sebastian@centricular.com>
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use gst_rtsp_server_sys;
+
+use glib::subclass::prelude::*;
+use glib::translate::*;
+
+use RTSPServer;
+use RTSPServerClass;
+
+pub trait RTSPServerImpl: RTSPServerImplExt + ObjectImpl + Send + Sync + 'static {
+    fn create_client(&self, server: &RTSPServer) -> Option<::RTSPClient> {
+        self.parent_create_client(server)
+    }
+
+    fn client_connected(&self, server: &RTSPServer, client: &::RTSPClient) {
+        self.parent_client_connected(server, client);
+    }
+}
+
+pub trait RTSPServerImplExt {
+    fn parent_create_client(&self, server: &RTSPServer) -> Option<::RTSPClient>;
+
+    fn parent_client_connected(&self, server: &RTSPServer, client: &::RTSPClient);
+}
+
+impl<T: RTSPServerImpl + ObjectImpl> RTSPServerImplExt for T {
+    fn parent_create_client(&self, server: &RTSPServer) -> Option<::RTSPClient> {
+        unsafe {
+            let data = self.get_type_data();
+            let parent_class =
+                data.as_ref().get_parent_class() as *mut gst_rtsp_server_sys::GstRTSPServerClass;
+            let f = (*parent_class)
+                .create_client
+                .expect("No `create_client` virtual method implementation in parent class");
+            from_glib_full(f(server.to_glib_none().0))
+        }
+    }
+
+    fn parent_client_connected(&self, server: &RTSPServer, client: &::RTSPClient) {
+        unsafe {
+            let data = self.get_type_data();
+            let parent_class =
+                data.as_ref().get_parent_class() as *mut gst_rtsp_server_sys::GstRTSPServerClass;
+            if let Some(f) = (*parent_class).client_connected {
+                f(server.to_glib_none().0, client.to_glib_none().0)
+            }
+        }
+    }
+}
+unsafe impl<T: ObjectSubclass + RTSPServerImpl> IsSubclassable<T> for RTSPServerClass {
+    fn override_vfuncs(&mut self) {
+        <glib::ObjectClass as IsSubclassable<T>>::override_vfuncs(self);
+        unsafe {
+            let klass = &mut *(self as *mut Self as *mut gst_rtsp_server_sys::GstRTSPServerClass);
+            klass.create_client = Some(server_create_client::<T>);
+            klass.client_connected = Some(server_client_connected::<T>);
+        }
+    }
+}
+
+unsafe extern "C" fn server_create_client<T: ObjectSubclass>(
+    ptr: *mut gst_rtsp_server_sys::GstRTSPServer,
+) -> *mut gst_rtsp_server_sys::GstRTSPClient
+where
+    T: RTSPServerImpl,
+{
+    let instance = &*(ptr as *mut T::Instance);
+    let imp = instance.get_impl();
+    let wrap: RTSPServer = from_glib_borrow(ptr);
+
+    imp.create_client(&wrap).to_glib_full()
+}
+
+unsafe extern "C" fn server_client_connected<T: ObjectSubclass>(
+    ptr: *mut gst_rtsp_server_sys::GstRTSPServer,
+    client: *mut gst_rtsp_server_sys::GstRTSPClient,
+) where
+    T: RTSPServerImpl,
+{
+    let instance = &*(ptr as *mut T::Instance);
+    let imp = instance.get_impl();
+    let wrap: RTSPServer = from_glib_borrow(ptr);
+
+    imp.client_connected(&wrap, &from_glib_borrow(client));
+}