--- /dev/null
+/*
+ *
+ * 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/gprpp/memory.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];
+ grpc_core::UniquePtr<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_moved_string(std::move(addr_text))));
+}
+
+static void do_read(internal_request* req) {
+ grpc_endpoint_read(req->ep, &req->incoming, &req->on_read, /*urgent=*/true);
+}
+
+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,
+ const 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;
+}
git://repos.xcallymotion.com / motion2.git / blobdiff