aboutsummaryrefslogtreecommitdiff
path: root/fallback.go
blob: 729a4e48c1affe9ded0629b8e9f51beb79b4ea22 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package proxy

import "github.com/HimbeerserverDE/mt"

func (cc *ClientConn) fallback() bool {
	if cc.fallbackFrom == "" {
		return false
	}

	fallback := config.Servers[cc.fallbackFrom].Fallback
	if fallback == "" {
		ack, _ := cc.SendCmd(cc.whyKicked)

		select {
		case <-cc.Closed():
		case <-ack:
			cc.Close()

			cc.mu.Lock()
			cc.srv = nil
			cc.mu.Unlock()

			if srv := cc.server(); srv != nil {
				srv.mu.Lock()
				srv.clt = nil
				srv.mu.Unlock()
			}
		}

		return true
	}

	// Use HopRaw so the fallback server doesn't get saved
	// as the last server.
	if err := cc.HopRaw(fallback); err != nil {
		cc.Log("<-", "fallback fail:", err)

		ack, _ := cc.SendCmd(&mt.ToCltKick{
			Reason: mt.Custom,
			Custom: "Fallback failed.",
		})

		select {
		case <-cc.Closed():
		case <-ack:
			cc.Close()

			cc.mu.Lock()
			cc.srv = nil
			cc.mu.Unlock()

			if srv := cc.server(); srv != nil {
				srv.mu.Lock()
				srv.clt = nil
				srv.mu.Unlock()
			}
		}
	}

	return true
}