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

YML config should override lesson class

Browse Source
This commit is contained in:
Nanne Baars 2016-06-29 12:08:13 +02:00
parent 0a26e05ca5
commit c350e86772
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
webgoat-container/src/main/java/org/owasp/webgoat/controller/StartLesson.java
View File

@ -80,7 +80,7 @@ public class StartLesson {
public ModelAndView lessonPage(HttpServletRequest request) {
// I will set here the thymeleaf fragment location based on the resource requested.
ModelAndView model = new ModelAndView();
SecurityContext context = SecurityContextHolder.getContext();
SecurityContext context = SecurityContextHolder.getContext(); //TODO this should work with the security roles of Spring
GrantedAuthority authority = context.getAuthentication().getAuthorities().iterator().next();
String path = request.getServletPath(); // we now got /a/b/c/AccessControlMatrix.lesson
String lessonName = path.substring(path.lastIndexOf('/') + 1, path.indexOf(".lesson"));
@ -88,8 +88,7 @@ public class StartLesson {
List<AbstractLesson> lessons = ws.getCourse()
.getLessons(ws, AbstractLesson.USER_ROLE);//TODO this should work with the security roles of Spring
Optional<AbstractLesson> lesson = lessons.stream()
.filter(l -> l instanceof YmlBasedLesson)
.filter(l -> ((YmlBasedLesson) l).getHtml().equals(lessonName))
.filter(l -> l.getHtml().equals(lessonName))
.findFirst();
model.setViewName("lesson_content");
model.addObject("lesson", lesson.get());

5
webgoat-container/src/main/java/org/owasp/webgoat/plugins/Plugin.java
View File

@ -68,9 +68,7 @@ public class Plugin {
}
//New code all lessons should work as below
if (this.lesson == null) {
readYmlLessonConfiguration();
}
readYmlLessonConfiguration();
}
private void readYmlLessonConfiguration() {
@ -87,6 +85,7 @@ public class Plugin {
final String title = (String) lessonYml.get("title");
final String html = (String) lessonYml.get("html");
this.ymlBasedLesson = new YmlBasedLesson(category, hints, title, html);
this.lesson = null;
} catch (IOException e) {
throw new PluginLoadingFailure("Unable to read yml file", e);
}