// Populate bans
func populateBans(server *Server, db *sql.DB) (err error) {
stmt, err := db.Prepare("SELECT base, mask, name, hash, reason, start, duration FROM bans WHERE server_id=?")
if err != nil {
return
}
rows, err := stmt.Query(server.Id)
if err != nil {
return err
}
for rows.Next() {
var (
Ban ban.Ban
IP []byte
StartDate string
Duration int64
)
err = rows.Scan(&IP, &Ban.Mask, &Ban.Username, &Ban.CertHash, &Ban.Reason, &StartDate, &Duration)
if err != nil {
return err
}
if len(IP) == 16 && IP[10] == 0xff && IP[11] == 0xff {
Ban.IP = net.IPv4(IP[12], IP[13], IP[14], IP[15])
} else {
Ban.IP = IP
}
Ban.SetISOStartDate(StartDate)
Ban.Duration = uint32(Duration)
server.Bans = append(server.Bans, Ban)
}
return
}
func (server *Server) handleBanListMessage(client *Client, msg *Message) {
banlist := &mumbleproto.BanList{}
err := proto.Unmarshal(msg.buf, banlist)
if err != nil {
client.Panic(err)
return
}
rootChan := server.RootChannel()
if !acl.HasPermission(&rootChan.ACL, client, acl.BanPermission) {
client.sendPermissionDenied(client, rootChan, acl.BanPermission)
return
}
if banlist.Query != nil && *banlist.Query != false {
banlist.Reset()
server.banlock.RLock()
defer server.banlock.RUnlock()
for _, ban := range server.Bans {
entry := &mumbleproto.BanList_BanEntry{}
entry.Address = ban.IP
entry.Mask = proto.Uint32(uint32(ban.Mask))
entry.Name = proto.String(ban.Username)
entry.Hash = proto.String(ban.CertHash)
entry.Reason = proto.String(ban.Reason)
entry.Start = proto.String(ban.ISOStartDate())
entry.Duration = proto.Uint32(ban.Duration)
banlist.Bans = append(banlist.Bans, entry)
}
if err := client.sendMessage(banlist); err != nil {
client.Panic("Unable to send BanList")
}
} else {
server.banlock.Lock()
defer server.banlock.Unlock()
server.Bans = server.Bans[0:0]
for _, entry := range banlist.Bans {
ban := ban.Ban{}
ban.IP = entry.Address
ban.Mask = int(*entry.Mask)
if entry.Name != nil {
ban.Username = *entry.Name
}
if entry.Hash != nil {
ban.CertHash = *entry.Hash
}
if entry.Reason != nil {
ban.Reason = *entry.Reason
}
if entry.Start != nil {
ban.SetISOStartDate(*entry.Start)
}
if entry.Duration != nil {
ban.Duration = *entry.Duration
}
server.Bans = append(server.Bans, ban)
}
server.UpdateFrozenBans(server.Bans)
client.Printf("Banlist updated")
}
}
