1 pub mod pb {
2     tonic::include_proto!("grpc.examples.echo");
3 }
4 
5 use futures::Stream;
6 use pb::{EchoRequest, EchoResponse};
7 use std::pin::Pin;
8 use tonic::transport::{Certificate, Identity, Server, ServerTlsConfig};
9 use tonic::{Request, Response, Status};
10 
11 type EchoResult<T> = Result<Response<T>, Status>;
12 type ResponseStream = Pin<Box<dyn Stream<Item = Result<EchoResponse, Status>> + Send + Sync>>;
13 
14 #[derive(Default)]
15 pub struct EchoServer;
16 
17 #[tonic::async_trait]
18 impl pb::echo_server::Echo for EchoServer {
19     async fn unary_echo(&self, request: Request<EchoRequest>) -> EchoResult<EchoResponse> {
20         if let Some(certs) = request.peer_certs() {
21             println!("Got {} peer certs!", certs.len());
22         }
23 
24         let message = request.into_inner().message;
25         Ok(Response::new(EchoResponse { message }))
26     }
27 
28     type ServerStreamingEchoStream = ResponseStream;
29 
30     async fn server_streaming_echo(
31         &self,
32         _: Request<EchoRequest>,
33     ) -> Result<Response<Self::ServerStreamingEchoStream>, Status> {
34         Err(Status::unimplemented("Not yet implemented"))
35     }
36 
37     async fn client_streaming_echo(
38         &self,
39         _: Request<tonic::Streaming<EchoRequest>>,
40     ) -> Result<Response<EchoResponse>, Status> {
41         Err(Status::unimplemented("Not yet implemented"))
42     }
43 
44     type BidirectionalStreamingEchoStream = ResponseStream;
45 
46     async fn bidirectional_streaming_echo(
47         &self,
48         _: Request<tonic::Streaming<EchoRequest>>,
49     ) -> Result<Response<Self::BidirectionalStreamingEchoStream>, Status> {
50         Err(Status::unimplemented("Not yet implemented"))
51     }
52 }
53 
54 #[tokio::main]
55 async fn main() -> Result<(), Box<dyn std::error::Error>> {
56     let cert = tokio::fs::read("examples/data/tls/server.pem").await?;
57     let key = tokio::fs::read("examples/data/tls/server.key").await?;
58     let server_identity = Identity::from_pem(cert, key);
59 
60     let client_ca_cert = tokio::fs::read("examples/data/tls/client_ca.pem").await?;
61     let client_ca_cert = Certificate::from_pem(client_ca_cert);
62 
63     let addr = "[::1]:50051".parse().unwrap();
64     let server = EchoServer::default();
65 
66     let tls = ServerTlsConfig::new()
67         .identity(server_identity)
68         .client_ca_root(client_ca_cert);
69 
70     Server::builder()
71         .tls_config(tls)
72         .add_service(pb::echo_server::EchoServer::new(server))
73         .serve(addr)
74         .await?;
75 
76     Ok(())
77 }
78