From: octo <octo>
Date: Sun, 16 Apr 2006 00:25:42 +0000 (+0000)
Subject: Fixed bugs in the Darwin code of the `disk' plugin. It mostly works now.
X-Git-Tag: liboping-0.1.0~77
X-Git-Url: https://git.tokkee.org/?a=commitdiff_plain;h=f66c1d1c2de579c64982b715e44591c6c0baeb7c;p=collectd.git

Fixed bugs in the Darwin code of the `disk' plugin. It mostly works now.
---

diff --git a/src/disk.c b/src/disk.c
index c4f919ab..beee7fa6 100644
--- a/src/disk.c
+++ b/src/disk.c
@@ -259,7 +259,10 @@ static signed long long dict_get_value (CFDictionaryRef dict, const char *key)
 	key_obj = CFStringCreateWithCString (kCFAllocatorDefault, key,
 		       	kCFStringEncodingASCII);
 	if (key_obj == NULL)
+	{
+		DBG ("CFStringCreateWithCString (%s) failed.", key);
 		return (-1LL);
+	}
 	
 	/* get => we don't need to release (== free) the object */
 	val_obj = (CFNumberRef) CFDictionaryGetValue (dict, key_obj);
@@ -267,11 +270,16 @@ static signed long long dict_get_value (CFDictionaryRef dict, const char *key)
 	CFRelease (key_obj);
 
 	if (val_obj == NULL)
+	{
+		DBG ("CFDictionaryGetValue (%s) failed.", key);
 		return (-1LL);
+	}
 
-	if (CFNumberGetValue (val_obj, kCFNumberSInt64Type, &val_int)
-			!= kIOReturnSuccess)
+	if (!CFNumberGetValue (val_obj, kCFNumberSInt64Type, &val_int))
+	{
+		DBG ("CFNumberGetValue (%s) failed.", key);
 		return (-1LL);
+	}
 
 	return (val_int);
 }
@@ -286,6 +294,7 @@ static void disk_read (void)
 	CFDictionaryRef		props_dict;
 	CFDictionaryRef		stats_dict;
 	CFDictionaryRef		child_dict;
+	kern_return_t           status;
 
 	signed long long read_ops;
 	signed long long read_byt;
@@ -313,6 +322,16 @@ static void disk_read (void)
 		stats_dict = NULL;
 		child_dict = NULL;
 
+		/* `disk_child' must be released */
+		if ((status = IORegistryEntryGetChildEntry (disk, kIOServicePlane, &disk_child))
+			       	!= kIOReturnSuccess)
+		{
+			/* This fails for example for DVD/CD drives.. */
+			DBG ("IORegistryEntryGetChildEntry (disk) failed: 0x%08x", status);
+			IOObjectRelease (disk);
+			continue;
+		}
+
 		/* We create `props_dict' => we need to release it later */
 		if (IORegistryEntryCreateCFProperties (disk,
 					(CFMutableDictionaryRef *) &props_dict,
@@ -321,6 +340,7 @@ static void disk_read (void)
 				!= kIOReturnSuccess)
 		{
 			syslog (LOG_ERR, "disk-plugin: IORegistryEntryCreateCFProperties failed.");
+			IOObjectRelease (disk_child);
 			IOObjectRelease (disk);
 			continue;
 		}
@@ -328,6 +348,7 @@ static void disk_read (void)
 		if (props_dict == NULL)
 		{
 			DBG ("IORegistryEntryCreateCFProperties (disk) failed.");
+			IOObjectRelease (disk_child);
 			IOObjectRelease (disk);
 			continue;
 		}
@@ -340,16 +361,7 @@ static void disk_read (void)
 			DBG ("CFDictionaryGetValue (%s) failed.",
 				       	kIOBlockStorageDriverStatisticsKey);
 			CFRelease (props_dict);
-			IOObjectRelease (disk);
-			continue;
-		}
-
-		/* `disk_child' must be released */
-		if (IORegistryEntryGetChildEntry (disk, kIOServicePlane, &disk_child)
-			       	!= kIOReturnSuccess)
-		{
-			DBG ("IORegistryEntryGetChildEntry (disk) failed.");
-			CFRelease (props_dict);
+			IOObjectRelease (disk_child);
 			IOObjectRelease (disk);
 			continue;
 		}
@@ -393,6 +405,7 @@ static void disk_read (void)
 			IOObjectRelease (disk);
 			continue;
 		}
+		DBG ("disk_name = %s", disk_name);
 
 		if ((read_ops != -1LL)
 				|| (read_byt != -1LL)