| /* |
| * |
| * 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 "src/core/ext/transport/chttp2/transport/frame_window_update.h" |
| #include "src/core/ext/transport/chttp2/transport/internal.h" |
| |
| #include <grpc/support/alloc.h> |
| #include <grpc/support/log.h> |
| #include <grpc/support/string_util.h> |
| |
| grpc_slice grpc_chttp2_window_update_create( |
| uint32_t id, uint32_t window_update, grpc_transport_one_way_stats *stats) { |
| static const size_t frame_size = 13; |
| grpc_slice slice = GRPC_SLICE_MALLOC(frame_size); |
| stats->header_bytes += frame_size; |
| uint8_t *p = GRPC_SLICE_START_PTR(slice); |
| |
| GPR_ASSERT(window_update); |
| |
| *p++ = 0; |
| *p++ = 0; |
| *p++ = 4; |
| *p++ = GRPC_CHTTP2_FRAME_WINDOW_UPDATE; |
| *p++ = 0; |
| *p++ = (uint8_t)(id >> 24); |
| *p++ = (uint8_t)(id >> 16); |
| *p++ = (uint8_t)(id >> 8); |
| *p++ = (uint8_t)(id); |
| *p++ = (uint8_t)(window_update >> 24); |
| *p++ = (uint8_t)(window_update >> 16); |
| *p++ = (uint8_t)(window_update >> 8); |
| *p++ = (uint8_t)(window_update); |
| |
| return slice; |
| } |
| |
| grpc_error *grpc_chttp2_window_update_parser_begin_frame( |
| grpc_chttp2_window_update_parser *parser, uint32_t length, uint8_t flags) { |
| if (flags || length != 4) { |
| char *msg; |
| gpr_asprintf(&msg, "invalid window update: length=%d, flags=%02x", length, |
| flags); |
| grpc_error *err = GRPC_ERROR_CREATE_FROM_COPIED_STRING(msg); |
| gpr_free(msg); |
| return err; |
| } |
| parser->byte = 0; |
| parser->amount = 0; |
| return GRPC_ERROR_NONE; |
| } |
| |
| grpc_error *grpc_chttp2_window_update_parser_parse( |
| grpc_exec_ctx *exec_ctx, void *parser, grpc_chttp2_transport *t, |
| grpc_chttp2_stream *s, grpc_slice slice, int is_last) { |
| uint8_t *const beg = GRPC_SLICE_START_PTR(slice); |
| uint8_t *const end = GRPC_SLICE_END_PTR(slice); |
| uint8_t *cur = beg; |
| grpc_chttp2_window_update_parser *p = |
| (grpc_chttp2_window_update_parser *)parser; |
| |
| while (p->byte != 4 && cur != end) { |
| p->amount |= ((uint32_t)*cur) << (8 * (3 - p->byte)); |
| cur++; |
| p->byte++; |
| } |
| |
| if (s != NULL) { |
| s->stats.incoming.framing_bytes += (uint32_t)(end - cur); |
| } |
| |
| if (p->byte == 4) { |
| uint32_t received_update = p->amount; |
| if (received_update == 0 || (received_update & 0x80000000u)) { |
| char *msg; |
| gpr_asprintf(&msg, "invalid window update bytes: %d", p->amount); |
| grpc_error *err = GRPC_ERROR_CREATE_FROM_COPIED_STRING(msg); |
| gpr_free(msg); |
| return err; |
| } |
| GPR_ASSERT(is_last); |
| |
| if (t->incoming_stream_id != 0) { |
| if (s != NULL) { |
| grpc_chttp2_flowctl_recv_stream_update( |
| &t->flow_control, &s->flow_control, received_update); |
| if (grpc_chttp2_list_remove_stalled_by_stream(t, s)) { |
| grpc_chttp2_mark_stream_writable(exec_ctx, t, s); |
| grpc_chttp2_initiate_write( |
| exec_ctx, t, |
| GRPC_CHTTP2_INITIATE_WRITE_FLOW_CONTROL_UNSTALLED_BY_UPDATE); |
| } |
| } |
| } else { |
| bool was_zero = t->flow_control.remote_window <= 0; |
| grpc_chttp2_flowctl_recv_transport_update(&t->flow_control, |
| received_update); |
| bool is_zero = t->flow_control.remote_window <= 0; |
| if (was_zero && !is_zero) { |
| grpc_chttp2_initiate_write( |
| exec_ctx, t, |
| GRPC_CHTTP2_INITIATE_WRITE_TRANSPORT_FLOW_CONTROL_UNSTALLED); |
| } |
| } |
| } |
| |
| return GRPC_ERROR_NONE; |
| } |