本文原创为freas_1990,转载请标明出处:http://blog.csdn.net/freas_1990/article/details/11539695
关于socket与sock的关系再简单说一下。
socket是面向文件与API的,拿来给application程序员fuck。
而sock是面向数据包的,用来发送用户的数据和接收网卡传到内核的数据。
我们再看一下sock的结构体定义。
struct sock { ...
volatile unsigned long wmem_alloc;
volatile unsigned long rmem_alloc; ...
unsigned short rcvbuf;
unsigned short sndbuf; ...
};
注意,这里的rcvbuf和sendbuf不是缓冲区,而只是一个可调整的参数。通过setsocketopt()可以这两个参数进行调整。
其中的wmem_alloc和rmem_alloc是动态变化的,实时记录了当前sock获取了多少rcvbuf。
那为什么需要设置wmem_alloc这样的一个值呢?
在计算机的内存管理里,有这样的一个技术——用一个值来记录当前(server)使用了多少内存,所有的内存分配,释放会记录到这个变量里,以做到对内存的实时反馈。
在Linux内核里,在redis里也有,相信在很多别的server代码里也会有。
如果不使用这样的变量会出现什么样的问题呢?
这个问题我暂时不给出答案,大家可以思考一下。
那什么时候这几个变量会起作用呢?
我们来看一下下面的代码。
struct sk_buff *
sock_wmalloc(struct sock *sk, unsigned long size, int force,
int priority)
{
if (sk) {
if (sk->wmem_alloc + size < sk->sndbuf || force) {
struct sk_buff * c = alloc_skb(size, priority);
if (c) {
cli();
sk->wmem_alloc+= size;
sti();
}
return c;
}
DPRINTF((DBG_INET, "sock_wmalloc(%X,%d,%d,%d) returning NULL\n",
sk, size, force, priority));
return(NULL);
}
return(alloc_skb(size, priority));
}
当我们发送一个数据包的时候,会调用sock_wmalloc函数,这个函数会把当前sock已经分配的内存(wmem_alloc)和马上会分配的数据包内存相加后与sndbuf比较,如果大于sndbuf的话,就不能发送这个包。
从这里,可以看出来,sndbuf其实只是一个参数,它是可以动态调整,也是可以忽视的(如果force参数被设置了)。
那这两个参数是何时被初始化的呢?
来看一下如下代码
static int
inet_create(struct socket *sock, int protocol)
{
struct sock *sk;
struct proto *prot;
int err; sk = (struct sock *) kmalloc(sizeof(*sk), GFP_KERNEL);
...
sk->sndbuf = SK_WMEM_MAX;
sk->rcvbuf = SK_RMEM_MAX;
...
if (sk->prot->init) {
err = sk->prot->init(sk);
if (err != 0) {
destroy_sock(sk);
return(err);
}
}
return(0);
}
我们再来看一下,setsocketopt是如何修改这个参数的。
/* * This is meant for all protocols to use and covers goings on * at the socket level. Everything here is generic. */
int sock_setsockopt(struct sock *sk, int level, int optname, char *optval, int optlen) { int val; int err; ... case SO_SNDBUF: if(val>32767) val=32767; if(val<256) val=256; sk->sndbuf=val; return 0; ... case SO_RCVBUF: if(val>32767) val=32767; if(val<256) val=256; sk->rcvbuf=val; return(0); ... default: return(-ENOPROTOOPT); } }
那sock里的skb数据结构是如何控制的呢?
那一张漂亮的图是如何画出来的呢?