From b7a82c980fe7e0c1f9029b55be97422428d65d5a Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Wed, 11 Jan 2017 09:11:51 +0100 Subject: Clean up code MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Mattias Andrée --- src/vu-read-head.c | 29 +++++++---------------------- 1 file changed, 7 insertions(+), 22 deletions(-) (limited to 'src/vu-read-head.c') diff --git a/src/vu-read-head.c b/src/vu-read-head.c index 2a22a77..b13ddb7 100644 --- a/src/vu-read-head.c +++ b/src/vu-read-head.c @@ -1,34 +1,22 @@ /* See LICENSE file for copyright and license details. */ -#include "arg.h" #include "stream.h" #include "util.h" #include #include -static void -usage(void) -{ - eprintf("usage: %s\n", argv0); -} +USAGE("") int main(int argc, char *argv[]) { char buf[2 + 3 * sizeof(size_t) + sizeof(((struct stream *)0)->pixfmt)]; char magic[] = {'\0', 'u', 'i', 'v', 'f'}; - char b; - char *p; + char b, *p; size_t i, ptr; ssize_t r; - ARGBEGIN { - default: - usage(); - } ARGEND; - - if (!argc) - usage(); + ENOFLAGS(!argc); for (ptr = 0; ptr < sizeof(buf);) { r = read(STDIN_FILENO, buf + ptr, 1); @@ -47,11 +35,11 @@ main(int argc, char *argv[]) r = read(STDIN_FILENO, &b, 1); if (r < 0) eprintf("read :"); - if (!r || b != magic[i]) + if (r == 0 || b != magic[i]) goto bad_format; } - for (i = 0; i < 2; i++) { + for (i = 0; i < 3; i++) { if (!isdigit(*p)) goto bad_format; while (isdigit(*p)) p++; @@ -66,13 +54,10 @@ main(int argc, char *argv[]) if (p[-1] == ' ' || p[0] != '\n') goto bad_format; - for (i = 0; i < ptr; i += (size_t)r) { - r = write(STDOUT_FILENO, buf + i, ptr - i); - if (r < 0) - eprintf("write :"); - } + ewriteall(STDOUT_FILENO, buf, ptr, ""); return 0; bad_format: eprintf(": file format not supported\n"); + return 0; } -- cgit v1.2.3-70-g09d2