+/* Convenience function for error checking in memory-tag commands. */
+
+static void
+show_addr_not_tagged (CORE_ADDR address)
+{
+ error (_("Address %s not in a region mapped with a memory tagging flag."),
+ paddress (target_gdbarch (), address));
+}
+
+/* Convenience function for error checking in memory-tag commands. */
+
+static void
+show_memory_tagging_unsupported (void)
+{
+ error (_("Memory tagging not supported or disabled by the current"
+ " architecture."));
+}
+
+/* Implement the "memory-tag" prefix command. */
+
+static void
+memory_tag_command (const char *arg, int from_tty)
+{
+ help_list (memory_tag_list, "memory-tag ", all_commands, gdb_stdout);
+}
+
+/* Helper for print-logical-tag and print-allocation-tag. */
+
+static void
+memory_tag_print_tag_command (const char *args, enum memtag_type tag_type)
+{
+ if (args == nullptr)
+ error_no_arg (_("address or pointer"));
+
+ /* Parse args into a value. If the value is a pointer or an address,
+ then fetch the logical or allocation tag. */
+ value_print_options print_opts;
+
+ struct value *val = process_print_command_args (args, &print_opts, true);
+
+ /* If the address is not in a region memory mapped with a memory tagging
+ flag, it is no use trying to access/manipulate its allocation tag.
+
+ It is OK to manipulate the logical tag though. */
+ if (tag_type == memtag_type::allocation
+ && !gdbarch_tagged_address_p (target_gdbarch (), val))
+ show_addr_not_tagged (value_as_address (val));
+
+ struct value *tag_value
+ = gdbarch_get_memtag (target_gdbarch (), val, tag_type);
+ std::string tag = gdbarch_memtag_to_string (target_gdbarch (), tag_value);
+
+ if (tag.empty ())
+ printf_filtered (_("%s tag unavailable.\n"),
+ tag_type
+ == memtag_type::logical? "Logical" : "Allocation");
+
+ struct value *v_tag = process_print_command_args (tag.c_str (),
+ &print_opts,
+ true);
+ print_opts.output_format = 'x';
+ print_value (v_tag, print_opts);
+}
+
+/* Implement the "memory-tag print-logical-tag" command. */
+
+static void
+memory_tag_print_logical_tag_command (const char *args, int from_tty)
+{
+ if (!target_supports_memory_tagging ())
+ show_memory_tagging_unsupported ();
+
+ memory_tag_print_tag_command (args, memtag_type::logical);
+}
+
+/* Implement the "memory-tag print-allocation-tag" command. */
+
+static void
+memory_tag_print_allocation_tag_command (const char *args, int from_tty)
+{
+ if (!target_supports_memory_tagging ())
+ show_memory_tagging_unsupported ();
+
+ memory_tag_print_tag_command (args, memtag_type::allocation);
+}
+
+/* Parse ARGS and extract ADDR and TAG.
+ ARGS should have format <expression> <tag bytes>. */
+
+static void
+parse_with_logical_tag_input (const char *args, struct value **val,
+ gdb::byte_vector &tags,
+ value_print_options *print_opts)
+{
+ /* Fetch the address. */
+ std::string address_string = extract_string_maybe_quoted (&args);
+
+ /* Parse the address into a value. */
+ *val = process_print_command_args (address_string.c_str (), print_opts,
+ true);
+
+ /* Fetch the tag bytes. */
+ std::string tag_string = extract_string_maybe_quoted (&args);
+
+ /* Validate the input. */
+ if (address_string.empty () || tag_string.empty ())
+ error (_("Missing arguments."));
+
+ if (tag_string.length () != 2)
+ error (_("Error parsing tags argument. The tag should be 2 digits."));
+
+ tags = hex2bin (tag_string.c_str ());
+}
+
+/* Implement the "memory-tag with-logical-tag" command. */
+
+static void
+memory_tag_with_logical_tag_command (const char *args, int from_tty)
+{
+ if (!target_supports_memory_tagging ())
+ show_memory_tagging_unsupported ();
+
+ if (args == nullptr)
+ error_no_arg (_("<address> <tag>"));
+
+ gdb::byte_vector tags;
+ struct value *val;
+ value_print_options print_opts;
+
+ /* Parse the input. */
+ parse_with_logical_tag_input (args, &val, tags, &print_opts);
+
+ /* Setting the logical tag is just a local operation that does not touch
+ any memory from the target. Given an input value, we modify the value
+ to include the appropriate tag.
+
+ For this reason we need to cast the argument value to a
+ (void *) pointer. This is so we have the right type for the gdbarch
+ hook to manipulate the value and insert the tag.
+
+ Otherwise, this would fail if, for example, GDB parsed the argument value
+ into an int-sized value and the pointer value has a type of greater
+ length. */
+
+ /* Cast to (void *). */
+ val = value_cast (builtin_type (target_gdbarch ())->builtin_data_ptr,
+ val);
+
+ /* Length doesn't matter for a logical tag. Pass 0. */
+ if (!gdbarch_set_memtags (target_gdbarch (), val, 0, tags,
+ memtag_type::logical))
+ printf_filtered (_("Could not update the logical tag data.\n"));
+ else
+ {
+ /* Always print it in hex format. */
+ print_opts.output_format = 'x';
+ print_value (val, print_opts);
+ }
+}
+
+/* Parse ARGS and extract ADDR, LENGTH and TAGS. */
+
+static void
+parse_set_allocation_tag_input (const char *args, struct value **val,
+ size_t *length, gdb::byte_vector &tags)
+{
+ /* Fetch the address. */
+ std::string address_string = extract_string_maybe_quoted (&args);
+
+ /* Parse the address into a value. */
+ value_print_options print_opts;
+ *val = process_print_command_args (address_string.c_str (), &print_opts,
+ true);
+
+ /* Fetch the length. */
+ std::string length_string = extract_string_maybe_quoted (&args);
+
+ /* Fetch the tag bytes. */
+ std::string tags_string = extract_string_maybe_quoted (&args);
+
+ /* Validate the input. */
+ if (address_string.empty () || length_string.empty () || tags_string.empty ())
+ error (_("Missing arguments."));
+
+ errno = 0;
+ const char *trailer = nullptr;
+ LONGEST parsed_length = strtoulst (length_string.c_str (), &trailer, 10);
+
+ if (errno != 0 || (trailer != nullptr && trailer[0] != '\0'))
+ error (_("Error parsing length argument."));
+
+ if (parsed_length <= 0)
+ error (_("Invalid zero or negative length."));
+
+ *length = parsed_length;
+
+ if (tags_string.length () % 2)
+ error (_("Error parsing tags argument. Tags should be 2 digits per byte."));
+
+ tags = hex2bin (tags_string.c_str ());
+
+ /* If the address is not in a region memory mapped with a memory tagging
+ flag, it is no use trying to access/manipulate its allocation tag. */
+ if (!gdbarch_tagged_address_p (target_gdbarch (), *val))
+ show_addr_not_tagged (value_as_address (*val));
+}
+
+/* Implement the "memory-tag set-allocation-tag" command.
+ ARGS should be in the format <address> <length> <tags>. */
+
+static void
+memory_tag_set_allocation_tag_command (const char *args, int from_tty)
+{
+ if (!target_supports_memory_tagging ())
+ show_memory_tagging_unsupported ();
+
+ if (args == nullptr)
+ error_no_arg (_("<starting address> <length> <tag bytes>"));
+
+ gdb::byte_vector tags;
+ size_t length = 0;
+ struct value *val;
+
+ /* Parse the input. */
+ parse_set_allocation_tag_input (args, &val, &length, tags);
+
+ if (!gdbarch_set_memtags (target_gdbarch (), val, length, tags,
+ memtag_type::allocation))
+ printf_filtered (_("Could not update the allocation tag(s).\n"));
+ else
+ printf_filtered (_("Allocation tag(s) updated successfully.\n"));
+}
+
+/* Implement the "memory-tag check" command. */
+
+static void
+memory_tag_check_command (const char *args, int from_tty)
+{
+ if (!target_supports_memory_tagging ())
+ show_memory_tagging_unsupported ();
+
+ if (args == nullptr)
+ error (_("Argument required (address or pointer)"));
+
+ /* Parse the expression into a value. If the value is an address or
+ pointer, then check its logical tag against the allocation tag. */
+ value_print_options print_opts;
+
+ struct value *val = process_print_command_args (args, &print_opts, true);
+
+ /* If the address is not in a region memory mapped with a memory tagging
+ flag, it is no use trying to access/manipulate its allocation tag. */
+ if (!gdbarch_tagged_address_p (target_gdbarch (), val))
+ show_addr_not_tagged (value_as_address (val));
+
+ CORE_ADDR addr = value_as_address (val);
+
+ /* Check if the tag is valid. */
+ if (!gdbarch_memtag_matches_p (target_gdbarch (), val))
+ {
+ struct value *tag
+ = gdbarch_get_memtag (target_gdbarch (), val, memtag_type::logical);
+ std::string ltag
+ = gdbarch_memtag_to_string (target_gdbarch (), tag);
+
+ tag = gdbarch_get_memtag (target_gdbarch (), val,
+ memtag_type::allocation);
+ std::string atag
+ = gdbarch_memtag_to_string (target_gdbarch (), tag);
+
+ printf_filtered (_("Logical tag (%s) does not match"
+ " the allocation tag (%s) for address %s.\n"),
+ ltag.c_str (), atag.c_str (),
+ paddress (target_gdbarch (), addr));
+ }
+ else
+ {
+ struct value *tag
+ = gdbarch_get_memtag (target_gdbarch (), val, memtag_type::logical);
+ std::string ltag
+ = gdbarch_memtag_to_string (target_gdbarch (), tag);
+
+ printf_filtered (_("Memory tags for address %s match (%s).\n"),
+ paddress (target_gdbarch (), addr), ltag.c_str ());
+ }
+}
+
+void _initialize_printcmd ();