1#include "builtin.h"
  2#include "perf.h"
  3
  4#include "util/parse-options.h"
  5#include "util/trace-event.h"
  6#include "util/tool.h"
  7#include "util/session.h"
  8#include "util/data.h"
  9
 10#define MEM_OPERATION_LOAD	"load"
 11#define MEM_OPERATION_STORE	"store"
 12
 13static const char	*mem_operation		= MEM_OPERATION_LOAD;
 14
 15struct perf_mem {
 16	struct perf_tool	tool;
 17	char const		*input_name;
 18	bool			hide_unresolved;
 19	bool			dump_raw;
 20	const char		*cpu_list;
 21	DECLARE_BITMAP(cpu_bitmap, MAX_NR_CPUS);
 22};
 23
 24static const char * const mem_usage[] = {
 25	"perf mem [<options>] {record <command> |report}",
 26	NULL
 27};
 28
 29static int __cmd_record(int argc, const char **argv)
 30{
 31	int rec_argc, i = 0, j;
 32	const char **rec_argv;
 33	char event[64];
 34	int ret;
 35
 36	rec_argc = argc + 4;
 37	rec_argv = calloc(rec_argc + 1, sizeof(char *));
 38	if (!rec_argv)
 39		return -1;
 40
 41	rec_argv[i++] = strdup("record");
 42	if (!strcmp(mem_operation, MEM_OPERATION_LOAD))
 43		rec_argv[i++] = strdup("-W");
 44	rec_argv[i++] = strdup("-d");
 45	rec_argv[i++] = strdup("-e");
 46
 47	if (strcmp(mem_operation, MEM_OPERATION_LOAD))
 48		sprintf(event, "cpu/mem-stores/pp");
 49	else
 50		sprintf(event, "cpu/mem-loads/pp");
 51
 52	rec_argv[i++] = strdup(event);
 53	for (j = 1; j < argc; j++, i++)
 54		rec_argv[i] = argv[j];
 55
 56	ret = cmd_record(i, rec_argv, NULL);
 57	free(rec_argv);
 58	return ret;
 59}
 60
 61static int
 62dump_raw_samples(struct perf_tool *tool,
 63		 union perf_event *event,
 64		 struct perf_sample *sample,
 65		 struct machine *machine)
 66{
 67	struct perf_mem *mem = container_of(tool, struct perf_mem, tool);
 68	struct addr_location al;
 69	const char *fmt;
 70
 71	if (perf_event__preprocess_sample(event, machine, &al, sample) < 0) {
 72		fprintf(stderr, "problem processing %d event, skipping it.\n",
 73				event->header.type);
 74		return -1;
 75	}
 76
 77	if (al.filtered || (mem->hide_unresolved && al.sym == NULL))
 78		return 0;
 79
 80	if (al.map != NULL)
 81		al.map->dso->hit = 1;
 82
 83	if (symbol_conf.field_sep) {
 84		fmt = "%d%s%d%s0x%"PRIx64"%s0x%"PRIx64"%s%"PRIu64
 85		      "%s0x%"PRIx64"%s%s:%s\n";
 86	} else {
 87		fmt = "%5d%s%5d%s0x%016"PRIx64"%s0x016%"PRIx64
 88		      "%s%5"PRIu64"%s0x%06"PRIx64"%s%s:%s\n";
 89		symbol_conf.field_sep = " ";
 90	}
 91
 92	printf(fmt,
 93		sample->pid,
 94		symbol_conf.field_sep,
 95		sample->tid,
 96		symbol_conf.field_sep,
 97		sample->ip,
 98		symbol_conf.field_sep,
 99		sample->addr,
100		symbol_conf.field_sep,
101		sample->weight,
102		symbol_conf.field_sep,
103		sample->data_src,
104		symbol_conf.field_sep,
105		al.map ? (al.map->dso ? al.map->dso->long_name : "???") : "???",
106		al.sym ? al.sym->name : "???");
107
108	return 0;
109}
110
111static int process_sample_event(struct perf_tool *tool,
112				union perf_event *event,
113				struct perf_sample *sample,
114				struct perf_evsel *evsel __maybe_unused,
115				struct machine *machine)
116{
117	return dump_raw_samples(tool, event, sample, machine);
118}
119
120static int report_raw_events(struct perf_mem *mem)
121{
122	struct perf_data_file file = {
123		.path = input_name,
124		.mode = PERF_DATA_MODE_READ,
125	};
126	int err = -EINVAL;
127	int ret;
128	struct perf_session *session = perf_session__new(&file, false,
129							 &mem->tool);
130
131	if (session == NULL)
132		return -ENOMEM;
133
134	if (mem->cpu_list) {
135		ret = perf_session__cpu_bitmap(session, mem->cpu_list,
136					       mem->cpu_bitmap);
137		if (ret)
138			goto out_delete;
139	}
140
141	if (symbol__init() < 0)
142		return -1;
143
144	printf("# PID, TID, IP, ADDR, LOCAL WEIGHT, DSRC, SYMBOL\n");
145
146	err = perf_session__process_events(session, &mem->tool);
147	if (err)
148		return err;
149
150	return 0;
151
152out_delete:
153	perf_session__delete(session);
154	return err;
155}
156
157static int report_events(int argc, const char **argv, struct perf_mem *mem)
158{
159	const char **rep_argv;
160	int ret, i = 0, j, rep_argc;
161
162	if (mem->dump_raw)
163		return report_raw_events(mem);
164
165	rep_argc = argc + 3;
166	rep_argv = calloc(rep_argc + 1, sizeof(char *));
167	if (!rep_argv)
168		return -1;
169
170	rep_argv[i++] = strdup("report");
171	rep_argv[i++] = strdup("--mem-mode");
172	rep_argv[i++] = strdup("-n"); /* display number of samples */
173
174	/*
175	 * there is no weight (cost) associated with stores, so don't print
176	 * the column
177	 */
178	if (strcmp(mem_operation, MEM_OPERATION_LOAD))
179		rep_argv[i++] = strdup("--sort=mem,sym,dso,symbol_daddr,"
180				       "dso_daddr,tlb,locked");
181
182	for (j = 1; j < argc; j++, i++)
183		rep_argv[i] = argv[j];
184
185	ret = cmd_report(i, rep_argv, NULL);
186	free(rep_argv);
187	return ret;
188}
189
190int cmd_mem(int argc, const char **argv, const char *prefix __maybe_unused)
191{
192	struct stat st;
193	struct perf_mem mem = {
194		.tool = {
195			.sample		= process_sample_event,
196			.mmap		= perf_event__process_mmap,
197			.mmap2		= perf_event__process_mmap2,
198			.comm		= perf_event__process_comm,
199			.lost		= perf_event__process_lost,
200			.fork		= perf_event__process_fork,
201			.build_id	= perf_event__process_build_id,
202			.ordered_samples = true,
203		},
204		.input_name		 = "perf.data",
205	};
206	const struct option mem_options[] = {
207	OPT_STRING('t', "type", &mem_operation,
208		   "type", "memory operations(load/store)"),
209	OPT_BOOLEAN('D', "dump-raw-samples", &mem.dump_raw,
210		    "dump raw samples in ASCII"),
211	OPT_BOOLEAN('U', "hide-unresolved", &mem.hide_unresolved,
212		    "Only display entries resolved to a symbol"),
213	OPT_STRING('i', "input", &input_name, "file",
214		   "input file name"),
215	OPT_STRING('C', "cpu", &mem.cpu_list, "cpu",
216		   "list of cpus to profile"),
217	OPT_STRING('x', "field-separator", &symbol_conf.field_sep,
218		   "separator",
219		   "separator for columns, no spaces will be added"
220		   " between columns '.' is reserved."),
221	OPT_END()
222	};
223
224	argc = parse_options(argc, argv, mem_options, mem_usage,
225			     PARSE_OPT_STOP_AT_NON_OPTION);
226
227	if (!argc || !(strncmp(argv[0], "rec", 3) || mem_operation))
228		usage_with_options(mem_usage, mem_options);
229
230	if (!mem.input_name || !strlen(mem.input_name)) {
231		if (!fstat(STDIN_FILENO, &st) && S_ISFIFO(st.st_mode))
232			mem.input_name = "-";
233		else
234			mem.input_name = "perf.data";
235	}
236
237	if (!strncmp(argv[0], "rec", 3))
238		return __cmd_record(argc, argv);
239	else if (!strncmp(argv[0], "rep", 3))
240		return report_events(argc, argv, &mem);
241	else
242		usage_with_options(mem_usage, mem_options);
243
244	return 0;
245}