1 /*
2  *
3  * Copyright 2015 gRPC authors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  */
18 
19 #ifndef GRPC_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H
20 #define GRPC_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H
21 
22 #include <memory>
23 
24 #include <grpcpp/security/server_credentials.h>
25 
26 #include <grpc/grpc_security.h>
27 
28 #include "src/cpp/server/thread_pool_interface.h"
29 
30 namespace grpc {
31 
32 class AuthMetadataProcessorAyncWrapper final {
33  public:
34   static void Destroy(void* wrapper);
35 
36   static void Process(void* wrapper, grpc_auth_context* context,
37                       const grpc_metadata* md, size_t num_md,
38                       grpc_process_auth_metadata_done_cb cb, void* user_data);
39 
AuthMetadataProcessorAyncWrapper(const std::shared_ptr<AuthMetadataProcessor> & processor)40   AuthMetadataProcessorAyncWrapper(
41       const std::shared_ptr<AuthMetadataProcessor>& processor)
42       : thread_pool_(CreateDefaultThreadPool()), processor_(processor) {}
43 
44  private:
45   void InvokeProcessor(grpc_auth_context* context, const grpc_metadata* md,
46                        size_t num_md, grpc_process_auth_metadata_done_cb cb,
47                        void* user_data);
48   std::unique_ptr<ThreadPoolInterface> thread_pool_;
49   std::shared_ptr<AuthMetadataProcessor> processor_;
50 };
51 
52 class SecureServerCredentials final : public ServerCredentials {
53  public:
SecureServerCredentials(grpc_server_credentials * creds)54   explicit SecureServerCredentials(grpc_server_credentials* creds)
55       : creds_(creds) {}
~SecureServerCredentials()56   ~SecureServerCredentials() override {
57     grpc_server_credentials_release(creds_);
58   }
59 
60   int AddPortToServer(const grpc::string& addr, grpc_server* server) override;
61 
62   void SetAuthMetadataProcessor(
63       const std::shared_ptr<AuthMetadataProcessor>& processor) override;
64 
65  private:
66   grpc_server_credentials* creds_;
67   std::unique_ptr<AuthMetadataProcessorAyncWrapper> processor_;
68 };
69 
70 }  // namespace grpc
71 
72 #endif  // GRPC_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H
73