124 lines
2.2 KiB
Go
124 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var ErrInvalidAddress = errors.New("invalid ip address format")
|
|
|
|
// BanList returns the list of banned players and IP addresses
|
|
func BanList() (map[string]string, error) {
|
|
db, err := authDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer db.Close()
|
|
|
|
rows, err := db.Query(`SELECT addr, name FROM ban;`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r := make(map[string]string)
|
|
|
|
for rows.Next() {
|
|
var addr, name string
|
|
|
|
if err = rows.Scan(&addr, &name); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r[addr] = name
|
|
}
|
|
|
|
return r, nil
|
|
}
|
|
|
|
// IsBanned reports whether an IP address is banned
|
|
func IsBanned(addr string) (bool, string, error) {
|
|
db, err := authDB()
|
|
if err != nil {
|
|
return true, "", err
|
|
}
|
|
defer db.Close()
|
|
|
|
var name string
|
|
err = db.QueryRow(`SELECT name FROM ban WHERE addr = $1;`, addr).Scan(&name)
|
|
if err != nil && !errors.Is(err, sql.ErrNoRows) {
|
|
return true, "", err
|
|
}
|
|
|
|
return name != "", name, nil
|
|
}
|
|
|
|
// IsBanned reports whether a Conn is banned
|
|
func (c *Conn) IsBanned() (bool, string, error) {
|
|
addr := c.Addr().(*net.UDPAddr).IP.String()
|
|
|
|
banned, name, err := IsBanned(addr)
|
|
if err != nil {
|
|
return true, "", err
|
|
}
|
|
|
|
return banned, name, nil
|
|
}
|
|
|
|
// Ban adds an IP address to the ban list
|
|
func Ban(addr, name string) error {
|
|
db, err := authDB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
if net.ParseIP(addr) == nil {
|
|
return ErrInvalidAddress
|
|
}
|
|
|
|
_, err = db.Exec(`INSERT INTO ban (
|
|
addr,
|
|
name
|
|
) VALUES (
|
|
$1,
|
|
$2
|
|
);`, addr, name)
|
|
return err
|
|
}
|
|
|
|
// Ban adds a Conn to the ban list
|
|
func (c *Conn) Ban() error {
|
|
banned, _, err := c.IsBanned()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if banned {
|
|
return fmt.Errorf("ip address %s is already banned", c.Addr().String())
|
|
}
|
|
|
|
name := c.Username()
|
|
addr := c.Addr().(*net.UDPAddr).IP.String()
|
|
|
|
Ban(addr, name)
|
|
|
|
c.CloseWith(AccessDeniedCustomString, "Banned.", false)
|
|
return nil
|
|
}
|
|
|
|
// Unban removes a player from the ban list
|
|
func Unban(id string) error {
|
|
db, err := authDB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
_, err = db.Exec(`DELETE FROM ban WHERE name = $1 OR addr = $2;`, id, id)
|
|
return err
|
|
}
|