/* Hey EMACS -*- linux-c -*- */
/* $Id$ */

/*  libticables2 - link cable library, a part of the TiLP project
 *  Copyright (C) 1999-2005  Romain Lievin
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software Foundation,
 *  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/* "TiEmu" virtual link cable unit */

/* 
 *  This unit use two pipes between 2 programs which use this lib.
 *  Convention used: 0 is an emulator and 1 is a linking program.
 *  One pipe is used for transferring information from 0 to 1 and the other
 *  pipe is used for transferring from 1 to 0.
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <errno.h>

#include "../ticables.h"
#include "../logging.h"
#include "../error.h"
#include "../gettext.h"
#include "detect.h"

#define BUFFER_SIZE 256
#define HIGH 666	        // upper limit (used for avoiding timeout)
#define LOW  333		// lower limit

static key_t ipc_key;
static int   shmid;
static int*  shmaddr;
#define ref_cnt (*shmaddr)      // counter of instances

static int rd[2] = { 0 };	// Pipe 0 <- 1 or 1 <- 0
static int wr[2] = { 0 };	// Pipe 0 -> 1 or 1 -> 0

static const char fifo_names[4][256] = 
{
  "/tmp/.vlc_1_0", "/tmp/.vlc_0_1",
  "/tmp/.vlc_0_1", "/tmp/.vlc_1_0"
};

static int shm_check(void)
{
    int shmid_;

    if ((ipc_key = ftok("/tmp", 0x1234)) == -1)
        return ERR_TIE_OPEN;
    shmid_ = shmget(ipc_key, 1, IPC_CREAT | IPC_EXCL | 0666);

    if((shmid_ == -1) && (errno == EEXIST))
	return 1;
    else
	shmctl(shmid_, IPC_RMID, NULL);
   
    return 0;
}

static int tie_prepare(CableHandle *h)
{
    // in fact, address & device are unused
    switch(h->port)
    {
    case PORT_0:	// automatic setting
	h->address = shm_check();
	break;
    case PORT_1:	// forced setting, for compatibility
    case PORT_3: 
	h->address = 0; h->device = strdup("0->1"); 
	break;
    case PORT_2:
    case PORT_4:
	h->address = 1; h->device = strdup("1->0"); 
	break;
    default: return ERR_ILLEGAL_ARG;
    }
    
    return 0;
}

static int tie_reset(CableHandle *h);
static int tie_open(CableHandle *h)
{
    int p = h->address;
    int new = 0;

    if ((ipc_key = ftok("/tmp", 0x1234)) == -1)
	return ERR_TIE_OPEN;
    shmid = shmget(ipc_key, 1, IPC_CREAT | IPC_EXCL | 0666);
    if(shmid != -1)
	new = 1;
    else if((shmid == -1) && (errno == EEXIST))
	if ((shmid = shmget(ipc_key, 1, IPC_CREAT | 0666)) < 0)
	    return ERR_TIE_OPEN;
    if ((shmaddr = shmat(shmid, NULL, 0)) == (int *)-1)
	return ERR_TIE_OPEN;
 
    /* Check if the pipes already exist else create them */
    if (access(fifo_names[0], F_OK) | access(fifo_names[1], F_OK)) 
    {
	mkfifo(fifo_names[0], O_RDONLY | O_WRONLY | O_NONBLOCK | S_IRWXU);
	mkfifo(fifo_names[1], O_RDONLY | O_WRONLY | O_NONBLOCK | S_IRWXU);
    }
    
    /* Open the pipes */
    // Open the 1->0 pipe in reading
    if ((rd[p] = open(fifo_names[2 * (p) + 0], O_RDONLY | O_NONBLOCK)) == -1) 
    {
	ticables_warning(_("error: %s\n"), strerror(errno));
	return ERR_TIE_OPEN;
    }
    // Open the 0->1 pipe in writing (in reading at first)
    if ((wr[p] = open(fifo_names[2 * (p) + 1], O_RDONLY | O_NONBLOCK)) == -1) 
    {
	return ERR_TIE_OPEN;
    }
    if ((wr[p] = open(fifo_names[2 * (p) + 1], O_WRONLY | O_NONBLOCK)) == -1) 
    {
	return ERR_TIE_OPEN;
    }

    ref_cnt = new ? 1 : 2;
    TRYC(tie_reset(h));

    return 0;
}

static int tie_close(CableHandle *h)
{
    int p = h->address;

    if (rd[p]) 
    {
	if (close(rd[p]) == -1) 
	    return ERR_TIE_CLOSE;
	rd[p] = 0;
    }
    if (wr[p]) 
    {
	if (close(wr[p]) == -1) 
	    return ERR_TIE_CLOSE;
	wr[p] = 0;
    }
    
    ref_cnt--;
    if (!ref_cnt)
    {
	struct shmid_ds shmid_ds;
	shmdt(shmaddr);
	shmctl(shmid, IPC_RMID, &shmid_ds);
    }
    else
	shmdt(shmaddr);
    
    return 0;
}

static int tie_reset(CableHandle *h)
{
    int p = h->address;
    uint8_t d;
        
    if(ref_cnt < 2)
	return 0;
    
    /* Flush the pipe */
    while(read(rd[p], (void *) (&d), 1) > 0);
    
    return 0;
}

static int tie_put(CableHandle *h, uint8_t *data, uint32_t len)
{
    int p = h->address;
    tiTIME clk;
    struct stat s;
    ssize_t n = 0;
    
    if(ref_cnt < 2)
	return 0;
    
    /* Transfer rate modulation */
    TO_START(clk);
    do 
    {
	if (TO_ELAPSED(clk, h->timeout))
	    return ERR_WRITE_TIMEOUT;
	fstat(wr[p], &s);
	if (s.st_size > HIGH)
	    n = 0;
	else if (s.st_size < LOW)
	    n = 1;
    }
    while (n <= 0);
    
    /* Write the data with byte timeout */
    TO_START(clk);
    do 
    {
	n = write(wr[p], (void *)data, len);

	if (TO_ELAPSED(clk, h->timeout))
	    return ERR_WRITE_TIMEOUT;
    }
    while (n <= 0);
    
    return 0;
}

static int tie_get(CableHandle *h, uint8_t *data, uint32_t len)
{
    int p = h->address;
    tiTIME clk;
    ssize_t i, ret;

    if(ref_cnt < 2)
	return 0;

    for(i = 0; i < len; )    
    {
	TO_START(clk);
	do 
	{
	    ret = read(rd[p], (void *)(data + i), len - i);

	    if (TO_ELAPSED(clk, h->timeout))
                return ERR_READ_TIMEOUT;

	    if (ret == -1 && errno != EAGAIN)
		return ERR_READ_ERROR;
	}
	while (ret <= 0);
	
	i += ret;
    }    

    return 0;
}

static int tie_probe(CableHandle *h)
{
    return shm_check() ? 0 : ERR_PROBE_FAILED;
}

static int tie_check(CableHandle *h, int *status)
{
    int p = h->address;
    fd_set rdfs;
    struct timeval tv;
    int retval;

    *status = STATUS_NONE;

    if(ref_cnt < 2)
	return 0;
    
    FD_ZERO(&rdfs);
    FD_SET(rd[p], &rdfs);
    tv.tv_sec = 0;
    tv.tv_usec = 0;
    
    retval = select(rd[p] + 1, &rdfs, NULL, NULL, &tv);
    switch (retval) 
    {
    case -1:			//error
	return ERR_READ_ERROR;
    case 0:			//no data
	return 0;
    default:			// data available
	*status = STATUS_RX;
	break;
    }
    
    return 0;
}

static int tie_set_red_wire(CableHandle *h, int b)
{
	return 0;
}

static int tie_set_white_wire(CableHandle *h, int b)
{
	return 0;
}

static int tie_get_red_wire(CableHandle *h)
{
	return 1;
}

static int tie_get_white_wire(CableHandle *h)
{
	return 1;
}

const CableFncts cable_tie = 
{
	CABLE_TIE,
	"TIE",
	N_("TiEmu"),
	N_("Virtual link for TiEmu"),
	0,
	&tie_prepare,
	&tie_open, &tie_close, &tie_reset, &tie_probe, NULL,
	&tie_put, &tie_get, &tie_check,
	&tie_set_red_wire, &tie_set_white_wire,
	&tie_get_red_wire, &tie_get_white_wire,
	NULL, NULL
};
