Creepso/ratio-spoof
1
0
Fork 0
mirror of https://github.com/ap-pauloafonso/ratio-spoof.git synced 2026-01-11 20:10:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

rand and empty result tracker gracefuly exit

Browse source
This commit is contained in:
ap-pauloafonso 2020-12-01 20:58:06 -03:00
parent 02ad394ac4
commit b3c03d6dac
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
assets/demo.gif
View file

Binary file not shown.
Side by side Swipe Overlay

Before

Width:  |  Height:  |  Size: 1.9 MiB

After

Width:  |  Height:  |  Size: 3.2 MiB

17
ratiospoof/ratio-spoof.go
View file

@ -8,6 +8,7 @@ import (
"fmt"
"io/ioutil"
"math"
"math/rand"
"net/http"
"net/url"
"os"
@ -235,6 +236,7 @@ func (R *ratioSPoofState) gracefullyExit() {
}
func (R *ratioSPoofState) Run() {
rand.Seed(time.Now().UnixNano())
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
R.firstAnnounce()
@ -286,8 +288,10 @@ func (R *ratioSPoofState) fireAnnounce() {
query := replacer.Replace(R.bitTorrentClient.Query())
trackerResp := R.tryMakeRequest(query)
R.updateSeedersAndLeechers(trackerResp)
R.updateInterval(trackerResp)
if trackerResp != nil {
R.updateSeedersAndLeechers(*trackerResp)
R.updateInterval(*trackerResp)
}
}
func (R *ratioSPoofState) generateNextAnnounce() {
R.resetTimer(R.announceInterval)
@ -416,7 +420,7 @@ func (R *ratioSPoofState) resetTimer(newAnnounceRate int) {
R.currentAnnounceTimer = R.announceRate
}
func (R *ratioSPoofState) tryMakeRequest(query string) trackerResponse {
func (R *ratioSPoofState) tryMakeRequest(query string) *trackerResponse {
for idx, url := range R.torrentInfo.trackerInfo.urls {
completeURL := url + "?" + strings.TrimLeft(query, "?")
R.lastAnounceRequest = completeURL
@ -428,6 +432,9 @@ func (R *ratioSPoofState) tryMakeRequest(query string) trackerResponse {
if err == nil {
if resp.StatusCode == http.StatusOK {
bytesR, _ := ioutil.ReadAll(resp.Body)
if len(bytesR) == 0 {
return nil
}
mimeType := http.DetectContentType(bytesR)
if mimeType == "application/x-gzip" {
gzipReader, _ := gzip.NewReader(bytes.NewReader(bytesR))
@ -440,7 +447,7 @@ func (R *ratioSPoofState) tryMakeRequest(query string) trackerResponse {
R.torrentInfo.trackerInfo.SwapFirst(idx)
}
ret := extractTrackerResponse(decodedResp)
return ret
return &ret
}
resp.Body.Close()
}
@ -455,7 +462,7 @@ func calculateNextTotalSizeByte(speedKbps, currentByte, pieceSizeByte, seconds,
}
total := currentByte + (speedKbps * 1024 * seconds)
closestPieceNumber := int(total / pieceSizeByte)
closestPieceNumber += 5
closestPieceNumber += rand.Intn(10-1) + 1
nextTotal := closestPieceNumber * pieceSizeByte
if limitTotalBytes != 0 && nextTotal > limitTotalBytes {