linux TCP/IP协议栈 ---ip_defrag() inet_frag_find

1. /* Process an incoming IP datagram fragment. */
2. int ip_defrag(struct sk_buff *skb, u32 user)
3. {
4.     struct ipq *qp;
5. 
   
6.     IP_INC_STATS_BH(IPSTATS_MIB_REASMREQDS);
7. 
   
8.     /* Start by cleaning up the memory. */
9.     /* 
10.      * 首先检查所有IP分片所消耗的内存是否大于系统允许的最高阀值，如果是，则调用
11.      * ip_evictor()丢弃未完全到达的IP分片，从最旧的分片开始释放。此举一来是为了节
12.      * 约内存，二来是未了防止黑客的恶意攻击。使分片在系统中累计，降低系统性能。
13.      */
14.     if (atomic_read(&ip4_frags.mem) > ip4_frags_ctl.high_thresh)
15.         ip_evictor();
16. 
    
17.     /* Lookup (or create) queue header */
18.     /* 如果该分片是数据报的第一个分片，则ip_find返回一个新的队列来搜集分片，否则
19.      * 返回其所属于的分片队列。 */
20.     if ((qp = ip_find(ip_hdr(skb), user)) != NULL) {
21.         int ret;
22. 
    
23.         spin_lock(&qp->q.lock);
24.     /* 将该分片加入到队列中，重组分片队列，如果所有的包都收到了，则该函数
25.      * 负责重组IP包 */
26.         ret = ip_frag_queue(qp, skb);
27. 
    
28.         spin_unlock(&qp->q.lock);
29.         ipq_put(qp);    /* 引用计数减1 */
30.         return ret;
31.     }
32. 
    
33.     IP_INC_STATS_BH(IPSTATS_MIB_REASMFAILS);
34.     kfree_skb(skb);
35.     return -ENOMEM;
36. }

1. struct inet_frag_queue *inet_frag_find(struct inet_frags *f, void *key,
2.         unsigned int hash)
3. {
4.     struct inet_frag_queue *q;
5.     struct hlist_node *n;
6. 
   
7.     /* f->lock是读写锁，先搜索是否存在该IP分段所属的队列 */
8.     read_lock(&f->lock);
9.     hlist_for_each_entry(q, n, &f->hash[hash], list) { /* 扫描该HASH槽中所有节点 */
10.     /* f->match中match字段在ipfrag_init中初始化为ip4_frag_match函数。*/
11.     /* 对比分片队列中的散列字段和user是否和key相等，key指向的是struct ip4_create_arg
12.      * 结构，包含IP头部和user字段。 */
13.         if (f->match(q, key)) {
14.             atomic_inc(&q->refcnt);     /* 若找到，则增加该队列引用计数。 */
15.             read_unlock(&f->lock);
16.             return q;                /* 返回该队列 */
17.         }
18.     }
19.     read_unlock(&f->lock);
20.     /* 该分片是第一个IP分片，创建一个新的分片队列并添加到合适的HASH队列 */
21.     return inet_frag_create(f, key, hash);
22. }