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
	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) {
	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
	}
	chunk_wrapper := NewChunkWrapper(conn)
	if !NegotiateConnect(chunk_wrapper) {
		return
	}
	if !CreateStream(chunk_wrapper) {
		return
	}
	*stream_live = false
}