codeql/rust/ql/test/library-tests/dataflow/taint/main.rs

// Tests for taint flow.

fn source(i: i64) -> i64 {
    1000 + i
}

fn sink(s: i64) {
    println!("{}", s);
}

fn addition() {
    let a = source(42);
    sink(a + 1); // $ hasTaintFlow=42

    let mut b = source(58);
    b += 2;
    sink(b); // $ hasTaintFlow=58

    let mut c = 0;
    c += source(99);
    sink(c); // $ hasTaintFlow=99
}

fn negation() {
    let a = source(17);
    sink(-a); // $ hasTaintFlow=17
}

fn cast() {
    let a = source(77);
    let b = a as u8;
    sink(b as i64); // $ hasTaintFlow=77
}

mod string {
    fn source(i: i64) -> String {
        format!("{}", i)
    }

    fn sink(s: &str) {
        println!("{}", s);
    }

    pub fn string_slice() {
        let s = source(35);
        let sliced = &s[1..3];
        sink(sliced); // $ hasTaintFlow=35
    }
}

mod array_source {
    fn source(i: i64) -> [i64; 3] {
        [i; 3]
    }

    fn sink(i: i64) {
        println!("{}", i);
    }

    pub fn array_tainted() {
        let arr = source(76);
        sink(arr[1]); // $ hasTaintFlow=76
    }
}

mod array_sink {
    fn source(i: i64) -> i64 {
        i
    }

    fn sink(s: [i64; 3]) {
        println!("{}", s[1]);
    }

    pub fn array_with_taint() {
        let mut arr2 = [1, 2, 3];
        arr2[1] = source(36);
        sink(arr2); // $ hasTaintFlow=36
    }
}

use string::*;

mod tuples {
    fn source_string(i: i64) -> String {
        "".to_string()
    }

    fn source_tuple(i: i64) -> (String, String) {
        ("".to_string(), "".to_string())
    }

    fn sink<T>(t: T) {
    }

    pub fn tuples() {
        sink((source_string(1), "".to_string()));
        sink((source_string(1), "".to_string()).0); // $ hasValueFlow=1
        sink((source_string(1), "".to_string()).1);

        sink(source_tuple(2)); // $ hasValueFlow=2
        sink(source_tuple(2).0); // $ hasTaintFlow=2
        sink(source_tuple(2).1); // $ hasTaintFlow=2

        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())));
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).0);
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).0.0);
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).0.1); // $ hasValueFlow=3
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).1);
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).1.0);
        sink((("".to_string(), source_string(3)), ("".to_string(), "".to_string())).1.1);

        sink((source_tuple(4), ("".to_string(), "".to_string())));
        sink((source_tuple(4), ("".to_string(), "".to_string())).0); // $ hasValueFlow=4
        sink((source_tuple(4), ("".to_string(), "".to_string())).0.0); // $ hasTaintFlow=4
        sink((source_tuple(4), ("".to_string(), "".to_string())).0.1); // $ hasTaintFlow=4
        sink((source_tuple(4), ("".to_string(), "".to_string())).1);
        sink((source_tuple(4), ("".to_string(), "".to_string())).1.0);
        sink((source_tuple(4), ("".to_string(), "".to_string())).1.1);
    }
}

fn main() {
    addition();
    negation();
    cast();
    string_slice();
    array_source::array_tainted();
    array_sink::array_with_taint();
    tuples::tuples();
}