1
2
3
4
5
6
7
8
9
10
11
12
13
14 package procfs
15
16 import (
17 "bufio"
18 "bytes"
19 "fmt"
20 "io"
21 "strconv"
22 "strings"
23
24 "github.com/prometheus/procfs/internal/fs"
25 "github.com/prometheus/procfs/internal/util"
26 )
27
28
29 type CPUStat struct {
30 User float64
31 Nice float64
32 System float64
33 Idle float64
34 Iowait float64
35 IRQ float64
36 SoftIRQ float64
37 Steal float64
38 Guest float64
39 GuestNice float64
40 }
41
42
43
44
45 type SoftIRQStat struct {
46 Hi uint64
47 Timer uint64
48 NetTx uint64
49 NetRx uint64
50 Block uint64
51 BlockIoPoll uint64
52 Tasklet uint64
53 Sched uint64
54 Hrtimer uint64
55 Rcu uint64
56 }
57
58
59 type Stat struct {
60
61 BootTime uint64
62
63 CPUTotal CPUStat
64
65 CPU map[int64]CPUStat
66
67 IRQTotal uint64
68
69 IRQ []uint64
70
71 ContextSwitches uint64
72
73 ProcessCreated uint64
74
75 ProcessesRunning uint64
76
77 ProcessesBlocked uint64
78
79 SoftIRQTotal uint64
80
81 SoftIRQ SoftIRQStat
82 }
83
84
85 func parseCPUStat(line string) (CPUStat, int64, error) {
86 cpuStat := CPUStat{}
87 var cpu string
88
89 count, err := fmt.Sscanf(line, "%s %f %f %f %f %f %f %f %f %f %f",
90 &cpu,
91 &cpuStat.User, &cpuStat.Nice, &cpuStat.System, &cpuStat.Idle,
92 &cpuStat.Iowait, &cpuStat.IRQ, &cpuStat.SoftIRQ, &cpuStat.Steal,
93 &cpuStat.Guest, &cpuStat.GuestNice)
94
95 if err != nil && err != io.EOF {
96 return CPUStat{}, -1, fmt.Errorf("%s: couldn't parse %q (cpu): %w", ErrFileParse, line, err)
97 }
98 if count == 0 {
99 return CPUStat{}, -1, fmt.Errorf("%w: couldn't parse %q (cpu): 0 elements parsed", ErrFileParse, line)
100 }
101
102 cpuStat.User /= userHZ
103 cpuStat.Nice /= userHZ
104 cpuStat.System /= userHZ
105 cpuStat.Idle /= userHZ
106 cpuStat.Iowait /= userHZ
107 cpuStat.IRQ /= userHZ
108 cpuStat.SoftIRQ /= userHZ
109 cpuStat.Steal /= userHZ
110 cpuStat.Guest /= userHZ
111 cpuStat.GuestNice /= userHZ
112
113 if cpu == "cpu" {
114 return cpuStat, -1, nil
115 }
116
117 cpuID, err := strconv.ParseInt(cpu[3:], 10, 64)
118 if err != nil {
119 return CPUStat{}, -1, fmt.Errorf("%s: couldn't parse %q (cpu/cpuid): %w", ErrFileParse, line, err)
120 }
121
122 return cpuStat, cpuID, nil
123 }
124
125
126 func parseSoftIRQStat(line string) (SoftIRQStat, uint64, error) {
127 softIRQStat := SoftIRQStat{}
128 var total uint64
129 var prefix string
130
131 _, err := fmt.Sscanf(line, "%s %d %d %d %d %d %d %d %d %d %d %d",
132 &prefix, &total,
133 &softIRQStat.Hi, &softIRQStat.Timer, &softIRQStat.NetTx, &softIRQStat.NetRx,
134 &softIRQStat.Block, &softIRQStat.BlockIoPoll,
135 &softIRQStat.Tasklet, &softIRQStat.Sched,
136 &softIRQStat.Hrtimer, &softIRQStat.Rcu)
137
138 if err != nil {
139 return SoftIRQStat{}, 0, fmt.Errorf("%s: couldn't parse %q (softirq): %w", ErrFileParse, line, err)
140 }
141
142 return softIRQStat, total, nil
143 }
144
145
146
147
148
149 func NewStat() (Stat, error) {
150 fs, err := NewFS(fs.DefaultProcMountPoint)
151 if err != nil {
152 return Stat{}, err
153 }
154 return fs.Stat()
155 }
156
157
158
159
160
161 func (fs FS) NewStat() (Stat, error) {
162 return fs.Stat()
163 }
164
165
166
167 func (fs FS) Stat() (Stat, error) {
168 fileName := fs.proc.Path("stat")
169 data, err := util.ReadFileNoStat(fileName)
170 if err != nil {
171 return Stat{}, err
172 }
173 procStat, err := parseStat(bytes.NewReader(data), fileName)
174 if err != nil {
175 return Stat{}, err
176 }
177 return procStat, nil
178 }
179
180
181 func parseStat(r io.Reader, fileName string) (Stat, error) {
182 var (
183 scanner = bufio.NewScanner(r)
184 stat = Stat{
185 CPU: make(map[int64]CPUStat),
186 }
187 err error
188 )
189
190
191 buf := make([]byte, 0, 8*1024)
192 scanner.Buffer(buf, 1024*1024)
193
194 for scanner.Scan() {
195 line := scanner.Text()
196 parts := strings.Fields(scanner.Text())
197
198 if len(parts) < 2 {
199 continue
200 }
201 switch {
202 case parts[0] == "btime":
203 if stat.BootTime, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
204 return Stat{}, fmt.Errorf("%s: couldn't parse %q (btime): %w", ErrFileParse, parts[1], err)
205 }
206 case parts[0] == "intr":
207 if stat.IRQTotal, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
208 return Stat{}, fmt.Errorf("%s: couldn't parse %q (intr): %w", ErrFileParse, parts[1], err)
209 }
210 numberedIRQs := parts[2:]
211 stat.IRQ = make([]uint64, len(numberedIRQs))
212 for i, count := range numberedIRQs {
213 if stat.IRQ[i], err = strconv.ParseUint(count, 10, 64); err != nil {
214 return Stat{}, fmt.Errorf("%s: couldn't parse %q (intr%d): %w", ErrFileParse, count, i, err)
215 }
216 }
217 case parts[0] == "ctxt":
218 if stat.ContextSwitches, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
219 return Stat{}, fmt.Errorf("%s: couldn't parse %q (ctxt): %w", ErrFileParse, parts[1], err)
220 }
221 case parts[0] == "processes":
222 if stat.ProcessCreated, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
223 return Stat{}, fmt.Errorf("%s: couldn't parse %q (processes): %w", ErrFileParse, parts[1], err)
224 }
225 case parts[0] == "procs_running":
226 if stat.ProcessesRunning, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
227 return Stat{}, fmt.Errorf("%s: couldn't parse %q (procs_running): %w", ErrFileParse, parts[1], err)
228 }
229 case parts[0] == "procs_blocked":
230 if stat.ProcessesBlocked, err = strconv.ParseUint(parts[1], 10, 64); err != nil {
231 return Stat{}, fmt.Errorf("%s: couldn't parse %q (procs_blocked): %w", ErrFileParse, parts[1], err)
232 }
233 case parts[0] == "softirq":
234 softIRQStats, total, err := parseSoftIRQStat(line)
235 if err != nil {
236 return Stat{}, err
237 }
238 stat.SoftIRQTotal = total
239 stat.SoftIRQ = softIRQStats
240 case strings.HasPrefix(parts[0], "cpu"):
241 cpuStat, cpuID, err := parseCPUStat(line)
242 if err != nil {
243 return Stat{}, err
244 }
245 if cpuID == -1 {
246 stat.CPUTotal = cpuStat
247 } else {
248 stat.CPU[cpuID] = cpuStat
249 }
250 }
251 }
252
253 if err := scanner.Err(); err != nil {
254 return Stat{}, fmt.Errorf("%s: couldn't parse %q: %w", ErrFileParse, fileName, err)
255 }
256
257 return stat, nil
258 }
259
View as plain text