summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorJesse Luehrs <doy@tozt.net>2019-04-13 04:30:50 -0400
committerJesse Luehrs <doy@tozt.net>2019-04-13 04:30:50 -0400
commit01e62e9baaf5860f0b9dee39e985a89333c0a854 (patch)
tree1305097c8943d56edfc7cf9a06016a30fb1ce195 /src
parentc48a2b579673d4f4034bb7275165b72077e60fbd (diff)
downloadmatasano-01e62e9baaf5860f0b9dee39e985a89333c0a854.tar.gz
matasano-01e62e9baaf5860f0b9dee39e985a89333c0a854.zip
adjust the signature of create_query_string a bit
Diffstat (limited to 'src')
-rw-r--r--src/http.rs14
1 files changed, 7 insertions, 7 deletions
diff --git a/src/http.rs b/src/http.rs
index 10b0eca..359502f 100644
--- a/src/http.rs
+++ b/src/http.rs
@@ -25,7 +25,7 @@ pub fn parse_query_string(string: &str) -> Option<HashMap<&str, &str>> {
return Some(map);
}
-pub fn create_query_string(params: HashMap<&str, &str>) -> String {
+pub fn create_query_string(params: &HashMap<&str, String>) -> String {
fn escape(s: &str) -> String {
s.replace("%", "%25")
.replace("&", "%26")
@@ -57,10 +57,10 @@ fn test_parse_query_string() {
#[test]
fn test_create_query_string() {
let mut params = HashMap::new();
- params.insert("foo", "bar");
- params.insert("baz", "qux");
- params.insert("zap", "zazzle");
- let got = create_query_string(params);
+ params.insert("foo", "bar".to_string());
+ params.insert("baz", "qux".to_string());
+ params.insert("zap", "zazzle".to_string());
+ let got = create_query_string(&params);
let expected1 = "foo=bar&baz=qux&zap=zazzle";
let expected2 = "foo=bar&zap=zazzle&baz=qux";
let expected3 = "baz=qux&foo=bar&zap=zazzle";
@@ -82,8 +82,8 @@ fn test_create_query_string() {
#[test]
fn test_create_query_string_malicious() {
let mut params = HashMap::new();
- params.insert("email", "foo@bar.com&role=admin");
- let got = create_query_string(params);
+ params.insert("email", "foo@bar.com&role=admin".to_string());
+ let got = create_query_string(&params);
let expected = "email=foo@bar.com%26role%3Dadmin";
assert_eq!(got, expected);
}