dubey/WebGoat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed #45 - multiple tracker for one user fixed

Browse Source
This commit is contained in:
TortugaAttack 2019-08-21 23:38:27 +02:00
parent 6d36e7db74
commit f0d1555a09
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
webgoat-container/src/main/java/org/owasp/webgoat/users/UserService.java
View File

@ -30,14 +30,26 @@ public class UserService implements UserDetailsService {
} }
public void addUser(String username, String password) { public void addUser(String username, String password) {
//get user if there exists one by the name
WebGoatUser webGoatUser = userRepository.findByUsername(username);
//if user exists it will be updated, otherwise created
userRepository.save(new WebGoatUser(username, password)); userRepository.save(new WebGoatUser(username, password));
//if user previously existed it will not get another tracker
if (webGoatUser == null) {
userTrackerRepository.save(new UserTracker(username)); userTrackerRepository.save(new UserTracker(username));
} }
}
public void addUser(String username, String password, String role) { public void addUser(String username, String password, String role) {
//get user if there exists one by the name
WebGoatUser webGoatUser = userRepository.findByUsername(username);
//if user exists it will be updated, otherwise created
userRepository.save(new WebGoatUser(username,password,role)); userRepository.save(new WebGoatUser(username,password,role));
//if user previously existed it will not get another tracker
if (webGoatUser == null) {
userTrackerRepository.save(new UserTracker(username)); userTrackerRepository.save(new UserTracker(username));
} }
}
public List<WebGoatUser> getAllUsers () { public List<WebGoatUser> getAllUsers () {
return userRepository.findAll(); return userRepository.findAll();