msgrabber

package main

import (
    "flag"
    "fmt"
    "io/ioutil"
    "net"
    "strings"

    "github.com/oschwald/geoip2-golang"
)

func main() {
    masterServerFlag := flag.String("master", "127.0.0.1:27010", "Адрес и порт мастер-сервера")
    portFilterFlag := flag.String("port", "", "Порт для фильтрации серверов (например, 27015)")
    countryFilterFlag := flag.String("country", "", "Код страны для фильтрации серверов (например, RU)")
    cityFilterFlag := flag.String("city", "", "Название города для фильтрации серверов")
    flag.Parse()

    masterServerAddr, err := net.ResolveUDPAddr("udp4", *masterServerFlag)
    if err != nil {
        fmt.Printf("Не удалось разрешить адрес мастер-сервера: %v\n", err)
        return
    }

    conn, err := net.DialUDP("udp4", nil, masterServerAddr)
    if err != nil {
        fmt.Printf("Не удалось подключиться к мастер-серверу: %v\n", err)
        return
    }
    defer conn.Close()

    // query := []byte("\x31 ... \x00")
    query := []byte("\x31")
    _, err = conn.Write(query)
    if err != nil {
        fmt.Printf("Не удалось отправить запрос: %v\n", err)
        return
    }

    buffer := make([]byte, 8192)
    n, _, err := conn.ReadFromUDP(buffer)
    if err != nil {
        fmt.Printf("Не удалось получить ответ: %v\n", err)
        return
    }

    servers := parseServerList(buffer[6:n])

    if *portFilterFlag != "" {
        servers = filterServersByPort(servers, *portFilterFlag)
    }

    if *countryFilterFlag != "" {
        servers, err = filterServersByCountry(servers, *countryFilterFlag)
        if err != nil {
            fmt.Printf("Ошибка при фильтрации серверов по стране: %v\n", err)
            return
        }
    }

    if *cityFilterFlag != "" {
        servers, err = filterServersByCity(servers, *cityFilterFlag)
        if err != nil {
            fmt.Printf("Ошибка при фильтрации серверов по городу: %v\n", err)
            return
        }
    }

    err = ioutil.WriteFile("ms-servers.txt", []byte(strings.Join(servers, "\n")), 0644)
    if err != nil {
        fmt.Printf("Не удалось сохранить список серверов: %v\n", err)
        return
    }

    fmt.Println("Список серверов успешно сохранен в файле ms-servers.txt")
}

func parseServerList(data []byte) []string {
    var servers []string

    for i := 6; i < len(data); {
        ip := fmt.Sprintf("%d.%d.%d.%d", data[i], data[i+1], data[i+2], data[i+3])
        port := int(data[i+4])<<8 + int(data[i+5])
        server := fmt.Sprintf("%s:%d", ip, port)
        servers = append(servers, server)

        i += 6
    }

    return servers
}

func filterServersByPort(servers []string, portFilter string) []string {
    var filteredServers []string

    for _, server := range servers {
        parts := strings.Split(server, ":")
        if len(parts) != 2 {
            continue
        }

        if parts[1] == portFilter {
            filteredServers = append(filteredServers, server)
        }
    }

    return filteredServers
}

func filterServersByCountry(servers []string, countryFilter string) ([]string, error) {
    db, err := geoip2.Open("GeoLite2-Country.mmdb")
    if err != nil {
        return nil, fmt.Errorf("не удалось открыть базу данных GeoIP: %v", err)
    }
    defer db.Close()

    var filteredServers []string

    for _, server := range servers {
        parts := strings.Split(server, ":")
        if len(parts) != 2 {
            continue
        }

        ip := net.ParseIP(parts[0])
        record, err := db.Country(ip)
        if err != nil {
            continue
        }

        if record.Country.IsoCode == countryFilter {
            filteredServers = append(filteredServers, server)
        }
    }

    return filteredServers, nil
}

func filterServersByCity(servers []string, cityFilter string) ([]string, error) {
    db, err := geoip2.Open("GeoLite2-City.mmdb")
    if err != nil {
        return nil, fmt.Errorf("не удалось открыть базу данных GeoIP: %v", err)
    }
    defer db.Close()

    var filteredServers []string

    for _, server := range servers {
        parts := strings.Split(server, ":")
        if len(parts) != 2 {
            continue
        }

        ip := net.ParseIP(parts[0])
        record, err := db.City(ip)
        if err != nil {
            continue
        }

        if record.City.Names["en"] == cityFilter {
            filteredServers = append(filteredServers, server)
        }
    }

    return filteredServers, nil
}