mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
382 lines
8.6 KiB
382 lines
8.6 KiB
/* cat - conCATenate STDIN to STDOUT
|
|
*
|
|
* 24-Apr 1998 hiroyama
|
|
*
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
#pragma hdrstop
|
|
|
|
#include "fnreg.h"
|
|
|
|
#if !defined(UNICODE)
|
|
#error please compile me for UNICODE
|
|
#endif
|
|
|
|
#ifndef _T
|
|
#define _T TEXT
|
|
#endif
|
|
|
|
#define LINENUMBER 0x0001
|
|
#define EOL_MARK 0x0002
|
|
#define TAB_SPECIAL 0x0004
|
|
#define CTRL_SPECIAL 0x0008
|
|
#define NO_BUFFERED_OUTPUT 0x0010
|
|
#define UNICODE_INPUT 0x0020
|
|
#define AUTO_UNICODE_DETECT 0x0040
|
|
|
|
struct InOutMode {
|
|
BOOLEAN fUnicodeInput;
|
|
BOOLEAN fMbcsOutput;
|
|
BOOLEAN fNeedSwab;
|
|
BOOLEAN fLineBuffer;
|
|
};
|
|
|
|
DWORD options = 0;
|
|
|
|
#define LARGE_BUFSIZE (512)
|
|
|
|
void usage()
|
|
{
|
|
static const char str[] =
|
|
"cat [-aenotuvV] [filename ...]\n"
|
|
"-a same as -u if input begins with BOM\n"
|
|
"-e add '$' at the end of line\n"
|
|
"-n add line number\n"
|
|
"-o suppress buffering for output\n"
|
|
"-t show tab character as '^I'\n"
|
|
"-u unicode text processing\n"
|
|
"-v show control characters as '^' + alphabet, except tab and newline.\n"
|
|
"-V show version\n"
|
|
"-- declare end of options\n";
|
|
|
|
fputs(str, stderr);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
template <class T>
|
|
inline void ntoa(T* p, int n, int width)
|
|
{
|
|
p += width;
|
|
*p-- = '\0';
|
|
for (--width; width >= 0; --width) {
|
|
*p-- = (n % 10) + '0';
|
|
n /= 10;
|
|
}
|
|
}
|
|
|
|
template <class T>
|
|
inline void swap(T& a, T& b)
|
|
{
|
|
a ^= b ^= a ^= b;
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////
|
|
// Complex cat UNICODE helpers
|
|
/////////////////////////////////////////////////////////
|
|
|
|
inline int getcaw(InOutMode mode, FILE* fp)
|
|
{
|
|
if (mode.fUnicodeInput) {
|
|
wchar_t c = fgetwc(fp);
|
|
if (c == WEOF)
|
|
return EOF;
|
|
return c;
|
|
}
|
|
return fgetc(fp);
|
|
}
|
|
|
|
inline void ungetaw(InOutMode mode, int c, FILE* fp)
|
|
{
|
|
if (mode.fUnicodeInput)
|
|
ungetwc((wchar_t)c, fp);
|
|
else
|
|
ungetc(c, fp);
|
|
}
|
|
|
|
inline void putcharaw(InOutMode mode, int c)
|
|
{
|
|
if (mode.fUnicodeInput) {
|
|
// if output is dbcs (i.e. tty output), we need to
|
|
// translate the unicode character
|
|
if (mode.fMbcsOutput) {
|
|
// prevent to print BOM
|
|
if (c != 0xfeff) {
|
|
// translate the output
|
|
char buf[2];
|
|
int n = wctomb(buf, (wchar_t)c);
|
|
putchar(buf[0]);
|
|
if (n == 2) {
|
|
putchar(buf[1]);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
putwchar((wchar_t)c);
|
|
}
|
|
}
|
|
else
|
|
putchar(c);
|
|
}
|
|
|
|
inline void swab(int& c)
|
|
{
|
|
c = ((c & 0xff00) >> 8) | ((unsigned char)c << 8);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////
|
|
// Complex cat
|
|
/////////////////////////////////////////////////////////
|
|
|
|
void complex_cat(const TCHAR* fname)
|
|
{
|
|
FILE* fp;
|
|
static char* large_buf;
|
|
|
|
if (fname) {
|
|
if ((fp = _tfopen(fname, _T("rb"))) == NULL) {
|
|
_tperror(fname);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
large_buf = (char*)alloca(LARGE_BUFSIZE);
|
|
if (setvbuf(fp, large_buf, _IOFBF, LARGE_BUFSIZE))
|
|
perror("setvbuf");
|
|
}
|
|
else {
|
|
// if fname == NULL, take input from stdin.
|
|
fp = stdin;
|
|
}
|
|
|
|
static bool tol = true; // Top Of Line
|
|
static long lineno = 0;
|
|
int c, c2;
|
|
|
|
// Initialize In/Out mode
|
|
InOutMode inOutMode = {
|
|
!!(options & UNICODE_INPUT),
|
|
false,
|
|
false,
|
|
// if buffered mode and stdout is tty, flush buffer at each EOL
|
|
!(options & NO_BUFFERED_OUTPUT) && _isatty(_fileno(stdout)),
|
|
};
|
|
|
|
// UNICODE initialization
|
|
if (inOutMode.fUnicodeInput) {
|
|
// sample the first word for BOM detection
|
|
c = fgetwc(fp);
|
|
init_unicode:
|
|
_setmode(_fileno(fp), _O_BINARY);
|
|
if (_isatty(_fileno(stdout))) {
|
|
// if the output is tty,
|
|
// need to convert UNICODE to MBCS on output
|
|
inOutMode.fMbcsOutput = true;
|
|
}
|
|
|
|
// try to process the BOM
|
|
if (c == 0xfeff) {
|
|
putcharaw(inOutMode, c);
|
|
}
|
|
else if (c == 0xfffe) {
|
|
inOutMode.fNeedSwab = true;
|
|
swab(c);
|
|
putcharaw(inOutMode, c);
|
|
}
|
|
else {
|
|
ungetwc((wchar_t)c, fp);
|
|
}
|
|
}
|
|
else if (options & AUTO_UNICODE_DETECT) {
|
|
// sample and examine the first word to see if it's UNICODE BOM
|
|
c = fgetwc(fp);
|
|
if (c == 0xfffe || c == 0xfeff) {
|
|
inOutMode.fUnicodeInput = true;
|
|
goto init_unicode;
|
|
}
|
|
ungetwc((wchar_t)c, fp);
|
|
}
|
|
|
|
#ifdef MEASURE_PERF
|
|
DWORD start = ::GetTickCount();
|
|
#endif
|
|
|
|
while ((c = getcaw(inOutMode, fp)) != EOF) {
|
|
if (tol) {
|
|
// process line number
|
|
tol = false;
|
|
if (options & LINENUMBER) {
|
|
if (inOutMode.fUnicodeInput && !inOutMode.fMbcsOutput) {
|
|
wchar_t buf[5];
|
|
ntoa(buf, ++lineno, 4);
|
|
fputws(buf, stdout);
|
|
fputws(L": ", stdout);
|
|
}
|
|
else {
|
|
char buf[5];
|
|
ntoa(buf, ++lineno, 4);
|
|
fputs(buf, stdout);
|
|
fputs(": ", stdout);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (inOutMode.fNeedSwab)
|
|
swab(c);
|
|
|
|
switch (c) {
|
|
case '\r':
|
|
c2 = getcaw(inOutMode, fp);
|
|
if (c2 != '\n') {
|
|
ungetaw(inOutMode, c2, fp);
|
|
goto normal_input;
|
|
}
|
|
// fall through
|
|
case '\n':
|
|
if (options & EOL_MARK) {
|
|
putcharaw(inOutMode, '$');
|
|
}
|
|
if (c != '\n') {
|
|
putcharaw(inOutMode, c);
|
|
c = c2;
|
|
}
|
|
putcharaw(inOutMode, c);
|
|
if (inOutMode.fLineBuffer) {
|
|
// if line buffer mode, flush it
|
|
fflush(stdout);
|
|
}
|
|
tol = true;
|
|
break;
|
|
|
|
case '\t':
|
|
if (options & TAB_SPECIAL) {
|
|
fputs("^I", stdout);
|
|
}
|
|
else {
|
|
putcharaw(inOutMode, c);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
normal_input:
|
|
if (c < 0x20 && (options & CTRL_SPECIAL)) {
|
|
putcharaw(inOutMode, '^');
|
|
c += '@';
|
|
}
|
|
putcharaw(inOutMode, c);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (fname) {
|
|
fclose(fp);
|
|
}
|
|
|
|
#ifdef MEASURE_PERF
|
|
DWORD end = ::GetTickCount();
|
|
fprintf(stderr, "delta=%u\n", end - start);
|
|
#endif
|
|
}
|
|
|
|
void cat(const TCHAR* fname = NULL)
|
|
{
|
|
static bool is1st = true;
|
|
if (is1st) {
|
|
is1st = false;
|
|
if (options & NO_BUFFERED_OUTPUT) {
|
|
// non buffered mode
|
|
if (setvbuf(stdout, NULL, _IONBF, 0))
|
|
perror("setvbuf");
|
|
}
|
|
else {
|
|
if (setvbuf(stdout, NULL, _IOFBF, LARGE_BUFSIZE))
|
|
perror("setvbuf");
|
|
}
|
|
}
|
|
|
|
complex_cat(fname);
|
|
}
|
|
|
|
|
|
void parse_option(const TCHAR* s, bool& eoo) // eoo: end of options
|
|
{
|
|
extern char version[];
|
|
|
|
while (*++s) {
|
|
switch (*s) {
|
|
case _T('-'):
|
|
eoo = true;
|
|
return;
|
|
case _T('a'):
|
|
options |= AUTO_UNICODE_DETECT;
|
|
break;
|
|
case _T('e'):
|
|
options |= EOL_MARK;
|
|
break;
|
|
case _T('n'):
|
|
options |= LINENUMBER;
|
|
break;
|
|
case _T('v'):
|
|
options |= CTRL_SPECIAL;
|
|
break;
|
|
case _T('t'):
|
|
options |= TAB_SPECIAL;
|
|
break;
|
|
case _T('o'):
|
|
options |= NO_BUFFERED_OUTPUT;
|
|
break;
|
|
case _T('u'):
|
|
options |= UNICODE_INPUT;
|
|
break;
|
|
case _T('V'):
|
|
fputs(version, stderr);
|
|
exit(EXIT_SUCCESS);
|
|
default:
|
|
usage(); // never returns
|
|
}
|
|
}
|
|
}
|
|
|
|
#ifdef UNICODE
|
|
#define main wmain
|
|
#endif
|
|
|
|
extern "C"
|
|
int __cdecl main(int argc, TCHAR** argv)
|
|
{
|
|
int n = 0;
|
|
bool eoo = false;
|
|
|
|
fnexpand(&argc, &argv);
|
|
|
|
setlocale(LC_ALL, "");
|
|
|
|
#if 1
|
|
// set stdout binary mode
|
|
_setmode(_fileno(stdout), _O_BINARY);
|
|
|
|
#else // TEST
|
|
for (int i = 0; i < argc; ++i) {
|
|
_putts(argv[i]);
|
|
}
|
|
exit(0);
|
|
#endif
|
|
|
|
while (--argc) {
|
|
if (**++argv == _T('-') && !eoo) {
|
|
parse_option(*argv, eoo);
|
|
}
|
|
else {
|
|
++n;
|
|
eoo = true;
|
|
cat(*argv);
|
|
}
|
|
}
|
|
|
|
if (n == 0) {
|
|
_setmode(_fileno(stdin), _O_BINARY);
|
|
cat();
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|