Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2015 gRPC authors. |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [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 |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [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. |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 19 | #include "src/core/lib/http/httpcli.h" |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 20 | |
| 21 | #include <string.h> |
| 22 | |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 23 | #include <grpc/support/alloc.h> |
| 24 | #include <grpc/support/log.h> |
Masood Malekghassemi | 701af60 | 2015-06-03 15:01:17 -0700 | [diff] [blame] | 25 | #include <grpc/support/string_util.h> |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 26 | #include <grpc/support/useful.h> |
Craig Tiller | 1d7704d | 2016-02-25 14:07:29 -0800 | [diff] [blame] | 27 | |
Mark D. Roth | 8d5e60b | 2017-06-09 09:08:23 -0700 | [diff] [blame] | 28 | #include "src/core/lib/channel/channel_args.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 29 | #include "src/core/lib/http/format_request.h" |
| 30 | #include "src/core/lib/http/parser.h" |
| 31 | #include "src/core/lib/iomgr/endpoint.h" |
| 32 | #include "src/core/lib/iomgr/iomgr_internal.h" |
| 33 | #include "src/core/lib/iomgr/resolve_address.h" |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 34 | #include "src/core/lib/iomgr/sockaddr_utils.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 35 | #include "src/core/lib/iomgr/tcp_client.h" |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 36 | #include "src/core/lib/slice/slice_internal.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 37 | #include "src/core/lib/support/string.h" |
Craig Tiller | 69b093b | 2016-02-25 19:04:07 -0800 | [diff] [blame] | 38 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 39 | typedef struct { |
Craig Tiller | d41a4a7 | 2016-10-26 16:16:06 -0700 | [diff] [blame] | 40 | grpc_slice request_text; |
Matthew Iselin | 1824f05 | 2016-02-10 12:16:06 +1100 | [diff] [blame] | 41 | grpc_http_parser parser; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 42 | grpc_resolved_addresses *addresses; |
| 43 | size_t next_address; |
| 44 | grpc_endpoint *ep; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 45 | char *host; |
Craig Tiller | cc0535d | 2015-12-08 15:14:47 -0800 | [diff] [blame] | 46 | char *ssl_host_override; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 47 | gpr_timespec deadline; |
| 48 | int have_read_byte; |
Craig Tiller | f53d9c8 | 2015-08-04 14:19:43 -0700 | [diff] [blame] | 49 | const grpc_httpcli_handshaker *handshaker; |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 50 | grpc_closure *on_done; |
Craig Tiller | a0abe37 | 2015-06-01 11:31:08 -0700 | [diff] [blame] | 51 | grpc_httpcli_context *context; |
David Garcia Quintas | 2a50dfe | 2016-05-31 15:09:12 -0700 | [diff] [blame] | 52 | grpc_polling_entity *pollent; |
Craig Tiller | f39f2d6 | 2015-06-06 14:43:06 -0700 | [diff] [blame] | 53 | grpc_iomgr_object iomgr_obj; |
Craig Tiller | d41a4a7 | 2016-10-26 16:16:06 -0700 | [diff] [blame] | 54 | grpc_slice_buffer incoming; |
| 55 | grpc_slice_buffer outgoing; |
Craig Tiller | 3382511 | 2015-09-18 07:44:19 -0700 | [diff] [blame] | 56 | grpc_closure on_read; |
| 57 | grpc_closure done_write; |
Craig Tiller | d1bec03 | 2015-09-18 17:29:00 -0700 | [diff] [blame] | 58 | grpc_closure connected; |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 59 | grpc_error *overall_error; |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 60 | grpc_resource_quota *resource_quota; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 61 | } internal_request; |
| 62 | |
ctiller | 40260c4 | 2014-12-17 17:03:15 -0800 | [diff] [blame] | 63 | static grpc_httpcli_get_override g_get_override = NULL; |
| 64 | static grpc_httpcli_post_override g_post_override = NULL; |
| 65 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 66 | static void plaintext_handshake(grpc_exec_ctx *exec_ctx, void *arg, |
| 67 | grpc_endpoint *endpoint, const char *host, |
Craig Tiller | 449c64b | 2016-06-13 16:26:50 -0700 | [diff] [blame] | 68 | gpr_timespec deadline, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 69 | void (*on_done)(grpc_exec_ctx *exec_ctx, |
| 70 | void *arg, |
| 71 | grpc_endpoint *endpoint)) { |
| 72 | on_done(exec_ctx, arg, endpoint); |
Craig Tiller | f53d9c8 | 2015-08-04 14:19:43 -0700 | [diff] [blame] | 73 | } |
| 74 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 75 | const grpc_httpcli_handshaker grpc_httpcli_plaintext = {"http", |
| 76 | plaintext_handshake}; |
Craig Tiller | f53d9c8 | 2015-08-04 14:19:43 -0700 | [diff] [blame] | 77 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 78 | void grpc_httpcli_context_init(grpc_httpcli_context *context) { |
Craig Tiller | 69b093b | 2016-02-25 19:04:07 -0800 | [diff] [blame] | 79 | context->pollset_set = grpc_pollset_set_create(); |
Craig Tiller | a0abe37 | 2015-06-01 11:31:08 -0700 | [diff] [blame] | 80 | } |
| 81 | |
Craig Tiller | 9e5ac1b | 2017-02-14 22:25:50 -0800 | [diff] [blame] | 82 | void grpc_httpcli_context_destroy(grpc_exec_ctx *exec_ctx, |
| 83 | grpc_httpcli_context *context) { |
| 84 | grpc_pollset_set_destroy(exec_ctx, context->pollset_set); |
Craig Tiller | a0abe37 | 2015-06-01 11:31:08 -0700 | [diff] [blame] | 85 | } |
| 86 | |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 87 | static void next_address(grpc_exec_ctx *exec_ctx, internal_request *req, |
| 88 | grpc_error *due_to_error); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 89 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 90 | static void finish(grpc_exec_ctx *exec_ctx, internal_request *req, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 91 | grpc_error *error) { |
David Garcia Quintas | c4d5112 | 2016-06-06 14:56:02 -0700 | [diff] [blame] | 92 | grpc_polling_entity_del_from_pollset_set(exec_ctx, req->pollent, |
David Garcia Quintas | 69ff63d | 2016-06-06 16:39:47 -0700 | [diff] [blame] | 93 | req->context->pollset_set); |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 94 | GRPC_CLOSURE_SCHED(exec_ctx, req->on_done, error); |
Matthew Iselin | 1824f05 | 2016-02-10 12:16:06 +1100 | [diff] [blame] | 95 | grpc_http_parser_destroy(&req->parser); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 96 | if (req->addresses != NULL) { |
| 97 | grpc_resolved_addresses_destroy(req->addresses); |
| 98 | } |
| 99 | if (req->ep != NULL) { |
| 100 | grpc_endpoint_destroy(exec_ctx, req->ep); |
| 101 | } |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 102 | grpc_slice_unref_internal(exec_ctx, req->request_text); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 103 | gpr_free(req->host); |
Craig Tiller | cc0535d | 2015-12-08 15:14:47 -0800 | [diff] [blame] | 104 | gpr_free(req->ssl_host_override); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 105 | grpc_iomgr_unregister_object(&req->iomgr_obj); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 106 | grpc_slice_buffer_destroy_internal(exec_ctx, &req->incoming); |
| 107 | grpc_slice_buffer_destroy_internal(exec_ctx, &req->outgoing); |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 108 | GRPC_ERROR_UNREF(req->overall_error); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 109 | grpc_resource_quota_unref_internal(exec_ctx, req->resource_quota); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 110 | gpr_free(req); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 111 | } |
| 112 | |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 113 | static void append_error(internal_request *req, grpc_error *error) { |
| 114 | if (req->overall_error == GRPC_ERROR_NONE) { |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 115 | req->overall_error = |
| 116 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("Failed HTTP/1 client request"); |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 117 | } |
| 118 | grpc_resolved_address *addr = &req->addresses->addrs[req->next_address - 1]; |
murgatroid99 | 7871f73 | 2016-09-23 13:49:05 -0700 | [diff] [blame] | 119 | char *addr_text = grpc_sockaddr_to_uri(addr); |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 120 | req->overall_error = grpc_error_add_child( |
| 121 | req->overall_error, |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 122 | grpc_error_set_str(error, GRPC_ERROR_STR_TARGET_ADDRESS, |
| 123 | grpc_slice_from_copied_string(addr_text))); |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 124 | gpr_free(addr_text); |
| 125 | } |
| 126 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 127 | static void do_read(grpc_exec_ctx *exec_ctx, internal_request *req) { |
| 128 | grpc_endpoint_read(exec_ctx, req->ep, &req->incoming, &req->on_read); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame] | 129 | } |
| 130 | |
Craig Tiller | c027e77 | 2016-05-03 16:27:00 -0700 | [diff] [blame] | 131 | static void on_read(grpc_exec_ctx *exec_ctx, void *user_data, |
| 132 | grpc_error *error) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 133 | internal_request *req = (internal_request *)user_data; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 134 | size_t i; |
| 135 | |
Craig Tiller | e6deeb1 | 2016-05-09 13:14:32 -0700 | [diff] [blame] | 136 | for (i = 0; i < req->incoming.count; i++) { |
Craig Tiller | 618e67d | 2016-10-26 21:08:10 -0700 | [diff] [blame] | 137 | if (GRPC_SLICE_LENGTH(req->incoming.slices[i])) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 138 | req->have_read_byte = 1; |
Craig Tiller | e6deeb1 | 2016-05-09 13:14:32 -0700 | [diff] [blame] | 139 | grpc_error *err = |
Mark D. Roth | 714c7ec | 2016-08-04 12:58:16 -0700 | [diff] [blame] | 140 | grpc_http_parser_parse(&req->parser, req->incoming.slices[i], NULL); |
Craig Tiller | e6deeb1 | 2016-05-09 13:14:32 -0700 | [diff] [blame] | 141 | if (err != GRPC_ERROR_NONE) { |
| 142 | finish(exec_ctx, req, err); |
| 143 | return; |
| 144 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 145 | } |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 146 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 147 | |
Craig Tiller | c027e77 | 2016-05-03 16:27:00 -0700 | [diff] [blame] | 148 | if (error == GRPC_ERROR_NONE) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 149 | do_read(exec_ctx, req); |
| 150 | } else if (!req->have_read_byte) { |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 151 | next_address(exec_ctx, req, GRPC_ERROR_REF(error)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 152 | } else { |
Craig Tiller | e6deeb1 | 2016-05-09 13:14:32 -0700 | [diff] [blame] | 153 | finish(exec_ctx, req, grpc_http_parser_eof(&req->parser)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 154 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 155 | } |
| 156 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 157 | static void on_written(grpc_exec_ctx *exec_ctx, internal_request *req) { |
| 158 | do_read(exec_ctx, req); |
Craig Tiller | d1bec03 | 2015-09-18 17:29:00 -0700 | [diff] [blame] | 159 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 160 | |
Craig Tiller | c027e77 | 2016-05-03 16:27:00 -0700 | [diff] [blame] | 161 | static void done_write(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 162 | internal_request *req = (internal_request *)arg; |
Craig Tiller | c027e77 | 2016-05-03 16:27:00 -0700 | [diff] [blame] | 163 | if (error == GRPC_ERROR_NONE) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 164 | on_written(exec_ctx, req); |
| 165 | } else { |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 166 | next_address(exec_ctx, req, GRPC_ERROR_REF(error)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 167 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 168 | } |
| 169 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 170 | static void start_write(grpc_exec_ctx *exec_ctx, internal_request *req) { |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 171 | grpc_slice_ref_internal(req->request_text); |
Craig Tiller | d41a4a7 | 2016-10-26 16:16:06 -0700 | [diff] [blame] | 172 | grpc_slice_buffer_add(&req->outgoing, req->request_text); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 173 | grpc_endpoint_write(exec_ctx, req->ep, &req->outgoing, &req->done_write); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 174 | } |
| 175 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 176 | static void on_handshake_done(grpc_exec_ctx *exec_ctx, void *arg, |
| 177 | grpc_endpoint *ep) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 178 | internal_request *req = (internal_request *)arg; |
Craig Tiller | f53d9c8 | 2015-08-04 14:19:43 -0700 | [diff] [blame] | 179 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 180 | if (!ep) { |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 181 | next_address(exec_ctx, req, GRPC_ERROR_CREATE_FROM_STATIC_STRING( |
| 182 | "Unexplained handshake failure")); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 183 | return; |
| 184 | } |
Craig Tiller | f53d9c8 | 2015-08-04 14:19:43 -0700 | [diff] [blame] | 185 | |
| 186 | req->ep = ep; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 187 | start_write(exec_ctx, req); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 188 | } |
| 189 | |
Craig Tiller | c027e77 | 2016-05-03 16:27:00 -0700 | [diff] [blame] | 190 | static void on_connected(grpc_exec_ctx *exec_ctx, void *arg, |
| 191 | grpc_error *error) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 192 | internal_request *req = (internal_request *)arg; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 193 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 194 | if (!req->ep) { |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 195 | next_address(exec_ctx, req, GRPC_ERROR_REF(error)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 196 | return; |
| 197 | } |
Craig Tiller | cc0535d | 2015-12-08 15:14:47 -0800 | [diff] [blame] | 198 | req->handshaker->handshake( |
| 199 | exec_ctx, req, req->ep, |
| 200 | req->ssl_host_override ? req->ssl_host_override : req->host, |
Craig Tiller | 449c64b | 2016-06-13 16:26:50 -0700 | [diff] [blame] | 201 | req->deadline, on_handshake_done); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 202 | } |
| 203 | |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 204 | static void next_address(grpc_exec_ctx *exec_ctx, internal_request *req, |
| 205 | grpc_error *error) { |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 206 | grpc_resolved_address *addr; |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 207 | if (error != GRPC_ERROR_NONE) { |
| 208 | append_error(req, error); |
| 209 | } |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 210 | if (req->next_address == req->addresses->naddrs) { |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 211 | finish(exec_ctx, req, |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 212 | GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING( |
| 213 | "Failed HTTP requests to all targets", &req->overall_error, 1)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 214 | return; |
| 215 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 216 | addr = &req->addresses->addrs[req->next_address++]; |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 217 | GRPC_CLOSURE_INIT(&req->connected, on_connected, req, |
Craig Tiller | 91031da | 2016-12-28 15:44:25 -0800 | [diff] [blame] | 218 | grpc_schedule_on_exec_ctx); |
Mark D. Roth | 8d5e60b | 2017-06-09 09:08:23 -0700 | [diff] [blame] | 219 | grpc_arg arg = grpc_channel_arg_pointer_create( |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 220 | (char *)GRPC_ARG_RESOURCE_QUOTA, req->resource_quota, |
Mark D. Roth | 8d5e60b | 2017-06-09 09:08:23 -0700 | [diff] [blame] | 221 | grpc_resource_quota_arg_vtable()); |
Craig Tiller | 8695876 | 2016-09-23 12:05:34 -0700 | [diff] [blame] | 222 | grpc_channel_args args = {1, &arg}; |
murgatroid99 | dedb923 | 2016-09-26 13:54:04 -0700 | [diff] [blame] | 223 | grpc_tcp_client_connect(exec_ctx, &req->connected, &req->ep, |
Craig Tiller | 3d357d9 | 2016-10-26 20:52:03 -0700 | [diff] [blame] | 224 | req->context->pollset_set, &args, addr, |
| 225 | req->deadline); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 226 | } |
| 227 | |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 228 | static void on_resolved(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 229 | internal_request *req = (internal_request *)arg; |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 230 | if (error != GRPC_ERROR_NONE) { |
Craig Tiller | d646687 | 2017-05-23 16:29:00 -0700 | [diff] [blame] | 231 | finish(exec_ctx, req, GRPC_ERROR_REF(error)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 232 | return; |
| 233 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 234 | req->next_address = 0; |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 235 | next_address(exec_ctx, req, GRPC_ERROR_NONE); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 236 | } |
| 237 | |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 238 | static void internal_request_begin(grpc_exec_ctx *exec_ctx, |
| 239 | grpc_httpcli_context *context, |
Craig Tiller | c97065d | 2016-06-07 13:00:14 -0700 | [diff] [blame] | 240 | grpc_polling_entity *pollent, |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 241 | grpc_resource_quota *resource_quota, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 242 | const grpc_httpcli_request *request, |
| 243 | gpr_timespec deadline, grpc_closure *on_done, |
| 244 | grpc_httpcli_response *response, |
Craig Tiller | d41a4a7 | 2016-10-26 16:16:06 -0700 | [diff] [blame] | 245 | const char *name, grpc_slice request_text) { |
Yash Tibrewal | 7cdd99c | 2017-09-08 16:04:12 -0700 | [diff] [blame] | 246 | internal_request *req = |
| 247 | (internal_request *)gpr_malloc(sizeof(internal_request)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 248 | memset(req, 0, sizeof(*req)); |
Craig Tiller | 47a708e | 2015-09-15 16:16:06 -0700 | [diff] [blame] | 249 | req->request_text = request_text; |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 250 | grpc_http_parser_init(&req->parser, GRPC_HTTP_RESPONSE, response); |
| 251 | req->on_done = on_done; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 252 | req->deadline = deadline; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 253 | req->handshaker = |
| 254 | request->handshaker ? request->handshaker : &grpc_httpcli_plaintext; |
Craig Tiller | a0abe37 | 2015-06-01 11:31:08 -0700 | [diff] [blame] | 255 | req->context = context; |
David Garcia Quintas | 2a50dfe | 2016-05-31 15:09:12 -0700 | [diff] [blame] | 256 | req->pollent = pollent; |
Craig Tiller | 6a64bfd | 2016-05-09 13:05:03 -0700 | [diff] [blame] | 257 | req->overall_error = GRPC_ERROR_NONE; |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 258 | req->resource_quota = grpc_resource_quota_ref_internal(resource_quota); |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 259 | GRPC_CLOSURE_INIT(&req->on_read, on_read, req, grpc_schedule_on_exec_ctx); |
| 260 | GRPC_CLOSURE_INIT(&req->done_write, done_write, req, |
Craig Tiller | 91031da | 2016-12-28 15:44:25 -0800 | [diff] [blame] | 261 | grpc_schedule_on_exec_ctx); |
Craig Tiller | d41a4a7 | 2016-10-26 16:16:06 -0700 | [diff] [blame] | 262 | grpc_slice_buffer_init(&req->incoming); |
| 263 | grpc_slice_buffer_init(&req->outgoing); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 264 | grpc_iomgr_register_object(&req->iomgr_obj, name); |
| 265 | req->host = gpr_strdup(request->host); |
Craig Tiller | cc0535d | 2015-12-08 15:14:47 -0800 | [diff] [blame] | 266 | req->ssl_host_override = gpr_strdup(request->ssl_host_override); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 267 | |
David Garcia Quintas | 2a50dfe | 2016-05-31 15:09:12 -0700 | [diff] [blame] | 268 | GPR_ASSERT(pollent); |
David Garcia Quintas | c4d5112 | 2016-06-06 14:56:02 -0700 | [diff] [blame] | 269 | grpc_polling_entity_add_to_pollset_set(exec_ctx, req->pollent, |
David Garcia Quintas | 69ff63d | 2016-06-06 16:39:47 -0700 | [diff] [blame] | 270 | req->context->pollset_set); |
Craig Tiller | 91031da | 2016-12-28 15:44:25 -0800 | [diff] [blame] | 271 | grpc_resolve_address( |
| 272 | exec_ctx, request->host, req->handshaker->default_port, |
| 273 | req->context->pollset_set, |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 274 | GRPC_CLOSURE_CREATE(on_resolved, req, grpc_schedule_on_exec_ctx), |
Craig Tiller | 91031da | 2016-12-28 15:44:25 -0800 | [diff] [blame] | 275 | &req->addresses); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 276 | } |
| 277 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 278 | void grpc_httpcli_get(grpc_exec_ctx *exec_ctx, grpc_httpcli_context *context, |
David Garcia Quintas | 2a50dfe | 2016-05-31 15:09:12 -0700 | [diff] [blame] | 279 | grpc_polling_entity *pollent, |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 280 | grpc_resource_quota *resource_quota, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 281 | const grpc_httpcli_request *request, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 282 | gpr_timespec deadline, grpc_closure *on_done, |
| 283 | grpc_httpcli_response *response) { |
Craig Tiller | 47a708e | 2015-09-15 16:16:06 -0700 | [diff] [blame] | 284 | char *name; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 285 | if (g_get_override && |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 286 | g_get_override(exec_ctx, request, deadline, on_done, response)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 287 | return; |
| 288 | } |
Matthew Iselin | 1824f05 | 2016-02-10 12:16:06 +1100 | [diff] [blame] | 289 | gpr_asprintf(&name, "HTTP:GET:%s:%s", request->host, request->http.path); |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 290 | internal_request_begin(exec_ctx, context, pollent, resource_quota, request, |
Craig Tiller | e34c285 | 2016-09-23 09:43:32 -0700 | [diff] [blame] | 291 | deadline, on_done, response, name, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 292 | grpc_httpcli_format_get_request(request)); |
| 293 | gpr_free(name); |
Craig Tiller | 47a708e | 2015-09-15 16:16:06 -0700 | [diff] [blame] | 294 | } |
| 295 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 296 | void grpc_httpcli_post(grpc_exec_ctx *exec_ctx, grpc_httpcli_context *context, |
David Garcia Quintas | 2a50dfe | 2016-05-31 15:09:12 -0700 | [diff] [blame] | 297 | grpc_polling_entity *pollent, |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 298 | grpc_resource_quota *resource_quota, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 299 | const grpc_httpcli_request *request, |
| 300 | const char *body_bytes, size_t body_size, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 301 | gpr_timespec deadline, grpc_closure *on_done, |
| 302 | grpc_httpcli_response *response) { |
Craig Tiller | f39f2d6 | 2015-06-06 14:43:06 -0700 | [diff] [blame] | 303 | char *name; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 304 | if (g_post_override && |
| 305 | g_post_override(exec_ctx, request, body_bytes, body_size, deadline, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 306 | on_done, response)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 307 | return; |
| 308 | } |
Matthew Iselin | 1824f05 | 2016-02-10 12:16:06 +1100 | [diff] [blame] | 309 | gpr_asprintf(&name, "HTTP:POST:%s:%s", request->host, request->http.path); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 310 | internal_request_begin( |
Craig Tiller | 20afa3d | 2016-10-17 14:52:14 -0700 | [diff] [blame] | 311 | exec_ctx, context, pollent, resource_quota, request, deadline, on_done, |
Craig Tiller | e34c285 | 2016-09-23 09:43:32 -0700 | [diff] [blame] | 312 | response, name, |
Craig Tiller | 5b15afd | 2016-05-04 15:00:14 -0700 | [diff] [blame] | 313 | grpc_httpcli_format_post_request(request, body_bytes, body_size)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 314 | gpr_free(name); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 315 | } |
ctiller | 40260c4 | 2014-12-17 17:03:15 -0800 | [diff] [blame] | 316 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 317 | void grpc_httpcli_set_override(grpc_httpcli_get_override get, |
| 318 | grpc_httpcli_post_override post) { |
ctiller | 40260c4 | 2014-12-17 17:03:15 -0800 | [diff] [blame] | 319 | g_get_override = get; |
| 320 | g_post_override = post; |
Craig Tiller | 190d360 | 2015-02-18 09:23:38 -0800 | [diff] [blame] | 321 | } |