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

Merge pull request #643 from TortugaAttack/multipleTracker45

Browse Source 
Fixed #642 - multiple tracker for one user fixed
This commit is contained in:
René Zubcevic 2019-08-22 15:07:59 +02:00 committed by GitHub
commit c93563da3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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();