Implemented callstack capture flow for a unit test

.github/workflows/ci.yml

@@ -33,6 +33,11 @@ jobs:
       - name: Install Protoc
         uses: arduino/setup-protoc@v2
 
+      - name: Install libunwind-dev
+        uses: ConorMacBride/install-package@v1
+        with:
+          apt: libunwind-dev
+
       - name: Environment
         run: |
           cargo --version
@@ -69,6 +74,10 @@ jobs:
       - name: Install Protoc
         uses: arduino/setup-protoc@v2
 
+      - name: Install libunwind-dev
+        run: |
+          sudo apt update && sudo apt install -y libunwind-dev
+
       - name: Environment
         run: |
           cargo --version

.gitignore

@@ -1,4 +1,4 @@
-/target
+**/target
 **/*.rs.bk
 .vs
 packages

Cargo.toml

@@ -41,11 +41,18 @@ wildmatch = "1"
 glob = "0.3"
 shell-words = "1"
 
+[target.'cfg(unix)'.dependencies]
+rstack = "0.3.3"
+
 [dev-dependencies]
 portpicker = "0.1.1"
 grpc-tester = { version = "0.1.0", path = "test/rust_grpc"}
 serial_test = "2.0.0"
 lazy_static = "1.4.0"
 
+[target.'cfg(unix)'.dev-dependencies]
+rstack-self = "0.3.0"
+rstack-launcher = { version = "0.1.0", path = "test/rstack-launcher" }
+
 [profile.release]
 debug = true

src/connection.rs

@@ -1,4 +1,6 @@
+use http::{HeaderMap, HeaderValue};
 use snafu::{ResultExt, Snafu};
+use std::convert::TryFrom;
 use std::net::SocketAddr;
 use std::sync::mpsc::Sender;
 use std::sync::Arc;
@@ -124,6 +126,58 @@ impl<TClient, TServer> Streams<TClient, TServer>
     }
 }
 
+/// When available, identifies the thread in the calling or client process.
+/// The client should reports its process id with the proxide-client-process-id" header and
+/// the thread id with the "proxide-client-thread-id" header.
+/// This enables the proxide proxy to capture client's callstack when it is making the call if the proxide
+/// and the client are running on the same host.
+pub struct ClientThreadId
+{
+    process_id: u32,
+    thread_id: i64,
+}
+
+impl ClientThreadId
+{
+    pub fn process_id(&self) -> u32
+    {
+        self.process_id
+    }
+
+    pub fn thread_id(&self) -> i64
+    {
+        self.thread_id
+    }
+}
+
+impl TryFrom<&MessageData> for ClientThreadId
+{
+    type Error = ();
+
+    fn try_from(value: &MessageData) -> std::result::Result<Self, Self::Error>
+    {
+        ClientThreadId::try_from(&value.headers)
+    }
+}
+
+impl TryFrom<&HeaderMap> for ClientThreadId
+{
+    type Error = ();
+
+    fn try_from(value: &HeaderMap) -> std::result::Result<Self, Self::Error>
+    {
+        let process_id: Option<u32> = number_or_none(&value.get("proxide-client-process-id"));
+        let thread_id: Option<i64> = number_or_none(&value.get("proxide-client-thread-id"));
+        match (process_id, thread_id) {
+            (Some(process_id), Some(thread_id)) => Ok(ClientThreadId {
+                process_id,
+                thread_id,
+            }),
+            _ => Err(()),
+        }
+    }
+}
+
 /// Handles a single client connection.
 ///
 /// The connection handling is split into multiple functions, but the functions are chained in a
@@ -311,3 +365,17 @@ where
         log::info!("Exit");
     });
 }
+
+fn number_or_none<N>(header: &Option<&HeaderValue>) -> Option<N>
+where
+    N: std::str::FromStr,
+{
+    if let Some(value) = header {
+        value
+            .to_str()
+            .map(|s| N::from_str(s).map(|n| Some(n)).unwrap_or(None))
+            .unwrap_or(None)
+    } else {
+        None
+    }
+}