X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/98711170da204fa86f2889672770c6a3feb4c4db..bf998c10baabbd95f2d59d29f9edea0eb107b156:/Driver.hpp

diff --git a/Driver.hpp b/Driver.hpp
index efd2bc6..adb03f2 100644
--- a/Driver.hpp
+++ b/Driver.hpp
@@ -39,7 +39,6 @@ struct CYScript;
 struct CYWord;
 
 enum CYMark {
-    CYMarkIgnore,
     CYMarkScript,
     CYMarkModule,
 };
@@ -60,12 +59,12 @@ class _visible CYDriver {
 
     std::stack<CYClassTail *> class_;
 
+    CYMark mark_;
+    int hold_;
     bool newline_;
     bool last_;
-    bool next_;
 
-    std::istream &data_;
-    CYMark mark_;
+    std::streambuf &data_;
 
     int debug_;
     bool strict_;
@@ -121,10 +120,10 @@ class _visible CYDriver {
     void ScannerDestroy();
 
   public:
-    CYDriver(CYPool &pool, std::istream &data, const std::string &filename = "");
+    CYDriver(CYPool &pool, std::streambuf &data, const std::string &filename = "");
     ~CYDriver();
 
-    bool Parse(CYMark mark = CYMarkScript);
+    bool Parse(CYMark mark = CYMarkModule);
     void Replace(CYOptions &options);
 
     void SetRegEx(bool equal);