From f6437cf28cc94d936d13c518a46e3587e61f25a6 Mon Sep 17 00:00:00 2001
From: lawson89 <>
Date: Mon, 11 Aug 2014 16:28:05 -0400
Subject: [PATCH] add new service to view source

---
 .../webgoat/lessons/model/SourceListing.java  | 31 +++++++++++++++++++
 .../owasp/webgoat/service/SourceService.java  |  7 +++--
 2 files changed, 36 insertions(+), 2 deletions(-)
 create mode 100644 java/org/owasp/webgoat/lessons/model/SourceListing.java

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;
     }
 
     /**