sqlancer.h2.H2Constant Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlancer Show documentation
SQLancer finds logic bugs in Database Management Systems through automatic testing
There is a newer version: 2.0.0
package sqlancer.h2;

import sqlancer.common.ast.newast.Node;

public class H2Constant implements Node {

    private H2Constant() {
    }

    public static class H2NullConstant extends H2Constant {

        @Override
        public String toString() {
            return "NULL";
        }

    }

    public static class H2IntConstant extends H2Constant {

        private final long value;

        public H2IntConstant(long value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.valueOf(value);
        }

    }

    public static class H2BoolConstant extends H2Constant {

        private final boolean value;

        public H2BoolConstant(boolean value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.valueOf(value);
        }

    }

    public static class H2StringConstant extends H2Constant {

        private final String value;

        public H2StringConstant(String value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.format("'%s'", value.replace("'", "''"));
        }

    }

    public static class H2DoubleConstant extends H2Constant {

        private final double value;

        public H2DoubleConstant(double value) {
            this.value = value;
        }

        @Override
        public String toString() {
            if (value == Double.POSITIVE_INFINITY) {
                return "POWER(0, -1)";
            } else if (value == Double.NEGATIVE_INFINITY) {
                return "(-POWER(0, -1))";
            } else if (Double.compare(value, -0.0) == 0) {
                return "(-CAST(0 AS DOUBLE))";
            } else if (Double.isNaN(value)) {
                return "SQRT(-1)";
            } else {
                return String.valueOf(value);
            }
        }

    }

    public static class H2BinaryConstant extends H2Constant {

        private String value;

        public H2BinaryConstant(long value) {
            this.value = Long.toHexString(value);
            if (this.value.length() % 2 == 1) {
                this.value = '0' + this.value; // pad with leading zero if needed
            }
        }

        public String getValue() {
            return value;
        }

        @Override
        public String toString() {
            return "X'" + value + "'";
        }

    }

    public static Node createIntConstant(long val) {
        return new H2IntConstant(val);
    }

    public static Node createNullConstant() {
        return new H2NullConstant();
    }

    public static Node createBoolConstant(boolean val) {
        return new H2BoolConstant(val);
    }

    public static Node createStringConstant(String val) {
        return new H2StringConstant(val);
    }

    public static Node createDoubleConstant(double val) {
        return new H2DoubleConstant(val);
    }

    public static Node createBinaryConstant(long val) {
        return new H2BinaryConstant(val);
    }

}