Merge pull request #6 from jwpage/use_resources

Use file resources for BinaryReader

README.md

@@ -10,15 +10,15 @@ You probably wouldn't be here if you hadn't run into a scenario where you needed
 binary data. The honest truth is PHP really stinks at this stuff, but as long as we're going to be using it we may as
 well do our best to make it as painless as possible.
 
-The purpose of this binary reader is to accept a string of file contents and provide a set of methods inspired by .NET
-to traverse it.
+The purpose of this binary reader is to accept a string of file contents or file resource and provide a set of methods 
+inspired by .NET to traverse it.
 
 Note on Endians
 ---
 The reader is designed to work on little endian machines, which is going to apply to most scenarios as all x86 and x86-64
 machines are little endian. If you have somehow found yourself on a big endian machine, you need to inform the class or
 you may not be able to properly read signed integers in the file you're parsing.
-```
+```php
 $fileData = file_get_contents('somefile.bin');
 $br = new BinaryReader($fileData);
 $br->setMachineByteOrder(Endian::ENDIAN_BIG);
@@ -27,9 +27,14 @@ $br->setMachineByteOrder(Endian::ENDIAN_BIG);
 
 Example Usage
 ---
-```
+
+```php
 $fileData = file_get_contents('somefile.bin');
 $br = new BinaryReader($fileData, Endian::ENDIAN_LITTLE);
+// or
+$fileResource = fopen('somefile.bin', 'rb');
+$br = new BinaryReader($fileResource, Endian::ENDIAN_LITTLE);
+
 $magic = $br->readUInt32();
 $offset = $br->readUInt16();
 $length = $br->readUInt16();
@@ -38,7 +43,7 @@ $length = $br->readUInt16();
 
 Methods
 ---
-**__construct($str, $endian)** a string must be provided to use this class, an endian is optional (string [big|little], or use the constants in the Endian class), it will default to little if not provided.
+**__construct($input, $endian)** a string or file resource must be provided to use this class, an endian is optional (string [big|little], or use the constants in the Endian class), it will default to little if not provided.
 
 **readUInt8()** returns a single 8 bit byte as an unsigned integer
 

src/BinaryReader.php

@@ -18,9 +18,9 @@ class BinaryReader
     private $machineByteOrder = Endian::ENDIAN_LITTLE;
 
     /**
-     * @var string
+     * @var resource 
      */
-    private $inputString;
+    private $inputHandle;
 
     /**
      * @var int
@@ -78,16 +78,21 @@ class BinaryReader
     private $int32Reader;
 
     /**
-     * @param  string                    $str
+     * @param  string|resource           $input
      * @param  int|string                $endian
      * @throws \InvalidArgumentException
      */
-    public function __construct($str, $endian = Endian::ENDIAN_LITTLE)
+    public function __construct($input, $endian = Endian::ENDIAN_LITTLE)
     {
-        $this->eofPosition = strlen($str);
+        if (!is_resource($input)) {
+            $this->setInputString($input);
+        } else {
+            $this->setInputHandle($input);
+        }
+
+        $this->eofPosition = fstat($this->getInputHandle())['size'];
 
         $this->setEndian($endian);
-        $this->setInputString($str);
         $this->setNextByte(false);
         $this->setCurrentBit(0);
         $this->setPosition(0);
@@ -227,12 +232,34 @@ public function getMachineByteOrder()
     }
 
     /**
-     * @param  string $inputString
+     * @param  resource $inputHandle
+     * @return $this
+     */
+    public function setInputHandle($inputHandle)
+    {
+        $this->inputHandle = $inputHandle;
+
+        return $this;
+    }
+
+    /**
+     * @return resource
+     */
+    public function getInputHandle()
+    {
+        return $this->inputHandle;
+    }
+
+    /**
+     * @param string $inputString
      * @return $this
      */
     public function setInputString($inputString)
     {
-        $this->inputString = $inputString;
+        $handle = fopen('php://memory', 'br+');
+        fwrite($handle, $inputString);
+        rewind($handle);
+        $this->inputHandle = $handle;
 
         return $this;
     }
@@ -242,7 +269,11 @@ public function setInputString($inputString)
      */
     public function getInputString()
     {
-        return $this->inputString;
+        $handle = $this->getInputHandle();
+        $str = stream_get_contents($handle);
+        rewind($handle);
+
+        return $str;
     }
 
     /**
@@ -271,6 +302,7 @@ public function getNextByte()
     public function setPosition($position)
     {
         $this->position = $position;
+        fseek($this->getInputHandle(), $position);
 
         return $this;
     }

src/Type/Bit.php

@@ -76,7 +76,7 @@ public function read(BinaryReader &$br, $length)
 
         if ($bits != 0) {
             $code = $this->getSigned() ? 'c' : 'C';
-            $data = unpack($code, substr($br->getInputString(), $br->getPosition(), 1));
+            $data = unpack($code, fread($br->getInputHandle(), 1));
             $br->setNextByte($data[1]);
             $br->setPosition($br->getPosition() + 1);
             $result |= $br->getNextByte() & $bitmask->getMask($bits, BitMask::MASK_LO);

src/Type/Byte.php

@@ -28,7 +28,7 @@ public function read(BinaryReader &$br, $length = null)
             throw new \OutOfBoundsException('Cannot read bytes, it exceeds the boundary of the file');
         }
 
-        $segment = substr($br->getInputString(), $br->getPosition(), $length);
+        $segment = fread($br->getInputHandle(), $length);
 
         $br->setPosition($br->getPosition() + $length);
 

src/Type/Int16.php

@@ -33,7 +33,7 @@ public function read(BinaryReader &$br, $length = null)
         }
 
         $endian = $br->getEndian() == Endian::ENDIAN_BIG ? $this->getEndianBig() : $this->getEndianLittle();
-        $segment = substr($br->getInputString(), $br->getPosition(), 2);
+        $segment = fread($br->getInputHandle(), 2);
 
         $data = unpack($endian, $segment);
         $data = $data[1];

src/Type/Int32.php

@@ -33,7 +33,7 @@ public function read(BinaryReader &$br, $length = null)
         }
 
         $endian = $br->getEndian() == Endian::ENDIAN_BIG ? $this->getEndianBig() : $this->getEndianLittle();
-        $segment = substr($br->getInputString(), $br->getPosition(), 4);
+        $segment = fread($br->getInputHandle(), 4);
 
         $data = unpack($endian, $segment);
         $data = $data[1];

src/Type/Int8.php

@@ -27,7 +27,7 @@ public function read(BinaryReader &$br, $length = null)
             throw new \OutOfBoundsException('Cannot read 32-bit int, it exceeds the boundary of the file');
         }
 
-        $segment = substr($br->getInputString(), $br->getPosition(), 1);
+        $segment = fread($br->getInputHandle(), 1);
 
         $data = unpack($this->getEndian(), $segment);
         $data = $data[1];

src/Type/String.php

@@ -24,7 +24,7 @@ public function read(BinaryReader &$br, $length)
             throw new \OutOfBoundsException('Cannot read string, it exceeds the boundary of the file');
         }
 
-        $str = substr($br->getInputString(), $br->getPosition(), $length);
+        $str = fread($br->getInputHandle(), $length);
         $br->setPosition($br->getPosition() + $length);
 
         return $str;

test/AbstractTestCase.php

@@ -0,0 +1,27 @@
+<?php
+
+namespace PhpBinaryReader;
+
+class AbstractTestCase extends \PHPUnit_Framework_TestCase
+{
+    // Data provider for creating both File and String based binary readers.
+    public function binaryReaders()
+    {
+        $fileBig = __DIR__ . '/asset/testfile-big.bin';
+        $fileLittle = __DIR__ . '/asset/testfile-little.bin';
+
+        $dataBig = fopen($fileBig, 'rb');
+        $dataLittle = fopen($fileLittle, 'rb');
+
+        $brBigFile = new BinaryReader(fopen($fileBig, 'rb'), Endian::ENDIAN_BIG);
+        $brLittleFile = new BinaryReader(fopen($fileLittle, 'rb'), Endian::ENDIAN_LITTLE);
+
+        $brBigStr = new BinaryReader(file_get_contents($fileBig), Endian::ENDIAN_BIG);
+        $brLittleStr = new BinaryReader(file_get_contents($fileLittle), Endian::ENDIAN_LITTLE);
+
+        return [
+            [$brBigFile, $brLittleFile],
+            [$brBigStr, $brLittleStr],
+        ];
+    }
+}