Use Oak labels

project-oak · Jun 8, 2020 · 3c216f0 · 3c216f0
1 parent f661f80
commit 3c216f0
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/examples/Cargo.lock b/examples/Cargo.lock
diff --git a/examples/trusted_information_retrieval/client/rust/Cargo.toml b/examples/trusted_information_retrieval/client/rust/Cargo.toml
@@ -12,6 +12,7 @@ futures-core = "*"
 futures-util = "*"
 log = "*"
 oak = "=0.1.0"
+oak_abi = "=0.1.0"
 prost = "*"
 structopt = "*"
 tokio = { version = "*", features = ["fs", "macros", "sync", "stream"] }

diff --git a/examples/trusted_information_retrieval/client/rust/src/main.rs b/examples/trusted_information_retrieval/client/rust/src/main.rs
@@ -22,6 +22,8 @@ pub mod proto {
 
 use anyhow::anyhow;
 use log::info;
+use oak_abi::label::Label;
+use prost::Message;
 use proto::{
     trusted_information_retrieval_client::TrustedInformationRetrievalClient,
     ListPointsOfInterestRequest, Location,
@@ -78,7 +80,20 @@ async fn main() -> anyhow::Result<()> {
         .await
         .expect("Could not connect to Oak Application");
 
-    let mut client = TrustedInformationRetrievalClient::new(channel);
+    let mut label = Vec::new();
+    Label::public_untrusted()
+        .encode(&mut label)
+        .expect("Error encoding label");
+    let mut client = TrustedInformationRetrievalClient::with_interceptor(
+        channel,
+        move |mut request: tonic::Request<()>| {
+            request.metadata_mut().insert_bin(
+                "x-oak-label-bin",
+                tonic::metadata::MetadataValue::from_bytes(label.as_ref()),
+            );
+            Ok(request)
+        },
+    );
 
     let request = tonic::Request::new(ListPointsOfInterestRequest {
         location: Some(Location {