socket.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:40k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * NET An implementation of the SOCKET network access protocol.
  3.  *
  4.  * Version: @(#)socket.c 1.1.93 18/02/95
  5.  *
  6.  * Authors: Orest Zborowski, <obz@Kodak.COM>
  7.  * Ross Biro, <bir7@leland.Stanford.Edu>
  8.  * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
  9.  *
  10.  * Fixes:
  11.  * Anonymous : NOTSOCK/BADF cleanup. Error fix in
  12.  * shutdown()
  13.  * Alan Cox : verify_area() fixes
  14.  * Alan Cox : Removed DDI
  15.  * Jonathan Kamens : SOCK_DGRAM reconnect bug
  16.  * Alan Cox : Moved a load of checks to the very
  17.  * top level.
  18.  * Alan Cox : Move address structures to/from user
  19.  * mode above the protocol layers.
  20.  * Rob Janssen : Allow 0 length sends.
  21.  * Alan Cox : Asynchronous I/O support (cribbed from the
  22.  * tty drivers).
  23.  * Niibe Yutaka : Asynchronous I/O for writes (4.4BSD style)
  24.  * Jeff Uphoff : Made max number of sockets command-line
  25.  * configurable.
  26.  * Matti Aarnio : Made the number of sockets dynamic,
  27.  * to be allocated when needed, and mr.
  28.  * Uphoff's max is used as max to be
  29.  * allowed to allocate.
  30.  * Linus : Argh. removed all the socket allocation
  31.  * altogether: it's in the inode now.
  32.  * Alan Cox : Made sock_alloc()/sock_release() public
  33.  * for NetROM and future kernel nfsd type
  34.  * stuff.
  35.  * Alan Cox : sendmsg/recvmsg basics.
  36.  * Tom Dyas : Export net symbols.
  37.  * Marcin Dalecki : Fixed problems with CONFIG_NET="n".
  38.  * Alan Cox : Added thread locking to sys_* calls
  39.  * for sockets. May have errors at the
  40.  * moment.
  41.  * Kevin Buhr : Fixed the dumb errors in the above.
  42.  * Andi Kleen : Some small cleanups, optimizations,
  43.  * and fixed a copy_from_user() bug.
  44.  * Tigran Aivazian : sys_send(args) calls sys_sendto(args, NULL, 0)
  45.  * Tigran Aivazian : Made listen(2) backlog sanity checks 
  46.  * protocol-independent
  47.  *
  48.  *
  49.  * This program is free software; you can redistribute it and/or
  50.  * modify it under the terms of the GNU General Public License
  51.  * as published by the Free Software Foundation; either version
  52.  * 2 of the License, or (at your option) any later version.
  53.  *
  54.  *
  55.  * This module is effectively the top level interface to the BSD socket
  56.  * paradigm. 
  57.  *
  58.  */
  59. #include <linux/config.h>
  60. #include <linux/mm.h>
  61. #include <linux/smp_lock.h>
  62. #include <linux/socket.h>
  63. #include <linux/file.h>
  64. #include <linux/net.h>
  65. #include <linux/interrupt.h>
  66. #include <linux/netdevice.h>
  67. #include <linux/proc_fs.h>
  68. #include <linux/wanrouter.h>
  69. #include <linux/netlink.h>
  70. #include <linux/rtnetlink.h>
  71. #include <linux/init.h>
  72. #include <linux/poll.h>
  73. #include <linux/cache.h>
  74. #include <linux/module.h>
  75. #include <linux/highmem.h>
  76. #if defined(CONFIG_KMOD) && defined(CONFIG_NET)
  77. #include <linux/kmod.h>
  78. #endif
  79. #include <asm/uaccess.h>
  80. #include <net/sock.h>
  81. #include <net/scm.h>
  82. #include <linux/netfilter.h>
  83. static int sock_no_open(struct inode *irrelevant, struct file *dontcare);
  84. static loff_t sock_lseek(struct file *file, loff_t offset, int whence);
  85. static ssize_t sock_read(struct file *file, char *buf,
  86.  size_t size, loff_t *ppos);
  87. static ssize_t sock_write(struct file *file, const char *buf,
  88.   size_t size, loff_t *ppos);
  89. static int sock_mmap(struct file *file, struct vm_area_struct * vma);
  90. static int sock_close(struct inode *inode, struct file *file);
  91. static unsigned int sock_poll(struct file *file,
  92.       struct poll_table_struct *wait);
  93. static int sock_ioctl(struct inode *inode, struct file *file,
  94.       unsigned int cmd, unsigned long arg);
  95. static int sock_fasync(int fd, struct file *filp, int on);
  96. static ssize_t sock_readv(struct file *file, const struct iovec *vector,
  97.   unsigned long count, loff_t *ppos);
  98. static ssize_t sock_writev(struct file *file, const struct iovec *vector,
  99.   unsigned long count, loff_t *ppos);
  100. static ssize_t sock_sendpage(struct file *file, struct page *page,
  101.      int offset, size_t size, loff_t *ppos, int more);
  102. /*
  103.  * Socket files have a set of 'special' operations as well as the generic file ones. These don't appear
  104.  * in the operation structures but are done directly via the socketcall() multiplexor.
  105.  */
  106. static struct file_operations socket_file_ops = {
  107. llseek: sock_lseek,
  108. read: sock_read,
  109. write: sock_write,
  110. poll: sock_poll,
  111. ioctl: sock_ioctl,
  112. mmap: sock_mmap,
  113. open: sock_no_open, /* special open code to disallow open via /proc */
  114. release: sock_close,
  115. fasync: sock_fasync,
  116. readv: sock_readv,
  117. writev: sock_writev,
  118. sendpage: sock_sendpage
  119. };
  120. /*
  121.  * The protocol list. Each protocol is registered in here.
  122.  */
  123. static struct net_proto_family *net_families[NPROTO];
  124. #ifdef CONFIG_SMP
  125. static atomic_t net_family_lockct = ATOMIC_INIT(0);
  126. static spinlock_t net_family_lock = SPIN_LOCK_UNLOCKED;
  127. /* The strategy is: modifications net_family vector are short, do not
  128.    sleep and veeery rare, but read access should be free of any exclusive
  129.    locks.
  130.  */
  131. static void net_family_write_lock(void)
  132. {
  133. spin_lock(&net_family_lock);
  134. while (atomic_read(&net_family_lockct) != 0) {
  135. spin_unlock(&net_family_lock);
  136. current->policy |= SCHED_YIELD;
  137. schedule();
  138. spin_lock(&net_family_lock);
  139. }
  140. }
  141. static __inline__ void net_family_write_unlock(void)
  142. {
  143. spin_unlock(&net_family_lock);
  144. }
  145. static __inline__ void net_family_read_lock(void)
  146. {
  147. atomic_inc(&net_family_lockct);
  148. spin_unlock_wait(&net_family_lock);
  149. }
  150. static __inline__ void net_family_read_unlock(void)
  151. {
  152. atomic_dec(&net_family_lockct);
  153. }
  154. #else
  155. #define net_family_write_lock() do { } while(0)
  156. #define net_family_write_unlock() do { } while(0)
  157. #define net_family_read_lock() do { } while(0)
  158. #define net_family_read_unlock() do { } while(0)
  159. #endif
  160. /*
  161.  * Statistics counters of the socket lists
  162.  */
  163. static union {
  164. int counter;
  165. char __pad[SMP_CACHE_BYTES];
  166. } sockets_in_use[NR_CPUS] __cacheline_aligned = {{0}};
  167. /*
  168.  * Support routines. Move socket addresses back and forth across the kernel/user
  169.  * divide and look after the messy bits.
  170.  */
  171. #define MAX_SOCK_ADDR 128 /* 108 for Unix domain - 
  172.    16 for IP, 16 for IPX,
  173.    24 for IPv6,
  174.    about 80 for AX.25 
  175.    must be at least one bigger than
  176.    the AF_UNIX size (see net/unix/af_unix.c
  177.    :unix_mkname()).  
  178.  */
  179.  
  180. /**
  181.  * move_addr_to_kernel - copy a socket address into kernel space
  182.  * @uaddr: Address in user space
  183.  * @kaddr: Address in kernel space
  184.  * @ulen: Length in user space
  185.  *
  186.  * The address is copied into kernel space. If the provided address is
  187.  * too long an error code of -EINVAL is returned. If the copy gives
  188.  * invalid addresses -EFAULT is returned. On a success 0 is returned.
  189.  */
  190. int move_addr_to_kernel(void *uaddr, int ulen, void *kaddr)
  191. {
  192. if(ulen<0||ulen>MAX_SOCK_ADDR)
  193. return -EINVAL;
  194. if(ulen==0)
  195. return 0;
  196. if(copy_from_user(kaddr,uaddr,ulen))
  197. return -EFAULT;
  198. return 0;
  199. }
  200. /**
  201.  * move_addr_to_user - copy an address to user space
  202.  * @kaddr: kernel space address
  203.  * @klen: length of address in kernel
  204.  * @uaddr: user space address
  205.  * @ulen: pointer to user length field
  206.  *
  207.  * The value pointed to by ulen on entry is the buffer length available.
  208.  * This is overwritten with the buffer space used. -EINVAL is returned
  209.  * if an overlong buffer is specified or a negative buffer size. -EFAULT
  210.  * is returned if either the buffer or the length field are not
  211.  * accessible.
  212.  * After copying the data up to the limit the user specifies, the true
  213.  * length of the data is written over the length limit the user
  214.  * specified. Zero is returned for a success.
  215.  */
  216.  
  217. int move_addr_to_user(void *kaddr, int klen, void *uaddr, int *ulen)
  218. {
  219. int err;
  220. int len;
  221. if((err=get_user(len, ulen)))
  222. return err;
  223. if(len>klen)
  224. len=klen;
  225. if(len<0 || len> MAX_SOCK_ADDR)
  226. return -EINVAL;
  227. if(len)
  228. {
  229. if(copy_to_user(uaddr,kaddr,len))
  230. return -EFAULT;
  231. }
  232. /*
  233.  * "fromlen shall refer to the value before truncation.."
  234.  * 1003.1g
  235.  */
  236. return __put_user(klen, ulen);
  237. }
  238. #define SOCKFS_MAGIC 0x534F434B
  239. static int sockfs_statfs(struct super_block *sb, struct statfs *buf)
  240. {
  241. buf->f_type = SOCKFS_MAGIC;
  242. buf->f_bsize = 1024;
  243. buf->f_namelen = 255;
  244. return 0;
  245. }
  246. static struct super_operations sockfs_ops = {
  247. statfs: sockfs_statfs,
  248. };
  249. static struct super_block * sockfs_read_super(struct super_block *sb, void *data, int silent)
  250. {
  251. struct inode *root = new_inode(sb);
  252. if (!root)
  253. return NULL;
  254. root->i_mode = S_IFDIR | S_IRUSR | S_IWUSR;
  255. root->i_uid = root->i_gid = 0;
  256. root->i_atime = root->i_mtime = root->i_ctime = CURRENT_TIME;
  257. sb->s_blocksize = 1024;
  258. sb->s_blocksize_bits = 10;
  259. sb->s_magic = SOCKFS_MAGIC;
  260. sb->s_op = &sockfs_ops;
  261. sb->s_root = d_alloc(NULL, &(const struct qstr) { "socket:", 7, 0 });
  262. if (!sb->s_root) {
  263. iput(root);
  264. return NULL;
  265. }
  266. sb->s_root->d_sb = sb;
  267. sb->s_root->d_parent = sb->s_root;
  268. d_instantiate(sb->s_root, root);
  269. return sb;
  270. }
  271. static struct vfsmount *sock_mnt;
  272. static DECLARE_FSTYPE(sock_fs_type, "sockfs", sockfs_read_super, FS_NOMOUNT);
  273. static int sockfs_delete_dentry(struct dentry *dentry)
  274. {
  275. return 1;
  276. }
  277. static struct dentry_operations sockfs_dentry_operations = {
  278. d_delete: sockfs_delete_dentry,
  279. };
  280. /*
  281.  * Obtains the first available file descriptor and sets it up for use.
  282.  *
  283.  * This functions creates file structure and maps it to fd space
  284.  * of current process. On success it returns file descriptor
  285.  * and file struct implicitly stored in sock->file.
  286.  * Note that another thread may close file descriptor before we return
  287.  * from this function. We use the fact that now we do not refer
  288.  * to socket after mapping. If one day we will need it, this
  289.  * function will inincrement ref. count on file by 1.
  290.  *
  291.  * In any case returned fd MAY BE not valid!
  292.  * This race condition is inavoidable
  293.  * with shared fd spaces, we cannot solve is inside kernel,
  294.  * but we take care of internal coherence yet.
  295.  */
  296. static int sock_map_fd(struct socket *sock)
  297. {
  298. int fd;
  299. struct qstr this;
  300. char name[32];
  301. /*
  302.  * Find a file descriptor suitable for return to the user. 
  303.  */
  304. fd = get_unused_fd();
  305. if (fd >= 0) {
  306. struct file *file = get_empty_filp();
  307. if (!file) {
  308. put_unused_fd(fd);
  309. fd = -ENFILE;
  310. goto out;
  311. }
  312. sprintf(name, "[%lu]", sock->inode->i_ino);
  313. this.name = name;
  314. this.len = strlen(name);
  315. this.hash = sock->inode->i_ino;
  316. file->f_dentry = d_alloc(sock_mnt->mnt_sb->s_root, &this);
  317. if (!file->f_dentry) {
  318. put_filp(file);
  319. put_unused_fd(fd);
  320. fd = -ENOMEM;
  321. goto out;
  322. }
  323. file->f_dentry->d_op = &sockfs_dentry_operations;
  324. d_add(file->f_dentry, sock->inode);
  325. file->f_vfsmnt = mntget(sock_mnt);
  326. sock->file = file;
  327. file->f_op = sock->inode->i_fop = &socket_file_ops;
  328. file->f_mode = 3;
  329. file->f_flags = O_RDWR;
  330. file->f_pos = 0;
  331. fd_install(fd, file);
  332. }
  333. out:
  334. return fd;
  335. }
  336. extern __inline__ struct socket *socki_lookup(struct inode *inode)
  337. {
  338. return &inode->u.socket_i;
  339. }
  340. /**
  341.  * sockfd_lookup -  Go from a file number to its socket slot
  342.  * @fd: file handle
  343.  * @err: pointer to an error code return
  344.  *
  345.  * The file handle passed in is locked and the socket it is bound
  346.  * too is returned. If an error occurs the err pointer is overwritten
  347.  * with a negative errno code and NULL is returned. The function checks
  348.  * for both invalid handles and passing a handle which is not a socket.
  349.  *
  350.  * On a success the socket object pointer is returned.
  351.  */
  352. struct socket *sockfd_lookup(int fd, int *err)
  353. {
  354. struct file *file;
  355. struct inode *inode;
  356. struct socket *sock;
  357. if (!(file = fget(fd)))
  358. {
  359. *err = -EBADF;
  360. return NULL;
  361. }
  362. inode = file->f_dentry->d_inode;
  363. if (!inode->i_sock || !(sock = socki_lookup(inode)))
  364. {
  365. *err = -ENOTSOCK;
  366. fput(file);
  367. return NULL;
  368. }
  369. if (sock->file != file) {
  370. printk(KERN_ERR "socki_lookup: socket file changed!n");
  371. sock->file = file;
  372. }
  373. return sock;
  374. }
  375. extern __inline__ void sockfd_put(struct socket *sock)
  376. {
  377. fput(sock->file);
  378. }
  379. /**
  380.  * sock_alloc - allocate a socket
  381.  *
  382.  * Allocate a new inode and socket object. The two are bound together
  383.  * and initialised. The socket is then returned. If we are out of inodes
  384.  * NULL is returned.
  385.  */
  386. struct socket *sock_alloc(void)
  387. {
  388. struct inode * inode;
  389. struct socket * sock;
  390. inode = get_empty_inode();
  391. if (!inode)
  392. return NULL;
  393. inode->i_sb = sock_mnt->mnt_sb;
  394. sock = socki_lookup(inode);
  395. inode->i_mode = S_IFSOCK|S_IRWXUGO;
  396. inode->i_sock = 1;
  397. inode->i_uid = current->fsuid;
  398. inode->i_gid = current->fsgid;
  399. sock->inode = inode;
  400. init_waitqueue_head(&sock->wait);
  401. sock->fasync_list = NULL;
  402. sock->state = SS_UNCONNECTED;
  403. sock->flags = 0;
  404. sock->ops = NULL;
  405. sock->sk = NULL;
  406. sock->file = NULL;
  407. sockets_in_use[smp_processor_id()].counter++;
  408. return sock;
  409. }
  410. /*
  411.  * In theory you can't get an open on this inode, but /proc provides
  412.  * a back door. Remember to keep it shut otherwise you'll let the
  413.  * creepy crawlies in.
  414.  */
  415.   
  416. static int sock_no_open(struct inode *irrelevant, struct file *dontcare)
  417. {
  418. return -ENXIO;
  419. }
  420. /**
  421.  * sock_release - close a socket
  422.  * @sock: socket to close
  423.  *
  424.  * The socket is released from the protocol stack if it has a release
  425.  * callback, and the inode is then released if the socket is bound to
  426.  * an inode not a file. 
  427.  */
  428.  
  429. void sock_release(struct socket *sock)
  430. {
  431. if (sock->ops) 
  432. sock->ops->release(sock);
  433. if (sock->fasync_list)
  434. printk(KERN_ERR "sock_release: fasync list not empty!n");
  435. sockets_in_use[smp_processor_id()].counter--;
  436. if (!sock->file) {
  437. iput(sock->inode);
  438. return;
  439. }
  440. sock->file=NULL;
  441. }
  442. int sock_sendmsg(struct socket *sock, struct msghdr *msg, int size)
  443. {
  444. int err;
  445. struct scm_cookie scm;
  446. err = scm_send(sock, msg, &scm);
  447. if (err >= 0) {
  448. err = sock->ops->sendmsg(sock, msg, size, &scm);
  449. scm_destroy(&scm);
  450. }
  451. return err;
  452. }
  453. int sock_recvmsg(struct socket *sock, struct msghdr *msg, int size, int flags)
  454. {
  455. struct scm_cookie scm;
  456. memset(&scm, 0, sizeof(scm));
  457. size = sock->ops->recvmsg(sock, msg, size, flags, &scm);
  458. if (size >= 0)
  459. scm_recv(sock, msg, &scm, flags);
  460. return size;
  461. }
  462. /*
  463.  * Sockets are not seekable.
  464.  */
  465. static loff_t sock_lseek(struct file *file, loff_t offset, int whence)
  466. {
  467. return -ESPIPE;
  468. }
  469. /*
  470.  * Read data from a socket. ubuf is a user mode pointer. We make sure the user
  471.  * area ubuf...ubuf+size-1 is writable before asking the protocol.
  472.  */
  473. static ssize_t sock_read(struct file *file, char *ubuf,
  474.  size_t size, loff_t *ppos)
  475. {
  476. struct socket *sock;
  477. struct iovec iov;
  478. struct msghdr msg;
  479. int flags;
  480. if (ppos != &file->f_pos)
  481. return -ESPIPE;
  482. if (size==0) /* Match SYS5 behaviour */
  483. return 0;
  484. sock = socki_lookup(file->f_dentry->d_inode); 
  485. msg.msg_name=NULL;
  486. msg.msg_namelen=0;
  487. msg.msg_iov=&iov;
  488. msg.msg_iovlen=1;
  489. msg.msg_control=NULL;
  490. msg.msg_controllen=0;
  491. iov.iov_base=ubuf;
  492. iov.iov_len=size;
  493. flags = !(file->f_flags & O_NONBLOCK) ? 0 : MSG_DONTWAIT;
  494. return sock_recvmsg(sock, &msg, size, flags);
  495. }
  496. /*
  497.  * Write data to a socket. We verify that the user area ubuf..ubuf+size-1
  498.  * is readable by the user process.
  499.  */
  500. static ssize_t sock_write(struct file *file, const char *ubuf,
  501.   size_t size, loff_t *ppos)
  502. {
  503. struct socket *sock;
  504. struct msghdr msg;
  505. struct iovec iov;
  506. if (ppos != &file->f_pos)
  507. return -ESPIPE;
  508. if(size==0) /* Match SYS5 behaviour */
  509. return 0;
  510. sock = socki_lookup(file->f_dentry->d_inode); 
  511. msg.msg_name=NULL;
  512. msg.msg_namelen=0;
  513. msg.msg_iov=&iov;
  514. msg.msg_iovlen=1;
  515. msg.msg_control=NULL;
  516. msg.msg_controllen=0;
  517. msg.msg_flags=!(file->f_flags & O_NONBLOCK) ? 0 : MSG_DONTWAIT;
  518. if (sock->type == SOCK_SEQPACKET)
  519. msg.msg_flags |= MSG_EOR;
  520. iov.iov_base=(void *)ubuf;
  521. iov.iov_len=size;
  522. return sock_sendmsg(sock, &msg, size);
  523. }
  524. ssize_t sock_sendpage(struct file *file, struct page *page,
  525.       int offset, size_t size, loff_t *ppos, int more)
  526. {
  527. struct socket *sock;
  528. int flags;
  529. if (ppos != &file->f_pos)
  530. return -ESPIPE;
  531. sock = socki_lookup(file->f_dentry->d_inode);
  532. flags = !(file->f_flags & O_NONBLOCK) ? 0 : MSG_DONTWAIT;
  533. if (more)
  534. flags |= MSG_MORE;
  535. return sock->ops->sendpage(sock, page, offset, size, flags);
  536. }
  537. int sock_readv_writev(int type, struct inode * inode, struct file * file,
  538.       const struct iovec * iov, long count, long size)
  539. {
  540. struct msghdr msg;
  541. struct socket *sock;
  542. sock = socki_lookup(inode);
  543. msg.msg_name = NULL;
  544. msg.msg_namelen = 0;
  545. msg.msg_control = NULL;
  546. msg.msg_controllen = 0;
  547. msg.msg_iov = (struct iovec *) iov;
  548. msg.msg_iovlen = count;
  549. msg.msg_flags = (file->f_flags & O_NONBLOCK) ? MSG_DONTWAIT : 0;
  550. /* read() does a VERIFY_WRITE */
  551. if (type == VERIFY_WRITE)
  552. return sock_recvmsg(sock, &msg, size, msg.msg_flags);
  553. if (sock->type == SOCK_SEQPACKET)
  554. msg.msg_flags |= MSG_EOR;
  555. return sock_sendmsg(sock, &msg, size);
  556. }
  557. static ssize_t sock_readv(struct file *file, const struct iovec *vector,
  558.   unsigned long count, loff_t *ppos)
  559. {
  560. size_t tot_len = 0;
  561. int i;
  562.         for (i = 0 ; i < count ; i++)
  563.                 tot_len += vector[i].iov_len;
  564. return sock_readv_writev(VERIFY_WRITE, file->f_dentry->d_inode,
  565.  file, vector, count, tot_len);
  566. }
  567. static ssize_t sock_writev(struct file *file, const struct iovec *vector,
  568.    unsigned long count, loff_t *ppos)
  569. {
  570. size_t tot_len = 0;
  571. int i;
  572.         for (i = 0 ; i < count ; i++)
  573.                 tot_len += vector[i].iov_len;
  574. return sock_readv_writev(VERIFY_READ, file->f_dentry->d_inode,
  575.  file, vector, count, tot_len);
  576. }
  577. /*
  578.  * With an ioctl arg may well be a user mode pointer, but we don't know what to do
  579.  * with it - that's up to the protocol still.
  580.  */
  581. int sock_ioctl(struct inode *inode, struct file *file, unsigned int cmd,
  582.    unsigned long arg)
  583. {
  584. struct socket *sock;
  585. int err;
  586. unlock_kernel();
  587. sock = socki_lookup(inode);
  588. err = sock->ops->ioctl(sock, cmd, arg);
  589. lock_kernel();
  590. return err;
  591. }
  592. /* No kernel lock held - perfect */
  593. static unsigned int sock_poll(struct file *file, poll_table * wait)
  594. {
  595. struct socket *sock;
  596. /*
  597.  * We can't return errors to poll, so it's either yes or no. 
  598.  */
  599. sock = socki_lookup(file->f_dentry->d_inode);
  600. return sock->ops->poll(file, sock, wait);
  601. }
  602. static int sock_mmap(struct file * file, struct vm_area_struct * vma)
  603. {
  604. struct socket *sock = socki_lookup(file->f_dentry->d_inode);
  605. return sock->ops->mmap(file, sock, vma);
  606. }
  607. int sock_close(struct inode *inode, struct file *filp)
  608. {
  609. /*
  610.  * It was possible the inode is NULL we were 
  611.  * closing an unfinished socket. 
  612.  */
  613. if (!inode)
  614. {
  615. printk(KERN_DEBUG "sock_close: NULL inoden");
  616. return 0;
  617. }
  618. sock_fasync(-1, filp, 0);
  619. sock_release(socki_lookup(inode));
  620. return 0;
  621. }
  622. /*
  623.  * Update the socket async list
  624.  *
  625.  * Fasync_list locking strategy.
  626.  *
  627.  * 1. fasync_list is modified only under process context socket lock
  628.  *    i.e. under semaphore.
  629.  * 2. fasync_list is used under read_lock(&sk->callback_lock)
  630.  *    or under socket lock.
  631.  * 3. fasync_list can be used from softirq context, so that
  632.  *    modification under socket lock have to be enhanced with
  633.  *    write_lock_bh(&sk->callback_lock).
  634.  * --ANK (990710)
  635.  */
  636. static int sock_fasync(int fd, struct file *filp, int on)
  637. {
  638. struct fasync_struct *fa, *fna=NULL, **prev;
  639. struct socket *sock;
  640. struct sock *sk;
  641. if (on)
  642. {
  643. fna=(struct fasync_struct *)kmalloc(sizeof(struct fasync_struct), GFP_KERNEL);
  644. if(fna==NULL)
  645. return -ENOMEM;
  646. }
  647. sock = socki_lookup(filp->f_dentry->d_inode);
  648. if ((sk=sock->sk) == NULL)
  649. return -EINVAL;
  650. lock_sock(sk);
  651. prev=&(sock->fasync_list);
  652. for (fa=*prev; fa!=NULL; prev=&fa->fa_next,fa=*prev)
  653. if (fa->fa_file==filp)
  654. break;
  655. if(on)
  656. {
  657. if(fa!=NULL)
  658. {
  659. write_lock_bh(&sk->callback_lock);
  660. fa->fa_fd=fd;
  661. write_unlock_bh(&sk->callback_lock);
  662. kfree(fna);
  663. goto out;
  664. }
  665. fna->fa_file=filp;
  666. fna->fa_fd=fd;
  667. fna->magic=FASYNC_MAGIC;
  668. fna->fa_next=sock->fasync_list;
  669. write_lock_bh(&sk->callback_lock);
  670. sock->fasync_list=fna;
  671. write_unlock_bh(&sk->callback_lock);
  672. }
  673. else
  674. {
  675. if (fa!=NULL)
  676. {
  677. write_lock_bh(&sk->callback_lock);
  678. *prev=fa->fa_next;
  679. write_unlock_bh(&sk->callback_lock);
  680. kfree(fa);
  681. }
  682. }
  683. out:
  684. release_sock(sock->sk);
  685. return 0;
  686. }
  687. /* This function may be called only under socket lock or callback_lock */
  688. int sock_wake_async(struct socket *sock, int how, int band)
  689. {
  690. if (!sock || !sock->fasync_list)
  691. return -1;
  692. switch (how)
  693. {
  694. case 1:
  695. if (test_bit(SOCK_ASYNC_WAITDATA, &sock->flags))
  696. break;
  697. goto call_kill;
  698. case 2:
  699. if (!test_and_clear_bit(SOCK_ASYNC_NOSPACE, &sock->flags))
  700. break;
  701. /* fall through */
  702. case 0:
  703. call_kill:
  704. __kill_fasync(sock->fasync_list, SIGIO, band);
  705. break;
  706. case 3:
  707. __kill_fasync(sock->fasync_list, SIGURG, band);
  708. }
  709. return 0;
  710. }
  711. int sock_create(int family, int type, int protocol, struct socket **res)
  712. {
  713. int i;
  714. struct socket *sock;
  715. /*
  716.  * Check protocol is in range
  717.  */
  718. if (family < 0 || family >= NPROTO)
  719. return -EAFNOSUPPORT;
  720. if (type < 0 || type >= SOCK_MAX)
  721. return -EINVAL;
  722. /* Compatibility.
  723.    This uglymoron is moved from INET layer to here to avoid
  724.    deadlock in module load.
  725.  */
  726. if (family == PF_INET && type == SOCK_PACKET) {
  727. static int warned; 
  728. if (!warned) {
  729. warned = 1;
  730. printk(KERN_INFO "%s uses obsolete (PF_INET,SOCK_PACKET)n", current->comm);
  731. }
  732. family = PF_PACKET;
  733. }
  734. #if defined(CONFIG_KMOD) && defined(CONFIG_NET)
  735. /* Attempt to load a protocol module if the find failed. 
  736.  * 
  737.  * 12/09/1996 Marcin: But! this makes REALLY only sense, if the user 
  738.  * requested real, full-featured networking support upon configuration.
  739.  * Otherwise module support will break!
  740.  */
  741. if (net_families[family]==NULL)
  742. {
  743. char module_name[30];
  744. sprintf(module_name,"net-pf-%d",family);
  745. request_module(module_name);
  746. }
  747. #endif
  748. net_family_read_lock();
  749. if (net_families[family] == NULL) {
  750. i = -EAFNOSUPPORT;
  751. goto out;
  752. }
  753. /*
  754.  * Allocate the socket and allow the family to set things up. if
  755.  * the protocol is 0, the family is instructed to select an appropriate
  756.  * default.
  757.  */
  758. if (!(sock = sock_alloc())) 
  759. {
  760. printk(KERN_WARNING "socket: no more socketsn");
  761. i = -ENFILE; /* Not exactly a match, but its the
  762.    closest posix thing */
  763. goto out;
  764. }
  765. sock->type  = type;
  766. if ((i = net_families[family]->create(sock, protocol)) < 0) 
  767. {
  768. sock_release(sock);
  769. goto out;
  770. }
  771. *res = sock;
  772. out:
  773. net_family_read_unlock();
  774. return i;
  775. }
  776. asmlinkage long sys_socket(int family, int type, int protocol)
  777. {
  778. int retval;
  779. struct socket *sock;
  780. retval = sock_create(family, type, protocol, &sock);
  781. if (retval < 0)
  782. goto out;
  783. retval = sock_map_fd(sock);
  784. if (retval < 0)
  785. goto out_release;
  786. out:
  787. /* It may be already another descriptor 8) Not kernel problem. */
  788. return retval;
  789. out_release:
  790. sock_release(sock);
  791. return retval;
  792. }
  793. /*
  794.  * Create a pair of connected sockets.
  795.  */
  796. asmlinkage long sys_socketpair(int family, int type, int protocol, int usockvec[2])
  797. {
  798. struct socket *sock1, *sock2;
  799. int fd1, fd2, err;
  800. /*
  801.  * Obtain the first socket and check if the underlying protocol
  802.  * supports the socketpair call.
  803.  */
  804. err = sock_create(family, type, protocol, &sock1);
  805. if (err < 0)
  806. goto out;
  807. err = sock_create(family, type, protocol, &sock2);
  808. if (err < 0)
  809. goto out_release_1;
  810. err = sock1->ops->socketpair(sock1, sock2);
  811. if (err < 0) 
  812. goto out_release_both;
  813. fd1 = fd2 = -1;
  814. err = sock_map_fd(sock1);
  815. if (err < 0)
  816. goto out_release_both;
  817. fd1 = err;
  818. err = sock_map_fd(sock2);
  819. if (err < 0)
  820. goto out_close_1;
  821. fd2 = err;
  822. /* fd1 and fd2 may be already another descriptors.
  823.  * Not kernel problem.
  824.  */
  825. err = put_user(fd1, &usockvec[0]); 
  826. if (!err)
  827. err = put_user(fd2, &usockvec[1]);
  828. if (!err)
  829. return 0;
  830. sys_close(fd2);
  831. sys_close(fd1);
  832. return err;
  833. out_close_1:
  834.         sock_release(sock2);
  835. sys_close(fd1);
  836. return err;
  837. out_release_both:
  838.         sock_release(sock2);
  839. out_release_1:
  840.         sock_release(sock1);
  841. out:
  842. return err;
  843. }
  844. /*
  845.  * Bind a name to a socket. Nothing much to do here since it's
  846.  * the protocol's responsibility to handle the local address.
  847.  *
  848.  * We move the socket address to kernel space before we call
  849.  * the protocol layer (having also checked the address is ok).
  850.  */
  851. asmlinkage long sys_bind(int fd, struct sockaddr *umyaddr, int addrlen)
  852. {
  853. struct socket *sock;
  854. char address[MAX_SOCK_ADDR];
  855. int err;
  856. if((sock = sockfd_lookup(fd,&err))!=NULL)
  857. {
  858. if((err=move_addr_to_kernel(umyaddr,addrlen,address))>=0)
  859. err = sock->ops->bind(sock, (struct sockaddr *)address, addrlen);
  860. sockfd_put(sock);
  861. }
  862. return err;
  863. }
  864. /*
  865.  * Perform a listen. Basically, we allow the protocol to do anything
  866.  * necessary for a listen, and if that works, we mark the socket as
  867.  * ready for listening.
  868.  */
  869. asmlinkage long sys_listen(int fd, int backlog)
  870. {
  871. struct socket *sock;
  872. int err;
  873. if ((sock = sockfd_lookup(fd, &err)) != NULL) {
  874. if ((unsigned) backlog > SOMAXCONN)
  875. backlog = SOMAXCONN;
  876. err=sock->ops->listen(sock, backlog);
  877. sockfd_put(sock);
  878. }
  879. return err;
  880. }
  881. /*
  882.  * For accept, we attempt to create a new socket, set up the link
  883.  * with the client, wake up the client, then return the new
  884.  * connected fd. We collect the address of the connector in kernel
  885.  * space and move it to user at the very end. This is unclean because
  886.  * we open the socket then return an error.
  887.  *
  888.  * 1003.1g adds the ability to recvmsg() to query connection pending
  889.  * status to recvmsg. We need to add that support in a way thats
  890.  * clean when we restucture accept also.
  891.  */
  892. asmlinkage long sys_accept(int fd, struct sockaddr *upeer_sockaddr, int *upeer_addrlen)
  893. {
  894. struct socket *sock, *newsock;
  895. int err, len;
  896. char address[MAX_SOCK_ADDR];
  897. sock = sockfd_lookup(fd, &err);
  898. if (!sock)
  899. goto out;
  900. err = -EMFILE;
  901. if (!(newsock = sock_alloc())) 
  902. goto out_put;
  903. newsock->type = sock->type;
  904. newsock->ops = sock->ops;
  905. err = sock->ops->accept(sock, newsock, sock->file->f_flags);
  906. if (err < 0)
  907. goto out_release;
  908. if (upeer_sockaddr) {
  909. if(newsock->ops->getname(newsock, (struct sockaddr *)address, &len, 2)<0) {
  910. err = -ECONNABORTED;
  911. goto out_release;
  912. }
  913. err = move_addr_to_user(address, len, upeer_sockaddr, upeer_addrlen);
  914. if (err < 0)
  915. goto out_release;
  916. }
  917. /* File flags are not inherited via accept() unlike another OSes. */
  918. if ((err = sock_map_fd(newsock)) < 0)
  919. goto out_release;
  920. out_put:
  921. sockfd_put(sock);
  922. out:
  923. return err;
  924. out_release:
  925. sock_release(newsock);
  926. goto out_put;
  927. }
  928. /*
  929.  * Attempt to connect to a socket with the server address.  The address
  930.  * is in user space so we verify it is OK and move it to kernel space.
  931.  *
  932.  * For 1003.1g we need to add clean support for a bind to AF_UNSPEC to
  933.  * break bindings
  934.  *
  935.  * NOTE: 1003.1g draft 6.3 is broken with respect to AX.25/NetROM and
  936.  * other SEQPACKET protocols that take time to connect() as it doesn't
  937.  * include the -EINPROGRESS status for such sockets.
  938.  */
  939. asmlinkage long sys_connect(int fd, struct sockaddr *uservaddr, int addrlen)
  940. {
  941. struct socket *sock;
  942. char address[MAX_SOCK_ADDR];
  943. int err;
  944. sock = sockfd_lookup(fd, &err);
  945. if (!sock)
  946. goto out;
  947. err = move_addr_to_kernel(uservaddr, addrlen, address);
  948. if (err < 0)
  949. goto out_put;
  950. err = sock->ops->connect(sock, (struct sockaddr *) address, addrlen,
  951.  sock->file->f_flags);
  952. out_put:
  953. sockfd_put(sock);
  954. out:
  955. return err;
  956. }
  957. /*
  958.  * Get the local address ('name') of a socket object. Move the obtained
  959.  * name to user space.
  960.  */
  961. asmlinkage long sys_getsockname(int fd, struct sockaddr *usockaddr, int *usockaddr_len)
  962. {
  963. struct socket *sock;
  964. char address[MAX_SOCK_ADDR];
  965. int len, err;
  966. sock = sockfd_lookup(fd, &err);
  967. if (!sock)
  968. goto out;
  969. err = sock->ops->getname(sock, (struct sockaddr *)address, &len, 0);
  970. if (err)
  971. goto out_put;
  972. err = move_addr_to_user(address, len, usockaddr, usockaddr_len);
  973. out_put:
  974. sockfd_put(sock);
  975. out:
  976. return err;
  977. }
  978. /*
  979.  * Get the remote address ('name') of a socket object. Move the obtained
  980.  * name to user space.
  981.  */
  982. asmlinkage long sys_getpeername(int fd, struct sockaddr *usockaddr, int *usockaddr_len)
  983. {
  984. struct socket *sock;
  985. char address[MAX_SOCK_ADDR];
  986. int len, err;
  987. if ((sock = sockfd_lookup(fd, &err))!=NULL)
  988. {
  989. err = sock->ops->getname(sock, (struct sockaddr *)address, &len, 1);
  990. if (!err)
  991. err=move_addr_to_user(address,len, usockaddr, usockaddr_len);
  992. sockfd_put(sock);
  993. }
  994. return err;
  995. }
  996. /*
  997.  * Send a datagram to a given address. We move the address into kernel
  998.  * space and check the user space data area is readable before invoking
  999.  * the protocol.
  1000.  */
  1001. asmlinkage long sys_sendto(int fd, void * buff, size_t len, unsigned flags,
  1002.    struct sockaddr *addr, int addr_len)
  1003. {
  1004. struct socket *sock;
  1005. char address[MAX_SOCK_ADDR];
  1006. int err;
  1007. struct msghdr msg;
  1008. struct iovec iov;
  1009. sock = sockfd_lookup(fd, &err);
  1010. if (!sock)
  1011. goto out;
  1012. iov.iov_base=buff;
  1013. iov.iov_len=len;
  1014. msg.msg_name=NULL;
  1015. msg.msg_iov=&iov;
  1016. msg.msg_iovlen=1;
  1017. msg.msg_control=NULL;
  1018. msg.msg_controllen=0;
  1019. msg.msg_namelen=0;
  1020. if(addr)
  1021. {
  1022. err = move_addr_to_kernel(addr, addr_len, address);
  1023. if (err < 0)
  1024. goto out_put;
  1025. msg.msg_name=address;
  1026. msg.msg_namelen=addr_len;
  1027. }
  1028. if (sock->file->f_flags & O_NONBLOCK)
  1029. flags |= MSG_DONTWAIT;
  1030. msg.msg_flags = flags;
  1031. err = sock_sendmsg(sock, &msg, len);
  1032. out_put:
  1033. sockfd_put(sock);
  1034. out:
  1035. return err;
  1036. }
  1037. /*
  1038.  * Send a datagram down a socket. 
  1039.  */
  1040. asmlinkage long sys_send(int fd, void * buff, size_t len, unsigned flags)
  1041. {
  1042. return sys_sendto(fd, buff, len, flags, NULL, 0);
  1043. }
  1044. /*
  1045.  * Receive a frame from the socket and optionally record the address of the 
  1046.  * sender. We verify the buffers are writable and if needed move the
  1047.  * sender address from kernel to user space.
  1048.  */
  1049. asmlinkage long sys_recvfrom(int fd, void * ubuf, size_t size, unsigned flags,
  1050.      struct sockaddr *addr, int *addr_len)
  1051. {
  1052. struct socket *sock;
  1053. struct iovec iov;
  1054. struct msghdr msg;
  1055. char address[MAX_SOCK_ADDR];
  1056. int err,err2;
  1057. sock = sockfd_lookup(fd, &err);
  1058. if (!sock)
  1059. goto out;
  1060. msg.msg_control=NULL;
  1061. msg.msg_controllen=0;
  1062. msg.msg_iovlen=1;
  1063. msg.msg_iov=&iov;
  1064. iov.iov_len=size;
  1065. iov.iov_base=ubuf;
  1066. msg.msg_name=address;
  1067. msg.msg_namelen=MAX_SOCK_ADDR;
  1068. if (sock->file->f_flags & O_NONBLOCK)
  1069. flags |= MSG_DONTWAIT;
  1070. err=sock_recvmsg(sock, &msg, size, flags);
  1071. if(err >= 0 && addr != NULL && msg.msg_namelen)
  1072. {
  1073. err2=move_addr_to_user(address, msg.msg_namelen, addr, addr_len);
  1074. if(err2<0)
  1075. err=err2;
  1076. }
  1077. sockfd_put(sock);
  1078. out:
  1079. return err;
  1080. }
  1081. /*
  1082.  * Receive a datagram from a socket. 
  1083.  */
  1084. asmlinkage long sys_recv(int fd, void * ubuf, size_t size, unsigned flags)
  1085. {
  1086. return sys_recvfrom(fd, ubuf, size, flags, NULL, NULL);
  1087. }
  1088. /*
  1089.  * Set a socket option. Because we don't know the option lengths we have
  1090.  * to pass the user mode parameter for the protocols to sort out.
  1091.  */
  1092. asmlinkage long sys_setsockopt(int fd, int level, int optname, char *optval, int optlen)
  1093. {
  1094. int err;
  1095. struct socket *sock;
  1096. if (optlen < 0)
  1097. return -EINVAL;
  1098. if ((sock = sockfd_lookup(fd, &err))!=NULL)
  1099. {
  1100. if (level == SOL_SOCKET)
  1101. err=sock_setsockopt(sock,level,optname,optval,optlen);
  1102. else
  1103. err=sock->ops->setsockopt(sock, level, optname, optval, optlen);
  1104. sockfd_put(sock);
  1105. }
  1106. return err;
  1107. }
  1108. /*
  1109.  * Get a socket option. Because we don't know the option lengths we have
  1110.  * to pass a user mode parameter for the protocols to sort out.
  1111.  */
  1112. asmlinkage long sys_getsockopt(int fd, int level, int optname, char *optval, int *optlen)
  1113. {
  1114. int err;
  1115. struct socket *sock;
  1116. if ((sock = sockfd_lookup(fd, &err))!=NULL)
  1117. {
  1118. if (level == SOL_SOCKET)
  1119. err=sock_getsockopt(sock,level,optname,optval,optlen);
  1120. else
  1121. err=sock->ops->getsockopt(sock, level, optname, optval, optlen);
  1122. sockfd_put(sock);
  1123. }
  1124. return err;
  1125. }
  1126. /*
  1127.  * Shutdown a socket.
  1128.  */
  1129. asmlinkage long sys_shutdown(int fd, int how)
  1130. {
  1131. int err;
  1132. struct socket *sock;
  1133. if ((sock = sockfd_lookup(fd, &err))!=NULL)
  1134. {
  1135. err=sock->ops->shutdown(sock, how);
  1136. sockfd_put(sock);
  1137. }
  1138. return err;
  1139. }
  1140. /*
  1141.  * BSD sendmsg interface
  1142.  */
  1143. asmlinkage long sys_sendmsg(int fd, struct msghdr *msg, unsigned flags)
  1144. {
  1145. struct socket *sock;
  1146. char address[MAX_SOCK_ADDR];
  1147. struct iovec iovstack[UIO_FASTIOV], *iov = iovstack;
  1148. unsigned char ctl[sizeof(struct cmsghdr) + 20]; /* 20 is size of ipv6_pktinfo */
  1149. unsigned char *ctl_buf = ctl;
  1150. struct msghdr msg_sys;
  1151. int err, ctl_len, iov_size, total_len;
  1152. err = -EFAULT;
  1153. if (copy_from_user(&msg_sys,msg,sizeof(struct msghdr)))
  1154. goto out; 
  1155. sock = sockfd_lookup(fd, &err);
  1156. if (!sock) 
  1157. goto out;
  1158. /* do not move before msg_sys is valid */
  1159. err = -EINVAL;
  1160. if (msg_sys.msg_iovlen > UIO_MAXIOV)
  1161. goto out_put;
  1162. /* Check whether to allocate the iovec area*/
  1163. err = -ENOMEM;
  1164. iov_size = msg_sys.msg_iovlen * sizeof(struct iovec);
  1165. if (msg_sys.msg_iovlen > UIO_FASTIOV) {
  1166. iov = sock_kmalloc(sock->sk, iov_size, GFP_KERNEL);
  1167. if (!iov)
  1168. goto out_put;
  1169. }
  1170. /* This will also move the address data into kernel space */
  1171. err = verify_iovec(&msg_sys, iov, address, VERIFY_READ);
  1172. if (err < 0) 
  1173. goto out_freeiov;
  1174. total_len = err;
  1175. err = -ENOBUFS;
  1176. if (msg_sys.msg_controllen > INT_MAX)
  1177. goto out_freeiov;
  1178. ctl_len = msg_sys.msg_controllen; 
  1179. if (ctl_len) 
  1180. {
  1181. if (ctl_len > sizeof(ctl))
  1182. {
  1183. ctl_buf = sock_kmalloc(sock->sk, ctl_len, GFP_KERNEL);
  1184. if (ctl_buf == NULL) 
  1185. goto out_freeiov;
  1186. }
  1187. err = -EFAULT;
  1188. if (copy_from_user(ctl_buf, msg_sys.msg_control, ctl_len))
  1189. goto out_freectl;
  1190. msg_sys.msg_control = ctl_buf;
  1191. }
  1192. msg_sys.msg_flags = flags;
  1193. if (sock->file->f_flags & O_NONBLOCK)
  1194. msg_sys.msg_flags |= MSG_DONTWAIT;
  1195. err = sock_sendmsg(sock, &msg_sys, total_len);
  1196. out_freectl:
  1197. if (ctl_buf != ctl)    
  1198. sock_kfree_s(sock->sk, ctl_buf, ctl_len);
  1199. out_freeiov:
  1200. if (iov != iovstack)
  1201. sock_kfree_s(sock->sk, iov, iov_size);
  1202. out_put:
  1203. sockfd_put(sock);
  1204. out:       
  1205. return err;
  1206. }
  1207. /*
  1208.  * BSD recvmsg interface
  1209.  */
  1210. asmlinkage long sys_recvmsg(int fd, struct msghdr *msg, unsigned int flags)
  1211. {
  1212. struct socket *sock;
  1213. struct iovec iovstack[UIO_FASTIOV];
  1214. struct iovec *iov=iovstack;
  1215. struct msghdr msg_sys;
  1216. unsigned long cmsg_ptr;
  1217. int err, iov_size, total_len, len;
  1218. /* kernel mode address */
  1219. char addr[MAX_SOCK_ADDR];
  1220. /* user mode address pointers */
  1221. struct sockaddr *uaddr;
  1222. int *uaddr_len;
  1223. err=-EFAULT;
  1224. if (copy_from_user(&msg_sys,msg,sizeof(struct msghdr)))
  1225. goto out;
  1226. sock = sockfd_lookup(fd, &err);
  1227. if (!sock)
  1228. goto out;
  1229. err = -EINVAL;
  1230. if (msg_sys.msg_iovlen > UIO_MAXIOV)
  1231. goto out_put;
  1232. /* Check whether to allocate the iovec area*/
  1233. err = -ENOMEM;
  1234. iov_size = msg_sys.msg_iovlen * sizeof(struct iovec);
  1235. if (msg_sys.msg_iovlen > UIO_FASTIOV) {
  1236. iov = sock_kmalloc(sock->sk, iov_size, GFP_KERNEL);
  1237. if (!iov)
  1238. goto out_put;
  1239. }
  1240. /*
  1241.  * Save the user-mode address (verify_iovec will change the
  1242.  * kernel msghdr to use the kernel address space)
  1243.  */
  1244.  
  1245. uaddr = msg_sys.msg_name;
  1246. uaddr_len = &msg->msg_namelen;
  1247. err = verify_iovec(&msg_sys, iov, addr, VERIFY_WRITE);
  1248. if (err < 0)
  1249. goto out_freeiov;
  1250. total_len=err;
  1251. cmsg_ptr = (unsigned long)msg_sys.msg_control;
  1252. msg_sys.msg_flags = 0;
  1253. if (sock->file->f_flags & O_NONBLOCK)
  1254. flags |= MSG_DONTWAIT;
  1255. err = sock_recvmsg(sock, &msg_sys, total_len, flags);
  1256. if (err < 0)
  1257. goto out_freeiov;
  1258. len = err;
  1259. if (uaddr != NULL && msg_sys.msg_namelen) {
  1260. err = move_addr_to_user(addr, msg_sys.msg_namelen, uaddr, uaddr_len);
  1261. if (err < 0)
  1262. goto out_freeiov;
  1263. }
  1264. err = __put_user(msg_sys.msg_flags, &msg->msg_flags);
  1265. if (err)
  1266. goto out_freeiov;
  1267. err = __put_user((unsigned long)msg_sys.msg_control-cmsg_ptr, 
  1268.  &msg->msg_controllen);
  1269. if (err)
  1270. goto out_freeiov;
  1271. err = len;
  1272. out_freeiov:
  1273. if (iov != iovstack)
  1274. sock_kfree_s(sock->sk, iov, iov_size);
  1275. out_put:
  1276. sockfd_put(sock);
  1277. out:
  1278. return err;
  1279. }
  1280. /*
  1281.  * Perform a file control on a socket file descriptor.
  1282.  *
  1283.  * Doesn't acquire a fd lock, because no network fcntl
  1284.  * function sleeps currently.
  1285.  */
  1286. int sock_fcntl(struct file *filp, unsigned int cmd, unsigned long arg)
  1287. {
  1288. struct socket *sock;
  1289. sock = socki_lookup (filp->f_dentry->d_inode);
  1290. if (sock && sock->ops)
  1291. return sock_no_fcntl(sock, cmd, arg);
  1292. return(-EINVAL);
  1293. }
  1294. /* Argument list sizes for sys_socketcall */
  1295. #define AL(x) ((x) * sizeof(unsigned long))
  1296. static unsigned char nargs[18]={AL(0),AL(3),AL(3),AL(3),AL(2),AL(3),
  1297. AL(3),AL(3),AL(4),AL(4),AL(4),AL(6),
  1298. AL(6),AL(2),AL(5),AL(5),AL(3),AL(3)};
  1299. #undef AL
  1300. /*
  1301.  * System call vectors. 
  1302.  *
  1303.  * Argument checking cleaned up. Saved 20% in size.
  1304.  *  This function doesn't need to set the kernel lock because
  1305.  *  it is set by the callees. 
  1306.  */
  1307. asmlinkage long sys_socketcall(int call, unsigned long *args)
  1308. {
  1309. unsigned long a[6];
  1310. unsigned long a0,a1;
  1311. int err;
  1312. if(call<1||call>SYS_RECVMSG)
  1313. return -EINVAL;
  1314. /* copy_from_user should be SMP safe. */
  1315. if (copy_from_user(a, args, nargs[call]))
  1316. return -EFAULT;
  1317. a0=a[0];
  1318. a1=a[1];
  1319. switch(call) 
  1320. {
  1321. case SYS_SOCKET:
  1322. err = sys_socket(a0,a1,a[2]);
  1323. break;
  1324. case SYS_BIND:
  1325. err = sys_bind(a0,(struct sockaddr *)a1, a[2]);
  1326. break;
  1327. case SYS_CONNECT:
  1328. err = sys_connect(a0, (struct sockaddr *)a1, a[2]);
  1329. break;
  1330. case SYS_LISTEN:
  1331. err = sys_listen(a0,a1);
  1332. break;
  1333. case SYS_ACCEPT:
  1334. err = sys_accept(a0,(struct sockaddr *)a1, (int *)a[2]);
  1335. break;
  1336. case SYS_GETSOCKNAME:
  1337. err = sys_getsockname(a0,(struct sockaddr *)a1, (int *)a[2]);
  1338. break;
  1339. case SYS_GETPEERNAME:
  1340. err = sys_getpeername(a0, (struct sockaddr *)a1, (int *)a[2]);
  1341. break;
  1342. case SYS_SOCKETPAIR:
  1343. err = sys_socketpair(a0,a1, a[2], (int *)a[3]);
  1344. break;
  1345. case SYS_SEND:
  1346. err = sys_send(a0, (void *)a1, a[2], a[3]);
  1347. break;
  1348. case SYS_SENDTO:
  1349. err = sys_sendto(a0,(void *)a1, a[2], a[3],
  1350.  (struct sockaddr *)a[4], a[5]);
  1351. break;
  1352. case SYS_RECV:
  1353. err = sys_recv(a0, (void *)a1, a[2], a[3]);
  1354. break;
  1355. case SYS_RECVFROM:
  1356. err = sys_recvfrom(a0, (void *)a1, a[2], a[3],
  1357.    (struct sockaddr *)a[4], (int *)a[5]);
  1358. break;
  1359. case SYS_SHUTDOWN:
  1360. err = sys_shutdown(a0,a1);
  1361. break;
  1362. case SYS_SETSOCKOPT:
  1363. err = sys_setsockopt(a0, a1, a[2], (char *)a[3], a[4]);
  1364. break;
  1365. case SYS_GETSOCKOPT:
  1366. err = sys_getsockopt(a0, a1, a[2], (char *)a[3], (int *)a[4]);
  1367. break;
  1368. case SYS_SENDMSG:
  1369. err = sys_sendmsg(a0, (struct msghdr *) a1, a[2]);
  1370. break;
  1371. case SYS_RECVMSG:
  1372. err = sys_recvmsg(a0, (struct msghdr *) a1, a[2]);
  1373. break;
  1374. default:
  1375. err = -EINVAL;
  1376. break;
  1377. }
  1378. return err;
  1379. }
  1380. /*
  1381.  * This function is called by a protocol handler that wants to
  1382.  * advertise its address family, and have it linked into the
  1383.  * SOCKET module.
  1384.  */
  1385. int sock_register(struct net_proto_family *ops)
  1386. {
  1387. int err;
  1388. if (ops->family >= NPROTO) {
  1389. printk(KERN_CRIT "protocol %d >= NPROTO(%d)n", ops->family, NPROTO);
  1390. return -ENOBUFS;
  1391. }
  1392. net_family_write_lock();
  1393. err = -EEXIST;
  1394. if (net_families[ops->family] == NULL) {
  1395. net_families[ops->family]=ops;
  1396. err = 0;
  1397. }
  1398. net_family_write_unlock();
  1399. return err;
  1400. }
  1401. /*
  1402.  * This function is called by a protocol handler that wants to
  1403.  * remove its address family, and have it unlinked from the
  1404.  * SOCKET module.
  1405.  */
  1406. int sock_unregister(int family)
  1407. {
  1408. if (family < 0 || family >= NPROTO)
  1409. return -1;
  1410. net_family_write_lock();
  1411. net_families[family]=NULL;
  1412. net_family_write_unlock();
  1413. return 0;
  1414. }
  1415. extern void sk_init(void);
  1416. #ifdef CONFIG_WAN_ROUTER
  1417. extern void wanrouter_init(void);
  1418. #endif
  1419. #ifdef CONFIG_BLUEZ
  1420. extern void bluez_init(void);
  1421. #endif
  1422. void __init sock_init(void)
  1423. {
  1424. int i;
  1425. printk(KERN_INFO "Linux NET4.0 for Linux 2.4n");
  1426. printk(KERN_INFO "Based upon Swansea University Computer Society NET3.039n");
  1427. /*
  1428.  * Initialize all address (protocol) families. 
  1429.  */
  1430.  
  1431. for (i = 0; i < NPROTO; i++) 
  1432. net_families[i] = NULL;
  1433. /*
  1434.  * Initialize sock SLAB cache.
  1435.  */
  1436.  
  1437. sk_init();
  1438. #ifdef SLAB_SKB
  1439. /*
  1440.  * Initialize skbuff SLAB cache 
  1441.  */
  1442. skb_init();
  1443. #endif
  1444. /*
  1445.  * Wan router layer. 
  1446.  */
  1447. #ifdef CONFIG_WAN_ROUTER  
  1448. wanrouter_init();
  1449. #endif
  1450. /*
  1451.  * Initialize the protocols module. 
  1452.  */
  1453. register_filesystem(&sock_fs_type);
  1454. sock_mnt = kern_mount(&sock_fs_type);
  1455. /* The real protocol initialization is performed when
  1456.  *  do_initcalls is run.  
  1457.  */
  1458. /*
  1459.  * The netlink device handler may be needed early.
  1460.  */
  1461. #ifdef CONFIG_NET
  1462. rtnetlink_init();
  1463. #endif
  1464. #ifdef CONFIG_NETLINK_DEV
  1465. init_netlink();
  1466. #endif
  1467. #ifdef CONFIG_NETFILTER
  1468. netfilter_init();
  1469. #endif
  1470. #ifdef CONFIG_BLUEZ
  1471. bluez_init();
  1472. #endif
  1473. }
  1474. int socket_get_info(char *buffer, char **start, off_t offset, int length)
  1475. {
  1476. int len, cpu;
  1477. int counter = 0;
  1478. for (cpu=0; cpu<smp_num_cpus; cpu++)
  1479. counter += sockets_in_use[cpu_logical_map(cpu)].counter;
  1480. /* It can be negative, by the way. 8) */
  1481. if (counter < 0)
  1482. counter = 0;
  1483. len = sprintf(buffer, "sockets: used %dn", counter);
  1484. if (offset >= len)
  1485. {
  1486. *start = buffer;
  1487. return 0;
  1488. }
  1489. *start = buffer + offset;
  1490. len -= offset;
  1491. if (len > length)
  1492. len = length;
  1493. if (len < 0)
  1494. len = 0;
  1495. return len;
  1496. }