/* * $Id: awk.c,v 1.177 2007-02-23 08:54:03 bacon Exp $ */ #include #include #include #include #include #include #include #include #include #include #include #if defined(_WIN32) #include #include #pragma warning (disable: 4996) #pragma warning (disable: 4296) #elif defined(ASE_CHAR_IS_MCHAR) #include #include #else #include #include #include #endif #if defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG) #define _CRTDBG_MAP_ALLOC #include #endif #if defined(__linux) && defined(_DEBUG) #include #endif struct awk_src_io { const ase_char_t* input_file; FILE* input_handle; }; #if defined(vms) || defined(__vms) /* it seems that the main function should be placed in the main object file * in OpenVMS. otherwise, the first function in the main object file seems * to become the main function resulting in program start-up failure. */ #include #endif static ase_real_t awk_pow (ase_real_t x, ase_real_t y) { return pow (x, y); } static void awk_abort (void* custom_data) { abort (); } static void awk_aprintf (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); } static void awk_dprintf (const ase_char_t* fmt, ...) { va_list ap; va_start (ap, fmt); ase_vfprintf (stderr, fmt, ap); va_end (ap); } static ase_ssize_t awk_srcio_in ( int cmd, void* arg, ase_char_t* data, ase_size_t size) { struct awk_src_io* src_io = (struct awk_src_io*)arg; ase_cint_t c; if (cmd == ASE_AWK_IO_OPEN) { if (src_io->input_file == ASE_NULL) return 0; src_io->input_handle = ase_fopen (src_io->input_file, ASE_T("r")); if (src_io->input_handle == NULL) return -1; return 1; } else if (cmd == ASE_AWK_IO_CLOSE) { if (src_io->input_file == ASE_NULL) return 0; fclose ((FILE*)src_io->input_handle); return 0; } else if (cmd == ASE_AWK_IO_READ) { if (size <= 0) return -1; c = ase_fgetc ((FILE*)src_io->input_handle); if (c == ASE_CHAR_EOF) return 0; *data = (ase_char_t)c; return 1; } return -1; } static ase_ssize_t awk_srcio_out ( int cmd, void* arg, ase_char_t* data, ase_size_t size) { /*struct awk_src_io* src_io = (struct awk_src_io*)arg;*/ if (cmd == ASE_AWK_IO_OPEN) return 1; else if (cmd == ASE_AWK_IO_CLOSE) { fflush (stdout); return 0; } else if (cmd == ASE_AWK_IO_WRITE) { int n = ase_fprintf (stdout, ASE_T("%.*s"), size, data); if (n < 0) return -1; return size; } return -1; } static ase_ssize_t awk_extio_pipe ( int cmd, void* arg, ase_char_t* data, ase_size_t size) { ase_awk_extio_t* epa = (ase_awk_extio_t*)arg; switch (cmd) { case ASE_AWK_IO_OPEN: { FILE* handle; const ase_char_t* mode; if (epa->mode == ASE_AWK_EXTIO_PIPE_READ) mode = ASE_T("r"); else if (epa->mode == ASE_AWK_EXTIO_PIPE_WRITE) mode = ASE_T("w"); else return -1; /* TODO: any way to set the error number? */ awk_dprintf (ASE_T("opening %s of type %d (pipe)\n"), epa->name, epa->type); handle = ase_popen (epa->name, mode); if (handle == NULL) return -1; epa->handle = (void*)handle; return 1; } case ASE_AWK_IO_CLOSE: { awk_dprintf (ASE_T("closing %s of type (pipe) %d\n"), epa->name, epa->type); fclose ((FILE*)epa->handle); epa->handle = NULL; return 0; } case ASE_AWK_IO_READ: { if (ase_fgets (data, size, (FILE*)epa->handle) == ASE_NULL) { if (ferror((FILE*)epa->handle)) return -1; return 0; } return ase_strlen(data); } case ASE_AWK_IO_WRITE: { #if defined(ASE_CHAR_IS_WCHAR) && defined(__linux) /* fwprintf seems to return an error with the file * pointer opened by popen, as of this writing. * anyway, hopefully the following replacement * will work all the way. */ int n = fprintf ( (FILE*)epa->handle, "%.*ls", size, data); #else int n = ase_fprintf ( (FILE*)epa->handle, ASE_T("%.*s"), size, data); #endif if (n < 0) return -1; return size; } case ASE_AWK_IO_FLUSH: { if (epa->mode == ASE_AWK_EXTIO_PIPE_READ) return -1; else return 0; } case ASE_AWK_IO_NEXT: { return -1; } } return -1; } static ase_ssize_t awk_extio_file ( int cmd, void* arg, ase_char_t* data, ase_size_t size) { ase_awk_extio_t* epa = (ase_awk_extio_t*)arg; switch (cmd) { case ASE_AWK_IO_OPEN: { FILE* handle; const ase_char_t* mode; if (epa->mode == ASE_AWK_EXTIO_FILE_READ) mode = ASE_T("r"); else if (epa->mode == ASE_AWK_EXTIO_FILE_WRITE) mode = ASE_T("w"); else if (epa->mode == ASE_AWK_EXTIO_FILE_APPEND) mode = ASE_T("a"); else return -1; /* TODO: any way to set the error number? */ awk_dprintf (ASE_T("opening %s of type %d (file)\n"), epa->name, epa->type); handle = ase_fopen (epa->name, mode); if (handle == NULL) return -1; epa->handle = (void*)handle; return 1; } case ASE_AWK_IO_CLOSE: { awk_dprintf (ASE_T("closing %s of type %d (file)\n"), epa->name, epa->type); fclose ((FILE*)epa->handle); epa->handle = NULL; return 0; } case ASE_AWK_IO_READ: { if (ase_fgets (data, size, (FILE*)epa->handle) == ASE_NULL) { if (ferror((FILE*)epa->handle)) return -1; return 0; } return ase_strlen(data); } case ASE_AWK_IO_WRITE: { int n = ase_fprintf ( (FILE*)epa->handle, ASE_T("%.*s"), size, data); if (n < 0) return -1; return size; } case ASE_AWK_IO_FLUSH: { if (fflush ((FILE*)epa->handle) == EOF) return -1; return 0; } case ASE_AWK_IO_NEXT: { return -1; } } return -1; } static int open_extio_console (ase_awk_extio_t* epa); static int close_extio_console (ase_awk_extio_t* epa); static int next_extio_console (ase_awk_extio_t* epa); static ase_size_t infile_no = 0; static const ase_char_t* infiles[1000] = { ASE_T(""), ASE_NULL }; static ase_ssize_t awk_extio_console ( int cmd, void* arg, ase_char_t* data, ase_size_t size) { ase_awk_extio_t* epa = (ase_awk_extio_t*)arg; if (cmd == ASE_AWK_IO_OPEN) { return open_extio_console (epa); } else if (cmd == ASE_AWK_IO_CLOSE) { return close_extio_console (epa); } else if (cmd == ASE_AWK_IO_READ) { while (ase_fgets (data, size, (FILE*)epa->handle) == ASE_NULL) { if (ferror((FILE*)epa->handle)) return -1; /* it has reached the end of the current file. * open the next file if available */ if (infiles[infile_no] == ASE_NULL) { /* no more input console */ /* is this correct??? */ /* if (epa->handle != ASE_NULL && epa->handle != stdin && epa->handle != stdout && epa->handle != stderr) { fclose ((FILE*)epa->handle); } epa->handle = ASE_NULL; */ return 0; } if (infiles[infile_no][0] == ASE_T('\0')) { if (epa->handle != ASE_NULL && epa->handle != stdin && epa->handle != stdout && epa->handle != stderr) { fclose ((FILE*)epa->handle); } epa->handle = stdin; } else { FILE* fp = ase_fopen (infiles[infile_no], ASE_T("r")); if (fp == ASE_NULL) { awk_dprintf (ASE_T("failed to open the next console of type %x - fopen failure\n"), epa->type); return -1; } if (epa->handle != ASE_NULL && epa->handle != stdin && epa->handle != stdout && epa->handle != stderr) { fclose ((FILE*)epa->handle); } awk_dprintf (ASE_T("open the next console [%s]\n"), infiles[infile_no]); epa->handle = fp; } infile_no++; } return ase_strlen(data); } else if (cmd == ASE_AWK_IO_WRITE) { int n = ase_fprintf ( (FILE*)epa->handle, ASE_T("%.*s"), size, data); if (n < 0) return -1; return size; } else if (cmd == ASE_AWK_IO_FLUSH) { if (fflush ((FILE*)epa->handle) == EOF) return -1; return 0; } else if (cmd == ASE_AWK_IO_NEXT) { return next_extio_console (epa); } return -1; } static int open_extio_console (ase_awk_extio_t* epa) { /* TODO: OpenConsole in GUI APPLICATION */ awk_dprintf (ASE_T("opening console[%s] of type %x\n"), epa->name, epa->type); if (epa->mode == ASE_AWK_EXTIO_CONSOLE_READ) { if (infiles[infile_no] == ASE_NULL) { /* no more input file */ awk_dprintf (ASE_T("console - no more file\n"));; return 0; } if (infiles[infile_no][0] == ASE_T('\0')) { awk_dprintf (ASE_T(" console(r) - \n")); epa->handle = stdin; } else { /* a temporary variable fp is used here not to change * any fields of epa when the open operation fails */ FILE* fp = ase_fopen (infiles[infile_no], ASE_T("r")); if (fp == ASE_NULL) { awk_dprintf (ASE_T("cannot open console of type %x - fopen failure\n"), epa->type); return -1; } awk_dprintf (ASE_T(" console(r) - %s\n"), infiles[infile_no]); if (ase_awk_setfilename ( epa->run, infiles[infile_no], ase_strlen(infiles[infile_no])) == -1) { fclose (fp); return -1; } epa->handle = fp; } infile_no++; return 1; } else if (epa->mode == ASE_AWK_EXTIO_CONSOLE_WRITE) { awk_dprintf (ASE_T(" console(w) - \n")); /* TODO: does output console has a name??? */ /*ase_awk_setconsolename (ASE_T(""));*/ epa->handle = stdout; return 1; } return -1; } static int close_extio_console (ase_awk_extio_t* epa) { awk_dprintf (ASE_T("closing console of type %x\n"), epa->type); if (epa->handle != ASE_NULL && epa->handle != stdin && epa->handle != stdout && epa->handle != stderr) { fclose ((FILE*)epa->handle); } /* TODO: CloseConsole in GUI APPLICATION */ return 0; } static int next_extio_console (ase_awk_extio_t* epa) { int n; FILE* fp = (FILE*)epa->handle; awk_dprintf (ASE_T("switching console[%s] of type %x\n"), epa->name, epa->type); n = open_extio_console(epa); if (n == -1) return -1; if (n == 0) { /* if there is no more file, keep the previous handle */ return 0; } if (fp != ASE_NULL && fp != stdin && fp != stdout && fp != stderr) fclose (fp); return n; } ase_awk_t* app_awk = NULL; ase_awk_run_t* app_run = NULL; #ifdef _WIN32 static BOOL WINAPI stop_run (DWORD ctrl_type) { if (ctrl_type == CTRL_C_EVENT || ctrl_type == CTRL_CLOSE_EVENT) { ase_awk_stop (ase_awk_getrunawk(app_run), app_run); return TRUE; } return FALSE; } #else static void stop_run (int sig) { signal (SIGINT, SIG_IGN); ase_awk_stop (ase_awk_getrunawk(app_run), app_run); /*ase_awk_stopall (app_awk); */ /*ase_awk_stopall (ase_awk_getrunawk(app_run)); */ signal (SIGINT, stop_run); } #endif static void on_run_start (ase_awk_run_t* run, void* custom_data) { app_run = run; awk_dprintf (ASE_T("[AWK ABOUT TO START]\n")); } static int __printval (ase_awk_pair_t* pair, void* arg) { ase_awk_run_t* run = (ase_awk_run_t*)arg; awk_dprintf (ASE_T("%s = "), (const ase_char_t*)pair->key); ase_awk_dprintval (run, (ase_awk_val_t*)pair->val); awk_dprintf (ASE_T("\n")); return 0; } static void on_run_return ( ase_awk_run_t* run, ase_awk_val_t* ret, void* custom_data) { app_run = run; awk_dprintf (ASE_T("[RETURN] - ")); ase_awk_dprintval (run, ret); awk_dprintf (ASE_T("\n")); awk_dprintf (ASE_T("[NAMED VARIABLES]\n")); ase_awk_map_walk (ase_awk_getrunnamedvarmap(run), __printval, run); awk_dprintf (ASE_T("[END NAMED VARIABLES]\n")); } static void on_run_end (ase_awk_run_t* run, int errnum, void* custom_data) { if (errnum != ASE_AWK_ENOERR) { awk_dprintf (ASE_T("[AWK ENDED WITH AN ERROR] - ")); awk_dprintf (ASE_T("CODE [%d] LINE [%u] %s\n"), errnum, (unsigned int)ase_awk_getrunerrlin(run), ase_awk_getrunerrmsg(run)); } else awk_dprintf (ASE_T("[AWK ENDED SUCCESSFULLY]\n")); app_run = NULL; } #ifdef _WIN32 typedef struct prmfns_data_t prmfns_data_t; struct prmfns_data_t { HANDLE heap; }; #endif static void* awk_malloc (ase_mmgr_t* mmgr, ase_size_t n) { #ifdef _WIN32 return HeapAlloc (((prmfns_data_t*)mmgr->custom_data)->heap, 0, n); #else return malloc (n); #endif } static void* awk_realloc (ase_mmgr_t* mmgr, void* ptr, ase_size_t n) { #ifdef _WIN32 /* HeapReAlloc behaves differently from realloc */ if (ptr == NULL) return HeapAlloc (((prmfns_data_t*)mmgr->custom_data)->heap, 0, n); else return HeapReAlloc (((prmfns_data_t*)mmgr->custom_data)->heap, 0, ptr, n); #else return realloc (ptr, n); #endif } static void awk_free (ase_mmgr_t* mmgr, void* ptr) { #ifdef _WIN32 HeapFree (((prmfns_data_t*)mmgr->custom_data)->heap, 0, ptr); #else free (ptr); #endif } static void print_usage (const ase_char_t* argv0) { ase_printf (ASE_T("Usage: %s [-m] [-d] [-a argument]* -f source-file [data-file]*\n"), argv0); } static int awk_main (int argc, ase_char_t* argv[]) { ase_awk_t* awk; ase_awk_srcios_t srcios; ase_awk_runcbs_t runcbs; ase_awk_runios_t runios; ase_awk_prmfns_t prmfns; struct awk_src_io src_io = { NULL, NULL }; int opt, i, file_count = 0, errnum; #ifdef _WIN32 prmfns_data_t prmfns_data; #endif const ase_char_t* mfn = ASE_NULL; int mode = 0; int runarg_count = 0; ase_awk_runarg_t runarg[128]; int deparse = 0; opt = ASE_AWK_IMPLICIT | ASE_AWK_EXPLICIT | ASE_AWK_UNIQUEFN | ASE_AWK_IDIV | ASE_AWK_SHADING | ASE_AWK_SHIFT | ASE_AWK_EXTIO | /*ASE_AWK_COPROC |*/ ASE_AWK_BLOCKLESS | ASE_AWK_STRBASEONE | ASE_AWK_STRIPSPACES | ASE_AWK_NEXTOFILE; if (argc <= 1) { print_usage (argv[0]); return -1; } for (i = 1; i < argc; i++) { if (mode == 0) { if (ase_strcmp(argv[i], ASE_T("-m")) == 0) { mfn = ASE_T("main"); } else if (ase_strcmp(argv[i], ASE_T("-d")) == 0) { deparse = 1; } else if (ase_strcmp(argv[i], ASE_T("-f")) == 0) { /* specify source file */ mode = 1; } else if (ase_strcmp(argv[i], ASE_T("-a")) == 0) { /* specify arguments */ mode = 2; } else if (argv[i][0] == ASE_T('-')) { print_usage (argv[0]); return -1; } else if (file_count < ASE_COUNTOF(infiles)-1) { infiles[file_count] = argv[i]; file_count++; } else { print_usage (argv[0]); return -1; } } else if (mode == 1) /* source mode */ { if (argv[i][0] == ASE_T('-')) { print_usage (argv[0]); return -1; } if (src_io.input_file != NULL) { print_usage (argv[0]); return -1; } src_io.input_file = argv[i]; mode = 0; } else if (mode == 2) /* argument mode */ { if (argv[i][0] == ASE_T('-')) { print_usage (argv[0]); return -1; } if (runarg_count >= ASE_COUNTOF(runarg)-1) { print_usage (argv[0]); return -1; } runarg[runarg_count].ptr = argv[i]; runarg[runarg_count].len = ase_strlen(argv[i]); runarg_count++; mode = 0; } } infiles[file_count] = ASE_NULL; runarg[runarg_count].ptr = NULL; runarg[runarg_count].len = 0; if (mode != 0 || src_io.input_file == NULL) { print_usage (argv[0]); return -1; } memset (&prmfns, 0, ASE_SIZEOF(prmfns)); prmfns.mmgr.malloc = awk_malloc; prmfns.mmgr.realloc = awk_realloc; prmfns.mmgr.free = awk_free; #ifdef _WIN32 prmfns_data.heap = HeapCreate (0, 1000000, 1000000); if (prmfns_data.heap == NULL) { ase_printf (ASE_T("Error: cannot create an awk heap\n")); return -1; } prmfns.mmgr.custom_data = &prmfns_data; #else prmfns.mmgr.custom_data = NULL; #endif prmfns.ccls.is_upper = ase_isupper; prmfns.ccls.is_lower = ase_islower; prmfns.ccls.is_alpha = ase_isalpha; prmfns.ccls.is_digit = ase_isdigit; prmfns.ccls.is_xdigit = ase_isxdigit; prmfns.ccls.is_alnum = ase_isalnum; prmfns.ccls.is_space = ase_isspace; prmfns.ccls.is_print = ase_isprint; prmfns.ccls.is_graph = ase_isgraph; prmfns.ccls.is_cntrl = ase_iscntrl; prmfns.ccls.is_punct = ase_ispunct; prmfns.ccls.to_upper = ase_toupper; prmfns.ccls.to_lower = ase_tolower; prmfns.ccls.custom_data = NULL; prmfns.misc.pow = awk_pow; prmfns.misc.sprintf = ase_sprintf; prmfns.misc.aprintf = awk_aprintf; prmfns.misc.dprintf = awk_dprintf; prmfns.misc.abort = awk_abort; prmfns.misc.lock = NULL; prmfns.misc.unlock = NULL; prmfns.misc.custom_data = NULL; if ((awk = ase_awk_open(&prmfns, ASE_NULL, &errnum)) == ASE_NULL) { #ifdef _WIN32 HeapDestroy (prmfns_data.heap); #endif ase_printf ( ASE_T("ERROR: cannot open awk [%d] %s\n"), errnum, ase_awk_geterrstr(errnum)); return -1; } app_awk = awk; ase_awk_setoption (awk, opt); srcios.in = awk_srcio_in; srcios.out = deparse? awk_srcio_out: NULL; srcios.custom_data = &src_io; ase_awk_setmaxdepth ( awk, ASE_AWK_DEPTH_BLOCK_PARSE | ASE_AWK_DEPTH_EXPR_PARSE, 20); ase_awk_setmaxdepth ( awk, ASE_AWK_DEPTH_BLOCK_RUN | ASE_AWK_DEPTH_EXPR_RUN, 50); if (ase_awk_parse (awk, &srcios) == -1) { int errnum = ase_awk_geterrnum(awk); ase_printf ( ASE_T("ERROR: cannot parse program - line %u [%d] %s\n"), (unsigned int)ase_awk_geterrlin(awk), errnum, ase_awk_geterrmsg(awk)); ase_awk_close (awk); return -1; } #ifdef _WIN32 SetConsoleCtrlHandler (stop_run, TRUE); #else signal (SIGINT, stop_run); #endif runios.pipe = awk_extio_pipe; runios.coproc = ASE_NULL; runios.file = awk_extio_file; runios.console = awk_extio_console; runcbs.on_start = on_run_start; runcbs.on_return = on_run_return; runcbs.on_end = on_run_end; runcbs.custom_data = ASE_NULL; if (ase_awk_run (awk, mfn, &runios, &runcbs, runarg, ASE_NULL) == -1) { int errnum = ase_awk_geterrnum(awk); ase_printf ( ASE_T("error: cannot run program - [%d] %s\n"), errnum, ase_awk_geterrstr(errnum)); ase_awk_close (awk); return -1; } ase_awk_close (awk); #ifdef _WIN32 HeapDestroy (prmfns_data.heap); #endif return 0; } int ase_main (int argc, ase_char_t* argv[]) { int n; #if defined(__linux) && defined(_DEBUG) mtrace (); #endif /*#if defined(_WIN32) && defined(_MSC_VER) && defined(_DEBUG) _CrtSetDbgFlag (_CRTDBG_LEAK_CHECK_DF | _CRTDBG_ALLOC_MEM_DF | _CRTDBG_CHECK_ALWAYS_DF); #endif*/ n = awk_main (argc, argv); #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; }