X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fzone.c;h=15eae6a222fa942116841c36c0dd20f6676417e1;hb=73de67494a4f5d9a698f2a5ab93f85e19bccb90a;hp=6a01b298bcced48ada82f0e08ae8f257c2eef005;hpb=c9adea50e71c3a73989b658d82712f02cea3c901;p=collectd.git

diff --git a/src/zone.c b/src/zone.c
index 6a01b298..15eae6a2 100644
--- a/src/zone.c
+++ b/src/zone.c
@@ -17,9 +17,18 @@
  *
  * Authors:
  *   Mathijs Mohlmann
+ *   Dagobert Michelsen (forward-porting)
  **/
 
-#define _BSD_SOURCE
+#if HAVE_CONFIG_H
+# include "config.h"
+# undef HAVE_CONFIG_H
+#endif
+/* avoid procfs.h error "Cannot use procfs in the large file compilation environment" */
+#if !defined(_LP64) && _FILE_OFFSET_BITS == 64
+#  undef _FILE_OFFSET_BITS
+#  undef _LARGEFILE64_SOURCE
+#endif
 
 #include "collectd.h"
 #include "common.h"
@@ -59,22 +68,24 @@ zone_compare(const zoneid_t *a, const zoneid_t *b)
 }
 
 static int
-zone_read_procfile(char *pidstr, char *file, void *buf, size_t bufsize)
+zone_read_procfile(char const *pidstr, char const *name, 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, name);
+	if ((fd = open(procfile, O_RDONLY)) == -1) {
+		return (1);
 	}
 
-	if (pread(fd, buf, bufsize, 0) != bufsize) {
+	if (sread(fd, buf, bufsize) != 0) {
+		char errbuf[1024];
+		ERROR ("zone plugin: Reading \"%s\" failed: %s", procfile,
+				sstrerror (errno, errbuf, sizeof (errbuf)));
 		close(fd);
 		return (1);
 	}
+
 	close(fd);
 	return (0);
 }
@@ -144,7 +155,6 @@ zone_submit_values(c_avl_tree_t *tree)
 static c_avl_tree_t *
 zone_scandir(DIR *procdir)
 {
-	char         *pidstr;
 	pid_t         pid;
 	dirent_t     *direntp;
 	psinfo_t      psinfo;
@@ -156,16 +166,19 @@ zone_scandir(DIR *procdir)
 		return(NULL);
 	}
 
-	for (rewinddir(procdir); (direntp = readdir(procdir)); ) {
-		pidstr = direntp->d_name;
+	rewinddir(procdir);
+	while ((direntp = readdir(procdir))) {
+		char const *pidstr = direntp->d_name;
 		if (pidstr[0] == '.')	/* skip "." and ".."  */
 			continue;
+
 		pid = atoi(pidstr);
 		if (pid == 0 || pid == 2 || pid == 3)
 			continue;	/* skip sched, pageout and fsflush */
-		if (zone_read_procfile(pidstr, "psinfo", &psinfo, 
-				  sizeof(psinfo_t)) != 0)
+
+		if (zone_read_procfile(pidstr, "psinfo", &psinfo, sizeof(psinfo_t)) != 0)
 			continue;
+
 		stats = zone_find_stats(tree, psinfo.pr_zoneid);
 		if( stats ) {
 			stats->pctcpu += psinfo.pr_pctcpu;
@@ -175,7 +188,6 @@ zone_scandir(DIR *procdir)
 	return(tree);
 }
 
-
 static int zone_read (void)
 {
 	DIR          *procdir;
@@ -183,11 +195,14 @@ static int zone_read (void)
 
 	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);
 }