av/rtmp/RTMP.go

/*
NAME
  RtpToTsConverter.go - provides utilities for the conversion of Rtp packets
  to equivalent MpegTs packets.

DESCRIPTION
  See Readme.md

AUTHOR
  Saxon Nelson-Milton <saxon.milton@gmail.com>

LICENSE
  RtpToTsConverter.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)

  It is free software: you can redistribute it and/or modify them
  under the terms of the GNU General Public License as published by the
  Free Software Foundation, either version 3 of the License, or (at your
  option) any later version.

  It is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 for more details.

  You should have received a copy of the GNU General Public License
  along with revid in gpl.txt.  If not, see [GNU licenses](http://www.gnu.org/licenses).
*/
package rtmp

// #cgo CFLAGS: -I/home/saxon/Desktop/AusOcean/av/rtmp/
// #cgo CFLAGS: -I/home/saxon/Desktop/AusOcean/av/rtmp/rtmp_c/librtmp
// #cgo LDFLAGS: /home/saxon/Desktop/AusOcean/av/rtmp/rtmp_c/librtmp/librtmp.a
// #cgo LDFLAGS: -lssl -lcrypto -lz
// #include <RTMPWrapper.h>
import "C"

import (
  "errors"
  _"unsafe"
  "fmt"
)

type RTMPSession interface {
  StartSession() error
  WriteFrame(data []byte, dataLength uint) error
  EndSession() error
}

type rtmpSession struct {
  rtmp *C.struct_RTMP
  url string
  timeout uint
}

func NewRTMPSession(url string, connectTimeout uint) (session *rtmpSession){
  session = new(rtmpSession)
  session.url = url
  session.timeout = connectTimeout
  return
}

func (s *rtmpSession) StartSession() error {
  fmt.Println("Starting session!")
  /*
  if !uintToBool(uint(C.RTMP_start_session(s.rtmp, C.CString(s.url), C.uint(s.timeout)))) {
    return errors.New("RTMP start error! Check rtmp log for details!")
  }
  */
  return nil
}

func (s *rtmpSession) WriteFrame(data []byte, dataLength uint) error {
  fmt.Println("writing frame!")
  /*
  if !uintToBool(uint(C.RTMP_write_frame(s.rtmp, (*C.char)(unsafe.Pointer(&data[0])), C.uint(dataLength)))) {
    return errors.New("RTMP write error! Check rtmp log for details!")
  }
  */
  return nil
}

func (s *rtmpSession) EndSession() error {
  if !uintToBool(uint(C.RTMP_end_session(s.rtmp))) {
    return errors.New("RTMP end session error! Check rtmp log for details!")
  }
  return nil
}

func uintToBool(x uint) bool {
  return x != 0
}