blueprint-tsa 1.0.2 → 1.0.4

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "blueprint-tsa",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "TSA plugin for TON Blueprint",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "files": [
     "dist",
-    "checkers"
+    "src/checkers"
   ],
   "license": "MIT",
   "scripts": {

package/src/checkers/authorization-check.fc

@@ -0,0 +1,37 @@
+#include "imports/stdlib.fc";
+#include "imports/tsa_functions.fc";
+
+global int opcode;
+global slice initial_sender;
+
+() on_internal_message_send(int balance, int msg_value, cell in_msg_full, slice msg_body, int input_id) impure method_id {
+    slice cs = in_msg_full.begin_parse();
+    cs~load_uint(4); ;; flags
+    initial_sender = cs~load_msg_addr(); ;; src
+    slice contract_address = cs~load_msg_addr(); ;; dst
+    cs~load_coins(); ;; value
+
+    int cur_opcode = msg_body~load_uint(32);
+    tsa_assert(cur_opcode == opcode);
+
+    cell c4 = tsa_get_c4(1);
+    tsa_make_slice_independent_from_random_addresses(c4.begin_parse());
+    tsa_make_slice_independent_from_random_addresses(contract_address);
+}
+
+() main() impure {
+    tsa_forbid_failures();
+
+    slice ds = get_data().begin_parse();
+    opcode = ds~load_uint(32);
+
+    tsa_send_internal_message(1, 0);
+
+    tsa_allow_failures();
+
+    int random_sender = tsa_mk_int(1, -1);
+    throw_if(1001, ~ random_sender);
+
+    tsa_make_address_random(initial_sender);
+    throw(1000);
+}

package/src/checkers/bounce-check-scheme.json

@@ -0,0 +1,20 @@
+[
+  {
+    "id": 1,
+    "inOpcodeToDestination": {},
+    "other": {
+      "type": "out_opcodes",
+      "outOpcodeToDestination": {},
+      "other": [2]
+    }
+  },
+  {
+    "id": 2,
+    "inOpcodeToDestination": {},
+    "other": {
+      "type": "out_opcodes",
+      "outOpcodeToDestination": {},
+      "other": [1]
+    }
+  }
+]

package/src/checkers/bounce-check.fc

@@ -0,0 +1,35 @@
+#include "imports/stdlib.fc";
+#include "imports/tsa_functions.fc";
+
+const int error::failed_to_handle_bounce = 1000;
+const int error::sent_messages_on_bounce = 1000;
+
+global int counter;
+
+() on_compute_phase_exit(cell c5, int exit_code, int compute_fee, int contract_id) impure method_id {
+    if (contract_id != 1) {
+        return ();
+    }
+    ;; on the first entrance, we are ending the execution triggered by checker
+    ;; on the consecutive entrance, we are ending the execution triggered by bounced message
+    if (counter > 0) {
+        throw_if(error::failed_to_handle_bounce, exit_code > 0);
+        int sent_messages_on_bounce = c5.begin_parse().slice_refs() > 0;
+        throw_if(error::sent_messages_on_bounce, sent_messages_on_bounce);
+    }
+    counter += 1;
+}
+
+() on_internal_message_send(int balance, int msg_value, cell in_msg_full, slice msg_body, int input_id) impure method_id {
+    counter = 0;
+    tsa_forbid_failures();
+    slice cs = in_msg_full.begin_parse();
+    int flags = cs~load_uint(4);
+    tsa_assert_not(flags & 1);
+
+    tsa_allow_failures();
+}
+
+() main() impure {
+    tsa_send_internal_message(1, 0);
+}

package/src/checkers/chameleon-contract.fc

@@ -0,0 +1,8 @@
+#include "imports/stdlib.fc";
+
+() recv_internal(int msg_value, cell in_msg_full, slice in_msg) impure {
+    var data = in_msg~load_ref();
+    var code = in_msg~load_ref();
+    set_data(data);
+    set_code(code);
+}

package/src/checkers/drain-check-concrete.fc

@@ -0,0 +1,50 @@
+#include "imports/stdlib.fc";
+#include "imports/tsa_functions.fc";
+
+global slice initial_sender;
+global int initial_value;
+global int received_value;
+global int max_value;
+
+() on_internal_message_send(int balance, int msg_value, cell in_msg_full, slice msg_body, int input_id) impure method_id {
+    slice cs = in_msg_full.begin_parse();
+    cs~load_uint(4); ;; flags
+    slice sender = cs~load_msg_addr(); ;; src
+    cs~load_msg_addr(); ;; dst
+    initial_value = cs~load_coins(); ;; value
+
+    tsa_assert(equal_slice_bits(sender, initial_sender));
+    tsa_assert(initial_value <= max_value);
+}
+
+() on_out_message(cell msg_full, slice msg_body, int receiver_contract_id, int sender_contract_id) impure method_id {
+    if ((sender_contract_id != 1) | (receiver_contract_id != -1)) {
+        return ();
+    }
+
+    slice cs = msg_full.begin_parse();
+    cs~skip_bits(4); ;; message headers
+    cs~load_msg_addr(); ;; src address
+    slice dest = cs~load_msg_addr();
+    int value = cs~load_grams();
+
+    if (equal_slice_bits(dest, initial_sender)) {
+        received_value += value;
+    }
+}
+
+() main() impure {
+    received_value = 0;
+
+    tsa_forbid_failures();
+
+    slice ds = get_data().begin_parse();
+    initial_sender = ds~load_msg_addr();
+    max_value = ds~load_coins();
+
+    tsa_send_internal_message(1, 0);
+
+    tsa_allow_failures();
+
+    throw_if(1000, received_value > initial_value);
+}

package/src/checkers/drain-check-symbolic.fc

@@ -0,0 +1,48 @@
+#include "imports/stdlib.fc";
+#include "imports/tsa_functions.fc";
+
+global slice initial_sender;
+global slice contract_address;
+global int initial_value;
+global int received_value;
+
+() on_internal_message_send(int balance, int msg_value, cell in_msg_full, slice msg_body, int input_id) impure method_id {
+    slice cs = in_msg_full.begin_parse();
+    cs~load_uint(4); ;; flags
+    initial_sender = cs~load_msg_addr(); ;; src
+    contract_address = cs~load_msg_addr(); ;; dst
+    initial_value = cs~load_coins(); ;; value
+
+    cell c4 = tsa_get_c4(1);
+    tsa_make_address_random(initial_sender);
+    tsa_make_slice_independent_from_random_addresses(c4.begin_parse());
+    tsa_make_slice_independent_from_random_addresses(contract_address);
+}
+
+() on_out_message(cell msg_full, slice msg_body, int receiver_contract_id, int sender_contract_id) impure method_id {
+    if ((sender_contract_id != 1) | (receiver_contract_id != -1)) {
+        return ();
+    }
+
+    slice cs = msg_full.begin_parse();
+    cs~skip_bits(4); ;; message headers
+    cs~load_msg_addr(); ;; src address
+    slice dest = cs~load_msg_addr();
+    int value = cs~load_grams();
+
+    if (equal_slice_bits(dest, initial_sender)) {
+        received_value += value;
+    }
+}
+
+() main() impure {
+    received_value = 0;
+
+    tsa_forbid_failures();
+
+    tsa_send_internal_message(1, 0);
+
+    tsa_allow_failures();
+
+    throw_if(1000, received_value > initial_value);
+}