Logger.java 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. package de.tudarmstadt.informatik.hostage.logging;
  2. import android.app.IntentService;
  3. import android.content.Context;
  4. import android.content.Intent;
  5. import android.os.Parcelable;
  6. import de.tudarmstadt.informatik.hostage.Services.MultiStage;
  7. import de.tudarmstadt.informatik.hostage.logging.MessageRecord.TYPE;
  8. import de.tudarmstadt.informatik.hostage.persistence.HostageDBOpenHelper;
  9. /**
  10. * The Logger is used to write the database in dedicated worker threads and implements a message queue.
  11. * @author Mihai Plasoianu
  12. * @author Lars Pandikow
  13. *
  14. */
  15. public class Logger extends IntentService {
  16. private static final String ACTION_LOG_MESSAGE = "de.tudarmstadt.informatik.hostage.action.LOG_MESSAGE";
  17. private static final String ACTION_LOG_ATTACK = "de.tudarmstadt.informatik.hostage.action.LOG_ATTACK";
  18. private static final String ACTION_LOG_NETWORK = "de.tudarmstadt.informatik.hostage.action.LOG_NETWORK";
  19. private static final String ACTION_LOG_PORTSCAN = "de.tudarmstadt.informatik.hostage.action.LOG_PORTSCAN";
  20. private static final String EXTRA_RECORD = "de.tudarmstadt.informatik.hostage.extra.RECORD";
  21. private static final String EXTRA_RECORD2 = "de.tudarmstadt.informatik.hostage.extra.RECORD2";
  22. private static final String EXTRA_TIMESTAMP = "de.tudarmstadt.informatik.hostage.extra.TIMESTAMP";
  23. /**
  24. * Adds a single MessageRecord to the Database.
  25. * @param context Context needed to access database
  26. * @param record The MessageRecord to be added
  27. */
  28. public static void log(Context context, MessageRecord record) {
  29. Intent intent = new Intent(context, Logger.class);
  30. intent.setAction(ACTION_LOG_MESSAGE);
  31. intent.putExtra(EXTRA_RECORD, (Parcelable)record);
  32. context.startService(intent);
  33. }
  34. /**
  35. * Adds a single AttackRecord to the Database.
  36. * @param context Context needed to access database
  37. * @param record The AttackRecord to be added
  38. */
  39. public static void log(Context context, AttackRecord record) {
  40. Intent intent = new Intent(context, Logger.class);
  41. intent.setAction(ACTION_LOG_ATTACK);
  42. intent.putExtra(EXTRA_RECORD, (Parcelable)record);
  43. context.startService(intent);
  44. }
  45. /**
  46. * Adds a single NetworkRecord to the Database.
  47. * @param context Context needed to access database
  48. * @param record The NetworkRecord to be added
  49. */
  50. public static void log(Context context, NetworkRecord record) {
  51. Intent intent = new Intent(context, Logger.class);
  52. intent.setAction(ACTION_LOG_NETWORK);
  53. intent.putExtra(EXTRA_RECORD, (Parcelable)record);
  54. context.startService(intent);
  55. }
  56. /**
  57. * Adds a port scan entry to the database
  58. * @param context Context needed to access database
  59. * @param attackRecord AttackRecord containing the attack information of the port scan
  60. * @param netRecord NetworkRecord containing the network information of the port scan
  61. * @param timestamp Timestamp of the port scan
  62. */
  63. public static void logPortscan(Context context, AttackRecord attackRecord, NetworkRecord netRecord, long timestamp){
  64. Intent intent = new Intent(context, Logger.class);
  65. intent.setAction(ACTION_LOG_PORTSCAN);
  66. intent.putExtra(EXTRA_RECORD, (Parcelable) attackRecord);
  67. intent.putExtra(EXTRA_RECORD2, (Parcelable) netRecord);
  68. intent.putExtra(EXTRA_TIMESTAMP, timestamp);
  69. context.startService(intent);
  70. }
  71. /**
  72. * Adds a File Injection entry to the database
  73. * @param context
  74. * @param attackRecord
  75. * @param netRecord
  76. * @param timestamp
  77. */
  78. public static void logFileInjection(Context context, AttackRecord attackRecord, NetworkRecord netRecord, long timestamp){
  79. Intent intent = new Intent(context, Logger.class);
  80. intent.setAction(ACTION_LOG_PORTSCAN);
  81. intent.putExtra(EXTRA_RECORD, (Parcelable)attackRecord);
  82. intent.putExtra(EXTRA_RECORD2, (Parcelable)netRecord);
  83. intent.putExtra(EXTRA_TIMESTAMP, timestamp);
  84. context.startService(intent);
  85. }
  86. /**
  87. * Adds a Multi Stage Attack entry to the database
  88. * @param context
  89. * @param attackRecord
  90. * @param networkRecord
  91. * @param timestamp
  92. */
  93. public static void logMultiStageAttack(Context context,AttackRecord attackRecord,NetworkRecord networkRecord, long timestamp){
  94. Intent intent = new Intent(context, Logger.class);
  95. intent.setAction(ACTION_LOG_PORTSCAN);
  96. intent.putExtra(EXTRA_RECORD, (Parcelable)attackRecord);
  97. intent.putExtra(EXTRA_RECORD2, (Parcelable)networkRecord);
  98. intent.putExtra(EXTRA_TIMESTAMP, timestamp);
  99. context.startService(intent);
  100. }
  101. private HostageDBOpenHelper mDbHelper;
  102. public Logger() {
  103. super("Logger");
  104. }
  105. @Override
  106. public void onCreate() {
  107. super.onCreate();
  108. mDbHelper = new HostageDBOpenHelper(getApplicationContext());
  109. }
  110. private void handleActionLog(MessageRecord record) {
  111. mDbHelper.addMessageRecord(record);
  112. }
  113. private void handleActionLog(AttackRecord record) {
  114. mDbHelper.addAttackRecord(record);
  115. mDbHelper.updateSyncAttackCounter(record);
  116. }
  117. private void handleActionLog(NetworkRecord record) {
  118. mDbHelper.updateNetworkInformation(record);
  119. }
  120. /**
  121. * Method to handle the Intent createt by the public interface.
  122. */
  123. @Override
  124. protected void onHandleIntent(Intent intent) {
  125. if (intent != null) {
  126. final String action = intent.getAction();
  127. if (ACTION_LOG_MESSAGE.equals(action)) {
  128. final MessageRecord record = intent.getParcelableExtra(EXTRA_RECORD);
  129. handleActionLog(record);
  130. }else if(ACTION_LOG_ATTACK.equals(action)){
  131. final AttackRecord record = intent.getParcelableExtra(EXTRA_RECORD);
  132. handleActionLog(record);
  133. }else if(ACTION_LOG_NETWORK.equals(action)){
  134. final NetworkRecord record = intent.getParcelableExtra(EXTRA_RECORD);
  135. handleActionLog(record);
  136. }else if(ACTION_LOG_PORTSCAN.equals(action)){
  137. final AttackRecord attackRecord = intent.getParcelableExtra(EXTRA_RECORD);
  138. final NetworkRecord networkRecord = intent.getParcelableExtra(EXTRA_RECORD2);
  139. MessageRecord messageRecord = new MessageRecord(true);
  140. messageRecord.setAttack_id(attackRecord.getAttack_id());
  141. //messageRecord.setId(0);
  142. messageRecord.setPacket("");
  143. messageRecord.setTimestamp(intent.getLongExtra(EXTRA_TIMESTAMP, 0));
  144. messageRecord.setType(TYPE.RECEIVE);
  145. handleActionLog(attackRecord);
  146. handleActionLog(networkRecord);
  147. handleActionLog(messageRecord);
  148. }
  149. }
  150. }
  151. }