--- /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/ext/transport/chttp2/transport/frame_ping.h"
+#include "src/core/ext/transport/chttp2/transport/internal.h"
+
+#include <string.h>
+
+#include <grpc/support/alloc.h>
+#include <grpc/support/log.h>
+#include <grpc/support/string_util.h>
+
+static bool g_disable_ping_ack = false;
+
+grpc_slice grpc_chttp2_ping_create(uint8_t ack, uint64_t opaque_8bytes) {
+ grpc_slice slice = GRPC_SLICE_MALLOC(9 + 8);
+ uint8_t* p = GRPC_SLICE_START_PTR(slice);
+
+ *p++ = 0;
+ *p++ = 0;
+ *p++ = 8;
+ *p++ = GRPC_CHTTP2_FRAME_PING;
+ *p++ = ack ? 1 : 0;
+ *p++ = 0;
+ *p++ = 0;
+ *p++ = 0;
+ *p++ = 0;
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 56);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 48);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 40);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 32);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 24);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 16);
+ *p++ = static_cast<uint8_t>(opaque_8bytes >> 8);
+ *p++ = static_cast<uint8_t>(opaque_8bytes);
+
+ return slice;
+}
+
+grpc_error* grpc_chttp2_ping_parser_begin_frame(grpc_chttp2_ping_parser* parser,
+ uint32_t length,
+ uint8_t flags) {
+ if (flags & 0xfe || length != 8) {
+ char* msg;
+ gpr_asprintf(&msg, "invalid ping: length=%d, flags=%02x", length, flags);
+ grpc_error* error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(msg);
+ gpr_free(msg);
+ return error;
+ }
+ parser->byte = 0;
+ parser->is_ack = flags;
+ parser->opaque_8bytes = 0;
+ return GRPC_ERROR_NONE;
+}
+
+grpc_error* grpc_chttp2_ping_parser_parse(void* parser,
+ grpc_chttp2_transport* t,
+ grpc_chttp2_stream* s,
+ const grpc_slice& slice,
+ int is_last) {
+ const uint8_t* const beg = GRPC_SLICE_START_PTR(slice);
+ const uint8_t* const end = GRPC_SLICE_END_PTR(slice);
+ const uint8_t* cur = beg;
+ grpc_chttp2_ping_parser* p = static_cast<grpc_chttp2_ping_parser*>(parser);
+
+ while (p->byte != 8 && cur != end) {
+ p->opaque_8bytes |= ((static_cast<uint64_t>(*cur)) << (56 - 8 * p->byte));
+ cur++;
+ p->byte++;
+ }
+
+ if (p->byte == 8) {
+ GPR_ASSERT(is_last);
+ if (p->is_ack) {
+ grpc_chttp2_ack_ping(t, p->opaque_8bytes);
+ } else {
+ if (!t->is_client) {
+ grpc_millis now = grpc_core::ExecCtx::Get()->Now();
+ grpc_millis next_allowed_ping =
+ t->ping_recv_state.last_ping_recv_time +
+ t->ping_policy.min_recv_ping_interval_without_data;
+
+ if (t->keepalive_permit_without_calls == 0 &&
+ grpc_chttp2_stream_map_size(&t->stream_map) == 0) {
+ /* According to RFC1122, the interval of TCP Keep-Alive is default to
+ no less than two hours. When there is no outstanding streams, we
+ restrict the number of PINGS equivalent to TCP Keep-Alive. */
+ next_allowed_ping =
+ t->ping_recv_state.last_ping_recv_time + 7200 * GPR_MS_PER_SEC;
+ }
+
+ if (next_allowed_ping > now) {
+ grpc_chttp2_add_ping_strike(t);
+ }
+
+ t->ping_recv_state.last_ping_recv_time = now;
+ }
+ if (!g_disable_ping_ack) {
+ if (t->ping_ack_count == t->ping_ack_capacity) {
+ t->ping_ack_capacity = GPR_MAX(t->ping_ack_capacity * 3 / 2, 3);
+ t->ping_acks = static_cast<uint64_t*>(gpr_realloc(
+ t->ping_acks, t->ping_ack_capacity * sizeof(*t->ping_acks)));
+ }
+ t->num_pending_induced_frames++;
+ t->ping_acks[t->ping_ack_count++] = p->opaque_8bytes;
+ grpc_chttp2_initiate_write(t, GRPC_CHTTP2_INITIATE_WRITE_PING_RESPONSE);
+ }
+ }
+ }
+
+ return GRPC_ERROR_NONE;
+}
+
+void grpc_set_disable_ping_ack(bool disable_ping_ack) {
+ g_disable_ping_ack = disable_ping_ack;
+}