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

diff --git a/src/zone.c b/src/zone.c
index 1781775f..78e91ef2 100644
--- a/src/zone.c
+++ b/src/zone.c
@@ -33,7 +33,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 +42,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 +64,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 +122,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 +148,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 +165,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 +179,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 */