vyre 0.4.0

GPU compute intermediate representation with a standard operation library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

//! Catalog entry for `detect_url`.

use crate::ops::security_detection::detector_support::{spans, ByteSpan, DetectionError};

/// Embedded operation spec formerly stored in metadata/spec.toml.
pub const SPEC_TOML: &str = r#"schema_version = 1
id = "security_detection.detect_url"
archetype = "match-bytes-pattern"
display_name = "Detect URL"
summary = "Returns offset-length spans for URL-like byte sequences."
category = "C"

[intrinsic]
wgsl = "security_detection_detect_url"

[signature]
inputs = ["Bytes", "Bytes"]
output = "Bytes"

laws = []
equivalence_classes = ["http", "https", "missing_host_dot", "t47_cap"]
workgroup_size = [64, 1, 1]
tags = ["security-detection", "url", "ioc", "t47"]
fixtures_dir = "fixtures/"
"#;

/// Embedded reference vectors formerly stored in fixtures/reference-vectors.toml.
pub const REFERENCE_VECTORS_TOML: &str = r#"[[case]]
name = "positive_https"
input = "fetch https://example.com/a?b=c now"
expected_spans = [{ offset = 6, len = 25 }]

[[case]]
name = "negative_missing_dot"
input = "fetch http://localhost/path now"
expected_spans = []
"#;

/// WGSL lowering source for this detector.
pub mod lowering {
    /// Return the detector-specific WGSL source.
    #[must_use]
    pub const fn source() -> &'static str {
        r#"@compute @workgroup_size(64)
pub fn detect_url(@builtin(global_invocation_id) gid: vec3<u32>) {
    let i = gid.x;
    if (i + 7u >= input_len) { return; }
    let http = eq_ci(i, "http://");
    let https = eq_ci(i, "https://");
    if (!http && !https) { return; }
    let scheme_len = select(7u, 8u, https);
    var j = i + scheme_len;
    var dot = false;
    loop {
        if (j >= input_len || !is_uri_char(input[j])) { break; }
        if (input[j] == 46u) { dot = true; }
        j = j + 1u;
    }
    if (dot && j > i + scheme_len) { emit_span(i, j - i); }
}"#
    }
}

/// Return URL-like spans with offset and length.
///
/// # Errors
///
/// Returns `Fix: ...` when input exceeds 64 MiB.
pub fn detect_url(input: &[u8]) -> Result<Vec<ByteSpan>, DetectionError> {
    spans::url_spans(input)
}

/// Compatibility surface for the previous generated implementation module.
pub mod implementation {
    pub use super::detect_url;
    /// Compatibility module for callers that used the generated kernel path.
    pub mod kernel {
        pub use super::super::detect_url;
    }

    /// Compatibility module for callers that used the generated lowering path.
    pub mod lowering {
        /// Compatibility module for callers that used `implementation::lowering::wgsl`.
        pub mod wgsl {
            pub use super::super::super::lowering::source;
        }
    }
}