mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 01:57:12 -05:00
fix loading of json config
This commit is contained in:
parent
34f51c6a0d
commit
f4ec85d320
@ -1,23 +1,35 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
|
||||||
"v2ray.com/core"
|
"v2ray.com/core"
|
||||||
|
"v2ray.com/core/common"
|
||||||
"v2ray.com/core/common/platform"
|
"v2ray.com/core/common/platform"
|
||||||
|
"v2ray.com/core/common/signal"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type logWriter struct{}
|
||||||
|
|
||||||
|
func (*logWriter) Write(b []byte) (int, error) {
|
||||||
|
n, err := os.Stderr.Write(b)
|
||||||
|
if err == nil {
|
||||||
|
os.Stderr.WriteString(platform.LineSeparator())
|
||||||
|
}
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
|
||||||
func jsonToProto(input io.Reader) (*core.Config, error) {
|
func jsonToProto(input io.Reader) (*core.Config, error) {
|
||||||
v2ctl := platform.GetToolLocation("v2ctl")
|
v2ctl := platform.GetToolLocation("v2ctl")
|
||||||
_, err := os.Stat(v2ctl)
|
if _, err := os.Stat(v2ctl); err != nil {
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
cmd := exec.Command(v2ctl, "config")
|
cmd := exec.Command(v2ctl, "config")
|
||||||
cmd.Stdin = input
|
cmd.Stdin = input
|
||||||
cmd.Stderr = os.Stderr
|
cmd.Stderr = &logWriter{}
|
||||||
cmd.SysProcAttr = getSysProcAttr()
|
cmd.SysProcAttr = getSysProcAttr()
|
||||||
|
|
||||||
stdoutReader, err := cmd.StdoutPipe()
|
stdoutReader, err := cmd.StdoutPipe()
|
||||||
@ -30,19 +42,34 @@ func jsonToProto(input io.Reader) (*core.Config, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
config, err := core.LoadConfig(core.ConfigFormat_Protobuf, stdoutReader)
|
var config *core.Config
|
||||||
|
|
||||||
cmd.Wait()
|
loadTask := signal.ExecuteAsync(func() error {
|
||||||
|
c, err := core.LoadConfig(core.ConfigFormat_Protobuf, stdoutReader)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
config = c
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
return config, err
|
waitTask := signal.ExecuteAsync(func() error {
|
||||||
|
return cmd.Wait()
|
||||||
|
})
|
||||||
|
|
||||||
|
if err := signal.ErrorOrFinish2(context.Background(), loadTask, waitTask); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return config, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
core.RegisterConfigLoader(core.ConfigFormat_JSON, func(input io.Reader) (*core.Config, error) {
|
common.Must(core.RegisterConfigLoader(core.ConfigFormat_JSON, func(input io.Reader) (*core.Config, error) {
|
||||||
config, err := jsonToProto(input)
|
config, err := jsonToProto(input)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, newError("failed to execute v2ctl to convert config file.").Base(err).AtWarning()
|
return nil, newError("failed to execute v2ctl to convert config file.").Base(err).AtWarning()
|
||||||
}
|
}
|
||||||
return config, nil
|
return config, nil
|
||||||
})
|
}))
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user