-/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-

- * vim: sts=0 sw=8 ts=8 tw=78 noexpandtab

- *

-

- gchar *path;

- gchar *id;

- gdouble temp;

- GDBusProxy *sensor_proxy; // dbus object

- GError *error;

-static void update_device(DevInfo *info)

-{

- GError *error = NULL;

- GVariant *tempgvar = NULL;

- GVariant *tempgvar2 = NULL;

- gdouble temp;

- // check valid input parameter

- g_return_if_fail(info != NULL);

- // check connection too

- g_return_if_fail(connection != NULL);

- g_clear_error(&info->error);

- // check for sensor_proxy, which should exist at this point, make one if necessary and save it into DevInfo

- // this is used to get the temp value the direct way

- if(NULL == info->sensor_proxy)

- {

- info->sensor_proxy = g_dbus_proxy_new_sync(connection, G_DBUS_PROXY_FLAGS_NONE, NULL,

- UDISKS2_BUS_NAME,

- info->path,

- UDISKS2_PROPERTIES_INTERFACE,

- NULL, &error

- );

-

- // check, just to be sure

- if (NULL == info->sensor_proxy)

- {

- g_debug("Failed to get drive temperature 1: %s",

- error->message);

- g_clear_error(&error);

- return;

- }

- }

- if (!g_dbus_proxy_call_sync(sensor_proxy, "Get", NULL,

- G_TYPE_STRING, UDISKS2_BUS_NAME,

- G_TYPE_STRING, "DriveAtaSmartTimeCollected", G_TYPE_INVALID,

- G_TYPE_VALUE, &smart_time,

- G_TYPE_INVALID) ||

- !g_value_get_uint64(&smart_time))

- {

- g_object_unref(sensor_proxy);

- return;

- }

- if (!g_dbus_proxy_call_sync(sensor_proxy, "Get", &info->error,

- G_TYPE_STRING, UDISKS2_BUS_NAME,

- G_TYPE_STRING, "DriveAtaSmartBlob", G_TYPE_INVALID,

- G_TYPE_VALUE, &smart_blob_val,

- G_TYPE_INVALID))

- {

- g_debug("Error getting DriveAtaSmartBlob %s",

- info->error ? info->error->message : "NULL");

- g_object_unref(sensor_proxy);

- return;

- }

- smart_blob = g_value_get_boxed(&smart_blob_val);

-

- sk_disk_open(NULL, &sk_disk);

- sk_disk_set_blob(sk_disk, smart_blob->data, smart_blob->len);

- if (sk_disk_smart_get_temperature(sk_disk, &temperature) < 0)

- {

- g_debug("Error getting temperature from AtaSmartBlob");

- g_free(sk_disk);

- g_array_free(smart_blob, TRUE);

- g_object_unref(sensor_proxy);

- return;

- }

- // directly asking the device's DBus object for the temp

- tempgvar = g_dbus_proxy_call_sync(info->sensor_proxy, "Get",

- g_variant_new ("(ss)",

- UDISKS2_DEVICE_INTERFACE2_NAME,

- "SmartTemperature"), // parameters

- G_DBUS_CALL_FLAGS_NONE, // flags

- -1, // timeout

- NULL, // cancellable

- &error);

- if (NULL == tempgvar)

- {

- g_debug("Failed to get drive temperature 2: %s",

- error->message);

- g_clear_error(&error);

- // throw away proxy, maybe next time it will be better

- g_clear_object(&info->sensor_proxy);

- return;

- }

- else

- {

- // tempgvar comes back as sg along the lines of array(gvariant(tempasdouble))

- // hence unpacking

- // need to free up every param / return value, so can't do it like:

- //temp = g_variant_get_double(g_variant_get_variant(g_variant_get_child_value(tempgvar, 0)));

- tempgvar2 = g_variant_get_child_value(tempgvar, 0);

- g_variant_unref(tempgvar);

- tempgvar = g_variant_get_variant(tempgvar2);

- g_variant_unref(tempgvar2);

- temp = g_variant_get_double(tempgvar);

- g_variant_unref(tempgvar);

- // temp in K

- info->temp = temp - 273.15;

- }

- const gchar *path;

- DevInfo *info;

- // fd2: Gets the object path sensor_proxy is for.

- path = g_dbus_proxy_get_object_path(sensor_proxy);

- g_debug("%s changed()", path);

- info = g_hash_table_lookup(devices, path);

-// update_device(info);

- GDBusProxy *proxy = NULL;

- GError *error = NULL;

- DevInfo *info;

- /* This connection will be used for everything, including the obtaining

- * of sensor data

- */

- // fd1: Returns a connection to the given bus. The connection is a global variable shared with other callers of this function.

- // fd2: Synchronously connects to the message bus specified by bus_type . Note that the returned object may shared with other callers, e.g. if two separate parts of a process calls this function with the same bus_type , they will share the same object.

- connection = g_bus_get_sync(G_BUS_TYPE_SYSTEM, NULL, &error);

- if (NULL == connection)

- {

- g_debug("Failed to open connection to DBUS: %s",

- error->message);

- g_clear_error(&error);

- return;

- }

- /* This was the proxy which is only used once during the enumeration of

- * the device object paths

- */

- // I use it to get all info of all devices at once

- // fd1: Creates a new proxy for a remote interface exported by a connection on a message bus.

- // fd2: Creates a proxy for accessing interface_name on the remote object at object_path owned by name at connection and synchronously loads D-Bus properties unless the G_DBUS_PROXY_FLAGS_DO_NOT_LOAD_PROPERTIES flag is used.

- proxy = g_dbus_proxy_new_sync(connection, G_DBUS_PROXY_FLAGS_NONE, NULL,

- UDISKS2_BUS_NAME,

- UDISKS2_OBJECT_PATH,

- UDISKS2_INTERFACE_NAME,

- NULL, &error

- );

-

- if (NULL == proxy)

- {

- g_debug("dbus conn proxy fail: %s",

- error->message);

- g_clear_error(&error);

- g_clear_object(&connection);

- return;

- }

- /* The object paths of the disks are enumerated and placed in an array

- * of object paths

- */

- // fd1: Function for synchronously invoking a method and receiving reply values.

- // fd2: Synchronously invokes the method_name method on proxy.

- // called "EnumerateDevices" method on UDisks object through dbus

- // which returned an array of objectpaths

- // "GetManagedObjects" returns dict of (objectpath, (dict of (string [ie. if. name], dict of(string [ie. property name], variant [ie. prop. value]))))

- // g_dbus_proxy_call_sync() returns NULL on error, GVariant * otherwise

- // need second param to prevent memory leak

- GVariant *managed_objects, *managed_objects2;

- managed_objects2 = g_dbus_proxy_call_sync(proxy, "GetManagedObjects",

- NULL, // parameters

- G_DBUS_CALL_FLAGS_NONE, // flags

- -1, // timeout

- NULL, // cancellable

- &error);

- if (NULL == managed_objects2)

- {

- g_debug("Failed to enumerate disk devices: %s",

- error->message);

- g_clear_error(&error);

- g_clear_object(&proxy);

- g_clear_object(&connection);

- return;

- }

- // the result dictionary is enclosed in an array, unpack

- managed_objects = g_variant_get_child_value(managed_objects2, 0);

- g_variant_unref(managed_objects2);

- // iterator for the result dictionary

- // iterator code is based on the g_variant_iter_next() documentation

- // iter is freed if the GVariant is, when using g_variant_iter_init()

- GVariantIter iter;

- gchar *key; // object path (like '/org/freedesktop/UDisks2/drives/Samsung_SSD_840_EVO_250GB_*insert drive serial nr.*')

- GVariant *value;

- g_variant_iter_init(&iter, managed_objects);

- // "{sv}" is a GVariant format string

- // {} dictionary of, s string, v GVariant

- // changed to "{oa{sa{sv}}}" on error message 'the GVariant format string '{sv}' has a type of '{sv}' but the given value has a type of 'a{oa{sa{sv}}}''

- // a is array, o is object path

- // NOO!! the right format string is "{o@*}", which means get an object path into the 1st variable (key)

- // and get 'everything else' (as a GVariant) into the 2nd variable (value)

- // needs the & before the key and value params!

- while (g_variant_iter_next(&iter, "{o@*}", &key, &value))

- {

- // level 2

- // create a dictionary of value

- // the two interface names that we are searching for are known and defined

- // can't use GVariantDict, it only supports '{sv}' but the given value has a type of '{sa{sv}}'

- // using general lookup

- GVariant *propdict; // drive data

- GVariant *propdict2; // drive smart data

- // make two dictionaries that contain the properties of the drive interfaces

- propdict = g_variant_lookup_value(value, UDISKS2_DEVICE_INTERFACE_NAME, G_VARIANT_TYPE_DICTIONARY);

- propdict2 = g_variant_lookup_value(value, UDISKS2_DEVICE_INTERFACE2_NAME, G_VARIANT_TYPE_DICTIONARY);

- // do we have the right ifname keys?

- if((NULL != propdict) && (NULL != propdict2))

- {

- // get data

- gchar *id;

- gchar *model;

- gboolean smartenabled;

- gdouble temp;

- // NULL, bc we don't care about the length of the string

- // typecast bc g_variant_get_string() returns const char*

- id = (gchar *) g_variant_get_string(g_variant_lookup_value(propdict, "Id", G_VARIANT_TYPE_STRING), NULL);

- model = (gchar *) g_variant_get_string(g_variant_lookup_value(propdict, "Model", G_VARIANT_TYPE_STRING), NULL);

- smartenabled = g_variant_get_boolean(g_variant_lookup_value(propdict2, "SmartEnabled", G_VARIANT_TYPE_BOOLEAN));

- temp = g_variant_get_double(g_variant_lookup_value(propdict2, "SmartTemperature", G_VARIANT_TYPE_DOUBLE));

- // only go on if smart is enabled

- // save data

- if(smartenabled)

- {

- info = g_malloc0(sizeof(DevInfo));

- if (NULL == devices)

- {

- devices = g_hash_table_new(g_str_hash,

- g_str_equal);

- }

- info->id = g_strdup(id);

- info->path = g_strdup(key);

- // temp in K

- // this could be left at 0.0, 2 seconds later it will be refreshed anyway

- info->temp = (gdouble)temp - 273.15;

- g_hash_table_insert(devices, info->id, info);

- // Write the sensor data

- sensors_applet_plugin_add_sensor(sensors,

- id,

- "Disk Temperature",

- model,

- TEMP_SENSOR,

- FALSE,

- HDD_ICON,

- DEFAULT_GRAPH_COLOR);

- g_debug("Added %s", id);

- }

- else

- {

- g_debug ("No temp data for device: %s\n", key);

- }

- g_free(id);

- g_free(model);

- }

- // free propdict, propdict2

- // g_variant_dict_unref() may not work a few times, gives error

- // this one seems to do fine

- if(NULL != propdict) {g_variant_unref(propdict);}

- if(NULL != propdict2) {g_variant_unref(propdict2);}

- g_free(key);

- g_variant_unref(value);

- } // end of while loop

- g_variant_unref(managed_objects);

- g_clear_object(&proxy);

- if (NULL == devices)

- {

- g_clear_object(&connection);

- }

-

- const gchar *id,

- SensorType type,

- GError **error) {

- DevInfo *info;

-

- // get device stuct from data store

- info = (DevInfo *)g_hash_table_lookup(devices, path);

- if (NULL == info)

- {

- g_set_error(error, SENSORS_APPLET_PLUGIN_ERROR, 0,

- "Error finding disk with path %s", path);

- return 0.0;

- }

-

- if (info->error)

- {

- *error = info->error;

- info->error = NULL;

- return 0.0;

- }

-

- // refresh device temp

- /* update value since Changed() signal doesn't fire manually enough so

- * poll instead */

- update_device(info);

- return info->temp;

- return plugin_name;

- GList *sensors = NULL;

- udisks2_plugin_get_sensors(&sensors);

- return sensors;

- return udisks2_plugin_init();

- const gchar *id,

- SensorType type,

- GError **error) {

- return udisks2_plugin_get_sensor_value(path, id, type, error);