stream-server/rtmp/server.go

package rtmp

import (
	"net"
)

// meta-struct, holding together params for each streaing peer
type ProtocolParams struct {
	peer_chunk_size uint32
	chunk_size uint32
	peer_ack_win uint32
	curr_read uint32
	stream_key string
	trans_id float64
}

func NewServer(port string) (error) {
	l, err := net.Listen("tcp", ":" + port)
	if err != nil {
		return err
	}
	go start(l)
	return nil
}

func start(l net.Listener) {
	// bool to block connections while a stream is currently active
	// should have it on a per user basis instead, but adding a db
	// system would make this even more bloated
	stream_live := false
	for {
		conn, err := l.Accept()
		if err != nil {
			continue
		} else if stream_live {
			conn.Close()
			continue
		}
		stream_live = true
		go handle_conn(conn, &stream_live)
	}
}

func handle_conn(conn net.Conn, stream_live *bool) {
	defer conn.Close()
	defer func(a *bool) {
		*a = false
	}(stream_live) // flip the stream state back when done
	if !DoHandshake(conn) {
		return
	}
	chunk_wrapper := NewChunkWrapper(conn)
	if !NegotiateConnect(chunk_wrapper) {
		return
	}
	if !CreateStream(chunk_wrapper) {
		return
	}
	if !PublishAsKey(chunk_wrapper) {
		return
	}
	HandleDataLoop(chunk_wrapper) // no error handle since the connection ends either way
	go StreamCleanup(chunk_wrapper.params.stream_key, 10, stream_live) // remove any media files left 1 min after stream ends
}