David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2015 gRPC authors. |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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 |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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. |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 19 | #include <string.h> |
| 20 | |
| 21 | #include <grpc/support/alloc.h> |
Mark D. Roth | fef585f | 2016-09-16 10:23:28 -0700 | [diff] [blame] | 22 | #include <grpc/support/string_util.h> |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 23 | |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 24 | #include "src/core/lib/channel/channel_args.h" |
| 25 | |
Craig Tiller | 9eb0fde | 2017-03-31 16:59:30 -0700 | [diff] [blame] | 26 | #include "src/core/ext/filters/client_channel/lb_policy_factory.h" |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 27 | #include "src/core/ext/filters/client_channel/parse_address.h" |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 28 | |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 29 | grpc_lb_addresses* grpc_lb_addresses_create( |
| 30 | size_t num_addresses, const grpc_lb_user_data_vtable* user_data_vtable) { |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 31 | grpc_lb_addresses* addresses = |
| 32 | (grpc_lb_addresses*)gpr_zalloc(sizeof(grpc_lb_addresses)); |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 33 | addresses->num_addresses = num_addresses; |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 34 | addresses->user_data_vtable = user_data_vtable; |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 35 | const size_t addresses_size = sizeof(grpc_lb_address) * num_addresses; |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 36 | addresses->addresses = (grpc_lb_address*)gpr_zalloc(addresses_size); |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 37 | return addresses; |
| 38 | } |
| 39 | |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 40 | grpc_lb_addresses* grpc_lb_addresses_copy(const grpc_lb_addresses* addresses) { |
| 41 | grpc_lb_addresses* new_addresses = grpc_lb_addresses_create( |
| 42 | addresses->num_addresses, addresses->user_data_vtable); |
Mark D. Roth | 7f7d165 | 2016-09-20 10:46:15 -0700 | [diff] [blame] | 43 | memcpy(new_addresses->addresses, addresses->addresses, |
Mark D. Roth | fef585f | 2016-09-16 10:23:28 -0700 | [diff] [blame] | 44 | sizeof(grpc_lb_address) * addresses->num_addresses); |
Mark D. Roth | ee5173f | 2016-09-16 10:14:18 -0700 | [diff] [blame] | 45 | for (size_t i = 0; i < addresses->num_addresses; ++i) { |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 46 | if (new_addresses->addresses[i].balancer_name != nullptr) { |
Mark D. Roth | 7f7d165 | 2016-09-20 10:46:15 -0700 | [diff] [blame] | 47 | new_addresses->addresses[i].balancer_name = |
| 48 | gpr_strdup(new_addresses->addresses[i].balancer_name); |
Mark D. Roth | fef585f | 2016-09-16 10:23:28 -0700 | [diff] [blame] | 49 | } |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 50 | if (new_addresses->addresses[i].user_data != nullptr) { |
Mark D. Roth | 557c990 | 2016-10-24 11:12:05 -0700 | [diff] [blame] | 51 | new_addresses->addresses[i].user_data = addresses->user_data_vtable->copy( |
| 52 | new_addresses->addresses[i].user_data); |
Mark D. Roth | ee5173f | 2016-09-16 10:14:18 -0700 | [diff] [blame] | 53 | } |
| 54 | } |
Mark D. Roth | 7f7d165 | 2016-09-20 10:46:15 -0700 | [diff] [blame] | 55 | return new_addresses; |
Mark D. Roth | ee5173f | 2016-09-16 10:14:18 -0700 | [diff] [blame] | 56 | } |
| 57 | |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 58 | void grpc_lb_addresses_set_address(grpc_lb_addresses* addresses, size_t index, |
Juanli Shen | fe40815 | 2017-09-27 12:27:20 -0700 | [diff] [blame] | 59 | const void* address, size_t address_len, |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 60 | bool is_balancer, const char* balancer_name, |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 61 | void* user_data) { |
| 62 | GPR_ASSERT(index < addresses->num_addresses); |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 63 | if (user_data != nullptr) GPR_ASSERT(addresses->user_data_vtable != nullptr); |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 64 | grpc_lb_address* target = &addresses->addresses[index]; |
| 65 | memcpy(target->address.addr, address, address_len); |
| 66 | target->address.len = address_len; |
| 67 | target->is_balancer = is_balancer; |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 68 | target->balancer_name = gpr_strdup(balancer_name); |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 69 | target->user_data = user_data; |
| 70 | } |
| 71 | |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 72 | bool grpc_lb_addresses_set_address_from_uri(grpc_lb_addresses* addresses, |
| 73 | size_t index, const grpc_uri* uri, |
| 74 | bool is_balancer, |
| 75 | const char* balancer_name, |
| 76 | void* user_data) { |
| 77 | grpc_resolved_address address; |
| 78 | if (!grpc_parse_uri(uri, &address)) return false; |
| 79 | grpc_lb_addresses_set_address(addresses, index, address.addr, address.len, |
| 80 | is_balancer, balancer_name, user_data); |
| 81 | return true; |
| 82 | } |
| 83 | |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 84 | int grpc_lb_addresses_cmp(const grpc_lb_addresses* addresses1, |
| 85 | const grpc_lb_addresses* addresses2) { |
| 86 | if (addresses1->num_addresses > addresses2->num_addresses) return 1; |
| 87 | if (addresses1->num_addresses < addresses2->num_addresses) return -1; |
| 88 | if (addresses1->user_data_vtable > addresses2->user_data_vtable) return 1; |
| 89 | if (addresses1->user_data_vtable < addresses2->user_data_vtable) return -1; |
| 90 | for (size_t i = 0; i < addresses1->num_addresses; ++i) { |
| 91 | const grpc_lb_address* target1 = &addresses1->addresses[i]; |
| 92 | const grpc_lb_address* target2 = &addresses2->addresses[i]; |
| 93 | if (target1->address.len > target2->address.len) return 1; |
| 94 | if (target1->address.len < target2->address.len) return -1; |
Mark D. Roth | 557c990 | 2016-10-24 11:12:05 -0700 | [diff] [blame] | 95 | int retval = memcmp(target1->address.addr, target2->address.addr, |
| 96 | target1->address.len); |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 97 | if (retval != 0) return retval; |
| 98 | if (target1->is_balancer > target2->is_balancer) return 1; |
| 99 | if (target1->is_balancer < target2->is_balancer) return -1; |
Mark D. Roth | 557c990 | 2016-10-24 11:12:05 -0700 | [diff] [blame] | 100 | const char* balancer_name1 = |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 101 | target1->balancer_name != nullptr ? target1->balancer_name : ""; |
Mark D. Roth | 557c990 | 2016-10-24 11:12:05 -0700 | [diff] [blame] | 102 | const char* balancer_name2 = |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 103 | target2->balancer_name != nullptr ? target2->balancer_name : ""; |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 104 | retval = strcmp(balancer_name1, balancer_name2); |
| 105 | if (retval != 0) return retval; |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 106 | if (addresses1->user_data_vtable != nullptr) { |
Mark D. Roth | 16883a3 | 2016-10-21 10:30:58 -0700 | [diff] [blame] | 107 | retval = addresses1->user_data_vtable->cmp(target1->user_data, |
| 108 | target2->user_data); |
| 109 | if (retval != 0) return retval; |
| 110 | } |
| 111 | } |
| 112 | return 0; |
| 113 | } |
| 114 | |
Craig Tiller | 87a7e1f | 2016-11-09 09:42:19 -0800 | [diff] [blame] | 115 | void grpc_lb_addresses_destroy(grpc_exec_ctx* exec_ctx, |
| 116 | grpc_lb_addresses* addresses) { |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 117 | for (size_t i = 0; i < addresses->num_addresses; ++i) { |
| 118 | gpr_free(addresses->addresses[i].balancer_name); |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 119 | if (addresses->addresses[i].user_data != nullptr) { |
Craig Tiller | 87a7e1f | 2016-11-09 09:42:19 -0800 | [diff] [blame] | 120 | addresses->user_data_vtable->destroy(exec_ctx, |
| 121 | addresses->addresses[i].user_data); |
Mark D. Roth | 7ce14d2 | 2016-09-16 13:03:46 -0700 | [diff] [blame] | 122 | } |
Mark D. Roth | c5c3878 | 2016-09-16 08:51:01 -0700 | [diff] [blame] | 123 | } |
| 124 | gpr_free(addresses->addresses); |
| 125 | gpr_free(addresses); |
| 126 | } |
| 127 | |
Mark D. Roth | 3686996 | 2016-10-21 10:43:45 -0700 | [diff] [blame] | 128 | static void* lb_addresses_copy(void* addresses) { |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 129 | return grpc_lb_addresses_copy((grpc_lb_addresses*)addresses); |
Mark D. Roth | 3686996 | 2016-10-21 10:43:45 -0700 | [diff] [blame] | 130 | } |
Craig Tiller | 87a7e1f | 2016-11-09 09:42:19 -0800 | [diff] [blame] | 131 | static void lb_addresses_destroy(grpc_exec_ctx* exec_ctx, void* addresses) { |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 132 | grpc_lb_addresses_destroy(exec_ctx, (grpc_lb_addresses*)addresses); |
Mark D. Roth | 3686996 | 2016-10-21 10:43:45 -0700 | [diff] [blame] | 133 | } |
| 134 | static int lb_addresses_cmp(void* addresses1, void* addresses2) { |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 135 | return grpc_lb_addresses_cmp((grpc_lb_addresses*)addresses1, |
| 136 | (grpc_lb_addresses*)addresses2); |
Mark D. Roth | 3686996 | 2016-10-21 10:43:45 -0700 | [diff] [blame] | 137 | } |
| 138 | static const grpc_arg_pointer_vtable lb_addresses_arg_vtable = { |
| 139 | lb_addresses_copy, lb_addresses_destroy, lb_addresses_cmp}; |
| 140 | |
| 141 | grpc_arg grpc_lb_addresses_create_channel_arg( |
Mark D. Roth | 557c990 | 2016-10-24 11:12:05 -0700 | [diff] [blame] | 142 | const grpc_lb_addresses* addresses) { |
Mark D. Roth | 8d5e60b | 2017-06-09 09:08:23 -0700 | [diff] [blame] | 143 | return grpc_channel_arg_pointer_create( |
Yash Tibrewal | 9eb8672 | 2017-09-17 23:43:30 -0700 | [diff] [blame] | 144 | (char*)GRPC_ARG_LB_ADDRESSES, (void*)addresses, &lb_addresses_arg_vtable); |
Mark D. Roth | 3686996 | 2016-10-21 10:43:45 -0700 | [diff] [blame] | 145 | } |
| 146 | |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 147 | grpc_lb_addresses* grpc_lb_addresses_find_channel_arg( |
| 148 | const grpc_channel_args* channel_args) { |
| 149 | const grpc_arg* lb_addresses_arg = |
| 150 | grpc_channel_args_find(channel_args, GRPC_ARG_LB_ADDRESSES); |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 151 | if (lb_addresses_arg == nullptr || lb_addresses_arg->type != GRPC_ARG_POINTER) |
| 152 | return nullptr; |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 153 | return (grpc_lb_addresses*)lb_addresses_arg->value.pointer.p; |
David Garcia Quintas | 53af23c | 2017-04-15 10:29:46 -0700 | [diff] [blame] | 154 | } |
| 155 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 156 | void grpc_lb_policy_factory_ref(grpc_lb_policy_factory* factory) { |
| 157 | factory->vtable->ref(factory); |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 158 | } |
| 159 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 160 | void grpc_lb_policy_factory_unref(grpc_lb_policy_factory* factory) { |
| 161 | factory->vtable->unref(factory); |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame] | 162 | } |
| 163 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 164 | grpc_lb_policy* grpc_lb_policy_factory_create_lb_policy( |
David Garcia Quintas | 67c0d04 | 2016-03-25 01:37:53 -0700 | [diff] [blame] | 165 | grpc_exec_ctx* exec_ctx, grpc_lb_policy_factory* factory, |
| 166 | grpc_lb_policy_args* args) { |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame] | 167 | if (factory == nullptr) return nullptr; |
David Garcia Quintas | 67c0d04 | 2016-03-25 01:37:53 -0700 | [diff] [blame] | 168 | return factory->vtable->create_lb_policy(exec_ctx, factory, args); |
David Garcia Quintas | 5c4543d | 2015-09-03 15:49:56 -0700 | [diff] [blame] | 169 | } |