feat: basic fuzzing

Author: Randoooom

.dockerignore

@@ -4,6 +4,7 @@ lib/
 target/
 common/target/
 codegen/target/
+fuzz/target/
 docs/
 Makefile.toml
 dashboard/.nuxt/

.github/workflows/nightly.yml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Install toolchain
-        uses: actions-rs/toolchain@v1
+        uses: actions-rust-lang/setup-rust-toolchain@v1
 
       - name: Checkout Sources
         uses: actions/checkout@v4

Cargo.lock

@@ -74,6 +74,7 @@ paste = "1.0.14"
 rand = "0.8.5"
 reqwest = { version = "0.12.3", features = ["json"] }
 test-log = "0.2.15"
+feedback_fusion_common = { path = "./common", features = ["arbitrary"] }
 
 [features]
 default = ["all-databases", "otlp", "caching-skytable"]
@@ -95,3 +96,6 @@ mssql = ["rbdc-mssql"]
 [[bench]]
 name = "grpc"
 harness = false
+
+[workspace]
+members = [".", "fuzz", "common", "codegen"]

Cargo.toml

@@ -1,10 +1,11 @@
 FROM rust:slim AS build
 
+COPY ./rust-toolchain.toml ./rust-toolchain.toml
+
 RUN apt-get update \ 
   && apt-get install libssl-dev protobuf-compiler libprotobuf-dev pkg-config -y --no-install-recommends \
   && apt-get clean \
-  && rustup toolchain install stable \
-  && rustup default stable
+  && rustup update
 
 ARG features=all-databases,otlp
 

Dockerfile

@@ -1,3 +1,5 @@
+.PHONY: fuzz
+
 pnpm:
 	pnpm i -C lib 
 	pnpm i -C docs
@@ -83,6 +85,9 @@ integration_test:
 	OIDC_PROVIDER="http://localhost:5151" OIDC_CLIENT_ID="client" OIDC_CLIENT_SECRET="secret" RUST_LOG="INFO" GRPC_ENDPOINT="http://localhost:8000" \
 	cargo test --no-fail-fast --test integration_test
 
+fuzz:
+	OIDC_PROVIDER="http://localhost:5151" OIDC_CLIENT_ID="client" OIDC_CLIENT_SECRET="secret" RUST_LOG="INFO" GRPC_ENDPOINT="http://localhost:8000" cargo fuzz run fuzz_create_and_export
+
 cleanup:
 	docker rm -f database;docker rm -f oidc-server-mock;docker rm -f feedback-fusion;docker rm -f skytable;docker network rm feedback-fusion; echo ""
 

Makefile

@@ -6,11 +6,17 @@ edition = "2021"
 [dependencies]
 feedback_fusion_codegen = { path = "../codegen" }
 
+arbitrary = { version = "1.4.1", features = ["derive"], optional = true }
+lazy_static = { version = "1.5.0", optional = true }
 rbatis = "4.5.21"
 tonic = "0.12.0"
+openidconnect = { version = "3.5.0", optional = true }
 prost = "0.13.0"
 prost-types = "0.13.0"
 validator = { version = "0.18.1", features = ["derive"] }
 
 [build-dependencies]
 tonic-build = "0.12.0"
+
+[features]
+arbitrary = ["dep:arbitrary", "dep:lazy_static", "dep:openidconnect"]

common/Cargo.toml

@@ -26,6 +26,58 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let out_dir = PathBuf::from(std::env::var("OUT_DIR").unwrap());
 
     tonic_build::configure()
+        .type_attribute(
+            "CreateTargetRequest",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "CreatePromptRequest",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "CreateFieldRequest",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "FieldOptions.options",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "FieldOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "FieldType",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "TextOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "RatingOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "CheckboxOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "SelectionOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "RangeOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "NumberOptions",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
+        .type_attribute(
+            "CheckboxStyle",
+            r#"#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]"#,
+        )
         .type_attribute("CreateTargetRequest", "#[derive(validator::Validate)]")
         .field_attribute("CreateTargetRequest.name", "#[validate(length(max = 255))]")
         .type_attribute("UpdateTargetRequest", "#[derive(validator::Validate)]")

common/build.rs

@@ -20,6 +20,9 @@
 //DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 //OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
+#[cfg(feature = "arbitrary")]
+pub mod tests;
+
 pub mod proto {
     tonic::include_proto!("feedback_fusion_v1");
     pub const FILE_DESCRIPTOR_SET: &[u8] =

common/src/lib.rs

@@ -1,4 +1,4 @@
-//SPDX-FileCopyrightText: 2024 OneLiteFeatherNet
+//SPDX-FileCopyrightText: 2025 OneLiteFeatherNet
 //SPDX-License-Identifier: MIT
 
 //MIT License
@@ -58,19 +58,19 @@ pub async fn authenticate() -> String {
 #[macro_export]
 macro_rules! connect {
     () => {{
-        let channel = tonic::transport::Channel::from_static(&crate::common::GRPC_ENDPOINT).connect().await.unwrap();
-        let token: tonic::metadata::MetadataValue<_> = format!("Bearer {}", crate::common::authenticate().await).parse().unwrap();
+        let channel = tonic::transport::Channel::from_static(&$crate::tests::GRPC_ENDPOINT).connect().await.unwrap();
+        let token: tonic::metadata::MetadataValue<_> = format!("Bearer {}", $crate::tests::authenticate().await).parse().unwrap();
 
         let client =
-            feedback_fusion_common::proto::feedback_fusion_v1_client::FeedbackFusionV1Client::with_interceptor(channel, move |mut request: tonic::Request<()>| {
+            $crate::proto::feedback_fusion_v1_client::FeedbackFusionV1Client::with_interceptor(channel, move |mut request: tonic::Request<()>| {
                 request
                     .metadata_mut()
                     .insert("authorization", token.clone());
 
                 Ok(request)
             });
 
-        let public_client = feedback_fusion_common::proto::public_feedback_fusion_v1_client::PublicFeedbackFusionV1Client::connect(crate::common::GRPC_ENDPOINT.as_str())
+        let public_client = $crate::proto::public_feedback_fusion_v1_client::PublicFeedbackFusionV1Client::connect($crate::tests::GRPC_ENDPOINT.as_str())
             .await
             .unwrap();