Author: scottl
Date: Sun Mar 22 12:49:57 2015
New Revision: 280344
URL: https://svnweb.freebsd.org/changeset/base/280344

Log:
  MFC r279842, r279875
  
  Implement basic reporting of cpuset info via the -S option
  
  Obtained from:        Netflix, Inc.

Added:
  stable/10/usr.bin/procstat/procstat_cs.c
     - copied unchanged from r279842, head/usr.bin/procstat/procstat_cs.c
Modified:
  stable/10/usr.bin/procstat/Makefile
  stable/10/usr.bin/procstat/procstat.1
  stable/10/usr.bin/procstat/procstat.c
  stable/10/usr.bin/procstat/procstat.h
Directory Properties:
  stable/10/   (props changed)

Modified: stable/10/usr.bin/procstat/Makefile
==============================================================================
--- stable/10/usr.bin/procstat/Makefile Sun Mar 22 09:12:44 2015        
(r280343)
+++ stable/10/usr.bin/procstat/Makefile Sun Mar 22 12:49:57 2015        
(r280344)
@@ -8,6 +8,7 @@ SRCS=   procstat.c              \
        procstat_basic.c        \
        procstat_bin.c          \
        procstat_cred.c         \
+       procstat_cs.c           \
        procstat_files.c        \
        procstat_kstack.c       \
        procstat_rlimit.c       \

Modified: stable/10/usr.bin/procstat/procstat.1
==============================================================================
--- stable/10/usr.bin/procstat/procstat.1       Sun Mar 22 09:12:44 2015        
(r280343)
+++ stable/10/usr.bin/procstat/procstat.1       Sun Mar 22 12:49:57 2015        
(r280344)
@@ -25,7 +25,7 @@
 .\"
 .\" $FreeBSD$
 .\"
-.Dd May 16, 2014
+.Dd March 10, 2015
 .Dt PROCSTAT 1
 .Os
 .Sh NAME
@@ -35,7 +35,7 @@
 .Nm
 .Op Fl CHhn
 .Op Fl w Ar interval
-.Op Fl b | c | e | f | i | j | k | l | r | s | t | v | x
+.Op Fl b | c | e | f | i | j | k | l | r | s | S | t | v | x
 .Op Fl a | Ar pid | Ar core ...
 .Sh DESCRIPTION
 The
@@ -75,6 +75,8 @@ Display resource limits for the process.
 Display resource usage information for the process.
 .It Fl s
 Display security credential information for the process.
+.It Fl S
+Display the cpuset information for the thread.
 .It Fl t
 Display thread information for the process.
 .It Fl v
@@ -108,9 +110,16 @@ flag may be used to request per-thread s
 statistics for some options.
 For those options, the second field in the table will list the thread ID
 to which the row of information corresponds.
+The
+.Fl H
+flag is implied for the
+.Fl S
+mode.
 .Pp
-Some information, such as VM and file descriptor information, is available
+Information for VM, file descriptor, and cpuset options is available
 only to the owner of a process or the superuser.
+A cpuset value displayed as -1 means that the information is either invalid
+or not available.
 .Ss Binary Information
 Display the process ID, command, and path to the process binary:
 .Pp

Modified: stable/10/usr.bin/procstat/procstat.c
==============================================================================
--- stable/10/usr.bin/procstat/procstat.c       Sun Mar 22 09:12:44 2015        
(r280343)
+++ stable/10/usr.bin/procstat/procstat.c       Sun Mar 22 12:49:57 2015        
(r280344)
@@ -40,7 +40,7 @@
 #include "procstat.h"
 
 static int aflag, bflag, cflag, eflag, fflag, iflag, jflag, kflag, lflag, 
rflag;
-static int sflag, tflag, vflag, xflag;
+static int sflag, tflag, vflag, xflag, Sflag;
 int    hflag, nflag, Cflag, Hflag;
 
 static void
@@ -50,7 +50,7 @@ usage(void)
        fprintf(stderr, "usage: procstat [-CHhn] [-M core] [-N system] "
            "[-w interval] \n");
        fprintf(stderr, "                [-b | -c | -e | -f | -i | -j | -k | "
-           "-l | -r | -s | -t | -v | -x]\n");
+           "-l | -r | -s | -S | -t | -v | -x]\n");
        fprintf(stderr, "                [-a | pid | core ...]\n");
        exit(EX_USAGE);
 }
@@ -85,6 +85,8 @@ procstat(struct procstat *prstat, struct
                procstat_vm(prstat, kipp);
        else if (xflag)
                procstat_auxv(prstat, kipp);
+       else if (Sflag)
+               procstat_cs(prstat, kipp);
        else
                procstat_basic(kipp);
 }
@@ -128,7 +130,7 @@ main(int argc, char *argv[])
 
        interval = 0;
        memf = nlistf = NULL;
-       while ((ch = getopt(argc, argv, "CHN:M:abcefijklhrstvw:x")) != -1) {
+       while ((ch = getopt(argc, argv, "CHN:M:abcefijklhrsStvw:x")) != -1) {
                switch (ch) {
                case 'C':
                        Cflag++;
@@ -144,6 +146,9 @@ main(int argc, char *argv[])
                case 'N':
                        nlistf = optarg;
                        break;
+               case 'S':
+                       Sflag++;
+                       break;
                case 'a':
                        aflag++;
                        break;
@@ -228,7 +233,7 @@ main(int argc, char *argv[])
 
        /* We require that either 0 or 1 mode flags be set. */
        tmp = bflag + cflag + eflag + fflag + iflag + jflag + (kflag ? 1 : 0) +
-           lflag + rflag + sflag + tflag + vflag + xflag;
+           lflag + rflag + sflag + tflag + vflag + xflag + Sflag;
        if (!(tmp == 0 || tmp == 1))
                usage();
 

Modified: stable/10/usr.bin/procstat/procstat.h
==============================================================================
--- stable/10/usr.bin/procstat/procstat.h       Sun Mar 22 09:12:44 2015        
(r280343)
+++ stable/10/usr.bin/procstat/procstat.h       Sun Mar 22 12:49:57 2015        
(r280344)
@@ -39,6 +39,7 @@ void  procstat_auxv(struct procstat *prst
 void   procstat_basic(struct kinfo_proc *kipp);
 void   procstat_bin(struct procstat *prstat, struct kinfo_proc *kipp);
 void   procstat_cred(struct procstat *prstat, struct kinfo_proc *kipp);
+void   procstat_cs(struct procstat *prstat, struct kinfo_proc *kipp);
 void   procstat_env(struct procstat *prstat, struct kinfo_proc *kipp);
 void   procstat_files(struct procstat *prstat, struct kinfo_proc *kipp);
 void   procstat_kstack(struct procstat *prstat, struct kinfo_proc *kipp,

Copied: stable/10/usr.bin/procstat/procstat_cs.c (from r279842, 
head/usr.bin/procstat/procstat_cs.c)
==============================================================================
--- /dev/null   00:00:00 1970   (empty, because file is newly added)
+++ stable/10/usr.bin/procstat/procstat_cs.c    Sun Mar 22 12:49:57 2015        
(r280344, copy of r279842, head/usr.bin/procstat/procstat_cs.c)
@@ -0,0 +1,108 @@
+/*-
+ * Copyright (c) 2007 Robert N. M. Watson
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $FreeBSD$
+ */
+
+#include <sys/param.h>
+#include <sys/cpuset.h>
+#include <sys/sysctl.h>
+#include <sys/user.h>
+
+#include <err.h>
+#include <errno.h>
+#include <libprocstat.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "procstat.h"
+
+void
+procstat_cs(struct procstat *procstat, struct kinfo_proc *kipp)
+{
+       cpusetid_t cs;
+       cpuset_t mask;
+       struct kinfo_proc *kip;
+       unsigned int count, i;
+       int once, twice, lastcpu, cpu;
+
+       if (!hflag)
+               printf("%5s %6s %-16s %-16s %2s %4s %-7s\n", "PID",
+                   "TID", "COMM", "TDNAME", "CPU", "CSID", "CPU MASK");
+
+       kip = procstat_getprocs(procstat, KERN_PROC_PID | KERN_PROC_INC_THREAD,
+           kipp->ki_pid, &count);
+       if (kip == NULL)
+               return;
+       kinfo_proc_sort(kip, count);
+       for (i = 0; i < count; i++) {
+               kipp = &kip[i];
+               printf("%5d ", kipp->ki_pid);
+               printf("%6d ", kipp->ki_tid);
+               printf("%-16s ", strlen(kipp->ki_comm) ?
+                   kipp->ki_comm : "-");
+               printf("%-16s ", (strlen(kipp->ki_tdname) &&
+                   (strcmp(kipp->ki_comm, kipp->ki_tdname) != 0)) ?
+                   kipp->ki_tdname : "-");
+               if (kipp->ki_oncpu != 255)
+                       printf("%3d ", kipp->ki_oncpu);
+               else if (kipp->ki_lastcpu != 255)
+                       printf("%3d ", kipp->ki_lastcpu);
+               else
+                       printf("%3s ", "-");
+               if (cpuset_getid(CPU_LEVEL_CPUSET, CPU_WHICH_TID,
+                   kipp->ki_tid, &cs) != 0) {
+                       cs = CPUSET_INVALID;
+               }
+               printf("%4d ", cs);
+               if ((cs != CPUSET_INVALID) && 
+                   (cpuset_getaffinity(CPU_LEVEL_WHICH, CPU_WHICH_TID,
+                   kipp->ki_tid, sizeof(mask), &mask) == 0)) {
+                       lastcpu = -1;
+                       once = 0;
+                       twice = 0;
+                       for (cpu = 0; cpu < CPU_SETSIZE; cpu++) {
+                               if (CPU_ISSET(cpu, &mask)) {
+                                       if (once == 0) {
+                                               printf("%d", cpu);
+                                               once = 1;
+                                       } else if (cpu == lastcpu + 1) {
+                                               twice = 1;
+                                       } else if (twice == 1) {
+                                               printf("-%d,%d", lastcpu, cpu);
+                                               twice = 0;
+                                       } else
+                                               printf(",%d", cpu);
+                                       lastcpu = cpu;
+                               }
+                       }
+                       if (once && twice)
+                               printf("-%d", lastcpu);
+               }
+               printf("\n");
+       }
+       procstat_freeprocs(procstat, kip);
+}
_______________________________________________
svn-src-all@freebsd.org mailing list
http://lists.freebsd.org/mailman/listinfo/svn-src-all
To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"

Reply via email to