*fixed a bug with while and do statements.

HISTORY

@@ -1,6 +1,7 @@
 Jan. 25 2016
 Fixed a memory leak when printing a referenced usertype
 Fixed a bug with nested IF statement
+Fixed a bug with WHILE and DO statements when returning from a sub routine
 Fixed an indexing bug with list
 Added a new sample script source file
 

core/my_basic.c

@@ -12380,7 +12380,12 @@ _loop_begin:
 				result = MB_FUNC_OK;
 
 				goto _exit;
-			} else if(result != MB_FUNC_OK && result != MB_SUB_RETURN) { /* Normally */
+			} else if(result == MB_SUB_RETURN) { /* RETURN */
+				if(ast) ast = ast->prev;
+				if(ast) ast = ast->prev;
+
+				goto _exit;
+			} else if(result != MB_FUNC_OK) { /* Normally */
 				goto _exit;
 			}
 
@@ -12454,7 +12459,12 @@ _loop_begin:
 			result = MB_FUNC_OK;
 
 			goto _exit;
-		} else if(result != MB_FUNC_OK && result != MB_SUB_RETURN) { /* Normally */
+		} else if(result == MB_SUB_RETURN) { /* RETURN */
+			if(ast) ast = ast->prev;
+			if(ast) ast = ast->prev;
+
+			goto _exit;
+		} else if(result != MB_FUNC_OK) { /* Normally */
 			goto _exit;
 		}