Merge pull request #1 from wyaode/patch-2

Update main.go
This commit is contained in:
wyaode 2020-04-13 09:10:42 +08:00 committed by GitHub
commit ff6a84038c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -22,11 +22,40 @@ import (
"github.com/fatedier/frp/cmd/frpc/sub" "github.com/fatedier/frp/cmd/frpc/sub"
"github.com/fatedier/golib/crypto" "github.com/fatedier/golib/crypto"
"github.com/kardianos/service"
"os"
"path/filepath"
) )
func main() { func main() {
crypto.DefaultSalt = "frp" crypto.DefaultSalt = "frp"
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
sub.Execute() svcConfig := &service.Config{
Name: "FRPC",
}
prg := &FRPC{}
s, err := service.New(prg, svcConfig)
if err != nil {
sub.Execute()
return
}
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
err = os.Chdir(dir)
_ = s.Run()
}
type FRPC struct {}
func (p *FRPC) Start(s service.Service) error {
_, _ = s.Status()
go sub.Execute()
return nil
}
func (p *FRPC) Stop(s service.Service) error {
_, _ = s.Status()
if service.Interactive() {
os.Exit(0)
}
return nil
} }