Mysql 源码阅读（二）登录连接调试

简介

在上篇中我们将源码在本地顺利的跑了起来，接下来我们需要连接我们的数据库，为后面的工作做一定的准备

使用docker容器尝试连接

我们首先使用docker容器去启动连接我们上篇文章中成功运行起来的MySQL服务器，但没能如愿，没有连接成功

下面的错误，以前我们也是经常见到了

PS C:\Users\lw> docker run -ti mysql mysql -h 192.168.1.4 -P 3306 -u root -p
Enter password:
ERROR 1130 (HY000): Host 'DESKTOP-8U69O9P' is not allowed to connect to this MySQL server


我们使用： is not allowed 去全局搜索下代码，找到下面的代码，通过打上断点，也走到了这步里面

成功的找到了代码入口

bool acl_check_host(const char *host, const char *ip)
{
 mysql_mutex_lock(&acl_cache->lock);
 if (allow_all_hosts)
 {
 
 mysql_mutex_unlock(&acl_cache->lock);
 return 0;
 }

 if ((host && my_hash_search(&acl_check_hosts,(uchar*) host,strlen(host))) ||
 (ip && my_hash_search(&acl_check_hosts,(uchar*) ip, strlen(ip))))
 {
 mysql_mutex_unlock(&acl_cache->lock);
 return 0; // Found host
 }
 for (ACL_HOST_AND_IP *acl= acl_wild_hosts->begin();
 acl != acl_wild_hosts->end(); ++acl)
 {
 if (acl->compare_hostname(host, ip))
 {
 mysql_mutex_unlock(&acl_cache->lock);
 return 0; // Host ok
 }
 }
 mysql_mutex_unlock(&acl_cache->lock);
 if (ip != NULL)
 {
 /* Increment HOST_CACHE.COUNT_HOST_ACL_ERRORS. */
 Host_errors errors;
 errors.m_host_acl= 1;
 inc_host_errors(ip, &errors);
 }
 return 1; // Host is not allowed
}


通过调试，我们发现acl_wild_hosts里面没有值导致的，导致ip==null，从而host没有匹配上，不允许连接

我们在当前文件中没有搜索到acl_wild_hosts，只能在整个解决方案中进行搜索，我们定位到下面一段代码：

static void init_check_host(void)
{
 DBUG_ENTER("init_check_host");
 if (acl_wild_hosts != NULL)
 acl_wild_hosts->clear();
 else
 acl_wild_hosts=
 new Prealloced_array<ACL_HOST_AND_IP, ACL_PREALLOC_SIZE>(key_memory_acl_mem);

 size_t acl_users_size= acl_users ? acl_users->size() : 0;

 (void) my_hash_init(&acl_check_hosts,system_charset_info,
 acl_users_size, 0, 0,
 (my_hash_get_key) check_get_key, 0, 0,
 key_memory_acl_mem);
 if (acl_users_size && !allow_all_hosts)
 {
 for (ACL_USER *acl_user= acl_users->begin();
 acl_user != acl_users->end(); ++acl_user)
 {
 if (acl_user->host.has_wildcard())
 { // Has wildcard
 ACL_HOST_AND_IP *acl= NULL;
 for (acl= acl_wild_hosts->begin(); acl != acl_wild_hosts->end(); ++acl)
 { // Check if host already exists
 if (!my_strcasecmp(system_charset_info,
 acl_user->host.get_host(), acl->get_host()))
 break; // already stored
 }
 if (acl == acl_wild_hosts->end()) // If new
 acl_wild_hosts->push_back(acl_user->host);
 }
 else if (!my_hash_search(&acl_check_hosts,(uchar*)
 acl_user->host.get_host(),
 acl_user->host.get_host_len()))
 {
 if (my_hash_insert(&acl_check_hosts,(uchar*) acl_user))
 { // End of memory
 allow_all_hosts=1; // Should never happen
 DBUG_VOID_RETURN;
 }
 }
 }
 }
 acl_wild_hosts->shrink_to_fit();
 freeze_size(&acl_check_hosts.array);
 DBUG_VOID_RETURN;
}


从名称和功能上看出，这个就是acl_check_hosts的初始化相关代码

其中acl_users通过断点调试，有三个用户：

• root：我们的经常用到的初始用户

• mysql.session: 这个还真没见过

• mysql.sys: 这个也没有见过

这个函数看下来，大致有两个地方会导致acl_check_hosts的增加：

(void) my_hash_init(&acl_check_hosts,system_charset_info,
 acl_users_size, 0, 0,
 (my_hash_get_key) check_get_key, 0, 0,
 key_memory_acl_mem);


上面这个是初始化，从里面的参数没有找到什么新增的

下面一段代码是明显新增的，但调试中，代码一直走不进去这段逻辑

if (acl == acl_wild_hosts->end()) // If new
 acl_wild_hosts->push_back(acl_user->host);


而走不下去的第一步就是

if (acl_user->host.has_wildcard())

bool has_wildcard()
 {
 return (strchr(hostname,wild_many) ||
 strchr(hostname,wild_one) || ip_mask );
 }


strchr是一个字符串搜索函数：该函数返回在字符串 str 中第一次出现字符 c 的位置，如果未找到该字符则返回 NULL。

其中的wild_many是我们熟悉的%，而wild_one是"_"，而hostname是localhost，所以没有匹配上

都走不下去，那我们只能看看acl_user的相关代码了

很幸运，在当前文件中，我们找到了acl_users的初始化函数：

static my_bool acl_load(THD *thd, TABLE_LIST *tables)

// 函数代码太长了，先粘贴一个关键的
acl_users->push_back(user);


而host的初始化在这步：

user.host.update_hostname(get_field(&global_acl_memory,
 table->field[table_schema->host_idx()]));


看着是从表中读取一些数据，有点头大，搞不动了，看不懂（超过认知负载了，我们继续挖下去也挖不动了，只能先往上跳一跳）

使用mysql workbench进行连接

为了避免一直卡住不动了，我们还是直接本地安装一个mysql workbench，然后使用localhost的域名和root用户进行连接

如果忘记了初始密码，想要再初始化一次，可能会遇到下面的错误：

2022-06-25T00:28:59.791538Z 0 [ERROR] --initialize specified but the data directory has files in it. Aborting.


我们把源码根目录下的: release/sql/data 文件夹删除即可

初次链接的使用需要重置密码，重置完成后，就链接上了

然后我们执行下面的脚本，赋予root其他ip的连接权限

SET SQL_SAFE_UPDATES = 0;
use mysql;
update user set host='%' where user='root';
flush privileges;


在我们运行上面的SQL的时候，用户的初始化和host检测的初始化，都重新执行一遍

再次使用mysql容器进行连接

然后我们再次运行命令进行连接

docker run -ti mysql mysql -h 192.168.1.4 -P 3306 -u root -p


通过调试，我们发现现在是直接：allow_all_hosts 直接允许访问了

bool acl_check_host(const char *host, const char *ip)
{
 mysql_mutex_lock(&acl_cache->lock);
 if (allow_all_hosts)
 {
 
 mysql_mutex_unlock(&acl_cache->lock);
 return 0;
 }
 ......
}


通过查看代码是在这里设置的，但也看不是太懂，留待日后，今天看的也有点晕了

static void init_check_host(void)
{
......
 if (acl_users_size && !allow_all_hosts)
 {
 {
 if (my_hash_insert(&acl_check_hosts,(uchar*) acl_user))
 { // End of memory
 // 这里直接设置了
 allow_all_hosts=1; // Should never happen
 DBUG_VOID_RETURN;
 }
 }
 }
 }
......
}


然后我们就成功访问了

总结

今天吧登录认证的代码走了一遍，对MySQL的连接认证有了一个初步的认知，但还是有很多的细节不清楚

使用docker容器连接成功后，我们一直debug下去，发现了tcp连接的监听处理函数：

extern "C" void *handle_connection(void *arg)
{
 // 连接准备，今天的host检测和用户认证都是在这个函数里面进行的
 if (thd_prepare_connection(thd))
 handler_manager->inc_aborted_connects();
 else
 {
 while (thd_connection_alive(thd))
 {
 // 看名字就知道是具体命令的处理函数，嘿嘿，找到了下篇文章的函数入口
 if (do_command(thd))
 break;
 }
 end_connection(thd);
 }
 close_connection(thd, 0, false, false);
 ······
}


参考链接

• MySQL error code: 1175 during UPDATE in MySQL Workbench

• MySQL篇之初始化数据库后修改root用户密码以及使用指定host登录

• C 库函数 - strchr()