[BACK]Return to dkstats.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / vmstat

Annotation of src/usr.bin/vmstat/dkstats.c, Revision 1.34

1.34    ! lum         1: /*     $OpenBSD: dkstats.c,v 1.33 2007/11/26 09:28:34 martynas Exp $   */
1.2       deraadt     2: /*     $NetBSD: dkstats.c,v 1.1 1996/05/10 23:19:27 thorpej Exp $      */
1.1       tholo       3:
                      4: /*
1.2       deraadt     5:  * Copyright (c) 1996 John M. Vinopal
1.1       tholo       6:  * All rights reserved.
                      7:  *
                      8:  * Redistribution and use in source and binary forms, with or without
                      9:  * modification, are permitted provided that the following conditions
                     10:  * are met:
                     11:  * 1. Redistributions of source code must retain the above copyright
                     12:  *    notice, this list of conditions and the following disclaimer.
                     13:  * 2. Redistributions in binary form must reproduce the above copyright
                     14:  *    notice, this list of conditions and the following disclaimer in the
                     15:  *    documentation and/or other materials provided with the distribution.
                     16:  * 3. All advertising materials mentioning features or use of this software
                     17:  *    must display the following acknowledgement:
                     18:  *      This product includes software developed for the NetBSD Project
                     19:  *      by John M. Vinopal.
                     20:  * 4. The name of the author may not be used to endorse or promote products
                     21:  *    derived from this software without specific prior written permission.
                     22:  *
                     23:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
                     24:  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
                     25:  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
                     26:  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
                     27:  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
                     28:  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
                     29:  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
                     30:  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
                     31:  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     32:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     33:  * SUCH DAMAGE.
                     34:  */
                     35:
1.10      angelos    36: #include <sys/param.h>
1.1       tholo      37: #include <sys/dkstat.h>
                     38: #include <sys/time.h>
                     39: #include <sys/disk.h>
1.10      angelos    40: #include <sys/sysctl.h>
                     41: #include <sys/tty.h>
1.1       tholo      42:
                     43: #include <err.h>
                     44: #include <fcntl.h>
                     45: #include <kvm.h>
                     46: #include <limits.h>
                     47: #include <nlist.h>
                     48: #include <stdio.h>
                     49: #include <stdlib.h>
                     50: #include <string.h>
                     51: #include <unistd.h>
                     52: #include "dkstats.h"
                     53:
1.16      deraadt    54: #if !defined(NOKVM)
1.1       tholo      55: static struct nlist namelist[] = {
1.18      deraadt    56: #define        X_TK_NIN        0               /* sysctl */
                     57:        { "_tk_nin" },
                     58: #define        X_TK_NOUT       1               /* sysctl */
                     59:        { "_tk_nout" },
                     60: #define        X_CP_TIME       2               /* sysctl */
                     61:        { "_cp_time" },
                     62: #define        X_HZ            3               /* sysctl */
                     63:        { "_hz" },
                     64: #define        X_STATHZ        4               /* sysctl */
1.1       tholo      65:        { "_stathz" },
1.18      deraadt    66: #define X_DISK_COUNT   5               /* sysctl */
                     67:        { "_disk_count" },
                     68: #define X_DISKLIST     6               /* sysctl */
                     69:        { "_disklist" },
1.1       tholo      70:        { NULL },
                     71: };
1.16      deraadt    72: #define        KVM_ERROR(_string) {                                            \
                     73:        warnx("%s", (_string));                                         \
                     74:        errx(1, "%s", kvm_geterr(kd));                                  \
                     75: }
                     76:
                     77: /*
1.30      deraadt    78:  * Dereference the namelist pointer `v' and fill in the local copy
1.16      deraadt    79:  * 'p' which is of size 's'.
                     80:  */
                     81: #define deref_nl(v, p, s) deref_kptr((void *)namelist[(v)].n_value, (p), (s));
                     82: static void deref_kptr(void *, void *, size_t);
                     83: #endif /* !defined(NOKVM) */
1.1       tholo      84:
                     85: /* Structures to hold the statistics. */
                     86: struct _disk   cur, last;
                     87:
                     88: /* Kernel pointers: nlistf and memf defined in calling program. */
1.19      deraadt    89: #if !defined(NOKVM)
1.20      deraadt    90: extern kvm_t   *kd;
1.19      deraadt    91: #endif
1.1       tholo      92: extern char    *nlistf;
                     93: extern char    *memf;
                     94:
1.19      deraadt    95: #if !defined(NOKVM)
1.1       tholo      96: /* Pointer to list of disks. */
                     97: static struct disk     *dk_drivehead = NULL;
1.19      deraadt    98: #endif
1.1       tholo      99:
                    100: /* Backward compatibility references. */
1.30      deraadt   101: int            dk_ndrive = 0;
1.1       tholo     102: int            *dk_select;
                    103: char           **dr_name;
                    104:
                    105: /* Missing from <sys/time.h> */
1.16      deraadt   106: #define timerset(tvp, uvp) \
                    107:        ((uvp)->tv_sec = (tvp)->tv_sec);                \
                    108:        ((uvp)->tv_usec = (tvp)->tv_usec)
                    109:
                    110: #define SWAP(fld)      tmp = cur.fld;                          \
                    111:                        cur.fld -= last.fld;                    \
                    112:                        last.fld = tmp
1.1       tholo     113:
                    114: /*
                    115:  * Take the delta between the present values and the last recorded
                    116:  * values, storing the present values in the 'last' structure, and
                    117:  * the delta values in the 'cur' structure.
                    118:  */
                    119: void
1.21      deraadt   120: dkswap(void)
1.1       tholo     121: {
                    122:        u_int64_t tmp;
                    123:        int     i;
                    124:
1.22      tdeval    125:        for (i = 0; i < cur.dk_ndrive; i++) {
1.1       tholo     126:                struct timeval  tmp_timer;
                    127:
                    128:                if (!cur.dk_select[i])
                    129:                        continue;
                    130:
                    131:                /* Delta Values. */
1.24      tedu      132:                SWAP(dk_rxfer[i]);
                    133:                SWAP(dk_wxfer[i]);
1.1       tholo     134:                SWAP(dk_seek[i]);
1.24      tedu      135:                SWAP(dk_rbytes[i]);
                    136:                SWAP(dk_wbytes[i]);
1.1       tholo     137:
                    138:                /* Delta Time. */
                    139:                timerclear(&tmp_timer);
                    140:                timerset(&(cur.dk_time[i]), &tmp_timer);
                    141:                timersub(&tmp_timer, &(last.dk_time[i]), &(cur.dk_time[i]));
                    142:                timerclear(&(last.dk_time[i]));
                    143:                timerset(&tmp_timer, &(last.dk_time[i]));
                    144:        }
                    145:        for (i = 0; i < CPUSTATES; i++) {
1.29      deraadt   146:                long ltmp;
                    147:
                    148:                ltmp = cur.cp_time[i];
                    149:                cur.cp_time[i] -= last.cp_time[i];
                    150:                last.cp_time[i] = ltmp;
1.1       tholo     151:        }
                    152:        SWAP(tk_nin);
                    153:        SWAP(tk_nout);
                    154:
                    155: #undef SWAP
                    156: }
                    157:
                    158: /*
1.30      deraadt   159:  * Read the disk statistics for each disk in the disk list.
1.1       tholo     160:  * Also collect statistics for tty i/o and cpu ticks.
                    161:  */
                    162: void
1.21      deraadt   163: dkreadstats(void)
1.1       tholo     164: {
1.19      deraadt   165: #if !defined(NOKVM)
1.1       tholo     166:        struct disk     cur_disk, *p;
1.19      deraadt   167: #endif
1.22      tdeval    168:        int             i, j, mib[3];
1.10      angelos   169:        size_t          size;
1.22      tdeval    170:        char            *disknames, *name, *bufpp, **dk_name;
1.13      angelos   171:        struct diskstats *q;
1.1       tholo     172:
1.22      tdeval    173:        last.dk_ndrive = cur.dk_ndrive;
                    174:
1.10      angelos   175:        if (nlistf == NULL && memf == NULL) {
1.22      tdeval    176:                /* Get the number of attached drives. */
                    177:                mib[0] = CTL_HW;
                    178:                mib[1] = HW_DISKCOUNT;
                    179:                size = sizeof(dk_ndrive);
                    180:                if (sysctl(mib, 2, &dk_ndrive, &size, NULL, 0) < 0 ) {
                    181:                        warn("could not read hw.diskcount");
                    182:                        dk_ndrive = 0;
                    183:                }
                    184:
                    185:                if (cur.dk_ndrive != dk_ndrive) {
                    186:                        /* Re-read the disk names. */
1.29      deraadt   187:                        dk_name = calloc((size_t)dk_ndrive, sizeof(char *));
1.22      tdeval    188:                        if (dk_name == NULL)
                    189:                                err(1, NULL);
                    190:                        mib[0] = CTL_HW;
                    191:                        mib[1] = HW_DISKNAMES;
                    192:                        size = 0;
                    193:                        if (sysctl(mib, 2, NULL, &size, NULL, 0) < 0)
                    194:                                err(1, "can't get hw.disknames");
                    195:                        disknames = malloc(size);
                    196:                        if (disknames == NULL)
                    197:                                err(1, NULL);
                    198:                        if (sysctl(mib, 2, disknames, &size, NULL, 0) < 0)
                    199:                                err(1, "can't get hw.disknames");
                    200:                        bufpp = disknames;
1.23      millert   201:                        for (i = 0; i < dk_ndrive && (name = strsep(&bufpp, ",")) != NULL; i++)
                    202:                                dk_name[i] = name;
1.22      tdeval    203:                        disknames = cur.dk_name[0];     /* To free old names. */
                    204:
                    205:                        if (dk_ndrive < cur.dk_ndrive) {
                    206:                                for (i = 0, j = 0; i < dk_ndrive; i++, j++) {
                    207:                                        while (j < cur.dk_ndrive &&
                    208:                                            strcmp(cur.dk_name[j], dk_name[i]))
                    209:                                                j++;
                    210:                                        if (i == j) continue;
                    211:
                    212:                                        if (j >= cur.dk_ndrive) {
                    213:                                                cur.dk_select[i] = 1;
1.24      tedu      214:                                                last.dk_rxfer[i] = 0;
                    215:                                                last.dk_wxfer[i] = 0;
1.22      tdeval    216:                                                last.dk_seek[i] = 0;
1.24      tedu      217:                                                last.dk_rbytes[i] = 0;
                    218:                                                last.dk_wbytes[i] = 0;
1.22      tdeval    219:                                                bzero(&last.dk_time[i],
                    220:                                                    sizeof(struct timeval));
                    221:                                                continue;
                    222:                                        }
                    223:
                    224:                                        cur.dk_select[i] = cur.dk_select[j];
1.24      tedu      225:                                        last.dk_rxfer[i] = last.dk_rxfer[j];
                    226:                                        last.dk_wxfer[i] = last.dk_wxfer[j];
1.22      tdeval    227:                                        last.dk_seek[i] = last.dk_seek[j];
1.24      tedu      228:                                        last.dk_rbytes[i] = last.dk_rbytes[j];
                    229:                                        last.dk_wbytes[i] = last.dk_wbytes[j];
1.22      tdeval    230:                                        last.dk_time[i] = last.dk_time[j];
                    231:                                }
                    232:
                    233:                                cur.dk_select = realloc(cur.dk_select,
                    234:                                    dk_ndrive * sizeof(*cur.dk_select));
1.24      tedu      235:                                cur.dk_rxfer = realloc(cur.dk_rxfer,
                    236:                                    dk_ndrive * sizeof(*cur.dk_rxfer));
                    237:                                cur.dk_wxfer = realloc(cur.dk_wxfer,
                    238:                                    dk_ndrive * sizeof(*cur.dk_wxfer));
1.22      tdeval    239:                                cur.dk_seek = realloc(cur.dk_seek,
                    240:                                    dk_ndrive * sizeof(*cur.dk_seek));
1.24      tedu      241:                                cur.dk_rbytes = realloc(cur.dk_rbytes,
                    242:                                    dk_ndrive * sizeof(*cur.dk_rbytes));
                    243:                                cur.dk_wbytes = realloc(cur.dk_wbytes,
                    244:                                    dk_ndrive * sizeof(*cur.dk_wbytes));
1.22      tdeval    245:                                cur.dk_time = realloc(cur.dk_time,
                    246:                                    dk_ndrive * sizeof(*cur.dk_time));
1.24      tedu      247:                                last.dk_rxfer = realloc(last.dk_rxfer,
                    248:                                    dk_ndrive * sizeof(*last.dk_rxfer));
1.25      millert   249:                                last.dk_wxfer = realloc(last.dk_wxfer,
                    250:                                    dk_ndrive * sizeof(*last.dk_wxfer));
1.22      tdeval    251:                                last.dk_seek = realloc(last.dk_seek,
                    252:                                    dk_ndrive * sizeof(*last.dk_seek));
1.24      tedu      253:                                last.dk_rbytes = realloc(last.dk_rbytes,
                    254:                                    dk_ndrive * sizeof(*last.dk_rbytes));
                    255:                                last.dk_wbytes = realloc(last.dk_wbytes,
                    256:                                    dk_ndrive * sizeof(*last.dk_wbytes));
1.22      tdeval    257:                                last.dk_time = realloc(last.dk_time,
                    258:                                    dk_ndrive * sizeof(*last.dk_time));
1.32      chl       259:
                    260:                                if (!cur.dk_select || !cur.dk_rxfer ||
                    261:                                    !cur.dk_wxfer || !cur.dk_seek ||
                    262:                                    !cur.dk_rbytes || !cur.dk_wbytes ||
                    263:                                    !cur.dk_time || !last.dk_rxfer ||
                    264:                                    !last.dk_wxfer || !last.dk_seek ||
                    265:                                    !last.dk_rbytes || !last.dk_wbytes ||
                    266:                                    !last.dk_time)
                    267:                                        errx(1, "Memory allocation failure.");
1.22      tdeval    268:                        } else {
                    269:                                cur.dk_select = realloc(cur.dk_select,
                    270:                                    dk_ndrive * sizeof(*cur.dk_select));
1.24      tedu      271:                                cur.dk_rxfer = realloc(cur.dk_rxfer,
                    272:                                    dk_ndrive * sizeof(*cur.dk_rxfer));
                    273:                                cur.dk_wxfer = realloc(cur.dk_wxfer,
                    274:                                    dk_ndrive * sizeof(*cur.dk_wxfer));
1.22      tdeval    275:                                cur.dk_seek = realloc(cur.dk_seek,
                    276:                                    dk_ndrive * sizeof(*cur.dk_seek));
1.24      tedu      277:                                cur.dk_rbytes = realloc(cur.dk_rbytes,
                    278:                                    dk_ndrive * sizeof(*cur.dk_rbytes));
                    279:                                cur.dk_wbytes = realloc(cur.dk_wbytes,
                    280:                                    dk_ndrive * sizeof(*cur.dk_wbytes));
1.22      tdeval    281:                                cur.dk_time = realloc(cur.dk_time,
                    282:                                    dk_ndrive * sizeof(*cur.dk_time));
1.24      tedu      283:                                last.dk_rxfer = realloc(last.dk_rxfer,
                    284:                                    dk_ndrive * sizeof(*last.dk_rxfer));
                    285:                                last.dk_wxfer = realloc(last.dk_wxfer,
                    286:                                    dk_ndrive * sizeof(*last.dk_wxfer));
1.22      tdeval    287:                                last.dk_seek = realloc(last.dk_seek,
                    288:                                    dk_ndrive * sizeof(*last.dk_seek));
1.24      tedu      289:                                last.dk_rbytes = realloc(last.dk_rbytes,
                    290:                                    dk_ndrive * sizeof(*last.dk_rbytes));
                    291:                                last.dk_wbytes = realloc(last.dk_wbytes,
                    292:                                    dk_ndrive * sizeof(*last.dk_wbytes));
1.22      tdeval    293:                                last.dk_time = realloc(last.dk_time,
                    294:                                    dk_ndrive * sizeof(*last.dk_time));
1.32      chl       295:
                    296:                                if (!cur.dk_select || !cur.dk_rxfer ||
                    297:                                    !cur.dk_wxfer || !cur.dk_seek ||
                    298:                                    !cur.dk_rbytes || !cur.dk_wbytes ||
                    299:                                    !cur.dk_time || !last.dk_rxfer ||
                    300:                                    !last.dk_wxfer || !last.dk_seek ||
                    301:                                    !last.dk_rbytes || !last.dk_wbytes ||
                    302:                                    !last.dk_time)
                    303:                                        errx(1, "Memory allocation failure.");
1.22      tdeval    304:
                    305:                                for (i = dk_ndrive - 1, j = cur.dk_ndrive - 1;
                    306:                                     i >= 0; i--) {
                    307:
                    308:                                        if (j < 0 ||
                    309:                                            strcmp(cur.dk_name[j], dk_name[i]))
                    310:                                        {
                    311:                                                cur.dk_select[i] = 1;
1.24      tedu      312:                                                last.dk_rxfer[i] = 0;
                    313:                                                last.dk_wxfer[i] = 0;
1.22      tdeval    314:                                                last.dk_seek[i] = 0;
1.24      tedu      315:                                                last.dk_rbytes[i] = 0;
                    316:                                                last.dk_wbytes[i] = 0;
1.22      tdeval    317:                                                bzero(&last.dk_time[i],
                    318:                                                    sizeof(struct timeval));
                    319:                                                continue;
                    320:                                        }
                    321:
                    322:                                        if (i > j) {
                    323:                                                cur.dk_select[i] =
                    324:                                                    cur.dk_select[j];
1.24      tedu      325:                                                last.dk_rxfer[i] =
                    326:                                                    last.dk_rxfer[j];
                    327:                                                last.dk_wxfer[i] =
                    328:                                                    last.dk_wxfer[j];
1.22      tdeval    329:                                                last.dk_seek[i] =
                    330:                                                    last.dk_seek[j];
1.24      tedu      331:                                                last.dk_rbytes[i] =
                    332:                                                    last.dk_rbytes[j];
                    333:                                                last.dk_wbytes[i] =
                    334:                                                    last.dk_wbytes[j];
1.22      tdeval    335:                                                last.dk_time[i] =
                    336:                                                    last.dk_time[j];
                    337:                                        }
                    338:                                        j--;
                    339:                                }
                    340:                        }
                    341:
                    342:                        cur.dk_ndrive = dk_ndrive;
                    343:                        free(disknames);
                    344:                        cur.dk_name = dk_name;
                    345:                        dr_name = cur.dk_name;
                    346:                        dk_select = cur.dk_select;
                    347:                }
                    348:
                    349:                size = cur.dk_ndrive * sizeof(struct diskstats);
1.10      angelos   350:                mib[0] = CTL_HW;
                    351:                mib[1] = HW_DISKSTATS;
1.13      angelos   352:                q = malloc(size);
                    353:                if (q == NULL)
1.10      angelos   354:                        err(1, NULL);
1.13      angelos   355:                if (sysctl(mib, 2, q, &size, NULL, 0) < 0) {
1.22      tdeval    356: #ifdef DEBUG
1.10      angelos   357:                        warn("could not read hw.diskstats");
1.22      tdeval    358: #endif /* DEBUG */
                    359:                        bzero(q, cur.dk_ndrive * sizeof(struct diskstats));
1.10      angelos   360:                }
                    361:
1.22      tdeval    362:                for (i = 0; i < cur.dk_ndrive; i++)     {
1.24      tedu      363:                        cur.dk_rxfer[i] = q[i].ds_rxfer;
                    364:                        cur.dk_wxfer[i] = q[i].ds_wxfer;
1.13      angelos   365:                        cur.dk_seek[i] = q[i].ds_seek;
1.24      tedu      366:                        cur.dk_rbytes[i] = q[i].ds_rbytes;
                    367:                        cur.dk_wbytes[i] = q[i].ds_wbytes;
1.13      angelos   368:                        timerset(&(q[i].ds_time), &(cur.dk_time[i]));
1.10      angelos   369:                }
1.14      deraadt   370:                free(q);
1.10      angelos   371:
1.30      deraadt   372:                size = sizeof(cur.cp_time);
1.10      angelos   373:                mib[0] = CTL_KERN;
                    374:                mib[1] = KERN_CPTIME;
                    375:                if (sysctl(mib, 2, cur.cp_time, &size, NULL, 0) < 0) {
1.11      angelos   376:                        warn("could not read kern.cp_time");
1.10      angelos   377:                        bzero(cur.cp_time, sizeof(cur.cp_time));
                    378:                }
                    379:                size = sizeof(cur.tk_nin);
                    380:                mib[0] = CTL_KERN;
                    381:                mib[1] = KERN_TTY;
                    382:                mib[2] = KERN_TTY_TKNIN;
                    383:                if (sysctl(mib, 3, &cur.tk_nin, &size, NULL, 0) < 0) {
1.11      angelos   384:                        warn("could not read kern.tty.tk_nin");
1.10      angelos   385:                        cur.tk_nin = 0;
                    386:                }
                    387:                size = sizeof(cur.tk_nin);
                    388:                mib[0] = CTL_KERN;
                    389:                mib[1] = KERN_TTY;
                    390:                mib[2] = KERN_TTY_TKNOUT;
                    391:                if (sysctl(mib, 3, &cur.tk_nout, &size, NULL, 0) < 0) {
1.11      angelos   392:                        warn("could not read kern.tty.tk_nout");
1.10      angelos   393:                        cur.tk_nout = 0;
                    394:                }
                    395:        } else {
1.16      deraadt   396: #if !defined(NOKVM)
1.10      angelos   397:                p = dk_drivehead;
                    398:
1.22      tdeval    399:                for (i = 0; i < cur.dk_ndrive; i++) {
1.10      angelos   400:                        deref_kptr(p, &cur_disk, sizeof(cur_disk));
1.24      tedu      401:                        cur.dk_rxfer[i] = cur_disk.dk_rxfer;
                    402:                        cur.dk_wxfer[i] = cur_disk.dk_wxfer;
1.10      angelos   403:                        cur.dk_seek[i] = cur_disk.dk_seek;
1.24      tedu      404:                        cur.dk_rbytes[i] = cur_disk.dk_rbytes;
                    405:                        cur.dk_wbytes[i] = cur_disk.dk_wbytes;
1.10      angelos   406:                        timerset(&(cur_disk.dk_time), &(cur.dk_time[i]));
1.27      otto      407:                        p = TAILQ_NEXT(&cur_disk, dk_link);
1.10      angelos   408:                }
                    409:                deref_nl(X_CP_TIME, cur.cp_time, sizeof(cur.cp_time));
                    410:                deref_nl(X_TK_NIN, &cur.tk_nin, sizeof(cur.tk_nin));
                    411:                deref_nl(X_TK_NOUT, &cur.tk_nout, sizeof(cur.tk_nout));
1.16      deraadt   412: #endif /* !defined(NOKVM) */
1.1       tholo     413:        }
                    414: }
                    415:
                    416: /*
                    417:  * Perform all of the initialization and memory allocation needed to
                    418:  * track disk statistics.
                    419:  */
                    420: int
1.28      otto      421: dkinit(int sel)
1.1       tholo     422: {
1.19      deraadt   423: #if !defined(NOKVM)
1.1       tholo     424:        struct disklist_head disk_head;
                    425:        struct disk     cur_disk, *p;
                    426:         char           errbuf[_POSIX2_LINE_MAX];
1.19      deraadt   427: #endif
1.1       tholo     428:        static int      once = 0;
                    429:        extern int      hz;
1.10      angelos   430:        int             i, mib[2];
                    431:        size_t          size;
                    432:        struct clockinfo clkinfo;
                    433:        char            *disknames, *name, *bufpp;
1.1       tholo     434:
                    435:        if (once)
                    436:                return(1);
                    437:
1.10      angelos   438:        if (nlistf != NULL || memf != NULL) {
1.16      deraadt   439: #if !defined(NOKVM)
1.10      angelos   440:                /* Open the kernel. */
1.20      deraadt   441:                if (kd == NULL &&
                    442:                    (kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY,
1.10      angelos   443:                    errbuf)) == NULL)
                    444:                        errx(1, "kvm_openfiles: %s", errbuf);
                    445:
                    446:                /* Obtain the namelist symbols from the kernel. */
                    447:                if (kvm_nlist(kd, namelist))
                    448:                        KVM_ERROR("kvm_nlist failed to read symbols.");
                    449:
                    450:                /* Get the number of attached drives. */
1.22      tdeval    451:                deref_nl(X_DISK_COUNT, &cur.dk_ndrive, sizeof(cur.dk_ndrive));
1.10      angelos   452:
1.22      tdeval    453:                if (cur.dk_ndrive < 0)
                    454:                        errx(1, "invalid _disk_count %d.", cur.dk_ndrive);
1.10      angelos   455:
1.1       tholo     456:                /* Get a pointer to the first disk. */
                    457:                deref_nl(X_DISKLIST, &disk_head, sizeof(disk_head));
1.27      otto      458:                dk_drivehead = TAILQ_FIRST(&disk_head);
1.10      angelos   459:
                    460:                /* Get ticks per second. */
                    461:                deref_nl(X_STATHZ, &hz, sizeof(hz));
                    462:                if (!hz)
                    463:                  deref_nl(X_HZ, &hz, sizeof(hz));
1.16      deraadt   464: #endif /* !defined(NOKVM) */
1.10      angelos   465:        } else {
                    466:                /* Get the number of attached drives. */
                    467:                mib[0] = CTL_HW;
                    468:                mib[1] = HW_DISKCOUNT;
1.22      tdeval    469:                size = sizeof(cur.dk_ndrive);
                    470:                if (sysctl(mib, 2, &cur.dk_ndrive, &size, NULL, 0) < 0 ) {
1.11      angelos   471:                        warn("could not read hw.diskcount");
1.22      tdeval    472:                        cur.dk_ndrive = 0;
1.10      angelos   473:                }
                    474:
                    475:                /* Get ticks per second. */
                    476:                mib[0] = CTL_KERN;
                    477:                mib[1] = KERN_CLOCKRATE;
                    478:                size = sizeof(clkinfo);
                    479:                if (sysctl(mib, 2, &clkinfo, &size, NULL, 0) < 0) {
1.11      angelos   480:                        warn("could not read kern.clockrate");
1.10      angelos   481:                        hz = 0;
                    482:                } else
                    483:                        hz = clkinfo.stathz;
1.1       tholo     484:        }
                    485:
                    486:        /* allocate space for the statistics */
1.29      deraadt   487:        cur.dk_time = calloc((size_t)cur.dk_ndrive, sizeof(struct timeval));
                    488:        cur.dk_rxfer = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    489:        cur.dk_wxfer = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    490:        cur.dk_seek = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    491:        cur.dk_rbytes = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    492:        cur.dk_wbytes = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
1.31      chl       493:        cur.dk_select = calloc((size_t)cur.dk_ndrive, sizeof(int));
                    494:        cur.dk_name = calloc((size_t)cur.dk_ndrive, sizeof(char *));
1.29      deraadt   495:        last.dk_time = calloc((size_t)cur.dk_ndrive, sizeof(struct timeval));
                    496:        last.dk_rxfer = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    497:        last.dk_wxfer = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    498:        last.dk_seek = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    499:        last.dk_rbytes = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
                    500:        last.dk_wbytes = calloc((size_t)cur.dk_ndrive, sizeof(u_int64_t));
1.30      deraadt   501:
1.24      tedu      502:        if (!cur.dk_time || !cur.dk_rxfer || !cur.dk_wxfer || !cur.dk_seek ||
1.31      chl       503:            !cur.dk_rbytes || !cur.dk_wbytes || !cur.dk_select ||
                    504:            !cur.dk_name || !last.dk_time || !last.dk_rxfer ||
                    505:            !last.dk_wxfer || !last.dk_seek || !last.dk_rbytes ||
                    506:            !last.dk_wbytes)
1.1       tholo     507:                errx(1, "Memory allocation failure.");
                    508:
                    509:        /* Set up the compatibility interfaces. */
1.22      tdeval    510:        dk_ndrive = cur.dk_ndrive;
1.1       tholo     511:        dk_select = cur.dk_select;
                    512:        dr_name = cur.dk_name;
                    513:
1.22      tdeval    514:        /* Read the disk names and set initial selection. */
1.10      angelos   515:        if (nlistf == NULL && memf == NULL) {
                    516:                mib[0] = CTL_HW;
                    517:                mib[1] = HW_DISKNAMES;
                    518:                size = 0;
                    519:                if (sysctl(mib, 2, NULL, &size, NULL, 0) < 0)
                    520:                        err(1, "can't get hw.disknames");
1.12      angelos   521:                disknames = malloc(size);
1.10      angelos   522:                if (disknames == NULL)
                    523:                        err(1, NULL);
                    524:                if (sysctl(mib, 2, disknames, &size, NULL, 0) < 0)
                    525:                        err(1, "can't get hw.disknames");
                    526:                bufpp = disknames;
1.23      millert   527:                for (i = 0; i < dk_ndrive && (name = strsep(&bufpp, ",")) != NULL; i++) {
                    528:                        cur.dk_name[i] = name;
1.28      otto      529:                        cur.dk_select[i] = sel;
1.10      angelos   530:                }
                    531:        } else {
1.16      deraadt   532: #if !defined(NOKVM)
1.10      angelos   533:                p = dk_drivehead;
1.22      tdeval    534:                for (i = 0; i < cur.dk_ndrive; i++) {
1.10      angelos   535:                        char    buf[10];
1.16      deraadt   536:
1.10      angelos   537:                        deref_kptr(p, &cur_disk, sizeof(cur_disk));
                    538:                        deref_kptr(cur_disk.dk_name, buf, sizeof(buf));
                    539:                        cur.dk_name[i] = strdup(buf);
                    540:                        if (!cur.dk_name[i])
                    541:                                errx(1, "Memory allocation failure.");
1.28      otto      542:                        cur.dk_select[i] = sel;
1.1       tholo     543:
1.27      otto      544:                        p = TAILQ_NEXT(&cur_disk, dk_link);
1.10      angelos   545:                }
1.16      deraadt   546: #endif /* !defined(NOKVM) */
1.1       tholo     547:        }
                    548:
1.33      martynas  549:        /* Never do this initialization again. */
1.1       tholo     550:        once = 1;
                    551:        return(1);
                    552: }
                    553:
1.16      deraadt   554: #if !defined(NOKVM)
1.1       tholo     555: /*
1.30      deraadt   556:  * Dereference the kernel pointer `kptr' and fill in the local copy
1.1       tholo     557:  * pointed to by `ptr'.  The storage space must be pre-allocated,
                    558:  * and the size of the copy passed in `len'.
                    559:  */
                    560: static void
1.21      deraadt   561: deref_kptr(void *kptr, void *ptr, size_t len)
1.1       tholo     562: {
                    563:        char buf[128];
                    564:
1.6       art       565:        if (kvm_read(kd, (u_long)kptr, ptr, len) != len) {
1.1       tholo     566:                bzero(buf, sizeof(buf));
                    567:                snprintf(buf, (sizeof(buf) - 1),
1.2       deraadt   568:                     "can't dereference kptr 0x%lx", (u_long)kptr);
1.1       tholo     569:                KVM_ERROR(buf);
                    570:        }
                    571: }
1.17      drahn     572: #endif /* !defined(NOKVM) */