mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-12 06:54:23 -04:00
49 lines
951 B
Go
49 lines
951 B
Go
|
package tls
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
"v2ray.com/core/common"
|
||
|
"v2ray.com/core/common/dice"
|
||
|
)
|
||
|
|
||
|
// DTLS writes header as DTLS. See https://tools.ietf.org/html/rfc6347
|
||
|
type DTLS struct {
|
||
|
epoch uint16
|
||
|
sequence uint32
|
||
|
}
|
||
|
|
||
|
// Size implements PacketHeader.
|
||
|
func (*DTLS) Size() int32 {
|
||
|
return 1 + 2 + 2 + 3 + 2
|
||
|
}
|
||
|
|
||
|
// Write implements PacketHeader.
|
||
|
func (d *DTLS) Write(b []byte) (int, error) {
|
||
|
b[0] = 23 // application data
|
||
|
b[1] = 254
|
||
|
b[2] = 253
|
||
|
b[3] = byte(d.epoch >> 8)
|
||
|
b[4] = byte(d.epoch)
|
||
|
b[5] = byte(d.sequence >> 16)
|
||
|
b[6] = byte(d.sequence >> 8)
|
||
|
b[7] = byte(d.sequence)
|
||
|
d.sequence++
|
||
|
l := dice.RollUint16()
|
||
|
b[8] = byte(l >> 8)
|
||
|
b[9] = byte(l)
|
||
|
return 10, nil
|
||
|
}
|
||
|
|
||
|
// New creates a new UTP header for the given config.
|
||
|
func New(ctx context.Context, config interface{}) (interface{}, error) {
|
||
|
return &DTLS{
|
||
|
epoch: dice.RollUint16(),
|
||
|
sequence: 0,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
common.Must(common.RegisterConfig((*PacketConfig)(nil), New))
|
||
|
}
|