kbs: api_server: add attestation_service to auth callback

Signed-off-by: Mikko Ylinen <mikko.ylinen@intel.com>
confidential-containers · May 29, 2024 · fdbaa41 · fdbaa41
1 parent 4b0b47f
commit fdbaa41
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/kbs/src/api/src/http/attest.rs b/kbs/src/api/src/http/attest.rs
@@ -17,6 +17,7 @@ pub(crate) async fn auth(
     request: web::Json<Request>,
     map: web::Data<SessionMap>,
     timeout: web::Data<i64>,
+    _attestation_service: web::Data<Arc<AttestationService>>,
 ) -> Result<HttpResponse> {
     info!("Auth API called.");
     debug!("Auth Request: {:?}", &request);

diff --git a/kbs/src/api/src/lib.rs b/kbs/src/api/src/lib.rs
@@ -290,7 +290,8 @@ impl ApiServer {
             cfg_if::cfg_if! {
                 if #[cfg(feature = "as")] {
                     server_app = server_app.app_data(web::Data::clone(&sessions))
-                    .app_data(web::Data::clone(&attestation_service)).service(web::resource(kbs_path!("auth")).route(web::post().to(http::auth)))
+                    .app_data(web::Data::clone(&attestation_service))
+                    .service(web::resource(kbs_path!("auth")).route(web::post().to(http::auth)))
                     .service(web::resource(kbs_path!("attest")).route(web::post().to(http::attest)))
                     .service(
                         web::resource(kbs_path!("attestation-policy"))