ucli no error on vsc

3 files changed, 176 insertions, 0 deletions

diff --git a/source/ucli/ucli-lib.cc b/source/ucli/ucli-lib.cc
new file mode 100644
index 0000000..92a3c65
--- /dev/null
+++ b/source/ucli/ucli-lib.cc
@@ -0,0 +1,32 @@
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+#include "ucli.h"
+
+/**
+ * @brief 调用ioctl
+ */
+long diag_call_ioctl(unsigned long request, unsigned long arg) {
+  long ret = 0;
+  int fd;
+
+  fd = open(DEVICE, O_RDWR, 0);
+  if (fd < 0) {
+    printf("open %s error!\n", DEVICE);
+    return -EEXIST;
+  }
+
+  ret = ioctl(fd, request, arg);
+  if (ret < 0) {
+    printf("call cmd %lx fail, ret is %ld\n", request, ret);
+    goto err;
+  }
+
+err:
+  close(fd);
+  return ret;
+}
+
diff --git a/source/ucli/ucli.cc b/source/ucli/ucli.cc
new file mode 100644
index 0000000..b949aa0
--- /dev/null
+++ b/source/ucli/ucli.cc
@@ -0,0 +1,119 @@
+/**
+ * various_monitor cli 命令行工具
+ */
+#include "ucli.h"
+
+#include <getopt.h>
+#include <string.h>
+
+#include <cstdio>
+
+static int buffer_extract(void *buf, unsigned int len) {
+  int *et_type;
+  variable_monitor_record *vm_record;
+  variable_monitor_task *tsk_info;
+  struct load_monitor_cpu_run *cpu_run;
+  static int seq = 0;
+
+  if (len == 0) return 0;
+
+  et_type = (int *)buf;
+  switch (*et_type) {
+    case 0:
+      if (len < sizeof(variable_monitor_record)) break;
+      vm_record = (variable_monitor_record *)buf;
+
+      printf("超出阈值：%d\n", vm_record->tv);
+
+      for (int i = 0; i < vm_record->threshold_num; i++) {
+        printf("\t: pid: %d, name: %s, ptr: %p, threshold:%d\n",
+               vm_record->threshold_record[i].task_id,
+               vm_record->threshold_record[i]
+                   .name,  // Assuming name is a null-terminated string
+               vm_record->threshold_record[i].ptr,
+               vm_record->threshold_record[i].threshold);
+      }
+      break;
+    case 1:
+      if (len < sizeof(variable_monitor_task)) break;
+      tsk_info = (variable_monitor_task *)buf;
+      seq++;
+
+      printf("##CGROUP:[%s]  %d      [%03d]  采样命中[%s]\n",
+             tsk_info->task.cgroup_buf, tsk_info->task.pid, seq,
+             tsk_info->task.state == 0 ? "R" : "D");
+
+      printk_task_brief(&tsk_info->task);
+      diag_printf_raw_stack(tsk_info->task.tgid, tsk_info->task.container_tgid,
+                            tsk_info->task.comm, &tsk_info->raw_stack);
+      diag_printf_kern_stack(&tsk_info->kern_stack);
+
+      printf("#*        0xffffffffffffff %s (UNKNOWN)\n", tsk_info->task.comm);
+      diag_printf_proc_chains(&tsk_info->proc_chains);
+      printf("##\n");
+
+      break;
+    default:
+      break;
+  }
+
+  return 0;
+}
+
+static void do_dump(const char *arg) {
+  static char variant_buf[50 * 1024 * 1024];
+  int len;
+  int ret = 0;
+
+  //  参数解析, 未实现/ 源码在 SOURCE/diagnose-tools/params_parse.h |
+  //  params_parse.cc
+  //   struct params_parser parse(arg);
+
+  struct diag_ioctl_dump_param dump_param = {
+      .user_ptr_len = &len,
+      .user_buf_len = 5 * 1024 * 1024,
+      .user_buf = variant_buf,
+  };
+
+  ret = diag_call_ioctl(1, (long)&dump_param);  // !todo arg -> #define
+  if (ret == 0) {
+    buffer_extract(variant_buf, len);
+  }
+}
+
+int main(int argc, char *argv[]) {
+  static const struct option long_options[] = {
+      {"help", no_argument, 0, 0},
+      {"report", no_argument, 0, 0},
+      //   {"pid", required_argument, 0, 0},
+      {0, 0, 0, 0}};
+  int c;
+
+  if (argc <= 1) {
+    // do something default
+    return 0;
+  }
+
+  while (1) {
+    int option_index = -1;
+
+    c = getopt_long_only(argc, argv, "", long_options, &option_index);
+    if (c == -1) {
+      break;
+    }
+
+    switch (option_index) {
+      case 0:  // help
+        // usage_pupil();
+        break;
+      case 1:  // report
+        do_dump(optarg ? optarg : "");
+        break;
+      default:
+        // usage_pupil();
+        break;
+    }
+  }
+
+  return 0;
+}
\ No newline at end of file
diff --git a/source/ucli/ucli.h b/source/ucli/ucli.h
new file mode 100644
index 0000000..f069537
--- /dev/null
+++ b/source/ucli/ucli.h
@@ -0,0 +1,25 @@
+#ifndef UAPI_H
+#define UAPI_H
+
+#include <cstddef>  // size_t
+
+#include "../module/monitor_trace.h"
+
+#define DEVICE "/dev/variable_monitor"
+
+struct diag_ioctl_dump_param {
+  int *user_ptr_len;
+  size_t user_buf_len;
+  void *user_buf;
+};
+
+long diag_call_ioctl(unsigned long request, unsigned long arg);
+
+// all print fun
+void printk_task_brief(task_detail *detail);
+void diag_printf_raw_stack(int pid, int ns_pid, const char *comm,
+                           raw_stack_detail *raw_stack);
+void diag_printf_kern_stack(kern_stack_detail *kern_stack);
+void diag_printf_proc_chains(proc_chains_detail *proc_chains);
+
+#endif /* UAPI_H */
\ No newline at end of file