mirror of
https://github.com/encounter/go-rtmp.git
synced 2026-03-30 11:12:49 -07:00
yutopp
74 lines
1.2 KiB
Go
74 lines
1.2 KiB
Go
//
|
|
// Copyright (c) 2018- yutopp (yutopp@gmail.com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
package handshake
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
)
|
|
|
|
type Encoder struct {
|
|
w io.Writer
|
|
}
|
|
|
|
func NewEncoder(w io.Writer) *Encoder {
|
|
return &Encoder{
|
|
w: w,
|
|
}
|
|
}
|
|
|
|
func (e *Encoder) EncodeS0C0(h *S0C0) error {
|
|
buf := [1]byte{byte(*h)}
|
|
|
|
_, err := e.w.Write(buf[:])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (e *Encoder) EncodeS1C1(h *S1C1) error {
|
|
buf := [4]byte{}
|
|
|
|
binary.BigEndian.PutUint32(buf[:], h.Time)
|
|
if _, err := e.w.Write(buf[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := e.w.Write(h.Version[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := e.w.Write(h.Random[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (e *Encoder) EncodeS2C2(h *S2C2) error {
|
|
buf := [4]byte{}
|
|
|
|
binary.BigEndian.PutUint32(buf[:], h.Time)
|
|
if _, err := e.w.Write(buf[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
binary.BigEndian.PutUint32(buf[:], h.Time2)
|
|
if _, err := e.w.Write(buf[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := e.w.Write(h.Random[:]); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|