[lttngtop.git] / src / common.c

/*
 * Copyright (C) 2011-2012 Julien Desfossez
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License Version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <babeltrace/ctf/events.h>
#include <stdlib.h>
#include <linux/unistd.h>
#include <string.h>
#include "common.h"

uint64_t get_cpu_id(const struct bt_ctf_event *event)
{
	const struct bt_definition *scope;
	uint64_t cpu_id;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_PACKET_CONTEXT);
	cpu_id = bt_ctf_get_uint64(bt_ctf_get_field(event, scope, "cpu_id"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "[error] get cpu_id\n");
		return -1ULL;
	}

	return cpu_id;
}

uint64_t get_context_tid(const struct bt_ctf_event *event)
{
	const struct bt_definition *scope;
	uint64_t tid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	tid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_tid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing tid context info\n");
		return -1ULL;
	}

	return tid;
}

uint64_t get_context_pid(const struct bt_ctf_event *event)
{
	const struct bt_definition *scope;
	uint64_t pid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	pid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_pid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing pid context info\n");
		return -1ULL;
	}

	return pid;
}

uint64_t get_context_ppid(const struct bt_ctf_event *event)
{
	const struct bt_definition *scope;
	uint64_t ppid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	ppid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_ppid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing ppid context info\n");
		return -1ULL;
	}

	return ppid;
}

uint64_t get_context_vtid(const struct bt_ctf_event *event)
{
	const struct definition *scope;
	uint64_t vtid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	vtid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_vtid"));
	if (bt_ctf_field_get_error()) {
		return -1ULL;
	}

	return vtid;
}

uint64_t get_context_vpid(const struct bt_ctf_event *event)
{
	const struct definition *scope;
	uint64_t vpid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	vpid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_vpid"));
	if (bt_ctf_field_get_error()) {
		return -1ULL;
	}

	return vpid;
}

uint64_t get_context_vppid(const struct bt_ctf_event *event)
{
	const struct definition *scope;
	uint64_t vppid;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	vppid = bt_ctf_get_int64(bt_ctf_get_field(event,
				scope, "_vppid"));
	if (bt_ctf_field_get_error()) {
		return -1ULL;
	}

	return vppid;
}

char *get_context_comm(const struct bt_ctf_event *event)
{
	const struct bt_definition *scope;
	char *comm;

	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	comm = bt_ctf_get_char_array(bt_ctf_get_field(event,
				scope, "_procname"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing comm context info\n");
		return NULL;
	}

	return comm;
}

/*
 * To get the parent process, put the pid in the tid field
 * because the parent process gets pid = tid
 */
struct processtop *find_process_tid(struct lttngtop *ctx, int tid, char *comm)
{
	struct processtop *tmp;

	tmp = g_hash_table_lookup(ctx->process_hash_table,
			(gconstpointer) (unsigned long) tid);

	return tmp;
}

struct processtop* add_proc(struct lttngtop *ctx, int tid, char *comm,
		unsigned long timestamp)
{
	struct processtop *newproc;

	/* if the PID already exists, we just rename the process */
	/* FIXME : need to integrate with clone/fork/exit to be accurate */
	newproc = find_process_tid(ctx, tid, comm);
	if (!newproc) {
		newproc = g_new0(struct processtop, 1);
		newproc->tid = tid;
		newproc->birth = timestamp;
		newproc->process_files_table = g_ptr_array_new();
		newproc->files_history = NULL;
		newproc->totalfileread = 0;
		newproc->totalfilewrite = 0;
		newproc->fileread = 0;
		newproc->filewrite = 0;
		newproc->syscall_info = NULL;
		newproc->threadparent = NULL;
		newproc->threads = g_ptr_array_new();
		newproc->perf = g_hash_table_new(g_str_hash, g_str_equal);
		g_ptr_array_add(ctx->process_table, newproc);
		g_hash_table_insert(ctx->process_hash_table,
				(gpointer) (unsigned long) tid, newproc);

		ctx->nbnewthreads++;
		ctx->nbthreads++;
	}
	newproc->comm = strdup(comm);

	return newproc;
}

struct processtop* update_proc(struct processtop* proc, int pid, int tid,
		int ppid, int vpid, int vtid, int vppid, char *comm)
{
	if (proc) {
		proc->pid = pid;
		proc->tid = tid;
		proc->ppid = ppid;
		proc->vpid = vpid;
		proc->vtid = vtid;
		proc->vppid = vppid;
		if (strcmp(proc->comm, comm) != 0) {
			free(proc->comm);
			proc->comm = strdup(comm);
		}
	}
	return proc;
}

/*
 * This function just sets the time of death of a process.
 * When we rotate the cputime we remove it from the process list.
 */
void death_proc(struct lttngtop *ctx, int tid, char *comm,
		unsigned long timestamp)
{
	struct processtop *tmp;
	tmp = find_process_tid(ctx, tid, comm);

	g_hash_table_remove(ctx->process_hash_table,
			(gpointer) (unsigned long) tid);
	if (tmp && strcmp(tmp->comm, comm) == 0) {
		tmp->death = timestamp;
		ctx->nbdeadthreads++;
		ctx->nbthreads--;
	}
}

struct processtop* get_proc(struct lttngtop *ctx, int tid, char *comm,
		unsigned long timestamp)
{
	struct processtop *tmp;
	tmp = find_process_tid(ctx, tid, comm);
	if (tmp && strcmp(tmp->comm, comm) == 0)
		return tmp;
	return add_proc(ctx, tid, comm, timestamp);
}

struct processtop *get_proc_pid(struct lttngtop *ctx, int tid, int pid,
		unsigned long timestamp)
{
	struct processtop *tmp;
	tmp = find_process_tid(ctx, tid, NULL);
	if (tmp && tmp->pid == pid)
		return tmp;
	return add_proc(ctx, tid, "Unknown", timestamp);
}

void add_thread(struct processtop *parent, struct processtop *thread)
{
	gint i;
	struct processtop *tmp;

	for (i = 0; i < parent->threads->len; i++) {
		tmp = g_ptr_array_index(parent->threads, i);
		if (tmp == thread)
			return;
	}
	g_ptr_array_add(parent->threads, thread);
}

struct cputime* add_cpu(int cpu)
{
	struct cputime *newcpu;

	newcpu = g_new0(struct cputime, 1);
	newcpu->id = cpu;
	newcpu->current_task = NULL;
	newcpu->perf = g_hash_table_new(g_str_hash, g_str_equal);

	g_ptr_array_add(lttngtop.cpu_table, newcpu);

	return newcpu;
}
struct cputime* get_cpu(int cpu)
{
	gint i;
	struct cputime *tmp;

	for (i = 0; i < lttngtop.cpu_table->len; i++) {
		tmp = g_ptr_array_index(lttngtop.cpu_table, i);
		if (tmp->id == cpu)
			return tmp;
	}

	return add_cpu(cpu);
}

/*
 * At the end of a sampling period, we need to display the cpu time for each
 * process and to reset it to zero for the next period
 */
void rotate_cputime(unsigned long end)
{
	gint i;
	struct cputime *tmp;
	unsigned long elapsed;

	for (i = 0; i < lttngtop.cpu_table->len; i++) {
		tmp = g_ptr_array_index(lttngtop.cpu_table, i);
		elapsed = end - tmp->task_start;
		if (tmp->current_task) {
			tmp->current_task->totalcpunsec += elapsed;
			tmp->current_task->threadstotalcpunsec += elapsed;
			if (tmp->current_task->pid != tmp->current_task->tid &&
					tmp->current_task->threadparent) {
				tmp->current_task->threadparent->threadstotalcpunsec += elapsed;
			}
		}
		tmp->task_start = end;
	}
}

void reset_perf_counter(gpointer key, gpointer value, gpointer user_data)
{
	((struct perfcounter*) value)->count = 0;
}

void copy_perf_counter(gpointer key, gpointer value, gpointer new_table)
{
	struct perfcounter *newperf;

	newperf = g_new0(struct perfcounter, 1);
	newperf->count = ((struct perfcounter *) value)->count;
	newperf->visible = ((struct perfcounter *) value)->visible;
	newperf->sort = ((struct perfcounter *) value)->sort;
	g_hash_table_insert((GHashTable *) new_table, strdup(key), newperf);
}

void copy_process_table(gpointer key, gpointer value, gpointer new_table)
{
	g_hash_table_insert((GHashTable *) new_table, key, value);
}

void rotate_perfcounter() {
	int i;
	struct processtop *tmp;
	for (i = 0; i < lttngtop.process_table->len; i++) {
		tmp = g_ptr_array_index(lttngtop.process_table, i);
		g_hash_table_foreach(tmp->perf, reset_perf_counter, NULL);
	}
}

void cleanup_processtop()
{
	gint i, j;
	struct processtop *tmp;
	struct files *tmpf; /* a temporary file */

	for (i = 0; i < lttngtop.process_table->len; i++) {
		tmp = g_ptr_array_index(lttngtop.process_table, i);
		tmp->totalcpunsec = 0;
		tmp->threadstotalcpunsec = 0;
		tmp->fileread = 0;
		tmp->filewrite = 0;

		for (j = 0; j < tmp->process_files_table->len; j++) {
			tmpf = g_ptr_array_index(tmp->process_files_table, j);
			if (tmpf != NULL) {
				tmpf->read = 0;
				tmpf->write = 0;

				if (tmpf->flag == __NR_close)
					g_ptr_array_index(
						tmp->process_files_table, j
					) = NULL;
			}
		}
	}
}

void reset_global_counters()
{
	lttngtop.nbnewproc = 0;
	lttngtop.nbdeadproc = 0;
	lttngtop.nbnewthreads = 0;
	lttngtop.nbdeadthreads = 0;
	lttngtop.nbnewfiles = 0;
	lttngtop.nbclosedfiles = 0;
}

void copy_global_counters(struct lttngtop *dst)
{
	dst->nbproc = lttngtop.nbproc;
	dst->nbnewproc = lttngtop.nbnewproc;
	dst->nbdeadproc = lttngtop.nbdeadproc;
	dst->nbthreads = lttngtop.nbthreads;
	dst->nbnewthreads = lttngtop.nbnewthreads;
	dst->nbdeadthreads = lttngtop.nbdeadthreads;
	dst->nbfiles = lttngtop.nbfiles;
	dst->nbnewfiles = lttngtop.nbnewfiles;
	dst->nbclosedfiles = lttngtop.nbclosedfiles;
	reset_global_counters();
}

struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
{
	gint i, j;
	unsigned long time;
	struct lttngtop *dst;
	struct processtop *tmp, *tmp2, *new;
	struct cputime *tmpcpu, *newcpu;
	struct files *tmpfile, *newfile;

	dst = g_new0(struct lttngtop, 1);
	dst->start = start;
	dst->end = end;
	copy_global_counters(dst);
	dst->process_table = g_ptr_array_new();
	dst->files_table = g_ptr_array_new();
	dst->cpu_table = g_ptr_array_new();
	dst->process_hash_table = g_hash_table_new(g_direct_hash, g_direct_equal);
	g_hash_table_foreach(lttngtop.process_hash_table, copy_process_table,
			dst->process_hash_table);

	rotate_cputime(end);

	for (i = 0; i < lttngtop.process_table->len; i++) {
		tmp = g_ptr_array_index(lttngtop.process_table, i);
		new = g_new0(struct processtop, 1);

		memcpy(new, tmp, sizeof(struct processtop));
		new->threads = g_ptr_array_new();
		new->comm = strdup(tmp->comm);
		new->process_files_table = g_ptr_array_new();
		new->files_history = tmp->files_history;
		new->perf = g_hash_table_new(g_str_hash, g_str_equal);
		g_hash_table_foreach(tmp->perf, copy_perf_counter, new->perf);

		/* compute the stream speed */
		if (end - start != 0) {
			time = (end - start) / NSEC_PER_SEC;
			new->fileread = new->fileread/(time);
			new->filewrite = new->filewrite/(time);
		}

		for (j = 0; j < tmp->process_files_table->len; j++) {
			tmpfile = g_ptr_array_index(tmp->process_files_table, j);

			newfile = malloc(sizeof(struct files));

			if (tmpfile != NULL) {
				memcpy(newfile, tmpfile, sizeof(struct files));
				newfile->name = strdup(tmpfile->name);
				newfile->ref = new;
				g_ptr_array_add(new->process_files_table,
						newfile);
				g_ptr_array_add(dst->files_table, newfile);
			} else {
				g_ptr_array_add(new->process_files_table, NULL);
				g_ptr_array_add(dst->files_table, NULL);
			}
			/*
			 * if the process died during the last period, we remove all
			 * files associated with if after the copy
			 */
			if (tmp->death > 0 && tmp->death < end) {
				/* FIXME : close the files before */
				g_ptr_array_remove(tmp->process_files_table, tmpfile);
				g_free(tmpfile);
			}
		}
		g_ptr_array_add(dst->process_table, new);

		/*
		 * if the process died during the last period, we remove it from
		 * the current process list after the copy
		 */
		if (tmp->death > 0 && tmp->death < end) {
			g_ptr_array_remove(lttngtop.process_table, tmp);
			/* FIXME : TRUE does not mean clears the object in it */
			g_ptr_array_free(tmp->threads, TRUE);
			free(tmp->comm);
			g_ptr_array_free(tmp->process_files_table, TRUE);
			/* FIXME : clear elements */
			g_hash_table_destroy(tmp->perf);
			g_free(tmp);
		}
	}
	rotate_perfcounter();

	for (i = 0; i < lttngtop.cpu_table->len; i++) {
		tmpcpu = g_ptr_array_index(lttngtop.cpu_table, i);
		newcpu = g_new0(struct cputime, 1);
		memcpy(newcpu, tmpcpu, sizeof(struct cputime));
		newcpu->perf = g_hash_table_new(g_str_hash, g_str_equal);
		g_hash_table_foreach(tmpcpu->perf, copy_perf_counter, newcpu->perf);
		/*
		 * note : we don't care about the current process pointer in the copy
		 * so the reference is invalid after the memcpy
		 */
		g_ptr_array_add(dst->cpu_table, newcpu);
	}
	/* FIXME : better algo */
	/* create the threads index if required */
	for (i = 0; i < dst->process_table->len; i++) {
		tmp = g_ptr_array_index(dst->process_table, i);
		if (tmp->pid == tmp->tid) {
			for (j = 0; j < dst->process_table->len; j++) {
				tmp2 = g_ptr_array_index(dst->process_table, j);
				if (tmp2->pid == tmp->pid) {
					tmp2->threadparent = tmp;
					g_ptr_array_add(tmp->threads, tmp2);
				}
			}
		}
	}

	//  update_global_stats(dst);
	cleanup_processtop();

	return dst;
}


enum bt_cb_ret handle_statedump_process_state(struct bt_ctf_event *call_data,
		void *private_data)
{
	const struct bt_definition *scope;
	struct processtop *proc;
	unsigned long timestamp;
	int64_t pid, tid, ppid, vtid, vpid, vppid;
	char *procname;

	timestamp = bt_ctf_get_timestamp(call_data);
	if (timestamp == -1ULL)
		goto error;

	scope = bt_ctf_get_top_level_scope(call_data,
			 BT_EVENT_FIELDS);
	pid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
			     scope, "_pid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing pid context info\n");
		goto error;
	}
	ppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
				scope, "_ppid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing ppid context info\n");
		goto error;
	}
	tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
				scope, "_tid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing tid context info\n");
		goto error;
	}
	vtid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
				scope, "_vtid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing vtid context info\n");
		goto error;
	}
	vpid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
				scope, "_vpid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing vpid context info\n");
		goto error;
	}
	vppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
				scope, "_vppid"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing vppid context info\n");
		goto error;
	}

	scope = bt_ctf_get_top_level_scope(call_data,
			BT_EVENT_FIELDS);
	procname = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
				scope, "_name"));
	if (bt_ctf_field_get_error()) {
		fprintf(stderr, "Missing process name context info\n");
		goto error;
	}

	proc = find_process_tid(&lttngtop, tid, procname);
	if (proc == NULL)
		proc = add_proc(&lttngtop, tid, procname, timestamp);
	update_proc(proc, pid, tid, ppid, vpid, vtid, vppid, procname);

	free(proc->comm);
	proc->comm = strdup(procname);
	proc->pid = pid;

	return BT_CB_OK;

error:
	return BT_CB_ERROR_STOP;
}

struct tm format_timestamp(uint64_t timestamp)
{
	struct tm tm;
	uint64_t ts_sec = 0, ts_nsec;
	time_t time_s;

	ts_nsec = timestamp;
	ts_sec += ts_nsec / NSEC_PER_SEC;
	ts_nsec = ts_nsec % NSEC_PER_SEC;

	time_s = (time_t) ts_sec;

	localtime_r(&time_s, &tm);

	return tm;
}
Commit	Line	Data
1fc22eb4	1	/*
aa15ac1c	2	* Copyright (C) 2011-2012 Julien Desfossez
1fc22eb4 JD	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License Version 2 as
	6	* published by the Free Software Foundation;
	7	*
	8	* This program is distributed in the hope that it will be useful,
	9	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	10	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	11	* GNU General Public License for more details.
	12	*
71bd7ce1 AM	13	* You should have received a copy of the GNU General Public License along
	14	* with this program; if not, write to the Free Software Foundation, Inc.,
	15	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
1fc22eb4 JD	16	*/
1fc22eb4 JD	17
1dec520a	18	#include <babeltrace/ctf/events.h>
1fc22eb4	19	#include <stdlib.h>
ceb3a221	20	#include <linux/unistd.h>
1fc22eb4 JD	21	#include <string.h>
	22	#include "common.h"
	23
4adc8274	24	uint64_t get_cpu_id(const struct bt_ctf_event *event)
d67167cd	25	{
2e0a1190	26	const struct bt_definition *scope;
d67167cd JD	27	uint64_t cpu_id;
	28
	29	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_PACKET_CONTEXT);
	30	cpu_id = bt_ctf_get_uint64(bt_ctf_get_field(event, scope, "cpu_id"));
	31	if (bt_ctf_field_get_error()) {
	32	fprintf(stderr, "[error] get cpu_id\n");
	33	return -1ULL;
	34	}
	35
	36	return cpu_id;
	37	}
	38
4adc8274	39	uint64_t get_context_tid(const struct bt_ctf_event *event)
1dec520a	40	{
2e0a1190	41	const struct bt_definition *scope;
1dec520a JD	42	uint64_t tid;
	43
	44	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	45	tid = bt_ctf_get_int64(bt_ctf_get_field(event,
	46	scope, "_tid"));
	47	if (bt_ctf_field_get_error()) {
	48	fprintf(stderr, "Missing tid context info\n");
	49	return -1ULL;
	50	}
	51
	52	return tid;
	53	}
	54
4adc8274	55	uint64_t get_context_pid(const struct bt_ctf_event *event)
1dec520a	56	{
2e0a1190	57	const struct bt_definition *scope;
1dec520a JD	58	uint64_t pid;
	59
	60	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	61	pid = bt_ctf_get_int64(bt_ctf_get_field(event,
	62	scope, "_pid"));
	63	if (bt_ctf_field_get_error()) {
	64	fprintf(stderr, "Missing pid context info\n");
	65	return -1ULL;
	66	}
	67
	68	return pid;
	69	}
	70
4adc8274	71	uint64_t get_context_ppid(const struct bt_ctf_event *event)
1dec520a	72	{
2e0a1190	73	const struct bt_definition *scope;
1dec520a JD	74	uint64_t ppid;
	75
	76	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	77	ppid = bt_ctf_get_int64(bt_ctf_get_field(event,
	78	scope, "_ppid"));
	79	if (bt_ctf_field_get_error()) {
	80	fprintf(stderr, "Missing ppid context info\n");
	81	return -1ULL;
	82	}
	83
	84	return ppid;
	85	}
	86
1402044a JD	87	uint64_t get_context_vtid(const struct bt_ctf_event *event)
	88	{
	89	const struct definition *scope;
	90	uint64_t vtid;
	91
	92	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	93	vtid = bt_ctf_get_int64(bt_ctf_get_field(event,
	94	scope, "_vtid"));
	95	if (bt_ctf_field_get_error()) {
	96	return -1ULL;
	97	}
	98
	99	return vtid;
	100	}
	101
	102	uint64_t get_context_vpid(const struct bt_ctf_event *event)
	103	{
	104	const struct definition *scope;
	105	uint64_t vpid;
	106
	107	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	108	vpid = bt_ctf_get_int64(bt_ctf_get_field(event,
	109	scope, "_vpid"));
	110	if (bt_ctf_field_get_error()) {
	111	return -1ULL;
	112	}
	113
	114	return vpid;
	115	}
	116
	117	uint64_t get_context_vppid(const struct bt_ctf_event *event)
	118	{
	119	const struct definition *scope;
	120	uint64_t vppid;
	121
	122	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	123	vppid = bt_ctf_get_int64(bt_ctf_get_field(event,
	124	scope, "_vppid"));
	125	if (bt_ctf_field_get_error()) {
	126	return -1ULL;
	127	}
	128
	129	return vppid;
	130	}
	131
4adc8274	132	char get_context_comm(const struct bt_ctf_event event)
1dec520a	133	{
2e0a1190	134	const struct bt_definition *scope;
1dec520a JD	135	char *comm;
	136
	137	scope = bt_ctf_get_top_level_scope(event, BT_STREAM_EVENT_CONTEXT);
	138	comm = bt_ctf_get_char_array(bt_ctf_get_field(event,
	139	scope, "_procname"));
	140	if (bt_ctf_field_get_error()) {
	141	fprintf(stderr, "Missing comm context info\n");
	142	return NULL;
	143	}
	144
	145	return comm;
	146	}
	147
59288610 MB	148	/*
	149	* To get the parent process, put the pid in the tid field
	150	* because the parent process gets pid = tid
59288610	151	*/
1fc22eb4 JD	152	struct processtop find_process_tid(struct lttngtop ctx, int tid, char *comm)
1fc22eb4 JD	153	{
1fc22eb4 JD	154	struct processtop *tmp;
1fc22eb4 JD	155
30b646c4 JD	156	tmp = g_hash_table_lookup(ctx->process_hash_table,
	157	(gconstpointer) (unsigned long) tid);
	158
	159	return tmp;
1fc22eb4 JD	160	}
	161
	162	struct processtop* add_proc(struct lttngtop ctx, int tid, char comm,
	163	unsigned long timestamp)
	164	{
	165	struct processtop *newproc;
	166
	167	/* if the PID already exists, we just rename the process */
	168	/* FIXME : need to integrate with clone/fork/exit to be accurate */
	169	newproc = find_process_tid(ctx, tid, comm);
	170	if (!newproc) {
559c9f86	171	newproc = g_new0(struct processtop, 1);
1fc22eb4 JD	172	newproc->tid = tid;
	173	newproc->birth = timestamp;
	174	newproc->process_files_table = g_ptr_array_new();
ceb3a221	175	newproc->files_history = NULL;
b093de8a MB	176	newproc->totalfileread = 0;
	177	newproc->totalfilewrite = 0;
	178	newproc->fileread = 0;
	179	newproc->filewrite = 0;
	180	newproc->syscall_info = NULL;
59288610	181	newproc->threadparent = NULL;
1fc22eb4	182	newproc->threads = g_ptr_array_new();
85db4618	183	newproc->perf = g_hash_table_new(g_str_hash, g_str_equal);
1fc22eb4	184	g_ptr_array_add(ctx->process_table, newproc);
30b646c4 JD	185	g_hash_table_insert(ctx->process_hash_table,
30b646c4 JD	186	(gpointer) (unsigned long) tid, newproc);
e05a35a6 JD	187
	188	ctx->nbnewthreads++;
	189	ctx->nbthreads++;
1fc22eb4 JD	190	}
	191	newproc->comm = strdup(comm);
	192
	193	return newproc;
	194	}
	195
	196	struct processtop* update_proc(struct processtop* proc, int pid, int tid,
1402044a	197	int ppid, int vpid, int vtid, int vppid, char *comm)
1fc22eb4 JD	198	{
	199	if (proc) {
	200	proc->pid = pid;
	201	proc->tid = tid;
	202	proc->ppid = ppid;
1402044a JD	203	proc->vpid = vpid;
	204	proc->vtid = vtid;
	205	proc->vppid = vppid;
1fc22eb4 JD	206	if (strcmp(proc->comm, comm) != 0) {
	207	free(proc->comm);
	208	proc->comm = strdup(comm);
	209	}
	210	}
	211	return proc;
	212	}
	213
	214	/*
	215	* This function just sets the time of death of a process.
	216	* When we rotate the cputime we remove it from the process list.
	217	*/
	218	void death_proc(struct lttngtop ctx, int tid, char comm,
	219	unsigned long timestamp)
	220	{
	221	struct processtop *tmp;
	222	tmp = find_process_tid(ctx, tid, comm);
30b646c4 JD	223
	224	g_hash_table_remove(ctx->process_hash_table,
	225	(gpointer) (unsigned long) tid);
e05a35a6	226	if (tmp && strcmp(tmp->comm, comm) == 0) {
1fc22eb4	227	tmp->death = timestamp;
e05a35a6 JD	228	ctx->nbdeadthreads++;
	229	ctx->nbthreads--;
	230	}
1fc22eb4 JD	231	}
	232
	233	struct processtop* get_proc(struct lttngtop ctx, int tid, char comm,
	234	unsigned long timestamp)
	235	{
	236	struct processtop *tmp;
	237	tmp = find_process_tid(ctx, tid, comm);
	238	if (tmp && strcmp(tmp->comm, comm) == 0)
	239	return tmp;
	240	return add_proc(ctx, tid, comm, timestamp);
	241	}
	242
59288610 MB	243	struct processtop get_proc_pid(struct lttngtop ctx, int tid, int pid,
	244	unsigned long timestamp)
	245	{
	246	struct processtop *tmp;
	247	tmp = find_process_tid(ctx, tid, NULL);
	248	if (tmp && tmp->pid == pid)
	249	return tmp;
	250	return add_proc(ctx, tid, "Unknown", timestamp);
	251	}
	252
1fc22eb4 JD	253	void add_thread(struct processtop parent, struct processtop thread)
	254	{
	255	gint i;
	256	struct processtop *tmp;
	257
	258	for (i = 0; i < parent->threads->len; i++) {
	259	tmp = g_ptr_array_index(parent->threads, i);
	260	if (tmp == thread)
	261	return;
	262	}
	263	g_ptr_array_add(parent->threads, thread);
	264	}
	265
	266	struct cputime* add_cpu(int cpu)
	267	{
	268	struct cputime *newcpu;
	269
559c9f86	270	newcpu = g_new0(struct cputime, 1);
1fc22eb4 JD	271	newcpu->id = cpu;
1fc22eb4 JD	272	newcpu->current_task = NULL;
85db4618	273	newcpu->perf = g_hash_table_new(g_str_hash, g_str_equal);
1fc22eb4 JD	274
	275	g_ptr_array_add(lttngtop.cpu_table, newcpu);
	276
	277	return newcpu;
	278	}
	279	struct cputime* get_cpu(int cpu)
	280	{
	281	gint i;
	282	struct cputime *tmp;
	283
	284	for (i = 0; i < lttngtop.cpu_table->len; i++) {
	285	tmp = g_ptr_array_index(lttngtop.cpu_table, i);
	286	if (tmp->id == cpu)
	287	return tmp;
	288	}
	289
	290	return add_cpu(cpu);
	291	}
	292
	293	/*
	294	* At the end of a sampling period, we need to display the cpu time for each
	295	* process and to reset it to zero for the next period
	296	*/
	297	void rotate_cputime(unsigned long end)
	298	{
	299	gint i;
	300	struct cputime *tmp;
	301	unsigned long elapsed;
	302
	303	for (i = 0; i < lttngtop.cpu_table->len; i++) {
	304	tmp = g_ptr_array_index(lttngtop.cpu_table, i);
	305	elapsed = end - tmp->task_start;
	306	if (tmp->current_task) {
	307	tmp->current_task->totalcpunsec += elapsed;
	308	tmp->current_task->threadstotalcpunsec += elapsed;
	309	if (tmp->current_task->pid != tmp->current_task->tid &&
	310	tmp->current_task->threadparent) {
	311	tmp->current_task->threadparent->threadstotalcpunsec += elapsed;
	312	}
	313	}
	314	tmp->task_start = end;
	315	}
	316	}
	317
	318	void reset_perf_counter(gpointer key, gpointer value, gpointer user_data)
	319	{
	320	((struct perfcounter*) value)->count = 0;
	321	}
	322
	323	void copy_perf_counter(gpointer key, gpointer value, gpointer new_table)
	324	{
	325	struct perfcounter *newperf;
59288610	326
559c9f86	327	newperf = g_new0(struct perfcounter, 1);
1fc22eb4 JD	328	newperf->count = ((struct perfcounter *) value)->count;
	329	newperf->visible = ((struct perfcounter *) value)->visible;
	330	newperf->sort = ((struct perfcounter *) value)->sort;
85db4618	331	g_hash_table_insert((GHashTable *) new_table, strdup(key), newperf);
1fc22eb4 JD	332	}
1fc22eb4 JD	333
30b646c4 JD	334	void copy_process_table(gpointer key, gpointer value, gpointer new_table)
	335	{
	336	g_hash_table_insert((GHashTable *) new_table, key, value);
	337	}
	338
1fc22eb4 JD	339	void rotate_perfcounter() {
	340	int i;
	341	struct processtop *tmp;
	342	for (i = 0; i < lttngtop.process_table->len; i++) {
	343	tmp = g_ptr_array_index(lttngtop.process_table, i);
	344	g_hash_table_foreach(tmp->perf, reset_perf_counter, NULL);
	345	}
	346	}
	347
	348	void cleanup_processtop()
	349	{
b093de8a	350	gint i, j;
1fc22eb4	351	struct processtop *tmp;
b093de8a	352	struct files tmpf; / a temporary file */
1fc22eb4 JD	353
	354	for (i = 0; i < lttngtop.process_table->len; i++) {
	355	tmp = g_ptr_array_index(lttngtop.process_table, i);
	356	tmp->totalcpunsec = 0;
	357	tmp->threadstotalcpunsec = 0;
b093de8a MB	358	tmp->fileread = 0;
	359	tmp->filewrite = 0;
	360
	361	for (j = 0; j < tmp->process_files_table->len; j++) {
	362	tmpf = g_ptr_array_index(tmp->process_files_table, j);
	363	if (tmpf != NULL) {
	364	tmpf->read = 0;
	365	tmpf->write = 0;
ceb3a221 MB	366
	367	if (tmpf->flag == __NR_close)
	368	g_ptr_array_index(
	369	tmp->process_files_table, j
	370	) = NULL;
b093de8a MB	371	}
b093de8a MB	372	}
1fc22eb4 JD	373	}
	374	}
	375
e05a35a6 JD	376	void reset_global_counters()
	377	{
	378	lttngtop.nbnewproc = 0;
	379	lttngtop.nbdeadproc = 0;
	380	lttngtop.nbnewthreads = 0;
	381	lttngtop.nbdeadthreads = 0;
	382	lttngtop.nbnewfiles = 0;
	383	lttngtop.nbclosedfiles = 0;
	384	}
	385
	386	void copy_global_counters(struct lttngtop *dst)
	387	{
	388	dst->nbproc = lttngtop.nbproc;
	389	dst->nbnewproc = lttngtop.nbnewproc;
	390	dst->nbdeadproc = lttngtop.nbdeadproc;
	391	dst->nbthreads = lttngtop.nbthreads;
	392	dst->nbnewthreads = lttngtop.nbnewthreads;
	393	dst->nbdeadthreads = lttngtop.nbdeadthreads;
	394	dst->nbfiles = lttngtop.nbfiles;
	395	dst->nbnewfiles = lttngtop.nbnewfiles;
	396	dst->nbclosedfiles = lttngtop.nbclosedfiles;
	397	reset_global_counters();
	398	}
	399
1fc22eb4 JD	400	struct lttngtop* get_copy_lttngtop(unsigned long start, unsigned long end)
	401	{
	402	gint i, j;
	403	unsigned long time;
	404	struct lttngtop *dst;
	405	struct processtop tmp, tmp2, *new;
	406	struct cputime tmpcpu, newcpu;
	407	struct files tmpfile, newfile;
	408
559c9f86	409	dst = g_new0(struct lttngtop, 1);
1fc22eb4 JD	410	dst->start = start;
1fc22eb4 JD	411	dst->end = end;
e05a35a6	412	copy_global_counters(dst);
1fc22eb4 JD	413	dst->process_table = g_ptr_array_new();
	414	dst->files_table = g_ptr_array_new();
	415	dst->cpu_table = g_ptr_array_new();
30b646c4 JD	416	dst->process_hash_table = g_hash_table_new(g_direct_hash, g_direct_equal);
	417	g_hash_table_foreach(lttngtop.process_hash_table, copy_process_table,
	418	dst->process_hash_table);
1fc22eb4 JD	419
	420	rotate_cputime(end);
	421
1fc22eb4 JD	422	for (i = 0; i < lttngtop.process_table->len; i++) {
1fc22eb4 JD	423	tmp = g_ptr_array_index(lttngtop.process_table, i);
559c9f86	424	new = g_new0(struct processtop, 1);
1fc22eb4 JD	425
	426	memcpy(new, tmp, sizeof(struct processtop));
	427	new->threads = g_ptr_array_new();
	428	new->comm = strdup(tmp->comm);
	429	new->process_files_table = g_ptr_array_new();
ceb3a221	430	new->files_history = tmp->files_history;
85db4618	431	new->perf = g_hash_table_new(g_str_hash, g_str_equal);
1fc22eb4 JD	432	g_hash_table_foreach(tmp->perf, copy_perf_counter, new->perf);
1fc22eb4 JD	433
1fc22eb4	434	/* compute the stream speed */
85db4618 JD	435	if (end - start != 0) {
85db4618 JD	436	time = (end - start) / NSEC_PER_SEC;
b093de8a MB	437	new->fileread = new->fileread/(time);
b093de8a MB	438	new->filewrite = new->filewrite/(time);
1fc22eb4 JD	439	}
	440
	441	for (j = 0; j < tmp->process_files_table->len; j++) {
	442	tmpfile = g_ptr_array_index(tmp->process_files_table, j);
1fc22eb4	443
b093de8a MB	444	newfile = malloc(sizeof(struct files));
	445
	446	if (tmpfile != NULL) {
	447	memcpy(newfile, tmpfile, sizeof(struct files));
	448	newfile->name = strdup(tmpfile->name);
	449	newfile->ref = new;
	450	g_ptr_array_add(new->process_files_table,
	451	newfile);
	452	g_ptr_array_add(dst->files_table, newfile);
	453	} else {
	454	g_ptr_array_add(new->process_files_table, NULL);
	455	g_ptr_array_add(dst->files_table, NULL);
	456	}
1fc22eb4 JD	457	/*
	458	* if the process died during the last period, we remove all
	459	* files associated with if after the copy
	460	*/
	461	if (tmp->death > 0 && tmp->death < end) {
e05a35a6	462	/* FIXME : close the files before */
1fc22eb4	463	g_ptr_array_remove(tmp->process_files_table, tmpfile);
559c9f86	464	g_free(tmpfile);
1fc22eb4 JD	465	}
	466	}
	467	g_ptr_array_add(dst->process_table, new);
	468
	469	/*
	470	* if the process died during the last period, we remove it from
	471	* the current process list after the copy
	472	*/
	473	if (tmp->death > 0 && tmp->death < end) {
	474	g_ptr_array_remove(lttngtop.process_table, tmp);
85db4618	475	/* FIXME : TRUE does not mean clears the object in it */
1fc22eb4 JD	476	g_ptr_array_free(tmp->threads, TRUE);
	477	free(tmp->comm);
	478	g_ptr_array_free(tmp->process_files_table, TRUE);
85db4618	479	/* FIXME : clear elements */
1fc22eb4	480	g_hash_table_destroy(tmp->perf);
559c9f86	481	g_free(tmp);
1fc22eb4 JD	482	}
	483	}
	484	rotate_perfcounter();
	485
	486	for (i = 0; i < lttngtop.cpu_table->len; i++) {
	487	tmpcpu = g_ptr_array_index(lttngtop.cpu_table, i);
559c9f86	488	newcpu = g_new0(struct cputime, 1);
1fc22eb4	489	memcpy(newcpu, tmpcpu, sizeof(struct cputime));
85db4618	490	newcpu->perf = g_hash_table_new(g_str_hash, g_str_equal);
1fc22eb4 JD	491	g_hash_table_foreach(tmpcpu->perf, copy_perf_counter, newcpu->perf);
	492	/*
	493	* note : we don't care about the current process pointer in the copy
	494	* so the reference is invalid after the memcpy
	495	*/
	496	g_ptr_array_add(dst->cpu_table, newcpu);
	497	}
85db4618	498	/* FIXME : better algo */
1fc22eb4 JD	499	/* create the threads index if required */
	500	for (i = 0; i < dst->process_table->len; i++) {
	501	tmp = g_ptr_array_index(dst->process_table, i);
	502	if (tmp->pid == tmp->tid) {
	503	for (j = 0; j < dst->process_table->len; j++) {
	504	tmp2 = g_ptr_array_index(dst->process_table, j);
	505	if (tmp2->pid == tmp->pid) {
	506	tmp2->threadparent = tmp;
	507	g_ptr_array_add(tmp->threads, tmp2);
	508	}
	509	}
	510	}
	511	}
	512
	513	// update_global_stats(dst);
	514	cleanup_processtop();
	515
	516	return dst;
	517	}
	518
928f18a6 MB	519
	520	enum bt_cb_ret handle_statedump_process_state(struct bt_ctf_event *call_data,
	521	void *private_data)
	522	{
2e0a1190	523	const struct bt_definition *scope;
928f18a6 MB	524	struct processtop *proc;
928f18a6 MB	525	unsigned long timestamp;
11d218ce	526	int64_t pid, tid, ppid, vtid, vpid, vppid;
928f18a6 MB	527	char *procname;
928f18a6 MB	528
c78f2cdc	529	timestamp = bt_ctf_get_timestamp(call_data);
928f18a6 MB	530	if (timestamp == -1ULL)
	531	goto error;
	532
	533	scope = bt_ctf_get_top_level_scope(call_data,
	534	BT_EVENT_FIELDS);
	535	pid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	536	scope, "_pid"));
	537	if (bt_ctf_field_get_error()) {
	538	fprintf(stderr, "Missing pid context info\n");
	539	goto error;
	540	}
11d218ce JD	541	ppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	542	scope, "_ppid"));
	543	if (bt_ctf_field_get_error()) {
b7194a4e	544	fprintf(stderr, "Missing ppid context info\n");
11d218ce JD	545	goto error;
11d218ce JD	546	}
928f18a6 MB	547	tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	548	scope, "_tid"));
	549	if (bt_ctf_field_get_error()) {
	550	fprintf(stderr, "Missing tid context info\n");
	551	goto error;
	552	}
11d218ce JD	553	vtid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	554	scope, "_vtid"));
	555	if (bt_ctf_field_get_error()) {
	556	fprintf(stderr, "Missing vtid context info\n");
	557	goto error;
	558	}
	559	vpid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	560	scope, "_vpid"));
	561	if (bt_ctf_field_get_error()) {
b7194a4e	562	fprintf(stderr, "Missing vpid context info\n");
11d218ce JD	563	goto error;
	564	}
	565	vppid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
	566	scope, "_vppid"));
	567	if (bt_ctf_field_get_error()) {
b7194a4e	568	fprintf(stderr, "Missing vppid context info\n");
11d218ce JD	569	goto error;
	570	}
	571
928f18a6 MB	572	scope = bt_ctf_get_top_level_scope(call_data,
	573	BT_EVENT_FIELDS);
	574	procname = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
	575	scope, "_name"));
	576	if (bt_ctf_field_get_error()) {
	577	fprintf(stderr, "Missing process name context info\n");
	578	goto error;
	579	}
	580
	581	proc = find_process_tid(&lttngtop, tid, procname);
	582	if (proc == NULL)
	583	proc = add_proc(&lttngtop, tid, procname, timestamp);
11d218ce	584	update_proc(proc, pid, tid, ppid, vpid, vtid, vppid, procname);
928f18a6 MB	585
	586	free(proc->comm);
	587	proc->comm = strdup(procname);
	588	proc->pid = pid;
	589
928f18a6 MB	590	return BT_CB_OK;
	591
	592	error:
	593	return BT_CB_ERROR_STOP;
	594	}
b520ab45 JD	595
	596	struct tm format_timestamp(uint64_t timestamp)
	597	{
	598	struct tm tm;
	599	uint64_t ts_sec = 0, ts_nsec;
	600	time_t time_s;
	601
	602	ts_nsec = timestamp;
	603	ts_sec += ts_nsec / NSEC_PER_SEC;
	604	ts_nsec = ts_nsec % NSEC_PER_SEC;
	605
	606	time_s = (time_t) ts_sec;
	607
	608	localtime_r(&time_s, &tm);
	609
	610	return tm;
	611	}