wasm_interface.cpp 37.6 KB
Newer Older
B
Bart Wyatt 已提交
1 2
#include <eosio/chain/wasm_interface.hpp>
#include <eosio/chain/apply_context.hpp>
3
#include <eosio/chain/chain_controller.hpp>
4
#include <eosio/chain/exceptions.hpp>
D
Daniel Larimer 已提交
5
#include <boost/core/ignore_unused.hpp>
6 7
#include <eosio/chain/wasm_interface_private.hpp>
#include <fc/exception/exception.hpp>
D
Daniel Larimer 已提交
8 9 10

#include <Runtime/Runtime.h>
#include "IR/Module.h"
11 12 13 14
#include "Platform/Platform.h"
#include "WAST/WAST.h"
#include "IR/Operators.h"
#include "IR/Validate.h"
15
#include "IR/Types.h"
16 17 18
#include "Runtime/Runtime.h"
#include "Runtime/Linker.h"
#include "Runtime/Intrinsics.h"
D
Daniel Larimer 已提交
19

20 21 22
#include <boost/asio.hpp>
#include <boost/bind.hpp>

23
#include <mutex>
24 25 26
#include <thread>
#include <condition_variable>

27

28

29 30
using namespace IR;
using namespace Runtime;
31
using boost::asio::io_service;
D
Daniel Larimer 已提交
32

33
#if 0
B
Brian Johnson 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
   // account.h/hpp expected account API balance interchange format
   // must match account.hpp account_balance definition
   PACKED_STRUCT(
   struct account_balance
   {
      /**
      * Name of the account who's balance this is
      */
      account_name account;

      /**
      * Balance for this account
      */
      asset eos_balance;

      /**
      * Staked balance for this account
      */
      asset staked_balance;

      /**
      * Unstaking balance for this account
      */
      asset unstaking_balance;

      /**
      * Time at which last unstaking occurred for this account
      */
      time last_unstaking_time;
   })
64
#endif
B
Brian Johnson 已提交
65

66
namespace eosio { namespace chain {
67
   using namespace contracts;
68

69 70 71
   /**
    * Integration with the WASM Linker to resolve our intrinsics
    */
D
Daniel Larimer 已提交
72
   struct root_resolver : Runtime::Resolver
73
   {
74 75
      bool resolve(const string& mod_name,
                   const string& export_name,
D
Daniel Larimer 已提交
76 77
                   ObjectType type,
                   ObjectInstance*& out) override
78
      {
79 80 81
         // Try to resolve an intrinsic first.
         if(IntrinsicResolver::singleton.resolve(mod_name,export_name,type, out)) {
            return true;
82
         }
83 84

         FC_ASSERT( !"unresolvable", "${module}.${export}", ("module",mod_name)("export",export_name) );
85
         return false;
86
      }
D
Daniel Larimer 已提交
87
   };
M
Matias Romeo 已提交
88

89 90 91 92 93 94 95 96 97 98
   /**
    *  Implementation class for the wasm cache
    *  it is responsible for compiling and storing instances of wasm code for use
    *
    */
   struct wasm_cache_impl {
      wasm_cache_impl()
      :_ios()
      ,_work(_ios)
      {
99 100
         Runtime::init();

101 102 103 104
         _utility_thread = std::thread([](io_service* ios){
            ios->run();
         }, &_ios);
      }
M
Matias Romeo 已提交
105

106 107 108 109 110 111 112 113 114 115
      /**
       * this must wait for all work to be done otherwise it may destroy memory
       * referenced by other threads
       *
       * Expectations on wasm_cache dictate that all available code has been
       * returned before this can be destroyed
       */
      ~wasm_cache_impl() {
         _work.reset();
         _ios.stop();
116 117
         _utility_thread.join();
         freeUnreferencedObjects({});
118
      }
M
Matias Romeo 已提交
119

120 121 122 123 124 125 126 127 128 129 130 131 132
      /**
       * internal tracking structure which deduplicates memory images
       * and tracks available vs in-use entries.
       *
       * The instances array has two sections, "available" instances
       * are in the front of the vector and anything at an index of
       * available_instances or greater is considered "in use"
       *
       * instances are stored as pointers so that their positions
       * in the array can be moved without invaliding references to
       * the instance handed out to other threads
       */
      struct code_info {
133 134 135 136
         code_info( size_t mem_end, vector<char>&& mem_image )
         :mem_end(mem_end),mem_image(std::forward<vector<char>>(mem_image))
         {}

137 138 139 140 141 142 143 144 145 146
         // a clean image of the memory used to sanitize things on checkin
         size_t mem_start           = 0;
         size_t mem_end             = 1<<16;
         vector<char> mem_image;

         // all existing instances of this code
         vector<unique_ptr<wasm_cache::entry>> instances;
         size_t available_instances = 0;
      };

147 148 149
      using optional_info_ref = optional<std::reference_wrapper<code_info>>;
      using optional_entry_ref = optional<std::reference_wrapper<wasm_cache::entry>>;

150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168
      /**
       * Convenience method for running code with the _cache_lock and releaseint that lock
       * when the code completes
       *
       * @param f - lambda to execute
       * @return - varies depending on the signature of the lambda
       */
      template<typename F>
      auto with_lock(F f) {
         std::lock_guard<std::mutex> lock(_cache_lock);
         return f();
      };

      /**
       * Fetch the tracking struct given a code_id if it exists
       *
       * @param code_id
       * @return
       */
169 170
      optional_info_ref fetch_info(const digest_type& code_id) {
         return with_lock([&,this](){
171 172
            auto iter = _cache.find(code_id);
            if (iter != _cache.end()) {
173
               return optional_info_ref(iter->second);
174 175
            }

176
            return optional_info_ref();
177 178
         });
      }
M
Matias Romeo 已提交
179

180 181 182 183 184
      /**
       * Opportunistically fetch an available instance of the code;
       * @param code_id - the id of the code to fetch
       * @return - reference to the entry when one is available
       */
185 186
      optional_entry_ref try_fetch_entry(const digest_type& code_id) {
         return with_lock([&,this](){
187 188
            auto iter = _cache.find(code_id);
            if (iter != _cache.end() && iter->second.available_instances > 0) {
189
               auto &ptr = iter->second.instances.at(--(iter->second.available_instances));
190
               return optional_entry_ref(*ptr);
191 192
            }

193
            return optional_entry_ref();
194
         });
D
Daniel Larimer 已提交
195
      }
M
Matias Romeo 已提交
196

197 198 199 200 201 202 203 204 205 206 207 208
      /**
       * Fetch a copy of the code, this is guaranteed to return an entry IF the code is compilable.
       * In order to do that in safe way this code may cause the calling thread to sleep while a new
       * version of the code is compiled and inserted into the cache
       *
       * @param code_id - the id of the code to fetch
       * @param wasm_binary - the binary for the wasm
       * @param wasm_binary_size - the size of the binary
       * @return reference to a usable cache entry
       */
      wasm_cache::entry& fetch_entry(const digest_type& code_id, const char* wasm_binary, size_t wasm_binary_size) {
         std::condition_variable condition;
209
         optional_entry_ref result;
210 211 212 213
         std::exception_ptr error;

         // compilation is not thread safe, so we dispatch it to a io_service running on a single thread to
         // queue up and synchronize compilations
214
         _ios.post([&,this](){
215 216 217 218 219 220
            // check to see if someone returned what we need before making a new one
            auto pending_result = try_fetch_entry(code_id);
            std::exception_ptr pending_error;

            if (!pending_result) {
               // time to compile a brand new (maybe first) copy of this code
221 222
               Module* module = new Module();
               ModuleInstance* instance = nullptr;
223
               size_t mem_end = 0;
224 225 226 227
               vector<char> mem_image;

               try {
                  Serialization::MemoryInputStream stream((const U8 *) wasm_binary, wasm_binary_size);
228 229
                  #warning TODO: restore checktime injection?
                  WASM::serialize(stream, *module);
230 231

                  root_resolver resolver;
232 233
                  LinkResult link_result = linkModule(*module, resolver);
                  instance = instantiateModule(*module, std::move(link_result.resolvedImports));
234 235
                  FC_ASSERT(instance != nullptr);

236
                  MemoryInstance* current_memory = Runtime::getDefaultMemory(instance);
237

238 239 240 241 242 243
                  if(current_memory) {
                     char *mem_ptr = &memoryRef<char>(current_memory, 0);
                     const auto allocated_memory = Runtime::getDefaultMemorySize(instance);
                     for (uint64_t i = 0; i < allocated_memory; ++i) {
                        if (mem_ptr[i])
                           mem_end = i + 1;
244
                     }
245 246
                     mem_image.resize(mem_end);
                     memcpy(mem_image.data(), mem_ptr, mem_end);
247
                  }
248
                  
249 250 251 252 253 254
               } catch (...) {
                  pending_error = std::current_exception();
               }

               if (pending_error == nullptr) {
                  // grab the lock and put this in the cache as unavailble
255
                  with_lock([&,this]() {
256
                     // find or create a new entry
257
                     auto iter = _cache.emplace(code_id, code_info(mem_end, std::move(mem_image))).first;
258

259 260
                     iter->second.instances.emplace_back(std::make_unique<wasm_cache::entry>(instance, module));
                     pending_result = optional_entry_ref(*iter->second.instances.back().get());
261 262 263 264 265 266 267 268 269 270 271 272
                  });
               }
            }

            // publish result under lock
            with_lock([&](){
               if (pending_error != nullptr) {
                  error = pending_error;
               } else {
                  result = pending_result;
               }
            });
273

274 275 276 277 278 279 280 281 282 283
            condition.notify_all();
         });

         // wait for the other thread to compile a copy for us
         {
            std::unique_lock<std::mutex> lock(_cache_lock);
            condition.wait(lock, [&]{
               return error != nullptr || result.valid();
            });
         }
284

285 286 287 288
         try {
            if (error != nullptr) {
               std::rethrow_exception(error);
            } else {
289
               return (*result).get();
290 291
            }
         } FC_RETHROW_EXCEPTIONS(error, "error compiling WASM for code with hash: ${code_id}", ("code_id", code_id));
D
Daniel Larimer 已提交
292
      }
293

294 295 296 297 298 299 300 301 302
      /**
       * return an entry to the cache.  The entry is presumed to come back in a "dirty" state and must be
       * sanitized before returning to the "available" state.  This sanitization is done asynchronously so
       * as not to delay the current executing thread.
       *
       * @param code_id - the code Id associated with the instance
       * @param entry - the entry to return
       */
      void return_entry(const digest_type& code_id, wasm_cache::entry& entry) {
303
         _ios.post([&,code_id,this](){
304
            // sanitize by reseting the memory that may now be dirty
305
            auto& info = (*fetch_info(code_id)).get();
306 307 308 309 310
            if(getDefaultMemory(entry.instance)) {
               char* memstart = &memoryRef<char>( getDefaultMemory(entry.instance), 0 );
               memset( memstart + info.mem_end, 0, ((1<<16) - info.mem_end) );
               memcpy( memstart, info.mem_image.data(), info.mem_end);
            }
311
            resetGlobalInstances(entry.instance);
312 313

            // under a lock, put this entry back in the available instances side of the instances vector
314
            with_lock([&,this](){
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
               // walk the vector and find this entry
               auto iter = info.instances.begin();
               while (iter->get() != &entry) {
                  ++iter;
               }

               FC_ASSERT(iter != info.instances.end(), "Checking in a WASM enty that was not created properly!");

               auto first_unavailable = (info.instances.begin() + info.available_instances);
               if (iter != first_unavailable) {
                  std::swap(iter, first_unavailable);
               }
               info.available_instances++;
            });
         });
D
Daniel Larimer 已提交
330
      }
331

332 333 334
      // mapping of digest to an entry for the code
      map<digest_type, code_info> _cache;
      std::mutex _cache_lock;
B
Brian Johnson 已提交
335

336 337 338 339 340
      // compilation and cleanup thread
      std::thread _utility_thread;
      io_service _ios;
      optional<io_service::work> _work;
   };
B
Brian Johnson 已提交
341

342 343 344
   wasm_cache::wasm_cache()
      :_my( new wasm_cache_impl() ) {
   }
345

346 347
   wasm_cache::~wasm_cache() = default;

348
   wasm_cache::entry &wasm_cache::checkout( const digest_type& code_id, const char* wasm_binary, size_t wasm_binary_size ) {
349
      // see if there is an available entry in the cache
350
      auto result = _my->try_fetch_entry(code_id);
351
      if (result) {
352
         return (*result).get();
D
Daniel Larimer 已提交
353
      }
354 355 356 357 358
      return _my->fetch_entry(code_id, wasm_binary, wasm_binary_size);
   }


   void wasm_cache::checkin(const digest_type& code_id, entry& code ) {
359 360 361
      MemoryInstance* default_mem = Runtime::getDefaultMemory(code.instance);
      if(default_mem)
         Runtime::shrinkMemory(default_mem, Runtime::getMemoryNumPages(default_mem) - 1);
362
      _my->return_entry(code_id, code);
363
   }
364

365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382
   /**
    * RAII wrapper to make sure that the context is cleaned up on exception
    */
   struct scoped_context {
      template<typename ...Args>
      scoped_context(optional<wasm_context> &context, Args&... args)
      :context(context)
      {
         context = wasm_context{ args... };
      }

      ~scoped_context() {
         context.reset();
      }

      optional<wasm_context>& context;
   };

383
   void wasm_interface_impl::call(const string& entry_point, const vector<Value>& args, wasm_cache::entry& code, apply_context& context)
384
   try {
385
      FunctionInstance* call = asFunctionNullable(getInstanceExport(code.instance,entry_point) );
386 387 388 389
      if( !call ) {
         return;
      }

390
      FC_ASSERT( getFunctionType(call)->parameters.size() == args.size() );
391

392
      auto context_guard = scoped_context(current_context, code, context);
393
      runInstanceStartFunc(code.instance);
394 395 396 397 398 399 400 401 402 403 404 405
      Runtime::invokeFunction(call,args);
   } catch( const Runtime::Exception& e ) {
      FC_THROW_EXCEPTION(wasm_execution_error,
                         "cause: ${cause}\n${callstack}",
                         ("cause", string(describeExceptionCause(e.cause)))
                         ("callstack", e.callStack));
   } FC_CAPTURE_AND_RETHROW()

   wasm_interface::wasm_interface()
      :my( new wasm_interface_impl() ) {
   }

406 407 408 409 410 411 412 413
   wasm_interface& wasm_interface::get() {
      thread_local wasm_interface* single = nullptr;
      if( !single ) {
         single = new wasm_interface();
      }
      return *single;
   }

414
   void wasm_interface::apply( wasm_cache::entry& code, apply_context& context ) {
B
Bart Wyatt 已提交
415
      if (context.act.account == config::system_account_name && context.act.name == N(setcode)) {
416 417
         my->call("init", {}, code, context);
      } else {
B
Bart Wyatt 已提交
418
         vector<Value> args = {Value(uint64_t(context.act.account)),
419 420 421
                               Value(uint64_t(context.act.name))};
         my->call("apply", args, code, context);
      }
422 423 424 425 426
   }

   void wasm_interface::error( wasm_cache::entry& code, apply_context& context ) {
      vector<Value> args = { /* */ };
      my->call("error", args, code, context);
427 428
   }

429 430
#if 0
  DEFINE_INTRINSIC_FUNCTION2(env,assert,assert,none,i32,test,i32,msg) {
D
Daniel Larimer 已提交
431 432 433 434 435 436 437
      elog( "assert" );
      /*
      const char* m = &Runtime::memoryRef<char>( wasm_interface::get().current_memory, msg );
     std::string message( m );
     if( !test ) edump((message));
     FC_ASSERT( test, "assertion failed: ${s}", ("s",message)("ptr",msg) );
     */
438
   }
439

D
Daniel Larimer 已提交
440 441 442 443 444
   DEFINE_INTRINSIC_FUNCTION1(env,printi,printi,none,i64,val) {
     std::cerr << uint64_t(val);
   }
   DEFINE_INTRINSIC_FUNCTION1(env,printd,printd,none,i64,val) {
     //std::cerr << DOUBLE(*reinterpret_cast<double *>(&val));
445
   }
446

D
Daniel Larimer 已提交
447
   DEFINE_INTRINSIC_FUNCTION1(env,printi128,printi128,none,i32,val) {
448 449 450 451 452 453 454
      /*
      auto& wasm  = wasm_interface::get();
      auto  mem   = wasm.memory();
      auto& value = memoryRef<unsigned __int128>( mem, val );
      fc::uint128_t v(value>>64, uint64_t(value) );
      std::cerr << fc::variant(v).get_string();
      */
455

D
Daniel Larimer 已提交
456 457 458 459
   }
   DEFINE_INTRINSIC_FUNCTION1(env,printn,printn,none,i64,val) {
     std::cerr << name(val).to_string();
   }
460

461 462 463 464




D
Daniel Larimer 已提交
465 466 467
   DEFINE_INTRINSIC_FUNCTION1(env,prints,prints,none,i32,charptr) {
     auto& wasm  = wasm_interface::get();
     auto  mem   = wasm.memory();
468

D
Daniel Larimer 已提交
469 470
     const char* str = &memoryRef<const char>( mem, charptr );
     const auto allocated_memory = wasm.memory_size(); //Runtime::getDefaultMemorySize(state.instance);
471

D
Daniel Larimer 已提交
472 473
     std::cerr << std::string( str, strnlen(str, allocated_memory-charptr) );
   }
474

D
Daniel Larimer 已提交
475
DEFINE_INTRINSIC_FUNCTION2(env,readMessage,readMessage,i32,i32,destptr,i32,destsize) {
476
   FC_ASSERT( destsize > 0 );
477

D
Daniel Larimer 已提交
478
   /*
479 480
   wasm_interface& wasm = wasm_interface::get();
   auto  mem   = wasm.current_memory;
481
   char* begin = memoryArrayPtr<char>( mem, destptr, uint32_t(destsize) );
482 483

   int minlen = std::min<int>(wasm.current_validate_context->msg.data.size(), destsize);
484

D
Daniel Larimer 已提交
485
//   wdump((destsize)(wasm.current_validate_context->msg.data.size()));
486
   memcpy( begin, wasm.current_validate_context->msg.data.data(), minlen );
D
Daniel Larimer 已提交
487 488
   */
   return 0;//minlen;
A
Andrianto Lie 已提交
489 490
}

491

492

493 494 495
DEFINE_INTRINSIC_FUNCTION1(env,printi128,printi128,none,i32,val) {
  auto& wasm  = wasm_interface::get();
  auto  mem   = wasm.current_memory;
496 497 498
  auto& value = memoryRef<unsigned __int128>( mem, val );
  fc::uint128_t v(value>>64, uint64_t(value) );
  std::cerr << fc::variant(v).get_string();
499 500
}
DEFINE_INTRINSIC_FUNCTION1(env,printn,printn,none,i64,val) {
501
  std::cerr << name(val).to_string();
502
}
503

504
DEFINE_INTRINSIC_FUNCTION1(env,prints,prints,none,i32,charptr) {
505 506 507
  auto& wasm  = wasm_interface::get();
  auto  mem   = wasm.current_memory;

508
  const char* str = &memoryRef<const char>( mem, charptr );
509

D
Daniel Larimer 已提交
510
  std::cerr << std::string( str, strnlen(str, wasm.current_state->mem_end-charptr) );
511 512
}

A
Andrianto Lie 已提交
513 514 515 516 517 518 519 520 521
DEFINE_INTRINSIC_FUNCTION2(env,prints_l,prints_l,none,i32,charptr,i32,len) {
  auto& wasm  = wasm_interface::get();
  auto  mem   = wasm.current_memory;

  const char* str = &memoryRef<const char>( mem, charptr );

  std::cerr << std::string( str, len );
}

M
Matias Romeo 已提交
522 523 524
DEFINE_INTRINSIC_FUNCTION2(env,printhex,printhex,none,i32,data,i32,datalen) {
  auto& wasm  = wasm_interface::get();
  auto  mem   = wasm.current_memory;
A
Andrianto Lie 已提交
525

M
Matias Romeo 已提交
526
  char* buff = memoryArrayPtr<char>(mem, data, datalen);
527
  std::cerr << fc::to_hex(buff, datalen);
M
Matias Romeo 已提交
528 529 530
}


531
DEFINE_INTRINSIC_FUNCTION1(env,free,free,none,i32,ptr) {
532 533
}

D
Daniel Larimer 已提交
534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554
#define DEFINE_RECORD_READ_FUNCTIONS(OBJTYPE, FUNCPREFIX, INDEX, SCOPE) \
   DEFINE_INTRINSIC_FUNCTION5(env,load_##FUNCPREFIX##OBJTYPE,load_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(load_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,front_##FUNCPREFIX##OBJTYPE,front_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(front_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,back_##FUNCPREFIX##OBJTYPE,back_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(back_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,next_##FUNCPREFIX##OBJTYPE,next_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(next_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,previous_##FUNCPREFIX##OBJTYPE,previous_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(previous_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,lower_bound_##FUNCPREFIX##OBJTYPE,lower_bound_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(lower_bound_record, INDEX, SCOPE); \
   } \
   DEFINE_INTRINSIC_FUNCTION5(env,upper_bound_##FUNCPREFIX##OBJTYPE,upper_bound_##FUNCPREFIX##OBJTYPE,i32,i64,scope,i64,code,i64,table,i32,valueptr,i32,valuelen) { \
      READ_RECORD(upper_bound_record, INDEX, SCOPE); \
555
   }
B
Brian Johnson 已提交
556 557 558 559 560 561 562 563 564
DEFINE_INTRINSIC_FUNCTION2(env,account_balance_get,account_balance_get,i32,i32,charptr,i32,len) {
  auto& wasm  = wasm_interface::get();
  auto  mem   = wasm.current_memory;

  const uint32_t account_balance_size = sizeof(account_balance);
  FC_ASSERT( len == account_balance_size, "passed in len ${len} is not equal to the size of an account_balance struct == ${real_len}", ("len",len)("real_len",account_balance_size) );

  account_balance& total_balance = memoryRef<account_balance>( mem, charptr );

565 566
  wasm.current_apply_context->require_scope(total_balance.account);

B
Brian Johnson 已提交
567 568 569 570 571 572 573 574 575 576 577 578 579 580
  auto& db = wasm.current_apply_context->db;
  auto* balance        = db.find< balance_object,by_owner_name >( total_balance.account );
  auto* staked_balance = db.find<staked_balance_object,by_owner_name>( total_balance.account );

  if (balance == nullptr || staked_balance == nullptr)
     return false;

  total_balance.eos_balance          = asset(balance->balance, EOS_SYMBOL);
  total_balance.staked_balance       = asset(staked_balance->staked_balance);
  total_balance.unstaking_balance    = asset(staked_balance->unstaking_balance);
  total_balance.last_unstaking_time  = staked_balance->last_unstaking_time;

  return true;
}
581

D
Daniel Larimer 已提交
582 583
#define UPDATE_RECORD(UPDATEFUNC, INDEX, DATASIZE) \
   return 0;
584

D
Daniel Larimer 已提交
585 586 587 588 589 590
   /*
   auto lambda = [&](apply_context* ctx, INDEX::value_type::key_type* keys, char *data, uint32_t datalen) -> int32_t { \
      return ctx->UPDATEFUNC<INDEX::value_type>( Name(scope), Name(ctx->code.value), Name(table), keys, data, datalen); \
   }; \
   return validate<decltype(lambda), INDEX::value_type::key_type, INDEX::value_type::number_of_keys>(valueptr, DATASIZE, lambda);
   */
591

D
Daniel Larimer 已提交
592 593 594 595 596 597 598 599 600
#define DEFINE_RECORD_UPDATE_FUNCTIONS(OBJTYPE, INDEX) \
   DEFINE_INTRINSIC_FUNCTION4(env,store_##OBJTYPE,store_##OBJTYPE,i32,i64,scope,i64,table,i32,valueptr,i32,valuelen) { \
      UPDATE_RECORD(store_record, INDEX, valuelen); \
   } \
   DEFINE_INTRINSIC_FUNCTION4(env,update_##OBJTYPE,update_##OBJTYPE,i32,i64,scope,i64,table,i32,valueptr,i32,valuelen) { \
      UPDATE_RECORD(update_record, INDEX, valuelen); \
   } \
   DEFINE_INTRINSIC_FUNCTION3(env,remove_##OBJTYPE,remove_##OBJTYPE,i32,i64,scope,i64,table,i32,valueptr) { \
      UPDATE_RECORD(remove_record, INDEX, sizeof(typename INDEX::value_type::key_type)*INDEX::value_type::number_of_keys); \
B
Brian Johnson 已提交
601 602
   }

D
Daniel Larimer 已提交
603 604
DEFINE_RECORD_READ_FUNCTIONS(i64,,key_value_index, by_scope_primary);
DEFINE_RECORD_UPDATE_FUNCTIONS(i64, key_value_index);
B
Brian Johnson 已提交
605

D
Daniel Larimer 已提交
606 607 608
DEFINE_INTRINSIC_FUNCTION1(env,requireAuth,requireAuth,none,i64,account) {
   //wasm_interface::get().current_validate_context->require_authorization( Name(account) );
}
B
Brian Johnson 已提交
609

D
Daniel Larimer 已提交
610 611 612 613 614 615 616 617 618
DEFINE_INTRINSIC_FUNCTION1(env,requireNotice,requireNotice,none,i64,account) {
   //wasm_interface::get().current_validate_context->require_authorization( Name(account) );
}
DEFINE_INTRINSIC_FUNCTION0(env,checktime,checktime,none) {
   /*
   auto dur = wasm_interface::get().current_execution_time();
   if (dur > CHECKTIME_LIMIT) {
      wlog("checktime called ${d}", ("d", dur));
      throw checktime_exceeded();
B
Brian Johnson 已提交
619
   }
D
Daniel Larimer 已提交
620 621
   */
}
622 623 624
#endif

#if defined(assert)
625
   #undef assert
626 627
#endif

B
Bart Wyatt 已提交
628
class context_aware_api {
629
   public:
B
Bart Wyatt 已提交
630
      context_aware_api(wasm_interface& wasm)
631 632
      :context(intrinsics_accessor::get_context(wasm).context), code(intrinsics_accessor::get_context(wasm).code),
       sbrk_bytes(intrinsics_accessor::get_context(wasm).sbrk_bytes)
633 634
      {}

B
Bart Wyatt 已提交
635
   protected:
636
      uint32_t&          sbrk_bytes;
B
Bucky Kittinger 已提交
637
      wasm_cache::entry& code;
B
Bucky Kittinger 已提交
638
      apply_context&     context;
B
Bart Wyatt 已提交
639 640 641 642 643 644 645 646 647 648 649
};

class system_api : public context_aware_api {
   public:
      using context_aware_api::context_aware_api;

      void assert(bool condition, const char* str) {
         std::string message( str );
         if( !condition ) edump((message));
         FC_ASSERT( condition, "assertion failed: ${s}", ("s",message));
      }
650 651 652 653

      fc::time_point_sec now() {
         return context.controller.head_block_time();
      }
B
Bart Wyatt 已提交
654 655 656 657 658 659
};

class action_api : public context_aware_api {
   public:
      using context_aware_api::context_aware_api;

660 661
      int read_action(array_ptr<char> memory, size_t size) {
         FC_ASSERT(size > 0);
662
         int minlen = std::min<size_t>(context.act.data.size(), size);
663 664 665 666
         memcpy((void *)memory, context.act.data.data(), minlen);
         return minlen;
      }

B
Bart Wyatt 已提交
667 668
      int action_size() {
         return context.act.data.size();
669 670
      }

B
Bart Wyatt 已提交
671 672 673
      const name& current_receiver() {
         return context.receiver;
      }
674 675

      fc::time_point_sec publication_time() {
B
Bart Wyatt 已提交
676
         return context.trx_meta.published;
677 678 679
      }

      name current_sender() {
B
Bart Wyatt 已提交
680 681
         if (context.trx_meta.sender) {
            return *context.trx_meta.sender;
682 683 684 685
         } else {
            return name();
         }
      }
686
};
B
Brian Johnson 已提交
687

B
Bart Wyatt 已提交
688 689 690 691 692
class console_api : public context_aware_api {
   public:
      using context_aware_api::context_aware_api;

      void prints(const char *str) {
693
         context.console_append(str);
B
Bart Wyatt 已提交
694 695 696
      }

      void prints_l(array_ptr<const char> str, size_t str_len ) {
697
         context.console_append(string(str, str_len));
B
Bart Wyatt 已提交
698 699 700
      }

      void printi(uint64_t val) {
701
         context.console_append(val);
B
Bart Wyatt 已提交
702 703 704 705
      }

      void printi128(const unsigned __int128& val) {
         fc::uint128_t v(val>>64, uint64_t(val) );
706
         context.console_append(fc::variant(v).get_string());
B
Bart Wyatt 已提交
707 708 709
      }

      void printd( wasm_double val ) {
710
         context.console_append(val.str());
B
Bart Wyatt 已提交
711 712 713
      }

      void printn(const name& value) {
714
         context.console_append(value.to_string());
B
Bart Wyatt 已提交
715 716 717
      }

      void printhex(array_ptr<const char> data, size_t data_len ) {
718
         context.console_append(fc::to_hex(data, data_len));
B
Bart Wyatt 已提交
719 720 721
      }
};

722 723 724 725 726 727 728 729 730 731 732 733 734 735 736
template<typename ObjectType>
class db_api : public context_aware_api {
   using KeyType = typename ObjectType::key_type;
   static constexpr int KeyCount = ObjectType::number_of_keys;
   using KeyArrayType = KeyType[KeyCount];
   using ContextMethodType = int(apply_context::*)(const table_id_object&, const KeyType*, const char*, size_t);

   private:
      int call(ContextMethodType method, const scope_name& scope, const name& table, array_ptr<const char> data, size_t data_len) {
         const auto& t_id = context.find_or_create_table(scope, context.receiver, table);
         FC_ASSERT(data_len >= KeyCount * sizeof(KeyType), "Data is not long enough to contain keys");
         const KeyType* keys = reinterpret_cast<const KeyType *>((const char *)data);

         const char* record_data =  ((const char*)data) + sizeof(KeyArrayType);
         size_t record_len = data_len - sizeof(KeyArrayType);
737
         return (context.*(method))(t_id, keys, record_data, record_len) + sizeof(KeyArrayType);
738 739 740 741 742 743
      }

   public:
      using context_aware_api::context_aware_api;

      int store(const scope_name& scope, const name& table, array_ptr<const char> data, size_t data_len) {
744 745 746 747
         auto res = call(&apply_context::store_record<ObjectType>, scope, table, data, data_len);
         //ilog("STORE [${scope},${code},${table}] => ${res} :: ${HEX}", ("scope",scope)("code",context.receiver)("table",table)("res",res)("HEX", fc::to_hex(data, data_len)));
         return res;

748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770
      }

      int update(const scope_name& scope, const name& table, array_ptr<const char> data, size_t data_len) {
         return call(&apply_context::update_record<ObjectType>, scope, table, data, data_len);
      }

      int remove(const scope_name& scope, const name& table, const KeyArrayType &keys) {
         const auto& t_id = context.find_or_create_table(scope, context.receiver, table);
         return context.remove_record<ObjectType>(t_id, keys);
      }
};

template<typename IndexType, typename Scope>
class db_index_api : public context_aware_api {
   using KeyType = typename IndexType::value_type::key_type;
   static constexpr int KeyCount = IndexType::value_type::number_of_keys;
   using KeyArrayType = KeyType[KeyCount];
   using ContextMethodType = int(apply_context::*)(const table_id_object&, KeyType*, char*, size_t);


   int call(ContextMethodType method, const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
      auto maybe_t_id = context.find_table(scope, context.receiver, table);
      if (maybe_t_id == nullptr) {
771
         return -1;
772 773 774 775 776 777 778 779 780
      }

      const auto& t_id = *maybe_t_id;
      FC_ASSERT(data_len >= KeyCount * sizeof(KeyType), "Data is not long enough to contain keys");
      KeyType* keys = reinterpret_cast<KeyType *>((char *)data);

      char* record_data =  ((char*)data) + sizeof(KeyArrayType);
      size_t record_len = data_len - sizeof(KeyArrayType);

781 782 783 784 785
      auto res = (context.*(method))(t_id, keys, record_data, record_len);
      if (res != -1) {
         res += sizeof(KeyArrayType);
      }
      return res;
786 787 788 789 790 791
   }

   public:
      using context_aware_api::context_aware_api;

      int load(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
792 793 794
         auto res = call(&apply_context::load_record<IndexType, Scope>, scope, code, table, data, data_len);
         //ilog("LOAD [${scope},${code},${table}] => ${res} :: ${HEX}", ("scope",scope)("code",code)("table",table)("res",res)("HEX", fc::to_hex(data, data_len)));
         return res;
795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822
      }

      int front(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::front_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

      int back(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::back_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

      int next(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::next_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

      int previous(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::previous_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

      int lower_bound(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::lower_bound_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

      int upper_bound(const scope_name& scope, const account_name& code, const name& table, array_ptr<char> data, size_t data_len) {
         return call(&apply_context::upper_bound_record<IndexType, Scope>, scope, code, table, data, data_len);
      }

};

823
class memory_api : public context_aware_api {
824
   public:
B
Bucky Kittinger 已提交
825
      using context_aware_api::context_aware_api;
826
     
827 828 829 830 831 832 833 834
      char* memcpy( array_ptr<char> dest, array_ptr<const char> src, size_t length) {
         return (char *)::memcpy(dest, src, length);
      }

      int memcmp( array_ptr<const char> dest, array_ptr<const char> src, size_t length) {
         return ::memcmp(dest, src, length);
      }

B
Bucky Kittinger 已提交
835
      char* memset( array_ptr<char> dest, int value, size_t length ) {
B
Bucky Kittinger 已提交
836
         return (char *)::memset( dest, value, length );
B
Bucky Kittinger 已提交
837 838 839
      }

      uint32_t sbrk(int num_bytes) {
B
Bucky Kittinger 已提交
840
         // TODO: omitted checktime function from previous version of sbrk, may need to be put back in at some point
841 842 843
         constexpr uint32_t NBPPL2  = IR::numBytesPerPageLog2;
         constexpr uint32_t MAX_MEM = 1024 * 1024;

844 845 846 847 848 849 850
         MemoryInstance*  default_mem    = Runtime::getDefaultMemory(code.instance);
         if(!default_mem)
            throw eosio::chain::page_memory_error();

         const uint32_t         num_pages      = Runtime::getMemoryNumPages(default_mem);
         const uint32_t         min_bytes      = (num_pages << NBPPL2) > UINT32_MAX ? UINT32_MAX : num_pages << NBPPL2;
         const uint32_t         prev_num_bytes = sbrk_bytes; //_num_bytes;
851
         
B
Bucky Kittinger 已提交
852 853 854
         // round the absolute value of num_bytes to an alignment boundary
         num_bytes = (num_bytes + 7) & ~7;

855
         if ((num_bytes > 0) && (prev_num_bytes > (MAX_MEM - num_bytes)))  // test if allocating too much memory (overflowed)
B
Bucky Kittinger 已提交
856 857 858 859 860
            throw eosio::chain::page_memory_error();
         else if ((num_bytes < 0) && (prev_num_bytes < (min_bytes - num_bytes))) // test for underflow
            throw eosio::chain::page_memory_error(); 

         // update the number of bytes allocated, and compute the number of pages needed
861 862
         sbrk_bytes += num_bytes;
         const uint32_t num_desired_pages = (sbrk_bytes + IR::numBytesPerPage - 1) >> NBPPL2;
B
Bucky Kittinger 已提交
863 864 865 866 867 868 869 870

         // grow or shrink the memory to the desired number of pages
         if (num_desired_pages > num_pages)
            Runtime::growMemory(default_mem, num_desired_pages - num_pages);
         else if (num_desired_pages < num_pages)
            Runtime::shrinkMemory(default_mem, num_pages - num_desired_pages);

         return prev_num_bytes;
871
      }
872 873
};

874 875 876 877 878 879 880 881 882 883 884 885 886 887 888
class transaction_api : public context_aware_api {
   public:
      using context_aware_api::context_aware_api;

      void send_inline( array_ptr<char> data, size_t data_len ) {
         // TODO: use global properties object for dynamic configuration of this default_max_gen_trx_size
         FC_ASSERT( data_len < config::default_max_inline_action_size, "inline action too big" );

         action act;
         fc::raw::unpack<action>(data, data_len, act);
         context.execute_inline(std::move(act));
      }


      void send_deferred( uint32_t sender_id, const fc::time_point_sec& execute_after, array_ptr<char> data, size_t data_len ) {
889 890 891
         try {
            // TODO: use global properties object for dynamic configuration of this default_max_gen_trx_size
            FC_ASSERT(data_len < config::default_max_gen_trx_size, "generated transaction too big");
892

893 894 895 896 897 898 899
            deferred_transaction dtrx;
            fc::raw::unpack<transaction>(data, data_len, dtrx);
            dtrx.sender = context.receiver;
            dtrx.sender_id = sender_id;
            dtrx.execute_after = execute_after;
            context.execute_deferred(std::move(dtrx));
         } FC_CAPTURE_AND_RETHROW((fc::to_hex(data, data_len)));
900 901 902 903
      }

};

B
Bart Wyatt 已提交
904 905
REGISTER_INTRINSICS(system_api,
   (assert,      void(int, int))
906
   (now,          int())
B
Bart Wyatt 已提交
907 908 909 910 911 912
);

REGISTER_INTRINSICS(action_api,
   (read_action,            int(int, int)  )
   (action_size,            int()          )
   (current_receiver,   int64_t()          )
913 914
   (publication_time,   int32_t()          )
   (current_sender,     int64_t()          )
B
Bart Wyatt 已提交
915 916 917
);

REGISTER_INTRINSICS(apply_context,
918 919
   (require_write_lock,    void(int64_t)   )
   (require_read_lock,     void(int64_t, int64_t)   )
B
Bart Wyatt 已提交
920
   (require_recipient,     void(int64_t)   )
921
   (require_authorization, void(int64_t), "require_auth", void(apply_context::*)(const account_name&)const)
B
Bart Wyatt 已提交
922 923 924 925 926 927 928 929 930 931 932 933
);

REGISTER_INTRINSICS(console_api,
   (prints,                void(int)       )
   (prints_l,              void(int, int)  )
   (printi,                void(int64_t)   )
   (printi128,             void(int)       )
   (printd,                void(int64_t)   )
   (printn,                void(int64_t)   )
   (printhex,              void(int, int)  )
);

934 935 936 937 938
REGISTER_INTRINSICS(transaction_api,
   (send_inline,           void(int, int)  )
   (send_deferred,         void(int, int, int, int)  )
);

939 940 941 942
REGISTER_INTRINSICS(memory_api,
   (memcpy,                 int(int, int, int)   )
   (memcmp,                 int(int, int, int)   )
   (memset,                 int(int, int, int)   )
B
Bucky Kittinger 已提交
943
   (sbrk,                   int(int)             )
944 945
);

946

947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982
#define DB_METHOD_SEQ(SUFFIX) \
   (store,        int32_t(int64_t, int64_t, int, int),            "store_"#SUFFIX )\
   (update,       int32_t(int64_t, int64_t, int, int),            "update_"#SUFFIX )\
   (remove,       int32_t(int64_t, int64_t, int),                 "remove_"#SUFFIX )

#define DB_INDEX_METHOD_SEQ(SUFFIX)\
   (load,         int32_t(int64_t, int64_t, int64_t, int, int),   "load_"#SUFFIX )\
   (front,        int32_t(int64_t, int64_t, int64_t, int, int),   "front_"#SUFFIX )\
   (back,         int32_t(int64_t, int64_t, int64_t, int, int),   "back_"#SUFFIX )\
   (next,         int32_t(int64_t, int64_t, int64_t, int, int),   "next_"#SUFFIX )\
   (previous,     int32_t(int64_t, int64_t, int64_t, int, int),   "previous_"#SUFFIX )\
   (lower_bound,  int32_t(int64_t, int64_t, int64_t, int, int),   "lower_bound_"#SUFFIX )\
   (upper_bound,  int32_t(int64_t, int64_t, int64_t, int, int),   "upper_bound_"#SUFFIX )\

using db_api_key_value_object                                 = db_api<key_value_object>;
using db_api_keystr_value_object                              = db_api<keystr_value_object>;
using db_api_key128x128_value_object                          = db_api<key128x128_value_object>;
using db_api_key64x64x64_value_object                         = db_api<key64x64x64_value_object>;
using db_index_api_key_value_index_by_scope_primary           = db_index_api<key_value_index,by_scope_primary>;
using db_index_api_keystr_value_index_by_scope_primary        = db_index_api<keystr_value_index,by_scope_primary>;
using db_index_api_key128x128_value_index_by_scope_primary    = db_index_api<key128x128_value_index,by_scope_primary>;
using db_index_api_key128x128_value_index_by_scope_secondary  = db_index_api<key128x128_value_index,by_scope_secondary>;
using db_index_api_key64x64x64_value_index_by_scope_primary   = db_index_api<key64x64x64_value_index,by_scope_primary>;
using db_index_api_key64x64x64_value_index_by_scope_secondary = db_index_api<key64x64x64_value_index,by_scope_secondary>;
using db_index_api_key64x64x64_value_index_by_scope_tertiary  = db_index_api<key64x64x64_value_index,by_scope_tertiary>;

REGISTER_INTRINSICS(db_api_key_value_object,         DB_METHOD_SEQ(i64));
REGISTER_INTRINSICS(db_api_keystr_value_object,      DB_METHOD_SEQ(str));
REGISTER_INTRINSICS(db_api_key128x128_value_object,  DB_METHOD_SEQ(i128i128));
REGISTER_INTRINSICS(db_api_key64x64x64_value_object, DB_METHOD_SEQ(i64i64i64));

REGISTER_INTRINSICS(db_index_api_key_value_index_by_scope_primary,           DB_INDEX_METHOD_SEQ(i64));
REGISTER_INTRINSICS(db_index_api_keystr_value_index_by_scope_primary,        DB_INDEX_METHOD_SEQ(str));
REGISTER_INTRINSICS(db_index_api_key128x128_value_index_by_scope_primary,    DB_INDEX_METHOD_SEQ(primary_i128i128));
REGISTER_INTRINSICS(db_index_api_key128x128_value_index_by_scope_secondary,  DB_INDEX_METHOD_SEQ(secondary_i128i128));
REGISTER_INTRINSICS(db_index_api_key64x64x64_value_index_by_scope_primary,   DB_INDEX_METHOD_SEQ(primary_i64i64i64));
983
REGISTER_INTRINSICS(db_index_api_key64x64x64_value_index_by_scope_secondary, DB_INDEX_METHOD_SEQ(secondary_i64i64i64));
984 985
REGISTER_INTRINSICS(db_index_api_key64x64x64_value_index_by_scope_tertiary,  DB_INDEX_METHOD_SEQ(tertiary_i64i64i64));

B
Brian Johnson 已提交
986

D
Daniel Larimer 已提交
987
} } /// eosio::chain