Return to file.c CVS log

Up to [local] / OpenXM_contrib2 / asir2000 / builtin

Corrected the behavior of load operation.
(CAUTION: may contain critical bugs.)

/*
 * Copyright (c) 1994-2000 FUJITSU LABORATORIES LIMITED 
 * All rights reserved.
 * 
 * FUJITSU LABORATORIES LIMITED ("FLL") hereby grants you a limited,
 * non-exclusive and royalty-free license to use, copy, modify and
 * redistribute, solely for non-commercial and non-profit purposes, the
 * computer program, "Risa/Asir" ("SOFTWARE"), subject to the terms and
 * conditions of this Agreement. For the avoidance of doubt, you acquire
 * only a limited right to use the SOFTWARE hereunder, and FLL or any
 * third party developer retains all rights, including but not limited to
 * copyrights, in and to the SOFTWARE.
 * 
 * (1) FLL does not grant you a license in any way for commercial
 * purposes. You may use the SOFTWARE only for non-commercial and
 * non-profit purposes only, such as academic, research and internal
 * business use.
 * (2) The SOFTWARE is protected by the Copyright Law of Japan and
 * international copyright treaties. If you make copies of the SOFTWARE,
 * with or without modification, as permitted hereunder, you shall affix
 * to all such copies of the SOFTWARE the above copyright notice.
 * (3) An explicit reference to this SOFTWARE and its copyright owner
 * shall be made on your publication or presentation in any form of the
 * results obtained by use of the SOFTWARE.
 * (4) In the event that you modify the SOFTWARE, you shall notify FLL by
 * e-mail at risa-admin@sec.flab.fujitsu.co.jp of the detailed specification
 * for such modification or the source code of the modified part of the
 * SOFTWARE.
 * 
 * THE SOFTWARE IS PROVIDED AS IS WITHOUT ANY WARRANTY OF ANY KIND. FLL
 * MAKES ABSOLUTELY NO WARRANTIES, EXPRESSED, IMPLIED OR STATUTORY, AND
 * EXPRESSLY DISCLAIMS ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE OR NONINFRINGEMENT OF THIRD PARTIES'
 * RIGHTS. NO FLL DEALER, AGENT, EMPLOYEES IS AUTHORIZED TO MAKE ANY
 * MODIFICATIONS, EXTENSIONS, OR ADDITIONS TO THIS WARRANTY.
 * UNDER NO CIRCUMSTANCES AND UNDER NO LEGAL THEORY, TORT, CONTRACT,
 * OR OTHERWISE, SHALL FLL BE LIABLE TO YOU OR ANY OTHER PERSON FOR ANY
 * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, PUNITIVE OR CONSEQUENTIAL
 * DAMAGES OF ANY CHARACTER, INCLUDING, WITHOUT LIMITATION, DAMAGES
 * ARISING OUT OF OR RELATING TO THE SOFTWARE OR THIS AGREEMENT, DAMAGES
 * FOR LOSS OF GOODWILL, WORK STOPPAGE, OR LOSS OF DATA, OR FOR ANY
 * DAMAGES, EVEN IF FLL SHALL HAVE BEEN INFORMED OF THE POSSIBILITY OF
 * SUCH DAMAGES, OR FOR ANY CLAIM BY ANY OTHER PARTY. EVEN IF A PART
 * OF THE SOFTWARE HAS BEEN DEVELOPED BY A THIRD PARTY, THE THIRD PARTY
 * DEVELOPER SHALL HAVE NO LIABILITY IN CONNECTION WITH THE USE,
 * PERFORMANCE OR NON-PERFORMANCE OF THE SOFTWARE.
 *
 * $OpenXM: OpenXM_contrib2/asir2000/builtin/file.c,v 1.22 2006/02/08 02:11:19 noro Exp $ 
*/
#include "ca.h"
#include "parse.h"
#include "ox.h"
#include "base.h"
#if !defined(VISUAL)
#include "unistd.h"
#endif
#if defined(PARI)
#include "genpari.h"
#endif

#if defined(VISUAL)
#include <windows.h>
#include <process.h>
#include <io.h>
/* #define ECGEN_KEYNAME "SoftWare\\Fujitsu\\WinECgen\\1.00.000" */
#define ECGEN_KEYNAME "SoftWare\\Fujitsu\\FSEcParamGen\\V1.0L10"
#define ASIR_KEYNAME "SoftWare\\Fujitsu\\Asir\\1999.03.31"
#endif

void Pget_rootdir();
void Paccess(),Premove_file();
void Pbsave_enc(), Pbload_enc();

void Pload(), Pwhich(), Ploadfiles(), Poutput();
void Pbsave(), Pbload(), Pbload27();
void Pbsave_compat(), Pbload_compat();
void Pbsave_cmo(), Pbload_cmo();
void Popen_file(), Pclose_file(), Pget_line(), Pget_byte(), Pput_byte();
void Ppurge_stdin();
void Pimport();

extern int des_encryption;
extern char *asir_libdir;

struct ftab file_tab[] = {
	{"purge_stdin",Ppurge_stdin,0},
	{"open_file",Popen_file,-2},
	{"close_file",Pclose_file,1},
	{"get_byte",Pget_byte,1},
	{"put_byte",Pput_byte,2},
	{"get_line",Pget_line,-1},
	{"remove_file",Premove_file,1},
	{"access",Paccess,1},
	{"load",Pload,-1},
	{"import",Pimport,-1},
	{"which",Pwhich,1},
	{"loadfiles",Ploadfiles,1},
	{"output",Poutput,-1},
	{"bsave",Pbsave,2},
	{"bload",Pbload,1},
	{"get_rootdir",Pget_rootdir,0},
#if defined(VISUAL) && defined(DES_ENC)
	{"bsave_enc",Pbsave_enc,2},
	{"bload_enc",Pbload_enc,1},
#endif
	{"bload27",Pbload27,1},
	{"bsave_compat",Pbsave_compat,2},
	{"bload_compat",Pbload_compat,1},
	{"bsave_cmo",Pbsave_cmo,2},
	{"bload_cmo",Pbload_cmo,1},
	{0,0,0},
};

static FILE *file_ptrs[BUFSIZ];

void Ppurge_stdin(Q *rp)
{
	purge_stdin(stdin);
	*rp = 0;
}

void Popen_file(NODE arg,Q *rp)
{
	char *name;
	FILE *fp;
	char errbuf[BUFSIZ];
	int i;

	asir_assert(ARG0(arg),O_STR,"open_file");
	for ( i = 0; i < BUFSIZ && file_ptrs[i]; i++ );
	if ( i == BUFSIZ )
		error("open_file : too many open files");
	name = BDY((STRING)ARG0(arg));
	if (strcmp(name,"unix://stdin") == 0) {
	  fp = stdin;
	}else if (strcmp(name,"unix://stdout") == 0) {
	  fp = stdout;
    }else if (strcmp(name,"unix://stderr") == 0) {
	  fp = stderr;
    }else{
	  if ( argc(arg) == 2 ) {
		asir_assert(ARG1(arg),O_STR,"open_file");
		fp = fopen(name,BDY((STRING)ARG1(arg)));
	  } else
		fp = fopen(name,"r");
    }
	if ( !fp ) {
		sprintf(errbuf,"open_file : cannot open \"%s\"",name);
		error(errbuf);
	}
	file_ptrs[i] = fp;
	STOQ(i,*rp);
}

void Pclose_file(NODE arg,Q *rp)
{
	int i;

	asir_assert(ARG0(arg),O_N,"close_file");
	i = QTOS((Q)ARG0(arg));
	if ( file_ptrs[i] ) {
		fclose(file_ptrs[i]);
		file_ptrs[i] = 0;
	} else
		error("close_file : invalid argument");
	*rp = ONE;
}

void Pget_line(NODE arg,STRING *rp)
{
	int i,j,c;
	FILE *fp;
	fpos_t head;
	char *str;
	char buf[BUFSIZ];

	if ( !arg ) {
#if defined(VISUAL_LIB)
		get_string(buf,sizeof(buf));
#else
		fgets(buf,sizeof(buf),stdin);
#endif
		i = strlen(buf);
		str = (char *)MALLOC_ATOMIC(i+1);	
		strcpy(str,buf);
		MKSTR(*rp,str);
		return;
	}

	asir_assert(ARG0(arg),O_N,"get_line");
	i = QTOS((Q)ARG0(arg));
	if ( fp = file_ptrs[i] ) {
		if ( feof(fp) ) {
			*rp = 0;
			return;
		}
		fgetpos(fp,&head);
		j = 0;
		while ( 1 ) {
			c = getc(fp);
			if ( c == EOF ) {
				if ( !j ) {
					*rp = 0;
					return;
				} else
					break;
			}
			j++;
			if ( c == '\n' )
				break;
		}
		fsetpos(fp,&head);
		str = (char *)MALLOC_ATOMIC(j+1);	
		fgets(str,j+1,fp);	
		MKSTR(*rp,str);
	} else
		error("get_line : invalid argument");
}

void Pget_byte(NODE arg,Q *rp)
{
	int i,c;
	FILE *fp;

	asir_assert(ARG0(arg),O_N,"get_byte");
	i = QTOS((Q)ARG0(arg));
	if ( fp = file_ptrs[i] ) {
		if ( feof(fp) ) {
			STOQ(-1,*rp);
			return;
		}
		c = getc(fp);
		STOQ(c,*rp);
	} else
		error("get_byte : invalid argument");
}

void Pput_byte(NODE arg,Obj *rp)
{
	int i,j,c;
	FILE *fp;
	Obj obj;
	TB tb;

	asir_assert(ARG0(arg),O_N,"put_byte");
	i = QTOS((Q)ARG0(arg));
	if ( !(fp = file_ptrs[i]) )
		error("put_byte : invalid argument");

	obj = (Obj)ARG1(arg);
	if ( !obj || OID(obj) == O_N ) {
		c = QTOS((Q)obj);
		putc(c,fp);
	} else if ( OID(obj) == O_STR )
		fputs(BDY((STRING)obj),fp);
	else if ( OID(obj) == O_TB ) {
		tb = (TB)obj;
		for ( j = 0; j < tb->next; j++ )
			fputs(tb->body[j],fp);
	}
	*rp = obj;
}

void Pload(NODE arg,Q *rp)
{
	int ret = 0;
	char *name,*name0;
	char errbuf[BUFSIZ];

	if ( ARG0(arg) ) {
		switch (OID(ARG0(arg))) {
			case O_STR:
				name0 = BDY((STRING)ARG0(arg));
				searchasirpath(name0,&name);
				if ( !name ) {
					sprintf(errbuf,"load : \"%s\" not found in the search path",name0);
					error(errbuf);
				}
				execasirfile(name);
				break;
			default:
				error("load : invalid argument");
				break;
		}
	}
	STOQ(ret,*rp);
}

NODE imported_files;

void Pimport(NODE arg,Q *rp)
{
	char *name;
	NODE t,p,opt;

	name = BDY((STRING)ARG0(arg));
	for ( t = imported_files; t; t = NEXT(t) )
		if ( !strcmp((char *)BDY(t),name) ) break;
	if ( !t ) {
		Pload(arg,rp);
		MKNODE(t,name,imported_files); 
		imported_files = t;
		return;
	} else if ( current_option ) {
		for ( opt = current_option; opt; opt = NEXT(opt) ) {
			p = BDY((LIST)BDY(opt));
			if ( !strcmp(BDY((STRING)BDY(p)),"reimport") && BDY(NEXT(p)) ) {
				Pload(arg,rp);
				return;
			}
		}
	}
	*rp = 0;
}

void Pwhich(NODE arg,STRING *rp)
{
	char *name;
	STRING str;

	switch (OID(ARG0(arg))) {
		case O_STR:
			searchasirpath(BDY((STRING)ARG0(arg)),&name);
			break;
		default:
			name = 0;
			break;
	}
	if ( name ) {
		MKSTR(str,name); *rp = str;
	} else
		*rp = 0;
}

void Ploadfiles(NODE arg,Q *rp)
{
	int ret;

	if ( ARG0(arg) )
		if ( OID(ARG0(arg)) != O_LIST )
			ret = 0;
		else
			ret = loadfiles(BDY((LIST)ARG0(arg)));
	else
		ret = 0;
	STOQ(ret,*rp);
}

void Poutput(NODE arg,Q *rp)
{
#if defined(PARI)
	extern FILE *outfile;
#endif
	FILE *fp;

	fflush(asir_out);
	if ( asir_out != stdout )
		fclose(asir_out);
	switch ( argc(arg) ) {
		case 0:
			fp = stdout; break;
		case 1:
			asir_assert(ARG0(arg),O_STR,"output");
			fp = fopen(((STRING)ARG0(arg))->body,"a+");
			if ( !fp )
				error("output : invalid filename");
			break;
	}
#if defined(PARI)
	pari_outfile =
#endif
	asir_out = fp;
	*rp = ONE;
}

extern int ox_file_io;

void Pbsave(NODE arg,Q *rp)
{
	FILE *fp;
	VL vl,t;

	asir_assert(ARG1(arg),O_STR,"bsave");
	get_vars_recursive(ARG0(arg),&vl);
	for ( t = vl; t; t = NEXT(t) )
		if ( t->v->attr == (pointer)V_UC )
			error("bsave : not implemented");
	fp = fopen(BDY((STRING)ARG1(arg)),"wb");
	if ( !fp )
		error("bsave : invalid filename");
	ox_file_io = 1; /* network byte order is used */
	savevl(fp,vl);
	saveobj(fp,ARG0(arg));
	fclose(fp);
	ox_file_io = 0;
	*rp = ONE;
}

void Pbload(NODE arg,Obj *rp)
{
	FILE *fp;

	asir_assert(ARG0(arg),O_STR,"bload");
	fp = fopen(BDY((STRING)ARG0(arg)),"rb");
	if ( !fp )
		error("bload : invalid filename");
	ox_file_io = 1; /* network byte order is used */
	loadvl(fp);
	loadobj(fp,rp);
	fclose(fp);
	ox_file_io = 0;
}

void Pbsave_cmo(NODE arg,Q *rp)
{
	FILE *fp;

	asir_assert(ARG1(arg),O_STR,"bsave_cmo");
	fp = fopen(BDY((STRING)ARG1(arg)),"wb");
	if ( !fp )
		error("bsave_cmo : invalid filename");
	ox_file_io = 1; /* network byte order is used */
	write_cmo(fp,ARG0(arg));
	fclose(fp);
	ox_file_io = 0;
	*rp = ONE;
}

void Pbload_cmo(NODE arg,Obj *rp)
{
	FILE *fp;

	asir_assert(ARG0(arg),O_STR,"bload_cmo");
	fp = fopen(BDY((STRING)ARG0(arg)),"rb");
	if ( !fp )
		error("bload_cmo : invalid filename");
	ox_file_io = 1; /* network byte order is used */
	read_cmo(fp,rp);
	fclose(fp);
	ox_file_io = 0;
}

static struct oSTRING rootdir;

#if defined(VISUAL)
void get_rootdir(char *name,int len)
{
	LONG	ret;
	HKEY	hOpenKey;
	DWORD	Type;
	char	*slash;

	if ( rootdir.body ) {
		strcpy(name,rootdir.body);
		return;
	}

	if ( access("UseCurrentDir",0) >= 0 ) {
		GetCurrentDirectory(BUFSIZ,name);
		slash = strrchr(name,'\\');
		if ( slash )
			*slash = 0;
		return;
	}
	name[0] = 0;
	ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, ECGEN_KEYNAME, 0,
		KEY_QUERY_VALUE, &hOpenKey);
	if ( ret != ERROR_SUCCESS )
		ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, ASIR_KEYNAME, 0,
			KEY_QUERY_VALUE, &hOpenKey);
	if( ret == ERROR_SUCCESS ) {
		RegQueryValueEx(hOpenKey, "Directory", NULL, &Type, name, &len);
		RegCloseKey(hOpenKey);
	} else {
		GetCurrentDirectory(len,name);
		slash = strrchr(name,'\\');
		if ( slash )
			*slash = 0;
	}
}

void set_rootdir(char *name)
{
	static char DirName[BUFSIZ];

	strcpy(DirName,name);
	rootdir.id = O_STR;
	rootdir.body = DirName;
	asir_libdir = DirName;
	/* XXX */
	env_init();
}

#else
void get_rootdir(char *name,int len)
{
	strcpy(name,asir_libdir);
}

void set_rootdir(char *name)
{
	static char DirName[BUFSIZ];

	strcpy(DirName,name);
	asir_libdir = DirName;
	/* XXX */
	env_init();
}

#endif

void Pget_rootdir(STRING *rp)
{
	static char DirName[BUFSIZ];

	if ( !rootdir.body ) {
		get_rootdir(DirName,sizeof(DirName));
		rootdir.id = O_STR;
		rootdir.body = DirName;
	}
	*rp = &rootdir;
}

#if defined(VISUAL) && defined(DES_ENC)
void Pbsave_enc(NODE arg,Obj *rp)
{
	init_deskey();
	des_encryption = 1;
	Pbsave(arg,rp);
	des_encryption = 0;
}

void Pbload_enc(NODE arg,Obj *rp)
{
	init_deskey();
	des_encryption = 1;
	Pbload(arg,rp);
	des_encryption = 0;
}
#endif

void Pbload27(NODE arg,Obj *rp)
{
	FILE *fp;
	Obj r;

	asir_assert(ARG0(arg),O_STR,"bload27");
	fp = fopen(BDY((STRING)ARG0(arg)),"rb");
	if ( !fp )
		error("bload : invalid filename");
	loadvl(fp);
	loadobj(fp,&r);
	fclose(fp);
	bobjtoobj(BASE27,r,rp);
}

void Pbsave_compat(NODE arg,Q *rp)
{
	FILE *fp;
	VL vl,t;

	asir_assert(ARG1(arg),O_STR,"bsave_compat");
	get_vars_recursive(ARG0(arg),&vl);
	for ( t = vl; t; t = NEXT(t) )
		if ( t->v->attr == (pointer)V_UC )
			error("bsave : not implemented");
	fp = fopen(BDY((STRING)ARG1(arg)),"wb");
	if ( !fp )
		error("bsave : invalid filename");
	/* indicator of an asir32 file */
	putw(0,fp); putw(0,fp);
	savevl(fp,vl);
	saveobj(fp,ARG0(arg));
	fclose(fp);
	*rp = ONE;
}

void Pbload_compat(NODE arg,Obj *rp)
{
	FILE *fp;
	unsigned int hdr[2];
	Obj r;
	int c;

	asir_assert(ARG0(arg),O_STR,"bload_compat");
	fp = fopen(BDY((STRING)ARG0(arg)),"rb");
	if ( !fp )
		error("bload : invalid filename");
	fread(hdr,sizeof(unsigned int),2,fp);
	if ( !hdr[0] && !hdr[1] ) {
		/* asir32 file or asir27 0 */
		c = fgetc(fp);
		if ( c == EOF ) {
			/* asir27 0 */
			*rp = 0;
		} else {
			/* asir32 file */
			ungetc(c,fp);
			loadvl(fp);
			loadobj(fp,rp);
		}
	} else {
		/* asir27 file */
		rewind(fp);
		loadvl(fp);
		loadobj(fp,&r);
		bobjtoobj(BASE27,r,rp);
	}
	fclose(fp);
}

void Premove_file(NODE arg,Q *rp)
{
	unlink((char *)BDY((STRING)ARG0(arg)));
	*rp = ONE;
}

void Paccess(NODE arg,Q *rp)
{
#if defined(VISUAL)
	if ( access(BDY((STRING)ARG0(arg)),04) >= 0 )
#else
	if ( access(BDY((STRING)ARG0(arg)),R_OK) >= 0 )
#endif
		*rp = ONE;
	else
		*rp = 0;
}

#if defined(VISUAL)
int process_id()
{
	return GetCurrentProcessId();
}

void call_exe(char *name,char **av)
{
	_spawnv(_P_WAIT,name,av);
}
#endif