Enhance the Python IO interface

Related to #832

api/python/PE/objects/pyParser.cpp

@@ -14,7 +14,9 @@
  * limitations under the License.
  */
 #include "pyPE.hpp"
+#include "logging.hpp"
 
+#include "PyIOStream.hpp"
 #include "LIEF/PE/Parser.hpp"
 
 #include <string>
@@ -39,39 +41,13 @@ void create<Parser>(py::module& m) {
 
 
     m.def("parse",
-      [] (py::object byteio, const std::string& name) {
-        const auto& io = py::module::import("io");
-        const auto& RawIOBase = io.attr("RawIOBase");
-        const auto& BufferedIOBase = io.attr("BufferedIOBase");
-        const auto& TextIOBase = io.attr("TextIOBase");
-
-        py::object rawio;
-
-
-        if (py::isinstance(byteio, RawIOBase)) {
-          rawio = byteio;
+      [] (py::object byteio, const std::string& name) -> py::object {
+        if (auto stream = PyIOStream::from_python(byteio)) {
+          auto ptr = std::make_unique<PyIOStream>(std::move(*stream));
+          return py::cast(PE::Parser::parse(std::move(ptr), name));
         }
-
-        else if (py::isinstance(byteio, BufferedIOBase)) {
-          rawio = byteio.attr("raw");
-        }
-
-        else if (py::isinstance(byteio, TextIOBase)) {
-          rawio = byteio.attr("buffer").attr("raw");
-        }
-
-        else {
-          throw py::type_error(py::repr(byteio).cast<std::string>().c_str());
-        }
-
-        std::string raw_str = static_cast<py::bytes>(rawio.attr("readall")());
-
-        std::vector<uint8_t> raw = {
-          std::make_move_iterator(std::begin(raw_str)),
-          std::make_move_iterator(std::end(raw_str))
-        };
-
-        return LIEF::PE::Parser::parse(std::move(raw), name);
+        LIEF_ERR("Can't create a LIEF stream interface over the provided io");
+        return py::none();
       },
       "Parse the PE binary from the given Python IO interface and return a :class:`lief.PE.Binary` object",
       "io"_a, "name"_a = "",