stream-server/rtmp/server.go

package rtmp

import (
	"net"
)

type ProtocolParams struct {
	peer_chunk_size uint32
	chunk_size uint32
	peer_ack_win uint32
	curr_read uint32
}

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) {
	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()
	if !DoHandshake(conn) {
		return
	}
	params := ProtocolParams{1024, 512, 0, 0}
	open_chnkstrms, open_msgs, chunk_buffers := OpenStreamsMapInit()
	if !NegotiateConnect(conn, &params, open_chnkstrms, open_msgs, chunk_buffers) {
		return
	}

	*stream_live = false
}