243 lines · 5.2 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0#include <stdbool.h>#include <assert.h>#include <errno.h>#include <stdlib.h>#include <string.h>#include "metricgroup.h"#include "debug.h"#include "expr.h"#include "expr-bison.h"#include "expr-flex.h"#include <linux/kernel.h>#include <linux/zalloc.h>#include <ctype.h> #ifdef PARSER_DEBUGextern int expr_debug;#endif static size_t key_hash(const void *key, void *ctx __maybe_unused){	const char *str = (const char *)key;	size_t hash = 0; 	while (*str != '\0') {		hash *= 31;		hash += *str;		str++;	}	return hash;} static bool key_equal(const void *key1, const void *key2,		    void *ctx __maybe_unused){	return !strcmp((const char *)key1, (const char *)key2);} /* Caller must make sure id is allocated */int expr__add_id(struct expr_parse_ctx *ctx, const char *id){	struct expr_id_data *data_ptr = NULL, *old_data = NULL;	char *old_key = NULL;	int ret; 	data_ptr = malloc(sizeof(*data_ptr));	if (!data_ptr)		return -ENOMEM; 	data_ptr->parent = ctx->parent; 	ret = hashmap__set(&ctx->ids, id, data_ptr,			   (const void **)&old_key, (void **)&old_data);	if (ret)		free(data_ptr);	free(old_key);	free(old_data);	return ret;} /* Caller must make sure id is allocated */int expr__add_id_val(struct expr_parse_ctx *ctx, const char *id, double val){	struct expr_id_data *data_ptr = NULL, *old_data = NULL;	char *old_key = NULL;	int ret; 	data_ptr = malloc(sizeof(*data_ptr));	if (!data_ptr)		return -ENOMEM;	data_ptr->val = val;	data_ptr->is_ref = false; 	ret = hashmap__set(&ctx->ids, id, data_ptr,			   (const void **)&old_key, (void **)&old_data);	if (ret)		free(data_ptr);	free(old_key);	free(old_data);	return ret;} int expr__add_ref(struct expr_parse_ctx *ctx, struct metric_ref *ref){	struct expr_id_data *data_ptr = NULL, *old_data = NULL;	char *old_key = NULL;	char *name, *p;	int ret; 	data_ptr = zalloc(sizeof(*data_ptr));	if (!data_ptr)		return -ENOMEM; 	name = strdup(ref->metric_name);	if (!name) {		free(data_ptr);		return -ENOMEM;	} 	/*	 * The jevents tool converts all metric expressions	 * to lowercase, including metric references, hence	 * we need to add lowercase name for metric, so it's	 * properly found.	 */	for (p = name; *p; p++)		*p = tolower(*p); 	/*	 * Intentionally passing just const char pointers,	 * originally from 'struct pmu_event' object.	 * We don't need to change them, so there's no	 * need to create our own copy.	 */	data_ptr->ref.metric_name = ref->metric_name;	data_ptr->ref.metric_expr = ref->metric_expr;	data_ptr->ref.counted = false;	data_ptr->is_ref = true; 	ret = hashmap__set(&ctx->ids, name, data_ptr,			   (const void **)&old_key, (void **)&old_data);	if (ret)		free(data_ptr); 	pr_debug2("adding ref metric %s: %s\n",		  ref->metric_name, ref->metric_expr); 	free(old_key);	free(old_data);	return ret;} int expr__get_id(struct expr_parse_ctx *ctx, const char *id,		 struct expr_id_data **data){	return hashmap__find(&ctx->ids, id, (void **)data) ? 0 : -1;} int expr__resolve_id(struct expr_parse_ctx *ctx, const char *id,		     struct expr_id_data **datap){	struct expr_id_data *data; 	if (expr__get_id(ctx, id, datap) || !*datap) {		pr_debug("%s not found\n", id);		return -1;	} 	data = *datap; 	pr_debug2("lookup: is_ref %d, counted %d, val %f: %s\n",		  data->is_ref, data->ref.counted, data->val, id); 	if (data->is_ref && !data->ref.counted) {		data->ref.counted = true;		pr_debug("processing metric: %s ENTRY\n", id);		if (expr__parse(&data->val, ctx, data->ref.metric_expr, 1)) {			pr_debug("%s failed to count\n", id);			return -1;		}		pr_debug("processing metric: %s EXIT: %f\n", id, data->val);	} 	return 0;} void expr__del_id(struct expr_parse_ctx *ctx, const char *id){	struct expr_id_data *old_val = NULL;	char *old_key = NULL; 	hashmap__delete(&ctx->ids, id,			(const void **)&old_key, (void **)&old_val);	free(old_key);	free(old_val);} void expr__ctx_init(struct expr_parse_ctx *ctx){	hashmap__init(&ctx->ids, key_hash, key_equal, NULL);} void expr__ctx_clear(struct expr_parse_ctx *ctx){	struct hashmap_entry *cur;	size_t bkt; 	hashmap__for_each_entry((&ctx->ids), cur, bkt) {		free((char *)cur->key);		free(cur->value);	}	hashmap__clear(&ctx->ids);} static int__expr__parse(double *val, struct expr_parse_ctx *ctx, const char *expr,	      int start, int runtime){	struct expr_scanner_ctx scanner_ctx = {		.start_token = start,		.runtime = runtime,	};	YY_BUFFER_STATE buffer;	void *scanner;	int ret; 	pr_debug2("parsing metric: %s\n", expr); 	ret = expr_lex_init_extra(&scanner_ctx, &scanner);	if (ret)		return ret; 	buffer = expr__scan_string(expr, scanner); #ifdef PARSER_DEBUG	expr_debug = 1;	expr_set_debug(1, scanner);#endif 	ret = expr_parse(val, ctx, scanner); 	expr__flush_buffer(buffer, scanner);	expr__delete_buffer(buffer, scanner);	expr_lex_destroy(scanner);	return ret;} int expr__parse(double *final_val, struct expr_parse_ctx *ctx,		const char *expr, int runtime){	return __expr__parse(final_val, ctx, expr, EXPR_PARSE, runtime) ? -1 : 0;} int expr__find_other(const char *expr, const char *one,		     struct expr_parse_ctx *ctx, int runtime){	int ret = __expr__parse(NULL, ctx, expr, EXPR_OTHER, runtime); 	if (one)		expr__del_id(ctx, one); 	return ret;}