123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470 |
- // Copyright (c) 2017 Shivaram Lingamneni <slingamn@cs.stanford.edu>
- // released under the MIT license
-
- package irc
-
- import (
- "net"
- "sync/atomic"
- "time"
- "unsafe"
-
- "github.com/oragono/oragono/irc/languages"
- "github.com/oragono/oragono/irc/modes"
- )
-
- func (server *Server) Config() (config *Config) {
- return (*Config)(atomic.LoadPointer(&server.config))
- }
-
- func (server *Server) SetConfig(config *Config) {
- atomic.StorePointer(&server.config, unsafe.Pointer(config))
- }
-
- func (server *Server) ChannelRegistrationEnabled() bool {
- return server.Config().Channels.Registration.Enabled
- }
-
- func (server *Server) GetOperator(name string) (oper *Oper) {
- name, err := CasefoldName(name)
- if err != nil {
- return
- }
- return server.Config().operators[name]
- }
-
- func (server *Server) Languages() (lm *languages.Manager) {
- return server.Config().languageManager
- }
-
- func (client *Client) Sessions() (sessions []*Session) {
- client.stateMutex.RLock()
- sessions = client.sessions
- client.stateMutex.RUnlock()
- return
- }
-
- func (client *Client) GetSessionByResumeID(resumeID string) (result *Session) {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
-
- for _, session := range client.sessions {
- if session.resumeID == resumeID {
- return session
- }
- }
- return
- }
-
- type SessionData struct {
- ctime time.Time
- atime time.Time
- ip net.IP
- hostname string
- certfp string
- }
-
- func (client *Client) AllSessionData(currentSession *Session) (data []SessionData, currentIndex int) {
- currentIndex = -1
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
-
- data = make([]SessionData, len(client.sessions))
- for i, session := range client.sessions {
- if session == currentSession {
- currentIndex = i
- }
- data[i] = SessionData{
- atime: session.lastActive,
- ctime: session.ctime,
- hostname: session.rawHostname,
- certfp: session.certfp,
- }
- if session.proxiedIP != nil {
- data[i].ip = session.proxiedIP
- } else {
- data[i].ip = session.realIP
- }
- }
- return
- }
-
- func (client *Client) AddSession(session *Session) (success bool, numSessions int, lastSeen time.Time) {
- client.stateMutex.Lock()
- defer client.stateMutex.Unlock()
-
- // client may be dying and ineligible to receive another session
- if client.destroyed {
- return
- }
- // success, attach the new session to the client
- session.client = client
- newSessions := make([]*Session, len(client.sessions)+1)
- copy(newSessions, client.sessions)
- newSessions[len(newSessions)-1] = session
- if client.accountSettings.AutoreplayMissed {
- lastSeen = client.lastSeen
- }
- client.sessions = newSessions
- return true, len(client.sessions), lastSeen
- }
-
- func (client *Client) removeSession(session *Session) (success bool, length int) {
- if len(client.sessions) == 0 {
- return
- }
- sessions := make([]*Session, 0, len(client.sessions)-1)
- for _, currentSession := range client.sessions {
- if session == currentSession {
- success = true
- } else {
- sessions = append(sessions, currentSession)
- }
- }
- client.sessions = sessions
- length = len(sessions)
- return
- }
-
- func (session *Session) SetResumeID(resumeID string) {
- session.client.stateMutex.Lock()
- session.resumeID = resumeID
- session.client.stateMutex.Unlock()
- }
-
- func (client *Client) Nick() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.nick
- }
-
- func (client *Client) NickMaskString() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.nickMaskString
- }
-
- func (client *Client) NickCasefolded() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.nickCasefolded
- }
-
- func (client *Client) NickMaskCasefolded() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.nickMaskCasefolded
- }
-
- func (client *Client) Username() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.username
- }
-
- func (client *Client) Hostname() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.hostname
- }
-
- func (client *Client) Away() (result bool) {
- client.stateMutex.Lock()
- result = client.away
- client.stateMutex.Unlock()
- return
- }
-
- func (client *Client) SetAway(away bool, awayMessage string) (changed bool) {
- client.stateMutex.Lock()
- changed = away != client.away
- client.away = away
- client.awayMessage = awayMessage
- client.stateMutex.Unlock()
- return
- }
-
- func (client *Client) SetExitedSnomaskSent() {
- client.stateMutex.Lock()
- client.exitedSnomaskSent = true
- client.stateMutex.Unlock()
- }
-
- func (client *Client) AlwaysOn() (alwaysOn bool) {
- client.stateMutex.Lock()
- alwaysOn = client.alwaysOn
- client.stateMutex.Unlock()
- return
- }
-
- // uniqueIdentifiers returns the strings for which the server enforces per-client
- // uniqueness/ownership; no two clients can have colliding casefolded nicks or
- // skeletons.
- func (client *Client) uniqueIdentifiers() (nickCasefolded string, skeleton string) {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.nickCasefolded, client.skeleton
- }
-
- func (client *Client) ResumeID() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.resumeID
- }
-
- func (client *Client) SetResumeID(id string) {
- client.stateMutex.Lock()
- defer client.stateMutex.Unlock()
- client.resumeID = id
- }
-
- func (client *Client) Oper() *Oper {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.oper
- }
-
- func (client *Client) Registered() bool {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.registered
- }
-
- func (client *Client) SetRegistered() {
- // `registered` is only written from the client's own goroutine, but may be
- // read from other goroutines; therefore, the client's own goroutine may read
- // the value without synchronization, but must write it with synchronization,
- // and other goroutines must read it with synchronization
- client.stateMutex.Lock()
- client.registered = true
- client.stateMutex.Unlock()
- }
-
- func (client *Client) RawHostname() (result string) {
- client.stateMutex.Lock()
- result = client.rawHostname
- client.stateMutex.Unlock()
- return
- }
-
- func (client *Client) AwayMessage() (result string) {
- client.stateMutex.RLock()
- result = client.awayMessage
- client.stateMutex.RUnlock()
- return
- }
-
- func (client *Client) SetAwayMessage(message string) {
- client.stateMutex.Lock()
- client.awayMessage = message
- client.stateMutex.Unlock()
- }
-
- func (client *Client) Account() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.account
- }
-
- func (client *Client) AccountName() string {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.accountName
- }
-
- func (client *Client) Login(account ClientAccount) {
- alwaysOn := persistenceEnabled(client.server.Config().Accounts.Multiclient.AlwaysOn, account.Settings.AlwaysOn)
- client.stateMutex.Lock()
- defer client.stateMutex.Unlock()
- client.account = account.NameCasefolded
- client.accountName = account.Name
- client.accountSettings = account.Settings
- // check `registered` to avoid incorrectly marking a temporary (pre-reattach),
- // SASL'ing client as always-on
- if client.registered {
- client.alwaysOn = alwaysOn
- }
- client.accountRegDate = account.RegisteredAt
- return
- }
-
- func (client *Client) historyCutoff() (cutoff time.Time) {
- client.stateMutex.Lock()
- if client.account != "" {
- cutoff = client.accountRegDate
- } else {
- cutoff = client.ctime
- }
- client.stateMutex.Unlock()
- return
- }
-
- func (client *Client) Logout() {
- client.stateMutex.Lock()
- client.account = ""
- client.accountName = ""
- client.alwaysOn = false
- client.accountRegDate = time.Time{}
- client.accountSettings = AccountSettings{}
- client.stateMutex.Unlock()
- }
-
- func (client *Client) AccountSettings() (result AccountSettings) {
- client.stateMutex.RLock()
- result = client.accountSettings
- client.stateMutex.RUnlock()
- return
- }
-
- func (client *Client) SetAccountSettings(settings AccountSettings) {
- // we mark dirty if the client is transitioning to always-on
- markDirty := false
- alwaysOn := persistenceEnabled(client.server.Config().Accounts.Multiclient.AlwaysOn, settings.AlwaysOn)
- client.stateMutex.Lock()
- client.accountSettings = settings
- if client.registered {
- markDirty = !client.alwaysOn && alwaysOn
- client.alwaysOn = alwaysOn
- }
- client.stateMutex.Unlock()
- if markDirty {
- client.markDirty(IncludeAllAttrs)
- }
- }
-
- func (client *Client) Languages() (languages []string) {
- client.stateMutex.RLock()
- languages = client.languages
- client.stateMutex.RUnlock()
- return languages
- }
-
- func (client *Client) SetLanguages(languages []string) {
- client.stateMutex.Lock()
- client.languages = languages
- client.stateMutex.Unlock()
- }
-
- func (client *Client) HasMode(mode modes.Mode) bool {
- // client.flags has its own synch
- return client.modes.HasMode(mode)
- }
-
- func (client *Client) SetMode(mode modes.Mode, on bool) bool {
- return client.modes.SetMode(mode, on)
- }
-
- func (client *Client) SetRealname(realname string) {
- client.stateMutex.Lock()
- client.realname = realname
- client.stateMutex.Unlock()
- }
-
- func (client *Client) Channels() (result []*Channel) {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- length := len(client.channels)
- result = make([]*Channel, length)
- i := 0
- for channel := range client.channels {
- result[i] = channel
- i++
- }
- return
- }
-
- func (client *Client) NumChannels() int {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return len(client.channels)
- }
-
- func (client *Client) WhoWas() (result WhoWas) {
- return client.Details().WhoWas
- }
-
- func (client *Client) Details() (result ClientDetails) {
- client.stateMutex.RLock()
- defer client.stateMutex.RUnlock()
- return client.detailsNoMutex()
- }
-
- func (client *Client) detailsNoMutex() (result ClientDetails) {
- result.nick = client.nick
- result.nickCasefolded = client.nickCasefolded
- result.username = client.username
- result.hostname = client.hostname
- result.realname = client.realname
- result.nickMask = client.nickMaskString
- result.nickMaskCasefolded = client.nickMaskCasefolded
- result.account = client.account
- result.accountName = client.accountName
- return
- }
-
- func (channel *Channel) Name() string {
- channel.stateMutex.RLock()
- defer channel.stateMutex.RUnlock()
- return channel.name
- }
-
- func (channel *Channel) NameCasefolded() string {
- channel.stateMutex.RLock()
- defer channel.stateMutex.RUnlock()
- return channel.nameCasefolded
- }
-
- func (channel *Channel) Rename(name, nameCasefolded string) {
- channel.stateMutex.Lock()
- channel.name = name
- channel.nameCasefolded = nameCasefolded
- if channel.registeredFounder != "" {
- channel.registeredTime = time.Now().UTC()
- }
- channel.stateMutex.Unlock()
- }
-
- func (channel *Channel) Members() (result []*Client) {
- channel.stateMutex.RLock()
- defer channel.stateMutex.RUnlock()
- return channel.membersCache
- }
-
- func (channel *Channel) setUserLimit(limit int) {
- channel.stateMutex.Lock()
- channel.userLimit = limit
- channel.stateMutex.Unlock()
- }
-
- func (channel *Channel) setKey(key string) {
- channel.stateMutex.Lock()
- defer channel.stateMutex.Unlock()
- channel.key = key
- }
-
- func (channel *Channel) Founder() string {
- channel.stateMutex.RLock()
- defer channel.stateMutex.RUnlock()
- return channel.registeredFounder
- }
-
- func (channel *Channel) HighestUserMode(client *Client) (result modes.Mode) {
- channel.stateMutex.RLock()
- clientModes := channel.members[client]
- channel.stateMutex.RUnlock()
- return clientModes.HighestChannelUserMode()
- }
-
- func (channel *Channel) Settings() (result ChannelSettings) {
- channel.stateMutex.RLock()
- result = channel.settings
- channel.stateMutex.RUnlock()
- return result
- }
-
- func (channel *Channel) SetSettings(settings ChannelSettings) {
- channel.stateMutex.Lock()
- channel.settings = settings
- channel.stateMutex.Unlock()
- channel.MarkDirty(IncludeSettings)
- }
|