mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package proxy
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/synctv-org/synctv/internal/settings"
|
|
"github.com/synctv-org/synctv/utils"
|
|
"github.com/zijiren233/go-uhc"
|
|
)
|
|
|
|
func ProxyURL(ctx *gin.Context, u string, headers map[string]string) error {
|
|
if !settings.AllowProxyToLocal.Get() {
|
|
if l, err := utils.ParseURLIsLocalIP(u); err != nil {
|
|
return fmt.Errorf("check url is local ip error: %w", err)
|
|
} else if l {
|
|
return errors.New("not allow proxy to local")
|
|
}
|
|
}
|
|
ctx2, cf := context.WithCancel(ctx)
|
|
defer cf()
|
|
req, err := http.NewRequestWithContext(ctx2, http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return fmt.Errorf("new request error: %w", err)
|
|
}
|
|
for k, v := range headers {
|
|
req.Header.Set(k, v)
|
|
}
|
|
req.Header.Set("Range", ctx.GetHeader("Range"))
|
|
req.Header.Set("Accept-Encoding", ctx.GetHeader("Accept-Encoding"))
|
|
if req.Header.Get("User-Agent") == "" {
|
|
req.Header.Set("User-Agent", utils.UA)
|
|
}
|
|
cli := http.Client{
|
|
Transport: uhc.DefaultTransport,
|
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
|
req.Header.Del("Referer")
|
|
for k, v := range headers {
|
|
req.Header.Set(k, v)
|
|
}
|
|
req.Header.Set("Range", ctx.GetHeader("Range"))
|
|
req.Header.Set("Accept-Encoding", ctx.GetHeader("Accept-Encoding"))
|
|
if req.Header.Get("User-Agent") == "" {
|
|
req.Header.Set("User-Agent", utils.UA)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
resp, err := cli.Do(req)
|
|
if err != nil {
|
|
return fmt.Errorf("request url error: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
ctx.Status(resp.StatusCode)
|
|
ctx.Header("Accept-Ranges", resp.Header.Get("Accept-Ranges"))
|
|
ctx.Header("Cache-Control", resp.Header.Get("Cache-Control"))
|
|
ctx.Header("Content-Length", resp.Header.Get("Content-Length"))
|
|
ctx.Header("Content-Range", resp.Header.Get("Content-Range"))
|
|
ctx.Header("Content-Type", resp.Header.Get("Content-Type"))
|
|
_, err = copyBuffer(ctx.Writer, resp.Body)
|
|
if err != nil && err != io.EOF {
|
|
return fmt.Errorf("copy response body error: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func copyBuffer(dst io.Writer, src io.Reader) (written int64, err error) {
|
|
buf := getBuffer()
|
|
defer putBuffer(buf)
|
|
for {
|
|
nr, er := src.Read(*buf)
|
|
if nr > 0 {
|
|
nw, ew := dst.Write((*buf)[0:nr])
|
|
if nw < 0 || nr < nw {
|
|
nw = 0
|
|
if ew == nil {
|
|
ew = errors.New("invalid write result")
|
|
}
|
|
}
|
|
written += int64(nw)
|
|
if ew != nil {
|
|
err = ew
|
|
break
|
|
}
|
|
if nr != nw {
|
|
err = io.ErrShortWrite
|
|
break
|
|
}
|
|
}
|
|
if er != nil {
|
|
if er != io.EOF {
|
|
err = er
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return written, err
|
|
}
|