1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
package main
import (
"fmt"
"log"
"syscall"
)
/*
func main() {
go listenForBlockEvents()
// Simulate other work
select {}
}
*/
func listenForBlockEvents() {
// Open netlink socket
fd, err := syscall.Socket(syscall.AF_NETLINK, syscall.SOCK_RAW, syscall.NETLINK_KOBJECT_UEVENT)
if err != nil {
log.Fatalf("Failed to open netlink socket: %v", err)
}
defer syscall.Close(fd)
sa := &syscall.SockaddrNetlink{
Family: syscall.AF_NETLINK,
Groups: 1, // receive broadcast group 1 (KOBJ events)
}
if err := syscall.Bind(fd, sa); err != nil {
log.Fatalf("Failed to bind netlink socket: %v", err)
}
buf := make([]byte, syscall.Getpagesize())
for {
n, _, err := syscall.Recvfrom(fd, buf, 0)
if err != nil {
log.Printf("Recvfrom error: %v", err)
continue
}
msg := parseUevent(buf[:n])
if msg["SUBSYSTEM"] == "block" && msg["ACTION"] == "add" {
fmt.Printf("New block device added: %s\n", msg["DEVNAME"])
}
}
}
// parseUevent parses the raw uevent buffer into a map
func parseUevent(buf []byte) map[string]string {
msg := make(map[string]string)
parts := bytesToStrings(buf)
for _, part := range parts {
kv := splitKV(part)
if len(kv) == 2 {
msg[kv[0]] = kv[1]
}
}
return msg
}
func bytesToStrings(buf []byte) []string {
var parts []string
start := 0
for i := 0; i < len(buf); i++ {
if buf[i] == 0 {
if i > start {
parts = append(parts, string(buf[start:i]))
}
start = i + 1
}
}
return parts
}
func splitKV(s string) []string {
if i := indexByte(s, '='); i >= 0 {
return []string{s[:i], s[i+1:]}
}
return []string{s}
}
func indexByte(s string, b byte) int {
for i := 0; i < len(s); i++ {
if s[i] == b {
return i
}
}
return -1
}
|
