您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

server.go 33KB

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