1 files changed, 17 insertions, 5 deletions
diff --git a/src/nix-log2xml/log2xml.cc b/src/nix-log2xml/log2xml.cc
index 9be3978a9274..8a3cbc8c42cd 100644
--- a/src/nix-log2xml/log2xml.cc
+++ b/src/nix-log2xml/log2xml.cc
@@ -34,6 +34,8 @@ struct Decoder
     void pushChar(char c);
 
     void finishLine();
+
+    void decodeFile(istream & st);
 };
 
 
@@ -158,16 +160,26 @@ void Decoder::finishLine()
 }
 
 
-int main(int argc, char * * argv)
+void Decoder::decodeFile(istream & st)
 {
-    Decoder dec;
     int c;
-
+    
     cout << "<logfile>" << endl;
     
-    while ((c = getchar()) != EOF) {
-        dec.pushChar(c);
+    while ((c = st.get()) != EOF) {
+        pushChar(c);
     }
 
+    if (line.size()) finishLine();
+
+    while (level--) cout << "</nest>" << endl;
+    
     cout << "</logfile>" << endl;
 }
+
+
+int main(int argc, char * * argv)
+{
+    Decoder dec;
+    dec.decodeFile(cin);
+}