2015-05-20 22:57:19 +01:00
|
|
|
package transport
|
|
|
|
|
|
|
|
import (
|
2015-06-01 18:55:27 +01:00
|
|
|
"bufio"
|
2015-05-20 22:57:19 +01:00
|
|
|
"bytes"
|
2016-01-18 00:10:04 +00:00
|
|
|
"crypto/tls"
|
2015-05-21 21:08:19 +01:00
|
|
|
"errors"
|
2015-10-22 15:14:56 +01:00
|
|
|
"io"
|
2015-05-20 22:57:19 +01:00
|
|
|
"io/ioutil"
|
|
|
|
"net"
|
|
|
|
"net/http"
|
|
|
|
"net/url"
|
2015-10-22 15:14:56 +01:00
|
|
|
"sync"
|
2016-04-01 13:10:18 +01:00
|
|
|
"time"
|
2016-01-18 00:10:04 +00:00
|
|
|
|
2018-04-08 12:37:45 +01:00
|
|
|
maddr "github.com/micro/util/go/lib/addr"
|
|
|
|
mnet "github.com/micro/util/go/lib/net"
|
|
|
|
mls "github.com/micro/util/go/lib/tls"
|
2018-07-29 10:55:46 +01:00
|
|
|
"golang.org/x/net/http2"
|
2019-01-31 17:14:36 +00:00
|
|
|
"golang.org/x/net/http2/h2c"
|
2015-05-20 22:57:19 +01:00
|
|
|
)
|
|
|
|
|
2015-11-25 00:17:15 +00:00
|
|
|
type buffer struct {
|
|
|
|
io.ReadWriter
|
|
|
|
}
|
|
|
|
|
2016-01-18 00:10:04 +00:00
|
|
|
type httpTransport struct {
|
|
|
|
opts Options
|
|
|
|
}
|
2015-05-20 22:57:19 +01:00
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
type httpTransportClient struct {
|
2015-06-01 18:55:27 +01:00
|
|
|
ht *httpTransport
|
|
|
|
addr string
|
|
|
|
conn net.Conn
|
2015-12-31 18:11:46 +00:00
|
|
|
dialOpts DialOptions
|
2015-10-22 15:14:56 +01:00
|
|
|
once sync.Once
|
2015-11-28 16:34:27 +00:00
|
|
|
|
2019-01-30 18:42:11 +00:00
|
|
|
sync.RWMutex
|
2015-12-18 20:28:50 +00:00
|
|
|
r chan *http.Request
|
|
|
|
bl []*http.Request
|
2015-11-28 16:34:27 +00:00
|
|
|
buff *bufio.Reader
|
2018-11-14 19:41:13 +00:00
|
|
|
|
|
|
|
// local/remote ip
|
|
|
|
local string
|
|
|
|
remote string
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
type httpTransportSocket struct {
|
2018-07-29 10:55:46 +01:00
|
|
|
ht *httpTransport
|
|
|
|
w http.ResponseWriter
|
|
|
|
r *http.Request
|
|
|
|
rw *bufio.ReadWriter
|
2015-12-18 20:28:50 +00:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
conn net.Conn
|
|
|
|
// for the first request
|
|
|
|
ch chan *http.Request
|
2018-11-14 19:41:13 +00:00
|
|
|
|
|
|
|
// local/remote ip
|
|
|
|
local string
|
|
|
|
remote string
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
type httpTransportListener struct {
|
2016-07-28 19:12:56 +01:00
|
|
|
ht *httpTransport
|
2015-05-20 22:57:19 +01:00
|
|
|
listener net.Listener
|
|
|
|
}
|
|
|
|
|
2015-11-25 00:17:15 +00:00
|
|
|
func (b *buffer) Close() error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-11-14 19:41:13 +00:00
|
|
|
func (h *httpTransportClient) Local() string {
|
|
|
|
return h.local
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTransportClient) Remote() string {
|
|
|
|
return h.remote
|
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
func (h *httpTransportClient) Send(m *Message) error {
|
2015-05-20 22:57:19 +01:00
|
|
|
header := make(http.Header)
|
|
|
|
|
|
|
|
for k, v := range m.Header {
|
|
|
|
header.Set(k, v)
|
|
|
|
}
|
|
|
|
|
|
|
|
reqB := bytes.NewBuffer(m.Body)
|
|
|
|
defer reqB.Reset()
|
|
|
|
buf := &buffer{
|
|
|
|
reqB,
|
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
req := &http.Request{
|
2015-05-20 22:57:19 +01:00
|
|
|
Method: "POST",
|
|
|
|
URL: &url.URL{
|
|
|
|
Scheme: "http",
|
2015-05-21 19:24:57 +01:00
|
|
|
Host: h.addr,
|
2015-05-20 22:57:19 +01:00
|
|
|
},
|
|
|
|
Header: header,
|
|
|
|
Body: buf,
|
|
|
|
ContentLength: int64(reqB.Len()),
|
2015-05-21 19:24:57 +01:00
|
|
|
Host: h.addr,
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-12-18 20:28:50 +00:00
|
|
|
h.Lock()
|
|
|
|
h.bl = append(h.bl, req)
|
|
|
|
select {
|
|
|
|
case h.r <- h.bl[0]:
|
|
|
|
h.bl = h.bl[1:]
|
|
|
|
default:
|
|
|
|
}
|
|
|
|
h.Unlock()
|
2015-06-01 18:55:27 +01:00
|
|
|
|
2016-08-01 16:31:27 +01:00
|
|
|
// set timeout if its greater than 0
|
|
|
|
if h.ht.opts.Timeout > time.Duration(0) {
|
|
|
|
h.conn.SetDeadline(time.Now().Add(h.ht.opts.Timeout))
|
2016-07-28 19:12:56 +01:00
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
return req.Write(h.conn)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTransportClient) Recv(m *Message) error {
|
2016-10-29 21:33:04 +01:00
|
|
|
if m == nil {
|
|
|
|
return errors.New("message passed in is nil")
|
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
var r *http.Request
|
2015-12-31 18:14:40 +00:00
|
|
|
if !h.dialOpts.Stream {
|
2015-10-22 15:14:56 +01:00
|
|
|
rc, ok := <-h.r
|
|
|
|
if !ok {
|
|
|
|
return io.EOF
|
|
|
|
}
|
|
|
|
r = rc
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
|
|
|
|
2016-08-01 16:31:27 +01:00
|
|
|
// set timeout if its greater than 0
|
|
|
|
if h.ht.opts.Timeout > time.Duration(0) {
|
|
|
|
h.conn.SetDeadline(time.Now().Add(h.ht.opts.Timeout))
|
2016-07-28 19:12:56 +01:00
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
rsp, err := http.ReadResponse(h.buff, r)
|
2015-05-20 22:57:19 +01:00
|
|
|
if err != nil {
|
2015-06-01 18:55:27 +01:00
|
|
|
return err
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
defer rsp.Body.Close()
|
|
|
|
|
|
|
|
b, err := ioutil.ReadAll(rsp.Body)
|
|
|
|
if err != nil {
|
2015-06-01 18:55:27 +01:00
|
|
|
return err
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2016-05-06 15:44:32 +01:00
|
|
|
if rsp.StatusCode != 200 {
|
|
|
|
return errors.New(rsp.Status + ": " + string(b))
|
|
|
|
}
|
|
|
|
|
2016-10-29 21:33:04 +01:00
|
|
|
m.Body = b
|
|
|
|
|
|
|
|
if m.Header == nil {
|
|
|
|
m.Header = make(map[string]string)
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
for k, v := range rsp.Header {
|
|
|
|
if len(v) > 0 {
|
2016-10-29 21:33:04 +01:00
|
|
|
m.Header[k] = v[0]
|
2015-05-20 22:57:19 +01:00
|
|
|
} else {
|
2016-10-29 21:33:04 +01:00
|
|
|
m.Header[k] = ""
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
return nil
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportClient) Close() error {
|
2015-11-28 16:34:27 +00:00
|
|
|
err := h.conn.Close()
|
2015-10-22 15:14:56 +01:00
|
|
|
h.once.Do(func() {
|
2015-11-28 16:34:27 +00:00
|
|
|
h.Lock()
|
|
|
|
h.buff.Reset(nil)
|
|
|
|
h.Unlock()
|
2015-10-22 15:14:56 +01:00
|
|
|
close(h.r)
|
|
|
|
})
|
2015-11-28 16:34:27 +00:00
|
|
|
return err
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2018-11-14 19:41:13 +00:00
|
|
|
func (h *httpTransportSocket) Local() string {
|
|
|
|
return h.local
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTransportSocket) Remote() string {
|
|
|
|
return h.remote
|
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportSocket) Recv(m *Message) error {
|
2015-05-21 21:08:19 +01:00
|
|
|
if m == nil {
|
|
|
|
return errors.New("message passed in is nil")
|
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
if m.Header == nil {
|
|
|
|
m.Header = make(map[string]string)
|
2016-07-28 19:12:56 +01:00
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// process http 1
|
|
|
|
if h.r.ProtoMajor == 1 {
|
|
|
|
// set timeout if its greater than 0
|
|
|
|
if h.ht.opts.Timeout > time.Duration(0) {
|
|
|
|
h.conn.SetDeadline(time.Now().Add(h.ht.opts.Timeout))
|
|
|
|
}
|
|
|
|
|
|
|
|
var r *http.Request
|
|
|
|
|
|
|
|
select {
|
|
|
|
// get first request
|
|
|
|
case r = <-h.ch:
|
|
|
|
// read next request
|
|
|
|
default:
|
|
|
|
rr, err := http.ReadRequest(h.rw.Reader)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
r = rr
|
|
|
|
}
|
|
|
|
|
|
|
|
// read body
|
|
|
|
b, err := ioutil.ReadAll(r.Body)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// set body
|
|
|
|
r.Body.Close()
|
|
|
|
m.Body = b
|
|
|
|
|
|
|
|
// set headers
|
2018-11-22 10:39:36 +00:00
|
|
|
for k, v := range r.Header {
|
2018-07-29 10:55:46 +01:00
|
|
|
if len(v) > 0 {
|
|
|
|
m.Header[k] = v[0]
|
|
|
|
} else {
|
|
|
|
m.Header[k] = ""
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// return early early
|
|
|
|
return nil
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
2015-12-18 20:28:50 +00:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// processing http2 request
|
|
|
|
// read streaming body
|
|
|
|
|
|
|
|
// set max buffer size
|
|
|
|
buf := make([]byte, 4*1024)
|
|
|
|
|
|
|
|
// read the request body
|
|
|
|
n, err := h.r.Body.Read(buf)
|
|
|
|
// not an eof error
|
2015-12-18 20:28:50 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// check if we have data
|
|
|
|
if n > 0 {
|
|
|
|
m.Body = buf[:n]
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// set headers
|
|
|
|
for k, v := range h.r.Header {
|
2015-05-20 22:57:19 +01:00
|
|
|
if len(v) > 0 {
|
2016-10-29 21:33:04 +01:00
|
|
|
m.Header[k] = v[0]
|
2015-05-20 22:57:19 +01:00
|
|
|
} else {
|
2016-10-29 21:33:04 +01:00
|
|
|
m.Header[k] = ""
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-02 12:55:06 +00:00
|
|
|
// set path
|
|
|
|
m.Header[":path"] = h.r.URL.Path
|
|
|
|
|
2015-05-21 21:08:19 +01:00
|
|
|
return nil
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportSocket) Send(m *Message) error {
|
2018-07-29 10:55:46 +01:00
|
|
|
if h.r.ProtoMajor == 1 {
|
|
|
|
rsp := &http.Response{
|
|
|
|
Header: h.r.Header,
|
|
|
|
Body: ioutil.NopCloser(bytes.NewReader(m.Body)),
|
|
|
|
Status: "200 OK",
|
|
|
|
StatusCode: 200,
|
|
|
|
Proto: "HTTP/1.1",
|
|
|
|
ProtoMajor: 1,
|
|
|
|
ProtoMinor: 1,
|
|
|
|
ContentLength: int64(len(m.Body)),
|
|
|
|
}
|
2015-06-01 18:55:27 +01:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
for k, v := range m.Header {
|
|
|
|
rsp.Header.Set(k, v)
|
|
|
|
}
|
2015-05-20 22:57:19 +01:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// set timeout if its greater than 0
|
|
|
|
if h.ht.opts.Timeout > time.Duration(0) {
|
|
|
|
h.conn.SetDeadline(time.Now().Add(h.ht.opts.Timeout))
|
|
|
|
}
|
|
|
|
|
|
|
|
return rsp.Write(h.conn)
|
2015-12-18 20:28:50 +00:00
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// http2 request
|
|
|
|
|
|
|
|
// set headers
|
|
|
|
for k, v := range m.Header {
|
|
|
|
h.w.Header().Set(k, v)
|
2016-07-28 19:12:56 +01:00
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// write request
|
|
|
|
_, err := h.w.Write(m.Body)
|
|
|
|
return err
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-11-24 00:52:59 +00:00
|
|
|
func (h *httpTransportSocket) error(m *Message) error {
|
2018-07-29 10:55:46 +01:00
|
|
|
if h.r.ProtoMajor == 1 {
|
|
|
|
rsp := &http.Response{
|
|
|
|
Header: make(http.Header),
|
|
|
|
Body: ioutil.NopCloser(bytes.NewReader(m.Body)),
|
|
|
|
Status: "500 Internal Server Error",
|
|
|
|
StatusCode: 500,
|
|
|
|
Proto: "HTTP/1.1",
|
|
|
|
ProtoMajor: 1,
|
|
|
|
ProtoMinor: 1,
|
|
|
|
ContentLength: int64(len(m.Body)),
|
|
|
|
}
|
2015-11-24 00:52:59 +00:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
for k, v := range m.Header {
|
|
|
|
rsp.Header.Set(k, v)
|
|
|
|
}
|
2015-11-24 00:52:59 +00:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
return rsp.Write(h.conn)
|
|
|
|
}
|
|
|
|
return nil
|
2015-11-24 00:52:59 +00:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportSocket) Close() error {
|
2018-07-29 10:55:46 +01:00
|
|
|
if h.r.ProtoMajor == 1 {
|
|
|
|
return h.conn.Close()
|
|
|
|
}
|
|
|
|
return nil
|
2015-05-21 21:08:19 +01:00
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportListener) Addr() string {
|
2015-05-20 22:57:19 +01:00
|
|
|
return h.listener.Addr().String()
|
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportListener) Close() error {
|
2015-05-20 22:57:19 +01:00
|
|
|
return h.listener.Close()
|
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
func (h *httpTransportListener) Accept(fn func(Socket)) error {
|
2018-07-29 10:55:46 +01:00
|
|
|
// create handler mux
|
|
|
|
mux := http.NewServeMux()
|
2018-11-18 16:32:53 +00:00
|
|
|
|
|
|
|
// register our transport handler
|
2018-07-29 10:55:46 +01:00
|
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
var buf *bufio.ReadWriter
|
|
|
|
var con net.Conn
|
|
|
|
|
|
|
|
// read a regular request
|
|
|
|
if r.ProtoMajor == 1 {
|
|
|
|
b, err := ioutil.ReadAll(r.Body)
|
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
r.Body = ioutil.NopCloser(bytes.NewReader(b))
|
|
|
|
// hijack the conn
|
|
|
|
hj, ok := w.(http.Hijacker)
|
|
|
|
if !ok {
|
|
|
|
// we're screwed
|
|
|
|
http.Error(w, "cannot serve conn", http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
2016-04-06 18:03:27 +01:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
conn, bufrw, err := hj.Hijack()
|
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
2016-04-01 13:10:18 +01:00
|
|
|
}
|
2018-07-29 10:55:46 +01:00
|
|
|
defer conn.Close()
|
|
|
|
buf = bufrw
|
|
|
|
con = conn
|
2015-12-18 20:28:50 +00:00
|
|
|
}
|
2015-11-24 00:52:59 +00:00
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// save the request
|
|
|
|
ch := make(chan *http.Request, 1)
|
|
|
|
ch <- r
|
|
|
|
|
|
|
|
fn(&httpTransportSocket{
|
2018-11-14 19:41:13 +00:00
|
|
|
ht: h.ht,
|
|
|
|
w: w,
|
|
|
|
r: r,
|
|
|
|
rw: buf,
|
|
|
|
ch: ch,
|
|
|
|
conn: con,
|
|
|
|
local: h.Addr(),
|
|
|
|
remote: r.RemoteAddr,
|
2018-07-29 10:55:46 +01:00
|
|
|
})
|
|
|
|
})
|
2015-12-18 20:28:50 +00:00
|
|
|
|
2018-11-18 16:32:53 +00:00
|
|
|
// get optional handlers
|
|
|
|
if h.ht.opts.Context != nil {
|
|
|
|
handlers, ok := h.ht.opts.Context.Value("http_handlers").(map[string]http.Handler)
|
|
|
|
if ok {
|
|
|
|
for pattern, handler := range handlers {
|
|
|
|
mux.Handle(pattern, handler)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-29 10:55:46 +01:00
|
|
|
// default http2 server
|
|
|
|
srv := &http.Server{
|
|
|
|
Handler: mux,
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
2018-07-29 10:55:46 +01:00
|
|
|
|
|
|
|
// insecure connection use h2c
|
|
|
|
if !(h.ht.opts.Secure || h.ht.opts.TLSConfig != nil) {
|
2019-01-31 17:14:36 +00:00
|
|
|
srv.Handler = h2c.NewHandler(mux, &http2.Server{})
|
2018-07-29 10:55:46 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// begin serving
|
|
|
|
return srv.Serve(h.listener)
|
2015-05-20 22:57:19 +01:00
|
|
|
}
|
|
|
|
|
2015-06-01 18:55:27 +01:00
|
|
|
func (h *httpTransport) Dial(addr string, opts ...DialOption) (Client, error) {
|
2016-01-03 21:25:03 +00:00
|
|
|
dopts := DialOptions{
|
|
|
|
Timeout: DefaultDialTimeout,
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
for _, opt := range opts {
|
|
|
|
opt(&dopts)
|
|
|
|
}
|
|
|
|
|
2016-01-18 00:10:04 +00:00
|
|
|
var conn net.Conn
|
|
|
|
var err error
|
|
|
|
|
|
|
|
// TODO: support dial option here rather than using internal config
|
|
|
|
if h.opts.Secure || h.opts.TLSConfig != nil {
|
|
|
|
config := h.opts.TLSConfig
|
|
|
|
if config == nil {
|
|
|
|
config = &tls.Config{
|
|
|
|
InsecureSkipVerify: true,
|
|
|
|
}
|
|
|
|
}
|
2019-05-24 17:05:31 +01:00
|
|
|
config.NextProtos = []string{"http/1.1"}
|
2019-01-02 15:24:17 +00:00
|
|
|
conn, err = newConn(func(addr string) (net.Conn, error) {
|
|
|
|
return tls.DialWithDialer(&net.Dialer{Timeout: dopts.Timeout}, "tcp", addr, config)
|
|
|
|
})(addr)
|
2016-01-18 00:10:04 +00:00
|
|
|
} else {
|
2019-01-02 15:24:17 +00:00
|
|
|
conn, err = newConn(func(addr string) (net.Conn, error) {
|
|
|
|
return net.DialTimeout("tcp", addr, dopts.Timeout)
|
|
|
|
})(addr)
|
2016-01-18 00:10:04 +00:00
|
|
|
}
|
|
|
|
|
2016-01-03 21:25:03 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
return &httpTransportClient{
|
2015-06-01 18:55:27 +01:00
|
|
|
ht: h,
|
|
|
|
addr: addr,
|
|
|
|
conn: conn,
|
|
|
|
buff: bufio.NewReader(conn),
|
|
|
|
dialOpts: dopts,
|
|
|
|
r: make(chan *http.Request, 1),
|
2018-11-14 19:41:13 +00:00
|
|
|
local: conn.LocalAddr().String(),
|
|
|
|
remote: conn.RemoteAddr().String(),
|
2015-05-20 22:57:19 +01:00
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
2016-01-18 00:10:04 +00:00
|
|
|
func (h *httpTransport) Listen(addr string, opts ...ListenOption) (Listener, error) {
|
|
|
|
var options ListenOptions
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
|
|
|
|
var l net.Listener
|
|
|
|
var err error
|
|
|
|
|
|
|
|
// TODO: support use of listen options
|
|
|
|
if h.opts.Secure || h.opts.TLSConfig != nil {
|
|
|
|
config := h.opts.TLSConfig
|
2016-03-02 14:20:28 +00:00
|
|
|
|
|
|
|
fn := func(addr string) (net.Listener, error) {
|
|
|
|
if config == nil {
|
2016-06-05 23:13:29 +08:00
|
|
|
hosts := []string{addr}
|
2016-06-06 14:05:02 +01:00
|
|
|
|
|
|
|
// check if its a valid host:port
|
|
|
|
if host, _, err := net.SplitHostPort(addr); err == nil {
|
|
|
|
if len(host) == 0 {
|
2017-01-12 13:52:49 +00:00
|
|
|
hosts = maddr.IPs()
|
2016-06-05 23:13:29 +08:00
|
|
|
} else {
|
2016-06-06 14:05:02 +01:00
|
|
|
hosts = []string{host}
|
2016-06-05 23:13:29 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-06-06 14:05:02 +01:00
|
|
|
// generate a certificate
|
2016-06-05 23:13:29 +08:00
|
|
|
cert, err := mls.Certificate(hosts...)
|
2016-03-02 14:20:28 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
config = &tls.Config{Certificates: []tls.Certificate{cert}}
|
2016-01-18 00:10:04 +00:00
|
|
|
}
|
2016-03-02 14:20:28 +00:00
|
|
|
return tls.Listen("tcp", addr, config)
|
2016-01-18 00:10:04 +00:00
|
|
|
}
|
2016-03-02 14:20:28 +00:00
|
|
|
|
2017-01-12 14:11:25 +00:00
|
|
|
l, err = mnet.Listen(addr, fn)
|
2016-01-18 00:10:04 +00:00
|
|
|
} else {
|
2016-03-02 14:20:28 +00:00
|
|
|
fn := func(addr string) (net.Listener, error) {
|
|
|
|
return net.Listen("tcp", addr)
|
|
|
|
}
|
|
|
|
|
2017-01-12 14:11:25 +00:00
|
|
|
l, err = mnet.Listen(addr, fn)
|
2016-01-18 00:10:04 +00:00
|
|
|
}
|
|
|
|
|
2015-05-20 22:57:19 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2015-05-23 20:04:16 +01:00
|
|
|
return &httpTransportListener{
|
2016-07-28 19:12:56 +01:00
|
|
|
ht: h,
|
2015-05-20 22:57:19 +01:00
|
|
|
listener: l,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
func (h *httpTransport) Init(opts ...Option) error {
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&h.opts)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *httpTransport) Options() Options {
|
|
|
|
return h.opts
|
|
|
|
}
|
|
|
|
|
2015-12-19 21:56:14 +00:00
|
|
|
func (h *httpTransport) String() string {
|
|
|
|
return "http"
|
|
|
|
}
|
|
|
|
|
2016-06-06 14:05:02 +01:00
|
|
|
func newHTTPTransport(opts ...Option) *httpTransport {
|
|
|
|
var options Options
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
return &httpTransport{opts: options}
|
|
|
|
}
|