feat: check that sender is correct

Signed-off-by: Pablo Maldonado <[email protected]>
UMAprotocol · Jul 19, 2024 · 79fd454 · 79fd454
1 parent 69b24fb
commit 79fd454
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -50,8 +50,8 @@ const provider = getProvider();
 const { ovalConfigs, ovalConfigsShared } = env;
 
 // Initialize unlocker wallets for each Oval instance.
-const walletManager = WalletManager.getInstance(provider);
-walletManager.initialize(ovalConfigs, ovalConfigsShared);
+const walletManager = WalletManager.getInstance();
+walletManager.initialize(provider, ovalConfigs, ovalConfigsShared);
 
 // Initialize Oval discovery
 const ovalDiscovery = OvalDiscovery.getInstance();

diff --git a/src/lib/walletManager.ts b/src/lib/walletManager.ts
@@ -4,6 +4,7 @@ import { env } from "./env";
 import { retrieveGckmsKey } from "./gckms";
 import { isMochaTest } from "./helpers";
 import { OvalConfigs, OvalConfigsShared } from "./types";
+import { PermissionProxy__factory } from "../contract-types";
 
 type WalletConfig = {
     unlockerKey?: string;
@@ -120,6 +121,13 @@ export class WalletManager {
             const wallet = await this.createWallet(config);
             if (wallet) {
                 const walletPubKey = await wallet.getAddress();
+
+                // Check if the wallet is a sender in the PermissionProxy contract.
+                const isSender = await PermissionProxy__factory.connect(env.permissionProxyAddress, this.provider!).senders(wallet.address);
+                if (!isSender) {
+                    throw new Error(`Wallet ${wallet.address} is not a sender in the PermissionProxy contract.`);
+                }
+
                 this.sharedWallets.set(walletPubKey, wallet);
             }
         }