fuzz/fuzz_targets/fuzz_parse_roundtrip.rs

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#![no_main]

use libfuzzer_sys::fuzz_target;
use sqlparser::dialect::{
    AnsiDialect, BigQueryDialect, ClickHouseDialect, DatabricksDialect, Dialect, DuckDbDialect,
    GenericDialect, HiveDialect, MsSqlDialect, MySqlDialect, OracleDialect, PostgreSqlDialect,
    RedshiftSqlDialect, SQLiteDialect, SnowflakeDialect, SparkSqlDialect, TeradataDialect,
};
use sqlparser::parser::Parser;

fuzz_target!(|sql: &str| {
    let dialects: [(&str, &dyn Dialect); 16] = [
        ("ansi", &AnsiDialect {}),
        ("bigquery", &BigQueryDialect {}),
        ("clickhouse", &ClickHouseDialect {}),
        ("databricks", &DatabricksDialect {}),
        ("duckdb", &DuckDbDialect {}),
        ("generic", &GenericDialect {}),
        ("hive", &HiveDialect {}),
        ("mssql", &MsSqlDialect {}),
        ("mysql", &MySqlDialect {}),
        ("oracle", &OracleDialect {}),
        ("postgres", &PostgreSqlDialect {}),
        ("redshift", &RedshiftSqlDialect {}),
        ("sqlite", &SQLiteDialect {}),
        ("snowflake", &SnowflakeDialect {}),
        ("spark", &SparkSqlDialect {}),
        ("teradata", &TeradataDialect {}),
    ];
    for (name, dialect) in dialects {
        let Ok(statements) = Parser::parse_sql(dialect, sql) else {
            continue;
        };
        for statement in &statements {
            let rendered = statement.to_string();
            // SQL the AST renders must parse back. A failure is a Display bug
            if let Err(err) = Parser::parse_sql(dialect, &rendered) {
                panic!(
                    "{name}: displayed SQL failed to re-parse\n  display: {rendered}\n  error: {err}"
                );
            }
        }
    }
});