1 files changed, 33 insertions, 14 deletions

diff --git a/listener/deal.go b/listener/deal.go
index 8f77431..8225224 100644
--- a/listener/deal.go
+++ b/listener/deal.go
@@ -3,6 +3,7 @@ package main
 import (
         "fmt"
         "os"
+        "strconv"
         "syscall"
         "time"
 
@@ -17,27 +18,45 @@ const (
 )
 
 var pidCol, fdCol, fileCol mongoClient
-var err error
+
+func initPidCol() (err error) {
+        // TODO: 这里是否需要补全一下进程信息？
+        dirs, err := os.ReadDir(fmt.Sprintf("/proc/%d/task", containerdPid))
+        if err != nil {
+                return err
+        }
+        for _, file := range dirs {
+                pid, _ := strconv.Atoi(file.Name())
+                process := Process{
+                        Ppid:       1,
+                        ParentTgid: 1,
+                        Pid:        pid,
+                        Tgid:       containerdPid,
+                        Cwd:        "/",
+                        Children:   make([]int, 0),
+                        Execve:     make([]Exec, 0),
+                        Args:       make([]string, 0),
+                }
+                if pid == containerdPid {
+                        process.Star = true
+                }
+                err = pidCol.InsertOne(process)
+        }
+        return nil
+}
 
 func deal() {
         defer wg.Done()
         var cooked Event
         var ok bool
+        var err error
 
         if err = pidCol.init(dbName, pidColName); err != nil {
                 fmt.Fprintf(os.Stderr, "Error while initing the mongodb: %v\n", err)
                 return
         }
-        err = pidCol.InsertOne(Process{
-                Ppid:     1,
-                Pid:      containerdPid,
-                Cwd:      "/",
-                Children: make([]int, 0),
-                Star:     true,
-        })
-        if err != nil {
-                fmt.Fprintf(os.Stderr, "Error while initing the mongodb: %v\n", err)
-                return
+        if err = initPidCol(); err != nil {
+                fmt.Fprintf(os.Stderr, "Err while initing pidcol: %v\n", err)
         }
 
         if err = fdCol.init(dbName, fdColName); err != nil {
@@ -96,7 +115,7 @@ func deletePid(cooked Event) {
 func dealNewPid(cooked Event) {
         // 自身是否已经记录
         var docRes []Process
-        err = pidCol.Finddoc(bson.M{"pid": cooked.pid}, &docRes)
+        err := pidCol.Finddoc(bson.M{"pid": cooked.pid}, &docRes)
         if err != nil {
                 fmt.Fprintf(os.Stderr, "Err finding: %v\n", err)
                 return
@@ -136,7 +155,7 @@ func dealNewPid(cooked Event) {
                 }
         }
 
-        err := pidCol.UpdateOne(bson.M{"pid": cooked.ppid}, bson.M{
+        err = pidCol.UpdateOne(bson.M{"pid": cooked.ppid}, bson.M{
                 "$push": bson.M{
                         "children": cooked.pid,
                 },
@@ -149,7 +168,7 @@ func dealNewPid(cooked Event) {
 func dealExecve(cooked Event) {
         var docRes []Process
         // 首先检查进程是否存在，如不存在则为之创建
-        err = pidCol.Finddoc(bson.M{"pid": cooked.pid}, &docRes)
+        err := pidCol.Finddoc(bson.M{"pid": cooked.pid}, &docRes)
         if err != nil {
                 return
         }