586 lines
10 KiB
C++
586 lines
10 KiB
C++
/*
|
|
* $Id: Awk.cpp,v 1.15 2007/05/16 14:44:13 bacon Exp $
|
|
*/
|
|
|
|
#include <ase/awk/StdAwk.hpp>
|
|
#include <ase/cmn/str.h>
|
|
#include <ase/utl/stdio.h>
|
|
#include <ase/utl/main.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <math.h>
|
|
|
|
#if defined(_WIN32)
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
class TestAwk: public ASE::StdAwk
|
|
{
|
|
public:
|
|
TestAwk (): srcInName(ASE_NULL), srcOutName(ASE_NULL),
|
|
numConInFiles(0), numConOutFiles(0)
|
|
{
|
|
#ifdef _WIN32
|
|
heap = ASE_NULL;
|
|
#endif
|
|
}
|
|
|
|
~TestAwk ()
|
|
{
|
|
close ();
|
|
}
|
|
|
|
int open ()
|
|
{
|
|
#ifdef _WIN32
|
|
ASE_ASSERT (heap == ASE_NULL);
|
|
heap = HeapCreate (0, 1000000, 1000000);
|
|
if (heap == ASE_NULL) return -1;
|
|
#endif
|
|
|
|
return StdAwk::open ();
|
|
}
|
|
|
|
void close ()
|
|
{
|
|
StdAwk::close ();
|
|
|
|
numConInFiles = 0;
|
|
numConOutFiles = 0;
|
|
|
|
#ifdef _WIN32
|
|
HeapDestroy (heap);
|
|
heap = ASE_NULL;
|
|
#endif
|
|
}
|
|
|
|
int addConsoleInput (const char_t* file)
|
|
{
|
|
if (numConInFiles < ASE_COUNTOF(conInFile))
|
|
{
|
|
conInFile[numConInFiles++] = file;
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int addConsoleOutput (const char_t* file)
|
|
{
|
|
if (numConOutFiles < ASE_COUNTOF(conOutFile))
|
|
{
|
|
conOutFile[numConOutFiles++] = file;
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int parse (const char_t* in, const char_t* out)
|
|
{
|
|
srcInName = in;
|
|
srcOutName = out;
|
|
return StdAwk::parse ();
|
|
}
|
|
|
|
protected:
|
|
|
|
void onRunStart (const Run& run)
|
|
{
|
|
wprintf (L"*** awk run started ***\n");
|
|
}
|
|
|
|
void onRunEnd (const Run& run, int errnum)
|
|
{
|
|
wprintf (L"*** awk run ended ***\n");
|
|
}
|
|
|
|
int openSource (Source& io)
|
|
{
|
|
Source::Mode mode = io.getMode();
|
|
FILE* fp = ASE_NULL;
|
|
|
|
if (mode == Source::READ)
|
|
{
|
|
if (srcInName == ASE_NULL)
|
|
{
|
|
io.setHandle (stdin);
|
|
return 0;
|
|
}
|
|
|
|
if (srcInName[0] == ASE_T('\0')) fp = stdin;
|
|
else fp = ase_fopen (srcInName, ASE_T("r"));
|
|
}
|
|
else if (mode == Source::WRITE)
|
|
{
|
|
if (srcOutName == ASE_NULL)
|
|
{
|
|
io.setHandle (stdout);
|
|
return 0;
|
|
}
|
|
|
|
if (srcOutName[0] == ASE_T('\0')) fp = stdout;
|
|
else fp = ase_fopen (srcOutName, ASE_T("w"));
|
|
}
|
|
|
|
if (fp == ASE_NULL) return -1;
|
|
io.setHandle (fp);
|
|
return 1;
|
|
}
|
|
|
|
int closeSource (Source& io)
|
|
{
|
|
Source::Mode mode = io.getMode();
|
|
FILE* fp = (FILE*)io.getHandle();
|
|
if (fp != stdin && fp != stdout && fp != stderr) fclose (fp);
|
|
io.setHandle (ASE_NULL);
|
|
return 0;
|
|
}
|
|
|
|
ssize_t readSource (Source& io, char_t* buf, size_t len)
|
|
{
|
|
FILE* fp = (FILE*)io.getHandle();
|
|
ssize_t n = 0;
|
|
|
|
while (n < len)
|
|
{
|
|
ase_cint_t c = ase_fgetc (fp);
|
|
if (c == ASE_CHAR_EOF) break;
|
|
|
|
buf[n++] = c;
|
|
if (c == ASE_T('\n')) break;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
ssize_t writeSource (Source& io, char_t* buf, size_t len)
|
|
{
|
|
FILE* fp = (FILE*)io.getHandle();
|
|
size_t left = len;
|
|
|
|
while (left > 0)
|
|
{
|
|
if (*buf == ASE_T('\0'))
|
|
{
|
|
if (ase_fputc (*buf, fp) == ASE_CHAR_EOF) return -1;
|
|
left -= 1; buf += 1;
|
|
}
|
|
else
|
|
{
|
|
int n = ase_fprintf (fp, ASE_T("%.*s"), left, buf);
|
|
if (n < 0) return -1;
|
|
left -= n; buf += n;
|
|
}
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
// console io handlers
|
|
int openConsole (Console& io)
|
|
{
|
|
Awk::Console::Mode mode = io.getMode();
|
|
FILE* fp = ASE_NULL;
|
|
const char_t* fn = ASE_NULL;
|
|
|
|
switch (mode)
|
|
{
|
|
case Awk::Console::READ:
|
|
if (numConInFiles == 0) fp = stdin;
|
|
else
|
|
{
|
|
fn = conInFile[0];
|
|
fp = ase_fopen (fn, ASE_T("r"));
|
|
}
|
|
break;
|
|
|
|
case Awk::Console::WRITE:
|
|
if (numConOutFiles == 0) fp = stdout;
|
|
else
|
|
{
|
|
fn = conOutFile[0];
|
|
fp = ase_fopen (fn, ASE_T("w"));
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (fp == NULL) return -1;
|
|
|
|
ConTrack* t = (ConTrack*) ase_awk_malloc (awk, ASE_SIZEOF(ConTrack));
|
|
if (t == ASE_NULL)
|
|
{
|
|
if (fp != stdin && fp != stdout) fclose (fp);
|
|
return -1;
|
|
}
|
|
|
|
t->handle = fp;
|
|
t->nextConIdx = 1;
|
|
|
|
if (fn != ASE_NULL)
|
|
{
|
|
if (io.setFileName (fn) == -1)
|
|
{
|
|
if (fp != stdin && fp != stdout) fclose (fp);
|
|
ase_awk_free (awk, t);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
io.setHandle (t);
|
|
return 1;
|
|
}
|
|
|
|
int closeConsole (Console& io)
|
|
{
|
|
ConTrack* t = (ConTrack*)io.getHandle();
|
|
FILE* fp = t->handle;
|
|
|
|
if (fp != stdin && fp != stdout && fp != stderr) fclose (fp);
|
|
|
|
ase_awk_free (awk, t);
|
|
return 0;
|
|
}
|
|
|
|
ssize_t readConsole (Console& io, char_t* buf, size_t len)
|
|
{
|
|
ConTrack* t = (ConTrack*)io.getHandle();
|
|
FILE* fp = t->handle;
|
|
ssize_t n = 0;
|
|
|
|
while (n < len)
|
|
{
|
|
ase_cint_t c = ase_fgetc (fp);
|
|
if (c == ASE_CHAR_EOF) break;
|
|
|
|
buf[n++] = c;
|
|
if (c == ASE_T('\n')) break;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
ssize_t writeConsole (Console& io, char_t* buf, size_t len)
|
|
{
|
|
ConTrack* t = (ConTrack*)io.getHandle();
|
|
FILE* fp = t->handle;
|
|
size_t left = len;
|
|
|
|
while (left > 0)
|
|
{
|
|
if (*buf == ASE_T('\0'))
|
|
{
|
|
if (ase_fputc (*buf, fp) == ASE_CHAR_EOF) return -1;
|
|
left -= 1; buf += 1;
|
|
}
|
|
else
|
|
{
|
|
int n = ase_fprintf (fp, ASE_T("%.*s"), left, buf);
|
|
if (n < 0) return -1;
|
|
left -= n; buf += n;
|
|
}
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
int flushConsole (Console& io)
|
|
{
|
|
ConTrack* t = (ConTrack*)io.getHandle();
|
|
FILE* fp = t->handle;
|
|
return ::fflush (fp);
|
|
}
|
|
|
|
int nextConsole (Console& io)
|
|
{
|
|
Awk::Console::Mode mode = io.getMode();
|
|
ConTrack* t = (ConTrack*)io.getHandle();
|
|
FILE* ofp = t->handle;
|
|
FILE* nfp = ASE_NULL;
|
|
const char_t* fn = ASE_NULL;
|
|
|
|
switch (mode)
|
|
{
|
|
case Awk::Console::READ:
|
|
if (t->nextConIdx >= numConInFiles) return 0;
|
|
fn = conInFile[t->nextConIdx];
|
|
nfp = ase_fopen (fn, ASE_T("r"));
|
|
break;
|
|
|
|
case Awk::Console::WRITE:
|
|
if (t->nextConIdx >= numConOutFiles) return 0;
|
|
fn = conOutFile[t->nextConIdx];
|
|
nfp = ase_fopen (fn, ASE_T("w"));
|
|
break;
|
|
}
|
|
|
|
if (nfp == ASE_NULL) return -1;
|
|
|
|
if (fn != ASE_NULL)
|
|
{
|
|
if (io.setFileName (fn) == -1)
|
|
{
|
|
fclose (nfp);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
fclose (ofp);
|
|
|
|
t->nextConIdx++;
|
|
t->handle = nfp;
|
|
|
|
return 1;
|
|
}
|
|
|
|
void* allocMem (size_t n)
|
|
{
|
|
#ifdef _WIN32
|
|
return ::HeapAlloc (heap, 0, n);
|
|
#else
|
|
return ::malloc (n);
|
|
#endif
|
|
}
|
|
|
|
void* reallocMem (void* ptr, size_t n)
|
|
{
|
|
#ifdef _WIN32
|
|
if (ptr == NULL)
|
|
return ::HeapAlloc (heap, 0, n);
|
|
else
|
|
return ::HeapReAlloc (heap, 0, ptr, n);
|
|
#else
|
|
return ::realloc (ptr, n);
|
|
#endif
|
|
}
|
|
|
|
void freeMem (void* ptr)
|
|
{
|
|
#ifdef _WIN32
|
|
::HeapFree (heap, 0, ptr);
|
|
#else
|
|
::free (ptr);
|
|
#endif
|
|
}
|
|
|
|
private:
|
|
const char_t* srcInName;
|
|
const char_t* srcOutName;
|
|
|
|
struct ConTrack
|
|
{
|
|
FILE* handle;
|
|
size_t nextConIdx;
|
|
};
|
|
|
|
size_t numConInFiles;
|
|
const char_t* conInFile[128];
|
|
|
|
size_t numConOutFiles;
|
|
const char_t* conOutFile[128];
|
|
|
|
#ifdef _WIN32
|
|
void* heap;
|
|
#endif
|
|
};
|
|
|
|
#ifndef NDEBUG
|
|
void ase_assert_abort (void)
|
|
{
|
|
abort ();
|
|
}
|
|
|
|
void ase_assert_printf (const ase_char_t* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
#ifdef _WIN32
|
|
int n;
|
|
ase_char_t buf[1024];
|
|
#endif
|
|
|
|
va_start (ap, fmt);
|
|
#if defined(_WIN32)
|
|
n = _vsntprintf (buf, ASE_COUNTOF(buf), fmt, ap);
|
|
if (n < 0) buf[ASE_COUNTOF(buf)-1] = ASE_T('\0');
|
|
|
|
#if defined(_MSC_VER) && (_MSC_VER<1400)
|
|
MessageBox (NULL, buf,
|
|
ASE_T("Assertion Failure"), MB_OK|MB_ICONERROR);
|
|
#else
|
|
MessageBox (NULL, buf,
|
|
ASE_T("\uB2DD\uAE30\uB9AC \uC870\uB610"), MB_OK|MB_ICONERROR);
|
|
#endif
|
|
#else
|
|
ase_vprintf (fmt, ap);
|
|
#endif
|
|
va_end (ap);
|
|
}
|
|
#endif
|
|
|
|
static void print_error (const ase_char_t* msg)
|
|
{
|
|
ase_printf (ASE_T("Error: %s\n"), msg);
|
|
}
|
|
|
|
static void print_usage (const ase_char_t* argv0)
|
|
{
|
|
ase_printf (ASE_T("Usage: %s [-m main-function] [-si source-in-file] [-so source-out-file] [-ci console-in-file]* [-co console-out-file]* [-a argument]*\n"), argv0);
|
|
}
|
|
|
|
int awk_main (int argc, ase_char_t* argv[])
|
|
{
|
|
TestAwk awk;
|
|
int mode = 0;
|
|
const ase_char_t* mainfn = NULL;
|
|
const ase_char_t* srcin = ASE_T("");
|
|
const ase_char_t* srcout = NULL;
|
|
const ase_char_t* args[256];
|
|
ase_size_t nargs = 0;
|
|
ase_size_t nsrcins = 0;
|
|
ase_size_t nsrcouts = 0;
|
|
|
|
for (int i = 1; i < argc; i++)
|
|
{
|
|
if (mode == 0)
|
|
{
|
|
if (ase_strcmp(argv[i], ASE_T("-si")) == 0) mode = 1;
|
|
else if (ase_strcmp(argv[i], ASE_T("-so")) == 0) mode = 2;
|
|
else if (ase_strcmp(argv[i], ASE_T("-ci")) == 0) mode = 3;
|
|
else if (ase_strcmp(argv[i], ASE_T("-co")) == 0) mode = 4;
|
|
else if (ase_strcmp(argv[i], ASE_T("-a")) == 0) mode = 5;
|
|
else if (ase_strcmp(argv[i], ASE_T("-m")) == 0) mode = 6;
|
|
else
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (argv[i][0] == ASE_T('-'))
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
if (mode == 1) // source input
|
|
{
|
|
if (nsrcins != 0)
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
srcin = argv[i];
|
|
nsrcins++;
|
|
mode = 0;
|
|
}
|
|
else if (mode == 2) // source output
|
|
{
|
|
if (nsrcouts != 0)
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
srcout = argv[i];
|
|
nsrcouts++;
|
|
mode = 0;
|
|
}
|
|
else if (mode == 3) // console input
|
|
{
|
|
if (awk.addConsoleInput (argv[i]) == -1)
|
|
{
|
|
print_error (ASE_T("too many console inputs"));
|
|
return -1;
|
|
}
|
|
|
|
mode = 0;
|
|
}
|
|
else if (mode == 4) // console output
|
|
{
|
|
if (awk.addConsoleOutput (argv[i]) == -1)
|
|
{
|
|
print_error (ASE_T("too many console outputs"));
|
|
return -1;
|
|
}
|
|
|
|
mode = 0;
|
|
}
|
|
else if (mode == 5) // argument mode
|
|
{
|
|
if (nargs >= ASE_COUNTOF(args))
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
args[nargs++] = argv[i];
|
|
mode = 0;
|
|
}
|
|
else if (mode == 6)
|
|
{
|
|
if (mainfn != NULL)
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
mainfn = argv[i];
|
|
mode = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (mode != 0)
|
|
{
|
|
print_usage (argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
if (awk.open() == -1)
|
|
{
|
|
ase_fprintf (stderr, ASE_T("cannot open awk\n"));
|
|
return -1;
|
|
}
|
|
|
|
if (awk.parse (srcin, srcout) == -1)
|
|
{
|
|
ase_fprintf (stderr, ASE_T("cannot parse\n"));
|
|
return -1;
|
|
}
|
|
|
|
if (awk.run (mainfn, args, nargs) == -1)
|
|
{
|
|
ase_fprintf (stderr, ASE_T("cannot run\n"));
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int ase_main (int argc, ase_achar_t* argv[])
|
|
{
|
|
|
|
int n;
|
|
|
|
#if defined(__linux) && defined(_DEBUG)
|
|
mtrace ();
|
|
#endif
|
|
|
|
n = ase_runmain (argc,argv,awk_main);
|
|
|
|
#if defined(__linux) && defined(_DEBUG)
|
|
muntrace ();
|
|
#endif
|
|
#if defined(_WIN32) && defined(_DEBUG)
|
|
#if defined(_MSC_VER)
|
|
_CrtDumpMemoryLeaks ();
|
|
#endif
|
|
_tprintf (_T("Press ENTER to quit\n"));
|
|
getchar ();
|
|
#endif
|
|
|
|
return n;
|
|
}
|