Multiple attacks logging for Hpfeeds.

aau-network-security · Aug 13, 2020 · df02055 · df02055
1 parent 96b77e4
commit df02055
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/src/main/java/de/tudarmstadt/informatik/hostage/commons/JSONHelper.java b/src/main/java/de/tudarmstadt/informatik/hostage/commons/JSONHelper.java
@@ -8,6 +8,7 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.OutputStreamWriter;
+import java.util.ArrayList;
 
 import de.tudarmstadt.informatik.hostage.logging.RecordAll;
 import de.tudarmstadt.informatik.hostage.ui.activity.MainActivity;
@@ -18,15 +19,17 @@ public class JSONHelper {
     File file = new File("/data/data/" + MainActivity.getContext().getPackageName() + "/" + PERSIST_FILENAME);
 
 
-    public void persistData(RecordAll record){
+    public void persistData(ArrayList<RecordAll> records){
         try {
             int BUFFER_SIZE = 8192;
             String UTF8 = "utf8";
             FileOutputStream fout = new FileOutputStream(file);
             BufferedWriter fnw = new BufferedWriter(new OutputStreamWriter(fout, UTF8), BUFFER_SIZE);
 
             JSONArray arr = new JSONArray();
-            arr.put(record.toJSON());
+            for(RecordAll record: records) {
+                arr.put(record.toJSON());
+            }
             fnw.write(arr.toString());
             fnw.close();
             fout.close();
@@ -37,7 +40,6 @@ public void persistData(RecordAll record){
     }
 
     public String getFilePath(){
-        System.out.println("Filepath!!! "+file.getAbsolutePath());
         return file.getAbsolutePath();
     }
 }
diff --git a/src/main/java/de/tudarmstadt/informatik/hostage/hpfeeds/publisher/PublishHelper.java b/src/main/java/de/tudarmstadt/informatik/hostage/hpfeeds/publisher/PublishHelper.java
@@ -1,7 +1,7 @@
 package de.tudarmstadt.informatik.hostage.hpfeeds.publisher;
 
-import java.io.File;
 import java.io.IOException;
+import java.util.ArrayList;
 
 import de.tudarmstadt.informatik.hostage.HostageApplication;
 import de.tudarmstadt.informatik.hostage.commons.JSONHelper;
@@ -14,9 +14,9 @@ public class PublishHelper {
     private DaoSession dbSession;
     private DAOHelper daoHelper;
     private int offset=0;
-    private int limit=10;
+    private int limit=20;
     private int attackRecordOffset=0;
-    private int attackRecordLimit=9999;
+    private int attackRecordLimit=999;
     LogFilter filter = null;
     JSONHelper jsonHelper = new JSONHelper();
 
@@ -49,8 +49,7 @@ private void publisher() throws Hpfeeds.ReadTimeOutException, Hpfeeds.EOSExcepti
 
     }
 
-    private RecordAll getLastInsertedRecord(){
-        RecordAll recordAll = daoHelper.getAttackRecordDAO().getRecordsForFilter(filter,offset,limit,attackRecordOffset,attackRecordLimit).get(0);
-        return recordAll;
+    private ArrayList<RecordAll> getLastInsertedRecord(){
+        return  daoHelper.getAttackRecordDAO().getRecordsForFilter(filter,offset,limit,attackRecordOffset,attackRecordLimit);
     }
 }