118 lines
2.1 KiB
Go
118 lines
2.1 KiB
Go
package p2p
|
|
|
|
// SpeedTest Runs a speed test and does updates IP tables accordingly
|
|
func (ip *IpAddresses)SpeedTest() error{
|
|
// Remove Duplicate IP addresses
|
|
var DoNotRead IpAddresses
|
|
|
|
for i, _ := range ip.IpAddress {
|
|
|
|
Exists := false
|
|
for k := range DoNotRead.IpAddress {
|
|
if DoNotRead.IpAddress[k].Ipv4 == ip.IpAddress[i].Ipv4 {
|
|
Exists = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if Exists {
|
|
continue
|
|
}
|
|
|
|
var err error
|
|
if len(ip.IpAddress) == 1 {
|
|
i = 0
|
|
}
|
|
|
|
// Ping Test
|
|
err = ip.IpAddress[i].PingTest()
|
|
|
|
if err != nil {
|
|
ip.IpAddress = remove(ip.IpAddress,i)
|
|
// Proceed to next element in the array
|
|
continue
|
|
}
|
|
|
|
|
|
//Upload Speed Test
|
|
err = ip.IpAddress[i].UploadSpeed()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = ip.IpAddress[i].DownloadSpeed()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
DoNotRead.IpAddress = append(DoNotRead.IpAddress, ip.IpAddress[i])
|
|
}
|
|
|
|
err := ip.WriteIpTable()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SpeedTestUpdatedIPTable Called when ip tables from client/server is also passed on
|
|
func (ip *IpAddresses)SpeedTestUpdatedIPTable() error{
|
|
targets, err := ReadIpTable()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// To ensure struct has no duplicates IP addresses
|
|
//DoNotRead := targets
|
|
|
|
// Appends all IP addresses
|
|
for i, _ := range targets.IpAddress {
|
|
|
|
// To ensure that there are no duplicate IP addresses
|
|
Exists := false
|
|
for k := range ip.IpAddress {
|
|
if ip.IpAddress[k].Ipv4 == targets.IpAddress[i].Ipv4 {
|
|
Exists = true
|
|
break
|
|
}
|
|
}
|
|
|
|
// If the struct exists then continues
|
|
if Exists {
|
|
continue
|
|
}
|
|
|
|
ip.IpAddress = append(ip.IpAddress, targets.IpAddress[i])
|
|
}
|
|
|
|
err = ip.SpeedTest()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// LocalSpeedTestIpTable Runs speed test in iptables locally only
|
|
func LocalSpeedTestIpTable() error {
|
|
targets, err := ReadIpTable()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = targets.SpeedTest()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Helper function to remove element from an array of a struct
|
|
func remove(s []IpAddress, i int) []IpAddress {
|
|
s[len(s)-1], s[i] = s[i], s[len(s)-1]
|
|
return s[:len(s)-1]
|
|
}
|