solana_sdk.h 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545
  1. // SPDX-License-Identifier: Apache-2.0
  2. #pragma once
  3. /**
  4. * Numeric types
  5. */
  6. #ifndef __LP64__
  7. #error LP64 data model required
  8. #endif
  9. /** Indicates the instruction was processed successfully */
  10. #define SUCCESS 0
  11. /**
  12. * Builtin program status values occupy the upper 32 bits of the program return
  13. * value. Programs may define their own error values but they must be confined
  14. * to the lower 32 bits.
  15. */
  16. #define TO_BUILTIN(error) ((uint64_t)(error) << 32)
  17. /** Note: Not applicable to program written in C */
  18. #define ERROR_CUSTOM_ZERO TO_BUILTIN(1)
  19. /** The arguments provided to a program instruction where invalid */
  20. #define ERROR_INVALID_ARGUMENT TO_BUILTIN(2)
  21. /** An instruction's data contents was invalid */
  22. #define ERROR_INVALID_INSTRUCTION_DATA TO_BUILTIN(3)
  23. /** An account's data contents was invalid */
  24. #define ERROR_INVALID_ACCOUNT_DATA TO_BUILTIN(4)
  25. /** An account's data was too small */
  26. #define ERROR_ACCOUNT_DATA_TOO_SMALL TO_BUILTIN(5)
  27. /** An account's balance was too small to complete the instruction */
  28. #define ERROR_INSUFFICIENT_FUNDS TO_BUILTIN(6)
  29. /** The account did not have the expected program id */
  30. #define ERROR_INCORRECT_PROGRAM_ID TO_BUILTIN(7)
  31. /** A signature was required but not found */
  32. #define ERROR_MISSING_REQUIRED_SIGNATURES TO_BUILTIN(8)
  33. /** An initialize instruction was sent to an account that has already been initialized */
  34. #define ERROR_ACCOUNT_ALREADY_INITIALIZED TO_BUILTIN(9)
  35. /** An attempt to operate on an account that hasn't been initialized */
  36. #define ERROR_UNINITIALIZED_ACCOUNT TO_BUILTIN(10)
  37. /** The instruction expected additional account keys */
  38. #define ERROR_NOT_ENOUGH_ACCOUNT_KEYS TO_BUILTIN(11)
  39. /** Note: Not applicable to program written in C */
  40. #define ERROR_ACCOUNT_BORROW_FAILED TO_BUILTIN(12)
  41. /** The length of the seed is too long for address generation */
  42. #define MAX_SEED_LENGTH_EXCEEDED TO_BUILTIN(13)
  43. /** Provided seeds do not result in a valid address */
  44. #define INVALID_SEEDS TO_BUILTIN(14)
  45. /** Need more account */
  46. #define ERROR_NEW_ACCOUNT_NEEDED TO_BUILTIN(15)
  47. /**
  48. * Boolean type
  49. */
  50. #ifndef __cplusplus
  51. #include <stdbool.h>
  52. #endif
  53. /**
  54. * Prints a string to stdout
  55. */
  56. void sol_log_(const char *, uint64_t);
  57. #define sol_log(message) sol_log_(message, sol_strlen(message))
  58. /**
  59. * Prints a 64 bit values represented in hexadecimal to stdout
  60. */
  61. void sol_log_64_(uint64_t, uint64_t, uint64_t, uint64_t, uint64_t);
  62. #define sol_log_64 sol_log_64_
  63. /**
  64. * Size of Public key in bytes
  65. */
  66. #define SIZE_PUBKEY 32
  67. /**
  68. * Public key
  69. */
  70. typedef struct
  71. {
  72. uint8_t x[SIZE_PUBKEY];
  73. } SolPubkey;
  74. /**
  75. * Compares two public keys
  76. *
  77. * @param one First public key
  78. * @param two Second public key
  79. * @return true if the same
  80. */
  81. static bool SolPubkey_same(const SolPubkey *one, const SolPubkey *two)
  82. {
  83. for (int i = 0; i < sizeof(*one); i++)
  84. {
  85. if (one->x[i] != two->x[i])
  86. {
  87. return false;
  88. }
  89. }
  90. return true;
  91. }
  92. /**
  93. * Keyed Account
  94. */
  95. typedef struct
  96. {
  97. SolPubkey *key; /** Public key of the account */
  98. uint64_t *lamports; /** Number of lamports owned by this account */
  99. uint64_t data_len; /** Length of data in bytes */
  100. uint8_t *data; /** On-chain data within this account */
  101. SolPubkey *owner; /** Program that owns this account */
  102. uint64_t rent_epoch; /** The epoch at which this account will next owe rent */
  103. bool is_signer; /** Transaction was signed by this account's key? */
  104. bool is_writable; /** Is the account writable? */
  105. bool executable; /** This account's data contains a loaded program (and is now read-only) */
  106. } SolAccountInfo;
  107. /**
  108. * Copies memory
  109. */
  110. static void sol_memcpy(void *dst, const void *src, int len)
  111. {
  112. for (int i = 0; i < len; i++)
  113. {
  114. *((uint8_t *)dst + i) = *((const uint8_t *)src + i);
  115. }
  116. }
  117. /**
  118. * Compares memory
  119. */
  120. static int sol_memcmp(const void *s1, const void *s2, int n)
  121. {
  122. for (int i = 0; i < n; i++)
  123. {
  124. uint8_t diff = *((uint8_t *)s1 + i) - *((const uint8_t *)s2 + i);
  125. if (diff)
  126. {
  127. return diff;
  128. }
  129. }
  130. return 0;
  131. }
  132. /**
  133. * Fill a byte string with a byte value
  134. */
  135. static void sol_memset(void *b, int c, size_t len)
  136. {
  137. uint8_t *a = (uint8_t *)b;
  138. while (len > 0)
  139. {
  140. *a = c;
  141. a++;
  142. len--;
  143. }
  144. }
  145. /**
  146. * Find length of string
  147. */
  148. static size_t sol_strlen(const char *s)
  149. {
  150. size_t len = 0;
  151. while (*s)
  152. {
  153. len++;
  154. s++;
  155. }
  156. return len;
  157. }
  158. /**
  159. * Computes the number of elements in an array
  160. */
  161. #define SOL_ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
  162. /**
  163. * Panics
  164. *
  165. * Prints the line number where the panic occurred and then causes
  166. * the BPF VM to immediately halt execution. No accounts' data are updated
  167. */
  168. void sol_panic_(const char *, uint64_t, uint64_t, uint64_t);
  169. #define sol_panic() sol_panic_(__FILE__, sizeof(__FILE__), __LINE__, 0)
  170. /**
  171. * Asserts
  172. */
  173. #define sol_assert(expr) \
  174. if (!(expr)) \
  175. { \
  176. sol_panic(); \
  177. }
  178. /**
  179. * Seed used to create a program address or passed to sol_invoke_signed
  180. */
  181. typedef struct
  182. {
  183. const uint8_t *addr; /** Seed bytes */
  184. uint64_t len; /** Length of the seed bytes */
  185. } SolSignerSeed;
  186. /**
  187. * Structure that the program's entrypoint input data is deserialized into.
  188. */
  189. typedef struct
  190. {
  191. SolAccountInfo ka[10]; /** Pointer to an array of SolAccountInfo, must already
  192. point to an array of SolAccountInfos */
  193. uint64_t ka_num; /** Number of SolAccountInfo entries in `ka` */
  194. const uint8_t *input; /** pointer to the instruction data */
  195. uint64_t input_len; /** Length in bytes of the instruction data */
  196. SolPubkey *program_id; /** program_id of the currently executing program */
  197. const SolAccountInfo *ka_clock;
  198. const SolAccountInfo *ka_instructions;
  199. } SolParameters;
  200. /**
  201. * Maximum number of bytes a program may add to an account during a single realloc
  202. */
  203. #define MAX_PERMITTED_DATA_INCREASE (1024 * 10)
  204. /**
  205. * De-serializes the input parameters into usable types
  206. *
  207. * Use this function to deserialize the buffer passed to the program entrypoint
  208. * into usable types. This function does not perform copy deserialization,
  209. * instead it populates the pointers and lengths in SolAccountInfo and data so
  210. * that any modification to lamports or account data take place on the original
  211. * buffer. Doing so also eliminates the need to serialize back into the buffer
  212. * at the end of the program.
  213. *
  214. * @param input Source buffer containing serialized input parameters
  215. * @param params Pointer to a SolParameters structure
  216. * @return Boolean true if successful.
  217. */
  218. static uint64_t sol_deserialize(const uint8_t *input, SolParameters *params)
  219. {
  220. if (NULL == input || NULL == params)
  221. {
  222. return ERROR_INVALID_ARGUMENT;
  223. }
  224. uint64_t max_accounts = SOL_ARRAY_SIZE(params->ka);
  225. params->ka_num = *(uint64_t *)input;
  226. input += sizeof(uint64_t);
  227. for (int i = 0; i < params->ka_num; i++)
  228. {
  229. uint8_t dup_info = input[0];
  230. input += sizeof(uint8_t);
  231. if (i >= max_accounts)
  232. {
  233. if (dup_info == UINT8_MAX)
  234. {
  235. input += sizeof(uint8_t);
  236. input += sizeof(uint8_t);
  237. input += sizeof(uint8_t);
  238. input += 4; // padding
  239. input += sizeof(SolPubkey);
  240. input += sizeof(SolPubkey);
  241. input += sizeof(uint64_t);
  242. uint64_t data_len = *(uint64_t *)input;
  243. input += sizeof(uint64_t);
  244. input += data_len;
  245. input += MAX_PERMITTED_DATA_INCREASE;
  246. input = (uint8_t *)(((uint64_t)input + 8 - 1) & ~(8 - 1)); // padding
  247. input += sizeof(uint64_t);
  248. }
  249. else
  250. {
  251. input += 7; // padding for the 64-bit alignment
  252. }
  253. continue;
  254. }
  255. if (dup_info == UINT8_MAX)
  256. {
  257. // is signer?
  258. params->ka[i].is_signer = *(uint8_t *)input != 0;
  259. input += sizeof(uint8_t);
  260. // is writable?
  261. params->ka[i].is_writable = *(uint8_t *)input != 0;
  262. input += sizeof(uint8_t);
  263. // executable?
  264. params->ka[i].executable = *(uint8_t *)input;
  265. input += sizeof(uint8_t);
  266. input += 4; // padding
  267. // key
  268. params->ka[i].key = (SolPubkey *)input;
  269. input += sizeof(SolPubkey);
  270. // owner
  271. params->ka[i].owner = (SolPubkey *)input;
  272. input += sizeof(SolPubkey);
  273. // lamports
  274. params->ka[i].lamports = (uint64_t *)input;
  275. input += sizeof(uint64_t);
  276. // account data
  277. params->ka[i].data_len = *(uint64_t *)input;
  278. input += sizeof(uint64_t);
  279. params->ka[i].data = (uint8_t *)input;
  280. input += params->ka[i].data_len;
  281. input += MAX_PERMITTED_DATA_INCREASE;
  282. input = (uint8_t *)(((uint64_t)input + 8 - 1) & ~(8 - 1)); // padding
  283. // rent epoch
  284. params->ka[i].rent_epoch = *(uint64_t *)input;
  285. input += sizeof(uint64_t);
  286. }
  287. else
  288. {
  289. params->ka[i].is_signer = params->ka[dup_info].is_signer;
  290. params->ka[i].is_writable = params->ka[dup_info].is_writable;
  291. params->ka[i].executable = params->ka[dup_info].executable;
  292. params->ka[i].key = params->ka[dup_info].key;
  293. params->ka[i].owner = params->ka[dup_info].owner;
  294. params->ka[i].lamports = params->ka[dup_info].lamports;
  295. params->ka[i].data_len = params->ka[dup_info].data_len;
  296. params->ka[i].data = params->ka[dup_info].data;
  297. params->ka[i].rent_epoch = params->ka[dup_info].rent_epoch;
  298. input += 7; // padding
  299. }
  300. }
  301. uint64_t data_len = *(uint64_t *)input;
  302. input += sizeof(uint64_t);
  303. params->input_len = data_len;
  304. params->input = input;
  305. input += data_len;
  306. params->program_id = (SolPubkey *)input;
  307. input += sizeof(SolPubkey);
  308. if (params->ka_num > max_accounts)
  309. params->ka_num = max_accounts;
  310. return 0;
  311. }
  312. /**
  313. * Byte array pointer and string
  314. */
  315. typedef struct
  316. {
  317. const uint8_t *addr; /** bytes */
  318. uint64_t len; /** number of bytes*/
  319. } SolBytes;
  320. /**
  321. * Length of a sha256 hash result
  322. */
  323. #define SHA256_RESULT_LENGTH 32
  324. /**
  325. * Sha256
  326. *
  327. * @param bytes Array of byte arrays
  328. * @param bytes_len Number of byte arrays
  329. * @param result 32 byte array to hold the result
  330. */
  331. static uint64_t sol_sha256(const SolBytes *bytes, int bytes_len, const uint8_t *result);
  332. /**
  333. * Account Meta
  334. */
  335. typedef struct
  336. {
  337. SolPubkey *pubkey; /** An account's public key */
  338. bool is_writable; /** True if the `pubkey` can be loaded as a read-write account */
  339. bool is_signer; /** True if an Instruction requires a Transaction signature matching `pubkey` */
  340. } SolAccountMeta;
  341. /**
  342. * Instruction
  343. */
  344. typedef struct
  345. {
  346. SolPubkey *program_id; /** Pubkey of the instruction processor that executes this instruction */
  347. SolAccountMeta *accounts; /** Metadata for what accounts should be passed to the instruction processor */
  348. uint64_t account_len; /** Number of SolAccountMetas */
  349. uint8_t *data; /** Opaque data passed to the instruction processor */
  350. uint64_t data_len; /** Length of the data in bytes */
  351. } SolInstruction;
  352. /**
  353. * Seeds used by a signer to create a program address or passed to
  354. * sol_invoke_signed
  355. */
  356. typedef struct
  357. {
  358. const SolSignerSeed *addr; /** An array of a signer's seeds */
  359. uint64_t len; /** Number of seeds */
  360. } SolSignerSeeds;
  361. /**
  362. * Create a program address
  363. *
  364. * @param seeds Seed bytes used to sign program accounts
  365. * @param seeds_len Length of the seeds array
  366. * @param Progam id of the signer
  367. * @param Program address created, filled on return
  368. */
  369. static uint64_t sol_create_program_address(const SolSignerSeed *seeds, int seeds_len, const SolPubkey *program_id,
  370. const SolPubkey *address);
  371. /**
  372. * Cross-program invocation
  373. * * @{
  374. */
  375. /**
  376. * Invoke another program and sign for some of the keys
  377. *
  378. * @param instruction Instruction to process
  379. * @param account_infos Accounts used by instruction
  380. * @param account_infos_len Length of account_infos array
  381. * @param seeds Seed bytes used to sign program accounts
  382. * @param seeds_len Length of the seeds array
  383. */
  384. static uint64_t sol_invoke_signed(const SolInstruction *instruction, const SolAccountInfo *account_infos,
  385. int account_infos_len, const SolSignerSeeds *signers_seeds, int signers_seeds_len)
  386. {
  387. uint64_t sol_invoke_signed_c(const SolInstruction *instruction, const SolAccountInfo *account_infos,
  388. int account_infos_len, const SolSignerSeeds *signers_seeds, int signers_seeds_len);
  389. return sol_invoke_signed_c(instruction, account_infos, account_infos_len, signers_seeds, signers_seeds_len);
  390. }
  391. /**
  392. * Invoke another program
  393. *
  394. * @param instruction Instruction to process
  395. * @param account_infos Accounts used by instruction
  396. * @param account_infos_len Length of account_infos array
  397. */
  398. static uint64_t sol_invoke(const SolInstruction *instruction, const SolAccountInfo *account_infos,
  399. int account_infos_len)
  400. {
  401. const SolSignerSeeds signers_seeds[] = {{}};
  402. return sol_invoke_signed(instruction, account_infos, account_infos_len, signers_seeds, 0);
  403. }
  404. /**@}*/
  405. /**
  406. * Debugging utilities
  407. * @{
  408. */
  409. /**
  410. * Prints the hexadecimal representation of a public key
  411. *
  412. * @param key The public key to print
  413. */
  414. void sol_log_pubkey(const SolPubkey *pubkey);
  415. /**
  416. * Prints the hexadecimal representation of an array
  417. *
  418. * @param array The array to print
  419. */
  420. static void sol_log_array(const uint8_t *array, int len)
  421. {
  422. for (int j = 0; j < len; j++)
  423. {
  424. sol_log_64(0, 0, 0, j, array[j]);
  425. }
  426. }
  427. /**
  428. * Prints the program's input parameters
  429. *
  430. * @param params Pointer to a SolParameters structure
  431. */
  432. static void sol_log_params(const SolParameters *params)
  433. {
  434. sol_log("- Program identifier:");
  435. sol_log_pubkey(params->program_id);
  436. sol_log("- Number of KeyedAccounts");
  437. sol_log_64(0, 0, 0, 0, params->ka_num);
  438. for (int i = 0; i < params->ka_num; i++)
  439. {
  440. sol_log(" - Is signer");
  441. sol_log_64(0, 0, 0, 0, params->ka[i].is_signer);
  442. sol_log(" - Is writable");
  443. sol_log_64(0, 0, 0, 0, params->ka[i].is_writable);
  444. sol_log(" - Key");
  445. sol_log_pubkey(params->ka[i].key);
  446. sol_log(" - Lamports");
  447. sol_log_64(0, 0, 0, 0, *params->ka[i].lamports);
  448. sol_log(" - data");
  449. sol_log_array(params->ka[i].data, params->ka[i].data_len);
  450. sol_log(" - Owner");
  451. sol_log_pubkey(params->ka[i].owner);
  452. sol_log(" - Executable");
  453. sol_log_64(0, 0, 0, 0, params->ka[i].executable);
  454. sol_log(" - Rent Epoch");
  455. sol_log_64(0, 0, 0, 0, params->ka[i].rent_epoch);
  456. }
  457. sol_log("- Eth abi Instruction data\0");
  458. sol_log_array(params->input, params->input_len);
  459. }
  460. /**@}*/
  461. /**
  462. * Program instruction entrypoint
  463. *
  464. * @param input Buffer of serialized input parameters. Use sol_deserialize() to decode
  465. * @return 0 if the instruction executed successfully
  466. */
  467. uint64_t entrypoint(const uint8_t *input);
  468. #ifdef SOL_TEST
  469. /**
  470. * Stub log functions when building tests
  471. */
  472. #include <stdio.h>
  473. void sol_log_(const char *s, uint64_t len)
  474. {
  475. printf("sol_log: %s\n", s);
  476. }
  477. void sol_log_64(uint64_t arg1, uint64_t arg2, uint64_t arg3, uint64_t arg4, uint64_t arg5)
  478. {
  479. printf("sol_log_64: %llu, %llu, %llu, %llu, %llu\n", arg1, arg2, arg3, arg4, arg5);
  480. }
  481. #endif
  482. #ifdef __cplusplus
  483. }
  484. #endif
  485. /**@}*/