From a0638d4da1fbb6eb78fc149a0bde9730e8bbcb66 Mon Sep 17 00:00:00 2001
From: sirjonasxx <36828922+sirjonasxx@users.noreply.github.com>
Date: Wed, 20 Jun 2018 19:10:58 +0200
Subject: [PATCH] fix for windows

---
 G-Earth.iml                                   | 11 ++++
 .../hostreplacer/UnixHostReplacer.java        | 59 +++++++++++++------
 2 files changed, 52 insertions(+), 18 deletions(-)
 create mode 100644 G-Earth.iml

diff --git a/G-Earth.iml b/G-Earth.iml
new file mode 100644
index 0000000..c90834f
--- /dev/null
+++ b/G-Earth.iml
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/src/main/protocol/hostreplacer/UnixHostReplacer.java b/src/main/protocol/hostreplacer/UnixHostReplacer.java
index 6436e92..f4fb5dd 100644
--- a/src/main/protocol/hostreplacer/UnixHostReplacer.java
+++ b/src/main/protocol/hostreplacer/UnixHostReplacer.java
@@ -16,14 +16,19 @@ class UnixHostReplacer implements HostReplacer {
 
     @Override
     public void addRedirect(String original, String redirect) {
-        String text = redirect + " " + original;
+        String text = redirect + " " + original + "\t# G-Earth replacement";
+
+        FileReader fr = null;
+        BufferedReader br = null;
+        FileWriter fw = null;
+        BufferedWriter out = null;
 
         try
         {
             ArrayList<String> lines = new ArrayList<String>();
             File f1 = new File(hostsFileLocation);
-            FileReader fr = new FileReader(f1);
-            BufferedReader br = new BufferedReader(fr);
+            fr = new FileReader(f1);
+            br = new BufferedReader(fr);
             String line = null;
             boolean containmmm = false;
             while ((line = br.readLine()) != null)
@@ -36,59 +41,77 @@ class UnixHostReplacer implements HostReplacer {
             fr.close();
             br.close();
 
-            FileWriter fw = new FileWriter(f1);
-            BufferedWriter out = new BufferedWriter(fw);
+            fw = new FileWriter(f1);
+            out = new BufferedWriter(fw);
 
             if (!containmmm)	{
                 out.write(text);
             }
 
             for (int i = 0; i < lines.size(); i++)	{
-                out.write("\n"+ lines.get(i));
+                out.write(System.getProperty("line.separator")+ lines.get(i));
             }
-
-            out.flush();
-            out.close();
         }
         catch (Exception ex)
         {
             ex.printStackTrace();
         }
+        finally {
+            try {
+                if (fr != null) fr.close();
+                if (br != null) br.close();
+                if (fw != null) fw.close();
+                if (out != null) out.close();
+            }
+            catch (Exception e) {}
+        }
     }
 
     @Override
     public void removeRedirect(String original, String redirect) {
-        String text = redirect + " " + original;
+        String text = redirect + " " + original + "\t# G-Earth replacement";
+
+        FileReader fr = null;
+        BufferedReader br = null;
+        FileWriter fw = null;
+        BufferedWriter out = null;
 
         try
         {
             ArrayList<String> lines = new ArrayList<String>();
             File f1 = new File(hostsFileLocation);
-            FileReader fr = new FileReader(f1);
-            BufferedReader br = new BufferedReader(fr);
+            fr = new FileReader(f1);
+            br = new BufferedReader(fr);
             String line = null;
             while ((line = br.readLine()) != null)
             {
-                if (!line.contains(text))
+                if (!line.equals(text))
                     lines.add(line);
 
             }
             fr.close();
             br.close();
 
-            FileWriter fw = new FileWriter(f1);
-            BufferedWriter out = new BufferedWriter(fw);
+            fw = new FileWriter(f1);
+            out = new BufferedWriter(fw);
 
             for (int i = 0; i < lines.size(); i++)	{
                 out.write(lines.get(i));
-                if (i != lines.size() - 1) out.write("\n");
+                if (i != lines.size() - 1) out.write(System.getProperty("line.separator"));
             }
-            out.flush();
-            out.close();
         }
         catch (Exception ex)
         {
             ex.printStackTrace();
         }
+        finally {
+            try {
+                if (fr != null) fr.close();
+                if (br != null) br.close();
+                if (fw != null) fw.close();
+                if (out != null) out.close();
+            }
+            catch (Exception e) {}
+        }
     }
 }