Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
| 2 | /* |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 3 | * Mini tr implementation for busybox |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 4 | * |
Erik Andersen | 5afc864 | 2000-05-02 00:07:56 +0000 | [diff] [blame] | 5 | * 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 Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 9 | * |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 10 | * 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 Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 25 | */ |
| 26 | |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 27 | #include <stdio.h> |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 28 | #include <string.h> |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 29 | #include <stdlib.h> |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 30 | #include <unistd.h> |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 31 | #include <sys/types.h> |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 32 | #include "busybox.h" |
Erik Andersen | 330fd2b | 2000-05-19 05:35:19 +0000 | [diff] [blame] | 33 | |
Eric Andersen | 22ecf04 | 2001-07-02 17:32:40 +0000 | [diff] [blame] | 34 | /* 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 Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 37 | |
Mark Whitley | 8b7a0d8 | 2001-05-24 21:31:09 +0000 | [diff] [blame] | 38 | /* some "globals" shared across this file */ |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 39 | static char com_fl, del_fl, sq_fl; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 40 | static short in_index, out_index; |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 41 | /* these last are pointers to static buffers declared in tr_main */ |
| 42 | static unsigned char *poutput, *pinput; |
| 43 | static unsigned char *pvector; |
| 44 | static char *pinvec, *poutvec; |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 45 | |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 46 | |
Eric Andersen | eaecbf3 | 2001-10-31 10:41:31 +0000 | [diff] [blame^] | 47 | static void convert(void) |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 48 | { |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 49 | short read_chars = 0; |
| 50 | short c, coded; |
| 51 | short last = -1; |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 52 | |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 53 | for (;;) { |
| 54 | if (in_index == read_chars) { |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 55 | if ((read_chars = read(0, (char *) pinput, BUFSIZ)) <= 0) { |
| 56 | if (write(1, (char *) poutput, out_index) != out_index) |
Eric Andersen | 5a4a46a | 2001-07-09 21:32:29 +0000 | [diff] [blame] | 57 | error_msg("%s", write_error); |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 58 | exit(0); |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 59 | } |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 60 | in_index = 0; |
| 61 | } |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 62 | c = pinput[in_index++]; |
| 63 | coded = pvector[c]; |
| 64 | if (del_fl && pinvec[c]) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 65 | continue; |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 66 | if (sq_fl && last == coded && (pinvec[c] || poutvec[coded])) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 67 | continue; |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 68 | poutput[out_index++] = last = coded; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 69 | if (out_index == BUFSIZ) { |
Eric Andersen | 5a4a46a | 2001-07-09 21:32:29 +0000 | [diff] [blame] | 70 | if (write(1, (char *) poutput, out_index) != out_index) |
| 71 | error_msg_and_die("%s", write_error); |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 72 | out_index = 0; |
| 73 | } |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 74 | } |
| 75 | |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 76 | /* NOTREACHED */ |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 77 | } |
| 78 | |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 79 | static void map(register unsigned char *string1, unsigned int string1_len, |
| 80 | register unsigned char *string2, unsigned int string2_len) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 81 | { |
| 82 | unsigned char last = '0'; |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 83 | unsigned int i, j; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 84 | |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 85 | for (j = 0, i = 0; i < string1_len; i++) { |
| 86 | if (string2_len <= j) |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 87 | pvector[string1[i]] = last; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 88 | else |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 89 | pvector[string1[i]] = last = string2[j++]; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 90 | } |
| 91 | } |
| 92 | |
Mark Whitley | 8b7a0d8 | 2001-05-24 21:31:09 +0000 | [diff] [blame] | 93 | /* supported constructs: |
| 94 | * Ranges, e.g., [0-9] ==> 0123456789 |
| 95 | * Escapes, e.g., \a ==> Control-G |
| 96 | */ |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 97 | static unsigned int expand(const char *arg, register unsigned char *buffer) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 98 | { |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 99 | unsigned char *buffer_start = buffer; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 100 | int i, ac; |
| 101 | |
| 102 | while (*arg) { |
| 103 | if (*arg == '\\') { |
| 104 | arg++; |
Eric Andersen | f7cf2f7 | 2000-07-05 17:26:35 +0000 | [diff] [blame] | 105 | *buffer++ = process_escape_sequence(&arg); |
Eric Andersen | 5a4a46a | 2001-07-09 21:32:29 +0000 | [diff] [blame] | 106 | } 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 Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 116 | } 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 Whitley | 8b7a0d8 | 2001-05-24 21:31:09 +0000 | [diff] [blame] | 127 | arg++; /* Skip the assumed ']' */ |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 128 | } else |
| 129 | *buffer++ = *arg++; |
| 130 | } |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 131 | |
| 132 | return (buffer - buffer_start); |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 133 | } |
| 134 | |
Eric Andersen | fad04fd | 2000-07-14 06:49:52 +0000 | [diff] [blame] | 135 | static int complement(unsigned char *buffer, int buffer_len) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 136 | { |
Eric Andersen | 1ca20a7 | 2001-03-21 07:34:27 +0000 | [diff] [blame] | 137 | register short i, j, ix; |
Eric Andersen | fad04fd | 2000-07-14 06:49:52 +0000 | [diff] [blame] | 138 | char conv[ASCII + 2]; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 139 | |
Eric Andersen | 1ca20a7 | 2001-03-21 07:34:27 +0000 | [diff] [blame] | 140 | ix = 0; |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 141 | for (i = 0; i <= ASCII; i++) { |
| 142 | for (j = 0; j < buffer_len; j++) |
| 143 | if (buffer[j] == i) |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 144 | break; |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 145 | if (j == buffer_len) |
Eric Andersen | 1ca20a7 | 2001-03-21 07:34:27 +0000 | [diff] [blame] | 146 | conv[ix++] = i & ASCII; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 147 | } |
Eric Andersen | 1ca20a7 | 2001-03-21 07:34:27 +0000 | [diff] [blame] | 148 | memcpy(buffer, conv, ix); |
| 149 | return ix; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 150 | } |
| 151 | |
| 152 | extern int tr_main(int argc, char **argv) |
| 153 | { |
| 154 | register unsigned char *ptr; |
Eric Andersen | fad04fd | 2000-07-14 06:49:52 +0000 | [diff] [blame] | 155 | int output_length=0, input_length; |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 156 | int idx = 1; |
Eric Andersen | fad04fd | 2000-07-14 06:49:52 +0000 | [diff] [blame] | 157 | int i; |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 158 | 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 Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 163 | |
| 164 | /* ... but make them available globally */ |
| 165 | poutput = output; |
| 166 | pinput = input; |
| 167 | pvector = vector; |
| 168 | pinvec = invec; |
| 169 | poutvec = outvec; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 170 | |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 171 | if (argc > 1 && argv[idx][0] == '-') { |
| 172 | for (ptr = (unsigned char *) &argv[idx][1]; *ptr; ptr++) { |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 173 | 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 Andersen | 67991cf | 2001-02-14 21:23:06 +0000 | [diff] [blame] | 184 | show_usage(); |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 185 | } |
| 186 | } |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 187 | idx++; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 188 | } |
| 189 | for (i = 0; i <= ASCII; i++) { |
| 190 | vector[i] = i; |
| 191 | invec[i] = outvec[i] = FALSE; |
| 192 | } |
| 193 | |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 194 | if (argv[idx] != NULL) { |
| 195 | input_length = expand(argv[idx++], input); |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 196 | if (com_fl) |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 197 | input_length = complement(input, input_length); |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 198 | if (argv[idx] != NULL) { |
| 199 | if (*argv[idx] == '\0') |
Matt Kraai | dd19c69 | 2001-01-31 19:00:21 +0000 | [diff] [blame] | 200 | error_msg_and_die("STRING2 cannot be empty"); |
Eric Andersen | f6aa13d | 2001-03-23 17:08:21 +0000 | [diff] [blame] | 201 | output_length = expand(argv[idx], output); |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 202 | map(input, input_length, output, output_length); |
Eric Andersen | a03d86c | 2000-07-10 16:38:50 +0000 | [diff] [blame] | 203 | } |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 204 | for (i = 0; i < input_length; i++) |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 205 | invec[(int)input[i]] = TRUE; |
Eric Andersen | 00143ba | 2000-07-13 16:40:41 +0000 | [diff] [blame] | 206 | for (i = 0; i < output_length; i++) |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 207 | outvec[(int)output[i]] = TRUE; |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 208 | } |
| 209 | convert(); |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 210 | return (0); |
| 211 | } |
| 212 | |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 213 | /* |
Erik Andersen | 8f8d6d5 | 2000-05-01 22:30:37 +0000 | [diff] [blame] | 214 | * 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 Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 247 | */ |
Erik Andersen | fb002d0 | 2000-03-05 08:07:00 +0000 | [diff] [blame] | 248 | |