114 lines
1.7 KiB
Go
114 lines
1.7 KiB
Go
|
|
package rtmp
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type dummyWriter struct {
|
|
}
|
|
|
|
func (m *dummyWriter) Write(p []byte) (n int, err error) {
|
|
return 0, nil
|
|
}
|
|
|
|
type logger int
|
|
|
|
func (l logger) Printf(format string, v ...interface{}) {
|
|
str := fmt.Sprintf(format, v...)
|
|
switch {
|
|
case strings.HasPrefix(str, "server") && l >= 0,
|
|
strings.HasPrefix(str, "stream") && l >= 0,
|
|
strings.HasPrefix(str, "event") && l >= 0,
|
|
strings.HasPrefix(str, "data") && l >= 0,
|
|
strings.HasPrefix(str, "msg") && l >= 1:
|
|
l2.Println(str)
|
|
}
|
|
}
|
|
|
|
var (
|
|
dummyW = &dummyWriter{}
|
|
l = logger(0)
|
|
l2 *log.Logger
|
|
)
|
|
|
|
func init() {
|
|
l2 = log.New(os.Stderr, "", 0)
|
|
l2.SetFlags(log.Lmicroseconds)
|
|
}
|
|
|
|
type stream struct {
|
|
r io.ReadWriteCloser
|
|
}
|
|
|
|
func (s stream) Read(p []byte) (n int, err error) {
|
|
n, err = s.r.Read(p)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s stream) Write(p []byte) (n int, err error) {
|
|
n, err = s.r.Write(p)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s stream) Close() {
|
|
s.r.Close()
|
|
}
|
|
|
|
func ReadBuf(r io.Reader, n int) (b []byte) {
|
|
b = make([]byte, n)
|
|
r.Read(b)
|
|
return
|
|
}
|
|
|
|
func ReadInt(r io.Reader, n int) (ret int) {
|
|
b := ReadBuf(r, n)
|
|
for i := 0; i < n; i++ {
|
|
ret <<= 8
|
|
ret += int(b[i])
|
|
}
|
|
return
|
|
}
|
|
|
|
func ReadIntLE(r io.Reader, n int) (ret int) {
|
|
b := ReadBuf(r, n)
|
|
for i := 0; i < n; i++ {
|
|
ret <<= 8
|
|
ret += int(b[n-i-1])
|
|
}
|
|
return
|
|
}
|
|
|
|
func WriteBuf(w io.Writer, buf []byte) {
|
|
w.Write(buf)
|
|
}
|
|
|
|
func WriteInt(w io.Writer, v int, n int) {
|
|
b := make([]byte, n)
|
|
for i := 0; i < n; i++ {
|
|
b[n-i-1] = byte(v&0xff)
|
|
v >>= 8
|
|
}
|
|
WriteBuf(w, b)
|
|
}
|
|
|
|
func WriteIntLE(w io.Writer, v int, n int) {
|
|
b := make([]byte, n)
|
|
for i := 0; i < n; i++ {
|
|
b[i] = byte(v&0xff)
|
|
v >>= 8
|
|
}
|
|
WriteBuf(w, b)
|
|
}
|
|
|