/*
   Copyright (C) 1995,96,97,99,2000,02,07,17 Free Software Foundation, Inc.
   Written by Michael I. Bushnell, p/BSG.

   This file is part of the GNU Hurd.

   The GNU Hurd 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, or (at
   your option) any later version.

   The GNU Hurd 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 the GNU Hurd.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <lwip-hurd.h>

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <error.h>
#include <fcntl.h>
#include <argp.h>
#include <sys/mman.h>
#include <hurd/trivfs.h>

#include <lwip_io_S.h>
#include <lwip_socket_S.h>
#include <lwip_pfinet_S.h>
#include <lwip_iioctl_S.h>
#include <lwip_rioctl_S.h>
#include <lwip_startup_notify_S.h>

#include <netif/hurdethif.h>
#include <netif/hurdtunif.h>
#include "startup.h"
#include "options.h"

/* Translator initialization */

struct port_bucket *lwip_bucket;
struct port_class *socketport_class;
struct port_class *addrport_class;
struct port_class *shutdown_notify_class;

struct port_class *lwip_protid_portclasses[2];
struct port_class *lwip_cntl_portclasses[2];

int lwip_bootstrap_portclass;

uid_t lwip_owner;
uid_t lwip_group;

mach_port_t fsys_identity;

struct trivfs_control *lwipcntl;

int trivfs_fstype = FSTYPE_MISC;
int trivfs_fsid = 0;
int trivfs_support_read = 0;
int trivfs_support_write = 0;
int trivfs_support_exec = 0;
int trivfs_allow_open = O_READ | O_WRITE;

void
trivfs_modify_stat (struct trivfs_protid *cred, io_statbuf_t * st)
{
}

error_t
trivfs_goaway (struct trivfs_control *fsys, int flags)
{
  if (flags & FSYS_GOAWAY_FORCE)
    exit (0);
  else
    {
      /* Stop new requests.  */
      ports_inhibit_class_rpcs (lwip_cntl_portclasses[0]);
      ports_inhibit_class_rpcs (lwip_protid_portclasses[0]);
      ports_inhibit_class_rpcs (lwip_cntl_portclasses[1]);
      ports_inhibit_class_rpcs (lwip_protid_portclasses[1]);
      ports_inhibit_class_rpcs (socketport_class);
      ports_inhibit_class_rpcs (addrport_class);

      if (ports_count_class (socketport_class) != 0
	  || ports_count_class (addrport_class) != 0)
	{
	  /* We won't go away, so start things going again...  */
	  ports_resume_class_rpcs (addrport_class);
	  ports_resume_class_rpcs (socketport_class);
	  ports_resume_class_rpcs (lwip_cntl_portclasses[1]);
	  ports_resume_class_rpcs (lwip_protid_portclasses[1]);
	  ports_resume_class_rpcs (lwip_cntl_portclasses[0]);
	  ports_resume_class_rpcs (lwip_protid_portclasses[0]);

	  return EBUSY;
	}

      /* There are no sockets, so we can die without breaking anybody
         too badly.  We don't let user ports on the /servers/socket/2
         file keep us alive because those get cached in every process
         that ever makes a PF_INET socket, libc copes with getting
         MACH_SEND_INVALID_DEST and looking up the new translator.  */
      exit (0);
    }
}

int
lwip_demuxer (mach_msg_header_t * inp, mach_msg_header_t * outp)
{
  struct port_info *pi;
  mig_routine_t routine = NULL;

  /* Clear errno to prevent raising previous errors again */
  errno = 0;

  /* We have several classes in one bucket, which need to be demuxed
     differently.  */
  if (MACH_MSGH_BITS_LOCAL (inp->msgh_bits) ==
      MACH_MSG_TYPE_PROTECTED_PAYLOAD)
    pi = ports_lookup_payload (lwip_bucket,
			       inp->msgh_protected_payload, socketport_class);
  else
    pi = ports_lookup_port (lwip_bucket,
			    inp->msgh_local_port, socketport_class);

  if (pi)
    {
      ports_port_deref (pi);
      routine = lwip_io_server_routine (inp);
    }

  if (routine || (routine = lwip_socket_server_routine (inp)) ||
      (routine = lwip_pfinet_server_routine (inp)) ||
      (routine = lwip_rioctl_server_routine (inp)) ||
      (routine = lwip_iioctl_server_routine (inp)) ||
      (routine = lwip_startup_notify_server_routine (inp)))
    {
      (*routine) (inp, outp);
      return TRUE;
    }
  else
    return trivfs_demuxer (inp, outp);
}

void
translator_bind (int portclass, const char *name)
{
  struct trivfs_control *cntl;
  error_t err = 0;
  mach_port_t right;
  file_t file = file_name_lookup (name, O_CREAT | O_NOTRANS, 0666);

  if (file == MACH_PORT_NULL)
    err = errno;

  if (!err)
    {
      if (lwip_protid_portclasses[portclass] != NULL)
	error (1, 0, "Cannot bind one protocol to multiple nodes.\n");

      err =
	trivfs_add_protid_port_class (&lwip_protid_portclasses[portclass]);
      if (err)
	error (1, 0, "error creating control port class");

      err = trivfs_add_control_port_class (&lwip_cntl_portclasses[portclass]);
      if (err)
	error (1, 0, "error creating control port class");

      err = trivfs_create_control (file, lwip_cntl_portclasses[portclass],
				   lwip_bucket,
				   lwip_protid_portclasses[portclass],
				   lwip_bucket, &cntl);
    }

  if (!err)
    {
      right = ports_get_send_right (cntl);
      err = file_set_translator (file, 0, FS_TRANS_EXCL | FS_TRANS_SET,
				 0, 0, 0, right, MACH_MSG_TYPE_COPY_SEND);
      mach_port_deallocate (mach_task_self (), right);
    }

  if (err)
    error (1, err, "%s", name);

  ports_port_deref (cntl);
}

int
main (int argc, char **argv)
{
  error_t err;
  struct stat st;
  mach_port_t bootstrap;

  lwip_bucket = ports_create_bucket ();
  addrport_class = ports_create_class (clean_addrport, 0);
  socketport_class = ports_create_class (clean_socketport, 0);
  lwip_bootstrap_portclass = PORTCLASS_INET;

  mach_port_allocate (mach_task_self (),
		      MACH_PORT_RIGHT_RECEIVE, &fsys_identity);

  /* Init the device modules */
  hurdethif_module_init ();
  hurdtunif_module_init ();

  /* Parse options.  When successful, this configures the interfaces
     before returning */
  argp_parse (&lwip_argp, argc, argv, 0, 0, 0);

  task_get_bootstrap_port (mach_task_self (), &bootstrap);
  if (bootstrap == MACH_PORT_NULL)
    error (1, 0, "Must be started as a translator");

  /* Create portclass to install on the bootstrap port. */
  if (lwip_protid_portclasses[lwip_bootstrap_portclass] != NULL)
    error (1, 0, "No portclass left to assign to bootstrap port");

  err =
    trivfs_add_protid_port_class (&lwip_protid_portclasses
				  [lwip_bootstrap_portclass]);
  if (err)
    error (1, err, "error creating control port class");

  err =
    trivfs_add_control_port_class (&lwip_cntl_portclasses
				   [lwip_bootstrap_portclass]);
  if (err)
    error (1, err, "error creating control port class");

  /* Reply to our parent */
  err = trivfs_startup (bootstrap, 0,
			lwip_cntl_portclasses[lwip_bootstrap_portclass],
			lwip_bucket,
			lwip_protid_portclasses[lwip_bootstrap_portclass],
			lwip_bucket, &lwipcntl);
  mach_port_deallocate (mach_task_self (), bootstrap);
  if (err)
    error (1, err, "trivfs_startup failed");

  /* Initialize status from underlying node.  */
  lwip_owner = lwip_group = 0;
  err = io_stat (lwipcntl->underlying, &st);
  if (!err)
    {
      lwip_owner = st.st_uid;
      lwip_group = st.st_gid;
    }

  /* Ask init to tell us when the system is going down,
     so we can try to be friendly to our correspondents on the network.  */
  arrange_shutdown_notification ();

  ports_manage_port_operations_multithread (lwip_bucket, lwip_demuxer,
					    30 * 1000, 2 * 60 * 1000, 0);

  return 0;
}