From 54714331e41eceb2591f704748d652e0054dd388 Mon Sep 17 00:00:00 2001
From: Tobias Brunner <tobias@strongswan.org>
Date: Wed, 21 Jun 2017 16:22:11 +0200
Subject: [PATCH] android: Add ability to remove a range set from another

---
 .../strongswan/android/utils/IPRangeSet.java  | 16 +++++++++++++++
 .../android/test/IPRangeSetTest.java          | 20 +++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/src/frontends/android/app/src/main/java/org/strongswan/android/utils/IPRangeSet.java b/src/frontends/android/app/src/main/java/org/strongswan/android/utils/IPRangeSet.java
index e6884c65ec..3749ac02c2 100644
--- a/src/frontends/android/app/src/main/java/org/strongswan/android/utils/IPRangeSet.java
+++ b/src/frontends/android/app/src/main/java/org/strongswan/android/utils/IPRangeSet.java
@@ -108,6 +108,22 @@ public class IPRangeSet
 		mRanges.addAll(additions);
 	}
 
+	/**
+	 * Remove the given ranges from ranges in this set.
+	 */
+	public void remove(IPRangeSet ranges)
+	{
+		if (ranges == this)
+		{
+			mRanges.clear();
+			return;
+		}
+		for (IPRange range : ranges.mRanges)
+		{
+			remove(range);
+		}
+	}
+
 	/**
 	 * Returns the subnets derived from all the ranges in this set.
 	 */
diff --git a/src/frontends/android/app/src/test/java/org/strongswan/android/test/IPRangeSetTest.java b/src/frontends/android/app/src/test/java/org/strongswan/android/test/IPRangeSetTest.java
index 3f0e33e62e..6c6e1593f8 100644
--- a/src/frontends/android/app/src/test/java/org/strongswan/android/test/IPRangeSetTest.java
+++ b/src/frontends/android/app/src/test/java/org/strongswan/android/test/IPRangeSetTest.java
@@ -166,6 +166,26 @@ public class IPRangeSetTest
 		assertSubnets(set.getSubnets(), new IPRange("192.168.1.0/25"), new IPRange("192.168.2.128/25"));
 	}
 
+	@Test
+	public void testRemoveRangesIdent() throws UnknownHostException
+	{
+		IPRangeSet set = IPRangeSet.fromString("192.168.1.0/24 192.168.4.0/24");
+		set.remove(set);
+		assertEquals("size", 0, set.size());
+		assertSubnets(set.getSubnets());
+	}
+
+	@Test
+	public void testRemoveRanges() throws UnknownHostException
+	{
+		IPRangeSet set = IPRangeSet.fromString("192.168.0.0/16");
+		IPRangeSet remove = IPRangeSet.fromString("192.168.1.0/24 192.168.3.0/24 192.168.16.0-192.168.255.255");
+		set.remove(remove);
+		assertEquals("size", 3, set.size());
+		assertSubnets(set.getSubnets(), new IPRange("192.168.0.0/24"), new IPRange("192.168.2.0/24"),
+					  new IPRange("192.168.4.0/22"), new IPRange("192.168.8.0/21"));
+	}
+
 	@Test
 	public void testFromStringSingle() throws UnknownHostException
 	{