TCP/IP源码(22)——tcp_sendmsg(1)

2022-01-17 10:53:26

本文的copyleft归gfree.wind@gmail.com所有，使用GPL发布，可以*拷贝，转载。但转载请保持文档的完整性，注明原作者及原链接，严禁用于任何商业用途。
作者：gfree.wind@gmail.com
博客：linuxfocus.blog.chinaunix.net    

今天开始学习tcp_sendmsg
  1. int tcp_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg,
  2.         size_t size)
  3. {
  4.     struct iovec *iov;
  5.     /*
  6.     从通用的struct sock *sk得到struct tcp_sock *tp
  7.     其实只是一个强制类型转换，因为strcut sock为所有其它socket类型的第一个成员，所有可以直接对指针进行     强制类型转换
  8.     */
  9.     struct tcp_sock *tp = tcp_sk(sk);
  10.     struct sk_buff *skb;
  11.     int iovlen, flags;
  12.     int mss_now, size_goal;
  13.     int sg, err, copied;
  14.     long timeo;

  16.     lock_sock(sk);
  17.     TCP_CHECK_TIMER(sk);

  19.     flags = msg->msg_flags;
  20.     /* 
  21.        设置发送等待时间，如果设置了DONTWAIT，则timeo为0. 
  22.        如果没有该标志，则timeo就为sock->sk_sndtimeo
  23.        */
  24.     timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);

  26.     /* Wait for a connection to finish. */
  27.     /* 等到直到连接完成 */
  28.     if ((1 sk->sk_state) & ~(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT))
  29.         if ((err = sk_stream_wait_connect(sk, &timeo)) != 0)
  30.             goto out_err;

  32.     /* This should be in poll */
  33.     clear_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
     /* 计算当前的MSS， size_goal为可发送数据的大小 */
  1.     mss_now = tcp_send_mss(sk, &size_goal, flags);

  3.     /* Ok commence sending. */
  4.     iovlen = msg->msg_iovlen;
  5.     iov = msg->msg_iov;
  6.     copied = 0;

  8.     err = -EPIPE;
  9.     if (sk->sk_err || (sk->sk_shutdown & SEND_SHUTDOWN))
  10.         goto out_err;

  12.     sg = sk->sk_route_caps & NETIF_F_SG;
     /* 开始发送数据 */
  1.     while (--iovlen >= 0) {
  2.         size_t seglen = iov->iov_len;
  3.         unsigned char __user *from = iov->iov_base;

  5.         iov++;

  7.         while (seglen > 0) {
  8.             int copy = 0;
  9.             int max = size_goal;

  11.             skb = tcp_write_queue_tail(sk);
  12.             /* 计算需要复制的字节数为copy */
  13.             if (tcp_send_head(sk)) {
  14.                 if (skb->ip_summed == CHECKSUM_NONE)
  15.                     max = mss_now;
  16.                 copy = max - skb->len;
  17.             }

  19.             if (copy = 0) {
  20.                 /* 已复制完以前的数据了 */
  21. new_segment:
  22.                 /* Allocate new segment. If the interface is SG,
  23.                  * allocate skb fitting to single page.
  24.                  */
                    /* 检验是否还有空闲send buffer，若没有则跳到wait_for_sndbuf */
  25.                 if (!sk_stream_memory_free(sk))
  26.                     goto wait_for_sndbuf;
                 /* 申请新的skb buffer */
  1.                 skb = sk_stream_alloc_skb(sk,
  2.                              select_size(sk, sg),
  3.                              sk->sk_allocation);
  4.                 /* 若分配失败，则跳转到wait_for_memory */
  5.                 if (!skb)
  6.                     goto wait_for_memory;

  8.                 /*
  9.                  * Check whether we can use HW checksum.
  10.                  */
  11.                 if (sk->sk_route_caps & NETIF_F_ALL_CSUM)
  12.                     skb->ip_summed = CHECKSUM_PARTIAL;
                 /* 将新分配的skb添加到sk上 */
  1.                 skb_entail(sk, skb);
  2.                 copy = size_goal;
  3.                 max = size_goal;
  4.             }
             /* 下面开始复制数据，未完待续 */

今天学习的代码不多，一方面是今天的空闲时间不多，家里有不少事情，另一方面，tcp的确实比较复杂。




上一篇：JQuery Tables 的应用（一）


下一篇：TCP/IP源码(59)——TCP中的三个接收队列