• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  *
3  * Copyright 2016 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 #include <grpc/support/port_platform.h>
20 
21 #include <grpc/support/alloc.h>
22 #include <grpc/support/log.h>
23 
24 #include "src/core/lib/iomgr/polling_entity.h"
25 
grpc_polling_entity_create_from_pollset_set(grpc_pollset_set * pollset_set)26 grpc_polling_entity grpc_polling_entity_create_from_pollset_set(
27     grpc_pollset_set* pollset_set) {
28   grpc_polling_entity pollent;
29   pollent.pollent.pollset_set = pollset_set;
30   pollent.tag = GRPC_POLLS_POLLSET_SET;
31   return pollent;
32 }
33 
grpc_polling_entity_create_from_pollset(grpc_pollset * pollset)34 grpc_polling_entity grpc_polling_entity_create_from_pollset(
35     grpc_pollset* pollset) {
36   grpc_polling_entity pollent;
37   pollent.pollent.pollset = pollset;
38   pollent.tag = GRPC_POLLS_POLLSET;
39   return pollent;
40 }
41 
grpc_polling_entity_pollset(grpc_polling_entity * pollent)42 grpc_pollset* grpc_polling_entity_pollset(grpc_polling_entity* pollent) {
43   if (pollent->tag == GRPC_POLLS_POLLSET) {
44     return pollent->pollent.pollset;
45   }
46   return nullptr;
47 }
48 
grpc_polling_entity_pollset_set(grpc_polling_entity * pollent)49 grpc_pollset_set* grpc_polling_entity_pollset_set(
50     grpc_polling_entity* pollent) {
51   if (pollent->tag == GRPC_POLLS_POLLSET_SET) {
52     return pollent->pollent.pollset_set;
53   }
54   return nullptr;
55 }
56 
grpc_polling_entity_is_empty(const grpc_polling_entity * pollent)57 bool grpc_polling_entity_is_empty(const grpc_polling_entity* pollent) {
58   return pollent->tag == GRPC_POLLS_NONE;
59 }
60 
grpc_polling_entity_add_to_pollset_set(grpc_polling_entity * pollent,grpc_pollset_set * pss_dst)61 void grpc_polling_entity_add_to_pollset_set(grpc_polling_entity* pollent,
62                                             grpc_pollset_set* pss_dst) {
63   if (pollent->tag == GRPC_POLLS_POLLSET) {
64     // CFStream does not use file destriptors. When CFStream is used, the fd
65     // pollset is possible to be null.
66     if (pollent->pollent.pollset != nullptr) {
67       grpc_pollset_set_add_pollset(pss_dst, pollent->pollent.pollset);
68     }
69   } else if (pollent->tag == GRPC_POLLS_POLLSET_SET) {
70     GPR_ASSERT(pollent->pollent.pollset_set != nullptr);
71     grpc_pollset_set_add_pollset_set(pss_dst, pollent->pollent.pollset_set);
72   } else {
73     gpr_log(GPR_ERROR, "Invalid grpc_polling_entity tag '%d'", pollent->tag);
74     abort();
75   }
76 }
77 
grpc_polling_entity_del_from_pollset_set(grpc_polling_entity * pollent,grpc_pollset_set * pss_dst)78 void grpc_polling_entity_del_from_pollset_set(grpc_polling_entity* pollent,
79                                               grpc_pollset_set* pss_dst) {
80   if (pollent->tag == GRPC_POLLS_POLLSET) {
81 #ifdef GRPC_CFSTREAM
82     if (pollent->pollent.pollset != nullptr) {
83       grpc_pollset_set_del_pollset(pss_dst, pollent->pollent.pollset);
84     }
85 #else
86     GPR_ASSERT(pollent->pollent.pollset != nullptr);
87     grpc_pollset_set_del_pollset(pss_dst, pollent->pollent.pollset);
88 #endif
89   } else if (pollent->tag == GRPC_POLLS_POLLSET_SET) {
90     GPR_ASSERT(pollent->pollent.pollset_set != nullptr);
91     grpc_pollset_set_del_pollset_set(pss_dst, pollent->pollent.pollset_set);
92   } else {
93     gpr_log(GPR_ERROR, "Invalid grpc_polling_entity tag '%d'", pollent->tag);
94     abort();
95   }
96 }
97