Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 2 | /* |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 3 | * Mini dd implementation for busybox |
| 4 | * |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 5 | * Copyright (C) 1999, 2000 by Lineo, inc. |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 6 | * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org> |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 7 | * |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 8 | * Based in part on code taken from sash. |
| 9 | * Copyright (c) 1999 by David I. Bell |
| 10 | * Permission is granted to use, distribute, or modify this source, |
| 11 | * provided that this copyright notice remains intact. |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 12 | * |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 13 | * Permission to distribute this code under the GPL has been granted. |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 14 | * |
| 15 | * This program is free software; you can redistribute it and/or modify |
| 16 | * it under the terms of the GNU General Public License as published by |
| 17 | * the Free Software Foundation; either version 2 of the License, or |
| 18 | * (at your option) any later version. |
| 19 | * |
| 20 | * This program is distributed in the hope that it will be useful, |
| 21 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 22 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 23 | * General Public License for more details. |
| 24 | * |
| 25 | * You should have received a copy of the GNU General Public License |
| 26 | * along with this program; if not, write to the Free Software |
| 27 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| 28 | * |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 29 | */ |
| 30 | |
Eric Andersen | c499601 | 1999-10-20 22:08:37 +0000 | [diff] [blame] | 31 | |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 32 | #include "internal.h" |
Eric Andersen | cb41c2e | 1999-11-22 07:41:00 +0000 | [diff] [blame] | 33 | #include <features.h> |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 34 | #include <stdio.h> |
| 35 | #include <fcntl.h> |
| 36 | #include <errno.h> |
Erik Andersen | 4d1d011 | 1999-12-17 18:44:15 +0000 | [diff] [blame] | 37 | #if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1) |
Eric Andersen | 6a76e65 | 1999-11-19 05:31:45 +0000 | [diff] [blame] | 38 | #include <inttypes.h> |
Eric Andersen | cb41c2e | 1999-11-22 07:41:00 +0000 | [diff] [blame] | 39 | #else |
| 40 | typedef unsigned long long int uintmax_t; |
| 41 | #endif |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 42 | |
Eric Andersen | e77ae3a | 1999-10-19 20:03:34 +0000 | [diff] [blame] | 43 | static const char dd_usage[] = |
Erik Andersen | 59b9e87 | 2000-05-10 05:05:45 +0000 | [diff] [blame] | 44 | "dd [if=FILE] [of=FILE] [bs=N] [count=N] [skip=N] [seek=N]\n" |
| 45 | #ifndef BB_FEATURE_TRIVIAL_HELP |
| 46 | "\nCopy a file, converting and formatting according to options\n\n" |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 47 | "\tif=FILE\tread from FILE instead of stdin\n" |
| 48 | "\tof=FILE\twrite to FILE instead of stdout\n" |
Erik Andersen | 59b9e87 | 2000-05-10 05:05:45 +0000 | [diff] [blame] | 49 | "\tbs=N\tread and write N bytes at a time\n" |
| 50 | "\tcount=N\tcopy only N input blocks\n" |
| 51 | "\tskip=N\tskip N input blocks\n" |
| 52 | "\tseek=N\tskip N output blocks\n" |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 53 | "\n" |
Erik Andersen | 59b9e87 | 2000-05-10 05:05:45 +0000 | [diff] [blame] | 54 | "Numbers may be suffixed by w (x2), k (x1024), b (x512), or M (x1024^2)\n" |
| 55 | #endif |
| 56 | ; |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 57 | |
| 58 | |
Eric Andersen | cc8ed39 | 1999-10-05 16:24:54 +0000 | [diff] [blame] | 59 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 60 | extern int dd_main(int argc, char **argv) |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 61 | { |
Erik Andersen | 1ad302a | 2000-03-24 00:54:46 +0000 | [diff] [blame] | 62 | char *inFile = NULL; |
| 63 | char *outFile = NULL; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 64 | char *cp; |
| 65 | int inFd; |
| 66 | int outFd; |
| 67 | int inCc = 0; |
| 68 | int outCc; |
| 69 | long blockSize = 512; |
| 70 | uintmax_t skipBlocks = 0; |
| 71 | uintmax_t seekBlocks = 0; |
| 72 | uintmax_t count = (uintmax_t) - 1; |
| 73 | uintmax_t intotal; |
| 74 | uintmax_t outTotal; |
| 75 | unsigned char *buf; |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 76 | |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 77 | argc--; |
| 78 | argv++; |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 79 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 80 | /* Parse any options */ |
| 81 | while (argc) { |
| 82 | if (inFile == NULL && (strncmp(*argv, "if", 2) == 0)) |
| 83 | inFile = ((strchr(*argv, '=')) + 1); |
| 84 | else if (outFile == NULL && (strncmp(*argv, "of", 2) == 0)) |
| 85 | outFile = ((strchr(*argv, '=')) + 1); |
| 86 | else if (strncmp("count", *argv, 5) == 0) { |
| 87 | count = getNum((strchr(*argv, '=')) + 1); |
| 88 | if (count <= 0) { |
Matt Kraai | d537a95 | 2000-07-14 01:51:25 +0000 | [diff] [blame^] | 89 | errorMsg("Bad count value %s\n", *argv); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 90 | goto usage; |
| 91 | } |
| 92 | } else if (strncmp(*argv, "bs", 2) == 0) { |
| 93 | blockSize = getNum((strchr(*argv, '=')) + 1); |
| 94 | if (blockSize <= 0) { |
Matt Kraai | d537a95 | 2000-07-14 01:51:25 +0000 | [diff] [blame^] | 95 | errorMsg("Bad block size value %s\n", *argv); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 96 | goto usage; |
| 97 | } |
| 98 | } else if (strncmp(*argv, "skip", 4) == 0) { |
| 99 | skipBlocks = getNum((strchr(*argv, '=')) + 1); |
| 100 | if (skipBlocks <= 0) { |
Matt Kraai | d537a95 | 2000-07-14 01:51:25 +0000 | [diff] [blame^] | 101 | errorMsg("Bad skip value %s\n", *argv); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 102 | goto usage; |
| 103 | } |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 104 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 105 | } else if (strncmp(*argv, "seek", 4) == 0) { |
| 106 | seekBlocks = getNum((strchr(*argv, '=')) + 1); |
| 107 | if (seekBlocks <= 0) { |
Matt Kraai | d537a95 | 2000-07-14 01:51:25 +0000 | [diff] [blame^] | 108 | errorMsg("Bad seek value %s\n", *argv); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 109 | goto usage; |
| 110 | } |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 111 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 112 | } else { |
| 113 | goto usage; |
| 114 | } |
| 115 | argc--; |
| 116 | argv++; |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 117 | } |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 118 | |
Erik Andersen | 0d068a2 | 2000-03-21 22:32:57 +0000 | [diff] [blame] | 119 | buf = xmalloc(blockSize); |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 120 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 121 | intotal = 0; |
| 122 | outTotal = 0; |
| 123 | |
| 124 | if (inFile == NULL) |
| 125 | inFd = fileno(stdin); |
| 126 | else |
| 127 | inFd = open(inFile, 0); |
| 128 | |
| 129 | if (inFd < 0) { |
Erik Andersen | 298854f | 2000-03-23 01:09:18 +0000 | [diff] [blame] | 130 | /* Note that we are not freeing buf or closing |
| 131 | * files here to save a few bytes. This exits |
| 132 | * here anyways... */ |
| 133 | |
| 134 | /* free(buf); */ |
| 135 | fatalError( inFile); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | if (outFile == NULL) |
| 139 | outFd = fileno(stdout); |
| 140 | else |
| 141 | outFd = open(outFile, O_WRONLY | O_CREAT | O_TRUNC, 0666); |
| 142 | |
| 143 | if (outFd < 0) { |
Erik Andersen | 298854f | 2000-03-23 01:09:18 +0000 | [diff] [blame] | 144 | /* Note that we are not freeing buf or closing |
| 145 | * files here to save a few bytes. This exits |
| 146 | * here anyways... */ |
| 147 | |
| 148 | /* close(inFd); |
| 149 | free(buf); */ |
| 150 | fatalError( outFile); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 151 | } |
| 152 | |
| 153 | lseek(inFd, skipBlocks * blockSize, SEEK_SET); |
| 154 | lseek(outFd, seekBlocks * blockSize, SEEK_SET); |
| 155 | // |
| 156 | //TODO: Convert to using fullRead & fullWrite |
| 157 | // from utility.c |
| 158 | // -Erik |
| 159 | while (outTotal < count * blockSize) { |
| 160 | inCc = read(inFd, buf, blockSize); |
| 161 | if (inCc < 0) { |
| 162 | perror(inFile); |
| 163 | goto cleanup; |
| 164 | } else if (inCc == 0) { |
| 165 | goto cleanup; |
| 166 | } |
| 167 | intotal += inCc; |
| 168 | cp = buf; |
| 169 | |
| 170 | while (intotal > outTotal) { |
| 171 | if (outTotal + inCc > count * blockSize) |
| 172 | inCc = count * blockSize - outTotal; |
| 173 | outCc = write(outFd, cp, inCc); |
| 174 | if (outCc < 0) { |
| 175 | perror(outFile); |
| 176 | goto cleanup; |
| 177 | } else if (outCc == 0) { |
| 178 | goto cleanup; |
| 179 | } |
| 180 | |
| 181 | inCc -= outCc; |
| 182 | cp += outCc; |
| 183 | outTotal += outCc; |
| 184 | } |
| 185 | } |
| 186 | |
| 187 | if (inCc < 0) |
| 188 | perror(inFile); |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 189 | |
| 190 | cleanup: |
Erik Andersen | 298854f | 2000-03-23 01:09:18 +0000 | [diff] [blame] | 191 | /* Note that we are not freeing memory or closing |
| 192 | * files here, to save a few bytes. */ |
| 193 | #if 0 |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 194 | close(inFd); |
| 195 | close(outFd); |
| 196 | free(buf); |
Erik Andersen | 298854f | 2000-03-23 01:09:18 +0000 | [diff] [blame] | 197 | #endif |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 198 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 199 | printf("%ld+%d records in\n", (long) (intotal / blockSize), |
| 200 | (intotal % blockSize) != 0); |
| 201 | printf("%ld+%d records out\n", (long) (outTotal / blockSize), |
| 202 | (outTotal % blockSize) != 0); |
| 203 | exit(TRUE); |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 204 | usage: |
| 205 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 206 | usage(dd_usage); |
Eric Andersen | 3cf52d1 | 1999-10-12 22:26:06 +0000 | [diff] [blame] | 207 | } |