summaryrefslogtreecommitdiffhomepage
path: root/app/src/main/java/com/wireguard
diff options
context:
space:
mode:
Diffstat (limited to 'app/src/main/java/com/wireguard')
-rw-r--r--app/src/main/java/com/wireguard/android/ProfileService.java118
1 files changed, 68 insertions, 50 deletions
diff --git a/app/src/main/java/com/wireguard/android/ProfileService.java b/app/src/main/java/com/wireguard/android/ProfileService.java
index 1f40a0a1..984cf4b1 100644
--- a/app/src/main/java/com/wireguard/android/ProfileService.java
+++ b/app/src/main/java/com/wireguard/android/ProfileService.java
@@ -54,46 +54,6 @@ public class ProfileService extends Service {
return START_STICKY;
}
- private class ProfileAdder extends AsyncTask<Void, Void, Boolean> {
- private final Profile profile;
- private final boolean shouldConnect;
-
- private ProfileAdder(Profile profile, boolean shouldConnect) {
- super();
- if (profiles.get(profile.getName()) != null)
- throw new IllegalStateException("Profile already exists: " + profile.getName());
- this.profile = profile;
- this.shouldConnect = shouldConnect;
- }
-
- @Override
- protected Boolean doInBackground(Void... voids) {
- Log.i(TAG, "Adding profile " + profile.getName());
- try {
- final String configFile = profile.getName() + ".conf";
- if (new File(getFilesDir(), configFile).exists())
- throw new IOException("Refusing to overwrite existing profile configuration");
- final FileOutputStream stream = openFileOutput(configFile, MODE_PRIVATE);
- stream.write(profile.toString().getBytes(StandardCharsets.UTF_8));
- stream.close();
- return true;
- } catch (IOException e) {
- Log.e(TAG, "Could not create profile " + profile.getName(), e);
- return false;
- }
- }
-
- @Override
- protected void onPostExecute(Boolean result) {
- if (!result)
- return;
- profile.setIsConnected(false);
- profiles.put(profile.getName(), profile);
- if (shouldConnect)
- new ProfileConnecter(profile).execute();
- }
- }
-
private class ProfileConnecter extends AsyncTask<Void, Void, Boolean> {
private final Profile profile;
@@ -184,14 +144,10 @@ public class ProfileService extends Service {
private class ProfileRemover extends AsyncTask<Void, Void, Boolean> {
private final Profile profile;
- private final Profile replaceWith;
- private final boolean shouldConnect;
- private ProfileRemover(Profile profile, Profile replaceWith, Boolean shouldConnect) {
+ private ProfileRemover(Profile profile) {
super();
this.profile = profile;
- this.replaceWith = replaceWith;
- this.shouldConnect = shouldConnect != null ? shouldConnect : false;
}
@Override
@@ -211,8 +167,70 @@ public class ProfileService extends Service {
if (!result)
return;
profiles.remove(profile.getName());
- if (replaceWith != null)
- new ProfileAdder(replaceWith, shouldConnect).execute();
+ }
+ }
+
+ private class ProfileUpdater extends AsyncTask<Void, Void, Boolean> {
+ private final String newName;
+ private Profile newProfile;
+ private final String oldName;
+ private final Boolean shouldConnect;
+
+ private ProfileUpdater(String oldName, Profile newProfile, Boolean shouldConnect) {
+ super();
+ this.newName = newProfile.getName();
+ this.newProfile = newProfile;
+ this.oldName = oldName;
+ this.shouldConnect = shouldConnect;
+ if (profiles.values().contains(newProfile))
+ throw new IllegalArgumentException("Profile " + newName + " modified directly");
+ if (!newName.equals(oldName) && profiles.get(newName) != null)
+ throw new IllegalStateException("Profile " + newName + " already exists");
+ }
+
+ @Override
+ protected Boolean doInBackground(Void... voids) {
+ Log.i(TAG, (oldName == null ? "Adding" : "Updating") + " profile " + newName);
+ final File newFile = new File(getFilesDir(), newName + ".conf");
+ final File oldFile = new File(getFilesDir(), oldName + ".conf");
+ if (!newName.equals(oldName) && newFile.exists()) {
+ Log.w(TAG, "Refusing to overwrite existing profile configuration");
+ return false;
+ }
+ try {
+ final FileOutputStream stream = openFileOutput(oldFile.getName(), MODE_PRIVATE);
+ stream.write(newProfile.toString().getBytes(StandardCharsets.UTF_8));
+ stream.close();
+ } catch (IOException e) {
+ Log.e(TAG, "Could not save configuration for profile " + oldName, e);
+ return false;
+ }
+ if (!newName.equals(oldName) && !oldFile.renameTo(newFile)) {
+ Log.e(TAG, "Could not rename " + oldFile.getName() + " to " + newFile.getName());
+ return false;
+ }
+ return true;
+ }
+
+ @Override
+ protected void onPostExecute(Boolean result) {
+ if (!result)
+ return;
+ final Profile oldProfile = profiles.remove(oldName);
+ if (oldProfile != null) {
+ try {
+ oldProfile.parseFrom(newProfile);
+ oldProfile.setName(newName);
+ newProfile = oldProfile;
+ } catch (IOException e) {
+ Log.e(TAG, "Could not replace profile " + oldName + " with " + newName, e);
+ return;
+ }
+ }
+ newProfile.setIsConnected(false);
+ profiles.put(newName, newProfile);
+ if (shouldConnect)
+ new ProfileConnecter(newProfile).execute();
}
}
@@ -251,7 +269,7 @@ public class ProfileService extends Service {
return;
if (profile.getIsConnected())
new ProfileDisconnecter(profile).execute();
- new ProfileRemover(profile, null, null).execute();
+ new ProfileRemover(profile).execute();
}
@Override
@@ -263,9 +281,9 @@ public class ProfileService extends Service {
final boolean wasConnected = oldProfile.getIsConnected();
if (wasConnected)
new ProfileDisconnecter(oldProfile).execute();
- new ProfileRemover(oldProfile, newProfile, wasConnected).execute();
+ new ProfileUpdater(oldName, newProfile, wasConnected).execute();
} else {
- new ProfileAdder(newProfile, false).execute();
+ new ProfileUpdater(null, newProfile, false).execute();
}
}
}