dns.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. package constant
  2. import (
  3. "encoding/json"
  4. "errors"
  5. )
  6. // DNSModeMapping is a mapping for EnhancedMode enum
  7. var DNSModeMapping = map[string]DNSMode{
  8. DNSNormal.String(): DNSNormal,
  9. DNSFakeIP.String(): DNSFakeIP,
  10. DNSMapping.String(): DNSMapping,
  11. }
  12. const (
  13. DNSNormal DNSMode = iota
  14. DNSFakeIP
  15. DNSMapping
  16. DNSHosts
  17. )
  18. type DNSMode int
  19. // UnmarshalYAML unserialize EnhancedMode with yaml
  20. func (e *DNSMode) UnmarshalYAML(unmarshal func(any) error) error {
  21. var tp string
  22. if err := unmarshal(&tp); err != nil {
  23. return err
  24. }
  25. mode, exist := DNSModeMapping[tp]
  26. if !exist {
  27. return errors.New("invalid mode")
  28. }
  29. *e = mode
  30. return nil
  31. }
  32. // MarshalYAML serialize EnhancedMode with yaml
  33. func (e DNSMode) MarshalYAML() (any, error) {
  34. return e.String(), nil
  35. }
  36. // UnmarshalJSON unserialize EnhancedMode with json
  37. func (e *DNSMode) UnmarshalJSON(data []byte) error {
  38. var tp string
  39. json.Unmarshal(data, &tp)
  40. mode, exist := DNSModeMapping[tp]
  41. if !exist {
  42. return errors.New("invalid mode")
  43. }
  44. *e = mode
  45. return nil
  46. }
  47. // MarshalJSON serialize EnhancedMode with json
  48. func (e DNSMode) MarshalJSON() ([]byte, error) {
  49. return json.Marshal(e.String())
  50. }
  51. func (e DNSMode) String() string {
  52. switch e {
  53. case DNSNormal:
  54. return "normal"
  55. case DNSFakeIP:
  56. return "fake-ip"
  57. case DNSMapping:
  58. return "redir-host"
  59. case DNSHosts:
  60. return "hosts"
  61. default:
  62. return "unknown"
  63. }
  64. }
  65. type DNSPrefer int
  66. const (
  67. DualStack DNSPrefer = iota
  68. IPv4Only
  69. IPv6Only
  70. IPv4Prefer
  71. IPv6Prefer
  72. )
  73. var dnsPreferMap = map[string]DNSPrefer{
  74. DualStack.String(): DualStack,
  75. IPv4Only.String(): IPv4Only,
  76. IPv6Only.String(): IPv6Only,
  77. IPv4Prefer.String(): IPv4Prefer,
  78. IPv6Prefer.String(): IPv6Prefer,
  79. }
  80. func (d DNSPrefer) String() string {
  81. switch d {
  82. case DualStack:
  83. return "dual"
  84. case IPv4Only:
  85. return "ipv4"
  86. case IPv6Only:
  87. return "ipv6"
  88. case IPv4Prefer:
  89. return "ipv4-prefer"
  90. case IPv6Prefer:
  91. return "ipv6-prefer"
  92. default:
  93. return "dual"
  94. }
  95. }
  96. func NewDNSPrefer(prefer string) DNSPrefer {
  97. if p, ok := dnsPreferMap[prefer]; ok {
  98. return p
  99. } else {
  100. return DualStack
  101. }
  102. }
  103. type HTTPVersion string
  104. const (
  105. // HTTPVersion11 is HTTP/1.1.
  106. HTTPVersion11 HTTPVersion = "http/1.1"
  107. // HTTPVersion2 is HTTP/2.
  108. HTTPVersion2 HTTPVersion = "h2"
  109. // HTTPVersion3 is HTTP/3.
  110. HTTPVersion3 HTTPVersion = "h3"
  111. )