granicus.if.org Git - fcron/blob - getloadavg.c

   1 /*
   2  *  gloadavg.c - get load average for Linux
   3  *  Copyright (C) 1993  Thomas Koenig
   4  *  Copyright 2000-2014 Thibault Godouet <fcron@free.fr>
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19  */
  20
  21 #ifndef _POSIX_SOURCE           /* Don't redefine if already exists */
  22 #define _POSIX_SOURCE 1
  23 #endif
  24
  25 #include "fcron.h"
  26
  27 /* Local headers */
  28
  29 #include "getloadavg.h"
  30
  31 /* Global functions */
  32
  33 #ifdef HAVE_KSTAT
  34
  35 #include <kstat.h>
  36
  37 int
  38 getloadavg(double *result, int n)
  39 /* return the current load average as a floating point number,
  40  * the number of load averages read, or <0 for error
  41  */
  42 {
  43     kstat_ctl_t *kc;
  44     kstat_t *ksp;
  45     kstat_named_t *knm;
  46     kstat_named_t knm_buf[20];
  47     int cnt;
  48     int ret;
  49
  50     if (n != 3) {
  51         return -1;
  52     }
  53
  54     ret = 0;
  55
  56     kc = kstat_open();
  57     for (ksp = kc->kc_chain; ksp != NULL; ksp = ksp->ks_next) {
  58         if (strcmp(ksp->ks_name, "system_misc") == 0) {
  59             kstat_read(kc, ksp, &knm_buf);
  60             for (cnt = 0; cnt < ksp->ks_ndata; cnt++) {
  61                 knm = &knm_buf[cnt];
  62                 if (strcmp(knm->name, "avenrun_1min") == 0) {
  63                     result[0] = knm->value.ui32 / 256.0;
  64                     ret++;
  65                 }
  66                 else if (strcmp(knm->name, "avenrun_5min") == 0) {
  67                     result[1] = knm->value.ui32 / 256.0;
  68                     ret++;
  69                 }
  70                 else if (strcmp(knm->name, "avenrun_15min") == 0) {
  71                     result[2] = knm->value.ui32 / 256.0;
  72                     ret++;
  73                 }
  74             }
  75         }
  76     }
  77     kstat_close(kc);
  78
  79     return ret;
  80 }
  81
  82 #else                           /* def HAVE_KSTAT */
  83
  84 int
  85 getloadavg(double *result, int n)
  86 /* return the current load average as a floating point number,
  87  * the number of load averages read, or <0 for error
  88  */
  89 {
  90     FILE *fp;
  91     int i;
  92     char loadavg_path = PROC "/loadavg";
  93
  94     if (n > 3)
  95         n = 3;
  96
  97     if ((fp = fopen(loadavg_path, "r")) == NULL) {
  98         error_e("could not open '%s' (make sure procfs is mounted)",
  99                 loadavg_path);
 100         i = -1;
 101     }
 102     else {
 103         for (i = 0; i < n; i++) {
 104             if (fscanf(fp, "%lf", result) != 1)
 105                 goto end;
 106             result++;
 107         }
 108     }
 109  end:
 110     xfclose_check(&fp, loadavg_path);
 111     return (i < 0) ? i : i;
 112 }
 113
 114 #endif                          /* def HAVE_KSTAT */