diff options
author | Timo Teras <timo.teras@iki.fi> | 2010-03-19 00:45:35 +0300 |
---|---|---|
committer | Denys Vlasenko <vda.linux@googlemail.com> | 2010-03-19 00:45:35 +0300 |
commit | 892d4b6b3d7e9950ff3708ca957e1e0590dd5a2d (patch) | |
tree | 7b3064dca217fdec81a15e150e7beb0cdc84b9dd /util-linux/flock.c | |
parent | 8531d76a15890c2c535908ce888b2e2aed35b172 (diff) |
flock: new applet
function old new delta
flock_main - 253 +253
packed_usage 26466 26498 +32
applet_names 2170 2176 +6
applet_main 1280 1284 +4
applet_nameofs 640 642 +2
applet_install_loc 160 161 +1
------------------------------------------------------------------------------
(add/remove: 1/0 grow/shrink: 5/0 up/down: 298/0) Total: 298 bytes
Signed-off-by: Timo Teras <timo.teras@iki.fi>
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
Diffstat (limited to 'util-linux/flock.c')
-rw-r--r-- | util-linux/flock.c | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/util-linux/flock.c b/util-linux/flock.c new file mode 100644 index 000000000..78b1e4ba3 --- /dev/null +++ b/util-linux/flock.c @@ -0,0 +1,87 @@ +/* + * Copyright (C) 2010 Timo Teras <timo.teras@iki.fi> + * + * This is free software, licensed under the GNU General Public License v2. + */ +#include <sys/file.h> +#include "libbb.h" + +int flock_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int flock_main(int argc UNUSED_PARAM, char **argv) +{ + int mode, opt, fd; + enum { + OPT_s = (1 << 0), + OPT_x = (1 << 1), + OPT_n = (1 << 2), + OPT_u = (1 << 3), + OPT_c = (1 << 4), + }; + +#if ENABLE_LONG_OPTS + static const char getopt_longopts[] ALIGN1 = + "shared\0" No_argument "s" + "exclusive\0" No_argument "x" + "unlock\0" No_argument "u" + "nonblock\0" No_argument "n" + ; + applet_long_options = getopt_longopts; +#endif + opt_complementary = "-1"; + + opt = getopt32(argv, "+sxnu"); + argv += optind; + + if (argv[1]) { + fd = open(argv[0], O_RDONLY|O_NOCTTY|O_CREAT, 0666); + if (fd < 0 && errno == EISDIR) + fd = open(argv[0], O_RDONLY|O_NOCTTY); + if (fd < 0) + bb_perror_msg_and_die("can't open '%s'", argv[0]); + //TODO? close_on_exec_on(fd); + } else { + fd = xatoi_u(argv[0]); + } + argv++; + + /* If it is "flock FILE -c PROG", then -c isn't caught by getopt32: + * we use "+" in order to support "flock -opt FILE PROG -with-opts", + * we need to remove -c by hand. + * TODO: in upstream, -c 'PROG ARGS' means "run sh -c 'PROG ARGS'" + */ + if (argv[0] + && argv[0][0] == '-' + && ( (argv[0][1] == 'c' && !argv[0][2]) + || (ENABLE_LONG_OPTS && strcmp(argv[0] + 1, "-command") == 0) + ) + ) { + argv++; + } + + if (OPT_s == LOCK_SH && OPT_x == LOCK_EX && OPT_n == LOCK_NB && OPT_u == LOCK_UN) { + /* With suitably matched constants, mode setting is much simpler */ + mode = opt & (LOCK_SH + LOCK_EX + LOCK_NB + LOCK_UN); + if (!(mode & ~LOCK_NB)) + mode |= LOCK_EX; + } else { + if (opt & OPT_u) + mode = LOCK_UN; + else if (opt & OPT_s) + mode = LOCK_SH; + else + mode = LOCK_EX; + if (opt & OPT_n) + mode |= LOCK_NB; + } + + if (flock(fd, mode) != 0) { + if (errno == EWOULDBLOCK) + return EXIT_FAILURE; + bb_perror_nomsg_and_die(); + } + + if (argv[0]) + return spawn_and_wait(argv); + + return EXIT_SUCCESS; +} |