Add #result for Return Refinements

liquidjava-example/src/main/java/testSuite/CorrectResult.java

@@ -0,0 +1,16 @@
+package testSuite;
+
+import liquidjava.specification.Refinement;
+
+public class CorrectResult {
+
+    @Refinement("$result > 10")
+    public int getLargeNumber() {
+        return 15;
+    }
+
+    @Refinement("$result == (a + b)")
+    public int sum(int a, int b) {
+        return a + b;
+    }
+}

liquidjava-example/src/main/java/testSuite/ErrorResultVariable.java

@@ -0,0 +1,10 @@
+// Not Found Error
+package testSuite;
+import liquidjava.specification.Refinement;
+
+public class ErrorResultVariable {
+    public void test() {
+        @Refinement("$result > 0") 
+        int x = 10; 
+    }
+}

liquidjava-verifier/src/main/antlr4/rj/grammar/RJ.g4

@@ -36,6 +36,7 @@ operand:
 literalExpression:
 		'(' literalExpression ')'	#litGroup
 	|	literal						#lit
+	|   RESULT                      #result
 	| 	ID 							#var
 	|	ID '.' functionCall			#targetInvocation
 	|	functionCall				#invocation
@@ -91,7 +92,8 @@ ARITHOP : '+'|'*'|'/'|'%';//|'-';
 BOOL    : 'true' | 'false';
 ID_UPPER: ([A-Z][a-zA-Z0-9]*);
 OBJECT_TYPE:
-		  (([a-zA-Z][a-zA-Z0-9]+) ('.' [a-zA-Z][a-zA-Z0-9]*)+);
+		  (([a-zA-Z][a-zA-Z0-9]+) ('.' [a-zA-Z][a-zA-Z0-9]*)+);	  
+RESULT  : '$result';		  
 ID     	: '#'*[a-zA-Z_][a-zA-Z0-9_#]*;
 STRING  : '"'(~["])*'"';
 INT     : 	(([0-9]+) |	([0-9]+('_'[0-9]+)*));

liquidjava-verifier/src/main/java/liquidjava/processor/refinement_checker/general_checkers/MethodsFunctionsChecker.java

@@ -161,6 +161,7 @@ private Predicate handleFunctionRefinements(RefinedFunction f, CtElement method,
         Optional<Predicate> oret = rtc.getRefinementFromAnnotation(method);
         Predicate ret = oret.orElse(new Predicate());
         ret = ret.substituteVariable("return", Keys.WILDCARD);
+        ret = ret.substituteVariable("$result", Keys.WILDCARD);// added for refinement
         f.setRefReturn(ret);
         return Predicate.createConjunction(joint, ret);
     }
@@ -180,6 +181,13 @@ public <R> void getReturnRefinements(CtReturn<R> ret) throws LJError {
                 RefinedFunction fi = rtc.getContext().getFunction(method.getSimpleName(),
                         ((CtClass<?>) method.getParent()).getQualifiedName(), method.getParameters().size());
 
+                if (fi == null)
+                    return;// null check refinement
+
+                if (fi.getRefReturn() == null) {
+                    return;
+                }
+
                 List<Variable> lv = fi.getArguments();
                 for (Variable v : lv) {
                     rtc.getContext().addVarToContext(v);

liquidjava-verifier/src/main/java/liquidjava/rj_language/visitors/CreateASTVisitor.java

@@ -49,6 +49,7 @@
 import rj.grammar.RJParser.PredLogicContext;
 import rj.grammar.RJParser.PredNegateContext;
 import rj.grammar.RJParser.ProgContext;
+import rj.grammar.RJParser.ResultContext;
 import rj.grammar.RJParser.StartContext;
 import rj.grammar.RJParser.StartPredContext;
 import rj.grammar.RJParser.TargetInvocationContext;
@@ -159,9 +160,18 @@ else if (rc instanceof VarContext) {
         } else if (rc instanceof TargetInvocationContext) {
             // TODO Finish Invocation with Target (a.len())
             return null;
-        } else {
+        }
+        // else {
+        // return create(((InvocationContext) rc).functionCall());
+        // }
+        else if (rc instanceof ResultContext) {
+            return new Var("$result");
+        } else if (rc instanceof InvocationContext) {
             return create(((InvocationContext) rc).functionCall());
+        } else {
+            throw new IllegalStateException("Unknown literalExpression: " + rc.getClass());
         }
+
     }
 
     private Expression functionCallCreate(FunctionCallContext rc) throws LJError {