surrealpatch/lib/fuzz/fuzz_targets/fuzz_executor.rs

#![no_main]

use libfuzzer_sys::fuzz_target;

fuzz_target!(|commands: &str| {
	let commands: Vec<&str> = commands.split_inclusive(";").collect();
	let blacklisted_command_strings = ["sleep", "SLEEP"];

	use surrealdb::{dbs::Session, kvs::Datastore};
	let max_commands = 500;
	if commands.len() > max_commands {
		return;
	}

	tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap().block_on(async {
		let dbs = Datastore::new("memory").await.unwrap();
		let ses = Session::for_kv().with_ns("test").with_db("test");
		for command in commands.iter() {
			for blacklisted_string in blacklisted_command_strings.iter() {
				if command.contains(blacklisted_string) {
					return;
				}
			}
			let _ignore_the_result = dbs.execute(command, &ses, None, false).await;

			// TODO: Add some async timeout and `tokio::select!` between it and the query
			// Alternatively, wrap future in `tokio::time::Timeout`.
		}
	})
});