stack_wallet/lib/services/cashfusion/comms.dart

import 'dart:async';
import 'dart:io';

import 'package:protobuf/protobuf.dart';
import 'package:stackwallet/services/cashfusion/connection.dart';
import 'package:stackwallet/services/cashfusion/fusion.dart';
import 'package:stackwallet/services/cashfusion/fusion.pb.dart';
import 'package:stackwallet/services/cashfusion/socketwrapper.dart';
import 'package:stackwallet/services/cashfusion/util.dart';

typedef PbCreateFunc = GeneratedMessage Function();

Map<Type, PbCreateFunc> pbClassCreators = {
  CovertResponse: () => CovertResponse(),
  ClientMessage: () => ClientMessage(),
  InputComponent: () => InputComponent(),
  OutputComponent: () => OutputComponent(),
  BlankComponent: () => BlankComponent(),
  Component: () => Component(),
  InitialCommitment: () => InitialCommitment(),
  Proof: () => Proof(),
  ClientHello: () => ClientHello(),
  ServerHello: () => ServerHello(),
  JoinPools: () => JoinPools(),
  TierStatusUpdate: () => TierStatusUpdate(),
  FusionBegin: () => FusionBegin(),
  StartRound: () => StartRound(),
  PlayerCommit: () => PlayerCommit(),
  BlindSigResponses: () => BlindSigResponses(),
  AllCommitments: () => AllCommitments(),
  CovertComponent: () => CovertComponent(),
  ShareCovertComponents: () => ShareCovertComponents(),
  CovertTransactionSignature: () => CovertTransactionSignature(),
  FusionResult: () => FusionResult(),
  MyProofsList: () => MyProofsList(),
  TheirProofsList: () => TheirProofsList(),
  Blames: () => Blames(),
  RestartRound: () => RestartRound(),
  Error: () => Error(),
  Ping: () => Ping(),
  OK: () => OK(),
  ClientMessage: () => ClientMessage(),
  ServerMessage: () => ServerMessage(),
  CovertMessage: () => CovertMessage(),
  CovertResponse: () => CovertResponse()
};

Future<void> sendPb(
    Connection connection, Type pbClass, GeneratedMessage subMsg,
    {Duration? timeout}) async {
  // Construct the outer message with the submessage.

  if (pbClassCreators[pbClass] == null) {
    print('pbClassCreators[pbClass] is null');
    return;
  }

  var pbMessage = pbClassCreators[pbClass]!()..mergeFromMessage(subMsg);
  final msgBytes = pbMessage.writeToBuffer();
  try {
    await connection.sendMessage(msgBytes, timeout: timeout);
  } on SocketException {
    throw FusionError('Connection closed by remote');
  } on TimeoutException {
    throw FusionError('Timed out during send');
  } catch (e) {
    throw FusionError('Communications error: ${e.runtimeType}: $e');
  }
}

Future<void> sendPb2(SocketWrapper socketwrapper, Connection connection,
    Type pbClass, GeneratedMessage subMsg,
    {Duration? timeout}) async {
  // Construct the outer message with the submessage.

  if (pbClassCreators[pbClass] == null) {
    print('pbClassCreators[pbClass] is null');
    return;
  }

  var pbMessage = pbClassCreators[pbClass]!()..mergeFromMessage(subMsg);
  final msgBytes = pbMessage.writeToBuffer();
  try {
    await connection.sendMessageWithSocketWrapper(socketwrapper, msgBytes,
        timeout: timeout);
  } on SocketException {
    throw FusionError('Connection closed by remote');
  } on TimeoutException {
    throw FusionError('Timed out during send');
  } catch (e) {
    throw FusionError('Communications error: ${e.runtimeType}: $e');
  }
}

Future<Tuple<GeneratedMessage, String>> recvPb2(SocketWrapper socketwrapper,
    Connection connection, Type pbClass, List<String> expectedFieldNames,
    {Duration? timeout}) async {
  try {
    List<int> blob =
        await connection.recv_message2(socketwrapper, timeout: timeout);

    var pbMessage = pbClassCreators[pbClass]!()..mergeFromBuffer(blob);

    if (!pbMessage.isInitialized()) {
      throw FusionError('Incomplete message received');
    }

    for (var name in expectedFieldNames) {
      var fieldInfo = pbMessage.info_.byName[name];

      if (fieldInfo == null) {
        throw FusionError('Expected field not found in message: $name');
      }

      if (pbMessage.hasField(fieldInfo.tagNumber)) {
        return Tuple(pbMessage, name);
      }
    }

    throw FusionError(
        'None of the expected fields found in the received message');
  } catch (e) {
    // Handle different exceptions here
    if (e is SocketException) {
      throw FusionError('Connection closed by remote');
    } else if (e is InvalidProtocolBufferException) {
      throw FusionError('Message decoding error: ' + e.toString());
    } else if (e is TimeoutException) {
      throw FusionError('Timed out during receive');
    } else if (e is OSError && e.errorCode == 9) {
      throw FusionError('Connection closed by local');
    } else {
      throw FusionError(
          'Communications error: ${e.runtimeType}: ${e.toString()}');
    }
  }
}

Future<Tuple<GeneratedMessage, String>> recvPb(
    Connection connection, Type pbClass, List<String> expectedFieldNames,
    {Duration? timeout}) async {
  try {
    List<int> blob = await connection.recv_message(timeout: timeout);

    var pbMessage = pbClassCreators[pbClass]!()..mergeFromBuffer(blob);

    if (!pbMessage.isInitialized()) {
      throw FusionError('Incomplete message received');
    }

    for (var name in expectedFieldNames) {
      var fieldInfo = pbMessage.info_.byName[name];

      if (fieldInfo == null) {
        throw FusionError('Expected field not found in message: $name');
      }

      if (pbMessage.hasField(fieldInfo.tagNumber)) {
        return Tuple(pbMessage, name);
      }
    }

    throw FusionError(
        'None of the expected fields found in the received message');
  } catch (e) {
    // Handle different exceptions here
    if (e is SocketException) {
      throw FusionError('Connection closed by remote');
    } else if (e is InvalidProtocolBufferException) {
      throw FusionError('Message decoding error: ' + e.toString());
    } else if (e is TimeoutException) {
      throw FusionError('Timed out during receive');
    } else if (e is OSError && e.errorCode == 9) {
      throw FusionError('Connection closed by local');
    } else {
      throw FusionError(
          'Communications error: ${e.runtimeType}: ${e.toString()}');
    }
  }
}