Replace StatusTuple::code() != 0 with !StatusTuple.ok() in examples/
authorYaxiong Zhao <yzhao@pixielabs.ai>
Mon, 13 Dec 2021 20:24:04 +0000 (12:24 -0800)
committeryonghong-song <ys114321@gmail.com>
Tue, 14 Dec 2021 02:45:57 +0000 (18:45 -0800)
13 files changed:
examples/cpp/CGroupTest.cc
examples/cpp/CPUDistribution.cc
examples/cpp/FollyRequestContextSwitch.cc
examples/cpp/HelloWorld.cc
examples/cpp/KFuncExample.cc
examples/cpp/KModRetExample.cc
examples/cpp/LLCStat.cc
examples/cpp/RandomRead.cc
examples/cpp/RecordMySQLQuery.cc
examples/cpp/SkLocalStorageIterator.cc
examples/cpp/TCPSendStack.cc
examples/cpp/TaskIterator.cc
examples/cpp/pyperf/PyPerfUtil.cc

index bfe156ca4bcf21c1a547c803d530c85b98821870..151b6d992687d47ba83993f86e4a880264f22f09 100644 (file)
@@ -49,21 +49,21 @@ int main(int argc, char** argv) {
 
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
 
   auto cgroup_array = bpf.get_cgroup_array("cgroup");
   auto update_res = cgroup_array.update_value(0, argv[1]);
-  if (update_res.code() != 0) {
+  if (!update_res.ok()) {
     std::cerr << update_res.msg() << std::endl;
     return 1;
   }
 
   auto attach_res =
       bpf.attach_kprobe("vfs_open", "on_vfs_open");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
@@ -76,7 +76,7 @@ int main(int argc, char** argv) {
     std::cout << line << std::endl;
 
   auto detach_res = bpf.detach_kprobe("vfs_open");
-  if (detach_res.code() != 0) {
+  if (!detach_res.ok()) {
     std::cerr << detach_res.msg() << std::endl;
     return 1;
   }
index 010339f66e7648ccb6e05d7e088ac89b954fd8e0..5f4acd9fc7947629c20a42eedc9f15296370052d 100644 (file)
@@ -61,14 +61,14 @@ int task_switch_event(struct pt_regs *ctx, struct task_struct *prev) {
 int main(int argc, char** argv) {
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
 
   auto attach_res =
       bpf.attach_kprobe("finish_task_switch", "task_switch_event");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
@@ -88,7 +88,7 @@ int main(int argc, char** argv) {
   }
 
   auto detach_res = bpf.detach_kprobe("finish_task_switch");
-  if (detach_res.code() != 0) {
+  if (!detach_res.ok()) {
     std::cerr << detach_res.msg() << std::endl;
     return 1;
   }
index d2ff02c549588fb0b0618dd68835e65e094ba207..2040a0c0d86f32b676cc806b9b6f7a3a0aaf1d7f 100644 (file)
@@ -93,13 +93,13 @@ int main(int argc, char** argv) {
   ebpf::BPF* bpf = new ebpf::BPF();
 
   auto init_res = bpf->init(BPF_PROGRAM, {}, {u});
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
 
   auto attach_res = bpf->attach_usdt_all();
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   } else {
@@ -107,7 +107,7 @@ int main(int argc, char** argv) {
   }
 
   auto open_res = bpf->open_perf_buffer("events", &handle_output);
-  if (open_res.code() != 0) {
+  if (!open_res.ok()) {
     std::cerr << open_res.msg() << std::endl;
     return 1;
   }
index e229ced3279cf5345aff029db075eaac291ac402..51356b34ec353fed22eba851585bc933e0627d12 100644 (file)
@@ -21,7 +21,7 @@ int on_sys_clone(void *ctx) {
 int main() {
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
@@ -31,7 +31,7 @@ int main() {
   std::string clone_fnname = bpf.get_syscall_fnname("clone");
 
   auto attach_res = bpf.attach_kprobe(clone_fnname, "on_sys_clone");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
@@ -43,7 +43,7 @@ int main() {
       std::cout << line << std::endl;
       // Detach the probe if we got at least one line.
       auto detach_res = bpf.detach_kprobe(clone_fnname);
-      if (detach_res.code() != 0) {
+      if (!detach_res.ok()) {
         std::cerr << detach_res.msg() << std::endl;
         return 1;
       }
index fed2b2cf59a918f9b2f57ae3d2b05d8c6366319c..9b6a2c090f9203f096b5e24eb10b4b0a508f4de2 100644 (file)
@@ -78,14 +78,14 @@ void handle_output(void *cb_cookie, void *data, int data_size) {
 int main() {
   ebpf::BPF bpf;
   auto res = bpf.init(BPF_PROGRAM);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
 
   int prog_fd;
   res = bpf.load_func("kfunc____x64_sys_openat", BPF_PROG_TYPE_TRACING, prog_fd);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -97,7 +97,7 @@ int main() {
   }
 
   res = bpf.load_func("kretfunc____x64_sys_openat", BPF_PROG_TYPE_TRACING, prog_fd);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -109,7 +109,7 @@ int main() {
   }
 
   auto open_res = bpf.open_perf_buffer("events", &handle_output);
-  if (open_res.code() != 0) {
+  if (!open_res.ok()) {
     std::cerr << open_res.msg() << std::endl;
     return 1;
   }
index b5c3a90dae0e06e15797c92e8c6e7d1c667ecc0e..3ebb7dc6c8a8452fb4e8ac8db37e1de4892ad2fd 100644 (file)
@@ -100,7 +100,7 @@ static int modify_return(ebpf::BPF &bpf) {
   int prog_fd;
   auto res = bpf.load_func("kmod_ret____x64_sys_openat",
                            BPF_PROG_TYPE_TRACING, prog_fd, BPF_F_SLEEPABLE);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -122,7 +122,7 @@ static int modify_return(ebpf::BPF &bpf) {
   uint32_t key = 0;
   struct fname_buf val;
   res = fname_table.get_value(key, val);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     close(attach_fd);
     std::cerr << res.msg() << std::endl;
     return 1;
@@ -138,7 +138,7 @@ static int not_modify_return(ebpf::BPF &bpf) {
   int prog_fd;
   auto res = bpf.load_func("kmod_ret__security_file_open",
                             BPF_PROG_TYPE_TRACING, prog_fd);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -159,7 +159,7 @@ static int not_modify_return(ebpf::BPF &bpf) {
   auto count_table = bpf.get_array_table<uint32_t>("count");
   uint32_t key = 0, val = 0;
   res = count_table.get_value(key, val);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     close(attach_fd);
     std::cerr << res.msg() << std::endl;
     return 1;
@@ -173,7 +173,7 @@ static int not_modify_return(ebpf::BPF &bpf) {
 int main() {
   ebpf::BPF bpf;
   auto res = bpf.init(BPF_PROGRAM);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -181,7 +181,7 @@ int main() {
   uint32_t key = 0, val = getpid();
   auto pid_table = bpf.get_array_table<uint32_t>("target_pid");
   res = pid_table.update_value(key, val);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
index b351f1dd8c756ac02dd756206edc716f3862dfdc..7e7e2082ecd711d7b670a68596a6a819cdaa8298 100644 (file)
@@ -73,7 +73,7 @@ struct event_t {
 int main(int argc, char** argv) {
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
@@ -81,13 +81,13 @@ int main(int argc, char** argv) {
   auto attach_ref_res =
       bpf.attach_perf_event(PERF_TYPE_HARDWARE, PERF_COUNT_HW_CACHE_REFERENCES,
                             "on_cache_ref", 100, 0);
-  if (attach_ref_res.code() != 0) {
+  if (!attach_ref_res.ok()) {
     std::cerr << attach_ref_res.msg() << std::endl;
     return 1;
   }
   auto attach_miss_res = bpf.attach_perf_event(
       PERF_TYPE_HARDWARE, PERF_COUNT_HW_CACHE_MISSES, "on_cache_miss", 100, 0);
-  if (attach_miss_res.code() != 0) {
+  if (!attach_miss_res.ok()) {
     std::cerr << attach_miss_res.msg() << std::endl;
     return 1;
   }
index 4851ddec6a7d8a4da8597549d17506d3154a91fa..f566a9f4efebe10e29b6f8b9bf165727c08fc51b 100644 (file)
@@ -106,14 +106,14 @@ int main(int argc, char** argv) {
 
   bpf = new ebpf::BPF(0, nullptr, true, "", allow_rlimit);
   auto init_res = bpf->init(BPF_PROGRAM, cflags, {});
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
   if (argc == 3) {
     auto cgroup_array = bpf->get_cgroup_array("cgroup");
     auto update_res = cgroup_array.update_value(0, argv[2]);
-    if (update_res.code() != 0) {
+    if (!update_res.ok()) {
       std::cerr << update_res.msg() << std::endl;
       return 1;
     }
@@ -121,13 +121,13 @@ int main(int argc, char** argv) {
 
   auto attach_res =
       bpf->attach_raw_tracepoint("urandom_read", "on_urandom_read");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
 
   auto open_res = bpf->open_perf_buffer("events", &handle_output);
-  if (open_res.code() != 0) {
+  if (!open_res.ok()) {
     std::cerr << open_res.msg() << std::endl;
     return 1;
   }
index 6d49eee9b6b63328be56be14fba4fa76a1b5392a..aea43b32764754d89fed154a818f154cf89ba7f7 100644 (file)
@@ -63,14 +63,14 @@ int main(int argc, char** argv) {
 
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
 
   auto attach_res =
       bpf.attach_uprobe(mysql_path, ALLOC_QUERY_FUNC, "probe_mysql_query");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
@@ -95,7 +95,7 @@ int main(int argc, char** argv) {
   }
 
   auto detach_res = bpf.detach_uprobe(mysql_path, ALLOC_QUERY_FUNC);
-  if (detach_res.code() != 0) {
+  if (!detach_res.ok()) {
     std::cerr << detach_res.msg() << std::endl;
     return 1;
   }
index fdcf1d5f079cb981f7922cbe9cb31827238986e9..88bead937fefa7ffae54f0097657a23c9e0891da 100644 (file)
@@ -88,7 +88,7 @@ struct info_t {
 int main() {
   ebpf::BPF bpf;
   auto res = bpf.init(BPF_PROGRAM);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
@@ -111,7 +111,7 @@ int main() {
   auto sk_table = bpf.get_sk_storage_table<unsigned long long>("sk_data_map");
 
   res = sk_table.update_value(sockfd1, v1);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << "sk_data_map sockfd1 update failure: " << res.msg() << std::endl;
     close(sockfd2);
     close(sockfd1);
@@ -119,7 +119,7 @@ int main() {
   }
 
   res = sk_table.update_value(sockfd2, v2);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << "sk_data_map sockfd2 update failure: " << res.msg() << std::endl;
     close(sockfd2);
     close(sockfd1);
@@ -128,7 +128,7 @@ int main() {
 
   int prog_fd;
   res = bpf.load_func("bpf_iter__bpf_sk_storage_map", BPF_PROG_TYPE_TRACING, prog_fd);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
index f7f150d5a234815093b1bd8bc65ccfe61af82cc5..68a1d216090480e21bbdf9f9c77a992be2642f66 100644 (file)
@@ -58,13 +58,13 @@ struct stack_key_t {
 int main(int argc, char** argv) {
   ebpf::BPF bpf;
   auto init_res = bpf.init(BPF_PROGRAM);
-  if (init_res.code() != 0) {
+  if (!init_res.ok()) {
     std::cerr << init_res.msg() << std::endl;
     return 1;
   }
 
   auto attach_res = bpf.attach_kprobe("tcp_sendmsg", "on_tcp_send");
-  if (attach_res.code() != 0) {
+  if (!attach_res.ok()) {
     std::cerr << attach_res.msg() << std::endl;
     return 1;
   }
@@ -77,7 +77,7 @@ int main(int argc, char** argv) {
   sleep(probe_time);
 
   auto detach_res = bpf.detach_kprobe("tcp_sendmsg");
-  if (detach_res.code() != 0) {
+  if (!detach_res.ok()) {
     std::cerr << detach_res.msg() << std::endl;
     return 1;
   }
index 3815e3aa141e1c9e69f0d125fedc2f16b81a9266..dc30663fb8f694fa7fb00e0d063a3235224ce9e4 100644 (file)
@@ -87,14 +87,14 @@ struct info_t {
 int main() {
   ebpf::BPF bpf;
   auto res = bpf.init(BPF_PROGRAM);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
 
   int prog_fd;
   res = bpf.load_func("bpf_iter__task", BPF_PROG_TYPE_TRACING, prog_fd);
-  if (res.code() != 0) {
+  if (!res.ok()) {
     std::cerr << res.msg() << std::endl;
     return 1;
   }
index 84af184088f0babcf988833c20e55ef1888671b8..312d01818ae29dcbbdfcf500da3e568e999ea1c5 100644 (file)
@@ -171,7 +171,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::init() {
                       std::to_string(kPythonStackProgIdx));
 
   auto initRes = bpf_.init(PYPERF_BPF_PROGRAM, cflags);
-  if (initRes.code() != 0) {
+  if (!initRes.ok()) {
     std::fprintf(stderr, "Failed to compiled PyPerf BPF programs: %s\n",
                  initRes.msg().c_str());
     return PyPerfResult::INIT_FAIL;
@@ -180,7 +180,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::init() {
   int progFd = -1;
   auto loadRes =
       bpf_.load_func(kPythonStackFuncName, BPF_PROG_TYPE_PERF_EVENT, progFd);
-  if (loadRes.code() != 0) {
+  if (!loadRes.ok()) {
     std::fprintf(stderr, "Failed to load BPF program %s: %s\n",
                  kPythonStackFuncName.c_str(), loadRes.msg().c_str());
     return PyPerfResult::INIT_FAIL;
@@ -188,7 +188,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::init() {
 
   auto progTable = bpf_.get_prog_table(kProgsTableName);
   auto updateRes = progTable.update_value(kPythonStackProgIdx, progFd);
-  if (updateRes.code() != 0) {
+  if (!updateRes.ok()) {
     std::fprintf(stderr,
                  "Failed to set BPF program %s FD %d to program table: %s\n",
                  kPythonStackFuncName.c_str(), progFd, updateRes.msg().c_str());
@@ -216,7 +216,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::init() {
   auto openRes = bpf_.open_perf_buffer(
       kSamplePerfBufName, &handleSampleCallback, &handleLostSamplesCallback,
       this, kPerfBufSizePages);
-  if (openRes.code() != 0) {
+  if (!openRes.ok()) {
     std::fprintf(stderr, "Unable to open Perf Buffer: %s\n",
                  openRes.msg().c_str());
     return PyPerfResult::PERF_BUF_OPEN_FAIL;
@@ -245,7 +245,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::profile(int64_t sampleRate,
   // Attach to CPU cycles
   auto attachRes =
       bpf_.attach_perf_event(0, 0, kOnEventFuncName, sampleRate, 0);
-  if (attachRes.code() != 0) {
+  if (!attachRes.ok()) {
     std::fprintf(stderr, "Attach to CPU cycles event failed: %s\n",
                  attachRes.msg().c_str());
     return PyPerfResult::EVENT_ATTACH_FAIL;
@@ -269,7 +269,7 @@ PyPerfUtil::PyPerfResult PyPerfUtil::profile(int64_t sampleRate,
 
   // Detach the event
   auto detachRes = bpf_.detach_perf_event(0, 0);
-  if (detachRes.code() != 0) {
+  if (!detachRes.ok()) {
     std::fprintf(stderr, "Detach CPU cycles event failed: %s\n",
                  detachRes.msg().c_str());
     return PyPerfResult::EVENT_DETACH_FAIL;