package geetest import ( "context" "crypto/tls" "errors" "io/ioutil" "net" "net/http" "net/url" "strconv" "strings" "time" "go-common/app/interface/main/answer/conf" "go-common/app/interface/main/answer/model" "go-common/library/log" bm "go-common/library/net/http/blademaster" ) const ( _register = "/register.php" _validate = "/validate.php" ) // Dao is account dao. type Dao struct { c *conf.Config // url registerURI string validateURI string // http client client *http.Client clientx *bm.Client } // New new a dao. func New(c *conf.Config) (d *Dao) { d = &Dao{ c: c, registerURI: c.Host.Geetest + _register, validateURI: c.Host.Geetest + _validate, // http client client: NewClient(c.HTTPClient), clientx: bm.NewClient(c.HTTPClient.Slow), } return } // PreProcess preprocessing the geetest and get to challenge func (d *Dao) PreProcess(c context.Context, mid int64, ip, geeType string, gc conf.GeetestConfig, newCaptcha int) (challenge string, err error) { var ( req *http.Request res *http.Response bs []byte params url.Values ) params = url.Values{} params.Set("user_id", strconv.FormatInt(mid, 10)) params.Set("new_captcha", strconv.Itoa(newCaptcha)) params.Set("ip_address", ip) params.Set("client_type", geeType) params.Set("gt", gc.CaptchaID) if req, err = http.NewRequest("GET", d.registerURI+"?"+params.Encode(), nil); err != nil { log.Error("d.preprocess uri(%s) params(%s) error(%v)", d.validateURI, params.Encode(), err) return } req.Header.Set("Content-Type", "application/x-www-form-urlencoded") if res, err = d.client.Do(req); err != nil { log.Error("client.Do(%s) error(%v)", d.validateURI+"?"+params.Encode(), err) return } defer res.Body.Close() if res.StatusCode >= http.StatusInternalServerError { err = errors.New("http status code 5xx") log.Error("gtServerErr uri(%s) error(%v)", d.validateURI+"?"+params.Encode(), err) return } if bs, err = ioutil.ReadAll(res.Body); err != nil { log.Error("ioutil.ReadAll(%s) uri(%s) error(%v)", bs, d.validateURI+"?"+params.Encode(), err) return } if len(bs) != 32 { log.Error("d.preprocess len(%s) the length not equate 32byte", string(bs)) return } challenge = string(bs) return } // Validate recheck the challenge code and get to seccode func (d *Dao) Validate(c context.Context, challenge, seccode, clientType, ip, captchaID string, mid int64) (res *model.ValidateRes, err error) { params := url.Values{} params.Set("seccode", seccode) params.Set("challenge", challenge) params.Set("captchaid", captchaID) params.Set("client_type", clientType) params.Set("ip_address", ip) params.Set("json_format", "1") params.Set("sdk", "golang_3.0.0") params.Set("user_id", strconv.FormatInt(mid, 10)) params.Set("timestamp", strconv.FormatInt(time.Now().Unix(), 10)) req, err := http.NewRequest("POST", d.validateURI, strings.NewReader(params.Encode())) if err != nil { log.Error("http.NewRequest error(%v) | uri(%s) params(%s)", err, d.validateURI, params.Encode()) return } log.Info("Validate(%v) start", params) req.Header.Set("Content-Type", "application/x-www-form-urlencoded") if err = d.clientx.Do(c, req, &res); err != nil { log.Error("d.client.Do error(%v) | uri(%s) params(%s) res(%v)", err, d.validateURI, params.Encode(), res) return } log.Info("Validate(%v) suc", res) return } // NewClient new a http client. func NewClient(c *conf.HTTPClient) (client *http.Client) { var ( transport *http.Transport dialer *net.Dialer ) dialer = &net.Dialer{ Timeout: time.Duration(c.Slow.Dial), KeepAlive: time.Duration(c.Slow.KeepAlive), } transport = &http.Transport{ DialContext: dialer.DialContext, TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, } client = &http.Client{ Transport: transport, } return } // GeeConfig get geetest config. func (d *Dao) GeeConfig(t string, c *conf.Geetest) (gc conf.GeetestConfig, geetype string) { switch t { case "pc": gc = c.PC case "h5": gc = c.H5 default: gc = c.PC } geetype = "web" return }