Context Navigation

upstream.go@ 237

Last change on this file since 237 was 235, checked in by contact, 5 years ago

Log TOPIC messages

See [1].

[1]: https://github.com/znc/znc/blob/82c4ad8362c6b2f4406770c54a5e10a40fdba548/modules/log.cpp#L486

File size: 34.0 KB

Rev	Line
[98]	1	package soju
[13]	2
	3	import (
	4	"crypto/tls"
[95]	5	"encoding/base64"
[155]	6	"errors"
[13]	7	"fmt"
	8	"io"
	9	"net"
[19]	10	"strconv"
[17]	11	"strings"
[19]	12	"time"
[13]	13
[95]	14	"github.com/emersion/go-sasl"
[13]	15	"gopkg.in/irc.v3"
	16	)
	17
[19]	18	type upstreamChannel struct {
[162]	19	Name string
	20	conn *upstreamConn
	21	Topic string
	22	TopicWho string
	23	TopicTime time.Time
	24	Status channelStatus
	25	modes channelModes
	26	creationTime string
	27	Members map[string]*membership
	28	complete bool
[19]	29	}
	30
[13]	31	type upstreamConn struct {
[210]	32	conn
[16]	33
[210]	34	network *network
	35	user *user
	36
[16]	37	serverName string
	38	availableUserModes string
[139]	39	availableChannelModes map[byte]channelModeType
	40	availableChannelTypes string
	41	availableMemberships []membership
[19]	42
	43	registered bool
[42]	44	nick string
[77]	45	username string
	46	realname string
[139]	47	modes userModes
[19]	48	channels map[string]*upstreamChannel
[92]	49	caps map[string]string
[153]	50	batches map[string]batch
[198]	51	away bool
[95]	52
[155]	53	tagsSupported bool
	54	labelsSupported bool
[161]	55	nextLabelID uint64
[152]	56
[95]	57	saslClient sasl.Client
	58	saslStarted bool
[177]	59
	60	// set of LIST commands in progress, per downstream
	61	pendingLISTDownstreamSet map[uint64]struct{}
[178]	62
[215]	63	messageLoggers map[string]*messageLogger
[13]	64	}
	65
[77]	66	func connectToUpstream(network network) (upstreamConn, error) {
	67	logger := &prefixLogger{network.user.srv.Logger, fmt.Sprintf("upstream %q: ", network.Addr)}
[33]	68
[77]	69	addr := network.Addr
	70	if !strings.ContainsRune(addr, ':') {
	71	addr = addr + ":6697"
	72	}
	73
[206]	74	dialer := net.Dialer{Timeout: connectTimeout}
	75
[77]	76	logger.Printf("connecting to TLS server at address %q", addr)
[206]	77	netConn, err := tls.DialWithDialer(&dialer, "tcp", addr, nil)
[33]	78	if err != nil {
[77]	79	return nil, fmt.Errorf("failed to dial %q: %v", addr, err)
[33]	80	}
	81
[55]	82	uc := &upstreamConn{
[210]	83	conn: *newConn(network.user.srv, netConn, logger),
[177]	84	network: network,
	85	user: network.user,
	86	channels: make(map[string]*upstreamChannel),
	87	caps: make(map[string]string),
	88	batches: make(map[string]batch),
	89	availableChannelTypes: stdChannelTypes,
	90	availableChannelModes: stdChannelModes,
	91	availableMemberships: stdMemberships,
	92	pendingLISTDownstreamSet: make(map[uint64]struct{}),
[215]	93	messageLoggers: make(map[string]*messageLogger),
[33]	94	}
	95
[55]	96	return uc, nil
[33]	97	}
	98
[73]	99	func (uc upstreamConn) forEachDownstream(f func(downstreamConn)) {
[218]	100	uc.network.forEachDownstream(f)
[73]	101	}
	102
[161]	103	func (uc upstreamConn) forEachDownstreamByID(id uint64, f func(downstreamConn)) {
[155]	104	uc.forEachDownstream(func(dc *downstreamConn) {
	105	if id != 0 && id != dc.id {
	106	return
	107	}
	108	f(dc)
	109	})
	110	}
	111
[55]	112	func (uc upstreamConn) getChannel(name string) (upstreamChannel, error) {
	113	ch, ok := uc.channels[name]
[19]	114	if !ok {
	115	return nil, fmt.Errorf("unknown channel %q", name)
	116	}
	117	return ch, nil
	118	}
	119
[129]	120	func (uc *upstreamConn) isChannel(entity string) bool {
[139]	121	if i := strings.IndexByte(uc.availableChannelTypes, entity[0]); i >= 0 {
	122	return true
[129]	123	}
	124	return false
	125	}
	126
[181]	127	func (uc upstreamConn) getPendingLIST() pendingLIST {
[177]	128	for _, pl := range uc.user.pendingLISTs {
	129	if _, ok := pl.pendingCommands[uc.network.ID]; !ok {
	130	continue
	131	}
	132	return &pl
	133	}
	134	return nil
	135	}
	136
[181]	137	func (uc *upstreamConn) endPendingLISTs(all bool) (found bool) {
[177]	138	found = false
	139	for i := 0; i < len(uc.user.pendingLISTs); i++ {
	140	pl := uc.user.pendingLISTs[i]
	141	if _, ok := pl.pendingCommands[uc.network.ID]; !ok {
	142	continue
	143	}
	144	delete(pl.pendingCommands, uc.network.ID)
	145	if len(pl.pendingCommands) == 0 {
	146	uc.user.pendingLISTs = append(uc.user.pendingLISTs[:i], uc.user.pendingLISTs[i+1:]...)
	147	i--
	148	uc.forEachDownstreamByID(pl.downstreamID, func(dc *downstreamConn) {
	149	dc.SendMessage(&irc.Message{
	150	Prefix: dc.srv.prefix(),
	151	Command: irc.RPL_LISTEND,
	152	Params: []string{dc.nick, "End of /LIST"},
	153	})
	154	})
	155	}
	156	found = true
	157	if !all {
	158	delete(uc.pendingLISTDownstreamSet, pl.downstreamID)
	159	uc.user.forEachUpstream(func(uc *upstreamConn) {
[181]	160	uc.trySendLIST(pl.downstreamID)
[177]	161	})
	162	return
	163	}
	164	}
	165	return
	166	}
	167
[181]	168	func (uc *upstreamConn) trySendLIST(downstreamID uint64) {
[177]	169	// must be called with a lock in uc.user.pendingLISTsLock
	170
	171	if _, ok := uc.pendingLISTDownstreamSet[downstreamID]; ok {
	172	// a LIST command is already pending
	173	// we will try again when that command is completed
	174	return
	175	}
	176
	177	for _, pl := range uc.user.pendingLISTs {
	178	if pl.downstreamID != downstreamID {
	179	continue
	180	}
	181	// this is the first pending LIST command list of the downstream
	182	listCommand, ok := pl.pendingCommands[uc.network.ID]
	183	if !ok {
	184	// there is no command for this upstream in these LIST commands
	185	// do not send anything
	186	continue
	187	}
	188	// there is a command for this upstream in these LIST commands
	189	// send it now
	190
	191	uc.SendMessageLabeled(downstreamID, listCommand)
	192
	193	uc.pendingLISTDownstreamSet[downstreamID] = struct{}{}
	194	return
	195	}
	196	}
	197
[139]	198	func (uc upstreamConn) parseMembershipPrefix(s string) (membership membership, nick string) {
	199	for _, m := range uc.availableMemberships {
	200	if m.Prefix == s[0] {
	201	return &m, s[1:]
	202	}
	203	}
	204	return nil, s
	205	}
	206
[55]	207	func (uc upstreamConn) handleMessage(msg irc.Message) error {
[155]	208	var label string
	209	if l, ok := msg.GetTag("label"); ok {
	210	label = l
	211	}
	212
[153]	213	var msgBatch *batch
	214	if batchName, ok := msg.GetTag("batch"); ok {
	215	b, ok := uc.batches[batchName]
	216	if !ok {
	217	return fmt.Errorf("unexpected batch reference: batch was not defined: %q", batchName)
	218	}
	219	msgBatch = &b
[155]	220	if label == "" {
	221	label = msgBatch.Label
	222	}
[153]	223	}
	224
[161]	225	var downstreamID uint64 = 0
[155]	226	if label != "" {
	227	var labelOffset uint64
[161]	228	n, err := fmt.Sscanf(label, "sd-%d-%d", &downstreamID, &labelOffset)
[155]	229	if err == nil && n < 2 {
	230	err = errors.New("not enough arguments")
	231	}
	232	if err != nil {
	233	return fmt.Errorf("unexpected message label: invalid downstream reference for label %q: %v", label, err)
	234	}
	235	}
	236
[216]	237	if _, ok := msg.Tags["time"]; !ok {
	238	msg.Tags["time"] = irc.TagValue(time.Now().Format(serverTimeLayout))
	239	}
	240
[13]	241	switch msg.Command {
	242	case "PING":
[60]	243	uc.SendMessage(&irc.Message{
[13]	244	Command: "PONG",
[68]	245	Params: msg.Params,
[60]	246	})
[33]	247	return nil
[18]	248	case "NOTICE":
[171]	249	if msg.Prefix.User == "" && msg.Prefix.Host == "" { // server message
[226]	250	uc.produce(msg)
[171]	251	} else { // regular user NOTICE
[217]	252	var entity, text string
	253	if err := parseMessageParams(msg, &entity, &text); err != nil {
[171]	254	return err
	255	}
	256
[217]	257	target := entity
	258	if target == uc.nick {
[178]	259	target = msg.Prefix.Name
	260	}
[215]	261	uc.appendLog(target, msg)
[178]	262
[226]	263	uc.produce(msg)
[171]	264	}
[92]	265	case "CAP":
[95]	266	var subCmd string
	267	if err := parseMessageParams(msg, nil, &subCmd); err != nil {
	268	return err
[92]	269	}
[95]	270	subCmd = strings.ToUpper(subCmd)
	271	subParams := msg.Params[2:]
	272	switch subCmd {
	273	case "LS":
	274	if len(subParams) < 1 {
	275	return newNeedMoreParamsError(msg.Command)
	276	}
	277	caps := strings.Fields(subParams[len(subParams)-1])
	278	more := len(subParams) >= 2 && msg.Params[len(subParams)-2] == "*"
[92]	279
[95]	280	for _, s := range caps {
	281	kv := strings.SplitN(s, "=", 2)
	282	k := strings.ToLower(kv[0])
	283	var v string
	284	if len(kv) == 2 {
	285	v = kv[1]
	286	}
	287	uc.caps[k] = v
[92]	288	}
	289
[95]	290	if more {
	291	break // wait to receive all capabilities
	292	}
	293
[152]	294	requestCaps := make([]string, 0, 16)
[193]	295	for _, c := range []string{"message-tags", "batch", "labeled-response", "server-time"} {
[152]	296	if _, ok := uc.caps[c]; ok {
	297	requestCaps = append(requestCaps, c)
	298	}
	299	}
	300
[95]	301	if uc.requestSASL() {
[152]	302	requestCaps = append(requestCaps, "sasl")
	303	}
	304
	305	if len(requestCaps) > 0 {
[95]	306	uc.SendMessage(&irc.Message{
	307	Command: "CAP",
[152]	308	Params: []string{"REQ", strings.Join(requestCaps, " ")},
[95]	309	})
[152]	310	}
	311
	312	if uc.requestSASL() {
[95]	313	break // we'll send CAP END after authentication is completed
	314	}
	315
[92]	316	uc.SendMessage(&irc.Message{
	317	Command: "CAP",
	318	Params: []string{"END"},
	319	})
[95]	320	case "ACK", "NAK":
	321	if len(subParams) < 1 {
	322	return newNeedMoreParamsError(msg.Command)
	323	}
	324	caps := strings.Fields(subParams[0])
	325
	326	for _, name := range caps {
	327	if err := uc.handleCapAck(strings.ToLower(name), subCmd == "ACK"); err != nil {
	328	return err
	329	}
	330	}
	331
	332	if uc.saslClient == nil {
	333	uc.SendMessage(&irc.Message{
	334	Command: "CAP",
	335	Params: []string{"END"},
	336	})
	337	}
	338	default:
	339	uc.logger.Printf("unhandled message: %v", msg)
[92]	340	}
[95]	341	case "AUTHENTICATE":
	342	if uc.saslClient == nil {
	343	return fmt.Errorf("received unexpected AUTHENTICATE message")
	344	}
	345
	346	// TODO: if a challenge is 400 bytes long, buffer it
	347	var challengeStr string
	348	if err := parseMessageParams(msg, &challengeStr); err != nil {
	349	uc.SendMessage(&irc.Message{
	350	Command: "AUTHENTICATE",
	351	Params: []string{"*"},
	352	})
	353	return err
	354	}
	355
	356	var challenge []byte
	357	if challengeStr != "+" {
	358	var err error
	359	challenge, err = base64.StdEncoding.DecodeString(challengeStr)
	360	if err != nil {
	361	uc.SendMessage(&irc.Message{
	362	Command: "AUTHENTICATE",
	363	Params: []string{"*"},
	364	})
	365	return err
	366	}
	367	}
	368
	369	var resp []byte
	370	var err error
	371	if !uc.saslStarted {
	372	_, resp, err = uc.saslClient.Start()
	373	uc.saslStarted = true
	374	} else {
	375	resp, err = uc.saslClient.Next(challenge)
	376	}
	377	if err != nil {
	378	uc.SendMessage(&irc.Message{
	379	Command: "AUTHENTICATE",
	380	Params: []string{"*"},
	381	})
	382	return err
	383	}
	384
	385	// TODO: send response in multiple chunks if >= 400 bytes
	386	var respStr = "+"
	387	if resp != nil {
	388	respStr = base64.StdEncoding.EncodeToString(resp)
	389	}
	390
	391	uc.SendMessage(&irc.Message{
	392	Command: "AUTHENTICATE",
	393	Params: []string{respStr},
	394	})
[125]	395	case irc.RPL_LOGGEDIN:
[95]	396	var account string
	397	if err := parseMessageParams(msg, nil, nil, &account); err != nil {
	398	return err
	399	}
	400	uc.logger.Printf("logged in with account %q", account)
[125]	401	case irc.RPL_LOGGEDOUT:
[95]	402	uc.logger.Printf("logged out")
[125]	403	case irc.ERR_NICKLOCKED, irc.RPL_SASLSUCCESS, irc.ERR_SASLFAIL, irc.ERR_SASLTOOLONG, irc.ERR_SASLABORTED:
[95]	404	var info string
	405	if err := parseMessageParams(msg, nil, &info); err != nil {
	406	return err
	407	}
	408	switch msg.Command {
[125]	409	case irc.ERR_NICKLOCKED:
[95]	410	uc.logger.Printf("invalid nick used with SASL authentication: %v", info)
[125]	411	case irc.ERR_SASLFAIL:
[95]	412	uc.logger.Printf("SASL authentication failed: %v", info)
[125]	413	case irc.ERR_SASLTOOLONG:
[95]	414	uc.logger.Printf("SASL message too long: %v", info)
	415	}
	416
	417	uc.saslClient = nil
	418	uc.saslStarted = false
	419
	420	uc.SendMessage(&irc.Message{
	421	Command: "CAP",
	422	Params: []string{"END"},
	423	})
[14]	424	case irc.RPL_WELCOME:
[55]	425	uc.registered = true
	426	uc.logger.Printf("connection registered")
[19]	427
[77]	428	channels, err := uc.srv.db.ListChannels(uc.network.ID)
	429	if err != nil {
	430	uc.logger.Printf("failed to list channels from database: %v", err)
	431	break
	432	}
	433
	434	for _, ch := range channels {
[146]	435	params := []string{ch.Name}
	436	if ch.Key != "" {
	437	params = append(params, ch.Key)
	438	}
[60]	439	uc.SendMessage(&irc.Message{
[19]	440	Command: "JOIN",
[146]	441	Params: params,
[60]	442	})
[19]	443	}
[16]	444	case irc.RPL_MYINFO:
[139]	445	if err := parseMessageParams(msg, nil, &uc.serverName, nil, &uc.availableUserModes, nil); err != nil {
[43]	446	return err
[16]	447	}
[139]	448	case irc.RPL_ISUPPORT:
	449	if err := parseMessageParams(msg, nil, nil); err != nil {
	450	return err
[16]	451	}
[139]	452	for _, token := range msg.Params[1 : len(msg.Params)-1] {
	453	negate := false
	454	parameter := token
	455	value := ""
	456	if strings.HasPrefix(token, "-") {
	457	negate = true
	458	token = token[1:]
	459	} else {
	460	if i := strings.IndexByte(token, '='); i >= 0 {
	461	parameter = token[:i]
	462	value = token[i+1:]
	463	}
	464	}
	465	if !negate {
	466	switch parameter {
	467	case "CHANMODES":
	468	parts := strings.SplitN(value, ",", 5)
	469	if len(parts) < 4 {
	470	return fmt.Errorf("malformed ISUPPORT CHANMODES value: %v", value)
	471	}
	472	modes := make(map[byte]channelModeType)
	473	for i, mt := range []channelModeType{modeTypeA, modeTypeB, modeTypeC, modeTypeD} {
	474	for j := 0; j < len(parts[i]); j++ {
	475	mode := parts[i][j]
	476	modes[mode] = mt
	477	}
	478	}
	479	uc.availableChannelModes = modes
	480	case "CHANTYPES":
	481	uc.availableChannelTypes = value
	482	case "PREFIX":
	483	if value == "" {
	484	uc.availableMemberships = nil
	485	} else {
	486	if value[0] != '(' {
	487	return fmt.Errorf("malformed ISUPPORT PREFIX value: %v", value)
	488	}
	489	sep := strings.IndexByte(value, ')')
	490	if sep < 0 \|\| len(value) != sep*2 {
	491	return fmt.Errorf("malformed ISUPPORT PREFIX value: %v", value)
	492	}
	493	memberships := make([]membership, len(value)/2-1)
	494	for i := range memberships {
	495	memberships[i] = membership{
	496	Mode: value[i+1],
	497	Prefix: value[sep+i+1],
	498	}
	499	}
	500	uc.availableMemberships = memberships
	501	}
	502	}
	503	} else {
	504	// TODO: handle ISUPPORT negations
	505	}
	506	}
[153]	507	case "BATCH":
	508	var tag string
	509	if err := parseMessageParams(msg, &tag); err != nil {
	510	return err
	511	}
	512
	513	if strings.HasPrefix(tag, "+") {
	514	tag = tag[1:]
	515	if _, ok := uc.batches[tag]; ok {
	516	return fmt.Errorf("unexpected BATCH reference tag: batch was already defined: %q", tag)
	517	}
	518	var batchType string
	519	if err := parseMessageParams(msg, nil, &batchType); err != nil {
	520	return err
	521	}
[155]	522	label := label
	523	if label == "" && msgBatch != nil {
	524	label = msgBatch.Label
	525	}
[153]	526	uc.batches[tag] = batch{
	527	Type: batchType,
	528	Params: msg.Params[2:],
	529	Outer: msgBatch,
[155]	530	Label: label,
[153]	531	}
	532	} else if strings.HasPrefix(tag, "-") {
	533	tag = tag[1:]
	534	if _, ok := uc.batches[tag]; !ok {
	535	return fmt.Errorf("unknown BATCH reference tag: %q", tag)
	536	}
	537	delete(uc.batches, tag)
	538	} else {
	539	return fmt.Errorf("unexpected BATCH reference tag: missing +/- prefix: %q", tag)
	540	}
[42]	541	case "NICK":
[83]	542	if msg.Prefix == nil {
	543	return fmt.Errorf("expected a prefix")
	544	}
	545
[43]	546	var newNick string
	547	if err := parseMessageParams(msg, &newNick); err != nil {
	548	return err
[42]	549	}
	550
[55]	551	if msg.Prefix.Name == uc.nick {
	552	uc.logger.Printf("changed nick from %q to %q", uc.nick, newNick)
	553	uc.nick = newNick
[42]	554	}
	555
[55]	556	for _, ch := range uc.channels {
[42]	557	if membership, ok := ch.Members[msg.Prefix.Name]; ok {
	558	delete(ch.Members, msg.Prefix.Name)
	559	ch.Members[newNick] = membership
[215]	560	uc.appendLog(ch.Name, msg)
[42]	561	}
	562	}
[82]	563
	564	if msg.Prefix.Name != uc.nick {
	565	uc.forEachDownstream(func(dc *downstreamConn) {
	566	dc.SendMessage(&irc.Message{
	567	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	568	Command: "NICK",
	569	Params: []string{newNick},
	570	})
	571	})
	572	}
[69]	573	case "JOIN":
	574	if msg.Prefix == nil {
	575	return fmt.Errorf("expected a prefix")
	576	}
[42]	577
[43]	578	var channels string
	579	if err := parseMessageParams(msg, &channels); err != nil {
	580	return err
[19]	581	}
[34]	582
[43]	583	for _, ch := range strings.Split(channels, ",") {
[55]	584	if msg.Prefix.Name == uc.nick {
	585	uc.logger.Printf("joined channel %q", ch)
	586	uc.channels[ch] = &upstreamChannel{
[34]	587	Name: ch,
[55]	588	conn: uc,
[139]	589	Members: make(map[string]*membership),
[34]	590	}
[139]	591
	592	uc.SendMessage(&irc.Message{
	593	Command: "MODE",
	594	Params: []string{ch},
	595	})
[34]	596	} else {
[55]	597	ch, err := uc.getChannel(ch)
[34]	598	if err != nil {
	599	return err
	600	}
[139]	601	ch.Members[msg.Prefix.Name] = nil
[19]	602	}
[69]	603
[215]	604	uc.appendLog(ch, msg)
[178]	605
[73]	606	uc.forEachDownstream(func(dc *downstreamConn) {
[69]	607	dc.SendMessage(&irc.Message{
	608	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	609	Command: "JOIN",
	610	Params: []string{dc.marshalChannel(uc, ch)},
	611	})
	612	})
[19]	613	}
[69]	614	case "PART":
	615	if msg.Prefix == nil {
	616	return fmt.Errorf("expected a prefix")
	617	}
[34]	618
[43]	619	var channels string
	620	if err := parseMessageParams(msg, &channels); err != nil {
	621	return err
[34]	622	}
	623
[178]	624	var reason string
	625	if len(msg.Params) > 1 {
	626	reason = msg.Params[1]
	627	}
	628
[43]	629	for _, ch := range strings.Split(channels, ",") {
[55]	630	if msg.Prefix.Name == uc.nick {
	631	uc.logger.Printf("parted channel %q", ch)
	632	delete(uc.channels, ch)
[34]	633	} else {
[55]	634	ch, err := uc.getChannel(ch)
[34]	635	if err != nil {
	636	return err
	637	}
	638	delete(ch.Members, msg.Prefix.Name)
	639	}
[69]	640
[215]	641	uc.appendLog(ch, msg)
[178]	642
[73]	643	uc.forEachDownstream(func(dc *downstreamConn) {
[215]	644	params := []string{dc.marshalChannel(uc, ch)}
	645	if reason != "" {
	646	params = append(params, reason)
	647	}
[69]	648	dc.SendMessage(&irc.Message{
	649	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	650	Command: "PART",
[215]	651	Params: params,
[69]	652	})
	653	})
[34]	654	}
[159]	655	case "KICK":
	656	if msg.Prefix == nil {
	657	return fmt.Errorf("expected a prefix")
	658	}
	659
	660	var channel, user string
	661	if err := parseMessageParams(msg, &channel, &user); err != nil {
	662	return err
	663	}
	664
	665	var reason string
	666	if len(msg.Params) > 2 {
[215]	667	reason = msg.Params[2]
[159]	668	}
	669
	670	if user == uc.nick {
	671	uc.logger.Printf("kicked from channel %q by %s", channel, msg.Prefix.Name)
	672	delete(uc.channels, channel)
	673	} else {
	674	ch, err := uc.getChannel(channel)
	675	if err != nil {
	676	return err
	677	}
	678	delete(ch.Members, user)
	679	}
	680
[215]	681	uc.appendLog(channel, msg)
[178]	682
[159]	683	uc.forEachDownstream(func(dc *downstreamConn) {
	684	params := []string{dc.marshalChannel(uc, channel), dc.marshalNick(uc, user)}
	685	if reason != "" {
	686	params = append(params, reason)
	687	}
	688	dc.SendMessage(&irc.Message{
	689	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	690	Command: "KICK",
	691	Params: params,
	692	})
	693	})
[83]	694	case "QUIT":
	695	if msg.Prefix == nil {
	696	return fmt.Errorf("expected a prefix")
	697	}
	698
	699	if msg.Prefix.Name == uc.nick {
	700	uc.logger.Printf("quit")
	701	}
	702
	703	for _, ch := range uc.channels {
[178]	704	if _, ok := ch.Members[msg.Prefix.Name]; ok {
	705	delete(ch.Members, msg.Prefix.Name)
	706
[215]	707	uc.appendLog(ch.Name, msg)
[178]	708	}
[83]	709	}
	710
	711	if msg.Prefix.Name != uc.nick {
	712	uc.forEachDownstream(func(dc *downstreamConn) {
	713	dc.SendMessage(&irc.Message{
	714	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	715	Command: "QUIT",
	716	Params: msg.Params,
	717	})
	718	})
	719	}
[19]	720	case irc.RPL_TOPIC, irc.RPL_NOTOPIC:
[43]	721	var name, topic string
	722	if err := parseMessageParams(msg, nil, &name, &topic); err != nil {
	723	return err
[19]	724	}
[55]	725	ch, err := uc.getChannel(name)
[19]	726	if err != nil {
	727	return err
	728	}
	729	if msg.Command == irc.RPL_TOPIC {
[43]	730	ch.Topic = topic
[19]	731	} else {
	732	ch.Topic = ""
	733	}
	734	case "TOPIC":
[43]	735	var name string
[74]	736	if err := parseMessageParams(msg, &name); err != nil {
[43]	737	return err
[19]	738	}
[55]	739	ch, err := uc.getChannel(name)
[19]	740	if err != nil {
	741	return err
	742	}
	743	if len(msg.Params) > 1 {
	744	ch.Topic = msg.Params[1]
	745	} else {
	746	ch.Topic = ""
	747	}
[235]	748	uc.appendLog(ch.Name, msg)
[74]	749	uc.forEachDownstream(func(dc *downstreamConn) {
	750	params := []string{dc.marshalChannel(uc, name)}
	751	if ch.Topic != "" {
	752	params = append(params, ch.Topic)
	753	}
	754	dc.SendMessage(&irc.Message{
	755	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	756	Command: "TOPIC",
	757	Params: params,
	758	})
	759	})
[139]	760	case "MODE":
	761	var name, modeStr string
	762	if err := parseMessageParams(msg, &name, &modeStr); err != nil {
	763	return err
	764	}
	765
	766	if !uc.isChannel(name) { // user mode change
	767	if name != uc.nick {
	768	return fmt.Errorf("received MODE message for unknown nick %q", name)
	769	}
	770	return uc.modes.Apply(modeStr)
	771	// TODO: notify downstreams about user mode change?
	772	} else { // channel mode change
	773	ch, err := uc.getChannel(name)
	774	if err != nil {
	775	return err
	776	}
	777
	778	if ch.modes != nil {
	779	if err := ch.modes.Apply(uc.availableChannelModes, modeStr, msg.Params[2:]...); err != nil {
	780	return err
	781	}
	782	}
	783
[215]	784	uc.appendLog(ch.Name, msg)
[178]	785
[139]	786	uc.forEachDownstream(func(dc *downstreamConn) {
	787	params := []string{dc.marshalChannel(uc, name), modeStr}
	788	params = append(params, msg.Params[2:]...)
	789
	790	dc.SendMessage(&irc.Message{
	791	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	792	Command: "MODE",
	793	Params: params,
	794	})
	795	})
	796	}
	797	case irc.RPL_UMODEIS:
	798	if err := parseMessageParams(msg, nil); err != nil {
	799	return err
	800	}
	801	modeStr := ""
	802	if len(msg.Params) > 1 {
	803	modeStr = msg.Params[1]
	804	}
	805
	806	uc.modes = ""
	807	if err := uc.modes.Apply(modeStr); err != nil {
	808	return err
	809	}
	810	// TODO: send RPL_UMODEIS to downstream connections when applicable
	811	case irc.RPL_CHANNELMODEIS:
	812	var channel string
	813	if err := parseMessageParams(msg, nil, &channel); err != nil {
	814	return err
	815	}
	816	modeStr := ""
	817	if len(msg.Params) > 2 {
	818	modeStr = msg.Params[2]
	819	}
	820
	821	ch, err := uc.getChannel(channel)
	822	if err != nil {
	823	return err
	824	}
	825
	826	firstMode := ch.modes == nil
	827	ch.modes = make(map[byte]string)
	828	if err := ch.modes.Apply(uc.availableChannelModes, modeStr, msg.Params[3:]...); err != nil {
	829	return err
	830	}
	831	if firstMode {
	832	modeStr, modeParams := ch.modes.Format()
	833
	834	uc.forEachDownstream(func(dc *downstreamConn) {
	835	params := []string{dc.nick, dc.marshalChannel(uc, channel), modeStr}
	836	params = append(params, modeParams...)
	837
	838	dc.SendMessage(&irc.Message{
	839	Prefix: dc.srv.prefix(),
	840	Command: irc.RPL_CHANNELMODEIS,
	841	Params: params,
	842	})
	843	})
	844	}
[162]	845	case rpl_creationtime:
	846	var channel, creationTime string
	847	if err := parseMessageParams(msg, nil, &channel, &creationTime); err != nil {
	848	return err
	849	}
	850
	851	ch, err := uc.getChannel(channel)
	852	if err != nil {
	853	return err
	854	}
	855
	856	firstCreationTime := ch.creationTime == ""
	857	ch.creationTime = creationTime
	858	if firstCreationTime {
	859	uc.forEachDownstream(func(dc *downstreamConn) {
	860	dc.SendMessage(&irc.Message{
	861	Prefix: dc.srv.prefix(),
	862	Command: rpl_creationtime,
	863	Params: []string{dc.nick, channel, creationTime},
	864	})
	865	})
	866	}
[19]	867	case rpl_topicwhotime:
[43]	868	var name, who, timeStr string
	869	if err := parseMessageParams(msg, nil, &name, &who, &timeStr); err != nil {
	870	return err
[19]	871	}
[55]	872	ch, err := uc.getChannel(name)
[19]	873	if err != nil {
	874	return err
	875	}
[43]	876	ch.TopicWho = who
	877	sec, err := strconv.ParseInt(timeStr, 10, 64)
[19]	878	if err != nil {
	879	return fmt.Errorf("failed to parse topic time: %v", err)
	880	}
	881	ch.TopicTime = time.Unix(sec, 0)
[177]	882	case irc.RPL_LIST:
	883	var channel, clients, topic string
	884	if err := parseMessageParams(msg, nil, &channel, &clients, &topic); err != nil {
	885	return err
	886	}
	887
[181]	888	pl := uc.getPendingLIST()
[177]	889	if pl == nil {
	890	return fmt.Errorf("unexpected RPL_LIST: no matching pending LIST")
	891	}
	892
	893	uc.forEachDownstreamByID(pl.downstreamID, func(dc *downstreamConn) {
	894	dc.SendMessage(&irc.Message{
	895	Prefix: dc.srv.prefix(),
	896	Command: irc.RPL_LIST,
	897	Params: []string{dc.nick, dc.marshalChannel(uc, channel), clients, topic},
	898	})
	899	})
	900	case irc.RPL_LISTEND:
[181]	901	ok := uc.endPendingLISTs(false)
[177]	902	if !ok {
	903	return fmt.Errorf("unexpected RPL_LISTEND: no matching pending LIST")
	904	}
[19]	905	case irc.RPL_NAMREPLY:
[43]	906	var name, statusStr, members string
	907	if err := parseMessageParams(msg, nil, &statusStr, &name, &members); err != nil {
	908	return err
[19]	909	}
[140]	910
	911	ch, ok := uc.channels[name]
	912	if !ok {
	913	// NAMES on a channel we have not joined, forward to downstream
[161]	914	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[140]	915	channel := dc.marshalChannel(uc, name)
[174]	916	members := splitSpace(members)
[140]	917	for i, member := range members {
	918	membership, nick := uc.parseMembershipPrefix(member)
	919	members[i] = membership.String() + dc.marshalNick(uc, nick)
	920	}
	921	memberStr := strings.Join(members, " ")
	922
	923	dc.SendMessage(&irc.Message{
	924	Prefix: dc.srv.prefix(),
	925	Command: irc.RPL_NAMREPLY,
	926	Params: []string{dc.nick, statusStr, channel, memberStr},
	927	})
	928	})
	929	return nil
[19]	930	}
	931
[43]	932	status, err := parseChannelStatus(statusStr)
[19]	933	if err != nil {
	934	return err
	935	}
	936	ch.Status = status
	937
[174]	938	for _, s := range splitSpace(members) {
[139]	939	membership, nick := uc.parseMembershipPrefix(s)
[19]	940	ch.Members[nick] = membership
	941	}
	942	case irc.RPL_ENDOFNAMES:
[43]	943	var name string
	944	if err := parseMessageParams(msg, nil, &name); err != nil {
	945	return err
[25]	946	}
[140]	947
	948	ch, ok := uc.channels[name]
	949	if !ok {
	950	// NAMES on a channel we have not joined, forward to downstream
[161]	951	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[140]	952	channel := dc.marshalChannel(uc, name)
	953
	954	dc.SendMessage(&irc.Message{
	955	Prefix: dc.srv.prefix(),
	956	Command: irc.RPL_ENDOFNAMES,
	957	Params: []string{dc.nick, channel, "End of /NAMES list"},
	958	})
	959	})
	960	return nil
[25]	961	}
	962
[34]	963	if ch.complete {
	964	return fmt.Errorf("received unexpected RPL_ENDOFNAMES")
	965	}
[25]	966	ch.complete = true
[27]	967
[73]	968	uc.forEachDownstream(func(dc *downstreamConn) {
[27]	969	forwardChannel(dc, ch)
[40]	970	})
[127]	971	case irc.RPL_WHOREPLY:
	972	var channel, username, host, server, nick, mode, trailing string
	973	if err := parseMessageParams(msg, nil, &channel, &username, &host, &server, &nick, &mode, &trailing); err != nil {
	974	return err
	975	}
	976
	977	parts := strings.SplitN(trailing, " ", 2)
	978	if len(parts) != 2 {
	979	return fmt.Errorf("received malformed RPL_WHOREPLY: wrong trailing parameter: %s", trailing)
	980	}
	981	realname := parts[1]
	982	hops, err := strconv.Atoi(parts[0])
	983	if err != nil {
	984	return fmt.Errorf("received malformed RPL_WHOREPLY: wrong hop count: %s", parts[0])
	985	}
	986	hops++
	987
	988	trailing = strconv.Itoa(hops) + " " + realname
	989
[161]	990	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[127]	991	channel := channel
	992	if channel != "*" {
	993	channel = dc.marshalChannel(uc, channel)
	994	}
	995	nick := dc.marshalNick(uc, nick)
	996	dc.SendMessage(&irc.Message{
	997	Prefix: dc.srv.prefix(),
	998	Command: irc.RPL_WHOREPLY,
	999	Params: []string{dc.nick, channel, username, host, server, nick, mode, trailing},
	1000	})
	1001	})
	1002	case irc.RPL_ENDOFWHO:
	1003	var name string
	1004	if err := parseMessageParams(msg, nil, &name); err != nil {
	1005	return err
	1006	}
	1007
[161]	1008	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[127]	1009	name := name
	1010	if name != "*" {
	1011	// TODO: support WHO masks
	1012	name = dc.marshalEntity(uc, name)
	1013	}
	1014	dc.SendMessage(&irc.Message{
	1015	Prefix: dc.srv.prefix(),
	1016	Command: irc.RPL_ENDOFWHO,
[142]	1017	Params: []string{dc.nick, name, "End of /WHO list"},
[127]	1018	})
	1019	})
[128]	1020	case irc.RPL_WHOISUSER:
	1021	var nick, username, host, realname string
	1022	if err := parseMessageParams(msg, nil, &nick, &username, &host, nil, &realname); err != nil {
	1023	return err
	1024	}
	1025
[161]	1026	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1027	nick := dc.marshalNick(uc, nick)
	1028	dc.SendMessage(&irc.Message{
	1029	Prefix: dc.srv.prefix(),
	1030	Command: irc.RPL_WHOISUSER,
	1031	Params: []string{dc.nick, nick, username, host, "*", realname},
	1032	})
	1033	})
	1034	case irc.RPL_WHOISSERVER:
	1035	var nick, server, serverInfo string
	1036	if err := parseMessageParams(msg, nil, &nick, &server, &serverInfo); err != nil {
	1037	return err
	1038	}
	1039
[161]	1040	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1041	nick := dc.marshalNick(uc, nick)
	1042	dc.SendMessage(&irc.Message{
	1043	Prefix: dc.srv.prefix(),
	1044	Command: irc.RPL_WHOISSERVER,
	1045	Params: []string{dc.nick, nick, server, serverInfo},
	1046	})
	1047	})
	1048	case irc.RPL_WHOISOPERATOR:
	1049	var nick string
	1050	if err := parseMessageParams(msg, nil, &nick); err != nil {
	1051	return err
	1052	}
	1053
[161]	1054	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1055	nick := dc.marshalNick(uc, nick)
	1056	dc.SendMessage(&irc.Message{
	1057	Prefix: dc.srv.prefix(),
	1058	Command: irc.RPL_WHOISOPERATOR,
	1059	Params: []string{dc.nick, nick, "is an IRC operator"},
	1060	})
	1061	})
	1062	case irc.RPL_WHOISIDLE:
	1063	var nick string
	1064	if err := parseMessageParams(msg, nil, &nick, nil); err != nil {
	1065	return err
	1066	}
	1067
[161]	1068	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1069	nick := dc.marshalNick(uc, nick)
	1070	params := []string{dc.nick, nick}
	1071	params = append(params, msg.Params[2:]...)
	1072	dc.SendMessage(&irc.Message{
	1073	Prefix: dc.srv.prefix(),
	1074	Command: irc.RPL_WHOISIDLE,
	1075	Params: params,
	1076	})
	1077	})
	1078	case irc.RPL_WHOISCHANNELS:
	1079	var nick, channelList string
	1080	if err := parseMessageParams(msg, nil, &nick, &channelList); err != nil {
	1081	return err
	1082	}
[174]	1083	channels := splitSpace(channelList)
[128]	1084
[161]	1085	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1086	nick := dc.marshalNick(uc, nick)
	1087	channelList := make([]string, len(channels))
	1088	for i, channel := range channels {
[139]	1089	prefix, channel := uc.parseMembershipPrefix(channel)
[128]	1090	channel = dc.marshalChannel(uc, channel)
	1091	channelList[i] = prefix.String() + channel
	1092	}
	1093	channels := strings.Join(channelList, " ")
	1094	dc.SendMessage(&irc.Message{
	1095	Prefix: dc.srv.prefix(),
	1096	Command: irc.RPL_WHOISCHANNELS,
	1097	Params: []string{dc.nick, nick, channels},
	1098	})
	1099	})
	1100	case irc.RPL_ENDOFWHOIS:
	1101	var nick string
	1102	if err := parseMessageParams(msg, nil, &nick); err != nil {
	1103	return err
	1104	}
	1105
[161]	1106	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
[128]	1107	nick := dc.marshalNick(uc, nick)
	1108	dc.SendMessage(&irc.Message{
	1109	Prefix: dc.srv.prefix(),
	1110	Command: irc.RPL_ENDOFWHOIS,
[142]	1111	Params: []string{dc.nick, nick, "End of /WHOIS list"},
[128]	1112	})
	1113	})
[36]	1114	case "PRIVMSG":
[117]	1115	if msg.Prefix == nil {
	1116	return fmt.Errorf("expected a prefix")
	1117	}
	1118
[217]	1119	var entity, text string
	1120	if err := parseMessageParams(msg, &entity, &text); err != nil {
[69]	1121	return err
	1122	}
[117]	1123
	1124	if msg.Prefix.Name == serviceNick {
	1125	uc.logger.Printf("skipping PRIVMSG from soju's service: %v", msg)
	1126	break
	1127	}
[217]	1128	if entity == serviceNick {
[117]	1129	uc.logger.Printf("skipping PRIVMSG to soju's service: %v", msg)
	1130	break
	1131	}
	1132
[217]	1133	target := entity
	1134	if target == uc.nick {
[178]	1135	target = msg.Prefix.Name
	1136	}
[215]	1137	uc.appendLog(target, msg)
[178]	1138
[226]	1139	uc.produce(msg)
[115]	1140	case "INVITE":
	1141	var nick string
	1142	var channel string
	1143	if err := parseMessageParams(msg, &nick, &channel); err != nil {
	1144	return err
	1145	}
	1146
	1147	uc.forEachDownstream(func(dc *downstreamConn) {
	1148	dc.SendMessage(&irc.Message{
	1149	Prefix: dc.marshalUserPrefix(uc, msg.Prefix),
	1150	Command: "INVITE",
	1151	Params: []string{dc.marshalNick(uc, nick), dc.marshalChannel(uc, channel)},
	1152	})
	1153	})
[163]	1154	case irc.RPL_INVITING:
	1155	var nick string
	1156	var channel string
	1157	if err := parseMessageParams(msg, &nick, &channel); err != nil {
	1158	return err
	1159	}
	1160
	1161	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
	1162	dc.SendMessage(&irc.Message{
	1163	Prefix: dc.srv.prefix(),
	1164	Command: irc.RPL_INVITING,
	1165	Params: []string{dc.nick, dc.marshalNick(uc, nick), dc.marshalChannel(uc, channel)},
	1166	})
	1167	})
[177]	1168	case irc.ERR_UNKNOWNCOMMAND, irc.RPL_TRYAGAIN:
	1169	var command, reason string
	1170	if err := parseMessageParams(msg, nil, &command, &reason); err != nil {
	1171	return err
	1172	}
	1173
	1174	if command == "LIST" {
[181]	1175	ok := uc.endPendingLISTs(false)
[177]	1176	if !ok {
	1177	return fmt.Errorf("unexpected response for LIST: %q: no matching pending LIST", msg.Command)
	1178	}
	1179	uc.forEachDownstreamByID(downstreamID, func(dc *downstreamConn) {
	1180	dc.SendMessage(&irc.Message{
	1181	Prefix: uc.srv.prefix(),
	1182	Command: msg.Command,
	1183	Params: []string{dc.nick, "LIST", reason},
	1184	})
	1185	})
	1186	}
[152]	1187	case "TAGMSG":
	1188	// TODO: relay to downstream connections that accept message-tags
[155]	1189	case "ACK":
	1190	// Ignore
[198]	1191	case irc.RPL_NOWAWAY, irc.RPL_UNAWAY:
	1192	// Ignore
[16]	1193	case irc.RPL_YOURHOST, irc.RPL_CREATED:
[14]	1194	// Ignore
	1195	case irc.RPL_LUSERCLIENT, irc.RPL_LUSEROP, irc.RPL_LUSERUNKNOWN, irc.RPL_LUSERCHANNELS, irc.RPL_LUSERME:
	1196	// Ignore
	1197	case irc.RPL_MOTDSTART, irc.RPL_MOTD, irc.RPL_ENDOFMOTD:
	1198	// Ignore
[177]	1199	case irc.RPL_LISTSTART:
	1200	// Ignore
[14]	1201	case rpl_localusers, rpl_globalusers:
	1202	// Ignore
[96]	1203	case irc.RPL_STATSVLINE, rpl_statsping, irc.RPL_STATSBLINE, irc.RPL_STATSDLINE:
[14]	1204	// Ignore
[13]	1205	default:
[95]	1206	uc.logger.Printf("unhandled message: %v", msg)
[13]	1207	}
[14]	1208	return nil
[13]	1209	}
	1210
[174]	1211	func splitSpace(s string) []string {
	1212	return strings.FieldsFunc(s, func(r rune) bool {
	1213	return r == ' '
	1214	})
	1215	}
	1216
[55]	1217	func (uc *upstreamConn) register() {
[77]	1218	uc.nick = uc.network.Nick
	1219	uc.username = uc.network.Username
	1220	if uc.username == "" {
	1221	uc.username = uc.nick
	1222	}
	1223	uc.realname = uc.network.Realname
	1224	if uc.realname == "" {
	1225	uc.realname = uc.nick
	1226	}
	1227
[60]	1228	uc.SendMessage(&irc.Message{
[92]	1229	Command: "CAP",
	1230	Params: []string{"LS", "302"},
	1231	})
	1232
[93]	1233	if uc.network.Pass != "" {
	1234	uc.SendMessage(&irc.Message{
	1235	Command: "PASS",
	1236	Params: []string{uc.network.Pass},
	1237	})
	1238	}
	1239
[92]	1240	uc.SendMessage(&irc.Message{
[13]	1241	Command: "NICK",
[69]	1242	Params: []string{uc.nick},
[60]	1243	})
	1244	uc.SendMessage(&irc.Message{
[13]	1245	Command: "USER",
[77]	1246	Params: []string{uc.username, "0", "*", uc.realname},
[60]	1247	})
[44]	1248	}
[13]	1249
[197]	1250	func (uc *upstreamConn) runUntilRegistered() error {
	1251	for !uc.registered {
[212]	1252	msg, err := uc.ReadMessage()
[197]	1253	if err != nil {
	1254	return fmt.Errorf("failed to read message: %v", err)
	1255	}
	1256
	1257	if err := uc.handleMessage(msg); err != nil {
	1258	return fmt.Errorf("failed to handle message %q: %v", msg, err)
	1259	}
	1260	}
	1261
	1262	return nil
	1263	}
	1264
[95]	1265	func (uc *upstreamConn) requestSASL() bool {
	1266	if uc.network.SASL.Mechanism == "" {
	1267	return false
	1268	}
	1269
	1270	v, ok := uc.caps["sasl"]
	1271	if !ok {
	1272	return false
	1273	}
	1274	if v != "" {
	1275	mechanisms := strings.Split(v, ",")
	1276	found := false
	1277	for _, mech := range mechanisms {
	1278	if strings.EqualFold(mech, uc.network.SASL.Mechanism) {
	1279	found = true
	1280	break
	1281	}
	1282	}
	1283	if !found {
	1284	return false
	1285	}
	1286	}
	1287
	1288	return true
	1289	}
	1290
	1291	func (uc *upstreamConn) handleCapAck(name string, ok bool) error {
	1292	auth := &uc.network.SASL
	1293	switch name {
	1294	case "sasl":
	1295	if !ok {
	1296	uc.logger.Printf("server refused to acknowledge the SASL capability")
	1297	return nil
	1298	}
	1299
	1300	switch auth.Mechanism {
	1301	case "PLAIN":
	1302	uc.logger.Printf("starting SASL PLAIN authentication with username %q", auth.Plain.Username)
	1303	uc.saslClient = sasl.NewPlainClient("", auth.Plain.Username, auth.Plain.Password)
	1304	default:
	1305	return fmt.Errorf("unsupported SASL mechanism %q", name)
	1306	}
	1307
	1308	uc.SendMessage(&irc.Message{
	1309	Command: "AUTHENTICATE",
	1310	Params: []string{auth.Mechanism},
	1311	})
[152]	1312	case "message-tags":
	1313	uc.tagsSupported = ok
[155]	1314	case "labeled-response":
	1315	uc.labelsSupported = ok
[193]	1316	case "batch", "server-time":
	1317	// Nothing to do
	1318	default:
	1319	uc.logger.Printf("received CAP ACK/NAK for a cap we don't support: %v", name)
[95]	1320	}
	1321	return nil
	1322	}
	1323
[165]	1324	func (uc *upstreamConn) readMessages(ch chan<- event) error {
[13]	1325	for {
[210]	1326	msg, err := uc.ReadMessage()
[13]	1327	if err == io.EOF {
	1328	break
	1329	} else if err != nil {
	1330	return fmt.Errorf("failed to read IRC command: %v", err)
	1331	}
	1332
[165]	1333	ch <- eventUpstreamMessage{msg, uc}
[13]	1334	}
	1335
[45]	1336	return nil
[13]	1337	}
[60]	1338
[176]	1339	func (uc upstreamConn) SendMessageLabeled(downstreamID uint64, msg irc.Message) {
[155]	1340	if uc.labelsSupported {
	1341	if msg.Tags == nil {
	1342	msg.Tags = make(map[string]irc.TagValue)
	1343	}
[176]	1344	msg.Tags["label"] = irc.TagValue(fmt.Sprintf("sd-%d-%d", downstreamID, uc.nextLabelID))
[161]	1345	uc.nextLabelID++
[155]	1346	}
	1347	uc.SendMessage(msg)
	1348	}
[178]	1349
	1350	// TODO: handle moving logs when a network name changes, when support for this is added
[215]	1351	func (uc upstreamConn) appendLog(entity string, msg irc.Message) {
[178]	1352	if uc.srv.LogPath == "" {
	1353	return
	1354	}
[215]	1355
	1356	ml, ok := uc.messageLoggers[entity]
	1357	if !ok {
	1358	ml = newMessageLogger(uc, entity)
	1359	uc.messageLoggers[entity] = ml
[178]	1360	}
	1361
[215]	1362	if err := ml.Append(msg); err != nil {
	1363	uc.logger.Printf("failed to log message: %v", err)
[178]	1364	}
	1365	}
[198]	1366
[226]	1367	func (uc upstreamConn) produce(msg irc.Message) {
	1368	uc.network.ring.Produce(msg)
[227]	1369
	1370	uc.forEachDownstream(func(dc *downstreamConn) {
[231]	1371	dc.sendFromUpstream(msg, uc)
[227]	1372	})
[226]	1373	}
	1374
[198]	1375	func (uc *upstreamConn) updateAway() {
	1376	away := true
	1377	uc.forEachDownstream(func(*downstreamConn) {
	1378	away = false
	1379	})
	1380	if away == uc.away {
	1381	return
	1382	}
	1383	if away {
	1384	uc.SendMessage(&irc.Message{
	1385	Command: "AWAY",
	1386	Params: []string{"Auto away"},
	1387	})
	1388	} else {
	1389	uc.SendMessage(&irc.Message{
	1390	Command: "AWAY",
	1391	})
	1392	}
	1393	uc.away = away
	1394	}

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: code/trunk/upstream.go@ 237

Download in other formats: