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 30KB

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