initial commit, authentication and user creation

This commit is contained in:
Colin McKechney
2023-04-25 05:44:08 +00:00
commit 1986941c23
3 changed files with 900 additions and 0 deletions

97
src/main.rs Normal file
View File

@@ -0,0 +1,97 @@
use sha2::{Sha256, Digest};
use rand::{prelude::Rng, distributions::Alphanumeric };
use oracle::{Connection, Error};
use log::{info, warn, error};
use env_logger::Env;
static SQL_USERNAME: &str = "group09_user";
static SQL_PASSWORD: &str = "group09_user";
static SALT_LEN: usize = 16;
fn main() {
//init logging system
let env = Env::default().filter_or("LOG_LEVEL", "info");
env_logger::init_from_env(env);
//Temporary for testing purposes, should write something to make a random salt
let username = "cmckechn";
let password = "password";
//proof of concept tests, create_user should fail in this instance because user was already
//created
println!("{}",authenticate(username, password).unwrap());
create_user("test", "test_create", "test_first", "test_last").unwrap();
println!("{}", authenticate("test", "test_create").unwrap());
}
fn authenticate(username: &str, password: &str) -> Result<bool, Error> {
info!("Authenticating user: {}", username);
let conn = Connection::connect(SQL_USERNAME,SQL_PASSWORD, "")?;
let mut stmt = conn.statement("select password, salt from student where net_id = :1").build()?;
let row = stmt.query_row_as::<(String, String)>(&[&username])?;
let true_pword = row.0;
let salt = row.1;
let mut hasher = Sha256::new();
hasher.update(password);
hasher.update(salt);
let hash = hasher.finalize();
let mut tmp: String = String::new();
for value in hash{
tmp += &format!("{:x}", value);
}
conn.close()?;
if true_pword.eq(&tmp) {
info!("User {} successfully authenticated", username);
Ok(true)
}
else{
warn!("User {} failed authentication", username);
Ok(false)
}
}
fn create_user(username: &str, password: &str, first_name: &str, last_name: &str) -> Result<(), Error> {
info!("Creating user: {}", username);
let conn = Connection::connect(SQL_USERNAME, SQL_PASSWORD, "")?;
let mut stmt = conn.statement("insert into student values(:net_id, :first_name, :last_name, :password, :salt)").build()?;
let salt: String = rand::thread_rng().sample_iter(&Alphanumeric).take(SALT_LEN).map(char::from).collect();
let mut hasher = Sha256::new();
hasher.update(&password);
hasher.update(&salt);
let hash = hasher.finalize();
let mut hash_string = String::new();
for value in hash{
hash_string += &format!("{:x}", value);
}
match stmt.execute_named(&[("net_id", &username), ("first_name", &first_name), ("last_name", &last_name), ("password", &hash_string), ("salt", &salt)]) {
Ok(_) => {
info!("User {} successfully created", username);
conn.commit()?;
},
Err(_) => {
warn!("Failed to create user {}", username);
conn.rollback()?;
}
};
conn.close()?;
Ok(())
}