This commit is contained in:
crusader
2018-06-14 16:17:03 +09:00
parent ea8d6ed43e
commit bdfaf81df9
6 changed files with 30 additions and 16 deletions

View File

@@ -1,10 +1,13 @@
package servlet
import (
"bytes"
"compress/gzip"
"context"
"encoding/base64"
"encoding/json"
"fmt"
"io/ioutil"
"github.com/valyala/fasthttp"
@@ -80,11 +83,23 @@ func (s *AuthServlets) Handshake(servletCtx server.ServletCtx, ctx *fasthttp.Req
if nil == bInfo {
return nil, fmt.Errorf("Unexpected Request Header Key NoAuthProbe Info")
}
rJSON := make([]byte, base64.StdEncoding.DecodedLen(len(bInfo)))
_, err := base64.StdEncoding.Decode(rJSON, bInfo)
buf := make([]byte, base64.StdEncoding.DecodedLen(len(bInfo)))
_, err := base64.StdEncoding.Decode(buf, bInfo)
if nil != err {
return nil, fmt.Errorf("Base64 Encoding Error: %s", err.Error())
}
bReader := bytes.NewReader(buf)
gReader, err := gzip.NewReader(bReader)
if nil != err {
return nil, fmt.Errorf("gzip Error: %s", err.Error())
}
rJSON, err := ioutil.ReadAll(gReader)
if nil != err {
return nil, fmt.Errorf("gzip Error: %s", err.Error())
}
grpcCTX := context.Background()
r, err := oeg.Exec(grpcCTX, "NoAuthProbeService.regist", string(rJSON))
if nil != err {