2016-06-15 21:50:56 +10:00
// Copyright (c) 2012-2014 Jeremy Latt
// Copyright (c) 2014-2015 Edmund Huber
2017-03-27 22:15:02 +10:00
// Copyright (c) 2016-2017 Daniel Oaks <daniel@danieloaks.net>
2016-06-15 21:50:56 +10:00
// released under the MIT license
2012-04-07 11:44:59 -07:00
package irc
import (
2012-04-17 22:11:35 -07:00
"fmt"
2012-04-07 11:44:59 -07:00
"net"
2016-10-16 20:35:50 +10:00
"runtime/debug"
2016-06-30 15:35:34 +10:00
"strconv"
2017-01-20 23:51:36 +10:00
"strings"
2017-04-18 22:26:01 +10:00
"sync"
2017-10-22 19:50:16 -04:00
"sync/atomic"
2012-12-11 23:12:35 -08:00
"time"
2016-06-17 22:17:42 +10:00
2017-06-15 10:14:19 -06:00
"github.com/goshuirc/irc-go/ircfmt"
"github.com/goshuirc/irc-go/ircmsg"
2017-06-14 12:00:53 -06:00
ident "github.com/oragono/go-ident"
2017-09-29 12:07:52 +10:00
"github.com/oragono/oragono/irc/caps"
2019-01-01 16:45:37 -05:00
"github.com/oragono/oragono/irc/connection_limits"
2018-11-26 05:23:27 -05:00
"github.com/oragono/oragono/irc/history"
2018-02-03 20:21:32 +10:00
"github.com/oragono/oragono/irc/modes"
2017-06-14 12:00:53 -06:00
"github.com/oragono/oragono/irc/sno"
2017-10-05 23:47:43 +10:00
"github.com/oragono/oragono/irc/utils"
2012-04-07 11:44:59 -07:00
)
2014-03-12 17:52:25 -07:00
const (
2017-06-19 14:53:16 -06:00
// IdentTimeoutSeconds is how many seconds before our ident (username) check times out.
2018-11-26 05:23:27 -05:00
IdentTimeoutSeconds = 1.5
2020-02-18 19:38:42 -05:00
IRCv3TimestampFormat = utils . IRCv3TimestampFormat
2016-06-30 15:35:34 +10:00
)
2018-11-26 05:23:27 -05:00
// ResumeDetails is a place to stash data at various stages of
// the resume process: when handling the RESUME command itself,
// when completing the registration, and when rejoining channels.
type ResumeDetails struct {
PresentedToken string
Timestamp time . Time
HistoryIncomplete bool
}
2016-10-23 11:48:57 +10:00
// Client is an IRC client.
2012-04-07 11:44:59 -07:00
type Client struct {
2018-02-11 05:30:40 -05:00
account string
2019-01-01 13:00:16 -05:00
accountName string // display name of the account: uncasefolded, '*' if not logged in
2020-02-18 19:38:42 -05:00
accountRegDate time . Time
2019-05-19 04:27:44 -04:00
accountSettings AccountSettings
2019-04-28 15:10:03 -04:00
away bool
2016-10-11 23:51:46 +10:00
awayMessage string
2019-05-21 21:40:25 -04:00
brbTimer BrbTimer
2016-10-11 23:51:46 +10:00
channels ChannelSet
ctime time . Time
2019-05-27 05:52:40 -04:00
destroyed bool
2017-06-11 10:01:39 -06:00
exitedSnomaskSent bool
2020-02-18 19:38:42 -05:00
modes modes . ModeSet
2016-10-11 23:51:46 +10:00
hostname string
2018-12-23 13:25:02 -05:00
invitedTo map [ string ] bool
2019-08-27 00:51:09 -04:00
isSTSOnly bool
2018-01-21 16:49:17 +10:00
languages [ ] string
2020-02-27 02:13:31 -05:00
lastActive time . Time // last time they sent a command that wasn't PONG or similar
lastSeen time . Time // last time they sent any kind of command
2019-01-01 16:45:37 -05:00
loginThrottle connection_limits . GenericThrottle
2016-10-11 23:51:46 +10:00
nick string
nickCasefolded string
nickMaskCasefolded string
2017-04-18 22:26:01 +10:00
nickMaskString string // cache for nickmask string since it's used with lots of replies
2019-03-08 03:12:21 -05:00
nickTimer NickTimer
2018-04-19 02:48:19 -04:00
oper * Oper
2018-02-26 21:44:03 -05:00
preregNick string
2018-02-01 15:53:49 -05:00
proxiedIP net . IP // actual remote IP if using the PROXY protocol
2017-04-18 22:26:01 +10:00
rawHostname string
2019-05-12 02:17:57 -04:00
cloakedHostname string
2016-10-11 23:51:46 +10:00
realname string
2019-02-05 00:19:03 -05:00
realIP net . IP
2016-10-11 23:51:46 +10:00
registered bool
2019-02-12 00:27:57 -05:00
resumeID string
2016-10-11 23:51:46 +10:00
server * Server
2019-01-30 18:59:49 -05:00
skeleton string
2019-04-12 00:08:46 -04:00
sessions [ ] * Session
2017-11-22 04:41:11 -05:00
stateMutex sync . RWMutex // tier 1
2020-02-18 19:38:42 -05:00
alwaysOn bool
2016-10-11 23:51:46 +10:00
username string
2017-04-18 22:26:01 +10:00
vhost string
2019-05-21 21:40:25 -04:00
history history . Buffer
2020-02-18 19:38:42 -05:00
dirtyBits uint
writerSemaphore utils . Semaphore // tier 1.5
2012-12-16 19:13:53 -08:00
}
2020-02-18 21:42:27 -05:00
type saslStatus struct {
mechanism string
value string
}
func ( s * saslStatus ) Clear ( ) {
* s = saslStatus { }
}
2019-04-12 00:08:46 -04:00
// Session is an individual client connection to the server (TCP connection
// and associated per-connection data, such as capabilities). There is a
// many-one relationship between sessions and clients.
type Session struct {
client * Client
2020-02-27 02:13:31 -05:00
ctime time . Time
lastActive time . Time
2019-05-08 04:11:54 -04:00
socket * Socket
realIP net . IP
proxiedIP net . IP
rawHostname string
2020-02-18 19:38:42 -05:00
isTor bool
2019-05-08 04:11:54 -04:00
2019-04-12 00:08:46 -04:00
idletimer IdleTimer
fakelag Fakelag
2019-05-27 04:18:07 -04:00
destroyed uint32
2019-04-12 00:08:46 -04:00
2020-02-18 21:42:27 -05:00
certfp string
sasl saslStatus
sentPassCommand bool
2019-12-23 15:26:37 -05:00
batchCounter uint32
2019-04-12 00:08:46 -04:00
quitMessage string
capabilities caps . Set
capState caps . State
capVersion caps . Version
2019-05-21 21:40:25 -04:00
2019-05-22 16:15:59 -04:00
registrationMessages int
2020-02-27 02:13:31 -05:00
resumeID string
resumeDetails * ResumeDetails
zncPlaybackTimes * zncPlaybackTimes
autoreplayMissedSince time . Time
2019-12-23 15:26:37 -05:00
batch MultilineBatch
}
// MultilineBatch tracks the state of a client-to-server multiline batch.
type MultilineBatch struct {
label string // this is the first param to BATCH (the "reference tag")
command string
target string
responseLabel string // this is the value of the labeled-response tag sent with BATCH
message utils . SplitMessage
tags map [ string ] string
2019-04-12 00:08:46 -04:00
}
// sets the session quit message, if there isn't one already
func ( sd * Session ) SetQuitMessage ( message string ) ( set bool ) {
if message == "" {
2019-05-22 20:29:19 -04:00
message = "Connection closed"
}
if sd . quitMessage == "" {
2019-04-12 00:08:46 -04:00
sd . quitMessage = message
return true
2019-05-22 20:29:19 -04:00
} else {
return false
2019-04-12 00:08:46 -04:00
}
}
2020-02-18 19:38:42 -05:00
func ( s * Session ) IP ( ) net . IP {
if s . proxiedIP != nil {
return s . proxiedIP
}
return s . realIP
}
2019-05-27 04:18:07 -04:00
// returns whether the session was actively destroyed (for example, by ping
// timeout or NS GHOST).
// avoids a race condition between asynchronous idle-timing-out of sessions,
// and a condition that allows implicit BRB on connection errors (since
// destroy()'s socket.Close() appears to socket.Read() as a connection error)
func ( session * Session ) Destroyed ( ) bool {
return atomic . LoadUint32 ( & session . destroyed ) == 1
}
// sets the timed-out flag
func ( session * Session ) SetDestroyed ( ) {
atomic . StoreUint32 ( & session . destroyed , 1 )
}
2019-05-29 19:23:46 -04:00
// returns whether the client supports a smart history replay cap,
// and therefore autoreplay-on-join and similar should be suppressed
func ( session * Session ) HasHistoryCaps ( ) bool {
2020-02-18 19:38:42 -05:00
return session . capabilities . Has ( caps . Chathistory ) || session . capabilities . Has ( caps . ZNCPlayback )
2019-05-29 19:23:46 -04:00
}
2019-12-23 15:26:37 -05:00
// generates a batch ID. the uniqueness requirements for this are fairly weak:
// any two batch IDs that are active concurrently (either through interleaving
// or nesting) on an individual session connection need to be unique.
// this allows ~4 billion such batches which should be fine.
func ( session * Session ) generateBatchID ( ) string {
id := atomic . AddUint32 ( & session . batchCounter , 1 )
return strconv . Itoa ( int ( id ) )
}
2019-01-01 13:00:16 -05:00
// WhoWas is the subset of client details needed to answer a WHOWAS query
type WhoWas struct {
nick string
nickCasefolded string
username string
hostname string
realname string
}
// ClientDetails is a standard set of details about a client
type ClientDetails struct {
WhoWas
nickMask string
nickMaskCasefolded string
account string
accountName string
}
2019-05-12 20:57:34 -04:00
// RunClient sets up a new client and runs its goroutine.
2019-11-20 17:14:42 -05:00
func ( server * Server ) RunClient ( conn clientConn , proxyLine string ) {
2019-05-12 20:57:34 -04:00
var isBanned bool
var banMsg string
var realIP net . IP
2019-11-20 17:43:40 -05:00
if conn . Config . Tor {
2019-05-12 20:57:34 -04:00
realIP = utils . IPv4LoopbackAddress
isBanned , banMsg = server . checkTorLimits ( )
} else {
realIP = utils . AddrToIP ( conn . Conn . RemoteAddr ( ) )
2020-01-09 04:38:59 -05:00
// skip the ban check for k8s-style proxy-before-TLS
if proxyLine == "" {
isBanned , banMsg = server . checkBans ( realIP )
}
2019-05-12 20:57:34 -04:00
}
if isBanned {
// this might not show up properly on some clients,
// but our objective here is just to close the connection out before it has a load impact on us
conn . Conn . Write ( [ ] byte ( fmt . Sprintf ( errorMsg , banMsg ) ) )
conn . Conn . Close ( )
return
}
server . logger . Info ( "localconnect-ip" , fmt . Sprintf ( "Client connecting from %v" , realIP ) )
2019-05-12 03:12:50 -04:00
now := time . Now ( ) . UTC ( )
2018-07-16 03:46:40 -04:00
config := server . Config ( )
2019-03-07 02:31:46 -05:00
// give them 1k of grace over the limit:
2020-01-18 23:47:05 -05:00
socket := NewSocket ( conn . Conn , ircmsg . MaxlenTagsFromClient + 512 + 1024 , config . Server . MaxSendQBytes )
2012-12-09 12:51:50 -08:00
client := & Client {
2020-02-27 02:13:31 -05:00
lastSeen : now ,
lastActive : now ,
channels : make ( ChannelSet ) ,
ctime : now ,
isSTSOnly : conn . Config . STSOnly ,
languages : server . Languages ( ) . Default ( ) ,
2019-01-01 16:45:37 -05:00
loginThrottle : connection_limits . GenericThrottle {
Duration : config . Accounts . LoginThrottling . Duration ,
Limit : config . Accounts . LoginThrottling . MaxAttempts ,
} ,
2016-09-19 22:30:29 +10:00
server : server ,
2019-01-01 13:00:16 -05:00
accountName : "*" ,
2016-10-11 23:51:46 +10:00
nick : "*" , // * is used until actual nick is given
nickCasefolded : "*" ,
2016-09-19 22:30:29 +10:00
nickMaskString : "*" , // * is used until actual nick is given
2012-12-09 12:51:50 -08:00
}
2020-03-02 01:32:08 -05:00
client . writerSemaphore . Initialize ( 1 )
2019-05-19 16:34:52 -04:00
client . history . Initialize ( config . History . ClientLength , config . History . AutoresizeWindow )
2019-05-21 21:40:25 -04:00
client . brbTimer . Initialize ( client )
2019-04-12 00:08:46 -04:00
session := & Session {
client : client ,
socket : socket ,
capVersion : caps . Cap301 ,
capState : caps . NoneState ,
2019-05-08 04:11:54 -04:00
ctime : now ,
2020-02-27 02:13:31 -05:00
lastActive : now ,
2019-05-12 20:57:34 -04:00
realIP : realIP ,
2020-02-18 19:38:42 -05:00
isTor : conn . Config . Tor ,
2019-04-12 00:08:46 -04:00
}
client . sessions = [ ] * Session { session }
2016-09-07 21:32:58 +10:00
2019-06-17 22:21:37 -04:00
if conn . Config . TLSConfig != nil {
2019-02-25 21:50:43 -05:00
client . SetMode ( modes . TLS , true )
2016-09-07 21:32:58 +10:00
// error is not useful to us here anyways so we can ignore it
2020-02-18 21:42:27 -05:00
session . certfp , _ = socket . CertFP ( )
2016-06-29 01:09:07 +10:00
}
2019-02-25 21:50:43 -05:00
2019-11-20 17:43:40 -05:00
if conn . Config . Tor {
2019-02-25 21:50:43 -05:00
client . SetMode ( modes . TLS , true )
2019-05-08 04:11:54 -04:00
// cover up details of the tor proxying infrastructure (not a user privacy concern,
// but a hardening measure):
session . proxiedIP = utils . IPv4LoopbackAddress
2019-12-17 15:10:23 -05:00
client . proxiedIP = session . proxiedIP
2019-05-08 04:11:54 -04:00
session . rawHostname = config . Server . TorListeners . Vhost
2019-12-17 15:10:23 -05:00
client . rawHostname = session . rawHostname
2019-02-25 21:50:43 -05:00
} else {
2019-05-12 20:57:34 -04:00
remoteAddr := conn . Conn . RemoteAddr ( )
2020-02-18 19:38:42 -05:00
if realIP . IsLoopback ( ) || utils . IPInNets ( realIP , config . Server . secureNets ) {
2019-05-08 04:11:54 -04:00
// treat local connections as secure (may be overridden later by WEBIRC)
client . SetMode ( modes . TLS , true )
}
2019-02-25 21:50:43 -05:00
if config . Server . CheckIdent && ! utils . AddrIsUnix ( remoteAddr ) {
client . doIdentLookup ( conn . Conn )
2016-06-30 19:28:34 +10:00
}
2019-02-25 21:50:43 -05:00
}
2019-05-08 04:11:54 -04:00
client . realIP = session . realIP
2019-02-25 21:50:43 -05:00
2019-07-01 09:21:38 -04:00
server . stats . Add ( )
2019-11-20 17:14:42 -05:00
client . run ( session , proxyLine )
2019-02-25 21:50:43 -05:00
}
2020-03-02 03:06:57 -05:00
func ( server * Server ) AddAlwaysOnClient ( account ClientAccount , chnames [ ] string , lastSeen time . Time ) {
2020-02-18 19:38:42 -05:00
now := time . Now ( ) . UTC ( )
config := server . Config ( )
2020-03-02 03:06:57 -05:00
if lastSeen . IsZero ( ) {
lastSeen = now
2020-02-27 02:13:31 -05:00
}
2020-02-18 19:38:42 -05:00
client := & Client {
2020-03-02 03:06:57 -05:00
lastSeen : lastSeen ,
lastActive : now ,
2020-02-27 02:13:31 -05:00
channels : make ( ChannelSet ) ,
ctime : now ,
languages : server . Languages ( ) . Default ( ) ,
server : server ,
2020-02-18 19:38:42 -05:00
// TODO figure out how to set these on reattach?
username : "~user" ,
rawHostname : server . name ,
realIP : utils . IPv4LoopbackAddress ,
2020-02-27 02:13:31 -05:00
alwaysOn : true ,
2020-02-18 19:38:42 -05:00
}
client . SetMode ( modes . TLS , true )
client . writerSemaphore . Initialize ( 1 )
client . history . Initialize ( 0 , 0 )
client . brbTimer . Initialize ( client )
server . accounts . Login ( client , account )
client . resizeHistory ( config )
_ , err := server . clients . SetNick ( client , nil , account . Name )
if err != nil {
server . logger . Error ( "internal" , "could not establish always-on client" , account . Name , err . Error ( ) )
return
} else {
server . logger . Debug ( "accounts" , "established always-on client" , account . Name )
}
// XXX set this last to avoid confusing SetNick:
client . registered = true
for _ , chname := range chnames {
// XXX we're using isSajoin=true, to make these joins succeed even without channel key
// this is *probably* ok as long as the persisted memberships are accurate
server . channels . Join ( client , chname , "" , true , nil )
}
}
func ( client * Client ) resizeHistory ( config * Config ) {
2020-02-24 14:09:00 -05:00
status , _ := client . historyStatus ( config )
if status == HistoryEphemeral {
2020-02-18 19:38:42 -05:00
client . history . Resize ( config . History . ClientLength , config . History . AutoresizeWindow )
} else {
client . history . Resize ( 0 , 0 )
}
}
2019-12-17 15:10:23 -05:00
// resolve an IP to an IRC-ready hostname, using reverse DNS, forward-confirming if necessary,
// and sending appropriate notices to the client
func ( client * Client ) lookupHostname ( session * Session , overwrite bool ) {
2020-02-18 19:38:42 -05:00
if session . isTor {
2019-12-17 15:10:23 -05:00
return
} // else: even if cloaking is enabled, look up the real hostname to show to operators
config := client . server . Config ( )
ip := session . realIP
if session . proxiedIP != nil {
ip = session . proxiedIP
}
ipString := ip . String ( )
var hostname , candidate string
if config . Server . lookupHostnames {
session . Notice ( "*** Looking up your hostname..." )
names , err := net . LookupAddr ( ipString )
if err == nil && 0 < len ( names ) {
candidate = strings . TrimSuffix ( names [ 0 ] , "." )
}
if utils . IsHostname ( candidate ) {
if config . Server . ForwardConfirmHostnames {
addrs , err := net . LookupHost ( candidate )
if err == nil {
for _ , addr := range addrs {
if addr == ipString {
hostname = candidate // successful forward confirmation
break
}
}
}
} else {
hostname = candidate
}
}
}
if hostname != "" {
session . Notice ( "*** Found your hostname" )
} else {
if config . Server . lookupHostnames {
session . Notice ( "*** Couldn't look up your hostname" )
}
hostname = utils . IPStringToHostname ( ipString )
}
session . rawHostname = hostname
2019-12-17 19:57:23 -05:00
cloakedHostname := config . Server . Cloaks . ComputeCloak ( ip )
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
2019-12-17 15:10:23 -05:00
// update the hostname if this is a new connection or a resume, but not if it's a reattach
if overwrite || client . rawHostname == "" {
client . rawHostname = hostname
client . cloakedHostname = cloakedHostname
client . updateNickMaskNoMutex ( )
}
}
2019-02-25 21:50:43 -05:00
func ( client * Client ) doIdentLookup ( conn net . Conn ) {
_ , serverPortString , err := net . SplitHostPort ( conn . LocalAddr ( ) . String ( ) )
if err != nil {
client . server . logger . Error ( "internal" , "bad server address" , err . Error ( ) )
return
}
serverPort , _ := strconv . Atoi ( serverPortString )
clientHost , clientPortString , err := net . SplitHostPort ( conn . RemoteAddr ( ) . String ( ) )
if err != nil {
client . server . logger . Error ( "internal" , "bad client address" , err . Error ( ) )
return
}
clientPort , _ := strconv . Atoi ( clientPortString )
2016-06-30 19:28:34 +10:00
2019-02-25 21:50:43 -05:00
client . Notice ( client . t ( "*** Looking up your username" ) )
resp , err := ident . Query ( clientHost , serverPort , clientPort , IdentTimeoutSeconds )
if err == nil {
err := client . SetNames ( resp . Identifier , "" , true )
2016-06-30 19:28:34 +10:00
if err == nil {
2019-02-25 21:50:43 -05:00
client . Notice ( client . t ( "*** Found your username" ) )
// we don't need to updateNickMask here since nickMask is not used for anything yet
2016-06-30 19:28:34 +10:00
} else {
2019-02-25 21:50:43 -05:00
client . Notice ( client . t ( "*** Got a malformed username, ignoring" ) )
2016-06-30 19:28:34 +10:00
}
2019-02-25 21:50:43 -05:00
} else {
client . Notice ( client . t ( "*** Could not find your username" ) )
2016-06-30 19:28:34 +10:00
}
2012-04-07 11:44:59 -07:00
}
2019-05-22 20:25:57 -04:00
type AuthOutcome uint
const (
authSuccess AuthOutcome = iota
authFailPass
authFailTorSaslRequired
authFailSaslRequired
)
2020-02-18 21:42:27 -05:00
func ( client * Client ) isAuthorized ( config * Config , session * Session ) AuthOutcome {
2019-02-05 00:19:03 -05:00
saslSent := client . account != ""
2019-02-25 21:50:43 -05:00
// PASS requirement
2020-02-18 21:42:27 -05:00
if ( config . Server . passwordBytes != nil ) && ! session . sentPassCommand && ! ( config . Accounts . SkipServerPassword && saslSent ) {
2019-05-22 20:25:57 -04:00
return authFailPass
2019-02-05 00:19:03 -05:00
}
2019-02-25 21:50:43 -05:00
// Tor connections may be required to authenticate with SASL
2020-02-18 21:42:27 -05:00
if session . isTor && config . Server . TorListeners . RequireSasl && ! saslSent {
2019-05-22 20:25:57 -04:00
return authFailTorSaslRequired
2019-02-25 21:50:43 -05:00
}
// finally, enforce require-sasl
2020-02-18 21:42:27 -05:00
if config . Accounts . RequireSasl . Enabled && ! saslSent && ! utils . IPInNets ( session . IP ( ) , config . Accounts . RequireSasl . exemptedNets ) {
2019-05-22 20:25:57 -04:00
return authFailSaslRequired
}
return authSuccess
2019-02-05 00:19:03 -05:00
}
2019-04-12 00:08:46 -04:00
func ( session * Session ) resetFakelag ( ) {
var flc FakelagConfig = session . client . server . Config ( ) . Fakelag
flc . Enabled = flc . Enabled && ! session . client . HasRoleCapabs ( "nofakelag" )
session . fakelag . Initialize ( flc )
2018-03-22 11:04:21 -04:00
}
2017-05-24 00:58:36 -06:00
// IP returns the IP address of this client.
func ( client * Client ) IP ( ) net . IP {
2019-02-05 00:19:03 -05:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2018-02-01 15:53:49 -05:00
if client . proxiedIP != nil {
return client . proxiedIP
2017-09-11 01:04:08 -04:00
}
2019-02-05 00:19:03 -05:00
return client . realIP
2017-05-24 00:58:36 -06:00
}
2017-06-23 05:15:10 +10:00
// IPString returns the IP address of this client as a string.
func ( client * Client ) IPString ( ) string {
ip := client . IP ( ) . String ( )
if 0 < len ( ip ) && ip [ 0 ] == ':' {
ip = "0" + ip
}
return ip
}
2019-07-01 09:21:38 -04:00
// t returns the translated version of the given string, based on the languages configured by the client.
func ( client * Client ) t ( originalString string ) string {
languageManager := client . server . Config ( ) . languageManager
if ! languageManager . Enabled ( ) {
return originalString
}
return languageManager . Translate ( client . Languages ( ) , originalString )
}
2019-11-20 17:14:42 -05:00
// main client goroutine: read lines and execute the corresponding commands
// `proxyLine` is the PROXY-before-TLS line, if there was one
func ( client * Client ) run ( session * Session , proxyLine string ) {
2014-04-15 08:49:52 -07:00
2017-10-23 18:38:32 -04:00
defer func ( ) {
2017-10-26 05:15:55 -04:00
if r := recover ( ) ; r != nil {
client . server . logger . Error ( "internal" ,
fmt . Sprintf ( "Client caused panic: %v\n%s" , r , debug . Stack ( ) ) )
2019-05-22 19:07:12 -04:00
if client . server . Config ( ) . Debug . recoverFromErrors {
2017-10-26 05:15:55 -04:00
client . server . logger . Error ( "internal" , "Disconnecting client and attempting to recover" )
} else {
panic ( r )
2017-10-26 04:19:01 -04:00
}
2017-10-23 18:38:32 -04:00
}
// ensure client connection gets closed
2019-05-21 21:40:25 -04:00
client . destroy ( session )
2017-10-23 18:38:32 -04:00
} ( )
2019-04-12 00:08:46 -04:00
session . idletimer . Initialize ( session )
session . resetFakelag ( )
2017-10-15 12:24:28 -04:00
2019-04-12 00:08:46 -04:00
isReattach := client . Registered ( )
2019-05-08 18:14:49 -04:00
if isReattach {
2019-05-21 21:40:25 -04:00
if session . resumeDetails != nil {
session . playResume ( )
session . resumeDetails = nil
client . brbTimer . Disable ( )
2019-07-04 06:59:08 -04:00
client . SetAway ( false , "" ) // clear BRB message if any
2019-05-21 21:40:25 -04:00
} else {
client . playReattachMessages ( session )
}
2019-05-08 18:14:49 -04:00
} else {
// don't reset the nick timer during a reattach
2019-04-12 00:08:46 -04:00
client . nickTimer . Initialize ( client )
}
2018-03-22 11:04:21 -04:00
2019-05-27 04:18:07 -04:00
firstLine := ! isReattach
2018-09-03 00:19:10 -04:00
2016-06-17 22:17:42 +10:00
for {
2019-11-20 17:14:42 -05:00
var line string
var err error
if proxyLine == "" {
line , err = session . socket . Read ( )
} else {
line = proxyLine // pretend we're just now receiving the proxy-before-TLS line
proxyLine = ""
}
2016-06-17 22:17:42 +10:00
if err != nil {
2018-03-17 21:32:12 -04:00
quitMessage := "connection closed"
if err == errReadQ {
quitMessage = "readQ exceeded"
}
2019-04-12 00:08:46 -04:00
client . Quit ( quitMessage , session )
2019-05-21 21:40:25 -04:00
// since the client did not actually send us a QUIT,
2019-05-27 04:18:07 -04:00
// give them a chance to resume if applicable:
if ! session . Destroyed ( ) {
client . brbTimer . Enable ( )
}
2016-06-17 22:17:42 +10:00
break
}
2018-10-28 15:44:13 -04:00
if client . server . logger . IsLoggingRawIO ( ) {
client . server . logger . Debug ( "userinput" , client . nick , "<- " , line )
}
2018-09-03 00:19:10 -04:00
// special-cased handling of PROXY protocol, see `handleProxyCommand` for details:
2019-05-27 04:18:07 -04:00
if firstLine {
2018-09-03 00:19:10 -04:00
firstLine = false
if strings . HasPrefix ( line , "PROXY" ) {
2019-04-12 00:08:46 -04:00
err = handleProxyCommand ( client . server , client , session , line )
2018-09-03 00:19:10 -04:00
if err != nil {
break
} else {
continue
}
}
}
2017-03-06 22:11:10 +10:00
2019-05-22 18:35:24 -04:00
if client . registered {
session . fakelag . Touch ( )
} else {
// DoS hardening, #505
2019-05-22 16:15:59 -04:00
session . registrationMessages ++
if client . server . Config ( ) . Limits . RegistrationMessages < session . registrationMessages {
client . Send ( nil , client . server . name , ERR_UNKNOWNERROR , "*" , client . t ( "You have sent too many registration messages" ) )
break
}
}
2020-01-18 23:47:05 -05:00
msg , err := ircmsg . ParseLineStrict ( line , true , 512 )
2017-01-19 07:56:33 +10:00
if err == ircmsg . ErrorLineIsEmpty {
continue
2019-03-07 02:31:46 -05:00
} else if err == ircmsg . ErrorLineTooLong {
2019-05-15 22:30:35 -04:00
session . Send ( nil , client . server . name , ERR_INPUTTOOLONG , client . Nick ( ) , client . t ( "Input line too long" ) )
2019-03-07 02:31:46 -05:00
continue
2017-01-19 07:56:33 +10:00
} else if err != nil {
2019-04-12 00:08:46 -04:00
client . Quit ( client . t ( "Received malformed line" ) , session )
2016-06-17 22:17:42 +10:00
break
2014-02-23 22:21:39 -08:00
}
2014-04-15 08:49:52 -07:00
2019-12-23 15:26:37 -05:00
// "Clients MUST NOT send messages other than PRIVMSG while a multiline batch is open."
// in future we might want to whitelist some commands that are allowed here, like PONG
if session . batch . label != "" && msg . Command != "BATCH" {
_ , batchTag := msg . GetTag ( "batch" )
if batchTag != session . batch . label {
if msg . Command != "NOTICE" {
session . Send ( nil , client . server . name , "FAIL" , "BATCH" , "MULTILINE_INVALID" , client . t ( "Incorrect batch tag sent" ) )
}
session . batch = MultilineBatch { }
continue
}
}
2016-06-19 10:01:30 +10:00
cmd , exists := Commands [ msg . Command ]
if ! exists {
2016-11-04 21:38:47 +10:00
if len ( msg . Command ) > 0 {
2019-05-15 22:30:35 -04:00
session . Send ( nil , client . server . name , ERR_UNKNOWNCOMMAND , client . Nick ( ) , msg . Command , client . t ( "Unknown command" ) )
2016-11-04 21:38:47 +10:00
} else {
2019-05-15 22:30:35 -04:00
session . Send ( nil , client . server . name , ERR_UNKNOWNCOMMAND , client . Nick ( ) , "lastcmd" , client . t ( "No command given" ) )
2016-11-04 21:38:47 +10:00
}
2016-06-20 22:53:45 +10:00
continue
2016-06-19 10:01:30 +10:00
}
2019-04-12 00:08:46 -04:00
isExiting := cmd . Run ( client . server , client , session , msg )
if isExiting {
break
} else if session . client != client {
// bouncer reattach
2019-11-20 17:14:42 -05:00
go session . client . run ( session , "" )
2016-06-17 22:17:42 +10:00
break
}
2014-02-23 22:21:39 -08:00
}
2014-04-15 08:49:52 -07:00
}
2019-05-08 18:14:49 -04:00
func ( client * Client ) playReattachMessages ( session * Session ) {
client . server . playRegistrationBurst ( session )
2020-02-18 19:38:42 -05:00
hasHistoryCaps := session . HasHistoryCaps ( )
2019-04-12 00:08:46 -04:00
for _ , channel := range session . client . Channels ( ) {
channel . playJoinForSession ( session )
2020-02-18 19:38:42 -05:00
// clients should receive autoreplay-on-join lines, if applicable:
if hasHistoryCaps {
continue
}
2019-05-29 19:23:46 -04:00
// if they negotiated znc.in/playback or chathistory, they will receive nothing,
// because those caps disable autoreplay-on-join and they haven't sent the relevant
// *playback PRIVMSG or CHATHISTORY command yet
rb := NewResponseBuffer ( session )
channel . autoReplayHistory ( client , rb , "" )
rb . Send ( true )
2019-04-12 00:08:46 -04:00
}
2020-02-27 02:13:31 -05:00
if ! session . autoreplayMissedSince . IsZero ( ) && ! hasHistoryCaps {
2020-02-18 19:38:42 -05:00
rb := NewResponseBuffer ( session )
2020-02-27 19:07:49 -05:00
zncPlayPrivmsgs ( client , rb , "*" , time . Now ( ) . UTC ( ) , session . autoreplayMissedSince )
2020-02-18 19:38:42 -05:00
rb . Send ( true )
}
2020-02-27 02:13:31 -05:00
session . autoreplayMissedSince = time . Time { }
2019-04-12 00:08:46 -04:00
}
2016-06-17 22:17:42 +10:00
//
2017-05-09 20:37:48 +10:00
// idle, quit, timers and timeouts
2016-06-17 22:17:42 +10:00
//
2014-04-15 08:49:52 -07:00
2020-02-27 02:13:31 -05:00
// Touch indicates that we received a line from the client (so the connection is healthy
// at this time, modulo network latency and fakelag). `active` means not a PING or suchlike
// (i.e. the user should be sitting in front of their client).
func ( client * Client ) Touch ( active bool , session * Session ) {
2019-05-12 03:12:50 -04:00
now := time . Now ( ) . UTC ( )
2017-12-02 20:05:06 -05:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
2020-02-27 02:13:31 -05:00
client . lastSeen = now
if active {
client . lastActive = now
session . lastActive = now
}
2014-02-18 13:25:21 -08:00
}
2014-02-13 18:39:33 -08:00
2017-10-15 12:24:28 -04:00
// Ping sends the client a PING message.
2019-04-12 00:08:46 -04:00
func ( session * Session ) Ping ( ) {
session . Send ( nil , "" , "PING" , session . client . Nick ( ) )
2017-05-09 20:37:48 +10:00
}
2019-02-09 20:01:47 -05:00
// tryResume tries to resume if the client asked us to.
2019-05-21 21:40:25 -04:00
func ( session * Session ) tryResume ( ) ( success bool ) {
var oldResumeID string
2018-01-21 11:23:33 +10:00
2019-02-09 20:01:47 -05:00
defer func ( ) {
2019-05-21 21:40:25 -04:00
if success {
// "On a successful request, the server [...] terminates the old client's connection"
oldSession := session . client . GetSessionByResumeID ( oldResumeID )
if oldSession != nil {
session . client . destroy ( oldSession )
}
} else {
session . resumeDetails = nil
2019-02-09 20:01:47 -05:00
}
} ( )
2019-05-21 21:40:25 -04:00
client := session . client
server := client . server
config := server . Config ( )
2018-01-21 11:23:33 +10:00
2019-05-21 21:40:25 -04:00
oldClient , oldResumeID := server . resumeManager . VerifyToken ( client , session . resumeDetails . PresentedToken )
2018-11-26 05:23:27 -05:00
if oldClient == nil {
2019-05-21 21:40:25 -04:00
session . Send ( nil , server . name , "FAIL" , "RESUME" , "INVALID_TOKEN" , client . t ( "Cannot resume connection, token is not valid" ) )
2018-01-21 12:23:47 +10:00
return
}
2018-11-26 05:23:27 -05:00
resumeAllowed := config . Server . AllowPlaintextResume || ( oldClient . HasMode ( modes . TLS ) && client . HasMode ( modes . TLS ) )
if ! resumeAllowed {
2019-05-21 21:40:25 -04:00
session . Send ( nil , server . name , "FAIL" , "RESUME" , "INSECURE_SESSION" , client . t ( "Cannot resume connection, old and new clients must have TLS" ) )
2018-01-21 11:23:33 +10:00
return
}
2019-05-21 21:40:25 -04:00
err := server . clients . Resume ( oldClient , session )
2018-11-26 05:23:27 -05:00
if err != nil {
2019-05-21 21:40:25 -04:00
session . Send ( nil , server . name , "FAIL" , "RESUME" , "CANNOT_RESUME" , client . t ( "Cannot resume connection" ) )
2018-01-21 11:23:33 +10:00
return
}
2019-02-09 20:01:47 -05:00
success = true
2019-05-21 21:40:25 -04:00
client . server . logger . Debug ( "quit" , fmt . Sprintf ( "%s is being resumed" , oldClient . Nick ( ) ) )
2019-02-09 20:01:47 -05:00
2019-05-21 21:40:25 -04:00
return
}
2018-01-21 11:23:33 +10:00
2019-05-21 21:40:25 -04:00
// playResume is called from the session's fresh goroutine after a resume;
// it sends notifications to friends, then plays the registration burst and replays
// stored history to the session
func ( session * Session ) playResume ( ) {
client := session . client
server := client . server
2020-02-18 19:38:42 -05:00
config := server . Config ( )
2018-11-26 05:23:27 -05:00
friends := make ( ClientSet )
2020-02-18 19:38:42 -05:00
var oldestLostMessage time . Time
2018-11-26 05:23:27 -05:00
// work out how much time, if any, is not covered by history buffers
2020-02-18 19:38:42 -05:00
// assume that a persistent buffer covers the whole resume period
2019-05-21 21:40:25 -04:00
for _ , channel := range client . Channels ( ) {
2018-11-26 05:23:27 -05:00
for _ , member := range channel . Members ( ) {
friends . Add ( member )
2020-02-18 19:38:42 -05:00
}
2020-02-24 14:09:00 -05:00
status , _ := channel . historyStatus ( config )
if status == HistoryEphemeral {
2018-11-26 05:23:27 -05:00
lastDiscarded := channel . history . LastDiscarded ( )
2020-02-18 19:38:42 -05:00
if oldestLostMessage . Before ( lastDiscarded ) {
2018-11-26 05:23:27 -05:00
oldestLostMessage = lastDiscarded
}
}
}
2020-02-24 14:09:00 -05:00
cHistoryStatus , _ := client . historyStatus ( config )
if cHistoryStatus == HistoryEphemeral {
2020-02-18 19:38:42 -05:00
lastDiscarded := client . history . LastDiscarded ( )
if oldestLostMessage . Before ( lastDiscarded ) {
oldestLostMessage = lastDiscarded
}
2018-11-26 05:23:27 -05:00
}
2020-02-18 19:38:42 -05:00
_ , privmsgSeq , _ := server . GetHistorySequence ( nil , client , "*" )
if privmsgSeq != nil {
privmsgs , _ , _ := privmsgSeq . Between ( history . Selector { } , history . Selector { } , config . History . ClientLength )
for _ , item := range privmsgs {
sender := server . clients . Get ( stripMaskFromNick ( item . Nick ) )
if sender != nil {
friends . Add ( sender )
}
2018-11-26 05:23:27 -05:00
}
}
2019-05-21 21:40:25 -04:00
timestamp := session . resumeDetails . Timestamp
2020-02-18 19:38:42 -05:00
gap := oldestLostMessage . Sub ( timestamp )
2019-05-29 07:34:23 -04:00
session . resumeDetails . HistoryIncomplete = gap > 0 || timestamp . IsZero ( )
2018-11-26 05:23:27 -05:00
gapSeconds := int ( gap . Seconds ( ) ) + 1 // round up to avoid confusion
2019-05-21 21:40:25 -04:00
details := client . Details ( )
oldNickmask := details . nickMask
2019-12-17 15:10:23 -05:00
client . lookupHostname ( session , true )
2019-05-21 21:40:25 -04:00
hostname := client . Hostname ( ) // may be a vhost
2019-05-29 07:34:23 -04:00
timestampString := timestamp . Format ( IRCv3TimestampFormat )
2019-05-21 21:40:25 -04:00
2018-11-26 05:23:27 -05:00
// send quit/resume messages to friends
for friend := range friends {
2019-05-21 21:40:25 -04:00
if friend == client {
continue
}
for _ , fSession := range friend . Sessions ( ) {
if fSession . capabilities . Has ( caps . Resume ) {
2019-05-29 07:34:23 -04:00
if ! session . resumeDetails . HistoryIncomplete {
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname , "ok" )
} else if session . resumeDetails . HistoryIncomplete && ! timestamp . IsZero ( ) {
2019-05-21 21:40:25 -04:00
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname , timestampString )
2019-05-27 04:40:24 -04:00
} else {
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname )
2019-04-12 00:08:46 -04:00
}
2018-11-26 05:23:27 -05:00
} else {
2019-05-29 07:34:23 -04:00
if ! session . resumeDetails . HistoryIncomplete {
2019-12-16 19:50:15 -05:00
fSession . Send ( nil , oldNickmask , "QUIT" , friend . t ( "Client reconnected" ) )
2019-05-29 07:34:23 -04:00
} else if session . resumeDetails . HistoryIncomplete && ! timestamp . IsZero ( ) {
fSession . Send ( nil , oldNickmask , "QUIT" , fmt . Sprintf ( friend . t ( "Client reconnected (up to %d seconds of message history lost)" ) , gapSeconds ) )
} else {
2019-12-16 19:50:15 -05:00
fSession . Send ( nil , oldNickmask , "QUIT" , friend . t ( "Client reconnected (message history may have been lost)" ) )
2019-04-12 00:08:46 -04:00
}
2018-11-26 05:23:27 -05:00
}
2018-01-21 11:23:33 +10:00
}
}
2020-02-18 19:38:42 -05:00
if session . resumeDetails . HistoryIncomplete {
if ! timestamp . IsZero ( ) {
session . Send ( nil , client . server . name , "WARN" , "RESUME" , "HISTORY_LOST" , fmt . Sprintf ( client . t ( "Resume may have lost up to %d seconds of history" ) , gapSeconds ) )
} else {
session . Send ( nil , client . server . name , "WARN" , "RESUME" , "HISTORY_LOST" , client . t ( "Resume may have lost some message history" ) )
}
2018-11-26 05:23:27 -05:00
}
2018-01-21 11:23:33 +10:00
2019-05-22 15:08:02 -04:00
session . Send ( nil , client . server . name , "RESUME" , "SUCCESS" , details . nick )
2018-04-24 03:11:11 -04:00
2019-05-21 21:40:25 -04:00
server . playRegistrationBurst ( session )
2018-01-21 11:23:33 +10:00
2019-05-21 21:40:25 -04:00
for _ , channel := range client . Channels ( ) {
channel . Resume ( session , timestamp )
2018-11-26 05:23:27 -05:00
}
2018-01-22 20:55:20 +10:00
2018-11-26 05:23:27 -05:00
// replay direct PRIVSMG history
2020-02-18 19:38:42 -05:00
if ! timestamp . IsZero ( ) && privmsgSeq != nil {
after := history . Selector { Time : timestamp }
items , complete , _ := privmsgSeq . Between ( after , history . Selector { } , config . History . ZNCMax )
2020-02-20 23:47:13 -05:00
if len ( items ) != 0 {
rb := NewResponseBuffer ( session )
client . replayPrivmsgHistory ( rb , items , "" , complete )
rb . Send ( true )
}
2018-04-24 03:11:11 -04:00
}
2018-01-21 11:23:33 +10:00
2019-05-21 21:40:25 -04:00
session . resumeDetails = nil
2018-11-26 05:23:27 -05:00
}
2020-02-18 19:38:42 -05:00
func ( client * Client ) replayPrivmsgHistory ( rb * ResponseBuffer , items [ ] history . Item , target string , complete bool ) {
2019-05-06 23:17:57 -04:00
var batchID string
2019-05-19 02:14:36 -04:00
details := client . Details ( )
nick := details . nick
2020-02-20 23:47:13 -05:00
if target == "" {
target = nick
2019-05-06 23:17:57 -04:00
}
2020-02-20 23:47:13 -05:00
batchID = rb . StartNestedHistoryBatch ( target )
2019-05-06 23:17:57 -04:00
allowTags := rb . session . capabilities . Has ( caps . MessageTags )
2019-02-04 05:18:17 -05:00
for _ , item := range items {
var command string
switch item . Type {
case history . Privmsg :
command = "PRIVMSG"
case history . Notice :
command = "NOTICE"
2019-05-06 23:17:57 -04:00
case history . Tagmsg :
if allowTags {
command = "TAGMSG"
} else {
continue
}
2019-02-04 05:18:17 -05:00
default :
continue
}
2019-03-07 02:31:46 -05:00
var tags map [ string ] string
2019-05-06 23:17:57 -04:00
if allowTags {
tags = item . Tags
2019-02-04 05:18:17 -05:00
}
2020-02-20 01:45:17 -05:00
// XXX: Params[0] is the message target. if the source of this message is an in-memory
// buffer, then it's "" for an incoming message and the recipient's nick for an outgoing
// message. if the source of the message is mysql, then mysql only sees one copy of the
// message, and it's the version with the recipient's nick filled in. so this is an
// incoming message if Params[0] (the recipient's nick) equals the client's nick:
2020-02-18 19:38:42 -05:00
if item . Params [ 0 ] == "" || item . Params [ 0 ] == nick {
2019-05-19 02:14:36 -04:00
rb . AddSplitMessageFromClient ( item . Nick , item . AccountName , tags , command , nick , item . Message )
} else {
// this message was sent *from* the client to another nick; the target is item.Params[0]
2020-02-18 19:38:42 -05:00
// substitute client's current nickmask in case client changed nick
2019-05-19 02:14:36 -04:00
rb . AddSplitMessageFromClient ( details . nickMask , item . AccountName , tags , command , item . Params [ 0 ] , item . Message )
}
2019-02-04 05:18:17 -05:00
}
2019-05-06 23:17:57 -04:00
rb . EndNestedBatch ( batchID )
2019-02-04 05:18:17 -05:00
if ! complete {
rb . Add ( nil , "HistServ" , "NOTICE" , nick , client . t ( "Some additional message history may have been lost" ) )
}
}
2016-10-23 11:48:57 +10:00
// IdleTime returns how long this client's been idle.
2014-02-17 15:25:32 -08:00
func ( client * Client ) IdleTime ( ) time . Duration {
2017-12-02 20:05:06 -05:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2020-02-27 02:13:31 -05:00
return time . Since ( client . lastActive )
2014-02-17 15:25:32 -08:00
}
2016-10-23 11:48:57 +10:00
// SignonTime returns this client's signon time as a unix timestamp.
2014-02-17 19:56:06 -08:00
func ( client * Client ) SignonTime ( ) int64 {
return client . ctime . Unix ( )
}
2016-10-23 11:48:57 +10:00
// IdleSeconds returns the number of seconds this client's been idle.
2014-02-17 19:08:57 -08:00
func ( client * Client ) IdleSeconds ( ) uint64 {
return uint64 ( client . IdleTime ( ) . Seconds ( ) )
}
2016-10-23 11:48:57 +10:00
// HasNick returns true if the client's nickname is set (used in registration).
2014-02-08 17:10:04 -08:00
func ( client * Client ) HasNick ( ) bool {
2017-11-22 04:41:11 -05:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2016-10-11 23:51:46 +10:00
return client . nick != "" && client . nick != "*"
2012-12-16 19:13:53 -08:00
}
2017-04-16 11:31:33 +10:00
// HasUsername returns true if the client's username is set (used in registration).
2014-02-08 17:10:04 -08:00
func ( client * Client ) HasUsername ( ) bool {
2017-11-22 04:41:11 -05:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2016-10-11 23:51:46 +10:00
return client . username != "" && client . username != "*"
2014-02-08 17:10:04 -08:00
}
2019-02-03 18:49:42 +10:00
// SetNames sets the client's ident and realname.
2019-02-05 02:40:49 -05:00
func ( client * Client ) SetNames ( username , realname string , fromIdent bool ) error {
limit := client . server . Config ( ) . Limits . IdentLen
if ! fromIdent {
limit -= 1 // leave room for the prepended ~
}
2019-02-05 03:04:52 -05:00
if limit < len ( username ) {
2019-02-05 02:40:49 -05:00
username = username [ : limit ]
}
2019-02-03 18:49:42 +10:00
if ! isIdent ( username ) {
2018-11-26 05:23:27 -05:00
return errInvalidUsername
}
2019-02-05 02:40:49 -05:00
if ! fromIdent {
username = "~" + username
}
2019-02-03 18:49:42 +10:00
2018-11-26 05:23:27 -05:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
if client . username == "" {
2019-02-05 02:40:49 -05:00
client . username = username
2018-11-26 05:23:27 -05:00
}
if client . realname == "" {
client . realname = realname
}
return nil
}
2017-09-29 12:11:06 +10:00
// HasRoleCapabs returns true if client has the given (role) capabilities.
func ( client * Client ) HasRoleCapabs ( capabs ... string ) bool {
2018-04-19 02:48:19 -04:00
oper := client . Oper ( )
if oper == nil {
2016-10-23 11:13:08 +10:00
return false
}
for _ , capab := range capabs {
2018-04-19 02:48:19 -04:00
if ! oper . Class . Capabilities [ capab ] {
2016-10-23 11:13:08 +10:00
return false
}
}
return true
}
2017-04-16 11:31:33 +10:00
// ModeString returns the mode string for this client.
func ( client * Client ) ModeString ( ) ( str string ) {
2020-02-18 19:38:42 -05:00
return "+" + client . modes . String ( )
2012-04-17 20:24:26 -07:00
}
2012-04-17 21:13:12 -07:00
2016-06-17 22:17:42 +10:00
// Friends refers to clients that share a channel with this client.
2019-04-12 00:08:46 -04:00
func ( client * Client ) Friends ( capabs ... caps . Capability ) ( result map [ * Session ] bool ) {
result = make ( map [ * Session ] bool )
2016-10-27 00:44:36 +10:00
2019-04-12 00:08:46 -04:00
// look at the client's own sessions
for _ , session := range client . Sessions ( ) {
if session . capabilities . HasAll ( capabs ... ) {
result [ session ] = true
2016-10-27 00:44:36 +10:00
}
}
2017-10-22 19:50:16 -04:00
for _ , channel := range client . Channels ( ) {
for _ , member := range channel . Members ( ) {
2019-04-12 00:08:46 -04:00
for _ , session := range member . Sessions ( ) {
if session . capabilities . HasAll ( capabs ... ) {
result [ session ] = true
2016-10-13 18:08:08 +10:00
}
}
2014-02-18 15:28:20 -08:00
}
2014-02-16 17:23:47 -08:00
}
2019-04-12 00:08:46 -04:00
return
2014-02-16 17:23:47 -08:00
}
2019-01-30 18:59:49 -05:00
func ( client * Client ) SetOper ( oper * Oper ) {
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
client . oper = oper
// operators typically get a vhost, update the nickmask
client . updateNickMaskNoMutex ( )
}
2018-04-19 02:48:19 -04:00
// XXX: CHGHOST requires prefix nickmask to have original hostname,
// this is annoying to do correctly
func ( client * Client ) sendChghost ( oldNickMask string , vhost string ) {
username := client . Username ( )
for fClient := range client . Friends ( caps . ChgHost ) {
2018-11-26 05:23:27 -05:00
fClient . sendFromClientInternal ( false , time . Time { } , "" , oldNickMask , client . AccountName ( ) , nil , "CHGHOST" , username , vhost )
2018-04-19 02:48:19 -04:00
}
}
// choose the correct vhost to display
func ( client * Client ) getVHostNoMutex ( ) string {
// hostserv vhost OR operclass vhost OR nothing (i.e., normal rdns hostmask)
if client . vhost != "" {
return client . vhost
} else if client . oper != nil {
return client . oper . Vhost
} else {
return ""
}
}
// SetVHost updates the client's hostserv-based vhost
func ( client * Client ) SetVHost ( vhost string ) ( updated bool ) {
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
updated = ( client . vhost != vhost )
client . vhost = vhost
if updated {
client . updateNickMaskNoMutex ( )
}
return
}
2017-10-04 00:57:03 -04:00
// updateNick updates `nick` and `nickCasefolded`.
2019-01-30 18:59:49 -05:00
func ( client * Client ) updateNick ( nick , nickCasefolded , skeleton string ) {
2017-10-04 00:57:03 -04:00
client . stateMutex . Lock ( )
2019-01-30 18:59:49 -05:00
defer client . stateMutex . Unlock ( )
2017-10-04 00:57:03 -04:00
client . nick = nick
2019-01-30 18:59:49 -05:00
client . nickCasefolded = nickCasefolded
client . skeleton = skeleton
client . updateNickMaskNoMutex ( )
2016-10-16 20:35:50 +10:00
}
2019-01-30 18:59:49 -05:00
// updateNickMaskNoMutex updates the casefolded nickname and nickmask, not acquiring any mutexes.
2018-01-22 20:55:20 +10:00
func ( client * Client ) updateNickMaskNoMutex ( ) {
2020-02-06 17:43:54 -05:00
if client . nick == "*" {
return // pre-registration, don't bother generating the hostname
}
2018-04-19 02:48:19 -04:00
client . hostname = client . getVHostNoMutex ( )
if client . hostname == "" {
2019-05-12 02:17:57 -04:00
client . hostname = client . cloakedHostname
if client . hostname == "" {
client . hostname = client . rawHostname
}
2016-10-23 11:28:31 +10:00
}
2019-12-18 07:01:26 -05:00
cfhostname := strings . ToLower ( client . hostname )
2019-01-28 13:36:15 -05:00
client . nickMaskString = fmt . Sprintf ( "%s!%s@%s" , client . nick , client . username , client . hostname )
2019-02-05 02:40:49 -05:00
client . nickMaskCasefolded = fmt . Sprintf ( "%s!%s@%s" , client . nickCasefolded , strings . ToLower ( client . username ) , cfhostname )
2016-06-19 15:37:29 +10:00
}
2017-01-11 22:38:16 +10:00
// AllNickmasks returns all the possible nickmasks for the client.
2019-01-28 23:03:30 -05:00
func ( client * Client ) AllNickmasks ( ) ( masks [ ] string ) {
2018-04-19 02:48:19 -04:00
client . stateMutex . RLock ( )
2019-01-28 23:03:30 -05:00
nick := client . nickCasefolded
2019-02-05 02:40:49 -05:00
username := client . username
2018-04-19 02:48:19 -04:00
rawHostname := client . rawHostname
2019-05-12 02:17:57 -04:00
cloakedHostname := client . cloakedHostname
2018-04-19 02:48:19 -04:00
vhost := client . getVHostNoMutex ( )
client . stateMutex . RUnlock ( )
2019-02-05 03:04:52 -05:00
username = strings . ToLower ( username )
2018-04-19 02:48:19 -04:00
if len ( vhost ) > 0 {
2019-12-18 07:01:26 -05:00
cfvhost := strings . ToLower ( vhost )
masks = append ( masks , fmt . Sprintf ( "%s!%s@%s" , nick , username , cfvhost ) )
2017-01-11 22:38:16 +10:00
}
2019-01-28 23:03:30 -05:00
var rawhostmask string
2019-12-18 07:01:26 -05:00
cfrawhost := strings . ToLower ( rawHostname )
rawhostmask = fmt . Sprintf ( "%s!%s@%s" , nick , username , cfrawhost )
masks = append ( masks , rawhostmask )
2017-01-11 22:38:16 +10:00
2019-05-12 02:17:57 -04:00
if cloakedHostname != "" {
masks = append ( masks , fmt . Sprintf ( "%s!%s@%s" , nick , username , cloakedHostname ) )
}
2017-01-11 22:38:16 +10:00
2019-01-28 23:03:30 -05:00
ipmask := fmt . Sprintf ( "%s!%s@%s" , nick , username , client . IPString ( ) )
if ipmask != rawhostmask {
masks = append ( masks , ipmask )
2017-01-11 22:38:16 +10:00
}
2019-01-28 23:03:30 -05:00
return
2017-01-11 22:38:16 +10:00
}
2017-09-28 15:49:01 +10:00
// LoggedIntoAccount returns true if this client is logged into an account.
func ( client * Client ) LoggedIntoAccount ( ) bool {
2018-02-11 05:30:40 -05:00
return client . Account ( ) != ""
2017-09-28 15:49:01 +10:00
}
2019-02-10 13:57:32 -05:00
// Quit sets the given quit message for the client.
// (You must ensure separately that destroy() is called, e.g., by returning `true` from
// the command handler or calling it yourself.)
2019-04-12 00:08:46 -04:00
func ( client * Client ) Quit ( message string , session * Session ) {
setFinalData := func ( sess * Session ) {
message := sess . quitMessage
var finalData [ ] byte
// #364: don't send QUIT lines to unregistered clients
if client . registered {
quitMsg := ircmsg . MakeMessage ( nil , client . nickMaskString , "QUIT" , message )
finalData , _ = quitMsg . LineBytesStrict ( false , 512 )
}
2017-10-10 20:49:29 -04:00
2019-04-12 00:08:46 -04:00
errorMsg := ircmsg . MakeMessage ( nil , "" , "ERROR" , message )
errorMsgBytes , _ := errorMsg . LineBytesStrict ( false , 512 )
finalData = append ( finalData , errorMsgBytes ... )
2017-10-10 20:49:29 -04:00
2019-04-12 00:08:46 -04:00
sess . socket . SetFinalData ( finalData )
2019-02-10 13:57:32 -05:00
}
2017-10-10 20:49:29 -04:00
2019-04-12 00:08:46 -04:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
var sessions [ ] * Session
if session != nil {
sessions = [ ] * Session { session }
} else {
sessions = client . sessions
}
2017-10-10 20:49:29 -04:00
2019-04-12 00:08:46 -04:00
for _ , session := range sessions {
if session . SetQuitMessage ( message ) {
setFinalData ( session )
}
}
2016-06-17 22:17:42 +10:00
}
2016-10-23 11:48:57 +10:00
// destroy gets rid of a client, removes them from server lists etc.
2019-04-12 00:08:46 -04:00
// if `session` is nil, destroys the client unconditionally, removing all sessions;
// otherwise, destroys one specific session, only destroying the client if it
// has no more sessions.
2019-05-21 21:40:25 -04:00
func ( client * Client ) destroy ( session * Session ) {
2019-04-12 00:08:46 -04:00
var sessionsToDestroy [ ] * Session
2018-11-26 05:23:27 -05:00
client . stateMutex . Lock ( )
2019-05-08 18:14:49 -04:00
details := client . detailsNoMutex ( )
2019-05-21 21:40:25 -04:00
brbState := client . brbTimer . state
2019-05-27 04:18:07 -04:00
brbAt := client . brbTimer . brbAt
2019-05-08 18:14:49 -04:00
wasReattach := session != nil && session . client != client
2019-04-12 00:08:46 -04:00
sessionRemoved := false
2020-02-18 19:38:42 -05:00
registered := client . registered
alwaysOn := client . alwaysOn
2020-02-27 02:13:31 -05:00
saveLastSeen := alwaysOn && client . accountSettings . AutoreplayMissed
2019-04-12 00:08:46 -04:00
var remainingSessions int
if session == nil {
sessionsToDestroy = client . sessions
client . sessions = nil
remainingSessions = 0
} else {
sessionRemoved , remainingSessions = client . removeSession ( session )
if sessionRemoved {
sessionsToDestroy = [ ] * Session { session }
}
}
2019-05-27 05:52:40 -04:00
// should we destroy the whole client this time?
2019-05-30 18:48:12 -04:00
// BRB is not respected if this is a destroy of the whole client (i.e., session == nil)
2020-03-06 04:21:21 -05:00
brbEligible := session != nil && brbState == BrbEnabled
shouldDestroy := ! client . destroyed && remainingSessions == 0 && ! brbEligible && ! alwaysOn
// decrement stats on a true destroy, or for the removal of the last connected session
// of an always-on client
shouldDecrement := shouldDestroy || ( alwaysOn && len ( sessionsToDestroy ) != 0 && len ( client . sessions ) == 0 )
2019-05-27 05:52:40 -04:00
if shouldDestroy {
// if it's our job to destroy it, don't let anyone else try
client . destroyed = true
}
2020-02-27 02:13:31 -05:00
if saveLastSeen {
client . dirtyBits |= IncludeLastSeen
2020-02-18 19:38:42 -05:00
}
2019-05-27 14:33:59 -04:00
exitedSnomaskSent := client . exitedSnomaskSent
2018-11-26 05:23:27 -05:00
client . stateMutex . Unlock ( )
2020-02-27 02:13:31 -05:00
// XXX there is no particular reason to persist this state here rather than
// any other place: it would be correct to persist it after every `Touch`. However,
// I'm not comfortable introducing that many database writes, and I don't want to
// design a throttle.
if saveLastSeen {
2020-02-20 02:33:49 -05:00
client . wakeWriter ( )
}
2019-05-08 18:14:49 -04:00
// destroy all applicable sessions:
var quitMessage string
2019-04-12 00:08:46 -04:00
for _ , session := range sessionsToDestroy {
if session . client != client {
// session has been attached to a new client; do not destroy it
continue
}
session . idletimer . Stop ( )
// send quit/error message to client if they haven't been sent already
client . Quit ( "" , session )
2019-05-08 18:14:49 -04:00
quitMessage = session . quitMessage
2019-05-27 04:18:07 -04:00
session . SetDestroyed ( )
2019-05-08 18:14:49 -04:00
session . socket . Close ( )
// remove from connection limits
var source string
2020-02-18 19:38:42 -05:00
if session . isTor {
2019-05-08 18:14:49 -04:00
client . server . torLimiter . RemoveClient ( )
source = "tor"
} else {
ip := session . realIP
if session . proxiedIP != nil {
ip = session . proxiedIP
}
client . server . connectionLimiter . RemoveClient ( ip )
source = ip . String ( )
}
client . server . logger . Info ( "localconnect-ip" , fmt . Sprintf ( "disconnecting session of %s from %s" , details . nick , source ) )
2019-04-12 00:08:46 -04:00
}
2020-02-18 19:38:42 -05:00
// decrement stats if we have no more sessions, even if the client will not be destroyed
2020-03-06 04:21:21 -05:00
if shouldDecrement {
2020-02-18 19:38:42 -05:00
invisible := client . HasMode ( modes . Invisible )
operator := client . HasMode ( modes . LocalOperator ) || client . HasMode ( modes . Operator )
client . server . stats . Remove ( registered , invisible , operator )
}
2019-05-21 21:40:25 -04:00
// do not destroy the client if it has either remaining sessions, or is BRB'ed
2019-05-27 05:52:40 -04:00
if ! shouldDestroy {
2018-01-22 20:55:20 +10:00
return
2014-02-18 13:25:21 -08:00
}
2014-02-19 18:46:46 -08:00
2020-02-18 19:38:42 -05:00
splitQuitMessage := utils . MakeMessage ( quitMessage )
quitItem := history . Item {
Type : history . Quit ,
Nick : details . nickMask ,
AccountName : details . accountName ,
Message : splitQuitMessage ,
}
var channels [ ] * Channel
2020-02-20 01:45:17 -05:00
// use a defer here to avoid writing to mysql while holding the destroy semaphore:
2020-02-18 19:38:42 -05:00
defer func ( ) {
for _ , channel := range channels {
channel . AddHistoryItem ( quitItem )
}
} ( )
2018-04-24 20:34:28 -04:00
// see #235: deduplicating the list of PART recipients uses (comparatively speaking)
// a lot of RAM, so limit concurrency to avoid thrashing
client . server . semaphores . ClientDestroy . Acquire ( )
defer client . server . semaphores . ClientDestroy . Release ( )
2019-05-21 21:40:25 -04:00
if ! wasReattach {
2019-05-06 23:17:57 -04:00
client . server . logger . Debug ( "quit" , fmt . Sprintf ( "%s is no longer on the server" , details . nick ) )
2018-01-21 11:23:33 +10:00
}
2017-03-06 22:11:10 +10:00
2019-05-21 21:40:25 -04:00
if registered {
2018-05-04 00:24:54 -04:00
client . server . whoWas . Append ( client . WhoWas ( ) )
2018-01-21 11:23:33 +10:00
}
2016-06-17 22:17:42 +10:00
2019-02-12 00:27:57 -05:00
client . server . resumeManager . Delete ( client )
2016-10-16 20:14:56 +10:00
// alert monitors
2019-05-08 18:14:49 -04:00
if registered {
client . server . monitorManager . AlertAbout ( client , false )
}
2017-10-04 00:57:03 -04:00
// clean up monitor state
2017-10-04 02:59:59 -04:00
client . server . monitorManager . RemoveAll ( client )
2016-10-16 20:14:56 +10:00
2016-06-17 22:17:42 +10:00
// clean up channels
2019-05-08 18:14:49 -04:00
// (note that if this is a reattach, client has no channels and therefore no friends)
2018-04-24 20:23:01 -04:00
friends := make ( ClientSet )
2020-02-18 19:38:42 -05:00
channels = client . Channels ( )
for _ , channel := range channels {
2019-05-21 21:40:25 -04:00
channel . Quit ( client )
2017-10-22 19:50:16 -04:00
for _ , member := range channel . Members ( ) {
friends . Add ( member )
}
2016-06-17 22:17:42 +10:00
}
2018-04-24 20:23:01 -04:00
friends . Remove ( client )
2016-06-17 22:17:42 +10:00
// clean up server
2019-05-21 21:40:25 -04:00
client . server . clients . Remove ( client )
2016-06-17 22:17:42 +10:00
// clean up self
2018-02-28 00:14:44 -05:00
client . nickTimer . Stop ( )
2019-05-21 21:40:25 -04:00
client . brbTimer . Disable ( )
2016-06-17 22:17:42 +10:00
2018-02-11 05:30:40 -05:00
client . server . accounts . Logout ( client )
2019-05-27 04:18:07 -04:00
// this happens under failure to return from BRB
if quitMessage == "" {
2019-10-05 23:50:11 -04:00
if brbState == BrbDead && ! brbAt . IsZero ( ) {
2019-05-27 04:18:07 -04:00
awayMessage := client . AwayMessage ( )
2019-07-04 06:59:08 -04:00
if awayMessage == "" {
awayMessage = "Disconnected" // auto-BRB
2019-05-27 04:18:07 -04:00
}
2019-07-04 06:59:08 -04:00
quitMessage = fmt . Sprintf ( "%s [%s ago]" , awayMessage , time . Since ( brbAt ) . Truncate ( time . Second ) . String ( ) )
2017-09-25 22:47:03 -04:00
}
2019-05-27 04:18:07 -04:00
}
if quitMessage == "" {
quitMessage = "Exited"
}
for friend := range friends {
2019-05-21 21:40:25 -04:00
friend . sendFromClientInternal ( false , splitQuitMessage . Time , splitQuitMessage . Msgid , details . nickMask , details . accountName , nil , "QUIT" , quitMessage )
2016-11-29 21:06:01 +10:00
}
2019-05-21 21:40:25 -04:00
2019-05-27 14:33:59 -04:00
if ! exitedSnomaskSent && registered {
2019-05-21 21:40:25 -04:00
client . server . snomasks . Send ( sno . LocalQuits , fmt . Sprintf ( ircfmt . Unescape ( "%s$r exited the network" ) , details . nick ) )
2017-06-11 10:01:39 -06:00
}
2016-06-19 10:01:30 +10:00
}
2014-02-18 13:25:21 -08:00
2017-01-14 15:28:50 +10:00
// SendSplitMsgFromClient sends an IRC PRIVMSG/NOTICE coming from a specific client.
// Adds account-tag to the line as well.
2019-05-13 00:39:59 -04:00
func ( session * Session ) sendSplitMsgFromClientInternal ( blocking bool , nickmask , accountName string , tags map [ string ] string , command , target string , message utils . SplitMessage ) {
2020-01-18 23:47:05 -05:00
if message . Is512 ( ) {
2019-05-13 00:39:59 -04:00
session . sendFromClientInternal ( blocking , message . Time , message . Msgid , nickmask , accountName , tags , command , target , message . Message )
2017-01-14 15:28:50 +10:00
} else {
2020-02-18 19:38:42 -05:00
if session . capabilities . Has ( caps . Multiline ) {
2019-12-23 15:26:37 -05:00
for _ , msg := range session . composeMultilineBatch ( nickmask , accountName , tags , command , target , message ) {
session . SendRawMessage ( msg , blocking )
}
} else {
2020-01-18 23:47:05 -05:00
for i , messagePair := range message . Split {
var msgid string
if i == 0 {
msgid = message . Msgid
}
session . sendFromClientInternal ( blocking , message . Time , msgid , nickmask , accountName , tags , command , target , messagePair . Message )
2019-12-23 15:26:37 -05:00
}
2017-01-14 15:28:50 +10:00
}
}
}
2019-05-06 23:17:57 -04:00
// Sends a line with `nickmask` as the prefix, adding `time` and `account` tags if supported
2019-04-12 00:08:46 -04:00
func ( client * Client ) sendFromClientInternal ( blocking bool , serverTime time . Time , msgid string , nickmask , accountName string , tags map [ string ] string , command string , params ... string ) ( err error ) {
for _ , session := range client . Sessions ( ) {
err_ := session . sendFromClientInternal ( blocking , serverTime , msgid , nickmask , accountName , tags , command , params ... )
if err_ != nil {
err = err_
}
}
return
}
func ( session * Session ) sendFromClientInternal ( blocking bool , serverTime time . Time , msgid string , nickmask , accountName string , tags map [ string ] string , command string , params ... string ) ( err error ) {
2019-03-07 02:31:46 -05:00
msg := ircmsg . MakeMessage ( tags , nickmask , command , params ... )
2016-09-12 11:25:31 +10:00
// attach account-tag
2019-04-12 00:08:46 -04:00
if session . capabilities . Has ( caps . AccountTag ) && accountName != "*" {
2019-03-07 02:31:46 -05:00
msg . SetTag ( "account" , accountName )
2016-09-12 11:25:31 +10:00
}
2017-01-14 19:52:47 +10:00
// attach message-id
2019-04-12 00:08:46 -04:00
if msgid != "" && session . capabilities . Has ( caps . MessageTags ) {
2019-05-15 01:30:21 -04:00
msg . SetTag ( "msgid" , msgid )
2019-03-07 02:31:46 -05:00
}
// attach server-time
2019-06-13 02:24:14 -04:00
session . setTimeTag ( & msg , serverTime )
2016-09-12 11:25:31 +10:00
2019-04-12 00:08:46 -04:00
return session . SendRawMessage ( msg , blocking )
2016-09-12 11:25:31 +10:00
}
2019-12-23 15:26:37 -05:00
func ( session * Session ) composeMultilineBatch ( fromNickMask , fromAccount string , tags map [ string ] string , command , target string , message utils . SplitMessage ) ( result [ ] ircmsg . IrcMessage ) {
batchID := session . generateBatchID ( )
2020-01-03 09:46:55 -05:00
batchStart := ircmsg . MakeMessage ( tags , fromNickMask , "BATCH" , "+" + batchID , caps . MultilineBatchType , target )
2019-12-23 15:26:37 -05:00
batchStart . SetTag ( "time" , message . Time . Format ( IRCv3TimestampFormat ) )
batchStart . SetTag ( "msgid" , message . Msgid )
if session . capabilities . Has ( caps . AccountTag ) && fromAccount != "*" {
batchStart . SetTag ( "account" , fromAccount )
}
result = append ( result , batchStart )
2020-01-18 23:47:05 -05:00
for _ , msg := range message . Split {
2019-12-23 15:26:37 -05:00
message := ircmsg . MakeMessage ( nil , fromNickMask , command , target , msg . Message )
message . SetTag ( "batch" , batchID )
if msg . Concat {
message . SetTag ( caps . MultilineConcatTag , "" )
}
result = append ( result , message )
}
result = append ( result , ircmsg . MakeMessage ( nil , fromNickMask , "BATCH" , "-" + batchID ) )
return
}
2017-01-21 00:07:10 +10:00
var (
2017-01-23 09:03:49 +10:00
// these are all the output commands that MUST have their last param be a trailing.
2017-10-08 11:05:05 +10:00
// this is needed because dumb clients like to treat trailing params separately from the
2017-01-23 09:03:49 +10:00
// other params in messages.
2017-01-21 00:07:10 +10:00
commandsThatMustUseTrailing = map [ string ] bool {
"PRIVMSG" : true ,
"NOTICE" : true ,
2017-01-23 09:03:49 +10:00
RPL_WHOISCHANNELS : true ,
2017-03-06 15:50:23 +10:00
RPL_USERHOST : true ,
2020-03-11 18:57:42 -04:00
// mirc's handling of RPL_NAMREPLY is broken:
// https://forums.mirc.com/ubbthreads.php/topics/266939/re-nick-list
RPL_NAMREPLY : true ,
2017-01-21 00:07:10 +10:00
}
)
2017-10-08 11:05:05 +10:00
// SendRawMessage sends a raw message to the client.
2019-04-12 00:08:46 -04:00
func ( session * Session ) SendRawMessage ( message ircmsg . IrcMessage , blocking bool ) error {
2017-10-08 11:05:05 +10:00
// use dumb hack to force the last param to be a trailing param if required
2019-05-09 14:18:30 -04:00
config := session . client . server . Config ( )
2020-03-11 18:57:25 -04:00
if config . Server . Compatibility . forceTrailing && commandsThatMustUseTrailing [ message . Command ] {
message . ForceTrailing ( )
2017-01-20 23:51:36 +10:00
}
2017-10-08 11:05:05 +10:00
// assemble message
2020-01-18 23:47:05 -05:00
line , err := message . LineBytesStrict ( false , 512 )
2016-06-19 10:01:30 +10:00
if err != nil {
2018-02-11 05:30:40 -05:00
logline := fmt . Sprintf ( "Error assembling message for sending: %v\n%s" , err , debug . Stack ( ) )
2019-04-12 00:08:46 -04:00
session . client . server . logger . Error ( "internal" , logline )
2016-11-04 21:38:47 +10:00
2019-04-12 00:08:46 -04:00
message = ircmsg . MakeMessage ( nil , session . client . server . name , ERR_UNKNOWNERROR , "*" , "Error assembling message for sending" )
2019-03-07 02:31:46 -05:00
line , _ := message . LineBytesStrict ( false , 0 )
2017-10-08 11:05:05 +10:00
2018-11-26 05:23:27 -05:00
if blocking {
2019-04-12 00:08:46 -04:00
session . socket . BlockingWrite ( line )
2018-11-26 05:23:27 -05:00
} else {
2019-04-12 00:08:46 -04:00
session . socket . Write ( line )
2018-11-26 05:23:27 -05:00
}
2016-06-19 10:01:30 +10:00
return err
2014-02-16 17:23:47 -08:00
}
2017-01-20 23:51:36 +10:00
2019-04-12 00:08:46 -04:00
if session . client . server . logger . IsLoggingRawIO ( ) {
2018-04-26 15:32:32 -04:00
logline := string ( line [ : len ( line ) - 2 ] ) // strip "\r\n"
2019-04-12 00:08:46 -04:00
session . client . server . logger . Debug ( "useroutput" , session . client . Nick ( ) , " ->" , logline )
2018-04-26 15:32:32 -04:00
}
2017-03-06 22:11:10 +10:00
2018-11-26 05:23:27 -05:00
if blocking {
2019-04-12 00:08:46 -04:00
return session . socket . BlockingWrite ( line )
2018-11-26 05:23:27 -05:00
} else {
2019-04-12 00:08:46 -04:00
return session . socket . Write ( line )
2018-11-26 05:23:27 -05:00
}
2017-10-08 11:05:05 +10:00
}
2018-11-26 05:23:27 -05:00
// Send sends an IRC line to the client.
2019-04-12 00:08:46 -04:00
func ( client * Client ) Send ( tags map [ string ] string , prefix string , command string , params ... string ) ( err error ) {
for _ , session := range client . Sessions ( ) {
err_ := session . Send ( tags , prefix , command , params ... )
if err_ != nil {
err = err_
}
}
return
}
func ( session * Session ) Send ( tags map [ string ] string , prefix string , command string , params ... string ) ( err error ) {
2019-03-07 02:31:46 -05:00
msg := ircmsg . MakeMessage ( tags , prefix , command , params ... )
2019-06-13 02:24:14 -04:00
session . setTimeTag ( & msg , time . Time { } )
return session . SendRawMessage ( msg , false )
}
func ( session * Session ) setTimeTag ( msg * ircmsg . IrcMessage , serverTime time . Time ) {
2019-04-12 00:08:46 -04:00
if session . capabilities . Has ( caps . ServerTime ) && ! msg . HasTag ( "time" ) {
2019-06-13 02:24:14 -04:00
if serverTime . IsZero ( ) {
serverTime = time . Now ( )
}
msg . SetTag ( "time" , serverTime . UTC ( ) . Format ( IRCv3TimestampFormat ) )
2019-03-07 02:31:46 -05:00
}
2018-11-26 05:23:27 -05:00
}
2016-06-19 10:01:30 +10:00
// Notice sends the client a notice from the server.
func ( client * Client ) Notice ( text string ) {
2019-04-12 00:08:46 -04:00
client . Send ( nil , client . server . name , "NOTICE" , client . Nick ( ) , text )
2014-02-16 17:23:47 -08:00
}
2017-10-22 19:50:16 -04:00
2019-12-17 15:10:23 -05:00
func ( session * Session ) Notice ( text string ) {
session . Send ( nil , session . client . server . name , "NOTICE" , session . client . Nick ( ) , text )
}
2020-03-02 01:22:00 -05:00
// `simulated` is for the fake join of an always-on client
// (we just read the channel name from the database, there's no need to write it back)
func ( client * Client ) addChannel ( channel * Channel , simulated bool ) {
2017-10-22 19:50:16 -04:00
client . stateMutex . Lock ( )
client . channels [ channel ] = true
2020-02-18 19:38:42 -05:00
alwaysOn := client . alwaysOn
2017-10-22 19:50:16 -04:00
client . stateMutex . Unlock ( )
2020-02-18 19:38:42 -05:00
2020-03-02 01:22:00 -05:00
if alwaysOn && ! simulated {
2020-02-18 19:38:42 -05:00
client . markDirty ( IncludeChannels )
}
2017-10-22 19:50:16 -04:00
}
func ( client * Client ) removeChannel ( channel * Channel ) {
client . stateMutex . Lock ( )
delete ( client . channels , channel )
2020-02-18 19:38:42 -05:00
alwaysOn := client . alwaysOn
2017-10-22 19:50:16 -04:00
client . stateMutex . Unlock ( )
2020-02-18 19:38:42 -05:00
if alwaysOn {
client . markDirty ( IncludeChannels )
}
2017-10-22 19:50:16 -04:00
}
2018-11-26 05:23:27 -05:00
2018-12-23 13:25:02 -05:00
// Records that the client has been invited to join an invite-only channel
func ( client * Client ) Invite ( casefoldedChannel string ) {
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
if client . invitedTo == nil {
client . invitedTo = make ( map [ string ] bool )
}
client . invitedTo [ casefoldedChannel ] = true
}
// Checks that the client was invited to join a given channel
func ( client * Client ) CheckInvited ( casefoldedChannel string ) ( invited bool ) {
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
invited = client . invitedTo [ casefoldedChannel ]
// joining an invited channel "uses up" your invite, so you can't rejoin on kick
delete ( client . invitedTo , casefoldedChannel )
return
}
2019-12-19 06:33:43 -05:00
// Implements auto-oper by certfp (scans for an auto-eligible operator block that matches
// the client's cert, then applies it).
func ( client * Client ) attemptAutoOper ( session * Session ) {
2020-02-18 21:42:27 -05:00
if session . certfp == "" || client . HasMode ( modes . Operator ) {
2019-12-19 06:33:43 -05:00
return
}
for _ , oper := range client . server . Config ( ) . operators {
2020-02-18 21:42:27 -05:00
if oper . Auto && oper . Pass == nil && oper . Fingerprint != "" && oper . Fingerprint == session . certfp {
2019-12-19 06:33:43 -05:00
rb := NewResponseBuffer ( session )
applyOper ( client , oper , rb )
rb . Send ( true )
2019-12-29 11:59:49 -05:00
return
2019-12-19 06:33:43 -05:00
}
}
}
2020-02-18 19:38:42 -05:00
2020-02-24 14:09:00 -05:00
func ( client * Client ) historyStatus ( config * Config ) ( status HistoryStatus , target string ) {
2020-02-18 19:38:42 -05:00
if ! config . History . Enabled {
2020-02-24 14:09:00 -05:00
return HistoryDisabled , ""
2020-02-18 19:38:42 -05:00
}
client . stateMutex . RLock ( )
2020-02-28 05:41:08 -05:00
target = client . account
2020-02-18 19:38:42 -05:00
historyStatus := client . accountSettings . DMHistory
client . stateMutex . RUnlock ( )
2020-02-28 05:41:08 -05:00
if target == "" {
2020-02-24 14:09:00 -05:00
return HistoryEphemeral , ""
2020-02-18 19:38:42 -05:00
}
2020-02-28 05:41:08 -05:00
status = historyEnabled ( config . History . Persistent . DirectMessages , historyStatus )
if status != HistoryPersistent {
target = ""
}
return
2020-02-18 19:38:42 -05:00
}
// these are bit flags indicating what part of the client status is "dirty"
// and needs to be read from memory and written to the db
const (
IncludeChannels uint = 1 << iota
2020-02-27 02:13:31 -05:00
IncludeLastSeen
2020-02-18 19:38:42 -05:00
)
func ( client * Client ) markDirty ( dirtyBits uint ) {
client . stateMutex . Lock ( )
alwaysOn := client . alwaysOn
client . dirtyBits = client . dirtyBits | dirtyBits
client . stateMutex . Unlock ( )
if alwaysOn {
client . wakeWriter ( )
}
}
func ( client * Client ) wakeWriter ( ) {
if client . writerSemaphore . TryAcquire ( ) {
go client . writeLoop ( )
}
}
func ( client * Client ) writeLoop ( ) {
for {
client . performWrite ( )
client . writerSemaphore . Release ( )
client . stateMutex . RLock ( )
isDirty := client . dirtyBits != 0
client . stateMutex . RUnlock ( )
if ! isDirty || ! client . writerSemaphore . TryAcquire ( ) {
return
}
}
}
func ( client * Client ) performWrite ( ) {
client . stateMutex . Lock ( )
2020-02-20 02:33:49 -05:00
dirtyBits := client . dirtyBits
2020-02-18 19:38:42 -05:00
client . dirtyBits = 0
account := client . account
2020-03-02 02:52:51 -05:00
lastSeen := client . lastSeen
2020-02-18 19:38:42 -05:00
client . stateMutex . Unlock ( )
if account == "" {
client . server . logger . Error ( "internal" , "attempting to persist logged-out client" , client . Nick ( ) )
return
}
2020-02-20 02:33:49 -05:00
if ( dirtyBits & IncludeChannels ) != 0 {
channels := client . Channels ( )
channelNames := make ( [ ] string , len ( channels ) )
for i , channel := range channels {
channelNames [ i ] = channel . Name ( )
}
client . server . accounts . saveChannels ( account , channelNames )
}
2020-02-27 02:13:31 -05:00
if ( dirtyBits & IncludeLastSeen ) != 0 {
client . server . accounts . saveLastSeen ( account , lastSeen )
2020-02-18 19:38:42 -05:00
}
}