PHP Websocket 在测试中验证用户身份(传递会话 Cookie)

2022-08-30 13:26:30

我正在尝试测试一个场景,一方面,匿名用户应该立即断开与Websocket连接的连接,另一方面,经过身份验证的用户应该留在websocket连接中。第一种情况很容易通过使用下面的代码进行测试。身份验证过程不起作用。

对于会话存储,我将Cookie身份验证与数据库结合使用:Symfony PDO会话存储。一切都很好,但是当涉及到使用身份验证来测试所描述的行为时,我不知道如何在测试中对用户进行身份验证。作为一个客户端,我正在使用Pawl异步Websocket客户端。这看起来如下:

\Ratchet\Client\connect('ws://127.0.0.1:8080')->then(function($conn) {
    $conn->on('message', function($msg) use ($conn) {
        echo "Received: {$msg}\n";
    });

    $conn->send('Hello World!');
}, function ($e) {
    echo "Could not connect: {$e->getMessage()}\n";
});

我知道作为第三个参数,我可以将标头信息传递给“connect”方法,但我找不到一种方法来连接客户端并在ws握手期间正确传递cookie。我想到了这样的事情:

  1. 通过创建身份验证令牌对客户端进行身份验证
  2. 我在具有序列化用户的数据库的会话表中创建一个新条目
  3. 我将创建的 cookie 作为第三个参数传递给 connect 方法

这是我认为可行的理论,但用户总是在websocket方面保持匿名。以下是到目前为止理论的代码:

// ...
use Symfony\Bundle\FrameworkBundle\Test\WebTestCase;

class WebsocketTest extends WebTestCase
{

    static $closed;

    protected function setUp()
    {
      self::$closed = null;
    }


    public function testWebsocketConnection()
    {
      $loop = Factory::create();
      $connector = new Connector($loop);

      // This user exists in database user tbl
      $symfClient = $this->createSession("testuser@test.com");

      $connector('ws://127.0.0.1:80', [], ['Origin' => 'http://127.0.0.1', 'Cookie' => 
                 $symfClient->getContainer()->get('session')->getName() . '=' 
                . $symfClient->getContainer()->get('session')->getId()])
        ->then(function(WebSocket $conn) use($loop){

            $conn->on('close', function($code = null, $reason = null) use($loop) {
                self::$closed = true;
                $loop->stop();
            });
            self::$closed = false;

        }, function(\Exception $e) use ($loop) {
            $this->fail("Websocket connection failed");
            $loop->stop();
        });

      $loop->run();

      // Check, that user stayed logged
      $this->assertFalse(self::$closed);
    }

    private function createSession($email)
    {
      $client = static::createClient();
      $container = $client->getContainer();

      $session = $container->get('session');
      $session->set('logged', true);

      $userManager = $container->get('fos_user.user_manager');
      $em = $container->get('doctrine.orm.entity_manager');
      $loginManager = $container->get('fos_user.security.login_manager');
      $firewallName = 'main';

      $user = $userManager->findUserByEmail($email);

      $loginManager->loginUser($firewallName, $user);

      // save the login token into the session and put it in a cookie
      $container->get('session')->set('_security_' . $firewallName,
        serialize($container->get('security.token_storage')->getToken()));
      $container->get('session')->save();
      $client->getCookieJar()->set(new Cookie($session->getName(), $session->getId()));


      // Create session in database
      $pdo = new PDOSessionStorage();
      $pdo->setSessId($session->getId());
      $pdo->setSessTime(time());
      $pdo->setSessData(serialize($container->get('security.token_storage')->getToken()));
      $pdo->setSessLifetime(1440);

      $em->persist($pdo);
      $em->flush();

      return $client;
  }

}

config_test.yml,我按以下方式配置会话:

session:
    storage_id:     session.storage.mock_file
    handler_id:     session.handler.pdo

对于服务器端websocket实现,我使用的是Ratchet,它被以下Symfony捆绑包包装:Gos Websocket Bundle

如何在测试 websockets 时对用户进行身份验证?在websocket服务器上,用户总是类似于“anon-15468850625756b3b424c94871115670”,但是当我手动测试时,他得到了正确的连接。

附加问题(次要):如何测试主题的订阅?(酒吧)互联网上没有关于此的博客文章或其他任何内容。

更新:没有人测试过他们的websockets的功能?这是不重要的,无用的,还是为什么没有人能在这个重要的话题上提供帮助?


答案 1

你有一辆马车先于马的情况在这里。当您在客户端连接上设置cookie时,只有在cookie限制(httpOnly,secure,域,路径等)匹配的后续请求(websockets或XHR,GET,POST等)上发送该cookie。

任何可用的 Cookie 都是在 websocket 连接的初始握手期间发送的。在打开的连接上设置cookie将在客户端上设置cookie,但由于套接字已经是打开的连接并已建立(握手后),因此服务器在该连接期间将对这些cookie视而不见。

有些人在握手时成功地设置了饼干。但是,这需要服务器和客户端套接字实现支持此行为并将凭据作为 get 参数传递(不好的做法)。

所以我认为你唯一真正的选择是:

  • 在打开 websocket 之前,通过 XHR 或其他请求处理身份验证
  • 使用 websocket 进行身份验证,但在成功登录后:
    • 设置您的身份验证 Cookie
    • 关闭现有套接字
    • 从客户端启动一个新的套接字(然后它将携带您的身份验证cookie)
  • 完全忘记 cookie,并根据打开连接的请求/资源 ID 处理服务器上的身份验证交换。

如果您选择最后一个选项,您仍然可以设置 Cookie 并查找 Cookie 以在重新连接时恢复连接。


答案 2

推荐