Chinaunix首页 | 论坛 | 博客
  • 博客访问: 641165
  • 博文数量: 363
  • 博客积分: 110
  • 博客等级: 民兵
  • 技术积分: 1347
  • 用 户 组: 普通用户
  • 注册时间: 2011-06-22 16:07
文章分类

全部博文(363)

文章存档

2018年(83)

2016年(1)

2014年(2)

2013年(34)

2012年(236)

2011年(7)

分类: LINUX

2018-01-31 12:15:27

作者:gfree.wind@gmail.com

sock_sendmsg的代码很简单
  1. int sock_sendmsg(struct socket *sock, struct msghdr *msg, size_t size)
  2. {
  3.      struct kiocb iocb;
  4.      struct sock_iocb siocb;
  5.      int ret;

  6.      init_sync_kiocb(&iocb, NULL);
  7.      iocb.private = &siocb;
  8.      ret = __sock_sendmsg(&iocb, sock, msg, size);
  9.      if (-EIOCBQUEUED == ret)
  10.           ret = wait_on_sync_kiocb(&iocb);
  11.      return ret;
  12. }
首先定义了一个struct kiocb类型的iocb——linux内核中所有I/O操作都要依赖于合格结构,然后初始化它。然后调用__sock_sendmsg,而__sock_sendmsg又调用UDP的sendmsg去做真正的发送。
也就是说,对于UDP的socket来说,sendto调用,真正去做工作的是udp_sendmsg这个函数。
  1. int udp_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
  2.           size_t len)
  3. {
  4.      struct inet_sock *inet = inet_sk(sk);
  5.      struct udp_sock *up = udp_sk(sk);
  6.      int ulen = len;
  7.      struct ipcm_cookie ipc;
  8.      struct rtable *rt = NULL;
  9.      int free = 0;
  10.      int connected = 0;
  11.      __be32 daddr, faddr, saddr;
  12.      __be16 dport;
  13.      u8 tos;
  14.      int err, is_udplite = IS_UDPLITE(sk);
  15.      int corkreq = up->corkflag || msg->msg_flags&MSG_MORE;
  16.      int (*getfrag)(void *, char *, int, int, int, struct sk_buff *);
将sk转为udp内部使用的类型指针,在TCP/IP中充斥了这样的转换。主要原因是因为对于上层来说,需要一个统一的类型,而到了底层的具体实现时,都会将上层抽象的数据类型,转为自己所需的类型。
并判断该UDP是否是udplite——udplite的定义见
  1.      if (len > 0xFFFF)
  2.           return -EMSGSIZE;

  3.      /*
  4.      * Check the flags.
  5.      */

  6.      if (msg->msg_flags & MSG_OOB) /* Mirror BSD error message compatibility */
  7.           return -EOPNOTSUPP;
进行一些错误检测。
  1.      ipc.opt = NULL;
  2.      ipc.shtx.flags = 0;

  3.      if (up->pending) {
  4.           /*
  5.           * There are pending frames.
  6.           * The socket lock must be held while it's corked.
  7.           */
  8.           lock_sock(sk);
  9.           if (likely(up->pending)) {
  10.                if (unlikely(up->pending != AF_INET)) {
  11.                     release_sock(sk);
  12.                     return -EINVAL;
  13.                }
  14.                goto do_append_data;
  15.           }
  16.           release_sock(sk);
  17.      }
  18.      ulen += sizeof(struct udphdr);
如果该socket有pending的frame,那么直接将数据追加。如果没有就ulen加上udp首部的长度。
  1.      /*
  2.      * Get and verify the address.
  3.      */
  4.      if (msg->msg_name) {
  5.           struct sockaddr_in * usin = (struct sockaddr_in *)msg->msg_name;
  6.           if (msg->msg_namelen < sizeof(*usin))
  7.                return -EINVAL;
  8.           if (usin->sin_family != AF_INET) {
  9.                if (usin->sin_family != AF_UNSPEC)
  10.                     return -EAFNOSUPPORT;
  11.           }

  12.           daddr = usin->sin_addr.s_addr;
  13.           dport = usin->sin_port;
  14.           if (dport == 0)
  15.                return -EINVAL;
  16.      } else {
  17.           if (sk->sk_state != TCP_ESTABLISHED)
  18.                return -EDESTADDRREQ;
  19.           daddr = inet->inet_daddr;
  20.           dport = inet->inet_dport;
  21.           /* Open fast path for connected socket.
  22.              Route will not be used, if at least one option is set.
  23.           */
  24.           connected = 1;
  25.      }
  26.      ipc.addr = inet->inet_saddr;
如果msg->msg_name不为空,就说明指定了目的地址,对其进行检验。如果为空,就就需要对sock的状态进行检验,查看其是否是连接状态——UDP的socket同样是可以调用connect,这样就不需要每次都指定发送地址了。

今天有点困了,就这样了。



阅读(494) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~