Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.

channel.go 33KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101
  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. "bytes"
  8. "fmt"
  9. "strconv"
  10. "strings"
  11. "time"
  12. "sync"
  13. "github.com/oragono/oragono/irc/caps"
  14. "github.com/oragono/oragono/irc/history"
  15. "github.com/oragono/oragono/irc/modes"
  16. "github.com/oragono/oragono/irc/utils"
  17. )
  18. // Channel represents a channel that clients can join.
  19. type Channel struct {
  20. flags modes.ModeSet
  21. lists map[modes.Mode]*UserMaskSet
  22. key string
  23. members MemberSet
  24. membersCache []*Client // allow iteration over channel members without holding the lock
  25. name string
  26. nameCasefolded string
  27. server *Server
  28. createdTime time.Time
  29. registeredFounder string
  30. registeredTime time.Time
  31. topic string
  32. topicSetBy string
  33. topicSetTime time.Time
  34. userLimit int
  35. accountToUMode map[string]modes.Mode
  36. history history.Buffer
  37. stateMutex sync.RWMutex // tier 1
  38. writerSemaphore Semaphore // tier 1.5
  39. joinPartMutex sync.Mutex // tier 3
  40. ensureLoaded utils.Once // manages loading stored registration info from the database
  41. dirtyBits uint
  42. }
  43. // NewChannel creates a new channel from a `Server` and a `name`
  44. // string, which must be unique on the server.
  45. func NewChannel(s *Server, name string, registered bool) *Channel {
  46. casefoldedName, err := CasefoldChannel(name)
  47. if err != nil {
  48. s.logger.Error("internal", "Bad channel name", name, err.Error())
  49. return nil
  50. }
  51. channel := &Channel{
  52. createdTime: time.Now(), // may be overwritten by applyRegInfo
  53. lists: map[modes.Mode]*UserMaskSet{
  54. modes.BanMask: NewUserMaskSet(),
  55. modes.ExceptMask: NewUserMaskSet(),
  56. modes.InviteMask: NewUserMaskSet(),
  57. },
  58. members: make(MemberSet),
  59. name: name,
  60. nameCasefolded: casefoldedName,
  61. server: s,
  62. accountToUMode: make(map[string]modes.Mode),
  63. }
  64. config := s.Config()
  65. channel.writerSemaphore.Initialize(1)
  66. channel.history.Initialize(config.History.ChannelLength)
  67. if !registered {
  68. for _, mode := range config.Channels.defaultModes {
  69. channel.flags.SetMode(mode, true)
  70. }
  71. // no loading to do, so "mark" the load operation as "done":
  72. channel.ensureLoaded.Do(func() {})
  73. } // else: modes will be loaded before first join
  74. return channel
  75. }
  76. // EnsureLoaded blocks until the channel's registration info has been loaded
  77. // from the database.
  78. func (channel *Channel) EnsureLoaded() {
  79. channel.ensureLoaded.Do(func() {
  80. nmc := channel.NameCasefolded()
  81. info, err := channel.server.channelRegistry.LoadChannel(nmc)
  82. if err == nil {
  83. channel.applyRegInfo(info)
  84. } else {
  85. channel.server.logger.Error("internal", "couldn't load channel", nmc, err.Error())
  86. }
  87. })
  88. }
  89. func (channel *Channel) IsLoaded() bool {
  90. return channel.ensureLoaded.Done()
  91. }
  92. // read in channel state that was persisted in the DB
  93. func (channel *Channel) applyRegInfo(chanReg RegisteredChannel) {
  94. channel.stateMutex.Lock()
  95. defer channel.stateMutex.Unlock()
  96. channel.registeredFounder = chanReg.Founder
  97. channel.registeredTime = chanReg.RegisteredAt
  98. channel.topic = chanReg.Topic
  99. channel.topicSetBy = chanReg.TopicSetBy
  100. channel.topicSetTime = chanReg.TopicSetTime
  101. channel.name = chanReg.Name
  102. channel.createdTime = chanReg.RegisteredAt
  103. channel.key = chanReg.Key
  104. for _, mode := range chanReg.Modes {
  105. channel.flags.SetMode(mode, true)
  106. }
  107. for _, mask := range chanReg.Banlist {
  108. channel.lists[modes.BanMask].Add(mask)
  109. }
  110. for _, mask := range chanReg.Exceptlist {
  111. channel.lists[modes.ExceptMask].Add(mask)
  112. }
  113. for _, mask := range chanReg.Invitelist {
  114. channel.lists[modes.InviteMask].Add(mask)
  115. }
  116. for account, mode := range chanReg.AccountToUMode {
  117. channel.accountToUMode[account] = mode
  118. }
  119. }
  120. // obtain a consistent snapshot of the channel state that can be persisted to the DB
  121. func (channel *Channel) ExportRegistration(includeFlags uint) (info RegisteredChannel) {
  122. channel.stateMutex.RLock()
  123. defer channel.stateMutex.RUnlock()
  124. info.Name = channel.name
  125. info.NameCasefolded = channel.nameCasefolded
  126. info.Founder = channel.registeredFounder
  127. info.RegisteredAt = channel.registeredTime
  128. if includeFlags&IncludeTopic != 0 {
  129. info.Topic = channel.topic
  130. info.TopicSetBy = channel.topicSetBy
  131. info.TopicSetTime = channel.topicSetTime
  132. }
  133. if includeFlags&IncludeModes != 0 {
  134. info.Key = channel.key
  135. info.Modes = channel.flags.AllModes()
  136. }
  137. if includeFlags&IncludeLists != 0 {
  138. for mask := range channel.lists[modes.BanMask].masks {
  139. info.Banlist = append(info.Banlist, mask)
  140. }
  141. for mask := range channel.lists[modes.ExceptMask].masks {
  142. info.Exceptlist = append(info.Exceptlist, mask)
  143. }
  144. for mask := range channel.lists[modes.InviteMask].masks {
  145. info.Invitelist = append(info.Invitelist, mask)
  146. }
  147. info.AccountToUMode = make(map[string]modes.Mode)
  148. for account, mode := range channel.accountToUMode {
  149. info.AccountToUMode[account] = mode
  150. }
  151. }
  152. return
  153. }
  154. // begin: asynchronous database writeback implementation, modeled on irc/socket.go
  155. // MarkDirty marks part (or all) of a channel's data as needing to be written back
  156. // to the database, then starts a writer goroutine if necessary.
  157. // This is the equivalent of Socket.Write().
  158. func (channel *Channel) MarkDirty(dirtyBits uint) {
  159. channel.stateMutex.Lock()
  160. isRegistered := channel.registeredFounder != ""
  161. channel.dirtyBits = channel.dirtyBits | dirtyBits
  162. channel.stateMutex.Unlock()
  163. if !isRegistered {
  164. return
  165. }
  166. channel.wakeWriter()
  167. }
  168. // IsClean returns whether a channel can be safely removed from the server.
  169. // To avoid the obvious TOCTOU race condition, it must be called while holding
  170. // ChannelManager's lock (that way, no one can join and make the channel dirty again
  171. // between this method exiting and the actual deletion).
  172. func (channel *Channel) IsClean() bool {
  173. if !channel.writerSemaphore.TryAcquire() {
  174. // a database write (which may fail) is in progress, the channel cannot be cleaned up
  175. return false
  176. }
  177. defer channel.writerSemaphore.Release()
  178. channel.stateMutex.RLock()
  179. defer channel.stateMutex.RUnlock()
  180. // the channel must be empty, and either be unregistered or fully written to the DB
  181. return len(channel.members) == 0 && (channel.registeredFounder == "" || channel.dirtyBits == 0)
  182. }
  183. func (channel *Channel) wakeWriter() {
  184. if channel.writerSemaphore.TryAcquire() {
  185. go channel.writeLoop()
  186. }
  187. }
  188. // equivalent of Socket.send()
  189. func (channel *Channel) writeLoop() {
  190. for {
  191. // TODO(#357) check the error value of this and implement timed backoff
  192. channel.performWrite(0)
  193. channel.writerSemaphore.Release()
  194. channel.stateMutex.RLock()
  195. isDirty := channel.dirtyBits != 0
  196. isEmpty := len(channel.members) == 0
  197. channel.stateMutex.RUnlock()
  198. if !isDirty {
  199. if isEmpty {
  200. channel.server.channels.Cleanup(channel)
  201. }
  202. return // nothing to do
  203. } // else: isDirty, so we need to write again
  204. if !channel.writerSemaphore.TryAcquire() {
  205. return
  206. }
  207. }
  208. }
  209. // Store writes part (or all) of the channel's data back to the database,
  210. // blocking until the write is complete. This is the equivalent of
  211. // Socket.BlockingWrite.
  212. func (channel *Channel) Store(dirtyBits uint) (err error) {
  213. defer func() {
  214. channel.stateMutex.Lock()
  215. isDirty := channel.dirtyBits != 0
  216. isEmpty := len(channel.members) == 0
  217. channel.stateMutex.Unlock()
  218. if isDirty {
  219. channel.wakeWriter()
  220. } else if isEmpty {
  221. channel.server.channels.Cleanup(channel)
  222. }
  223. }()
  224. channel.writerSemaphore.Acquire()
  225. defer channel.writerSemaphore.Release()
  226. return channel.performWrite(dirtyBits)
  227. }
  228. // do an individual write; equivalent of Socket.send()
  229. func (channel *Channel) performWrite(additionalDirtyBits uint) (err error) {
  230. channel.stateMutex.Lock()
  231. dirtyBits := channel.dirtyBits | additionalDirtyBits
  232. channel.dirtyBits = 0
  233. isRegistered := channel.registeredFounder != ""
  234. channel.stateMutex.Unlock()
  235. if !isRegistered || dirtyBits == 0 {
  236. return
  237. }
  238. info := channel.ExportRegistration(dirtyBits)
  239. err = channel.server.channelRegistry.StoreChannel(info, dirtyBits)
  240. if err != nil {
  241. channel.stateMutex.Lock()
  242. channel.dirtyBits = channel.dirtyBits | dirtyBits
  243. channel.stateMutex.Unlock()
  244. }
  245. return
  246. }
  247. // SetRegistered registers the channel, returning an error if it was already registered.
  248. func (channel *Channel) SetRegistered(founder string) error {
  249. channel.stateMutex.Lock()
  250. defer channel.stateMutex.Unlock()
  251. if channel.registeredFounder != "" {
  252. return errChannelAlreadyRegistered
  253. }
  254. channel.registeredFounder = founder
  255. channel.registeredTime = time.Now()
  256. channel.accountToUMode[founder] = modes.ChannelFounder
  257. return nil
  258. }
  259. // SetUnregistered deletes the channel's registration information.
  260. func (channel *Channel) SetUnregistered(expectedFounder string) {
  261. channel.stateMutex.Lock()
  262. defer channel.stateMutex.Unlock()
  263. if channel.registeredFounder != expectedFounder {
  264. return
  265. }
  266. channel.registeredFounder = ""
  267. var zeroTime time.Time
  268. channel.registeredTime = zeroTime
  269. channel.accountToUMode = make(map[string]modes.Mode)
  270. }
  271. // IsRegistered returns whether the channel is registered.
  272. func (channel *Channel) IsRegistered() bool {
  273. channel.stateMutex.RLock()
  274. defer channel.stateMutex.RUnlock()
  275. return channel.registeredFounder != ""
  276. }
  277. func (channel *Channel) regenerateMembersCache() {
  278. channel.stateMutex.RLock()
  279. result := make([]*Client, len(channel.members))
  280. i := 0
  281. for client := range channel.members {
  282. result[i] = client
  283. i++
  284. }
  285. channel.stateMutex.RUnlock()
  286. channel.stateMutex.Lock()
  287. channel.membersCache = result
  288. channel.stateMutex.Unlock()
  289. }
  290. // Names sends the list of users joined to the channel to the given client.
  291. func (channel *Channel) Names(client *Client, rb *ResponseBuffer) {
  292. isMultiPrefix := client.capabilities.Has(caps.MultiPrefix)
  293. isUserhostInNames := client.capabilities.Has(caps.UserhostInNames)
  294. maxNamLen := 480 - len(client.server.name) - len(client.Nick())
  295. var namesLines []string
  296. var buffer bytes.Buffer
  297. for _, target := range channel.Members() {
  298. var nick string
  299. if isUserhostInNames {
  300. nick = target.NickMaskString()
  301. } else {
  302. nick = target.Nick()
  303. }
  304. channel.stateMutex.RLock()
  305. modes := channel.members[target]
  306. channel.stateMutex.RUnlock()
  307. if modes == nil {
  308. continue
  309. }
  310. prefix := modes.Prefixes(isMultiPrefix)
  311. if buffer.Len()+len(nick)+len(prefix)+1 > maxNamLen {
  312. namesLines = append(namesLines, buffer.String())
  313. buffer.Reset()
  314. }
  315. if buffer.Len() > 0 {
  316. buffer.WriteString(" ")
  317. }
  318. buffer.WriteString(prefix)
  319. buffer.WriteString(nick)
  320. }
  321. if buffer.Len() > 0 {
  322. namesLines = append(namesLines, buffer.String())
  323. }
  324. for _, line := range namesLines {
  325. if buffer.Len() > 0 {
  326. rb.Add(nil, client.server.name, RPL_NAMREPLY, client.nick, "=", channel.name, line)
  327. }
  328. }
  329. rb.Add(nil, client.server.name, RPL_ENDOFNAMES, client.nick, channel.name, client.t("End of NAMES list"))
  330. }
  331. func channelUserModeIsAtLeast(clientModes *modes.ModeSet, permission modes.Mode) bool {
  332. if clientModes == nil {
  333. return false
  334. }
  335. for _, mode := range modes.ChannelUserModes {
  336. if clientModes.HasMode(mode) {
  337. return true
  338. }
  339. if mode == permission {
  340. break
  341. }
  342. }
  343. return false
  344. }
  345. // ClientIsAtLeast returns whether the client has at least the given channel privilege.
  346. func (channel *Channel) ClientIsAtLeast(client *Client, permission modes.Mode) bool {
  347. channel.stateMutex.RLock()
  348. clientModes := channel.members[client]
  349. channel.stateMutex.RUnlock()
  350. return channelUserModeIsAtLeast(clientModes, permission)
  351. }
  352. func (channel *Channel) ClientPrefixes(client *Client, isMultiPrefix bool) string {
  353. channel.stateMutex.RLock()
  354. defer channel.stateMutex.RUnlock()
  355. modes, present := channel.members[client]
  356. if !present {
  357. return ""
  358. } else {
  359. return modes.Prefixes(isMultiPrefix)
  360. }
  361. }
  362. func (channel *Channel) ClientHasPrivsOver(client *Client, target *Client) bool {
  363. channel.stateMutex.RLock()
  364. clientModes := channel.members[client]
  365. targetModes := channel.members[target]
  366. channel.stateMutex.RUnlock()
  367. if clientModes.HasMode(modes.ChannelFounder) {
  368. // founder can kick anyone
  369. return true
  370. } else if clientModes.HasMode(modes.ChannelAdmin) {
  371. // admins cannot kick other admins
  372. return !channelUserModeIsAtLeast(targetModes, modes.ChannelAdmin)
  373. } else if clientModes.HasMode(modes.ChannelOperator) {
  374. // operators *can* kick other operators
  375. return !channelUserModeIsAtLeast(targetModes, modes.ChannelAdmin)
  376. } else if clientModes.HasMode(modes.Halfop) {
  377. // halfops cannot kick other halfops
  378. return !channelUserModeIsAtLeast(targetModes, modes.Halfop)
  379. } else {
  380. // voice and unprivileged cannot kick anyone
  381. return false
  382. }
  383. }
  384. func (channel *Channel) hasClient(client *Client) bool {
  385. channel.stateMutex.RLock()
  386. defer channel.stateMutex.RUnlock()
  387. _, present := channel.members[client]
  388. return present
  389. }
  390. // <mode> <mode params>
  391. func (channel *Channel) modeStrings(client *Client) (result []string) {
  392. isMember := client.HasMode(modes.Operator) || channel.hasClient(client)
  393. showKey := isMember && (channel.key != "")
  394. showUserLimit := channel.userLimit > 0
  395. mods := "+"
  396. // flags with args
  397. if showKey {
  398. mods += modes.Key.String()
  399. }
  400. if showUserLimit {
  401. mods += modes.UserLimit.String()
  402. }
  403. mods += channel.flags.String()
  404. channel.stateMutex.RLock()
  405. defer channel.stateMutex.RUnlock()
  406. result = []string{mods}
  407. // args for flags with args: The order must match above to keep
  408. // positional arguments in place.
  409. if showKey {
  410. result = append(result, channel.key)
  411. }
  412. if showUserLimit {
  413. result = append(result, strconv.Itoa(channel.userLimit))
  414. }
  415. return
  416. }
  417. func (channel *Channel) IsEmpty() bool {
  418. channel.stateMutex.RLock()
  419. defer channel.stateMutex.RUnlock()
  420. return len(channel.members) == 0
  421. }
  422. // Join joins the given client to this channel (if they can be joined).
  423. func (channel *Channel) Join(client *Client, key string, isSajoin bool, rb *ResponseBuffer) {
  424. details := client.Details()
  425. channel.stateMutex.RLock()
  426. chname := channel.name
  427. chcfname := channel.nameCasefolded
  428. founder := channel.registeredFounder
  429. chkey := channel.key
  430. limit := channel.userLimit
  431. chcount := len(channel.members)
  432. _, alreadyJoined := channel.members[client]
  433. persistentMode := channel.accountToUMode[details.account]
  434. channel.stateMutex.RUnlock()
  435. if alreadyJoined {
  436. // no message needs to be sent
  437. return
  438. }
  439. // the founder can always join (even if they disabled auto +q on join);
  440. // anyone who automatically receives halfop or higher can always join
  441. hasPrivs := isSajoin || (founder != "" && founder == details.account) || (persistentMode != 0 && persistentMode != modes.Voice)
  442. if !hasPrivs && limit != 0 && chcount >= limit {
  443. rb.Add(nil, client.server.name, ERR_CHANNELISFULL, chname, fmt.Sprintf(client.t("Cannot join channel (+%s)"), "l"))
  444. return
  445. }
  446. if !hasPrivs && chkey != "" && !utils.SecretTokensMatch(chkey, key) {
  447. rb.Add(nil, client.server.name, ERR_BADCHANNELKEY, chname, fmt.Sprintf(client.t("Cannot join channel (+%s)"), "k"))
  448. return
  449. }
  450. isInvited := client.CheckInvited(chcfname) || channel.lists[modes.InviteMask].Match(details.nickMaskCasefolded)
  451. if !hasPrivs && channel.flags.HasMode(modes.InviteOnly) && !isInvited {
  452. rb.Add(nil, client.server.name, ERR_INVITEONLYCHAN, chname, fmt.Sprintf(client.t("Cannot join channel (+%s)"), "i"))
  453. return
  454. }
  455. if !hasPrivs && channel.lists[modes.BanMask].Match(details.nickMaskCasefolded) &&
  456. !isInvited &&
  457. !channel.lists[modes.ExceptMask].Match(details.nickMaskCasefolded) {
  458. rb.Add(nil, client.server.name, ERR_BANNEDFROMCHAN, chname, fmt.Sprintf(client.t("Cannot join channel (+%s)"), "b"))
  459. return
  460. }
  461. client.server.logger.Debug("join", fmt.Sprintf("%s joined channel %s", details.nick, chname))
  462. givenMode := func() (givenMode modes.Mode) {
  463. channel.joinPartMutex.Lock()
  464. defer channel.joinPartMutex.Unlock()
  465. func() {
  466. channel.stateMutex.Lock()
  467. defer channel.stateMutex.Unlock()
  468. channel.members.Add(client)
  469. firstJoin := len(channel.members) == 1
  470. newChannel := firstJoin && channel.registeredFounder == ""
  471. if newChannel {
  472. givenMode = modes.ChannelOperator
  473. } else {
  474. givenMode = persistentMode
  475. }
  476. if givenMode != 0 {
  477. channel.members[client].SetMode(givenMode, true)
  478. }
  479. }()
  480. channel.regenerateMembersCache()
  481. message := utils.SplitMessage{}
  482. message.Msgid = details.realname
  483. channel.history.Add(history.Item{
  484. Type: history.Join,
  485. Nick: details.nickMask,
  486. AccountName: details.accountName,
  487. Message: message,
  488. })
  489. return
  490. }()
  491. client.addChannel(channel)
  492. var modestr string
  493. if givenMode != 0 {
  494. modestr = fmt.Sprintf("+%v", givenMode)
  495. }
  496. for _, member := range channel.Members() {
  497. if member == client {
  498. continue
  499. }
  500. if member.capabilities.Has(caps.ExtendedJoin) {
  501. member.Send(nil, details.nickMask, "JOIN", chname, details.accountName, details.realname)
  502. } else {
  503. member.Send(nil, details.nickMask, "JOIN", chname)
  504. }
  505. if givenMode != 0 {
  506. member.Send(nil, client.server.name, "MODE", chname, modestr, details.nick)
  507. }
  508. }
  509. if client.capabilities.Has(caps.ExtendedJoin) {
  510. rb.Add(nil, details.nickMask, "JOIN", chname, details.accountName, details.realname)
  511. } else {
  512. rb.Add(nil, details.nickMask, "JOIN", chname)
  513. }
  514. channel.SendTopic(client, rb, false)
  515. channel.Names(client, rb)
  516. // TODO #259 can be implemented as Flush(false) (i.e., nonblocking) while holding joinPartMutex
  517. rb.Flush(true)
  518. replayLimit := channel.server.Config().History.AutoreplayOnJoin
  519. if replayLimit > 0 {
  520. items := channel.history.Latest(replayLimit)
  521. channel.replayHistoryItems(rb, items)
  522. rb.Flush(true)
  523. }
  524. }
  525. // Part parts the given client from this channel, with the given message.
  526. func (channel *Channel) Part(client *Client, message string, rb *ResponseBuffer) {
  527. chname := channel.Name()
  528. if !channel.hasClient(client) {
  529. rb.Add(nil, client.server.name, ERR_NOTONCHANNEL, chname, client.t("You're not on that channel"))
  530. return
  531. }
  532. channel.Quit(client)
  533. details := client.Details()
  534. for _, member := range channel.Members() {
  535. member.Send(nil, details.nickMask, "PART", chname, message)
  536. }
  537. rb.Add(nil, details.nickMask, "PART", chname, message)
  538. channel.history.Add(history.Item{
  539. Type: history.Part,
  540. Nick: details.nickMask,
  541. AccountName: details.accountName,
  542. Message: utils.MakeSplitMessage(message, true),
  543. })
  544. client.server.logger.Debug("part", fmt.Sprintf("%s left channel %s", details.nick, chname))
  545. }
  546. // Resume is called after a successful global resume to:
  547. // 1. Replace the old client with the new in the channel's data structures
  548. // 2. Send JOIN and MODE lines to channel participants (including the new client)
  549. // 3. Replay missed message history to the client
  550. func (channel *Channel) Resume(newClient, oldClient *Client, timestamp time.Time) {
  551. now := time.Now()
  552. channel.resumeAndAnnounce(newClient, oldClient)
  553. if !timestamp.IsZero() {
  554. channel.replayHistoryForResume(newClient, timestamp, now)
  555. }
  556. }
  557. func (channel *Channel) resumeAndAnnounce(newClient, oldClient *Client) {
  558. var oldModeSet *modes.ModeSet
  559. func() {
  560. channel.joinPartMutex.Lock()
  561. defer channel.joinPartMutex.Unlock()
  562. defer channel.regenerateMembersCache()
  563. channel.stateMutex.Lock()
  564. defer channel.stateMutex.Unlock()
  565. newClient.channels[channel] = true
  566. oldModeSet = channel.members[oldClient]
  567. if oldModeSet == nil {
  568. oldModeSet = modes.NewModeSet()
  569. }
  570. channel.members.Remove(oldClient)
  571. channel.members[newClient] = oldModeSet
  572. }()
  573. // construct fake modestring if necessary
  574. oldModes := oldModeSet.String()
  575. if 0 < len(oldModes) {
  576. oldModes = "+" + oldModes
  577. }
  578. // send join for old clients
  579. nick := newClient.Nick()
  580. nickMask := newClient.NickMaskString()
  581. accountName := newClient.AccountName()
  582. realName := newClient.Realname()
  583. for _, member := range channel.Members() {
  584. if member.capabilities.Has(caps.Resume) {
  585. continue
  586. }
  587. if member.capabilities.Has(caps.ExtendedJoin) {
  588. member.Send(nil, nickMask, "JOIN", channel.name, accountName, realName)
  589. } else {
  590. member.Send(nil, nickMask, "JOIN", channel.name)
  591. }
  592. if 0 < len(oldModes) {
  593. member.Send(nil, channel.server.name, "MODE", channel.name, oldModes, nick)
  594. }
  595. }
  596. rb := NewResponseBuffer(newClient)
  597. // use blocking i/o to synchronize with the later history replay
  598. if newClient.capabilities.Has(caps.ExtendedJoin) {
  599. rb.Add(nil, nickMask, "JOIN", channel.name, accountName, realName)
  600. } else {
  601. rb.Add(nil, nickMask, "JOIN", channel.name)
  602. }
  603. channel.SendTopic(newClient, rb, false)
  604. channel.Names(newClient, rb)
  605. if 0 < len(oldModes) {
  606. rb.Add(nil, newClient.server.name, "MODE", channel.name, oldModes, nick)
  607. }
  608. rb.Send(true)
  609. }
  610. func (channel *Channel) replayHistoryForResume(newClient *Client, after time.Time, before time.Time) {
  611. items, complete := channel.history.Between(after, before, false, 0)
  612. rb := NewResponseBuffer(newClient)
  613. channel.replayHistoryItems(rb, items)
  614. if !complete && !newClient.resumeDetails.HistoryIncomplete {
  615. // warn here if we didn't warn already
  616. rb.Add(nil, "HistServ", "NOTICE", channel.Name(), newClient.t("Some additional message history may have been lost"))
  617. }
  618. rb.Send(true)
  619. }
  620. func stripMaskFromNick(nickMask string) (nick string) {
  621. index := strings.Index(nickMask, "!")
  622. if index == -1 {
  623. return
  624. }
  625. return nickMask[0:index]
  626. }
  627. func (channel *Channel) replayHistoryItems(rb *ResponseBuffer, items []history.Item) {
  628. chname := channel.Name()
  629. client := rb.target
  630. serverTime := client.capabilities.Has(caps.ServerTime)
  631. for _, item := range items {
  632. var tags map[string]string
  633. if serverTime {
  634. tags = map[string]string{"time": item.Time.Format(IRCv3TimestampFormat)}
  635. }
  636. // TODO(#437) support history.Tagmsg
  637. switch item.Type {
  638. case history.Privmsg:
  639. rb.AddSplitMessageFromClient(item.Nick, item.AccountName, tags, "PRIVMSG", chname, item.Message)
  640. case history.Notice:
  641. rb.AddSplitMessageFromClient(item.Nick, item.AccountName, tags, "NOTICE", chname, item.Message)
  642. case history.Join:
  643. nick := stripMaskFromNick(item.Nick)
  644. var message string
  645. if item.AccountName == "*" {
  646. message = fmt.Sprintf(client.t("%s joined the channel"), nick)
  647. } else {
  648. message = fmt.Sprintf(client.t("%[1]s [account: %[2]s] joined the channel"), nick, item.AccountName)
  649. }
  650. rb.Add(tags, "HistServ", "PRIVMSG", chname, message)
  651. case history.Part:
  652. nick := stripMaskFromNick(item.Nick)
  653. message := fmt.Sprintf(client.t("%[1]s left the channel (%[2]s)"), nick, item.Message.Message)
  654. rb.Add(tags, "HistServ", "PRIVMSG", chname, message)
  655. case history.Quit:
  656. nick := stripMaskFromNick(item.Nick)
  657. message := fmt.Sprintf(client.t("%[1]s quit (%[2]s)"), nick, item.Message.Message)
  658. rb.Add(tags, "HistServ", "PRIVMSG", chname, message)
  659. case history.Kick:
  660. nick := stripMaskFromNick(item.Nick)
  661. // XXX Msgid is the kick target
  662. message := fmt.Sprintf(client.t("%[1]s kicked %[2]s (%[3]s)"), nick, item.Message.Msgid, item.Message.Message)
  663. rb.Add(tags, "HistServ", "PRIVMSG", chname, message)
  664. }
  665. }
  666. }
  667. // SendTopic sends the channel topic to the given client.
  668. // `sendNoTopic` controls whether RPL_NOTOPIC is sent when the topic is unset
  669. func (channel *Channel) SendTopic(client *Client, rb *ResponseBuffer, sendNoTopic bool) {
  670. if !channel.hasClient(client) {
  671. rb.Add(nil, client.server.name, ERR_NOTONCHANNEL, client.nick, channel.name, client.t("You're not on that channel"))
  672. return
  673. }
  674. channel.stateMutex.RLock()
  675. name := channel.name
  676. topic := channel.topic
  677. topicSetBy := channel.topicSetBy
  678. topicSetTime := channel.topicSetTime
  679. channel.stateMutex.RUnlock()
  680. if topic == "" {
  681. if sendNoTopic {
  682. rb.Add(nil, client.server.name, RPL_NOTOPIC, client.nick, name, client.t("No topic is set"))
  683. }
  684. return
  685. }
  686. rb.Add(nil, client.server.name, RPL_TOPIC, client.nick, name, topic)
  687. rb.Add(nil, client.server.name, RPL_TOPICTIME, client.nick, name, topicSetBy, strconv.FormatInt(topicSetTime.Unix(), 10))
  688. }
  689. // SetTopic sets the topic of this channel, if the client is allowed to do so.
  690. func (channel *Channel) SetTopic(client *Client, topic string, rb *ResponseBuffer) {
  691. if !(client.HasMode(modes.Operator) || channel.hasClient(client)) {
  692. rb.Add(nil, client.server.name, ERR_NOTONCHANNEL, channel.name, client.t("You're not on that channel"))
  693. return
  694. }
  695. if channel.flags.HasMode(modes.OpOnlyTopic) && !channel.ClientIsAtLeast(client, modes.ChannelOperator) {
  696. rb.Add(nil, client.server.name, ERR_CHANOPRIVSNEEDED, channel.name, client.t("You're not a channel operator"))
  697. return
  698. }
  699. topicLimit := client.server.Limits().TopicLen
  700. if len(topic) > topicLimit {
  701. topic = topic[:topicLimit]
  702. }
  703. channel.stateMutex.Lock()
  704. channel.topic = topic
  705. channel.topicSetBy = client.nickMaskString
  706. channel.topicSetTime = time.Now()
  707. channel.stateMutex.Unlock()
  708. for _, member := range channel.Members() {
  709. if member == client {
  710. rb.Add(nil, client.nickMaskString, "TOPIC", channel.name, topic)
  711. } else {
  712. member.Send(nil, client.nickMaskString, "TOPIC", channel.name, topic)
  713. }
  714. }
  715. channel.MarkDirty(IncludeTopic)
  716. }
  717. // CanSpeak returns true if the client can speak on this channel.
  718. func (channel *Channel) CanSpeak(client *Client) bool {
  719. channel.stateMutex.RLock()
  720. defer channel.stateMutex.RUnlock()
  721. _, hasClient := channel.members[client]
  722. if channel.flags.HasMode(modes.NoOutside) && !hasClient {
  723. return false
  724. }
  725. if channel.flags.HasMode(modes.Moderated) && !channel.ClientIsAtLeast(client, modes.Voice) {
  726. return false
  727. }
  728. if channel.flags.HasMode(modes.RegisteredOnly) && client.Account() == "" {
  729. return false
  730. }
  731. return true
  732. }
  733. func (channel *Channel) SendSplitMessage(command string, minPrefix *modes.Mode, clientOnlyTags map[string]string, client *Client, message utils.SplitMessage, rb *ResponseBuffer) {
  734. var histType history.ItemType
  735. switch command {
  736. case "PRIVMSG":
  737. histType = history.Privmsg
  738. case "NOTICE":
  739. histType = history.Notice
  740. case "TAGMSG":
  741. histType = history.Tagmsg
  742. default:
  743. channel.server.logger.Error("internal", "unrecognized Channel.SendSplitMessage command", command)
  744. return
  745. }
  746. if !channel.CanSpeak(client) {
  747. rb.Add(nil, client.server.name, ERR_CANNOTSENDTOCHAN, channel.name, client.t("Cannot send to channel"))
  748. return
  749. }
  750. // for STATUSMSG
  751. var minPrefixMode modes.Mode
  752. if minPrefix != nil {
  753. minPrefixMode = *minPrefix
  754. }
  755. // send echo-message
  756. if client.capabilities.Has(caps.EchoMessage) {
  757. var tagsToUse map[string]string
  758. if client.capabilities.Has(caps.MessageTags) {
  759. tagsToUse = clientOnlyTags
  760. }
  761. nickMaskString := client.NickMaskString()
  762. accountName := client.AccountName()
  763. if command == "TAGMSG" && client.capabilities.Has(caps.MessageTags) {
  764. rb.AddFromClient(message.Msgid, nickMaskString, accountName, tagsToUse, command, channel.name)
  765. } else {
  766. rb.AddSplitMessageFromClient(nickMaskString, accountName, tagsToUse, command, channel.name, message)
  767. }
  768. }
  769. nickmask := client.NickMaskString()
  770. account := client.AccountName()
  771. now := time.Now().UTC()
  772. for _, member := range channel.Members() {
  773. if minPrefix != nil && !channel.ClientIsAtLeast(member, minPrefixMode) {
  774. // STATUSMSG
  775. continue
  776. }
  777. // echo-message is handled above, so skip sending the msg to the user themselves as well
  778. if member == client {
  779. continue
  780. }
  781. var tagsToUse map[string]string
  782. if member.capabilities.Has(caps.MessageTags) {
  783. tagsToUse = clientOnlyTags
  784. } else if command == "TAGMSG" {
  785. continue
  786. }
  787. if command == "TAGMSG" {
  788. member.sendFromClientInternal(false, now, message.Msgid, nickmask, account, tagsToUse, command, channel.name)
  789. } else {
  790. member.sendSplitMsgFromClientInternal(false, now, nickmask, account, tagsToUse, command, channel.name, message)
  791. }
  792. }
  793. channel.history.Add(history.Item{
  794. Type: histType,
  795. Message: message,
  796. Nick: nickmask,
  797. AccountName: account,
  798. Time: now,
  799. })
  800. }
  801. func (channel *Channel) applyModeToMember(client *Client, mode modes.Mode, op modes.ModeOp, nick string, rb *ResponseBuffer) (result *modes.ModeChange) {
  802. casefoldedName, err := CasefoldName(nick)
  803. target := channel.server.clients.Get(casefoldedName)
  804. if err != nil || target == nil {
  805. rb.Add(nil, client.server.name, ERR_NOSUCHNICK, client.Nick(), nick, client.t("No such nick"))
  806. return nil
  807. }
  808. channel.stateMutex.Lock()
  809. modeset, exists := channel.members[target]
  810. if exists {
  811. if modeset.SetMode(mode, op == modes.Add) {
  812. result = &modes.ModeChange{
  813. Op: op,
  814. Mode: mode,
  815. Arg: nick,
  816. }
  817. }
  818. }
  819. channel.stateMutex.Unlock()
  820. if !exists {
  821. rb.Add(nil, client.server.name, ERR_USERNOTINCHANNEL, client.Nick(), channel.Name(), client.t("They aren't on that channel"))
  822. }
  823. return
  824. }
  825. // ShowMaskList shows the given list to the client.
  826. func (channel *Channel) ShowMaskList(client *Client, mode modes.Mode, rb *ResponseBuffer) {
  827. // choose appropriate modes
  828. var rpllist, rplendoflist string
  829. if mode == modes.BanMask {
  830. rpllist = RPL_BANLIST
  831. rplendoflist = RPL_ENDOFBANLIST
  832. } else if mode == modes.ExceptMask {
  833. rpllist = RPL_EXCEPTLIST
  834. rplendoflist = RPL_ENDOFEXCEPTLIST
  835. } else if mode == modes.InviteMask {
  836. rpllist = RPL_INVITELIST
  837. rplendoflist = RPL_ENDOFINVITELIST
  838. }
  839. nick := client.Nick()
  840. channel.stateMutex.RLock()
  841. // XXX don't acquire any new locks in this section, besides Socket.Write
  842. for mask := range channel.lists[mode].masks {
  843. rb.Add(nil, client.server.name, rpllist, nick, channel.name, mask)
  844. }
  845. channel.stateMutex.RUnlock()
  846. rb.Add(nil, client.server.name, rplendoflist, nick, channel.name, client.t("End of list"))
  847. }
  848. func (channel *Channel) applyModeMask(client *Client, mode modes.Mode, op modes.ModeOp, mask string, rb *ResponseBuffer) bool {
  849. list := channel.lists[mode]
  850. if list == nil {
  851. // This should never happen, but better safe than panicky.
  852. return false
  853. }
  854. if (op == modes.List) || (mask == "") {
  855. channel.ShowMaskList(client, mode, rb)
  856. return false
  857. }
  858. if !channel.ClientIsAtLeast(client, modes.ChannelOperator) {
  859. rb.Add(nil, client.server.name, ERR_CHANOPRIVSNEEDED, channel.name, client.t("You're not a channel operator"))
  860. return false
  861. }
  862. if op == modes.Add {
  863. return list.Add(mask)
  864. }
  865. if op == modes.Remove {
  866. return list.Remove(mask)
  867. }
  868. return false
  869. }
  870. // Quit removes the given client from the channel
  871. func (channel *Channel) Quit(client *Client) {
  872. channelEmpty := func() bool {
  873. channel.joinPartMutex.Lock()
  874. defer channel.joinPartMutex.Unlock()
  875. channel.stateMutex.Lock()
  876. channel.members.Remove(client)
  877. channelEmpty := len(channel.members) == 0
  878. channel.stateMutex.Unlock()
  879. channel.regenerateMembersCache()
  880. return channelEmpty
  881. }()
  882. if channelEmpty {
  883. client.server.channels.Cleanup(channel)
  884. }
  885. client.removeChannel(channel)
  886. }
  887. func (channel *Channel) Kick(client *Client, target *Client, comment string, rb *ResponseBuffer) {
  888. if !(client.HasMode(modes.Operator) || channel.hasClient(client)) {
  889. rb.Add(nil, client.server.name, ERR_NOTONCHANNEL, channel.name, client.t("You're not on that channel"))
  890. return
  891. }
  892. if !channel.hasClient(target) {
  893. rb.Add(nil, client.server.name, ERR_USERNOTINCHANNEL, client.nick, channel.name, client.t("They aren't on that channel"))
  894. return
  895. }
  896. if !channel.ClientHasPrivsOver(client, target) {
  897. rb.Add(nil, client.server.name, ERR_CHANOPRIVSNEEDED, channel.name, client.t("You don't have enough channel privileges"))
  898. return
  899. }
  900. kicklimit := client.server.Limits().KickLen
  901. if len(comment) > kicklimit {
  902. comment = comment[:kicklimit]
  903. }
  904. clientMask := client.NickMaskString()
  905. targetNick := target.Nick()
  906. for _, member := range channel.Members() {
  907. member.Send(nil, clientMask, "KICK", channel.name, targetNick, comment)
  908. }
  909. message := utils.SplitMessage{}
  910. message.Message = comment
  911. message.Msgid = targetNick // XXX abuse this field
  912. channel.history.Add(history.Item{
  913. Type: history.Kick,
  914. Nick: clientMask,
  915. AccountName: target.AccountName(),
  916. Message: message,
  917. })
  918. channel.Quit(target)
  919. }
  920. // Invite invites the given client to the channel, if the inviter can do so.
  921. func (channel *Channel) Invite(invitee *Client, inviter *Client, rb *ResponseBuffer) {
  922. chname := channel.Name()
  923. if channel.flags.HasMode(modes.InviteOnly) && !channel.ClientIsAtLeast(inviter, modes.ChannelOperator) {
  924. rb.Add(nil, inviter.server.name, ERR_CHANOPRIVSNEEDED, chname, inviter.t("You're not a channel operator"))
  925. return
  926. }
  927. if !channel.hasClient(inviter) {
  928. rb.Add(nil, inviter.server.name, ERR_NOTONCHANNEL, chname, inviter.t("You're not on that channel"))
  929. return
  930. }
  931. if channel.flags.HasMode(modes.InviteOnly) {
  932. invitee.Invite(channel.NameCasefolded())
  933. }
  934. for _, member := range channel.Members() {
  935. if member.capabilities.Has(caps.InviteNotify) && member != inviter && member != invitee && channel.ClientIsAtLeast(member, modes.Halfop) {
  936. member.Send(nil, inviter.NickMaskString(), "INVITE", invitee.Nick(), chname)
  937. }
  938. }
  939. cnick := inviter.Nick()
  940. tnick := invitee.Nick()
  941. rb.Add(nil, inviter.server.name, RPL_INVITING, cnick, tnick, chname)
  942. invitee.Send(nil, inviter.NickMaskString(), "INVITE", tnick, chname)
  943. if invitee.HasMode(modes.Away) {
  944. rb.Add(nil, inviter.server.name, RPL_AWAY, cnick, tnick, invitee.AwayMessage())
  945. }
  946. }