feat(app sec): Add AppSecRequest class (#13)

crowdsecurity · Sep 5, 2024 · 85f25f5 · 85f25f5
1 parent fe3afd8
commit 85f25f5
Show file tree

Hide file tree

Showing 11 changed files with 385 additions and 60 deletions.
diff --git a/src/Client/AbstractClient.php b/src/Client/AbstractClient.php
@@ -4,11 +4,11 @@
 
 namespace CrowdSec\Common\Client;
 
+use CrowdSec\Common\Client\HttpMessage\AppSecRequest;
 use CrowdSec\Common\Client\HttpMessage\Request;
 use CrowdSec\Common\Client\HttpMessage\Response;
 use CrowdSec\Common\Client\RequestHandler\Curl;
 use CrowdSec\Common\Client\RequestHandler\RequestHandlerInterface;
-use CrowdSec\Common\Constants;
 use Monolog\Handler\NullHandler;
 use Monolog\Logger;
 use Psr\Log\LoggerInterface;
@@ -33,6 +33,10 @@ abstract class AbstractClient
      * @var string[]
      */
     private $allowedMethods = ['POST', 'GET', 'DELETE'];
+    /**
+     * @var string[]
+     */
+    private $allowedAppSecMethods = ['POST', 'GET'];
     /**
      * @var LoggerInterface
      */
@@ -53,7 +57,7 @@ abstract class AbstractClient
     public function __construct(
         array $configs,
         ?RequestHandlerInterface $requestHandler = null,
-        ?LoggerInterface $logger = null
+        ?LoggerInterface $logger = null,
     ) {
         $this->configs = $configs;
         $this->requestHandler = ($requestHandler) ?: new Curl($this->configs);
@@ -88,11 +92,14 @@ public function getRequestHandler(): RequestHandlerInterface
         return $this->requestHandler;
     }
 
-    public function getUrl(string $type = Constants::TYPE_REST): string
+    public function getUrl(): string
     {
-        $url = Constants::TYPE_APPSEC === $type ? $this->appSecUrl : $this->url;
+        return rtrim($this->url, '/') . '/';
+    }
 
-        return rtrim($url, '/') . '/';
+    public function getAppSecUrl(): string
+    {
+        return rtrim($this->appSecUrl, '/') . '/';
     }
 
     /**
@@ -105,7 +112,6 @@ protected function request(
         string $endpoint,
         array $parameters = [],
         array $headers = [],
-        string $type = Constants::TYPE_REST
     ): array {
         $method = strtoupper($method);
         if (!in_array($method, $this->allowedMethods)) {
@@ -115,7 +121,31 @@ protected function request(
         }
 
         $response = $this->sendRequest(
-            new Request($this->getFullUrl($endpoint, $type), $method, $headers, $parameters)
+            new Request($this->getFullUrl($endpoint), $method, $headers, $parameters)
+        );
+
+        return $this->formatResponseBody($response);
+    }
+
+    /**
+     * Performs an HTTP request (POST, GET) to AppSec and returns its response body as an array.
+     *
+     * @throws ClientException
+     */
+    protected function requestAppSec(
+        string $method,
+        array $headers = [],
+        string $rawBody = '',
+    ): array {
+        $method = strtoupper($method);
+        if (!in_array($method, $this->allowedAppSecMethods)) {
+            $message = "Method ($method) is not allowed.";
+            $this->logger->error($message, ['type' => 'CLIENT_APPSEC_REQUEST']);
+            throw new ClientException($message);
+        }
+
+        $response = $this->sendRequest(
+            new AppSecRequest($this->getAppSecUrl(), $method, $headers, $rawBody)
         );
 
         return $this->formatResponseBody($response);
@@ -160,8 +190,8 @@ private function formatResponseBody(Response $response): array
         return $decoded;
     }
 
-    private function getFullUrl(string $endpoint, string $type = Constants::TYPE_REST): string
+    private function getFullUrl(string $endpoint): string
     {
-        return $this->getUrl($type) . ltrim($endpoint, '/');
+        return $this->getUrl() . ltrim($endpoint, '/');
     }
 }
diff --git a/src/Client/HttpMessage/AppSecRequest.php b/src/Client/HttpMessage/AppSecRequest.php
@@ -0,0 +1,42 @@
+<?php
+
+declare(strict_types=1);
+
+namespace CrowdSec\Common\Client\HttpMessage;
+
+/**
+ * Request that will be sent to CrowdSec AppSec component.
+ *
+ * @author    CrowdSec team
+ *
+ * @see      https://crowdsec.net CrowdSec Official Website
+ *
+ * @copyright Copyright (c) 2024+ CrowdSec
+ * @license   MIT License
+ */
+class AppSecRequest extends Request
+{
+    /**
+     * @var array
+     */
+    protected $headers = [];
+    /**
+     * @var string
+     */
+    private $rawBody = '';
+
+    public function __construct(
+        string $uri,
+        string $method,
+        array $headers = [],
+        string $rawBody = '',
+    ) {
+        $this->rawBody = $rawBody;
+        parent::__construct($uri, $method, $headers);
+    }
+
+    public function getRawBody(): string
+    {
+        return $this->rawBody;
+    }
+}
diff --git a/src/Client/HttpMessage/Request.php b/src/Client/HttpMessage/Request.php
@@ -4,8 +4,6 @@
 
 namespace CrowdSec\Common\Client\HttpMessage;
 
-use CrowdSec\Common\Constants;
-
 /**
  * Request that will be sent to CrowdSec.
  *
@@ -43,11 +41,10 @@ public function __construct(
         string $method,
         array $headers = [],
         array $parameters = [],
-        string $type = Constants::TYPE_REST
     ) {
         $this->uri = $uri;
         $this->method = $method;
-        $this->headers = Constants::TYPE_REST === $type ? array_merge($this->headers, $headers) : $headers;
+        $this->headers = array_merge($this->headers, $headers);
         $this->parameters = $parameters;
     }
 

diff --git a/src/Client/RequestHandler/Curl.php b/src/Client/RequestHandler/Curl.php
@@ -5,6 +5,7 @@
 namespace CrowdSec\Common\Client\RequestHandler;
 
 use CrowdSec\Common\Client\ClientException;
+use CrowdSec\Common\Client\HttpMessage\AppSecRequest;
 use CrowdSec\Common\Client\HttpMessage\Request;
 use CrowdSec\Common\Client\HttpMessage\Response;
 use CrowdSec\Common\Constants;
@@ -62,22 +63,9 @@ protected function getResponseHttpCode($handle)
         return curl_getinfo($handle, \CURLINFO_HTTP_CODE);
     }
 
-    private function handleConfigs(): array
+    private function handleTimeout(): array
     {
-        $result = [\CURLOPT_SSL_VERIFYPEER => false];
-        $authType = $this->getConfig('auth_type');
-        if ($authType && Constants::AUTH_TLS === $authType) {
-            $verifyPeer = $this->getConfig('tls_verify_peer') ?? true;
-            $result[\CURLOPT_SSL_VERIFYPEER] = $verifyPeer;
-            // The --cert option
-            $result[\CURLOPT_SSLCERT] = $this->getConfig('tls_cert_path') ?? '';
-            // The --key option
-            $result[\CURLOPT_SSLKEY] = $this->getConfig('tls_key_path') ?? '';
-            if ($verifyPeer) {
-                // The --cacert option
-                $result[\CURLOPT_CAINFO] = $this->getConfig('tls_ca_cert_path') ?? '';
-            }
-        }
+        $result = [];
         $timeout = $this->getConfig('api_timeout') ?? Constants::API_TIMEOUT;
         /**
          * To obtain an unlimited timeout, we don't pass the option (as it is the default behavior).
@@ -100,13 +88,38 @@ private function handleConfigs(): array
         return $result;
     }
 
-    private function handleMethod(string $method, string $url, array $parameters = []): array
+    private function handleSSL(Request $request): array
+    {
+        $result = [\CURLOPT_SSL_VERIFYPEER => false];
+        if ($request instanceof AppSecRequest) {
+            // AppSec does not require SSL verification
+            return $result;
+        }
+
+        $authType = $this->getConfig('auth_type');
+        if ($authType && Constants::AUTH_TLS === $authType) {
+            $verifyPeer = $this->getConfig('tls_verify_peer') ?? true;
+            $result[\CURLOPT_SSL_VERIFYPEER] = $verifyPeer;
+            // The --cert option
+            $result[\CURLOPT_SSLCERT] = $this->getConfig('tls_cert_path') ?? '';
+            // The --key option
+            $result[\CURLOPT_SSLKEY] = $this->getConfig('tls_key_path') ?? '';
+            if ($verifyPeer) {
+                // The --cacert option
+                $result[\CURLOPT_CAINFO] = $this->getConfig('tls_ca_cert_path') ?? '';
+            }
+        }
+
+        return $result;
+    }
+
+    private function handleMethod(string $method, string $url, array $parameters = [], $rawBody = ''): array
     {
         $result = [];
         if ('POST' === strtoupper($method)) {
             $result[\CURLOPT_POST] = true;
             $result[\CURLOPT_CUSTOMREQUEST] = 'POST';
-            $result[\CURLOPT_POSTFIELDS] = json_encode($parameters);
+            $result[\CURLOPT_POSTFIELDS] = $rawBody ?: json_encode($parameters);
         } elseif ('GET' === strtoupper($method)) {
             $result[\CURLOPT_POST] = false;
             $result[\CURLOPT_CUSTOMREQUEST] = 'GET';
@@ -132,19 +145,27 @@ private function handleMethod(string $method, string $url, array $parameters = [
      */
     private function createOptions(Request $request): array
     {
+        $isAppSec = $request instanceof AppSecRequest;
         $headers = $request->getHeaders();
         $method = $request->getMethod();
         $url = $request->getUri();
         $parameters = $request->getParams();
-        if (!isset($headers['User-Agent'])) {
+        if (!isset($headers['User-Agent']) && !$isAppSec) {
             throw new ClientException('User agent is required', 400);
         }
+        $rawBody = '';
+        if ($isAppSec) {
+            /** @var AppSecRequest $request */
+            $rawBody = $request->getRawBody();
+        }
         $options = [
             \CURLOPT_HEADER => false,
             \CURLOPT_RETURNTRANSFER => true,
-            \CURLOPT_USERAGENT => $headers['User-Agent'],
             \CURLOPT_ENCODING => '',
         ];
+        if (isset($headers['User-Agent'])) {
+            $options[\CURLOPT_USERAGENT] = $headers['User-Agent'];
+        }
 
         $options[\CURLOPT_HTTPHEADER] = [];
         foreach ($headers as $key => $values) {
@@ -153,8 +174,9 @@ private function createOptions(Request $request): array
             }
         }
         // We need to keep keys indexes (array_merge not keeping indexes)
-        $options += $this->handleConfigs();
-        $options += $this->handleMethod($method, $url, $parameters);
+        $options += $this->handleSSL($request);
+        $options += $this->handleTimeout();
+        $options += $this->handleMethod($method, $url, $parameters, $rawBody);
 
         return $options;
     }

diff --git a/src/Client/RequestHandler/FileGetContents.php b/src/Client/RequestHandler/FileGetContents.php
@@ -5,6 +5,7 @@
 namespace CrowdSec\Common\Client\RequestHandler;
 
 use CrowdSec\Common\Client\ClientException;
+use CrowdSec\Common\Client\HttpMessage\AppSecRequest;
 use CrowdSec\Common\Client\HttpMessage\Request;
 use CrowdSec\Common\Client\HttpMessage\Response;
 use CrowdSec\Common\Constants;
@@ -95,9 +96,15 @@ protected function convertHeadersToString(array $headers): string
     private function createContextConfig(Request $request): array
     {
         $headers = $request->getHeaders();
-        if (!isset($headers['User-Agent'])) {
+        $isAppSec = $request instanceof AppSecRequest;
+        if (!isset($headers['User-Agent']) && !$isAppSec) {
             throw new ClientException('User agent is required', 400);
         }
+        $rawBody = '';
+        if ($isAppSec) {
+            /** @var AppSecRequest $request */
+            $rawBody = $request->getRawBody();
+        }
         $header = $this->convertHeadersToString($headers);
         $method = $request->getMethod();
         $timeout = $this->getConfig('api_timeout');
@@ -112,24 +119,36 @@ private function createContextConfig(Request $request): array
             ],
         ];
 
-        $config['ssl'] = ['verify_peer' => false];
+        $config += $this->handleSSL($request);
+
+        if ('POST' === strtoupper($method)) {
+            $config['http']['content'] =
+                $isAppSec ? $rawBody : json_encode($request->getParams());
+        }
+
+        return $config;
+    }
+
+    private function handleSSL(Request $request): array
+    {
+        $result = ['ssl' => ['verify_peer' => false]];
+        if ($request instanceof AppSecRequest) {
+            // AppSec does not require SSL verification
+            return $result;
+        }
         $authType = $this->getConfig('auth_type');
         if ($authType && Constants::AUTH_TLS === $authType) {
             $verifyPeer = $this->getConfig('tls_verify_peer') ?? true;
-            $config['ssl'] = [
+            $result['ssl'] = [
                 'verify_peer' => $verifyPeer,
                 'local_cert' => $this->getConfig('tls_cert_path') ?? '',
                 'local_pk' => $this->getConfig('tls_key_path') ?? '',
             ];
             if ($verifyPeer) {
-                $config['ssl']['cafile'] = $this->getConfig('tls_ca_cert_path') ?? '';
+                $result['ssl']['cafile'] = $this->getConfig('tls_ca_cert_path') ?? '';
             }
         }
 
-        if ('POST' === strtoupper($method)) {
-            $config['http']['content'] = json_encode($request->getParams());
-        }
-
-        return $config;
+        return $result;
     }
 }
diff --git a/src/Constants.php b/src/Constants.php
@@ -56,6 +56,10 @@ class Constants
      * @var string The LISTS origin for decisions
      */
     public const ORIGIN_LISTS = 'lists';
+    /**
+     * @var string The CrowdSec App Sec raw body param
+     */
+    public const APPSEC_RAW_BODY_PARAM = 'app_sec_raw_body';
     /**
      * @var string The ban remediation
      */

diff --git a/tests/MockedData.php b/tests/MockedData.php
@@ -32,5 +32,9 @@ class MockedData
 
     public const UNAUTHORIZED = <<<EOT
 {"message":"Unauthorized"}
+EOT;
+
+    public const APPSEC_ALLOWED = <<<EOT
+{"action":"allow"}
 EOT;
 }