Fixes in mac build script, some gofmt works.
This commit is contained in:
@@ -10,160 +10,160 @@
|
||||
package requester
|
||||
|
||||
import (
|
||||
// stdlib
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
// stdlib
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Requester struct {
|
||||
// Pooler.
|
||||
Pooler *Pooler
|
||||
// Packet prefix.
|
||||
pp string
|
||||
// Showstopper for delimiting IP addresses.
|
||||
// As we are receiving bytes, we will put bytes representation of "\"
|
||||
// character.
|
||||
ip_delimiter int
|
||||
// Pooler.
|
||||
Pooler *Pooler
|
||||
// Packet prefix.
|
||||
pp string
|
||||
// Showstopper for delimiting IP addresses.
|
||||
// As we are receiving bytes, we will put bytes representation of "\"
|
||||
// character.
|
||||
ip_delimiter int
|
||||
}
|
||||
|
||||
// Requester's initialization.
|
||||
func (r *Requester) Initialize() {
|
||||
fmt.Println("Initializing Requester...")
|
||||
r.pp = "\377\377\377\377"
|
||||
r.ip_delimiter = 92
|
||||
r.Pooler = &Pooler{}
|
||||
r.Pooler.Initialize()
|
||||
fmt.Println("Initializing Requester...")
|
||||
r.pp = "\377\377\377\377"
|
||||
r.ip_delimiter = 92
|
||||
r.Pooler = &Pooler{}
|
||||
r.Pooler.Initialize()
|
||||
}
|
||||
|
||||
// Gets all available servers from master server.
|
||||
// This isn't in pooler, because it have no need to be pooled.
|
||||
func (r *Requester) getServers() error {
|
||||
// Get master server address and port from configuration.
|
||||
var master_server string = ""
|
||||
var master_server_port string = ""
|
||||
master_server_raw, ok := Cfg.Cfg["/servers_updating/master_server"]
|
||||
if ok {
|
||||
master_server = strings.Split(master_server_raw, ":")[0]
|
||||
master_server_port = strings.Split(master_server_raw, ":")[1]
|
||||
} else {
|
||||
master_server = "master.urbanterror.info"
|
||||
master_server_port = "27900"
|
||||
}
|
||||
fmt.Println("Using master server address: " + master_server + ":" + master_server_port)
|
||||
// IP addresses we will compose to return.
|
||||
conn, err1 := net.Dial("udp", master_server + ":" + master_server_port)
|
||||
if err1 != nil {
|
||||
fmt.Println("Error dialing to master server!")
|
||||
Eventer.LaunchEvent("setToolbarLabelText", map[string]string{"text": "Failed to connect to master server!"})
|
||||
return errors.New("Failed to connect to master server!")
|
||||
}
|
||||
defer conn.Close()
|
||||
// Get master server address and port from configuration.
|
||||
var master_server string = ""
|
||||
var master_server_port string = ""
|
||||
master_server_raw, ok := Cfg.Cfg["/servers_updating/master_server"]
|
||||
if ok {
|
||||
master_server = strings.Split(master_server_raw, ":")[0]
|
||||
master_server_port = strings.Split(master_server_raw, ":")[1]
|
||||
} else {
|
||||
master_server = "master.urbanterror.info"
|
||||
master_server_port = "27900"
|
||||
}
|
||||
fmt.Println("Using master server address: " + master_server + ":" + master_server_port)
|
||||
// IP addresses we will compose to return.
|
||||
conn, err1 := net.Dial("udp", master_server+":"+master_server_port)
|
||||
if err1 != nil {
|
||||
fmt.Println("Error dialing to master server!")
|
||||
Eventer.LaunchEvent("setToolbarLabelText", map[string]string{"text": "Failed to connect to master server!"})
|
||||
return errors.New("Failed to connect to master server!")
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
// Set deadline, so we won't wait forever.
|
||||
ddl := time.Now()
|
||||
// This should be enough. Maybe, you should'n run URTrator on modem
|
||||
// connections? :)
|
||||
ddl = ddl.Add(time.Second * 2)
|
||||
conn.SetDeadline(ddl)
|
||||
fmt.Println("Master server connection deadline set: " + ddl.String())
|
||||
// Set deadline, so we won't wait forever.
|
||||
ddl := time.Now()
|
||||
// This should be enough. Maybe, you should'n run URTrator on modem
|
||||
// connections? :)
|
||||
ddl = ddl.Add(time.Second * 2)
|
||||
conn.SetDeadline(ddl)
|
||||
fmt.Println("Master server connection deadline set: " + ddl.String())
|
||||
|
||||
// Request.
|
||||
// 68 - protocol version for 4.3
|
||||
// 70 - protocol version for 4.2.x
|
||||
msg := []byte(r.pp + "getservers 68 full empty")
|
||||
conn.Write(msg)
|
||||
// Request.
|
||||
// 68 - protocol version for 4.3
|
||||
// 70 - protocol version for 4.2.x
|
||||
msg := []byte(r.pp + "getservers 68 full empty")
|
||||
conn.Write(msg)
|
||||
|
||||
// UDP Buffer.
|
||||
var received_buf []byte = make([]byte, 4096)
|
||||
// Received buffer.
|
||||
var raw_received []byte
|
||||
// Received IP addresses.
|
||||
//var received []string
|
||||
fmt.Println("Receiving servers list...")
|
||||
for {
|
||||
_, err := conn.Read(received_buf)
|
||||
raw_received = append(raw_received, received_buf...)
|
||||
fmt.Println("Received " + strconv.Itoa(len(raw_received)) + " bytes")
|
||||
// UDP Buffer.
|
||||
var received_buf []byte = make([]byte, 4096)
|
||||
// Received buffer.
|
||||
var raw_received []byte
|
||||
// Received IP addresses.
|
||||
//var received []string
|
||||
fmt.Println("Receiving servers list...")
|
||||
for {
|
||||
_, err := conn.Read(received_buf)
|
||||
raw_received = append(raw_received, received_buf...)
|
||||
fmt.Println("Received " + strconv.Itoa(len(raw_received)) + " bytes")
|
||||
|
||||
if err != nil {
|
||||
// A bit hacky - if we have received data length lower or
|
||||
// equal to 4k - we have an error. Looks like conn.Read()
|
||||
// reads by 4k.
|
||||
if len(raw_received) < 4097 {
|
||||
fmt.Println("Error dialing to master server!")
|
||||
Eventer.LaunchEvent("setToolbarLabelText", map[string]string{"text": "Failed to connect to master server!"})
|
||||
return errors.New("Failed to connect to master server!")
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
// A bit hacky - if we have received data length lower or
|
||||
// equal to 4k - we have an error. Looks like conn.Read()
|
||||
// reads by 4k.
|
||||
if len(raw_received) < 4097 {
|
||||
fmt.Println("Error dialing to master server!")
|
||||
Eventer.LaunchEvent("setToolbarLabelText", map[string]string{"text": "Failed to connect to master server!"})
|
||||
return errors.New("Failed to connect to master server!")
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// Obtaining list of slices.
|
||||
var raw_received_slices [][]byte = bytes.Split(raw_received, []byte("\\"))
|
||||
// Obtaining list of slices.
|
||||
var raw_received_slices [][]byte = bytes.Split(raw_received, []byte("\\"))
|
||||
|
||||
// Every ip-port pair contains:
|
||||
// 1. IP as first 4 bytes
|
||||
// 2. Port as last 2 bytes.
|
||||
// Every package is a 7-bytes sequence, which starts with "\"
|
||||
// (code 92), which we used before to obtain list of slices.
|
||||
for _, slice := range raw_received_slices {
|
||||
// We need only 6-bytes slices. All other aren't represent
|
||||
// server's address.
|
||||
if len(slice) != 6 {
|
||||
continue
|
||||
}
|
||||
// Generate IP.
|
||||
ip := strconv.Itoa(int(slice[0])) + "." + strconv.Itoa(int(slice[1])) + "." + strconv.Itoa(int(slice[2])) + "." + strconv.Itoa(int(slice[3]))
|
||||
// Generate port from last two bytes.
|
||||
// This is a very shitty thing. Don't do this in real world.
|
||||
// Maybe bitshifting will help here, but I can't get it to work :(
|
||||
// Get first byte as integer and multiply it on 256 and summing with
|
||||
// second byte.
|
||||
p1 := int(slice[4]) * 256
|
||||
port := strconv.Itoa(p1 + int(slice[5]))
|
||||
addr := ip + ":" + port
|
||||
// Every ip-port pair contains:
|
||||
// 1. IP as first 4 bytes
|
||||
// 2. Port as last 2 bytes.
|
||||
// Every package is a 7-bytes sequence, which starts with "\"
|
||||
// (code 92), which we used before to obtain list of slices.
|
||||
for _, slice := range raw_received_slices {
|
||||
// We need only 6-bytes slices. All other aren't represent
|
||||
// server's address.
|
||||
if len(slice) != 6 {
|
||||
continue
|
||||
}
|
||||
// Generate IP.
|
||||
ip := strconv.Itoa(int(slice[0])) + "." + strconv.Itoa(int(slice[1])) + "." + strconv.Itoa(int(slice[2])) + "." + strconv.Itoa(int(slice[3]))
|
||||
// Generate port from last two bytes.
|
||||
// This is a very shitty thing. Don't do this in real world.
|
||||
// Maybe bitshifting will help here, but I can't get it to work :(
|
||||
// Get first byte as integer and multiply it on 256 and summing with
|
||||
// second byte.
|
||||
p1 := int(slice[4]) * 256
|
||||
port := strconv.Itoa(p1 + int(slice[5]))
|
||||
addr := ip + ":" + port
|
||||
|
||||
// Check if we already have this server added previously. If so - do nothing.
|
||||
_, ok := Cache.Servers[addr]
|
||||
if !ok {
|
||||
// Create cached server.
|
||||
Cache.CreateServer(addr)
|
||||
Cache.ServersMutex.Lock()
|
||||
Cache.Servers[addr].Server.Ip = ip
|
||||
Cache.Servers[addr].Server.Port = port
|
||||
Cache.ServersMutex.Unlock()
|
||||
}
|
||||
}
|
||||
// Check if we already have this server added previously. If so - do nothing.
|
||||
_, ok := Cache.Servers[addr]
|
||||
if !ok {
|
||||
// Create cached server.
|
||||
Cache.CreateServer(addr)
|
||||
Cache.ServersMutex.Lock()
|
||||
Cache.Servers[addr].Server.Ip = ip
|
||||
Cache.Servers[addr].Server.Port = port
|
||||
Cache.ServersMutex.Unlock()
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
return nil
|
||||
}
|
||||
|
||||
// Updates information about all available servers from master server and
|
||||
// parses it to usable format.
|
||||
func (r *Requester) UpdateAllServers(task bool) {
|
||||
fmt.Println("Starting all servers updating procedure...")
|
||||
err := r.getServers()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
r.Pooler.UpdateServers("all")
|
||||
fmt.Println("Starting all servers updating procedure...")
|
||||
err := r.getServers()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
r.Pooler.UpdateServers("all")
|
||||
|
||||
if task {
|
||||
Eventer.LaunchEvent("taskDone", map[string]string{"task_name": "Server's autoupdating"})
|
||||
}
|
||||
if task {
|
||||
Eventer.LaunchEvent("taskDone", map[string]string{"task_name": "Server's autoupdating"})
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Requester) UpdateFavoriteServers() {
|
||||
fmt.Println("Updating favorites servers...")
|
||||
r.Pooler.UpdateServers("favorites")
|
||||
fmt.Println("Updating favorites servers...")
|
||||
r.Pooler.UpdateServers("favorites")
|
||||
}
|
||||
|
||||
func (r *Requester) UpdateOneServer(server_address string) {
|
||||
fmt.Println("Updating server " + server_address)
|
||||
r.Pooler.UpdateOneServer(server_address)
|
||||
fmt.Println("Updating server " + server_address)
|
||||
r.Pooler.UpdateOneServer(server_address)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user