fix a bug

include/file_parser.h

@@ -17,9 +17,11 @@ struct RegexEntry {
   string handler;
 
   friend inline istream& operator>>(istream& is, RegexEntry& t) {
-    is >> t.regex;
     char buffer[256];
     is.getline(buffer, 256);
+    string buffer_tmp(buffer);
+    t.regex = buffer_tmp;
+    is.getline(buffer, 256);
     while (string(buffer).substr(0, 4) != "====") {
       t.handler.append(buffer);
       is.getline(buffer, 256);
@@ -28,12 +30,14 @@ struct RegexEntry {
   }
 
   friend inline ostream& operator<<(ostream& os, const RegexEntry& t) {
+    os << "/*" << endl;
     os << "====" << endl;
-    os << "regex: " << t.regex << endl;
+    os << "regex: \'" << t.regex << "'"<< endl;
     os << "priority: " << t.priority << endl;
     os << "hander: " << endl;
     os << t.handler << endl;
     os << "====" << endl;
+    os << "*/" << endl;
     return os;
   }
 };

include/mylex.h

@@ -16,6 +16,7 @@
 #include <string>
 #include <iostream>
 #include <fstream>
+#include <sstream>
 #include <boost/foreach.hpp>
 
 #endif

src/dfa.cpp

@@ -287,6 +287,8 @@ void DFA::_c_state_change(ostream& os, DFAState* state) {
   }
   cout   << "    default:" << endl;
   if (state->is_end) {
+    cout << "      fseek(shm_file, -1, SEEK_CUR);" << endl;
+    cout << "      buffer_index--;buffer[buffer_index] = '\\0';" << endl;
     _c_state_end(os, state->end_id);
     _c_state_reset(os);
   } else {

src/file_parser.cpp

@@ -8,8 +8,11 @@ void RegexEntry::to_c(ostream& os) {
 
 FileParser::FileParser(string& filename) {
   ifstream ifs(filename.c_str());
+  char buffer[32];
+  ifs.getline(buffer, 32);
+  stringstream ss(buffer);
   int N;
-  ifs >> N;
+  ss >> N;
   for (int i = 0; i < N; i++) {
     RegexEntry re;
     ifs >> re;