/*-
 * Copyright (c) 2002-2018 The UbixOS Project.
 * All rights reserved.
 *
 * This was developed by Christopher W. Olsen for the UbixOS Project.
 *
 * Redistribution and use in source and binary forms, with or without modification, are permitted
 * provided that the following conditions are met:
 *
 * 1) Redistributions of source code must retain the above copyright notice, this list of
 *    conditions, the following disclaimer and the list of authors.
 * 2) Redistributions in binary form must reproduce the above copyright notice, this list of
 *    conditions, the following disclaimer and the list of authors in the documentation and/or
 *    other materials provided with the distribution.
 * 3) Neither the name of the UbixOS Project nor the names of its contributors may be used to
 *    endorse or promote products derived from this software without specific prior written
 *    permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/descrip.h>
#include <sys/sysproto_posix.h>
#include <sys/thread.h>
#include <lib/kprintf.h>
#include <ubixos/endtask.h>
#include <lib/kmalloc.h>
#include <assert.h>
#include <sys/ioctl.h>
#include <sys/select.h>

static struct file *kern_files = 0x0;

int fcntl(struct thread *td, struct fcntl_args *uap) {
  struct file *fp = 0x0;

#ifdef DEBUG
  kprintf("[%s:%i]",__FILE__,__LINE__);
#endif

  if (td->o_files[uap->fd] == 0x0) {
    kprintf("ERROR!!!\n");
    return (-1);
  }

  fp = (struct file *) td->o_files[uap->fd];
  switch (uap->cmd) {
  case 3:
    td->td_retval[0] = fp->f_flag;
    break;
  case 4:
    fp->f_flag &= ~FCNTLFLAGS;
    fp->f_flag |= FFLAGS(uap->arg & ~O_ACCMODE) & FCNTLFLAGS;
    break;
  default:
    kprintf("ERROR DEFAULT");
  }

  return (0x0);
}

int falloc(struct thread *td, struct file **resultfp, int *resultfd) {

  struct file *fp = 0x0;
  int i = 0;

  fp = (struct file *) kmalloc(sizeof(struct file));

  /* First 5 Descriptors Are Reserved */
  for (i = 5; i < MAX_FILES; i++) {
    if (td->o_files[i] == 0x0) {
      td->o_files[i] = (void *) fp;
      if (resultfd)
        *resultfd = i;
      if (resultfp)
        *resultfp = fp;
      goto allocated;
    }
  }

  kfree(fp);

  *resultfp = 0x0;
  *resultfd = 0x0;

  allocated:

  return (0x0);
}

int fdestroy(struct thread *td, struct file *fp, int fd) {
  int error = 0;

  if (td->o_files[fd] != fp) {
    error = -1;
  } else {
    kfree(td->o_files[fd]);
    td->o_files[fd] = 0x0;
  }

  return (error);
}

int close(struct thread *td, struct close_args *uap) {
#ifdef DEBUG
  kprintf("[%s:%i]",__FILE__,__LINE__);
#endif
  kfree((void *) td->o_files[uap->fd]);
  td->o_files[uap->fd] = 0x0;
  td->td_retval[0] = 0x0;
  return (0x0);
}

/*!
 * \brief return data table size
 */
int getdtablesize(struct thread *td, struct getdtablesize_args *uap) {
#ifdef DEBUG
  kprintf("[%s:%i]",__FILE__,__LINE__);
#endif
  td->td_retval[0] = O_FILES;
  return (0);
}

/* HACK */
int fstat(struct thread *td, struct sys_fstat_args *uap) {
  struct file *fp = 0x0;

#ifdef DEBUG
  kprintf("[%s:%i]",__FILE__,__LINE__);
#endif

  fp = (struct file *) _current->td.o_files[uap->fd];
  uap->sb->st_mode = 0x2180;
  uap->sb->st_blksize = 0x1000;
  kprintf("fstat: %i", uap->fd);
  return (0x0);
}

/*!
 * \brief ioctl functionality not implimented yet
 *
 * \returns NULL for now
 */
int ioctl(struct thread *td, struct ioctl_args *uap) {
  td->td_retval[0] = 0x0;
  return (0x0);
}

/*!
 * \brief get pointer to file fd in specified thread
 *
 * \return returns fp
 */
int getfd(struct thread *td, struct file **fp, int fd) {
  int error = 0x0;

#ifdef DEBUG
  kprintf("[%s:%i]",__FILE__,__LINE__);
#endif

  *fp = (struct file *) td->o_files[fd];

  if (fp == 0x0)
    error = -1;

  return (error);
}

int sys_ioctl(struct thread *td, struct sys_ioctl_args *args) {
  switch (args->com) {
  case TIOCGETA:
    if (args->fd == 0 || args->fd == 1) {
      struct termios *t = (struct termios *) args->data;

      t->c_iflag = 0x2B02;
      t->c_oflag = 0x3;
      t->c_cflag = 0x4B00;
      t->c_lflag = 0x5CB;

      t->c_cc[0] = 4;
      t->c_cc[1] = 255;
      t->c_cc[2] = 255;
      t->c_cc[3] = 127;
      t->c_cc[4] = 23;
      t->c_cc[5] = 21;
      t->c_cc[6] = 18;
      t->c_cc[7] = 8;
      t->c_cc[8] = 3;
      t->c_cc[9] = 28;
      t->c_cc[10] = 26;
      t->c_cc[11] = 25;
      t->c_cc[12] = 17;
      t->c_cc[13] = 19;
      t->c_cc[14] = 22;
      t->c_cc[15] = 15;
      t->c_cc[16] = 1;
      t->c_cc[17] = 0;
      t->c_cc[18] = 20;
      t->c_cc[19] = 255;

      t->c_ispeed = 0x9600;
      t->c_ospeed = 0x9600;

      td->td_retval[0] = 0;
    } else {
      td->td_retval[0] = -1;
    }
    break;
  default:
    kprintf("ioFD:%i:%i!", args->fd, args->com);
    break;
  }
  td->td_retval[0] = 0x0;
  return (0x0);
}

int sys_select(struct thread *td, struct sys_select_args *args) {
  int error = 0x0;
  /*
  int i = 0x0;

  fd_set sock_rfds;
  fd_set sock_wfds;
  fd_set sock_efds;

  FD_ZERO(&sock_rfds);
  FD_ZERO(&sock_wfds);
  FD_ZERO(&sock_efds);

  if (args->in != 0x0) {
    for (i = 0; i < args->nd; i++) {
      FD_SET(((struct file * ) td->o_files[args->in[0]]).socket, &sock_rfds);
    }
  }

  if (args->ou != 0x0) {
    for (i = 0; i < args->nd; i++) {
      FD_SET(((struct file * ) td->o_files[args->ou[0]]).socket, &sock_wfds);
    }
  }

  if (args->ex != 0x0) {
    for (i = 0; i < args->nd; i++) {
    FD_SET(((struct file * ) td->o_files[args->ex[0]]).socket, &sock_efds);
    }
  }

  if ((td->td_retval[0] = lwip_select(args->nd, &sock_rfds, &sock_wfds, &sock_efds, args->tv)) == -1)
    error = -1;

  if (args->in != 0x0) {
    for (i = 0; i < MAX_FILES; i++) {
      if (!FD_ISSET(((struct file * ) td->o_files[args->ou[0]]).socket, &sock_rfds))
        FD_CLR(((struct file * ) td->o_files[args->ou[0]]).socket, args->in);
    }
  }

  if (args->ou != 0x0) {
    for (i = 0; i < MAX_FILES; i++) {
      if (!FD_ISSET(((struct file * ) td->o_files[args->ou[0]]).socket, &sock_wfds))
        FD_CLR(((struct file * ) td->o_files[args->ou[0]]).socket, args->ou);
    }
  }

  if (args->ex != 0x0) {
    for (i = 0; i < MAX_FILES; i++) {
      if (!FD_ISSET(((struct file * ) td->o_files[args->ou[0]]).socket, &sock_efds))
        FD_CLR(((struct file * ) td->o_files[args->ou[0]]).socket, args->ex);
    }
  }

   */

  if ((td->td_retval[0] = lwip_select(args->nd, args->in, args->ou, args->ex, args->tv)) == -1)
    error = -1;

  return (error);
}