aler9
6 years ago
16 changed files with 670 additions and 222 deletions
@ -0,0 +1,96 @@ |
|||||||
|
package servertls |
||||||
|
|
||||||
|
import ( |
||||||
|
"crypto/tls" |
||||||
|
"strconv" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/aler9/gortsplib" |
||||||
|
|
||||||
|
"github.com/aler9/rtsp-simple-server/internal/logger" |
||||||
|
) |
||||||
|
|
||||||
|
// Parent is implemented by program.
|
||||||
|
type Parent interface { |
||||||
|
Log(logger.Level, string, ...interface{}) |
||||||
|
} |
||||||
|
|
||||||
|
// Server is a TCP/TLS/RTSPS listener.
|
||||||
|
type Server struct { |
||||||
|
parent Parent |
||||||
|
|
||||||
|
srv *gortsplib.Server |
||||||
|
|
||||||
|
// out
|
||||||
|
accept chan *gortsplib.ServerConn |
||||||
|
done chan struct{} |
||||||
|
} |
||||||
|
|
||||||
|
// New allocates a Server.
|
||||||
|
func New(port int, |
||||||
|
readTimeout time.Duration, |
||||||
|
writeTimeout time.Duration, |
||||||
|
serverKey string, |
||||||
|
serverCert string, |
||||||
|
parent Parent) (*Server, error) { |
||||||
|
|
||||||
|
cert, err := tls.LoadX509KeyPair(serverCert, serverKey) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
conf := gortsplib.ServerConf{ |
||||||
|
TLSConfig: &tls.Config{Certificates: []tls.Certificate{cert}}, |
||||||
|
ReadTimeout: readTimeout, |
||||||
|
WriteTimeout: writeTimeout, |
||||||
|
ReadBufferCount: 1, |
||||||
|
} |
||||||
|
|
||||||
|
srv, err := conf.Serve(":" + strconv.FormatInt(int64(port), 10)) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
s := &Server{ |
||||||
|
parent: parent, |
||||||
|
srv: srv, |
||||||
|
accept: make(chan *gortsplib.ServerConn), |
||||||
|
done: make(chan struct{}), |
||||||
|
} |
||||||
|
|
||||||
|
parent.Log(logger.Info, "[TCP/TLS/RTSPS server] opened on :%d", port) |
||||||
|
|
||||||
|
go s.run() |
||||||
|
return s, nil |
||||||
|
} |
||||||
|
|
||||||
|
// Close closes a Server.
|
||||||
|
func (s *Server) Close() { |
||||||
|
go func() { |
||||||
|
for co := range s.accept { |
||||||
|
co.Close() |
||||||
|
} |
||||||
|
}() |
||||||
|
s.srv.Close() |
||||||
|
<-s.done |
||||||
|
} |
||||||
|
|
||||||
|
func (s *Server) run() { |
||||||
|
defer close(s.done) |
||||||
|
|
||||||
|
for { |
||||||
|
conn, err := s.srv.Accept() |
||||||
|
if err != nil { |
||||||
|
break |
||||||
|
} |
||||||
|
|
||||||
|
s.accept <- conn |
||||||
|
} |
||||||
|
|
||||||
|
close(s.accept) |
||||||
|
} |
||||||
|
|
||||||
|
// Accept returns a channel to accept incoming connections.
|
||||||
|
func (s *Server) Accept() chan *gortsplib.ServerConn { |
||||||
|
return s.accept |
||||||
|
} |
||||||
@ -0,0 +1,95 @@ |
|||||||
|
|
||||||
|
#include <gst/gst.h> |
||||||
|
#include <gst/video/video.h> |
||||||
|
|
||||||
|
GType gst_exitafterframe_get_type (); |
||||||
|
|
||||||
|
#define GST_TYPE_EXITAFTERFRAME (gst_exitafterframe_get_type()) |
||||||
|
#define GST_EXITAFTERFRAME(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_EXITAFTERFRAME,GstExitAfterFrame)) |
||||||
|
|
||||||
|
typedef struct |
||||||
|
{ |
||||||
|
GstElement element; |
||||||
|
GstPad *srcpad; |
||||||
|
GstPad *sinkpad; |
||||||
|
|
||||||
|
} GstExitAfterFrame; |
||||||
|
|
||||||
|
typedef struct |
||||||
|
{ |
||||||
|
GstElementClass parent_class; |
||||||
|
|
||||||
|
} GstExitAfterFrameClass; |
||||||
|
|
||||||
|
#define gst_exitafterframe_parent_class parent_class |
||||||
|
G_DEFINE_TYPE (GstExitAfterFrame, gst_exitafterframe, GST_TYPE_ELEMENT); |
||||||
|
|
||||||
|
static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE( |
||||||
|
"sink", |
||||||
|
GST_PAD_SINK, |
||||||
|
GST_PAD_ALWAYS, |
||||||
|
GST_STATIC_CAPS("video/x-raw") |
||||||
|
); |
||||||
|
|
||||||
|
static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE( |
||||||
|
"src", |
||||||
|
GST_PAD_SRC, |
||||||
|
GST_PAD_ALWAYS, |
||||||
|
GST_STATIC_CAPS("video/x-raw") |
||||||
|
); |
||||||
|
|
||||||
|
static GstFlowReturn |
||||||
|
gst_exitafterframe_chain (GstPad * pad, GstObject * parent, GstBuffer * buf) |
||||||
|
{ |
||||||
|
GstExitAfterFrame *filter = GST_EXITAFTERFRAME (parent); |
||||||
|
exit(0); |
||||||
|
return gst_pad_push (filter->srcpad, buf); |
||||||
|
} |
||||||
|
|
||||||
|
static void |
||||||
|
gst_exitafterframe_class_init(GstExitAfterFrameClass* klass) { |
||||||
|
GstElementClass* element_class = (GstElementClass*)klass; |
||||||
|
|
||||||
|
gst_element_class_set_details_simple( |
||||||
|
element_class, |
||||||
|
"Plugin", |
||||||
|
"FIXME:Generic", |
||||||
|
"FIXME:Generic Template Element", |
||||||
|
"AUTHOR_NAME AUTHOR_EMAIL" |
||||||
|
); |
||||||
|
|
||||||
|
gst_element_class_add_pad_template(element_class, |
||||||
|
gst_static_pad_template_get(&src_factory)); |
||||||
|
gst_element_class_add_pad_template(element_class, |
||||||
|
gst_static_pad_template_get(&sink_factory)); |
||||||
|
} |
||||||
|
|
||||||
|
static void |
||||||
|
gst_exitafterframe_init (GstExitAfterFrame* filter) |
||||||
|
{ |
||||||
|
GstElement* element = GST_ELEMENT(filter); |
||||||
|
|
||||||
|
g_print("[lasermqtt] init()\n"); |
||||||
|
|
||||||
|
filter->sinkpad = gst_pad_new_from_static_template(&sink_factory, "sink"); |
||||||
|
gst_pad_set_chain_function(filter->sinkpad, gst_exitafterframe_chain); |
||||||
|
GST_PAD_SET_PROXY_CAPS(filter->sinkpad); |
||||||
|
gst_element_add_pad(element, filter->sinkpad); |
||||||
|
|
||||||
|
filter->srcpad = gst_pad_new_from_static_template(&src_factory, "src"); |
||||||
|
GST_PAD_SET_PROXY_CAPS(filter->srcpad); |
||||||
|
gst_element_add_pad(element, filter->srcpad); |
||||||
|
} |
||||||
|
|
||||||
|
static gboolean |
||||||
|
plugin_init (GstPlugin * plugin) |
||||||
|
{ |
||||||
|
return gst_element_register (plugin, "exitafterframe", GST_RANK_NONE, |
||||||
|
GST_TYPE_EXITAFTERFRAME); |
||||||
|
} |
||||||
|
|
||||||
|
#define PACKAGE "exitafterframe" |
||||||
|
|
||||||
|
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR, exitafterframe, |
||||||
|
"exitafterframe", plugin_init, "1.0", "LGPL", "exitafterframe", |
||||||
|
"http://example.com") |
||||||
Loading…
Reference in new issue