/*-
 * 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: [%i]", uap->fd);
  }

  return (0x0);
}

int sys_fcntl(struct thread *td, struct sys_fcntl_args *uap) {
  return (fcntl(td, uap));
}

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;
    case TIOCGWINSZ:
      asm("nop");
      struct winsize *win = (struct winsize *) args->data;
      win->ws_row = 50;
      win->ws_col = 80;
    break;
    default:
      kprintf("ioFD:%i:0x%X!", 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);
}