/*
 * 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/asir2018/builtin/file.c,v 1.3 2020/10/06 06:31:19 noro Exp $
*/
#include "ca.h"
#include "parse.h"
#include "ox.h"
#include "base.h"
#include <stdlib.h>
#if !defined(VISUAL) && !defined(__MINGW32__)
#include "unistd.h"
#endif
#if defined(PARI)
#include "genpari.h"
#endif

#if defined(VISUAL) || defined(__MINGW32__)
#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"
#define R_OK 0x04
#endif

void Psprintf(NODE,STRING *);

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

void Pload(), Pwhich(), Ploadfiles(), Poutput();
void Pbsave(), Pbload();
void Pbsave_cmo(), Pbload_cmo();
void Popen_file(), Pclose_file(), Pget_line(), Pget_byte(), Pput_byte();
void Pput_word(), Pget_word();
void Ppurge_stdin();
void Pfprintf();
void Pimport();
void Pgetpid();

extern int des_encryption;
extern char *asir_libdir;

struct ftab file_tab[] = {
  {"fprintf",Pfprintf,-99999999},
  {"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_word",Pget_word,1},
  {"put_word",Pput_word,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(DES_ENC)
  {"bsave_enc",Pbsave_enc,2},
  {"bload_enc",Pbload_enc,1},
#endif
  {"bsave_cmo",Pbsave_cmo,2},
  {"bload_cmo",Pbload_cmo,1},
  {"getpid",Pgetpid,0},
  {0,0,0},
};

static FILE *file_ptrs[BUFSIZ];

void Pfprintf(NODE arg,pointer *rp)
{
  FILE *fp;
  STRING s;
  asir_assert(ARG0(arg),O_N,"fprintf");
  fp = file_ptrs[ZTOS((Q)ARG0(arg))];
  if ( !fp ) {
    error("fprintf : invalid argument");
  }
  arg = NEXT(arg);
  if ( arg ) {
    Psprintf(arg,&s);
    fputs(BDY(s),fp);
    /* fflush(fp); */
  }
  *rp = 0;
  return;
}

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

void Popen_file(NODE arg,Z *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,"rb");
    }
  if ( !fp ) {
    sprintf(errbuf,"open_file : cannot open \"%s\"",name);
    error(errbuf);
  }
  file_ptrs[i] = fp;
  STOZ(i,*rp);
}

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

  asir_assert(ARG0(arg),O_N,"close_file");
  i = ZTOS((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 = ZTOS((Q)ARG0(arg));
  if ( ( fp = file_ptrs[i] ) != 0 ) {
    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);  
                for ( i = 0; i < j; i++ )
                  if ( str[i] == '\r' ) {
                     str[i] = '\n';
                     str[i+1] = 0;
                     break;
                  }
    MKSTR(*rp,str);
  } else
    error("get_line : invalid argument");
}

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

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

void Pget_word(NODE arg,Z *rp)
{
  int i,c;
  FILE *fp;

  asir_assert(ARG0(arg),O_N,"get_word");
  i = ZTOS((Q)ARG0(arg));
  if ( ( fp = file_ptrs[i] ) != 0 ) {
    if ( feof(fp) ) {
      error("get_word : end of file");
      return;
    }
    read_int(fp,(unsigned int *)&c);
    STOZ(c,*rp);
  } else
    error("get_word : 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 = ZTOS((Q)ARG0(arg));
  if ( !(fp = file_ptrs[i]) )
    error("put_byte : invalid argument");

  obj = (Obj)ARG1(arg);
  if ( !obj || OID(obj) == O_N ) {
    c = ZTOS((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 Pput_word(NODE arg,Obj *rp)
{
  int i,c;
  FILE *fp;
  Obj obj;

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

  obj = (Obj)ARG1(arg);
  c = ZTOS((Q)obj);
  write_int(fp,(unsigned int *)&c);
  *rp = obj;
}

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

  if ( !arg ) error("load : invalid argument");
  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;
    }
  }
  STOZ(ret,*rp);
}

NODE imported_files;

void Pimport(NODE arg,Z *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,Z *rp)
{
  int ret;

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

void Poutput(NODE arg,Z *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,Z *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,Z *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) || defined(__MINGW32__)
static void chop_delim(char *s)
{
    int n=strlen(s);
    if(n>1 && s[n-1]=='\\') {
        s[n-1]=0;
    }
}

void get_rootdir(char *name,int len)
{
  LONG  ret;
  HKEY  hOpenKey;
  DWORD  Type;
  char    drive[_MAX_DRIVE], dir0[_MAX_DIR], dir[_MAX_DIR];
  char    fullname[_MAX_PATH];

  if ( rootdir.body ) {
    strcpy_s(name,len,rootdir.body);
    return;
  }
  if ( GetModuleFileName(NULL,fullname,_MAX_PATH) ) {
        _splitpath(fullname,drive,dir0,NULL,NULL);
        chop_delim(dir0);
        _splitpath(dir0,NULL,dir,NULL,NULL);
        chop_delim(dir);
        strcpy_s(name,len,drive);
        strcat_s(name,len,dir);
    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(_MAX_PATH,fullname);
        _splitpath(fullname,drive,dir,NULL,NULL);
        chop_delim(dir);
        strcpy_s(name,len,drive);
        strcat_s(name,len,dir);
  }
}

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;
}

void Pgetpid(Z *rp)
{
  int id;

#if defined(VISUAL) || defined(__MINGW32__)
  id = GetCurrentProcessId();
#else
  id = getpid();
#endif
  STOZ(id,*rp);
}

#if 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 Premove_file(NODE arg,Z *rp)
{
  unlink((char *)BDY((STRING)ARG0(arg)));
  *rp = ONE;
}

void Paccess(NODE arg,Z *rp)
{
  if ( access(BDY((STRING)ARG0(arg)),R_OK) >= 0 )
    *rp = ONE;
  else
    *rp = 0;
}

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

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