diff options
author | Anton Tananaev <anton.tananaev@gmail.com> | 2016-04-20 14:21:02 +1200 |
---|---|---|
committer | Anton Tananaev <anton.tananaev@gmail.com> | 2016-04-20 14:21:02 +1200 |
commit | a427f3f6644a62366e4c2ed6c69f2f4479b2bed5 (patch) | |
tree | 035f0954a7ebc8a3d18bfcc04819baced97a32b6 /src/org/traccar | |
parent | 4af79323eff0aa57c44e4069bcb46cb8388c9524 (diff) | |
download | traccar-server-a427f3f6644a62366e4c2ed6c69f2f4479b2bed5.tar.gz traccar-server-a427f3f6644a62366e4c2ed6c69f2f4479b2bed5.tar.bz2 traccar-server-a427f3f6644a62366e4c2ed6c69f2f4479b2bed5.zip |
Protect against cyclic hierarchy
Diffstat (limited to 'src/org/traccar')
-rw-r--r-- | src/org/traccar/database/DataManager.java | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/org/traccar/database/DataManager.java b/src/org/traccar/database/DataManager.java index 7baa1df47..eb0be9274 100644 --- a/src/org/traccar/database/DataManager.java +++ b/src/org/traccar/database/DataManager.java @@ -25,7 +25,9 @@ import java.util.ArrayList; import java.util.Collection; import java.util.Date; import java.util.HashMap; +import java.util.HashSet; import java.util.Map; +import java.util.Set; import java.util.concurrent.locks.ReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock; import javax.naming.InitialContext; @@ -391,7 +393,24 @@ public class DataManager implements IdentityManager { return groups; } + private void checkGroupCycles(Group group) { + groupsLock.readLock().lock(); + try { + Set<Long> groups = new HashSet<>(); + while (group != null) { + if (groups.contains(group.getId())) { + throw new IllegalArgumentException("Cycle in group hierarchy"); + } + groups.add(group.getId()); + group = groupsById.get(group.getGroupId()); + } + } finally { + groupsLock.readLock().unlock(); + } + } + public void addGroup(Group group) throws SQLException { + checkGroupCycles(group); group.setId(QueryBuilder.create(dataSource, getQuery("database.insertGroup"), true) .setObject(group) .executeUpdate()); @@ -399,6 +418,7 @@ public class DataManager implements IdentityManager { } public void updateGroup(Group group) throws SQLException { + checkGroupCycles(group); QueryBuilder.create(dataSource, getQuery("database.updateGroup")) .setObject(group) .executeUpdate(); |