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

feat: func to rotate the IP address from gluetun automatically depending of the traffic
All checks were successful
CI / build (push) Successful in 48s
Details

Browse source
This commit is contained in:
Fijxu 2025-02-13 02:00:57 -03:00
parent 53d04ac79b
commit 2f88b8487b
Signed by: Fijxu
GPG key ID: 32C1DDF333EDA6A4
1 changed files with 51 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
main.go
View file

@ -5,6 +5,7 @@ import (
"encoding/json"
"errors"
"flag"
"fmt"
"io"
"log"
"net"
@ -21,6 +22,7 @@ import (
"github.com/conduitio/bwlimit"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/prometheus/procfs"
"github.com/quic-go/quic-go"
"github.com/quic-go/quic-go/http3"
)
@ -325,23 +327,45 @@ func requestPerMinute() {
}
}
func forbiddenRequestsPerSec() {
var last int64
// Detects if a backend has been blocked based on the amount of bandwidth
// reported by procfs.
// This may be the best way to detect if the IP has been blocked from googlevideo
// servers. I would like to detect blockages using the status code that googlevideo
// returns, which most of the time is 403 (Forbidden). But this error code is not
// exclusive to IP blocks, it's also returned for other reasons like a wrong
// query parameter like `pot` (po_token) or anything like that.
func blockChecker(gh string) {
// Sleep for 60 seconds before commencing the loop
time.Sleep(60 * time.Second)
url := "http://" + gh + "/v1/openvpn/status"
p, err := procfs.Self()
if err != nil {
log.Printf("[ERROR] [procfs]: Could not get process: %s\n", err)
log.Println("[INFO] Blockchecker will not run, so if the VPN IP used on gluetun gets blocked, it will not be rotated!")
return
}
var last uint64
for {
time.Sleep(1 * time.Second)
current := stats_.RequestsForbidden.Videoplayback
stats_.RequestsForbiddenPerSec.Videoplayback = current - last
// fmt.Println(aux)
// if aux > 15 {
// body := "{\"status\":\"stopped\"}\""
// request, err := http.NewRequest("PUT", "http://gluetun:8000", strings.NewReader(body))
// if err != nil {
// log.Panic(err)
// }
// client.Do(request)
// }
// metrics.RequestPerSecond.Set(float64(stats_.RequestPerSecond))
// time.Sleep(60 * time.Second)
// p.NetDev should never fail.
stat, _ := p.NetDev()
current := stat.Total().TxBytes
aux := current - last
if float64(aux)*0.000008 < 3.0 {
fmt.Printf("xd\n")
body := "{\"status\":\"stopped\"}\""
// This should never fail too
request, _ := http.NewRequest("PUT", url, strings.NewReader(body))
_, err = client.Do(request)
if err != nil {
log.Printf("[ERROR] Failed to send request to gluetun.")
} else {
log.Printf("[INFO] Request to change IP sent to gluetun successfully")
}
time.Sleep(60 * time.Second)
}
last = current
}
}
@ -408,6 +432,7 @@ func main() {
var https bool = false
var h3c bool = false
var ipv6 bool = false
var bc bool = true
if strings.ToLower(os.Getenv("HTTPS")) == "true" {
https = true
@ -421,6 +446,9 @@ func main() {
if strings.ToLower(os.Getenv("IPV6_ONLY")) == "true" {
ipv6 = true
}
if strings.ToLower(os.Getenv("BLOCK_CHECKER")) == "false" {
bc = false
}
if strings.ToLower(os.Getenv("DOMAIN_ONLY_ACCESS")) == "true" {
domain_only_access = true
}
@ -445,6 +473,11 @@ func main() {
if host == "" {
host = defaultHost
}
// gh is where the gluetun api is located
gh := os.Getenv("GLUETUN_HOSTNAME")
if gh == "" {
gh = "127.0.0.1:8000"
}
proxy = os.Getenv("PROXY")
flag.BoolVar(&https, "https", https, "Use built-in https server (recommended)")
@ -508,7 +541,9 @@ func main() {
go requestPerSecond()
go requestPerMinute()
go forbiddenRequestsPerSec()
if bc {
go blockChecker(gh)
}
ln, err := net.Listen("tcp", host+":"+port)
if err != nil {