Merge pull request #128 from ostrolucky/handle-binary-in-full-http-formatter

Omit binary body in FullHttpMessageFormatter

Author: dbu

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## Unreleased
 
+- Omitted binary body in FullHttpMessageFormatter. `[binary stream omitted]` will be shown instead.
+
 ### Added
 
 - New Header authentication method for arbitrary header authentication.

spec/Formatter/FullHttpMessageFormatterSpec.php

@@ -227,4 +227,25 @@ function it_does_not_format_no_seekable_response(ResponseInterface $response, St
 STR;
         $this->formatResponse($response)->shouldReturn($expectedMessage);
     }
+
+    function it_omits_body_with_null_bytes(RequestInterface $request, StreamInterface $stream)
+    {
+        $this->beConstructedWith(1);
+
+        $stream->isSeekable()->willReturn(true);
+        $stream->rewind()->shouldBeCalled();
+        $stream->__toString()->willReturn("\0");
+        $request->getBody()->willReturn($stream);
+        $request->getMethod()->willReturn('GET');
+        $request->getRequestTarget()->willReturn('/foo');
+        $request->getProtocolVersion()->willReturn('1.1');
+        $request->getHeaders()->willReturn([]);
+
+        $expectedMessage = <<<STR
+GET /foo HTTP/1.1
+
+[binary stream omitted]
+STR;
+        $this->formatRequest($request)->shouldReturn($expectedMessage);
+    }
 }

src/Formatter/FullHttpMessageFormatter.php

@@ -76,20 +76,24 @@ public function formatResponse(ResponseInterface $response)
      */
     private function addBody(MessageInterface $request, $message)
     {
+        $message .= "\n";
         $stream = $request->getBody();
         if (!$stream->isSeekable() || 0 === $this->maxBodyLength) {
             // Do not read the stream
-            return $message."\n";
+            return $message;
         }
 
-        if (null === $this->maxBodyLength) {
-            $message .= "\n".$stream->__toString();
-        } else {
-            $message .= "\n".mb_substr($stream->__toString(), 0, $this->maxBodyLength);
+        $data = $stream->__toString();
+        $stream->rewind();
+
+        if (preg_match('/[\x00-\x1F\x7F]/', $data)) {
+            return $message.'[binary stream omitted]';
         }
 
-        $stream->rewind();
+        if (null === $this->maxBodyLength) {
+            return $message.$data;
+        }
 
-        return $message;
+        return $message.mb_substr($data, 0, $this->maxBodyLength);
     }
 }