Add whitelist referrer config

.env.sample

@@ -28,6 +28,7 @@ POLYGON_2ND_RPC="https://polygon-rpc.com/"
 
 DISABLE_UNLOCKER="false"
 BLACKLISTED_REFERRERS=""
+WHITELISTED_REFERRERS=""
 
 SOLANA_TX_MODE="NORMAL" # [NORMAL | JITO]
 JITO_ENDPOINT="https://frankfurt.mainnet.block-engine.jito.wtf"

src/config/global.ts

@@ -27,6 +27,7 @@ export interface SwiftFeeParams {
 export type GlobalConfig = {
 	ignoreReferrers: Set<string>;
 	blackListedReferrerAddresses: Set<string>;
+	whiteListedReferrerAddresses: Set<string>;
 	auctionTimeSeconds: number;
 	batchUnlockThreshold: number; // Optimal Number of swaps to select for unlocking
 	singleBatchChainIds: number[]; // Expensive chain-ids that use direct vaa post instead of batch (e.g ethereum)

src/driver/evm.ts

@@ -75,7 +75,7 @@ export class EvmFulfiller {
 						this.contractsConfig.contracts[chainId],
 					);
 
-					if (current < ethers.MaxUint256 - 1_000_000_000_000_000_000n) {
+					if (current < ethers.MaxUint256 - 1_000_000_000_000_000_000_000_000_000n) {
 						logger.info(`Setting allowance for ${driverToken.contract} on chain ${chainId}`);
 						await giveErc20Allowance(
 							this.walletHelper.getDriverWallet(chainId),
@@ -104,7 +104,7 @@ export class EvmFulfiller {
 						this.contractsConfig.evmFulfillHelpers[chainId],
 					);
 
-					if (current < ethers.MaxUint256 - 1_000_000_000_000_000_000n) {
+					if (current < ethers.MaxUint256 - 1_000_000_000_000_000_000_000_000_000n) {
 						logger.info(`Setting allowance for helper ${driverToken.contract} on chain ${chainId}`);
 						await giveErc20Allowance(
 							this.walletHelper.getDriverWallet(chainId),

src/index.ts

@@ -43,6 +43,11 @@ export async function main() {
 	const initialDynamicConfig = await fetchDynamicSdkParams();
 	rpcConfig.wormholeGuardianRpcs = initialDynamicConfig.wormholeGuardianRpcs.split(',');
 
+	let whiteListedReferrerAddresses: Set<string> = new Set();
+	if (process.env.WHITELISTED_REFERRERS) {
+		whiteListedReferrerAddresses = new Set(process.env.WHITELISTED_REFERRERS.split(','));
+	}
+
 	const globalConfig: GlobalConfig = {
 		ignoreReferrers: new Set(initialDynamicConfig.ignoreReferrers),
 		auctionTimeSeconds: initialDynamicConfig.auctionTimeSeconds,
@@ -59,6 +64,8 @@ export async function main() {
 			acc.add(x);
 			return acc;
 		}, new Set<string>()),
+
+		whiteListedReferrerAddresses: whiteListedReferrerAddresses,
 	};
 
 	const contracts: ContractsConfig = {

src/relayer.ts

@@ -124,6 +124,16 @@ export class Relayer {
 				return;
 			}
 
+			if (
+				this.gConf.whiteListedReferrerAddresses.size > 0 &&
+				!this.gConf.whiteListedReferrerAddresses.has(swap.referrerAddress)
+			) {
+				logger.warn(
+					`Whitelist enabled and referrer address ${swap.referrerAddress} is not whitelisted for ${swap.sourceTxHash}. discarding...`,
+				);
+				return;
+			}
+
 			if (
 				this.gConf.ignoreReferrers.has(swap.referrerAddress) ||
 				this.gConf.blackListedReferrerAddresses.has(swap.referrerAddress)