1 files changed, 27 insertions, 13 deletions

diff --git a/src/basefunc.go b/src/basefunc.go
index 5fff3e8..2f39507 100644
--- a/src/basefunc.go
+++ b/src/basefunc.go
@@ -4,32 +4,46 @@ import (
         "bufio"
         "fmt"
         "os"
+        "os/exec"
         "path/filepath"
-        "regexp"
         "strconv"
         "strings"
         "time"
 )
 
 func figureOutSyscalls() error {
-        NRRegex := regexp.MustCompile(`#define __NR_(.*?) (\d+)$`)
-        file, err := os.Open("/usr/include/asm/unistd_64.h")
+        cmd := exec.Command("ausyscall", "--dump")
+        stdout, err := cmd.StdoutPipe()
         if err != nil {
                 return err
         }
-        defer file.Close()
 
-        scanner := bufio.NewScanner(file)
-        for scanner.Scan() {
+        if err := cmd.Start(); err != nil {
+                return err
+        }
+
+        scanner := bufio.NewScanner(stdout)
+        for i := 0; scanner.Scan(); i++ {
+                if i == 0 {
+                        continue
+                }
                 line := scanner.Text()
-                if NRRegex.MatchString(line) {
-                        match := NRRegex.FindStringSubmatch(line)
-                        num, err := strconv.Atoi(match[2])
-                        if err != nil {
-                                return err
-                        }
-                        syscallTable[num] = match[1]
+                parts := strings.Split(line, "\t")
+                if len(parts) != 2 {
+                        return fmt.Errorf("invalid ausyscall format")
                 }
+                num, err := strconv.Atoi(parts[0])
+                if err != nil {
+                        return err
+                }
+                syscallTable[num] = parts[1]
+        }
+
+        if err := scanner.Err(); err != nil {
+                return err
+        }
+        if err := cmd.Wait(); err != nil {
+                return err
         }
         return nil
 }