67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
|
package constant
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
var StackTypeMapping = map[string]TUNStack{
|
||
|
|
strings.ToUpper(TunGvisor.String()): TunGvisor,
|
||
|
|
strings.ToUpper(TunSystem.String()): TunSystem,
|
||
|
|
}
|
||
|
|
|
||
|
|
const (
|
||
|
|
TunGvisor TUNStack = iota
|
||
|
|
TunSystem
|
||
|
|
)
|
||
|
|
|
||
|
|
type TUNStack int
|
||
|
|
|
||
|
|
// UnmarshalYAML unserialize TUNStack with yaml
|
||
|
|
func (e *TUNStack) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
||
|
|
var tp string
|
||
|
|
if err := unmarshal(&tp); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
mode, exist := StackTypeMapping[strings.ToUpper(tp)]
|
||
|
|
if !exist {
|
||
|
|
return errors.New("invalid tun stack")
|
||
|
|
}
|
||
|
|
*e = mode
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalYAML serialize TUNStack with yaml
|
||
|
|
func (e TUNStack) MarshalYAML() (interface{}, error) {
|
||
|
|
return e.String(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnmarshalJSON unserialize TUNStack with json
|
||
|
|
func (e *TUNStack) UnmarshalJSON(data []byte) error {
|
||
|
|
var tp string
|
||
|
|
json.Unmarshal(data, &tp)
|
||
|
|
mode, exist := StackTypeMapping[strings.ToUpper(tp)]
|
||
|
|
if !exist {
|
||
|
|
return errors.New("invalid tun stack")
|
||
|
|
}
|
||
|
|
*e = mode
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalJSON serialize TUNStack with json
|
||
|
|
func (e TUNStack) MarshalJSON() ([]byte, error) {
|
||
|
|
return json.Marshal(e.String())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e TUNStack) String() string {
|
||
|
|
switch e {
|
||
|
|
case TunGvisor:
|
||
|
|
return "gVisor"
|
||
|
|
case TunSystem:
|
||
|
|
return "System"
|
||
|
|
default:
|
||
|
|
return "unknown"
|
||
|
|
}
|
||
|
|
}
|