#include <string.h>
#include <stdbool.h>
#include <unistd.h>
+#include <math.h>
#include "report-generator.h"
#include "log.h"
#include "err-check.h"
#include "appinfo-provider.h"
-#include "stats.h"
+#include "sys-stats.h"
#include "clock.h"
+#include "proc-scanner.h"
+#include "process.h"
+#include "json-schema-defs.h"
struct report_generator_system {
/** system cpu usage statistics */
- struct stats_system previous;
+ struct sys_stats stats;
};
struct report_generator_process
{
- /** process pid */
- int pid;
- /** process cpu usage statistics */
- struct stats_process previous;
+ proc_scanner_t *scanner;
+ report_generator_filter_cb filter;
+ void *user_data;
+ JsonBuilder *current_builder;
};
-struct report_generator_app
+struct report_generator_top
{
- char *app_id;
- report_generator_process_t *process_gen;
+ struct sys_stats sys_stats;
+ proc_scanner_t *scanner;
+ report_generator_top_type_e type;
+ int limit;
};
-int _app_report_generator_setup_process_generator(report_generator_app_t *generator);
+struct report_generator_top_closure {
+ struct sys_stats sys_stats;
+ int current_index; int max_index;
+ struct process_usage_report *usage_report;
+};
+
+static int _report_generator_top_report_generator_scan(report_generator_top_t *generator);
report_generator_system_t *report_generator_new_system_report_generator()
{
if (!ret)
return NULL;
- if (stats_update_system_stats(&ret->previous) != 0) {
+ if (sys_stats_update(&ret->stats) != 0) {
ERR("stats_update_system_stats failed");
free(ret);
return NULL;
free(generator);
}
-report_generator_process_t *report_generator_new_process_report_generator(int pid)
+report_generator_process_t *report_generator_new_process_report_generator()
{
- ON_TRUE_RETURN_VAL(pid <= 0, NULL);
-
report_generator_process_t *ret = calloc(1, sizeof(struct report_generator_process));
if (!ret)
return NULL;
- if (stats_update_process_stats(pid, &ret->previous) != 0) {
- ERR("stats_update_process_stats failed.");
- free(ret);
- return NULL;
- };
+ ret->scanner = proc_scanner_new();
+ if (!ret->scanner) goto err;
- ret->pid = pid;
+ if (proc_scanner_scan(ret->scanner) != 0) {
+ ERR("proc_scanner_scan failed");
+ goto err;
+ }
return ret;
+
+err:
+ proc_scanner_free(ret->scanner);
+ free(ret);
+ return NULL;
}
void report_generator_free_process_generator(report_generator_process_t *generator)
{
- if (!generator)
- return;
-
+ if (!generator) return;
+ proc_scanner_free(generator->scanner);
free(generator);
}
-report_generator_app_t *report_generator_new_app_report_generator(const char *app_id)
+static bool _match_proc_and_append(struct process *proc, void *data)
{
- ON_NULL_RETURN_VAL(app_id, NULL);
+ report_generator_process_t *gen = data;
- report_generator_app_t *ret = calloc(1, sizeof(struct report_generator_app));
- if (!ret) {
- ERR("malloc failed");
- return NULL;
+ if (gen->filter && gen->filter(proc, gen->user_data)) {
+ process_serialize(proc, gen->current_builder);
}
- ret->app_id = strdup(app_id);
- _app_report_generator_setup_process_generator(ret);
- return ret;
+ return true;
}
-void report_generator_free_app_generator(report_generator_app_t *generator)
+int report_generator_generate_process_report(
+ report_generator_process_t *generator,
+ JsonBuilder *builder)
{
- if (!generator) return;
- report_generator_free_process_generator(generator->process_gen);
- free(generator->app_id);
- free(generator);
+ ON_NULL_RETURN_VAL(generator, -1);
+ ON_NULL_RETURN_VAL(builder, -1);
+
+ // refresh data
+ if (proc_scanner_scan(generator->scanner) != 0) {
+ ERR("proc_scanner_scan failed");
+ return -1;
+ }
+
+ generator->current_builder = builder;
+ json_builder_begin_object(builder);
+
+ json_builder_set_member_name(builder, SCHEMA_TYPE);
+ json_builder_add_string_value(builder, SCHEMA_TYPE_PROCESS);
+
+ json_builder_set_member_name(builder, "time");
+ json_builder_add_int_value(builder, clock_realtime_get());
+
+ json_builder_set_member_name(builder, SCHEMA_RESULT_DATA_PROCESS);
+ json_builder_begin_array(builder);
+
+ proc_scanner_foreach_process(generator->scanner, _match_proc_and_append, generator);
+ json_builder_end_array(builder);
+
+ json_builder_end_object(builder); // end report
+
+ generator->current_builder = NULL;
+ return 0;
+}
+
+void report_generator_set_filter(
+ report_generator_process_t *generator,
+ report_generator_filter_cb cb, const void *user_data)
+{
+ ON_NULL_RETURN(generator);
+ ON_NULL_RETURN(cb);
+
+ generator->filter = cb;
+ generator->user_data = (void*)user_data;
}
-int report_generator_generate_system_cpu_usage_report(
+int report_generator_generate_system_report(
report_generator_system_t *generator,
- int interval,
- struct system_cpu_usage_report *report)
+ struct system_report *report)
{
ON_NULL_RETURN_VAL(generator, -1);
- ON_TRUE_RETURN_VAL(interval < 0, -1);
ON_NULL_RETURN_VAL(report, -1);
- float usage;
- bool block = interval > 0;
- struct stats_system current;
-
- if (block) {
- if (stats_update_system_stats(&generator->previous) != 0) {
- ERR("stats_update_system_stats failed.");
- return -1;
- }
- sleep(interval);
- }
-
- if (stats_update_system_stats(¤t) != 0) {
+ if (sys_stats_update(&generator->stats) != 0) {
ERR("stats_update_system_stats failed.");
return -1;
}
- if (stats_get_system_cpu_usage_average(&generator->previous, ¤t, &usage))
+ if (sys_stats_get_cpu_usage_percentage(&generator->stats, &report->cpu))
{
ERR("stats_get_system_cpu_usage_average failed");
return -1;
}
- report->usage = usage;
- report->time = clock_realtime_get().tv_sec;
+ if (sys_stats_get_memory_usage_percentage(&generator->stats, &report->memory) != 0) {
+ ERR("stats_get_system_memory_usage failed.");
+ return -1;
+ }
- generator->previous = current;
+ report->time = clock_realtime_get();
return 0;
}
-int report_generator_generate_system_memory_usage_report(
- report_generator_system_t *generator,
- struct system_memory_usage_report *report) {
- ON_NULL_RETURN_VAL(generator, -1);
+int report_generator_generate_load_average_report(struct system_load_average_report *report)
+{
ON_NULL_RETURN_VAL(report, -1);
- float usage;
+ float a1, a5, a15;
- if (stats_get_system_memory_usage(&usage) != 0) {
- ERR("stats_get_system_memory_usage failed.");
+ if (sys_stats_get_load_averages(&a1, &a5, &a15) != 0) {
+ ERR("stats_get_load_averages failed.");
return -1;
}
- report->time = clock_realtime_get().tv_sec;
- report->usage = usage;
+ report->time = clock_realtime_get();
+ report->one_min_avg = a1;
+ report->five_min_avg = a5;
+ report->fifteen_min_avg = a15;
return 0;
}
-int report_generator_generate_process_cpu_usage_report(
- report_generator_process_t *generator,
- int interval,
- struct process_cpu_usage_report *report)
+report_generator_top_t *report_generator_new_top_report_generator(report_generator_top_type_e type, int limit)
{
- ON_NULL_RETURN_VAL(generator, -1);
- ON_TRUE_RETURN_VAL(interval < 0, -1);
- ON_NULL_RETURN_VAL(report, -1);
-
- struct stats_process current = {0,};
- bool block = interval > 0;
- float usage;
-
- if (block) {
- if (stats_update_process_stats(generator->pid, &generator->previous) != 0) {
- ERR("stats_update_process_stats failed.");
- return -1;
- }
- sleep(interval);
+ report_generator_top_t *gen = calloc(1, sizeof(report_generator_top_t));
+ if (!gen) {
+ ERR("calloc failed.");
+ return NULL;
}
- if (stats_update_process_stats(generator->pid, ¤t) != 0) {
- ERR("stats_update_process_stats failed.");
- return -1;
+ gen->type = type;
+ gen->limit = limit;
+ gen->scanner = proc_scanner_new();
+ if (!gen->scanner) {
+ report_generator_free_top_generator(gen);
+ return NULL;
}
-
- if (stats_get_process_cpu_usage_average(&generator->previous, ¤t, &usage) ) {
- ERR("stats_update_process_stats failed.");
- return -1;
+ // run initial scan, so other next report will have valid data.
+ if (_report_generator_top_report_generator_scan(gen) != 0) {
+ report_generator_free_top_generator(gen);
+ return NULL;
+ }
+ if (sys_stats_update(&gen->sys_stats) != 0) {
+ report_generator_free_top_generator(gen);
+ return NULL;
}
- report->time = clock_realtime_get().tv_sec;
- report->pid = generator->pid;
- report->usage = usage;
+ return gen;
+}
+
+void report_generator_free_top_generator(report_generator_top_t *generator)
+{
+ if (!generator) return;
+ proc_scanner_free(generator->scanner);
+ free(generator);
+}
+
+static bool _append_to_mem_report(struct process *proc, void *data)
+{
+ struct report_generator_top_closure *closure = data;
+ struct process_usage_report report = {0,};
+ const char *appid;
+ int mem;
+
+ if (closure->current_index >= closure->max_index)
+ return false;
+
+ if (process_get_memory_usage(proc, &mem) != 0) {
+ report.usage = NAN;
+ } else {
+ report.usage = (float)mem / closure->sys_stats.total_memory;
+ }
- generator->previous = current;
+ appid = process_get_appid(proc);
+ if (appid) strncpy(report.app_id, appid, sizeof(report.app_id));
+ report.pid = process_get_pid(proc);
+ report.time = clock_realtime_get();
+ closure->usage_report[closure->current_index++] = report;
- return 0;
+ return true;
}
-int report_generator_generate_process_memory_usage_report(
- report_generator_process_t *generator,
- struct process_memory_usage_report *report)
+static bool _append_to_cpu_report(struct process *proc, void *data)
{
- ON_NULL_RETURN_VAL(generator, -1);
- ON_NULL_RETURN_VAL(report, -1);
+ struct report_generator_top_closure *closure = data;
+ struct process_usage_report report = {0,};
+ const char *appid;
- float usage;
+ if (closure->current_index >= closure->max_index)
+ return false;
- if (stats_get_process_memory_usage(generator->pid, &usage) != 0) {
- ERR("stats_get_process_memory_usage failed.");
- return -1;
+ if (process_get_cpu_usage_percentage(proc, &report.usage) != 0) {
+ report.usage = NAN;
}
- report->time = clock_realtime_get().tv_sec;
- report->usage = usage;
+ appid = process_get_appid(proc);
+ if (appid) strncpy(report.app_id, appid, sizeof(report.app_id));
+ report.pid = process_get_pid(proc);
+ report.time = clock_realtime_get();
+ closure->usage_report[closure->current_index++] = report;
- return 0;
+ return true;
}
-int _app_report_generator_setup_process_generator(report_generator_app_t *generator)
+static int _sort_by_cpu_usage(struct process *proc1, struct process *proc2)
{
- int pid = app_info_provider_find_main_pid(generator->app_id);
- if (pid < 0) {
+ unsigned long long usage1, usage2;
+ if (process_get_cpu_usage(proc1, &usage1) != 0) {
+ return 1;
+ }
+ if (process_get_cpu_usage(proc2, &usage2) != 0) {
return -1;
}
+ return usage2 - usage1;
+}
- if (!generator->process_gen || generator->process_gen->pid != pid) {
- if (generator->process_gen)
- report_generator_free_process_generator(generator->process_gen);
- generator->process_gen = report_generator_new_process_report_generator(pid);
+static int _sort_by_memory_usage(struct process *proc1, struct process *proc2)
+{
+ int usage1, usage2;
+ if (process_get_memory_usage(proc1, &usage1) != 0) {
+ return 1;
}
- return 0;
+ if (process_get_memory_usage(proc2, &usage2) != 0) {
+ return -1;
+ }
+ return usage2 - usage1;
}
-int report_generator_generate_app_cpu_usage_report(
- report_generator_app_t *generator,
- int interval,
- struct app_cpu_usage_report *report)
+static int _report_generator_top_report_generator_scan_apps(
+ report_generator_top_t *generator)
{
- ON_NULL_RETURN_VAL(generator, -1);
- ON_TRUE_RETURN_VAL(interval < 0, -1);
- ON_NULL_RETURN_VAL(report, -1);
+ app_info_iterator_t *iter = app_info_provider_get_running_applications();
+ int i = 0;
- if (_app_report_generator_setup_process_generator(generator) != 0) {
- ERR("_app_report_generator_setup_process_generator failed.");
+ if (!iter) {
return -1;
}
- if (report_generator_generate_process_cpu_usage_report(
- generator->process_gen, interval, &report->process_report) != 0)
- {
- ERR("report_generator_generate_process_cpu_usage_report failed.");
- return 0;
+ int count = app_info_iterator_get_count(iter);
+ if (count < 1) {
+ app_info_iterator_free(iter);
+ return -1;
}
- strncpy(report->app_id, generator->app_id, sizeof(report->app_id));
- report->process_report.pid = generator->process_gen->pid;
+ int *pids = calloc(count, sizeof(int));
+ if (!pids) {
+ app_info_iterator_free(iter);
+ return -1;
+ }
- return 0;
+ do {
+ pids[i++] = app_info_iterator_get_pid(iter);
+ }
+ while (app_info_iterator_next(iter));
+
+ app_info_iterator_free(iter);
+
+ int ret = proc_scanner_scan_pids(generator->scanner, pids, i);
+ free(pids);
+ return ret;
+}
+
+static int _report_generator_top_report_generator_scan_all(
+ report_generator_top_t *generator)
+{
+ return proc_scanner_scan(generator->scanner);
}
-int report_generator_generate_app_memory_usage_report(
- report_generator_app_t *generator,
- struct app_memory_usage_report *report)
+static int _report_generator_top_report_generator_scan(
+ report_generator_top_t *generator)
+{
+ switch (generator->type) {
+ case REPORT_GENERATOR_TOP_TYPE_APPS:
+ return _report_generator_top_report_generator_scan_apps(generator);
+ case REPORT_GENERATOR_TOP_TYPE_ALL:
+ return _report_generator_top_report_generator_scan_all(generator);
+ break;
+ }
+
+ return -1;
+}
+
+int report_generator_generate_top_cpu_report(
+ report_generator_top_t *generator,
+ struct process_usage_report **report,
+ int *n_reports)
{
ON_NULL_RETURN_VAL(generator, -1);
ON_NULL_RETURN_VAL(report, -1);
+ ON_NULL_RETURN_VAL(n_reports, -1);
+
+ struct report_generator_top_closure closure = {0,};
- if (_app_report_generator_setup_process_generator(generator)) {
- ERR("_app_report_generator_setup_process_generator failed.");
+ if (sys_stats_update(&generator->sys_stats) != 0) {
return -1;
}
- if (report_generator_generate_process_memory_usage_report(
- generator->process_gen, &report->process_report) != 0)
- {
- ERR("report_generator_generate_process_memory_usage_report failed.");
- return 0;
+ if (_report_generator_top_report_generator_scan(generator) != 0) {
+ return -1;
}
- strncpy(report->app_id, generator->app_id, sizeof(report->app_id));
- report->process_report.pid = generator->process_gen->pid;
+ if (proc_scanner_sort_processes(generator->scanner, _sort_by_cpu_usage) != 0) {
+ return -1;
+ }
+
+ closure.max_index = generator->limit;
+ closure.sys_stats = generator->sys_stats;
+ closure.usage_report = calloc(generator->limit, sizeof(struct process_usage_report));
+ if (!closure.usage_report) {
+ return -1;
+ }
+
+ if (proc_scanner_foreach_process(generator->scanner, _append_to_cpu_report, &closure) != 0) {
+ free(closure.usage_report);
+ return -1;
+ }
+
+ *report = closure.usage_report;
+ *n_reports = closure.current_index;
return 0;
}
-int report_generator_generate_load_average_report(struct system_load_average_report *report)
+int report_generator_generate_top_memory_report(
+ report_generator_top_t *generator,
+ struct process_usage_report **report,
+ int *n_reports)
{
+ ON_NULL_RETURN_VAL(generator, -1);
ON_NULL_RETURN_VAL(report, -1);
+ ON_NULL_RETURN_VAL(n_reports, -1);
- float a1, a5, a15;
+ struct report_generator_top_closure closure = {0,};
- if (stats_get_load_averages(&a1, &a5, &a15) != 0) {
- ERR("stats_get_load_averages failed.");
+ if (sys_stats_update(&generator->sys_stats) != 0) {
return -1;
}
- report->time = clock_realtime_get().tv_sec;
- report->one_min_avg = a1;
- report->five_min_avg = a5;
- report->fifteen_min_avg = a15;
+ if (_report_generator_top_report_generator_scan(generator) != 0) {
+ return -1;
+ }
+
+ if (proc_scanner_sort_processes(generator->scanner, _sort_by_memory_usage) != 0) {
+ return -1;
+ }
+
+ closure.max_index = generator->limit;
+ closure.sys_stats = generator->sys_stats;
+ closure.usage_report = calloc(generator->limit, sizeof(struct process_usage_report));
+ if (!closure.usage_report) {
+ return -1;
+ }
+
+ if (proc_scanner_foreach_process(generator->scanner, _append_to_mem_report, &closure) != 0) {
+ free(closure.usage_report);
+ return -1;
+ }
+
+ *report = closure.usage_report;
+ *n_reports = closure.current_index;
return 0;
}
-