Generate a notification when an update includes new tests

Author: SandroMachado

app/src/main/AndroidManifest.xml

@@ -11,16 +11,31 @@
         android:icon="@drawable/ic_launcher"
         android:label="@string/app_name"
         android:theme="@style/AppTheme" >
+
         <activity
             android:name="fuzion24.device.vulnerability.test.ui.MainActivity"
             android:configChanges="orientation|screenSize|keyboardHidden"
+            android:launchMode="singleTask"
             android:label="@string/app_name" >
             <intent-filter>
                 <action android:name="android.intent.action.MAIN" />
 
                 <category android:name="android.intent.category.LAUNCHER" />
             </intent-filter>
         </activity>
+
+        <receiver android:name="fuzion24.device.vulnerability.broadcastreceiver.ApplicationUpdateBroadcastReceiver">
+            <intent-filter>
+                <action android:name="android.intent.action.PACKAGE_REPLACED"/>
+                <data android:scheme="package"  />
+            </intent-filter>
+
+            <intent-filter>
+                <action android:name="android.intent.action.PACKAGE_ADDED"/>
+                <data android:scheme="package"  />
+            </intent-filter>
+        </receiver>
+
     </application>
 
 </manifest>

app/src/main/java/fuzion24/device/vulnerability/broadcastreceiver/ApplicationUpdateBroadcastReceiver.java

@@ -0,0 +1,83 @@
+package fuzion24.device.vulnerability.broadcastreceiver;
+
+import android.app.NotificationManager;
+import android.app.PendingIntent;
+import android.content.BroadcastReceiver;
+import android.content.Context;
+import android.content.Intent;
+import android.support.v4.app.NotificationCompat;
+import android.util.Log;
+
+import com.nowsecure.android.vts.R;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import fuzion24.device.vulnerability.test.ui.MainActivity;
+import fuzion24.device.vulnerability.util.SharedPreferencesUtils;
+import fuzion24.device.vulnerability.vulnerabilities.VulnerabilityOrganizer;
+import fuzion24.device.vulnerability.vulnerabilities.VulnerabilityTest;
+
+/**
+ * Handles the application update/installation to show a notification to the user if there are
+ * new vulnerabilities scans available.
+ */
+
+public class ApplicationUpdateBroadcastReceiver extends BroadcastReceiver {
+
+    private static final String TAG = "VULN_TEST_BROAD_UPDATE";
+
+    private static final int NOTIFICATION_ID = 1;
+
+    @Override
+    public void onReceive(final Context context, final Intent intent) {
+        List<String> currentTestsAvailable = new ArrayList<>();
+
+        for (VulnerabilityTest test : VulnerabilityOrganizer.getTests(context)) {
+            currentTestsAvailable.add(test.getCVEorID());
+        }
+
+        List<String> testsAvailableOnPreviousBuild = SharedPreferencesUtils.getTheListOfScansAvailable(context);
+
+        // First run, save the current list of scans.
+        if (testsAvailableOnPreviousBuild == null) {
+            Log.d(TAG, "This is the  first time this detection is running.");
+            SharedPreferencesUtils.setTheListOfScansAvailable(context, currentTestsAvailable);
+
+            return;
+        }
+
+        // Update application with the tests available.
+        SharedPreferencesUtils.setTheListOfScansAvailable(context, currentTestsAvailable);
+
+        for (String testName : currentTestsAvailable) {
+            if (!testsAvailableOnPreviousBuild.contains(testName)) {
+
+                // This test is new, show a notification to the user.
+                buildNotification(context);
+                Log.d(TAG, String.format("New test available:, %s", testName));
+
+                return;
+            }
+        }
+
+        Log.d(TAG, "Without new tests detected.");
+    }
+
+    private void buildNotification(Context context) {
+        NotificationCompat.Builder builder = new NotificationCompat.Builder(context).setSmallIcon(R.drawable.ic_notification)
+            .setAutoCancel(true)
+            .setContentTitle(context.getString(R.string.app_name))
+            .setContentText(context.getString(R.string.notification_new_tests));
+
+        Intent notificationIntent = new Intent(context, MainActivity.class);
+
+        PendingIntent contentIntent = PendingIntent.getActivity(context, 0, notificationIntent, PendingIntent.FLAG_UPDATE_CURRENT);
+        builder.setContentIntent(contentIntent);
+
+        // Add the notification.
+        NotificationManager manager = (NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE);
+        manager.notify(NOTIFICATION_ID, builder.build());
+    }
+
+}

app/src/main/java/fuzion24/device/vulnerability/util/SharedPreferencesUtils.java

@@ -0,0 +1,42 @@
+package fuzion24.device.vulnerability.util;
+
+import android.content.Context;
+import android.content.SharedPreferences;
+
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+public class SharedPreferencesUtils {
+
+    private static final String PREFS_LAST_LIST_OF_SCANS = "PREFS_LIST_OF_SCANS";
+
+
+    private static final String PREFS_NAME = "com.nowsecure.android.vts.PREFERENCE_FILE_KEY";
+
+    private static SharedPreferences getPreferences(Context context) {
+        return context.getSharedPreferences(PREFS_NAME, Context.MODE_PRIVATE);
+    }
+
+    public static List<String> getTheListOfScansAvailable(Context context) {
+        Set<String> listOfPreviousScans = getPreferences(context).getStringSet(PREFS_LAST_LIST_OF_SCANS, null);
+
+        if (listOfPreviousScans == null) {
+            return null;
+        }
+
+        List<String> listOfScans = Arrays.asList(listOfPreviousScans.toArray(new String[listOfPreviousScans.size()]));
+
+        return listOfScans;
+    }
+
+    public static void setTheListOfScansAvailable(Context context, List<String> listOfScans) {
+        SharedPreferences.Editor editor = getPreferences(context).edit();
+
+        editor.putStringSet(PREFS_LAST_LIST_OF_SCANS, new HashSet<>(listOfScans));
+
+        editor.apply();
+    }
+
+}

app/src/main/res/drawable-hdpi/ic_notification.png

@@ -34,5 +34,6 @@
     <string name="patches">Patches</string>
     <string name="cvssv2">CVSSV2: </string>
     <string name="cve_date">CVE Date: </string>
+    <string name="notification_new_tests">New tests are available</string>
 
 </resources>