X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fzone.c;h=b8cc9197c2de5eea70990a1e4209b7f284a8d292;hb=2d4ef8aa6e7c9e2139b2bd2e22d790948980d62b;hp=1781775ff90b68a5a4abba5af4792ea086175b40;hpb=47761781efc410c30c2735cb10713b833e2a5148;p=collectd.git

diff --git a/src/zone.c b/src/zone.c
index 1781775f..b8cc9197 100644
--- a/src/zone.c
+++ b/src/zone.c
@@ -17,6 +17,7 @@
  *
  * Authors:
  *   Mathijs Mohlmann
+ *   Dagobert Michelsen (forward-porting)
  **/
 
 #define _BSD_SOURCE
@@ -33,7 +34,6 @@
 #include "utils_avltree.h"
 
 #define	MAX_PROCFS_PATH	40
-#define MAX_ZONE_NAME 20
 #define FRC2PCT(pp)(((float)(pp))/0x8000*100)
 
 typedef struct zone_stats {
@@ -43,6 +43,12 @@ typedef struct zone_stats {
 
 static long pagesize;
 
+static int zone_init (void)
+{
+	pagesize = sysconf(_SC_PAGESIZE);
+	return (0);
+}
+
 static int
 zone_compare(const zoneid_t *a, const zoneid_t *b)
 {
@@ -59,11 +65,9 @@ zone_read_procfile(char *pidstr, char *file, void *buf, size_t bufsize)
 	int fd;
 
 	char procfile[MAX_PROCFS_PATH];
-	(void)snprintf(procfile, MAX_PROCFS_PATH, "/proc/%s/%s", pidstr, file);
-	while ((fd = open(procfile, O_RDONLY)) == -1) {
-		if ((errno != EMFILE) || (errno != ENFILE)) {
-			return(1);
-		}
+	(void)snprintf(procfile, sizeof(procfile), "/proc/%s/%s", pidstr, file);
+	if ((fd = open(procfile, O_RDONLY)) == -1) {
+		return (1);
 	}
 
 	if (pread(fd, buf, bufsize, 0) != bufsize) {
@@ -119,14 +123,17 @@ zone_find_stats(c_avl_tree_t *tree, zoneid_t zoneid)
 static void
 zone_submit_values(c_avl_tree_t *tree)
 {
-	char          zonename[MAX_ZONE_NAME];
+	char          zonename[ZONENAME_MAX];
 	zoneid_t     *zoneid = NULL;
 	zone_stats_t *stats  = NULL;
 
 	while (c_avl_pick (tree, (void **)&zoneid, (void **)&stats) == 0)
 	{
-		getzonenamebyid(*zoneid, zonename, MAX_ZONE_NAME-1);
-		zone_submit_value(zonename, (gauge_t)FRC2PCT(stats->pctcpu));
+		if (getzonenamebyid(*zoneid, zonename, sizeof( zonename )) == -1) {
+			WARNING("zone plugin: error retreiving zonename");
+		} else {
+			zone_submit_value(zonename, (gauge_t)FRC2PCT(stats->pctcpu));
+		}
 		free(stats);
 		free(zoneid);
 	}
@@ -142,7 +149,6 @@ zone_scandir(DIR *procdir)
 	psinfo_t      psinfo;
 	c_avl_tree_t *tree;
 	zone_stats_t *stats;
-/*	size_t    physmem = sysconf(_SC_PHYS_PAGES) * pagesize;*/
 
 	if (!(tree=c_avl_create((void *) zone_compare))) {
 		WARNING("zone plugin: Failed to create tree");
@@ -160,8 +166,10 @@ zone_scandir(DIR *procdir)
 				  sizeof(psinfo_t)) != 0)
 			continue;
 		stats = zone_find_stats(tree, psinfo.pr_zoneid);
-		stats->pctcpu += psinfo.pr_pctcpu;
-		stats->pctmem += psinfo.pr_pctmem;
+		if( stats ) {
+			stats->pctcpu += psinfo.pr_pctcpu;
+			stats->pctmem += psinfo.pr_pctmem;
+		}
 	}
 	return(tree);
 }
@@ -172,19 +180,22 @@ static int zone_read (void)
 	DIR          *procdir;
 	c_avl_tree_t *tree;
 
-	pagesize = sysconf(_SC_PAGESIZE);
 	if ((procdir = opendir("/proc")) == NULL) {
 		ERROR("zone plugin: cannot open /proc directory\n");
-		exit(1);
+		return (-1);
 	}
 
 	tree=zone_scandir(procdir);
 	closedir(procdir);
+	if (tree == NULL) {
+		return (-1);
+	}
 	zone_submit_values(tree); /* this also frees tree */
 	return (0);
 }
 
 void module_register (void)
 {
+	plugin_register_init ("zone", zone_init);
 	plugin_register_read ("zone", zone_read);
 } /* void module_register */