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

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048
  1. // Copyright (c) 2012-2014 Jeremy Latt
  2. // Copyright (c) 2014-2015 Edmund Huber
  3. // Copyright (c) 2016-2017 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. "errors"
  11. "fmt"
  12. "log"
  13. "math/rand"
  14. "net"
  15. "net/http"
  16. "os"
  17. "os/signal"
  18. "strconv"
  19. "strings"
  20. "sync"
  21. "syscall"
  22. "time"
  23. "github.com/DanielOaks/girc-go/ircfmt"
  24. "github.com/DanielOaks/girc-go/ircmsg"
  25. "github.com/DanielOaks/oragono/irc/logger"
  26. "github.com/DanielOaks/oragono/irc/sno"
  27. "github.com/tidwall/buntdb"
  28. )
  29. var (
  30. // cached because this may be used lots
  31. tooManyClientsMsg = ircmsg.MakeMessage(nil, "", "ERROR", "Too many clients from your network")
  32. tooManyClientsBytes, _ = tooManyClientsMsg.Line()
  33. bannedFromServerMsg = ircmsg.MakeMessage(nil, "", "ERROR", "You are banned from this server (%s)")
  34. bannedFromServerBytes, _ = bannedFromServerMsg.Line()
  35. errDbOutOfDate = errors.New("Database schema is old")
  36. )
  37. // Limits holds the maximum limits for various things such as topic lengths.
  38. type Limits struct {
  39. AwayLen int
  40. ChannelLen int
  41. KickLen int
  42. MonitorEntries int
  43. NickLen int
  44. TopicLen int
  45. ChanListModes int
  46. LineLen LineLenLimits
  47. }
  48. // LineLenLimits holds the maximum limits for IRC lines.
  49. type LineLenLimits struct {
  50. Tags int
  51. Rest int
  52. }
  53. // ListenerInterface represents an interface for a listener.
  54. type ListenerInterface struct {
  55. Listener net.Listener
  56. Events chan ListenerEvent
  57. }
  58. const (
  59. // DestroyListener instructs the listener to destroy itself.
  60. DestroyListener ListenerEventType = iota
  61. // UpdateListener instructs the listener to update itself (grab new certs, etc).
  62. UpdateListener = iota
  63. )
  64. // ListenerEventType is the type of event this is.
  65. type ListenerEventType int
  66. // ListenerEvent is an event that's passed to the listener.
  67. type ListenerEvent struct {
  68. Type ListenerEventType
  69. NewConfig *tls.Config
  70. }
  71. // Server is the main Oragono server.
  72. type Server struct {
  73. accountAuthenticationEnabled bool
  74. accountRegistration *AccountRegistration
  75. accounts map[string]*ClientAccount
  76. channelRegistrationEnabled bool
  77. channels ChannelNameMap
  78. channelJoinPartMutex sync.Mutex // used when joining/parting channels to prevent stomping over each others' access and all
  79. checkIdent bool
  80. clients *ClientLookupSet
  81. commands chan Command
  82. configFilename string
  83. connectionLimits *ConnectionLimits
  84. connectionLimitsMutex sync.Mutex // used when affecting the connection limiter, to make sure rehashing doesn't make things go out-of-whack
  85. connectionThrottle *ConnectionThrottle
  86. connectionThrottleMutex sync.Mutex // used when affecting the connection limiter, to make sure rehashing doesn't make things go out-of-whack
  87. ctime time.Time
  88. currentOpers map[*Client]bool
  89. dlines *DLineManager
  90. isupport *ISupportList
  91. klines *KLineManager
  92. limits Limits
  93. listenerEventActMutex sync.Mutex
  94. listeners map[string]ListenerInterface
  95. listenerUpdateMutex sync.Mutex
  96. logger *logger.Manager
  97. MaxSendQBytes uint64
  98. monitoring map[string][]*Client
  99. motdLines []string
  100. name string
  101. nameCasefolded string
  102. networkName string
  103. newConns chan clientConn
  104. operators map[string]Oper
  105. operclasses map[string]OperClass
  106. password []byte
  107. passwords *PasswordManager
  108. registeredChannels map[string]*RegisteredChannel
  109. registeredChannelsMutex sync.RWMutex
  110. rehashMutex sync.Mutex
  111. rehashSignal chan os.Signal
  112. restAPI *RestAPIConfig
  113. signals chan os.Signal
  114. snomasks *SnoManager
  115. store *buntdb.DB
  116. stsEnabled bool
  117. whoWas *WhoWasList
  118. }
  119. var (
  120. // ServerExitSignals are the signals the server will exit on.
  121. ServerExitSignals = []os.Signal{
  122. syscall.SIGINT,
  123. syscall.SIGTERM,
  124. syscall.SIGQUIT,
  125. }
  126. )
  127. type clientConn struct {
  128. Conn net.Conn
  129. IsTLS bool
  130. }
  131. // NewServer returns a new Oragono server.
  132. func NewServer(configFilename string, config *Config, logger *logger.Manager) (*Server, error) {
  133. casefoldedName, err := Casefold(config.Server.Name)
  134. if err != nil {
  135. return nil, fmt.Errorf("Server name isn't valid [%s]: %s", config.Server.Name, err.Error())
  136. }
  137. // startup check that we have HELP entries for every command
  138. for name := range Commands {
  139. _, exists := Help[strings.ToLower(name)]
  140. if !exists {
  141. return nil, fmt.Errorf("Help entry does not exist for command %s", name)
  142. }
  143. }
  144. // generate help indexes
  145. HelpIndex = GenerateHelpIndex(false)
  146. HelpIndexOpers = GenerateHelpIndex(true)
  147. if config.Accounts.AuthenticationEnabled {
  148. SupportedCapabilities[SASL] = true
  149. }
  150. if config.Server.STS.Enabled {
  151. SupportedCapabilities[STS] = true
  152. CapValues[STS] = config.Server.STS.Value()
  153. }
  154. if config.Limits.LineLen.Tags > 512 || config.Limits.LineLen.Rest > 512 {
  155. SupportedCapabilities[MaxLine] = true
  156. CapValues[MaxLine] = fmt.Sprintf("%d,%d", config.Limits.LineLen.Tags, config.Limits.LineLen.Rest)
  157. }
  158. operClasses, err := config.OperatorClasses()
  159. if err != nil {
  160. return nil, fmt.Errorf("Error loading oper classes: %s", err.Error())
  161. }
  162. opers, err := config.Operators(operClasses)
  163. if err != nil {
  164. return nil, fmt.Errorf("Error loading operators: %s", err.Error())
  165. }
  166. connectionLimits, err := NewConnectionLimits(config.Server.ConnectionLimits)
  167. if err != nil {
  168. return nil, fmt.Errorf("Error loading connection limits: %s", err.Error())
  169. }
  170. connectionThrottle, err := NewConnectionThrottle(config.Server.ConnectionThrottle)
  171. if err != nil {
  172. return nil, fmt.Errorf("Error loading connection throttler: %s", err.Error())
  173. }
  174. server := &Server{
  175. accountAuthenticationEnabled: config.Accounts.AuthenticationEnabled,
  176. accounts: make(map[string]*ClientAccount),
  177. channelRegistrationEnabled: config.Channels.Registration.Enabled,
  178. channels: *NewChannelNameMap(),
  179. checkIdent: config.Server.CheckIdent,
  180. clients: NewClientLookupSet(),
  181. commands: make(chan Command),
  182. configFilename: configFilename,
  183. connectionLimits: connectionLimits,
  184. connectionThrottle: connectionThrottle,
  185. ctime: time.Now(),
  186. currentOpers: make(map[*Client]bool),
  187. limits: Limits{
  188. AwayLen: int(config.Limits.AwayLen),
  189. ChannelLen: int(config.Limits.ChannelLen),
  190. KickLen: int(config.Limits.KickLen),
  191. MonitorEntries: int(config.Limits.MonitorEntries),
  192. NickLen: int(config.Limits.NickLen),
  193. TopicLen: int(config.Limits.TopicLen),
  194. ChanListModes: int(config.Limits.ChanListModes),
  195. LineLen: LineLenLimits{
  196. Tags: config.Limits.LineLen.Tags,
  197. Rest: config.Limits.LineLen.Rest,
  198. },
  199. },
  200. listeners: make(map[string]ListenerInterface),
  201. logger: logger,
  202. MaxSendQBytes: config.Server.MaxSendQBytes,
  203. monitoring: make(map[string][]*Client),
  204. name: config.Server.Name,
  205. nameCasefolded: casefoldedName,
  206. networkName: config.Network.Name,
  207. newConns: make(chan clientConn),
  208. operators: opers,
  209. operclasses: *operClasses,
  210. registeredChannels: make(map[string]*RegisteredChannel),
  211. rehashSignal: make(chan os.Signal, 1),
  212. restAPI: &config.Server.RestAPI,
  213. signals: make(chan os.Signal, len(ServerExitSignals)),
  214. snomasks: NewSnoManager(),
  215. stsEnabled: config.Server.STS.Enabled,
  216. whoWas: NewWhoWasList(config.Limits.WhowasEntries),
  217. }
  218. // open data store
  219. server.logger.Debug("startup", "Opening datastore")
  220. db, err := buntdb.Open(config.Datastore.Path)
  221. if err != nil {
  222. return nil, fmt.Errorf("Failed to open datastore: %s", err.Error())
  223. }
  224. server.store = db
  225. // check db version
  226. err = server.store.View(func(tx *buntdb.Tx) error {
  227. version, _ := tx.Get(keySchemaVersion)
  228. if version != latestDbSchema {
  229. logger.Error("startup", "server", fmt.Sprintf("Database must be updated. Expected schema v%s, got v%s.", latestDbSchema, version))
  230. return errDbOutOfDate
  231. }
  232. return nil
  233. })
  234. if err != nil {
  235. // close the db
  236. db.Close()
  237. return nil, errDbOutOfDate
  238. }
  239. // load *lines
  240. server.logger.Debug("startup", "Loading D/Klines")
  241. server.loadDLines()
  242. server.loadKLines()
  243. // load password manager
  244. server.logger.Debug("startup", "Loading passwords")
  245. err = server.store.View(func(tx *buntdb.Tx) error {
  246. saltString, err := tx.Get(keySalt)
  247. if err != nil {
  248. return fmt.Errorf("Could not retrieve salt string: %s", err.Error())
  249. }
  250. salt, err := base64.StdEncoding.DecodeString(saltString)
  251. if err != nil {
  252. return err
  253. }
  254. pwm := NewPasswordManager(salt)
  255. server.passwords = &pwm
  256. return nil
  257. })
  258. if err != nil {
  259. return nil, fmt.Errorf("Could not load salt: %s", err.Error())
  260. }
  261. server.logger.Debug("startup", "Loading MOTD")
  262. if config.Server.MOTD != "" {
  263. file, err := os.Open(config.Server.MOTD)
  264. if err == nil {
  265. defer file.Close()
  266. reader := bufio.NewReader(file)
  267. for {
  268. line, err := reader.ReadString('\n')
  269. if err != nil {
  270. break
  271. }
  272. line = strings.TrimRight(line, "\r\n")
  273. // "- " is the required prefix for MOTD, we just add it here to make
  274. // bursting it out to clients easier
  275. line = fmt.Sprintf("- %s", line)
  276. server.motdLines = append(server.motdLines, line)
  277. }
  278. }
  279. }
  280. if config.Server.Password != "" {
  281. server.password = config.Server.PasswordBytes()
  282. }
  283. for _, addr := range config.Server.Listen {
  284. server.createListener(addr, config.TLSListeners())
  285. }
  286. if config.Server.Wslisten != "" {
  287. server.wslisten(config.Server.Wslisten, config.Server.TLSListeners)
  288. }
  289. // registration
  290. accountReg := NewAccountRegistration(config.Accounts.Registration)
  291. server.accountRegistration = &accountReg
  292. // Attempt to clean up when receiving these signals.
  293. signal.Notify(server.signals, ServerExitSignals...)
  294. signal.Notify(server.rehashSignal, syscall.SIGHUP)
  295. server.setISupport()
  296. // start API if enabled
  297. if server.restAPI.Enabled {
  298. logger.Info("startup", "server", fmt.Sprintf("%s rest API started on %s.", server.name, server.restAPI.Listen))
  299. server.startRestAPI()
  300. }
  301. return server, nil
  302. }
  303. // setISupport sets up our RPL_ISUPPORT reply.
  304. func (server *Server) setISupport() {
  305. maxTargetsString := strconv.Itoa(maxTargets)
  306. // add RPL_ISUPPORT tokens
  307. server.isupport = NewISupportList()
  308. server.isupport.Add("AWAYLEN", strconv.Itoa(server.limits.AwayLen))
  309. server.isupport.Add("CASEMAPPING", casemappingName)
  310. server.isupport.Add("CHANMODES", strings.Join([]string{Modes{BanMask, ExceptMask, InviteMask}.String(), "", Modes{UserLimit, Key}.String(), Modes{InviteOnly, Moderated, NoOutside, OpOnlyTopic, ChanRoleplaying, Secret}.String()}, ","))
  311. server.isupport.Add("CHANNELLEN", strconv.Itoa(server.limits.ChannelLen))
  312. server.isupport.Add("CHANTYPES", "#")
  313. server.isupport.Add("EXCEPTS", "")
  314. server.isupport.Add("INVEX", "")
  315. server.isupport.Add("KICKLEN", strconv.Itoa(server.limits.KickLen))
  316. server.isupport.Add("MAXLIST", fmt.Sprintf("beI:%s", strconv.Itoa(server.limits.ChanListModes)))
  317. server.isupport.Add("MAXTARGETS", maxTargetsString)
  318. server.isupport.Add("MODES", "")
  319. server.isupport.Add("MONITOR", strconv.Itoa(server.limits.MonitorEntries))
  320. server.isupport.Add("NETWORK", server.networkName)
  321. server.isupport.Add("NICKLEN", strconv.Itoa(server.limits.NickLen))
  322. server.isupport.Add("PREFIX", "(qaohv)~&@%+")
  323. server.isupport.Add("RPCHAN", "E")
  324. server.isupport.Add("RPUSER", "E")
  325. server.isupport.Add("STATUSMSG", "~&@%+")
  326. server.isupport.Add("TARGMAX", fmt.Sprintf("NAMES:1,LIST:1,KICK:1,WHOIS:1,USERHOST:10,PRIVMSG:%s,TAGMSG:%s,NOTICE:%s,MONITOR:", maxTargetsString, maxTargetsString, maxTargetsString))
  327. server.isupport.Add("TOPICLEN", strconv.Itoa(server.limits.TopicLen))
  328. // account registration
  329. if server.accountRegistration.Enabled {
  330. // 'none' isn't shown in the REGCALLBACKS vars
  331. var enabledCallbacks []string
  332. for _, name := range server.accountRegistration.EnabledCallbacks {
  333. if name != "*" {
  334. enabledCallbacks = append(enabledCallbacks, name)
  335. }
  336. }
  337. server.isupport.Add("REGCOMMANDS", "CREATE,VERIFY")
  338. server.isupport.Add("REGCALLBACKS", strings.Join(enabledCallbacks, ","))
  339. server.isupport.Add("REGCREDTYPES", "passphrase,certfp")
  340. }
  341. server.isupport.RegenerateCachedReply()
  342. }
  343. func loadChannelList(channel *Channel, list string, maskMode Mode) {
  344. if list == "" {
  345. return
  346. }
  347. channel.lists[maskMode].AddAll(strings.Split(list, " "))
  348. }
  349. // Shutdown shuts down the server.
  350. func (server *Server) Shutdown() {
  351. //TODO(dan): Make sure we disallow new nicks
  352. server.clients.ByNickMutex.RLock()
  353. for _, client := range server.clients.ByNick {
  354. client.Notice("Server is shutting down")
  355. }
  356. server.clients.ByNickMutex.RUnlock()
  357. if err := server.store.Close(); err != nil {
  358. server.logger.Error("shutdown", fmt.Sprintln("Could not close datastore:", err))
  359. }
  360. }
  361. // Run starts the server.
  362. func (server *Server) Run() {
  363. // defer closing db/store
  364. defer server.store.Close()
  365. done := false
  366. for !done {
  367. select {
  368. case <-server.signals:
  369. server.Shutdown()
  370. done = true
  371. case <-server.rehashSignal:
  372. server.logger.Info("rehash", "Rehashing due to SIGHUP")
  373. err := server.rehash()
  374. if err != nil {
  375. server.logger.Error("rehash", fmt.Sprintln("Failed to rehash:", err.Error()))
  376. }
  377. case conn := <-server.newConns:
  378. // check connection limits
  379. ipaddr := net.ParseIP(IPString(conn.Conn.RemoteAddr()))
  380. if ipaddr != nil {
  381. // check DLINEs
  382. isBanned, info := server.dlines.CheckIP(ipaddr)
  383. if isBanned {
  384. banMessage := fmt.Sprintf(bannedFromServerBytes, info.Reason)
  385. if info.Time != nil {
  386. banMessage += fmt.Sprintf(" [%s]", info.Time.Duration.String())
  387. }
  388. conn.Conn.Write([]byte(banMessage))
  389. conn.Conn.Close()
  390. continue
  391. }
  392. // check connection limits
  393. server.connectionLimitsMutex.Lock()
  394. err := server.connectionLimits.AddClient(ipaddr, false)
  395. server.connectionLimitsMutex.Unlock()
  396. if err != nil {
  397. // too many connections from one client, tell the client and close the connection
  398. // this might not show up properly on some clients, but our objective here is just to close it out before it has a load impact on us
  399. conn.Conn.Write([]byte(tooManyClientsBytes))
  400. conn.Conn.Close()
  401. continue
  402. }
  403. // check connection throttle
  404. server.connectionThrottleMutex.Lock()
  405. err = server.connectionThrottle.AddClient(ipaddr)
  406. server.connectionThrottleMutex.Unlock()
  407. if err != nil {
  408. // too many connections too quickly from client, tell them and close the connection
  409. length := &IPRestrictTime{
  410. Duration: server.connectionThrottle.BanDuration,
  411. Expires: time.Now().Add(server.connectionThrottle.BanDuration),
  412. }
  413. server.dlines.AddIP(ipaddr, length, server.connectionThrottle.BanMessage, "Exceeded automated connection throttle")
  414. // reset ban on connectionThrottle
  415. server.connectionThrottle.ResetFor(ipaddr)
  416. // this might not show up properly on some clients, but our objective here is just to close it out before it has a load impact on us
  417. conn.Conn.Write([]byte(server.connectionThrottle.BanMessageBytes))
  418. conn.Conn.Close()
  419. continue
  420. }
  421. server.logger.Debug("localconnect-ip", fmt.Sprintf("Client connecting from %v", ipaddr))
  422. // prolly don't need to alert snomasks on this, only on connection reg
  423. go NewClient(server, conn.Conn, conn.IsTLS)
  424. continue
  425. }
  426. }
  427. }
  428. }
  429. //
  430. // IRC protocol listeners
  431. //
  432. // createListener starts the given listeners.
  433. func (server *Server) createListener(addr string, tlsMap map[string]*tls.Config) {
  434. config, listenTLS := tlsMap[addr]
  435. _, alreadyExists := server.listeners[addr]
  436. if alreadyExists {
  437. log.Fatal(server, "listener already exists:", addr)
  438. }
  439. // make listener event channel
  440. listenerEventChannel := make(chan ListenerEvent, 1)
  441. // make listener
  442. listener, err := net.Listen("tcp", addr)
  443. if err != nil {
  444. log.Fatal(server, "listen error: ", err)
  445. }
  446. tlsString := "plaintext"
  447. if listenTLS {
  448. config.ClientAuth = tls.RequestClientCert
  449. listener = tls.NewListener(listener, config)
  450. tlsString = "TLS"
  451. }
  452. // throw our details to the server so we can be modified/killed later
  453. li := ListenerInterface{
  454. Events: listenerEventChannel,
  455. Listener: listener,
  456. }
  457. server.listeners[addr] = li
  458. // start listening
  459. server.logger.Info("listeners", fmt.Sprintf("listening on %s using %s.", addr, tlsString))
  460. // setup accept goroutine
  461. go func() {
  462. for {
  463. conn, err := listener.Accept()
  464. if err == nil {
  465. newConn := clientConn{
  466. Conn: conn,
  467. IsTLS: listenTLS,
  468. }
  469. server.newConns <- newConn
  470. }
  471. select {
  472. case event := <-server.listeners[addr].Events:
  473. // this is used to confirm that whoever passed us this event has closed the existing listener correctly (in an attempt to get us to notice the event).
  474. // this is required to keep REHASH from having a very small race possibility of killing the primary listener
  475. server.listenerEventActMutex.Lock()
  476. server.listenerEventActMutex.Unlock()
  477. if event.Type == DestroyListener {
  478. // listener should already be closed, this is just for safety
  479. listener.Close()
  480. return
  481. } else if event.Type == UpdateListener {
  482. // close old listener
  483. listener.Close()
  484. // make new listener
  485. listener, err = net.Listen("tcp", addr)
  486. if err != nil {
  487. log.Fatal(server, "listen error: ", err)
  488. }
  489. tlsString := "plaintext"
  490. if event.NewConfig != nil {
  491. config = event.NewConfig
  492. config.ClientAuth = tls.RequestClientCert
  493. listener = tls.NewListener(listener, config)
  494. tlsString = "TLS"
  495. }
  496. // update server ListenerInterface
  497. li.Listener = listener
  498. server.listenerUpdateMutex.Lock()
  499. server.listeners[addr] = li
  500. server.listenerUpdateMutex.Unlock()
  501. // print notice
  502. server.logger.Info("listeners", fmt.Sprintf("updated listener %s using %s.", addr, tlsString))
  503. }
  504. default:
  505. // no events waiting for us, fall-through and continue
  506. }
  507. }
  508. }()
  509. }
  510. //
  511. // websocket listen goroutine
  512. //
  513. func (server *Server) wslisten(addr string, tlsMap map[string]*TLSListenConfig) {
  514. http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  515. if r.Method != "GET" {
  516. server.logger.Error("ws", addr, fmt.Sprintf("%s method not allowed", r.Method))
  517. return
  518. }
  519. // We don't have any subprotocols, so if someone attempts to `new
  520. // WebSocket(server, "subprotocol")` they'll break here, instead of
  521. // getting the default, ambiguous, response from gorilla.
  522. if v, ok := r.Header["Sec-Websocket-Protocol"]; ok {
  523. http.Error(w, fmt.Sprintf("WebSocket subprocotols (e.g. %s) not supported", v), 400)
  524. }
  525. ws, err := upgrader.Upgrade(w, r, nil)
  526. if err != nil {
  527. server.logger.Error("ws", addr, fmt.Sprintf("%s websocket upgrade error: %s", server.name, err))
  528. return
  529. }
  530. newConn := clientConn{
  531. Conn: WSContainer{ws},
  532. IsTLS: false, //TODO(dan): track TLS or not here properly
  533. }
  534. server.newConns <- newConn
  535. })
  536. go func() {
  537. config, listenTLS := tlsMap[addr]
  538. tlsString := "plaintext"
  539. var err error
  540. if listenTLS {
  541. tlsString = "TLS"
  542. }
  543. server.logger.Info("listeners", fmt.Sprintf("websocket listening on %s using %s.", addr, tlsString))
  544. if listenTLS {
  545. err = http.ListenAndServeTLS(addr, config.Cert, config.Key, nil)
  546. } else {
  547. err = http.ListenAndServe(addr, nil)
  548. }
  549. if err != nil {
  550. server.logger.Error("listeners", fmt.Sprintf("listenAndServe error [%s]: %s", tlsString, err))
  551. }
  552. }()
  553. }
  554. // generateMessageID returns a network-unique message ID.
  555. func (server *Server) generateMessageID() string {
  556. return fmt.Sprintf("%s-%s", strconv.FormatInt(time.Now().UTC().UnixNano(), 10), strconv.FormatInt(rand.Int63(), 10))
  557. }
  558. //
  559. // server functionality
  560. //
  561. func (server *Server) tryRegister(c *Client) {
  562. if c.registered || !c.HasNick() || !c.HasUsername() ||
  563. (c.capState == CapNegotiating) {
  564. return
  565. }
  566. // check KLINEs
  567. isBanned, info := server.klines.CheckMasks(c.AllNickmasks()...)
  568. if isBanned {
  569. reason := info.Reason
  570. if info.Time != nil {
  571. reason += fmt.Sprintf(" [%s]", info.Time.Duration.String())
  572. }
  573. c.Send(nil, "", "ERROR", fmt.Sprintf("You are banned from this server (%s)", reason))
  574. c.quitMessageSent = true
  575. c.destroy()
  576. return
  577. }
  578. // continue registration
  579. server.logger.Debug("localconnect", fmt.Sprintf("Client registered [%s] [u:%s] [r:%s]", c.nick, c.username, c.realname))
  580. server.snomasks.Send(sno.LocalConnects, fmt.Sprintf(ircfmt.Unescape("Client registered $c[grey][$r%s$c[grey]] [u:$r%s$c[grey]] [h:$r%s$c[grey]] [r:$r%s$c[grey]]"), c.nick, c.username, c.rawHostname, c.realname))
  581. c.Register()
  582. // send welcome text
  583. //NOTE(dan): we specifically use the NICK here instead of the nickmask
  584. // see http://modern.ircdocs.horse/#rplwelcome-001 for details on why we avoid using the nickmask
  585. c.Send(nil, server.name, RPL_WELCOME, c.nick, fmt.Sprintf("Welcome to the Internet Relay Network %s", c.nick))
  586. c.Send(nil, server.name, RPL_YOURHOST, c.nick, fmt.Sprintf("Your host is %s, running version %s", server.name, Ver))
  587. c.Send(nil, server.name, RPL_CREATED, c.nick, fmt.Sprintf("This server was created %s", server.ctime.Format(time.RFC1123)))
  588. //TODO(dan): Look at adding last optional [<channel modes with a parameter>] parameter
  589. c.Send(nil, server.name, RPL_MYINFO, c.nick, server.name, Ver, supportedUserModesString, supportedChannelModesString)
  590. c.RplISupport()
  591. server.MOTD(c)
  592. c.Send(nil, c.nickMaskString, RPL_UMODEIS, c.nick, c.ModeString())
  593. if server.logger.DumpingRawInOut {
  594. c.Notice("This server is in debug mode and is logging all user I/O. If you do not wish for everything you send to be readable by the server owner(s), please disconnect.")
  595. }
  596. }
  597. // MOTD serves the Message of the Day.
  598. func (server *Server) MOTD(client *Client) {
  599. if len(server.motdLines) < 1 {
  600. client.Send(nil, server.name, ERR_NOMOTD, client.nick, "MOTD File is missing")
  601. return
  602. }
  603. client.Send(nil, server.name, RPL_MOTDSTART, client.nick, fmt.Sprintf("- %s Message of the day - ", server.name))
  604. for _, line := range server.motdLines {
  605. client.Send(nil, server.name, RPL_MOTD, client.nick, line)
  606. }
  607. client.Send(nil, server.name, RPL_ENDOFMOTD, client.nick, "End of MOTD command")
  608. }
  609. //
  610. // registration commands
  611. //
  612. // PASS <password>
  613. func passHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  614. if client.registered {
  615. client.Send(nil, server.name, ERR_ALREADYREGISTRED, client.nick, "You may not reregister")
  616. return false
  617. }
  618. // if no password exists, skip checking
  619. if len(server.password) == 0 {
  620. client.authorized = true
  621. return false
  622. }
  623. // check the provided password
  624. password := []byte(msg.Params[0])
  625. if ComparePassword(server.password, password) != nil {
  626. client.Send(nil, server.name, ERR_PASSWDMISMATCH, client.nick, "Password incorrect")
  627. client.Send(nil, server.name, "ERROR", "Password incorrect")
  628. return true
  629. }
  630. client.authorized = true
  631. return false
  632. }
  633. // USER <username> * 0 <realname>
  634. func userHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  635. if client.registered {
  636. client.Send(nil, server.name, ERR_ALREADYREGISTRED, client.nick, "You may not reregister")
  637. return false
  638. }
  639. if !client.authorized {
  640. client.Quit("Bad password")
  641. return true
  642. }
  643. if client.username != "" && client.realname != "" {
  644. return false
  645. }
  646. // confirm that username is valid
  647. //
  648. _, err := CasefoldName(msg.Params[0])
  649. if err != nil {
  650. client.Send(nil, "", "ERROR", "Malformed username")
  651. return true
  652. }
  653. if !client.HasUsername() {
  654. client.username = "~" + msg.Params[0]
  655. // don't bother updating nickmask here, it's not valid anyway
  656. }
  657. if client.realname == "" {
  658. client.realname = msg.Params[3]
  659. }
  660. server.tryRegister(client)
  661. return false
  662. }
  663. // QUIT [<reason>]
  664. func quitHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  665. reason := "Quit"
  666. if len(msg.Params) > 0 {
  667. reason += ": " + msg.Params[0]
  668. }
  669. client.Quit(reason)
  670. return true
  671. }
  672. //
  673. // normal commands
  674. //
  675. // PING <server1> [<server2>]
  676. func pingHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  677. client.Send(nil, server.name, "PONG", msg.Params...)
  678. return false
  679. }
  680. // PONG <server> [ <server2> ]
  681. func pongHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  682. // client gets touched when they send this command, so we don't need to do anything
  683. return false
  684. }
  685. // JOIN <channel>{,<channel>} [<key>{,<key>}]
  686. // JOIN 0
  687. func joinHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  688. // handle JOIN 0
  689. if msg.Params[0] == "0" {
  690. for channel := range client.channels {
  691. channel.Part(client, client.nickCasefolded)
  692. }
  693. return false
  694. }
  695. // handle regular JOINs
  696. channels := strings.Split(msg.Params[0], ",")
  697. var keys []string
  698. if len(msg.Params) > 1 {
  699. keys = strings.Split(msg.Params[1], ",")
  700. }
  701. // get lock
  702. server.channelJoinPartMutex.Lock()
  703. defer server.channelJoinPartMutex.Unlock()
  704. for i, name := range channels {
  705. casefoldedName, err := CasefoldChannel(name)
  706. if err != nil {
  707. if len(name) > 0 {
  708. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, name, "No such channel")
  709. }
  710. continue
  711. }
  712. channel := server.channels.Get(casefoldedName)
  713. if channel == nil {
  714. if len(casefoldedName) > server.limits.ChannelLen {
  715. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, name, "No such channel")
  716. continue
  717. }
  718. channel = NewChannel(server, name, true)
  719. }
  720. var key string
  721. if len(keys) > i {
  722. key = keys[i]
  723. }
  724. channel.Join(client, key)
  725. }
  726. return false
  727. }
  728. // PART <channel>{,<channel>} [<reason>]
  729. func partHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  730. channels := strings.Split(msg.Params[0], ",")
  731. 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
  732. if len(msg.Params) > 1 {
  733. reason = msg.Params[1]
  734. }
  735. // get lock
  736. server.channelJoinPartMutex.Lock()
  737. defer server.channelJoinPartMutex.Unlock()
  738. for _, chname := range channels {
  739. casefoldedChannelName, err := CasefoldChannel(chname)
  740. channel := server.channels.Get(casefoldedChannelName)
  741. if err != nil || channel == nil {
  742. if len(chname) > 0 {
  743. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, chname, "No such channel")
  744. }
  745. continue
  746. }
  747. channel.Part(client, reason)
  748. }
  749. return false
  750. }
  751. // TOPIC <channel> [<topic>]
  752. func topicHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  753. name, err := CasefoldChannel(msg.Params[0])
  754. channel := server.channels.Get(name)
  755. if err != nil || channel == nil {
  756. if len(msg.Params[0]) > 0 {
  757. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, msg.Params[0], "No such channel")
  758. }
  759. return false
  760. }
  761. if len(msg.Params) > 1 {
  762. channel.SetTopic(client, msg.Params[1])
  763. } else {
  764. channel.GetTopic(client)
  765. }
  766. return false
  767. }
  768. // wordWrap wraps the given text into a series of lines that don't exceed lineWidth characters.
  769. func wordWrap(text string, lineWidth int) []string {
  770. var lines []string
  771. var cacheLine, cacheWord string
  772. for _, char := range text {
  773. if char == '\r' {
  774. continue
  775. } else if char == '\n' {
  776. cacheLine += cacheWord
  777. lines = append(lines, cacheLine)
  778. cacheWord = ""
  779. cacheLine = ""
  780. } else if (char == ' ' || char == '-') && len(cacheLine)+len(cacheWord)+1 < lineWidth {
  781. // natural word boundary
  782. cacheLine += cacheWord + string(char)
  783. cacheWord = ""
  784. } else if lineWidth <= len(cacheLine)+len(cacheWord)+1 {
  785. // time to wrap to next line
  786. if len(cacheLine) < (lineWidth / 2) {
  787. // this word takes up more than half a line... just split in the middle of the word
  788. cacheLine += cacheWord + string(char)
  789. cacheWord = ""
  790. } else {
  791. cacheWord += string(char)
  792. }
  793. lines = append(lines, cacheLine)
  794. cacheLine = ""
  795. } else {
  796. // normal character
  797. cacheWord += string(char)
  798. }
  799. }
  800. if 0 < len(cacheWord) {
  801. cacheLine += cacheWord
  802. }
  803. if 0 < len(cacheLine) {
  804. lines = append(lines, cacheLine)
  805. }
  806. return lines
  807. }
  808. // SplitMessage represents a message that's been split for sending.
  809. type SplitMessage struct {
  810. For512 []string
  811. ForMaxLine string
  812. }
  813. func (server *Server) splitMessage(original string, origIs512 bool) SplitMessage {
  814. var newSplit SplitMessage
  815. newSplit.ForMaxLine = original
  816. if !origIs512 {
  817. newSplit.For512 = wordWrap(original, 400)
  818. } else {
  819. newSplit.For512 = []string{original}
  820. }
  821. return newSplit
  822. }
  823. // PRIVMSG <target>{,<target>} <message>
  824. func privmsgHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  825. clientOnlyTags := GetClientOnlyTags(msg.Tags)
  826. targets := strings.Split(msg.Params[0], ",")
  827. message := msg.Params[1]
  828. // split privmsg
  829. splitMsg := server.splitMessage(message, !client.capabilities[MaxLine])
  830. for i, targetString := range targets {
  831. // max of four targets per privmsg
  832. if i > maxTargets-1 {
  833. break
  834. }
  835. prefixes, targetString := SplitChannelMembershipPrefixes(targetString)
  836. lowestPrefix := GetLowestChannelModePrefix(prefixes)
  837. // eh, no need to notify them
  838. if len(targetString) < 1 {
  839. continue
  840. }
  841. target, err := CasefoldChannel(targetString)
  842. if err == nil {
  843. channel := server.channels.Get(target)
  844. if channel == nil {
  845. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, targetString, "No such channel")
  846. continue
  847. }
  848. if !channel.CanSpeak(client) {
  849. client.Send(nil, client.server.name, ERR_CANNOTSENDTOCHAN, channel.name, "Cannot send to channel")
  850. continue
  851. }
  852. msgid := server.generateMessageID()
  853. channel.SplitPrivMsg(msgid, lowestPrefix, clientOnlyTags, client, splitMsg)
  854. } else {
  855. target, err = CasefoldName(targetString)
  856. if target == "chanserv" {
  857. server.chanservReceivePrivmsg(client, message)
  858. continue
  859. } else if target == "nickserv" {
  860. server.nickservReceivePrivmsg(client, message)
  861. continue
  862. }
  863. user := server.clients.Get(target)
  864. if err != nil || user == nil {
  865. if len(target) > 0 {
  866. client.Send(nil, server.name, ERR_NOSUCHNICK, target, "No such nick")
  867. }
  868. continue
  869. }
  870. if !user.capabilities[MessageTags] {
  871. clientOnlyTags = nil
  872. }
  873. msgid := server.generateMessageID()
  874. user.SendSplitMsgFromClient(msgid, client, clientOnlyTags, "PRIVMSG", user.nick, splitMsg)
  875. if client.capabilities[EchoMessage] {
  876. client.SendSplitMsgFromClient(msgid, client, clientOnlyTags, "PRIVMSG", user.nick, splitMsg)
  877. }
  878. if user.flags[Away] {
  879. //TODO(dan): possibly implement cooldown of away notifications to users
  880. client.Send(nil, server.name, RPL_AWAY, user.nick, user.awayMessage)
  881. }
  882. }
  883. }
  884. return false
  885. }
  886. // TAGMSG <target>{,<target>}
  887. func tagmsgHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  888. clientOnlyTags := GetClientOnlyTags(msg.Tags)
  889. // no client-only tags, so we can drop it
  890. if clientOnlyTags == nil {
  891. return false
  892. }
  893. targets := strings.Split(msg.Params[0], ",")
  894. for i, targetString := range targets {
  895. // max of four targets per privmsg
  896. if i > maxTargets-1 {
  897. break
  898. }
  899. prefixes, targetString := SplitChannelMembershipPrefixes(targetString)
  900. lowestPrefix := GetLowestChannelModePrefix(prefixes)
  901. // eh, no need to notify them
  902. if len(targetString) < 1 {
  903. continue
  904. }
  905. target, err := CasefoldChannel(targetString)
  906. if err == nil {
  907. channel := server.channels.Get(target)
  908. if channel == nil {
  909. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, targetString, "No such channel")
  910. continue
  911. }
  912. if !channel.CanSpeak(client) {
  913. client.Send(nil, client.server.name, ERR_CANNOTSENDTOCHAN, channel.name, "Cannot send to channel")
  914. continue
  915. }
  916. msgid := server.generateMessageID()
  917. channel.TagMsg(msgid, lowestPrefix, clientOnlyTags, client)
  918. } else {
  919. target, err = CasefoldName(targetString)
  920. user := server.clients.Get(target)
  921. if err != nil || user == nil {
  922. if len(target) > 0 {
  923. client.Send(nil, server.name, ERR_NOSUCHNICK, target, "No such nick")
  924. }
  925. continue
  926. }
  927. msgid := server.generateMessageID()
  928. // end user can't receive tagmsgs
  929. if !user.capabilities[MessageTags] {
  930. continue
  931. }
  932. user.SendFromClient(msgid, client, clientOnlyTags, "TAGMSG", user.nick)
  933. if client.capabilities[EchoMessage] {
  934. client.SendFromClient(msgid, client, clientOnlyTags, "TAGMSG", user.nick)
  935. }
  936. if user.flags[Away] {
  937. //TODO(dan): possibly implement cooldown of away notifications to users
  938. client.Send(nil, server.name, RPL_AWAY, user.nick, user.awayMessage)
  939. }
  940. }
  941. }
  942. return false
  943. }
  944. // WhoisChannelsNames returns the common channel names between two users.
  945. func (client *Client) WhoisChannelsNames(target *Client) []string {
  946. isMultiPrefix := target.capabilities[MultiPrefix]
  947. var chstrs []string
  948. index := 0
  949. for channel := range client.channels {
  950. channel.membersMutex.RLock()
  951. defer channel.membersMutex.RUnlock()
  952. // channel is secret and the target can't see it
  953. if !target.flags[Operator] && channel.flags[Secret] && !channel.members.Has(target) {
  954. continue
  955. }
  956. chstrs = append(chstrs, channel.members[client].Prefixes(isMultiPrefix)+channel.name)
  957. index++
  958. }
  959. return chstrs
  960. }
  961. // WHOIS [ <target> ] <mask> *( "," <mask> )
  962. func whoisHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  963. var masksString string
  964. //var target string
  965. if len(msg.Params) > 1 {
  966. //target = msg.Params[0]
  967. masksString = msg.Params[1]
  968. } else {
  969. masksString = msg.Params[0]
  970. }
  971. if len(strings.TrimSpace(masksString)) < 1 {
  972. client.Send(nil, server.name, ERR_UNKNOWNERROR, client.nick, msg.Command, "No masks given")
  973. return false
  974. }
  975. if client.flags[Operator] {
  976. masks := strings.Split(masksString, ",")
  977. for _, mask := range masks {
  978. casefoldedMask, err := Casefold(mask)
  979. if err != nil {
  980. client.Send(nil, client.server.name, ERR_NOSUCHNICK, mask, "No such nick")
  981. continue
  982. }
  983. matches := server.clients.FindAll(casefoldedMask)
  984. if len(matches) == 0 {
  985. client.Send(nil, client.server.name, ERR_NOSUCHNICK, mask, "No such nick")
  986. continue
  987. }
  988. for mclient := range matches {
  989. client.getWhoisOf(mclient)
  990. }
  991. }
  992. } else {
  993. // only get the first request
  994. casefoldedMask, err := Casefold(strings.Split(masksString, ",")[0])
  995. mclient := server.clients.Get(casefoldedMask)
  996. if err != nil || mclient == nil {
  997. client.Send(nil, client.server.name, ERR_NOSUCHNICK, masksString, "No such nick")
  998. // fall through, ENDOFWHOIS is always sent
  999. } else {
  1000. client.getWhoisOf(mclient)
  1001. }
  1002. }
  1003. client.Send(nil, server.name, RPL_ENDOFWHOIS, client.nick, masksString, "End of /WHOIS list")
  1004. return false
  1005. }
  1006. func (client *Client) getWhoisOf(target *Client) {
  1007. client.Send(nil, client.server.name, RPL_WHOISUSER, client.nick, target.nick, target.username, target.hostname, "*", target.realname)
  1008. whoischannels := client.WhoisChannelsNames(target)
  1009. if whoischannels != nil {
  1010. client.Send(nil, client.server.name, RPL_WHOISCHANNELS, client.nick, target.nick, strings.Join(whoischannels, " "))
  1011. }
  1012. if target.class != nil {
  1013. client.Send(nil, client.server.name, RPL_WHOISOPERATOR, client.nick, target.nick, target.whoisLine)
  1014. }
  1015. if target.certfp != "" && (client.flags[Operator] || client == target) {
  1016. client.Send(nil, client.server.name, RPL_WHOISCERTFP, client.nick, target.nick, fmt.Sprintf("has client certificate fingerprint %s", target.certfp))
  1017. }
  1018. client.Send(nil, client.server.name, RPL_WHOISIDLE, client.nick, target.nick, strconv.FormatUint(target.IdleSeconds(), 10), strconv.FormatInt(target.SignonTime(), 10), "seconds idle, signon time")
  1019. }
  1020. // RplWhoReplyNoMutex returns the WHO reply between one user and another channel/user.
  1021. // <channel> <user> <host> <server> <nick> ( "H" / "G" ) ["*"] [ ( "@" / "+" ) ]
  1022. // :<hopcount> <real name>
  1023. func (target *Client) RplWhoReplyNoMutex(channel *Channel, client *Client) {
  1024. channelName := "*"
  1025. flags := ""
  1026. if client.flags[Away] {
  1027. flags = "G"
  1028. } else {
  1029. flags = "H"
  1030. }
  1031. if client.flags[Operator] {
  1032. flags += "*"
  1033. }
  1034. if channel != nil {
  1035. flags += channel.members[client].Prefixes(target.capabilities[MultiPrefix])
  1036. channelName = channel.name
  1037. }
  1038. target.Send(nil, target.server.name, RPL_WHOREPLY, target.nick, channelName, client.username, client.hostname, client.server.name, client.nick, flags, strconv.Itoa(client.hops)+" "+client.realname)
  1039. }
  1040. func whoChannel(client *Client, channel *Channel, friends ClientSet) {
  1041. channel.membersMutex.RLock()
  1042. defer channel.membersMutex.RUnlock()
  1043. for member := range channel.members {
  1044. if !client.flags[Invisible] || friends[client] {
  1045. client.RplWhoReplyNoMutex(channel, member)
  1046. }
  1047. }
  1048. }
  1049. // WHO [ <mask> [ "o" ] ]
  1050. func whoHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1051. friends := client.Friends()
  1052. var mask string
  1053. if len(msg.Params) > 0 {
  1054. casefoldedMask, err := Casefold(msg.Params[0])
  1055. if err != nil {
  1056. client.Send(nil, server.name, ERR_UNKNOWNERROR, "WHO", "Mask isn't valid")
  1057. return false
  1058. }
  1059. mask = casefoldedMask
  1060. }
  1061. //TODO(dan): is this used and would I put this param in the Modern doc?
  1062. // if not, can we remove it?
  1063. //var operatorOnly bool
  1064. //if len(msg.Params) > 1 && msg.Params[1] == "o" {
  1065. // operatorOnly = true
  1066. //}
  1067. if mask == "" {
  1068. server.channels.ChansLock.RLock()
  1069. for _, channel := range server.channels.Chans {
  1070. whoChannel(client, channel, friends)
  1071. }
  1072. server.channels.ChansLock.RUnlock()
  1073. } else if mask[0] == '#' {
  1074. // TODO implement wildcard matching
  1075. //TODO(dan): ^ only for opers
  1076. channel := server.channels.Get(mask)
  1077. if channel != nil {
  1078. whoChannel(client, channel, friends)
  1079. }
  1080. } else {
  1081. for mclient := range server.clients.FindAll(mask) {
  1082. client.RplWhoReplyNoMutex(nil, mclient)
  1083. }
  1084. }
  1085. client.Send(nil, server.name, RPL_ENDOFWHO, client.nick, mask, "End of WHO list")
  1086. return false
  1087. }
  1088. // OPER <name> <password>
  1089. func operHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1090. name, err := CasefoldName(msg.Params[0])
  1091. if err != nil {
  1092. client.Send(nil, server.name, ERR_PASSWDMISMATCH, client.nick, "Password incorrect")
  1093. return true
  1094. }
  1095. hash := server.operators[name].Pass
  1096. password := []byte(msg.Params[1])
  1097. err = ComparePassword(hash, password)
  1098. if (hash == nil) || (err != nil) {
  1099. client.Send(nil, server.name, ERR_PASSWDMISMATCH, client.nick, "Password incorrect")
  1100. return true
  1101. }
  1102. client.flags[Operator] = true
  1103. client.operName = name
  1104. client.class = server.operators[name].Class
  1105. server.currentOpers[client] = true
  1106. client.whoisLine = server.operators[name].WhoisLine
  1107. // push new vhost if one is set
  1108. if len(server.operators[name].Vhost) > 0 {
  1109. for fClient := range client.Friends(ChgHost) {
  1110. fClient.SendFromClient("", client, nil, "CHGHOST", client.username, server.operators[name].Vhost)
  1111. }
  1112. // CHGHOST requires prefix nickmask to have original hostname, so do that before updating nickmask
  1113. client.vhost = server.operators[name].Vhost
  1114. client.updateNickMask()
  1115. }
  1116. // set new modes
  1117. var applied ModeChanges
  1118. if 0 < len(server.operators[name].Modes) {
  1119. modeChanges, unknownChanges := ParseUserModeChanges(strings.Split(server.operators[name].Modes, " ")...)
  1120. applied = client.applyUserModeChanges(true, modeChanges)
  1121. if 0 < len(unknownChanges) {
  1122. var runes string
  1123. for r := range unknownChanges {
  1124. runes += string(r)
  1125. }
  1126. client.Notice(fmt.Sprintf("Could not apply mode changes: +%s", runes))
  1127. }
  1128. }
  1129. client.Send(nil, server.name, RPL_YOUREOPER, client.nick, "You are now an IRC operator")
  1130. applied = append(applied, ModeChange{
  1131. mode: Operator,
  1132. op: Add,
  1133. })
  1134. client.Send(nil, server.name, "MODE", client.nick, applied.String())
  1135. return false
  1136. }
  1137. // rehash reloads the config and applies the changes from the config file.
  1138. func (server *Server) rehash() error {
  1139. server.logger.Debug("rehash", "Starting rehash")
  1140. // only let one REHASH go on at a time
  1141. server.rehashMutex.Lock()
  1142. defer server.rehashMutex.Unlock()
  1143. server.logger.Debug("rehash", "Got rehash lock")
  1144. config, err := LoadConfig(server.configFilename)
  1145. if err != nil {
  1146. return fmt.Errorf("Error rehashing config file config: %s", err.Error())
  1147. }
  1148. // line lengths cannot be changed after launching the server
  1149. if server.limits.LineLen.Tags != config.Limits.LineLen.Tags || server.limits.LineLen.Rest != config.Limits.LineLen.Rest {
  1150. return fmt.Errorf("Maximum line length (linelen) cannot be changed after launching the server, rehash aborted")
  1151. }
  1152. // confirm connectionLimits are fine
  1153. connectionLimits, err := NewConnectionLimits(config.Server.ConnectionLimits)
  1154. if err != nil {
  1155. return fmt.Errorf("Error rehashing config file connection-limits: %s", err.Error())
  1156. }
  1157. // confirm connectionThrottler is fine
  1158. connectionThrottle, err := NewConnectionThrottle(config.Server.ConnectionThrottle)
  1159. if err != nil {
  1160. return fmt.Errorf("Error rehashing config file connection-throttle: %s", err.Error())
  1161. }
  1162. // confirm operator stuff all exists and is fine
  1163. operclasses, err := config.OperatorClasses()
  1164. if err != nil {
  1165. return fmt.Errorf("Error rehashing config file operclasses: %s", err.Error())
  1166. }
  1167. opers, err := config.Operators(operclasses)
  1168. if err != nil {
  1169. return fmt.Errorf("Error rehashing config file opers: %s", err.Error())
  1170. }
  1171. for client := range server.currentOpers {
  1172. _, exists := opers[client.operName]
  1173. if !exists {
  1174. return fmt.Errorf("Oper [%s] no longer exists (used by client [%s])", client.operName, client.nickMaskString)
  1175. }
  1176. }
  1177. // apply new connectionlimits
  1178. server.connectionLimitsMutex.Lock()
  1179. server.connectionLimits = connectionLimits
  1180. server.connectionThrottleMutex.Lock()
  1181. server.connectionThrottle = connectionThrottle
  1182. server.clients.ByNickMutex.RLock()
  1183. for _, client := range server.clients.ByNick {
  1184. ipaddr := net.ParseIP(IPString(client.socket.conn.RemoteAddr()))
  1185. if ipaddr != nil {
  1186. server.connectionLimits.AddClient(ipaddr, true)
  1187. }
  1188. }
  1189. server.clients.ByNickMutex.RUnlock()
  1190. server.connectionThrottleMutex.Unlock()
  1191. server.connectionLimitsMutex.Unlock()
  1192. // setup new and removed caps
  1193. addedCaps := make(CapabilitySet)
  1194. removedCaps := make(CapabilitySet)
  1195. updatedCaps := make(CapabilitySet)
  1196. // SASL
  1197. if config.Accounts.AuthenticationEnabled && !server.accountAuthenticationEnabled {
  1198. // enabling SASL
  1199. SupportedCapabilities[SASL] = true
  1200. addedCaps[SASL] = true
  1201. }
  1202. if !config.Accounts.AuthenticationEnabled && server.accountAuthenticationEnabled {
  1203. // disabling SASL
  1204. SupportedCapabilities[SASL] = false
  1205. removedCaps[SASL] = true
  1206. }
  1207. server.accountAuthenticationEnabled = config.Accounts.AuthenticationEnabled
  1208. // STS
  1209. stsValue := config.Server.STS.Value()
  1210. var stsDisabled bool
  1211. server.logger.Debug("rehash", "STS Vals", CapValues[STS], stsValue, fmt.Sprintf("server[%v] config[%v]", server.stsEnabled, config.Server.STS.Enabled))
  1212. if config.Server.STS.Enabled && !server.stsEnabled {
  1213. // enabling STS
  1214. SupportedCapabilities[STS] = true
  1215. addedCaps[STS] = true
  1216. CapValues[STS] = stsValue
  1217. } else if !config.Server.STS.Enabled && server.stsEnabled {
  1218. // disabling STS
  1219. SupportedCapabilities[STS] = false
  1220. removedCaps[STS] = true
  1221. stsDisabled = true
  1222. } else if config.Server.STS.Enabled && server.stsEnabled && stsValue != CapValues[STS] {
  1223. // STS policy updated
  1224. CapValues[STS] = stsValue
  1225. updatedCaps[STS] = true
  1226. }
  1227. server.stsEnabled = config.Server.STS.Enabled
  1228. // burst new and removed caps
  1229. var capBurstClients ClientSet
  1230. added := make(map[CapVersion]string)
  1231. var removed string
  1232. // updated caps get DEL'd and then NEW'd
  1233. // so, we can just add updated ones to both removed and added lists here and they'll be correctly handled
  1234. server.logger.Debug("rehash", "Updated Caps", updatedCaps.String(Cap301), strconv.Itoa(len(updatedCaps)))
  1235. if len(updatedCaps) > 0 {
  1236. for capab := range updatedCaps {
  1237. addedCaps[capab] = true
  1238. removedCaps[capab] = true
  1239. }
  1240. }
  1241. if len(addedCaps) > 0 || len(removedCaps) > 0 {
  1242. capBurstClients = server.clients.AllWithCaps(CapNotify)
  1243. added[Cap301] = addedCaps.String(Cap301)
  1244. added[Cap302] = addedCaps.String(Cap302)
  1245. // removed never has values
  1246. removed = removedCaps.String(Cap301)
  1247. }
  1248. for sClient := range capBurstClients {
  1249. if stsDisabled {
  1250. // remove STS policy
  1251. //TODO(dan): this is an ugly hack. we can write this better.
  1252. stsPolicy := "sts=duration=0"
  1253. if len(addedCaps) > 0 {
  1254. added[Cap302] = added[Cap302] + " " + stsPolicy
  1255. } else {
  1256. addedCaps[STS] = true
  1257. added[Cap302] = stsPolicy
  1258. }
  1259. }
  1260. // DEL caps and then send NEW ones so that updated caps get removed/added correctly
  1261. if len(removedCaps) > 0 {
  1262. sClient.Send(nil, server.name, "CAP", sClient.nick, "DEL", removed)
  1263. }
  1264. if len(addedCaps) > 0 {
  1265. sClient.Send(nil, server.name, "CAP", sClient.nick, "NEW", added[sClient.capVersion])
  1266. }
  1267. }
  1268. // set server options
  1269. lineLenConfig := LineLenLimits{
  1270. Tags: config.Limits.LineLen.Tags,
  1271. Rest: config.Limits.LineLen.Rest,
  1272. }
  1273. server.limits = Limits{
  1274. AwayLen: int(config.Limits.AwayLen),
  1275. ChannelLen: int(config.Limits.ChannelLen),
  1276. KickLen: int(config.Limits.KickLen),
  1277. MonitorEntries: int(config.Limits.MonitorEntries),
  1278. NickLen: int(config.Limits.NickLen),
  1279. TopicLen: int(config.Limits.TopicLen),
  1280. ChanListModes: int(config.Limits.ChanListModes),
  1281. LineLen: lineLenConfig,
  1282. }
  1283. server.operclasses = *operclasses
  1284. server.operators = opers
  1285. server.checkIdent = config.Server.CheckIdent
  1286. // registration
  1287. accountReg := NewAccountRegistration(config.Accounts.Registration)
  1288. server.accountRegistration = &accountReg
  1289. server.channelRegistrationEnabled = config.Channels.Registration.Enabled
  1290. // set new sendqueue size
  1291. if config.Server.MaxSendQBytes != server.MaxSendQBytes {
  1292. server.MaxSendQBytes = config.Server.MaxSendQBytes
  1293. // update on all clients
  1294. server.clients.ByNickMutex.RLock()
  1295. for _, sClient := range server.clients.ByNick {
  1296. sClient.socket.MaxSendQBytes = config.Server.MaxSendQBytes
  1297. }
  1298. server.clients.ByNickMutex.RUnlock()
  1299. }
  1300. // set RPL_ISUPPORT
  1301. oldISupportList := server.isupport
  1302. server.setISupport()
  1303. newISupportReplies := oldISupportList.GetDifference(server.isupport)
  1304. // push new info to all of our clients
  1305. server.clients.ByNickMutex.RLock()
  1306. for _, sClient := range server.clients.ByNick {
  1307. for _, tokenline := range newISupportReplies {
  1308. // ugly trickery ahead
  1309. sClient.Send(nil, server.name, RPL_ISUPPORT, append([]string{sClient.nick}, tokenline...)...)
  1310. }
  1311. }
  1312. server.clients.ByNickMutex.RUnlock()
  1313. // destroy old listeners
  1314. tlsListeners := config.TLSListeners()
  1315. for addr := range server.listeners {
  1316. var exists bool
  1317. for _, newaddr := range config.Server.Listen {
  1318. if newaddr == addr {
  1319. exists = true
  1320. break
  1321. }
  1322. }
  1323. server.listenerEventActMutex.Lock()
  1324. if exists {
  1325. // update old listener
  1326. server.listeners[addr].Events <- ListenerEvent{
  1327. Type: UpdateListener,
  1328. NewConfig: tlsListeners[addr],
  1329. }
  1330. } else {
  1331. // destroy nonexistent listener
  1332. server.listeners[addr].Events <- ListenerEvent{
  1333. Type: DestroyListener,
  1334. }
  1335. }
  1336. // force listener to apply the event right away
  1337. server.listeners[addr].Listener.Close()
  1338. server.listenerEventActMutex.Unlock()
  1339. }
  1340. for _, newaddr := range config.Server.Listen {
  1341. _, exists := server.listeners[newaddr]
  1342. if !exists {
  1343. // make new listener
  1344. server.createListener(newaddr, tlsListeners)
  1345. }
  1346. }
  1347. return nil
  1348. }
  1349. // REHASH
  1350. func rehashHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1351. server.logger.Info("rehash", fmt.Sprintf("REHASH command used by %s", client.nick))
  1352. err := server.rehash()
  1353. if err == nil {
  1354. client.Send(nil, server.name, RPL_REHASHING, client.nick, "ircd.yaml", "Rehashing")
  1355. } else {
  1356. server.logger.Error("rehash", fmt.Sprintln("Failed to rehash:", err.Error()))
  1357. client.Send(nil, server.name, ERR_UNKNOWNERROR, client.nick, "REHASH", err.Error())
  1358. }
  1359. return false
  1360. }
  1361. // AWAY [<message>]
  1362. func awayHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1363. var isAway bool
  1364. var text string
  1365. if len(msg.Params) > 0 {
  1366. isAway = true
  1367. text = msg.Params[0]
  1368. if len(text) > server.limits.AwayLen {
  1369. text = text[:server.limits.AwayLen]
  1370. }
  1371. }
  1372. if isAway {
  1373. client.flags[Away] = true
  1374. } else {
  1375. delete(client.flags, Away)
  1376. }
  1377. client.awayMessage = text
  1378. var op ModeOp
  1379. if client.flags[Away] {
  1380. op = Add
  1381. client.Send(nil, server.name, RPL_NOWAWAY, client.nick, "You have been marked as being away")
  1382. } else {
  1383. op = Remove
  1384. client.Send(nil, server.name, RPL_UNAWAY, client.nick, "You are no longer marked as being away")
  1385. }
  1386. //TODO(dan): Should this be sent automagically as part of setting the flag/mode?
  1387. modech := ModeChanges{ModeChange{
  1388. mode: Away,
  1389. op: op,
  1390. }}
  1391. client.Send(nil, server.name, "MODE", client.nick, client.nick, modech.String())
  1392. // dispatch away-notify
  1393. for friend := range client.Friends(AwayNotify) {
  1394. if client.flags[Away] {
  1395. friend.SendFromClient("", client, nil, "AWAY", client.awayMessage)
  1396. } else {
  1397. friend.SendFromClient("", client, nil, "AWAY")
  1398. }
  1399. }
  1400. return false
  1401. }
  1402. // ISON <nick>{ <nick>}
  1403. func isonHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1404. var nicks = msg.Params
  1405. var err error
  1406. var casefoldedNick string
  1407. ison := make([]string, 0)
  1408. for _, nick := range nicks {
  1409. casefoldedNick, err = CasefoldName(nick)
  1410. if err != nil {
  1411. continue
  1412. }
  1413. if iclient := server.clients.Get(casefoldedNick); iclient != nil {
  1414. ison = append(ison, iclient.nick)
  1415. }
  1416. }
  1417. client.Send(nil, server.name, RPL_ISON, client.nick, strings.Join(nicks, " "))
  1418. return false
  1419. }
  1420. // MOTD [<target>]
  1421. func motdHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1422. //TODO(dan): hook this up when we have multiple servers I guess???
  1423. //var target string
  1424. //if len(msg.Params) > 0 {
  1425. // target = msg.Params[0]
  1426. //}
  1427. server.MOTD(client)
  1428. return false
  1429. }
  1430. // NOTICE <target>{,<target>} <message>
  1431. func noticeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1432. clientOnlyTags := GetClientOnlyTags(msg.Tags)
  1433. targets := strings.Split(msg.Params[0], ",")
  1434. message := msg.Params[1]
  1435. // split privmsg
  1436. splitMsg := server.splitMessage(message, !client.capabilities[MaxLine])
  1437. for i, targetString := range targets {
  1438. // max of four targets per privmsg
  1439. if i > maxTargets-1 {
  1440. break
  1441. }
  1442. prefixes, targetString := SplitChannelMembershipPrefixes(targetString)
  1443. lowestPrefix := GetLowestChannelModePrefix(prefixes)
  1444. target, cerr := CasefoldChannel(targetString)
  1445. if cerr == nil {
  1446. channel := server.channels.Get(target)
  1447. if channel == nil {
  1448. // errors silently ignored with NOTICE as per RFC
  1449. continue
  1450. }
  1451. if !channel.CanSpeak(client) {
  1452. // errors silently ignored with NOTICE as per RFC
  1453. continue
  1454. }
  1455. msgid := server.generateMessageID()
  1456. channel.SplitNotice(msgid, lowestPrefix, clientOnlyTags, client, splitMsg)
  1457. } else {
  1458. target, err := CasefoldName(targetString)
  1459. if err != nil {
  1460. continue
  1461. }
  1462. if target == "chanserv" {
  1463. server.chanservReceiveNotice(client, message)
  1464. continue
  1465. } else if target == "nickserv" {
  1466. server.nickservReceiveNotice(client, message)
  1467. continue
  1468. }
  1469. user := server.clients.Get(target)
  1470. if user == nil {
  1471. // errors silently ignored with NOTICE as per RFC
  1472. continue
  1473. }
  1474. if !user.capabilities[MessageTags] {
  1475. clientOnlyTags = nil
  1476. }
  1477. msgid := server.generateMessageID()
  1478. user.SendSplitMsgFromClient(msgid, client, clientOnlyTags, "NOTICE", user.nick, splitMsg)
  1479. if client.capabilities[EchoMessage] {
  1480. client.SendSplitMsgFromClient(msgid, client, clientOnlyTags, "NOTICE", user.nick, splitMsg)
  1481. }
  1482. }
  1483. }
  1484. return false
  1485. }
  1486. // KICK <channel>{,<channel>} <user>{,<user>} [<comment>]
  1487. func kickHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1488. channels := strings.Split(msg.Params[0], ",")
  1489. users := strings.Split(msg.Params[1], ",")
  1490. if (len(channels) != len(users)) && (len(users) != 1) {
  1491. client.Send(nil, server.name, ERR_NEEDMOREPARAMS, client.nick, "KICK", "Not enough parameters")
  1492. return false
  1493. }
  1494. var kicks [][]string
  1495. for index, channel := range channels {
  1496. if len(users) == 1 {
  1497. kicks = append(kicks, []string{channel, users[0]})
  1498. } else {
  1499. kicks = append(kicks, []string{channel, users[index]})
  1500. }
  1501. }
  1502. var comment string
  1503. if len(msg.Params) > 2 {
  1504. comment = msg.Params[2]
  1505. }
  1506. for _, info := range kicks {
  1507. chname := info[0]
  1508. nickname := info[1]
  1509. casefoldedChname, err := CasefoldChannel(chname)
  1510. channel := server.channels.Get(casefoldedChname)
  1511. if err != nil || channel == nil {
  1512. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, chname, "No such channel")
  1513. continue
  1514. }
  1515. casefoldedNickname, err := CasefoldName(nickname)
  1516. target := server.clients.Get(casefoldedNickname)
  1517. if err != nil || target == nil {
  1518. client.Send(nil, server.name, ERR_NOSUCHNICK, nickname, "No such nick")
  1519. continue
  1520. }
  1521. // make sure client has privs to kick the given user
  1522. //TODO(dan): split this into a separate function that checks if users have privs
  1523. // over other users, useful for things like -aoh as well
  1524. channel.membersMutex.Lock()
  1525. var hasPrivs bool
  1526. for _, mode := range ChannelPrivModes {
  1527. if channel.members[client][mode] {
  1528. hasPrivs = true
  1529. // admins cannot kick other admins
  1530. if mode == ChannelAdmin && channel.members[target][ChannelAdmin] {
  1531. hasPrivs = false
  1532. }
  1533. break
  1534. } else if channel.members[target][mode] {
  1535. break
  1536. }
  1537. }
  1538. if hasPrivs {
  1539. if comment == "" {
  1540. comment = nickname
  1541. }
  1542. channel.kickNoMutex(client, target, comment)
  1543. } else {
  1544. client.Send(nil, client.server.name, ERR_CHANOPRIVSNEEDED, chname, "You're not a channel operator")
  1545. }
  1546. channel.membersMutex.Unlock()
  1547. }
  1548. return false
  1549. }
  1550. // LIST [<channel>{,<channel>} [<server>]]
  1551. func listHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1552. var channels []string
  1553. if len(msg.Params) > 0 {
  1554. channels = strings.Split(msg.Params[0], ",")
  1555. }
  1556. var target string
  1557. if len(msg.Params) > 1 {
  1558. target = msg.Params[1]
  1559. }
  1560. //TODO(dan): target server when we have multiple servers
  1561. //TODO(dan): we should continue just fine if it's this current server though
  1562. if target != "" {
  1563. client.Send(nil, server.name, ERR_NOSUCHSERVER, client.nick, target, "No such server")
  1564. return false
  1565. }
  1566. if len(channels) == 0 {
  1567. server.channels.ChansLock.RLock()
  1568. for _, channel := range server.channels.Chans {
  1569. if !client.flags[Operator] && channel.flags[Secret] {
  1570. continue
  1571. }
  1572. client.RplList(channel)
  1573. }
  1574. server.channels.ChansLock.RUnlock()
  1575. } else {
  1576. // limit regular users to only listing one channel
  1577. if !client.flags[Operator] {
  1578. channels = channels[:1]
  1579. }
  1580. for _, chname := range channels {
  1581. casefoldedChname, err := CasefoldChannel(chname)
  1582. channel := server.channels.Get(casefoldedChname)
  1583. if err != nil || channel == nil || (!client.flags[Operator] && channel.flags[Secret]) {
  1584. if len(chname) > 0 {
  1585. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, chname, "No such channel")
  1586. }
  1587. continue
  1588. }
  1589. client.RplList(channel)
  1590. }
  1591. }
  1592. client.Send(nil, server.name, RPL_LISTEND, client.nick, "End of LIST")
  1593. return false
  1594. }
  1595. func (target *Client) RplList(channel *Channel) {
  1596. channel.membersMutex.RLock()
  1597. defer channel.membersMutex.RUnlock()
  1598. // get the correct number of channel members
  1599. var memberCount int
  1600. if target.flags[Operator] || channel.members.Has(target) {
  1601. memberCount = len(channel.members)
  1602. } else {
  1603. for member := range channel.members {
  1604. if !member.flags[Invisible] {
  1605. memberCount++
  1606. }
  1607. }
  1608. }
  1609. target.Send(nil, target.server.name, RPL_LIST, target.nick, channel.name, strconv.Itoa(memberCount), channel.topic)
  1610. }
  1611. // NAMES [<channel>{,<channel>}]
  1612. func namesHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1613. var channels []string
  1614. if len(msg.Params) > 0 {
  1615. channels = strings.Split(msg.Params[0], ",")
  1616. }
  1617. //var target string
  1618. //if len(msg.Params) > 1 {
  1619. // target = msg.Params[1]
  1620. //}
  1621. if len(channels) == 0 {
  1622. server.channels.ChansLock.RLock()
  1623. for _, channel := range server.channels.Chans {
  1624. channel.Names(client)
  1625. }
  1626. server.channels.ChansLock.RUnlock()
  1627. return false
  1628. }
  1629. // limit regular users to only listing one channel
  1630. if !client.flags[Operator] {
  1631. channels = channels[:1]
  1632. }
  1633. for _, chname := range channels {
  1634. casefoldedChname, err := CasefoldChannel(chname)
  1635. channel := server.channels.Get(casefoldedChname)
  1636. if err != nil || channel == nil {
  1637. if len(chname) > 0 {
  1638. client.Send(nil, server.name, ERR_NOSUCHCHANNEL, client.nick, chname, "No such channel")
  1639. }
  1640. continue
  1641. }
  1642. channel.Names(client)
  1643. }
  1644. return false
  1645. }
  1646. // VERSION [<server>]
  1647. func versionHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1648. var target string
  1649. if len(msg.Params) > 0 {
  1650. target = msg.Params[0]
  1651. }
  1652. casefoldedTarget, err := Casefold(target)
  1653. if target != "" && (err != nil || casefoldedTarget != server.nameCasefolded) {
  1654. client.Send(nil, server.name, ERR_NOSUCHSERVER, client.nick, target, "No such server")
  1655. return false
  1656. }
  1657. client.Send(nil, server.name, RPL_VERSION, client.nick, Ver, server.name)
  1658. client.RplISupport()
  1659. return false
  1660. }
  1661. // INVITE <nickname> <channel>
  1662. func inviteHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1663. nickname := msg.Params[0]
  1664. channelName := msg.Params[1]
  1665. casefoldedNickname, err := CasefoldName(nickname)
  1666. target := server.clients.Get(casefoldedNickname)
  1667. if err != nil || target == nil {
  1668. client.Send(nil, server.name, ERR_NOSUCHNICK, client.nick, nickname, "No such nick")
  1669. return false
  1670. }
  1671. casefoldedChannelName, err := CasefoldChannel(channelName)
  1672. channel := server.channels.Get(casefoldedChannelName)
  1673. if err != nil || channel == nil {
  1674. client.Send(nil, server.name, RPL_INVITING, client.nick, target.nick, channelName)
  1675. target.Send(nil, client.nickMaskString, "INVITE", target.nick, channel.name)
  1676. return true
  1677. }
  1678. channel.Invite(target, client)
  1679. return false
  1680. }
  1681. // TIME [<server>]
  1682. func timeHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1683. var target string
  1684. if len(msg.Params) > 0 {
  1685. target = msg.Params[0]
  1686. }
  1687. casefoldedTarget, err := Casefold(target)
  1688. if (target != "") && err != nil || (casefoldedTarget != server.nameCasefolded) {
  1689. client.Send(nil, server.name, ERR_NOSUCHSERVER, client.nick, target, "No such server")
  1690. return false
  1691. }
  1692. client.Send(nil, server.name, RPL_TIME, client.nick, server.name, time.Now().Format(time.RFC1123))
  1693. return false
  1694. }
  1695. // KILL <nickname> <comment>
  1696. func killHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1697. nickname := msg.Params[0]
  1698. comment := "<no reason supplied>"
  1699. if len(msg.Params) > 1 {
  1700. comment = msg.Params[1]
  1701. }
  1702. casefoldedNickname, err := CasefoldName(nickname)
  1703. target := server.clients.Get(casefoldedNickname)
  1704. if err != nil || target == nil {
  1705. client.Send(nil, client.server.name, ERR_NOSUCHNICK, nickname, "No such nick")
  1706. return false
  1707. }
  1708. quitMsg := fmt.Sprintf("Killed (%s (%s))", client.nick, comment)
  1709. target.Quit(quitMsg)
  1710. target.destroy()
  1711. return false
  1712. }
  1713. // WHOWAS <nickname> [<count> [<server>]]
  1714. func whowasHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1715. nicknames := strings.Split(msg.Params[0], ",")
  1716. var count int64
  1717. if len(msg.Params) > 1 {
  1718. count, _ = strconv.ParseInt(msg.Params[1], 10, 64)
  1719. }
  1720. //var target string
  1721. //if len(msg.Params) > 2 {
  1722. // target = msg.Params[2]
  1723. //}
  1724. for _, nickname := range nicknames {
  1725. results := server.whoWas.Find(nickname, count)
  1726. if len(results) == 0 {
  1727. if len(nickname) > 0 {
  1728. client.Send(nil, server.name, ERR_WASNOSUCHNICK, client.nick, nickname, "There was no such nickname")
  1729. }
  1730. } else {
  1731. for _, whoWas := range results {
  1732. client.Send(nil, server.name, RPL_WHOWASUSER, client.nick, whoWas.nickname, whoWas.username, whoWas.hostname, "*", whoWas.realname)
  1733. }
  1734. }
  1735. if len(nickname) > 0 {
  1736. client.Send(nil, server.name, RPL_ENDOFWHOWAS, client.nick, nickname, "End of WHOWAS")
  1737. }
  1738. }
  1739. return false
  1740. }
  1741. // LUSERS [<mask> [<server>]]
  1742. func lusersHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1743. //TODO(vegax87) Fix network statistics and additional parameters
  1744. var totalcount, invisiblecount, opercount int
  1745. server.clients.ByNickMutex.RLock()
  1746. defer server.clients.ByNickMutex.RUnlock()
  1747. for _, onlineusers := range server.clients.ByNick {
  1748. totalcount++
  1749. if onlineusers.flags[Invisible] {
  1750. invisiblecount++
  1751. }
  1752. if onlineusers.flags[Operator] {
  1753. opercount++
  1754. }
  1755. }
  1756. client.Send(nil, server.name, RPL_LUSERCLIENT, client.nick, fmt.Sprintf("There are %d users and %d invisible on %d server(s)", totalcount, invisiblecount, 1))
  1757. client.Send(nil, server.name, RPL_LUSEROP, client.nick, fmt.Sprintf("%d IRC Operators online", opercount))
  1758. client.Send(nil, server.name, RPL_LUSERCHANNELS, client.nick, fmt.Sprintf("%d channels formed", server.channels.Len()))
  1759. client.Send(nil, server.name, RPL_LUSERME, client.nick, fmt.Sprintf("I have %d clients and %d servers", totalcount, 1))
  1760. return false
  1761. }
  1762. // USERHOST <nickname> [<nickname> <nickname> ...]
  1763. func userhostHandler(server *Server, client *Client, msg ircmsg.IrcMessage) bool {
  1764. returnedNicks := make(map[string]bool)
  1765. for i, nickname := range msg.Params {
  1766. if i >= 10 {
  1767. break
  1768. }
  1769. casefoldedNickname, err := CasefoldName(nickname)
  1770. target := server.clients.Get(casefoldedNickname)
  1771. if err != nil || target == nil {
  1772. client.Send(nil, client.server.name, ERR_NOSUCHNICK, nickname, "No such nick")
  1773. return false
  1774. }
  1775. if returnedNicks[casefoldedNickname] {
  1776. continue
  1777. }
  1778. // to prevent returning multiple results for a single nick
  1779. returnedNicks[casefoldedNickname] = true
  1780. var isOper, isAway string
  1781. if target.flags[Operator] {
  1782. isOper = "*"
  1783. }
  1784. if target.flags[Away] {
  1785. isAway = "-"
  1786. } else {
  1787. isAway = "+"
  1788. }
  1789. client.Send(nil, client.server.name, RPL_USERHOST, client.nick, fmt.Sprintf("%s%s=%s%s@%s", target.nick, isOper, isAway, target.username, target.hostname))
  1790. }
  1791. return false
  1792. }