2 * Copyright (C) 2021 Simon Marchi <simon.marchi@efficios.com>
4 * SPDX-License-Identifier: GPL-2.0-only
8 #include "../command.hpp"
9 #include "common/argpar/argpar.h"
10 #include "common/argpar-utils/argpar-utils.hpp"
11 #include "common/mi-lttng.hpp"
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
, 1, argv
,
116 if (status
== PARSE_NEXT_ITEM_STATUS_ERROR
||
117 status
== PARSE_NEXT_ITEM_STATUS_ERROR_MEMORY
) {
119 } else if (status
== PARSE_NEXT_ITEM_STATUS_END
) {
123 assert(status
== PARSE_NEXT_ITEM_STATUS_OK
);
125 if (argpar_item_type(argpar_item
) == ARGPAR_ITEM_TYPE_OPT
) {
126 const struct argpar_opt_descr
*descr
=
127 argpar_item_opt_descr(argpar_item
);
128 const char *arg
= argpar_item_opt_arg(argpar_item
);
135 case OPT_LIST_OPTIONS
:
136 list_cmd_options_argpar(stdout
,
137 remove_trigger_options
);
142 if (!assign_string(&owner_uid
, arg
,
152 const char *arg
= argpar_item_non_opt_arg(argpar_item
);
155 ERR("Unexpected argument '%s'", arg
);
164 ERR("Missing `name` argument.");
172 uid
= strtol(owner_uid
, &end
, 10);
173 if (end
== owner_uid
|| *end
!= '\0' || errno
!= 0) {
174 ERR("Failed to parse `%s` as an integer.", owner_uid
);
180 ret
= lttng_list_triggers(&triggers
);
181 if (ret
!= LTTNG_OK
) {
182 ERR("Failed to get the list of triggers.");
186 trigger_status
= lttng_triggers_get_count(triggers
, &triggers_count
);
187 LTTNG_ASSERT(trigger_status
== LTTNG_TRIGGER_STATUS_OK
);
189 for (i
= 0; i
< triggers_count
; i
++) {
190 const struct lttng_trigger
*trigger
;
191 const char *trigger_name
;
194 trigger
= lttng_triggers_get_at_index(triggers
, i
);
195 trigger_status
= lttng_trigger_get_name(trigger
, &trigger_name
);
196 switch (trigger_status
) {
197 case LTTNG_TRIGGER_STATUS_OK
:
199 case LTTNG_TRIGGER_STATUS_UNSET
:
200 /* Don't compare against anonymous triggers. */
206 trigger_status
= lttng_trigger_get_owner_uid(
207 trigger
, &trigger_uid
);
208 LTTNG_ASSERT(trigger_status
== LTTNG_TRIGGER_STATUS_OK
);
210 if (trigger_uid
== uid
&& strcmp(trigger_name
, name
) == 0) {
211 trigger_to_remove
= trigger
;
216 if (!trigger_to_remove
) {
217 ERR("Couldn't find trigger with name `%s`.", name
);
221 ret
= lttng_unregister_trigger(trigger_to_remove
);
223 ERR("Failed to unregister trigger `%s`.", name
);
228 ret_code
= lttng_trigger_mi_serialize(
229 trigger_to_remove
, mi_writer
, NULL
);
230 if (ret_code
!= LTTNG_OK
) {
234 MSG("Removed trigger `%s`.", name
);
244 if (lttng_opt_mi
&& mi_writer
) {
245 /* Close output element. */
246 int mi_ret
= mi_lttng_writer_close_element(mi_writer
);
252 mi_ret
= mi_lttng_writer_write_element_bool(mi_writer
,
253 mi_lttng_element_command_success
, ret
? 0 : 1);
259 /* Command element close. */
260 mi_ret
= mi_lttng_writer_command_close(mi_writer
);
268 argpar_item_destroy(argpar_item
);
269 argpar_iter_destroy(argpar_iter
);
270 lttng_triggers_destroy(triggers
);
273 if (mi_writer
&& mi_lttng_writer_destroy(mi_writer
)) {
274 /* Preserve original error code. */
275 ret
= ret
? ret
: CMD_ERROR
;