发布于 由admin

剖析 WordPress `is_user_logged_in()` 函数源码：如何检查 Cookie 与全局变量判断登录状态。

咳咳，麦克风测试，1，2，3… 大家好，我是今天的讲师，人称“代码界的段子手”老码。今天咱们就来聊聊WordPress里那个看起来人畜无害，实则暗藏玄机的is_user_logged_in()函数。

这函数，说白了，就是判断用户是不是登录了。但它怎么判断的呢？难道是靠第六感？当然不是，它是靠检查你的浏览器里有没有特殊的“身份证”——Cookie，再结合WordPress全局变量里的一些蛛丝马迹来判断的。

准备好了吗？咱们开始解剖这个函数，看看它肚子里到底装了些什么宝贝。

1. 函数的庐山真面目

首先，让我们先睹为快，看看is_user_logged_in()函数的源码：

function is_user_logged_in() {
    /**
     * Filters whether the current user is logged in.
     *
     * @since 2.5.0
     *
     * @param bool $is_logged_in Whether the user is logged in. Default false.
     */
    $logged_in = apply_filters( 'logged_in', false );

    if ( $logged_in ) {
        return true;
    }

    if ( isset( $GLOBALS['current_user'] ) && ( $GLOBALS['current_user'] instanceof WP_User ) && ( $GLOBALS['current_user']->ID ) ) {
        return true;
    }

    return false;
}

是不是感觉代码不多，挺简单的？别被它的外表迷惑了，麻雀虽小，五脏俱全。

2. 第一道防线：apply_filters( 'logged_in', false )

第一行代码，就用到了一个过滤器apply_filters( 'logged_in', false )。 这个东西，就像一个“拦截器”，允许你在函数执行之前，通过钩子函数来改变它的行为。

简单来说，就是如果有人注册了一个叫做logged_in的过滤器，并且让它返回true，那么is_user_logged_in()函数就会直接返回true，根本不需要进行后续的检查。

这种机制非常灵活，允许插件或者主题在某些特殊情况下，人为地“伪造”用户登录状态。 比如说，你可以根据用户的IP地址，或者其他自定义的逻辑，来判断用户是否“应该”被认为是登录状态。

// 示例：使用过滤器强制用户登录状态为true
add_filter( 'logged_in', 'my_custom_login_check' );

function my_custom_login_check( $logged_in ) {
  // 这里可以添加你的自定义逻辑
  // 例如，根据IP地址判断是否应该被认为是登录状态
  if ( $_SERVER['REMOTE_ADDR'] == '127.0.0.1' ) {
    return true; // 如果是本地访问，则认为是登录状态
  }
  return $logged_in; // 否则，保持原有的登录状态
}

3. 主力军：检查全局变量$GLOBALS['current_user']

如果过滤器没有起作用（或者返回了false），那么is_user_logged_in()函数就会进入正题，开始检查全局变量$GLOBALS['current_user']。

这个全局变量，是WordPress用来存储当前用户信息的地方。 当用户登录成功后，WordPress会将用户的各种信息（用户名、ID、邮箱等等）封装到一个WP_User对象里，然后把这个对象赋值给$GLOBALS['current_user']。

所以，is_user_logged_in()函数会检查以下几个条件：

  • isset( $GLOBALS['current_user'] ): 判断$GLOBALS['current_user']这个变量是否存在。 如果不存在，说明用户肯定没有登录。
  • ( $GLOBALS['current_user'] instanceof WP_User ): 判断$GLOBALS['current_user']这个变量是不是一个WP_User对象。 如果不是，说明它可能是一个空变量，或者是一个其他类型的变量，同样说明用户没有登录。
  • ( $GLOBALS['current_user']->ID ): 判断$GLOBALS['current_user']对象的ID属性是否为空。 ID是用户的唯一标识符，如果为空，说明WP_User对象可能是一个“空壳”，没有包含任何有用的用户信息，同样说明用户没有登录。

只有当以上三个条件都满足时，is_user_logged_in()函数才会返回true，表示用户已经登录。

4. Cookie的秘密：wp_set_auth_cookie()和wp_validate_auth_cookie()

你可能会问，$GLOBALS['current_user']这个变量是怎么被赋值的呢？ 这就涉及到Cookie的知识了。

当用户在登录表单中输入正确的用户名和密码后，WordPress会使用wp_set_auth_cookie()函数，在用户的浏览器中设置一些特殊的Cookie，这些Cookie包含了用户的认证信息。

然后，在每次用户访问WordPress网站时，WordPress会使用wp_validate_auth_cookie()函数来验证这些Cookie的有效性。 如果Cookie有效，WordPress就会根据Cookie中的信息，创建一个WP_User对象，并把它赋值给$GLOBALS['current_user']。

这两个函数，就像是WordPress登录认证的“左右护法”，负责生成和验证Cookie，确保用户的登录状态能够被正确地识别。

让我们简单看看这两个函数的作用：

  • wp_set_auth_cookie( $user_id, $remember = false, $secure = '' ): 这个函数负责设置认证Cookie。

    • $user_id: 用户的ID。
    • $remember: 是否记住登录状态（如果为true，则Cookie的有效期会更长）。
    • $secure: 是否使用HTTPS连接（如果为空，则会根据当前连接类型自动判断）。

    这个函数会将用户的ID、登录时间、以及一些加密信息，写入到Cookie中。 这些Cookie会被保存在用户的浏览器中，直到过期或者被清除。

  • wp_validate_auth_cookie( $cookie = '', $scheme = '' ): 这个函数负责验证认证Cookie。

    • $cookie: 要验证的Cookie的值（通常是$_COOKIE[AUTH_COOKIE]）。
    • $scheme: 认证方案（例如，logged_in、auth、secure_auth）。

    这个函数会从Cookie中提取用户信息，并验证Cookie的有效性。 如果Cookie有效，它会返回用户的ID； 如果Cookie无效，它会返回false。

5. 登录流程的简要总结

为了更好地理解is_user_logged_in()函数的工作原理，我们来简单总结一下WordPress的登录流程：

  1. 用户在登录页面输入用户名和密码，并提交表单。
  2. WordPress验证用户名和密码是否正确。
  3. 如果用户名和密码正确，WordPress会调用wp_set_auth_cookie()函数，在用户的浏览器中设置认证Cookie。
  4. 用户访问WordPress网站的其他页面。
  5. WordPress会调用wp_validate_auth_cookie()函数，验证Cookie的有效性。
  6. 如果Cookie有效，WordPress会根据Cookie中的信息，创建一个WP_User对象，并把它赋值给$GLOBALS['current_user']。
  7. 当调用is_user_logged_in()函数时，它会检查$GLOBALS['current_user']变量，判断用户是否已经登录。

用表格来表示可能更清晰：

| 步骤 | 动作 | 涉及函数 | 作用 |
|—-|—————————————-|————————————————|—————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————- ——————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————–>

6. 总结与注意事项

is_user_logged_in()函数，虽然看起来简单，但是它却是WordPress用户认证体系中的一个重要环节。它通过结合Cookie和全局变量的信息，来判断用户是否已经登录。

需要注意的是：

  • 安全性： Cookie的安全性至关重要。 WordPress使用加密技术来保护Cookie，防止被篡改。 因此，不要轻易修改WordPress的Cookie相关的代码，除非你非常清楚你在做什么。
  • 缓存： 有些缓存插件可能会缓存is_user_logged_in()函数的结果，导致登录状态的判断不准确。 因此，在使用缓存插件时，需要注意相关的配置，确保登录状态能够被正确地缓存和更新。
  • HTTPS： 如果你的网站使用了HTTPS连接，那么Cookie也会被标记为secure，只能通过HTTPS连接进行传输，这样可以有效地防止Cookie被窃取。

7. 课后思考题

  1. 如果用户在登录后，手动清除了浏览器中的Cookie，那么is_user_logged_in()函数的返回值会是什么？为什么？
  2. 如何在不修改WordPress核心代码的情况下，使用过滤器来改变is_user_logged_in()函数的返回值？
  3. 如果你的网站需要实现一个“记住我”的功能，应该如何使用wp_set_auth_cookie()函数来实现？

希望今天的讲解对大家有所帮助。 以后遇到和登录状态相关的问题，不妨回头看看这篇文章，说不定能给你带来新的思路。

好了，今天的课就到这里，下课！ 各位码农们，继续加油！

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注