#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/time.h>
#ifdef __sgi__
#include <bstring.h>
#endif
#include <stdlib.h>
#include <string.h>

int uhOpenGame_pt(void);
int uhCloseGame_pt(int);
int uhReadGame_pt(int hfd,void *buf,int count);
int uhWriteGame_pt(int hfd,void *buf,int count);
int uhReadRamrom_pt(int hfd,void *ramrom_adr,void *buf,int count);
int uhWriteRamrom_pt(int hfd,void *ramrom_adr,void *buf,int count);
int uhGload_pt(int hfd,char *loadfile);

int execl_pt(char *path,...);

static int n64read(int fd,void *buf,int count);
static int n64write(int fd,void *buf,int count);
static int n64_init(void);
static int n64_SetFlag(int val);
static int n64_GetFlag(void);
static int n64_sleep(int t);
static void n64_gload_wait(void);

int rfd;
int wfd;
int hfd;

#define HOST_FD 1234

/*
FIFO Packet format

	long	command
	long	data1
	long	data2
	long	size;
	buf[size]

	read mem
	 SEND
		command	1
		data1	addr
		data2	read len
		size	0
	 RECV
		command	1
		data1	0
		data2	0
		size	n byte + 4
		data	(n byte)
		long	status

	write mem
	 SEND
		command	2
		data1	addr
		data2	0
		siz	n byte
		data	n byte
	 RECV
		command	2
		data1	status
		data2	0
		size	0
*/

#define	MEMREAD		1
#define	MEMWRITE	2
#define	BLKREAD		3
#define	BLKWRITE	4
#define	PTN64CMD	5

int uhReadRamrom_pt(int hfd,void *ramrom_adr,void *buf,int count)
{
/*	return		< 0	error
			< -4	n64 target bus error
			> 0	read count
*/

	unsigned	pipe_buf[4];

	if(hfd!=HOST_FD) return -1;
	if(count==0) return 0;
	pipe_buf[0] = MEMREAD;
	pipe_buf[1] = (unsigned)ramrom_adr+0xb0000000;
	pipe_buf[2] = count;
	pipe_buf[3] = 0;
	if(n64write(wfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -1;
	if(n64read(rfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -2;
	if(n64read(rfd,buf,pipe_buf[3] - 4) < 0)
		return -3;
	if(n64read(rfd,pipe_buf,4) < 0)
		return -4;
	if(pipe_buf[0])
		return (-4 - pipe_buf[0]);
	return (pipe_buf[3] - 4);		/* read count */
}

int uhWriteRamrom_pt(int hfd,void *ramrom_adr,void *buf,int count)
{
/*	return		< 0	error
			< -4	n64 target bus error
			> 0	write count
*/

	unsigned	pipe_buf[4];

	if(hfd!=HOST_FD) return -1;
	if(count==0) return 0;
	pipe_buf[0] = MEMWRITE;
	pipe_buf[1] = (unsigned)ramrom_adr+0xb0000000;
	pipe_buf[2] = 0;
	pipe_buf[3] = count;
	if(n64write(wfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -1;
	if(n64write(wfd,buf,count) < 0)
		return -2;
	if(n64read(rfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -3;
	if(pipe_buf[1])
		return (-4 - pipe_buf[1]);
	return count;
}

int uhReadGame_pt(int hfd,void *buf,int count)
{
/*	return		< 0	error
			< -4	n64 target bus error
			> 0	read count
*/

	unsigned	pipe_buf[4];

	if(hfd!=HOST_FD) return -1;
	if(count==0) return 0;
	n64_gload_wait();
	pipe_buf[0] = BLKREAD | 0x80000000;
	pipe_buf[1] = count;
	pipe_buf[2] = 0;
	pipe_buf[3] = 0;
	if(n64write(wfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -1;
	if(n64read(rfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -2;
	if(n64read(rfd,buf,pipe_buf[3] - 4) < 0)
		return -3;
	if(n64read(rfd,pipe_buf,4) < 0)
		return -4;
	if(pipe_buf[0])
		return (-4 - pipe_buf[0]);
	return (pipe_buf[3] - 4);		/* read count */
}

int uhWriteGame_pt(int hfd,void *buf,int count)
{
/*	return		< 0	error
			< -4	n64 target bus error
			> 0	write count
*/

	unsigned	pipe_buf[4];

	if(hfd!=HOST_FD) return -1;
	if(count==0) return 0;
	n64_gload_wait();
	pipe_buf[0] = BLKWRITE | 0x80000000;
	pipe_buf[1] = 0;
	pipe_buf[2] = 0;
	pipe_buf[3] = count;
	if(n64write(wfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -1;
	if(n64write(wfd,buf,count) < 0)
		return -2;
	if(n64read(rfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -3;
	if(pipe_buf[1])
		return (-4 - pipe_buf[1]);
	return count;
}

int uhPartnerCmd_pt(int hfd,void *buf)
{
/*	return		< 0	error
			< -4	n64 target bus error
			> 0	write count
*/
	int count;
	unsigned pipe_buf[4];

	if(hfd!=HOST_FD) return -1;
	count=strlen(buf);
	if(count>=512) count=512-1;
	pipe_buf[0] = PTN64CMD;
	pipe_buf[1] = 0;
	pipe_buf[2] = 0;
	pipe_buf[3] = count;
	if(n64write(wfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -1;
	if(n64write(wfd,buf,count) < 0)
		return -2;
	if(n64read(rfd,pipe_buf,sizeof(pipe_buf)) < 0)
		return -3;
	if(pipe_buf[1])
		return (-4 - pipe_buf[1]);
	return count;
}

static int n64read(int fd,void *buf,int count)
{
	int	rc;
	int	count_bak = count;

	while(count != 0){
		if((rc = read(fd,buf,count)) < 0)
			return rc;
		buf = (char *)buf + rc;
		count -= rc;
	}
	return count_bak;
}

static int n64write(int fd,void *buf,int count)
{
	int	rc;
	int count_bak = count;

	while(count != 0){
		if((rc = write(fd,buf,count)) < 0)
			return rc;
		buf = (char *)buf + rc;
		count -= rc;
	}
	return count_bak;
}


static int	shmid;
static int	*shmptr;
static char	commemname[] = "/tmp/ptn64_c";
/*
	*(shmptr)	status
	*(shmptr+1)	not use
	*(shmptr+2)	not use
	*(shmptr+3)	cancel
*/


static int n64_init(void)
{
	int	i;

	if(*(shmptr+3)) return -1;
	*(shmptr+3) = 1;
	for(i = 0 ; i < 20 ; i++){
		if(*(shmptr+3) == 0) return 0;
		n64_sleep(100);		/* 100 mS sleep */
	}
	return -2;			/* 2 Sec wait error */
}

static int n64_SetFlag(int val)
{
	*shmptr = val;
	return 0;
}

static int n64_GetFlag(void)
{
	n64_sleep(100);			/* 100mS wait */
	return *shmptr;
}

static int n64_sleep(int t)		/* t mS wait */
{
	struct timeval timed;

	timed.tv_sec = 0;
	timed.tv_usec = 1000*t;
	select(0,(fd_set *)NULL,(fd_set *)NULL,(fd_set *)NULL,(struct timeval *)&timed);
	return 0;
}

static void n64_gload_wait(void)
{
    static int gload_ok=0;

    if(gload_ok==0){
	while(n64_GetFlag()!=0){
	    printf("\n***** WAIT ******\n");
	}
	gload_ok=1;
    }
}

int uhOpenGame_pt(void)
{
	int	rc;
	char	buf[256];
	struct	timeval timed;
	char	svrname[] = "/tmp/ptn64";
	char	recvname[] = "/tmp/ptn64_s";
	int	shmfd;

	if((wfd = open(svrname,O_WRONLY)) < 0)
		return -1;
	if((rfd = open(recvname,O_RDONLY)) < 0)
		return -2;
	if((rc = fcntl(rfd,F_SETFL,O_RDONLY | O_NDELAY)) < 0)
		return -4;

	do{
	    rc = 1;
	    while(rc){
		if((rc = read(rfd,buf,256)) < 0)
		return -3;
	    }

	    timed.tv_sec = 0;			/* wait 100 uS */
	    timed.tv_usec = 100;
	    select(1,(fd_set * )NULL,(fd_set *)NULL,
		(fd_set *)NULL,(struct timeval *)&timed);

	}while(read(rfd,buf,256) != 0);
	if((rc = fcntl(rfd,F_SETFL,O_RDONLY)) < 0)
		return -5;

	if((shmfd = open(commemname,O_RDONLY)) < 0){
	    printf("[%s] open error !!(Eno = %d)",commemname,errno);
	    return -6;
	}
	if(read(shmfd,buf,256) < 0){
	    printf("[%s] read error !!(Eno = %d)",commemname,errno);
	    return -7;
	}
	shmid = atoi(buf);
	if((shmptr = shmat(shmid,0,0)) == (void *)-1){
	    printf("shmptr error !!(Eno = %d)",errno);
	    return -8;
	}
/*	printf("\nshm ID = %d addr = %08x",shmid,shmptr); */
	n64_init();
	hfd=HOST_FD;
	return HOST_FD;
}


int uhCloseGame_pt(int hfd)
{
	if(hfd!=HOST_FD) return -1;
	hfd=0;
	close(rfd);
	close(wfd);
	return 0;
}

int uhGload_pt(int hfd,char *loadfile)
{
    int len;
    char *p,*q,*loadname;
    char cmd[512];
    struct stat file_stat;

    if(hfd!=HOST_FD) return -1;
    strcpy(cmd,"rd ");
    len=strlen(cmd);
    loadname=cmd+len;
    getcwd(loadname,sizeof(cmd)-len);
    if(loadfile==0){
	p="rom";
    }
    else{
	p=loadfile;
    }
    if(*p!='/' && *p!='~'){
	q=cmd+strlen(cmd);
	if(*(q-1)!='/'){
	    *(q++)='/';
	}
	strcpy(q,p);
    }
    else{
	strcpy(loadname,p);
    }
    if(stat(loadname,&file_stat)!=0){
	return -1;
    }
/*
    PARTNER-N64 Command
	rd loadfile,0xb0000000
	reset
	g
*/
    strcat(cmd,",0xb0000000\nreset\ng\n");
    uhPartnerCmd_pt(hfd,cmd);
    n64_SetFlag(0);
    return 0;
}

#include <stdarg.h>

int execl_pt(char *path,...)
{
    va_list ap;
    int len;
    char *p,*q,*loadname;
    char cmd[512];
    struct stat file_stat;

    va_start(ap,path);
    if(strcmp(path,"/usr/sbin/gload")==0){
	p=va_arg(ap,char *);
	p=va_arg(ap,char *);
	strcpy(cmd,"rd ");
	len=strlen(cmd);
	loadname=cmd+len;
	getcwd(loadname,sizeof(cmd)-len);
	if(p==0){
	    p="rom";
	}
	if(*p!='/' && *p!='~'){
	    q=cmd+strlen(cmd);
	    if(*(q-1)!='/'){
		*(q++)='/';
	    }
	    strcpy(q,p);
	}
	else{
	     strcpy(loadname,p);
	}
        if(stat(loadname,&file_stat)!=0){
	    return -1;
	}
/*
    PARTNER-N64 Command
	rd loadfile,0xb0000000
	reset
	g
*/
	strcat(cmd,",0xb0000000\nreset\n\g\n");
	uhPartnerCmd_pt(hfd,cmd);
	n64_SetFlag(0);
	for(;;){
	    n64_sleep(100);			/* 100mS wait */
	}
/*	exit(0); */
	return 0;
    }
    else{
	return execv(path,(char **)ap);
     }
}