| /* |
| * |
| * 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 <grpc/support/port_platform.h> |
| |
| #include "src/core/lib/http/httpcli.h" |
| |
| #include <string.h> |
| |
| #include <grpc/support/alloc.h> |
| #include <grpc/support/log.h> |
| #include <grpc/support/string_util.h> |
| |
| #include "src/core/lib/channel/channel_args.h" |
| #include "src/core/lib/gpr/string.h" |
| #include "src/core/lib/http/format_request.h" |
| #include "src/core/lib/http/parser.h" |
| #include "src/core/lib/iomgr/endpoint.h" |
| #include "src/core/lib/iomgr/iomgr_internal.h" |
| #include "src/core/lib/iomgr/resolve_address.h" |
| #include "src/core/lib/iomgr/sockaddr_utils.h" |
| #include "src/core/lib/iomgr/tcp_client.h" |
| #include "src/core/lib/slice/slice_internal.h" |
| |
| typedef struct { |
| grpc_slice request_text; |
| grpc_http_parser parser; |
| grpc_resolved_addresses* addresses; |
| size_t next_address; |
| grpc_endpoint* ep; |
| char* host; |
| char* ssl_host_override; |
| grpc_millis deadline; |
| int have_read_byte; |
| const grpc_httpcli_handshaker* handshaker; |
| grpc_closure* on_done; |
| grpc_httpcli_context* context; |
| grpc_polling_entity* pollent; |
| grpc_iomgr_object iomgr_obj; |
| grpc_slice_buffer incoming; |
| grpc_slice_buffer outgoing; |
| grpc_closure on_read; |
| grpc_closure done_write; |
| grpc_closure connected; |
| grpc_error* overall_error; |
| grpc_resource_quota* resource_quota; |
| } internal_request; |
| |
| static grpc_httpcli_get_override g_get_override = nullptr; |
| static grpc_httpcli_post_override g_post_override = nullptr; |
| |
| static void plaintext_handshake(void* arg, grpc_endpoint* endpoint, |
| const char* host, grpc_millis deadline, |
| void (*on_done)(void* arg, |
| grpc_endpoint* endpoint)) { |
| on_done(arg, endpoint); |
| } |
| |
| const grpc_httpcli_handshaker grpc_httpcli_plaintext = {"http", |
| plaintext_handshake}; |
| |
| void grpc_httpcli_context_init(grpc_httpcli_context* context) { |
| context->pollset_set = grpc_pollset_set_create(); |
| } |
| |
| void grpc_httpcli_context_destroy(grpc_httpcli_context* context) { |
| grpc_pollset_set_destroy(context->pollset_set); |
| } |
| |
| static void next_address(internal_request* req, grpc_error* due_to_error); |
| |
| static void finish(internal_request* req, grpc_error* error) { |
| grpc_polling_entity_del_from_pollset_set(req->pollent, |
| req->context->pollset_set); |
| GRPC_CLOSURE_SCHED(req->on_done, error); |
| grpc_http_parser_destroy(&req->parser); |
| if (req->addresses != nullptr) { |
| grpc_resolved_addresses_destroy(req->addresses); |
| } |
| if (req->ep != nullptr) { |
| grpc_endpoint_destroy(req->ep); |
| } |
| grpc_slice_unref_internal(req->request_text); |
| gpr_free(req->host); |
| gpr_free(req->ssl_host_override); |
| grpc_iomgr_unregister_object(&req->iomgr_obj); |
| grpc_slice_buffer_destroy_internal(&req->incoming); |
| grpc_slice_buffer_destroy_internal(&req->outgoing); |
| GRPC_ERROR_UNREF(req->overall_error); |
| grpc_resource_quota_unref_internal(req->resource_quota); |
| gpr_free(req); |
| } |
| |
| static void append_error(internal_request* req, grpc_error* error) { |
| if (req->overall_error == GRPC_ERROR_NONE) { |
| req->overall_error = |
| GRPC_ERROR_CREATE_FROM_STATIC_STRING("Failed HTTP/1 client request"); |
| } |
| grpc_resolved_address* addr = &req->addresses->addrs[req->next_address - 1]; |
| char* addr_text = grpc_sockaddr_to_uri(addr); |
| req->overall_error = grpc_error_add_child( |
| req->overall_error, |
| grpc_error_set_str(error, GRPC_ERROR_STR_TARGET_ADDRESS, |
| grpc_slice_from_copied_string(addr_text))); |
| gpr_free(addr_text); |
| } |
| |
| static void do_read(internal_request* req) { |
| grpc_endpoint_read(req->ep, &req->incoming, &req->on_read); |
| } |
| |
| static void on_read(void* user_data, grpc_error* error) { |
| internal_request* req = static_cast<internal_request*>(user_data); |
| size_t i; |
| |
| for (i = 0; i < req->incoming.count; i++) { |
| if (GRPC_SLICE_LENGTH(req->incoming.slices[i])) { |
| req->have_read_byte = 1; |
| grpc_error* err = grpc_http_parser_parse( |
| &req->parser, req->incoming.slices[i], nullptr); |
| if (err != GRPC_ERROR_NONE) { |
| finish(req, err); |
| return; |
| } |
| } |
| } |
| |
| if (error == GRPC_ERROR_NONE) { |
| do_read(req); |
| } else if (!req->have_read_byte) { |
| next_address(req, GRPC_ERROR_REF(error)); |
| } else { |
| finish(req, grpc_http_parser_eof(&req->parser)); |
| } |
| } |
| |
| static void on_written(internal_request* req) { do_read(req); } |
| |
| static void done_write(void* arg, grpc_error* error) { |
| internal_request* req = static_cast<internal_request*>(arg); |
| if (error == GRPC_ERROR_NONE) { |
| on_written(req); |
| } else { |
| next_address(req, GRPC_ERROR_REF(error)); |
| } |
| } |
| |
| static void start_write(internal_request* req) { |
| grpc_slice_ref_internal(req->request_text); |
| grpc_slice_buffer_add(&req->outgoing, req->request_text); |
| grpc_endpoint_write(req->ep, &req->outgoing, &req->done_write, nullptr); |
| } |
| |
| static void on_handshake_done(void* arg, grpc_endpoint* ep) { |
| internal_request* req = static_cast<internal_request*>(arg); |
| |
| if (!ep) { |
| next_address(req, GRPC_ERROR_CREATE_FROM_STATIC_STRING( |
| "Unexplained handshake failure")); |
| return; |
| } |
| |
| req->ep = ep; |
| start_write(req); |
| } |
| |
| static void on_connected(void* arg, grpc_error* error) { |
| internal_request* req = static_cast<internal_request*>(arg); |
| |
| if (!req->ep) { |
| next_address(req, GRPC_ERROR_REF(error)); |
| return; |
| } |
| req->handshaker->handshake( |
| req, req->ep, req->ssl_host_override ? req->ssl_host_override : req->host, |
| req->deadline, on_handshake_done); |
| } |
| |
| static void next_address(internal_request* req, grpc_error* error) { |
| grpc_resolved_address* addr; |
| if (error != GRPC_ERROR_NONE) { |
| append_error(req, error); |
| } |
| if (req->next_address == req->addresses->naddrs) { |
| finish(req, |
| GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING( |
| "Failed HTTP requests to all targets", &req->overall_error, 1)); |
| return; |
| } |
| addr = &req->addresses->addrs[req->next_address++]; |
| GRPC_CLOSURE_INIT(&req->connected, on_connected, req, |
| grpc_schedule_on_exec_ctx); |
| grpc_arg arg = grpc_channel_arg_pointer_create( |
| (char*)GRPC_ARG_RESOURCE_QUOTA, req->resource_quota, |
| grpc_resource_quota_arg_vtable()); |
| grpc_channel_args args = {1, &arg}; |
| grpc_tcp_client_connect(&req->connected, &req->ep, req->context->pollset_set, |
| &args, addr, req->deadline); |
| } |
| |
| static void on_resolved(void* arg, grpc_error* error) { |
| internal_request* req = static_cast<internal_request*>(arg); |
| if (error != GRPC_ERROR_NONE) { |
| finish(req, GRPC_ERROR_REF(error)); |
| return; |
| } |
| req->next_address = 0; |
| next_address(req, GRPC_ERROR_NONE); |
| } |
| |
| static void internal_request_begin(grpc_httpcli_context* context, |
| grpc_polling_entity* pollent, |
| grpc_resource_quota* resource_quota, |
| const grpc_httpcli_request* request, |
| grpc_millis deadline, grpc_closure* on_done, |
| grpc_httpcli_response* response, |
| const char* name, grpc_slice request_text) { |
| internal_request* req = |
| static_cast<internal_request*>(gpr_malloc(sizeof(internal_request))); |
| memset(req, 0, sizeof(*req)); |
| req->request_text = request_text; |
| grpc_http_parser_init(&req->parser, GRPC_HTTP_RESPONSE, response); |
| req->on_done = on_done; |
| req->deadline = deadline; |
| req->handshaker = |
| request->handshaker ? request->handshaker : &grpc_httpcli_plaintext; |
| req->context = context; |
| req->pollent = pollent; |
| req->overall_error = GRPC_ERROR_NONE; |
| req->resource_quota = grpc_resource_quota_ref_internal(resource_quota); |
| GRPC_CLOSURE_INIT(&req->on_read, on_read, req, grpc_schedule_on_exec_ctx); |
| GRPC_CLOSURE_INIT(&req->done_write, done_write, req, |
| grpc_schedule_on_exec_ctx); |
| grpc_slice_buffer_init(&req->incoming); |
| grpc_slice_buffer_init(&req->outgoing); |
| grpc_iomgr_register_object(&req->iomgr_obj, name); |
| req->host = gpr_strdup(request->host); |
| req->ssl_host_override = gpr_strdup(request->ssl_host_override); |
| |
| GPR_ASSERT(pollent); |
| grpc_polling_entity_add_to_pollset_set(req->pollent, |
| req->context->pollset_set); |
| grpc_resolve_address( |
| request->host, req->handshaker->default_port, req->context->pollset_set, |
| GRPC_CLOSURE_CREATE(on_resolved, req, grpc_schedule_on_exec_ctx), |
| &req->addresses); |
| } |
| |
| void grpc_httpcli_get(grpc_httpcli_context* context, |
| grpc_polling_entity* pollent, |
| grpc_resource_quota* resource_quota, |
| const grpc_httpcli_request* request, grpc_millis deadline, |
| grpc_closure* on_done, grpc_httpcli_response* response) { |
| char* name; |
| if (g_get_override && g_get_override(request, deadline, on_done, response)) { |
| return; |
| } |
| gpr_asprintf(&name, "HTTP:GET:%s:%s", request->host, request->http.path); |
| internal_request_begin(context, pollent, resource_quota, request, deadline, |
| on_done, response, name, |
| grpc_httpcli_format_get_request(request)); |
| gpr_free(name); |
| } |
| |
| void grpc_httpcli_post(grpc_httpcli_context* context, |
| grpc_polling_entity* pollent, |
| grpc_resource_quota* resource_quota, |
| const grpc_httpcli_request* request, |
| const char* body_bytes, size_t body_size, |
| grpc_millis deadline, grpc_closure* on_done, |
| grpc_httpcli_response* response) { |
| char* name; |
| if (g_post_override && g_post_override(request, body_bytes, body_size, |
| deadline, on_done, response)) { |
| return; |
| } |
| gpr_asprintf(&name, "HTTP:POST:%s:%s", request->host, request->http.path); |
| internal_request_begin( |
| context, pollent, resource_quota, request, deadline, on_done, response, |
| name, grpc_httpcli_format_post_request(request, body_bytes, body_size)); |
| gpr_free(name); |
| } |
| |
| void grpc_httpcli_set_override(grpc_httpcli_get_override get, |
| grpc_httpcli_post_override post) { |
| g_get_override = get; |
| g_post_override = post; |
| } |