Added ARGC, ARGV, ENVIRON, PROCINFO to StdAwk. Still pending setioattr and getioattr
This commit is contained in:
parent
02a72ee8e7
commit
1c1cb83bb1
@ -119,6 +119,12 @@ public:
|
||||
void clearConsoleOutputs ();
|
||||
|
||||
protected:
|
||||
int make_additional_globals (Run* run);
|
||||
int build_argcv (Run* run);
|
||||
int build_environ (Run* run);
|
||||
int __build_environ (Run* run, void* envptr);
|
||||
int build_procinfo (Run* run);
|
||||
|
||||
// intrinsic functions
|
||||
int rand (Run& run, Value& ret, const Value* args, size_t nargs,
|
||||
const char_t* name, size_t len);
|
||||
|
||||
@ -1180,7 +1180,7 @@ Awk::Run* Awk::parse (Source& in, Source& out)
|
||||
}
|
||||
|
||||
if (init_runctx() <= -1) return QSE_NULL;
|
||||
return &runctx;
|
||||
return &this->runctx;
|
||||
}
|
||||
|
||||
Awk::Run* Awk::resetRunContext ()
|
||||
@ -1196,18 +1196,18 @@ Awk::Run* Awk::resetRunContext ()
|
||||
|
||||
int Awk::loop (Value* ret)
|
||||
{
|
||||
QSE_ASSERT (awk != QSE_NULL);
|
||||
QSE_ASSERT (runctx.rtx != QSE_NULL);
|
||||
QSE_ASSERT (this->awk != QSE_NULL);
|
||||
QSE_ASSERT (this->runctx.rtx != QSE_NULL);
|
||||
|
||||
val_t* rv = qse_awk_rtx_loop (runctx.rtx);
|
||||
val_t* rv = qse_awk_rtx_loop (this->runctx.rtx);
|
||||
if (rv == QSE_NULL)
|
||||
{
|
||||
retrieveError (&runctx);
|
||||
retrieveError (&this->runctx);
|
||||
return -1;
|
||||
}
|
||||
|
||||
ret->setVal (&runctx, rv);
|
||||
qse_awk_rtx_refdownval (runctx.rtx, rv);
|
||||
ret->setVal (&this->runctx, rv);
|
||||
qse_awk_rtx_refdownval (this->runctx.rtx, rv);
|
||||
|
||||
return 0;
|
||||
}
|
||||
@ -1216,8 +1216,8 @@ int Awk::call (
|
||||
const char_t* name, Value* ret,
|
||||
const Value* args, size_t nargs)
|
||||
{
|
||||
QSE_ASSERT (awk != QSE_NULL);
|
||||
QSE_ASSERT (runctx.rtx != QSE_NULL);
|
||||
QSE_ASSERT (this->awk != QSE_NULL);
|
||||
QSE_ASSERT (this->runctx.rtx != QSE_NULL);
|
||||
|
||||
val_t* buf[16];
|
||||
val_t** ptr = QSE_NULL;
|
||||
@ -1231,8 +1231,8 @@ int Awk::call (
|
||||
awk, QSE_SIZEOF(val_t*) * nargs);
|
||||
if (ptr == QSE_NULL)
|
||||
{
|
||||
runctx.setError (QSE_AWK_ENOMEM);
|
||||
retrieveError (&runctx);
|
||||
this->runctx.setError (QSE_AWK_ENOMEM);
|
||||
retrieveError (&this->runctx);
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
@ -1240,19 +1240,19 @@ int Awk::call (
|
||||
for (size_t i = 0; i < nargs; i++) ptr[i] = (val_t*)args[i];
|
||||
}
|
||||
|
||||
val_t* rv = qse_awk_rtx_call (runctx.rtx, name, ptr, nargs);
|
||||
val_t* rv = qse_awk_rtx_call (this->runctx.rtx, name, ptr, nargs);
|
||||
|
||||
if (ptr != QSE_NULL && ptr != buf) qse_awk_freemem (awk, ptr);
|
||||
|
||||
if (rv == QSE_NULL)
|
||||
{
|
||||
retrieveError (&runctx);
|
||||
retrieveError (&this->runctx);
|
||||
return -1;
|
||||
}
|
||||
|
||||
ret->setVal (&runctx, rv);
|
||||
ret->setVal (&this->runctx, rv);
|
||||
|
||||
qse_awk_rtx_refdownval (runctx.rtx, rv);
|
||||
qse_awk_rtx_refdownval (this->runctx.rtx, rv);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
@ -31,10 +31,33 @@
|
||||
#include <stdlib.h>
|
||||
#include <math.h>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <tchar.h>
|
||||
#if defined(_WIN32)
|
||||
# include <windows.h>
|
||||
# include <tchar.h>
|
||||
#elif defined(__OS2__)
|
||||
# define INCL_DOSPROCESS
|
||||
# define INCL_DOSERRORS
|
||||
# include <os2.h>
|
||||
#elif defined(__DOS__)
|
||||
/* anything ? */
|
||||
#else
|
||||
#include <wchar.h>
|
||||
# include <unistd.h>
|
||||
#endif
|
||||
|
||||
#ifndef QSE_HAVE_CONFIG_H
|
||||
# if defined(_WIN32) || defined(__OS2__) || defined(__DOS__)
|
||||
# define HAVE_POW
|
||||
# define HAVE_FMOD
|
||||
# define HAVE_SIN
|
||||
# define HAVE_COS
|
||||
# define HAVE_TAN
|
||||
# define HAVE_ATAN
|
||||
# define HAVE_ATAN2
|
||||
# define HAVE_LOG
|
||||
# define HAVE_LOG10
|
||||
# define HAVE_EXP
|
||||
# define HAVE_SQRT
|
||||
# endif
|
||||
#endif
|
||||
|
||||
/////////////////////////////////
|
||||
@ -172,11 +195,205 @@ StdAwk::Run* StdAwk::parse (Source& in, Source& out)
|
||||
this->cmgrtab_inited = true;
|
||||
}
|
||||
|
||||
|
||||
if (run && make_additional_globals (run) <= -1) return QSE_NULL;
|
||||
|
||||
return run;
|
||||
}
|
||||
|
||||
int StdAwk::build_argcv (Run* run)
|
||||
{
|
||||
Value argv (run);
|
||||
|
||||
for (size_t i = 0; i < this->runarg.len; i++)
|
||||
{
|
||||
if (argv.setIndexedStr (
|
||||
Value::IntIndex(i),
|
||||
this->runarg.ptr[i].ptr,
|
||||
this->runarg.ptr[i].len) <= -1) return -1;
|
||||
}
|
||||
|
||||
run->setGlobal (this->gbl_argc, (long_t)this->runarg.len);
|
||||
run->setGlobal (this->gbl_argv, argv);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int StdAwk::__build_environ (Run* run, void* envptr)
|
||||
{
|
||||
qse_env_char_t** envarr = (qse_env_char_t**)envptr;
|
||||
Value v_env (run);
|
||||
|
||||
if (envarr)
|
||||
{
|
||||
qse_env_char_t* eq;
|
||||
qse_char_t* kptr, * vptr;
|
||||
qse_size_t klen, count;
|
||||
qse_mmgr_t* mmgr = ((Awk*)*run)->getMmgr();
|
||||
|
||||
for (count = 0; envarr[count]; count++)
|
||||
{
|
||||
#if ((defined(QSE_ENV_CHAR_IS_MCHAR) && defined(QSE_CHAR_IS_MCHAR)) || \
|
||||
(defined(QSE_ENV_CHAR_IS_WCHAR) && defined(QSE_CHAR_IS_WCHAR)))
|
||||
eq = qse_strchr (envarr[count], QSE_T('='));
|
||||
if (eq == QSE_NULL || eq == envarr[count]) continue;
|
||||
|
||||
kptr = envarr[count];
|
||||
klen = eq - envarr[count];
|
||||
vptr = eq + 1;
|
||||
#elif defined(QSE_ENV_CHAR_IS_MCHAR)
|
||||
eq = qse_mbschr (envarr[count], QSE_MT('='));
|
||||
if (eq == QSE_NULL || eq == envarr[count]) continue;
|
||||
|
||||
*eq = QSE_MT('\0');
|
||||
|
||||
kptr = qse_mbstowcsdup (envarr[count], mmgr);
|
||||
vptr = qse_mbstowcsdup (eq + 1, mmgr);
|
||||
if (kptr == QSE_NULL || vptr == QSE_NULL)
|
||||
{
|
||||
if (kptr) QSE_MMGR_FREE (mmgr, kptr);
|
||||
|
||||
/* mbstowcsdup() may fail for invalid encoding.
|
||||
* so setting the error code to ENOMEM may not
|
||||
* be really accurate */
|
||||
setError (QSE_AWK_ENOMEM);
|
||||
return -1;
|
||||
}
|
||||
|
||||
klen = qse_wcslen (kptr);
|
||||
*eq = QSE_MT('=');
|
||||
#else
|
||||
eq = qse_wcschr (envarr[count], QSE_WT('='));
|
||||
if (eq == QSE_NULL || eq == envarr[count]) continue;
|
||||
|
||||
*eq = QSE_WT('\0');
|
||||
|
||||
kptr = qse_wcstombsdup (envarr[count], mmgr);
|
||||
vptr = qse_wcstombsdup (eq + 1, mmgr);
|
||||
if (kptr == QSE_NULL || vptr == QSE_NULL)
|
||||
{
|
||||
if (kptr) QSE_MMGR_FREE (mmgr, kptr);
|
||||
|
||||
/* mbstowcsdup() may fail for invalid encoding.
|
||||
* so setting the error code to ENOMEM may not
|
||||
* be really accurate */
|
||||
setError (QSE_AWK_ENOMEM);
|
||||
return -1;
|
||||
}
|
||||
|
||||
klen = qse_mbslen (kptr);
|
||||
*eq = QSE_WT('=');
|
||||
#endif
|
||||
|
||||
v_env.setIndexedStr (Value::Index (kptr, klen), vptr);
|
||||
|
||||
#if ((defined(QSE_ENV_CHAR_IS_MCHAR) && defined(QSE_CHAR_IS_MCHAR)) || \
|
||||
(defined(QSE_ENV_CHAR_IS_WCHAR) && defined(QSE_CHAR_IS_WCHAR)))
|
||||
/* nothing to do */
|
||||
#else
|
||||
if (vptr) QSE_MMGR_FREE (mmgr, vptr);
|
||||
if (kptr) QSE_MMGR_FREE (mmgr, kptr);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
return run->setGlobal (this->gbl_environ, v_env);
|
||||
}
|
||||
|
||||
int StdAwk::build_environ (Run* run)
|
||||
{
|
||||
qse_env_t env;
|
||||
int xret;
|
||||
|
||||
if (qse_env_init (&env, ((Awk*)*run)->getMmgr(), 1) <= -1)
|
||||
{
|
||||
setError (QSE_AWK_ENOMEM);
|
||||
return -1;
|
||||
}
|
||||
|
||||
xret = __build_environ (run, qse_env_getarr(&env));
|
||||
|
||||
qse_env_fini (&env);
|
||||
return xret;
|
||||
}
|
||||
|
||||
int StdAwk::build_procinfo (Run* run)
|
||||
{
|
||||
static qse_cstr_t names[] =
|
||||
{
|
||||
{ QSE_T("pid"), 3 },
|
||||
{ QSE_T("ppid"), 5 },
|
||||
{ QSE_T("pgrp"), 4 },
|
||||
{ QSE_T("uid"), 3 },
|
||||
{ QSE_T("gid"), 3 },
|
||||
{ QSE_T("euid"), 4 },
|
||||
{ QSE_T("egid"), 4 },
|
||||
{ QSE_T("tid"), 3 }
|
||||
};
|
||||
|
||||
#if defined(__OS2__)
|
||||
PTIB tib;
|
||||
PPIB pib;
|
||||
|
||||
if (DosGetInfoBlocks (&tib, &pib) != NO_ERROR)
|
||||
{
|
||||
tib = QSE_NULL;
|
||||
pib = QSE_NULL;
|
||||
}
|
||||
#endif
|
||||
|
||||
Value v_procinfo (run);
|
||||
|
||||
for (size_t i = 0; i < QSE_COUNTOF(names); i++)
|
||||
{
|
||||
qse_long_t val = -99931; /* -99931 randomly chosen */
|
||||
|
||||
#if defined(_WIN32)
|
||||
switch (i)
|
||||
{
|
||||
case 0: val = GetCurrentProcessId(); break;
|
||||
case 7: val = GetCurrentThreadId(); break;
|
||||
}
|
||||
#elif defined(__OS2__)
|
||||
switch (i)
|
||||
{
|
||||
case 0: if (pib) val = pib->pib_ulpid; break;
|
||||
case 7: if (tib && tib->tib_ptib2) val = tib->tib_ptib2->tib2_ultid; break;
|
||||
}
|
||||
#elif defined(__DOS__)
|
||||
/* TODO: */
|
||||
#else
|
||||
switch (i)
|
||||
{
|
||||
case 0: val = getpid(); break;
|
||||
case 1: val = getppid(); break;
|
||||
case 2: val = getpgrp(); break;
|
||||
case 3: val = getuid(); break;
|
||||
case 4: val = getgid(); break;
|
||||
case 5: val = geteuid(); break;
|
||||
case 6: val = getegid(); break;
|
||||
#if defined(HAVE_GETTID)
|
||||
case 7: val = gettid(); break;
|
||||
#endif
|
||||
}
|
||||
#endif
|
||||
if (val == -99931) continue;
|
||||
|
||||
if (v_procinfo.setIndexedInt (
|
||||
Value::Index (names[i].ptr, names[i].len), val) <= -1) return -1;
|
||||
|
||||
}
|
||||
|
||||
return run->setGlobal (this->gbl_procinfo, v_procinfo);
|
||||
}
|
||||
|
||||
int StdAwk::make_additional_globals (Run* run)
|
||||
{
|
||||
if (build_argcv (run) <= -1 ||
|
||||
build_environ (run) <= -1 ||
|
||||
build_procinfo (run) <= -1) return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int StdAwk::rand (Run& run, Value& ret, const Value* args, size_t nargs,
|
||||
const char_t* name, size_t len)
|
||||
{
|
||||
|
||||
@ -1773,12 +1773,6 @@ static int build_procinfo (qse_awk_rtx_t* rtx, int gbl_id)
|
||||
|
||||
qse_awk_rtx_refupval (rtx, v_info);
|
||||
|
||||
if (qse_awk_rtx_setgbl (rtx, gbl_id, v_info) == -1)
|
||||
{
|
||||
qse_awk_rtx_refdownval (rtx, v_info);
|
||||
return -1;
|
||||
}
|
||||
|
||||
#if defined(__OS2__)
|
||||
if (DosGetInfoBlocks (&tib, &pib) != NO_ERROR)
|
||||
{
|
||||
@ -1850,6 +1844,12 @@ static int build_procinfo (qse_awk_rtx_t* rtx, int gbl_id)
|
||||
}
|
||||
}
|
||||
|
||||
if (qse_awk_rtx_setgbl (rtx, gbl_id, v_info) == -1)
|
||||
{
|
||||
qse_awk_rtx_refdownval (rtx, v_info);
|
||||
return -1;
|
||||
}
|
||||
|
||||
qse_awk_rtx_refdownval (rtx, v_info);
|
||||
return 0;
|
||||
}
|
||||
@ -2418,11 +2418,11 @@ static int add_globals (qse_awk_t* awk)
|
||||
|
||||
static int add_functions (qse_awk_t* awk)
|
||||
{
|
||||
if (qse_awk_addfnc (awk, QSE_T("rand"), 4, 0, 0, 0, QSE_NULL, fnc_rand) <= -1 ||
|
||||
qse_awk_addfnc (awk, QSE_T("srand"), 5, 0, 0, 1, QSE_NULL, fnc_srand) <= -1 ||
|
||||
qse_awk_addfnc (awk, QSE_T("system"), 6, 0, 1, 1, QSE_NULL, fnc_system) <= -1 ||
|
||||
qse_awk_addfnc (awk, QSE_T("time"), 4, 0, 0, 0, QSE_NULL, fnc_time) <= -1 ||
|
||||
qse_awk_addfnc (awk, QSE_T("setioattr"), 9, QSE_AWK_RIO, 3, 3, QSE_NULL, fnc_setioattr) <= -1 ||
|
||||
qse_awk_addfnc (awk, QSE_T("getioattr"), 9, QSE_AWK_RIO, 2, 2, QSE_NULL, fnc_getioattr) <= -1) return -1;
|
||||
if (qse_awk_addfnc (awk, QSE_T("rand"), 4, 0, 0, 0, QSE_NULL, fnc_rand) == QSE_NULL ||
|
||||
qse_awk_addfnc (awk, QSE_T("srand"), 5, 0, 0, 1, QSE_NULL, fnc_srand) == QSE_NULL ||
|
||||
qse_awk_addfnc (awk, QSE_T("system"), 6, 0, 1, 1, QSE_NULL, fnc_system) == QSE_NULL ||
|
||||
qse_awk_addfnc (awk, QSE_T("time"), 4, 0, 0, 0, QSE_NULL, fnc_time) == QSE_NULL ||
|
||||
qse_awk_addfnc (awk, QSE_T("setioattr"), 9, QSE_AWK_RIO, 3, 3, QSE_NULL, fnc_setioattr) == QSE_NULL ||
|
||||
qse_awk_addfnc (awk, QSE_T("getioattr"), 9, QSE_AWK_RIO, 2, 2, QSE_NULL, fnc_getioattr) == QSE_NULL) return -1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user