1 files changed, 143 insertions, 0 deletions

diff --git a/mfi-mqtt/reporting.c b/mfi-mqtt/reporting.c
new file mode 100644
index 0000000..7b4131a
--- /dev/null
+++ b/mfi-mqtt/reporting.c
@@ -0,0 +1,143 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <stdbool.h>
+#include <json-c/json.h>
+
+#include <net/if.h>
+#include <sys/socket.h>
+#include <sys/ioctl.h>
+#include <arpa/inet.h>
+
+#include "reporting.h"
+
+power_statistics * init_power_statistics(int relay_num) {
+    power_statistics *stats = malloc(sizeof(power_statistics));
+    stats->relay_num = relay_num;
+    stats->engaged = calloc(strlen("false") + 1, sizeof(char));
+    return stats;
+}
+
+void free_power_statistics(power_statistics *stats) {
+    free(stats->engaged);
+    free(stats);
+}
+
+bool power_statistics_is_engaged(power_statistics *stats) {
+    return strcmp(stats->engaged, "on") == 0;
+}
+
+int power_statistics_load_from_file(power_statistics *stats) {
+    FILE *fp;
+    char filename[12];
+
+    snprintf(filename, 12, "/dev/power%d", stats->relay_num);
+
+    fp = fopen(filename, "r");
+    if (!fp) {
+        return 1;
+    }
+
+    if (fscanf(fp, "%s %f\n %f\n %f\n %f\n %f\n",
+                stats->engaged,
+                &stats->active_power,
+                &stats->energy_sum,
+                &stats->current_rms,
+                &stats->voltage_rms,
+                &stats->power_factor) != 6) {
+        fclose(fp);
+        return 1;
+    }
+
+    fclose(fp);
+    return 0;
+}
+
+int get_output_count() {
+    return access("/dev/power8", F_OK) != -1 ? 8 : 3;
+}
+
+int get_stats_all_outputs(power_statistics ***out_stats) {
+    int i;
+    int output_count;
+    power_statistics *stat;
+    power_statistics **stats;
+
+    output_count = get_output_count();
+    stats = malloc(sizeof(power_statistics *) * (output_count + 1));
+
+    for (i = output_count; i > 0; i--) {
+        stat = init_power_statistics(i);
+
+        if (power_statistics_load_from_file(stat)) {
+            continue;
+        }
+
+        stats[i] = stat;
+    }
+
+    *out_stats = stats;
+
+    return output_count;
+}
+
+json_object * format_power_satistics_output_json(power_statistics *stats) {
+    json_object *top_object;
+
+    top_object = json_object_new_object();
+    json_object_object_add(top_object, "output", json_object_new_int(stats->relay_num));
+    json_object_object_add(top_object, "engaged", json_object_new_boolean(power_statistics_is_engaged(stats)));
+    json_object_object_add(top_object, "active_power", json_object_new_double(stats->active_power));
+    json_object_object_add(top_object, "energy_sum", json_object_new_double(stats->energy_sum));
+    json_object_object_add(top_object, "current_rms", json_object_new_double(stats->current_rms));
+    json_object_object_add(top_object, "voltage_rms", json_object_new_double(stats->voltage_rms));
+    json_object_object_add(top_object, "power_factor", json_object_new_double(stats->power_factor));
+
+    return top_object;
+}
+
+char * format_report_all_outputs(power_statistics **stats, int report_count) {
+    int i;
+    char *output;
+        char hostname[256];
+    char ip_address[15];
+    const char *tmp;
+    json_object *top_object, *report_array;
+
+    memset(hostname, 0, sizeof(hostname));
+    gethostname(hostname, 255);
+
+    memset(ip_address, 0, sizeof(ip_address));
+    get_primary_ip_address(ip_address);
+
+    report_array = json_object_new_array();
+    top_object = json_object_new_object();
+    json_object_object_add(top_object, "hostname", json_object_new_string(hostname));
+    json_object_object_add(top_object, "ip_address", json_object_new_string(ip_address));
+    json_object_object_add(top_object, "reports", report_array);
+
+    for (i = report_count; i > 0; i--) {
+        json_object_array_add(report_array, format_power_satistics_output_json(stats[i]));
+        free_power_statistics(stats[i]);
+    }
+
+    tmp = json_object_to_json_string_ext(top_object, JSON_C_TO_STRING_PLAIN);
+    output = strdup(tmp);
+    json_object_put(top_object);
+
+    return output;
+}
+
+void get_primary_ip_address(char ip_address[15])
+{
+    int fd;
+    struct ifreq ifr;
+
+    fd = socket(AF_INET, SOCK_DGRAM, 0);
+    memcpy(ifr.ifr_name, "ath0", IFNAMSIZ-1);
+    ioctl(fd, SIOCGIFADDR, &ifr);
+    close(fd);
+
+    strncpy(ip_address, inet_ntoa(((struct sockaddr_in *)&ifr.ifr_addr)->sin_addr), 15);
+}