| /* |
| * |
| * Copyright 2018 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. |
| * |
| */ |
| |
| #ifndef GRPC_CORE_LIB_IOMGR_BUFFER_LIST_H |
| #define GRPC_CORE_LIB_IOMGR_BUFFER_LIST_H |
| |
| #include <grpc/support/port_platform.h> |
| |
| #include "src/core/lib/iomgr/port.h" |
| |
| #include <grpc/support/time.h> |
| |
| #include "src/core/lib/gprpp/memory.h" |
| #include "src/core/lib/iomgr/error.h" |
| #include "src/core/lib/iomgr/internal_errqueue.h" |
| |
| namespace grpc_core { |
| struct Timestamps { |
| /* TODO(yashykt): This would also need to store OPTSTAT once support is added |
| */ |
| gpr_timespec sendmsg_time; |
| gpr_timespec scheduled_time; |
| gpr_timespec sent_time; |
| gpr_timespec acked_time; |
| |
| uint32_t byte_offset; /* byte offset relative to the start of the RPC */ |
| }; |
| |
| /** TracedBuffer is a class to keep track of timestamps for a specific buffer in |
| * the TCP layer. We are only tracking timestamps for Linux kernels and hence |
| * this class would only be used by Linux platforms. For all other platforms, |
| * TracedBuffer would be an empty class. |
| * |
| * The timestamps collected are according to grpc_core::Timestamps declared |
| * above. |
| * |
| * A TracedBuffer list is kept track of using the head element of the list. If |
| * the head element of the list is nullptr, then the list is empty. |
| */ |
| #ifdef GRPC_LINUX_ERRQUEUE |
| class TracedBuffer { |
| public: |
| /** Add a new entry in the TracedBuffer list pointed to by head. Also saves |
| * sendmsg_time with the current timestamp. */ |
| static void AddNewEntry(grpc_core::TracedBuffer** head, uint32_t seq_no, |
| void* arg); |
| |
| /** Processes a received timestamp based on sock_extended_err and |
| * scm_timestamping structures. It will invoke the timestamps callback if the |
| * timestamp type is SCM_TSTAMP_ACK. */ |
| static void ProcessTimestamp(grpc_core::TracedBuffer** head, |
| struct sock_extended_err* serr, |
| struct scm_timestamping* tss); |
| |
| /** Cleans the list by calling the callback for each traced buffer in the list |
| * with timestamps that it has. */ |
| static void Shutdown(grpc_core::TracedBuffer** head, void* remaining, |
| grpc_error* shutdown_err); |
| |
| private: |
| GPRC_ALLOW_CLASS_TO_USE_NON_PUBLIC_NEW |
| |
| TracedBuffer(uint32_t seq_no, void* arg) |
| : seq_no_(seq_no), arg_(arg), next_(nullptr) {} |
| |
| uint32_t seq_no_; /* The sequence number for the last byte in the buffer */ |
| void* arg_; /* The arg to pass to timestamps_callback */ |
| grpc_core::Timestamps ts_; /* The timestamps corresponding to this buffer */ |
| grpc_core::TracedBuffer* next_; /* The next TracedBuffer in the list */ |
| }; |
| #else /* GRPC_LINUX_ERRQUEUE */ |
| class TracedBuffer { |
| public: |
| /* Dummy shutdown function */ |
| static void Shutdown(grpc_core::TracedBuffer** head, void* remaining, |
| grpc_error* shutdown_err) {} |
| }; |
| #endif /* GRPC_LINUX_ERRQUEUE */ |
| |
| /** Sets the callback function to call when timestamps for a write are |
| * collected. The callback does not own a reference to error. */ |
| void grpc_tcp_set_write_timestamps_callback(void (*fn)(void*, |
| grpc_core::Timestamps*, |
| grpc_error* error)); |
| |
| }; /* namespace grpc_core */ |
| |
| #endif /* GRPC_CORE_LIB_IOMGR_BUFFER_LIST_H */ |