blax-software
/
laravel-websockets
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wip

This commit is contained in:
Marcel Pociot 2018-11-23 15:09:54 +01:00
parent 1cfe2c8776
commit a632b43b56
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/Server/OriginCheck.php
View File

@ -8,7 +8,8 @@ use Ratchet\Http\HttpServerInterface;
use Ratchet\MessageComponentInterface; use Ratchet\MessageComponentInterface;
use Psr\Http\Message\RequestInterface; use Psr\Http\Message\RequestInterface;
class OriginCheck implements HttpServerInterface { class OriginCheck implements HttpServerInterface
{
use CloseResponseTrait; use CloseResponseTrait;
@ -17,13 +18,14 @@ class OriginCheck implements HttpServerInterface {
protected $allowedOrigins = []; protected $allowedOrigins = [];
public function __construct(MessageComponentInterface $component, array $allowedOrigins = []) { public function __construct(MessageComponentInterface $component, array $allowedOrigins = [])
{
$this->_component = $component; $this->_component = $component;
$this->allowedOrigins = $allowedOrigins; $this->allowedOrigins = $allowedOrigins;
} }
public function onOpen(ConnectionInterface $connection, RequestInterface $request = null) { public function onOpen(ConnectionInterface $connection, RequestInterface $request = null)
{
if ($request->hasHeader('Origin')) { if ($request->hasHeader('Origin')) {
$this->verifyOrigin($connection, $request); $this->verifyOrigin($connection, $request);
} }
@ -31,15 +33,18 @@ class OriginCheck implements HttpServerInterface {
return $this->_component->onOpen($connection, $request); return $this->_component->onOpen($connection, $request);
} }
function onMessage(ConnectionInterface $from, $msg) { function onMessage(ConnectionInterface $from, $msg)
{
return $this->_component->onMessage($from, $msg); return $this->_component->onMessage($from, $msg);
} }
function onClose(ConnectionInterface $connection) { function onClose(ConnectionInterface $connection)
{
return $this->_component->onClose($connection); return $this->_component->onClose($connection);
} }
function onError(ConnectionInterface $connection, \Exception $e) { function onError(ConnectionInterface $connection, \Exception $e)
{
return $this->_component->onError($connection, $e); return $this->_component->onError($connection, $e);
} }
@ -48,7 +53,7 @@ class OriginCheck implements HttpServerInterface {
$header = (string)$request->getHeader('Origin')[0]; $header = (string)$request->getHeader('Origin')[0];
$origin = parse_url($header, PHP_URL_HOST) ?: $header; $origin = parse_url($header, PHP_URL_HOST) ?: $header;
if (! empty($this->allowedOrigins) && !in_array($origin, $this->allowedOrigins)) { if (!empty($this->allowedOrigins) && !in_array($origin, $this->allowedOrigins)) {
return $this->close($connection, 403); return $this->close($connection, 403);
} }
} }