release/src/router/httpd/cpuinfo.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <unistd.h>
   4 #include <string.h>
   5 #include "tomato.h"
   6
   7 struct occupy
   8 {
   9         char name[20];
  10         unsigned int user;
  11         unsigned int nice;
  12         unsigned int system;
  13         unsigned int idle;
  14         unsigned int io;
  15         unsigned int irq;
  16         unsigned int sirq;
  17 };
  18
  19 void trim( char *str)
  20 {
  21         char *copied, *tail = NULL;
  22         if ( str == NULL )
  23                 return ;
  24
  25         for( copied = str; *str; str++ )
  26         {
  27                 if ( (unsigned char)*str > 0x20 )
  28                 {
  29                         *copied++ = *str;
  30                         tail = copied;
  31                 }
  32                 else {
  33                         if ( tail )
  34                                 *copied++ = *str;
  35                 }
  36         }
  37
  38         if ( tail )
  39                 *tail = 0;
  40         else
  41                 *copied = 0;
  42         return ;
  43 }
  44
  45 static float g_cpu_used;
  46 static int cpu_num;
  47
  48 static void cal_occupy(struct occupy *, struct occupy *);
  49 static void get_occupy(struct occupy *);
  50
  51
  52 /*
  53 system type             : Broadcom BCM5354 chip rev 2 pkg 0
  54 processor               : 0
  55 cpu model               : BCM3302 V2.9
  56 BogoMIPS                : 238.38
  57 */
  58 int strncmp_ex(char *str1, char *str2)
  59 {
  60         return strncmp(str1, str2, strlen(str2));
  61 }
  62
  63 int get_cpuinfo(char *system_type, char *cpu_model, char *bogomips, char *cpuclk)
  64
  65 {
  66         FILE *fd;
  67         char *next;
  68         char buff[1024];
  69         char title[128], value[512];
  70         int okcount=0;
  71
  72         fd = fopen ("/proc/cpuinfo", "r");
  73         while (fgets(buff, sizeof(buff), fd)) {
  74                 next = buff;
  75                 strcpy(title, strsep(&next, ":"));
  76                 if (next == NULL) continue;
  77                 strcpy(value, next);
  78                 trim(value);
  79                 if (strncmp_ex(title, "system type")==0) {
  80                         okcount++;
  81                         //printf("system type: %s\n", value);
  82                         if (strncmp_ex(value, "Broadcom BCM5354")==0) {
  83                                 strcpy(system_type, "Broadcom BCM5354");
  84                         }
  85                         else {
  86                                 strcpy(system_type, value);
  87                         }
  88                 }
  89                 if (strncmp_ex(title, "cpu model")==0) {
  90                         okcount++;
  91                         //printf("cpu model: %s\n", value);
  92                         strcpy(cpu_model, value);
  93                 }
  94                 if (strncmp_ex(title, "BogoMIPS")==0) {
  95                         okcount++;
  96                         //printf("bogomips: %s\n", value);
  97                         strcpy(bogomips, value);
  98                 }
  99                 if (strncmp_ex(title, "cpu MHz")==0) {
 100                         okcount++;
 101                         //printf("cpuclk: %s\n", value);
 102                         strcpy(cpuclk, value);
 103                 }
 104
 105                 //fprintf (stderr, "%s - %s", title, value);
 106         }
 107         fclose(fd);
 108         return (okcount==3);
 109 }
 110
 111 /*
 112 int main()
 113 {
 114         char system_type[64];
 115         char cpu_model[64];
 116         char bogomips[64];
 117         char cpuclk[64];
 118         if get_cpuinfo(system_type, cpu_model, bogomips, cpuclk) {
 119                 printf("system type: %s, cpu model: %s, bogomips: %s, cpuclk: %s\n", system_type, cpu_model, bogomips, cpuclk);
 120         }
 121 }
 122 */
 123
 124 float get_cpupercent()
 125 {
 126         struct occupy ocpu[10];
 127         struct occupy ncpu[10];
 128         int i;
 129
 130 //      cpu_num = sysconf(_SC_NPROCESSORS_ONLN);
 131         cpu_num = 1;
 132         get_occupy(ocpu);
 133         sleep(1);
 134         get_occupy(ncpu);
 135         for (i=0; i<cpu_num; i++)
 136         {
 137                 cal_occupy(&ocpu[i], &ncpu[i]);
 138                 //printf("%f \n", g_cpu_used);
 139         }
 140         return g_cpu_used;
 141 }
 142
 143 static void cal_occupy (struct occupy *o, struct occupy *n)
 144 {
 145         double od, nd;
 146         double id, sd;
 147         double scale;
 148
 149         od = (double) (o->user + o->nice + o->system + o->idle + o->io + o->irq + o->sirq);
 150         nd = (double) (n->user + n->nice + n->system + n->idle + n->io + n->irq + n->sirq);
 151         scale = 100.0 / (float)(nd-od);
 152         id = (double) (n->user - o->user);
 153         sd = (double) (n->system - o->system);
 154         g_cpu_used = ((sd+id)*100.0)/(nd-od);
 155 }
 156
 157 static void get_occupy (struct occupy *o)
 158 {
 159         FILE *fd;
 160         int n;
 161         char buff[1024];
 162
 163         fd = fopen ("/proc/stat", "r");
 164         fgets (buff, sizeof(buff), fd);
 165         for(n=0;n<cpu_num;n++)
 166         {
 167                 fgets (buff, sizeof(buff),fd);
 168                 sscanf (buff, "%s %u %u %u %u %u %u %u", o[n].name, &o[n].user, &o[n].nice, &o[n].system, &o[n].idle, &o[n].io, &o[n].irq, &o[n].sirq);
 169                 //fprintf (stderr, "%s %u %u %u %u %u %u %u\n", o[n].name, o[n].user, o[n].nice, o[n].system, o[n].idle, o[n].io, o[n].irq, o[n].sirq);
 170         }
 171         fclose(fd);
 172 }