| /* |
| * |
| * Copyright 2015 gRPC authors. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| #include "src/core/lib/iomgr/socket_mutator.h" |
| |
| #include "src/core/lib/channel/channel_args.h" |
| |
| #include <grpc/impl/codegen/grpc_types.h> |
| #include <grpc/support/sync.h> |
| #include <grpc/support/useful.h> |
| |
| void grpc_socket_mutator_init(grpc_socket_mutator* mutator, |
| const grpc_socket_mutator_vtable* vtable) { |
| mutator->vtable = vtable; |
| gpr_ref_init(&mutator->refcount, 1); |
| } |
| |
| grpc_socket_mutator* grpc_socket_mutator_ref(grpc_socket_mutator* mutator) { |
| gpr_ref(&mutator->refcount); |
| return mutator; |
| } |
| |
| bool grpc_socket_mutator_mutate_fd(grpc_socket_mutator* mutator, int fd) { |
| return mutator->vtable->mutate_fd(fd, mutator); |
| } |
| |
| int grpc_socket_mutator_compare(grpc_socket_mutator* a, |
| grpc_socket_mutator* b) { |
| int c = GPR_ICMP(a, b); |
| if (c != 0) { |
| grpc_socket_mutator* sma = a; |
| grpc_socket_mutator* smb = b; |
| c = GPR_ICMP(sma->vtable, smb->vtable); |
| if (c == 0) { |
| c = sma->vtable->compare(sma, smb); |
| } |
| } |
| return c; |
| } |
| |
| void grpc_socket_mutator_unref(grpc_socket_mutator* mutator) { |
| if (gpr_unref(&mutator->refcount)) { |
| mutator->vtable->destory(mutator); |
| } |
| } |
| |
| static void* socket_mutator_arg_copy(void* p) { |
| return grpc_socket_mutator_ref((grpc_socket_mutator*)p); |
| } |
| |
| static void socket_mutator_arg_destroy(void* p) { |
| grpc_socket_mutator_unref((grpc_socket_mutator*)p); |
| } |
| |
| static int socket_mutator_cmp(void* a, void* b) { |
| return grpc_socket_mutator_compare((grpc_socket_mutator*)a, |
| (grpc_socket_mutator*)b); |
| } |
| |
| static const grpc_arg_pointer_vtable socket_mutator_arg_vtable = { |
| socket_mutator_arg_copy, socket_mutator_arg_destroy, socket_mutator_cmp}; |
| |
| grpc_arg grpc_socket_mutator_to_arg(grpc_socket_mutator* mutator) { |
| return grpc_channel_arg_pointer_create((char*)GRPC_ARG_SOCKET_MUTATOR, |
| mutator, &socket_mutator_arg_vtable); |
| } |