Rust: Expand tokio-postgres sources.

Author: geoffw0

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

@@ -21,3 +21,5 @@ extensions:
     data:
       - ["<tokio_postgres::row::Row>::get", "ReturnValue", "database", "manual"]
       - ["<tokio_postgres::row::Row>::try_get", "ReturnValue.Field[core::result::Result::Ok(0)]", "database", "manual"]
+      - ["<tokio_postgres::row::SimpleQueryRow>::get", "ReturnValue.Field[core::option::Option::Some(0)]", "database", "manual"]
+      - ["<tokio_postgres::row::SimpleQueryRow>::try_get", "ReturnValue.Field[core::result::Result::Ok(0)].Field[core::option::Option::Some(0)]", "database", "manual"]

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

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