3
0
mirror of https://github.com/ergochat/ergo.git synced 2024-11-14 16:09:32 +01:00
ergo/irc/server.go

1159 lines
33 KiB
Go
Raw Normal View History

// Copyright (c) 2012-2014 Jeremy Latt
// Copyright (c) 2014-2015 Edmund Huber
// Copyright (c) 2016- Daniel Oaks <daniel@danieloaks.net>
// released under the MIT license
2012-04-07 20:44:59 +02:00
package irc
import (
2014-02-12 01:35:32 +01:00
"bufio"
"crypto/tls"
"database/sql"
"encoding/base64"
"fmt"
2012-04-07 20:44:59 +02:00
"log"
"net"
"net/http"
2014-02-12 01:35:32 +01:00
"os"
"os/signal"
"regexp"
"strconv"
"strings"
"syscall"
2012-04-18 07:21:41 +02:00
"time"
"github.com/DanielOaks/girc-go/ircmsg"
2016-08-19 15:21:52 +02:00
"github.com/tidwall/buntdb"
)
2016-09-12 04:22:50 +02:00
// Limits holds the maximum limits for various things such as topic lengths
type Limits struct {
2016-09-12 04:40:09 +02:00
AwayLen int
KickLen int
TopicLen int
2016-09-12 04:22:50 +02:00
}
2012-04-07 20:44:59 +02:00
type Server struct {
accounts map[string]*ClientAccount
channels ChannelNameMap
clients *ClientLookupSet
commands chan Command
ctime time.Time
db *sql.DB
store buntdb.DB
idle chan *Client
2016-09-12 04:22:50 +02:00
limits Limits
motdLines []string
name Name
nameString string // cache for server name string since it's used with almost every reply
newConns chan clientConn
operators map[Name][]byte
password []byte
passwords *PasswordManager
accountRegistration *AccountRegistration
signals chan os.Signal
proxyAllowedFrom []string
whoWas *WhoWasList
isupport *ISupportList
checkIdent bool
}
var (
SERVER_SIGNALS = []os.Signal{
syscall.SIGINT,
syscall.SIGHUP, // eventually we expect to use HUP to reload config
syscall.SIGTERM,
syscall.SIGQUIT,
}
)
2016-06-28 17:09:07 +02:00
type clientConn struct {
Conn net.Conn
IsTLS bool
}
2014-02-09 16:53:42 +01:00
func NewServer(config *Config) *Server {
2012-12-09 21:51:50 +01:00
server := &Server{
2016-09-12 04:22:50 +02:00
accounts: make(map[string]*ClientAccount),
channels: make(ChannelNameMap),
clients: NewClientLookupSet(),
commands: make(chan Command),
ctime: time.Now(),
db: OpenDB(config.Datastore.SQLitePath),
idle: make(chan *Client),
limits: Limits{
2016-09-12 04:40:09 +02:00
AwayLen: config.Limits.AwayLen,
KickLen: config.Limits.KickLen,
TopicLen: config.Limits.TopicLen,
2016-09-12 04:22:50 +02:00
},
name: NewName(config.Server.Name),
nameString: NewName(config.Server.Name).String(),
2016-06-28 17:09:07 +02:00
newConns: make(chan clientConn),
operators: config.Operators(),
signals: make(chan os.Signal, len(SERVER_SIGNALS)),
proxyAllowedFrom: config.Server.ProxyAllowedFrom,
whoWas: NewWhoWasList(config.Limits.WhowasEntries),
checkIdent: config.Server.CheckIdent,
}
2014-02-09 19:07:40 +01:00
2016-08-19 15:21:52 +02:00
// open data store
db, err := buntdb.Open(config.Datastore.Path)
if err != nil {
log.Fatal(fmt.Sprintf("Failed to open datastore: %s", err.Error()))
}
server.store = *db
// load password manager
err = server.store.View(func(tx *buntdb.Tx) error {
saltString, err := tx.Get(keySalt)
if err != nil {
return fmt.Errorf("Could not retrieve salt string: %s", err.Error())
}
salt, err := base64.StdEncoding.DecodeString(saltString)
if err != nil {
return err
}
pwm := NewPasswordManager(salt)
server.passwords = &pwm
return nil
})
if err != nil {
log.Fatal(fmt.Sprintf("Could not load salt: %s", err.Error()))
}
if config.Server.MOTD != "" {
file, err := os.Open(config.Server.MOTD)
if err == nil {
defer file.Close()
reader := bufio.NewReader(file)
for {
line, err := reader.ReadString('\n')
if err != nil {
break
}
line = strings.TrimRight(line, "\r\n")
2016-06-19 13:59:18 +02:00
// "- " is the required prefix for MOTD, we just add it here to make
// bursting it out to clients easier
line = fmt.Sprintf("- %s", line)
server.motdLines = append(server.motdLines, line)
}
}
}
//TODO(dan): Hot damn this is an ugly hack. Fix it properly at some point.
ChannelNameExpr = regexp.MustCompile(fmt.Sprintf(`^[#][\pL\pN\pP\pS]{1,%d}$`, config.Limits.ChannelLen))
NicknameExpr = regexp.MustCompile(fmt.Sprintf("^[\\pL\\pN\\pP\\pS]{1,%d}$", config.Limits.NickLen))
2014-03-06 20:00:48 +01:00
if config.Server.Password != "" {
server.password = config.Server.PasswordBytes()
}
2014-02-26 00:57:35 +01:00
server.loadChannels()
for _, addr := range config.Server.Listen {
server.listen(addr, config.TLSListeners())
2014-02-26 00:57:35 +01:00
}
if config.Server.Wslisten != "" {
server.wslisten(config.Server.Wslisten, config.Server.TLSListeners)
}
// registration
accountReg := NewAccountRegistration(config.Registration.Accounts)
server.accountRegistration = &accountReg
// Attempt to clean up when receiving these signals.
signal.Notify(server.signals, SERVER_SIGNALS...)
2014-03-02 20:51:29 +01:00
// add RPL_ISUPPORT tokens
server.isupport = NewISupportList()
2016-09-12 04:40:09 +02:00
server.isupport.Add("AWAYLEN", strconv.Itoa(server.limits.AwayLen))
server.isupport.Add("CASEMAPPING", "ascii")
2016-09-13 14:05:38 +02:00
server.isupport.Add("CHANMODES", strings.Join([]string{ChannelModes{BanMask, ExceptMask, InviteMask}.String(), "", ChannelModes{UserLimit, Key}.String(), ChannelModes{InviteOnly, Moderated, NoOutside, OpOnlyTopic, Persistent, ReOp, Secret}.String()}, ","))
server.isupport.Add("CHANNELLEN", strconv.Itoa(config.Limits.ChannelLen))
server.isupport.Add("CHANTYPES", "#")
server.isupport.Add("EXCEPTS", "")
server.isupport.Add("INVEX", "")
2016-09-12 04:40:09 +02:00
server.isupport.Add("KICKLEN", strconv.Itoa(server.limits.KickLen))
// server.isupport.Add("MAXLIST", "") //TODO(dan): Support max list length?
// server.isupport.Add("MODES", "") //TODO(dan): Support max modes?
2016-04-12 07:44:00 +02:00
server.isupport.Add("NETWORK", config.Network.Name)
server.isupport.Add("NICKLEN", strconv.Itoa(config.Limits.NickLen))
server.isupport.Add("PREFIX", "(qaohv)~&@%+")
// server.isupport.Add("STATUSMSG", "@+") //TODO(dan): Support STATUSMSG
// server.isupport.Add("TARGMAX", "") //TODO(dan): Support this
2016-09-12 04:40:09 +02:00
server.isupport.Add("TOPICLEN", strconv.Itoa(server.limits.TopicLen))
// account registration
if server.accountRegistration.Enabled {
// 'none' isn't shown in the REGCALLBACKS vars
var enabledCallbacks []string
for _, name := range server.accountRegistration.EnabledCallbacks {
if name != "*" {
enabledCallbacks = append(enabledCallbacks, name)
}
}
server.isupport.Add("REGCOMMANDS", "CREATE,VERIFY")
server.isupport.Add("REGCALLBACKS", strings.Join(enabledCallbacks, ","))
server.isupport.Add("REGCREDTYPES", "passphrase,certfp")
}
server.isupport.RegenerateCachedReply()
2014-02-26 00:57:35 +01:00
return server
}
2014-03-08 03:14:02 +01:00
func loadChannelList(channel *Channel, list string, maskMode ChannelMode) {
if list == "" {
return
}
2014-03-09 21:45:36 +01:00
channel.lists[maskMode].AddAll(NewNames(strings.Split(list, " ")))
2014-03-08 03:14:02 +01:00
}
2014-02-26 00:57:35 +01:00
func (server *Server) loadChannels() {
rows, err := server.db.Query(`
2014-03-08 03:14:02 +01:00
SELECT name, flags, key, topic, user_limit, ban_list, except_list,
invite_list
FROM channel`)
if err != nil {
2014-03-06 08:07:55 +01:00
log.Fatal("error loading channels: ", err)
}
for rows.Next() {
var name, flags, key, topic string
var userLimit uint64
2014-03-08 03:14:02 +01:00
var banList, exceptList, inviteList string
err = rows.Scan(&name, &flags, &key, &topic, &userLimit, &banList,
&exceptList, &inviteList)
if err != nil {
2014-03-08 03:14:02 +01:00
log.Println("Server.loadChannels:", err)
continue
}
2016-04-21 11:29:50 +02:00
channel := NewChannel(server, NewName(name), false)
2014-02-26 00:57:35 +01:00
for _, flag := range flags {
channel.flags[ChannelMode(flag)] = true
}
2016-06-19 13:59:18 +02:00
channel.key = key
channel.topic = topic
channel.userLimit = userLimit
2014-03-08 03:14:02 +01:00
loadChannelList(channel, banList, BanMask)
loadChannelList(channel, exceptList, ExceptMask)
loadChannelList(channel, inviteList, InviteMask)
}
2013-06-03 07:07:50 +02:00
}
func (server *Server) Shutdown() {
//TODO(dan): Make sure we disallow new nicks
for _, client := range server.clients.byNick {
2016-06-19 13:59:18 +02:00
client.Notice("Server is shutting down")
}
if err := server.db.Close(); err != nil {
Log.error.Println("Server.Shutdown db.Close: error:", err)
}
if err := server.store.Close(); err != nil {
Log.error.Println("Server.Shutdown store.Close: error:", err)
}
}
2014-02-24 04:13:45 +01:00
func (server *Server) Run() {
// defer closing db/store
defer server.db.Close()
defer server.store.Close()
done := false
for !done {
select {
case <-server.signals:
server.Shutdown()
done = true
case conn := <-server.newConns:
2016-06-28 17:09:07 +02:00
NewClient(server, conn.Conn, conn.IsTLS)
2016-06-19 13:59:18 +02:00
/*TODO(dan): LOOK AT THIS MORE CLOSELY
2014-02-24 07:21:39 +01:00
case cmd := <-server.commands:
2014-02-25 02:45:04 +01:00
server.processCommand(cmd)
2016-06-19 13:59:18 +02:00
*/
2014-02-24 07:21:39 +01:00
case client := <-server.idle:
client.Idle()
2014-02-15 03:28:36 +01:00
}
}
}
//
// listen goroutine
//
func (s *Server) listen(addr string, tlsMap map[Name]*tls.Config) {
config, listenTLS := tlsMap[NewName(addr)]
listener, err := net.Listen("tcp", addr)
2012-04-07 20:44:59 +02:00
if err != nil {
log.Fatal(s, "listen error: ", err)
2012-04-07 20:44:59 +02:00
}
tlsString := "plaintext"
if listenTLS {
config.ClientAuth = tls.RequestClientCert
listener = tls.NewListener(listener, config)
tlsString = "TLS"
}
Log.info.Printf("%s listening on %s using %s.", s, addr, tlsString)
go func() {
for {
conn, err := listener.Accept()
if err != nil {
Log.error.Printf("%s accept error: %s", s, err)
continue
}
Log.debug.Printf("%s accept: %s", s, conn.RemoteAddr())
2016-06-28 17:09:07 +02:00
newConn := clientConn{
Conn: conn,
IsTLS: listenTLS,
}
s.newConns <- newConn
}
}()
}
//
// websocket listen goroutine
//
func (s *Server) wslisten(addr string, tlsMap map[string]*TLSListenConfig) {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
Log.error.Printf("%s method not allowed", s)
return
}
// We don't have any subprotocols, so if someone attempts to `new
// WebSocket(server, "subprotocol")` they'll break here, instead of
// getting the default, ambiguous, response from gorilla.
if v, ok := r.Header["Sec-Websocket-Protocol"]; ok {
http.Error(w, fmt.Sprintf("WebSocket subprocotols (e.g. %s) not supported", v), 400)
}
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
Log.error.Printf("%s websocket upgrade error: %s", s, err)
return
}
2016-06-28 17:09:07 +02:00
newConn := clientConn{
Conn: WSContainer{ws},
IsTLS: false, //TODO(dan): track TLS or not here properly
}
s.newConns <- newConn
})
go func() {
config, listenTLS := tlsMap[addr]
tlsString := "plaintext"
var err error
if listenTLS {
tlsString = "TLS"
}
Log.info.Printf("%s websocket listening on %s using %s.", s, addr, tlsString)
if listenTLS {
err = http.ListenAndServeTLS(addr, config.Cert, config.Key, nil)
} else {
err = http.ListenAndServe(addr, nil)
}
if err != nil {
Log.error.Printf("%s listenAndServe (%s) error: %s", s, tlsString, err)
}
}()
}
2014-02-15 03:28:36 +01:00
//
// server functionality
2014-02-15 03:28:36 +01:00
//
func (s *Server) tryRegister(c *Client) {
if c.registered || !c.HasNick() || !c.HasUsername() ||
(c.capState == CapNegotiating) {
return
}
c.Register()
2016-06-19 13:59:18 +02:00
// send welcome text
//NOTE(dan): we specifically use the NICK here instead of the nickmask
// see http://modern.ircdocs.horse/#rplwelcome-001 for details on why we avoid using the nickmask
c.Send(nil, s.nameString, RPL_WELCOME, c.nickString, fmt.Sprintf("Welcome to the Internet Relay Network %s", c.nickString))
c.Send(nil, s.nameString, RPL_YOURHOST, c.nickString, fmt.Sprintf("Your host is %s, running version %s", s.nameString, VER))
c.Send(nil, s.nameString, RPL_CREATED, c.nickString, fmt.Sprintf("This server was created %s", s.ctime.Format(time.RFC1123)))
2016-06-19 13:59:18 +02:00
//TODO(dan): Look at adding last optional [<channel modes with a parameter>] parameter
c.Send(nil, s.nameString, RPL_MYINFO, c.nickString, s.nameString, VER, supportedUserModesString, supportedChannelModesString)
c.RplISupport()
s.MOTD(c)
2016-06-28 17:09:07 +02:00
c.Send(nil, c.nickMaskString, RPL_UMODEIS, c.nickString, c.ModeString())
}
2014-02-12 00:33:02 +01:00
func (server *Server) MOTD(client *Client) {
if len(server.motdLines) < 1 {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOMOTD, client.nickString, "MOTD File is missing")
2014-02-12 01:35:32 +01:00
return
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_MOTDSTART, client.nickString, fmt.Sprintf("- %s Message of the day - ", server.nameString))
for _, line := range server.motdLines {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_MOTD, client.nickString, line)
2014-02-12 01:35:32 +01:00
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_ENDOFMOTD, client.nickString, "End of MOTD command")
2014-02-12 00:33:02 +01:00
}
2014-03-09 21:45:36 +01:00
func (s *Server) Id() Name {
2013-05-11 22:55:01 +02:00
return s.name
}
func (s *Server) String() string {
2014-03-09 21:45:36 +01:00
return s.name.String()
}
2014-03-09 21:45:36 +01:00
func (s *Server) Nick() Name {
return s.Id()
}
//
// registration commands
//
// PASS <password>
func passHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
if client.registered {
client.Send(nil, server.nameString, ERR_ALREADYREGISTRED, client.nickString, "You may not reregister")
return false
}
// if no password exists, skip checking
if len(server.password) == 0 {
client.authorized = true
return false
}
// check the provided password
2016-06-19 13:59:18 +02:00
password := []byte(msg.Params[0])
if ComparePassword(server.password, password) != nil {
2016-06-28 08:06:17 +02:00
client.Send(nil, server.nameString, ERR_PASSWDMISMATCH, client.nickString, "Password incorrect")
client.Send(nil, server.nameString, "ERROR", "Password incorrect")
return true
2014-02-15 03:28:36 +01:00
}
client.authorized = true
return false
}
// PROXY TCP4/6 SOURCEIP DESTIP SOURCEPORT DESTPORT
// http://www.haproxy.org/download/1.5/doc/proxy-protocol.txt
func proxyHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
clientAddress := IPString(client.socket.conn.RemoteAddr()).String()
clientHostname := client.hostname.String()
for _, address := range server.proxyAllowedFrom {
if clientHostname == address || clientAddress == address {
client.hostname = LookupHostname(NewName(msg.Params[1]))
return false
}
}
client.Quit("PROXY command is not usable from your address")
return true
}
// USER <username> * 0 <realname>
func userHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
if client.registered {
client.Send(nil, server.nameString, ERR_ALREADYREGISTRED, client.nickString, "You may not reregister")
return false
}
if !client.authorized {
2016-06-19 13:59:18 +02:00
client.Quit("Bad password")
return true
}
if client.username != "" && client.realname != "" {
return false
}
// confirm that username is valid
//
if !Name(msg.Params[0]).IsNickname() {
client.Send(nil, "", "ERROR", "Malformed username")
return true
}
// set user info and log client in
//TODO(dan): Could there be a race condition here with adding/removing the client?
//TODO(dan): we should do something like server.clients.Replace(client) instead
// we do it this way to ONLY replace what hasn't already been set
server.clients.Remove(client)
if !client.HasUsername() {
client.username = Name("~" + msg.Params[0])
2016-06-19 13:59:18 +02:00
client.updateNickMask()
}
if client.realname == "" {
2016-06-19 13:59:18 +02:00
client.realname = msg.Params[3]
}
server.clients.Add(client)
server.tryRegister(client)
2016-06-19 13:59:18 +02:00
return false
2014-02-15 03:28:36 +01:00
}
// QUIT [<reason>]
func quitHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
reason := "Quit"
if len(msg.Params) > 0 {
reason += ": " + msg.Params[0]
}
2016-06-19 13:59:18 +02:00
client.Quit(reason)
return true
2014-02-18 22:25:21 +01:00
}
2014-02-15 03:28:36 +01:00
//
// normal commands
//
// PING <server1> [<server2>]
func pingHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-22 14:04:13 +02:00
client.Send(nil, server.nameString, "PONG", msg.Params...)
return false
2014-02-15 03:28:36 +01:00
}
// PONG <server> [ <server2> ]
func pongHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-22 14:04:13 +02:00
// client gets touched when they send this command, so we don't need to do anything
return false
}
// JOIN <channel>{,<channel>} [<key>{,<key>}]
// JOIN 0
func joinHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
// handle JOIN 0
if msg.Params[0] == "0" {
for channel := range client.channels {
2016-06-19 13:59:18 +02:00
channel.Part(client, client.nickString)
}
return false
}
// handle regular JOINs
channels := strings.Split(msg.Params[0], ",")
var keys []string
if len(msg.Params) > 1 {
keys = strings.Split(msg.Params[1], ",")
}
2016-06-19 13:59:18 +02:00
var name Name
for i, nameString := range channels {
name = Name(nameString)
2014-03-09 21:45:36 +01:00
if !name.IsChannel() {
fmt.Println("ISN'T CHANNEL NAME:", nameString)
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, nameString, "No such channel")
2014-02-22 02:19:02 +01:00
continue
}
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(name)
2014-02-22 02:19:02 +01:00
if channel == nil {
2016-06-19 13:59:18 +02:00
channel = NewChannel(server, name, true)
2014-02-22 02:19:02 +01:00
}
var key string
if len(keys) > i {
key = keys[i]
}
2014-02-19 04:31:59 +01:00
channel.Join(client, key)
}
2016-06-19 13:59:18 +02:00
return false
}
// PART <channel>{,<channel>} [<reason>]
func partHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
channels := strings.Split(msg.Params[0], ",")
2016-06-19 13:59:18 +02:00
var reason string //TODO(dan): if this isn't supplied here, make sure the param doesn't exist in the PART message sent to other users
if len(msg.Params) > 1 {
reason = msg.Params[1]
}
for _, chname := range channels {
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(chname))
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
continue
}
2016-06-19 13:59:18 +02:00
channel.Part(client, reason)
}
2016-06-19 13:59:18 +02:00
return false
}
// TOPIC <channel> [<topic>]
func topicHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(msg.Params[0]))
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, msg.Params[0], "No such channel")
return false
}
if len(msg.Params) > 1 {
channel.SetTopic(client, msg.Params[1])
} else {
channel.GetTopic(client)
}
2016-06-19 13:59:18 +02:00
return false
}
// PRIVMSG <target>{,<target>} <message>
func privmsgHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
targets := strings.Split(msg.Params[0], ",")
message := msg.Params[1]
2016-06-19 13:59:18 +02:00
var target Name
for _, targetString := range targets {
target = Name(targetString)
if target.IsChannel() {
channel := server.channels.Get(target)
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, targetString, "No such channel")
continue
}
channel.PrivMsg(client, message)
} else {
user := server.clients.Get(target)
if user == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHNICK, targetString, "No such nick")
continue
}
2016-06-19 13:59:18 +02:00
user.Send(nil, client.nickMaskString, "PRIVMSG", user.nickString, message)
if user.flags[Away] {
2016-06-19 13:59:18 +02:00
//TODO(dan): possibly implement cooldown of away notifications to users
client.Send(nil, server.nameString, RPL_AWAY, user.nickString, user.awayMessage)
}
}
2014-02-12 00:44:58 +01:00
}
2016-06-19 13:59:18 +02:00
return false
2012-04-07 20:44:59 +02:00
}
2012-12-13 08:27:17 +01:00
func (client *Client) WhoisChannelsNames(target *Client) []string {
isMultiPrefix := target.capabilities[MultiPrefix]
var chstrs []string
index := 0
for channel := range client.channels {
// channel is secret and the target can't see it
if !target.flags[Operator] && channel.flags[Secret] && !channel.members.Has(target) {
continue
}
chstrs = append(chstrs, channel.members[client].Prefixes(isMultiPrefix)+channel.name.String())
index++
}
return chstrs
}
// WHOIS [ <target> ] <mask> *( "," <mask> )
func whoisHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var masksString string
//var target string
2014-02-09 02:43:59 +01:00
if len(msg.Params) > 1 {
//target = msg.Params[0]
2016-06-19 13:59:18 +02:00
masksString = msg.Params[1]
} else {
2016-06-19 13:59:18 +02:00
masksString = msg.Params[0]
}
2014-02-09 02:43:59 +01:00
2016-06-19 13:59:18 +02:00
if client.flags[Operator] {
masks := strings.Split(masksString, ",")
for _, mask := range masks {
matches := server.clients.FindAll(Name(mask))
if len(matches) == 0 {
client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, mask, "No such nick")
continue
}
for mclient := range matches {
2016-09-05 06:23:57 +02:00
client.getWhoisOf(mclient)
2016-06-19 13:59:18 +02:00
}
2014-02-18 04:08:57 +01:00
}
2016-06-19 13:59:18 +02:00
} else {
// specifically treat this as a single lookup rather than splitting as we do above
// this is by design
mclient := server.clients.Get(Name(masksString))
if mclient == nil {
client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, masksString, "No such nick")
// fall through, ENDOFWHOIS is always sent
} else {
client.getWhoisOf(mclient)
2014-02-09 02:43:59 +01:00
}
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_ENDOFWHOIS, client.nickString, masksString, "End of /WHOIS list")
return false
}
func (client *Client) getWhoisOf(target *Client) {
client.Send(nil, client.server.nameString, RPL_WHOISUSER, client.nickString, target.nickString, target.username.String(), target.hostname.String(), "*", target.realname)
2016-09-05 06:23:57 +02:00
//TODO(dan): ...one channel per reply? really?
for _, line := range client.WhoisChannelsNames(target) {
client.Send(nil, client.server.nameString, RPL_WHOISCHANNELS, client.nickString, target.nickString, line)
}
2016-06-19 13:59:18 +02:00
if target.flags[Operator] {
client.Send(nil, client.server.nameString, RPL_WHOISOPERATOR, client.nickString, target.nickString, "is an IRC operator")
}
2016-09-05 06:23:57 +02:00
if target.certfp != "" && (client.flags[Operator] || client == target) {
client.Send(nil, client.server.nameString, RPL_WHOISCERTFP, client.nickString, target.nickString, fmt.Sprintf("has client certificate fingerprint %s", target.certfp))
2016-06-19 13:59:18 +02:00
}
2016-09-05 06:23:57 +02:00
client.Send(nil, client.server.nameString, RPL_WHOISIDLE, client.nickString, target.nickString, strconv.FormatUint(target.IdleSeconds(), 10), strconv.FormatInt(target.SignonTime(), 10), "seconds idle, signon time")
2016-06-19 13:59:18 +02:00
}
// <channel> <user> <host> <server> <nick> ( "H" / "G" ) ["*"] [ ( "@" / "+" ) ]
// :<hopcount> <real name>
func (target *Client) RplWhoReply(channel *Channel, client *Client) {
channelName := "*"
flags := ""
if client.flags[Away] {
flags = "G"
} else {
flags = "H"
}
if client.flags[Operator] {
flags += "*"
}
if channel != nil {
flags += channel.members[client].Prefixes(target.capabilities[MultiPrefix])
channelName = channel.name.String()
}
target.Send(nil, target.server.nameString, RPL_WHOREPLY, target.nickString, channelName, client.username.String(), client.hostname.String(), client.server.nameString, client.nickString, flags, string(client.hops), client.realname)
2014-02-09 02:43:59 +01:00
}
2014-02-09 03:14:39 +01:00
func whoChannel(client *Client, channel *Channel, friends ClientSet) {
2014-02-09 03:49:52 +01:00
for member := range channel.members {
if !client.flags[Invisible] || friends[client] {
client.RplWhoReply(channel, member)
2014-02-18 06:30:14 +01:00
}
2014-02-09 03:49:52 +01:00
}
}
// WHO [ <mask> [ "o" ] ]
func whoHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
friends := client.Friends()
2016-06-19 13:59:18 +02:00
var mask Name
if len(msg.Params) > 0 {
2016-06-19 13:59:18 +02:00
mask = Name(msg.Params[0])
}
//TODO(dan): is this used and would I put this param in the Modern doc?
// if not, can we remove it?
//var operatorOnly bool
//if len(msg.Params) > 1 && msg.Params[1] == "o" {
// operatorOnly = true
//}
2014-02-09 03:49:52 +01:00
2014-02-09 07:42:14 +01:00
if mask == "" {
2014-02-09 03:49:52 +01:00
for _, channel := range server.channels {
whoChannel(client, channel, friends)
2014-02-09 03:49:52 +01:00
}
2014-03-09 21:45:36 +01:00
} else if mask.IsChannel() {
// TODO implement wildcard matching
2016-06-19 13:59:18 +02:00
//TODO(dan): ^ only for opers
channel := server.channels.Get(mask)
2014-02-09 03:49:52 +01:00
if channel != nil {
whoChannel(client, channel, friends)
2014-02-09 03:49:52 +01:00
}
} else {
for mclient := range server.clients.FindAll(mask) {
client.RplWhoReply(nil, mclient)
2014-02-09 03:49:52 +01:00
}
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_ENDOFWHO, client.nickString, mask.String(), "End of WHO list")
return false
2014-02-09 03:49:52 +01:00
}
2014-02-09 19:07:40 +01:00
// OPER <name> <password>
func operHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
name := NewName(msg.Params[0])
hash := server.operators[name]
password := []byte(msg.Params[1])
2014-02-09 19:07:40 +01:00
2016-06-19 13:59:18 +02:00
err := ComparePassword(hash, password)
if (hash == nil) || (err != nil) {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_PASSWDMISMATCH, client.nickString, "Password incorrect")
return true
2014-02-24 07:21:39 +01:00
}
2014-02-09 19:07:40 +01:00
2014-02-24 07:21:39 +01:00
client.flags[Operator] = true
client.Send(nil, server.nameString, RPL_YOUREOPER, client.nickString, "You are now an IRC operator")
2016-06-19 13:59:18 +02:00
//TODO(dan): Should this be sent automagically as part of setting the flag/mode?
modech := ModeChanges{&ModeChange{
mode: Operator,
op: Add,
2016-06-19 13:59:18 +02:00
}}
2016-08-12 14:26:45 +02:00
client.Send(nil, server.nameString, "MODE", client.nickString, modech.String())
2016-06-19 13:59:18 +02:00
return false
2014-02-09 19:07:40 +01:00
}
2014-02-10 20:14:34 +01:00
// AWAY [<message>]
func awayHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
var isAway bool
var text string
if len(msg.Params) > 0 {
2016-06-19 13:59:18 +02:00
isAway = true
text = msg.Params[0]
2016-09-12 04:40:09 +02:00
if len(text) > server.limits.AwayLen {
text = text[:server.limits.AwayLen]
}
}
if isAway {
2014-02-17 22:22:35 +01:00
client.flags[Away] = true
} else {
delete(client.flags, Away)
}
client.awayMessage = text
2014-02-12 00:44:58 +01:00
var op ModeOp
2014-02-17 22:22:35 +01:00
if client.flags[Away] {
op = Add
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_NOWAWAY, client.nickString, "You have been marked as being away")
2014-02-12 00:44:58 +01:00
} else {
op = Remove
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_UNAWAY, client.nickString, "You are no longer marked as being away")
2014-02-12 00:44:58 +01:00
}
2016-06-19 13:59:18 +02:00
//TODO(dan): Should this be sent automagically as part of setting the flag/mode?
modech := ModeChanges{&ModeChange{
mode: Away,
op: op,
2016-06-19 13:59:18 +02:00
}}
client.Send(nil, server.nameString, "MODE", client.nickString, client.nickString, modech.String())
2016-09-12 03:56:20 +02:00
// dispatch away-notify
for friend := range client.Friends() {
if client.flags[Away] {
friend.SendFromClient(client, nil, client.nickMaskString, "AWAY", client.awayMessage)
} else {
friend.SendFromClient(client, nil, client.nickMaskString, "AWAY")
}
}
2016-06-19 13:59:18 +02:00
return false
2014-02-12 00:44:58 +01:00
}
2014-02-12 00:58:54 +01:00
// ISON <nick>{ <nick>}
func isonHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var nicks = msg.Params
2014-02-12 00:58:54 +01:00
ison := make([]string, 0)
for _, nick := range nicks {
2016-06-19 13:59:18 +02:00
if iclient := server.clients.Get(Name(nick)); iclient != nil {
2014-03-09 21:45:36 +01:00
ison = append(ison, iclient.Nick().String())
2014-02-12 00:58:54 +01:00
}
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_ISON, client.nickString, strings.Join(nicks, " "))
return false
2014-02-12 00:58:54 +01:00
}
2014-02-12 01:35:32 +01:00
// MOTD [<target>]
func motdHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
//TODO(dan): hook this up when we have multiple servers I guess???
//var target string
//if len(msg.Params) > 0 {
// target = msg.Params[0]
//}
2016-06-19 13:59:18 +02:00
server.MOTD(client)
return false
2014-02-12 01:35:32 +01:00
}
2014-02-12 02:11:59 +01:00
// NOTICE <target>{,<target>} <message>
func noticeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
targets := strings.Split(msg.Params[0], ",")
message := msg.Params[1]
2016-06-19 13:59:18 +02:00
var target Name
for _, targetString := range targets {
target = Name(targetString)
if target.IsChannel() {
channel := server.channels.Get(target)
if channel == nil {
// errors silently ignored with NOTICE as per RFC
continue
}
channel.PrivMsg(client, message)
} else {
user := server.clients.Get(target)
if user == nil {
// errors silently ignored with NOTICE as per RFC
continue
}
user.Send(nil, client.nickMaskString, "NOTICE", user.nickString, message)
2014-02-12 02:11:59 +01:00
}
}
2016-06-19 13:59:18 +02:00
return false
}
2014-02-17 08:29:11 +01:00
// KICK <channel>{,<channel>} <user>{,<user>} [<comment>]
func kickHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
channels := strings.Split(msg.Params[0], ",")
users := strings.Split(msg.Params[1], ",")
if (len(channels) != len(users)) && (len(users) != 1) {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NEEDMOREPARAMS, client.nickString, "KICK", "Not enough parameters")
return false
}
2016-06-19 13:59:18 +02:00
kicks := make(map[string]string)
for index, channel := range channels {
if len(users) == 1 {
kicks[channel] = users[0]
} else {
kicks[channel] = users[index]
}
}
var comment string
if len(msg.Params) > 2 {
comment = msg.Params[2]
}
for chname, nickname := range kicks {
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(chname))
2014-02-17 08:29:11 +01:00
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
2014-02-17 08:29:11 +01:00
continue
}
2016-06-19 13:59:18 +02:00
target := server.clients.Get(Name(nickname))
2014-02-17 08:29:11 +01:00
if target == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHNICK, nickname, "No such nick")
2014-02-17 08:29:11 +01:00
continue
}
// make sure client has privs to kick the given user
2016-06-30 07:43:27 +02:00
//TODO(dan): split this into a separate function that checks if users have privs
// over other users, useful for things like -aoh as well
var hasPrivs bool
for _, mode := range ChannelPrivModes {
if channel.members[client][mode] {
hasPrivs = true
// admins cannot kick other admins
if mode == ChannelAdmin && channel.members[target][ChannelAdmin] {
hasPrivs = false
}
break
} else if channel.members[target][mode] {
break
}
}
if hasPrivs {
if comment == "" {
2016-06-30 07:43:27 +02:00
comment = nickname
}
2016-06-30 07:43:27 +02:00
channel.Kick(client, target, comment)
} else {
2016-06-19 13:59:18 +02:00
client.Send(nil, client.server.nameString, ERR_CHANOPRIVSNEEDED, chname, "You're not a channel operator")
}
2014-02-17 08:29:11 +01:00
}
2016-06-19 13:59:18 +02:00
return false
2014-02-17 08:29:11 +01:00
}
2014-02-17 08:51:27 +01:00
// LIST [<channel>{,<channel>} [<server>]]
func listHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var channels []string
if len(msg.Params) > 0 {
channels = strings.Split(msg.Params[0], ",")
}
2016-06-19 13:59:18 +02:00
var target string
if len(msg.Params) > 1 {
target = msg.Params[1]
}
2014-02-17 08:51:27 +01:00
//TODO(dan): target server when we have multiple servers
//TODO(dan): we should continue just fine if it's this current server though
if target != "" {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
return false
2014-02-17 08:51:27 +01:00
}
if len(channels) == 0 {
2014-02-17 08:51:27 +01:00
for _, channel := range server.channels {
if !client.flags[Operator] && channel.flags[Secret] {
2014-02-17 08:51:27 +01:00
continue
}
client.RplList(channel)
2014-02-17 08:51:27 +01:00
}
} else {
for _, chname := range channels {
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(chname))
if channel == nil || (!client.flags[Operator] && channel.flags[Secret]) {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
2014-02-17 08:51:27 +01:00
continue
}
client.RplList(channel)
2014-02-17 08:51:27 +01:00
}
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_LISTEND, client.nickString, "End of LIST")
return false
}
func (target *Client) RplList(channel *Channel) {
// get the correct number of channel members
var memberCount int
if target.flags[Operator] || channel.members.Has(target) {
memberCount = len(channel.members)
} else {
for member := range channel.members {
if !member.flags[Invisible] {
memberCount += 1
}
}
}
target.Send(nil, target.server.nameString, RPL_LIST, target.nickString, channel.nameString, string(memberCount), channel.topic)
2014-02-17 08:51:27 +01:00
}
// NAMES [<channel>{,<channel>}]
func namesHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var channels []string
if len(msg.Params) > 0 {
channels = strings.Split(msg.Params[0], ",")
}
//var target string
//if len(msg.Params) > 1 {
// target = msg.Params[1]
//}
if len(channels) == 0 {
2014-02-18 06:02:03 +01:00
for _, channel := range server.channels {
channel.Names(client)
}
return false
2014-02-18 06:02:03 +01:00
}
for _, chname := range channels {
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(chname))
2014-02-18 06:02:03 +01:00
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
2014-02-18 06:02:03 +01:00
continue
}
channel.Names(client)
}
2016-06-19 13:59:18 +02:00
return false
2014-02-18 06:02:03 +01:00
}
// VERSION [<server>]
func versionHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var target string
if len(msg.Params) > 0 {
target = msg.Params[0]
}
2016-06-19 13:59:18 +02:00
if (target != "") && (Name(target) != server.name) {
client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
return false
2014-02-25 07:04:11 +01:00
}
client.Send(nil, server.nameString, RPL_VERSION, client.nickString, VER, server.nameString)
client.RplISupport()
2016-06-19 13:59:18 +02:00
return false
2014-02-25 07:04:11 +01:00
}
2014-02-25 16:28:09 +01:00
// INVITE <nickname> <channel>
func inviteHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
nickname := msg.Params[0]
channelName := msg.Params[1]
2014-02-25 16:28:09 +01:00
2016-06-19 13:59:18 +02:00
target := server.clients.Get(Name(nickname))
2014-02-25 16:28:09 +01:00
if target == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_NOSUCHNICK, client.nickString, nickname, "No such nick")
return false
2014-02-25 16:28:09 +01:00
}
2016-06-19 13:59:18 +02:00
channel := server.channels.Get(Name(channelName))
2014-02-25 16:28:09 +01:00
if channel == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_INVITING, client.nickString, target.nickString, channelName)
target.Send(nil, client.nickMaskString, "INVITE", target.nickString, channel.nameString)
return true
2014-02-25 16:28:09 +01:00
}
channel.Invite(target, client)
2016-06-19 13:59:18 +02:00
return false
2014-02-25 16:28:09 +01:00
}
2014-02-25 16:45:40 +01:00
// TIME [<server>]
func timeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
var target string
if len(msg.Params) > 0 {
2016-06-19 13:59:18 +02:00
target = msg.Params[0]
}
2016-06-19 13:59:18 +02:00
if (target != "") && (Name(target) != server.name) {
client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
return false
2014-02-25 18:10:16 +01:00
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_TIME, client.nickString, server.nameString, time.Now().Format(time.RFC1123))
return false
2014-02-25 18:10:16 +01:00
}
// KILL <nickname> <comment>
func killHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
nickname := msg.Params[0]
comment := msg.Params[1]
2016-06-19 13:59:18 +02:00
target := server.clients.Get(Name(nickname))
2014-02-25 18:10:16 +01:00
if target == nil {
2016-06-19 13:59:18 +02:00
client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, nickname, "No such nick")
return false
2014-02-25 18:10:16 +01:00
}
2016-06-19 13:59:18 +02:00
quitMsg := fmt.Sprintf("Killed (%s (%s))", client.nickString, comment)
target.Quit(quitMsg)
target.destroy()
return false
2014-02-25 16:45:40 +01:00
}
// WHOWAS <nickname> [<count> [<server>]]
func whowasHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
2016-06-19 13:59:18 +02:00
nicknames := strings.Split(msg.Params[0], ",")
2016-06-19 13:59:18 +02:00
var count int64
if len(msg.Params) > 1 {
count, _ = strconv.ParseInt(msg.Params[1], 10, 64)
}
//var target string
//if len(msg.Params) > 2 {
// target = msg.Params[2]
//}
for _, nickname := range nicknames {
2016-06-19 13:59:18 +02:00
results := server.whoWas.Find(Name(nickname), count)
if len(results) == 0 {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, ERR_WASNOSUCHNICK, client.nickString, nickname, "There was no such nickname")
} else {
for _, whoWas := range results {
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_WHOWASUSER, client.nickString, whoWas.nickname.String(), whoWas.username.String(), whoWas.hostname.String(), "*", whoWas.realname)
}
}
2016-06-19 13:59:18 +02:00
client.Send(nil, server.nameString, RPL_ENDOFWHOWAS, client.nickString, nickname, "End of WHOWAS")
2014-03-06 21:14:21 +01:00
}
2016-06-19 13:59:18 +02:00
return false
2014-03-06 21:14:21 +01:00
}