1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-10-29 15:27:32 -04:00
v2fly/proxy/vmess/io/writer.go
2016-07-17 23:11:05 +02:00

37 lines
675 B
Go

package io
import (
"hash/fnv"
"github.com/v2ray/v2ray-core/common/alloc"
v2io "github.com/v2ray/v2ray-core/common/io"
)
type AuthChunkWriter struct {
writer v2io.Writer
}
func NewAuthChunkWriter(writer v2io.Writer) *AuthChunkWriter {
return &AuthChunkWriter{
writer: writer,
}
}
func (this *AuthChunkWriter) Write(buffer *alloc.Buffer) error {
Authenticate(buffer)
return this.writer.Write(buffer)
}
func (this *AuthChunkWriter) Release() {
this.writer.Release()
this.writer = nil
}
func Authenticate(buffer *alloc.Buffer) {
fnvHash := fnv.New32a()
fnvHash.Write(buffer.Value)
buffer.PrependHash(fnvHash)
buffer.PrependUint16(uint16(buffer.Len()))
}