addition.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. package inbound
  2. import (
  3. "net"
  4. C "github.com/metacubex/mihomo/constant"
  5. )
  6. type Addition func(metadata *C.Metadata)
  7. func ApplyAdditions(metadata *C.Metadata, additions ...Addition) {
  8. for _, addition := range additions {
  9. addition(metadata)
  10. }
  11. }
  12. func WithInName(name string) Addition {
  13. return func(metadata *C.Metadata) {
  14. metadata.InName = name
  15. }
  16. }
  17. func WithInUser(user string) Addition {
  18. return func(metadata *C.Metadata) {
  19. metadata.InUser = user
  20. }
  21. }
  22. func WithSpecialRules(specialRules string) Addition {
  23. return func(metadata *C.Metadata) {
  24. metadata.SpecialRules = specialRules
  25. }
  26. }
  27. func WithSpecialProxy(specialProxy string) Addition {
  28. return func(metadata *C.Metadata) {
  29. metadata.SpecialProxy = specialProxy
  30. }
  31. }
  32. func WithDstAddr(addr net.Addr) Addition {
  33. return func(metadata *C.Metadata) {
  34. _ = metadata.SetRemoteAddr(addr)
  35. }
  36. }
  37. func WithSrcAddr(addr net.Addr) Addition {
  38. return func(metadata *C.Metadata) {
  39. m := C.Metadata{}
  40. if err := m.SetRemoteAddr(addr); err == nil {
  41. metadata.SrcIP = m.DstIP
  42. metadata.SrcPort = m.DstPort
  43. }
  44. }
  45. }
  46. func WithInAddr(addr net.Addr) Addition {
  47. return func(metadata *C.Metadata) {
  48. m := C.Metadata{}
  49. if err := m.SetRemoteAddr(addr); err == nil {
  50. metadata.InIP = m.DstIP
  51. metadata.InPort = m.DstPort
  52. }
  53. }
  54. }
  55. func WithDSCP(dscp uint8) Addition {
  56. return func(metadata *C.Metadata) {
  57. metadata.DSCP = dscp
  58. }
  59. }
  60. func Placeholder(metadata *C.Metadata) {}