49e93c71c12c627871d5de0f0596f05deb75e4d9
[dhcpcd-ui] / src / dhcpcd-decode / dhcpcd-decode.c
1 /*
2  * dhcpcd-decode
3  * Copyright 2014 Roy Marples <roy@marples.name>
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26
27 #include <err.h>
28 #include <errno.h>
29 #include <libgen.h>
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <unistd.h>
34
35 #include "dhcpcd.h"
36
37 static ssize_t
38 dhcpcd_decode_shell(char *dst, size_t dlen, const char *src)
39 {
40         char *tmp;
41         ssize_t l;
42
43         tmp = malloc(dlen);
44         if (tmp == NULL)
45                 return -1;
46         if ((l = dhcpcd_decode(tmp, dlen, src)) != -1)
47                 l = dhcpcd_encode(dst, dlen, tmp, (size_t)l);
48
49         free(tmp);
50         return l;
51 }
52
53 static void
54 process(char *src, ssize_t (*decode)(char *, size_t, const char *))
55 {
56         char *buf;
57         size_t buflen;
58         ssize_t dl;
59
60         buflen = strlen(src) + 1;
61         buf = malloc(buflen);
62         if (buf == NULL)
63                 err(EXIT_FAILURE, "malloc");
64         if ((dl = decode(buf, buflen, src)) == -1) {
65                 free(buf);
66                 err(EXIT_FAILURE, "decode");
67         }
68         if (fwrite(buf, 1, (size_t)dl, stdout) != (size_t)dl) {
69                 free(buf);
70                 err(EXIT_FAILURE, "fwrite");
71         }
72         free(buf);
73         fputc('\n', stdout);
74 }
75
76 static void
77 usage(char *progname)
78 {
79
80         fprintf(stderr, "usage: %s [-sx] [data ...]\n", basename(progname));
81 }
82
83 int
84 main(int argc, char **argv)
85 {
86         int opt;
87         ssize_t (*decode)(char *, size_t, const char *);
88
89         decode = dhcpcd_decode;
90         while ((opt = getopt(argc, argv, "sx")) != -1) {
91                 switch (opt) {
92                 case 's':
93                         decode = dhcpcd_decode_shell;
94                         break;
95                 case 'x':
96                         decode = dhcpcd_decode_hex;
97                         break;
98                 case '?':
99                         usage(argv[0]);
100                         return EXIT_FAILURE;
101                 }
102         }
103
104         if (optind >= argc && isatty(fileno(stdin))) {
105                 usage(argv[0]);
106                 return EXIT_FAILURE;
107         }
108
109         for (; optind < argc; optind++)
110                 process(argv[optind], decode);
111
112         if (!isatty(fileno(stdin))) {
113                 char *arg;
114                 size_t len;
115                 ssize_t argl;
116
117                 arg = NULL;
118                 len = 0;
119                 while ((argl = getline(&arg, &len, stdin)) != -1) {
120                         if (arg[argl - 1] == '\n')
121                                 arg[argl - 1] = '\0';
122                         process(arg, decode);
123                 }
124                 free(arg);
125         }
126
127         return EXIT_SUCCESS;
128 }