discovery_darwin.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. //go:build darwin
  2. package discovery
  3. import (
  4. "os/exec"
  5. "regexp"
  6. "sort"
  7. "strconv"
  8. "strings"
  9. )
  10. func findByPID() []int {
  11. var ports []int
  12. pidRegex := regexp.MustCompile(`\bopencode\b`)
  13. out, err := exec.Command("ps", "aux").Output()
  14. if err != nil {
  15. return ports
  16. }
  17. var pids []int
  18. for _, line := range strings.Split(string(out), "\n") {
  19. if pidRegex.MatchString(line) && !strings.Contains(line, "opencode-monitor") && !strings.Contains(line, "grep") {
  20. fields := strings.Fields(line)
  21. if len(fields) > 1 {
  22. if pid, err := strconv.Atoi(fields[1]); err == nil {
  23. pids = append(pids, pid)
  24. }
  25. }
  26. }
  27. }
  28. for _, pid := range pids {
  29. lsofOut, err := exec.Command("lsof", "-p", strconv.Itoa(pid), "-i", "TCP", "-n").Output()
  30. if err != nil {
  31. continue
  32. }
  33. for _, line := range strings.Split(string(lsofOut), "\n") {
  34. if strings.Contains(line, "LISTEN") {
  35. fields := strings.Fields(line)
  36. for _, f := range fields {
  37. if strings.Contains(f, ":") {
  38. parts := strings.Split(f, ":")
  39. if len(parts) >= 2 {
  40. if port, err := strconv.Atoi(parts[len(parts)-1]); err == nil {
  41. if !contains(ports, port) {
  42. ports = append(ports, port)
  43. }
  44. }
  45. }
  46. }
  47. }
  48. }
  49. }
  50. netstatOut, err := exec.Command("netstat", "-an").Output()
  51. if err == nil {
  52. for _, line := range strings.Split(string(netstatOut), "\n") {
  53. if strings.Contains(line, "LISTEN") {
  54. fields := strings.Fields(line)
  55. if len(fields) >= 4 {
  56. addr := fields[3]
  57. if strings.Contains(addr, ":") {
  58. parts := strings.Split(addr, ":")
  59. if len(parts) >= 2 {
  60. if port, err := strconv.Atoi(parts[len(parts)-1]); err == nil {
  61. if port > 1024 && port < 65535 && !contains(ports, port) {
  62. ports = append(ports, port)
  63. }
  64. }
  65. }
  66. }
  67. }
  68. }
  69. }
  70. }
  71. }
  72. sort.Ints(ports)
  73. return ports
  74. }
  75. func findByCmdline() []int {
  76. var ports []int
  77. out, err := exec.Command("ps", "aux").Output()
  78. if err != nil {
  79. return ports
  80. }
  81. for _, line := range strings.Split(string(out), "\n") {
  82. if strings.Contains(line, "opencode") && !strings.Contains(line, "opencode-monitor") && !strings.Contains(line, "grep") {
  83. if strings.Contains(line, "--port") {
  84. fields := strings.Fields(line)
  85. for i, f := range fields {
  86. if f == "--port" && i+1 < len(fields) {
  87. if port, err := strconv.Atoi(fields[i+1]); err == nil {
  88. if !contains(ports, port) {
  89. ports = append(ports, port)
  90. }
  91. }
  92. }
  93. }
  94. }
  95. }
  96. }
  97. sort.Ints(ports)
  98. return ports
  99. }
  100. func getListeningPorts(pid int) []int {
  101. var ports []int
  102. lsofOut, err := exec.Command("lsof", "-p", strconv.Itoa(pid), "-i", "TCP", "-n", "-P").Output()
  103. if err != nil {
  104. return ports
  105. }
  106. for _, line := range strings.Split(string(lsofOut), "\n") {
  107. if strings.Contains(line, "LISTEN") {
  108. fields := strings.Fields(line)
  109. for _, f := range fields {
  110. if strings.Contains(f, ":") {
  111. parts := strings.Split(f, ":")
  112. if len(parts) >= 2 {
  113. if port, err := strconv.Atoi(parts[len(parts)-1]); err == nil {
  114. if !contains(ports, port) {
  115. ports = append(ports, port)
  116. }
  117. }
  118. }
  119. }
  120. }
  121. }
  122. }
  123. return ports
  124. }