[PATCH 2/9] setpriv: prepare option parsing logic for additional opts
Patrick Steinhardt
ps at pks.im
Thu Jun 29 17:34:29 UTC 2017
The current option parsing logic of setpriv only supports the case where
we want to execute a sub-program and have at most one argument. Refactor
handling of options to solve these shortcomings to make it easy to
support `setpriv --dump`, which does not accept any additional
arguments, as well as the case where additional options are passed to
setpriv. This is done by handling `argc` ourselves, throwing an error
when no program is specified, as well as introducing an enum for the
different option bitmasks.
---
util-linux/setpriv.c | 15 ++++++++++++---
1 file changed, 12 insertions(+), 3 deletions(-)
diff --git a/util-linux/setpriv.c b/util-linux/setpriv.c
index 686ad45d5..24e577131 100644
--- a/util-linux/setpriv.c
+++ b/util-linux/setpriv.c
@@ -51,6 +51,10 @@
#define PR_SET_NO_NEW_PRIVS 38
#endif
+enum {
+ OPT_NNP = (1 << 0),
+};
+
int setpriv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int setpriv_main(int argc UNUSED_PARAM, char **argv)
{
@@ -60,15 +64,20 @@ int setpriv_main(int argc UNUSED_PARAM, char **argv)
;
int opts;
- opt_complementary = "-1";
+ opt_complementary = "";
applet_long_options = setpriv_longopts;
opts = getopt32(argv, "+");
- if (opts) {
+ argc -= optind;
+ argv += optind;
+
+ if (!argc)
+ bb_error_msg_and_die("no program specified");
+
+ if (opts & OPT_NNP) {
if (prctl(PR_SET_NO_NEW_PRIVS, 1, 0, 0, 0))
bb_simple_perror_msg_and_die("prctl: NO_NEW_PRIVS");
}
- argv += optind;
BB_EXECVP_or_die(argv);
}
--
2.13.2
More information about the busybox
mailing list