update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package jsonpb 
 
import ( 
    "bytes" 
    "io" 
 
    "github.com/golang/protobuf/jsonpb" 
    "github.com/golang/protobuf/proto" 
 
    core "github.com/v2fly/v2ray-core/v5" 
    "github.com/v2fly/v2ray-core/v5/common" 
    "github.com/v2fly/v2ray-core/v5/common/buf" 
    "github.com/v2fly/v2ray-core/v5/common/cmdarg" 
    "github.com/v2fly/v2ray-core/v5/common/serial" 
) 
 
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen 
 
func loadJSONPB(data io.Reader) (*core.Config, error) { 
    coreconf := &core.Config{} 
    jsonpbloader := &jsonpb.Unmarshaler{AnyResolver: serial.GetResolver()} 
    err := jsonpbloader.Unmarshal(data, coreconf) 
    if err != nil { 
        return nil, err 
    } 
    return coreconf, nil 
} 
 
func dumpJSONPb(config proto.Message, w io.Writer) error { 
    jsonpbdumper := &jsonpb.Marshaler{AnyResolver: serial.GetResolver()} 
    err := jsonpbdumper.Marshal(w, config) 
    if err != nil { 
        return err 
    } 
    return nil 
} 
 
func DumpJSONPb(config proto.Message, w io.Writer) error { 
    return dumpJSONPb(config, w) 
} 
 
const FormatProtobufJSONPB = "jsonpb" 
 
func init() { 
    common.Must(core.RegisterConfigLoader(&core.ConfigFormat{ 
        Name:      []string{FormatProtobufJSONPB}, 
        Extension: []string{".pb.json", ".pbjson"}, 
        Loader: func(input interface{}) (*core.Config, error) { 
            switch v := input.(type) { 
            case string: 
                r, err := cmdarg.LoadArg(v) 
                if err != nil { 
                    return nil, err 
                } 
                data, err := buf.ReadAllToBytes(r) 
                if err != nil { 
                    return nil, err 
                } 
                return loadJSONPB(bytes.NewReader(data)) 
            case []byte: 
                return loadJSONPB(bytes.NewReader(v)) 
            case io.Reader: 
                data, err := buf.ReadAllToBytes(v) 
                if err != nil { 
                    return nil, err 
                } 
                return loadJSONPB(bytes.NewReader(data)) 
            default: 
                return nil, newError("unknown type") 
            } 
        }, 
    })) 
}