config_json.go 912 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. // +build json
  2. package blackhole
  3. import (
  4. "encoding/json"
  5. "errors"
  6. "github.com/golang/protobuf/proto"
  7. "v2ray.com/core/common/loader"
  8. )
  9. func (this *Config) UnmarshalJSON(data []byte) error {
  10. type JSONConfig struct {
  11. Response json.RawMessage `json:"response"`
  12. }
  13. jsonConfig := new(JSONConfig)
  14. if err := json.Unmarshal(data, jsonConfig); err != nil {
  15. return errors.New("Blackhole: Failed to parse config: " + err.Error())
  16. }
  17. if jsonConfig.Response != nil {
  18. response, _, err := configLoader.Load(jsonConfig.Response)
  19. if err != nil {
  20. return errors.New("Blackhole: Failed to parse response config: " + err.Error())
  21. }
  22. this.Response = loader.NewTypedSettings(response.(proto.Message))
  23. }
  24. return nil
  25. }
  26. var (
  27. configLoader = loader.NewJSONConfigLoader(
  28. loader.NamedTypeMap{
  29. "none": loader.GetType(new(NoneResponse)),
  30. "http": loader.GetType(new(HTTPResponse)),
  31. },
  32. "type",
  33. "")
  34. )