mirror of
https://github.com/cmz0228/hysteria-dev.git
synced 2025-06-08 21:39:53 +00:00
105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/tobyxdd/hysteria/pkg/core/pb"
|
|
"io"
|
|
)
|
|
|
|
const (
|
|
closeErrorCodeGeneric = 0
|
|
closeErrorCodeProtocolFailure = 1
|
|
)
|
|
|
|
func readDataBlock(r io.Reader) ([]byte, error) {
|
|
var sz uint32
|
|
if err := binary.Read(r, controlProtocolEndian, &sz); err != nil {
|
|
return nil, err
|
|
}
|
|
buf := make([]byte, sz)
|
|
_, err := io.ReadFull(r, buf)
|
|
return buf, err
|
|
}
|
|
|
|
func writeDataBlock(w io.Writer, data []byte) error {
|
|
sz := uint32(len(data))
|
|
if err := binary.Write(w, controlProtocolEndian, &sz); err != nil {
|
|
return err
|
|
}
|
|
_, err := w.Write(data)
|
|
return err
|
|
}
|
|
|
|
func readClientAuthRequest(r io.Reader) (*pb.ClientAuthRequest, error) {
|
|
bs, err := readDataBlock(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var req pb.ClientAuthRequest
|
|
err = proto.Unmarshal(bs, &req)
|
|
return &req, err
|
|
}
|
|
|
|
func writeClientAuthRequest(w io.Writer, req *pb.ClientAuthRequest) error {
|
|
bs, err := proto.Marshal(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return writeDataBlock(w, bs)
|
|
}
|
|
|
|
func readServerAuthResponse(r io.Reader) (*pb.ServerAuthResponse, error) {
|
|
bs, err := readDataBlock(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var resp pb.ServerAuthResponse
|
|
err = proto.Unmarshal(bs, &resp)
|
|
return &resp, err
|
|
}
|
|
|
|
func writeServerAuthResponse(w io.Writer, resp *pb.ServerAuthResponse) error {
|
|
bs, err := proto.Marshal(resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return writeDataBlock(w, bs)
|
|
}
|
|
|
|
func readClientConnectRequest(r io.Reader) (*pb.ClientConnectRequest, error) {
|
|
bs, err := readDataBlock(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var req pb.ClientConnectRequest
|
|
err = proto.Unmarshal(bs, &req)
|
|
return &req, err
|
|
}
|
|
|
|
func writeClientConnectRequest(w io.Writer, req *pb.ClientConnectRequest) error {
|
|
bs, err := proto.Marshal(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return writeDataBlock(w, bs)
|
|
}
|
|
|
|
func readServerConnectResponse(r io.Reader) (*pb.ServerConnectResponse, error) {
|
|
bs, err := readDataBlock(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var resp pb.ServerConnectResponse
|
|
err = proto.Unmarshal(bs, &resp)
|
|
return &resp, err
|
|
}
|
|
|
|
func writeServerConnectResponse(w io.Writer, resp *pb.ServerConnectResponse) error {
|
|
bs, err := proto.Marshal(resp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return writeDataBlock(w, bs)
|
|
}
|