mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-07-04 13:05:24 +00:00
33 lines
669 B
Go
33 lines
669 B
Go
package internet
|
|
|
|
import "v2ray.com/core/common"
|
|
|
|
type PacketHeader interface {
|
|
Size() int
|
|
Write([]byte) (int, error)
|
|
}
|
|
|
|
type PacketHeaderFactory interface {
|
|
Create(interface{}) PacketHeader
|
|
}
|
|
|
|
var (
|
|
headerCache = make(map[string]PacketHeaderFactory)
|
|
)
|
|
|
|
func RegisterPacketHeader(name string, factory PacketHeaderFactory) error {
|
|
if _, found := headerCache[name]; found {
|
|
return common.ErrDuplicatedName
|
|
}
|
|
headerCache[name] = factory
|
|
return nil
|
|
}
|
|
|
|
func CreatePacketHeader(name string, config interface{}) (PacketHeader, error) {
|
|
factory, found := headerCache[name]
|
|
if !found {
|
|
return nil, common.ErrObjectNotFound
|
|
}
|
|
return factory.Create(config), nil
|
|
}
|