From f00e11699edea8c7c9a707b11fc497097086c741 Mon Sep 17 00:00:00 2001
From: mutouyun <orz@orzz.org>
Date: Sun, 4 Jun 2023 18:17:36 +0800
Subject: [PATCH] add: (linux) benchmark for named pipe

---
 benchmark/benchmark_ipc.cpp | 57 +++++++++++++++++++++++++++
 test/ipc/test_ipc_shm.cpp   | 78 ++++++++++++++++++-------------------
 2 files changed, 96 insertions(+), 39 deletions(-)

diff --git a/benchmark/benchmark_ipc.cpp b/benchmark/benchmark_ipc.cpp
index 5d7a698..4415312 100644
--- a/benchmark/benchmark_ipc.cpp
+++ b/benchmark/benchmark_ipc.cpp
@@ -8,6 +8,7 @@
 #include <netinet/in.h>
 
 #include <string.h>
+#include <unistd.h>
 #include <fcntl.h>
 #include <mqueue.h>
 
@@ -95,6 +96,47 @@ struct mqueue_reader {
   }
 } mq_r__;
 
+struct pipe_reader {
+  pid_t pid_ {-1};
+
+  pipe_reader() {
+    auto shm = ipc::shared_memory("shm-pipe_reader", sizeof(flag_t));
+    shm.as<flag_t>()->store(false, std::memory_order_relaxed);
+
+    pid_ = test::subproc([this] {
+      auto shm = ipc::shared_memory("shm-pipe_reader", sizeof(flag_t));
+      auto flag = shm.as<flag_t>();
+      printf("pipe_reader start.\n");
+      mkfifo("/tmp/shm-pipe.w", S_IFIFO|0666);
+      mkfifo("/tmp/shm-pipe.r", S_IFIFO|0666);
+      int wfd = open("/tmp/shm-pipe.w", O_RDWR);
+      int rfd = open("/tmp/shm-pipe.r", O_RDWR);
+      while (!flag->load(std::memory_order_relaxed)) {
+        char n {};
+        // read
+        read(wfd, &n, sizeof(n));
+        // write
+        write(rfd, &n, sizeof(n));
+      }
+      printf("pipe_reader exit.\n");
+      close(wfd);
+      close(rfd);
+    });
+  }
+
+  ~pipe_reader() {
+    auto shm = ipc::shared_memory("shm-pipe_reader", sizeof(flag_t));
+    shm.as<flag_t>()->store(true, std::memory_order_seq_cst);
+    {
+      mqd_t wfd = open("/tmp/shm-pipe.w", O_WRONLY);
+      char n {};
+      write(wfd, &n, sizeof(n));
+      close(wfd);
+    }
+    test::join_subproc(pid_);
+  }
+} pipe_r__;
+
 struct sock_reader {
   pid_t pid_ {-1};
 
@@ -227,6 +269,20 @@ void ipc_mqueue_rtt(benchmark::State &state) {
   mq_close(rfd);
 }
 
+void ipc_npipe_rtt(benchmark::State &state) {
+  int wfd = open("/tmp/shm-pipe.w", O_WRONLY);
+  int rfd = open("/tmp/shm-pipe.r", O_RDONLY);
+  for (auto _ : state) {
+    char n {};
+    // write
+    write(wfd, &n, sizeof(n));
+    // read
+    read(rfd, &n, sizeof(n));
+  }
+  close(wfd);
+  close(rfd);
+}
+
 void ipc_sock_rtt(benchmark::State &state) {
   auto sfd = sock_r__.start_client();
   for (auto _ : state) {
@@ -259,5 +315,6 @@ void ipc_udp_rtt(benchmark::State &state) {
 
 BENCHMARK(ipc_eventfd_rtt);
 BENCHMARK(ipc_mqueue_rtt);
+BENCHMARK(ipc_npipe_rtt);
 BENCHMARK(ipc_sock_rtt);
 BENCHMARK(ipc_udp_rtt);
diff --git a/test/ipc/test_ipc_shm.cpp b/test/ipc/test_ipc_shm.cpp
index de8cb5d..6449431 100644
--- a/test/ipc/test_ipc_shm.cpp
+++ b/test/ipc/test_ipc_shm.cpp
@@ -80,56 +80,56 @@ TEST(shm, shared_memory) {
 #include <sys/socket.h>
 #include <sys/un.h>
 #include <sys/types.h>
+#include <sys/stat.h>
 #include <netinet/in.h>
 #include <string.h>
+#include <unistd.h>
+#include <fcntl.h>
 
 #include "test_util.h"
 
-TEST(shm, sock) {
-  auto reader = test::subproc([] {
-    int lfd = socket(AF_INET, SOCK_DGRAM, 0);
-    struct sockaddr_in ser {};
-    ser.sin_family = AF_INET;
-    ser.sin_addr.s_addr = htonl(INADDR_ANY);
-    ser.sin_port = htons(8888);
-    bind(lfd, (struct sockaddr *)&ser, sizeof(ser));
-    printf("reader prepared...\n");
-    struct sockaddr_in cli {};
-    socklen_t cli_len = sizeof(cli);
-    char c {'\0'};
-    printf("reading...\n");
-    for (;;) {
-      recvfrom(lfd, &c, sizeof(c), 0, (struct sockaddr *)&cli, &cli_len);
-      printf("read %c\n", c);
-      sendto(lfd, &c, sizeof(c), 0, (struct sockaddr *)&cli, cli_len);
-      if (c == 'Z') break;
-    }
-    close(lfd);
-  });
-
+TEST(shm, pipe) {
   auto writer = test::subproc([] {
-    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
-    struct sockaddr_in ser {};
-    ser.sin_family = AF_INET;
-    ser.sin_addr.s_addr = htonl(INADDR_ANY);
-    ser.sin_port = htons(8888);
-
+    mkfifo("/tmp/shm-pipe.w", S_IFIFO|0666);
+    // mkfifo("/tmp/shm-pipe.r", S_IFIFO|0666);
+    int wfd = open("/tmp/shm-pipe.w", O_WRONLY);
+    // int rfd = open("/tmp/shm-pipe.r", O_RDONLY);
     printf("writer prepared...\n");
-    sleep(1);
-
     for (char c = 'A'; c <= 'Z'; ++c) {
-      printf("write %c\n", c);
-      sendto(sfd, &c, sizeof(c), 0, (struct sockaddr *)&ser, sizeof(ser));
-      struct sockaddr_in cli {};
-      socklen_t len = sizeof(cli);
-      char n {};
-      recvfrom(sfd, &n, sizeof(n), 0, (struct sockaddr *)&cli, &len);
-      printf("echo %c\n", c);
+      sleep(1);
+      printf("\nwrite %c\n", c);
+      write(wfd, &c, sizeof(c));
+      // char n {};
+      // read(rfd, &n, sizeof(n));
+      // printf("write echo %c\n", n);
     }
-    close(sfd);
+    close(wfd);
+    // close(rfd);
   });
 
+  auto reader_maker = [](int k) {
+    return [k] {
+      printf("r%d prepared...\n", k);
+      sleep(1);
+      int wfd = open("/tmp/shm-pipe.w", O_RDONLY);
+      // int rfd = open("/tmp/shm-pipe.r", O_WRONLY);
+      for (;;) {
+        char n {};
+        read(wfd, &n, sizeof(n));
+        printf("r%d %c\n", k, n);
+        // write(rfd, &n, sizeof(n));
+        if (n == 'Z') break;
+      }
+      close(wfd);
+      // close(rfd);
+    };
+  };
+
+  auto r1 = test::subproc(reader_maker(1));
+  auto r2 = test::subproc(reader_maker(2));
+
   test::join_subproc(writer);
-  test::join_subproc(reader);
+  test::join_subproc(r1);
+  test::join_subproc(r2);
 }
 #endif
\ No newline at end of file