Add square sizes flag to the payForMessage cli command

x/payment/client/cli/wirepayformessage.go

@@ -58,10 +58,7 @@ func CmdWirePayForMessage() *cobra.Command {
 				return err
 			}
 
-			squareSizes64 := make([]uint64, len(squareSizes))
-			for i := range squareSizes {
-				squareSizes64[i] = uint64(squareSizes[i])
-			}
+			squareSizes64 := parseSquareSizes(squareSizes)
 			pfmMsg, err := types.NewWirePayForMessage(namespace, message, squareSizes64...)
 
 			if err != nil {
@@ -116,3 +113,11 @@ func CmdWirePayForMessage() *cobra.Command {
 
 	return cmd
 }
+
+func parseSquareSizes(squareSizes []uint) []uint64 {
+	squareSizes64 := make([]uint64, len(squareSizes))
+	for i := range squareSizes {
+		squareSizes64[i] = uint64(squareSizes[i])
+	}
+	return squareSizes64
+}

x/payment/client/testutil/integration_test.go

@@ -96,6 +96,19 @@ func (s *IntegrationTestSuite) TestSubmitWirePayForMessage() {
 			},
 			false, 0, &sdk.TxResponse{},
 		},
+		{
+			"invalid transaction list of square sizes",
+			[]string{
+				hexNS,
+				hexMsg,
+				fmt.Sprintf("--from=%s", username),
+				fmt.Sprintf("--%s=%s", flags.FlagBroadcastMode, flags.BroadcastBlock),
+				fmt.Sprintf("--%s=%s", flags.FlagFees, sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdk.NewInt(2))).String()),
+				fmt.Sprintf("--%s=true", flags.FlagSkipConfirmation),
+				fmt.Sprintf("--%s=%s", cflags.FlagSquareSizes, "256,123,64"),
+			},
+			true, 0, &sdk.TxResponse{},
+		},
 	}
 
 	for _, tc := range testCases {

x/payment/types/wirepayformessage.go

@@ -30,6 +30,9 @@ func NewWirePayForMessage(namespace, message []byte, sizes ...uint64) (*MsgWireP
 
 	// generate the share commitments
 	for i, size := range sizes {
+		if !powerOf2(size) {
+			return nil, fmt.Errorf("Invalid square size, the size must be power of 2: %d", size)
+		}
 		commit, err := CreateCommitment(size, namespace, message)
 		if err != nil {
 			return nil, err
@@ -106,6 +109,10 @@ func (msg *MsgWirePayForMessage) ValidateBasic() error {
 			return err
 		}
 
+		if !powerOf2(commit.K) {
+			return fmt.Errorf("invalid square size, the size must be power of 2: %d", commit.K)
+		}
+
 		if string(calculatedCommit) != string(commit.ShareCommitment) {
 			return fmt.Errorf("invalid commit for square size %d", commit.K)
 		}
@@ -206,3 +213,13 @@ func ProcessWirePayForMessage(msg *MsgWirePayForMessage, squareSize uint64) (*tm
 
 	return &coreMsg, pfm, shareCommit.Signature, nil
 }
+
+
+// Check if number is power of 2
+func powerOf2(size uint64) bool {
+	if size & (size-1) == 0 {
+		return true
+	} else {
+		return false
+	}
+}