1 | /* |
2 | * Copyright 2015 Red Hat, Inc. |
3 | * |
4 | * This library is free software; you can redistribute it and/or |
5 | * modify it under the terms of the GNU Lesser General Public |
6 | * License as published by the Free Software Foundation; either |
7 | * version 2.1 of the License, or (at your option) any later version. |
8 | * |
9 | * This library is distributed in the hope that it will be useful, |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
12 | * Lesser General Public License for more details. |
13 | * |
14 | * You should have received a copy of the GNU Lesser General Public |
15 | * License along with this library; if not, see <http://www.gnu.org/licenses/>. |
16 | * |
17 | * Author: Matthias Clasen <mclasen@redhat.com> |
18 | */ |
19 | |
20 | #include "config.h" |
21 | |
22 | #include <gio/gio.h> |
23 | #include <gi18n.h> |
24 | #include <errno.h> |
25 | |
26 | #ifdef G_OS_WIN32 |
27 | #include <io.h> |
28 | #endif |
29 | |
30 | #ifndef STDOUT_FILENO |
31 | #define STDOUT_FILENO 1 |
32 | #endif |
33 | |
34 | #ifdef HAVE_UNISTD_H |
35 | #include <unistd.h> |
36 | #endif |
37 | |
38 | #include "gio-tool.h" |
39 | |
40 | |
41 | static const GOptionEntry entries[] = { |
42 | { NULL } |
43 | }; |
44 | |
45 | /* 256k minus malloc overhead */ |
46 | #define STREAM_BUFFER_SIZE (1024*256 - 2*sizeof(gpointer)) |
47 | |
48 | static gboolean |
49 | cat (GFile *file) |
50 | { |
51 | GInputStream *in; |
52 | char *buffer; |
53 | char *p; |
54 | gssize res; |
55 | gboolean close_res; |
56 | GError *error; |
57 | gboolean success; |
58 | |
59 | error = NULL; |
60 | in = (GInputStream *) g_file_read (file, NULL, error: &error); |
61 | if (in == NULL) |
62 | { |
63 | print_file_error (file, message: error->message); |
64 | g_error_free (error); |
65 | return FALSE; |
66 | } |
67 | |
68 | buffer = g_malloc (STREAM_BUFFER_SIZE); |
69 | success = TRUE; |
70 | while (1) |
71 | { |
72 | res = g_input_stream_read (stream: in, buffer, STREAM_BUFFER_SIZE, NULL, error: &error); |
73 | if (res > 0) |
74 | { |
75 | gssize written; |
76 | |
77 | p = buffer; |
78 | while (res > 0) |
79 | { |
80 | int errsv; |
81 | |
82 | written = write (STDOUT_FILENO, buf: p, n: res); |
83 | errsv = errno; |
84 | |
85 | if (written == -1 && errsv != EINTR) |
86 | { |
87 | print_error (format: "%s" , _("Error writing to stdout" )); |
88 | success = FALSE; |
89 | goto out; |
90 | } |
91 | res -= written; |
92 | p += written; |
93 | } |
94 | } |
95 | else if (res < 0) |
96 | { |
97 | print_file_error (file, message: error->message); |
98 | g_error_free (error); |
99 | error = NULL; |
100 | success = FALSE; |
101 | break; |
102 | } |
103 | else if (res == 0) |
104 | break; |
105 | } |
106 | |
107 | out: |
108 | close_res = g_input_stream_close (stream: in, NULL, error: &error); |
109 | if (!close_res) |
110 | { |
111 | print_file_error (file, message: error->message); |
112 | g_error_free (error); |
113 | success = FALSE; |
114 | } |
115 | |
116 | g_free (mem: buffer); |
117 | |
118 | return success; |
119 | } |
120 | |
121 | int |
122 | handle_cat (int argc, char *argv[], gboolean do_help) |
123 | { |
124 | GOptionContext *context; |
125 | gchar *param; |
126 | GError *error = NULL; |
127 | int i; |
128 | gboolean res; |
129 | GFile *file; |
130 | |
131 | g_set_prgname (prgname: "gio cat" ); |
132 | /* Translators: commandline placeholder */ |
133 | param = g_strdup_printf (format: "%s…" , _("LOCATION" )); |
134 | context = g_option_context_new (parameter_string: param); |
135 | g_free (mem: param); |
136 | g_option_context_set_help_enabled (context, FALSE); |
137 | g_option_context_set_summary (context, |
138 | _("Concatenate files and print to standard output." )); |
139 | g_option_context_set_description (context, |
140 | _("gio cat works just like the traditional cat utility, but using GIO\n" |
141 | "locations instead of local files: for example, you can use something\n" |
142 | "like smb://server/resource/file.txt as location." )); |
143 | g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE); |
144 | |
145 | if (do_help) |
146 | { |
147 | show_help (context, NULL); |
148 | g_option_context_free (context); |
149 | return 0; |
150 | } |
151 | |
152 | if (!g_option_context_parse (context, argc: &argc, argv: &argv, error: &error)) |
153 | { |
154 | show_help (context, message: error->message); |
155 | g_error_free (error); |
156 | g_option_context_free (context); |
157 | return 1; |
158 | } |
159 | |
160 | if (argc < 2) |
161 | { |
162 | show_help (context, _("No locations given" )); |
163 | g_option_context_free (context); |
164 | return 1; |
165 | } |
166 | |
167 | g_option_context_free (context); |
168 | |
169 | res = TRUE; |
170 | for (i = 1; i < argc; i++) |
171 | { |
172 | file = g_file_new_for_commandline_arg (arg: argv[i]); |
173 | res &= cat (file); |
174 | g_object_unref (object: file); |
175 | } |
176 | |
177 | return res ? 0 : 2; |
178 | } |
179 | |