@encharm/cws 4.5.3 → 4.8.0

package/src/headers/26/tracing/node_trace_writer.h

@@ -0,0 +1,79 @@
+#ifndef SRC_TRACING_NODE_TRACE_WRITER_H_
+#define SRC_TRACING_NODE_TRACE_WRITER_H_
+
+#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
+
+#include <sstream>
+#include <queue>
+
+#include "libplatform/v8-tracing.h"
+#include "tracing/agent_legacy.h"
+#include "uv.h"
+
+namespace node {
+namespace tracing {
+
+using v8::platform::tracing::TraceObject;
+using v8::platform::tracing::TraceWriter;
+
+class NodeTraceWriter : public AsyncTraceWriter {
+ public:
+  explicit NodeTraceWriter(const std::string& log_file_pattern);
+  ~NodeTraceWriter() override;
+
+  void InitializeOnThread(uv_loop_t* loop) override;
+  void AppendTraceEvent(TraceObject* trace_event) override;
+  void Flush(bool blocking) override;
+
+  static const int kTracesPerFile = 1 << 19;
+
+ private:
+  struct WriteRequest {
+    std::string str;
+    int highest_request_id;
+  };
+
+  void AfterWrite();
+  void StartWrite(uv_buf_t buf);
+  void OpenNewFileForStreaming();
+  void WriteToFile(std::string&& str, int highest_request_id);
+  void WriteSuffix();
+  void FlushPrivate();
+  static void ExitSignalCb(uv_async_t* signal);
+
+  uv_loop_t* tracing_loop_ = nullptr;
+  // Triggers callback to initiate writing the contents of stream_ to disk.
+  uv_async_t flush_signal_;
+  // Triggers callback to close async objects, ending the tracing thread.
+  uv_async_t exit_signal_;
+  // Prevents concurrent R/W on state related to serialized trace data
+  // before it's written to disk, namely stream_ and total_traces_
+  // as well as json_trace_writer_.
+  Mutex stream_mutex_;
+  // Prevents concurrent R/W on state related to write requests.
+  // If both mutexes are locked, request_mutex_ has to be locked first.
+  Mutex request_mutex_;
+  // Allows blocking calls to Flush() to wait on a condition for
+  // trace events to be written to disk.
+  ConditionVariable request_cond_;
+  // Used to wait until async handles have been closed.
+  ConditionVariable exit_cond_;
+  int fd_ = -1;
+  uv_fs_t write_req_;
+  std::queue<WriteRequest> write_req_queue_;
+  int num_write_requests_ = 0;
+  int highest_request_id_completed_ = 0;
+  int total_traces_ = 0;
+  int file_num_ = 0;
+  std::string log_file_pattern_;
+  std::ostringstream stream_;
+  std::unique_ptr<TraceWriter> json_trace_writer_;
+  bool exited_ = false;
+};
+
+}  // namespace tracing
+}  // namespace node
+
+#endif  // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
+
+#endif  // SRC_TRACING_NODE_TRACE_WRITER_H_

package/src/headers/26/tracing/trace_event.h

@@ -0,0 +1,27 @@
+#ifndef SRC_TRACING_TRACE_EVENT_H_
+#define SRC_TRACING_TRACE_EVENT_H_
+
+#include "tracing/agent.h"
+
+#if defined(V8_USE_PERFETTO)
+
+#define TRACING_CATEGORY_NODE "node"
+#define TRACING_CATEGORY_NODE1(one) TRACING_CATEGORY_NODE "." #one
+#define TRACING_CATEGORY_NODE2(one, two) TRACING_CATEGORY_NODE "." #one "." #two
+
+#include "tracing/trace_event_perfetto.h"
+
+#else  // defined(V8_USE_PERFETTO)
+
+#define TRACING_CATEGORY_NODE "node"
+#define TRACING_CATEGORY_NODE1(one)                                            \
+  TRACING_CATEGORY_NODE "," TRACING_CATEGORY_NODE "." #one
+#define TRACING_CATEGORY_NODE2(one, two)                                       \
+  TRACING_CATEGORY_NODE "," TRACING_CATEGORY_NODE "." #one                     \
+                        "," TRACING_CATEGORY_NODE "." #one "." #two
+
+#include "tracing/trace_event_legacy_inl.h"
+
+#endif
+
+#endif  // SRC_TRACING_TRACE_EVENT_H_

package/src/headers/26/tracing/trace_event_helper.h

@@ -0,0 +1,40 @@
+#ifndef SRC_TRACING_TRACE_EVENT_HELPER_H_
+#define SRC_TRACING_TRACE_EVENT_HELPER_H_
+
+#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
+
+#include "v8-platform.h"
+
+#include <string>
+#include <string_view>
+
+namespace node::tracing {
+
+// Expands a trace log file pattern into a concrete path, substituting ${pid}
+// and ${rotation}.
+std::string GetTraceFilePath(std::string_view log_file_pattern, int file_num);
+
+class TraceEventHelper {
+ public:
+  static v8::TracingController* GetTracingController();
+  static void SetTracingController(v8::TracingController* controller);
+
+  static inline const uint8_t* GetCategoryGroupEnabled(const char* group) {
+#if !defined(V8_USE_PERFETTO)
+    v8::TracingController* controller = GetTracingController();
+    static const uint8_t disabled = 0;
+    if (controller == nullptr) [[unlikely]] {
+      return &disabled;
+    }
+    return controller->GetCategoryGroupEnabled(group);
+#else
+    return nullptr;
+#endif
+  }
+};
+
+}  // namespace node::tracing
+
+#endif  // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
+
+#endif  // SRC_TRACING_TRACE_EVENT_HELPER_H_