plugins/freebsd/freebsd.c

#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;

  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];
  nice-=cp_time[CP_NICE];
  kernel-=cp_time[CP_SYS];
  idle-=cp_time[CP_IDLE];
  
  total=user+nice+kernel+idle;

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