mirror of https://gitee.com/IrisVega/frp.git
42 lines
1.0 KiB
Go
42 lines
1.0 KiB
Go
package plugin
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
plugin "github.com/fatedier/frp/pkg/plugin/server"
|
|
"github.com/fatedier/frp/pkg/util/log"
|
|
"github.com/fatedier/frp/test/e2e/mock/server/httpserver"
|
|
)
|
|
|
|
type PluginHandler func(req *plugin.Request) *plugin.Response
|
|
|
|
type NewPluginRequest func() *plugin.Request
|
|
|
|
func NewHTTPPluginServer(port int, newFunc NewPluginRequest, handler PluginHandler, tlsConfig *tls.Config) *httpserver.Server {
|
|
return httpserver.New(
|
|
httpserver.WithBindPort(port),
|
|
httpserver.WithTlsConfig(tlsConfig),
|
|
httpserver.WithHandler(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
r := newFunc()
|
|
buf, err := ioutil.ReadAll(req.Body)
|
|
if err != nil {
|
|
w.WriteHeader(500)
|
|
return
|
|
}
|
|
log.Trace("plugin request: %s", string(buf))
|
|
err = json.Unmarshal(buf, &r)
|
|
if err != nil {
|
|
w.WriteHeader(500)
|
|
return
|
|
}
|
|
resp := handler(r)
|
|
buf, _ = json.Marshal(resp)
|
|
log.Trace("plugin response: %s", string(buf))
|
|
w.Write(buf)
|
|
})),
|
|
)
|
|
}
|