diff options
| author | Felix Fietkau <nbd@nbd.name> | 2016-08-05 13:08:53 +0200 |
|---|---|---|
| committer | Felix Fietkau <nbd@nbd.name> | 2016-09-22 17:04:33 +0200 |
| commit | 297552aa40da2bb0a0b9fbfd65c5a02b224a592a (patch) | |
| tree | b8fd1a11f45f4a7214a1c2a577f4bc607cf264fa /getrandom.c | |
| parent | 34c597b8f478a4937fa041bf9da988990314ea85 (diff) | |
| download | ubox-297552aa40da2bb0a0b9fbfd65c5a02b224a592a.tar.gz | |
getrandom: fix indentation, add missing return statement
Signed-off-by: Felix Fietkau <nbd@nbd.name>
Diffstat (limited to 'getrandom.c')
| -rw-r--r-- | getrandom.c | 50 |
1 files changed, 26 insertions, 24 deletions
diff --git a/getrandom.c b/getrandom.c index 9671202..73a37b0 100644 --- a/getrandom.c +++ b/getrandom.c @@ -21,38 +21,40 @@ #include <unistd.h> #define ERROR_EXIT(fmt, ...) do { \ - fprintf(stderr, fmt, ## __VA_ARGS__); \ - return EXIT_FAILURE; \ - } while (0) + fprintf(stderr, fmt, ## __VA_ARGS__); \ + return EXIT_FAILURE; \ + } while (0) static int usage(char *name) { - fprintf(stderr, "Usage: %s <nb>\n", name); - fprintf(stderr, " => return <nb> bytes from getrandom()\n"); - return EXIT_FAILURE; + fprintf(stderr, "Usage: %s <nb>\n", name); + fprintf(stderr, " => return <nb> bytes from getrandom()\n"); + return EXIT_FAILURE; } int main(int argc, char *argv[]) { - if (argc != 2) - return usage(argv[0]); + if (argc != 2) + return usage(argv[0]); - if (isatty(STDOUT_FILENO)) - ERROR_EXIT("Not outputting random to a tty\n"); + if (isatty(STDOUT_FILENO)) + ERROR_EXIT("Not outputting random to a tty\n"); - int nbtot = atoi(argv[1]); - if (nbtot < 1) - ERROR_EXIT("Invalid <nb> param (must be > 0)\n"); + int nbtot = atoi(argv[1]); + if (nbtot < 1) + ERROR_EXIT("Invalid <nb> param (must be > 0)\n"); - char buf[256]; - int len = sizeof(buf); - while (nbtot > 0) { - if (nbtot <= sizeof(buf)) - len = nbtot; - if (syscall(SYS_getrandom, buf, len, 0) == -1) - ERROR_EXIT("getrandom() failed: %s\n", strerror(errno)); - if (write(STDOUT_FILENO, buf, len) != len) - ERROR_EXIT("write() failed: %s\n", strerror(errno)); - nbtot -= sizeof(buf); - } + char buf[256]; + int len = sizeof(buf); + while (nbtot > 0) { + if (nbtot <= sizeof(buf)) + len = nbtot; + if (syscall(SYS_getrandom, buf, len, 0) == -1) + ERROR_EXIT("getrandom() failed: %s\n", strerror(errno)); + if (write(STDOUT_FILENO, buf, len) != len) + ERROR_EXIT("write() failed: %s\n", strerror(errno)); + nbtot -= sizeof(buf); + } + + return 0; } |