Tito shrank dos2unix.

This commit is contained in:
Rob Landley 2006-03-14 21:49:18 +00:00
parent 31642d75e1
commit 330ac853e3

View File

@ -10,21 +10,7 @@
* Without arguments it reverts the format (e.i. if source is in UNIX format,
* output is in DOS format and vice versa).
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
* See the COPYING file for license information.
* Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
*/
#include <string.h>
@ -36,55 +22,34 @@
#define CT_UNIX2DOS 1
#define CT_DOS2UNIX 2
/* We are making a lame pseudo-random string generator here. in
* convert(), each pass through the while loop will add more and more
* stuff into value, which is _supposed_ to wrap. We don't care about
* it being accurate. We care about it being messy, since we use it
* to pick a random letter to add to out temporary file. */
typedef unsigned long int bb_uint64_t;
#define tempFn bb_common_bufsiz1
/* if fn is NULL then input is stdin and output is stdout */
static int convert(char *fn, int ConvType)
{
int c, fd;
struct timeval tv;
static bb_uint64_t value=0;
FILE *in, *out;
if (fn != NULL) {
in = bb_xfopen(fn, "rw");
safe_strncpy(tempFn, fn, sizeof(tempFn));
c = strlen(tempFn);
tempFn[c] = '.';
while(1) {
/* tempFn is BUFSIZ so the last addressable spot it BUFSIZ-1.
* The loop increments by 2. So this must check for BUFSIZ-3. */
if (c >=BUFSIZ-3)
bb_error_msg_and_die("unique name not found");
/* Get some semi random stuff to try and make a
* random filename based (and in the same dir as)
* the input file... */
gettimeofday (&tv, NULL);
value += ((bb_uint64_t) tv.tv_usec << 16) ^ tv.tv_sec ^ getpid ();
tempFn[++c] = ((value%62) < 26)?(value%62)+97:
((value%62) < 52)?(value%62)+39:
(value%62)-4;
tempFn[c+1] = '\0';
value /= 62;
if ((fd = open(tempFn, O_RDWR | O_CREAT | O_EXCL, 0600)) < 0 ) {
continue;
/*
The file is then created with mode read/write and
permissions 0666 for glibc 2.0.6 and earlier or
0600 for glibc 2.0.7 and later.
*/
snprintf(tempFn, sizeof(tempFn), "%sXXXXXX", fn);
/*
sizeof tempFn is 4096, so it should be big enough to hold the full
path. however if the output is truncated the subsequent call to
mkstemp would fail.
*/
if ((fd = mkstemp(&tempFn[0])) == -1 || chmod(tempFn, 0600) == -1) {
bb_perror_nomsg_and_die();
}
out = fdopen(fd, "w+");
if (!out) {
close(fd);
remove(tempFn);
continue;
}
break;
}
} else {
in = stdin;
@ -92,37 +57,6 @@ static int convert(char *fn, int ConvType)
}
while ((c = fgetc(in)) != EOF) {
if (c == '\r') {
if ((ConvType == CT_UNIX2DOS) && (fn != NULL)) {
/* file is already in DOS format so it is
* not necessary to touch it. */
remove(tempFn);
if (fclose(in) < 0 || fclose(out) < 0) {
bb_perror_nomsg();
return -2;
}
return 0;
}
break;
}
if (c == '\n') {
if ((ConvType == CT_DOS2UNIX) && (fn != NULL)) {
/* file is already in DOS format so it is
* not necessary to touch it. */
remove(tempFn);
if ((fclose(in) < 0) || (fclose(out) < 0)) {
bb_perror_nomsg();
return -2;
}
return 0;
}
if (ConvType == CT_UNIX2DOS) {
fputc('\r', out);
}
}
fputc(c, out);
}
while (c != EOF && (c = fgetc(in)) != EOF) {
if (c == '\r')
continue;
if (c == '\n') {
@ -145,7 +79,7 @@ static int convert(char *fn, int ConvType)
* should be true since we put them into the same directory
* so we _should_ be ok, but you never know... */
if (rename(tempFn, fn) < 0) {
bb_perror_msg("unable to rename '%s' as '%s'", tempFn, fn);
bb_perror_msg("cannot rename '%s' as '%s'", tempFn, fn);
return -1;
}
}
@ -160,13 +94,16 @@ int dos2unix_main(int argc, char *argv[])
/* See if we are supposed to be doing dos2unix or unix2dos */
if (argv[0][0]=='d') {
ConvType = CT_DOS2UNIX;
ConvType = CT_DOS2UNIX; /*1*/
} else {
ConvType = CT_UNIX2DOS;
ConvType = CT_UNIX2DOS; /*2*/
}
/* -u and -d are mutally exclusive */
bb_opt_complementally = "?:u--d:d--u";
/* process parameters */
o = bb_getopt_ulflags(argc, argv, "ud");
/* -u convert to unix */
/* -d convert to dos */
o = bb_getopt_ulflags(argc, argv, "du");
/* Do the conversion requested by an argument else do the default
* conversion depending on our name. */
@ -183,4 +120,3 @@ int dos2unix_main(int argc, char *argv[])
return o;
}