Rust: Fix a couple more gaps.

Author: geoffw0

rust/ql/lib/codeql/rust/frameworks/postgres.model.yml

@@ -13,6 +13,7 @@ extensions:
       - ["<postgres::client::Client>::query_raw", "Argument[0]", "sql-injection", "manual"]
       - ["<postgres::client::Client>::query_typed", "Argument[0]", "sql-injection", "manual"]
       - ["<postgres::client::Client>::query_typed_raw", "Argument[0]", "sql-injection", "manual"]
+      - ["<postgres::client::Client>::simple_query", "Argument[0]", "sql-injection", "manual"]
   - addsTo:
       pack: codeql/rust-all
       extensible: sourceModel

rust/ql/lib/codeql/rust/frameworks/tokio-postgres.model.yml

@@ -9,6 +9,7 @@ extensions:
       - ["<tokio_postgres::client::Client>::prepare", "Argument[0]", "sql-injection", "manual"]
       - ["<tokio_postgres::client::Client>::prepare_typed", "Argument[0]", "sql-injection", "manual"]
       - ["<tokio_postgres::client::Client>::query", "Argument[0]", "sql-injection", "manual"]
+      - ["<tokio_postgres::client::Client>::query_one", "Argument[0]", "sql-injection", "manual"]
       - ["<tokio_postgres::client::Client>::query_opt", "Argument[0]", "sql-injection", "manual"]
       - ["<tokio_postgres::client::Client>::query_raw", "Argument[0]", "sql-injection", "manual"]
       - ["<tokio_postgres::client::Client>::query_typed", "Argument[0]", "sql-injection", "manual"]

rust/ql/test/library-tests/frameworks/postgres/main.rs

@@ -41,7 +41,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         println!("found person: {} {} {}", id, name, age);
     }
 
-    for message in &conn.simple_query("SELECT id, name, age FROM person")? {  // $ MISSING: sql-sink
+    for message in &conn.simple_query("SELECT id, name, age FROM person")? {  // $ sql-sink
         if let postgres::SimpleQueryMessage::Row(row) = message {
             let id: i32 = row.get(0).unwrap().parse().unwrap(); // $ database-read
             let name: &str = row.get(1).unwrap(); // $ database-read