2016-06-15 13:50:56 +02:00
// Copyright (c) 2012-2014 Jeremy Latt
// Copyright (c) 2014-2015 Edmund Huber
2017-03-27 14:15:02 +02:00
// Copyright (c) 2016-2017 Daniel Oaks <daniel@danieloaks.net>
2016-06-15 13:50:56 +02:00
// released under the MIT license
2012-04-07 20:44:59 +02:00
package irc
import (
2012-04-18 07:11:35 +02:00
"fmt"
2012-04-07 20:44:59 +02:00
"net"
2016-10-16 12:35:50 +02:00
"runtime/debug"
2016-06-30 07:35:34 +02:00
"strconv"
2017-01-20 14:51:36 +01:00
"strings"
2017-04-18 14:26:01 +02:00
"sync"
2017-10-23 01:50:16 +02:00
"sync/atomic"
2012-12-12 08:12:35 +01:00
"time"
2016-06-17 14:17:42 +02:00
2017-06-15 18:14:19 +02:00
"github.com/goshuirc/irc-go/ircfmt"
"github.com/goshuirc/irc-go/ircmsg"
2017-06-14 20:00:53 +02:00
ident "github.com/oragono/go-ident"
2017-09-29 04:07:52 +02:00
"github.com/oragono/oragono/irc/caps"
2019-01-01 22:45:37 +01:00
"github.com/oragono/oragono/irc/connection_limits"
2018-11-26 11:23:27 +01:00
"github.com/oragono/oragono/irc/history"
2018-02-03 11:21:32 +01:00
"github.com/oragono/oragono/irc/modes"
2017-06-14 20:00:53 +02:00
"github.com/oragono/oragono/irc/sno"
2017-10-05 15:47:43 +02:00
"github.com/oragono/oragono/irc/utils"
2012-04-07 20:44:59 +02:00
)
2014-03-13 01:52:25 +01:00
const (
2017-06-19 22:53:16 +02:00
// IdentTimeoutSeconds is how many seconds before our ident (username) check times out.
2018-11-26 11:23:27 +01:00
IdentTimeoutSeconds = 1.5
2020-02-19 01:38:42 +01:00
IRCv3TimestampFormat = utils . IRCv3TimestampFormat
2016-06-30 07:35:34 +02:00
)
2018-11-26 11:23:27 +01: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 03:48:57 +02:00
// Client is an IRC client.
2012-04-07 20:44:59 +02:00
type Client struct {
2018-02-11 11:30:40 +01:00
account string
2019-01-01 19:00:16 +01:00
accountName string // display name of the account: uncasefolded, '*' if not logged in
2020-02-19 01:38:42 +01:00
accountRegDate time . Time
2019-05-19 10:27:44 +02:00
accountSettings AccountSettings
2019-04-28 21:10:03 +02:00
away bool
2016-10-11 15:51:46 +02:00
awayMessage string
2019-05-22 03:40:25 +02:00
brbTimer BrbTimer
2016-10-11 15:51:46 +02:00
channels ChannelSet
ctime time . Time
2019-05-27 11:52:40 +02:00
destroyed bool
2017-06-11 18:01:39 +02:00
exitedSnomaskSent bool
2020-02-19 01:38:42 +01:00
modes modes . ModeSet
2016-10-11 15:51:46 +02:00
hostname string
2018-12-23 19:25:02 +01:00
invitedTo map [ string ] bool
2019-08-27 06:51:09 +02:00
isSTSOnly bool
2018-01-21 07:49:17 +01:00
languages [ ] string
2020-02-27 08:13:31 +01: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 22:45:37 +01:00
loginThrottle connection_limits . GenericThrottle
2016-10-11 15:51:46 +02:00
nick string
nickCasefolded string
nickMaskCasefolded string
2017-04-18 14:26:01 +02:00
nickMaskString string // cache for nickmask string since it's used with lots of replies
2019-03-08 09:12:21 +01:00
nickTimer NickTimer
2018-04-19 08:48:19 +02:00
oper * Oper
2018-02-27 03:44:03 +01:00
preregNick string
2018-02-01 21:53:49 +01:00
proxiedIP net . IP // actual remote IP if using the PROXY protocol
2017-04-18 14:26:01 +02:00
rawHostname string
2019-05-12 08:17:57 +02:00
cloakedHostname string
2016-10-11 15:51:46 +02:00
realname string
2019-02-05 06:19:03 +01:00
realIP net . IP
2016-10-11 15:51:46 +02:00
registered bool
2019-02-12 06:27:57 +01:00
resumeID string
2016-10-11 15:51:46 +02:00
server * Server
2019-01-31 00:59:49 +01:00
skeleton string
2019-04-12 06:08:46 +02:00
sessions [ ] * Session
2017-11-22 10:41:11 +01:00
stateMutex sync . RWMutex // tier 1
2020-02-19 01:38:42 +01:00
alwaysOn bool
2016-10-11 15:51:46 +02:00
username string
2017-04-18 14:26:01 +02:00
vhost string
2019-05-22 03:40:25 +02:00
history history . Buffer
2020-02-19 01:38:42 +01:00
dirtyBits uint
writerSemaphore utils . Semaphore // tier 1.5
2012-12-17 04:13:53 +01:00
}
2020-02-19 03:42:27 +01:00
type saslStatus struct {
mechanism string
value string
}
func ( s * saslStatus ) Clear ( ) {
* s = saslStatus { }
}
2019-04-12 06:08:46 +02: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 08:13:31 +01:00
ctime time . Time
lastActive time . Time
2019-05-08 10:11:54 +02:00
socket * Socket
realIP net . IP
proxiedIP net . IP
rawHostname string
2020-02-19 01:38:42 +01:00
isTor bool
2019-05-08 10:11:54 +02:00
2019-04-12 06:08:46 +02:00
idletimer IdleTimer
fakelag Fakelag
2019-05-27 10:18:07 +02:00
destroyed uint32
2019-04-12 06:08:46 +02:00
2020-02-19 03:42:27 +01:00
certfp string
sasl saslStatus
sentPassCommand bool
2019-12-23 21:26:37 +01:00
batchCounter uint32
2019-04-12 06:08:46 +02:00
quitMessage string
capabilities caps . Set
capState caps . State
capVersion caps . Version
2019-05-22 03:40:25 +02:00
2019-05-22 22:15:59 +02:00
registrationMessages int
2020-02-27 08:13:31 +01:00
resumeID string
resumeDetails * ResumeDetails
zncPlaybackTimes * zncPlaybackTimes
autoreplayMissedSince time . Time
2019-12-23 21:26:37 +01: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 06:08:46 +02:00
}
// sets the session quit message, if there isn't one already
func ( sd * Session ) SetQuitMessage ( message string ) ( set bool ) {
if message == "" {
2019-05-23 02:29:19 +02:00
message = "Connection closed"
}
if sd . quitMessage == "" {
2019-04-12 06:08:46 +02:00
sd . quitMessage = message
return true
2019-05-23 02:29:19 +02:00
} else {
return false
2019-04-12 06:08:46 +02:00
}
}
2020-02-19 01:38:42 +01:00
func ( s * Session ) IP ( ) net . IP {
if s . proxiedIP != nil {
return s . proxiedIP
}
return s . realIP
}
2019-05-27 10:18:07 +02: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-30 01:23:46 +02: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-19 01:38:42 +01:00
return session . capabilities . Has ( caps . Chathistory ) || session . capabilities . Has ( caps . ZNCPlayback )
2019-05-30 01:23:46 +02:00
}
2019-12-23 21:26:37 +01: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 19:00:16 +01: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-13 02:57:34 +02:00
// RunClient sets up a new client and runs its goroutine.
2019-11-20 23:14:42 +01:00
func ( server * Server ) RunClient ( conn clientConn , proxyLine string ) {
2019-05-13 02:57:34 +02:00
var isBanned bool
var banMsg string
var realIP net . IP
2019-11-20 23:43:40 +01:00
if conn . Config . Tor {
2019-05-13 02:57:34 +02:00
realIP = utils . IPv4LoopbackAddress
isBanned , banMsg = server . checkTorLimits ( )
} else {
realIP = utils . AddrToIP ( conn . Conn . RemoteAddr ( ) )
2020-01-09 10:38:59 +01:00
// skip the ban check for k8s-style proxy-before-TLS
if proxyLine == "" {
isBanned , banMsg = server . checkBans ( realIP )
}
2019-05-13 02:57:34 +02: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 09:12:50 +02:00
now := time . Now ( ) . UTC ( )
2018-07-16 09:46:40 +02:00
config := server . Config ( )
2019-03-07 08:31:46 +01:00
// give them 1k of grace over the limit:
2020-01-19 05:47:05 +01:00
socket := NewSocket ( conn . Conn , ircmsg . MaxlenTagsFromClient + 512 + 1024 , config . Server . MaxSendQBytes )
2012-12-09 21:51:50 +01:00
client := & Client {
2020-02-27 08:13:31 +01:00
lastSeen : now ,
lastActive : now ,
channels : make ( ChannelSet ) ,
ctime : now ,
isSTSOnly : conn . Config . STSOnly ,
languages : server . Languages ( ) . Default ( ) ,
2019-01-01 22:45:37 +01:00
loginThrottle : connection_limits . GenericThrottle {
Duration : config . Accounts . LoginThrottling . Duration ,
Limit : config . Accounts . LoginThrottling . MaxAttempts ,
} ,
2016-09-19 14:30:29 +02:00
server : server ,
2019-01-01 19:00:16 +01:00
accountName : "*" ,
2016-10-11 15:51:46 +02:00
nick : "*" , // * is used until actual nick is given
nickCasefolded : "*" ,
2016-09-19 14:30:29 +02:00
nickMaskString : "*" , // * is used until actual nick is given
2012-12-09 21:51:50 +01:00
}
2019-05-19 22:34:52 +02:00
client . history . Initialize ( config . History . ClientLength , config . History . AutoresizeWindow )
2019-05-22 03:40:25 +02:00
client . brbTimer . Initialize ( client )
2019-04-12 06:08:46 +02:00
session := & Session {
client : client ,
socket : socket ,
capVersion : caps . Cap301 ,
capState : caps . NoneState ,
2019-05-08 10:11:54 +02:00
ctime : now ,
2020-02-27 08:13:31 +01:00
lastActive : now ,
2019-05-13 02:57:34 +02:00
realIP : realIP ,
2020-02-19 01:38:42 +01:00
isTor : conn . Config . Tor ,
2019-04-12 06:08:46 +02:00
}
client . sessions = [ ] * Session { session }
2016-09-07 13:32:58 +02:00
2019-06-18 04:21:37 +02:00
if conn . Config . TLSConfig != nil {
2019-02-26 03:50:43 +01:00
client . SetMode ( modes . TLS , true )
2016-09-07 13:32:58 +02:00
// error is not useful to us here anyways so we can ignore it
2020-02-19 03:42:27 +01:00
session . certfp , _ = socket . CertFP ( )
2016-06-28 17:09:07 +02:00
}
2019-02-26 03:50:43 +01:00
2019-11-20 23:43:40 +01:00
if conn . Config . Tor {
2019-02-26 03:50:43 +01:00
client . SetMode ( modes . TLS , true )
2019-05-08 10:11:54 +02: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 21:10:23 +01:00
client . proxiedIP = session . proxiedIP
2019-05-08 10:11:54 +02:00
session . rawHostname = config . Server . TorListeners . Vhost
2019-12-17 21:10:23 +01:00
client . rawHostname = session . rawHostname
2019-02-26 03:50:43 +01:00
} else {
2019-05-13 02:57:34 +02:00
remoteAddr := conn . Conn . RemoteAddr ( )
2020-02-19 01:38:42 +01:00
if realIP . IsLoopback ( ) || utils . IPInNets ( realIP , config . Server . secureNets ) {
2019-05-08 10:11:54 +02:00
// treat local connections as secure (may be overridden later by WEBIRC)
client . SetMode ( modes . TLS , true )
}
2019-02-26 03:50:43 +01:00
if config . Server . CheckIdent && ! utils . AddrIsUnix ( remoteAddr ) {
client . doIdentLookup ( conn . Conn )
2016-06-30 11:28:34 +02:00
}
2019-02-26 03:50:43 +01:00
}
2019-05-08 10:11:54 +02:00
client . realIP = session . realIP
2019-02-26 03:50:43 +01:00
2019-07-01 15:21:38 +02:00
server . stats . Add ( )
2019-11-20 23:14:42 +01:00
client . run ( session , proxyLine )
2019-02-26 03:50:43 +01:00
}
2020-02-27 08:13:31 +01:00
func ( server * Server ) AddAlwaysOnClient ( account ClientAccount , chnames [ ] string , lastActive time . Time ) {
2020-02-19 01:38:42 +01:00
now := time . Now ( ) . UTC ( )
config := server . Config ( )
2020-02-27 08:13:31 +01:00
if lastActive . IsZero ( ) {
lastActive = now
}
2020-02-19 01:38:42 +01:00
client := & Client {
2020-02-27 08:13:31 +01:00
lastSeen : now ,
lastActive : lastActive ,
channels : make ( ChannelSet ) ,
ctime : now ,
languages : server . Languages ( ) . Default ( ) ,
server : server ,
2020-02-19 01:38:42 +01:00
// TODO figure out how to set these on reattach?
username : "~user" ,
rawHostname : server . name ,
realIP : utils . IPv4LoopbackAddress ,
2020-02-27 08:13:31 +01:00
alwaysOn : true ,
2020-02-19 01:38:42 +01: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 20:09:00 +01:00
status , _ := client . historyStatus ( config )
if status == HistoryEphemeral {
2020-02-19 01:38:42 +01:00
client . history . Resize ( config . History . ClientLength , config . History . AutoresizeWindow )
} else {
client . history . Resize ( 0 , 0 )
}
}
2019-12-17 21:10:23 +01: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-19 01:38:42 +01:00
if session . isTor {
2019-12-17 21:10:23 +01: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-18 01:57:23 +01:00
cloakedHostname := config . Server . Cloaks . ComputeCloak ( ip )
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
2019-12-17 21:10:23 +01: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-26 03:50:43 +01: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 11:28:34 +02:00
2019-02-26 03:50:43 +01: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 11:28:34 +02:00
if err == nil {
2019-02-26 03:50:43 +01: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 11:28:34 +02:00
} else {
2019-02-26 03:50:43 +01:00
client . Notice ( client . t ( "*** Got a malformed username, ignoring" ) )
2016-06-30 11:28:34 +02:00
}
2019-02-26 03:50:43 +01:00
} else {
client . Notice ( client . t ( "*** Could not find your username" ) )
2016-06-30 11:28:34 +02:00
}
2012-04-07 20:44:59 +02:00
}
2019-05-23 02:25:57 +02:00
type AuthOutcome uint
const (
authSuccess AuthOutcome = iota
authFailPass
authFailTorSaslRequired
authFailSaslRequired
)
2020-02-19 03:42:27 +01:00
func ( client * Client ) isAuthorized ( config * Config , session * Session ) AuthOutcome {
2019-02-05 06:19:03 +01:00
saslSent := client . account != ""
2019-02-26 03:50:43 +01:00
// PASS requirement
2020-02-19 03:42:27 +01:00
if ( config . Server . passwordBytes != nil ) && ! session . sentPassCommand && ! ( config . Accounts . SkipServerPassword && saslSent ) {
2019-05-23 02:25:57 +02:00
return authFailPass
2019-02-05 06:19:03 +01:00
}
2019-02-26 03:50:43 +01:00
// Tor connections may be required to authenticate with SASL
2020-02-19 03:42:27 +01:00
if session . isTor && config . Server . TorListeners . RequireSasl && ! saslSent {
2019-05-23 02:25:57 +02:00
return authFailTorSaslRequired
2019-02-26 03:50:43 +01:00
}
// finally, enforce require-sasl
2020-02-19 03:42:27 +01:00
if config . Accounts . RequireSasl . Enabled && ! saslSent && ! utils . IPInNets ( session . IP ( ) , config . Accounts . RequireSasl . exemptedNets ) {
2019-05-23 02:25:57 +02:00
return authFailSaslRequired
}
return authSuccess
2019-02-05 06:19:03 +01:00
}
2019-04-12 06:08:46 +02: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 16:04:21 +01:00
}
2017-05-24 08:58:36 +02:00
// IP returns the IP address of this client.
func ( client * Client ) IP ( ) net . IP {
2019-02-05 06:19:03 +01:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2018-02-01 21:53:49 +01:00
if client . proxiedIP != nil {
return client . proxiedIP
2017-09-11 07:04:08 +02:00
}
2019-02-05 06:19:03 +01:00
return client . realIP
2017-05-24 08:58:36 +02:00
}
2017-06-22 21:15:10 +02: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 15:21:38 +02: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 23:14:42 +01: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 17:49:52 +02:00
2017-10-24 00:38:32 +02:00
defer func ( ) {
2017-10-26 11:15:55 +02:00
if r := recover ( ) ; r != nil {
client . server . logger . Error ( "internal" ,
fmt . Sprintf ( "Client caused panic: %v\n%s" , r , debug . Stack ( ) ) )
2019-05-23 01:07:12 +02:00
if client . server . Config ( ) . Debug . recoverFromErrors {
2017-10-26 11:15:55 +02:00
client . server . logger . Error ( "internal" , "Disconnecting client and attempting to recover" )
} else {
panic ( r )
2017-10-26 10:19:01 +02:00
}
2017-10-24 00:38:32 +02:00
}
// ensure client connection gets closed
2019-05-22 03:40:25 +02:00
client . destroy ( session )
2017-10-24 00:38:32 +02:00
} ( )
2019-04-12 06:08:46 +02:00
session . idletimer . Initialize ( session )
session . resetFakelag ( )
2017-10-15 18:24:28 +02:00
2019-04-12 06:08:46 +02:00
isReattach := client . Registered ( )
2019-05-09 00:14:49 +02:00
if isReattach {
2019-05-22 03:40:25 +02:00
if session . resumeDetails != nil {
session . playResume ( )
session . resumeDetails = nil
client . brbTimer . Disable ( )
2019-07-04 12:59:08 +02:00
client . SetAway ( false , "" ) // clear BRB message if any
2019-05-22 03:40:25 +02:00
} else {
client . playReattachMessages ( session )
}
2019-05-09 00:14:49 +02:00
} else {
// don't reset the nick timer during a reattach
2019-04-12 06:08:46 +02:00
client . nickTimer . Initialize ( client )
}
2018-03-22 16:04:21 +01:00
2019-05-27 10:18:07 +02:00
firstLine := ! isReattach
2018-09-03 06:19:10 +02:00
2016-06-17 14:17:42 +02:00
for {
2019-11-20 23:14:42 +01: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 14:17:42 +02:00
if err != nil {
2018-03-18 02:32:12 +01:00
quitMessage := "connection closed"
if err == errReadQ {
quitMessage = "readQ exceeded"
}
2019-04-12 06:08:46 +02:00
client . Quit ( quitMessage , session )
2019-05-22 03:40:25 +02:00
// since the client did not actually send us a QUIT,
2019-05-27 10:18:07 +02:00
// give them a chance to resume if applicable:
if ! session . Destroyed ( ) {
client . brbTimer . Enable ( )
}
2016-06-17 14:17:42 +02:00
break
}
2018-10-28 20:44:13 +01:00
if client . server . logger . IsLoggingRawIO ( ) {
client . server . logger . Debug ( "userinput" , client . nick , "<- " , line )
}
2018-09-03 06:19:10 +02:00
// special-cased handling of PROXY protocol, see `handleProxyCommand` for details:
2019-05-27 10:18:07 +02:00
if firstLine {
2018-09-03 06:19:10 +02:00
firstLine = false
if strings . HasPrefix ( line , "PROXY" ) {
2019-04-12 06:08:46 +02:00
err = handleProxyCommand ( client . server , client , session , line )
2018-09-03 06:19:10 +02:00
if err != nil {
break
} else {
continue
}
}
}
2017-03-06 13:11:10 +01:00
2019-05-23 00:35:24 +02:00
if client . registered {
session . fakelag . Touch ( )
} else {
// DoS hardening, #505
2019-05-22 22:15:59 +02: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-19 05:47:05 +01:00
msg , err := ircmsg . ParseLineStrict ( line , true , 512 )
2017-01-18 22:56:33 +01:00
if err == ircmsg . ErrorLineIsEmpty {
continue
2019-03-07 08:31:46 +01:00
} else if err == ircmsg . ErrorLineTooLong {
2019-05-16 04:30:35 +02:00
session . Send ( nil , client . server . name , ERR_INPUTTOOLONG , client . Nick ( ) , client . t ( "Input line too long" ) )
2019-03-07 08:31:46 +01:00
continue
2017-01-18 22:56:33 +01:00
} else if err != nil {
2019-04-12 06:08:46 +02:00
client . Quit ( client . t ( "Received malformed line" ) , session )
2016-06-17 14:17:42 +02:00
break
2014-02-24 07:21:39 +01:00
}
2014-04-15 17:49:52 +02:00
2019-12-23 21:26:37 +01: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 02:01:30 +02:00
cmd , exists := Commands [ msg . Command ]
if ! exists {
2016-11-04 12:38:47 +01:00
if len ( msg . Command ) > 0 {
2019-05-16 04:30:35 +02:00
session . Send ( nil , client . server . name , ERR_UNKNOWNCOMMAND , client . Nick ( ) , msg . Command , client . t ( "Unknown command" ) )
2016-11-04 12:38:47 +01:00
} else {
2019-05-16 04:30:35 +02:00
session . Send ( nil , client . server . name , ERR_UNKNOWNCOMMAND , client . Nick ( ) , "lastcmd" , client . t ( "No command given" ) )
2016-11-04 12:38:47 +01:00
}
2016-06-20 14:53:45 +02:00
continue
2016-06-19 02:01:30 +02:00
}
2019-04-12 06:08:46 +02:00
isExiting := cmd . Run ( client . server , client , session , msg )
if isExiting {
break
} else if session . client != client {
// bouncer reattach
2019-11-20 23:14:42 +01:00
go session . client . run ( session , "" )
2016-06-17 14:17:42 +02:00
break
}
2014-02-24 07:21:39 +01:00
}
2014-04-15 17:49:52 +02:00
}
2019-05-09 00:14:49 +02:00
func ( client * Client ) playReattachMessages ( session * Session ) {
client . server . playRegistrationBurst ( session )
2020-02-19 01:38:42 +01:00
hasHistoryCaps := session . HasHistoryCaps ( )
2019-04-12 06:08:46 +02:00
for _ , channel := range session . client . Channels ( ) {
channel . playJoinForSession ( session )
2020-02-19 01:38:42 +01:00
// clients should receive autoreplay-on-join lines, if applicable:
if hasHistoryCaps {
continue
}
2019-05-30 01:23:46 +02: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 06:08:46 +02:00
}
2020-02-27 08:13:31 +01:00
if ! session . autoreplayMissedSince . IsZero ( ) && ! hasHistoryCaps {
2020-02-19 01:38:42 +01:00
rb := NewResponseBuffer ( session )
2020-02-28 01:07:49 +01:00
zncPlayPrivmsgs ( client , rb , "*" , time . Now ( ) . UTC ( ) , session . autoreplayMissedSince )
2020-02-19 01:38:42 +01:00
rb . Send ( true )
}
2020-02-27 08:13:31 +01:00
session . autoreplayMissedSince = time . Time { }
2019-04-12 06:08:46 +02:00
}
2016-06-17 14:17:42 +02:00
//
2017-05-09 12:37:48 +02:00
// idle, quit, timers and timeouts
2016-06-17 14:17:42 +02:00
//
2014-04-15 17:49:52 +02:00
2020-02-27 08:13:31 +01: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 09:12:50 +02:00
now := time . Now ( ) . UTC ( )
2017-12-03 02:05:06 +01:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
2020-02-27 08:13:31 +01:00
client . lastSeen = now
if active {
client . lastActive = now
session . lastActive = now
}
2014-02-18 22:25:21 +01:00
}
2014-02-14 03:39:33 +01:00
2017-10-15 18:24:28 +02:00
// Ping sends the client a PING message.
2019-04-12 06:08:46 +02:00
func ( session * Session ) Ping ( ) {
session . Send ( nil , "" , "PING" , session . client . Nick ( ) )
2017-05-09 12:37:48 +02:00
}
2019-02-10 02:01:47 +01:00
// tryResume tries to resume if the client asked us to.
2019-05-22 03:40:25 +02:00
func ( session * Session ) tryResume ( ) ( success bool ) {
var oldResumeID string
2018-01-21 02:23:33 +01:00
2019-02-10 02:01:47 +01:00
defer func ( ) {
2019-05-22 03:40:25 +02: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-10 02:01:47 +01:00
}
} ( )
2019-05-22 03:40:25 +02:00
client := session . client
server := client . server
config := server . Config ( )
2018-01-21 02:23:33 +01:00
2019-05-22 03:40:25 +02:00
oldClient , oldResumeID := server . resumeManager . VerifyToken ( client , session . resumeDetails . PresentedToken )
2018-11-26 11:23:27 +01:00
if oldClient == nil {
2019-05-22 03:40:25 +02:00
session . Send ( nil , server . name , "FAIL" , "RESUME" , "INVALID_TOKEN" , client . t ( "Cannot resume connection, token is not valid" ) )
2018-01-21 03:23:47 +01:00
return
}
2018-11-26 11:23:27 +01:00
resumeAllowed := config . Server . AllowPlaintextResume || ( oldClient . HasMode ( modes . TLS ) && client . HasMode ( modes . TLS ) )
if ! resumeAllowed {
2019-05-22 03:40:25 +02: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 02:23:33 +01:00
return
}
2019-05-22 03:40:25 +02:00
err := server . clients . Resume ( oldClient , session )
2018-11-26 11:23:27 +01:00
if err != nil {
2019-05-22 03:40:25 +02:00
session . Send ( nil , server . name , "FAIL" , "RESUME" , "CANNOT_RESUME" , client . t ( "Cannot resume connection" ) )
2018-01-21 02:23:33 +01:00
return
}
2019-02-10 02:01:47 +01:00
success = true
2019-05-22 03:40:25 +02:00
client . server . logger . Debug ( "quit" , fmt . Sprintf ( "%s is being resumed" , oldClient . Nick ( ) ) )
2019-02-10 02:01:47 +01:00
2019-05-22 03:40:25 +02:00
return
}
2018-01-21 02:23:33 +01:00
2019-05-22 03:40:25 +02: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-19 01:38:42 +01:00
config := server . Config ( )
2018-11-26 11:23:27 +01:00
friends := make ( ClientSet )
2020-02-19 01:38:42 +01:00
var oldestLostMessage time . Time
2018-11-26 11:23:27 +01:00
// work out how much time, if any, is not covered by history buffers
2020-02-19 01:38:42 +01:00
// assume that a persistent buffer covers the whole resume period
2019-05-22 03:40:25 +02:00
for _ , channel := range client . Channels ( ) {
2018-11-26 11:23:27 +01:00
for _ , member := range channel . Members ( ) {
friends . Add ( member )
2020-02-19 01:38:42 +01:00
}
2020-02-24 20:09:00 +01:00
status , _ := channel . historyStatus ( config )
if status == HistoryEphemeral {
2018-11-26 11:23:27 +01:00
lastDiscarded := channel . history . LastDiscarded ( )
2020-02-19 01:38:42 +01:00
if oldestLostMessage . Before ( lastDiscarded ) {
2018-11-26 11:23:27 +01:00
oldestLostMessage = lastDiscarded
}
}
}
2020-02-24 20:09:00 +01:00
cHistoryStatus , _ := client . historyStatus ( config )
if cHistoryStatus == HistoryEphemeral {
2020-02-19 01:38:42 +01:00
lastDiscarded := client . history . LastDiscarded ( )
if oldestLostMessage . Before ( lastDiscarded ) {
oldestLostMessage = lastDiscarded
}
2018-11-26 11:23:27 +01:00
}
2020-02-19 01:38:42 +01: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 11:23:27 +01:00
}
}
2019-05-22 03:40:25 +02:00
timestamp := session . resumeDetails . Timestamp
2020-02-19 01:38:42 +01:00
gap := oldestLostMessage . Sub ( timestamp )
2019-05-29 13:34:23 +02:00
session . resumeDetails . HistoryIncomplete = gap > 0 || timestamp . IsZero ( )
2018-11-26 11:23:27 +01:00
gapSeconds := int ( gap . Seconds ( ) ) + 1 // round up to avoid confusion
2019-05-22 03:40:25 +02:00
details := client . Details ( )
oldNickmask := details . nickMask
2019-12-17 21:10:23 +01:00
client . lookupHostname ( session , true )
2019-05-22 03:40:25 +02:00
hostname := client . Hostname ( ) // may be a vhost
2019-05-29 13:34:23 +02:00
timestampString := timestamp . Format ( IRCv3TimestampFormat )
2019-05-22 03:40:25 +02:00
2018-11-26 11:23:27 +01:00
// send quit/resume messages to friends
for friend := range friends {
2019-05-22 03:40:25 +02:00
if friend == client {
continue
}
for _ , fSession := range friend . Sessions ( ) {
if fSession . capabilities . Has ( caps . Resume ) {
2019-05-29 13:34:23 +02:00
if ! session . resumeDetails . HistoryIncomplete {
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname , "ok" )
} else if session . resumeDetails . HistoryIncomplete && ! timestamp . IsZero ( ) {
2019-05-22 03:40:25 +02:00
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname , timestampString )
2019-05-27 10:40:24 +02:00
} else {
fSession . Send ( nil , oldNickmask , "RESUMED" , hostname )
2019-04-12 06:08:46 +02:00
}
2018-11-26 11:23:27 +01:00
} else {
2019-05-29 13:34:23 +02:00
if ! session . resumeDetails . HistoryIncomplete {
2019-12-17 01:50:15 +01:00
fSession . Send ( nil , oldNickmask , "QUIT" , friend . t ( "Client reconnected" ) )
2019-05-29 13:34:23 +02: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-17 01:50:15 +01:00
fSession . Send ( nil , oldNickmask , "QUIT" , friend . t ( "Client reconnected (message history may have been lost)" ) )
2019-04-12 06:08:46 +02:00
}
2018-11-26 11:23:27 +01:00
}
2018-01-21 02:23:33 +01:00
}
}
2020-02-19 01:38:42 +01: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 11:23:27 +01:00
}
2018-01-21 02:23:33 +01:00
2019-05-22 21:08:02 +02:00
session . Send ( nil , client . server . name , "RESUME" , "SUCCESS" , details . nick )
2018-04-24 09:11:11 +02:00
2019-05-22 03:40:25 +02:00
server . playRegistrationBurst ( session )
2018-01-21 02:23:33 +01:00
2019-05-22 03:40:25 +02:00
for _ , channel := range client . Channels ( ) {
channel . Resume ( session , timestamp )
2018-11-26 11:23:27 +01:00
}
2018-01-22 11:55:20 +01:00
2018-11-26 11:23:27 +01:00
// replay direct PRIVSMG history
2020-02-19 01:38:42 +01:00
if ! timestamp . IsZero ( ) && privmsgSeq != nil {
after := history . Selector { Time : timestamp }
items , complete , _ := privmsgSeq . Between ( after , history . Selector { } , config . History . ZNCMax )
2020-02-21 05:47:13 +01:00
if len ( items ) != 0 {
rb := NewResponseBuffer ( session )
client . replayPrivmsgHistory ( rb , items , "" , complete )
rb . Send ( true )
}
2018-04-24 09:11:11 +02:00
}
2018-01-21 02:23:33 +01:00
2019-05-22 03:40:25 +02:00
session . resumeDetails = nil
2018-11-26 11:23:27 +01:00
}
2020-02-19 01:38:42 +01:00
func ( client * Client ) replayPrivmsgHistory ( rb * ResponseBuffer , items [ ] history . Item , target string , complete bool ) {
2019-05-07 05:17:57 +02:00
var batchID string
2019-05-19 08:14:36 +02:00
details := client . Details ( )
nick := details . nick
2020-02-21 05:47:13 +01:00
if target == "" {
target = nick
2019-05-07 05:17:57 +02:00
}
2020-02-21 05:47:13 +01:00
batchID = rb . StartNestedHistoryBatch ( target )
2019-05-07 05:17:57 +02:00
allowTags := rb . session . capabilities . Has ( caps . MessageTags )
2019-02-04 11:18:17 +01:00
for _ , item := range items {
var command string
switch item . Type {
case history . Privmsg :
command = "PRIVMSG"
case history . Notice :
command = "NOTICE"
2019-05-07 05:17:57 +02:00
case history . Tagmsg :
if allowTags {
command = "TAGMSG"
} else {
continue
}
2019-02-04 11:18:17 +01:00
default :
continue
}
2019-03-07 08:31:46 +01:00
var tags map [ string ] string
2019-05-07 05:17:57 +02:00
if allowTags {
tags = item . Tags
2019-02-04 11:18:17 +01:00
}
2020-02-20 07:45:17 +01: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-19 01:38:42 +01:00
if item . Params [ 0 ] == "" || item . Params [ 0 ] == nick {
2019-05-19 08:14:36 +02: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-19 01:38:42 +01:00
// substitute client's current nickmask in case client changed nick
2019-05-19 08:14:36 +02:00
rb . AddSplitMessageFromClient ( details . nickMask , item . AccountName , tags , command , item . Params [ 0 ] , item . Message )
}
2019-02-04 11:18:17 +01:00
}
2019-05-07 05:17:57 +02:00
rb . EndNestedBatch ( batchID )
2019-02-04 11:18:17 +01:00
if ! complete {
rb . Add ( nil , "HistServ" , "NOTICE" , nick , client . t ( "Some additional message history may have been lost" ) )
}
}
2016-10-23 03:48:57 +02:00
// IdleTime returns how long this client's been idle.
2014-02-18 00:25:32 +01:00
func ( client * Client ) IdleTime ( ) time . Duration {
2017-12-03 02:05:06 +01:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2020-02-27 08:13:31 +01:00
return time . Since ( client . lastActive )
2014-02-18 00:25:32 +01:00
}
2016-10-23 03:48:57 +02:00
// SignonTime returns this client's signon time as a unix timestamp.
2014-02-18 04:56:06 +01:00
func ( client * Client ) SignonTime ( ) int64 {
return client . ctime . Unix ( )
}
2016-10-23 03:48:57 +02:00
// IdleSeconds returns the number of seconds this client's been idle.
2014-02-18 04:08:57 +01:00
func ( client * Client ) IdleSeconds ( ) uint64 {
return uint64 ( client . IdleTime ( ) . Seconds ( ) )
}
2016-10-23 03:48:57 +02:00
// HasNick returns true if the client's nickname is set (used in registration).
2014-02-09 02:10:04 +01:00
func ( client * Client ) HasNick ( ) bool {
2017-11-22 10:41:11 +01:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2016-10-11 15:51:46 +02:00
return client . nick != "" && client . nick != "*"
2012-12-17 04:13:53 +01:00
}
2017-04-16 03:31:33 +02:00
// HasUsername returns true if the client's username is set (used in registration).
2014-02-09 02:10:04 +01:00
func ( client * Client ) HasUsername ( ) bool {
2017-11-22 10:41:11 +01:00
client . stateMutex . RLock ( )
defer client . stateMutex . RUnlock ( )
2016-10-11 15:51:46 +02:00
return client . username != "" && client . username != "*"
2014-02-09 02:10:04 +01:00
}
2019-02-03 09:49:42 +01:00
// SetNames sets the client's ident and realname.
2019-02-05 08:40:49 +01: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 09:04:52 +01:00
if limit < len ( username ) {
2019-02-05 08:40:49 +01:00
username = username [ : limit ]
}
2019-02-03 09:49:42 +01:00
if ! isIdent ( username ) {
2018-11-26 11:23:27 +01:00
return errInvalidUsername
}
2019-02-05 08:40:49 +01:00
if ! fromIdent {
username = "~" + username
}
2019-02-03 09:49:42 +01:00
2018-11-26 11:23:27 +01:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
if client . username == "" {
2019-02-05 08:40:49 +01:00
client . username = username
2018-11-26 11:23:27 +01:00
}
if client . realname == "" {
client . realname = realname
}
return nil
}
2017-09-29 04:11:06 +02:00
// HasRoleCapabs returns true if client has the given (role) capabilities.
func ( client * Client ) HasRoleCapabs ( capabs ... string ) bool {
2018-04-19 08:48:19 +02:00
oper := client . Oper ( )
if oper == nil {
2016-10-23 03:13:08 +02:00
return false
}
for _ , capab := range capabs {
2018-04-19 08:48:19 +02:00
if ! oper . Class . Capabilities [ capab ] {
2016-10-23 03:13:08 +02:00
return false
}
}
return true
}
2017-04-16 03:31:33 +02:00
// ModeString returns the mode string for this client.
func ( client * Client ) ModeString ( ) ( str string ) {
2020-02-19 01:38:42 +01:00
return "+" + client . modes . String ( )
2012-04-18 05:24:26 +02:00
}
2012-04-18 06:13:12 +02:00
2016-06-17 14:17:42 +02:00
// Friends refers to clients that share a channel with this client.
2019-04-12 06:08:46 +02:00
func ( client * Client ) Friends ( capabs ... caps . Capability ) ( result map [ * Session ] bool ) {
result = make ( map [ * Session ] bool )
2016-10-26 16:44:36 +02:00
2019-04-12 06:08:46 +02:00
// look at the client's own sessions
for _ , session := range client . Sessions ( ) {
if session . capabilities . HasAll ( capabs ... ) {
result [ session ] = true
2016-10-26 16:44:36 +02:00
}
}
2017-10-23 01:50:16 +02:00
for _ , channel := range client . Channels ( ) {
for _ , member := range channel . Members ( ) {
2019-04-12 06:08:46 +02:00
for _ , session := range member . Sessions ( ) {
if session . capabilities . HasAll ( capabs ... ) {
result [ session ] = true
2016-10-13 10:08:08 +02:00
}
}
2014-02-19 00:28:20 +01:00
}
2014-02-17 02:23:47 +01:00
}
2019-04-12 06:08:46 +02:00
return
2014-02-17 02:23:47 +01:00
}
2019-01-31 00:59:49 +01: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 08:48:19 +02: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 11:23:27 +01:00
fClient . sendFromClientInternal ( false , time . Time { } , "" , oldNickMask , client . AccountName ( ) , nil , "CHGHOST" , username , vhost )
2018-04-19 08:48:19 +02: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 06:57:03 +02:00
// updateNick updates `nick` and `nickCasefolded`.
2019-01-31 00:59:49 +01:00
func ( client * Client ) updateNick ( nick , nickCasefolded , skeleton string ) {
2017-10-04 06:57:03 +02:00
client . stateMutex . Lock ( )
2019-01-31 00:59:49 +01:00
defer client . stateMutex . Unlock ( )
2017-10-04 06:57:03 +02:00
client . nick = nick
2019-01-31 00:59:49 +01:00
client . nickCasefolded = nickCasefolded
client . skeleton = skeleton
client . updateNickMaskNoMutex ( )
2016-10-16 12:35:50 +02:00
}
2019-01-31 00:59:49 +01:00
// updateNickMaskNoMutex updates the casefolded nickname and nickmask, not acquiring any mutexes.
2018-01-22 11:55:20 +01:00
func ( client * Client ) updateNickMaskNoMutex ( ) {
2020-02-06 23:43:54 +01:00
if client . nick == "*" {
return // pre-registration, don't bother generating the hostname
}
2018-04-19 08:48:19 +02:00
client . hostname = client . getVHostNoMutex ( )
if client . hostname == "" {
2019-05-12 08:17:57 +02:00
client . hostname = client . cloakedHostname
if client . hostname == "" {
client . hostname = client . rawHostname
}
2016-10-23 03:28:31 +02:00
}
2019-12-18 13:01:26 +01:00
cfhostname := strings . ToLower ( client . hostname )
2019-01-28 19:36:15 +01:00
client . nickMaskString = fmt . Sprintf ( "%s!%s@%s" , client . nick , client . username , client . hostname )
2019-02-05 08:40:49 +01:00
client . nickMaskCasefolded = fmt . Sprintf ( "%s!%s@%s" , client . nickCasefolded , strings . ToLower ( client . username ) , cfhostname )
2016-06-19 07:37:29 +02:00
}
2017-01-11 13:38:16 +01:00
// AllNickmasks returns all the possible nickmasks for the client.
2019-01-29 05:03:30 +01:00
func ( client * Client ) AllNickmasks ( ) ( masks [ ] string ) {
2018-04-19 08:48:19 +02:00
client . stateMutex . RLock ( )
2019-01-29 05:03:30 +01:00
nick := client . nickCasefolded
2019-02-05 08:40:49 +01:00
username := client . username
2018-04-19 08:48:19 +02:00
rawHostname := client . rawHostname
2019-05-12 08:17:57 +02:00
cloakedHostname := client . cloakedHostname
2018-04-19 08:48:19 +02:00
vhost := client . getVHostNoMutex ( )
client . stateMutex . RUnlock ( )
2019-02-05 09:04:52 +01:00
username = strings . ToLower ( username )
2018-04-19 08:48:19 +02:00
if len ( vhost ) > 0 {
2019-12-18 13:01:26 +01:00
cfvhost := strings . ToLower ( vhost )
masks = append ( masks , fmt . Sprintf ( "%s!%s@%s" , nick , username , cfvhost ) )
2017-01-11 13:38:16 +01:00
}
2019-01-29 05:03:30 +01:00
var rawhostmask string
2019-12-18 13:01:26 +01:00
cfrawhost := strings . ToLower ( rawHostname )
rawhostmask = fmt . Sprintf ( "%s!%s@%s" , nick , username , cfrawhost )
masks = append ( masks , rawhostmask )
2017-01-11 13:38:16 +01:00
2019-05-12 08:17:57 +02:00
if cloakedHostname != "" {
masks = append ( masks , fmt . Sprintf ( "%s!%s@%s" , nick , username , cloakedHostname ) )
}
2017-01-11 13:38:16 +01:00
2019-01-29 05:03:30 +01:00
ipmask := fmt . Sprintf ( "%s!%s@%s" , nick , username , client . IPString ( ) )
if ipmask != rawhostmask {
masks = append ( masks , ipmask )
2017-01-11 13:38:16 +01:00
}
2019-01-29 05:03:30 +01:00
return
2017-01-11 13:38:16 +01:00
}
2017-09-28 07:49:01 +02:00
// LoggedIntoAccount returns true if this client is logged into an account.
func ( client * Client ) LoggedIntoAccount ( ) bool {
2018-02-11 11:30:40 +01:00
return client . Account ( ) != ""
2017-09-28 07:49:01 +02:00
}
2019-02-10 19:57:32 +01: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 06:08:46 +02: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-11 02:49:29 +02:00
2019-04-12 06:08:46 +02:00
errorMsg := ircmsg . MakeMessage ( nil , "" , "ERROR" , message )
errorMsgBytes , _ := errorMsg . LineBytesStrict ( false , 512 )
finalData = append ( finalData , errorMsgBytes ... )
2017-10-11 02:49:29 +02:00
2019-04-12 06:08:46 +02:00
sess . socket . SetFinalData ( finalData )
2019-02-10 19:57:32 +01:00
}
2017-10-11 02:49:29 +02:00
2019-04-12 06:08:46 +02:00
client . stateMutex . Lock ( )
defer client . stateMutex . Unlock ( )
var sessions [ ] * Session
if session != nil {
sessions = [ ] * Session { session }
} else {
sessions = client . sessions
}
2017-10-11 02:49:29 +02:00
2019-04-12 06:08:46 +02:00
for _ , session := range sessions {
if session . SetQuitMessage ( message ) {
setFinalData ( session )
}
}
2016-06-17 14:17:42 +02:00
}
2016-10-23 03:48:57 +02:00
// destroy gets rid of a client, removes them from server lists etc.
2019-04-12 06:08:46 +02: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-22 03:40:25 +02:00
func ( client * Client ) destroy ( session * Session ) {
2019-04-12 06:08:46 +02:00
var sessionsToDestroy [ ] * Session
2018-11-26 11:23:27 +01:00
client . stateMutex . Lock ( )
2019-05-09 00:14:49 +02:00
details := client . detailsNoMutex ( )
2019-05-22 03:40:25 +02:00
brbState := client . brbTimer . state
2019-05-27 10:18:07 +02:00
brbAt := client . brbTimer . brbAt
2019-05-09 00:14:49 +02:00
wasReattach := session != nil && session . client != client
2019-04-12 06:08:46 +02:00
sessionRemoved := false
2020-02-19 01:38:42 +01:00
registered := client . registered
alwaysOn := client . alwaysOn
2020-02-27 08:13:31 +01:00
saveLastSeen := alwaysOn && client . accountSettings . AutoreplayMissed
2019-04-12 06:08:46 +02: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 11:52:40 +02:00
// should we destroy the whole client this time?
2019-05-31 00:48:12 +02:00
// BRB is not respected if this is a destroy of the whole client (i.e., session == nil)
2020-02-19 01:38:42 +01:00
brbEligible := session != nil && ( brbState == BrbEnabled || alwaysOn )
2020-02-25 20:04:50 +01:00
alreadyDestroyed := client . destroyed
shouldDestroy := ! alreadyDestroyed && remainingSessions == 0 && ! brbEligible
2019-05-27 11:52:40 +02:00
if shouldDestroy {
// if it's our job to destroy it, don't let anyone else try
client . destroyed = true
}
2020-02-27 08:13:31 +01:00
if saveLastSeen {
client . dirtyBits |= IncludeLastSeen
2020-02-19 01:38:42 +01:00
}
2019-05-27 20:33:59 +02:00
exitedSnomaskSent := client . exitedSnomaskSent
2018-11-26 11:23:27 +01:00
client . stateMutex . Unlock ( )
2020-02-27 08:13:31 +01: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 08:33:49 +01:00
client . wakeWriter ( )
}
2019-05-09 00:14:49 +02:00
// destroy all applicable sessions:
var quitMessage string
2019-04-12 06:08:46 +02: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-09 00:14:49 +02:00
quitMessage = session . quitMessage
2019-05-27 10:18:07 +02:00
session . SetDestroyed ( )
2019-05-09 00:14:49 +02:00
session . socket . Close ( )
// remove from connection limits
var source string
2020-02-19 01:38:42 +01:00
if session . isTor {
2019-05-09 00:14:49 +02: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 06:08:46 +02:00
}
2020-02-19 01:38:42 +01:00
// decrement stats if we have no more sessions, even if the client will not be destroyed
2020-02-25 20:04:50 +01:00
if shouldDestroy || ( ! alreadyDestroyed && remainingSessions == 0 ) {
2020-02-19 01:38:42 +01:00
invisible := client . HasMode ( modes . Invisible )
operator := client . HasMode ( modes . LocalOperator ) || client . HasMode ( modes . Operator )
client . server . stats . Remove ( registered , invisible , operator )
}
2019-05-22 03:40:25 +02:00
// do not destroy the client if it has either remaining sessions, or is BRB'ed
2019-05-27 11:52:40 +02:00
if ! shouldDestroy {
2018-01-22 11:55:20 +01:00
return
2014-02-18 22:25:21 +01:00
}
2014-02-20 03:46:46 +01:00
2020-02-19 01:38:42 +01: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 07:45:17 +01:00
// use a defer here to avoid writing to mysql while holding the destroy semaphore:
2020-02-19 01:38:42 +01:00
defer func ( ) {
for _ , channel := range channels {
channel . AddHistoryItem ( quitItem )
}
} ( )
2018-04-25 02:34:28 +02: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-22 03:40:25 +02:00
if ! wasReattach {
2019-05-07 05:17:57 +02:00
client . server . logger . Debug ( "quit" , fmt . Sprintf ( "%s is no longer on the server" , details . nick ) )
2018-01-21 02:23:33 +01:00
}
2017-03-06 13:11:10 +01:00
2019-05-22 03:40:25 +02:00
if registered {
2018-05-04 06:24:54 +02:00
client . server . whoWas . Append ( client . WhoWas ( ) )
2018-01-21 02:23:33 +01:00
}
2016-06-17 14:17:42 +02:00
2019-02-12 06:27:57 +01:00
client . server . resumeManager . Delete ( client )
2016-10-16 12:14:56 +02:00
// alert monitors
2019-05-09 00:14:49 +02:00
if registered {
client . server . monitorManager . AlertAbout ( client , false )
}
2017-10-04 06:57:03 +02:00
// clean up monitor state
2017-10-04 08:59:59 +02:00
client . server . monitorManager . RemoveAll ( client )
2016-10-16 12:14:56 +02:00
2016-06-17 14:17:42 +02:00
// clean up channels
2019-05-09 00:14:49 +02:00
// (note that if this is a reattach, client has no channels and therefore no friends)
2018-04-25 02:23:01 +02:00
friends := make ( ClientSet )
2020-02-19 01:38:42 +01:00
channels = client . Channels ( )
for _ , channel := range channels {
2019-05-22 03:40:25 +02:00
channel . Quit ( client )
2017-10-23 01:50:16 +02:00
for _ , member := range channel . Members ( ) {
friends . Add ( member )
}
2016-06-17 14:17:42 +02:00
}
2018-04-25 02:23:01 +02:00
friends . Remove ( client )
2016-06-17 14:17:42 +02:00
// clean up server
2019-05-22 03:40:25 +02:00
client . server . clients . Remove ( client )
2016-06-17 14:17:42 +02:00
// clean up self
2018-02-28 06:14:44 +01:00
client . nickTimer . Stop ( )
2019-05-22 03:40:25 +02:00
client . brbTimer . Disable ( )
2016-06-17 14:17:42 +02:00
2018-02-11 11:30:40 +01:00
client . server . accounts . Logout ( client )
2019-05-27 10:18:07 +02:00
// this happens under failure to return from BRB
if quitMessage == "" {
2019-10-06 05:50:11 +02:00
if brbState == BrbDead && ! brbAt . IsZero ( ) {
2019-05-27 10:18:07 +02:00
awayMessage := client . AwayMessage ( )
2019-07-04 12:59:08 +02:00
if awayMessage == "" {
awayMessage = "Disconnected" // auto-BRB
2019-05-27 10:18:07 +02:00
}
2019-07-04 12:59:08 +02:00
quitMessage = fmt . Sprintf ( "%s [%s ago]" , awayMessage , time . Since ( brbAt ) . Truncate ( time . Second ) . String ( ) )
2017-09-26 04:47:03 +02:00
}
2019-05-27 10:18:07 +02:00
}
if quitMessage == "" {
quitMessage = "Exited"
}
for friend := range friends {
2019-05-22 03:40:25 +02:00
friend . sendFromClientInternal ( false , splitQuitMessage . Time , splitQuitMessage . Msgid , details . nickMask , details . accountName , nil , "QUIT" , quitMessage )
2016-11-29 12:06:01 +01:00
}
2019-05-22 03:40:25 +02:00
2019-05-27 20:33:59 +02:00
if ! exitedSnomaskSent && registered {
2019-05-22 03:40:25 +02:00
client . server . snomasks . Send ( sno . LocalQuits , fmt . Sprintf ( ircfmt . Unescape ( "%s$r exited the network" ) , details . nick ) )
2017-06-11 18:01:39 +02:00
}
2016-06-19 02:01:30 +02:00
}
2014-02-18 22:25:21 +01:00
2017-01-14 06:28:50 +01:00
// SendSplitMsgFromClient sends an IRC PRIVMSG/NOTICE coming from a specific client.
// Adds account-tag to the line as well.
2019-05-13 06:39:59 +02:00
func ( session * Session ) sendSplitMsgFromClientInternal ( blocking bool , nickmask , accountName string , tags map [ string ] string , command , target string , message utils . SplitMessage ) {
2020-01-19 05:47:05 +01:00
if message . Is512 ( ) {
2019-05-13 06:39:59 +02:00
session . sendFromClientInternal ( blocking , message . Time , message . Msgid , nickmask , accountName , tags , command , target , message . Message )
2017-01-14 06:28:50 +01:00
} else {
2020-02-19 01:38:42 +01:00
if session . capabilities . Has ( caps . Multiline ) {
2019-12-23 21:26:37 +01:00
for _ , msg := range session . composeMultilineBatch ( nickmask , accountName , tags , command , target , message ) {
session . SendRawMessage ( msg , blocking )
}
} else {
2020-01-19 05:47:05 +01: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 21:26:37 +01:00
}
2017-01-14 06:28:50 +01:00
}
}
}
2019-05-07 05:17:57 +02:00
// Sends a line with `nickmask` as the prefix, adding `time` and `account` tags if supported
2019-04-12 06:08:46 +02: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 08:31:46 +01:00
msg := ircmsg . MakeMessage ( tags , nickmask , command , params ... )
2016-09-12 03:25:31 +02:00
// attach account-tag
2019-04-12 06:08:46 +02:00
if session . capabilities . Has ( caps . AccountTag ) && accountName != "*" {
2019-03-07 08:31:46 +01:00
msg . SetTag ( "account" , accountName )
2016-09-12 03:25:31 +02:00
}
2017-01-14 10:52:47 +01:00
// attach message-id
2019-04-12 06:08:46 +02:00
if msgid != "" && session . capabilities . Has ( caps . MessageTags ) {
2019-05-15 07:30:21 +02:00
msg . SetTag ( "msgid" , msgid )
2019-03-07 08:31:46 +01:00
}
// attach server-time
2019-06-13 08:24:14 +02:00
session . setTimeTag ( & msg , serverTime )
2016-09-12 03:25:31 +02:00
2019-04-12 06:08:46 +02:00
return session . SendRawMessage ( msg , blocking )
2016-09-12 03:25:31 +02:00
}
2019-12-23 21:26:37 +01: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 15:46:55 +01:00
batchStart := ircmsg . MakeMessage ( tags , fromNickMask , "BATCH" , "+" + batchID , caps . MultilineBatchType , target )
2019-12-23 21:26:37 +01: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-19 05:47:05 +01:00
for _ , msg := range message . Split {
2019-12-23 21:26:37 +01: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-20 15:07:10 +01:00
var (
2017-01-23 00:03:49 +01:00
// these are all the output commands that MUST have their last param be a trailing.
2017-10-08 03:05:05 +02:00
// this is needed because dumb clients like to treat trailing params separately from the
2017-01-23 00:03:49 +01:00
// other params in messages.
2017-01-20 15:07:10 +01:00
commandsThatMustUseTrailing = map [ string ] bool {
"PRIVMSG" : true ,
"NOTICE" : true ,
2017-01-23 00:03:49 +01:00
RPL_WHOISCHANNELS : true ,
2017-03-06 06:50:23 +01:00
RPL_USERHOST : true ,
2017-01-20 15:07:10 +01:00
}
)
2017-10-08 03:05:05 +02:00
// SendRawMessage sends a raw message to the client.
2019-04-12 06:08:46 +02:00
func ( session * Session ) SendRawMessage ( message ircmsg . IrcMessage , blocking bool ) error {
2017-10-08 03:05:05 +02:00
// use dumb hack to force the last param to be a trailing param if required
2017-05-09 12:37:48 +02:00
var usedTrailingHack bool
2019-05-09 20:18:30 +02:00
config := session . client . server . Config ( )
if config . Server . Compatibility . forceTrailing && commandsThatMustUseTrailing [ message . Command ] && len ( message . Params ) > 0 {
2017-10-08 03:05:05 +02:00
lastParam := message . Params [ len ( message . Params ) - 1 ]
2017-05-09 12:37:48 +02:00
// to force trailing, we ensure the final param contains a space
2019-03-07 08:31:46 +01:00
if strings . IndexByte ( lastParam , ' ' ) == - 1 {
2017-10-08 03:05:05 +02:00
message . Params [ len ( message . Params ) - 1 ] = lastParam + " "
2017-05-09 12:37:48 +02:00
usedTrailingHack = true
2017-01-20 14:51:36 +01:00
}
}
2017-10-08 03:05:05 +02:00
// assemble message
2020-01-19 05:47:05 +01:00
line , err := message . LineBytesStrict ( false , 512 )
2016-06-19 02:01:30 +02:00
if err != nil {
2018-02-11 11:30:40 +01:00
logline := fmt . Sprintf ( "Error assembling message for sending: %v\n%s" , err , debug . Stack ( ) )
2019-04-12 06:08:46 +02:00
session . client . server . logger . Error ( "internal" , logline )
2016-11-04 12:38:47 +01:00
2019-04-12 06:08:46 +02:00
message = ircmsg . MakeMessage ( nil , session . client . server . name , ERR_UNKNOWNERROR , "*" , "Error assembling message for sending" )
2019-03-07 08:31:46 +01:00
line , _ := message . LineBytesStrict ( false , 0 )
2017-10-08 03:05:05 +02:00
2018-11-26 11:23:27 +01:00
if blocking {
2019-04-12 06:08:46 +02:00
session . socket . BlockingWrite ( line )
2018-11-26 11:23:27 +01:00
} else {
2019-04-12 06:08:46 +02:00
session . socket . Write ( line )
2018-11-26 11:23:27 +01:00
}
2016-06-19 02:01:30 +02:00
return err
2014-02-17 02:23:47 +01:00
}
2017-01-20 14:51:36 +01:00
2018-04-15 18:28:25 +02:00
// if we used the trailing hack, we need to strip the final space we appended earlier on
if usedTrailingHack {
2019-03-07 08:31:46 +01:00
copy ( line [ len ( line ) - 3 : ] , "\r\n" )
2018-04-26 21:32:32 +02:00
line = line [ : len ( line ) - 1 ]
2018-04-15 18:28:25 +02:00
}
2019-04-12 06:08:46 +02:00
if session . client . server . logger . IsLoggingRawIO ( ) {
2018-04-26 21:32:32 +02:00
logline := string ( line [ : len ( line ) - 2 ] ) // strip "\r\n"
2019-04-12 06:08:46 +02:00
session . client . server . logger . Debug ( "useroutput" , session . client . Nick ( ) , " ->" , logline )
2018-04-26 21:32:32 +02:00
}
2017-03-06 13:11:10 +01:00
2018-11-26 11:23:27 +01:00
if blocking {
2019-04-12 06:08:46 +02:00
return session . socket . BlockingWrite ( line )
2018-11-26 11:23:27 +01:00
} else {
2019-04-12 06:08:46 +02:00
return session . socket . Write ( line )
2018-11-26 11:23:27 +01:00
}
2017-10-08 03:05:05 +02:00
}
2018-11-26 11:23:27 +01:00
// Send sends an IRC line to the client.
2019-04-12 06:08:46 +02: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 08:31:46 +01:00
msg := ircmsg . MakeMessage ( tags , prefix , command , params ... )
2019-06-13 08:24:14 +02:00
session . setTimeTag ( & msg , time . Time { } )
return session . SendRawMessage ( msg , false )
}
func ( session * Session ) setTimeTag ( msg * ircmsg . IrcMessage , serverTime time . Time ) {
2019-04-12 06:08:46 +02:00
if session . capabilities . Has ( caps . ServerTime ) && ! msg . HasTag ( "time" ) {
2019-06-13 08:24:14 +02:00
if serverTime . IsZero ( ) {
serverTime = time . Now ( )
}
msg . SetTag ( "time" , serverTime . UTC ( ) . Format ( IRCv3TimestampFormat ) )
2019-03-07 08:31:46 +01:00
}
2018-11-26 11:23:27 +01:00
}
2016-06-19 02:01:30 +02:00
// Notice sends the client a notice from the server.
func ( client * Client ) Notice ( text string ) {
2019-04-12 06:08:46 +02:00
client . Send ( nil , client . server . name , "NOTICE" , client . Nick ( ) , text )
2014-02-17 02:23:47 +01:00
}
2017-10-23 01:50:16 +02:00
2019-12-17 21:10:23 +01:00
func ( session * Session ) Notice ( text string ) {
session . Send ( nil , session . client . server . name , "NOTICE" , session . client . Nick ( ) , text )
}
2017-10-23 01:50:16 +02:00
func ( client * Client ) addChannel ( channel * Channel ) {
client . stateMutex . Lock ( )
client . channels [ channel ] = true
2020-02-19 01:38:42 +01:00
alwaysOn := client . alwaysOn
2017-10-23 01:50:16 +02:00
client . stateMutex . Unlock ( )
2020-02-19 01:38:42 +01:00
if alwaysOn {
client . markDirty ( IncludeChannels )
}
2017-10-23 01:50:16 +02:00
}
func ( client * Client ) removeChannel ( channel * Channel ) {
client . stateMutex . Lock ( )
delete ( client . channels , channel )
2020-02-19 01:38:42 +01:00
alwaysOn := client . alwaysOn
2017-10-23 01:50:16 +02:00
client . stateMutex . Unlock ( )
2020-02-19 01:38:42 +01:00
if alwaysOn {
client . markDirty ( IncludeChannels )
}
2017-10-23 01:50:16 +02:00
}
2018-11-26 11:23:27 +01:00
2018-12-23 19:25:02 +01: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 12:33:43 +01: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-19 03:42:27 +01:00
if session . certfp == "" || client . HasMode ( modes . Operator ) {
2019-12-19 12:33:43 +01:00
return
}
for _ , oper := range client . server . Config ( ) . operators {
2020-02-19 03:42:27 +01:00
if oper . Auto && oper . Pass == nil && oper . Fingerprint != "" && oper . Fingerprint == session . certfp {
2019-12-19 12:33:43 +01:00
rb := NewResponseBuffer ( session )
applyOper ( client , oper , rb )
rb . Send ( true )
2019-12-29 17:59:49 +01:00
return
2019-12-19 12:33:43 +01:00
}
}
}
2020-02-19 01:38:42 +01:00
2020-02-24 20:09:00 +01:00
func ( client * Client ) historyStatus ( config * Config ) ( status HistoryStatus , target string ) {
2020-02-19 01:38:42 +01:00
if ! config . History . Enabled {
2020-02-24 20:09:00 +01:00
return HistoryDisabled , ""
2020-02-19 01:38:42 +01:00
}
client . stateMutex . RLock ( )
2020-02-28 11:41:08 +01:00
target = client . account
2020-02-19 01:38:42 +01:00
historyStatus := client . accountSettings . DMHistory
client . stateMutex . RUnlock ( )
2020-02-28 11:41:08 +01:00
if target == "" {
2020-02-24 20:09:00 +01:00
return HistoryEphemeral , ""
2020-02-19 01:38:42 +01:00
}
2020-02-28 11:41:08 +01:00
status = historyEnabled ( config . History . Persistent . DirectMessages , historyStatus )
if status != HistoryPersistent {
target = ""
}
return
2020-02-19 01:38:42 +01: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 08:13:31 +01:00
IncludeLastSeen
2020-02-19 01:38:42 +01: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 08:33:49 +01:00
dirtyBits := client . dirtyBits
2020-02-19 01:38:42 +01:00
client . dirtyBits = 0
account := client . account
client . stateMutex . Unlock ( )
if account == "" {
client . server . logger . Error ( "internal" , "attempting to persist logged-out client" , client . Nick ( ) )
return
}
2020-02-20 08:33:49 +01: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 08:13:31 +01:00
if ( dirtyBits & IncludeLastSeen ) != 0 {
2020-02-20 08:33:49 +01:00
client . stateMutex . RLock ( )
2020-02-27 08:13:31 +01:00
lastSeen := client . lastSeen
2020-02-20 08:33:49 +01:00
client . stateMutex . RUnlock ( )
2020-02-27 08:13:31 +01:00
client . server . accounts . saveLastSeen ( account , lastSeen )
2020-02-19 01:38:42 +01:00
}
}