add file black list

loopeer · Sep 5, 2016 · b9018d3 · b9018d3
1 parent 0924edf
commit b9018d3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/app/src/main/java/com/loopeer/codereader/ui/fragment/CodeReadFragment.java b/app/src/main/java/com/loopeer/codereader/ui/fragment/CodeReadFragment.java
@@ -189,7 +189,13 @@ public void call(Subscriber<? super String> subscriber) {
                 }
                 final InputStream finalStream = stream;
                 String[] names = mNode.name.split("\\.");
-                String jsFile = BrushMap.getJsFileForExtension(names[names.length - 1]);
+                String fileTypeName = names[names.length - 1];
+                if (BrushMap.isBlackFile(fileTypeName)) {
+                    subscriber.onError(new Throwable("Can not open this file!"));
+                    subscriber.onCompleted();
+                    return;
+                }
+                String jsFile = BrushMap.getJsFileForExtension(fileTypeName);
                 if (jsFile == null) {
                     jsFile = "Txt";
                 }

diff --git a/app/src/main/java/com/loopeer/codereader/utils/BrushMap.java b/app/src/main/java/com/loopeer/codereader/utils/BrushMap.java
@@ -1,6 +1,7 @@
 package com.loopeer.codereader.utils;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.List;
@@ -20,6 +21,13 @@ public BrushMap() {
         }
     }
 
+    public static final String[] FILE_BLACKLIST = new String[]{"hprof", "apk", "jar"};
+
+    public static boolean isBlackFile(String name) {
+        if (name != null) return Arrays.asList(FILE_BLACKLIST).contains(name.toLowerCase());
+        return false;
+    }
+
     public static void init() {
         List list = new ArrayList();
         list.add("actionscript3");