plugins/freebsd/freebsd.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org.uk
 * Copyright (C) 2000-2002 i-scream
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#include <stdio.h>
#include "ukcprog.h"
#include <sys/param.h>
#include <sys/ucred.h>
#include <sys/mount.h>
#include <string.h>
#include <stdlib.h>
#include <sys/utsname.h>
#include <sys/types.h>
#include <utmp.h>
#include <sys/dkstat.h>
#include <kvm.h>
#include <unistd.h>
#include <sys/sysctl.h>
#include <fcntl.h>
#include <limits.h>
#include <time.h>
#include <sys/time.h>

uid_t uid;
uid_t euid;
gid_t gid;
gid_t egid;

void die(){
  exit(1);
}

void diskStats(){
  struct statfs *mp;
  int nummnt;
  int counter=0;
  char *mntpnt;
  char *devicename;

  nummnt=getmntinfo(&mp , MNT_LOCAL);
  if (nummnt<=0){
    errf("Failed to get disk stats (%m)");
    die();
  }

  mntpnt=malloc(MNAMELEN);
  devicename=malloc(MNAMELEN);
  
  for(;counter<nummnt;counter++){
    strncpy(mntpnt, mp->f_mntonname ,MNAMELEN); 
    strncpy(devicename, mp->f_mntfromname ,MNAMELEN); 
    printf("packet.disk.p%d.attributes.mount %s\n", counter, mntpnt);
    printf("packet.disk.p%d.attributes.name %s\n", counter, devicename);
    printf("packet.disk.p%d.attributes.kbytes %lu\n",counter, ((mp->f_bsize/1024) * mp->f_blocks));
    printf("packet.disk.p%d.attributes.used %lu\n",counter, (((mp->f_bsize/1024) * mp->f_blocks) -((mp->f_bsize/1024) * mp->f_bfree)));
    printf("packet.disk.p%d.attributes.avail %lu\n",counter, (mp->f_bsize/1024) * mp->f_bavail);
    printf("packet.disk.p%d.attributes.totalinodes %lu\n", counter, mp->f_files);
    printf("packet.disk.p%d.attributes.freeinodes %lu\n",counter, mp->f_ffree);

    mp++;
  }
  

}

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 loadStats(){
  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 userStats(){
  struct utmp users;
  FILE *f;
  int numusers=0;

  if ((f=fopen(_PATH_UTMP, "r")) == NULL){
    errf("Failed to get user stats(%m)");
    die();
  }

  printf("packet.users.list");

  while((fread(&users, sizeof(users),1,f)) != 0){
    if (users.ut_name[0] == '\0') continue;
    printf(" %s",users.ut_name);
    numusers++;
  }

  printf("\npacket.users.count %d\n",numusers);

}

void systemStats(){

  long cp_time[CPUSTATES];
  long total, user, idle, kernel, nice;
  long totalmem, freemem, swaptotal, swapused;
  float stat;

  static char *uptimename = "kern.boottime";
  static char *cpname = "kern.cp_time";
  static char *tmemname = "hw.physmem";
  static char *fmemname = "vm.stats.vm.v_free_count";
  int pagesize=-1;
  size_t size;

  kvm_t *kvmd = NULL;
  struct kvm_swap swapinfo;
  char errbuf[_POSIX2_LINE_MAX];
  time_t current;
  struct timeval uptime;

  time(&current);
  
  if (sysctlbyname(uptimename, NULL, &size, NULL, NULL) < 0){
    errf("sysctlbyname (%m)");
    die();
  }


  if (sysctlbyname(uptimename, &uptime, &size, NULL, NULL) < 0){
    errf("Failed to get cpu stats (%m)");
    die();
  }
  
  printf("packet.os.uptime %ld\n",(current-(uptime.tv_sec)));


  if (sysctlbyname(cpname, NULL, &size, NULL, NULL) < 0){
    errf("sysctlbyname (%m)");
    die();
  }

  if (size != sizeof cp_time){
    errf("bailing out, trying to write into something to small");
    die();
  }

  if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
    errf("Failed to get cpu stats (%m)");
    die();
  }

  user=cp_time[CP_USER];
  nice=cp_time[CP_NICE];
  kernel=cp_time[CP_SYS];
  idle=cp_time[CP_IDLE];

  sleep(1);

  if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
    errf("Failed to get cpu stats (%m)");
    die();
  }
  
  user=cp_time[CP_USER]-user;
  nice=cp_time[CP_NICE]-nice;
  kernel=cp_time[CP_SYS]-kernel;
  idle=cp_time[CP_IDLE]-idle;
  
  total=user+nice+kernel+idle;
  stat=((float)(user+nice)/(float)total)*100.0;
  printf("packet.cpu.user %.2f\n",stat);
  stat=((float)(kernel)/(float)total)*100.0;
  printf("packet.cpu.kernel %.2f\n",stat);
  stat=((float)(idle)/(float)total)*100.0;
  printf("packet.cpu.idle %.2f\n",stat);
 
   /* Cos i-scream's expects this to be sent :/ */
  printf("packet.cpu.iowait 0\n");
  printf("packet.cpu.swap 0\n");
   
  /* MEMORY STATS */
  pagesize=getpagesize();
  if (pagesize==-1){
    errf("Failed to get pagesize (%m)");
    die();
  }

  if (sysctlbyname(tmemname, NULL, &size, NULL, NULL) < 0){
      errf("sysctlbyname (%m)");
      die();
    }

  if (sysctlbyname(tmemname, &totalmem, &size, NULL, NULL) < 0){
    errf("Failed to get memory stats (%m)");
    die();
  }

   if (sysctlbyname(fmemname, NULL, &size, NULL, NULL) < 0){
      errf("sysctlbyname (%m)");
      die();
    }

  if (sysctlbyname(fmemname, &freemem, &size, NULL, NULL) < 0){
    errf("Failed to get memory stats (%m)");
    die();
  }
  
  totalmem/=(1024*1024);
  freemem=(freemem*pagesize)/(1024*1024);
  
  printf("packet.memory.total %ld\n", totalmem);
  printf("packet.memory.free %ld\n", freemem);
  printf("packet.memory.used %ld\n", (totalmem-freemem));

  /* Swap stats */

  /* Get sufficent privilages to do this */
  if (gid!=egid){
    /* Means we are setgid something, hopefully kmem :) */
    if ((setegid(egid)) != 0){
      errf("Failed to gain sufficent privilages (%m)");
      die();
    }
  }

  kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
  if (kvmd == NULL){
    errf("Failed to open kvm info to get swap stats (%m)");
    die();
  }
  
  /* Lose our setgid'ness */

  if ((setegid(gid)) != 0){
    errf("Failed to release permissions, refusing to keep setgid. (%m)");
    die();
  }


  /* ok, just for proof of concept atm, ideally this will need to handle more 
     than one swap device */

  if ((kvm_getswapinfo(kvmd, &swapinfo, 1,0)) == -1){
    errf("Failed to get swap info (%m)");
    die();
  }
  
  swaptotal=((((long)swapinfo.ksw_total)*pagesize)/1024)/1024;
  swapused=((((long)swapinfo.ksw_used)*pagesize)/1024)/1024;

  printf("packet.swap.total %ld\n" , swaptotal);
  printf("packet.swap.used %ld\n", swapused);
  printf("packet.swap.free %lu\n", (swaptotal-swapused));
  
}

void processStats(){
  int sleeping=-1;
  int zombie=0;
  int stopped=0;
  int running=0;
  char *line;
  char *line_p;
  
  FILE *f;

  if((f=popen("/bin/ps -ax" , "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++);
    line_p=strchr(line_p, ' ');
    for(; (*line_p == ' ') && (*line_p != '\0'); line_p++);
    if (line_p==NULL) abort();

    if (*line_p=='S') sleeping++;
    if (*line_p=='I') sleeping++;
    if (*line_p=='R') running++;
    if (*line_p=='D') running++;
    if (*line_p=='Z') zombie++;
    if (*line_p=='T') stopped++;
    if (*line_p=='J') running++;

  }

  if((pclose(f)) != 0) {
    errf("Failed to close process stats (%m)");
    die();
  }
  
  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));


}


int main(){
  uid=getuid();
  euid=geteuid();
  gid=getgid();
  egid=getegid();

  /* We dont want to run with more permissions than we need, until we need em */
  if ((setegid(gid)) != 0){
    errf("Failed to release permissions, refusing to keep setgid. (%m)");
    die();
  }

 if ((seteuid(uid)) != 0){
    errf("Failed to release permissions, refusing to keep setuid. (%m)");
    die();
  }

  diskStats();
  osStats();
  loadStats();
  userStats();
  systemStats();
  processStats();
  exit(0);
}
Revision:	1.11
Committed:	Fri Mar 28 16:30:31 2003 UTC (21 years, 7 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.10:	+0 -0 lines
State:	*FILE REMOVED*
Log Message:	Removed some un-used code from CVS. We can always resurrect this later if someone feels they want to work on it. Gone are the old perl ihost which isn't needed now, winhost which is broken and shows no sign of being fixed, and DBReporter. If someone wants to revive them, I'll undelete them :-)
#	User	Rev	Content
1	tdb	1.9	/*
2			* i-scream central monitoring system
3	tdb	1.10	* http://www.i-scream.org.uk
4	tdb	1.9	* Copyright (C) 2000-2002 i-scream
5			*
6			* This program is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU General Public License
8			* as published by the Free Software Foundation; either version 2
9			* of the License, or (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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19			*/
20
21	pajs	1.1	#include <stdio.h>
22			#include "ukcprog.h"
23			#include <sys/param.h>
24			#include <sys/ucred.h>
25			#include <sys/mount.h>
26			#include <string.h>
27			#include <stdlib.h>
28	pajs	1.2	#include <sys/utsname.h>
29			#include <sys/types.h>
30			#include <utmp.h>
31	pajs	1.4	#include <sys/dkstat.h>
32			#include <kvm.h>
33			#include <unistd.h>
34			#include <sys/sysctl.h>
35			#include <fcntl.h>
36			#include <limits.h>
37	pajs	1.7	#include <time.h>
38			#include <sys/time.h>
39	pajs	1.4
40			uid_t uid;
41			uid_t euid;
42			gid_t gid;
43			gid_t egid;
44	pajs	1.1
45			void die(){
46			exit(1);
47			}
48
49			void diskStats(){
50			struct statfs *mp;
51			int nummnt;
52			int counter=0;
53			char *mntpnt;
54			char *devicename;
55
56			nummnt=getmntinfo(&mp , MNT_LOCAL);
57			if (nummnt<=0){
58			errf("Failed to get disk stats (%m)");
59			die();
60			}
61
62			mntpnt=malloc(MNAMELEN);
63			devicename=malloc(MNAMELEN);
64
65			for(;counter<nummnt;counter++){
66			strncpy(mntpnt, mp->f_mntonname ,MNAMELEN);
67			strncpy(devicename, mp->f_mntfromname ,MNAMELEN);
68			printf("packet.disk.p%d.attributes.mount %s\n", counter, mntpnt);
69			printf("packet.disk.p%d.attributes.name %s\n", counter, devicename);
70			printf("packet.disk.p%d.attributes.kbytes %lu\n",counter, ((mp->f_bsize/1024) * mp->f_blocks));
71			printf("packet.disk.p%d.attributes.used %lu\n",counter, (((mp->f_bsize/1024) * mp->f_blocks) -((mp->f_bsize/1024) * mp->f_bfree)));
72			printf("packet.disk.p%d.attributes.avail %lu\n",counter, (mp->f_bsize/1024) * mp->f_bavail);
73			printf("packet.disk.p%d.attributes.totalinodes %lu\n", counter, mp->f_files);
74			printf("packet.disk.p%d.attributes.freeinodes %lu\n",counter, mp->f_ffree);
75
76			mp++;
77			}
78
79
80			}
81
82	pajs	1.2	void osStats(){
83			struct utsname os;
84
85			if((uname(&os)) != 0){
86			errf("Failed to get os stats (%m)");
87			die();
88			}
89
90			printf("packet.os.name %s\n", os.sysname);
91			printf("packet.os.release %s\n" , os.release);
92			printf("packet.os.version %s\n", os.version);
93			printf("packet.os.sysname %s\n" , os.nodename);
94			printf("packet.os.platform %s\n", os.machine);
95
96			}
97			void loadStats(){
98			double loadav[3];
99
100			if((getloadavg(loadav,3)) == -1){
101			errf("Failed to get load averages (%m)");
102			die();
103			}
104
105			printf("packet.load.load1 %.2f\n",loadav[0]);
106			printf("packet.load.load5 %.2f\n",loadav[1]);
107			printf("packet.load.load15 %.2f\n",loadav[2]);
108			}
109
110	pajs	1.3	void userStats(){
111			struct utmp users;
112			FILE *f;
113			int numusers=0;
114
115			if ((f=fopen(_PATH_UTMP, "r")) == NULL){
116			errf("Failed to get user stats(%m)");
117			die();
118			}
119
120			printf("packet.users.list");
121
122			while((fread(&users, sizeof(users),1,f)) != 0){
123			if (users.ut_name[0] == '\0') continue;
124			printf(" %s",users.ut_name);
125			numusers++;
126			}
127
128			printf("\npacket.users.count %d\n",numusers);
129
130	pajs	1.4	}
131
132			void systemStats(){
133
134			long cp_time[CPUSTATES];
135			long total, user, idle, kernel, nice;
136			long totalmem, freemem, swaptotal, swapused;
137	pajs	1.8	float stat;
138	pajs	1.4
139	pajs	1.7	static char *uptimename = "kern.boottime";
140	pajs	1.4	static char *cpname = "kern.cp_time";
141			static char *tmemname = "hw.physmem";
142			static char *fmemname = "vm.stats.vm.v_free_count";
143			int pagesize=-1;
144			size_t size;
145
146			kvm_t *kvmd = NULL;
147			struct kvm_swap swapinfo;
148			char errbuf[_POSIX2_LINE_MAX];
149	pajs	1.7	time_t current;
150			struct timeval uptime;
151
152			time(&current);
153
154			if (sysctlbyname(uptimename, NULL, &size, NULL, NULL) < 0){
155			errf("sysctlbyname (%m)");
156			die();
157			}
158
159
160			if (sysctlbyname(uptimename, &uptime, &size, NULL, NULL) < 0){
161			errf("Failed to get cpu stats (%m)");
162			die();
163			}
164	pajs	1.4
165	pajs	1.7	printf("packet.os.uptime %ld\n",(current-(uptime.tv_sec)));
166
167
168	pajs	1.4	if (sysctlbyname(cpname, NULL, &size, NULL, NULL) < 0){
169			errf("sysctlbyname (%m)");
170			die();
171			}
172
173			if (size != sizeof cp_time){
174			errf("bailing out, trying to write into something to small");
175			die();
176			}
177
178			if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
179			errf("Failed to get cpu stats (%m)");
180			die();
181			}
182
183			user=cp_time[CP_USER];
184			nice=cp_time[CP_NICE];
185			kernel=cp_time[CP_SYS];
186			idle=cp_time[CP_IDLE];
187
188			sleep(1);
189
190			if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
191			errf("Failed to get cpu stats (%m)");
192			die();
193			}
194	pajs	1.3
195	pajs	1.8	user=cp_time[CP_USER]-user;
196			nice=cp_time[CP_NICE]-nice;
197			kernel=cp_time[CP_SYS]-kernel;
198			idle=cp_time[CP_IDLE]-idle;
199	pajs	1.4
200			total=user+nice+kernel+idle;
201	pajs	1.8	stat=((float)(user+nice)/(float)total)*100.0;
202			printf("packet.cpu.user %.2f\n",stat);
203			stat=((float)(kernel)/(float)total)*100.0;
204			printf("packet.cpu.kernel %.2f\n",stat);
205			stat=((float)(idle)/(float)total)*100.0;
206			printf("packet.cpu.idle %.2f\n",stat);
207	pajs	1.4
208			/* Cos i-scream's expects this to be sent :/ */
209			printf("packet.cpu.iowait 0\n");
210			printf("packet.cpu.swap 0\n");
211
212			/* MEMORY STATS */
213			pagesize=getpagesize();
214			if (pagesize==-1){
215			errf("Failed to get pagesize (%m)");
216			die();
217			}
218
219			if (sysctlbyname(tmemname, NULL, &size, NULL, NULL) < 0){
220			errf("sysctlbyname (%m)");
221			die();
222			}
223	pajs	1.3
224	pajs	1.4	if (sysctlbyname(tmemname, &totalmem, &size, NULL, NULL) < 0){
225			errf("Failed to get memory stats (%m)");
226			die();
227			}
228
229			if (sysctlbyname(fmemname, NULL, &size, NULL, NULL) < 0){
230			errf("sysctlbyname (%m)");
231			die();
232			}
233
234			if (sysctlbyname(fmemname, &freemem, &size, NULL, NULL) < 0){
235			errf("Failed to get memory stats (%m)");
236			die();
237			}
238
239			totalmem/=(1024*1024);
240			freemem=(freemempagesize)/(10241024);
241
242			printf("packet.memory.total %ld\n", totalmem);
243			printf("packet.memory.free %ld\n", freemem);
244			printf("packet.memory.used %ld\n", (totalmem-freemem));
245
246			/* Swap stats */
247
248			/* Get sufficent privilages to do this */
249			if (gid!=egid){
250			/* Means we are setgid something, hopefully kmem :) */
251			if ((setegid(egid)) != 0){
252			errf("Failed to gain sufficent privilages (%m)");
253			die();
254			}
255			}
256
257			kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
258			if (kvmd == NULL){
259			errf("Failed to open kvm info to get swap stats (%m)");
260			die();
261			}
262
263	pajs	1.8	/* Lose our setgid'ness */
264	pajs	1.5
265			if ((setegid(gid)) != 0){
266			errf("Failed to release permissions, refusing to keep setgid. (%m)");
267			die();
268			}
269
270	pajs	1.4
271			/* ok, just for proof of concept atm, ideally this will need to handle more
272			than one swap device */
273
274			if ((kvm_getswapinfo(kvmd, &swapinfo, 1,0)) == -1){
275			errf("Failed to get swap info (%m)");
276			die();
277			}
278
279			swaptotal=((((long)swapinfo.ksw_total)*pagesize)/1024)/1024;
280			swapused=((((long)swapinfo.ksw_used)*pagesize)/1024)/1024;
281
282			printf("packet.swap.total %ld\n" , swaptotal);
283			printf("packet.swap.used %ld\n", swapused);
284			printf("packet.swap.free %lu\n", (swaptotal-swapused));
285
286	pajs	1.3	}
287
288	pajs	1.6	void processStats(){
289			int sleeping=-1;
290			int zombie=0;
291			int stopped=0;
292			int running=0;
293			char *line;
294			char *line_p;
295
296			FILE *f;
297
298			if((f=popen("/bin/ps -ax" , "r")) == NULL) {
299			errf("Failed to get process stats (%m)");
300			die();
301			}
302
303			while((line=fpgetline(f)) != NULL) {
304			line_p=line;
305			for(; (line_p == ' ') && (line_p != '\0'); line_p++);
306			line_p=strchr(line_p, ' ');
307			for(; (line_p == ' ') && (line_p != '\0'); line_p++);
308			line_p=strchr(line_p, ' ');
309			for(; (line_p == ' ') && (line_p != '\0'); line_p++);
310			if (line_p==NULL) abort();
311
312			if (*line_p=='S') sleeping++;
313			if (*line_p=='I') sleeping++;
314			if (*line_p=='R') running++;
315			if (*line_p=='D') running++;
316			if (*line_p=='Z') zombie++;
317			if (*line_p=='T') stopped++;
318			if (*line_p=='J') running++;
319
320			}
321
322			if((pclose(f)) != 0) {
323			errf("Failed to close process stats (%m)");
324			die();
325			}
326
327			printf("packet.processes.sleeping %d\n",sleeping);
328			printf("packet.processes.cpu %d\n",running);
329			printf("packet.processes.zombie %d\n",zombie);
330			printf("packet.processes.stopped %d\n", stopped);
331			printf("packet.processes.total %d\n", (sleeping+running+zombie+stopped));
332
333
334			}
335
336
337	pajs	1.1	int main(){
338	pajs	1.4	uid=getuid();
339			euid=geteuid();
340			gid=getgid();
341			egid=getegid();
342
343			/* We dont want to run with more permissions than we need, until we need em */
344	pajs	1.5	if ((setegid(gid)) != 0){
345			errf("Failed to release permissions, refusing to keep setgid. (%m)");
346			die();
347			}
348
349			if ((seteuid(uid)) != 0){
350			errf("Failed to release permissions, refusing to keep setuid. (%m)");
351			die();
352			}
353	pajs	1.4
354	pajs	1.1	diskStats();
355	pajs	1.2	osStats();
356			loadStats();
357	pajs	1.3	userStats();
358	pajs	1.4	systemStats();
359	pajs	1.6	processStats();
360	pajs	1.1	exit(0);
361			}