Add more duplicate account tests (#15746)

programs/bpf/rust/dup_accounts/src/lib.rs

@@ -2,46 +2,92 @@
 
 extern crate solana_program;
 use solana_program::{
-    account_info::AccountInfo, entrypoint, entrypoint::ProgramResult, msg,
-    program_error::ProgramError, pubkey::Pubkey,
+    account_info::AccountInfo,
+    entrypoint,
+    entrypoint::ProgramResult,
+    instruction::{AccountMeta, Instruction},
+    msg,
+    program::invoke,
+    program_error::ProgramError,
+    pubkey::Pubkey,
 };
 
 entrypoint!(process_instruction);
 fn process_instruction(
-    _program_id: &Pubkey,
+    program_id: &Pubkey,
     accounts: &[AccountInfo],
     instruction_data: &[u8],
 ) -> ProgramResult {
     match instruction_data[0] {
         1 => {
-            msg!("modify first account data");
+            msg!("modify account (2) data");
             accounts[2].data.borrow_mut()[0] = 1;
         }
         2 => {
-            msg!("modify first account data");
+            msg!("modify account (3) data");
             accounts[3].data.borrow_mut()[0] = 2;
         }
         3 => {
-            msg!("modify both account data");
+            msg!("modify account (2,3) data");
             accounts[2].data.borrow_mut()[0] += 1;
             accounts[3].data.borrow_mut()[0] += 2;
         }
         4 => {
-            msg!("modify first account lamports");
+            msg!("modify account (1,2) lamports");
             **accounts[1].lamports.borrow_mut() -= 1;
             **accounts[2].lamports.borrow_mut() += 1;
         }
         5 => {
-            msg!("modify first account lamports");
+            msg!("modify account (1,3) lamports");
             **accounts[1].lamports.borrow_mut() -= 2;
             **accounts[3].lamports.borrow_mut() += 2;
         }
         6 => {
-            msg!("modify both account lamports");
+            msg!("modify account (1,2,3) lamports");
             **accounts[1].lamports.borrow_mut() -= 3;
             **accounts[2].lamports.borrow_mut() += 1;
             **accounts[3].lamports.borrow_mut() += 2;
         }
+        7 => {
+            msg!("check account (0,1,2,3) privs");
+            assert!(accounts[0].is_signer);
+            assert!(!accounts[1].is_signer);
+            assert!(accounts[2].is_signer);
+            assert!(accounts[3].is_signer);
+
+            assert!(accounts[0].is_writable);
+            assert!(accounts[1].is_writable);
+            assert!(accounts[2].is_writable);
+            assert!(accounts[3].is_writable);
+
+            if accounts.len() > 4 {
+                let instruction = Instruction::new_with_bytes(
+                    *program_id,
+                    &[7],
+                    vec![
+                        AccountMeta::new(*accounts[0].key, true),
+                        AccountMeta::new(*accounts[1].key, false),
+                        AccountMeta::new(*accounts[2].key, false),
+                        AccountMeta::new_readonly(*accounts[3].key, true),
+                    ],
+                );
+                invoke(&instruction, &accounts)?;
+
+                let instruction = Instruction::new_with_bytes(
+                    *program_id,
+                    &[3],
+                    vec![
+                        AccountMeta::new(*accounts[0].key, true),
+                        AccountMeta::new(*accounts[1].key, false),
+                        AccountMeta::new(*accounts[2].key, false),
+                        AccountMeta::new(*accounts[3].key, false),
+                    ],
+                );
+                invoke(&instruction, &accounts)?;
+                assert_eq!(accounts[2].try_borrow_mut_data()?[0], 3);
+                assert_eq!(accounts[3].try_borrow_mut_data()?[0], 3);
+            }
+        }
         _ => {
             msg!("Unrecognized command");
             return Err(ProgramError::InvalidArgument);