362 lines
7.4 KiB
C
362 lines
7.4 KiB
C
|
|
/********************************************
|
|
init.c
|
|
copyright 1991, Michael D. Brennan
|
|
|
|
This is a source file for mawk, an implementation of
|
|
the AWK programming language.
|
|
|
|
Mawk is distributed without warranty under the terms of
|
|
the GNU General Public License, version 2, 1991.
|
|
********************************************/
|
|
|
|
|
|
/* $Log: init.c,v $
|
|
/* Revision 1.2 1993/07/02 23:57:29 jtc
|
|
/* Updated to mawk 1.1.4
|
|
/*
|
|
* Revision 5.4 1992/12/24 01:58:19 mike
|
|
* 1.1.2d changes for MsDOS
|
|
*
|
|
* Revision 5.3 1992/07/10 16:17:10 brennan
|
|
* MsDOS: remove NO_BINMODE macro
|
|
*
|
|
* Revision 5.2 1992/01/09 08:46:14 brennan
|
|
* small change for MSC
|
|
*
|
|
* Revision 5.1 91/12/05 07:56:07 brennan
|
|
* 1.1 pre-release
|
|
*
|
|
*/
|
|
|
|
|
|
/* init.c */
|
|
#include "mawk.h"
|
|
#include "code.h"
|
|
#include "memory.h"
|
|
#include "symtype.h"
|
|
#include "init.h"
|
|
#include "bi_vars.h"
|
|
#include "field.h"
|
|
|
|
#ifdef THINK_C
|
|
#include <console.h>
|
|
#endif
|
|
|
|
#if MSDOS
|
|
#include <fcntl.h>
|
|
#endif
|
|
|
|
static void PROTO( process_cmdline , (int, char **) ) ;
|
|
static void PROTO( set_ARGV, (int, char **, int)) ;
|
|
static void PROTO( bad_option, (char *)) ;
|
|
|
|
extern void PROTO( print_version, (void) ) ;
|
|
extern int PROTO( is_cmdline_assign, (char*)) ;
|
|
|
|
#if MSDOS
|
|
void PROTO(stdout_init,(void)) ;
|
|
#if HAVE_REARGV
|
|
void PROTO(reargv, (int*,char***)) ;
|
|
#endif
|
|
#endif
|
|
|
|
char *progname ;
|
|
|
|
void initialize(argc, argv)
|
|
int argc ; char **argv ;
|
|
{
|
|
SET_PROGNAME() ;
|
|
|
|
bi_vars_init() ; /* load the builtin variables */
|
|
bi_funct_init() ; /* load the builtin functions */
|
|
kw_init() ; /* load the keywords */
|
|
field_init() ;
|
|
|
|
#ifdef THINK_C
|
|
fputc('\n',stderr); /* (causes Think C console window to appear) */
|
|
SetWTitle( FrontWindow(), "\pMacMAWK" );
|
|
argc = ccommand(&argv);
|
|
#endif
|
|
|
|
#if MSDOS
|
|
{ char *p = getenv("MAWKBINMODE") ;
|
|
|
|
if ( p ) set_binmode( atoi(p) ) ;
|
|
}
|
|
#endif
|
|
|
|
|
|
process_cmdline(argc, argv) ;
|
|
|
|
code_init() ;
|
|
fpe_init() ;
|
|
set_stderr() ;
|
|
|
|
#if MSDOS
|
|
stdout_init() ;
|
|
#endif
|
|
}
|
|
|
|
void compile_cleanup()
|
|
/* program has parsed OK, free some memory
|
|
we don't need anymore */
|
|
{
|
|
scan_cleanup() ;
|
|
code_cleanup() ;
|
|
}
|
|
|
|
int dump_code ; /* if on dump internal code */
|
|
int posix_space_flag ;
|
|
|
|
#ifdef DEBUG
|
|
int dump_RE ; /* if on dump compiled REs */
|
|
#endif
|
|
|
|
|
|
static void bad_option(s)
|
|
char *s ;
|
|
{ errmsg(0, "not an option: %s", s) ; mawk_exit(1) ; }
|
|
|
|
|
|
static void process_cmdline(argc, argv)
|
|
int argc ;
|
|
char **argv ;
|
|
{
|
|
int i , nextarg ;
|
|
char *optarg ;
|
|
PFILE dummy ; /* starts linked list of filenames */
|
|
PFILE *tail = &dummy ;
|
|
|
|
for( i = 1 ; i < argc && argv[i][0] == '-' ; i = nextarg )
|
|
{
|
|
if ( argv[i][1] == 0 ) /* - alone */
|
|
{
|
|
if ( ! pfile_name )
|
|
{ errmsg(0, "no program") ; exit(1) ; }
|
|
break ; /* the for loop */
|
|
}
|
|
/* safe to look at argv[i][2] */
|
|
|
|
if ( argv[i][2] == 0 )
|
|
{
|
|
if ( i == argc-1 && argv[i][1] != '-' )
|
|
{
|
|
if ( strchr("WFvf", argv[i][1]) )
|
|
{ errmsg(0, "option %s lacks argument", argv[i]) ;
|
|
mawk_exit(1) ;
|
|
}
|
|
bad_option(argv[i]) ;
|
|
}
|
|
|
|
optarg = argv[i+1] ;
|
|
nextarg = i+2 ;
|
|
}
|
|
else /* argument glued to option */
|
|
{
|
|
optarg = &argv[i][2] ;
|
|
nextarg = i+1 ;
|
|
}
|
|
|
|
switch( argv[i][1] )
|
|
{
|
|
case 'W' :
|
|
|
|
if ( optarg[0] >= 'a' && optarg[0] <= 'z')
|
|
optarg[0] += 'A' - 'a' ;
|
|
if ( optarg[0] == 'V' ) print_version() ;
|
|
else
|
|
if ( optarg[0] == 'D' )
|
|
{ dump_code = 1 ;
|
|
#if SM_DOS
|
|
errmsg(0, "-W dump option unavailable in small model") ;
|
|
dump_code=0 ;
|
|
#endif
|
|
}
|
|
else
|
|
if ( optarg[0] == 'S' )
|
|
{
|
|
char *p = strchr(optarg,'=') ;
|
|
int x = p ? atoi(p+1) : 0 ;
|
|
|
|
if ( x > SPRINTF_SZ )
|
|
{
|
|
sprintf_buff = (char *) zmalloc(x) ;
|
|
sprintf_limit = sprintf_buff + x ;
|
|
}
|
|
}
|
|
#if MSDOS
|
|
else
|
|
if ( optarg[0] == 'B' )
|
|
{
|
|
char *p = strchr(optarg,'=') ;
|
|
int x = p ? atoi(p+1) : 0 ;
|
|
|
|
set_binmode(x) ;
|
|
}
|
|
#endif
|
|
else
|
|
if ( optarg[0] == 'P' )
|
|
{
|
|
posix_space_flag = 1 ;
|
|
}
|
|
else
|
|
errmsg(0,"vacuous option: -W %s", optarg) ;
|
|
|
|
|
|
break ;
|
|
|
|
case 'v' :
|
|
if ( ! is_cmdline_assign(optarg) )
|
|
{ errmsg(0 , "improper assignment: -v %s" , optarg) ;
|
|
exit(1) ;
|
|
}
|
|
break ;
|
|
|
|
case 'F' :
|
|
|
|
rm_escape(optarg) ; /* recognize escape sequences */
|
|
cell_destroy(FS) ;
|
|
FS->type = C_STRING ;
|
|
FS->ptr = (PTR) new_STRING(optarg) ;
|
|
cast_for_split( cellcpy(&fs_shadow, FS) ) ;
|
|
break ;
|
|
|
|
case '-' :
|
|
if ( argv[i][2] != 0 ) bad_option(argv[i]) ;
|
|
i++ ;
|
|
goto no_more_opts ;
|
|
|
|
case 'f' :
|
|
/* first file goes in pfile_name ; any more go
|
|
on a list */
|
|
if ( ! pfile_name ) pfile_name = optarg ;
|
|
else
|
|
{ tail = tail->link = ZMALLOC(PFILE) ;
|
|
tail->fname = optarg ;
|
|
}
|
|
break ;
|
|
|
|
default : bad_option(argv[i]) ;
|
|
|
|
}
|
|
|
|
i = nextarg ;
|
|
}
|
|
|
|
no_more_opts:
|
|
|
|
tail->link = (PFILE*) 0 ;
|
|
pfile_list = dummy.link ;
|
|
|
|
if ( pfile_name )
|
|
{
|
|
set_ARGV(argc, argv, i) ;
|
|
scan_init( (char *) 0 ) ;
|
|
}
|
|
else /* program on command line */
|
|
{
|
|
#if 0
|
|
#if MSDOS && ! HAVE_REARGV /* prompt for program */
|
|
set_ARGV(argc, argv, i) ;
|
|
emit_prompt() ;
|
|
pfile_name = "CON" ;
|
|
scan_init( (char *) 0 ) ;
|
|
#else /* the real world */
|
|
#endif
|
|
#endif /* 0 */
|
|
if ( i == argc )
|
|
{ errmsg(0, "no program") ; mawk_exit(1) ; }
|
|
set_ARGV(argc, argv, i+1) ;
|
|
|
|
#if MSDOS && ! HAVE_REARGV /* reversed quotes */
|
|
{ char *p ;
|
|
|
|
for( p = argv[i] ; *p ; p++ )
|
|
if ( *p == '\'' ) *p = '\"' ;
|
|
}
|
|
#endif
|
|
scan_init( argv[i]) ;
|
|
/* #endif */
|
|
}
|
|
}
|
|
|
|
|
|
static void set_ARGV(argc, argv, i)
|
|
int argc ; char **argv ;
|
|
int i ; /* argv[i] = ARGV[1] */
|
|
{
|
|
SYMTAB *st_p ;
|
|
CELL argi ;
|
|
register CELL *cp ;
|
|
|
|
st_p = insert( "ARGV" ) ;
|
|
st_p->type = ST_ARRAY ;
|
|
Argv = st_p->stval.array = new_ARRAY() ;
|
|
argi.type = C_DOUBLE ;
|
|
argi.dval = 0.0 ;
|
|
cp = array_find( st_p->stval.array, &argi, CREATE) ;
|
|
cp->type = C_STRING ;
|
|
cp->ptr = (PTR) new_STRING( progname ) ;
|
|
|
|
/* ARGV[0] is set, do the rest
|
|
The type of ARGV[1] ... should be C_MBSTRN
|
|
because the user might enter numbers from the command line */
|
|
|
|
for( argi.dval = 1.0 ; i < argc ; i++, argi.dval += 1.0 )
|
|
{
|
|
cp = array_find( st_p->stval.array, &argi, CREATE) ;
|
|
cp->type = C_MBSTRN ;
|
|
cp->ptr = (PTR) new_STRING( argv[i] ) ;
|
|
}
|
|
ARGC->type = C_DOUBLE ;
|
|
ARGC->dval = argi.dval ;
|
|
}
|
|
|
|
#if 0
|
|
#if MSDOS && ! HAVE_REARGV
|
|
|
|
static void emit_prompt()
|
|
{ static char prompt[] = "mawk> " ;
|
|
int fd = open("CON", O_WRONLY, 0) ;
|
|
|
|
(void) write(fd, prompt, strlen(prompt)) ;
|
|
(void) close(fd) ;
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
|
|
/*----- ENVIRON ----------*/
|
|
|
|
void load_environ( ENV )
|
|
ARRAY ENV ;
|
|
{
|
|
CELL c ;
|
|
#ifndef MSDOS_MSC /* MSC declares it near */
|
|
extern char **environ ;
|
|
#endif
|
|
register char **p = environ ; /* walks environ */
|
|
char *s ; /* looks for the '=' */
|
|
CELL *cp ; /* pts at ENV[&c] */
|
|
|
|
c.type = C_STRING ;
|
|
|
|
while ( *p )
|
|
{
|
|
if ( s = strchr(*p, '=') ) /* shouldn't fail */
|
|
{
|
|
*s = 0 ;
|
|
c.ptr = (PTR) new_STRING(*p) ;
|
|
*s++ = '=' ;
|
|
|
|
cp = array_find(ENV, &c, CREATE) ;
|
|
cp->type = C_MBSTRN ;
|
|
cp->ptr = (PTR) new_STRING(s) ;
|
|
|
|
free_STRING(string(&c)) ;
|
|
}
|
|
p++ ;
|
|
}
|
|
}
|