source: code/trunk/upstream.go@ 719

Last change on this file since 719 was 711, checked in by contact, 4 years ago

Add message counter metrics

File size: 51.1 KB
RevLine 
[98]1package soju
[13]2
3import (
[652]4 "context"
[307]5 "crypto"
6 "crypto/sha256"
[13]7 "crypto/tls"
[307]8 "crypto/x509"
[95]9 "encoding/base64"
[155]10 "errors"
[13]11 "fmt"
12 "io"
13 "net"
[19]14 "strconv"
[17]15 "strings"
[19]16 "time"
[13]17
[95]18 "github.com/emersion/go-sasl"
[13]19 "gopkg.in/irc.v3"
20)
21
[282]22// permanentUpstreamCaps is the static list of upstream capabilities always
23// requested when supported.
24var permanentUpstreamCaps = map[string]bool{
[559]25 "account-tag": true,
[282]26 "away-notify": true,
27 "batch": true,
[419]28 "extended-join": true,
[448]29 "invite-notify": true,
[282]30 "labeled-response": true,
31 "message-tags": true,
[292]32 "multi-prefix": true,
[282]33 "server-time": true,
[540]34 "setname": true,
[685]35
36 "draft/extended-monitor": true,
[282]37}
38
[399]39type registrationError string
40
41func (err registrationError) Error() string {
42 return fmt.Sprintf("registration error: %v", string(err))
43}
44
[19]45type upstreamChannel struct {
[162]46 Name string
47 conn *upstreamConn
48 Topic string
[405]49 TopicWho *irc.Prefix
[162]50 TopicTime time.Time
51 Status channelStatus
52 modes channelModes
53 creationTime string
[478]54 Members membershipsCasemapMap
[162]55 complete bool
[435]56 detachTimer *time.Timer
[19]57}
58
[435]59func (uc *upstreamChannel) updateAutoDetach(dur time.Duration) {
60 if uc.detachTimer != nil {
61 uc.detachTimer.Stop()
62 uc.detachTimer = nil
63 }
64
65 if dur == 0 {
66 return
67 }
68
69 uc.detachTimer = time.AfterFunc(dur, func() {
70 uc.conn.network.user.events <- eventChannelDetach{
71 uc: uc.conn,
72 name: uc.Name,
73 }
74 })
75}
76
[681]77type pendingUpstreamCommand struct {
78 downstreamID uint64
[682]79 msg *irc.Message
[681]80}
81
[13]82type upstreamConn struct {
[210]83 conn
[16]84
[210]85 network *network
86 user *user
87
[16]88 serverName string
89 availableUserModes string
[139]90 availableChannelModes map[byte]channelModeType
91 availableChannelTypes string
92 availableMemberships []membership
[460]93 isupport map[string]*string
[19]94
[277]95 registered bool
96 nick string
[478]97 nickCM string
[277]98 username string
99 realname string
100 modes userModes
[478]101 channels upstreamChannelCasemapMap
[277]102 supportedCaps map[string]string
[278]103 caps map[string]bool
[277]104 batches map[string]batch
105 away bool
[559]106 account string
[278]107 nextLabelID uint64
[684]108 monitored monitorCasemapMap
[95]109
110 saslClient sasl.Client
111 saslStarted bool
[177]112
[478]113 casemapIsSet bool
114
[682]115 // Queue of commands in progress, indexed by type. The first entry has been
116 // sent to the server and is awaiting reply. The following entries have not
117 // been sent yet.
118 pendingCmds map[string][]pendingUpstreamCommand
[552]119
120 gotMotd bool
[13]121}
122
[77]123func connectToUpstream(network *network) (*upstreamConn, error) {
[502]124 logger := &prefixLogger{network.user.logger, fmt.Sprintf("upstream %q: ", network.GetName())}
[33]125
[352]126 dialer := net.Dialer{Timeout: connectTimeout}
[269]127
[457]128 u, err := network.URL()
[352]129 if err != nil {
[457]130 return nil, err
[352]131 }
[206]132
[269]133 var netConn net.Conn
[352]134 switch u.Scheme {
[269]135 case "ircs":
[352]136 addr := u.Host
[381]137 host, _, err := net.SplitHostPort(u.Host)
138 if err != nil {
139 host = u.Host
140 addr = u.Host + ":6697"
[269]141 }
142
[705]143 dialer.LocalAddr, err = network.user.localTCPAddrForHost(host)
144 if err != nil {
145 return nil, fmt.Errorf("failed to pick local IP for remote host %q: %v", host, err)
146 }
147
[269]148 logger.Printf("connecting to TLS server at address %q", addr)
[307]149
[455]150 tlsConfig := &tls.Config{ServerName: host, NextProtos: []string{"irc"}}
[307]151 if network.SASL.Mechanism == "EXTERNAL" {
152 if network.SASL.External.CertBlob == nil {
153 return nil, fmt.Errorf("missing certificate for authentication")
154 }
155 if network.SASL.External.PrivKeyBlob == nil {
156 return nil, fmt.Errorf("missing private key for authentication")
157 }
158 key, err := x509.ParsePKCS8PrivateKey(network.SASL.External.PrivKeyBlob)
159 if err != nil {
160 return nil, fmt.Errorf("failed to parse private key: %v", err)
161 }
[381]162 tlsConfig.Certificates = []tls.Certificate{
163 {
164 Certificate: [][]byte{network.SASL.External.CertBlob},
165 PrivateKey: key.(crypto.PrivateKey),
[307]166 },
167 }
168 logger.Printf("using TLS client certificate %x", sha256.Sum256(network.SASL.External.CertBlob))
169 }
170
[381]171 netConn, err = dialer.Dial("tcp", addr)
[352]172 if err != nil {
173 return nil, fmt.Errorf("failed to dial %q: %v", addr, err)
174 }
[381]175
176 // Don't do the TLS handshake immediately, because we need to register
177 // the new connection with identd ASAP. See:
178 // https://todo.sr.ht/~emersion/soju/69#event-41859
179 netConn = tls.Client(netConn, tlsConfig)
[270]180 case "irc+insecure":
[352]181 addr := u.Host
[705]182 host, _, err := net.SplitHostPort(addr)
183 if err != nil {
184 host = u.Host
185 addr = u.Host + ":6667"
[270]186 }
187
[705]188 dialer.LocalAddr, err = network.user.localTCPAddrForHost(host)
189 if err != nil {
190 return nil, fmt.Errorf("failed to pick local IP for remote host %q: %v", host, err)
191 }
192
[270]193 logger.Printf("connecting to plain-text server at address %q", addr)
194 netConn, err = dialer.Dial("tcp", addr)
[352]195 if err != nil {
196 return nil, fmt.Errorf("failed to dial %q: %v", addr, err)
197 }
[369]198 case "irc+unix", "unix":
[353]199 logger.Printf("connecting to Unix socket at path %q", u.Path)
200 netConn, err = dialer.Dial("unix", u.Path)
201 if err != nil {
202 return nil, fmt.Errorf("failed to connect to Unix socket %q: %v", u.Path, err)
203 }
[269]204 default:
[352]205 return nil, fmt.Errorf("failed to dial %q: unknown scheme: %v", network.Addr, u.Scheme)
[269]206 }
[33]207
[398]208 options := connOptions{
[402]209 Logger: logger,
[398]210 RateLimitDelay: upstreamMessageDelay,
211 RateLimitBurst: upstreamMessageBurst,
212 }
213
[55]214 uc := &upstreamConn{
[681]215 conn: *newConn(network.user.srv, newNetIRCConn(netConn), &options),
216 network: network,
217 user: network.user,
218 channels: upstreamChannelCasemapMap{newCasemapMap(0)},
219 supportedCaps: make(map[string]string),
220 caps: make(map[string]bool),
221 batches: make(map[string]batch),
222 availableChannelTypes: stdChannelTypes,
223 availableChannelModes: stdChannelModes,
224 availableMemberships: stdMemberships,
225 isupport: make(map[string]*string),
[682]226 pendingCmds: make(map[string][]pendingUpstreamCommand),
[684]227 monitored: monitorCasemapMap{newCasemapMap(0)},
[33]228 }
[55]229 return uc, nil
[33]230}
231
[73]232func (uc *upstreamConn) forEachDownstream(f func(*downstreamConn)) {
[218]233 uc.network.forEachDownstream(f)
[73]234}
235
[161]236func (uc *upstreamConn) forEachDownstreamByID(id uint64, f func(*downstreamConn)) {
[155]237 uc.forEachDownstream(func(dc *downstreamConn) {
238 if id != 0 && id != dc.id {
239 return
240 }
241 f(dc)
242 })
243}
244
[682]245func (uc *upstreamConn) downstreamByID(id uint64) *downstreamConn {
246 for _, dc := range uc.user.downstreamConns {
247 if dc.id == id {
248 return dc
249 }
250 }
251 return nil
252}
253
[55]254func (uc *upstreamConn) getChannel(name string) (*upstreamChannel, error) {
[478]255 ch := uc.channels.Value(name)
256 if ch == nil {
[19]257 return nil, fmt.Errorf("unknown channel %q", name)
258 }
259 return ch, nil
260}
261
[129]262func (uc *upstreamConn) isChannel(entity string) bool {
[454]263 return strings.ContainsRune(uc.availableChannelTypes, rune(entity[0]))
[129]264}
265
[478]266func (uc *upstreamConn) isOurNick(nick string) bool {
267 return uc.nickCM == uc.network.casemap(nick)
268}
269
[682]270func (uc *upstreamConn) endPendingCommands() {
271 for _, l := range uc.pendingCmds {
272 for _, pendingCmd := range l {
273 dc := uc.downstreamByID(pendingCmd.downstreamID)
274 if dc == nil {
275 continue
276 }
277
278 switch pendingCmd.msg.Command {
279 case "LIST":
280 dc.SendMessage(&irc.Message{
281 Prefix: dc.srv.prefix(),
282 Command: irc.RPL_LISTEND,
283 Params: []string{dc.nick, "End of /LIST"},
284 })
285 case "WHO":
286 mask := "*"
287 if len(pendingCmd.msg.Params) > 0 {
288 mask = pendingCmd.msg.Params[0]
289 }
290 dc.SendMessage(&irc.Message{
291 Prefix: dc.srv.prefix(),
292 Command: irc.RPL_ENDOFWHO,
293 Params: []string{dc.nick, mask, "End of /WHO"},
294 })
295 default:
296 panic(fmt.Errorf("Unsupported pending command %q", pendingCmd.msg.Command))
297 }
298 }
[177]299 }
[682]300
301 uc.pendingCmds = make(map[string][]pendingUpstreamCommand)
[177]302}
303
[682]304func (uc *upstreamConn) sendNextPendingCommand(cmd string) {
305 if len(uc.pendingCmds[cmd]) == 0 {
[681]306 return
[177]307 }
[682]308 uc.SendMessage(uc.pendingCmds[cmd][0].msg)
[177]309}
310
[682]311func (uc *upstreamConn) enqueueCommand(dc *downstreamConn, msg *irc.Message) {
312 switch msg.Command {
313 case "LIST", "WHO":
314 // Supported
315 default:
316 panic(fmt.Errorf("Unsupported pending command %q", msg.Command))
317 }
318
319 uc.pendingCmds[msg.Command] = append(uc.pendingCmds[msg.Command], pendingUpstreamCommand{
[681]320 downstreamID: dc.id,
[682]321 msg: msg,
[681]322 })
323
[682]324 if len(uc.pendingCmds[msg.Command]) == 1 {
325 uc.sendNextPendingCommand(msg.Command)
[177]326 }
[681]327}
[177]328
[682]329func (uc *upstreamConn) currentPendingCommand(cmd string) (*downstreamConn, *irc.Message) {
330 if len(uc.pendingCmds[cmd]) == 0 {
[681]331 return nil, nil
332 }
333
[682]334 pendingCmd := uc.pendingCmds[cmd][0]
335 return uc.downstreamByID(pendingCmd.downstreamID), pendingCmd.msg
[681]336}
[177]337
[682]338func (uc *upstreamConn) dequeueCommand(cmd string) (*downstreamConn, *irc.Message) {
339 dc, msg := uc.currentPendingCommand(cmd)
[681]340
[682]341 if len(uc.pendingCmds[cmd]) > 0 {
342 copy(uc.pendingCmds[cmd], uc.pendingCmds[cmd][1:])
343 uc.pendingCmds[cmd] = uc.pendingCmds[cmd][:len(uc.pendingCmds[cmd])-1]
[177]344 }
[681]345
[682]346 uc.sendNextPendingCommand(cmd)
[681]347
[682]348 return dc, msg
[177]349}
350
[292]351func (uc *upstreamConn) parseMembershipPrefix(s string) (ms *memberships, nick string) {
352 memberships := make(memberships, 0, 4)
353 i := 0
[139]354 for _, m := range uc.availableMemberships {
[292]355 if i >= len(s) {
356 break
[139]357 }
[292]358 if s[i] == m.Prefix {
359 memberships = append(memberships, m)
360 i++
361 }
[139]362 }
[292]363 return &memberships, s[i:]
[139]364}
365
[55]366func (uc *upstreamConn) handleMessage(msg *irc.Message) error {
[155]367 var label string
368 if l, ok := msg.GetTag("label"); ok {
369 label = l
[526]370 delete(msg.Tags, "label")
[155]371 }
372
[153]373 var msgBatch *batch
374 if batchName, ok := msg.GetTag("batch"); ok {
375 b, ok := uc.batches[batchName]
376 if !ok {
377 return fmt.Errorf("unexpected batch reference: batch was not defined: %q", batchName)
378 }
379 msgBatch = &b
[155]380 if label == "" {
381 label = msgBatch.Label
382 }
[443]383 delete(msg.Tags, "batch")
[153]384 }
385
[161]386 var downstreamID uint64 = 0
[155]387 if label != "" {
388 var labelOffset uint64
[161]389 n, err := fmt.Sscanf(label, "sd-%d-%d", &downstreamID, &labelOffset)
[155]390 if err == nil && n < 2 {
391 err = errors.New("not enough arguments")
392 }
393 if err != nil {
394 return fmt.Errorf("unexpected message label: invalid downstream reference for label %q: %v", label, err)
395 }
396 }
397
[216]398 if _, ok := msg.Tags["time"]; !ok {
[240]399 msg.Tags["time"] = irc.TagValue(time.Now().UTC().Format(serverTimeLayout))
[216]400 }
401
[13]402 switch msg.Command {
403 case "PING":
[60]404 uc.SendMessage(&irc.Message{
[13]405 Command: "PONG",
[68]406 Params: msg.Params,
[60]407 })
[33]408 return nil
[303]409 case "NOTICE", "PRIVMSG", "TAGMSG":
[273]410 if msg.Prefix == nil {
411 return fmt.Errorf("expected a prefix")
412 }
413
[286]414 var entity, text string
[303]415 if msg.Command != "TAGMSG" {
416 if err := parseMessageParams(msg, &entity, &text); err != nil {
417 return err
418 }
419 } else {
420 if err := parseMessageParams(msg, &entity); err != nil {
421 return err
422 }
[286]423 }
424
425 if msg.Prefix.Name == serviceNick {
426 uc.logger.Printf("skipping %v from soju's service: %v", msg.Command, msg)
427 break
428 }
429 if entity == serviceNick {
430 uc.logger.Printf("skipping %v to soju's service: %v", msg.Command, msg)
431 break
432 }
433
[171]434 if msg.Prefix.User == "" && msg.Prefix.Host == "" { // server message
[239]435 uc.produce("", msg, nil)
[303]436 } else { // regular user message
[217]437 target := entity
[478]438 if uc.isOurNick(target) {
[178]439 target = msg.Prefix.Name
440 }
[287]441
[478]442 ch := uc.network.channels.Value(target)
[505]443 if ch != nil && msg.Command != "TAGMSG" {
[435]444 if ch.Detached {
[499]445 uc.handleDetachedMessage(ch, msg)
[435]446 }
447
[499]448 highlight := uc.network.isHighlight(msg)
[435]449 if ch.DetachOn == FilterMessage || ch.DetachOn == FilterDefault || (ch.DetachOn == FilterHighlight && highlight) {
450 uc.updateChannelAutoDetach(target)
451 }
[287]452 }
[435]453
454 uc.produce(target, msg, nil)
[171]455 }
[92]456 case "CAP":
[95]457 var subCmd string
458 if err := parseMessageParams(msg, nil, &subCmd); err != nil {
459 return err
[92]460 }
[95]461 subCmd = strings.ToUpper(subCmd)
462 subParams := msg.Params[2:]
463 switch subCmd {
464 case "LS":
465 if len(subParams) < 1 {
466 return newNeedMoreParamsError(msg.Command)
467 }
[281]468 caps := subParams[len(subParams)-1]
[95]469 more := len(subParams) >= 2 && msg.Params[len(subParams)-2] == "*"
[92]470
[281]471 uc.handleSupportedCaps(caps)
[92]472
[95]473 if more {
474 break // wait to receive all capabilities
475 }
476
[281]477 uc.requestCaps()
[152]478
[95]479 if uc.requestSASL() {
480 break // we'll send CAP END after authentication is completed
481 }
482
[92]483 uc.SendMessage(&irc.Message{
484 Command: "CAP",
485 Params: []string{"END"},
486 })
[95]487 case "ACK", "NAK":
488 if len(subParams) < 1 {
489 return newNeedMoreParamsError(msg.Command)
490 }
491 caps := strings.Fields(subParams[0])
492
493 for _, name := range caps {
494 if err := uc.handleCapAck(strings.ToLower(name), subCmd == "ACK"); err != nil {
495 return err
496 }
497 }
498
[281]499 if uc.registered {
500 uc.forEachDownstream(func(dc *downstreamConn) {
501 dc.updateSupportedCaps()
502 })
503 }
504 case "NEW":
505 if len(subParams) < 1 {
506 return newNeedMoreParamsError(msg.Command)
507 }
508 uc.handleSupportedCaps(subParams[0])
509 uc.requestCaps()
510 case "DEL":
511 if len(subParams) < 1 {
512 return newNeedMoreParamsError(msg.Command)
513 }
514 caps := strings.Fields(subParams[0])
515
516 for _, c := range caps {
517 delete(uc.supportedCaps, c)
518 delete(uc.caps, c)
519 }
520
521 if uc.registered {
522 uc.forEachDownstream(func(dc *downstreamConn) {
523 dc.updateSupportedCaps()
524 })
525 }
[95]526 default:
527 uc.logger.Printf("unhandled message: %v", msg)
[92]528 }
[95]529 case "AUTHENTICATE":
530 if uc.saslClient == nil {
531 return fmt.Errorf("received unexpected AUTHENTICATE message")
532 }
533
534 // TODO: if a challenge is 400 bytes long, buffer it
535 var challengeStr string
536 if err := parseMessageParams(msg, &challengeStr); err != nil {
537 uc.SendMessage(&irc.Message{
538 Command: "AUTHENTICATE",
539 Params: []string{"*"},
540 })
541 return err
542 }
543
544 var challenge []byte
545 if challengeStr != "+" {
546 var err error
547 challenge, err = base64.StdEncoding.DecodeString(challengeStr)
548 if err != nil {
549 uc.SendMessage(&irc.Message{
550 Command: "AUTHENTICATE",
551 Params: []string{"*"},
552 })
553 return err
554 }
555 }
556
557 var resp []byte
558 var err error
559 if !uc.saslStarted {
560 _, resp, err = uc.saslClient.Start()
561 uc.saslStarted = true
562 } else {
563 resp, err = uc.saslClient.Next(challenge)
564 }
565 if err != nil {
566 uc.SendMessage(&irc.Message{
567 Command: "AUTHENTICATE",
568 Params: []string{"*"},
569 })
570 return err
571 }
572
573 // TODO: send response in multiple chunks if >= 400 bytes
574 var respStr = "+"
[318]575 if len(resp) != 0 {
[95]576 respStr = base64.StdEncoding.EncodeToString(resp)
577 }
578
579 uc.SendMessage(&irc.Message{
580 Command: "AUTHENTICATE",
581 Params: []string{respStr},
582 })
[125]583 case irc.RPL_LOGGEDIN:
[559]584 if err := parseMessageParams(msg, nil, nil, &uc.account); err != nil {
[95]585 return err
586 }
[559]587 uc.logger.Printf("logged in with account %q", uc.account)
[125]588 case irc.RPL_LOGGEDOUT:
[559]589 uc.account = ""
[95]590 uc.logger.Printf("logged out")
[125]591 case irc.ERR_NICKLOCKED, irc.RPL_SASLSUCCESS, irc.ERR_SASLFAIL, irc.ERR_SASLTOOLONG, irc.ERR_SASLABORTED:
[95]592 var info string
593 if err := parseMessageParams(msg, nil, &info); err != nil {
594 return err
595 }
596 switch msg.Command {
[125]597 case irc.ERR_NICKLOCKED:
[95]598 uc.logger.Printf("invalid nick used with SASL authentication: %v", info)
[125]599 case irc.ERR_SASLFAIL:
[95]600 uc.logger.Printf("SASL authentication failed: %v", info)
[125]601 case irc.ERR_SASLTOOLONG:
[95]602 uc.logger.Printf("SASL message too long: %v", info)
603 }
604
605 uc.saslClient = nil
606 uc.saslStarted = false
607
608 uc.SendMessage(&irc.Message{
609 Command: "CAP",
610 Params: []string{"END"},
611 })
[14]612 case irc.RPL_WELCOME:
[55]613 uc.registered = true
614 uc.logger.Printf("connection registered")
[19]615
[478]616 if uc.network.channels.Len() > 0 {
[350]617 var channels, keys []string
[478]618 for _, entry := range uc.network.channels.innerMap {
619 ch := entry.value.(*Channel)
[350]620 channels = append(channels, ch.Name)
[310]621 keys = append(keys, ch.Key)
622 }
[350]623
624 for _, msg := range join(channels, keys) {
625 uc.SendMessage(msg)
626 }
[19]627 }
[16]628 case irc.RPL_MYINFO:
[139]629 if err := parseMessageParams(msg, nil, &uc.serverName, nil, &uc.availableUserModes, nil); err != nil {
[43]630 return err
[16]631 }
[139]632 case irc.RPL_ISUPPORT:
633 if err := parseMessageParams(msg, nil, nil); err != nil {
634 return err
[16]635 }
[463]636
637 var downstreamIsupport []string
[139]638 for _, token := range msg.Params[1 : len(msg.Params)-1] {
639 parameter := token
[460]640 var negate, hasValue bool
641 var value string
[139]642 if strings.HasPrefix(token, "-") {
643 negate = true
644 token = token[1:]
[459]645 } else if i := strings.IndexByte(token, '='); i >= 0 {
646 parameter = token[:i]
647 value = token[i+1:]
[460]648 hasValue = true
[139]649 }
[460]650
651 if hasValue {
652 uc.isupport[parameter] = &value
653 } else if !negate {
654 uc.isupport[parameter] = nil
655 } else {
656 delete(uc.isupport, parameter)
657 }
658
[462]659 var err error
660 switch parameter {
[478]661 case "CASEMAPPING":
662 casemap, ok := parseCasemappingToken(value)
663 if !ok {
664 casemap = casemapRFC1459
665 }
666 uc.network.updateCasemapping(casemap)
667 uc.nickCM = uc.network.casemap(uc.nick)
668 uc.casemapIsSet = true
[462]669 case "CHANMODES":
670 if !negate {
671 err = uc.handleChanModes(value)
672 } else {
673 uc.availableChannelModes = stdChannelModes
674 }
675 case "CHANTYPES":
676 if !negate {
[139]677 uc.availableChannelTypes = value
[462]678 } else {
679 uc.availableChannelTypes = stdChannelTypes
[139]680 }
[462]681 case "PREFIX":
682 if !negate {
683 err = uc.handleMemberships(value)
684 } else {
685 uc.availableMemberships = stdMemberships
686 }
[139]687 }
[462]688 if err != nil {
689 return err
690 }
[463]691
692 if passthroughIsupport[parameter] {
693 downstreamIsupport = append(downstreamIsupport, token)
694 }
[139]695 }
[463]696
697 uc.forEachDownstream(func(dc *downstreamConn) {
698 if dc.network == nil {
699 return
700 }
701 msgs := generateIsupport(dc.srv.prefix(), dc.nick, downstreamIsupport)
702 for _, msg := range msgs {
703 dc.SendMessage(msg)
704 }
705 })
[478]706 case irc.ERR_NOMOTD, irc.RPL_ENDOFMOTD:
707 if !uc.casemapIsSet {
708 // upstream did not send any CASEMAPPING token, thus
709 // we assume it implements the old RFCs with rfc1459.
710 uc.casemapIsSet = true
711 uc.network.updateCasemapping(casemapRFC1459)
712 uc.nickCM = uc.network.casemap(uc.nick)
713 }
[552]714
715 if !uc.gotMotd {
716 // Ignore the initial MOTD upon connection, but forward
717 // subsequent MOTD messages downstream
718 uc.gotMotd = true
719 return nil
720 }
721
[553]722 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
723 dc.SendMessage(&irc.Message{
724 Prefix: uc.srv.prefix(),
[552]725 Command: msg.Command,
[553]726 Params: msg.Params,
[552]727 })
728 })
[153]729 case "BATCH":
730 var tag string
731 if err := parseMessageParams(msg, &tag); err != nil {
732 return err
733 }
734
735 if strings.HasPrefix(tag, "+") {
736 tag = tag[1:]
737 if _, ok := uc.batches[tag]; ok {
738 return fmt.Errorf("unexpected BATCH reference tag: batch was already defined: %q", tag)
739 }
740 var batchType string
741 if err := parseMessageParams(msg, nil, &batchType); err != nil {
742 return err
743 }
[155]744 label := label
745 if label == "" && msgBatch != nil {
746 label = msgBatch.Label
747 }
[153]748 uc.batches[tag] = batch{
749 Type: batchType,
750 Params: msg.Params[2:],
751 Outer: msgBatch,
[155]752 Label: label,
[153]753 }
754 } else if strings.HasPrefix(tag, "-") {
755 tag = tag[1:]
756 if _, ok := uc.batches[tag]; !ok {
757 return fmt.Errorf("unknown BATCH reference tag: %q", tag)
758 }
759 delete(uc.batches, tag)
760 } else {
761 return fmt.Errorf("unexpected BATCH reference tag: missing +/- prefix: %q", tag)
762 }
[42]763 case "NICK":
[83]764 if msg.Prefix == nil {
765 return fmt.Errorf("expected a prefix")
766 }
767
[43]768 var newNick string
769 if err := parseMessageParams(msg, &newNick); err != nil {
770 return err
[42]771 }
772
[244]773 me := false
[478]774 if uc.isOurNick(msg.Prefix.Name) {
[55]775 uc.logger.Printf("changed nick from %q to %q", uc.nick, newNick)
[244]776 me = true
[55]777 uc.nick = newNick
[478]778 uc.nickCM = uc.network.casemap(uc.nick)
[42]779 }
780
[478]781 for _, entry := range uc.channels.innerMap {
782 ch := entry.value.(*upstreamChannel)
783 memberships := ch.Members.Value(msg.Prefix.Name)
784 if memberships != nil {
785 ch.Members.Delete(msg.Prefix.Name)
786 ch.Members.SetValue(newNick, memberships)
[215]787 uc.appendLog(ch.Name, msg)
[42]788 }
789 }
[82]790
[244]791 if !me {
[82]792 uc.forEachDownstream(func(dc *downstreamConn) {
[261]793 dc.SendMessage(dc.marshalMessage(msg, uc.network))
[82]794 })
[296]795 } else {
796 uc.forEachDownstream(func(dc *downstreamConn) {
797 dc.updateNick()
798 })
[82]799 }
[540]800 case "SETNAME":
801 if msg.Prefix == nil {
802 return fmt.Errorf("expected a prefix")
803 }
804
805 var newRealname string
806 if err := parseMessageParams(msg, &newRealname); err != nil {
807 return err
808 }
809
810 // TODO: consider appending this message to logs
811
812 if uc.isOurNick(msg.Prefix.Name) {
813 uc.logger.Printf("changed realname from %q to %q", uc.realname, newRealname)
814 uc.realname = newRealname
815
816 uc.forEachDownstream(func(dc *downstreamConn) {
817 dc.updateRealname()
818 })
819 } else {
820 uc.forEachDownstream(func(dc *downstreamConn) {
821 dc.SendMessage(dc.marshalMessage(msg, uc.network))
822 })
823 }
[69]824 case "JOIN":
825 if msg.Prefix == nil {
826 return fmt.Errorf("expected a prefix")
827 }
[42]828
[43]829 var channels string
830 if err := parseMessageParams(msg, &channels); err != nil {
831 return err
[19]832 }
[34]833
[43]834 for _, ch := range strings.Split(channels, ",") {
[478]835 if uc.isOurNick(msg.Prefix.Name) {
[55]836 uc.logger.Printf("joined channel %q", ch)
[478]837 members := membershipsCasemapMap{newCasemapMap(0)}
838 members.casemap = uc.network.casemap
839 uc.channels.SetValue(ch, &upstreamChannel{
[34]840 Name: ch,
[55]841 conn: uc,
[478]842 Members: members,
843 })
[435]844 uc.updateChannelAutoDetach(ch)
[139]845
846 uc.SendMessage(&irc.Message{
847 Command: "MODE",
848 Params: []string{ch},
849 })
[34]850 } else {
[55]851 ch, err := uc.getChannel(ch)
[34]852 if err != nil {
853 return err
854 }
[478]855 ch.Members.SetValue(msg.Prefix.Name, &memberships{})
[19]856 }
[69]857
[245]858 chMsg := msg.Copy()
859 chMsg.Params[0] = ch
860 uc.produce(ch, chMsg, nil)
[19]861 }
[69]862 case "PART":
863 if msg.Prefix == nil {
864 return fmt.Errorf("expected a prefix")
865 }
[34]866
[43]867 var channels string
868 if err := parseMessageParams(msg, &channels); err != nil {
869 return err
[34]870 }
871
[43]872 for _, ch := range strings.Split(channels, ",") {
[478]873 if uc.isOurNick(msg.Prefix.Name) {
[55]874 uc.logger.Printf("parted channel %q", ch)
[478]875 uch := uc.channels.Value(ch)
876 if uch != nil {
877 uc.channels.Delete(ch)
[435]878 uch.updateAutoDetach(0)
879 }
[34]880 } else {
[55]881 ch, err := uc.getChannel(ch)
[34]882 if err != nil {
883 return err
884 }
[478]885 ch.Members.Delete(msg.Prefix.Name)
[34]886 }
[69]887
[245]888 chMsg := msg.Copy()
889 chMsg.Params[0] = ch
890 uc.produce(ch, chMsg, nil)
[34]891 }
[159]892 case "KICK":
893 if msg.Prefix == nil {
894 return fmt.Errorf("expected a prefix")
895 }
896
897 var channel, user string
898 if err := parseMessageParams(msg, &channel, &user); err != nil {
899 return err
900 }
901
[478]902 if uc.isOurNick(user) {
[159]903 uc.logger.Printf("kicked from channel %q by %s", channel, msg.Prefix.Name)
[478]904 uc.channels.Delete(channel)
[159]905 } else {
906 ch, err := uc.getChannel(channel)
907 if err != nil {
908 return err
909 }
[478]910 ch.Members.Delete(user)
[159]911 }
912
[245]913 uc.produce(channel, msg, nil)
[83]914 case "QUIT":
915 if msg.Prefix == nil {
916 return fmt.Errorf("expected a prefix")
917 }
918
[478]919 if uc.isOurNick(msg.Prefix.Name) {
[83]920 uc.logger.Printf("quit")
921 }
922
[478]923 for _, entry := range uc.channels.innerMap {
924 ch := entry.value.(*upstreamChannel)
925 if ch.Members.Has(msg.Prefix.Name) {
926 ch.Members.Delete(msg.Prefix.Name)
[178]927
[215]928 uc.appendLog(ch.Name, msg)
[178]929 }
[83]930 }
931
932 if msg.Prefix.Name != uc.nick {
933 uc.forEachDownstream(func(dc *downstreamConn) {
[261]934 dc.SendMessage(dc.marshalMessage(msg, uc.network))
[83]935 })
936 }
[19]937 case irc.RPL_TOPIC, irc.RPL_NOTOPIC:
[43]938 var name, topic string
939 if err := parseMessageParams(msg, nil, &name, &topic); err != nil {
940 return err
[19]941 }
[55]942 ch, err := uc.getChannel(name)
[19]943 if err != nil {
944 return err
945 }
946 if msg.Command == irc.RPL_TOPIC {
[43]947 ch.Topic = topic
[19]948 } else {
949 ch.Topic = ""
950 }
951 case "TOPIC":
[405]952 if msg.Prefix == nil {
953 return fmt.Errorf("expected a prefix")
954 }
955
[43]956 var name string
[74]957 if err := parseMessageParams(msg, &name); err != nil {
[43]958 return err
[19]959 }
[55]960 ch, err := uc.getChannel(name)
[19]961 if err != nil {
962 return err
963 }
964 if len(msg.Params) > 1 {
965 ch.Topic = msg.Params[1]
[405]966 ch.TopicWho = msg.Prefix.Copy()
967 ch.TopicTime = time.Now() // TODO use msg.Tags["time"]
[19]968 } else {
969 ch.Topic = ""
970 }
[245]971 uc.produce(ch.Name, msg, nil)
[139]972 case "MODE":
973 var name, modeStr string
974 if err := parseMessageParams(msg, &name, &modeStr); err != nil {
975 return err
976 }
977
978 if !uc.isChannel(name) { // user mode change
979 if name != uc.nick {
980 return fmt.Errorf("received MODE message for unknown nick %q", name)
981 }
[553]982
983 if err := uc.modes.Apply(modeStr); err != nil {
984 return err
985 }
986
987 uc.forEachDownstream(func(dc *downstreamConn) {
988 if dc.upstream() == nil {
989 return
990 }
991
992 dc.SendMessage(msg)
993 })
[139]994 } else { // channel mode change
995 ch, err := uc.getChannel(name)
996 if err != nil {
997 return err
998 }
999
[293]1000 needMarshaling, err := applyChannelModes(ch, modeStr, msg.Params[2:])
1001 if err != nil {
1002 return err
[139]1003 }
1004
[293]1005 uc.appendLog(ch.Name, msg)
1006
[478]1007 c := uc.network.channels.Value(name)
1008 if c == nil || !c.Detached {
[338]1009 uc.forEachDownstream(func(dc *downstreamConn) {
1010 params := make([]string, len(msg.Params))
1011 params[0] = dc.marshalEntity(uc.network, name)
1012 params[1] = modeStr
1013
1014 copy(params[2:], msg.Params[2:])
1015 for i, modeParam := range params[2:] {
1016 if _, ok := needMarshaling[i]; ok {
1017 params[2+i] = dc.marshalEntity(uc.network, modeParam)
1018 }
[293]1019 }
1020
[338]1021 dc.SendMessage(&irc.Message{
1022 Prefix: dc.marshalUserPrefix(uc.network, msg.Prefix),
1023 Command: "MODE",
1024 Params: params,
1025 })
[293]1026 })
[338]1027 }
[139]1028 }
1029 case irc.RPL_UMODEIS:
1030 if err := parseMessageParams(msg, nil); err != nil {
1031 return err
1032 }
1033 modeStr := ""
1034 if len(msg.Params) > 1 {
1035 modeStr = msg.Params[1]
1036 }
1037
1038 uc.modes = ""
1039 if err := uc.modes.Apply(modeStr); err != nil {
1040 return err
1041 }
[553]1042
1043 uc.forEachDownstream(func(dc *downstreamConn) {
1044 if dc.upstream() == nil {
1045 return
1046 }
1047
1048 dc.SendMessage(msg)
1049 })
[139]1050 case irc.RPL_CHANNELMODEIS:
1051 var channel string
1052 if err := parseMessageParams(msg, nil, &channel); err != nil {
1053 return err
1054 }
1055 modeStr := ""
1056 if len(msg.Params) > 2 {
1057 modeStr = msg.Params[2]
1058 }
1059
1060 ch, err := uc.getChannel(channel)
1061 if err != nil {
1062 return err
1063 }
1064
1065 firstMode := ch.modes == nil
1066 ch.modes = make(map[byte]string)
[293]1067 if _, err := applyChannelModes(ch, modeStr, msg.Params[3:]); err != nil {
[139]1068 return err
1069 }
1070 if firstMode {
[478]1071 c := uc.network.channels.Value(channel)
1072 if c == nil || !c.Detached {
[338]1073 modeStr, modeParams := ch.modes.Format()
[139]1074
[338]1075 uc.forEachDownstream(func(dc *downstreamConn) {
1076 params := []string{dc.nick, dc.marshalEntity(uc.network, channel), modeStr}
1077 params = append(params, modeParams...)
[139]1078
[338]1079 dc.SendMessage(&irc.Message{
1080 Prefix: dc.srv.prefix(),
1081 Command: irc.RPL_CHANNELMODEIS,
1082 Params: params,
1083 })
[139]1084 })
[338]1085 }
[139]1086 }
[162]1087 case rpl_creationtime:
1088 var channel, creationTime string
1089 if err := parseMessageParams(msg, nil, &channel, &creationTime); err != nil {
1090 return err
1091 }
1092
1093 ch, err := uc.getChannel(channel)
1094 if err != nil {
1095 return err
1096 }
1097
1098 firstCreationTime := ch.creationTime == ""
1099 ch.creationTime = creationTime
1100 if firstCreationTime {
1101 uc.forEachDownstream(func(dc *downstreamConn) {
1102 dc.SendMessage(&irc.Message{
1103 Prefix: dc.srv.prefix(),
1104 Command: rpl_creationtime,
[403]1105 Params: []string{dc.nick, dc.marshalEntity(uc.network, ch.Name), creationTime},
[162]1106 })
1107 })
1108 }
[19]1109 case rpl_topicwhotime:
[43]1110 var name, who, timeStr string
1111 if err := parseMessageParams(msg, nil, &name, &who, &timeStr); err != nil {
1112 return err
[19]1113 }
[55]1114 ch, err := uc.getChannel(name)
[19]1115 if err != nil {
1116 return err
1117 }
[405]1118 firstTopicWhoTime := ch.TopicWho == nil
1119 ch.TopicWho = irc.ParsePrefix(who)
[43]1120 sec, err := strconv.ParseInt(timeStr, 10, 64)
[19]1121 if err != nil {
1122 return fmt.Errorf("failed to parse topic time: %v", err)
1123 }
1124 ch.TopicTime = time.Unix(sec, 0)
[405]1125 if firstTopicWhoTime {
1126 uc.forEachDownstream(func(dc *downstreamConn) {
1127 topicWho := dc.marshalUserPrefix(uc.network, ch.TopicWho)
1128 dc.SendMessage(&irc.Message{
1129 Prefix: dc.srv.prefix(),
1130 Command: rpl_topicwhotime,
1131 Params: []string{
1132 dc.nick,
1133 dc.marshalEntity(uc.network, ch.Name),
1134 topicWho.String(),
1135 timeStr,
1136 },
1137 })
1138 })
1139 }
[177]1140 case irc.RPL_LIST:
1141 var channel, clients, topic string
1142 if err := parseMessageParams(msg, nil, &channel, &clients, &topic); err != nil {
1143 return err
1144 }
1145
[682]1146 dc, cmd := uc.currentPendingCommand("LIST")
[681]1147 if cmd == nil {
[177]1148 return fmt.Errorf("unexpected RPL_LIST: no matching pending LIST")
[681]1149 } else if dc == nil {
1150 return nil
[177]1151 }
1152
[681]1153 dc.SendMessage(&irc.Message{
1154 Prefix: dc.srv.prefix(),
1155 Command: irc.RPL_LIST,
1156 Params: []string{dc.nick, dc.marshalEntity(uc.network, channel), clients, topic},
[177]1157 })
1158 case irc.RPL_LISTEND:
[682]1159 dc, cmd := uc.dequeueCommand("LIST")
[681]1160 if cmd == nil {
[177]1161 return fmt.Errorf("unexpected RPL_LISTEND: no matching pending LIST")
[681]1162 } else if dc == nil {
1163 return nil
[177]1164 }
[681]1165
1166 dc.SendMessage(&irc.Message{
1167 Prefix: dc.srv.prefix(),
1168 Command: irc.RPL_LISTEND,
1169 Params: []string{dc.nick, "End of /LIST"},
1170 })
[19]1171 case irc.RPL_NAMREPLY:
[43]1172 var name, statusStr, members string
1173 if err := parseMessageParams(msg, nil, &statusStr, &name, &members); err != nil {
1174 return err
[19]1175 }
[140]1176
[478]1177 ch := uc.channels.Value(name)
1178 if ch == nil {
[140]1179 // NAMES on a channel we have not joined, forward to downstream
[161]1180 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1181 channel := dc.marshalEntity(uc.network, name)
[174]1182 members := splitSpace(members)
[140]1183 for i, member := range members {
[292]1184 memberships, nick := uc.parseMembershipPrefix(member)
1185 members[i] = memberships.Format(dc) + dc.marshalEntity(uc.network, nick)
[140]1186 }
1187 memberStr := strings.Join(members, " ")
1188
1189 dc.SendMessage(&irc.Message{
1190 Prefix: dc.srv.prefix(),
1191 Command: irc.RPL_NAMREPLY,
1192 Params: []string{dc.nick, statusStr, channel, memberStr},
1193 })
1194 })
1195 return nil
[19]1196 }
1197
[43]1198 status, err := parseChannelStatus(statusStr)
[19]1199 if err != nil {
1200 return err
1201 }
1202 ch.Status = status
1203
[174]1204 for _, s := range splitSpace(members) {
[292]1205 memberships, nick := uc.parseMembershipPrefix(s)
[478]1206 ch.Members.SetValue(nick, memberships)
[19]1207 }
1208 case irc.RPL_ENDOFNAMES:
[43]1209 var name string
1210 if err := parseMessageParams(msg, nil, &name); err != nil {
1211 return err
[25]1212 }
[140]1213
[478]1214 ch := uc.channels.Value(name)
1215 if ch == nil {
[140]1216 // NAMES on a channel we have not joined, forward to downstream
[161]1217 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1218 channel := dc.marshalEntity(uc.network, name)
[140]1219
1220 dc.SendMessage(&irc.Message{
1221 Prefix: dc.srv.prefix(),
1222 Command: irc.RPL_ENDOFNAMES,
1223 Params: []string{dc.nick, channel, "End of /NAMES list"},
1224 })
1225 })
1226 return nil
[25]1227 }
1228
[34]1229 if ch.complete {
1230 return fmt.Errorf("received unexpected RPL_ENDOFNAMES")
1231 }
[25]1232 ch.complete = true
[27]1233
[478]1234 c := uc.network.channels.Value(name)
1235 if c == nil || !c.Detached {
[338]1236 uc.forEachDownstream(func(dc *downstreamConn) {
1237 forwardChannel(dc, ch)
1238 })
1239 }
[127]1240 case irc.RPL_WHOREPLY:
1241 var channel, username, host, server, nick, mode, trailing string
1242 if err := parseMessageParams(msg, nil, &channel, &username, &host, &server, &nick, &mode, &trailing); err != nil {
1243 return err
1244 }
1245
[682]1246 dc, cmd := uc.currentPendingCommand("WHO")
1247 if cmd == nil {
1248 return fmt.Errorf("unexpected RPL_WHOREPLY: no matching pending WHO")
1249 } else if dc == nil {
1250 return nil
1251 }
1252
[127]1253 parts := strings.SplitN(trailing, " ", 2)
1254 if len(parts) != 2 {
1255 return fmt.Errorf("received malformed RPL_WHOREPLY: wrong trailing parameter: %s", trailing)
1256 }
1257 realname := parts[1]
1258 hops, err := strconv.Atoi(parts[0])
1259 if err != nil {
1260 return fmt.Errorf("received malformed RPL_WHOREPLY: wrong hop count: %s", parts[0])
1261 }
1262 hops++
1263
1264 trailing = strconv.Itoa(hops) + " " + realname
1265
[682]1266 if channel != "*" {
1267 channel = dc.marshalEntity(uc.network, channel)
1268 }
1269 nick = dc.marshalEntity(uc.network, nick)
1270 dc.SendMessage(&irc.Message{
1271 Prefix: dc.srv.prefix(),
1272 Command: irc.RPL_WHOREPLY,
1273 Params: []string{dc.nick, channel, username, host, server, nick, mode, trailing},
[127]1274 })
[682]1275 case rpl_whospcrpl:
1276 dc, cmd := uc.currentPendingCommand("WHO")
1277 if cmd == nil {
1278 return fmt.Errorf("unexpected RPL_WHOSPCRPL: no matching pending WHO")
1279 } else if dc == nil {
1280 return nil
1281 }
1282
1283 // Only supported in single-upstream mode, so forward as-is
1284 dc.SendMessage(msg)
[127]1285 case irc.RPL_ENDOFWHO:
1286 var name string
1287 if err := parseMessageParams(msg, nil, &name); err != nil {
1288 return err
1289 }
1290
[682]1291 dc, cmd := uc.dequeueCommand("WHO")
1292 if cmd == nil {
1293 return fmt.Errorf("unexpected RPL_ENDOFWHO: no matching pending WHO")
1294 } else if dc == nil {
1295 return nil
1296 }
1297
1298 mask := "*"
1299 if len(cmd.Params) > 0 {
1300 mask = cmd.Params[0]
1301 }
1302 dc.SendMessage(&irc.Message{
1303 Prefix: dc.srv.prefix(),
1304 Command: irc.RPL_ENDOFWHO,
1305 Params: []string{dc.nick, mask, "End of /WHO list"},
[127]1306 })
[128]1307 case irc.RPL_WHOISUSER:
1308 var nick, username, host, realname string
1309 if err := parseMessageParams(msg, nil, &nick, &username, &host, nil, &realname); err != nil {
1310 return err
1311 }
1312
[161]1313 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1314 nick := dc.marshalEntity(uc.network, nick)
[128]1315 dc.SendMessage(&irc.Message{
1316 Prefix: dc.srv.prefix(),
1317 Command: irc.RPL_WHOISUSER,
1318 Params: []string{dc.nick, nick, username, host, "*", realname},
1319 })
1320 })
1321 case irc.RPL_WHOISSERVER:
1322 var nick, server, serverInfo string
1323 if err := parseMessageParams(msg, nil, &nick, &server, &serverInfo); err != nil {
1324 return err
1325 }
1326
[161]1327 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1328 nick := dc.marshalEntity(uc.network, nick)
[128]1329 dc.SendMessage(&irc.Message{
1330 Prefix: dc.srv.prefix(),
1331 Command: irc.RPL_WHOISSERVER,
1332 Params: []string{dc.nick, nick, server, serverInfo},
1333 })
1334 })
1335 case irc.RPL_WHOISOPERATOR:
1336 var nick string
1337 if err := parseMessageParams(msg, nil, &nick); err != nil {
1338 return err
1339 }
1340
[161]1341 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1342 nick := dc.marshalEntity(uc.network, nick)
[128]1343 dc.SendMessage(&irc.Message{
1344 Prefix: dc.srv.prefix(),
1345 Command: irc.RPL_WHOISOPERATOR,
1346 Params: []string{dc.nick, nick, "is an IRC operator"},
1347 })
1348 })
1349 case irc.RPL_WHOISIDLE:
1350 var nick string
1351 if err := parseMessageParams(msg, nil, &nick, nil); err != nil {
1352 return err
1353 }
1354
[161]1355 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1356 nick := dc.marshalEntity(uc.network, nick)
[128]1357 params := []string{dc.nick, nick}
1358 params = append(params, msg.Params[2:]...)
1359 dc.SendMessage(&irc.Message{
1360 Prefix: dc.srv.prefix(),
1361 Command: irc.RPL_WHOISIDLE,
1362 Params: params,
1363 })
1364 })
1365 case irc.RPL_WHOISCHANNELS:
1366 var nick, channelList string
1367 if err := parseMessageParams(msg, nil, &nick, &channelList); err != nil {
1368 return err
1369 }
[174]1370 channels := splitSpace(channelList)
[128]1371
[161]1372 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1373 nick := dc.marshalEntity(uc.network, nick)
[128]1374 channelList := make([]string, len(channels))
1375 for i, channel := range channels {
[139]1376 prefix, channel := uc.parseMembershipPrefix(channel)
[260]1377 channel = dc.marshalEntity(uc.network, channel)
[292]1378 channelList[i] = prefix.Format(dc) + channel
[128]1379 }
1380 channels := strings.Join(channelList, " ")
1381 dc.SendMessage(&irc.Message{
1382 Prefix: dc.srv.prefix(),
1383 Command: irc.RPL_WHOISCHANNELS,
1384 Params: []string{dc.nick, nick, channels},
1385 })
1386 })
1387 case irc.RPL_ENDOFWHOIS:
1388 var nick string
1389 if err := parseMessageParams(msg, nil, &nick); err != nil {
1390 return err
1391 }
1392
[161]1393 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[260]1394 nick := dc.marshalEntity(uc.network, nick)
[128]1395 dc.SendMessage(&irc.Message{
1396 Prefix: dc.srv.prefix(),
1397 Command: irc.RPL_ENDOFWHOIS,
[142]1398 Params: []string{dc.nick, nick, "End of /WHOIS list"},
[128]1399 })
1400 })
[115]1401 case "INVITE":
[273]1402 var nick, channel string
[115]1403 if err := parseMessageParams(msg, &nick, &channel); err != nil {
1404 return err
1405 }
1406
[478]1407 weAreInvited := uc.isOurNick(nick)
[448]1408
[115]1409 uc.forEachDownstream(func(dc *downstreamConn) {
[448]1410 if !weAreInvited && !dc.caps["invite-notify"] {
1411 return
1412 }
[115]1413 dc.SendMessage(&irc.Message{
[260]1414 Prefix: dc.marshalUserPrefix(uc.network, msg.Prefix),
[115]1415 Command: "INVITE",
[260]1416 Params: []string{dc.marshalEntity(uc.network, nick), dc.marshalEntity(uc.network, channel)},
[115]1417 })
1418 })
[163]1419 case irc.RPL_INVITING:
[273]1420 var nick, channel string
[304]1421 if err := parseMessageParams(msg, nil, &nick, &channel); err != nil {
[163]1422 return err
1423 }
1424
1425 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1426 dc.SendMessage(&irc.Message{
1427 Prefix: dc.srv.prefix(),
1428 Command: irc.RPL_INVITING,
[260]1429 Params: []string{dc.nick, dc.marshalEntity(uc.network, nick), dc.marshalEntity(uc.network, channel)},
[163]1430 })
1431 })
[684]1432 case irc.RPL_MONONLINE, irc.RPL_MONOFFLINE:
1433 var targetsStr string
1434 if err := parseMessageParams(msg, nil, &targetsStr); err != nil {
1435 return err
1436 }
1437 targets := strings.Split(targetsStr, ",")
1438
1439 online := msg.Command == irc.RPL_MONONLINE
1440 for _, target := range targets {
1441 prefix := irc.ParsePrefix(target)
1442 uc.monitored.SetValue(prefix.Name, online)
1443 }
1444
1445 uc.forEachDownstream(func(dc *downstreamConn) {
1446 for _, target := range targets {
1447 prefix := irc.ParsePrefix(target)
1448 if dc.monitored.Has(prefix.Name) {
1449 dc.SendMessage(&irc.Message{
1450 Prefix: dc.srv.prefix(),
1451 Command: msg.Command,
1452 Params: []string{dc.nick, target},
1453 })
1454 }
1455 }
1456 })
1457 case irc.ERR_MONLISTFULL:
1458 var limit, targetsStr string
1459 if err := parseMessageParams(msg, nil, &limit, &targetsStr); err != nil {
1460 return err
1461 }
1462
1463 targets := strings.Split(targetsStr, ",")
1464 uc.forEachDownstream(func(dc *downstreamConn) {
1465 for _, target := range targets {
1466 if dc.monitored.Has(target) {
1467 dc.SendMessage(&irc.Message{
1468 Prefix: dc.srv.prefix(),
1469 Command: msg.Command,
1470 Params: []string{dc.nick, limit, target},
1471 })
1472 }
1473 }
1474 })
[272]1475 case irc.RPL_AWAY:
1476 var nick, reason string
1477 if err := parseMessageParams(msg, nil, &nick, &reason); err != nil {
1478 return err
1479 }
1480
[274]1481 uc.forEachDownstream(func(dc *downstreamConn) {
[272]1482 dc.SendMessage(&irc.Message{
1483 Prefix: dc.srv.prefix(),
1484 Command: irc.RPL_AWAY,
1485 Params: []string{dc.nick, dc.marshalEntity(uc.network, nick), reason},
1486 })
1487 })
[649]1488 case "AWAY", "ACCOUNT":
[276]1489 if msg.Prefix == nil {
1490 return fmt.Errorf("expected a prefix")
1491 }
1492
1493 uc.forEachDownstream(func(dc *downstreamConn) {
1494 dc.SendMessage(&irc.Message{
1495 Prefix: dc.marshalUserPrefix(uc.network, msg.Prefix),
[648]1496 Command: msg.Command,
1497 Params: msg.Params,
1498 })
1499 })
[300]1500 case irc.RPL_BANLIST, irc.RPL_INVITELIST, irc.RPL_EXCEPTLIST:
1501 var channel, mask string
1502 if err := parseMessageParams(msg, nil, &channel, &mask); err != nil {
1503 return err
1504 }
1505 var addNick, addTime string
1506 if len(msg.Params) >= 5 {
1507 addNick = msg.Params[3]
1508 addTime = msg.Params[4]
1509 }
1510
1511 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1512 channel := dc.marshalEntity(uc.network, channel)
1513
1514 var params []string
1515 if addNick != "" && addTime != "" {
1516 addNick := dc.marshalEntity(uc.network, addNick)
1517 params = []string{dc.nick, channel, mask, addNick, addTime}
1518 } else {
1519 params = []string{dc.nick, channel, mask}
1520 }
1521
1522 dc.SendMessage(&irc.Message{
1523 Prefix: dc.srv.prefix(),
1524 Command: msg.Command,
1525 Params: params,
1526 })
1527 })
1528 case irc.RPL_ENDOFBANLIST, irc.RPL_ENDOFINVITELIST, irc.RPL_ENDOFEXCEPTLIST:
1529 var channel, trailing string
1530 if err := parseMessageParams(msg, nil, &channel, &trailing); err != nil {
1531 return err
1532 }
1533
1534 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1535 upstreamChannel := dc.marshalEntity(uc.network, channel)
1536 dc.SendMessage(&irc.Message{
1537 Prefix: dc.srv.prefix(),
1538 Command: msg.Command,
1539 Params: []string{dc.nick, upstreamChannel, trailing},
1540 })
1541 })
[302]1542 case irc.ERR_UNKNOWNCOMMAND, irc.RPL_TRYAGAIN:
1543 var command, reason string
1544 if err := parseMessageParams(msg, nil, &command, &reason); err != nil {
1545 return err
1546 }
1547
[682]1548 if command == "LIST" || command == "WHO" {
1549 dc, _ := uc.dequeueCommand(command)
1550 if dc != nil && downstreamID == 0 {
1551 downstreamID = dc.id
1552 }
[302]1553 }
1554
[355]1555 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1556 dc.SendMessage(&irc.Message{
1557 Prefix: uc.srv.prefix(),
1558 Command: msg.Command,
1559 Params: []string{dc.nick, command, reason},
[302]1560 })
[355]1561 })
[155]1562 case "ACK":
1563 // Ignore
[198]1564 case irc.RPL_NOWAWAY, irc.RPL_UNAWAY:
1565 // Ignore
[16]1566 case irc.RPL_YOURHOST, irc.RPL_CREATED:
[14]1567 // Ignore
1568 case irc.RPL_LUSERCLIENT, irc.RPL_LUSEROP, irc.RPL_LUSERUNKNOWN, irc.RPL_LUSERCHANNELS, irc.RPL_LUSERME:
[561]1569 fallthrough
1570 case irc.RPL_STATSVLINE, rpl_statsping, irc.RPL_STATSBLINE, irc.RPL_STATSDLINE:
1571 fallthrough
1572 case rpl_localusers, rpl_globalusers:
1573 fallthrough
[478]1574 case irc.RPL_MOTDSTART, irc.RPL_MOTD:
[561]1575 // Ignore these messages if they're part of the initial registration
1576 // message burst. Forward them if the user explicitly asked for them.
[552]1577 if !uc.gotMotd {
1578 return nil
1579 }
1580
[553]1581 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1582 dc.SendMessage(&irc.Message{
1583 Prefix: uc.srv.prefix(),
[552]1584 Command: msg.Command,
[553]1585 Params: msg.Params,
[552]1586 })
1587 })
[177]1588 case irc.RPL_LISTSTART:
1589 // Ignore
[390]1590 case "ERROR":
1591 var text string
1592 if err := parseMessageParams(msg, &text); err != nil {
1593 return err
1594 }
1595 return fmt.Errorf("fatal server error: %v", text)
[389]1596 case irc.ERR_PASSWDMISMATCH, irc.ERR_ERRONEUSNICKNAME, irc.ERR_NICKNAMEINUSE, irc.ERR_NICKCOLLISION, irc.ERR_UNAVAILRESOURCE, irc.ERR_NOPERMFORHOST, irc.ERR_YOUREBANNEDCREEP:
[342]1597 if !uc.registered {
[399]1598 text := msg.Params[len(msg.Params)-1]
1599 return registrationError(text)
[342]1600 }
1601 fallthrough
[13]1602 default:
[95]1603 uc.logger.Printf("unhandled message: %v", msg)
[355]1604
1605 uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
1606 // best effort marshaling for unknown messages, replies and errors:
1607 // most numerics start with the user nick, marshal it if that's the case
1608 // otherwise, conservately keep the params without marshaling
1609 params := msg.Params
1610 if _, err := strconv.Atoi(msg.Command); err == nil { // numeric
1611 if len(msg.Params) > 0 && isOurNick(uc.network, msg.Params[0]) {
1612 params[0] = dc.nick
[302]1613 }
[355]1614 }
1615 dc.SendMessage(&irc.Message{
1616 Prefix: uc.srv.prefix(),
1617 Command: msg.Command,
1618 Params: params,
[302]1619 })
[355]1620 })
[13]1621 }
[14]1622 return nil
[13]1623}
1624
[499]1625func (uc *upstreamConn) handleDetachedMessage(ch *Channel, msg *irc.Message) {
1626 if uc.network.detachedMessageNeedsRelay(ch, msg) {
[435]1627 uc.forEachDownstream(func(dc *downstreamConn) {
[499]1628 dc.relayDetachedMessage(uc.network, msg)
[435]1629 })
1630 }
[499]1631 if ch.ReattachOn == FilterMessage || (ch.ReattachOn == FilterHighlight && uc.network.isHighlight(msg)) {
[435]1632 uc.network.attach(ch)
[652]1633 if err := uc.srv.db.StoreChannel(context.TODO(), uc.network.ID, ch); err != nil {
[435]1634 uc.logger.Printf("failed to update channel %q: %v", ch.Name, err)
1635 }
1636 }
1637}
1638
[458]1639func (uc *upstreamConn) handleChanModes(s string) error {
1640 parts := strings.SplitN(s, ",", 5)
1641 if len(parts) < 4 {
1642 return fmt.Errorf("malformed ISUPPORT CHANMODES value: %v", s)
1643 }
1644 modes := make(map[byte]channelModeType)
1645 for i, mt := range []channelModeType{modeTypeA, modeTypeB, modeTypeC, modeTypeD} {
1646 for j := 0; j < len(parts[i]); j++ {
1647 mode := parts[i][j]
1648 modes[mode] = mt
1649 }
1650 }
1651 uc.availableChannelModes = modes
1652 return nil
1653}
1654
1655func (uc *upstreamConn) handleMemberships(s string) error {
1656 if s == "" {
1657 uc.availableMemberships = nil
1658 return nil
1659 }
1660
1661 if s[0] != '(' {
1662 return fmt.Errorf("malformed ISUPPORT PREFIX value: %v", s)
1663 }
1664 sep := strings.IndexByte(s, ')')
1665 if sep < 0 || len(s) != sep*2 {
1666 return fmt.Errorf("malformed ISUPPORT PREFIX value: %v", s)
1667 }
1668 memberships := make([]membership, len(s)/2-1)
1669 for i := range memberships {
1670 memberships[i] = membership{
1671 Mode: s[i+1],
1672 Prefix: s[sep+i+1],
1673 }
1674 }
1675 uc.availableMemberships = memberships
1676 return nil
1677}
1678
[281]1679func (uc *upstreamConn) handleSupportedCaps(capsStr string) {
1680 caps := strings.Fields(capsStr)
1681 for _, s := range caps {
1682 kv := strings.SplitN(s, "=", 2)
1683 k := strings.ToLower(kv[0])
1684 var v string
1685 if len(kv) == 2 {
1686 v = kv[1]
1687 }
1688 uc.supportedCaps[k] = v
1689 }
1690}
1691
1692func (uc *upstreamConn) requestCaps() {
1693 var requestCaps []string
[282]1694 for c := range permanentUpstreamCaps {
[281]1695 if _, ok := uc.supportedCaps[c]; ok && !uc.caps[c] {
1696 requestCaps = append(requestCaps, c)
1697 }
1698 }
1699
1700 if uc.requestSASL() && !uc.caps["sasl"] {
1701 requestCaps = append(requestCaps, "sasl")
1702 }
1703
[282]1704 if len(requestCaps) == 0 {
1705 return
1706 }
1707
1708 uc.SendMessage(&irc.Message{
1709 Command: "CAP",
1710 Params: []string{"REQ", strings.Join(requestCaps, " ")},
1711 })
1712}
1713
1714func (uc *upstreamConn) requestSASL() bool {
1715 if uc.network.SASL.Mechanism == "" {
1716 return false
1717 }
1718
1719 v, ok := uc.supportedCaps["sasl"]
1720 if !ok {
1721 return false
1722 }
1723 if v != "" {
1724 mechanisms := strings.Split(v, ",")
1725 found := false
1726 for _, mech := range mechanisms {
1727 if strings.EqualFold(mech, uc.network.SASL.Mechanism) {
1728 found = true
1729 break
1730 }
1731 }
1732 if !found {
1733 return false
1734 }
1735 }
1736
1737 return true
1738}
1739
1740func (uc *upstreamConn) handleCapAck(name string, ok bool) error {
1741 uc.caps[name] = ok
1742
1743 switch name {
1744 case "sasl":
1745 if !ok {
1746 uc.logger.Printf("server refused to acknowledge the SASL capability")
1747 return nil
1748 }
1749
1750 auth := &uc.network.SASL
1751 switch auth.Mechanism {
1752 case "PLAIN":
1753 uc.logger.Printf("starting SASL PLAIN authentication with username %q", auth.Plain.Username)
1754 uc.saslClient = sasl.NewPlainClient("", auth.Plain.Username, auth.Plain.Password)
[307]1755 case "EXTERNAL":
1756 uc.logger.Printf("starting SASL EXTERNAL authentication")
1757 uc.saslClient = sasl.NewExternalClient("")
[282]1758 default:
1759 return fmt.Errorf("unsupported SASL mechanism %q", name)
1760 }
1761
[281]1762 uc.SendMessage(&irc.Message{
[282]1763 Command: "AUTHENTICATE",
1764 Params: []string{auth.Mechanism},
[281]1765 })
[282]1766 default:
1767 if permanentUpstreamCaps[name] {
1768 break
1769 }
1770 uc.logger.Printf("received CAP ACK/NAK for a cap we don't support: %v", name)
[281]1771 }
[282]1772 return nil
[281]1773}
1774
[174]1775func splitSpace(s string) []string {
1776 return strings.FieldsFunc(s, func(r rune) bool {
1777 return r == ' '
1778 })
1779}
1780
[55]1781func (uc *upstreamConn) register() {
[664]1782 uc.nick = GetNick(&uc.user.User, &uc.network.Network)
[478]1783 uc.nickCM = uc.network.casemap(uc.nick)
[674]1784 uc.username = GetUsername(&uc.user.User, &uc.network.Network)
[568]1785 uc.realname = GetRealname(&uc.user.User, &uc.network.Network)
[77]1786
[60]1787 uc.SendMessage(&irc.Message{
[92]1788 Command: "CAP",
1789 Params: []string{"LS", "302"},
1790 })
1791
[93]1792 if uc.network.Pass != "" {
1793 uc.SendMessage(&irc.Message{
1794 Command: "PASS",
1795 Params: []string{uc.network.Pass},
1796 })
1797 }
1798
[92]1799 uc.SendMessage(&irc.Message{
[13]1800 Command: "NICK",
[69]1801 Params: []string{uc.nick},
[60]1802 })
1803 uc.SendMessage(&irc.Message{
[13]1804 Command: "USER",
[77]1805 Params: []string{uc.username, "0", "*", uc.realname},
[60]1806 })
[44]1807}
[13]1808
[711]1809func (uc *upstreamConn) ReadMessage() (*irc.Message, error) {
1810 msg, err := uc.conn.ReadMessage()
1811 if err != nil {
1812 return nil, err
1813 }
1814 uc.srv.metrics.upstreamInMessagesTotal.Inc()
1815 return msg, nil
1816}
1817
[197]1818func (uc *upstreamConn) runUntilRegistered() error {
1819 for !uc.registered {
[212]1820 msg, err := uc.ReadMessage()
[197]1821 if err != nil {
1822 return fmt.Errorf("failed to read message: %v", err)
1823 }
1824
1825 if err := uc.handleMessage(msg); err != nil {
[399]1826 if _, ok := err.(registrationError); ok {
1827 return err
1828 } else {
1829 msg.Tags = nil // prevent message tags from cluttering logs
1830 return fmt.Errorf("failed to handle message %q: %v", msg, err)
1831 }
[197]1832 }
1833 }
1834
[263]1835 for _, command := range uc.network.ConnectCommands {
1836 m, err := irc.ParseMessage(command)
1837 if err != nil {
1838 uc.logger.Printf("failed to parse connect command %q: %v", command, err)
1839 } else {
1840 uc.SendMessage(m)
1841 }
1842 }
1843
[197]1844 return nil
1845}
1846
[165]1847func (uc *upstreamConn) readMessages(ch chan<- event) error {
[13]1848 for {
[210]1849 msg, err := uc.ReadMessage()
[655]1850 if errors.Is(err, io.EOF) {
[13]1851 break
1852 } else if err != nil {
1853 return fmt.Errorf("failed to read IRC command: %v", err)
1854 }
1855
[165]1856 ch <- eventUpstreamMessage{msg, uc}
[13]1857 }
1858
[45]1859 return nil
[13]1860}
[60]1861
[303]1862func (uc *upstreamConn) SendMessage(msg *irc.Message) {
1863 if !uc.caps["message-tags"] {
1864 msg = msg.Copy()
1865 msg.Tags = nil
1866 }
1867
[711]1868 uc.srv.metrics.upstreamOutMessagesTotal.Inc()
[303]1869 uc.conn.SendMessage(msg)
1870}
1871
[176]1872func (uc *upstreamConn) SendMessageLabeled(downstreamID uint64, msg *irc.Message) {
[278]1873 if uc.caps["labeled-response"] {
[155]1874 if msg.Tags == nil {
1875 msg.Tags = make(map[string]irc.TagValue)
1876 }
[176]1877 msg.Tags["label"] = irc.TagValue(fmt.Sprintf("sd-%d-%d", downstreamID, uc.nextLabelID))
[161]1878 uc.nextLabelID++
[155]1879 }
1880 uc.SendMessage(msg)
1881}
[178]1882
[428]1883// appendLog appends a message to the log file.
1884//
1885// The internal message ID is returned. If the message isn't recorded in the
1886// log file, an empty string is returned.
1887func (uc *upstreamConn) appendLog(entity string, msg *irc.Message) (msgID string) {
[423]1888 if uc.user.msgStore == nil {
[428]1889 return ""
[178]1890 }
[486]1891
[563]1892 // Don't store messages with a server mask target
1893 if strings.HasPrefix(entity, "$") {
1894 return ""
1895 }
1896
[486]1897 entityCM := uc.network.casemap(entity)
1898 if entityCM == "nickserv" {
[468]1899 // The messages sent/received from NickServ may contain
1900 // security-related information (like passwords). Don't store these.
1901 return ""
1902 }
[215]1903
[485]1904 if !uc.network.delivered.HasTarget(entity) {
[482]1905 // This is the first message we receive from this target. Save the last
1906 // message ID in delivery receipts, so that we can send the new message
1907 // in the backlog if an offline client reconnects.
[666]1908 lastID, err := uc.user.msgStore.LastMsgID(&uc.network.Network, entityCM, time.Now())
[409]1909 if err != nil {
1910 uc.logger.Printf("failed to log message: failed to get last message ID: %v", err)
[428]1911 return ""
[409]1912 }
1913
[489]1914 uc.network.delivered.ForEachClient(func(clientName string) {
[485]1915 uc.network.delivered.StoreID(entity, clientName, lastID)
[489]1916 })
[253]1917 }
1918
[666]1919 msgID, err := uc.user.msgStore.Append(&uc.network.Network, entityCM, msg)
[409]1920 if err != nil {
1921 uc.logger.Printf("failed to log message: %v", err)
[428]1922 return ""
[409]1923 }
[406]1924
[428]1925 return msgID
[253]1926}
1927
[409]1928// produce appends a message to the logs and forwards it to connected downstream
1929// connections.
[245]1930//
1931// If origin is not nil and origin doesn't support echo-message, the message is
1932// forwarded to all connections except origin.
[239]1933func (uc *upstreamConn) produce(target string, msg *irc.Message, origin *downstreamConn) {
[428]1934 var msgID string
[239]1935 if target != "" {
[428]1936 msgID = uc.appendLog(target, msg)
[239]1937 }
1938
[284]1939 // Don't forward messages if it's a detached channel
[478]1940 ch := uc.network.channels.Value(target)
[499]1941 detached := ch != nil && ch.Detached
[284]1942
[227]1943 uc.forEachDownstream(func(dc *downstreamConn) {
[499]1944 if !detached && (dc != origin || dc.caps["echo-message"]) {
[428]1945 dc.sendMessageWithID(dc.marshalMessage(msg, uc.network), msgID)
1946 } else {
1947 dc.advanceMessageWithID(msg, msgID)
[238]1948 }
[227]1949 })
[226]1950}
1951
[198]1952func (uc *upstreamConn) updateAway() {
1953 away := true
1954 uc.forEachDownstream(func(*downstreamConn) {
1955 away = false
1956 })
1957 if away == uc.away {
1958 return
1959 }
1960 if away {
1961 uc.SendMessage(&irc.Message{
1962 Command: "AWAY",
1963 Params: []string{"Auto away"},
1964 })
1965 } else {
1966 uc.SendMessage(&irc.Message{
1967 Command: "AWAY",
1968 })
1969 }
1970 uc.away = away
1971}
[435]1972
1973func (uc *upstreamConn) updateChannelAutoDetach(name string) {
[478]1974 uch := uc.channels.Value(name)
1975 if uch == nil {
1976 return
[435]1977 }
[478]1978 ch := uc.network.channels.Value(name)
1979 if ch == nil || ch.Detached {
1980 return
1981 }
1982 uch.updateAutoDetach(ch.DetachAfter)
[435]1983}
[684]1984
1985func (uc *upstreamConn) updateMonitor() {
1986 add := make(map[string]struct{})
1987 var addList []string
1988 seen := make(map[string]struct{})
1989 uc.forEachDownstream(func(dc *downstreamConn) {
1990 for targetCM := range dc.monitored.innerMap {
1991 if !uc.monitored.Has(targetCM) {
1992 if _, ok := add[targetCM]; !ok {
1993 addList = append(addList, targetCM)
1994 }
1995 add[targetCM] = struct{}{}
1996 } else {
1997 seen[targetCM] = struct{}{}
1998 }
1999 }
2000 })
2001
2002 removeAll := true
2003 var removeList []string
2004 for targetCM, entry := range uc.monitored.innerMap {
2005 if _, ok := seen[targetCM]; ok {
2006 removeAll = false
2007 } else {
2008 removeList = append(removeList, entry.originalKey)
2009 }
2010 }
2011
2012 // TODO: better handle the case where len(uc.monitored) + len(addList)
2013 // exceeds the limit, probably by immediately sending ERR_MONLISTFULL?
2014
2015 if removeAll && len(addList) == 0 && len(removeList) > 0 {
2016 // Optimization when the last MONITOR-aware downstream disconnects
2017 uc.SendMessage(&irc.Message{
2018 Command: "MONITOR",
2019 Params: []string{"C"},
2020 })
2021 } else {
2022 msgs := generateMonitor("-", removeList)
2023 msgs = append(msgs, generateMonitor("+", addList)...)
2024 for _, msg := range msgs {
2025 uc.SendMessage(msg)
2026 }
2027 }
2028
2029 for _, target := range removeList {
2030 uc.monitored.Delete(target)
2031 }
2032}
Note: See TracBrowser for help on using the repository browser.