wip

app/Server/Messages/TunnelMessage.php

@@ -0,0 +1,96 @@
+<?php
+
+namespace App\Server\Messages;
+
+use App\Server\Connections\Connection;
+use App\Server\Connections\ConnectionManager;
+use App\Server\Connections\IoConnection;
+use BFunky\HttpParser\HttpRequestParser;
+use Illuminate\Support\Arr;
+use Illuminate\Support\Str;
+use Ratchet\ConnectionInterface;
+use React\Stream\Util;
+use function GuzzleHttp\Psr7\parse_request;
+
+class TunnelMessage implements Message
+{
+    /** string */
+    protected $payload;
+
+    /** @var \Ratchet\ConnectionInterface */
+    protected $connection;
+
+    /** @var ConnectionManager */
+    private $connectionManager;
+
+    public function __construct($payload, ConnectionInterface $connection, ConnectionManager $connectionManager)
+    {
+        $this->payload = $payload;
+
+        $this->connection = $connection;
+
+        $this->connectionManager = $connectionManager;
+    }
+
+    public function respond()
+    {
+        $clientConnection = $this->connectionManager->findConnectionForSubdomain($this->detectSubdomain());
+
+        if (is_null($clientConnection)) {
+            return;
+        }
+
+        if ($this->hasBufferedAllData()) {
+            $this->copyDataToClient($clientConnection);
+        }
+    }
+
+    protected function getContentLength(): ?int
+    {
+        $request = parse_request($this->connection->buffer);
+
+        return Arr::first($request->getHeader('Content-Length'));
+    }
+
+    protected function detectSubdomain(): ?string
+    {
+        $subdomain = '';
+
+        $headers = collect(explode("\r\n", $this->connection->buffer))->map(function ($header) use (&$subdomain) {
+            $headerData = explode(':', $header);
+            if ($headerData[0] === 'Host') {
+                $domainParts = explode('.', $headerData[1]);
+                $subdomain = trim($domainParts[0]);
+            }
+        });
+
+        return $subdomain;
+    }
+
+    private function copyDataToClient(Connection $clientConnection)
+    {
+        $data = $clientConnection->rewriteHostInformation($this->connectionManager->host(), $this->connectionManager->port(), $this->connection->buffer);
+
+        $requestId = uniqid();
+
+        // Ask client to create a new proxy
+        $clientConnection->socket->send(json_encode([
+                'event' => 'createProxy',
+                'request_id' => $requestId,
+                'client_id' => $clientConnection->client_id,
+            ]) . "||");
+
+        $clientConnection->socket->getConnection()->once('proxy_ready_' . $requestId, function (IoConnection $proxy) use ($data, $requestId) {
+            Util::pipe($proxy->getConnection(), $this->connection->getConnection());
+
+            $proxy->send($data);
+        });
+
+        unset($this->connection->buffer);
+    }
+
+    protected function hasBufferedAllData()
+    {
+        return is_null($this->getContentLength()) || strlen(Str::after($this->connection->buffer, "\r\n\r\n")) === $this->getContentLength();
+    }
+}