You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

server.go 33KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161
  1. // Copyright (c) 2012-2014 Jeremy Latt
  2. // Copyright (c) 2014-2015 Edmund Huber
  3. // Copyright (c) 2016- Daniel Oaks <daniel@danieloaks.net>
  4. // released under the MIT license
  5. package irc
  6. import (
  7. "bufio"
  8. "crypto/tls"
  9. "database/sql"
  10. "encoding/base64"
  11. "fmt"
  12. "log"
  13. "net"
  14. "net/http"
  15. "os"
  16. "os/signal"
  17. "regexp"
  18. "strconv"
  19. "strings"
  20. "syscall"
  21. "time"
  22. "github.com/DanielOaks/girc-go/ircmsg"
  23. "github.com/tidwall/buntdb"
  24. )
  25. // Limits holds the maximum limits for various things such as topic lengths
  26. type Limits struct {
  27. AwayLen int
  28. KickLen int
  29. TopicLen int
  30. }
  31. type Server struct {
  32. accounts map[string]*ClientAccount
  33. channels ChannelNameMap
  34. clients *ClientLookupSet
  35. commands chan Command
  36. ctime time.Time
  37. db *sql.DB
  38. store buntdb.DB
  39. idle chan *Client
  40. limits Limits
  41. motdLines []string
  42. name Name
  43. nameString string // cache for server name string since it's used with almost every reply
  44. newConns chan clientConn
  45. operators map[Name][]byte
  46. password []byte
  47. passwords *PasswordManager
  48. accountRegistration *AccountRegistration
  49. signals chan os.Signal
  50. proxyAllowedFrom []string
  51. whoWas *WhoWasList
  52. isupport *ISupportList
  53. checkIdent bool
  54. }
  55. var (
  56. SERVER_SIGNALS = []os.Signal{
  57. syscall.SIGINT,
  58. syscall.SIGHUP, // eventually we expect to use HUP to reload config
  59. syscall.SIGTERM,
  60. syscall.SIGQUIT,
  61. }
  62. )
  63. type clientConn struct {
  64. Conn net.Conn
  65. IsTLS bool
  66. }
  67. func NewServer(config *Config) *Server {
  68. server := &Server{
  69. accounts: make(map[string]*ClientAccount),
  70. channels: make(ChannelNameMap),
  71. clients: NewClientLookupSet(),
  72. commands: make(chan Command),
  73. ctime: time.Now(),
  74. db: OpenDB(config.Datastore.SQLitePath),
  75. idle: make(chan *Client),
  76. limits: Limits{
  77. AwayLen: config.Limits.AwayLen,
  78. KickLen: config.Limits.KickLen,
  79. TopicLen: config.Limits.TopicLen,
  80. },
  81. name: NewName(config.Server.Name),
  82. nameString: NewName(config.Server.Name).String(),
  83. newConns: make(chan clientConn),
  84. operators: config.Operators(),
  85. signals: make(chan os.Signal, len(SERVER_SIGNALS)),
  86. proxyAllowedFrom: config.Server.ProxyAllowedFrom,
  87. whoWas: NewWhoWasList(config.Limits.WhowasEntries),
  88. checkIdent: config.Server.CheckIdent,
  89. }
  90. // open data store
  91. db, err := buntdb.Open(config.Datastore.Path)
  92. if err != nil {
  93. log.Fatal(fmt.Sprintf("Failed to open datastore: %s", err.Error()))
  94. }
  95. server.store = *db
  96. // load password manager
  97. err = server.store.View(func(tx *buntdb.Tx) error {
  98. saltString, err := tx.Get(keySalt)
  99. if err != nil {
  100. return fmt.Errorf("Could not retrieve salt string: %s", err.Error())
  101. }
  102. salt, err := base64.StdEncoding.DecodeString(saltString)
  103. if err != nil {
  104. return err
  105. }
  106. pwm := NewPasswordManager(salt)
  107. server.passwords = &pwm
  108. return nil
  109. })
  110. if err != nil {
  111. log.Fatal(fmt.Sprintf("Could not load salt: %s", err.Error()))
  112. }
  113. if config.Server.MOTD != "" {
  114. file, err := os.Open(config.Server.MOTD)
  115. if err == nil {
  116. defer file.Close()
  117. reader := bufio.NewReader(file)
  118. for {
  119. line, err := reader.ReadString('\n')
  120. if err != nil {
  121. break
  122. }
  123. line = strings.TrimRight(line, "\r\n")
  124. // "- " is the required prefix for MOTD, we just add it here to make
  125. // bursting it out to clients easier
  126. line = fmt.Sprintf("- %s", line)
  127. server.motdLines = append(server.motdLines, line)
  128. }
  129. }
  130. }
  131. //TODO(dan): Hot damn this is an ugly hack. Fix it properly at some point.
  132. ChannelNameExpr = regexp.MustCompile(fmt.Sprintf(`^[#][\pL\pN\pP\pS]{1,%d}$`, config.Limits.ChannelLen))
  133. NicknameExpr = regexp.MustCompile(fmt.Sprintf("^[\\pL\\pN\\pP\\pS]{1,%d}$", config.Limits.NickLen))
  134. if config.Server.Password != "" {
  135. server.password = config.Server.PasswordBytes()
  136. }
  137. server.loadChannels()
  138. for _, addr := range config.Server.Listen {
  139. server.listen(addr, config.TLSListeners())
  140. }
  141. if config.Server.Wslisten != "" {
  142. server.wslisten(config.Server.Wslisten, config.Server.TLSListeners)
  143. }
  144. // registration
  145. accountReg := NewAccountRegistration(config.Registration.Accounts)
  146. server.accountRegistration = &accountReg
  147. // Attempt to clean up when receiving these signals.
  148. signal.Notify(server.signals, SERVER_SIGNALS...)
  149. // add RPL_ISUPPORT tokens
  150. server.isupport = NewISupportList()
  151. server.isupport.Add("AWAYLEN", strconv.Itoa(server.limits.AwayLen))
  152. server.isupport.Add("CASEMAPPING", "ascii")
  153. 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()}, ","))
  154. server.isupport.Add("CHANNELLEN", strconv.Itoa(config.Limits.ChannelLen))
  155. server.isupport.Add("CHANTYPES", "#")
  156. server.isupport.Add("EXCEPTS", "")
  157. server.isupport.Add("INVEX", "")
  158. server.isupport.Add("KICKLEN", strconv.Itoa(server.limits.KickLen))
  159. // server.isupport.Add("MAXLIST", "") //TODO(dan): Support max list length?
  160. // server.isupport.Add("MODES", "") //TODO(dan): Support max modes?
  161. server.isupport.Add("NETWORK", config.Network.Name)
  162. server.isupport.Add("NICKLEN", strconv.Itoa(config.Limits.NickLen))
  163. server.isupport.Add("PREFIX", "(qaohv)~&@%+")
  164. // server.isupport.Add("STATUSMSG", "@+") //TODO(dan): Support STATUSMSG
  165. // server.isupport.Add("TARGMAX", "") //TODO(dan): Support this
  166. server.isupport.Add("TOPICLEN", strconv.Itoa(server.limits.TopicLen))
  167. // account registration
  168. if server.accountRegistration.Enabled {
  169. // 'none' isn't shown in the REGCALLBACKS vars
  170. var enabledCallbacks []string
  171. for _, name := range server.accountRegistration.EnabledCallbacks {
  172. if name != "*" {
  173. enabledCallbacks = append(enabledCallbacks, name)
  174. }
  175. }
  176. server.isupport.Add("REGCOMMANDS", "CREATE,VERIFY")
  177. server.isupport.Add("REGCALLBACKS", strings.Join(enabledCallbacks, ","))
  178. server.isupport.Add("REGCREDTYPES", "passphrase,certfp")
  179. }
  180. server.isupport.RegenerateCachedReply()
  181. return server
  182. }
  183. func loadChannelList(channel *Channel, list string, maskMode ChannelMode) {
  184. if list == "" {
  185. return
  186. }
  187. channel.lists[maskMode].AddAll(NewNames(strings.Split(list, " ")))
  188. }
  189. func (server *Server) loadChannels() {
  190. rows, err := server.db.Query(`
  191. SELECT name, flags, key, topic, user_limit, ban_list, except_list,
  192. invite_list
  193. FROM channel`)
  194. if err != nil {
  195. log.Fatal("error loading channels: ", err)
  196. }
  197. for rows.Next() {
  198. var name, flags, key, topic string
  199. var userLimit uint64
  200. var banList, exceptList, inviteList string
  201. err = rows.Scan(&name, &flags, &key, &topic, &userLimit, &banList,
  202. &exceptList, &inviteList)
  203. if err != nil {
  204. log.Println("Server.loadChannels:", err)
  205. continue
  206. }
  207. channel := NewChannel(server, NewName(name), false)
  208. for _, flag := range flags {
  209. channel.flags[ChannelMode(flag)] = true
  210. }
  211. channel.key = key
  212. channel.topic = topic
  213. channel.userLimit = userLimit
  214. loadChannelList(channel, banList, BanMask)
  215. loadChannelList(channel, exceptList, ExceptMask)
  216. loadChannelList(channel, inviteList, InviteMask)
  217. }
  218. }
  219. func (server *Server) Shutdown() {
  220. //TODO(dan): Make sure we disallow new nicks
  221. for _, client := range server.clients.byNick {
  222. client.Notice("Server is shutting down")
  223. }
  224. if err := server.db.Close(); err != nil {
  225. Log.error.Println("Server.Shutdown db.Close: error:", err)
  226. }
  227. if err := server.store.Close(); err != nil {
  228. Log.error.Println("Server.Shutdown store.Close: error:", err)
  229. }
  230. }
  231. func (server *Server) Run() {
  232. // defer closing db/store
  233. defer server.db.Close()
  234. defer server.store.Close()
  235. done := false
  236. for !done {
  237. select {
  238. case <-server.signals:
  239. server.Shutdown()
  240. done = true
  241. case conn := <-server.newConns:
  242. NewClient(server, conn.Conn, conn.IsTLS)
  243. /*TODO(dan): LOOK AT THIS MORE CLOSELY
  244. case cmd := <-server.commands:
  245. server.processCommand(cmd)
  246. */
  247. case client := <-server.idle:
  248. client.Idle()
  249. }
  250. }
  251. }
  252. //
  253. // listen goroutine
  254. //
  255. func (s *Server) listen(addr string, tlsMap map[Name]*tls.Config) {
  256. config, listenTLS := tlsMap[NewName(addr)]
  257. listener, err := net.Listen("tcp", addr)
  258. if err != nil {
  259. log.Fatal(s, "listen error: ", err)
  260. }
  261. tlsString := "plaintext"
  262. if listenTLS {
  263. config.ClientAuth = tls.RequestClientCert
  264. listener = tls.NewListener(listener, config)
  265. tlsString = "TLS"
  266. }
  267. Log.info.Printf("%s listening on %s using %s.", s, addr, tlsString)
  268. go func() {
  269. for {
  270. conn, err := listener.Accept()
  271. if err != nil {
  272. Log.error.Printf("%s accept error: %s", s, err)
  273. continue
  274. }
  275. Log.debug.Printf("%s accept: %s", s, conn.RemoteAddr())
  276. newConn := clientConn{
  277. Conn: conn,
  278. IsTLS: listenTLS,
  279. }
  280. s.newConns <- newConn
  281. }
  282. }()
  283. }
  284. //
  285. // websocket listen goroutine
  286. //
  287. func (s *Server) wslisten(addr string, tlsMap map[string]*TLSListenConfig) {
  288. http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  289. if r.Method != "GET" {
  290. Log.error.Printf("%s method not allowed", s)
  291. return
  292. }
  293. // We don't have any subprotocols, so if someone attempts to `new
  294. // WebSocket(server, "subprotocol")` they'll break here, instead of
  295. // getting the default, ambiguous, response from gorilla.
  296. if v, ok := r.Header["Sec-Websocket-Protocol"]; ok {
  297. http.Error(w, fmt.Sprintf("WebSocket subprocotols (e.g. %s) not supported", v), 400)
  298. }
  299. ws, err := upgrader.Upgrade(w, r, nil)
  300. if err != nil {
  301. Log.error.Printf("%s websocket upgrade error: %s", s, err)
  302. return
  303. }
  304. newConn := clientConn{
  305. Conn: WSContainer{ws},
  306. IsTLS: false, //TODO(dan): track TLS or not here properly
  307. }
  308. s.newConns <- newConn
  309. })
  310. go func() {
  311. config, listenTLS := tlsMap[addr]
  312. tlsString := "plaintext"
  313. var err error
  314. if listenTLS {
  315. tlsString = "TLS"
  316. }
  317. Log.info.Printf("%s websocket listening on %s using %s.", s, addr, tlsString)
  318. if listenTLS {
  319. err = http.ListenAndServeTLS(addr, config.Cert, config.Key, nil)
  320. } else {
  321. err = http.ListenAndServe(addr, nil)
  322. }
  323. if err != nil {
  324. Log.error.Printf("%s listenAndServe (%s) error: %s", s, tlsString, err)
  325. }
  326. }()
  327. }
  328. //
  329. // server functionality
  330. //
  331. func (s *Server) tryRegister(c *Client) {
  332. if c.registered || !c.HasNick() || !c.HasUsername() ||
  333. (c.capState == CapNegotiating) {
  334. return
  335. }
  336. c.Register()
  337. // send welcome text
  338. //NOTE(dan): we specifically use the NICK here instead of the nickmask
  339. // see http://modern.ircdocs.horse/#rplwelcome-001 for details on why we avoid using the nickmask
  340. c.Send(nil, s.nameString, RPL_WELCOME, c.nickString, fmt.Sprintf("Welcome to the Internet Relay Network %s", c.nickString))
  341. c.Send(nil, s.nameString, RPL_YOURHOST, c.nickString, fmt.Sprintf("Your host is %s, running version %s", s.nameString, VER))
  342. c.Send(nil, s.nameString, RPL_CREATED, c.nickString, fmt.Sprintf("This server was created %s", s.ctime.Format(time.RFC1123)))
  343. //TODO(dan): Look at adding last optional [<channel modes with a parameter>] parameter
  344. c.Send(nil, s.nameString, RPL_MYINFO, c.nickString, s.nameString, VER, supportedUserModesString, supportedChannelModesString)
  345. c.RplISupport()
  346. s.MOTD(c)
  347. c.Send(nil, c.nickMaskString, RPL_UMODEIS, c.nickString, c.ModeString())
  348. }
  349. func (server *Server) MOTD(client *Client) {
  350. if len(server.motdLines) < 1 {
  351. client.Send(nil, server.nameString, ERR_NOMOTD, client.nickString, "MOTD File is missing")
  352. return
  353. }
  354. client.Send(nil, server.nameString, RPL_MOTDSTART, client.nickString, fmt.Sprintf("- %s Message of the day - ", server.nameString))
  355. for _, line := range server.motdLines {
  356. client.Send(nil, server.nameString, RPL_MOTD, client.nickString, line)
  357. }
  358. client.Send(nil, server.nameString, RPL_ENDOFMOTD, client.nickString, "End of MOTD command")
  359. }
  360. func (s *Server) Id() Name {
  361. return s.name
  362. }
  363. func (s *Server) String() string {
  364. return s.name.String()
  365. }
  366. func (s *Server) Nick() Name {
  367. return s.Id()
  368. }
  369. //
  370. // registration commands
  371. //
  372. // PASS <password>
  373. func passHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  374. if client.registered {
  375. client.Send(nil, server.nameString, ERR_ALREADYREGISTRED, client.nickString, "You may not reregister")
  376. return false
  377. }
  378. // if no password exists, skip checking
  379. if len(server.password) == 0 {
  380. client.authorized = true
  381. return false
  382. }
  383. // check the provided password
  384. password := []byte(msg.Params[0])
  385. if ComparePassword(server.password, password) != nil {
  386. client.Send(nil, server.nameString, ERR_PASSWDMISMATCH, client.nickString, "Password incorrect")
  387. client.Send(nil, server.nameString, "ERROR", "Password incorrect")
  388. return true
  389. }
  390. client.authorized = true
  391. return false
  392. }
  393. // PROXY TCP4/6 SOURCEIP DESTIP SOURCEPORT DESTPORT
  394. // http://www.haproxy.org/download/1.5/doc/proxy-protocol.txt
  395. func proxyHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  396. clientAddress := IPString(client.socket.conn.RemoteAddr()).String()
  397. clientHostname := client.hostname.String()
  398. for _, address := range server.proxyAllowedFrom {
  399. if clientHostname == address || clientAddress == address {
  400. client.hostname = LookupHostname(NewName(msg.Params[1]))
  401. return false
  402. }
  403. }
  404. client.Quit("PROXY command is not usable from your address")
  405. return true
  406. }
  407. // USER <username> * 0 <realname>
  408. func userHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  409. if client.registered {
  410. client.Send(nil, server.nameString, ERR_ALREADYREGISTRED, client.nickString, "You may not reregister")
  411. return false
  412. }
  413. if !client.authorized {
  414. client.Quit("Bad password")
  415. return true
  416. }
  417. if client.username != "" && client.realname != "" {
  418. return false
  419. }
  420. // confirm that username is valid
  421. //
  422. if !Name(msg.Params[0]).IsNickname() {
  423. client.Send(nil, "", "ERROR", "Malformed username")
  424. return true
  425. }
  426. // set user info and log client in
  427. //TODO(dan): Could there be a race condition here with adding/removing the client?
  428. //TODO(dan): we should do something like server.clients.Replace(client) instead
  429. // we do it this way to ONLY replace what hasn't already been set
  430. server.clients.Remove(client)
  431. if !client.HasUsername() {
  432. client.username = Name("~" + msg.Params[0])
  433. client.updateNickMask()
  434. }
  435. if client.realname == "" {
  436. client.realname = msg.Params[3]
  437. }
  438. server.clients.Add(client)
  439. server.tryRegister(client)
  440. return false
  441. }
  442. // QUIT [<reason>]
  443. func quitHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  444. reason := "Quit"
  445. if len(msg.Params) > 0 {
  446. reason += ": " + msg.Params[0]
  447. }
  448. client.Quit(reason)
  449. return true
  450. }
  451. //
  452. // normal commands
  453. //
  454. // PING <server1> [<server2>]
  455. func pingHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  456. client.Send(nil, server.nameString, "PONG", msg.Params...)
  457. return false
  458. }
  459. // PONG <server> [ <server2> ]
  460. func pongHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  461. // client gets touched when they send this command, so we don't need to do anything
  462. return false
  463. }
  464. // JOIN <channel>{,<channel>} [<key>{,<key>}]
  465. // JOIN 0
  466. func joinHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  467. // handle JOIN 0
  468. if msg.Params[0] == "0" {
  469. for channel := range client.channels {
  470. channel.Part(client, client.nickString)
  471. }
  472. return false
  473. }
  474. // handle regular JOINs
  475. channels := strings.Split(msg.Params[0], ",")
  476. var keys []string
  477. if len(msg.Params) > 1 {
  478. keys = strings.Split(msg.Params[1], ",")
  479. }
  480. var name Name
  481. for i, nameString := range channels {
  482. name = Name(nameString)
  483. if !name.IsChannel() {
  484. fmt.Println("ISN'T CHANNEL NAME:", nameString)
  485. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, nameString, "No such channel")
  486. continue
  487. }
  488. channel := server.channels.Get(name)
  489. if channel == nil {
  490. channel = NewChannel(server, name, true)
  491. }
  492. var key string
  493. if len(keys) > i {
  494. key = keys[i]
  495. }
  496. channel.Join(client, key)
  497. }
  498. return false
  499. }
  500. // PART <channel>{,<channel>} [<reason>]
  501. func partHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  502. channels := strings.Split(msg.Params[0], ",")
  503. 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
  504. if len(msg.Params) > 1 {
  505. reason = msg.Params[1]
  506. }
  507. for _, chname := range channels {
  508. channel := server.channels.Get(Name(chname))
  509. if channel == nil {
  510. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
  511. continue
  512. }
  513. channel.Part(client, reason)
  514. }
  515. return false
  516. }
  517. // TOPIC <channel> [<topic>]
  518. func topicHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  519. channel := server.channels.Get(Name(msg.Params[0]))
  520. if channel == nil {
  521. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, msg.Params[0], "No such channel")
  522. return false
  523. }
  524. if len(msg.Params) > 1 {
  525. channel.SetTopic(client, msg.Params[1])
  526. } else {
  527. channel.GetTopic(client)
  528. }
  529. return false
  530. }
  531. // PRIVMSG <target>{,<target>} <message>
  532. func privmsgHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  533. targets := strings.Split(msg.Params[0], ",")
  534. message := msg.Params[1]
  535. var target Name
  536. for _, targetString := range targets {
  537. target = Name(targetString)
  538. if target.IsChannel() {
  539. channel := server.channels.Get(target)
  540. if channel == nil {
  541. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, targetString, "No such channel")
  542. continue
  543. }
  544. channel.PrivMsg(client, message)
  545. } else {
  546. user := server.clients.Get(target)
  547. if user == nil {
  548. client.Send(nil, server.nameString, ERR_NOSUCHNICK, targetString, "No such nick")
  549. continue
  550. }
  551. user.Send(nil, client.nickMaskString, "PRIVMSG", user.nickString, message)
  552. if user.flags[Away] {
  553. //TODO(dan): possibly implement cooldown of away notifications to users
  554. client.Send(nil, server.nameString, RPL_AWAY, user.nickString, user.awayMessage)
  555. }
  556. }
  557. }
  558. return false
  559. }
  560. func (client *Client) WhoisChannelsNames(target *Client) []string {
  561. isMultiPrefix := target.capabilities[MultiPrefix]
  562. var chstrs []string
  563. index := 0
  564. for channel := range client.channels {
  565. // channel is secret and the target can't see it
  566. if !target.flags[Operator] && channel.flags[Secret] && !channel.members.Has(target) {
  567. continue
  568. }
  569. chstrs = append(chstrs, channel.members[client].Prefixes(isMultiPrefix)+channel.name.String())
  570. index++
  571. }
  572. return chstrs
  573. }
  574. // WHOIS [ <target> ] <mask> *( "," <mask> )
  575. func whoisHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  576. var masksString string
  577. //var target string
  578. if len(msg.Params) > 1 {
  579. //target = msg.Params[0]
  580. masksString = msg.Params[1]
  581. } else {
  582. masksString = msg.Params[0]
  583. }
  584. if client.flags[Operator] {
  585. masks := strings.Split(masksString, ",")
  586. for _, mask := range masks {
  587. matches := server.clients.FindAll(Name(mask))
  588. if len(matches) == 0 {
  589. client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, mask, "No such nick")
  590. continue
  591. }
  592. for mclient := range matches {
  593. client.getWhoisOf(mclient)
  594. }
  595. }
  596. } else {
  597. // specifically treat this as a single lookup rather than splitting as we do above
  598. // this is by design
  599. mclient := server.clients.Get(Name(masksString))
  600. if mclient == nil {
  601. client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, masksString, "No such nick")
  602. // fall through, ENDOFWHOIS is always sent
  603. } else {
  604. client.getWhoisOf(mclient)
  605. }
  606. }
  607. client.Send(nil, server.nameString, RPL_ENDOFWHOIS, client.nickString, masksString, "End of /WHOIS list")
  608. return false
  609. }
  610. func (client *Client) getWhoisOf(target *Client) {
  611. client.Send(nil, client.server.nameString, RPL_WHOISUSER, client.nickString, target.nickString, target.username.String(), target.hostname.String(), "*", target.realname)
  612. //TODO(dan): ...one channel per reply? really?
  613. for _, line := range client.WhoisChannelsNames(target) {
  614. client.Send(nil, client.server.nameString, RPL_WHOISCHANNELS, client.nickString, target.nickString, line)
  615. }
  616. if target.flags[Operator] {
  617. client.Send(nil, client.server.nameString, RPL_WHOISOPERATOR, client.nickString, target.nickString, "is an IRC operator")
  618. }
  619. if target.certfp != "" && (client.flags[Operator] || client == target) {
  620. client.Send(nil, client.server.nameString, RPL_WHOISCERTFP, client.nickString, target.nickString, fmt.Sprintf("has client certificate fingerprint %s", target.certfp))
  621. }
  622. 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")
  623. }
  624. // <channel> <user> <host> <server> <nick> ( "H" / "G" ) ["*"] [ ( "@" / "+" ) ]
  625. // :<hopcount> <real name>
  626. func (target *Client) RplWhoReply(channel *Channel, client *Client) {
  627. channelName := "*"
  628. flags := ""
  629. if client.flags[Away] {
  630. flags = "G"
  631. } else {
  632. flags = "H"
  633. }
  634. if client.flags[Operator] {
  635. flags += "*"
  636. }
  637. if channel != nil {
  638. flags += channel.members[client].Prefixes(target.capabilities[MultiPrefix])
  639. channelName = channel.name.String()
  640. }
  641. 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)
  642. }
  643. func whoChannel(client *Client, channel *Channel, friends ClientSet) {
  644. for member := range channel.members {
  645. if !client.flags[Invisible] || friends[client] {
  646. client.RplWhoReply(channel, member)
  647. }
  648. }
  649. }
  650. // WHO [ <mask> [ "o" ] ]
  651. func whoHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  652. friends := client.Friends()
  653. var mask Name
  654. if len(msg.Params) > 0 {
  655. mask = Name(msg.Params[0])
  656. }
  657. //TODO(dan): is this used and would I put this param in the Modern doc?
  658. // if not, can we remove it?
  659. //var operatorOnly bool
  660. //if len(msg.Params) > 1 && msg.Params[1] == "o" {
  661. // operatorOnly = true
  662. //}
  663. if mask == "" {
  664. for _, channel := range server.channels {
  665. whoChannel(client, channel, friends)
  666. }
  667. } else if mask.IsChannel() {
  668. // TODO implement wildcard matching
  669. //TODO(dan): ^ only for opers
  670. channel := server.channels.Get(mask)
  671. if channel != nil {
  672. whoChannel(client, channel, friends)
  673. }
  674. } else {
  675. for mclient := range server.clients.FindAll(mask) {
  676. client.RplWhoReply(nil, mclient)
  677. }
  678. }
  679. client.Send(nil, server.nameString, RPL_ENDOFWHO, client.nickString, mask.String(), "End of WHO list")
  680. return false
  681. }
  682. // OPER <name> <password>
  683. func operHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  684. name := NewName(msg.Params[0])
  685. hash := server.operators[name]
  686. password := []byte(msg.Params[1])
  687. err := ComparePassword(hash, password)
  688. if (hash == nil) || (err != nil) {
  689. client.Send(nil, server.nameString, ERR_PASSWDMISMATCH, client.nickString, "Password incorrect")
  690. return true
  691. }
  692. client.flags[Operator] = true
  693. client.Send(nil, server.nameString, RPL_YOUREOPER, client.nickString, "You are now an IRC operator")
  694. //TODO(dan): Should this be sent automagically as part of setting the flag/mode?
  695. modech := ModeChanges{&ModeChange{
  696. mode: Operator,
  697. op: Add,
  698. }}
  699. client.Send(nil, server.nameString, "MODE", client.nickString, modech.String())
  700. return false
  701. }
  702. // AWAY [<message>]
  703. func awayHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  704. var isAway bool
  705. var text string
  706. if len(msg.Params) > 0 {
  707. isAway = true
  708. text = msg.Params[0]
  709. if len(text) > server.limits.AwayLen {
  710. text = text[:server.limits.AwayLen]
  711. }
  712. }
  713. if isAway {
  714. client.flags[Away] = true
  715. } else {
  716. delete(client.flags, Away)
  717. }
  718. client.awayMessage = text
  719. var op ModeOp
  720. if client.flags[Away] {
  721. op = Add
  722. client.Send(nil, server.nameString, RPL_NOWAWAY, client.nickString, "You have been marked as being away")
  723. } else {
  724. op = Remove
  725. client.Send(nil, server.nameString, RPL_UNAWAY, client.nickString, "You are no longer marked as being away")
  726. }
  727. //TODO(dan): Should this be sent automagically as part of setting the flag/mode?
  728. modech := ModeChanges{&ModeChange{
  729. mode: Away,
  730. op: op,
  731. }}
  732. client.Send(nil, server.nameString, "MODE", client.nickString, client.nickString, modech.String())
  733. // dispatch away-notify
  734. for friend := range client.Friends() {
  735. if client.flags[Away] {
  736. friend.SendFromClient(client, nil, client.nickMaskString, "AWAY", client.awayMessage)
  737. } else {
  738. friend.SendFromClient(client, nil, client.nickMaskString, "AWAY")
  739. }
  740. }
  741. return false
  742. }
  743. // ISON <nick>{ <nick>}
  744. func isonHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  745. var nicks = msg.Params
  746. ison := make([]string, 0)
  747. for _, nick := range nicks {
  748. if iclient := server.clients.Get(Name(nick)); iclient != nil {
  749. ison = append(ison, iclient.Nick().String())
  750. }
  751. }
  752. client.Send(nil, server.nameString, RPL_ISON, client.nickString, strings.Join(nicks, " "))
  753. return false
  754. }
  755. // MOTD [<target>]
  756. func motdHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  757. //TODO(dan): hook this up when we have multiple servers I guess???
  758. //var target string
  759. //if len(msg.Params) > 0 {
  760. // target = msg.Params[0]
  761. //}
  762. server.MOTD(client)
  763. return false
  764. }
  765. // NOTICE <target>{,<target>} <message>
  766. func noticeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  767. targets := strings.Split(msg.Params[0], ",")
  768. message := msg.Params[1]
  769. var target Name
  770. for _, targetString := range targets {
  771. target = Name(targetString)
  772. if target.IsChannel() {
  773. channel := server.channels.Get(target)
  774. if channel == nil {
  775. // errors silently ignored with NOTICE as per RFC
  776. continue
  777. }
  778. channel.PrivMsg(client, message)
  779. } else {
  780. user := server.clients.Get(target)
  781. if user == nil {
  782. // errors silently ignored with NOTICE as per RFC
  783. continue
  784. }
  785. user.Send(nil, client.nickMaskString, "NOTICE", user.nickString, message)
  786. }
  787. }
  788. return false
  789. }
  790. // KICK <channel>{,<channel>} <user>{,<user>} [<comment>]
  791. func kickHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  792. channels := strings.Split(msg.Params[0], ",")
  793. users := strings.Split(msg.Params[1], ",")
  794. if (len(channels) != len(users)) && (len(users) != 1) {
  795. client.Send(nil, server.nameString, ERR_NEEDMOREPARAMS, client.nickString, "KICK", "Not enough parameters")
  796. return false
  797. }
  798. kicks := make(map[string]string)
  799. for index, channel := range channels {
  800. if len(users) == 1 {
  801. kicks[channel] = users[0]
  802. } else {
  803. kicks[channel] = users[index]
  804. }
  805. }
  806. var comment string
  807. if len(msg.Params) > 2 {
  808. comment = msg.Params[2]
  809. }
  810. for chname, nickname := range kicks {
  811. channel := server.channels.Get(Name(chname))
  812. if channel == nil {
  813. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
  814. continue
  815. }
  816. target := server.clients.Get(Name(nickname))
  817. if target == nil {
  818. client.Send(nil, server.nameString, ERR_NOSUCHNICK, nickname, "No such nick")
  819. continue
  820. }
  821. // make sure client has privs to kick the given user
  822. //TODO(dan): split this into a separate function that checks if users have privs
  823. // over other users, useful for things like -aoh as well
  824. var hasPrivs bool
  825. for _, mode := range ChannelPrivModes {
  826. if channel.members[client][mode] {
  827. hasPrivs = true
  828. // admins cannot kick other admins
  829. if mode == ChannelAdmin && channel.members[target][ChannelAdmin] {
  830. hasPrivs = false
  831. }
  832. break
  833. } else if channel.members[target][mode] {
  834. break
  835. }
  836. }
  837. if hasPrivs {
  838. if comment == "" {
  839. comment = nickname
  840. }
  841. channel.Kick(client, target, comment)
  842. } else {
  843. client.Send(nil, client.server.nameString, ERR_CHANOPRIVSNEEDED, chname, "You're not a channel operator")
  844. }
  845. }
  846. return false
  847. }
  848. // LIST [<channel>{,<channel>} [<server>]]
  849. func listHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  850. var channels []string
  851. if len(msg.Params) > 0 {
  852. channels = strings.Split(msg.Params[0], ",")
  853. }
  854. var target string
  855. if len(msg.Params) > 1 {
  856. target = msg.Params[1]
  857. }
  858. //TODO(dan): target server when we have multiple servers
  859. //TODO(dan): we should continue just fine if it's this current server though
  860. if target != "" {
  861. client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
  862. return false
  863. }
  864. if len(channels) == 0 {
  865. for _, channel := range server.channels {
  866. if !client.flags[Operator] && channel.flags[Secret] {
  867. continue
  868. }
  869. client.RplList(channel)
  870. }
  871. } else {
  872. for _, chname := range channels {
  873. channel := server.channels.Get(Name(chname))
  874. if channel == nil || (!client.flags[Operator] && channel.flags[Secret]) {
  875. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
  876. continue
  877. }
  878. client.RplList(channel)
  879. }
  880. }
  881. client.Send(nil, server.nameString, RPL_LISTEND, client.nickString, "End of LIST")
  882. return false
  883. }
  884. func (target *Client) RplList(channel *Channel) {
  885. // get the correct number of channel members
  886. var memberCount int
  887. if target.flags[Operator] || channel.members.Has(target) {
  888. memberCount = len(channel.members)
  889. } else {
  890. for member := range channel.members {
  891. if !member.flags[Invisible] {
  892. memberCount += 1
  893. }
  894. }
  895. }
  896. target.Send(nil, target.server.nameString, RPL_LIST, target.nickString, channel.nameString, string(memberCount), channel.topic)
  897. }
  898. // NAMES [<channel>{,<channel>}]
  899. func namesHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  900. var channels []string
  901. if len(msg.Params) > 0 {
  902. channels = strings.Split(msg.Params[0], ",")
  903. }
  904. //var target string
  905. //if len(msg.Params) > 1 {
  906. // target = msg.Params[1]
  907. //}
  908. if len(channels) == 0 {
  909. for _, channel := range server.channels {
  910. channel.Names(client)
  911. }
  912. return false
  913. }
  914. for _, chname := range channels {
  915. channel := server.channels.Get(Name(chname))
  916. if channel == nil {
  917. client.Send(nil, server.nameString, ERR_NOSUCHCHANNEL, client.nickString, chname, "No such channel")
  918. continue
  919. }
  920. channel.Names(client)
  921. }
  922. return false
  923. }
  924. // VERSION [<server>]
  925. func versionHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  926. var target string
  927. if len(msg.Params) > 0 {
  928. target = msg.Params[0]
  929. }
  930. if (target != "") && (Name(target) != server.name) {
  931. client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
  932. return false
  933. }
  934. client.Send(nil, server.nameString, RPL_VERSION, client.nickString, VER, server.nameString)
  935. client.RplISupport()
  936. return false
  937. }
  938. // INVITE <nickname> <channel>
  939. func inviteHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  940. nickname := msg.Params[0]
  941. channelName := msg.Params[1]
  942. target := server.clients.Get(Name(nickname))
  943. if target == nil {
  944. client.Send(nil, server.nameString, ERR_NOSUCHNICK, client.nickString, nickname, "No such nick")
  945. return false
  946. }
  947. channel := server.channels.Get(Name(channelName))
  948. if channel == nil {
  949. client.Send(nil, server.nameString, RPL_INVITING, client.nickString, target.nickString, channelName)
  950. target.Send(nil, client.nickMaskString, "INVITE", target.nickString, channel.nameString)
  951. return true
  952. }
  953. channel.Invite(target, client)
  954. return false
  955. }
  956. // TIME [<server>]
  957. func timeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  958. var target string
  959. if len(msg.Params) > 0 {
  960. target = msg.Params[0]
  961. }
  962. if (target != "") && (Name(target) != server.name) {
  963. client.Send(nil, server.nameString, ERR_NOSUCHSERVER, client.nickString, target, "No such server")
  964. return false
  965. }
  966. client.Send(nil, server.nameString, RPL_TIME, client.nickString, server.nameString, time.Now().Format(time.RFC1123))
  967. return false
  968. }
  969. // KILL <nickname> <comment>
  970. func killHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  971. nickname := msg.Params[0]
  972. comment := "<no reason supplied>"
  973. if len(msg.Params) > 1 {
  974. comment = msg.Params[1]
  975. }
  976. target := server.clients.Get(Name(nickname))
  977. if target == nil {
  978. client.Send(nil, client.server.nameString, ERR_NOSUCHNICK, nickname, "No such nick")
  979. return false
  980. }
  981. quitMsg := fmt.Sprintf("Killed (%s (%s))", client.nickString, comment)
  982. target.Quit(quitMsg)
  983. target.destroy()
  984. return false
  985. }
  986. // WHOWAS <nickname> [<count> [<server>]]
  987. func whowasHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  988. nicknames := strings.Split(msg.Params[0], ",")
  989. var count int64
  990. if len(msg.Params) > 1 {
  991. count, _ = strconv.ParseInt(msg.Params[1], 10, 64)
  992. }
  993. //var target string
  994. //if len(msg.Params) > 2 {
  995. // target = msg.Params[2]
  996. //}
  997. for _, nickname := range nicknames {
  998. results := server.whoWas.Find(Name(nickname), count)
  999. if len(results) == 0 {
  1000. client.Send(nil, server.nameString, ERR_WASNOSUCHNICK, client.nickString, nickname, "There was no such nickname")
  1001. } else {
  1002. for _, whoWas := range results {
  1003. client.Send(nil, server.nameString, RPL_WHOWASUSER, client.nickString, whoWas.nickname.String(), whoWas.username.String(), whoWas.hostname.String(), "*", whoWas.realname)
  1004. }
  1005. }
  1006. client.Send(nil, server.nameString, RPL_ENDOFWHOWAS, client.nickString, nickname, "End of WHOWAS")
  1007. }
  1008. return false
  1009. }