78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package rpc
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.loafle.net/commons_go/logging"
|
|
|
|
cc "git.loafle.net/commons_go/config"
|
|
ooccn "git.loafle.net/overflow/overflow_commons_go/config/noauthprobe"
|
|
oocmn "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe"
|
|
noauthprobeM "git.loafle.net/overflow/overflow_commons_go/modules/noauthprobe/model"
|
|
probeM "git.loafle.net/overflow/overflow_commons_go/modules/probe/model"
|
|
oopai "git.loafle.net/overflow/overflow_probes/auth/info"
|
|
"git.loafle.net/overflow/overflow_probes/config"
|
|
)
|
|
|
|
type NoAuthProbeService struct {
|
|
DoneChan chan error
|
|
|
|
ConfigPath string
|
|
Config ooccn.NoAuthProbeConfig
|
|
}
|
|
|
|
func (s *NoAuthProbeService) Accept(probe *probeM.Probe) {
|
|
config.Config.Probe.Key = &probe.ProbeKey
|
|
|
|
if err := cc.Save(*config.Config, *config.ConfigFilePath, true); nil != err {
|
|
s.DoneChan <- fmt.Errorf("Auth: Saving of config file[%s] failed error[%v]", *config.ConfigFilePath, err)
|
|
return
|
|
}
|
|
|
|
s.DoneChan <- nil
|
|
}
|
|
|
|
func (s *NoAuthProbeService) Deny(noAuthProbe *noauthprobeM.NoAuthProbe) {
|
|
n := time.Now()
|
|
s.Config.DenyDate = &n
|
|
if err := cc.Save(s.Config, s.ConfigPath, true); nil != err {
|
|
s.DoneChan <- fmt.Errorf("Auth: Saving of NoAuth config file[%s] failed error[%v]", s.ConfigPath, err)
|
|
return
|
|
}
|
|
|
|
s.DoneChan <- nil
|
|
}
|
|
|
|
func (s *NoAuthProbeService) GetRequestHeader() map[string][]string {
|
|
header := make(map[string][]string)
|
|
switch s.Config.State() {
|
|
case ooccn.NoAuthProbeStateTypeNotRegisterd:
|
|
header[oocmn.HTTPRequestHeaderKey_NoAuthProbe_Method] = []string{oocmn.HTTPRequestHeaderValue_NoAuthProbe_Method_Regist}
|
|
header[oocmn.HTTPRequestHeaderKey_NoAuthProbe_Info] = []string{s.GetRegisterRequestHeader()}
|
|
|
|
case ooccn.NoAuthProbeStateTypeRegisterd:
|
|
header[oocmn.HTTPRequestHeaderKey_NoAuthProbe_Method] = []string{oocmn.HTTPRequestHeaderValue_NoAuthProbe_Method_Connect}
|
|
header[oocmn.HTTPRequestHeaderKey_NoAuthProbe_TempProbeKey] = []string{oocmn.HTTPRequestHeaderValue_NoAuthProbe_Method_Connect}
|
|
}
|
|
|
|
return header
|
|
}
|
|
|
|
func (s *NoAuthProbeService) GetRegisterRequestHeader() string {
|
|
i, err := oopai.GetRegistHeader()
|
|
if nil != err {
|
|
logging.Logger().Error(fmt.Sprintf("Auth: Gathering system information has been failed %v", err))
|
|
}
|
|
return i
|
|
}
|
|
|
|
func (s *NoAuthProbeService) SetTempProbeKey(tempProbeKey string) {
|
|
s.Config.TempKey = &tempProbeKey
|
|
if err := cc.Save(s.Config, s.ConfigPath, true); nil != err {
|
|
s.DoneChan <- fmt.Errorf("Auth: Saving of NoAuth config file[%s] failed error[%v]", s.ConfigPath, err)
|
|
return
|
|
}
|
|
|
|
}
|