@ignitionfi/fogo-stake-pool 1.0.3 → 1.1.0

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 import { AccountInfo, Connection, PublicKey, Signer, TransactionInstruction } from '@solana/web3.js';
 import BN from 'bn.js';
+import { AmountInput } from './instructions';
 import { StakeAccount, StakePool, ValidatorList } from './layouts';
 import { ValidatorAccount } from './utils';
 export { DEVNET_STAKE_POOL_PROGRAM_ID, STAKE_POOL_PROGRAM_ID, } from './constants';
@@ -57,7 +58,7 @@ export declare function depositStake(connection: Connection, stakePoolAddress: P
 /**
  * Creates instructions required to deposit sol to stake pool.
  */
-export declare function depositWsolWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, lamports: number, minimumPoolTokensOut?: number, destinationTokenAccount?: PublicKey, referrerTokenAccount?: PublicKey, depositAuthority?: PublicKey, payer?: PublicKey, 
+export declare function depositWsolWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, lamports: AmountInput, minimumPoolTokensOut?: AmountInput, destinationTokenAccount?: PublicKey, referrerTokenAccount?: PublicKey, depositAuthority?: PublicKey, payer?: PublicKey, 
 /**
  * Skip WSOL balance validation. Set to true when adding wrap instructions
  * in the same transaction that will fund the WSOL account before deposit.
@@ -69,7 +70,7 @@ skipBalanceCheck?: boolean): Promise<{
 /**
  * Creates instructions required to deposit sol to stake pool.
  */
-export declare function depositSol(connection: Connection, stakePoolAddress: PublicKey, from: PublicKey, lamports: number, destinationTokenAccount?: PublicKey, referrerTokenAccount?: PublicKey, depositAuthority?: PublicKey): Promise<{
+export declare function depositSol(connection: Connection, stakePoolAddress: PublicKey, from: PublicKey, lamports: AmountInput, destinationTokenAccount?: PublicKey, referrerTokenAccount?: PublicKey, depositAuthority?: PublicKey): Promise<{
     instructions: TransactionInstruction[];
     signers: Signer[];
 }>;
@@ -173,13 +174,13 @@ export declare function removeValidatorFromPool(connection: Connection, stakePoo
 /**
  * Creates instructions required to increase validator stake.
  */
-export declare function increaseValidatorStake(connection: Connection, stakePoolAddress: PublicKey, validatorVote: PublicKey, lamports: number, ephemeralStakeSeed?: number): Promise<{
+export declare function increaseValidatorStake(connection: Connection, stakePoolAddress: PublicKey, validatorVote: PublicKey, lamports: AmountInput, ephemeralStakeSeed?: number): Promise<{
     instructions: TransactionInstruction[];
 }>;
 /**
  * Creates instructions required to decrease validator stake.
  */
-export declare function decreaseValidatorStake(connection: Connection, stakePoolAddress: PublicKey, validatorVote: PublicKey, lamports: number, ephemeralStakeSeed?: number): Promise<{
+export declare function decreaseValidatorStake(connection: Connection, stakePoolAddress: PublicKey, validatorVote: PublicKey, lamports: AmountInput, ephemeralStakeSeed?: number): Promise<{
     instructions: TransactionInstruction[];
 }>;
 /**

package/dist/index.esm.js

@@ -438,6 +438,139 @@ const USER_STAKE_SEED_PREFIX = Buffer$1.from('user_stake');
 // for merges without a mismatch on credits observed
 const MINIMUM_ACTIVE_STAKE = 1000000;
 
+/**
+ * BN-based layout for data decoding using buffer-layout.
+ * Used for decoding on-chain account data (StakePool, ValidatorList, etc.)
+ */
+class BNDataLayout extends Layout {
+    constructor(span, signed, property) {
+        super(span, property);
+        this.blobLayout = blob(span);
+        this.signed = signed;
+    }
+    decode(b, offset = 0) {
+        const num = new BN(this.blobLayout.decode(b, offset), 10, 'le');
+        if (this.signed) {
+            return num.fromTwos(this.span * 8).clone();
+        }
+        return num;
+    }
+    encode(src, b, offset = 0) {
+        if (this.signed) {
+            src = src.toTwos(this.span * 8);
+        }
+        return this.blobLayout.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
+    }
+}
+/**
+ * Creates a u64 layout for data decoding (account layouts).
+ * Used in StakePoolLayout, ValidatorListLayout, etc.
+ */
+function u64(property) {
+    return new BNDataLayout(8, false, property);
+}
+/**
+ * BN-based layout for 64-bit unsigned integers using @solana/buffer-layout.
+ * Used for encoding instruction data with support for values > MAX_SAFE_INTEGER.
+ */
+class BNInstructionLayout extends BufferLayout.Layout {
+    constructor(span, signed, property) {
+        super(span, property);
+        this.blobLayout = BufferLayout.blob(span);
+        this.signed = signed;
+    }
+    decode(b, offset = 0) {
+        const num = new BN(this.blobLayout.decode(b, offset), 10, 'le');
+        if (this.signed) {
+            return num.fromTwos(this.span * 8).clone();
+        }
+        return num;
+    }
+    encode(src, b, offset = 0) {
+        if (this.signed) {
+            src = src.toTwos(this.span * 8);
+        }
+        return this.blobLayout.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
+    }
+    getSpan(_b, _offset) {
+        return this.span;
+    }
+}
+/**
+ * Creates a u64 layout for instruction encoding.
+ * Properly handles BN values larger than Number.MAX_SAFE_INTEGER.
+ * Compatible with @solana/buffer-layout.struct().
+ */
+// eslint-disable-next-line ts/no-explicit-any
+function u64Instruction(property) {
+    return new BNInstructionLayout(8, false, property);
+}
+class WrappedLayout extends Layout {
+    constructor(layout, decoder, encoder, property) {
+        super(layout.span, property);
+        this.layout = layout;
+        this.decoder = decoder;
+        this.encoder = encoder;
+    }
+    decode(b, offset) {
+        return this.decoder(this.layout.decode(b, offset));
+    }
+    encode(src, b, offset) {
+        return this.layout.encode(this.encoder(src), b, offset);
+    }
+    getSpan(b, offset) {
+        return this.layout.getSpan(b, offset);
+    }
+}
+function publicKey(property) {
+    return new WrappedLayout(blob(32), (b) => new PublicKey(b), (key) => key.toBuffer(), property);
+}
+class OptionLayout extends Layout {
+    constructor(layout, property) {
+        super(-1, property);
+        this.layout = layout;
+        this.discriminator = u8();
+    }
+    encode(src, b, offset = 0) {
+        if (src === null || src === undefined) {
+            return this.discriminator.encode(0, b, offset);
+        }
+        this.discriminator.encode(1, b, offset);
+        return this.layout.encode(src, b, offset + 1) + 1;
+    }
+    decode(b, offset = 0) {
+        const discriminator = this.discriminator.decode(b, offset);
+        if (discriminator === 0) {
+            return null;
+        }
+        else if (discriminator === 1) {
+            return this.layout.decode(b, offset + 1);
+        }
+        throw new Error(`Invalid option ${this.property}`);
+    }
+    getSpan(b, offset = 0) {
+        const discriminator = this.discriminator.decode(b, offset);
+        if (discriminator === 0) {
+            return 1;
+        }
+        else if (discriminator === 1) {
+            return this.layout.getSpan(b, offset + 1) + 1;
+        }
+        throw new Error(`Invalid option ${this.property}`);
+    }
+}
+function option(layout, property) {
+    return new OptionLayout(layout, property);
+}
+function vec(elementLayout, property) {
+    const length = u32('length');
+    const layout = struct([
+        length,
+        seq(elementLayout, offset(length, -length.span), 'values'),
+    ]);
+    return new WrappedLayout(layout, ({ values }) => values, values => ({ values }), property);
+}
+
 /**
  * Populate a buffer of instruction data using an InstructionType
  * @internal
@@ -556,95 +689,6 @@ function findUserStakeProgramAddress(programId, userWallet, seed) {
     return publicKey;
 }
 
-class BNLayout extends Layout {
-    constructor(span, signed, property) {
-        super(span, property);
-        this.blob = blob(span);
-        this.signed = signed;
-    }
-    decode(b, offset = 0) {
-        const num = new BN(this.blob.decode(b, offset), 10, 'le');
-        if (this.signed) {
-            return num.fromTwos(this.span * 8).clone();
-        }
-        return num;
-    }
-    encode(src, b, offset = 0) {
-        if (this.signed) {
-            src = src.toTwos(this.span * 8);
-        }
-        return this.blob.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
-    }
-}
-function u64(property) {
-    return new BNLayout(8, false, property);
-}
-class WrappedLayout extends Layout {
-    constructor(layout, decoder, encoder, property) {
-        super(layout.span, property);
-        this.layout = layout;
-        this.decoder = decoder;
-        this.encoder = encoder;
-    }
-    decode(b, offset) {
-        return this.decoder(this.layout.decode(b, offset));
-    }
-    encode(src, b, offset) {
-        return this.layout.encode(this.encoder(src), b, offset);
-    }
-    getSpan(b, offset) {
-        return this.layout.getSpan(b, offset);
-    }
-}
-function publicKey(property) {
-    return new WrappedLayout(blob(32), (b) => new PublicKey(b), (key) => key.toBuffer(), property);
-}
-class OptionLayout extends Layout {
-    constructor(layout, property) {
-        super(-1, property);
-        this.layout = layout;
-        this.discriminator = u8();
-    }
-    encode(src, b, offset = 0) {
-        if (src === null || src === undefined) {
-            return this.discriminator.encode(0, b, offset);
-        }
-        this.discriminator.encode(1, b, offset);
-        return this.layout.encode(src, b, offset + 1) + 1;
-    }
-    decode(b, offset = 0) {
-        const discriminator = this.discriminator.decode(b, offset);
-        if (discriminator === 0) {
-            return null;
-        }
-        else if (discriminator === 1) {
-            return this.layout.decode(b, offset + 1);
-        }
-        throw new Error(`Invalid option ${this.property}`);
-    }
-    getSpan(b, offset = 0) {
-        const discriminator = this.discriminator.decode(b, offset);
-        if (discriminator === 0) {
-            return 1;
-        }
-        else if (discriminator === 1) {
-            return this.layout.getSpan(b, offset + 1) + 1;
-        }
-        throw new Error(`Invalid option ${this.property}`);
-    }
-}
-function option(layout, property) {
-    return new OptionLayout(layout, property);
-}
-function vec(elementLayout, property) {
-    const length = u32('length');
-    const layout = struct([
-        length,
-        seq(elementLayout, offset(length, -length.span), 'values'),
-    ]);
-    return new WrappedLayout(layout, ({ values }) => values, values => ({ values }), property);
-}
-
 const feeFields = [u64('denominator'), u64('numerator')];
 var AccountType;
 (function (AccountType) {
@@ -987,11 +1031,50 @@ function arrayChunk(array, size) {
     return result;
 }
 
+/**
+ * Converts various numeric types to BN for safe large number handling.
+ * @internal
+ */
+function toBN(value) {
+    if (BN.isBN(value)) {
+        return value;
+    }
+    if (typeof value === 'bigint') {
+        return new BN(value.toString());
+    }
+    if (typeof value === 'string') {
+        // Validate string is a valid non-negative integer
+        const trimmed = value.trim();
+        if (!/^\d+$/.test(trimmed)) {
+            throw new Error(`Invalid amount string: "${value}". Must be a non-negative integer.`);
+        }
+        return new BN(trimmed);
+    }
+    if (typeof value === 'number') {
+        if (!Number.isFinite(value)) {
+            throw new Error('Invalid amount: must be a finite number');
+        }
+        if (value < 0) {
+            throw new Error('Invalid amount: must be non-negative');
+        }
+        if (!Number.isInteger(value)) {
+            throw new Error('Invalid amount: must be an integer (lamports)');
+        }
+        // CRITICAL: Numbers > MAX_SAFE_INTEGER have already lost precision
+        // We throw an error instead of silently corrupting data
+        if (value > Number.MAX_SAFE_INTEGER) {
+            throw new Error(`Amount ${value} exceeds Number.MAX_SAFE_INTEGER (9,007,199,254,740,991). ` +
+                `Use BigInt or BN for large values to avoid precision loss.`);
+        }
+        return new BN(value);
+    }
+    throw new Error(`Invalid amount type: ${typeof value}`);
+}
 // 'UpdateTokenMetadata' and 'CreateTokenMetadata' have dynamic layouts
 const MOVE_STAKE_LAYOUT = BufferLayout.struct([
     BufferLayout.u8('instruction'),
-    BufferLayout.ns64('lamports'),
-    BufferLayout.ns64('transientStakeSeed'),
+    u64Instruction('lamports'),
+    u64Instruction('transientStakeSeed'),
 ]);
 const UPDATE_VALIDATOR_LIST_BALANCE_LAYOUT = BufferLayout.struct([
     BufferLayout.u8('instruction'),
@@ -1066,7 +1149,7 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 10,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('poolTokens'),
+            u64Instruction('poolTokens'),
         ]),
     },
     /// Deposit SOL directly into the pool's reserve account. The output is a "pool" token
@@ -1075,7 +1158,7 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 14,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     /// Withdraw SOL directly from the pool's reserve account. Fails if the
@@ -1084,25 +1167,25 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 16,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('poolTokens'),
+            u64Instruction('poolTokens'),
         ]),
     },
     IncreaseAdditionalValidatorStake: {
         index: 19,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
-            BufferLayout.ns64('transientStakeSeed'),
-            BufferLayout.ns64('ephemeralStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('transientStakeSeed'),
+            u64Instruction('ephemeralStakeSeed'),
         ]),
     },
     DecreaseAdditionalValidatorStake: {
         index: 20,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
-            BufferLayout.ns64('transientStakeSeed'),
-            BufferLayout.ns64('ephemeralStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('transientStakeSeed'),
+            u64Instruction('ephemeralStakeSeed'),
         ]),
     },
     DecreaseValidatorStakeWithReserve: {
@@ -1117,62 +1200,62 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 23,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     WithdrawStakeWithSlippage: {
         index: 24,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('poolTokensIn'),
-            BufferLayout.ns64('minimumLamportsOut'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
         ]),
     },
     DepositSolWithSlippage: {
         index: 25,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     WithdrawSolWithSlippage: {
         index: 26,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     DepositWsolWithSession: {
         index: 27,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamportsIn'),
-            BufferLayout.ns64('minimumPoolTokensOut'),
+            u64Instruction('lamportsIn'),
+            u64Instruction('minimumPoolTokensOut'),
         ]),
     },
     WithdrawWsolWithSession: {
         index: 28,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('poolTokensIn'),
-            BufferLayout.ns64('minimumLamportsOut'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
         ]),
     },
     WithdrawStakeWithSession: {
         index: 29,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('poolTokensIn'),
-            BufferLayout.ns64('minimumLamportsOut'),
-            BufferLayout.ns64('userStakeSeed'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
+            u64Instruction('userStakeSeed'),
         ]),
     },
     WithdrawFromStakeAccountWithSession: {
         index: 30,
         layout: BufferLayout.struct([
             BufferLayout.u8('instruction'),
-            BufferLayout.ns64('lamports'),
-            BufferLayout.ns64('userStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('userStakeSeed'),
         ]),
     },
 });
@@ -1304,7 +1387,10 @@ class StakePoolInstruction {
     static increaseValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, transientStake, validatorStake, validatorVote, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.IncreaseValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1334,7 +1420,11 @@ class StakePoolInstruction {
     static increaseAdditionalValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, transientStake, validatorStake, validatorVote, lamports, transientStakeSeed, ephemeralStake, ephemeralStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.IncreaseAdditionalValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed, ephemeralStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+            ephemeralStakeSeed: toBN(ephemeralStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1364,7 +1454,10 @@ class StakePoolInstruction {
     static decreaseValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, validatorStake, transientStake, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1390,7 +1483,10 @@ class StakePoolInstruction {
     static decreaseValidatorStakeWithReserve(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, validatorStake, transientStake, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseValidatorStakeWithReserve;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1417,7 +1513,11 @@ class StakePoolInstruction {
     static decreaseAdditionalValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, validatorStake, transientStake, lamports, transientStakeSeed, ephemeralStakeSeed, ephemeralStake, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseAdditionalValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed, ephemeralStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+            ephemeralStakeSeed: toBN(ephemeralStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1474,7 +1574,7 @@ class StakePoolInstruction {
     static depositSol(params) {
         const { programId, stakePool, withdrawAuthority, depositAuthority, reserveStake, fundingAccount, destinationPoolAccount, managerFeeAccount, referralPoolAccount, poolMint, lamports, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DepositSol;
-        const data = encodeData(type, { lamports });
+        const data = encodeData(type, { lamports: toBN(lamports) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: withdrawAuthority, isSigner: false, isWritable: false },
@@ -1507,8 +1607,8 @@ class StakePoolInstruction {
         var _a;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DepositWsolWithSession;
         const data = encodeData(type, {
-            lamportsIn: params.lamportsIn,
-            minimumPoolTokensOut: params.minimumPoolTokensOut,
+            lamportsIn: toBN(params.lamportsIn),
+            minimumPoolTokensOut: toBN(params.minimumPoolTokensOut),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1550,7 +1650,7 @@ class StakePoolInstruction {
     static withdrawStake(params) {
         const { programId, stakePool, validatorList, withdrawAuthority, validatorStake, destinationStake, destinationStakeAuthority, sourceTransferAuthority, sourcePoolAccount, managerFeeAccount, poolMint, poolTokens, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStake;
-        const data = encodeData(type, { poolTokens });
+        const data = encodeData(type, { poolTokens: toBN(poolTokens) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: validatorList, isSigner: false, isWritable: true },
@@ -1578,7 +1678,7 @@ class StakePoolInstruction {
     static withdrawSol(params) {
         const { programId, stakePool, withdrawAuthority, sourceTransferAuthority, sourcePoolAccount, reserveStake, destinationSystemAccount, managerFeeAccount, solWithdrawAuthority, poolMint, poolTokens, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawSol;
-        const data = encodeData(type, { poolTokens });
+        const data = encodeData(type, { poolTokens: toBN(poolTokens) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: withdrawAuthority, isSigner: false, isWritable: false },
@@ -1613,8 +1713,8 @@ class StakePoolInstruction {
     static withdrawWsolWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawWsolWithSession;
         const data = encodeData(type, {
-            poolTokensIn: params.poolTokensIn,
-            minimumLamportsOut: params.minimumLamportsOut,
+            poolTokensIn: toBN(params.poolTokensIn),
+            minimumLamportsOut: toBN(params.minimumLamportsOut),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1656,9 +1756,9 @@ class StakePoolInstruction {
     static withdrawStakeWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStakeWithSession;
         const data = encodeData(type, {
-            poolTokensIn: params.poolTokensIn,
-            minimumLamportsOut: params.minimumLamportsOut,
-            userStakeSeed: params.userStakeSeed,
+            poolTokensIn: toBN(params.poolTokensIn),
+            minimumLamportsOut: toBN(params.minimumLamportsOut),
+            userStakeSeed: toBN(params.userStakeSeed),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1693,8 +1793,8 @@ class StakePoolInstruction {
     static withdrawFromStakeAccountWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawFromStakeAccountWithSession;
         const data = encodeData(type, {
-            lamports: params.lamports,
-            userStakeSeed: params.userStakeSeed,
+            lamports: toBN(params.lamports),
+            userStakeSeed: toBN(params.userStakeSeed),
         });
         const keys = [
             { pubkey: params.userStakeAccount, isSigner: false, isWritable: true },
@@ -1983,10 +2083,18 @@ skipBalanceCheck = false) {
     if (!skipBalanceCheck) {
         const tokenAccountInfo = await connection.getTokenAccountBalance(wsolTokenAccount, 'confirmed');
         const wsolBalance = tokenAccountInfo
-            ? parseInt(tokenAccountInfo.value.amount)
-            : 0;
-        if (wsolBalance < lamports) {
-            throw new Error(`Not enough WSOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(wsolBalance)} WSOL.`);
+            ? BigInt(tokenAccountInfo.value.amount)
+            : BigInt(0);
+        // Convert lamports to BigInt for comparison
+        const lamportsBigInt = typeof lamports === 'bigint'
+            ? lamports
+            : typeof lamports === 'string'
+                ? BigInt(lamports)
+                : BN.isBN(lamports)
+                    ? BigInt(lamports.toString())
+                    : BigInt(lamports);
+        if (wsolBalance < lamportsBigInt) {
+            throw new Error(`Not enough WSOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(Number(wsolBalance))} WSOL.`);
         }
     }
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
@@ -2032,7 +2140,15 @@ skipBalanceCheck = false) {
  */
 async function depositSol(connection, stakePoolAddress, from, lamports, destinationTokenAccount, referrerTokenAccount, depositAuthority) {
     const fromBalance = await connection.getBalance(from, 'confirmed');
-    if (fromBalance < lamports) {
+    // Convert lamports to BigInt for comparison
+    const lamportsBigInt = typeof lamports === 'bigint'
+        ? lamports
+        : typeof lamports === 'string'
+            ? BigInt(lamports)
+            : BN.isBN(lamports)
+                ? BigInt(lamports.toString())
+                : BigInt(lamports);
+    if (BigInt(fromBalance) < lamportsBigInt) {
         throw new Error(`Not enough SOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(fromBalance)} SOL.`);
     }
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
@@ -2046,7 +2162,7 @@ async function depositSol(connection, stakePoolAddress, from, lamports, destinat
     instructions.push(SystemProgram.transfer({
         fromPubkey: from,
         toPubkey: userSolTransfer.publicKey,
-        lamports,
+        lamports: lamportsBigInt,
     }));
     // Create token account if not specified
     if (!destinationTokenAccount) {