FilesExpand file tree

master

/

spot.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

146 lines (132 loc) · 3.47 KB

master

/

spot.go

Copy path

Top

File metadata and controls

• Code
• Blame

146 lines (132 loc) · 3.47 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

package fleet

import (

"bytes"

"context"

"fmt"

"log/slog"

"path"

"runtime/debug"

"strings"

"time"

"github.com/BottleFmt/gobottle"

"github.com/KarpelesLab/goupd"

"github.com/KarpelesLab/spotlib"

"github.com/KarpelesLab/spotproto"

"github.com/fxamacker/cbor/v2"

"github.com/quic-go/quic-go"

)

func (a *Agent) initSpot() {

k, err := a.getLocalKey()

if err != nil {

slog.Debug(fmt.Sprintf("failed to fetch local key: %s", err), "event", "fleet:spot:key_fail")

return

}

kc := gobottle.NewKeychain()

if tk, ok := k.(interface{ Keychain() *gobottle.Keychain }); ok {

kc = tk.Keychain()

} else {

kc.AddKey(k)

}

meta := map[string]string{"agent": "go-fleet"}

if info, ok := debug.ReadBuildInfo(); ok {

meta["project"] = path.Base(info.Path)

}

handlers := map[string]spotlib.MessageHandler{

"ping": a.spotPingHandler,

"fleet-announce": a.spotAnnounceHandler,

"fleet-fbin": a.spotFbinHandler,

"fleet-info": a.spotHandshakeHandler,

}

a.spot, err = spotlib.New(kc, meta, handlers)

if err != nil {

slog.Debug(fmt.Sprintf("failed to init spot: %s", err), "event", "fleet:spot:init_fail")

return

}

// we use spot's events handler so some events like "status" (0|1) are easily available

a.Events = a.spot.Events

a.id = a.spot.TargetId()

pkt, err := a.spot.ListenPacket("fleet-packet")

if err != nil {

slog.Error(fmt.Sprintf("[fleet] failed to listen for fleet-packet: %s", err), "event", "fleet:spot:packet_listen_fail")

} else {

a.quicT = &quic.Transport{

Conn: pkt,

ConnectionIDLength: 4,

}

}

}

func (a *Agent) Spot() *spotlib.Client {

return a.spot

}

func (a *Agent) shutdownSpot() {

if a.spot != nil {

a.spot.Close()

a.spot = nil

}

}

func (a *Agent) spotPingHandler(msg *spotproto.Message) ([]byte, error) {

return msg.Body, nil

}

func (a *Agent) spotAnnounceHandler(msg *spotproto.Message) ([]byte, error) {

// return announce packet

pkt := a.makeAnnouncePacket()

return cbor.Marshal(pkt)

}

func (a *Agent) spotHandshakeHandler(msg *spotproto.Message) ([]byte, error) {

// return handshake

pkt := &PacketHandshake{

Id: a.id,

Name: a.name,

Division: a.division,

Now: time.Now(),

Git: goupd.GIT_TAG,

Build: goupd.DATE_TAG,

Channel: goupd.CHANNEL,

}

return cbor.Marshal(pkt)

}

func (a *Agent) spotFbinHandler(msg *spotproto.Message) ([]byte, error) {

// got a fbin message from another peer, find it first

s := msg.Sender // k:xxx/yyy

if pos := strings.IndexByte(s, '/'); pos > 0 {

s = s[:pos]

}

p := a.GetPeer(s)

if p == nil {

// check if the id is part of the group

if a.group != nil {

ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)

defer cancel()

id, err := a.spot.GetIDCardForRecipient(ctx, s)

if err == nil {

if a.isFriend(id) {

// instanciate peer

p = a.makePeer(id)

}

}

}

// p might not be nil anymore at this point, check

if p == nil {

slog.Debug(fmt.Sprintf("[fleet] failed to locate peer %s", s), "event", "fleet:spot:peer_not_found")

return nil, nil

}

}

err := p.handleIncomingFbin(msg.Body)

if err != nil {

slog.Debug(fmt.Sprintf("[fleet] incoming packet handling failed: %s", err), "event", "fleet:spot:fbin_err")

}

return nil, nil

}

// isFriend returns true if id is member of our group

func (a *Agent) isFriend(id *gobottle.IDCard) bool {

grp := a.group

if grp == nil {

return false

}

for _, m := range id.Groups {

if bytes.Equal(m.Key, grp) {

return true

}

}

return false

}