From 3f5bcbc5569840fae74b6aa3485d511598451f91 Mon Sep 17 00:00:00 2001
From: Benjamin Wenger <ben@elixxir.ioo>
Date: Mon, 19 Sep 2022 09:38:40 -0700
Subject: [PATCH] fixed build issues

---
 channels/send_test.go         | 53 +++++++------------------------
 channels/userListener.go      | 11 +------
 channels/userListener_test.go | 59 +++++------------------------------
 3 files changed, 21 insertions(+), 102 deletions(-)

diff --git a/channels/send_test.go b/channels/send_test.go
index dff3abcdd..5a6e1999c 100644
--- a/channels/send_test.go
+++ b/channels/send_test.go
@@ -15,12 +15,12 @@ import (
 	"gitlab.com/elixxir/client/cmix/rounds"
 	"gitlab.com/elixxir/client/storage/versioned"
 	cryptoChannel "gitlab.com/elixxir/crypto/channel"
+	"gitlab.com/elixxir/crypto/rsa"
 	"gitlab.com/elixxir/ekv"
 	"gitlab.com/xx_network/crypto/csprng"
 	"testing"
 	"time"
 
-	"gitlab.com/xx_network/crypto/multicastRSA"
 	"gitlab.com/xx_network/primitives/id"
 	"gitlab.com/xx_network/primitives/id/ephemeral"
 
@@ -37,7 +37,7 @@ type mockBroadcastChannel struct {
 	payload []byte
 	params  cmix.CMIXParams
 
-	pk multicastRSA.PrivateKey
+	pk rsa.PrivateKey
 
 	crypto *cryptoBroadcast.Channel
 }
@@ -77,7 +77,7 @@ func (m *mockBroadcastChannel) BroadcastWithAssembler(assembler broadcast.Assemb
 	return rounds.Round{ID: 123}, ephemeral.Id{}, err
 }
 
-func (m *mockBroadcastChannel) BroadcastRSAtoPublic(pk multicastRSA.PrivateKey, payload []byte,
+func (m *mockBroadcastChannel) BroadcastRSAtoPublic(pk rsa.PrivateKey, payload []byte,
 	cMixParams cmix.CMIXParams) (rounds.Round, ephemeral.Id, error) {
 	m.hasRun = true
 
@@ -89,7 +89,7 @@ func (m *mockBroadcastChannel) BroadcastRSAtoPublic(pk multicastRSA.PrivateKey,
 }
 
 func (m *mockBroadcastChannel) BroadcastRSAToPublicWithAssembler(
-	pk multicastRSA.PrivateKey, assembler broadcast.Assembler,
+	pk rsa.PrivateKey, assembler broadcast.Assembler,
 	cMixParams cmix.CMIXParams) (rounds.Round, ephemeral.Id, error) {
 
 	m.hasRun = true
@@ -181,14 +181,8 @@ func TestSendGeneric(t *testing.T) {
 
 	//verify the message was handled correctly
 
-	//Unsize the broadcast
-	unsized, err := broadcast.DecodeSizedBroadcast(mbc.payload)
-	if err != nil {
-		t.Fatalf("Failed to decode the sized broadcast: %s", err)
-	}
-
 	//decode the user message
-	umi, err := unmarshalUserMessageInternal(unsized)
+	umi, err := unmarshalUserMessageInternal(mbc.payload)
 	if err != nil {
 		t.Fatalf("Failed to decode the user message: %s", err)
 	}
@@ -240,7 +234,8 @@ func TestAdminGeneric(t *testing.T) {
 	validUntil := time.Hour
 
 	rng := &csprng.SystemRNG{}
-	ch, priv, err := cryptoBroadcast.NewChannel("test", "test", rng)
+	ch, priv, err := cryptoBroadcast.NewChannel("test", "test",
+		1000, rng)
 	if err != nil {
 		t.Fatalf("Failed to generate channel: %+v", err)
 	}
@@ -260,13 +255,7 @@ func TestAdminGeneric(t *testing.T) {
 
 	//verify the message was handled correctly
 
-	//Unsize the broadcast
-	unsized, err := broadcast.DecodeSizedBroadcast(mbc.payload)
-	if err != nil {
-		t.Fatalf("Failed to decode the sized broadcast: %s", err)
-	}
-
-	msgID := cryptoChannel.MakeMessageID(unsized)
+	msgID := cryptoChannel.MakeMessageID(mbc.payload)
 
 	if !msgID.Equals(messageId) {
 		t.Errorf("The message IDs do not match. %s vs %s ",
@@ -275,7 +264,7 @@ func TestAdminGeneric(t *testing.T) {
 
 	//decode the channel message
 	chMgs := &ChannelMessage{}
-	err = proto.Unmarshal(unsized, chMgs)
+	err = proto.Unmarshal(mbc.payload, chMgs)
 	if err != nil {
 		t.Fatalf("Failed to decode the channel message: %s", err)
 	}
@@ -339,14 +328,8 @@ func TestSendMessage(t *testing.T) {
 
 	//verify the message was handled correctly
 
-	//Unsize the broadcast
-	unsized, err := broadcast.DecodeSizedBroadcast(mbc.payload)
-	if err != nil {
-		t.Fatalf("Failed to decode the sized broadcast: %s", err)
-	}
-
 	//decode the user message
-	umi, err := unmarshalUserMessageInternal(unsized)
+	umi, err := unmarshalUserMessageInternal(mbc.payload)
 	if err != nil {
 		t.Fatalf("Failed to decode the user message: %s", err)
 	}
@@ -427,14 +410,8 @@ func TestSendReply(t *testing.T) {
 
 	//verify the message was handled correctly
 
-	//Unsize the broadcast
-	unsized, err := broadcast.DecodeSizedBroadcast(mbc.payload)
-	if err != nil {
-		t.Fatalf("Failed to decode the sized broadcast: %s", err)
-	}
-
 	//decode the user message
-	umi, err := unmarshalUserMessageInternal(unsized)
+	umi, err := unmarshalUserMessageInternal(mbc.payload)
 	if err != nil {
 		t.Fatalf("Failed to decode the user message: %s", err)
 	}
@@ -514,14 +491,8 @@ func TestSendReaction(t *testing.T) {
 
 	//verify the message was handled correctly
 
-	//Unsize the broadcast
-	unsized, err := broadcast.DecodeSizedBroadcast(mbc.payload)
-	if err != nil {
-		t.Fatalf("Failed to decode the sized broadcast: %s", err)
-	}
-
 	//decode the user message
-	umi, err := unmarshalUserMessageInternal(unsized)
+	umi, err := unmarshalUserMessageInternal(mbc.payload)
 	if err != nil {
 		t.Fatalf("Failed to decode the user message: %s", err)
 	}
diff --git a/channels/userListener.go b/channels/userListener.go
index 227afb742..575c05e54 100644
--- a/channels/userListener.go
+++ b/channels/userListener.go
@@ -10,7 +10,6 @@ package channels
 import (
 	"crypto/ed25519"
 	jww "github.com/spf13/jwalterweatherman"
-	"gitlab.com/elixxir/client/broadcast"
 	"gitlab.com/elixxir/client/cmix/identity/receptionID"
 	"gitlab.com/elixxir/client/cmix/rounds"
 	"gitlab.com/elixxir/primitives/states"
@@ -31,16 +30,8 @@ type userListener struct {
 func (ul *userListener) Listen(payload []byte,
 	receptionID receptionID.EphemeralIdentity, round rounds.Round) {
 
-	//Remove the padding
-	payloadUnpadded, err := broadcast.DecodeSizedBroadcast(payload)
-	if err != nil {
-		jww.WARN.Printf("Failed to strip the padding on User Message "+
-			"on channel %s", ul.chID)
-		return
-	}
-
 	//Decode the message as a user message
-	umi, err := unmarshalUserMessageInternal(payloadUnpadded)
+	umi, err := unmarshalUserMessageInternal(payload)
 	if err != nil {
 		jww.WARN.Printf("Failed to unmarshal User Message on "+
 			"channel %s", ul.chID)
diff --git a/channels/userListener_test.go b/channels/userListener_test.go
index fcf69fbff..d58d540ff 100644
--- a/channels/userListener_test.go
+++ b/channels/userListener_test.go
@@ -16,7 +16,6 @@ import (
 
 	"github.com/golang/protobuf/proto"
 
-	"gitlab.com/elixxir/client/broadcast"
 	"gitlab.com/elixxir/client/cmix/identity/receptionID"
 	"gitlab.com/elixxir/client/cmix/rounds"
 	cryptoChannel "gitlab.com/elixxir/crypto/channel"
@@ -92,12 +91,6 @@ func TestUserListener_Listen(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -109,7 +102,7 @@ func TestUserListener_Listen(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if !dummy.gotData {
@@ -186,12 +179,6 @@ func TestUserListener_Listen_BadUserSig(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -203,7 +190,7 @@ func TestUserListener_Listen_BadUserSig(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
@@ -258,12 +245,6 @@ func TestUserListener_Listen_BadValidSig(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -275,7 +256,7 @@ func TestUserListener_Listen_BadValidSig(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
@@ -329,12 +310,6 @@ func TestUserListener_Listen_BadUnameTs(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -346,7 +321,7 @@ func TestUserListener_Listen_BadUnameTs(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
@@ -401,12 +376,6 @@ func TestUserListener_Listen_BadRound(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -418,7 +387,7 @@ func TestUserListener_Listen_BadRound(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
@@ -440,12 +409,6 @@ func TestUserListener_Listen_BadMessage(t *testing.T) {
 
 	umSerial := []byte("malformed")
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//build the listener
 	dummy := &triggerEventDummy{}
 
@@ -457,7 +420,7 @@ func TestUserListener_Listen_BadMessage(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
@@ -511,14 +474,8 @@ func TestUserListener_Listen_BadSizedBroadcast(t *testing.T) {
 		t.Fatalf("Failed to marshal proto: %+v", err)
 	}
 
-	umMsgSerialSized, err := broadcast.NewSizedBroadcast(
-		512, umSerial)
-	if err != nil {
-		t.Fatalf("Failed to size channel message: %+v", err)
-	}
-
 	//remove half the sized broadcast to make it malformed
-	umMsgSerialSized = umMsgSerialSized[:broadcast.GetSizedBroadcastSize(umMsgSerialSized)/2]
+	umSerial = umSerial[:len(umSerial)/2]
 
 	//build the listener
 	dummy := &triggerEventDummy{}
@@ -531,7 +488,7 @@ func TestUserListener_Listen_BadSizedBroadcast(t *testing.T) {
 	}
 
 	//call the listener
-	al.Listen(umMsgSerialSized, receptionID.EphemeralIdentity{}, r)
+	al.Listen(umSerial, receptionID.EphemeralIdentity{}, r)
 
 	//check the results
 	if dummy.gotData {
-- 
GitLab