On 2022/12/06 17:40, Tao Liu wrote:
 Maple tree is a new data structure for crash, so cmd_tree support
is
 needed for users to dump and view the content of maple tree. This patch
 achieves this by porting mt_dump() and its related functions from kernel,
 and adapting them with tree cmd.
 
 We introduced a new -v arg specifically for dumping the complete
 content of maple tree:
 
      crash> tree -t maple 0xffff9034c006aec0 -v
 
      maple_tree(ffff9034c006aec0) flags 309, height 2 root 0xffff9034de70041e
 
      0-18446744073709551615: node 0xffff9034de700400 depth 0 type 3 parent
0xffff9034c006aec1 contents:...
        0-140112331583487: node 0xffff9034c01e8800 depth 1 type 1 parent
0xffff9034de700406 contents:...
          0-94643156942847: (nil)
          94643156942848-94643158024191: 0xffff9035131754c0
          94643158024192-94643160117247: (nil)
          ...
 
 The old tree args can work as well:
 
      crash> tree -t maple -r mm_struct.mm_mt 0xffff9034c006aec0 -p
      ffff9035131754c0
        index: 0  position: root/0/1
      ffff9035131751c8
        index: 1  position: root/0/3
      ffff9035131757b8
        index: 2  position: root/0/4
      ...
 
      crash> tree -t maple 0xffff9034c006aec0 -p -x -s vm_area_struct.vm_start,vm_end
      ffff9035131754c0
        index: 0  position: root/0/1
        vm_start = 0x5613d3c00000,
        vm_end = 0x5613d3d08000,
      ffff9035131751c8
        index: 1  position: root/0/3
        vm_start = 0x5613d3f07000,
        vm_end = 0x5613d3f0b000,
      ffff9035131757b8
        index: 2  position: root/0/4
        vm_start = 0x5613d3f0b000,
        vm_end = 0x5613d3f14000,
      ....
 
 Signed-off-by: Tao Liu <ltao(a)redhat.com>
 ---
   defs.h       |  1 +
   maple_tree.c | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++
   tools.c      | 64 ++++++++++++++++++++++++++++++++++++++--------------
   3 files changed, 110 insertions(+), 17 deletions(-)
 
 diff --git a/defs.h b/defs.h
 index 792b007..cc40cba 100644
 --- a/defs.h
 +++ b/defs.h
 @@ -2704,6 +2704,7 @@ struct tree_data {
   #define TREE_PARSE_MEMBER         (VERBOSE << 7)
   #define TREE_READ_MEMBER          (VERBOSE << 8)
   #define TREE_LINEAR_ORDER         (VERBOSE << 9)
 +#define TREE_STRUCT_VERBOSE       (VERBOSE << 10)
   
   #define ALIAS_RUNTIME  (1)
   #define ALIAS_RCLOCAL  (2)
 diff --git a/maple_tree.c b/maple_tree.c
 index e27369b..756dc0a 100644
 --- a/maple_tree.c
 +++ b/maple_tree.c
 @@ -186,6 +186,10 @@ static void do_mt_range64(void *maple_tree_mt, void *entry,
   
   	maple_range_64_node = tmp_node + OFFSET(maple_node_mr64);
   
 +	if (td && td->flags & TREE_STRUCT_VERBOSE) {
 +		dump_mt_range64(maple_range_64_node);
 +	}
 +
   	for (i = 0; i < mt_slots[maple_range_64]; i++) {
   		last = max;
   
 @@ -246,6 +250,10 @@ static void do_mt_arange64(void *maple_tree_mt, void *entry,
   
   	maple_arange_64_node = tmp_node + OFFSET(maple_node_ma64);
   
 +	if (td && td->flags & TREE_STRUCT_VERBOSE) {
 +		dump_mt_arange64(maple_arange_64_node);
 +	}
 +
   	for (i = 0; i < mt_slots[maple_arange_64]; i++) {
   		last = max;
   
 @@ -294,6 +302,50 @@ static void do_mt_entry(void *entry, unsigned long min, unsigned
long max,
   
   	if (!td)
   		return;
 +
 +	if (!td->count && td->structname_args) {
 +		/*
 +		 * Retrieve all members' info only once (count == 0)
 +		 * After last iteration all memory will be freed up
 +		 */
 +		e = (struct req_entry **)GETBUF(sizeof(*e) * td->structname_args);
 +		for (i = 0; i < td->structname_args; i++)
 +			e[i] = fill_member_offsets(td->structname[i]);
 +	}
 +
 +	td->count++;
 +
 +	if (td->flags & TREE_STRUCT_VERBOSE) {
 +		dump_mt_entry(entry, min, max, depth);
 +	} else if (td->flags & VERBOSE && entry)
 +		fprintf(fp, "%lx\n", (ulong)entry);
 +	if (td->flags & TREE_POSITION_DISPLAY && entry)
 +		fprintf(fp, "  index: %ld  position: %s/%u\n",
 +			(*global_index)++, path, index);
 +
 +	if (td->structname) {
 +		if (td->flags & TREE_STRUCT_RADIX_10)
 +			print_radix = 10;
 +		else if (td->flags & TREE_STRUCT_RADIX_16)
 +			print_radix = 16;
 +		else
 +			print_radix = 0;
 +
 +		for (i = 0; i < td->structname_args; i++) {
 +			switch (count_chars(td->structname[i], '.')) {
 +			case 0:
 +				dump_struct(td->structname[i],
 +					    (ulong)entry, print_radix);
 +				break;
 +			default:
 +				if (td->flags & TREE_PARSE_MEMBER)
 +					dump_struct_members_for_tree(td, i, (ulong)entry);
 +				else if (td->flags & TREE_READ_MEMBER)
 +					dump_struct_members_fast(e[i], print_radix, (ulong)entry);
 +				break;
 +			}
 +		}
 +	}
   }
   
   static void do_mt_node(void *maple_tree_mt, void *entry,
 @@ -313,6 +365,10 @@ static void do_mt_node(void *maple_tree_mt, void *entry,
   	readmem((ulong)maple_node, KVADDR, tmp_node, SIZE(maple_node_struct),
   		"mt_dump_node read maple_node", FAULT_ON_ERROR);
   
 +	if (td && td->flags & TREE_STRUCT_VERBOSE) {
 +		dump_mt_node(maple_node, tmp_node, type, min, max, depth);
 +	}
 +
   	switch (type) {
   	case maple_dense:
   		for (i = 0; i < mt_slots[maple_dense]; i++) {
 @@ -359,6 +415,12 @@ static int do_maple_tree_traverse(ulong ptr, int is_root,
   			"mt_dump read maple_tree", FAULT_ON_ERROR);
   		entry = *(void **)(tmp_tree + OFFSET(maple_tree_ma_root));
   
 +		if (td && td->flags & TREE_STRUCT_VERBOSE) {
 +			fprintf(fp, "maple_tree(%lx) flags %X, height %u root %p\n\n",
 +				ptr, *(unsigned int *)(tmp_tree + OFFSET(maple_tree_ma_flags)),
 +				mt_height(tmp_tree), entry);
 +		}
 +
   		if (!xa_is_node(entry))
   			do_mt_entry(entry, 0, 0, 0, 0, path, &global_index, ops);
   		else if (entry) {
 diff --git a/tools.c b/tools.c
 index 39306c1..4e72c60 100644
 --- a/tools.c
 +++ b/tools.c
 @@ -30,7 +30,7 @@ static void dealloc_hq_entry(struct hq_entry *);
   static void show_options(void);
   static void dump_struct_members(struct list_data *, int, ulong);
   static void rbtree_iteration(ulong, struct tree_data *, char *);
 -static void dump_struct_members_for_tree(struct tree_data *, int, ulong);
 +void dump_struct_members_for_tree(struct tree_data *, int, ulong);
   
   struct req_entry {
   	char *arg, *name, **member;
 @@ -40,8 +40,8 @@ struct req_entry {
   };
   
   static void print_value(struct req_entry *, unsigned int, ulong, unsigned int);
 -static struct req_entry *fill_member_offsets(char *);
 -static void dump_struct_members_fast(struct req_entry *, int, ulong);
 +struct req_entry *fill_member_offsets(char *);
 +void dump_struct_members_fast(struct req_entry *, int, ulong);
   
   FILE *
   set_error(char *target)
 @@ -3666,7 +3666,7 @@ dump_struct_members_fast(struct req_entry *e, int radix, ulong p)
   	}
   }
   
 -static struct req_entry *
 +struct req_entry *
   fill_member_offsets(char *arg)
   {
   	int j;
 @@ -4307,6 +4307,7 @@ dump_struct_members(struct list_data *ld, int idx, ulong next)
   #define RADIXTREE_REQUEST (0x1)
   #define RBTREE_REQUEST    (0x2)
   #define XARRAY_REQUEST    (0x4)
 +#define MAPLE_REQUEST     (0x8)
   
   void
   cmd_tree()
 @@ -4317,6 +4318,7 @@ cmd_tree()
   	struct datatype_member struct_member, *sm;
   	struct syment *sp;
   	ulong value;
 +	char *type_name = NULL;
   
   	type_flag = 0;
   	root_offset = 0;
 @@ -4324,25 +4326,33 @@ cmd_tree()
   	td = &tree_data;
   	BZERO(td, sizeof(struct tree_data));
   
 -	while ((c = getopt(argcnt, args, "xdt:r:o:s:S:plN")) != EOF) {
 +	while ((c = getopt(argcnt, args, "xdt:r:o:s:S:plNv")) != EOF) {
   		switch (c)
   		{
   		case 't':
 -			if (type_flag & (RADIXTREE_REQUEST|RBTREE_REQUEST|XARRAY_REQUEST)) {
 +			if (type_flag & (RADIXTREE_REQUEST|RBTREE_REQUEST|XARRAY_REQUEST|MAPLE_REQUEST))
{
   				error(INFO, "multiple tree types may not be entered\n");
   				cmd_usage(pc->curcmd, SYNOPSIS);
   			}
   
   			if (STRNEQ(optarg, "ra"))
 -				if (MEMBER_EXISTS("radix_tree_root", "xa_head"))
 +				if (MEMBER_EXISTS("radix_tree_root", "xa_head")) {
   					type_flag = XARRAY_REQUEST;
 -				else
 +					type_name = "Xarrays";
 +				} else {
   					type_flag = RADIXTREE_REQUEST;
 -			else if (STRNEQ(optarg, "rb"))
 +					type_name = "radix trees";
 +				}
 +			else if (STRNEQ(optarg, "rb")) {
   				type_flag = RBTREE_REQUEST;
 -			else if (STRNEQ(optarg, "x"))
 +				type_name = "rbtrees";
 +			} else if (STRNEQ(optarg, "x")) {
   				type_flag = XARRAY_REQUEST;
 -			else {
 +				type_name = "Xarrays";
 +			} else if (STRNEQ(optarg, "m")) {
 +				type_flag = MAPLE_REQUEST;
 +				type_name = "maple trees";
 +			} else {
   				error(INFO, "invalid tree type: %s\n", optarg);
   				cmd_usage(pc->curcmd, SYNOPSIS);
   			}
 @@ -4417,6 +4427,9 @@ cmd_tree()
   					"-d and -x are mutually exclusive\n");
   			td->flags |= TREE_STRUCT_RADIX_10;
   			break;
 +		case 'v':
 +			td->flags |= TREE_STRUCT_VERBOSE;
 +			break;
   		default:
   			argerrs++;
   			break;
 @@ -4426,13 +4439,14 @@ cmd_tree()
   	if (argerrs)
   		cmd_usage(pc->curcmd, SYNOPSIS);
   
 -	if ((type_flag & (XARRAY_REQUEST|RADIXTREE_REQUEST)) && (td->flags &
TREE_LINEAR_ORDER))
 -		error(FATAL, "-l option is not applicable to %s\n",
 -			type_flag & RADIXTREE_REQUEST ? "radix trees" : "Xarrays");
 +	if ((type_flag & (XARRAY_REQUEST|RADIXTREE_REQUEST|MAPLE_REQUEST)) &&
(td->flags & TREE_LINEAR_ORDER))
 +		error(FATAL, "-l option is not applicable to %s\n", type_name);
   
 -	if ((type_flag & (XARRAY_REQUEST|RADIXTREE_REQUEST)) && (td->flags &
TREE_NODE_OFFSET_ENTERED))
 -		error(FATAL, "-o option is not applicable to %s\n",
 -			type_flag & RADIXTREE_REQUEST ? "radix trees" : "Xarrays");
 +	if ((type_flag & (XARRAY_REQUEST|RADIXTREE_REQUEST|MAPLE_REQUEST)) &&
(td->flags & TREE_NODE_OFFSET_ENTERED))
 +		error(FATAL, "-o option is not applicable to %s\n", type_name);
 +
 +	if ((type_flag & (RBTREE_REQUEST|XARRAY_REQUEST|RADIXTREE_REQUEST)) &&
(td->flags & TREE_STRUCT_VERBOSE))
 +		error(FATAL, "-v option is not applicable to %s\n", type_name);
 
Just a nit, a bit too long, please wrap after &&.
Otherwise, the v3 patch set looks good and works as expected.
Thanks,
Kazu
>   
>   	if ((td->flags & TREE_ROOT_OFFSET_ENTERED) &&
>   	    (td->flags & TREE_NODE_POINTER))
> @@ -4506,12 +4520,26 @@ next_arg:
>   		if (td->flags & TREE_STRUCT_RADIX_16)
>   			fprintf(fp, "%sTREE_STRUCT_RADIX_16",
>   				others++ ? "|" : "");
> +		if (td->flags & TREE_PARSE_MEMBER)
> +			fprintf(fp, "%sTREE_PARSE_MEMBER",
> +				others++ ? "|" : "");
> +		if (td->flags & TREE_READ_MEMBER)
> +			fprintf(fp, "%sTREE_READ_MEMBER",
> +				others++ ? "|" : "");
> +		if (td->flags & TREE_LINEAR_ORDER)
> +			fprintf(fp, "%sTREE_LINEAR_ORDER",
> +				others++ ? "|" : "");
> +		if (td->flags & TREE_STRUCT_VERBOSE)
> +			fprintf(fp, "%sTREE_STRUCT_VERBOSE",
> +				others++ ? "|" : "");
>   		fprintf(fp, ")\n");
>   		fprintf(fp, "              type: ");
>   			if (type_flag & RADIXTREE_REQUEST)
>   				fprintf(fp, "radix\n");
>   			else if (type_flag & XARRAY_REQUEST)
>   				fprintf(fp, "xarray\n");
> +			else if (type_flag & MAPLE_REQUEST)
> +				fprintf(fp, "maple\n");
>   			else
>   				fprintf(fp, "red-black%s",
>   					type_flag & RBTREE_REQUEST ?
> @@ -4532,6 +4560,8 @@ next_arg:
>   		do_rdtree(td);
>   	else if (type_flag & XARRAY_REQUEST)
>   		do_xatree(td);
> +	else if (type_flag & MAPLE_REQUEST)
> +		do_mptree(td);
>   	else
>   		do_rbtree(td);
>   	hq_close();