feat: msg.data payload management

Author: matteoettam09

contracts/interfaces/IInternalStateOrchestrator.sol

@@ -60,13 +60,7 @@ interface IInternalStateOrchestrator is AutomationCompatibleInterface {
 
     /// @notice Callback function to decrypt the encrypted values
     /// @param requestID The request ID
-    /// @param decryptedValues The decrypted values
-    /// @param signatures The signatures
-    function callbackPreProcessDecrypt(
-        uint256 requestID,
-        uint256[] memory decryptedValues,
-        bytes[] memory signatures
-    ) external;
+    function callbackPreProcessDecrypt(uint256 requestID) external;
 
     /// @notice Get the selling orders
     /// @return tokens The tokens to sell

contracts/orchestrators/InternalStatesOrchestrator.sol

@@ -17,7 +17,7 @@ import { EventsLib } from "../libraries/EventsLib.sol";
 import { UtilitiesLib } from "../libraries/UtilitiesLib.sol";
 import { FHE, euint128 } from "@fhevm/solidity/lib/FHE.sol";
 import { SepoliaConfig } from "@fhevm/solidity/config/ZamaConfig.sol";
-
+import "hardhat/console.sol";
 /**
  * @title Internal States Orchestrator
  * @notice Contract that orchestrates internal state management
@@ -458,7 +458,7 @@ contract InternalStatesOrchestrator is SepoliaConfig, Ownable, ReentrancyGuard,
             (address[] memory portfolioTokens, euint128[] memory sharesPerAsset) = vault.getPortfolio();
 
             // STEP 1: LIVE PORTFOLIO
-            euint128 totalAssets = _ezero;
+            euint128 totalAssets = _ezero; // TODO: ensure no accidental _ezero modification.
             for (uint16 j = 0; j < portfolioTokens.length; ++j) {
                 address token = portfolioTokens[j];
                 euint128 shares = sharesPerAsset[j];
@@ -492,7 +492,10 @@ contract InternalStatesOrchestrator is SepoliaConfig, Ownable, ReentrancyGuard,
                 totalAssets = FHE.add(totalAssets, value);
                 _addTokenIfNotExists(token);
             }
-            _currentEpoch.encryptedVaultsTotalAssets[address(vault)] = totalAssets;
+            uint128 tmp = uint128(696969) + i;
+            console.log("tmp in the main function", tmp);
+            _currentEpoch.encryptedVaultsTotalAssets[address(vault)] = FHE.asEuint128(tmp); // TODO: fix this.
+            FHE.allowThis(_currentEpoch.encryptedVaultsTotalAssets[address(vault)]);
         }
 
         if (i1 == nVaults) {
@@ -502,6 +505,7 @@ contract InternalStatesOrchestrator is SepoliaConfig, Ownable, ReentrancyGuard,
                 currentMinibatchIndex = 0;
             } else {
                 uint16 mTokens = uint16(_currentEpoch.tokens.length);
+                encryptedValuesLength = nVaults + mTokens;
                 bytes32[] memory cypherTexts = new bytes32[](nVaults + mTokens);
                 for (uint16 i = 0; i < nVaults; ++i) {
                     address vault = encryptedVaultsEpoch[i];
@@ -518,24 +522,34 @@ contract InternalStatesOrchestrator is SepoliaConfig, Ownable, ReentrancyGuard,
                     cypherTexts[nVaults + i] = FHE.toBytes32(position);
                 }
 
-                // slither-disable-next-line unused-return
-                FHE.requestDecryption(cypherTexts, this.callbackPreProcessDecrypt.selector);
+                requestId = FHE.requestDecryption(cypherTexts, this.callbackPreProcessDecrypt.selector);
             }
         }
     }
     // slither-disable-end reentrancy-no-eth
 
+    uint256 requestId;
+    uint16 encryptedValuesLength;
+
     /// @inheritdoc IInternalStateOrchestrator
-    function callbackPreProcessDecrypt(
-        uint256 requestID,
-        uint256[] calldata decryptedValues,
-        bytes[] calldata signatures
-    ) external {
-        FHE.checkSignatures(requestID, signatures);
+    function callbackPreProcessDecrypt(uint256 requestID) external {
+        if (requestID != requestId) revert ErrorsLib.InvalidArguments();
+
+        // Read multiple uint128 values and signatures from msg.data
+        // Skip function selector (4 bytes) + requestID (32 bytes) = 36 bytes
+        uint256 offset = 36;
+
+        // Read multiple uint128 values using encryptedValuesLength
+        uint128[] memory values = new uint128[](encryptedValuesLength);
+        for (uint16 i = 0; i < encryptedValuesLength; i++) {
+            values[i] = uint128(uint256(bytes32(msg.data[offset:offset + 32])));
+            console.log("value in the callback", values[i]);
+            offset += 32;
+        }
 
         // Store decrypted values for processing in the next phase.
         // TODO: avoid breaking down this into two phases, consider letting Zama callback do all the work.
-        _decryptedValues = decryptedValues;
+        // _decryptedValues = decryptedValues;
 
         currentPhase = InternalUpkeepPhase.ProcessingDecryptedValues;
         currentMinibatchIndex = 0;