diff --git a/cli/src/invocation/mod.rs b/cli/src/invocation/mod.rs index 194fc9cfc..685c96cf2 100644 --- a/cli/src/invocation/mod.rs +++ b/cli/src/invocation/mod.rs @@ -108,20 +108,20 @@ fn get_replica(settings: &Config) -> Fallible { fn get_server(settings: &Config) -> Fallible> { // if server_client_id and server_origin are both set, use // the remote server - if let (Ok(client_id), Ok(origin)) = ( + if let (Ok(client_key), Ok(origin)) = ( settings.get_str("server_client_id"), settings.get_str("server_origin"), ) { - let client_id = Uuid::parse_str(&client_id)?; + let client_key = Uuid::parse_str(&client_key)?; let encryption_secret = settings .get_str("encryption_secret") .map_err(|_| format_err!("Could not read `encryption_secret` configuration"))?; log::debug!("Using sync-server with origin {}", origin); - log::debug!("Sync client ID: {}", client_id); + log::debug!("Sync client ID: {}", client_key); Ok(server::from_config(ServerConfig::Remote { origin, - client_id, + client_key, encryption_secret: encryption_secret.as_bytes().to_vec(), })?) } else { diff --git a/docs/src/sync-protocol.md b/docs/src/sync-protocol.md index 9a5caa247..e96f7cc83 100644 --- a/docs/src/sync-protocol.md +++ b/docs/src/sync-protocol.md @@ -9,6 +9,8 @@ The protocol builds on the model presented in the previous chapter, and in parti From the server's perspective, replicas are indistinguishable, so this protocol uses the term "client" to refer generically to all replicas replicating a single task history. +Each client is identified and authenticated with a "client key", known only to the server and to the replicas replicating the task history. + ## Server For each client, the server is responsible for storing the task history, in the form of a branch-free sequence of versions. @@ -66,11 +68,12 @@ If not found, the server returns a negative response. The transactions above are realized for an HTTP server at `` using the HTTP requests and responses described here. The `origin` *should* be an HTTPS endpoint on general principle, but nothing in the functonality or security of the protocol depends on connection encryption. -The replica identifies itself to the server using a `clientId` in the form of a UUID. +The replica identifies itself to the server using a `clientKey` in the form of a UUID. +This value is passed with every request in the `X-Client-Id` header, in its dashed-hex format. ### AddVersion -The request is a `POST` to `/client//add-version/`. +The request is a `POST` to `/client/add-version/`. The request body contains the history segment, optionally encoded using any encoding supported by actix-web. The content-type must be `application/vnd.taskchampion.history-segment`. @@ -84,7 +87,7 @@ Other error responses (4xx or 5xx) may be returned and should be treated appropr ### GetChildVersion -The request is a `GET` to `/client//get-child-version/`. +The request is a `GET` to `/client/get-child-version/`. The response is 404 NOT FOUND if no such version exists. Otherwise, the response is a 200 OK. The version's history segment is returned in the response body, with content-type `application/vnd.taskchampion.history-segment`. diff --git a/sync-server/src/api/add_version.rs b/sync-server/src/api/add_version.rs index 10a36155b..4ca7cbdb1 100644 --- a/sync-server/src/api/add_version.rs +++ b/sync-server/src/api/add_version.rs @@ -1,8 +1,8 @@ use crate::api::{ - failure_to_ise, ServerState, HISTORY_SEGMENT_CONTENT_TYPE, PARENT_VERSION_ID_HEADER, - VERSION_ID_HEADER, + client_key_header, failure_to_ise, ServerState, HISTORY_SEGMENT_CONTENT_TYPE, + PARENT_VERSION_ID_HEADER, VERSION_ID_HEADER, }; -use crate::server::{add_version, AddVersionResult, ClientId, VersionId, NO_VERSION_ID}; +use crate::server::{add_version, AddVersionResult, VersionId, NO_VERSION_ID}; use actix_web::{error, post, web, HttpMessage, HttpRequest, HttpResponse, Result}; use futures::StreamExt; @@ -19,11 +19,11 @@ const MAX_SIZE: usize = 100 * 1024 * 1024; /// parent version ID in the `X-Parent-Version-Id` header. /// /// Returns other 4xx or 5xx responses on other errors. -#[post("/client/{client_id}/add-version/{parent_version_id}")] +#[post("/client/add-version/{parent_version_id}")] pub(crate) async fn service( req: HttpRequest, server_state: web::Data, - web::Path((client_id, parent_version_id)): web::Path<(ClientId, VersionId)>, + web::Path((parent_version_id,)): web::Path<(VersionId,)>, mut payload: web::Payload, ) -> Result { // check content-type @@ -31,6 +31,8 @@ pub(crate) async fn service( return Err(error::ErrorBadRequest("Bad content-type")); } + let client_key = client_key_header(&req)?; + // read the body in its entirety let mut body = web::BytesMut::new(); while let Some(chunk) = payload.next().await { @@ -52,16 +54,16 @@ pub(crate) async fn service( let mut txn = server_state.txn().map_err(failure_to_ise)?; // get, or create, the client - let client = match txn.get_client(client_id).map_err(failure_to_ise)? { + let client = match txn.get_client(client_key).map_err(failure_to_ise)? { Some(client) => client, None => { - txn.new_client(client_id, NO_VERSION_ID) + txn.new_client(client_key, NO_VERSION_ID) .map_err(failure_to_ise)?; - txn.get_client(client_id).map_err(failure_to_ise)?.unwrap() + txn.get_client(client_key).map_err(failure_to_ise)?.unwrap() } }; - let result = add_version(txn, client_id, client, parent_version_id, body.to_vec()) + let result = add_version(txn, client_key, client, parent_version_id, body.to_vec()) .map_err(failure_to_ise)?; Ok(match result { AddVersionResult::Ok(version_id) => HttpResponse::Ok() @@ -83,7 +85,7 @@ mod test { #[actix_rt::test] async fn test_success() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let version_id = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); @@ -91,19 +93,20 @@ mod test { // set up the storage contents.. { let mut txn = server_box.txn().unwrap(); - txn.new_client(client_id, Uuid::nil()).unwrap(); + txn.new_client(client_key, Uuid::nil()).unwrap(); } let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!("/client/{}/add-version/{}", client_id, parent_version_id); + let uri = format!("/client/add-version/{}", parent_version_id); let req = test::TestRequest::post() .uri(&uri) .header( "Content-Type", "application/vnd.taskchampion.history-segment", ) + .header("X-Client-Key", client_key.to_string()) .set_payload(b"abcd".to_vec()) .to_request(); let resp = test::call_service(&mut app, req).await; @@ -119,7 +122,7 @@ mod test { #[actix_rt::test] async fn test_conflict() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let version_id = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); @@ -127,19 +130,20 @@ mod test { // set up the storage contents.. { let mut txn = server_box.txn().unwrap(); - txn.new_client(client_id, version_id).unwrap(); + txn.new_client(client_key, version_id).unwrap(); } let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!("/client/{}/add-version/{}", client_id, parent_version_id); + let uri = format!("/client/add-version/{}", parent_version_id); let req = test::TestRequest::post() .uri(&uri) .header( "Content-Type", "application/vnd.taskchampion.history-segment", ) + .header("X-Client-Key", client_key.to_string()) .set_payload(b"abcd".to_vec()) .to_request(); let resp = test::call_service(&mut app, req).await; @@ -153,16 +157,17 @@ mod test { #[actix_rt::test] async fn test_bad_content_type() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!("/client/{}/add-version/{}", client_id, parent_version_id); + let uri = format!("/client/add-version/{}", parent_version_id); let req = test::TestRequest::post() .uri(&uri) .header("Content-Type", "not/correct") + .header("X-Client-Key", client_key.to_string()) .set_payload(b"abcd".to_vec()) .to_request(); let resp = test::call_service(&mut app, req).await; @@ -171,19 +176,20 @@ mod test { #[actix_rt::test] async fn test_empty_body() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!("/client/{}/add-version/{}", client_id, parent_version_id); + let uri = format!("/client/add-version/{}", parent_version_id); let req = test::TestRequest::post() .uri(&uri) .header( "Content-Type", "application/vnd.taskchampion.history-segment", ) + .header("X-Client-Key", client_key.to_string()) .to_request(); let resp = test::call_service(&mut app, req).await; assert_eq!(resp.status(), StatusCode::BAD_REQUEST); diff --git a/sync-server/src/api/get_child_version.rs b/sync-server/src/api/get_child_version.rs index 901c46440..0bf04d96a 100644 --- a/sync-server/src/api/get_child_version.rs +++ b/sync-server/src/api/get_child_version.rs @@ -1,9 +1,9 @@ use crate::api::{ - failure_to_ise, ServerState, HISTORY_SEGMENT_CONTENT_TYPE, PARENT_VERSION_ID_HEADER, - VERSION_ID_HEADER, + client_key_header, failure_to_ise, ServerState, HISTORY_SEGMENT_CONTENT_TYPE, + PARENT_VERSION_ID_HEADER, VERSION_ID_HEADER, }; -use crate::server::{get_child_version, ClientId, VersionId}; -use actix_web::{error, get, web, HttpResponse, Result}; +use crate::server::{get_child_version, VersionId}; +use actix_web::{error, get, web, HttpRequest, HttpResponse, Result}; /// Get a child version. /// @@ -13,18 +13,21 @@ use actix_web::{error, get, web, HttpResponse, Result}; /// /// If no such child exists, returns a 404 with no content. /// Returns other 4xx or 5xx responses on other errors. -#[get("/client/{client_id}/get-child-version/{parent_version_id}")] +#[get("/client/get-child-version/{parent_version_id}")] pub(crate) async fn service( + req: HttpRequest, server_state: web::Data, - web::Path((client_id, parent_version_id)): web::Path<(ClientId, VersionId)>, + web::Path((parent_version_id,)): web::Path<(VersionId,)>, ) -> Result { let mut txn = server_state.txn().map_err(failure_to_ise)?; - txn.get_client(client_id) + let client_key = client_key_header(&req)?; + + txn.get_client(client_key) .map_err(failure_to_ise)? .ok_or_else(|| error::ErrorNotFound("no such client"))?; - let result = get_child_version(txn, client_id, parent_version_id).map_err(failure_to_ise)?; + let result = get_child_version(txn, client_key, parent_version_id).map_err(failure_to_ise)?; if let Some(result) = result { Ok(HttpResponse::Ok() .content_type(HISTORY_SEGMENT_CONTENT_TYPE) @@ -49,7 +52,7 @@ mod test { #[actix_rt::test] async fn test_success() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let version_id = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); @@ -57,19 +60,19 @@ mod test { // set up the storage contents.. { let mut txn = server_box.txn().unwrap(); - txn.new_client(client_id, Uuid::new_v4()).unwrap(); - txn.add_version(client_id, version_id, parent_version_id, b"abcd".to_vec()) + txn.new_client(client_key, Uuid::new_v4()).unwrap(); + txn.add_version(client_key, version_id, parent_version_id, b"abcd".to_vec()) .unwrap(); } let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!( - "/client/{}/get-child-version/{}", - client_id, parent_version_id - ); - let req = test::TestRequest::get().uri(&uri).to_request(); + let uri = format!("/client/get-child-version/{}", parent_version_id); + let req = test::TestRequest::get() + .uri(&uri) + .header("X-Client-Key", client_key.to_string()) + .to_request(); let mut resp = test::call_service(&mut app, req).await; assert_eq!(resp.status(), StatusCode::OK); assert_eq!( @@ -92,17 +95,17 @@ mod test { #[actix_rt::test] async fn test_client_not_found() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!( - "/client/{}/get-child-version/{}", - client_id, parent_version_id - ); - let req = test::TestRequest::get().uri(&uri).to_request(); + let uri = format!("/client/get-child-version/{}", parent_version_id); + let req = test::TestRequest::get() + .uri(&uri) + .header("X-Client-Key", client_key.to_string()) + .to_request(); let resp = test::call_service(&mut app, req).await; assert_eq!(resp.status(), StatusCode::NOT_FOUND); assert_eq!(resp.headers().get("X-Version-Id"), None); @@ -111,23 +114,23 @@ mod test { #[actix_rt::test] async fn test_version_not_found() { - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let server_box: Box = Box::new(InMemoryStorage::new()); // create the client, but not the version { let mut txn = server_box.txn().unwrap(); - txn.new_client(client_id, Uuid::new_v4()).unwrap(); + txn.new_client(client_key, Uuid::new_v4()).unwrap(); } let server_state = ServerState::new(server_box); let mut app = test::init_service(App::new().service(app_scope(server_state))).await; - let uri = format!( - "/client/{}/get-child-version/{}", - client_id, parent_version_id - ); - let req = test::TestRequest::get().uri(&uri).to_request(); + let uri = format!("/client/get-child-version/{}", parent_version_id); + let req = test::TestRequest::get() + .uri(&uri) + .header("X-Client-Key", client_key.to_string()) + .to_request(); let resp = test::call_service(&mut app, req).await; assert_eq!(resp.status(), StatusCode::NOT_FOUND); assert_eq!(resp.headers().get("X-Version-Id"), None); diff --git a/sync-server/src/api/mod.rs b/sync-server/src/api/mod.rs index cddcab59c..5929f28c1 100644 --- a/sync-server/src/api/mod.rs +++ b/sync-server/src/api/mod.rs @@ -1,5 +1,6 @@ +use crate::server::ClientKey; use crate::storage::Storage; -use actix_web::{error, http::StatusCode, web, Scope}; +use actix_web::{error, http::StatusCode, web, HttpRequest, Result, Scope}; use std::sync::Arc; mod add_version; @@ -9,10 +10,13 @@ mod get_child_version; pub(crate) const HISTORY_SEGMENT_CONTENT_TYPE: &str = "application/vnd.taskchampion.history-segment"; -/// The header names for version ID +/// The header name for version ID pub(crate) const VERSION_ID_HEADER: &str = "X-Version-Id"; -/// The header names for parent version ID +/// The header name for client key +pub(crate) const CLIENT_KEY_HEADER: &str = "X-Client-Key"; + +/// The header name for parent version ID pub(crate) const PARENT_VERSION_ID_HEADER: &str = "X-Parent-Version-Id"; /// The type containing a reference to the Storage object in the Actix state. @@ -28,3 +32,17 @@ pub(crate) fn api_scope() -> Scope { fn failure_to_ise(err: failure::Error) -> impl actix_web::ResponseError { error::InternalError::new(err, StatusCode::INTERNAL_SERVER_ERROR) } + +/// Get the client key +fn client_key_header(req: &HttpRequest) -> Result { + fn badrequest() -> error::Error { + error::ErrorBadRequest("bad x-client-id") + } + if let Some(client_key_hdr) = req.headers().get(CLIENT_KEY_HEADER) { + let client_key = client_key_hdr.to_str().map_err(|_| badrequest())?; + let client_key = ClientKey::parse_str(client_key).map_err(|_| badrequest())?; + Ok(client_key) + } else { + Err(badrequest()) + } +} diff --git a/sync-server/src/server.rs b/sync-server/src/server.rs index a518189ce..6c0e183cc 100644 --- a/sync-server/src/server.rs +++ b/sync-server/src/server.rs @@ -8,7 +8,7 @@ use uuid::Uuid; pub const NO_VERSION_ID: VersionId = Uuid::nil(); pub(crate) type HistorySegment = Vec; -pub(crate) type ClientId = Uuid; +pub(crate) type ClientKey = Uuid; pub(crate) type VersionId = Uuid; /// Response to get_child_version @@ -21,11 +21,11 @@ pub(crate) struct GetVersionResult { pub(crate) fn get_child_version<'a>( mut txn: Box, - client_id: ClientId, + client_key: ClientKey, parent_version_id: VersionId, ) -> Fallible> { Ok(txn - .get_version_by_parent(client_id, parent_version_id)? + .get_version_by_parent(client_key, parent_version_id)? .map(|version| GetVersionResult { version_id: version.version_id, parent_version_id: version.parent_version_id, @@ -44,14 +44,14 @@ pub(crate) enum AddVersionResult { pub(crate) fn add_version<'a>( mut txn: Box, - client_id: ClientId, + client_key: ClientKey, client: Client, parent_version_id: VersionId, history_segment: HistorySegment, ) -> Fallible { log::debug!( - "add_version(client_id: {}, parent_version_id: {})", - client_id, + "add_version(client_key: {}, parent_version_id: {})", + client_key, parent_version_id, ); @@ -71,8 +71,8 @@ pub(crate) fn add_version<'a>( ); // update the DB - txn.add_version(client_id, version_id, parent_version_id, history_segment)?; - txn.set_client_latest_version_id(client_id, version_id)?; + txn.add_version(client_key, version_id, parent_version_id, history_segment)?; + txn.set_client_latest_version_id(client_key, version_id)?; txn.commit()?; Ok(AddVersionResult::Ok(version_id)) @@ -87,9 +87,9 @@ mod test { fn gcv_not_found() -> Fallible<()> { let storage = InMemoryStorage::new(); let txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); - assert_eq!(get_child_version(txn, client_id, parent_version_id)?, None); + assert_eq!(get_child_version(txn, client_key, parent_version_id)?, None); Ok(()) } @@ -97,20 +97,20 @@ mod test { fn gcv_found() -> Fallible<()> { let storage = InMemoryStorage::new(); let mut txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let version_id = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let history_segment = b"abcd".to_vec(); txn.add_version( - client_id, + client_key, version_id, parent_version_id, history_segment.clone(), )?; assert_eq!( - get_child_version(txn, client_id, parent_version_id)?, + get_child_version(txn, client_key, parent_version_id)?, Some(GetVersionResult { version_id, parent_version_id, @@ -124,7 +124,7 @@ mod test { fn av_conflict() -> Fallible<()> { let storage = InMemoryStorage::new(); let mut txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let history_segment = b"abcd".to_vec(); let existing_parent_version_id = Uuid::new_v4(); @@ -135,7 +135,7 @@ mod test { assert_eq!( add_version( txn, - client_id, + client_key, client, parent_version_id, history_segment.clone() @@ -145,9 +145,9 @@ mod test { // verify that the storage wasn't updated txn = storage.txn()?; - assert_eq!(txn.get_client(client_id)?, None); + assert_eq!(txn.get_client(client_key)?, None); assert_eq!( - txn.get_version_by_parent(client_id, parent_version_id)?, + txn.get_version_by_parent(client_key, parent_version_id)?, None ); @@ -157,7 +157,7 @@ mod test { fn test_av_success(latest_version_id_nil: bool) -> Fallible<()> { let storage = InMemoryStorage::new(); let mut txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let history_segment = b"abcd".to_vec(); let latest_version_id = if latest_version_id_nil { @@ -166,12 +166,12 @@ mod test { parent_version_id }; - txn.new_client(client_id, latest_version_id)?; - let client = txn.get_client(client_id)?.unwrap(); + txn.new_client(client_key, latest_version_id)?; + let client = txn.get_client(client_key)?.unwrap(); let result = add_version( txn, - client_id, + client_key, client, parent_version_id, history_segment.clone(), @@ -182,10 +182,10 @@ mod test { // verify that the storage was updated txn = storage.txn()?; - let client = txn.get_client(client_id)?.unwrap(); + let client = txn.get_client(client_key)?.unwrap(); assert_eq!(client.latest_version_id, new_version_id); let version = txn - .get_version_by_parent(client_id, parent_version_id)? + .get_version_by_parent(client_key, parent_version_id)? .unwrap(); assert_eq!(version.version_id, new_version_id); assert_eq!(version.parent_version_id, parent_version_id); diff --git a/sync-server/src/storage/inmemory.rs b/sync-server/src/storage/inmemory.rs index 68f25cf92..fbd56db65 100644 --- a/sync-server/src/storage/inmemory.rs +++ b/sync-server/src/storage/inmemory.rs @@ -4,10 +4,10 @@ use std::collections::HashMap; use std::sync::{Mutex, MutexGuard}; struct Inner { - /// Clients, indexed by client_id + /// Clients, indexed by client_key clients: HashMap, - /// Versions, indexed by (client_id, parent_version_id) + /// Versions, indexed by (client_key, parent_version_id) versions: HashMap<(Uuid, Uuid), Version>, } @@ -34,48 +34,48 @@ impl Storage for InMemoryStorage { } impl<'a> StorageTxn for InnerTxn<'a> { - fn get_client(&mut self, client_id: Uuid) -> Fallible> { - Ok(self.0.clients.get(&client_id).cloned()) + fn get_client(&mut self, client_key: Uuid) -> Fallible> { + Ok(self.0.clients.get(&client_key).cloned()) } - fn new_client(&mut self, client_id: Uuid, latest_version_id: Uuid) -> Fallible<()> { - if self.0.clients.get(&client_id).is_some() { - return Err(format_err!("Client {} already exists", client_id)); + fn new_client(&mut self, client_key: Uuid, latest_version_id: Uuid) -> Fallible<()> { + if self.0.clients.get(&client_key).is_some() { + return Err(format_err!("Client {} already exists", client_key)); } self.0 .clients - .insert(client_id, Client { latest_version_id }); + .insert(client_key, Client { latest_version_id }); Ok(()) } fn set_client_latest_version_id( &mut self, - client_id: Uuid, + client_key: Uuid, latest_version_id: Uuid, ) -> Fallible<()> { - if let Some(client) = self.0.clients.get_mut(&client_id) { + if let Some(client) = self.0.clients.get_mut(&client_key) { client.latest_version_id = latest_version_id; Ok(()) } else { - Err(format_err!("Client {} does not exist", client_id)) + Err(format_err!("Client {} does not exist", client_key)) } } fn get_version_by_parent( &mut self, - client_id: Uuid, + client_key: Uuid, parent_version_id: Uuid, ) -> Fallible> { Ok(self .0 .versions - .get(&(client_id, parent_version_id)) + .get(&(client_key, parent_version_id)) .cloned()) } fn add_version( &mut self, - client_id: Uuid, + client_key: Uuid, version_id: Uuid, parent_version_id: Uuid, history_segment: Vec, @@ -88,7 +88,7 @@ impl<'a> StorageTxn for InnerTxn<'a> { }; self.0 .versions - .insert((client_id, version.parent_version_id), version); + .insert((client_key, version.parent_version_id), version); Ok(()) } diff --git a/sync-server/src/storage/kv.rs b/sync-server/src/storage/kv.rs index d258da8f3..f3cc178b2 100644 --- a/sync-server/src/storage/kv.rs +++ b/sync-server/src/storage/kv.rs @@ -4,28 +4,28 @@ use kv::msgpack::Msgpack; use kv::{Bucket, Config, Error, Serde, Store, ValueBuf}; use std::path::Path; -/// Key for versions: concatenation of client_id and parent_version_id -type VersionKey = [u8; 32]; +/// DB Key for versions: concatenation of client_key and parent_version_id +type VersionDbKey = [u8; 32]; -fn version_key(client_id: Uuid, parent_version_id: Uuid) -> VersionKey { +fn version_db_key(client_key: Uuid, parent_version_id: Uuid) -> VersionDbKey { let mut key = [0u8; 32]; - key[..16].clone_from_slice(client_id.as_bytes()); + key[..16].clone_from_slice(client_key.as_bytes()); key[16..].clone_from_slice(parent_version_id.as_bytes()); key } -/// Key for clients: just the client_id -type ClientKey = [u8; 16]; +/// Key for clients: just the client_key +type ClientDbKey = [u8; 16]; -fn client_key(client_id: Uuid) -> ClientKey { - *client_id.as_bytes() +fn client_db_key(client_key: Uuid) -> ClientDbKey { + *client_key.as_bytes() } /// KVStorage is an on-disk storage backend which uses LMDB via the `kv` crate. pub(crate) struct KVStorage<'t> { store: Store, - clients_bucket: Bucket<'t, ClientKey, ValueBuf>>, - versions_bucket: Bucket<'t, VersionKey, ValueBuf>>, + clients_bucket: Bucket<'t, ClientDbKey, ValueBuf>>, + versions_bucket: Bucket<'t, VersionDbKey, ValueBuf>>, } impl<'t> KVStorage<'t> { @@ -37,9 +37,9 @@ impl<'t> KVStorage<'t> { let store = Store::new(config)?; let clients_bucket = - store.bucket::>>(Some("clients"))?; + store.bucket::>>(Some("clients"))?; let versions_bucket = - store.bucket::>>(Some("versions"))?; + store.bucket::>>(Some("versions"))?; Ok(KVStorage { store, @@ -73,17 +73,17 @@ impl<'t> Txn<'t> { } } - fn clients_bucket(&self) -> &'t Bucket<'t, ClientKey, ValueBuf>> { + fn clients_bucket(&self) -> &'t Bucket<'t, ClientDbKey, ValueBuf>> { &self.storage.clients_bucket } - fn versions_bucket(&self) -> &'t Bucket<'t, VersionKey, ValueBuf>> { + fn versions_bucket(&self) -> &'t Bucket<'t, VersionDbKey, ValueBuf>> { &self.storage.versions_bucket } } impl<'t> StorageTxn for Txn<'t> { - fn get_client(&mut self, client_id: Uuid) -> Fallible> { - let key = client_key(client_id); + fn get_client(&mut self, client_key: Uuid) -> Fallible> { + let key = client_db_key(client_key); let bucket = self.clients_bucket(); let kvtxn = self.kvtxn(); @@ -97,8 +97,8 @@ impl<'t> StorageTxn for Txn<'t> { Ok(Some(client)) } - fn new_client(&mut self, client_id: Uuid, latest_version_id: Uuid) -> Fallible<()> { - let key = client_key(client_id); + fn new_client(&mut self, client_key: Uuid, latest_version_id: Uuid) -> Fallible<()> { + let key = client_db_key(client_key); let bucket = self.clients_bucket(); let kvtxn = self.kvtxn(); let client = Client { latest_version_id }; @@ -108,19 +108,19 @@ impl<'t> StorageTxn for Txn<'t> { fn set_client_latest_version_id( &mut self, - client_id: Uuid, + client_key: Uuid, latest_version_id: Uuid, ) -> Fallible<()> { // implementation is the same as new_client.. - self.new_client(client_id, latest_version_id) + self.new_client(client_key, latest_version_id) } fn get_version_by_parent( &mut self, - client_id: Uuid, + client_key: Uuid, parent_version_id: Uuid, ) -> Fallible> { - let key = version_key(client_id, parent_version_id); + let key = version_db_key(client_key, parent_version_id); let bucket = self.versions_bucket(); let kvtxn = self.kvtxn(); let version = match kvtxn.get(&bucket, key) { @@ -135,12 +135,12 @@ impl<'t> StorageTxn for Txn<'t> { fn add_version( &mut self, - client_id: Uuid, + client_key: Uuid, version_id: Uuid, parent_version_id: Uuid, history_segment: Vec, ) -> Fallible<()> { - let key = version_key(client_id, parent_version_id); + let key = version_db_key(client_key, parent_version_id); let bucket = self.versions_bucket(); let kvtxn = self.kvtxn(); let version = Version { @@ -184,17 +184,17 @@ mod test { let storage = KVStorage::new(&tmp_dir.path())?; let mut txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let latest_version_id = Uuid::new_v4(); - txn.new_client(client_id, latest_version_id)?; + txn.new_client(client_key, latest_version_id)?; - let client = txn.get_client(client_id)?.unwrap(); + let client = txn.get_client(client_key)?.unwrap(); assert_eq!(client.latest_version_id, latest_version_id); let latest_version_id = Uuid::new_v4(); - txn.set_client_latest_version_id(client_id, latest_version_id)?; + txn.set_client_latest_version_id(client_key, latest_version_id)?; - let client = txn.get_client(client_id)?.unwrap(); + let client = txn.get_client(client_key)?.unwrap(); assert_eq!(client.latest_version_id, latest_version_id); Ok(()) @@ -216,18 +216,18 @@ mod test { let storage = KVStorage::new(&tmp_dir.path())?; let mut txn = storage.txn()?; - let client_id = Uuid::new_v4(); + let client_key = Uuid::new_v4(); let version_id = Uuid::new_v4(); let parent_version_id = Uuid::new_v4(); let history_segment = b"abc".to_vec(); txn.add_version( - client_id, + client_key, version_id, parent_version_id, history_segment.clone(), )?; let version = txn - .get_version_by_parent(client_id, parent_version_id)? + .get_version_by_parent(client_key, parent_version_id)? .unwrap(); assert_eq!( diff --git a/sync-server/src/storage/mod.rs b/sync-server/src/storage/mod.rs index 2a95917b9..c02d24fba 100644 --- a/sync-server/src/storage/mod.rs +++ b/sync-server/src/storage/mod.rs @@ -24,29 +24,29 @@ pub(crate) struct Version { pub(crate) trait StorageTxn { /// Get information about the given client - fn get_client(&mut self, client_id: Uuid) -> Fallible>; + fn get_client(&mut self, client_key: Uuid) -> Fallible>; /// Create a new client with the given latest_version_id - fn new_client(&mut self, client_id: Uuid, latest_version_id: Uuid) -> Fallible<()>; + fn new_client(&mut self, client_key: Uuid, latest_version_id: Uuid) -> Fallible<()>; /// Set the client's latest_version_id fn set_client_latest_version_id( &mut self, - client_id: Uuid, + client_key: Uuid, latest_version_id: Uuid, ) -> Fallible<()>; /// Get a version, indexed by parent version id fn get_version_by_parent( &mut self, - client_id: Uuid, + client_key: Uuid, parent_version_id: Uuid, ) -> Fallible>; /// Add a version (that must not already exist) fn add_version( &mut self, - client_id: Uuid, + client_key: Uuid, version_id: Uuid, parent_version_id: Uuid, history_segment: Vec, diff --git a/taskchampion/src/config.rs b/taskchampion/src/config.rs index 7347bcd5a..63fef6527 100644 --- a/taskchampion/src/config.rs +++ b/taskchampion/src/config.rs @@ -20,8 +20,8 @@ pub enum ServerConfig { /// Sync server "origin"; a URL with schema and hostname but no path or trailing `/` origin: String, - /// Client ID to identify this replica to the server - client_id: Uuid, + /// Client Key to identify and authenticate this replica to the server + client_key: Uuid, /// Private encryption secret used to encrypt all data sent to the server. This can /// be any suitably un-guessable string of bytes. diff --git a/taskchampion/src/server/mod.rs b/taskchampion/src/server/mod.rs index 4d214ee7e..40a8d4d2c 100644 --- a/taskchampion/src/server/mod.rs +++ b/taskchampion/src/server/mod.rs @@ -18,8 +18,8 @@ pub fn from_config(config: ServerConfig) -> Fallible> { ServerConfig::Local { server_dir } => Box::new(LocalServer::new(server_dir)?), ServerConfig::Remote { origin, - client_id, + client_key, encryption_secret, - } => Box::new(RemoteServer::new(origin, client_id, encryption_secret)), + } => Box::new(RemoteServer::new(origin, client_key, encryption_secret)), }) } diff --git a/taskchampion/src/server/remote/mod.rs b/taskchampion/src/server/remote/mod.rs index 51da87072..cd20618ec 100644 --- a/taskchampion/src/server/remote/mod.rs +++ b/taskchampion/src/server/remote/mod.rs @@ -8,7 +8,7 @@ use crypto::{HistoryCiphertext, HistoryCleartext, Secret}; pub struct RemoteServer { origin: String, - client_id: Uuid, + client_key: Uuid, encryption_secret: Secret, agent: ureq::Agent, } @@ -17,13 +17,13 @@ pub struct RemoteServer { /// taskchampion-sync-server). impl RemoteServer { /// Construct a new RemoteServer. The `origin` is the sync server's protocol and hostname - /// without a trailing slash, such as `https://tcsync.example.com`. Pass a client_id to + /// without a trailing slash, such as `https://tcsync.example.com`. Pass a client_key to /// identify this client to the server. Multiple replicas synchronizing the same task history - /// should use the same client_id. - pub fn new(origin: String, client_id: Uuid, encryption_secret: Vec) -> RemoteServer { + /// should use the same client_key. + pub fn new(origin: String, client_key: Uuid, encryption_secret: Vec) -> RemoteServer { RemoteServer { origin, - client_id, + client_key, encryption_secret: encryption_secret.into(), agent: ureq::agent(), } @@ -56,10 +56,7 @@ impl Server for RemoteServer { parent_version_id: VersionId, history_segment: HistorySegment, ) -> Fallible { - let url = format!( - "{}/client/{}/add-version/{}", - self.origin, self.client_id, parent_version_id - ); + let url = format!("{}/client/add-version/{}", self.origin, parent_version_id); let history_cleartext = HistoryCleartext { parent_version_id, history_segment, @@ -74,6 +71,7 @@ impl Server for RemoteServer { "Content-Type", "application/vnd.taskchampion.history-segment", ) + .set("X-Client-Key", &self.client_key.to_string()) .send_bytes(history_ciphertext.as_ref()); if resp.ok() { let version_id = get_uuid_header(&resp, "X-Version-Id")?; @@ -88,14 +86,15 @@ impl Server for RemoteServer { fn get_child_version(&mut self, parent_version_id: VersionId) -> Fallible { let url = format!( - "{}/client/{}/get-child-version/{}", - self.origin, self.client_id, parent_version_id + "{}/client/get-child-version/{}", + self.origin, parent_version_id ); let resp = self .agent .get(&url) .timeout_connect(10_000) .timeout_read(60_000) + .set("X-Client-Key", &self.client_key.to_string()) .call(); if resp.ok() {