mirror of
https://github.com/chatmail/core.git
synced 2026-04-19 06:26:30 +03:00
b6b0849bce
Add a trait for str.strdup() to replace to_cstring() which avoid the signature ambiguity with .to_string(). Also instruduce CString::yolo() as a shortcut to CString::new().unwrap() and use it whenever the variable does can be deallocated by going out of scope. This is less error prone. Use some Path.to_c_string() functions where possible.
66 lines
1.6 KiB
Rust
66 lines
1.6 KiB
Rust
use crate::context::Context;
|
|
use crate::dc_tools::*;
|
|
use crate::sql;
|
|
|
|
// Token namespaces
|
|
#[allow(non_camel_case_types)]
|
|
pub type dc_tokennamespc_t = usize;
|
|
pub const DC_TOKEN_AUTH: dc_tokennamespc_t = 110;
|
|
pub const DC_TOKEN_INVITENUMBER: dc_tokennamespc_t = 100;
|
|
|
|
// Functions to read/write token from/to the database. A token is any string associated with a key.
|
|
|
|
pub fn dc_token_save(
|
|
context: &Context,
|
|
namespc: dc_tokennamespc_t,
|
|
foreign_id: u32,
|
|
token: *const libc::c_char,
|
|
) -> bool {
|
|
if token.is_null() {
|
|
return false;
|
|
}
|
|
// foreign_id may be 0
|
|
sql::execute(
|
|
context,
|
|
&context.sql,
|
|
"INSERT INTO tokens (namespc, foreign_id, token, timestamp) VALUES (?, ?, ?, ?);",
|
|
params![namespc as i32, foreign_id as i32, as_str(token), time()],
|
|
)
|
|
.is_ok()
|
|
}
|
|
|
|
pub fn dc_token_lookup(
|
|
context: &Context,
|
|
namespc: dc_tokennamespc_t,
|
|
foreign_id: u32,
|
|
) -> *mut libc::c_char {
|
|
context
|
|
.sql
|
|
.query_row_col::<_, String>(
|
|
context,
|
|
"SELECT token FROM tokens WHERE namespc=? AND foreign_id=?;",
|
|
params![namespc as i32, foreign_id as i32],
|
|
0,
|
|
)
|
|
.map(|s| unsafe { s.strdup() })
|
|
.unwrap_or_else(|| std::ptr::null_mut())
|
|
}
|
|
|
|
pub fn dc_token_exists(
|
|
context: &Context,
|
|
namespc: dc_tokennamespc_t,
|
|
token: *const libc::c_char,
|
|
) -> bool {
|
|
if token.is_null() {
|
|
return false;
|
|
}
|
|
|
|
context
|
|
.sql
|
|
.exists(
|
|
"SELECT id FROM tokens WHERE namespc=? AND token=?;",
|
|
params![namespc as i32, as_str(token)],
|
|
)
|
|
.unwrap_or_default()
|
|
}
|