src/sqlancer/postgres/ast/PostgresBinaryComparisonOperation.java - master - NikolayIvkin/sqlancer2

src/sqlancer/postgres/ast/PostgresBinaryComparisonOperation.java
139 строк5 KB
Manuel Rigger
[Postgres] Let getExpectedValue() return null when no PQS implementation is available
25 авг 2020, 14:08
25 авг 2020, 14:08205b394
package sqlancer.postgres.ast;

import sqlancer.Randomly;
import sqlancer.common.ast.BinaryOperatorNode;
import sqlancer.common.ast.BinaryOperatorNode.Operator;
import sqlancer.postgres.PostgresSchema.PostgresDataType;
import sqlancer.postgres.ast.PostgresBinaryComparisonOperation.PostgresBinaryComparisonOperator;

public class PostgresBinaryComparisonOperation
        extends BinaryOperatorNode<PostgresExpression, PostgresBinaryComparisonOperator> implements PostgresExpression {

    public enum PostgresBinaryComparisonOperator implements Operator {
        EQUALS("=") {
            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                return leftVal.isEquals(rightVal);
            }
        },
        IS_DISTINCT("IS DISTINCT FROM") {
            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                return PostgresConstant
                        .createBooleanConstant(!IS_NOT_DISTINCT.getExpectedValue(leftVal, rightVal).asBoolean());
            }
        },
        IS_NOT_DISTINCT("IS NOT DISTINCT FROM") {
            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                if (leftVal.isNull()) {
                    return PostgresConstant.createBooleanConstant(rightVal.isNull());
                } else if (rightVal.isNull()) {
                    return PostgresConstant.createFalse();
                } else {
                    return leftVal.isEquals(rightVal);
                }
            }
        },
        NOT_EQUALS("!=") {
            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                PostgresConstant isEquals = leftVal.isEquals(rightVal);
                if (isEquals.isBoolean()) {
                    return PostgresConstant.createBooleanConstant(!isEquals.asBoolean());
                }
                return isEquals;
            }
        },
        LESS("<") {

            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                return leftVal.isLessThan(rightVal);
            }
        },
        LESS_EQUALS("<=") {

            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                PostgresConstant lessThan = leftVal.isLessThan(rightVal);
                if (lessThan.isBoolean() && !lessThan.asBoolean()) {
                    return leftVal.isEquals(rightVal);
                } else {
                    return lessThan;
                }
            }
        },
        GREATER(">") {
            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                PostgresConstant equals = leftVal.isEquals(rightVal);
                if (equals.isBoolean() && equals.asBoolean()) {
                    return PostgresConstant.createFalse();
                } else {
                    PostgresConstant applyLess = leftVal.isLessThan(rightVal);
                    if (applyLess.isNull()) {
                        return PostgresConstant.createNullConstant();
                    }
                    return PostgresPrefixOperation.PrefixOperator.NOT.getExpectedValue(applyLess);
                }
            }
        },
        GREATER_EQUALS(">=") {

            @Override
            public PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal) {
                PostgresConstant equals = leftVal.isEquals(rightVal);
                if (equals.isBoolean() && equals.asBoolean()) {
                    return PostgresConstant.createTrue();
                } else {
                    PostgresConstant applyLess = leftVal.isLessThan(rightVal);
                    if (applyLess.isNull()) {
                        return PostgresConstant.createNullConstant();
                    }
                    return PostgresPrefixOperation.PrefixOperator.NOT.getExpectedValue(applyLess);
                }
            }

        };

        private final String textRepresentation;

        @Override
        public String getTextRepresentation() {
            return textRepresentation;
        }

        PostgresBinaryComparisonOperator(String textRepresentation) {
            this.textRepresentation = textRepresentation;
        }

        public abstract PostgresConstant getExpectedValue(PostgresConstant leftVal, PostgresConstant rightVal);

        public static PostgresBinaryComparisonOperator getRandom() {
            return Randomly.fromOptions(PostgresBinaryComparisonOperator.values());
        }

    }

    public PostgresBinaryComparisonOperation(PostgresExpression left, PostgresExpression right,
            PostgresBinaryComparisonOperator op) {
        super(left, right, op);
    }

    @Override
    public PostgresConstant getExpectedValue() {
        PostgresConstant leftExpectedValue = getLeft().getExpectedValue();
        PostgresConstant rightExpectedValue = getRight().getExpectedValue();
        if (leftExpectedValue == null || rightExpectedValue == null) {
            return null;
        }
        return getOp().getExpectedValue(leftExpectedValue, rightExpectedValue);
    }

    @Override
    public PostgresDataType getExpressionType() {
        return PostgresDataType.BOOLEAN;
    }

}