3 * Utility to get per-pid and per-tgid delay accounting statistics
4 * Also illustrates usage of the taskstats interface
6 * Copyright (C) Shailabh Nagar, IBM Corp. 2005
7 * Copyright (C) Balbir Singh, IBM Corp. 2006
8 * Copyright (c) Jay Lan, SGI. 2006
11 * gcc -I/usr/src/linux/include getdelays.c -o getdelays
21 #include <sys/types.h>
23 #include <sys/socket.h>
24 #include <sys/types.h>
27 #include <linux/genetlink.h>
28 #include <linux/taskstats.h>
31 * Generic macros for dealing with netlink sockets. Might be duplicated
32 * elsewhere. It is recommended that commercial grade applications use
33 * libnl or libnetlink and use the interfaces provided by the library
35 #define GENLMSG_DATA(glh) ((void *)(NLMSG_DATA(glh) + GENL_HDRLEN))
36 #define GENLMSG_PAYLOAD(glh) (NLMSG_PAYLOAD(glh, 0) - GENL_HDRLEN)
37 #define NLA_DATA(na) ((void *)((char*)(na) + NLA_HDRLEN))
38 #define NLA_PAYLOAD(len) (len - NLA_HDRLEN)
40 #define err(code, fmt, arg...) \
42 fprintf(stderr, fmt, ##arg); \
51 int print_io_accounting
;
54 #define PRINTF(fmt, arg...) { \
60 /* Maximum size of response requested or message sent */
61 #define MAX_MSG_SIZE 1024
62 /* Maximum number of cpus expected to be specified in a cpumask */
64 /* Maximum length of pathname to log file */
65 #define MAX_FILENAME 256
70 char buf
[MAX_MSG_SIZE
];
73 char cpumask
[100+6*MAX_CPUS
];
75 static void usage(void)
77 fprintf(stderr
, "getdelays [-dilv] [-w logfile] [-r bufsize] "
78 "[-m cpumask] [-t tgid] [-p pid]\n");
79 fprintf(stderr
, " -d: print delayacct stats\n");
80 fprintf(stderr
, " -i: print IO accounting (works only with -p)\n");
81 fprintf(stderr
, " -l: listen forever\n");
82 fprintf(stderr
, " -v: debug on\n");
86 * Create a raw netlink socket and bind
88 static int create_nl_socket(int protocol
)
91 struct sockaddr_nl local
;
93 fd
= socket(AF_NETLINK
, SOCK_RAW
, protocol
);
98 if (setsockopt(fd
, SOL_SOCKET
, SO_RCVBUF
,
99 &rcvbufsz
, sizeof(rcvbufsz
)) < 0) {
100 fprintf(stderr
, "Unable to set socket rcv buf size "
106 memset(&local
, 0, sizeof(local
));
107 local
.nl_family
= AF_NETLINK
;
109 if (bind(fd
, (struct sockaddr
*) &local
, sizeof(local
)) < 0)
119 int send_cmd(int sd
, __u16 nlmsg_type
, __u32 nlmsg_pid
,
120 __u8 genl_cmd
, __u16 nla_type
,
121 void *nla_data
, int nla_len
)
124 struct sockaddr_nl nladdr
;
128 struct msgtemplate msg
;
130 msg
.n
.nlmsg_len
= NLMSG_LENGTH(GENL_HDRLEN
);
131 msg
.n
.nlmsg_type
= nlmsg_type
;
132 msg
.n
.nlmsg_flags
= NLM_F_REQUEST
;
134 msg
.n
.nlmsg_pid
= nlmsg_pid
;
135 msg
.g
.cmd
= genl_cmd
;
137 na
= (struct nlattr
*) GENLMSG_DATA(&msg
);
138 na
->nla_type
= nla_type
;
139 na
->nla_len
= nla_len
+ 1 + NLA_HDRLEN
;
140 memcpy(NLA_DATA(na
), nla_data
, nla_len
);
141 msg
.n
.nlmsg_len
+= NLMSG_ALIGN(na
->nla_len
);
144 buflen
= msg
.n
.nlmsg_len
;
145 memset(&nladdr
, 0, sizeof(nladdr
));
146 nladdr
.nl_family
= AF_NETLINK
;
147 while ((r
= sendto(sd
, buf
, buflen
, 0, (struct sockaddr
*) &nladdr
,
148 sizeof(nladdr
))) < buflen
) {
152 } else if (errno
!= EAGAIN
)
160 * Probe the controller in genetlink to find the family id
161 * for the TASKSTATS family
163 int get_family_id(int sd
)
175 strcpy(name
, TASKSTATS_GENL_NAME
);
176 rc
= send_cmd(sd
, GENL_ID_CTRL
, getpid(), CTRL_CMD_GETFAMILY
,
177 CTRL_ATTR_FAMILY_NAME
, (void *)name
,
178 strlen(TASKSTATS_GENL_NAME
)+1);
180 rep_len
= recv(sd
, &ans
, sizeof(ans
), 0);
181 if (ans
.n
.nlmsg_type
== NLMSG_ERROR
||
182 (rep_len
< 0) || !NLMSG_OK((&ans
.n
), rep_len
))
185 na
= (struct nlattr
*) GENLMSG_DATA(&ans
);
186 na
= (struct nlattr
*) ((char *) na
+ NLA_ALIGN(na
->nla_len
));
187 if (na
->nla_type
== CTRL_ATTR_FAMILY_ID
) {
188 id
= *(__u16
*) NLA_DATA(na
);
193 void print_delayacct(struct taskstats
*t
)
195 printf("\n\nCPU %15s%15s%15s%15s\n"
196 " %15llu%15llu%15llu%15llu\n"
201 "count", "real total", "virtual total", "delay total",
202 t
->cpu_count
, t
->cpu_run_real_total
, t
->cpu_run_virtual_total
,
204 "count", "delay total",
205 t
->blkio_count
, t
->blkio_delay_total
,
206 "count", "delay total", t
->swapin_count
, t
->swapin_delay_total
);
209 void print_ioacct(struct taskstats
*t
)
211 printf("%s: read=%llu, write=%llu, cancelled_write=%llu\n",
213 (unsigned long long)t
->read_bytes
,
214 (unsigned long long)t
->write_bytes
,
215 (unsigned long long)t
->cancelled_write_bytes
);
218 int main(int argc
, char *argv
[])
220 int c
, rc
, rep_len
, aggr_len
, len2
, cmd_type
;
237 struct msgtemplate msg
;
240 c
= getopt(argc
, argv
, "diw:r:m:t:p:vl");
246 printf("print delayacct stats ON\n");
250 printf("printing IO accounting\n");
251 print_io_accounting
= 1;
254 strncpy(logfile
, optarg
, MAX_FILENAME
);
255 printf("write to file %s\n", logfile
);
259 rcvbufsz
= atoi(optarg
);
260 printf("receive buf size %d\n", rcvbufsz
);
262 err(1, "Invalid rcv buf size\n");
265 strncpy(cpumask
, optarg
, sizeof(cpumask
));
267 printf("cpumask %s maskset %d\n", cpumask
, maskset
);
272 err(1, "Invalid tgid\n");
273 cmd_type
= TASKSTATS_CMD_ATTR_TGID
;
278 err(1, "Invalid pid\n");
279 cmd_type
= TASKSTATS_CMD_ATTR_PID
;
282 printf("debug on\n");
286 printf("listen forever\n");
296 fd
= open(logfile
, O_WRONLY
| O_CREAT
| O_TRUNC
,
297 S_IRUSR
| S_IWUSR
| S_IRGRP
| S_IROTH
);
299 perror("Cannot open output file\n");
304 if ((nl_sd
= create_nl_socket(NETLINK_GENERIC
)) < 0)
305 err(1, "error creating Netlink socket\n");
309 id
= get_family_id(nl_sd
);
311 fprintf(stderr
, "Error getting family id, errno %d\n", errno
);
314 PRINTF("family id %d\n", id
);
317 rc
= send_cmd(nl_sd
, id
, mypid
, TASKSTATS_CMD_GET
,
318 TASKSTATS_CMD_ATTR_REGISTER_CPUMASK
,
319 &cpumask
, strlen(cpumask
) + 1);
320 PRINTF("Sent register cpumask, retval %d\n", rc
);
322 fprintf(stderr
, "error sending register cpumask\n");
328 rc
= send_cmd(nl_sd
, id
, mypid
, TASKSTATS_CMD_GET
,
329 cmd_type
, &tid
, sizeof(__u32
));
330 PRINTF("Sent pid/tgid, retval %d\n", rc
);
332 fprintf(stderr
, "error sending tid/tgid cmd\n");
340 rep_len
= recv(nl_sd
, &msg
, sizeof(msg
), 0);
341 PRINTF("received %d bytes\n", rep_len
);
344 fprintf(stderr
, "nonfatal reply error: errno %d\n",
348 if (msg
.n
.nlmsg_type
== NLMSG_ERROR
||
349 !NLMSG_OK((&msg
.n
), rep_len
)) {
350 struct nlmsgerr
*err
= NLMSG_DATA(&msg
);
351 fprintf(stderr
, "fatal reply error, errno %d\n",
356 PRINTF("nlmsghdr size=%d, nlmsg_len=%d, rep_len=%d\n",
357 sizeof(struct nlmsghdr
), msg
.n
.nlmsg_len
, rep_len
);
360 rep_len
= GENLMSG_PAYLOAD(&msg
.n
);
362 na
= (struct nlattr
*) GENLMSG_DATA(&msg
);
365 while (len
< rep_len
) {
366 len
+= NLA_ALIGN(na
->nla_len
);
367 switch (na
->nla_type
) {
368 case TASKSTATS_TYPE_AGGR_TGID
:
370 case TASKSTATS_TYPE_AGGR_PID
:
371 aggr_len
= NLA_PAYLOAD(na
->nla_len
);
373 /* For nested attributes, na follows */
374 na
= (struct nlattr
*) NLA_DATA(na
);
376 while (len2
< aggr_len
) {
377 switch (na
->nla_type
) {
378 case TASKSTATS_TYPE_PID
:
379 rtid
= *(int *) NLA_DATA(na
);
381 printf("PID\t%d\n", rtid
);
383 case TASKSTATS_TYPE_TGID
:
384 rtid
= *(int *) NLA_DATA(na
);
386 printf("TGID\t%d\n", rtid
);
388 case TASKSTATS_TYPE_STATS
:
391 print_delayacct((struct taskstats
*) NLA_DATA(na
));
392 if (print_io_accounting
)
393 print_ioacct((struct taskstats
*) NLA_DATA(na
));
395 if (write(fd
, NLA_DATA(na
), na
->nla_len
) < 0) {
396 err(1,"write error\n");
403 fprintf(stderr
, "Unknown nested"
408 len2
+= NLA_ALIGN(na
->nla_len
);
409 na
= (struct nlattr
*) ((char *) na
+ len2
);
414 fprintf(stderr
, "Unknown nla_type %d\n",
418 na
= (struct nlattr
*) (GENLMSG_DATA(&msg
) + len
);
423 rc
= send_cmd(nl_sd
, id
, mypid
, TASKSTATS_CMD_GET
,
424 TASKSTATS_CMD_ATTR_DEREGISTER_CPUMASK
,
425 &cpumask
, strlen(cpumask
) + 1);
426 printf("Sent deregister mask, retval %d\n", rc
);
428 err(rc
, "error sending deregister cpumask\n");