Fijxu/http3-ytproxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Add option to disable the host restriction
Some checks are pending
CI / build (push) Waiting to run
Details

Browse source
This commit is contained in:
Fijxu 2024-12-21 14:54:46 -03:00
parent b156a52420
commit 71abe2ae58
Signed by: Fijxu
GPG key ID: 32C1DDF333EDA6A4
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
main.go
View file

@ -100,6 +100,8 @@ var version string
var h3s bool var h3s bool
var domain_only_access bool = false
var programInit = time.Now() var programInit = time.Now()
type ConnectionWatcher struct { type ConnectionWatcher struct {
@ -309,7 +311,7 @@ func beforeMisc(next http.HandlerFunc) http.HandlerFunc {
defer panicHandler(w) defer panicHandler(w)
// To prevent accessing from the bare IP address // To prevent accessing from the bare IP address
if req.Host == "" || net.ParseIP(strings.Split(req.Host, ":")[0]) != nil { if domain_only_access && (req.Host == "" || net.ParseIP(strings.Split(req.Host, ":")[0]) != nil) {
w.WriteHeader(444) w.WriteHeader(444)
return return
} }
@ -323,7 +325,7 @@ func beforeProxy(next http.HandlerFunc) http.HandlerFunc {
defer panicHandler(w) defer panicHandler(w)
// To prevent accessing from the bare IP address // To prevent accessing from the bare IP address
if req.Host == "" || net.ParseIP(strings.Split(req.Host, ":")[0]) != nil { if domain_only_access && (req.Host == "" || net.ParseIP(strings.Split(req.Host, ":")[0]) != nil) {
w.WriteHeader(444) w.WriteHeader(444)
return return
} }
@ -363,9 +365,8 @@ func main() {
defaultTLSCert := "/data/cert.pem" defaultTLSCert := "/data/cert.pem"
defaultTLSKey := "/data/key.key" defaultTLSKey := "/data/key.key"
var https bool = true var https bool = false
var h3c bool = false var h3c bool = false
var h3s bool = true
var ipv6 bool = false var ipv6 bool = false
if strings.ToLower(os.Getenv("HTTPS")) == "true" { if strings.ToLower(os.Getenv("HTTPS")) == "true" {
@ -380,6 +381,9 @@ func main() {
if strings.ToLower(os.Getenv("IPV6_ONLY")) == "true" { if strings.ToLower(os.Getenv("IPV6_ONLY")) == "true" {
ipv6 = true ipv6 = true
} }
if strings.ToLower(os.Getenv("DOMAIN_ONLY_ACCESS")) == "true" {
domain_only_access = true
}
tls_cert := os.Getenv("TLS_CERT") tls_cert := os.Getenv("TLS_CERT")
if tls_cert == "" { if tls_cert == "" {