[mirror_ubuntu-artful-kernel.git] / tools / perf / util / comm.c

#include "comm.h"
#include "util.h"
#include <stdlib.h>
#include <stdio.h>

struct comm_str {
	char *str;
	struct rb_node rb_node;
	int ref;
};

/* Should perhaps be moved to struct machine */
static struct rb_root comm_str_root;

static void comm_str__get(struct comm_str *cs)
{
	cs->ref++;
}

static void comm_str__put(struct comm_str *cs)
{
	if (!--cs->ref) {
		rb_erase(&cs->rb_node, &comm_str_root);
		zfree(&cs->str);
		free(cs);
	}
}

static struct comm_str *comm_str__alloc(const char *str)
{
	struct comm_str *cs;

	cs = zalloc(sizeof(*cs));
	if (!cs)
		return NULL;

	cs->str = strdup(str);
	if (!cs->str) {
		free(cs);
		return NULL;
	}

	return cs;
}

static struct comm_str *comm_str__findnew(const char *str, struct rb_root *root)
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct comm_str *iter, *new;
	int cmp;

	while (*p != NULL) {
		parent = *p;
		iter = rb_entry(parent, struct comm_str, rb_node);

		cmp = strcmp(str, iter->str);
		if (!cmp)
			return iter;

		if (cmp < 0)
			p = &(*p)->rb_left;
		else
			p = &(*p)->rb_right;
	}

	new = comm_str__alloc(str);
	if (!new)
		return NULL;

	rb_link_node(&new->rb_node, parent, p);
	rb_insert_color(&new->rb_node, root);

	return new;
}

struct comm *comm__new(const char *str, u64 timestamp, bool exec)
{
	struct comm *comm = zalloc(sizeof(*comm));

	if (!comm)
		return NULL;

	comm->start = timestamp;
	comm->exec = exec;

	comm->comm_str = comm_str__findnew(str, &comm_str_root);
	if (!comm->comm_str) {
		free(comm);
		return NULL;
	}

	comm_str__get(comm->comm_str);

	return comm;
}

int comm__override(struct comm *comm, const char *str, u64 timestamp, bool exec)
{
	struct comm_str *new, *old = comm->comm_str;

	new = comm_str__findnew(str, &comm_str_root);
	if (!new)
		return -ENOMEM;

	comm_str__get(new);
	comm_str__put(old);
	comm->comm_str = new;
	comm->start = timestamp;
	if (exec)
		comm->exec = true;

	return 0;
}

void comm__free(struct comm *comm)
{
	comm_str__put(comm->comm_str);
	free(comm);
}

const char *comm__str(const struct comm *comm)
{
	return comm->comm_str->str;
}
Commit	Line	Data
1902efe7 FW	1	#include "comm.h"
	2	#include "util.h"
	3	#include <stdlib.h>
	4	#include <stdio.h>
	5
	6	struct comm_str {
	7	char *str;
	8	struct rb_node rb_node;
	9	int ref;
	10	};
	11
	12	/* Should perhaps be moved to struct machine */
	13	static struct rb_root comm_str_root;
	14
	15	static void comm_str__get(struct comm_str *cs)
	16	{
	17	cs->ref++;
	18	}
	19
	20	static void comm_str__put(struct comm_str *cs)
	21	{
	22	if (!--cs->ref) {
	23	rb_erase(&cs->rb_node, &comm_str_root);
74cf249d	24	zfree(&cs->str);
1902efe7 FW	25	free(cs);
	26	}
	27	}
	28
	29	static struct comm_str comm_str__alloc(const char str)
	30	{
	31	struct comm_str *cs;
	32
	33	cs = zalloc(sizeof(*cs));
	34	if (!cs)
	35	return NULL;
	36
	37	cs->str = strdup(str);
	38	if (!cs->str) {
	39	free(cs);
	40	return NULL;
	41	}
	42
	43	return cs;
	44	}
	45
	46	static struct comm_str comm_str__findnew(const char str, struct rb_root *root)
	47	{
	48	struct rb_node **p = &root->rb_node;
	49	struct rb_node *parent = NULL;
	50	struct comm_str iter, new;
	51	int cmp;
	52
	53	while (*p != NULL) {
	54	parent = *p;
	55	iter = rb_entry(parent, struct comm_str, rb_node);
	56
	57	cmp = strcmp(str, iter->str);
	58	if (!cmp)
	59	return iter;
	60
	61	if (cmp < 0)
	62	p = &(*p)->rb_left;
	63	else
	64	p = &(*p)->rb_right;
	65	}
	66
	67	new = comm_str__alloc(str);
	68	if (!new)
	69	return NULL;
	70
	71	rb_link_node(&new->rb_node, parent, p);
	72	rb_insert_color(&new->rb_node, root);
	73
	74	return new;
	75	}
	76
65de51f9	77	struct comm comm__new(const char str, u64 timestamp, bool exec)
1902efe7 FW	78	{
	79	struct comm comm = zalloc(sizeof(comm));
	80
	81	if (!comm)
	82	return NULL;
	83
	84	comm->start = timestamp;
65de51f9	85	comm->exec = exec;
1902efe7 FW	86
	87	comm->comm_str = comm_str__findnew(str, &comm_str_root);
	88	if (!comm->comm_str) {
	89	free(comm);
	90	return NULL;
	91	}
	92
	93	comm_str__get(comm->comm_str);
	94
	95	return comm;
	96	}
	97
65de51f9	98	int comm__override(struct comm comm, const char str, u64 timestamp, bool exec)
4dfced35	99	{
3178f58b	100	struct comm_str new, old = comm->comm_str;
4dfced35	101
3178f58b FW	102	new = comm_str__findnew(str, &comm_str_root);
	103	if (!new)
	104	return -ENOMEM;
4dfced35	105
3178f58b	106	comm_str__get(new);
4dfced35	107	comm_str__put(old);
3178f58b FW	108	comm->comm_str = new;
3178f58b FW	109	comm->start = timestamp;
65de51f9 AH	110	if (exec)
65de51f9 AH	111	comm->exec = true;
3178f58b FW	112
3178f58b FW	113	return 0;
4dfced35 NK	114	}
4dfced35 NK	115
1902efe7 FW	116	void comm__free(struct comm *comm)
	117	{
	118	comm_str__put(comm->comm_str);
	119	free(comm);
	120	}
	121
	122	const char comm__str(const struct comm comm)
	123	{
	124	return comm->comm_str->str;
	125	}