gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add more duplicate account tests (#15746)

Browse Source
This commit is contained in:
Jack May
2021-03-05 20:36:27 -08:00
committed by GitHub
parent d09112fa6d
commit efcb58092e
3 changed files with 148 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

113
programs/bpf/c/src/dup_accounts/dup_accounts.c
View File

@ -1,5 +1,5 @@
/**
* @brief Example C-based BPF program that exercises duplicate keyed ka
* @brief Example C-based BPF program that exercises duplicate keyed accounts
* passed to it
*/
#include <solana_sdk.h>
@ -9,46 +9,87 @@
*/
extern uint64_t entrypoint(const uint8_t *input) {
SolAccountInfo ka[4];
SolParameters params = (SolParameters) { .ka = ka };
SolAccountInfo accounts[5];
SolParameters params = (SolParameters){.ka = accounts};
if (!sol_deserialize(input, &params, SOL_ARRAY_SIZE(ka))) {
if (!sol_deserialize(input, &params, SOL_ARRAY_SIZE(accounts))) {
return ERROR_INVALID_ARGUMENT;
}
switch (params.data[0]) {
case(1):
sol_log("modify first account data");
ka[2].data[0] = 1;
break;
case(2):
sol_log("modify first account data");
ka[3].data[0] = 2;
break;
case(3):
sol_log("modify both account data");
ka[2].data[0] += 1;
ka[3].data[0] += 2;
break;
case(4):
sol_log("modify first account lamports");
*ka[1].lamports -= 1;
*ka[2].lamports += 1;
break;
case(5):
sol_log("modify first account lamports");
*ka[1].lamports -= 2;
*ka[3].lamports += 2;
break;
case(6):
sol_log("modify both account lamports");
*ka[1].lamports -= 3;
*ka[2].lamports += 1;
*ka[3].lamports += 2;
break;
default:
sol_log("Unrecognized command");
return ERROR_INVALID_INSTRUCTION_DATA;
case (1):
sol_log("modify first account data");
accounts[2].data[0] = 1;
break;
case (2):
sol_log("modify first account data");
accounts[3].data[0] = 2;
break;
case (3):
sol_log("modify both account data");
accounts[2].data[0] += 1;
accounts[3].data[0] += 2;
break;
case (4):
sol_log("modify first account lamports");
*accounts[1].lamports -= 1;
*accounts[2].lamports += 1;
break;
case (5):
sol_log("modify first account lamports");
*accounts[1].lamports -= 2;
*accounts[3].lamports += 2;
break;
case (6):
sol_log("modify both account lamports");
*accounts[1].lamports -= 3;
*accounts[2].lamports += 1;
*accounts[3].lamports += 2;
break;
case (7):
sol_log("check account (0,1,2,3) privs");
sol_assert(accounts[0].is_signer);
sol_assert(!accounts[1].is_signer);
sol_assert(accounts[2].is_signer);
sol_assert(accounts[3].is_signer);
sol_assert(accounts[0].is_writable);
sol_assert(accounts[1].is_writable);
sol_assert(accounts[2].is_writable);
sol_assert(accounts[3].is_writable);
if (params.ka_num > 4) {
{
SolAccountMeta arguments[] = {{accounts[0].key, true, true},
{accounts[1].key, true, false},
{accounts[2].key, true, false},
{accounts[3].key, false, true}};
uint8_t data[] = {7};
const SolInstruction instruction = {
(SolPubkey *)params.program_id, arguments,
SOL_ARRAY_SIZE(arguments), data, SOL_ARRAY_SIZE(data)};
sol_assert(SUCCESS ==
sol_invoke(&instruction, accounts, params.ka_num));
}
{
SolAccountMeta arguments[] = {{accounts[0].key, true, true},
{accounts[1].key, true, false},
{accounts[2].key, true, false},
{accounts[3].key, true, false}};
uint8_t data[] = {3};
const SolInstruction instruction = {
(SolPubkey *)params.program_id, arguments,
SOL_ARRAY_SIZE(arguments), data, SOL_ARRAY_SIZE(data)};
sol_assert(SUCCESS ==
sol_invoke(&instruction, accounts, params.ka_num));
}
sol_assert(accounts[2].data[0] == 3);
sol_assert(accounts[3].data[0] == 3);
}
break;
default:
sol_log("Unrecognized command");
return ERROR_INVALID_INSTRUCTION_DATA;
}
return SUCCESS;
}