From cb794dcb509df05b3466a60af02ba23b46c1a95b Mon Sep 17 00:00:00 2001
From: "rogan.dawes" <rogan.dawes@4033779f-a91e-0410-96ef-6bf7bf53c507>
Date: Wed, 11 Jul 2007 12:52:23 +0000
Subject: [PATCH] Calculate the stage changes correctly

git-svn-id: http://webgoat.googlecode.com/svn/trunk@161 4033779f-a91e-0410-96ef-6bf7bf53c507
---
 .../JavaSource/org/owasp/webgoat/session/WebSession.java       | 2 +-
  webgoat/main/project/WebContent/main.jsp                      | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/WebSession.java b/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/WebSession.java
index 8602fbb26..d6b1711cc 100644
--- a/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/WebSession.java	
+++ b/ webgoat/main/project/JavaSource/org/owasp/webgoat/session/WebSession.java	
@@ -924,7 +924,7 @@ public class WebSession
 			RandomLessonAdapter rla = (RandomLessonAdapter) al;
 			int stage = myParser.getIntParameter(STAGE) - 1;
 			String[] stages = rla.getStages();
-			if (stage>0 && stage <= stages.length)
+			if (stage>=0 && stage < stages.length)
 				rla.setStage(this, stages[stage]);
 			} catch (ParameterNotFoundException pnfe) {}
 			}
diff --git a/ webgoat/main/project/WebContent/main.jsp b/ webgoat/main/project/WebContent/main.jsp
index a5071ea85..f06c6ccd3 100644
--- a/ webgoat/main/project/WebContent/main.jsp	
+++ b/ webgoat/main/project/WebContent/main.jsp	
@@ -200,7 +200,6 @@ StringBuffer buildList = new StringBuffer();
 				</div>
 				<%
 					AbstractLesson al = webSession.getCurrentLesson();
-					System.out.println("AL is a " + al.getClass().getName());
 					if (al instanceof SequentialLessonAdapter)
 					{
 					SequentialLessonAdapter sla = (SequentialLessonAdapter) al;
@@ -227,7 +226,7 @@ StringBuffer buildList = new StringBuffer();
 						<%
 						String stage = rla.getStage(webSession);
 						for (int i=0; i<stages.length;i++) {
-						%><option <% if (stages[i].equals(stage)) out.print("selected"); %> value="<%= i+1 %>">Stage <%= i+1 %></option>
+						%><option <% if (stages[i].equals(stage)) out.print("selected"); %> value="<%= i+1 %>"><%= stages[i] %><%= rla.getLessonTracker(webSession).hasCompleted(stages[i]) ? "*" : "" %></option>
 						<%
 						}
 						%></select></form><%