lzhr
/
v2fly
mirror of https://github.com/v2fly/v2ray-core.git
1
0
Fork 0
Code Issues Releases Wiki Activity

stack allocated buffer

This commit is contained in:
Darien Raymond 2018-11-15 16:04:13 +01:00
parent 24288a74a2
commit 770a20d266
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
2 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
common/protocol/address.go
View File

@ -2,11 +2,13 @@ package protocol
import (
"io"
"unsafe"
"v2ray.com/core/common"
"v2ray.com/core/common/buf"
"v2ray.com/core/common/net"
"v2ray.com/core/common/serial"
"v2ray.com/core/common/stack"
)
type AddressOption func(*option)
@ -232,7 +234,13 @@ func (p *addressParser) writeAddress(writer io.Writer, address net.Address) erro
switch address.Family() {
case net.AddressFamilyIPv4, net.AddressFamilyIPv6:
if _, err := writer.Write([]byte{tb}); err != nil {
var bytes stack.TwoBytes
s := bytes[:1]
s[0] = tb
p := uintptr(unsafe.Pointer(&s))
v := (*[]byte)(unsafe.Pointer(p))
if _, err := writer.Write(*v); err != nil {
return err
}
if _, err := writer.Write(address.IP()); err != nil {
@ -243,7 +251,15 @@ func (p *addressParser) writeAddress(writer io.Writer, address net.Address) erro
if isDomainTooLong(domain) {
return newError("Super long domain is not supported: ", domain)
}
if _, err := writer.Write([]byte{tb, byte(len(domain))}); err != nil {
var bytes stack.TwoBytes
s := bytes[:]
s[0] = tb
s[1] = byte(len(domain))
p := uintptr(unsafe.Pointer(&s))
v := (*[]byte)(unsafe.Pointer(p))
if _, err := writer.Write(*v); err != nil {
return err
}
if _, err := io.WriteString(writer, domain); err != nil {

11
common/stack/bytes.go Normal file
View File

@ -0,0 +1,11 @@
package stack
// TwoBytes is a [8]byte which is always allocated on stack.
//
//go:notinheap
type TwoBytes [2]byte
// EightBytes is a [8]byte which is always allocated on stack.
//
//go:notinheap
type EightBytes [8]byte