plugins/linux/linux.c

#include <stdio.h>
#include <stdlib.h>
#include "ukcprog.h"
#include <string.h>
#include <unistd.h>
#include <sys/utsname.h>
#include <sys/vfs.h>
#include <utmp.h>
#include <pwd.h>
#include <mntent.h>

int die() {
    exit (1);
}

void getLoadAv() {

  double loadav[3];

  if((getloadavg(loadav,3)) == -1){
    errf("Failed to get load averages (%m)");
    die();
  }
  
  printf("packet.load.load1 %.2f\n",loadav[0]);
  printf("packet.load.load5 %.2f\n",loadav[1]);
  printf("packet.load.load15 %.2f\n",loadav[2]);
  
}

void getMemInfo() {
    char *line;
    char *mem=NULL;
    char *swap=NULL;
    char *ch;
    long memstat[6];
    long swapstat[3];
    long tmp;
    int counter=0;
    
    FILE *f;
    
    if ((f=fopen("/proc/meminfo", "r" ))==NULL) {
        errf("Failed to open memory stats (%m)");
        die();
    }
    
    while(((line=fpgetline(f)) != NULL) && (counter < 2)) {
        if (((strncmp("Mem: ",line,5)) == 0)) {
            mem=strdup(line);
            counter++;
        }
        if (((strncmp(line,"Swap: ",6)) == 0)) {
            swap=strdup(line);
            counter++;
        }
    }
    
    if ((fclose(f)) != 0) {
        errf("Failed to close file (%m).");
        die();
    }
   
    if (mem==NULL || swap==NULL){
        errf("Failed to obtain information required for memory and swap stats");
        die();
    }
 
    /* Get the info we want from the 2 read in lines */
    
    ch = strchr(mem, ' ');
    if (ch == NULL) abort();
    *ch = '\0';
    ch++;
    
    /* By now we should of skipped the mem: bit.. onto the numbers */
    
    for(counter=0;counter<6;counter++) {
        for (; (*ch == ' '); ch++);
        if (ch == NULL) abort();
        memstat[counter]=atol(ch);
        ch++;
        for(; (*ch != ' ') && (*ch != '\0'); ch++);
        if (ch == NULL) abort();
    }
     
    /* Now swap.. */
    ch = strchr(swap, ' ');
    if (ch == NULL) abort();
    *ch = '\0';
    ch++;
    
    for(counter=0;counter<3;counter++) {
        for (; (*ch == ' '); ch++);
        if (ch == NULL) abort();
        swapstat[counter]=atol(ch);
        ch++;
        for(; (*ch != ' ') && (*ch != '\0'); ch++);
        if (ch == NULL) abort();
    }
    
    printf("packet.memory.total %ld\n",((memstat[0]/1024)/1024));
    
    /* Due to batty linux we do some maths to work out roughly what the free ram is */
    tmp=((memstat[1] - memstat[4])/1024)/1024;
    printf("packet.memory.used %ld\n",tmp);
    tmp=((memstat[2] + memstat[4])/1024)/1024;
    printf("packet.memory.free %ld\n",tmp);
    
    printf("packet.swap.total %ld\n",((swapstat[0]/1024)/1024));
    printf("packet.swap.used %ld\n",((swapstat[1]/1024)/1024));
    printf("packet.swap.free %ld\n",((swapstat[2])/1024)/1024);
       
    free(mem);
    free(swap);
}

void cpustats() {
    char *tmp;
    char *line[2];
    char *line_p[2];
    long cpustats[4][2];
    long user, kernel, idle;
    long total;
    int x,y;
    float usage;
    FILE *f;
    
    if ((f=fopen("/proc/stat", "r" ))==NULL) {
        errf("Failed to open cpu stats (%m)");
        die();
    }
    
    if((tmp=fpgetline(f)) == NULL) {
        errf("Failed to read cpu stats (%m)");
        die();
    }
    
    if((line[0]=strdup(tmp)) == NULL) {
        errf("strdup failed (%m)");
        die();
    }
    
    if ((fclose(f)) != 0) {
        errf("Failed to close file (%m).");
        die();
    }
     
    sleep(1);
    
    if ((f=fopen("/proc/stat", "r" ))==NULL) {
        errf("Failed to open cpu stats (%m)");
        die();
    }
    
    if((tmp=fpgetline(f)) == NULL) {
        errf("Failed to read cpu stats (%m)");
        die();
    }
    
    if((line[1]=strdup(tmp)) == NULL) {
        errf("strdup failed (%m)");
        die();
    }
      
    if ((fclose(f)) != 0) {
        errf("Failed to close file (%m).");
        die();
    }
    
    for(x=0;x<2;x++) {
        line_p[x] = strchr(line[x], ' ');
        if (line_p[x] == NULL) abort();
        *line_p[x] = '\0';
        line_p[x]++;
    }
    
    /* Now should be passed "cpu " */
    
    for(x=0;x<2;x++) {
        for(y=0;y<4;y++) {
            for (; (*line_p[x] == ' '); line_p[x]++);
            if (line_p[x] == NULL) abort();
            cpustats[y][x]=atol(line_p[x]);
            line_p[x]++;    
            for(; (*line_p[x] != ' ') && (*line_p[x] != '\0'); line_p[x]++);
            if (line_p[x] == NULL) abort();
            for (; (*line_p[x] == ' '); line_p[x]++);
        }
    }
    
    user=cpustats[0][1]-cpustats[0][0]+cpustats[1][1]-cpustats[1][0];
    kernel=cpustats[2][1]-cpustats[2][0];
    idle=cpustats[3][1]-cpustats[3][0];
    
    /* use total to get the total number and then work out the percentages */
    total=user+kernel+idle;
    
    usage=((((float)user)/((float)total))*100.00);
    printf("packet.cpu.user %3.2f\n", usage);
    usage=((((float)kernel)/((float)total))*100.00);
    printf("packet.cpu.kernel %3.2f\n", usage);
    usage=((((float)idle)/((float)total))*100.00);
    printf("packet.cpu.idle %3.2f\n", usage);
    
    /* Cos i-scream's expects this to be sent :/ */
    printf("packet.cpu.iowait 0\n");
    printf("packet.cpu.swap 0\n");   
 
    free(line[0]);
    free(line[1]);
}

void processStats() {
    int sleeping=-1;
    int zombie=0;
    int stopped=0;
    int running=0;
    int nousers=0;
    char *line;
    char *line_p;
    struct utmp *entry;
    
    FILE *f;
    
    if((f=popen("/bin/ps -Al" , "r")) == NULL) {
        errf("Failed to get process stats (%m)");
        die();
    }
    
    while((line=fpgetline(f)) != NULL) {

      line_p=line;
      for(; (*line_p == ' ') && (*line_p != '\0'); line_p++);
      line_p=strchr(line_p, ' ');
      for(; (*line_p == ' ') && (*line_p != '\0'); line_p++);
      if (line_p==NULL) abort();
      /* Ok, we should now be at the state :) .. */
      if (*line_p=='S') sleeping++;
      if (*line_p=='R') running++;
      if (*line_p=='Z') zombie++;
      if (*line_p=='T') stopped++;
    }
    
    if((pclose(f)) == -1) {
        errf("Failed to close process stats (%m)");
        die();
    }
    
    printf("packet.users.list");
    
    while((entry=getutent()) != NULL) {
        if(entry->ut_type==USER_PROCESS) {
            printf(" %s",entry->ut_user);
            nousers++;
        }
    }
    
    printf("\npacket.users.count %d\n", nousers);
       
    printf("packet.processes.sleeping %d\n",sleeping);
    printf("packet.processes.cpu %d\n",running);
    printf("packet.processes.zombie %d\n",zombie);
    printf("packet.processes.stopped %d\n", stopped);
    printf("packet.processes.total %d\n", (sleeping+running+zombie+stopped));
}

void uptimeStats() {
    char *line;
    char *line_p;
    FILE *f;
    
    if ((f=fopen("/proc/uptime", "r")) == NULL) {
        errf("Failed to get uptime stats (%m)");
        die();
    }
    
    if ((line=fpgetline(f)) == NULL) {
        errf("Failed to read uptime stats (%m)");
        die();
    }
    
    if ((fclose(f)) != 0) {
        errf("Failed to close file (%m).");
        die();
    }
    
    line_p=strchr(line, '.');
    if (line_p==NULL) abort();
    *line_p='\0';
    
    printf("packet.os.uptime %s\n", line);
}

void osStats() {
    struct utsname os;
    
    if((uname(&os)) != 0) {
        errf("Failed to get os stats (%m)");
        die();
    }
    
    printf("packet.os.name %s\n", os.sysname);
    printf("packet.os.release %s\n" , os.release);
    printf("packet.os.version %s\n", os.version);
    printf("packet.os.sysname %s\n" , os.nodename);
    printf("packet.os.platform %s\n", os.machine);
}

void diskStats() {

  struct mntent *mp;
  struct statfs df;
  FILE *f;
  int counter=0;

  if ((f=fopen("/etc/mtab", "r" ))==NULL){
      errf("Failed to open mounts (%m)");
      die();
  }

  while((mp=getmntent(f))){
    if ((statfs(mp->mnt_dir, &df)) !=0){
      errf("Failed to gets fs stats (%m)");
      continue;
    }

    if((((strcmp(mp->mnt_type, MNTTYPE_NFS))==0) ||  (strcmp(mp->mnt_type,MNTTYPE_IGNORE)) ==0)) continue;
    
      printf("packet.disk.p%d.attributes.mount %s\n", counter, mp->mnt_dir);
      printf("packet.disk.p%d.attributes.name %s\n", counter, mp->mnt_fsname);
      printf("packet.disk.p%d.attributes.kbytes %lu\n",counter, ((df.f_bsize/1024) * df.f_blocks));
      printf("packet.disk.p%d.attributes.used %lu\n",counter, (((df.f_bsize/1024) * df.f_blocks) -((df.f_bsize/1024) * df.f_bfree)));
      printf("packet.disk.p%d.attributes.avail %lu\n",counter, (df.f_bsize/1024) * df.f_bavail);
      printf("packet.disk.p%d.attributes.totalinodes %lu\n", counter, df.f_files);
      printf("packet.disk.p%d.attributes.freeinodes %lu\n",counter, df.f_ffree);
    
      counter++;
  }

  

}

int main() {
   getLoadAv();
   getMemInfo();
   cpustats();
   processStats();
   uptimeStats();
   osStats();
   diskStats();
   fflush(stdout);
   return 0;
}
Revision:	1.8
Committed:	Tue Apr 2 12:43:15 2002 UTC (23 years, 8 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.7:	+43 -66 lines
Log Message:	Improved the linux one to rely slightly less on /proc where it was sane not to use it.
#	Content
1	#include <stdio.h>
2	#include <stdlib.h>
3	#include "ukcprog.h"
4	#include <string.h>
5	#include <unistd.h>
6	#include <sys/utsname.h>
7	#include <sys/vfs.h>
8	#include <utmp.h>
9	#include <pwd.h>
10	#include <mntent.h>
11
12	int die() {
13	exit (1);
14	}
15
16	void getLoadAv() {
17
18	double loadav[3];
19
20	if((getloadavg(loadav,3)) == -1){
21	errf("Failed to get load averages (%m)");
22	die();
23	}
24
25	printf("packet.load.load1 %.2f\n",loadav[0]);
26	printf("packet.load.load5 %.2f\n",loadav[1]);
27	printf("packet.load.load15 %.2f\n",loadav[2]);
28
29	}
30
31	void getMemInfo() {
32	char *line;
33	char *mem=NULL;
34	char *swap=NULL;
35	char *ch;
36	long memstat[6];
37	long swapstat[3];
38	long tmp;
39	int counter=0;
40
41	FILE *f;
42
43	if ((f=fopen("/proc/meminfo", "r" ))==NULL) {
44	errf("Failed to open memory stats (%m)");
45	die();
46	}
47
48	while(((line=fpgetline(f)) != NULL) && (counter < 2)) {
49	if (((strncmp("Mem: ",line,5)) == 0)) {
50	mem=strdup(line);
51	counter++;
52	}
53	if (((strncmp(line,"Swap: ",6)) == 0)) {
54	swap=strdup(line);
55	counter++;
56	}
57	}
58
59	if ((fclose(f)) != 0) {
60	errf("Failed to close file (%m).");
61	die();
62	}
63
64	if (mem==NULL \|\| swap==NULL){
65	errf("Failed to obtain information required for memory and swap stats");
66	die();
67	}
68
69	/* Get the info we want from the 2 read in lines */
70
71	ch = strchr(mem, ' ');
72	if (ch == NULL) abort();
73	*ch = '\0';
74	ch++;
75
76	/* By now we should of skipped the mem: bit.. onto the numbers */
77
78	for(counter=0;counter<6;counter++) {
79	for (; (*ch == ' '); ch++);
80	if (ch == NULL) abort();
81	memstat[counter]=atol(ch);
82	ch++;
83	for(; (ch != ' ') && (ch != '\0'); ch++);
84	if (ch == NULL) abort();
85	}
86
87	/* Now swap.. */
88	ch = strchr(swap, ' ');
89	if (ch == NULL) abort();
90	*ch = '\0';
91	ch++;
92
93	for(counter=0;counter<3;counter++) {
94	for (; (*ch == ' '); ch++);
95	if (ch == NULL) abort();
96	swapstat[counter]=atol(ch);
97	ch++;
98	for(; (ch != ' ') && (ch != '\0'); ch++);
99	if (ch == NULL) abort();
100	}
101
102	printf("packet.memory.total %ld\n",((memstat[0]/1024)/1024));
103
104	/* Due to batty linux we do some maths to work out roughly what the free ram is */
105	tmp=((memstat[1] - memstat[4])/1024)/1024;
106	printf("packet.memory.used %ld\n",tmp);
107	tmp=((memstat[2] + memstat[4])/1024)/1024;
108	printf("packet.memory.free %ld\n",tmp);
109
110	printf("packet.swap.total %ld\n",((swapstat[0]/1024)/1024));
111	printf("packet.swap.used %ld\n",((swapstat[1]/1024)/1024));
112	printf("packet.swap.free %ld\n",((swapstat[2])/1024)/1024);
113
114	free(mem);
115	free(swap);
116	}
117
118	void cpustats() {
119	char *tmp;
120	char *line[2];
121	char *line_p[2];
122	long cpustats[4][2];
123	long user, kernel, idle;
124	long total;
125	int x,y;
126	float usage;
127	FILE *f;
128
129	if ((f=fopen("/proc/stat", "r" ))==NULL) {
130	errf("Failed to open cpu stats (%m)");
131	die();
132	}
133
134	if((tmp=fpgetline(f)) == NULL) {
135	errf("Failed to read cpu stats (%m)");
136	die();
137	}
138
139	if((line[0]=strdup(tmp)) == NULL) {
140	errf("strdup failed (%m)");
141	die();
142	}
143
144	if ((fclose(f)) != 0) {
145	errf("Failed to close file (%m).");
146	die();
147	}
148
149	sleep(1);
150
151	if ((f=fopen("/proc/stat", "r" ))==NULL) {
152	errf("Failed to open cpu stats (%m)");
153	die();
154	}
155
156	if((tmp=fpgetline(f)) == NULL) {
157	errf("Failed to read cpu stats (%m)");
158	die();
159	}
160
161	if((line[1]=strdup(tmp)) == NULL) {
162	errf("strdup failed (%m)");
163	die();
164	}
165
166	if ((fclose(f)) != 0) {
167	errf("Failed to close file (%m).");
168	die();
169	}
170
171	for(x=0;x<2;x++) {
172	line_p[x] = strchr(line[x], ' ');
173	if (line_p[x] == NULL) abort();
174	*line_p[x] = '\0';
175	line_p[x]++;
176	}
177
178	/* Now should be passed "cpu " */
179
180	for(x=0;x<2;x++) {
181	for(y=0;y<4;y++) {
182	for (; (*line_p[x] == ' '); line_p[x]++);
183	if (line_p[x] == NULL) abort();
184	cpustats[y][x]=atol(line_p[x]);
185	line_p[x]++;
186	for(; (line_p[x] != ' ') && (line_p[x] != '\0'); line_p[x]++);
187	if (line_p[x] == NULL) abort();
188	for (; (*line_p[x] == ' '); line_p[x]++);
189	}
190	}
191
192	user=cpustats[0][1]-cpustats[0][0]+cpustats[1][1]-cpustats[1][0];
193	kernel=cpustats[2][1]-cpustats[2][0];
194	idle=cpustats[3][1]-cpustats[3][0];
195
196	/* use total to get the total number and then work out the percentages */
197	total=user+kernel+idle;
198
199	usage=((((float)user)/((float)total))*100.00);
200	printf("packet.cpu.user %3.2f\n", usage);
201	usage=((((float)kernel)/((float)total))*100.00);
202	printf("packet.cpu.kernel %3.2f\n", usage);
203	usage=((((float)idle)/((float)total))*100.00);
204	printf("packet.cpu.idle %3.2f\n", usage);
205
206	/* Cos i-scream's expects this to be sent :/ */
207	printf("packet.cpu.iowait 0\n");
208	printf("packet.cpu.swap 0\n");
209
210	free(line[0]);
211	free(line[1]);
212	}
213
214	void processStats() {
215	int sleeping=-1;
216	int zombie=0;
217	int stopped=0;
218	int running=0;
219	int nousers=0;
220	char *line;
221	char *line_p;
222	struct utmp *entry;
223
224	FILE *f;
225
226	if((f=popen("/bin/ps -Al" , "r")) == NULL) {
227	errf("Failed to get process stats (%m)");
228	die();
229	}
230
231	while((line=fpgetline(f)) != NULL) {
232
233	line_p=line;
234	for(; (line_p == ' ') && (line_p != '\0'); line_p++);
235	line_p=strchr(line_p, ' ');
236	for(; (line_p == ' ') && (line_p != '\0'); line_p++);
237	if (line_p==NULL) abort();
238	/* Ok, we should now be at the state :) .. */
239	if (*line_p=='S') sleeping++;
240	if (*line_p=='R') running++;
241	if (*line_p=='Z') zombie++;
242	if (*line_p=='T') stopped++;
243	}
244
245	if((pclose(f)) == -1) {
246	errf("Failed to close process stats (%m)");
247	die();
248	}
249
250	printf("packet.users.list");
251
252	while((entry=getutent()) != NULL) {
253	if(entry->ut_type==USER_PROCESS) {
254	printf(" %s",entry->ut_user);
255	nousers++;
256	}
257	}
258
259	printf("\npacket.users.count %d\n", nousers);
260
261	printf("packet.processes.sleeping %d\n",sleeping);
262	printf("packet.processes.cpu %d\n",running);
263	printf("packet.processes.zombie %d\n",zombie);
264	printf("packet.processes.stopped %d\n", stopped);
265	printf("packet.processes.total %d\n", (sleeping+running+zombie+stopped));
266	}
267
268	void uptimeStats() {
269	char *line;
270	char *line_p;
271	FILE *f;
272
273	if ((f=fopen("/proc/uptime", "r")) == NULL) {
274	errf("Failed to get uptime stats (%m)");
275	die();
276	}
277
278	if ((line=fpgetline(f)) == NULL) {
279	errf("Failed to read uptime stats (%m)");
280	die();
281	}
282
283	if ((fclose(f)) != 0) {
284	errf("Failed to close file (%m).");
285	die();
286	}
287
288	line_p=strchr(line, '.');
289	if (line_p==NULL) abort();
290	*line_p='\0';
291
292	printf("packet.os.uptime %s\n", line);
293	}
294
295	void osStats() {
296	struct utsname os;
297
298	if((uname(&os)) != 0) {
299	errf("Failed to get os stats (%m)");
300	die();
301	}
302
303	printf("packet.os.name %s\n", os.sysname);
304	printf("packet.os.release %s\n" , os.release);
305	printf("packet.os.version %s\n", os.version);
306	printf("packet.os.sysname %s\n" , os.nodename);
307	printf("packet.os.platform %s\n", os.machine);
308	}
309
310	void diskStats() {
311
312	struct mntent *mp;
313	struct statfs df;
314	FILE *f;
315	int counter=0;
316
317	if ((f=fopen("/etc/mtab", "r" ))==NULL){
318	errf("Failed to open mounts (%m)");
319	die();
320	}
321
322	while((mp=getmntent(f))){
323	if ((statfs(mp->mnt_dir, &df)) !=0){
324	errf("Failed to gets fs stats (%m)");
325	continue;
326	}
327
328	if((((strcmp(mp->mnt_type, MNTTYPE_NFS))==0) \|\| (strcmp(mp->mnt_type,MNTTYPE_IGNORE)) ==0)) continue;
329
330	printf("packet.disk.p%d.attributes.mount %s\n", counter, mp->mnt_dir);
331	printf("packet.disk.p%d.attributes.name %s\n", counter, mp->mnt_fsname);
332	printf("packet.disk.p%d.attributes.kbytes %lu\n",counter, ((df.f_bsize/1024) * df.f_blocks));
333	printf("packet.disk.p%d.attributes.used %lu\n",counter, (((df.f_bsize/1024) * df.f_blocks) -((df.f_bsize/1024) * df.f_bfree)));
334	printf("packet.disk.p%d.attributes.avail %lu\n",counter, (df.f_bsize/1024) * df.f_bavail);
335	printf("packet.disk.p%d.attributes.totalinodes %lu\n", counter, df.f_files);
336	printf("packet.disk.p%d.attributes.freeinodes %lu\n",counter, df.f_ffree);
337
338	counter++;
339	}
340
341
342
343	}
344
345	int main() {
346	getLoadAv();
347	getMemInfo();
348	cpustats();
349	processStats();
350	uptimeStats();
351	osStats();
352	diskStats();
353	fflush(stdout);
354	return 0;
355	}