2 * Copyright (C) 2021 Simon Marchi <simon.marchi@efficios.com>
4 * SPDX-License-Identifier: GPL-2.0-only
8 #include "../command.h"
9 #include "common/argpar/argpar.h"
10 #include "common/argpar-utils/argpar-utils.h"
11 #include "common/mi-lttng.h"
12 #include <lttng/lttng.h>
15 #ifdef LTTNG_EMBED_HELP
16 static const char help_msg
[] =
17 #include <lttng-remove-trigger.1.h>
28 struct argpar_opt_descr remove_trigger_options
[] = {
29 { OPT_HELP
, 'h', "help", false },
30 { OPT_LIST_OPTIONS
, '\0', "list-options", false },
31 { OPT_OWNER_UID
, '\0', "owner-uid", true },
32 ARGPAR_OPT_DESCR_SENTINEL
,
36 bool assign_string(char **dest
, const char *src
, const char *opt_name
)
41 ERR("Duplicate option '%s' given.", opt_name
);
47 ERR("Failed to allocate '%s' string.", opt_name
);
61 int cmd_remove_trigger(int argc
, const char **argv
)
63 enum lttng_error_code ret_code
;
65 struct argpar_iter
*argpar_iter
= NULL
;
66 const struct argpar_item
*argpar_item
= NULL
;
67 const char *name
= NULL
;
69 struct lttng_triggers
*triggers
= NULL
;
70 unsigned int triggers_count
;
71 enum lttng_trigger_status trigger_status
;
72 const struct lttng_trigger
*trigger_to_remove
= NULL
;
73 char *owner_uid
= NULL
;
75 struct mi_writer
*mi_writer
= NULL
;
78 mi_writer
= mi_lttng_writer_create(
79 fileno(stdout
), lttng_opt_mi
);
85 /* Open command element. */
86 ret
= mi_lttng_writer_command_open(mi_writer
,
87 mi_lttng_element_command_remove_trigger
);
93 /* Open output element. */
94 ret
= mi_lttng_writer_open_element(
95 mi_writer
, mi_lttng_element_command_output
);
105 argpar_iter
= argpar_iter_create(argc
, argv
, remove_trigger_options
);
107 ERR("Failed to allocate an argpar iter.");
112 enum parse_next_item_status status
;
114 status
= parse_next_item(argpar_iter
, &argpar_item
, argv
,
116 if (status
== PARSE_NEXT_ITEM_STATUS_ERROR
) {
118 } else if (status
== PARSE_NEXT_ITEM_STATUS_END
) {
122 assert(status
== PARSE_NEXT_ITEM_STATUS_OK
);
124 if (argpar_item_type(argpar_item
) == ARGPAR_ITEM_TYPE_OPT
) {
125 const struct argpar_opt_descr
*descr
=
126 argpar_item_opt_descr(argpar_item
);
127 const char *arg
= argpar_item_opt_arg(argpar_item
);
134 case OPT_LIST_OPTIONS
:
135 list_cmd_options_argpar(stdout
,
136 remove_trigger_options
);
141 if (!assign_string(&owner_uid
, arg
,
151 const char *arg
= argpar_item_non_opt_arg(argpar_item
);
154 ERR("Unexpected argument '%s'", arg
);
163 ERR("Missing `name` argument.");
171 uid
= strtol(owner_uid
, &end
, 10);
172 if (end
== owner_uid
|| *end
!= '\0' || errno
!= 0) {
173 ERR("Failed to parse `%s` as an integer.", owner_uid
);
179 ret
= lttng_list_triggers(&triggers
);
180 if (ret
!= LTTNG_OK
) {
181 ERR("Failed to get the list of triggers.");
185 trigger_status
= lttng_triggers_get_count(triggers
, &triggers_count
);
186 LTTNG_ASSERT(trigger_status
== LTTNG_TRIGGER_STATUS_OK
);
188 for (i
= 0; i
< triggers_count
; i
++) {
189 const struct lttng_trigger
*trigger
;
190 const char *trigger_name
;
193 trigger
= lttng_triggers_get_at_index(triggers
, i
);
194 trigger_status
= lttng_trigger_get_name(trigger
, &trigger_name
);
195 switch (trigger_status
) {
196 case LTTNG_TRIGGER_STATUS_OK
:
198 case LTTNG_TRIGGER_STATUS_UNSET
:
199 /* Don't compare against anonymous triggers. */
205 trigger_status
= lttng_trigger_get_owner_uid(
206 trigger
, &trigger_uid
);
207 LTTNG_ASSERT(trigger_status
== LTTNG_TRIGGER_STATUS_OK
);
209 if (trigger_uid
== uid
&& strcmp(trigger_name
, name
) == 0) {
210 trigger_to_remove
= trigger
;
215 if (!trigger_to_remove
) {
216 ERR("Couldn't find trigger with name `%s`.", name
);
220 ret
= lttng_unregister_trigger(trigger_to_remove
);
222 ERR("Failed to unregister trigger `%s`.", name
);
227 ret_code
= lttng_trigger_mi_serialize(
228 trigger_to_remove
, mi_writer
, NULL
);
229 if (ret_code
!= LTTNG_OK
) {
233 MSG("Removed trigger `%s`.", name
);
243 if (lttng_opt_mi
&& mi_writer
) {
244 /* Close output element. */
245 int mi_ret
= mi_lttng_writer_close_element(mi_writer
);
251 mi_ret
= mi_lttng_writer_write_element_bool(mi_writer
,
252 mi_lttng_element_command_success
, ret
? 0 : 1);
258 /* Command element close. */
259 mi_ret
= mi_lttng_writer_command_close(mi_writer
);
267 argpar_item_destroy(argpar_item
);
268 argpar_iter_destroy(argpar_iter
);
269 lttng_triggers_destroy(triggers
);
272 if (mi_writer
&& mi_lttng_writer_destroy(mi_writer
)) {
273 /* Preserve original error code. */
274 ret
= ret
? ret
: CMD_ERROR
;