LINUX TC:HTB相关源码 收藏

HTB(hierarchy token buffer)是linux tc(traffic control)模块中的排队队列的一种。它的配置比CBQ要简单.同时实现功能也很强大。下面，就来看看，HTB在linux中的源码。

1、             Qdisc_ops的注册

先从module_init函数看起(遵从fudan_abc的建议)

static int __init htb_module_init(void)

{

return register_qdisc(&htb_qdisc_ops);

}

上面的code会调用register_qdisc函数，将htb_qdisc_ops注册到系统中，那么htb_qdisc_ops包含了那些内容：

static struct Qdisc_ops htb_qdisc_ops __read_mostly = {

.cl_ops        =     &htb_class_ops,

.id          =     "htb",

.priv_size   =     sizeof(struct htb_sched),

.enqueue     =     htb_enqueue,

.dequeue     =     htb_dequeue,

.peek            =     qdisc_peek_dequeued,

.drop            =     htb_drop,

.init              =     htb_init,

.reset           =     htb_reset,

.destroy      =     htb_destroy,

.dump          =     htb_dump,

.owner         =     THIS_MODULE,

};

可以看出，htb_qdisc_ops其实就是注册了htb管理queue的函数，最重要的莫过于：enqueue 和dequeue函数，它们作用如同它们的名字一样。那么到底将htb_qdisc_ops注册到那了呢？

这就要看看register函数了

static struct Qdisc_ops *qdisc_base;

/*

*qdisc_base 就是系统维护所以qdisc所使用的变量，系统中的所有的qdisc都要

*注册到这变量变量中

*在struct Qdisc_ops中，包含了成员(struct Qdisc_ops *)next

*也就是，所以的qdisc是以链表的形式存在的

*/

int register_qdisc(struct Qdisc_ops *qops)

{

struct Qdisc_ops *q, **qp;

int rc = -EEXIST;

write_lock(&qdisc_mod_lock);

/*

*首先，检测这个qdisc是否已经注册过了，这是通过比较id实现的，

*id的类型是char 数组：char                 id[IFNAMSIZ];IFNAMESIZ=16

*htb的id=”htb”

*/

for (qp = &qdisc_base; (q = *qp) != NULL; qp = &q->next)

if (!strcmp(qops->id, q->id))

goto out;

/*

*然后检测ops中的enqueue、dequeue、peek函数，

*如果这些函数都没有被初始化，将使用noop_qdisc_ops函数来初始化

*noop_qdisc_ops也是Qdisc_ops结构，

*它的作用就像是定义了Qdisc_ops的默认值

*/

if (qops->enqueue == NULL)

qops->enqueue = noop_qdisc_ops.enqueue;

if (qops->peek == NULL) {

if (qops->dequeue == NULL)

qops->peek = noop_qdisc_ops.peek;

else

goto out_einval;

}

if (qops->dequeue == NULL)

qops->dequeue = noop_qdisc_ops.dequeue;

/*

*然后检测cl_ops成员。

*cl_ops是结构Qdisc_class_ops，

*它定义了用于管理挂载到这个qdisc下的所有class(或者qdisc)

*/

if (qops->cl_ops) {

const struct Qdisc_class_ops *cops = qops->cl_ops;

if (!(cops->get && cops->put && cops->walk && cops->leaf))

goto out_einval;

if (cops->tcf_chain && !(cops->bind_tcf && cops->unbind_tcf))

goto out_einval;

}

/*

*最后将新的qops插入到链表的尾部：*qp = qops;

*这样就注册完成了

*/

qops->next = NULL;

*qp = qops;

rc = 0;

out:

write_unlock(&qdisc_mod_lock);

return rc;

out_einval:

rc = -EINVAL;

goto out;

}

EXPORT_SYMBOL(register_qdisc);

Qdisc_class_ops是管理这个tree的，那么看看htb的cl_ops有哪些函数：

static const struct Qdisc_class_ops htb_class_ops = {

.graft           =     htb_graft,

.leaf             =     htb_leaf,

.qlen_notify      =     htb_qlen_notify,

.get              =     htb_get,

.put              =     htb_put,

.change              =     htb_change_class,

.delete         =     htb_delete,

.walk           =     htb_walk,

.tcf_chain   =     htb_find_tcf,

.bind_tcf     =     htb_bind_filter,

.unbind_tcf =     htb_unbind_filter,

.dump          =     htb_dump_class,

.dump_stats      =     htb_dump_class_stats,

};

我们知道，tc qdisc命令添加qdisc到某个设备后，为了对数据包进行分类，需要使用tc filter 来添加fitler到某个qdisc， 当数据包来时，通过fitler来区分数据包，并转发到不同的subqdisc 或者subclass。而绑定fitler都是通过函数：bind_tcf来实现的。