X-Git-Url: https://git.wpitchoune.net/gitweb/?p=psensor.git;a=blobdiff_plain;f=src%2Flib%2Fnvidia.c;h=685f97da9311b1e50a082400e9bf1be22a1c2592;hp=d50774ca8bb9112aa902674fc0feb98b5b6d7926;hb=c1e20f2631a1249720e9c75d753eacfcb0f6c7b9;hpb=e19b7aae2f107f501cdec4f942223152a96054c5

diff --git a/src/lib/nvidia.c b/src/lib/nvidia.c
index d50774c..685f97d 100644
--- a/src/lib/nvidia.c
+++ b/src/lib/nvidia.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2010-2014 jeanfi@gmail.com
+ * Copyright (C) 2010-2016 jeanfi@gmail.com
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License as
@@ -20,6 +20,8 @@
 #include <libintl.h>
 #define _(str) gettext(str)
 
+#include <limits.h>
+#include <math.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -29,15 +31,31 @@
 #include <NVCtrl/NVCtrl.h>
 #include <NVCtrl/NVCtrlLib.h>
 
+#include <nvidia.h>
 #include <psensor.h>
 
-Display *display;
+static Display *display;
 
-static char *get_product_name(int id)
+static const char *PROVIDER_NAME = "nvctrl";
+
+static void set_nvidia_id(struct psensor *s, int id)
+{
+	*(int *)s->provider_data = id;
+}
+
+static int get_nvidia_id(struct psensor *s)
+{
+	return *(int *)s->provider_data;
+}
+
+static char *get_product_name(int id, int type)
 {
 	char *name;
 	Bool res;
 
+	if (type & SENSOR_TYPE_FAN)
+		return strdup("NVIDIA");
+
 	res = XNVCTRLQueryTargetStringAttribute(display,
 						NV_CTRL_TARGET_TYPE_GPU,
 						id,
@@ -45,66 +63,37 @@ static char *get_product_name(int id)
 						NV_CTRL_STRING_PRODUCT_NAME,
 						&name);
 	if (res == True) {
-		if (strcmp(name, "Unknown")) {
+		if (strcmp(name, "Unknown"))
 			return name;
-		} else {
-			log_err(_("Unknown NVIDIA product name for GPU %d"),
-				id);
-			free(name);
-		}
+
+		log_err(_("%s: Unknown NVIDIA product name for GPU %d"),
+			PROVIDER_NAME,
+			id);
+		free(name);
 	} else {
-		log_err(_("Failed to retrieve NVIDIA product name for GPU %d"),
+		log_err(_("%s: "
+			  "Failed to retrieve NVIDIA product name for GPU %d"),
+			PROVIDER_NAME,
 			id);
 	}
 
 	return strdup("NVIDIA");
 }
 
-static double nv_get_temp(int id)
+static double get_att(int target, int id, int att)
 {
 	Bool res;
 	int temp;
 
-	res = XNVCTRLQueryTargetAttribute(display,
-					  NV_CTRL_TARGET_TYPE_GPU,
-					  id,
-					  0,
-					  NV_CTRL_GPU_CORE_TEMPERATURE,
-					  &temp);
+	res = XNVCTRLQueryTargetAttribute(display, target, id, 0, att, &temp);
 
 	if (res == True)
 		return temp;
-	else
-		return UNKNOWN_DBL_VALUE;
-}
 
-static double get_ambient_temp(int id)
-{
-	Bool res;
-	int temp;
-
-	res = XNVCTRLQueryTargetAttribute(display,
-					  NV_CTRL_TARGET_TYPE_GPU,
-					  id,
-					  0,
-					  NV_CTRL_AMBIENT_TEMPERATURE,
-					  &temp);
-
-	if (res == True)
-		return temp;
-	else
-		return UNKNOWN_DBL_VALUE;
+	return UNKNOWN_DBL_VALUE;
 }
 
-static double get_temp(int id, int type)
-{
-	if (type & SENSOR_TYPE_AMBIENT)
-		return get_ambient_temp(id);
-	else
-		return nv_get_temp(id);
-}
-
-static double get_usage_att(char *atts, char *att)
+static double get_usage_att(char *atts, const char *att)
 {
 	char *c, *key, *strv, *s;
 	size_t n;
@@ -154,21 +143,46 @@ static double get_usage_att(char *atts, char *att)
 	return v;
 }
 
+static const char *get_nvidia_type_str(int type)
+{
+	if (type & SENSOR_TYPE_GRAPHICS)
+		return "graphics";
+
+	if (type & SENSOR_TYPE_VIDEO)
+		return "video";
+
+	if (type & SENSOR_TYPE_MEMORY)
+		return "memory";
+
+	if (type & SENSOR_TYPE_PCIE)
+		return "PCIe";
+
+	if (type & SENSOR_TYPE_AMBIENT)
+		return "ambient";
+
+	if (type & SENSOR_TYPE_TEMP)
+		return "temp";
+
+	if (type & SENSOR_TYPE_FAN) {
+		if (type & SENSOR_TYPE_RPM)
+			return "fan rpm";
+
+		return "fan level";
+	}
+
+	return "unknown";
+}
+
 static double get_usage(int id, int type)
 {
-	char *stype, *atts;
+	const char *stype;
+	char *atts;
 	double v;
 	Bool res;
 
-	if (type & SENSOR_TYPE_GRAPHICS)
-		stype = "graphics";
-	else if (type & SENSOR_TYPE_VIDEO)
-		stype = "video";
-	else if (type & SENSOR_TYPE_MEMORY)
-		stype = "memory";
-	else if (type & SENSOR_TYPE_PCIE)
-		stype = "PCIe";
-	else
+	stype = get_nvidia_type_str(type);
+
+	if (!stype)
 		return UNKNOWN_DBL_VALUE;
 
 	res = XNVCTRLQueryTargetStringAttribute(display,
@@ -188,183 +202,201 @@ static double get_usage(int id, int type)
 	return v;
 }
 
+static double get_value(int id, int type)
+{
+	int att;
+
+	if (type & SENSOR_TYPE_TEMP) {
+		if (type & SENSOR_TYPE_AMBIENT)
+			att = NV_CTRL_AMBIENT_TEMPERATURE;
+		else
+			att = NV_CTRL_GPU_CORE_TEMPERATURE;
+
+		return get_att(NV_CTRL_TARGET_TYPE_GPU, id, att);
+	} else if (type & SENSOR_TYPE_FAN) {
+		if (type & SENSOR_TYPE_RPM)
+			return get_att(NV_CTRL_TARGET_TYPE_COOLER,
+				       id,
+				       NV_CTRL_THERMAL_COOLER_SPEED);
+		else /* SENSOR_TYPE_PERCENT */
+			return get_att(NV_CTRL_TARGET_TYPE_COOLER,
+				       id,
+				       NV_CTRL_THERMAL_COOLER_LEVEL);
+	} else { /* SENSOR_TYPE_PERCENT */
+		return get_usage(id, type);
+	}
+}
+
 static void update(struct psensor *sensor)
 {
 	double v;
+	int id;
 
-	if (sensor->type & SENSOR_TYPE_TEMP) {
-		v = get_temp(sensor->nvidia_id, sensor->type);
-	} else { /* SENSOR_TYPE_USAGE */
-		v = get_usage(sensor->nvidia_id, sensor->type);
-	}
+	id = get_nvidia_id(sensor);
+
+	v = get_value(id, sensor->type);
 
 	if (v == UNKNOWN_DBL_VALUE)
-		log_err(_("Failed to retrieve measure of type %x "
+		log_err(_("%s: Failed to retrieve measure of type %x "
 			  "for NVIDIA GPU %d"),
+			PROVIDER_NAME,
 			sensor->type,
-			sensor->nvidia_id);
+			id);
 	psensor_set_current_value(sensor, v);
 }
 
-static struct psensor *create_temp_sensor(int id, int subtype, int values_len)
+static int check_sensor(int id, int type)
 {
-	char name[200];
-	char *sid, *pname;
-	struct psensor *s;
-	int t;
-
-	pname = get_product_name(id);
-
-	if (subtype & SENSOR_TYPE_AMBIENT)
-		sprintf(name, "%s %d ambient", pname, id);
-	else
-		sprintf(name, "%s %d", pname, id);
-	free(pname);
-
-	sid = malloc(strlen("NVIDIA") + 1 + strlen(name) + 1);
-	sprintf(sid, "NVIDIA %s", name);
+	return get_value(id, type) != UNKNOWN_DBL_VALUE;
+}
 
-	t = SENSOR_TYPE_NVCTRL | SENSOR_TYPE_GPU | SENSOR_TYPE_TEMP | subtype;
+static char *i2str(int i)
+{
+	char *str;
+	size_t n;
 
-	s = psensor_create(sid,
-			   strdup(name),
-			   strdup(_("NVIDIA GPU")),
-			   t,
-			   values_len);
+	/* second +1 to avoid issue about the conversion of a double
+	 * to a lower int
+	 */
+	n = 1 + (ceil(log10(INT_MAX)) + 1) + 1;
 
-	s->nvidia_id = id;
+	str = malloc(n);
+	snprintf(str, n, "%d", i);
 
-	return s;
+	return str;
 }
 
-static struct psensor *create_usage_sensor(int id,
-					   int subtype,
-					   int values_len)
+static struct psensor *create_nvidia_sensor(int id, int subtype, int value_len)
 {
-	char name[200];
-	char *sid;
+	char *pname, *name, *strnid, *sid;
+	const char *stype;
+	int type;
+	size_t n;
 	struct psensor *s;
-	int t;
+	double v;
+
+	type = SENSOR_TYPE_NVCTRL | subtype;
 
-	if (subtype & SENSOR_TYPE_GRAPHICS)
-		sprintf(name, "GPU%d graphics", id);
-	else if (subtype & SENSOR_TYPE_MEMORY)
-		sprintf(name, "GPU%d memory", id);
-	else if (subtype & SENSOR_TYPE_VIDEO)
-		sprintf(name, "GPU%d video", id);
-	else /* if (subtype & SENSOR_TYPE_PCIE) */
-		sprintf(name, "GPU%d PCIe", id);
+	if (!check_sensor(id, type))
+		return NULL;
 
+	pname = get_product_name(id, type);
+	strnid = i2str(id);
+	stype = get_nvidia_type_str(type);
 
-	sid = malloc(strlen("NVIDIA") + 1 + strlen(name) + 1);
-	sprintf(sid, "NVIDIA %s", name);
+	n = strlen(pname) + 1 + strlen(strnid) + 1 + strlen(stype) + 1;
+	name = malloc(n);
+	sprintf(name, "%s %s %s", pname, strnid, stype);
 
-	t = SENSOR_TYPE_NVCTRL | SENSOR_TYPE_GPU | SENSOR_TYPE_USAGE | subtype;
+	sid = malloc(strlen(PROVIDER_NAME) + 1 + strlen(name) + 1);
+	sprintf(sid, "%s %s", PROVIDER_NAME, name);
 
-	s = psensor_create(sid,
-			   strdup(name),
-			   strdup(_("NVIDIA GPU")),
-			   t,
-			   values_len);
+	s = psensor_create(sid, name, pname, type, value_len);
+	s->provider_data = malloc(sizeof(int));
+	set_nvidia_id(s, id);
+
+	if ((type & SENSOR_TYPE_GPU) && (type & SENSOR_TYPE_TEMP)) {
+		v = get_att(NV_CTRL_TARGET_TYPE_GPU,
+			    id,
+			    NV_CTRL_GPU_CORE_THRESHOLD);
+		s->max = v;
+	}
 
-	s->nvidia_id = id;
+	free(strnid);
 
 	return s;
 }
 
-/*
-  Opens connection to X server and returns the number
-  of NVIDIA GPUs.
-
-  Return 0 if no NVIDIA gpus or cannot get information.
-*/
-static int init()
+static int init(void)
 {
-	int evt, err, n;
+	int evt, err;
 
 	display = XOpenDisplay(NULL);
 
 	if (!display) {
-		log_err(_("Cannot open connection to X11 server."));
+		log_err(_("%s: Cannot open connection to X11 server."),
+			PROVIDER_NAME);
 		return 0;
 	}
 
-	if (XNVCTRLQueryExtension(display, &evt, &err) &&
-	    XNVCTRLQueryTargetCount(display, NV_CTRL_TARGET_TYPE_GPU, &n))
-		return n;
+	if (XNVCTRLQueryExtension(display, &evt, &err))
+		return 1;
 
-	log_err(_("Failed to retrieve NVIDIA information."));
+	log_err(_("%s: Failed to retrieve NVIDIA information."),
+		PROVIDER_NAME);
 
 	return 0;
 }
 
 void nvidia_psensor_list_update(struct psensor **sensors)
 {
-	struct psensor **ss, *s;
+	struct psensor *s;
 
-	ss = sensors;
-	while (*ss) {
-		s = *ss;
+	while (*sensors) {
+		s = *sensors;
 
-		if (s->type & SENSOR_TYPE_NVCTRL)
+		if (!(s->type & SENSOR_TYPE_REMOTE)
+		    && s->type & SENSOR_TYPE_NVCTRL)
 			update(s);
 
-		ss++;
+		sensors++;
 	}
 }
 
-struct psensor **nvidia_psensor_list_add(struct psensor **sensors,
-					 int values_len)
+static void add(struct psensor ***sensors, int id, int type, int values_len)
+{
+	struct psensor *s;
+
+	s = create_nvidia_sensor(id, type, values_len);
+
+	if (s)
+		psensor_list_append(sensors, s);
+}
+
+void nvidia_psensor_list_append(struct psensor ***ss, int values_len)
 {
-	int i, n;
-	struct psensor **tmp, **ss, *s;
-
-	n = init();
-
-	ss = sensors;
-	for (i = 0; i < n; i++) {
-		s = create_temp_sensor(i, 0, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
-		s = create_temp_sensor(i, SENSOR_TYPE_AMBIENT, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
-		s = create_usage_sensor(i, SENSOR_TYPE_GRAPHICS, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
-		s = create_usage_sensor(i, SENSOR_TYPE_VIDEO, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
-		s = create_usage_sensor(i, SENSOR_TYPE_MEMORY, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
-		s = create_usage_sensor(i, SENSOR_TYPE_PCIE, values_len);
-		tmp = psensor_list_add(ss, s);
-		if (ss != tmp)
-			free(ss);
-
-		ss = tmp;
+	int i, n, utype;
+	Bool ret;
+
+	if (!init())
+		return;
+
+	ret = XNVCTRLQueryTargetCount(display, NV_CTRL_TARGET_TYPE_GPU, &n);
+	if (ret == True) {
+		for (i = 0; i < n; i++) {
+			add(ss,
+			    i,
+			    SENSOR_TYPE_GPU | SENSOR_TYPE_TEMP,
+			    values_len);
+
+			utype = SENSOR_TYPE_GPU | SENSOR_TYPE_PERCENT;
+			add(ss, i, utype | SENSOR_TYPE_AMBIENT, values_len);
+			add(ss, i, utype | SENSOR_TYPE_GRAPHICS, values_len);
+			add(ss, i, utype | SENSOR_TYPE_VIDEO, values_len);
+			add(ss, i, utype | SENSOR_TYPE_MEMORY, values_len);
+			add(ss, i, utype | SENSOR_TYPE_PCIE, values_len);
+		}
 	}
 
-	return ss;
+	ret = XNVCTRLQueryTargetCount(display, NV_CTRL_TARGET_TYPE_COOLER, &n);
+	if (ret == True) {
+		log_fct("%s: Number of fans: %d", PROVIDER_NAME, n);
+		for (i = 0; i < n; i++) {
+			utype = SENSOR_TYPE_FAN | SENSOR_TYPE_RPM;
+			if (check_sensor(i, utype))
+				add(ss, i, utype, values_len);
+
+			utype = SENSOR_TYPE_FAN | SENSOR_TYPE_PERCENT;
+			if (check_sensor(i, utype))
+				add(ss, i, utype, values_len);
+		}
+	} else {
+		log_err(_("%s: Failed to retrieve number of fans."),
+			PROVIDER_NAME);
+	}
 }
 
-void nvidia_cleanup()
+void nvidia_cleanup(void)
 {
 	if (display) {
 		XCloseDisplay(display);