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>

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 *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];
  
  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));
  
}

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();
  exit(0);
}
Revision:	1.5
Committed:	Thu Mar 28 17:46:52 2002 UTC (23 years, 7 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.4:	+15 -3 lines
Log Message:	Added a few more highly unlikely to happen, but check anyway, security checks.
#	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
18			uid_t uid;
19			uid_t euid;
20			gid_t gid;
21			gid_t egid;
22	pajs	1.1
23			void die(){
24			exit(1);
25			}
26
27			void diskStats(){
28			struct statfs *mp;
29			int nummnt;
30			int counter=0;
31			char *mntpnt;
32			char *devicename;
33
34			nummnt=getmntinfo(&mp , MNT_LOCAL);
35			if (nummnt<=0){
36			errf("Failed to get disk stats (%m)");
37			die();
38			}
39
40			mntpnt=malloc(MNAMELEN);
41			devicename=malloc(MNAMELEN);
42
43			for(;counter<nummnt;counter++){
44			strncpy(mntpnt, mp->f_mntonname ,MNAMELEN);
45			strncpy(devicename, mp->f_mntfromname ,MNAMELEN);
46			printf("packet.disk.p%d.attributes.mount %s\n", counter, mntpnt);
47			printf("packet.disk.p%d.attributes.name %s\n", counter, devicename);
48			printf("packet.disk.p%d.attributes.kbytes %lu\n",counter, ((mp->f_bsize/1024) * mp->f_blocks));
49			printf("packet.disk.p%d.attributes.used %lu\n",counter, (((mp->f_bsize/1024) * mp->f_blocks) -((mp->f_bsize/1024) * mp->f_bfree)));
50			printf("packet.disk.p%d.attributes.avail %lu\n",counter, (mp->f_bsize/1024) * mp->f_bavail);
51			printf("packet.disk.p%d.attributes.totalinodes %lu\n", counter, mp->f_files);
52			printf("packet.disk.p%d.attributes.freeinodes %lu\n",counter, mp->f_ffree);
53
54			mp++;
55			}
56
57
58			}
59
60	pajs	1.2	void osStats(){
61			struct utsname os;
62
63			if((uname(&os)) != 0){
64			errf("Failed to get os stats (%m)");
65			die();
66			}
67
68			printf("packet.os.name %s\n", os.sysname);
69			printf("packet.os.release %s\n" , os.release);
70			printf("packet.os.version %s\n", os.version);
71			printf("packet.os.sysname %s\n" , os.nodename);
72			printf("packet.os.platform %s\n", os.machine);
73
74			}
75			void loadStats(){
76			double loadav[3];
77
78			if((getloadavg(loadav,3)) == -1){
79			errf("Failed to get load averages (%m)");
80			die();
81			}
82
83			printf("packet.load.load1 %.2f\n",loadav[0]);
84			printf("packet.load.load5 %.2f\n",loadav[1]);
85			printf("packet.load.load15 %.2f\n",loadav[2]);
86			}
87
88	pajs	1.3	void userStats(){
89			struct utmp users;
90			FILE *f;
91			int numusers=0;
92
93			if ((f=fopen(_PATH_UTMP, "r")) == NULL){
94			errf("Failed to get user stats(%m)");
95			die();
96			}
97
98			printf("packet.users.list");
99
100			while((fread(&users, sizeof(users),1,f)) != 0){
101			if (users.ut_name[0] == '\0') continue;
102			printf(" %s",users.ut_name);
103			numusers++;
104			}
105
106			printf("\npacket.users.count %d\n",numusers);
107
108	pajs	1.4	}
109
110			void systemStats(){
111
112			long cp_time[CPUSTATES];
113			long total, user, idle, kernel, nice;
114			long totalmem, freemem, swaptotal, swapused;
115
116			static char *cpname = "kern.cp_time";
117			static char *tmemname = "hw.physmem";
118			static char *fmemname = "vm.stats.vm.v_free_count";
119			int pagesize=-1;
120			size_t size;
121
122			kvm_t *kvmd = NULL;
123			struct kvm_swap swapinfo;
124			char errbuf[_POSIX2_LINE_MAX];
125
126			if (sysctlbyname(cpname, NULL, &size, NULL, NULL) < 0){
127			errf("sysctlbyname (%m)");
128			die();
129			}
130
131			if (size != sizeof cp_time){
132			errf("bailing out, trying to write into something to small");
133			die();
134			}
135
136			if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
137			errf("Failed to get cpu stats (%m)");
138			die();
139			}
140
141			user=cp_time[CP_USER];
142			nice=cp_time[CP_NICE];
143			kernel=cp_time[CP_SYS];
144			idle=cp_time[CP_IDLE];
145
146			sleep(1);
147
148			if (sysctlbyname(cpname, &cp_time, &size, NULL, NULL) < 0){
149			errf("Failed to get cpu stats (%m)");
150			die();
151			}
152	pajs	1.3
153	pajs	1.4	user-=cp_time[CP_USER];
154			nice-=cp_time[CP_NICE];
155			kernel-=cp_time[CP_SYS];
156			idle-=cp_time[CP_IDLE];
157
158			total=user+nice+kernel+idle;
159
160			printf("packet.cpu.user %ld\n",((user+nice)*100)/total);
161			printf("packet.cpu.kernel %ld\n",(kernel*100)/total);
162			printf("packet.cpu.idle %ld\n",(idle*100)/total);
163
164			/* Cos i-scream's expects this to be sent :/ */
165			printf("packet.cpu.iowait 0\n");
166			printf("packet.cpu.swap 0\n");
167
168			/* MEMORY STATS */
169			pagesize=getpagesize();
170			if (pagesize==-1){
171			errf("Failed to get pagesize (%m)");
172			die();
173			}
174
175			if (sysctlbyname(tmemname, NULL, &size, NULL, NULL) < 0){
176			errf("sysctlbyname (%m)");
177			die();
178			}
179	pajs	1.3
180	pajs	1.4	if (sysctlbyname(tmemname, &totalmem, &size, NULL, NULL) < 0){
181			errf("Failed to get memory stats (%m)");
182			die();
183			}
184
185			if (sysctlbyname(fmemname, NULL, &size, NULL, NULL) < 0){
186			errf("sysctlbyname (%m)");
187			die();
188			}
189
190			if (sysctlbyname(fmemname, &freemem, &size, NULL, NULL) < 0){
191			errf("Failed to get memory stats (%m)");
192			die();
193			}
194
195			totalmem/=(1024*1024);
196			freemem=(freemempagesize)/(10241024);
197
198			printf("packet.memory.total %ld\n", totalmem);
199			printf("packet.memory.free %ld\n", freemem);
200			printf("packet.memory.used %ld\n", (totalmem-freemem));
201
202			/* Swap stats */
203
204			/* Get sufficent privilages to do this */
205			if (gid!=egid){
206			/* Means we are setgid something, hopefully kmem :) */
207			if ((setegid(egid)) != 0){
208			errf("Failed to gain sufficent privilages (%m)");
209			die();
210			}
211			}
212
213			kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, errbuf);
214			if (kvmd == NULL){
215			errf("Failed to open kvm info to get swap stats (%m)");
216			die();
217			}
218
219			/* Lose are setgid'ness */
220	pajs	1.5
221			if ((setegid(gid)) != 0){
222			errf("Failed to release permissions, refusing to keep setgid. (%m)");
223			die();
224			}
225
226	pajs	1.4
227			/* ok, just for proof of concept atm, ideally this will need to handle more
228			than one swap device */
229
230			if ((kvm_getswapinfo(kvmd, &swapinfo, 1,0)) == -1){
231			errf("Failed to get swap info (%m)");
232			die();
233			}
234
235			swaptotal=((((long)swapinfo.ksw_total)*pagesize)/1024)/1024;
236			swapused=((((long)swapinfo.ksw_used)*pagesize)/1024)/1024;
237
238			printf("packet.swap.total %ld\n" , swaptotal);
239			printf("packet.swap.used %ld\n", swapused);
240			printf("packet.swap.free %lu\n", (swaptotal-swapused));
241
242	pajs	1.3	}
243
244	pajs	1.1	int main(){
245	pajs	1.4	uid=getuid();
246			euid=geteuid();
247			gid=getgid();
248			egid=getegid();
249
250			/* We dont want to run with more permissions than we need, until we need em */
251	pajs	1.5	if ((setegid(gid)) != 0){
252			errf("Failed to release permissions, refusing to keep setgid. (%m)");
253			die();
254			}
255
256			if ((seteuid(uid)) != 0){
257			errf("Failed to release permissions, refusing to keep setuid. (%m)");
258			die();
259			}
260	pajs	1.4
261	pajs	1.1	diskStats();
262	pajs	1.2	osStats();
263			loadStats();
264	pajs	1.3	userStats();
265	pajs	1.4	systemStats();
266	pajs	1.1	exit(0);
267			}