/*
 * NET          An implementation of the SOCKET network access protocol.
 *
 * Version:     @(#)socket.c    1.1.93  18/02/95
 *
 * Authors:     Orest Zborowski, <obz@Kodak.COM>
 *              Ross Biro, <bir7@leland.Stanford.Edu>
 *              Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
 *
 * Fixes:
 *              Anonymous       :       NOTSOCK/BADF cleanup. Error fix in
 *                                      shutdown()
 *              Alan Cox        :       verify_area() fixes
 *              Alan Cox        :       Removed DDI
 *              Jonathan Kamens :       SOCK_DGRAM reconnect bug
 *              Alan Cox        :       Moved a load of checks to the very
 *                                      top level.
 *              Alan Cox        :       Move address structures to/from user
 *                                      mode above the protocol layers.
 *              Rob Janssen     :       Allow 0 length sends.
 *              Alan Cox        :       Asynchronous I/O support (cribbed from the
 *                                      tty drivers).
 *              Niibe Yutaka    :       Asynchronous I/O for writes (4.4BSD style)
 *              Jeff Uphoff     :       Made max number of sockets command-line
 *                                      configurable.
 *              Matti Aarnio    :       Made the number of sockets dynamic,
 *                                      to be allocated when needed, and mr.
 *                                      Uphoff's max is used as max to be
 *                                      allowed to allocate.
 *              Linus           :       Argh. removed all the socket allocation
 *                                      altogether: it's in the inode now.
 *              Alan Cox        :       Made sock_alloc()/sock_release() public
 *                                      for NetROM and future kernel nfsd type
 *                                      stuff.
 *              Alan Cox        :       sendmsg/recvmsg basics.
 *              Tom Dyas        :       Export net symbols.
 *              Marcin Dalecki  :       Fixed problems with CONFIG_NET="n".
 *              Alan Cox        :       Added thread locking to sys_* calls
 *                                      for sockets. May have errors at the
 *                                      moment.
 *              Kevin Buhr      :       Fixed the dumb errors in the above.
 *              Andi Kleen      :       Some small cleanups, optimizations,
 *                                      and fixed a copy_from_user() bug.
 *
 *
 *              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 module is effectively the top level interface to the BSD socket
 *      paradigm. 
 *
 */

#include <linux/config.h>
#include <linux/signal.h>
#include <linux/errno.h>
#include <linux/sched.h>
#include <linux/mm.h>
#include <linux/smp.h>
#include <linux/smp_lock.h>
#include <linux/kernel.h>
#include <linux/major.h>
#include <linux/stat.h>
#include <linux/socket.h>
#include <linux/fcntl.h>
#include <linux/file.h>
#include <linux/net.h>
#include <linux/interrupt.h>
#include <linux/netdevice.h>
#include <linux/proc_fs.h>
#include <linux/firewall.h>
#include <linux/wanrouter.h>
#include <linux/init.h>

#if defined(CONFIG_KERNELD) && defined(CONFIG_NET)
#include <linux/kerneld.h>
#endif

#include <net/netlink.h>

#include <asm/system.h>
#include <asm/uaccess.h>

#include <linux/inet.h>
#include <linux/netdevice.h>
#include <net/ip.h>
#include <net/protocol.h>
#include <net/rarp.h>
#include <net/tcp.h>
#include <net/udp.h>
#include <linux/skbuff.h>
#include <net/sock.h>
#include <net/scm.h>


static long long sock_lseek(struct file *file, long long offset, int whence);
static long sock_read(struct inode *inode, struct file *file,
                      char *buf, unsigned long size);
static long sock_write(struct inode *inode, struct file *file,
                       const char *buf, unsigned long size);

static int sock_close(struct inode *inode, struct file *file);
static unsigned int sock_poll(struct file *file, poll_table *wait);
static int sock_ioctl(struct inode *inode, struct file *file,
                      unsigned int cmd, unsigned long arg);
static int sock_fasync(struct file *filp, int on);


/*
 *      Socket files have a set of 'special' operations as well as the generic file ones. These don't appear
 *      in the operation structures but are done directly via the socketcall() multiplexor.
 */

static struct file_operations socket_file_ops = {
        sock_lseek,
        sock_read,
        sock_write,
        NULL,                   /* readdir */
        sock_poll,
        sock_ioctl,
        NULL,                   /* mmap */
        NULL,                   /* no special open code... */
        sock_close,
        NULL,                   /* no fsync */
        sock_fasync
};

/*
 *      The protocol list. Each protocol is registered in here.
 */

struct net_proto_family *net_families[NPROTO];

/*
 *      Statistics counters of the socket lists
 */

static int sockets_in_use  = 0;

/*
 *      Support routines. Move socket addresses back and forth across the kernel/user
 *      divide and look after the messy bits.
 */

#define MAX_SOCK_ADDR   128             /* 108 for Unix domain - 
                                           16 for IP, 16 for IPX,
                                           24 for IPv6,
                                           about 80 for AX.25 
                                           must be at least one bigger than
                                           the AF_UNIX size (see net/unix/af_unix.c
                                           :unix_mkname()).  
                                         */
 
int move_addr_to_kernel(void *uaddr, int ulen, void *kaddr)
{
        if(ulen<0||ulen>MAX_SOCK_ADDR)
                return -EINVAL;
        if(ulen==0)
                return 0;
        if(copy_from_user(kaddr,uaddr,ulen))
                return -EFAULT;
        return 0;
}

int move_addr_to_user(void *kaddr, int klen, void *uaddr, int *ulen)
{
        int err;
        int len;

        if((err=get_user(len, ulen)))
                return err;
        if(len>klen)
                len=klen;
        if(len<0 || len> MAX_SOCK_ADDR)
                return -EINVAL;
        if(len)
        {
                if(copy_to_user(uaddr,kaddr,len))
                        return -EFAULT;
        }
        /*
         *      "fromlen shall refer to the value before truncation.."
         *                      1003.1g
         */
        return __put_user(klen, ulen);
}

/*
 *      Obtains the first available file descriptor and sets it up for use. 
 */

static int get_fd(struct inode *inode)
{
        int fd;

        /*
         *      Find a file descriptor suitable for return to the user. 
         */

        fd = get_unused_fd();
        if (fd >= 0) {
                struct file *file = get_empty_filp();

                if (!file) {
                        put_unused_fd(fd);
                        return -ENFILE;
                }

                file->f_dentry = d_alloc_root(inode, NULL);
                if (!file->f_dentry) {
                        put_filp(file);
                        put_unused_fd(fd);
                        return -ENOMEM;
                }

                /*
                 * The socket maintains a reference to the inode, so we
                 * have to increment the count.
                 */
                inode->i_count++;

                current->files->fd[fd] = file;
                file->f_op = &socket_file_ops;
                file->f_mode = 3;
                file->f_flags = O_RDWR;
                file->f_pos = 0;
        }
        return fd;
}

extern __inline__ struct socket *socki_lookup(struct inode *inode)
{
        return &inode->u.socket_i;
}

/*
 *      Go from a file number to its socket slot.
 */

extern __inline__ struct socket *sockfd_lookup(int fd, int *err)
{
        struct file *file;
        struct inode *inode;

        if (!(file = fget(fd)))
        {
                *err = -EBADF;
                return NULL;
        }

        inode = file->f_dentry->d_inode;
        if (!inode || !inode->i_sock || !socki_lookup(inode))
        {
                *err = -ENOTSOCK;
                fput(file);
                return NULL;
        }

        return socki_lookup(inode);
}

extern __inline__ void sockfd_put(struct socket *sock)
{
        fput(sock->file);
}

/*
 *      Allocate a socket.
 */

struct socket *sock_alloc(void)
{
        struct inode * inode;
        struct socket * sock;

        inode = get_empty_inode();
        if (!inode)
                return NULL;

        sock = socki_lookup(inode);

        inode->i_mode = S_IFSOCK;
        inode->i_sock = 1;
        inode->i_uid = current->uid;
        inode->i_gid = current->gid;

        sock->inode = inode;
        init_waitqueue(&sock->wait);
        sock->fasync_list = NULL;
        sock->state = SS_UNCONNECTED;
        sock->flags = 0;
        sock->ops = NULL;
        sock->sk = NULL;
        sock->file = NULL;

        sockets_in_use++;
        return sock;
}

void sock_release(struct socket *sock)
{
        int oldstate;

        if ((oldstate = sock->state) != SS_UNCONNECTED)
                sock->state = SS_DISCONNECTING;

        if (sock->ops) 
                sock->ops->release(sock, NULL);

        --sockets_in_use;       /* Bookkeeping.. */
        sock->file=NULL;
        iput(sock->inode);
}

int sock_sendmsg(struct socket *sock, struct msghdr *msg, int size)
{
        int err;
        struct scm_cookie scm;

        err = scm_send(sock, msg, &scm);
        if (err < 0)
                return err;

        err = sock->ops->sendmsg(sock, msg, size, &scm);

        scm_destroy(&scm);

        return err;
}

int sock_recvmsg(struct socket *sock, struct msghdr *msg, int size, int flags)
{
        struct scm_cookie scm;

        memset(&scm, 0, sizeof(scm));

        size = sock->ops->recvmsg(sock, msg, size, flags, &scm);

        if (size < 0)
                return size;

        scm_recv(sock, msg, &scm, flags);

        return size;
}


/*
 *      Sockets are not seekable.
 */

static long long sock_lseek(struct file *file,long long offset, int whence)
{
        return -ESPIPE;
}

/*
 *      Read data from a socket. ubuf is a user mode pointer. We make sure the user
 *      area ubuf...ubuf+size-1 is writable before asking the protocol.
 */

static long sock_read(struct inode *inode, struct file *file,
                      char *ubuf, unsigned long size)
{
        struct socket *sock;
        struct iovec iov;
        struct msghdr msg;

        sock = socki_lookup(inode); 
  
        if (size==0)            /* Match SYS5 behaviour */
                return 0;

        msg.msg_name=NULL;
        msg.msg_namelen=0;
        msg.msg_iov=&iov;
        msg.msg_iovlen=1;
        msg.msg_control=NULL;
        msg.msg_controllen=0;
        iov.iov_base=ubuf;
        iov.iov_len=size;

        return sock_recvmsg(sock, &msg, size,
                            !(file->f_flags & O_NONBLOCK) ? 0 : MSG_DONTWAIT);
}


/*
 *      Write data to a socket. We verify that the user area ubuf..ubuf+size-1 is
 *      readable by the user process.
 */

static long sock_write(struct inode *inode, struct file *file,
                       const char *ubuf, unsigned long size)
{
        struct socket *sock;
        struct msghdr msg;
        struct iovec iov;
        
        sock = socki_lookup(inode); 

        if(size==0)             /* Match SYS5 behaviour */
                return 0;

        msg.msg_name=NULL;
        msg.msg_namelen=0;
        msg.msg_iov=&iov;
        msg.msg_iovlen=1;
        msg.msg_control=NULL;
        msg.msg_controllen=0;
        msg.msg_flags=!(file->f_flags & O_NONBLOCK) ? 0 : MSG_DONTWAIT;
        iov.iov_base=(void *)ubuf;
        iov.iov_len=size;
        
        return sock_sendmsg(sock, &msg, size);
}

int sock_readv_writev(int type, struct inode * inode, struct file * file,
                      const struct iovec * iov, long count, long size)
{
        struct msghdr msg;
        struct socket *sock;

        sock = socki_lookup(inode);

        msg.msg_name = NULL;
        msg.msg_namelen = 0;
        msg.msg_control = NULL;
        msg.msg_controllen = 0;
        msg.msg_iov = (struct iovec *) iov;
        msg.msg_iovlen = count;
        msg.msg_flags = (file->f_flags & O_NONBLOCK) ? MSG_DONTWAIT : 0;

        /* read() does a VERIFY_WRITE */
        if (type == VERIFY_WRITE)
                return sock_recvmsg(sock, &msg, size, msg.msg_flags);
        return sock_sendmsg(sock, &msg, size);
}


/*
 *      With an ioctl arg may well be a user mode pointer, but we don't know what to do
 *      with it - that's up to the protocol still.
 */

int sock_ioctl(struct inode *inode, struct file *file, unsigned int cmd,
           unsigned long arg)
{
        struct socket *sock = socki_lookup(inode);
        return sock->ops->ioctl(sock, cmd, arg);
}


static unsigned int sock_poll(struct file *file, poll_table * wait)
{
        struct socket *sock;

        sock = socki_lookup(file->f_dentry->d_inode);

        /*
         *      We can't return errors to poll, so it's either yes or no. 
         */

        return sock->ops->poll(sock, wait);
}


int sock_close(struct inode *inode, struct file *filp)
{
        /*
         *      It was possible the inode is NULL we were 
         *      closing an unfinished socket. 
         */

        if (!inode)
        {
                printk(KERN_DEBUG "sock_close: NULL inode\n");
                return 0;
        }
        sock_fasync(filp, 0);
        sock_release(socki_lookup(inode));
        return 0;
}

/*
 *      Update the socket async list
 */
 
static int sock_fasync(struct file *filp, int on)
{
        struct fasync_struct *fa, *fna=NULL, **prev;
        struct socket *sock;
        unsigned long flags;
        
        if (on)
        {
                fna=(struct fasync_struct *)kmalloc(sizeof(struct fasync_struct), GFP_KERNEL);
                if(fna==NULL)
                        return -ENOMEM;
        }

        sock = socki_lookup(filp->f_dentry->d_inode);
        
        prev=&(sock->fasync_list);
        
        save_flags(flags);
        cli();
        
        for (fa=*prev; fa!=NULL; prev=&fa->fa_next,fa=*prev)
                if (fa->fa_file==filp)
                        break;
        
        if(on)
        {
                if(fa!=NULL)
                {
                        kfree_s(fna,sizeof(struct fasync_struct));
                        restore_flags(flags);
                        return 0;
                }
                fna->fa_file=filp;
                fna->magic=FASYNC_MAGIC;
                fna->fa_next=sock->fasync_list;
                sock->fasync_list=fna;
        }
        else
        {
                if (fa!=NULL)
                {
                        *prev=fa->fa_next;
                        kfree_s(fa,sizeof(struct fasync_struct));
                }
        }
        restore_flags(flags);
        return 0;
}

int sock_wake_async(struct socket *sock, int how)
{
        if (!sock || !sock->fasync_list)
                return -1;
        switch (how)
        {
                case 0:
                        kill_fasync(sock->fasync_list, SIGIO);
                        break;
                case 1:
                        if (!(sock->flags & SO_WAITDATA))
                                kill_fasync(sock->fasync_list, SIGIO);
                        break;
                case 2:
                        if (sock->flags & SO_NOSPACE)
                        {
                                kill_fasync(sock->fasync_list, SIGIO);
                                sock->flags &= ~SO_NOSPACE;
                        }
                        break;
        }
        return 0;
}


int sock_create(int family, int type, int protocol, struct socket **res)
{
        int i;
        struct socket *sock;

        /*
         *      Check protocol is in range
         */
        if(family<0||family>=NPROTO)
                return -EINVAL;
                
#if defined(CONFIG_KERNELD) && defined(CONFIG_NET)
        /* Attempt to load a protocol module if the find failed. 
         * 
         * 12/09/1996 Marcin: But! this makes REALLY only sense, if the user 
         * requested real, full-featured networking support upon configuration.
         * Otherwise module support will break!
         */
        if (net_families[family]==NULL)
        {
                char module_name[30];
                sprintf(module_name,"net-pf-%d",family);
                request_module(module_name);
        }
#endif

        if (net_families[family]==NULL)
                return -EINVAL;

/*
 *      Check that this is a type that we know how to manipulate and
 *      the protocol makes sense here. The family can still reject the
 *      protocol later.
 */
  
        if ((type != SOCK_STREAM && type != SOCK_DGRAM &&
             type != SOCK_SEQPACKET && type != SOCK_RAW && type != SOCK_RDM &&
#ifdef CONFIG_XTP
                type != SOCK_WEB  &&
#endif
             type != SOCK_PACKET) || protocol < 0)
                        return -EINVAL;

/*
 *      Allocate the socket and allow the family to set things up. if
 *      the protocol is 0, the family is instructed to select an appropriate
 *      default.
 */

        if (!(sock = sock_alloc())) 
        {
                printk(KERN_WARNING "socket: no more sockets\n");
                return -ENFILE;         /* Not exactly a match, but its the
                                           closest posix thing */
        }

        sock->type   = type;

        if ((i = net_families[family]->create(sock, protocol)) < 0) 
        {
                sock_release(sock);
                return i;
        }

        *res = sock;
        return 0;
}

asmlinkage int sys_socket(int family, int type, int protocol)
{
        int retval;
        struct socket *sock;

        lock_kernel();

        retval = sock_create(family, type, protocol, &sock);
        if (retval < 0)
                goto out;

        retval = get_fd(sock->inode);
        if (retval < 0) {
                sock_release(sock);
                goto out;
        }

        sock->file = current->files->fd[retval];
out:
        unlock_kernel();
        return retval;
}

/*
 *      Create a pair of connected sockets.
 */

asmlinkage int sys_socketpair(int family, int type, int protocol, int usockvec[2])
{
        int fd1, fd2, i;
        struct socket *sock1=NULL, *sock2=NULL;
        int err;

        lock_kernel();

        /*
         * Obtain the first socket and check if the underlying protocol
         * supports the socketpair call.
         */

        if ((fd1 = sys_socket(family, type, protocol)) < 0) {
                err = fd1;
                goto out;
        }

        sock1 = sockfd_lookup(fd1, &err);
        if (!sock1)
                goto out;
        /*
         *      Now grab another socket and try to connect the two together. 
         */
        err = -EINVAL;
        if ((fd2 = sys_socket(family, type, protocol)) < 0) 
        {
                sys_close(fd1);
                goto out;
        }

        sock2 = sockfd_lookup(fd2,&err);
        if (!sock2)
                goto out;
        if ((i = sock1->ops->socketpair(sock1, sock2)) < 0) 
        {
                sys_close(fd1);
                sys_close(fd2);
                err = i;
        }
        else
        {
                err = put_user(fd1, &usockvec[0]); 
                if (!err) 
                        err = put_user(fd2, &usockvec[1]);
                if (err) {
                        sys_close(fd1);
                        sys_close(fd2);
                }
        }
out:
        if(sock1)
                sockfd_put(sock1);
        if(sock2)
                sockfd_put(sock2);
        unlock_kernel();
        return err;
}


/*
 *      Bind a name to a socket. Nothing much to do here since it's
 *      the protocol's responsibility to handle the local address.
 *
 *      We move the socket address to kernel space before we call
 *      the protocol layer (having also checked the address is ok).
 */
 
asmlinkage int sys_bind(int fd, struct sockaddr *umyaddr, int addrlen)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        int err;

        lock_kernel();
        if((sock = sockfd_lookup(fd,&err))!=NULL)
        {
                if((err=move_addr_to_kernel(umyaddr,addrlen,address))>=0)
                        err = sock->ops->bind(sock, (struct sockaddr *)address, addrlen);
                sockfd_put(sock);
        }                       
        unlock_kernel();
        return err;
}


/*
 *      Perform a listen. Basically, we allow the protocol to do anything
 *      necessary for a listen, and if that works, we mark the socket as
 *      ready for listening.
 */

asmlinkage int sys_listen(int fd, int backlog)
{
        struct socket *sock;
        int err;
        
        lock_kernel();
        if((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                err=sock->ops->listen(sock, backlog);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}


/*
 *      For accept, we attempt to create a new socket, set up the link
 *      with the client, wake up the client, then return the new
 *      connected fd. We collect the address of the connector in kernel
 *      space and move it to user at the very end. This is unclean because
 *      we open the socket then return an error.
 *
 *      1003.1g adds the ability to recvmsg() to query connection pending
 *      status to recvmsg. We need to add that support in a way thats
 *      clean when we restucture accept also.
 */

asmlinkage int sys_accept(int fd, struct sockaddr *upeer_sockaddr, int *upeer_addrlen)
{
        struct inode *inode;
        struct socket *sock, *newsock;
        int err;
        char address[MAX_SOCK_ADDR];
        int len;

        lock_kernel();
restart:
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if (!(newsock = sock_alloc())) 
                {
                        err=-EMFILE;
                        goto out;
                }

                inode = newsock->inode;
                newsock->type = sock->type;

                if ((err = sock->ops->dup(newsock, sock)) < 0) 
                {
                        sock_release(newsock);
                        goto out;
                }

                err = newsock->ops->accept(sock, newsock, current->files->fd[fd]->f_flags);

                if (err < 0)
                {
                        sock_release(newsock);
                        goto out;
                }
                newsock = socki_lookup(inode);

                if ((err = get_fd(inode)) < 0) 
                {
                        sock_release(newsock);
                        err=-EINVAL;
                        goto out;
                }

                newsock->file = current->files->fd[err];
        
                if (upeer_sockaddr)
                {
                        /* Handle the race where the accept works and we
                           then getname after it has closed again */
                        if(newsock->ops->getname(newsock, (struct sockaddr *)address, &len, 1)<0)
                        {
                                sys_close(err);
                                goto restart;
                        }
                        move_addr_to_user(address,len, upeer_sockaddr, upeer_addrlen);
                }
out:
                sockfd_put(sock);               
        }
        unlock_kernel();
        return err;
}


/*
 *      Attempt to connect to a socket with the server address.  The address
 *      is in user space so we verify it is OK and move it to kernel space.
 *
 *      For 1003.1g we need to add clean support for a bind to AF_UNSPEC to
 *      break bindings
 *
 *      NOTE: 1003.1g draft 6.3 is broken with respect to AX.25/NetROM and
 *      other SEQPACKET protocols that take time to connect() as it doesn't
 *      include the -EINPROGRESS status for such sockets.
 */
 
asmlinkage int sys_connect(int fd, struct sockaddr *uservaddr, int addrlen)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        int err;

        lock_kernel();
        if ((sock = sockfd_lookup(fd,&err))!=NULL)
        {
                if((err=move_addr_to_kernel(uservaddr,addrlen,address))>=0)
                        err = sock->ops->connect(sock, (struct sockaddr *)address, addrlen,
                             current->files->fd[fd]->f_flags);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      Get the local address ('name') of a socket object. Move the obtained
 *      name to user space.
 */

asmlinkage int sys_getsockname(int fd, struct sockaddr *usockaddr, int *usockaddr_len)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        int len;
        int err;
        
        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if((err=sock->ops->getname(sock, (struct sockaddr *)address, &len, 0))==0)
                        err=move_addr_to_user(address,len, usockaddr, usockaddr_len);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      Get the remote address ('name') of a socket object. Move the obtained
 *      name to user space.
 */
 
asmlinkage int sys_getpeername(int fd, struct sockaddr *usockaddr, int *usockaddr_len)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        int len;
        int err;

        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if((err=sock->ops->getname(sock, (struct sockaddr *)address, &len, 1))==0)
                        err=move_addr_to_user(address,len, usockaddr, usockaddr_len);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      Send a datagram down a socket. The datagram as with write() is
 *      in user space. We check it can be read.
 */

asmlinkage int sys_send(int fd, void * buff, size_t len, unsigned flags)
{
        struct socket *sock;
        int err;
        struct msghdr msg;
        struct iovec iov;

        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if(len>=0)
                {
                        iov.iov_base=buff;
                        iov.iov_len=len;
                        msg.msg_name=NULL;
                        msg.msg_namelen=0;
                        msg.msg_iov=&iov;
                        msg.msg_iovlen=1;
                        msg.msg_control=NULL;
                        msg.msg_controllen=0;
                        if (current->files->fd[fd]->f_flags & O_NONBLOCK)
                                flags |= MSG_DONTWAIT;
                        msg.msg_flags=flags;
                        err=sock_sendmsg(sock, &msg, len);
                }
                else
                        err=-EINVAL;
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      Send a datagram to a given address. We move the address into kernel
 *      space and check the user space data area is readable before invoking
 *      the protocol.
 */

asmlinkage int sys_sendto(int fd, void * buff, size_t len, unsigned flags,
           struct sockaddr *addr, int addr_len)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        int err;
        struct msghdr msg;
        struct iovec iov;
        
        lock_kernel();
        if ((sock = sockfd_lookup(fd,&err))!=NULL)
        {
                iov.iov_base=buff;
                iov.iov_len=len;
                msg.msg_name=NULL;
                msg.msg_iov=&iov;
                msg.msg_iovlen=1;
                msg.msg_control=NULL;
                msg.msg_controllen=0;
                msg.msg_namelen=addr_len;
                if(addr)
                {
                        err=move_addr_to_kernel(addr,addr_len,address);
                        if (err < 0)
                                goto bad;
                        msg.msg_name=address;
                }
                if (current->files->fd[fd]->f_flags & O_NONBLOCK)
                        flags |= MSG_DONTWAIT;
                msg.msg_flags=flags;
                err=sock_sendmsg(sock, &msg, len);
bad:            
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;

}



/*
 *      Receive a frame from the socket and optionally record the address of the 
 *      sender. We verify the buffers are writable and if needed move the
 *      sender address from kernel to user space.
 */

asmlinkage int sys_recvfrom(int fd, void * ubuf, size_t size, unsigned flags,
             struct sockaddr *addr, int *addr_len)
{
        struct socket *sock;
        struct iovec iov;
        struct msghdr msg;
        char address[MAX_SOCK_ADDR];
        int err,err2;

        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {  
                msg.msg_control=NULL;
                msg.msg_controllen=0;
                msg.msg_iovlen=1;
                msg.msg_iov=&iov;
                iov.iov_len=size;
                iov.iov_base=ubuf;
                msg.msg_name=address;
                msg.msg_namelen=MAX_SOCK_ADDR;
                err=sock_recvmsg(sock, &msg, size,
                          (current->files->fd[fd]->f_flags & O_NONBLOCK) ? (flags | MSG_DONTWAIT) : flags);
                if(err>=0 && addr!=NULL)
                {
                        err2=move_addr_to_user(address, msg.msg_namelen, addr, addr_len);
                        if(err2<0)
                                err=err2;
                }
                sockfd_put(sock);                       
        }               
        unlock_kernel();
        return err;
}

/*
 *      Receive a datagram from a socket. 
 */

asmlinkage int sys_recv(int fd, void * ubuf, size_t size, unsigned flags)
{
        return sys_recvfrom(fd,ubuf,size,flags, NULL, NULL);
}

/*
 *      Set a socket option. Because we don't know the option lengths we have
 *      to pass the user mode parameter for the protocols to sort out.
 */
 
asmlinkage int sys_setsockopt(int fd, int level, int optname, char *optval, int optlen)
{
        int err;
        struct socket *sock;
        
        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if (level == SOL_SOCKET)
                        err=sock_setsockopt(sock,level,optname,optval,optlen);
                else
                        err=sock->ops->setsockopt(sock, level, optname, optval, optlen);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      Get a socket option. Because we don't know the option lengths we have
 *      to pass a user mode parameter for the protocols to sort out.
 */

asmlinkage int sys_getsockopt(int fd, int level, int optname, char *optval, int *optlen)
{
        int err;
        struct socket *sock;

        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                if (level == SOL_SOCKET)
                        err=sock_getsockopt(sock,level,optname,optval,optlen);
                else
                        err=sock->ops->getsockopt(sock, level, optname, optval, optlen);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}


/*
 *      Shutdown a socket.
 */
 
asmlinkage int sys_shutdown(int fd, int how)
{
        int err;
        struct socket *sock;

        lock_kernel();
        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                err=sock->ops->shutdown(sock, how);
                sockfd_put(sock);
        }
        unlock_kernel();
        return err;
}

/*
 *      BSD sendmsg interface
 */
 
asmlinkage int sys_sendmsg(int fd, struct msghdr *msg, unsigned flags)
{
        struct socket *sock;
        char address[MAX_SOCK_ADDR];
        struct iovec iov[UIO_FASTIOV];
        unsigned char ctl[sizeof(struct cmsghdr) + 20]; /* 20 is size of ipv6_pktinfo */
        struct msghdr msg_sys;
        int err= -EINVAL;
        int total_len;
        unsigned char *ctl_buf = ctl;
        
        lock_kernel();

        if (copy_from_user(&msg_sys,msg,sizeof(struct msghdr)))
        {
                err=-EFAULT;
                goto out; 
        }
        /* do not move before msg_sys is valid */
        if (msg_sys.msg_iovlen>UIO_MAXIOV)
                goto out;
        /* This will also move the address data into kernel space */
        err = verify_iovec(&msg_sys, iov, address, VERIFY_READ);
        if (err < 0)
                goto out;
        total_len=err;

        if (msg_sys.msg_controllen) 
        {
                /* XXX We just limit the buffer and assume that the 
                 * skbuff accounting stops it from going too far.
                 * I hope this is correct.
                 */
                if (msg_sys.msg_controllen > sizeof(ctl) &&
                        msg_sys.msg_controllen <= 256)
                {
                        ctl_buf = kmalloc(msg_sys.msg_controllen, GFP_KERNEL);
                        if (ctl_buf == NULL) 
                        {
                                err = -ENOBUFS;
                                goto failed2;
                        }
                }
                if (copy_from_user(ctl_buf, msg_sys.msg_control, 
                                            msg_sys.msg_controllen)) {
                        err = -EFAULT;
                        goto failed;
                }
                msg_sys.msg_control = ctl_buf;
        }
        msg_sys.msg_flags = flags;
        if (current->files->fd[fd]->f_flags & O_NONBLOCK)
                msg_sys.msg_flags |= MSG_DONTWAIT;

        if ((sock = sockfd_lookup(fd,&err))!=NULL)
        {
                err = sock_sendmsg(sock, &msg_sys, total_len);
                sockfd_put(sock);
        }

failed:
        if (ctl_buf != ctl)
                kfree_s(ctl_buf, msg_sys.msg_controllen);
failed2:
        if (msg_sys.msg_iov != iov)
                kfree(msg_sys.msg_iov);
out:
        unlock_kernel();
        return err;
}

/*
 *      BSD recvmsg interface
 */
 
asmlinkage int sys_recvmsg(int fd, struct msghdr *msg, unsigned int flags)
{
        struct socket *sock;
        struct iovec iovstack[UIO_FASTIOV];
        struct iovec *iov=iovstack;
        struct msghdr msg_sys;
        unsigned long cmsg_ptr;
        int err;
        int total_len;
        int len = 0;

        /* kernel mode address */
        char addr[MAX_SOCK_ADDR];

        /* user mode address pointers */
        struct sockaddr *uaddr;
        int *uaddr_len;
        
        lock_kernel();
        if (copy_from_user(&msg_sys,msg,sizeof(struct msghdr)))
        {
                err=-EFAULT;
                goto out;
        }
        if (msg_sys.msg_iovlen>UIO_MAXIOV)
        {
                err=-EINVAL;
                goto out;
        }
        
        /*
         *      Save the user-mode address (verify_iovec will change the
         *      kernel msghdr to use the kernel address space)
         */
         
        uaddr = msg_sys.msg_name;
        uaddr_len = &msg->msg_namelen;
        err=verify_iovec(&msg_sys, iov, addr, VERIFY_WRITE);
        if (err<0)
                goto out;

        total_len=err;

        cmsg_ptr = (unsigned long)msg_sys.msg_control;
        msg_sys.msg_flags = 0;
        
        if (current->files->fd[fd]->f_flags&O_NONBLOCK)
                flags |= MSG_DONTWAIT;

        if ((sock = sockfd_lookup(fd, &err))!=NULL)
        {
                err=sock_recvmsg(sock, &msg_sys, total_len, flags);
                if(err>=0)
                        len=err;
                sockfd_put(sock);
        }
        if (msg_sys.msg_iov != iov)
                kfree(msg_sys.msg_iov);

        if (uaddr != NULL && err>=0)
                err = move_addr_to_user(addr, msg_sys.msg_namelen, uaddr, uaddr_len);
        if (err>=0) {
                err = __put_user(msg_sys.msg_flags, &msg->msg_flags);
                if (!err)
                        err = __put_user((unsigned long)msg_sys.msg_control-cmsg_ptr, 
                                                         &msg->msg_controllen);
        }
out:
        unlock_kernel();
        if(err<0)
                return err;
        return len;
}


/*
 *      Perform a file control on a socket file descriptor.
 *
 *      FIXME: does this need an fd lock ?
 */

int sock_fcntl(struct file *filp, unsigned int cmd, unsigned long arg)
{
        struct socket *sock;

        sock = socki_lookup (filp->f_dentry->d_inode);
        if (sock && sock->ops)
                return sock->ops->fcntl(sock, cmd, arg);
        return(-EINVAL);
}

/* Argument list sizes for sys_socketcall */
#define AL(x) ((x) * sizeof(unsigned long))
static unsigned char nargs[18]={AL(0),AL(3),AL(3),AL(3),AL(2),AL(3),
                                                                AL(3),AL(3),AL(4),AL(4),AL(4),AL(6),
                                                                AL(6),AL(2),AL(5),AL(5),AL(3),AL(3)};
#undef AL

/*
 *      System call vectors. 
 *
 *      Argument checking cleaned up. Saved 20% in size.
 *  This function doesn't need to set the kernel lock because
 *  it is set by the callees. 
 */

asmlinkage int sys_socketcall(int call, unsigned long *args)
{
        unsigned long a[6];
        unsigned long a0,a1;
        int err;

        if(call<1||call>SYS_RECVMSG)
                return -EINVAL;

        /* copy_from_user should be SMP safe. */
        if (copy_from_user(a, args, nargs[call]))
                return -EFAULT;
                
        a0=a[0];
        a1=a[1];
        
        switch(call) 
        {
                case SYS_SOCKET:
                        err = sys_socket(a0,a1,a[2]);
                        break;
                case SYS_BIND:
                        err = sys_bind(a0,(struct sockaddr *)a1, a[2]);
                        break;
                case SYS_CONNECT:
                        err = sys_connect(a0, (struct sockaddr *)a1, a[2]);
                        break;
                case SYS_LISTEN:
                        err = sys_listen(a0,a1);
                        break;
                case SYS_ACCEPT:
                        err = sys_accept(a0,(struct sockaddr *)a1, (int *)a[2]);
                        break;
                case SYS_GETSOCKNAME:
                        err = sys_getsockname(a0,(struct sockaddr *)a1, (int *)a[2]);
                        break;
                case SYS_GETPEERNAME:
                        err = sys_getpeername(a0, (struct sockaddr *)a1, (int *)a[2]);
                        break;
                case SYS_SOCKETPAIR:
                        err = sys_socketpair(a0,a1, a[2], (int *)a[3]);
                        break;
                case SYS_SEND:
                        err = sys_send(a0, (void *)a1, a[2], a[3]);
                        break;
                case SYS_SENDTO:
                        err = sys_sendto(a0,(void *)a1, a[2], a[3],
                                         (struct sockaddr *)a[4], a[5]);
                        break;
                case SYS_RECV:
                        err = sys_recv(a0, (void *)a1, a[2], a[3]);
                        break;
                case SYS_RECVFROM:
                        err = sys_recvfrom(a0, (void *)a1, a[2], a[3],
                                           (struct sockaddr *)a[4], (int *)a[5]);
                        break;
                case SYS_SHUTDOWN:
                        err = sys_shutdown(a0,a1);
                        break;
                case SYS_SETSOCKOPT:
                        err = sys_setsockopt(a0, a1, a[2], (char *)a[3], a[4]);
                        break;
                case SYS_GETSOCKOPT:
                        err = sys_getsockopt(a0, a1, a[2], (char *)a[3], (int *)a[4]);
                        break;
                case SYS_SENDMSG:
                        err = sys_sendmsg(a0, (struct msghdr *) a1, a[2]);
                        break;
                case SYS_RECVMSG:
                        err = sys_recvmsg(a0, (struct msghdr *) a1, a[2]);
                        break;
                default:
                        err = -EINVAL;
                        break;
        }
        return err;
}

/*
 *      This function is called by a protocol handler that wants to
 *      advertise its address family, and have it linked into the
 *      SOCKET module.
 */
 
int sock_register(struct net_proto_family *ops)
{
        if (ops->family < 0 || ops->family >= NPROTO)
                return -1;

        net_families[ops->family]=ops;
        return 0;
}

/*
 *      This function is called by a protocol handler that wants to
 *      remove its address family, and have it unlinked from the
 *      SOCKET module.
 */
 
int sock_unregister(int family)
{
        if (family < 0 || family >= NPROTO)
                return -1;

        net_families[family]=NULL;
        return 0;
}

__initfunc(void proto_init(void))
{
        extern struct net_proto protocols[];    /* Network protocols */
        struct net_proto *pro;

        /* Kick all configured protocols. */
        pro = protocols;
        while (pro->name != NULL) 
        {
                (*pro->init_func)(pro);
                pro++;
        }
        /* We're all done... */
}

extern void sk_init(void);

__initfunc(void sock_init(void))
{
        int i;

        printk(KERN_INFO "Swansea University Computer Society NET3.039 for Linux 2.1\n");

        /*
         *      Initialize all address (protocol) families. 
         */
         
        for (i = 0; i < NPROTO; i++) 
                net_families[i] = NULL;

        /*
         *      Initialize sock SLAB cache.
         */
         
        sk_init();
        
        /*
         *      The netlink device handler may be needed early.
         */

#ifdef CONFIG_NETLINK
        init_netlink();
#endif

        /*
         *      Wan router layer. 
         */

#ifdef CONFIG_WAN_ROUTER         
        wanrouter_init();
#endif

        /*
         *      Attach the firewall module if configured
         */
         
#ifdef CONFIG_FIREWALL   
        fwchain_init();
#endif

        /*
         *      Initialize the protocols module. 
         */

        proto_init();
}

int socket_get_info(char *buffer, char **start, off_t offset, int length)
{
        int len = sprintf(buffer, "sockets: used %d\n", sockets_in_use);
        if (offset >= len)
        {
                *start = buffer;
                return 0;
        }
        *start = buffer + offset;
        len -= offset;
        if (len > length)
                len = length;
        return len;
}