source: code/trunk/upstream.go@ 457

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

Add Network.{URL,GetUsername,GetRealname}

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