diff --git a/java/org/owasp/webgoat/lessons/model/SourceListing.java b/java/org/owasp/webgoat/lessons/model/SourceListing.java new file mode 100644 index 000000000..7833d2776 --- /dev/null +++ b/java/org/owasp/webgoat/lessons/model/SourceListing.java @@ -0,0 +1,31 @@ +/* + * To change this license header, choose License Headers in Project Properties. + * To change this template file, choose Tools | Templates + * and open the template in the editor. + */ + +package org.owasp.webgoat.lessons.model; + +/** + * + * @author rlawson + */ +public class SourceListing { + + private String source; + + /** + * @return the source + */ + public String getSource() { + return source; + } + + /** + * @param source the source to set + */ + public void setSource(String source) { + this.source = source; + } + +} diff --git a/java/org/owasp/webgoat/service/SourceService.java b/java/org/owasp/webgoat/service/SourceService.java index 7a7f0a4d9..5b3baf157 100644 --- a/java/org/owasp/webgoat/service/SourceService.java +++ b/java/org/owasp/webgoat/service/SourceService.java @@ -34,6 +34,7 @@ import javax.servlet.http.HttpSession; import static org.owasp.webgoat.LessonSource.END_SOURCE_SKIP; import static org.owasp.webgoat.LessonSource.START_SOURCE_SKIP; import org.owasp.webgoat.lessons.AbstractLesson; +import org.owasp.webgoat.lessons.model.SourceListing; import org.owasp.webgoat.session.Course; import org.owasp.webgoat.session.WebSession; import org.springframework.stereotype.Controller; @@ -55,10 +56,12 @@ public class SourceService extends BaseService { */ @RequestMapping(value = "/source.mvc", produces = "application/json") public @ResponseBody - String showSource(HttpSession session) { + SourceListing showSource(HttpSession session) { WebSession ws = getWebSesion(session); String source = getSource(ws); - return source; + SourceListing sl = new SourceListing(); + sl.setSource(source); + return sl; } /**