More relax instruction data length check

febo · febo · commit 2f37967727df · 2025-04-25T17:08:20.000+01:00
diff --git a/p-token/src/processor/approve_checked.rs b/p-token/src/processor/approve_checked.rs
@@ -7,7 +7,7 @@ use {
 #[inline(always)]
 pub fn process_approve_checked(accounts: &[AccountInfo], instruction_data: &[u8]) -> ProgramResult {
     // expected u64 (8) + u8 (1)
-    let (amount, decimals) = if instruction_data.len() == 9 {
+    let (amount, decimals) = if instruction_data.len() >= 9 {
         let (amount, decimals) = instruction_data.split_at(U64_BYTES);
         (
             // SAFETY: The size of `amount` is `U64_BYTES` bytes.
diff --git a/p-token/src/processor/burn_checked.rs b/p-token/src/processor/burn_checked.rs
@@ -7,7 +7,7 @@ use {
 #[inline(always)]
 pub fn process_burn_checked(accounts: &[AccountInfo], instruction_data: &[u8]) -> ProgramResult {
     // expected u64 (8) + u8 (1)
-    let (amount, decimals) = if instruction_data.len() == 9 {
+    let (amount, decimals) = if instruction_data.len() >= 9 {
         let (amount, decimals) = instruction_data.split_at(U64_BYTES);
         (
             // SAFETY: The size of `amount` is `U64_BYTES` bytes.
diff --git a/p-token/src/processor/transfer_checked.rs b/p-token/src/processor/transfer_checked.rs
@@ -10,7 +10,7 @@ pub fn process_transfer_checked(
     instruction_data: &[u8],
 ) -> ProgramResult {
     // expected u64 (8) + u8 (1)
-    let (amount, decimals) = if instruction_data.len() == 9 {
+    let (amount, decimals) = if instruction_data.len() >= 9 {
         let (amount, decimals) = instruction_data.split_at(U64_BYTES);
         (
             // SAFETY: The size of `amount` is `U64_BYTES` bytes.

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ use {`
`7`	`7`	`#[inline(always)]`
`8`	`8`	`pub fn process_approve_checked(accounts: &[AccountInfo], instruction_data: &[u8]) -> ProgramResult {`
`9`	`9`	`// expected u64 (8) + u8 (1)`
`10`		`- let (amount, decimals) = if instruction_data.len() == 9 {`
	`10`	`+ let (amount, decimals) = if instruction_data.len() >= 9 {`
`11`	`11`	`let (amount, decimals) = instruction_data.split_at(U64_BYTES);`
`12`	`12`	`(`
`13`	`13`	// SAFETY: The size of `amount` is `U64_BYTES` bytes.
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ use {`
`7`	`7`	`#[inline(always)]`
`8`	`8`	`pub fn process_burn_checked(accounts: &[AccountInfo], instruction_data: &[u8]) -> ProgramResult {`
`9`	`9`	`// expected u64 (8) + u8 (1)`
`10`		`- let (amount, decimals) = if instruction_data.len() == 9 {`
	`10`	`+ let (amount, decimals) = if instruction_data.len() >= 9 {`
`11`	`11`	`let (amount, decimals) = instruction_data.split_at(U64_BYTES);`
`12`	`12`	`(`
`13`	`13`	// SAFETY: The size of `amount` is `U64_BYTES` bytes.
Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ pub fn process_transfer_checked(`
`10`	`10`	`instruction_data: &[u8],`
`11`	`11`	`) -> ProgramResult {`
`12`	`12`	`// expected u64 (8) + u8 (1)`
`13`		`- let (amount, decimals) = if instruction_data.len() == 9 {`
	`13`	`+ let (amount, decimals) = if instruction_data.len() >= 9 {`
`14`	`14`	`let (amount, decimals) = instruction_data.split_at(U64_BYTES);`
`15`	`15`	`(`
`16`	`16`	// SAFETY: The size of `amount` is `U64_BYTES` bytes.