blob: e2a4ef6c50478bb0931f42408ce372883eef9b10 [file] [log] [blame]
Erik Andersenfb002d02000-03-05 08:07:00 +00001/* vi: set sw=4 ts=4: */
2/*
Erik Andersen8f8d6d52000-05-01 22:30:37 +00003 * Mini tr implementation for busybox
Erik Andersenfb002d02000-03-05 08:07:00 +00004 *
Erik Andersen5afc8642000-05-02 00:07:56 +00005 * Copyright (c) Michiel Huisjes
6 *
7 * This version of tr is adapted from Minix tr and was modified
8 * by Erik Andersen <andersee@debian.org> to be used in busybox.
Erik Andersenfb002d02000-03-05 08:07:00 +00009 *
Erik Andersen8f8d6d52000-05-01 22:30:37 +000010 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 *
24 * Original copyright notice is retained at the end of this file.
Erik Andersenfb002d02000-03-05 08:07:00 +000025 */
26
Erik Andersenfb002d02000-03-05 08:07:00 +000027#include <stdio.h>
Erik Andersenfb002d02000-03-05 08:07:00 +000028#include <string.h>
Erik Andersen8f8d6d52000-05-01 22:30:37 +000029#include <stdlib.h>
Erik Andersenfb002d02000-03-05 08:07:00 +000030#include <unistd.h>
Erik Andersen8f8d6d52000-05-01 22:30:37 +000031#include <sys/types.h>
Eric Andersencbe31da2001-02-20 06:14:08 +000032#include "busybox.h"
Erik Andersen330fd2b2000-05-19 05:35:19 +000033
Eric Andersen22ecf042001-07-02 17:32:40 +000034/* This must be a #define, since when DODEBUG and BUFFERS_GO_IN_BSS are
35 * enabled, we otherwise get a "storage size isn't constant error. */
36#define ASCII 0377
Erik Andersenfb002d02000-03-05 08:07:00 +000037
Mark Whitley8b7a0d82001-05-24 21:31:09 +000038/* some "globals" shared across this file */
Erik Andersen8f8d6d52000-05-01 22:30:37 +000039static char com_fl, del_fl, sq_fl;
Erik Andersen8f8d6d52000-05-01 22:30:37 +000040static short in_index, out_index;
Mark Whitley59ab0252001-01-23 22:30:04 +000041/* these last are pointers to static buffers declared in tr_main */
42static unsigned char *poutput, *pinput;
43static unsigned char *pvector;
44static char *pinvec, *poutvec;
Erik Andersenfb002d02000-03-05 08:07:00 +000045
Erik Andersenfb002d02000-03-05 08:07:00 +000046
Eric Anderseneaecbf32001-10-31 10:41:31 +000047static void convert(void)
Erik Andersenfb002d02000-03-05 08:07:00 +000048{
Erik Andersen8f8d6d52000-05-01 22:30:37 +000049 short read_chars = 0;
50 short c, coded;
51 short last = -1;
Erik Andersenfb002d02000-03-05 08:07:00 +000052
Erik Andersen8f8d6d52000-05-01 22:30:37 +000053 for (;;) {
54 if (in_index == read_chars) {
Mark Whitley59ab0252001-01-23 22:30:04 +000055 if ((read_chars = read(0, (char *) pinput, BUFSIZ)) <= 0) {
56 if (write(1, (char *) poutput, out_index) != out_index)
Eric Andersen5a4a46a2001-07-09 21:32:29 +000057 error_msg("%s", write_error);
Erik Andersen8f8d6d52000-05-01 22:30:37 +000058 exit(0);
Erik Andersenfb002d02000-03-05 08:07:00 +000059 }
Erik Andersen8f8d6d52000-05-01 22:30:37 +000060 in_index = 0;
61 }
Mark Whitley59ab0252001-01-23 22:30:04 +000062 c = pinput[in_index++];
63 coded = pvector[c];
64 if (del_fl && pinvec[c])
Erik Andersen8f8d6d52000-05-01 22:30:37 +000065 continue;
Mark Whitley59ab0252001-01-23 22:30:04 +000066 if (sq_fl && last == coded && (pinvec[c] || poutvec[coded]))
Erik Andersen8f8d6d52000-05-01 22:30:37 +000067 continue;
Mark Whitley59ab0252001-01-23 22:30:04 +000068 poutput[out_index++] = last = coded;
Erik Andersen8f8d6d52000-05-01 22:30:37 +000069 if (out_index == BUFSIZ) {
Eric Andersen5a4a46a2001-07-09 21:32:29 +000070 if (write(1, (char *) poutput, out_index) != out_index)
71 error_msg_and_die("%s", write_error);
Erik Andersen8f8d6d52000-05-01 22:30:37 +000072 out_index = 0;
73 }
Erik Andersenfb002d02000-03-05 08:07:00 +000074 }
75
Erik Andersenfb002d02000-03-05 08:07:00 +000076 /* NOTREACHED */
Erik Andersen8f8d6d52000-05-01 22:30:37 +000077}
78
Eric Andersen00143ba2000-07-13 16:40:41 +000079static void map(register unsigned char *string1, unsigned int string1_len,
80 register unsigned char *string2, unsigned int string2_len)
Erik Andersen8f8d6d52000-05-01 22:30:37 +000081{
82 unsigned char last = '0';
Eric Andersen00143ba2000-07-13 16:40:41 +000083 unsigned int i, j;
Erik Andersen8f8d6d52000-05-01 22:30:37 +000084
Eric Andersen00143ba2000-07-13 16:40:41 +000085 for (j = 0, i = 0; i < string1_len; i++) {
86 if (string2_len <= j)
Mark Whitley59ab0252001-01-23 22:30:04 +000087 pvector[string1[i]] = last;
Erik Andersen8f8d6d52000-05-01 22:30:37 +000088 else
Mark Whitley59ab0252001-01-23 22:30:04 +000089 pvector[string1[i]] = last = string2[j++];
Erik Andersen8f8d6d52000-05-01 22:30:37 +000090 }
91}
92
Mark Whitley8b7a0d82001-05-24 21:31:09 +000093/* supported constructs:
94 * Ranges, e.g., [0-9] ==> 0123456789
95 * Escapes, e.g., \a ==> Control-G
96 */
Eric Andersene5dfced2001-04-09 22:48:12 +000097static unsigned int expand(const char *arg, register unsigned char *buffer)
Erik Andersen8f8d6d52000-05-01 22:30:37 +000098{
Eric Andersen00143ba2000-07-13 16:40:41 +000099 unsigned char *buffer_start = buffer;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000100 int i, ac;
101
102 while (*arg) {
103 if (*arg == '\\') {
104 arg++;
Eric Andersenf7cf2f72000-07-05 17:26:35 +0000105 *buffer++ = process_escape_sequence(&arg);
Eric Andersen5a4a46a2001-07-09 21:32:29 +0000106 } else if (*(arg+1) == '-') {
107 ac = *(arg+2);
108 if(ac == 0) {
109 *buffer++ = *arg++;
110 continue;
111 }
112 i = *arg;
113 while (i <= ac)
114 *buffer++ = i++;
115 arg += 3; /* Skip the assumed a-z */
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000116 } else if (*arg == '[') {
117 arg++;
118 i = *arg++;
119 if (*arg++ != '-') {
120 *buffer++ = '[';
121 arg -= 2;
122 continue;
123 }
124 ac = *arg++;
125 while (i <= ac)
126 *buffer++ = i++;
Mark Whitley8b7a0d82001-05-24 21:31:09 +0000127 arg++; /* Skip the assumed ']' */
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000128 } else
129 *buffer++ = *arg++;
130 }
Eric Andersen00143ba2000-07-13 16:40:41 +0000131
132 return (buffer - buffer_start);
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000133}
134
Eric Andersenfad04fd2000-07-14 06:49:52 +0000135static int complement(unsigned char *buffer, int buffer_len)
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000136{
Eric Andersen1ca20a72001-03-21 07:34:27 +0000137 register short i, j, ix;
Eric Andersenfad04fd2000-07-14 06:49:52 +0000138 char conv[ASCII + 2];
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000139
Eric Andersen1ca20a72001-03-21 07:34:27 +0000140 ix = 0;
Eric Andersen00143ba2000-07-13 16:40:41 +0000141 for (i = 0; i <= ASCII; i++) {
142 for (j = 0; j < buffer_len; j++)
143 if (buffer[j] == i)
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000144 break;
Eric Andersen00143ba2000-07-13 16:40:41 +0000145 if (j == buffer_len)
Eric Andersen1ca20a72001-03-21 07:34:27 +0000146 conv[ix++] = i & ASCII;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000147 }
Eric Andersen1ca20a72001-03-21 07:34:27 +0000148 memcpy(buffer, conv, ix);
149 return ix;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000150}
151
152extern int tr_main(int argc, char **argv)
153{
154 register unsigned char *ptr;
Eric Andersenfad04fd2000-07-14 06:49:52 +0000155 int output_length=0, input_length;
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000156 int idx = 1;
Eric Andersenfad04fd2000-07-14 06:49:52 +0000157 int i;
Eric Andersenbdfd0d72001-10-24 05:00:29 +0000158 RESERVE_CONFIG_BUFFER(output, BUFSIZ);
159 RESERVE_CONFIG_BUFFER(input, BUFSIZ);
160 RESERVE_CONFIG_UBUFFER(vector, ASCII+1);
161 RESERVE_CONFIG_BUFFER(invec, ASCII+1);
162 RESERVE_CONFIG_BUFFER(outvec, ASCII+1);
Mark Whitley59ab0252001-01-23 22:30:04 +0000163
164 /* ... but make them available globally */
165 poutput = output;
166 pinput = input;
167 pvector = vector;
168 pinvec = invec;
169 poutvec = outvec;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000170
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000171 if (argc > 1 && argv[idx][0] == '-') {
172 for (ptr = (unsigned char *) &argv[idx][1]; *ptr; ptr++) {
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000173 switch (*ptr) {
174 case 'c':
175 com_fl = TRUE;
176 break;
177 case 'd':
178 del_fl = TRUE;
179 break;
180 case 's':
181 sq_fl = TRUE;
182 break;
183 default:
Eric Andersen67991cf2001-02-14 21:23:06 +0000184 show_usage();
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000185 }
186 }
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000187 idx++;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000188 }
189 for (i = 0; i <= ASCII; i++) {
190 vector[i] = i;
191 invec[i] = outvec[i] = FALSE;
192 }
193
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000194 if (argv[idx] != NULL) {
195 input_length = expand(argv[idx++], input);
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000196 if (com_fl)
Eric Andersen00143ba2000-07-13 16:40:41 +0000197 input_length = complement(input, input_length);
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000198 if (argv[idx] != NULL) {
199 if (*argv[idx] == '\0')
Matt Kraaidd19c692001-01-31 19:00:21 +0000200 error_msg_and_die("STRING2 cannot be empty");
Eric Andersenf6aa13d2001-03-23 17:08:21 +0000201 output_length = expand(argv[idx], output);
Eric Andersen00143ba2000-07-13 16:40:41 +0000202 map(input, input_length, output, output_length);
Eric Andersena03d86c2000-07-10 16:38:50 +0000203 }
Eric Andersen00143ba2000-07-13 16:40:41 +0000204 for (i = 0; i < input_length; i++)
Eric Andersened3ef502001-01-27 08:24:39 +0000205 invec[(int)input[i]] = TRUE;
Eric Andersen00143ba2000-07-13 16:40:41 +0000206 for (i = 0; i < output_length; i++)
Eric Andersened3ef502001-01-27 08:24:39 +0000207 outvec[(int)output[i]] = TRUE;
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000208 }
209 convert();
Erik Andersenfb002d02000-03-05 08:07:00 +0000210 return (0);
211}
212
Erik Andersenfb002d02000-03-05 08:07:00 +0000213/*
Erik Andersen8f8d6d52000-05-01 22:30:37 +0000214 * Copyright (c) 1987,1997, Prentice Hall
215 * All rights reserved.
216 *
217 * Redistribution and use of the MINIX operating system in source and
218 * binary forms, with or without modification, are permitted provided
219 * that the following conditions are met:
220 *
221 * Redistributions of source code must retain the above copyright
222 * notice, this list of conditions and the following disclaimer.
223 *
224 * Redistributions in binary form must reproduce the above
225 * copyright notice, this list of conditions and the following
226 * disclaimer in the documentation and/or other materials provided
227 * with the distribution.
228 *
229 * Neither the name of Prentice Hall nor the names of the software
230 * authors or contributors may be used to endorse or promote
231 * products derived from this software without specific prior
232 * written permission.
233 *
234 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS, AUTHORS, AND
235 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
236 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
237 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
238 * IN NO EVENT SHALL PRENTICE HALL OR ANY AUTHORS OR CONTRIBUTORS BE
239 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
240 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
241 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
242 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
243 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
244 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
245 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
246 *
Erik Andersenfb002d02000-03-05 08:07:00 +0000247 */
Erik Andersenfb002d02000-03-05 08:07:00 +0000248