idk/go-i2p
1
0
Fork 0
mirror of https://github.com/go-i2p/go-i2p.git synced 2025-06-09 02:43:30 -04:00
Code Issues Packages Projects Releases Wiki Activity

expanded logging in reseed.go

Browse Source
This commit is contained in:
Haris Khan
2024-10-18 22:54:08 -04:00
parent be35267079
commit 8e97eb5f77
1 changed files with 35 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
lib/netdb/reseed/reseed.go
View File

@ -2,8 +2,9 @@ package reseed
import (
"fmt"
"github.com/go-i2p/go-i2p/lib/util/logger"
"github.com/sirupsen/logrus"
"io"
"log"
"net"
"net/http"
"net/url"
@ -16,6 +17,8 @@ import (
"github.com/go-i2p/go-i2p/lib/su3"
)
var log = logger.GetLogger()
const (
I2pUserAgent = "Wget/1.11.4"
)
@ -25,6 +28,8 @@ type Reseed struct {
}
func (r Reseed) SingleReseed(uri string) ([]router_info.RouterInfo, error) {
log.WithField("uri", uri).Debug("Starting single reseed operation")
transport := http.Transport{
DialContext: r.DialContext,
}
@ -33,6 +38,7 @@ func (r Reseed) SingleReseed(uri string) ([]router_info.RouterInfo, error) {
}
URL, err := url.Parse(uri)
if err != nil {
log.WithError(err).Error("Failed to parse reseed URI")
return nil, err
}
header := http.Header{}
@ -43,12 +49,23 @@ func (r Reseed) SingleReseed(uri string) ([]router_info.RouterInfo, error) {
}
response, err := client.Do(&request)
if err != nil {
log.WithError(err).Error("Failed to perform HTTP request")
return nil, err
}
log.Debug("Successfully received response from reseed server")
su3file, err := su3.Read(response.Body)
if err != nil {
log.WithError(err).Error("Failed to read SU3 file")
return nil, err
}
log.WithFields(logrus.Fields{
"file_type": su3file.FileType,
"content_type": su3file.ContentType,
}).Debug("Successfully read SU3 file")
if su3file.FileType == su3.ZIP {
if su3file.ContentType == su3.RESEED {
if err == nil {
@ -59,36 +76,53 @@ func (r Reseed) SingleReseed(uri string) ([]router_info.RouterInfo, error) {
return nil, err
}
log.Println("warning: this doesn't validate the signature yet", signature)
log.Warn("Doesn't validate the signature yet", logrus.Fields{"signature": signature})
}
zip := filepath.Join(config.RouterConfigProperties.NetDb.Path, "reseed.zip")
err = os.WriteFile(zip, content, 0o644)
if err != nil {
log.WithError(err).Error("Failed to write reseed zip file")
return nil, err
}
// content is a zip file, unzip it and get the files
files, err := unzip.New().Extract(zip, config.RouterConfigProperties.NetDb.Path)
if err != nil {
log.WithError(err).Error("Failed to extract reseed zip file")
return nil, err
}
if len(files) <= 0 {
log.Error("Reseed appears to have no content")
return nil, fmt.Errorf("error: reseed appears to have no content")
}
log.WithField("file_count", len(files)).Debug("Successfully extracted reseed files")
var ris []router_info.RouterInfo
for _, f := range files {
riB, err := os.ReadFile(f)
if err != nil {
log.WithError(err).WithField("file", f).Warn("Failed to read router info file")
continue
}
ri, _, err := router_info.ReadRouterInfo(riB)
if err != nil {
log.WithError(err).WithField("file", f).Warn("Failed to parse router info")
continue
}
ris = append(ris, ri)
}
err = os.Remove(zip)
if err != nil {
log.WithError(err).Warn("Failed to remove reseed zip file")
}
log.WithField("router_info_count", len(ris)).Debug("Successfully processed reseed data")
return ris, err
} else {
log.WithError(err).Error("Failed to read SU3 file signature")
return nil, err
}
}
}
log.Error("Undefined reseed error")
return nil, fmt.Errorf("error: undefined reseed error")
}