Skip to content
Snippets Groups Projects
Commit 4ce3e79b authored by Jono Wenger's avatar Jono Wenger
Browse files

Fix file transfer e2e wrapper listener registration (used my ID instead of zero user)

parent 992c6f14
No related branches found
No related tags found
1 merge request!510Release
......@@ -39,11 +39,11 @@ const (
// NewFileTransfer is transmitted first on the initialization of a file
// transfer to inform the receiver about the incoming file.
NewFileTransfer = 50
NewFileTransfer MessageType = 50
// EndFileTransfer is sent once all file parts have been transmitted to
// inform the receiver that the file transfer has ended.
EndFileTransfer = 51
EndFileTransfer MessageType = 51
// ConnectionAuthenticationRequest is sent by the recipient
// of an authenticated connection request
......
......@@ -148,13 +148,13 @@ func (l *mockListener) Name() string {
type mockE2eHandler struct {
msgMap map[id.ID]map[catalog.MessageType][][]byte
listeners map[id.ID]map[catalog.MessageType]receive.Listener
listeners map[catalog.MessageType]receive.Listener
}
func newMockE2eHandler() *mockE2eHandler {
return &mockE2eHandler{
msgMap: make(map[id.ID]map[catalog.MessageType][][]byte),
listeners: make(map[id.ID]map[catalog.MessageType]receive.Listener),
listeners: make(map[catalog.MessageType]receive.Listener),
}
}
......@@ -178,7 +178,7 @@ func newMockE2e(myID *id.ID, handler *mockE2eHandler) *mockE2e {
func (m *mockE2e) SendE2E(mt catalog.MessageType, recipient *id.ID, payload []byte,
_ e2e.Params) ([]id.Round, e2eCrypto.MessageID, time.Time, error) {
m.handler.listeners[*recipient][mt].Hear(receive.Message{
m.handler.listeners[mt].Hear(receive.Message{
MessageType: mt,
Payload: payload,
Sender: m.myID,
......@@ -188,14 +188,9 @@ func (m *mockE2e) SendE2E(mt catalog.MessageType, recipient *id.ID, payload []by
return []id.Round{42}, e2eCrypto.MessageID{}, netTime.Now(), nil
}
func (m *mockE2e) RegisterListener(senderID *id.ID, mt catalog.MessageType,
func (m *mockE2e) RegisterListener(_ *id.ID, mt catalog.MessageType,
listener receive.Listener) receive.ListenerID {
if _, exists := m.handler.listeners[*senderID]; !exists {
m.handler.listeners[*senderID] =
map[catalog.MessageType]receive.Listener{mt: listener}
} else if _, exists = m.handler.listeners[*senderID][mt]; !exists {
m.handler.listeners[*senderID][mt] = listener
}
m.handler.listeners[mt] = listener
return receive.ListenerID{}
}
......
......@@ -57,7 +57,7 @@ func NewWrapper(receiveCB ft.ReceiveCallback, p Params, ft ft.FileTransfer,
}
// Register listener to receive new file transfers
w.e2e.RegisterListener(w.myID, catalog.NewFileTransfer, &listener{w})
w.e2e.RegisterListener(&id.ZeroUser, catalog.NewFileTransfer, &listener{w})
return w, nil
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment