关键词:

socket是一种IPC方法,它允许位于同一主机或使用网络连接起来的不同主机上的程序之间交换数据。

关于Socket及后续章节介绍Socket的用法:

1. socket基础

一个典型的客户端/服务器场景中,应用程序使用socket进行通信的方式如下:

  • 各个应用程序创建一个socket。socket是一个允许通信的设备,两个应用程序都需要用到它。
  • 服务器将自己的socket绑定到一个众所周知的地址上是的客户端能够定位到它的位置。

关键socket API包括以下下几种:

  • socket()创建一个新的socket。
  • bind()将一个socket绑定到一个地址上。通常服务器需要使用这个调用来将其socket绑定到一个众所周知的地址上使得客户端能够定位到该socket上。
  • listen()允许一个流socket接受来自其他socket的接入连接。
  • accept()在一个监听流上接受来自一个对等应用程序的连接,并可选地返回对等socket的地址。
  • connect()建立与另一个socket之间的连接。
  • read()/write()/close()基于socket的读写和关闭。
  • recv()/send()/recvfrom()/sendto()分别通过socket发送或接收数据,类似read()/write()但是功能更丰富。

1.1 socket API介绍

1.1.1 socket domain

socket存在于一个通信domain中:识别出一个socket的方法(socket地址格式);通信范围(是统一主机不同应用之间;还是一个网络连接的不同主机上应用之间)。

domain都是以AF_开头,表示Address Family;PF_开头的表示Protocol Family。

在socket.h中定义如下,可以看出AF_和PF_基本一对一。

  1. /* Protocol families. */
  2. #define PF_UNSPEC 0 /* Unspecified. */
  3. #define PF_LOCAL 1 /* Local to host (pipes and file-domain). */
  4. #define PF_UNIX PF_LOCAL /* POSIX name for PF_LOCAL. */
  5. #define PF_FILE PF_LOCAL /* Another non-standard name for PF_LOCAL. */
  6. #define PF_INET 2 /* IP protocol family. */
  7. #define PF_AX25 3 /* Amateur Radio AX.25. */
  8. #define PF_IPX 4 /* Novell Internet Protocol. */
  9. #define PF_APPLETALK 5 /* Appletalk DDP. */
  10. #define PF_NETROM 6 /* Amateur radio NetROM. */
  11. #define PF_BRIDGE 7 /* Multiprotocol bridge. */
  12. #define PF_ATMPVC 8 /* ATM PVCs. */
  13. #define PF_X25 9 /* Reserved for X.25 project. */
  14. #define PF_INET6 10 /* IP version 6. */
  15. ...
  16. #define PF_MAX 44 /* For now.. */
  17.  
  18. /* Address families. */
  19. #define AF_UNSPEC PF_UNSPEC
  20. #define AF_LOCAL PF_LOCAL
  21. #define AF_UNIX PF_UNIX
  22. #define AF_FILE PF_FILE
  23. #define AF_INET PF_INET
  24. #define AF_AX25 PF_AX25
  25. #define AF_IPX PF_IPX
  26. #define AF_APPLETALK PF_APPLETALK
  27. #define AF_NETROM PF_NETROM
  28. #define AF_BRIDGE PF_BRIDGE
  29. #define AF_ATMPVC PF_ATMPVC
  30. #define AF_X25 PF_X25
  31. #define AF_INET6 PF_INET6
  32. ...
  33. #define AF_MAX PF_MAX

常用的AF_有AF_UNIX、AF_INET、AF_INET6三种。

  • AF_UNIX domain允许在同一主机上的应用程序之间进行通信。
  • AF_INET domain允许在使用IPv4网络连接起来的主机上的应用程序之间进行通信。
  • AF_INET6 domain允许在使用IPv6网络连接起来的主机上的应用程序之间进行通信。

1.1.2 socket type

每个socket实现都至少提供了两种socket:流和数据报。

  1. /* Types of sockets. */
  2. enum __socket_type
  3. {
  4. SOCK_STREAM = , /* Sequenced, reliable, connection-based
  5. byte streams. */
  6. SOCK_DGRAM = , /* Connectionless, unreliable datagrams
  7. of fixed maximum length. */
  8. SOCK_RAW = , /* Raw protocol interface. */
  9. SOCK_RDM = , /* Reliably-delivered messages. */
  10. SOCK_SEQPACKET = , /* Sequenced, reliable, connection-based,
  11. datagrams of fixed maximum length. */
  12. SOCK_DCCP = , /* Datagram Congestion Control Protocol. */
  13. SOCK_PACKET = , /* Linux specific way of getting packets
  14. at the dev level. For writing rarp and
  15. other similar things on the user level. */
  16. /* Flags to be ORed into the type parameter of socket and socketpair and
  17. used for the flags parameter of paccept. */
  18.  
  19. SOCK_CLOEXEC = , /* Atomically set close-on-exec flag for the
  20. new descriptor(s). */
  21. SOCK_NONBLOCK = /* Atomically mark descriptor(s) as
  22. non-blocking. */
  23. };

流socket(SOCK_STREAM)提供了一个可靠的双向的字节流通信信道。

流socket的正常工作需要一对相互连接的socket,因此流socket通常被称为面向连接的。

数据报socket(SOCK_DGRAM)允许数据以被称为数据报的消息的形式进行交换。

数据报socket是更一般的无连接socket概念,一个数据报socket在使用时无需与另一个socket连接。

在Internet domain中,数据报socket使用UDP,流socke通则使用TCP。

1.1.3 struct sockaddr

各种socket domain使用了不同的地址格式,对于各种socket domain都需要定义一个不同的结构类型来存储socket地址。

然而由于bind()调用适用于所有socket domain,因此他们必须要能够接受任意类型的地址结构。

为此,socket API定义了一个通用的地址结构struct sockaddr。

这个类型的唯一用途是将各种domain特定的地址结构转换成单个类型以供socket各个参数使用。

  1. /* Structure describing a generic socket address. */
  2. struct sockaddr
  3. {
  4. __SOCKADDR_COMMON (sa_); /* Common data: address family and length. */
  5. char sa_data[]; /* Address data. */
  6. };

这个结构是所有domain特定的地址结构的模板,其中每个地址结构均以与sockadr结构中的sa_family打头。

通过sa_family字段值足以确定存储在这个结构的剩余部分中地址大小和格式了。

1.1.4 socket():创建一个socket

  1. #include <sys/socket.h>
  2. int socket(int domain, int type, int protocol);
  3.     Returns file descriptor on success, or on error

domain指定了socket通信domain,常用的有AF_UNIX、AF_INET、AF_INET6;type指定了socket类型,常用的有SOCK_STREAM、SOCK_DGRAM;protocol一般被指为0。

socket()在成功时会返回一个引用在后续调用中会用到的新创建的socket文件描述符;错误则返回-1。

1.1.5 bind():将socket绑定到地址

  1. #include <sys/socket.h>
  2. int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
  3.     Returns on success, or on error

sockfd是由socket()返回的文件描述符。

addr是指向指定socket绑定到的地址的结构体指针,传输参数的类型取决于socket domain。

addrlen参数指定了地址结构的大小。

其中addr传入到内核,最终被不同socket domain的proto_ops->bind()调用的时候,会被强制转换成不同数据结构。

比如AF_UNIX、AF_INET、AF_INET6对应的地址结构体分别为struct sockaddr_un、struct sockaddr_in、struct sockaddr_in6:

  1. #define UNIX_PATH_MAX 108
  2.  
  3. struct sockaddr_un {
  4. __kernel_sa_family_t sun_family; /* AF_UNIX */---------------------填入AF_UNIX
  5. char sun_path[UNIX_PATH_MAX]; /* pathname */--------------------本地socket的路径。
  6. };
  7.  
  8. #define __SOCK_SIZE__ 16 /* sizeof(struct sockaddr) */
  9. struct sockaddr_in {
  10. __kernel_sa_family_t sin_family; /* Address family */---填入AF_INET
  11. __be16 sin_port; /* Port number */--------------端口号。
  12. struct in_addr sin_addr; /* Internet address */---------IPv4的地址。
  13.  
  14. /* Pad to size of `struct sockaddr'. */
  15. unsigned char __pad[__SOCK_SIZE__ - sizeof(short int) -
  16. sizeof(unsigned short int) - sizeof(struct in_addr)];
  17. };
  18. #define sin_zero __pad /* for BSD UNIX comp. -FvK */
  19.  
  20. struct sockaddr_in6 {
  21. unsigned short int sin6_family; /* AF_INET6 */---------------填入AF_INET6
  22. __be16 sin6_port; /* Transport layer port # */
  23. __be32 sin6_flowinfo; /* IPv6 flow information */
  24. struct in6_addr sin6_addr; /* IPv6 address */
  25. __u32 sin6_scope_id; /* scope id (new in RFC2553) */
  26. };

1.1.6 listen():监听接入连接

  1. #include <sys/socket.h>
  2. int listen(int sockfd, int backlog);
  3.     Returns on success, or on error

listen()将sockfd引用的流socket标记为被动,这个socket后面会被用来接受来自其他socket连接。

如果客户端在服务器调用accept()之前调用connect(),这将会产生一个未决的连接。

内核必须记录所有未决的连接请求,在后续accept()就能够处理这些请求。

backlog参数允许限制这种未决连接数量。在这个限制内的连接请求会立即成功。之外的连接请求就会阻塞直到一个未决连接被接受,并从未决队列中删除为止。

1.1.7 accept():接收连接

  1. #include <sys/socket.h>
  2. int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
  3.     Returns file descriptor on success, or on error

accept()用在sockfd引用的监听流socket上接受一个接入连接。

如果在accept()时不存在未决的连接,那么调用就会阻塞直到有连接请求到达为止。

理解accept()的关键点是它会创建一个新socket,并且这是这个新socket会与执行connect()的对等socket进行连接。

accept()返回结果是已经连接的socket文件描述符,其会保持打开状态,并且可以被用来接受后续的连接。

addr参数指向了一个用来返回socket地址的结构。

addrlen在调用之前必须要将其初始化为addr指向的缓冲区大小;返回之后被设置成实际被复制进缓冲区中的数据的字节数。

1.1.8 connect():(客户端)连接到(服务器)对等socket

  1. #include <sys/socket.h>
  2. int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
  3.     Returns on success, or on error

connect()将sockfd引用的socket连接到地址通过addr和addrlen指定的监听socket上。

其中addr和addrlen参数指定方式与bind()对应参数指定方式相同。

1.1.9 sendto()/recvfrom():交换数据报

  1. #include <sys/socket.h>
  2. ssize_t recvfrom(int sockfd, void *buffer, size_t length, int flags,struct sockaddr *src_addr, socklen_t *addrlen);
  3.     Returns number of bytes received, on EOF, or on error
  4. ssize_t sendto(int sockfd, const void *buffer, size_t length, int flags,const struct sockaddr *dest_addr, socklen_t addrlen);
  5.     Returns number of bytes sent, or on error

flags是一个位掩码,控制着特定的IO特性。

src_addr和addrlen用来获取或指定与之同行的对等socket地址。

对于recvfrom()来说,src_addr和addlen会返回用来发送数据报的远程socket地址。如果不关心,可以将src_addr和addrlen都指定为NULL。

对于sendto()来说,dest_addr和addrlen制定了数据报发送到的socket地址。

1.1.10 read()/write()/close():读、写、关闭socket文件

如果作为客户端,socket()返回的sockfd,在connect之后就就可以对sockfd进行read()/write()/close()操作。

如果作为服务端,在accept()之后产生新的sockfd,之后sockfd就保持打开状态。可以对其进行read()/write()/close()操作。

1.2 流socket

1.3 数据报socket

2. 从socket API到系统调用socketcall()

上面介绍了一系列socket相关API,但是这些C函数并没有对应的系统调用。

下面就看看这些scoket API是如何转到内核调用的,以socket()为例。

  1. int
  2. __socket (int fd, int type, int domain)
  3. {
  4. #ifdef __ASSUME_SOCKET_SYSCALL
  5. return INLINE_SYSCALL (socket, , fd, type, domain);
  6. #else
  7. returnSOCKETCALL (socket, fd, type, domain);----------根据SOCKETCALL()定义可知,socket通过连接符号便变成其对应的call,作为socketcall()的第一个参数。
  8. #endif
  9. }
  10. libc_hidden_def (__socket)-------------------------------对libc之外屏蔽__socket()函数访问。
  11. weak_alias (__socket, socket)----------------------------如果没有定义socket()函数,那么对socket()的调用将会转到调用__socket()。

2.1 SOCKETCALL():

从下面SOCKETCALL()宏定义可知,最终是通过socketcall()系统调用实现的。

具体对应connect()对应的是SOCKOP_socket,即socketcall()系统调用的第一个参数为1。

在socketcall()系统调用中,根据第一个参数执行对应的操作。

所以下面SOCKOP_对应的socket API都是通过socketcall()实现的,然后在socketcall()里面进行处理。

  1. #define SOCKOP_invalid -1
  2. #define SOCKOP_socket 1
  3. #define SOCKOP_bind 2
  4. #define SOCKOP_connect 3
  5. #define SOCKOP_listen 4
  6. #define SOCKOP_accept 5
  7. #define SOCKOP_getsockname 6
  8. #define SOCKOP_getpeername 7
  9. #define SOCKOP_socketpair 8
  10. #define SOCKOP_send 9
  11. #define SOCKOP_recv 10
  12. #define SOCKOP_sendto 11
  13. #define SOCKOP_recvfrom 12
  14. #define SOCKOP_shutdown 13
  15. #define SOCKOP_setsockopt 14
  16. #define SOCKOP_getsockopt 15
  17. #define SOCKOP_sendmsg 16
  18. #define SOCKOP_recvmsg 17
  19. #define SOCKOP_accept4 18
  20. #define SOCKOP_recvmmsg 19
  21. #define SOCKOP_sendmmsg 20
  22.  
  23. #define __SOCKETCALL1(name, a1) \
  24. INLINE_SYSCALL (socketcall, , name, \
  25. ((long int []) { (long int) (a1) }))
  26. ...
  27. #define __SOCKETCALL6(name, a1, a2, a3, a4, a5, a6) \
  28. INLINE_SYSCALL (socketcall, , name, \
  29. ((long int []) { (long int) (a1), (long int) (a2), (long int) (a3), \
  30. (long int) (a4), (long int) (a5), (long int) (a6) }))
  31.  
  32. #define SOCKETCALL(name, args...) \
  33. ({ \
  34. long int sc_ret = __SOCKETCALL (SOCKOP_##name, args); \
  35. sc_ret; \
  36. })

2.2 weak_alias()

weak_alias()是一个宏,其目的是为函数添加一个“弱”别名,与“强”符号进行区分。

如果调用函数对应的函数无“强”符号对应的函数,则会调用该别名对应的函数。所谓“强”符号的函数名就是普通声明定义的函数对应的函数名。

这里如果没有定义connect()函数,调用connect()实际就会转到__socket()。

  1. # define weak_alias(name, aliasname) _weak_alias (name, aliasname)
  2. # define _weak_alias(name, aliasname) \
  3. extern __typeof (name) aliasname __attribute__ ((weak, alias (#name)));

2.3 libc_hidden_def()

libc_hidden_def()的定义在libc-symbols.h中。

  1. # define libc_hidden_def(name) hidden_def (name) 

3. socketcall及socket系统调用分析

可以说socketcall()是所有socket调用的入口,socketcall()根据call的值switch-case到对应的函数中。这些函数和单独系统调用基本一致。

下面就先来分析一下socketcall()函数。

  1. #define SYS_SOCKET 1 /* sys_socket(2) */
  2. #define SYS_BIND 2 /* sys_bind(2) */
  3. #define SYS_CONNECT 3 /* sys_connect(2) */
  4. ...
  5. #define SYS_SENDMMSG 20 /* sys_sendmmsg(2) */
  6.  
  7. /* Argument list sizes for compat_sys_socketcall */
  8. #define AL(x) ((x) * sizeof(u32))
  9. static unsigned char nas[] = {----------------------------------------nas[]将call id作为下标,得到对应系统调用参数的总大小。这里是上面call id和系统调用的一座桥梁。
  10. AL(), AL(), AL(), AL(), AL(), AL(),
  11. AL(), AL(), AL(), AL(), AL(), AL(),
  12. AL(), AL(), AL(), AL(), AL(), AL(),
  13. AL(), AL(), AL()
  14. };
  15.  
  16. COMPAT_SYSCALL_DEFINE2(socketcall, int, call, u32 __user *, args)
  17. {
  18. u32 a[AUDITSC_ARGS];
  19. unsigned int len;
  20. u32 a0, a1;
  21. int ret;
  22.  
  23. if (call < SYS_SOCKET || call > SYS_SENDMMSG)-----------------------判断call范围,从SYS_SOCKETSYS_SOCKET
  24. return -EINVAL;
  25. len = nas[call];----------------------------------------------------根据call id获取args大小。
  26. if (len > sizeof(a))
  27. return -EINVAL;
  28.  
  29. if (copy_from_user(a, args, len))
  30. return -EFAULT;
  31.  
  32. ret = audit_socketcall_compat(len / sizeof(a[]), a);---------------未定义CONFIG_AUDITSYSCALL直接返回0
  33. if (ret)
  34. return ret;
  35.  
  36. a0 = a[];
  37. a1 = a[];
  38.  
  39. switch (call) {
  40. case SYS_SOCKET:
  41. ret = sys_socket(a0, a1, a[]);
  42. break;
  43. case SYS_BIND:
  44. ret = sys_bind(a0, compat_ptr(a1), a[]);
  45. break;
  46. case SYS_CONNECT:
  47. ret = sys_connect(a0, compat_ptr(a1), a[]);
  48. break;
  49. ...
  50. default:
  51. ret = -EINVAL;
  52. break;
  53. }
  54. return ret;
  55. }

结合socketcall()系统调用的实现和API单独系统调用,可以看出两者的实现是一致的。

3.1 sys_socket()

struct socket在内核中表示一个socket,struct sock在网络层表示一个socket。

  1. struct socket {
  2. socket_state state;-----------------------------表示当前socket的状态。
  3.  
  4. kmemcheck_bitfield_begin(type);
  5. short type;---------------------------------对应enum socket_type,等于sys_socket()传入的type参数。
  6. kmemcheck_bitfield_end(type);
  7.  
  8. unsigned long flags;
  9.  
  10. struct socket_wq __rcu *wq;
  11.  
  12. struct file *file;
  13. struct sock *sk;
  14. const struct proto_ops *ops;------------------------是familytype两者综合的操作函数集。
  15. };

sys_socket()创建一个struct socket,根据family从net_families[]找到对应协议族;然后在从type找到具体使用哪种类型struct proto_ops。

  1. SYSCALL_DEFINE3(socket, int, family, int, type, int, protocol)
  2. {
  3. int retval;
  4. struct socket *sock;
  5. int flags;
  6. ...
  7. flags = type & ~SOCK_TYPE_MASK;
  8. if (flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK))
  9. return -EINVAL;
  10. type &= SOCK_TYPE_MASK;---------------------------------------------通过SOCK_TYPE_MASK将传入的type分开,一部分是flags,另一部分是0~15之间的socket type
  11.  
  12. if (SOCK_NONBLOCK != O_NONBLOCK && (flags & SOCK_NONBLOCK))
  13. flags = (flags & ~SOCK_NONBLOCK) | O_NONBLOCK;
  14.  
  15. retval = sock_create(family, type, protocol, &sock);----------------根据family/type/protocol创建一个struct socket
  16. if (retval < )
  17. goto out;
  18.  
  19. retval = sock_map_fd(sock, flags & (O_CLOEXEC | O_NONBLOCK));-------为新创建的struct socket分配一个文件描述符。
  20. if (retval < )
  21. goto out_release;
  22.  
  23. out:
  24. /* It may be already another descriptor 8) Not kernel problem. */
  25. return retval;
  26.  
  27. out_release:
  28. sock_release(sock);
  29. return retval;
  30. }
  31.  
  32. int sock_create(int family, int type, int protocol, struct socket **res)
  33. {
  34. return __sock_create(current->nsproxy->net_ns, family, type, protocol, res, );
  35. }
  36.  
  37. int __sock_create(struct net *net, int family, int type, int protocol,
  38. struct socket **res, int kern)
  39. {
  40. int err;
  41. struct socket *sock;
  42. const struct net_proto_family *pf;
  43. ...
  44. if (family == PF_INET && type == SOCK_PACKET) {
  45. pr_info_once("%s uses obsolete (PF_INET,SOCK_PACKET)\n",
  46. current->comm);
  47. family = PF_PACKET;
  48. }
  49.  
  50. err = security_socket_create(family, type, protocol, kern);
  51. if (err)
  52. return err;
  53.  
  54. sock = sock_alloc();--------------------------------------------------创建一个struct inodestruct socket,并将两者绑定起来。
  55. if (!sock) {
  56. net_warn_ratelimited("socket: no more sockets\n");
  57. return -ENFILE; /* Not exactly a match, but its the
  58. closest posix thing */
  59. }
  60.  
  61. sock->type = type;----------------------------------------------------设置socket类型。
  62.  
  63. #ifdef CONFIG_MODULES
  64. if (rcu_access_pointer(net_families[family]) == NULL)
  65. request_module("net-pf-%d", family);
  66. #endif
  67.  
  68. rcu_read_lock();
  69. pf = rcu_dereference(net_families[family]);---------------------------net_families使用下标对应AF_XXX,通过net_families[family]可以获得对应struct net_proto_family
  70. err = -EAFNOSUPPORT;
  71. if (!pf)
  72. goto out_release;
  73.  
  74. if (!try_module_get(pf->owner))
  75. goto out_release;
  76.  
  77. rcu_read_unlock();
  78.  
  79. err = pf->create(net, sock, protocol, kern);--------------------------调用具体AF_XX对应的create成员,比如AF_UNIX对应unix_create()
  80. if (err < )
  81. goto out_module_put;
  82.  
  83. if (!try_module_get(sock->ops->owner))
  84. goto out_module_busy;
  85.  
  86. module_put(pf->owner);
  87. err = security_socket_post_create(sock, family, type, protocol, kern);
  88. if (err)
  89. goto out_sock_release;
  90. *res = sock;
  91.  
  92. return ;
  93. ...
  94. }

net_families[]保存了所有AF_XXX对应的struct net_proto_family。

这些struct net_proto_family通过sock_register()注册,通过sock_unregister()去注册。

  1. static const struct net_proto_family __rcu *net_families[NPROTO] __read_mostly;
  2.  
  3. int sock_register(const struct net_proto_family *ops)
  4. {
  5. int err;
  6. ...
  7. spin_lock(&net_family_lock);
  8. if (rcu_dereference_protected(net_families[ops->family],
  9. lockdep_is_held(&net_family_lock)))
  10. err = -EEXIST;
  11. else {
  12. rcu_assign_pointer(net_families[ops->family], ops);---------------主要就是讲struct net_proto_family赋给net_families[]。
  13. err = ;
  14. }
  15. spin_unlock(&net_family_lock);
  16. return err;
  17. }
  18.  
  19. void sock_unregister(int family)
  20. {
  21. BUG_ON(family < || family >= NPROTO);
  22.  
  23. spin_lock(&net_family_lock);
  24. RCU_INIT_POINTER(net_families[family], NULL);
  25. spin_unlock(&net_family_lock);
  26.  
  27. synchronize_rcu();
  28. }

下面以AF_UNIX为例,看看不同type的处理。

  1. static int __init af_unix_init(void)
  2. {
  3. ...
  4. sock_register(&unix_family_ops);
  5. ...
  6. }
  7.  
  8. static const struct net_proto_family unix_family_ops = {
  9. .family = PF_UNIX,
  10. .create =unix_create,
  11. .owner = THIS_MODULE,
  12. };
  13.  
  14. static int unix_create(struct net *net, struct socket *sock, int protocol,
  15. int kern)
  16. {
  17. if (protocol && protocol != PF_UNIX)
  18. return -EPROTONOSUPPORT;
  19.  
  20. sock->state = SS_UNCONNECTED;
  21.  
  22. switch (sock->type) {-------------------------------------------------可以看出AF_UNIX仅支持SOCK_STREAMSOCK_STREAMSOCK_STREAMSOCK_STREAM几种形式type
  23. case SOCK_STREAM:
  24. sock->ops = &unix_stream_ops;
  25. break;
  26. case SOCK_RAW:
  27. sock->type = SOCK_DGRAM;
  28. case SOCK_DGRAM:
  29. sock->ops = &unix_dgram_ops;
  30. break;
  31. case SOCK_SEQPACKET:
  32. sock->ops = &unix_seqpacket_ops;
  33. break;
  34. default:
  35. return -ESOCKTNOSUPPORT;
  36. }
  37. return unix_create1(net, sock, kern) ? : -ENOMEM;--------------------创建并初始化struct sock
  38. }
  39.  
  40. static const struct proto_ops unix_stream_ops = {
  41. .family = PF_UNIX,
  42. .owner = THIS_MODULE,
  43. .release = unix_release,
  44. ...
  45. .set_peek_off = unix_set_peek_off,
  46. };
  47.  
  48. static const struct proto_ops unix_dgram_ops = {
  49. .family = PF_UNIX,
  50. .owner = THIS_MODULE,
  51. .release = unix_release,
  52. ...
  53. .set_peek_off = unix_set_peek_off,
  54. };
  55.  
  56. static const struct proto_ops unix_seqpacket_ops = {
  57. .family = PF_UNIX,
  58. .owner = THIS_MODULE,
  59. .release = unix_release,
  60. ...
  61. .set_peek_off = unix_set_peek_off,
  62. };

unix_create1()函数分配并且初始化struct sock,然后作为struct socket的成员sk。

  1. static struct sock *unix_create1(struct net *net, struct socket *sock, int kern)
  2. {
  3. struct sock *sk = NULL;
  4. struct unix_sock *u;
  5.  
  6. atomic_long_inc(&unix_nr_socks);
  7. if (atomic_long_read(&unix_nr_socks) > * get_max_files())
  8. goto out;
  9.  
  10. sk = sk_alloc(net, PF_UNIX, GFP_KERNEL, &unix_proto, kern);
  11. if (!sk)
  12. goto out;
  13.  
  14. sock_init_data(sock, sk);
  15. lockdep_set_class(&sk->sk_receive_queue.lock,
  16. &af_unix_sk_receive_queue_lock_key);
  17.  
  18. sk->sk_allocation = GFP_KERNEL_ACCOUNT;
  19. sk->sk_write_space = unix_write_space;
  20. sk->sk_max_ack_backlog = net->unx.sysctl_max_dgram_qlen;
  21. sk->sk_destruct = unix_sock_destructor;
  22. u = unix_sk(sk);
  23. u->path.dentry = NULL;
  24. u->path.mnt = NULL;
  25. spin_lock_init(&u->lock);
  26. atomic_long_set(&u->inflight, );
  27. INIT_LIST_HEAD(&u->link);
  28. mutex_init(&u->iolock); /* single task reading lock */
  29. mutex_init(&u->bindlock); /* single task binding lock */
  30. init_waitqueue_head(&u->peer_wait);
  31. init_waitqueue_func_entry(&u->peer_wake, unix_dgram_peer_wake_relay);
  32. unix_insert_socket(unix_sockets_unbound(sk), sk);
  33. out:
  34. if (sk == NULL)
  35. atomic_long_dec(&unix_nr_socks);
  36. else {
  37. local_bh_disable();
  38. sock_prot_inuse_add(sock_net(sk), sk->sk_prot, );
  39. local_bh_enable();
  40. }
  41. return sk;
  42. }

综上所述,sys_socket() 主要完善内核中struct socket结构体,尤其是struct proto_ops结构体。然后返回对应文件描述符给用户空间。

后续关于socket的API都是通过文件描述符找到内核中对应的struct socket,然后调用struct proto_ops中成员来完成工作。

3.2 sys_bind()

sys_bind()通过入参fd找到内核中表示socket对应的struct socket。

然后调用struct socket->ops->bind()进行umyaddr和fd绑定。在AF_UNIX和SOCK_STREAM情况下,调用unix_bind()。

  1. SYSCALL_DEFINE3(bind, int, fd, struct sockaddr __user *, umyaddr, int, addrlen)
  2. {
  3. struct socket *sock;
  4. struct sockaddr_storage address;
  5. int err, fput_needed;
  6.  
  7. sock = sockfd_lookup_light(fd, &err, &fput_needed);-------------------------根据fd找到对应的struct socket结构体。以fs为索引从当前进程的文件描述符表files_struct中找到对应的file实例,然后从file实例中的private_data成员中获取socket实例。
  8. if (sock) {
  9. err = move_addr_to_kernel(umyaddr, addrlen, &address);
  10. if (err >= ) {
  11. err = security_socket_bind(sock,
  12. (struct sockaddr *)&address,
  13. addrlen);
  14. if (!err)
  15. err = sock->ops->bind(sock,
  16. (struct sockaddr *)
  17. &address, addrlen);-------------------------------调用struct socket->ops->bind()完成地址与socket的绑定,进而和fd绑定。
  18. }
  19. fput_light(sock->file, fput_needed);
  20. }
  21. return err;
  22. }
  23.  
  24. static struct socket *sockfd_lookup_light(int fd, int *err, int *fput_needed)
  25. {
  26. struct fd f = fdget(fd);
  27. struct socket *sock;
  28.  
  29. *err = -EBADF;
  30. if (f.file) {
  31. sock = sock_from_file(f.file, err);
  32. if (likely(sock)) {
  33. *fput_needed = f.flags;
  34. return sock;
  35. }
  36. fdput(f);
  37. }
  38. return NULL;
  39. }

3.3 sys_listen()

类似sys_bind(),sys_listen()也是同样的通过fd找到struct socket,然后调用struct socket->ops->listen()完成主要工作。

在AF_UNIX和SOCK_STREAM情况下,调用unix_listen()。

  1. SYSCALL_DEFINE2(listen, int, fd, int, backlog)
  2. {
  3. struct socket *sock;
  4. int err, fput_needed;
  5. int somaxconn;
  6.  
  7. sock = sockfd_lookup_light(fd, &err, &fput_needed);
  8. if (sock) {
  9. somaxconn = sock_net(sock->sk)->core.sysctl_somaxconn;
  10. if ((unsigned int)backlog > somaxconn)
  11. backlog = somaxconn;
  12.  
  13. err = security_socket_listen(sock, backlog);
  14. if (!err)
  15. err = sock->ops->listen(sock, backlog);
  16.  
  17. fput_light(sock->file, fput_needed);
  18. }
  19. return err;
  20. }

3.4 sys_connect()

服务器端socket使用bind()来绑定IP和端口,客户端使用connect()让系统自动选择IP和端口。

核心也是调用struct socket->ops->connect()。

  1. SYSCALL_DEFINE3(connect, int, fd, struct sockaddr __user *, uservaddr,
  2. int, addrlen)
  3. {
  4. struct socket *sock;
  5. struct sockaddr_storage address;
  6. int err, fput_needed;
  7.  
  8. sock = sockfd_lookup_light(fd, &err, &fput_needed);----------------------------------通过文件描述符fd找到对应的socket实例。
  9. if (!sock)
  10. goto out;
  11. err = move_addr_to_kernel(uservaddr, addrlen, &address);
  12. if (err < )
  13. goto out_put;
  14.  
  15. err =
  16. security_socket_connect(sock, (struct sockaddr *)&address, addrlen);--------------将socket地址从用户空间拷贝到内核。
  17. if (err)
  18. goto out_put;
  19.  
  20. err = sock->ops->connect(sock, (struct sockaddr *)&address, addrlen,-----------------调用connect()成员函数,对于AF_UNIXSOCK_STREAM即调用unix_stream_connect()。
  21. sock->file->f_flags);
  22. out_put:
  23. fput_light(sock->file, fput_needed);
  24. out:
  25. return err;
  26. }

3.5 sys_accept()/sys_accept4()

sys_accept()作为accept()在内核中的实现,返回一个新的句柄,建立新的操作上下文。

  1. SYSCALL_DEFINE4(accept4, int, fd, struct sockaddr __user *, upeer_sockaddr,
  2. int __user *, upeer_addrlen, int, flags)
  3. {
  4. struct socket *sock, *newsock;
  5. struct file *newfile;
  6. int err, len, newfd, fput_needed;
  7. struct sockaddr_storage address;
  8.  
  9. if (flags & ~(SOCK_CLOEXEC | SOCK_NONBLOCK))--------------------------------------不允许使用这两个flags
  10. return -EINVAL;
  11.  
  12. if (SOCK_NONBLOCK != O_NONBLOCK && (flags & SOCK_NONBLOCK))
  13. flags = (flags & ~SOCK_NONBLOCK) | O_NONBLOCK;
  14.  
  15. sock = sockfd_lookup_light(fd, &err, &fput_needed);--------------------------------根据fd找到struct socket
  16. if (!sock)
  17. goto out;
  18.  
  19. err = -ENFILE;
  20. newsock = sock_alloc();------------------------------------------------------------创建一个新的struct socket
  21. if (!newsock)
  22. goto out_put;
  23.  
  24. newsock->type = sock->type;--------------------------------------------------------新的socket类型和socket层操作。
  25. newsock->ops = sock->ops;
  26.  
  27. /*
  28. * We don't need try_module_get here, as the listening socket (sock)
  29. * has the protocol module (sock->ops->owner) held.
  30. */
  31. __module_get(newsock->ops->owner);
  32.  
  33. newfd = get_unused_fd_flags(flags);-------------------------------------------------分配一个空闲的文件句柄。
  34. if (unlikely(newfd < )) {
  35. err = newfd;
  36. sock_release(newsock);
  37. goto out_put;
  38. }
  39. newfile = sock_alloc_file(newsock, flags, sock->sk->sk_prot_creator->name);---------为新创建的struct socket分配一个文件描述符。
  40. if (IS_ERR(newfile)) {
  41. err = PTR_ERR(newfile);
  42. put_unused_fd(newfd);
  43. sock_release(newsock);
  44. goto out_put;
  45. }
  46.  
  47. err = security_socket_accept(sock, newsock);
  48. if (err)
  49. goto out_fd;
  50.  
  51. err = sock->ops->accept(sock, newsock, sock->file->f_flags);-------------------------对于AF_UNIXSOCK_STREAM则是调用unix_accept()。
  52. if (err < )
  53. goto out_fd;
  54.  
  55. if (upeer_sockaddr) {----------------------------------------------------------------如果accept需要返回对端socket地址,调用newsock->ops->getname()获取struct sockaddr并返还给用户空间upeer_sockaddr
  56. if (newsock->ops->getname(newsock, (struct sockaddr *)&address,
  57. &len, ) < ) {
  58. err = -ECONNABORTED;
  59. goto out_fd;
  60. }
  61. err = move_addr_to_user(&address,
  62. len, upeer_sockaddr, upeer_addrlen);
  63. if (err < )
  64. goto out_fd;
  65. }
  66.  
  67. /* File flags are not inherited via accept() unlike another OSes. */
  68.  
  69. fd_install(newfd, newfile);---------------------------------------------------------以newfd为索引,把newfile加入当前进程的文件描述符标files_struct中。
  70. err = newfd;
  71.  
  72. out_put:
  73. fput_light(sock->file, fput_needed);
  74. out:
  75. return err;
  76. out_fd:
  77. fput(newfile);
  78. put_unused_fd(newfd);
  79. goto out_put;
  80. }
  81.  
  82. SYSCALL_DEFINE3(accept, int, fd, struct sockaddr __user *, upeer_sockaddr,
  83. int __user *, upeer_addrlen)
  84. {
  85. return sys_accept4(fd, upeer_sockaddr, upeer_addrlen, );
  86. }

所以sys_accept ()主要作用就是:创建新的socket和inode并初始化完成;调用原socket->ops->accept();保存新创建socket的地址到用户空间。

3.6 其他API

sys_getsockname()/sys_getpeername()调用相应proto_ops->getname()。

sys_send()/sys_sendto()/sys_sendmsg()/sys_sendmmsg()最终都是通过___sys_sendmsg()实现。

sys_recv()/sys_recvfrom()/sys_recvmsg()/sys_recvmmsg()最终都是通过___sys_recvmsg实现。

sys_setsockopt()/sys_getsockopt()分别调用proto_ops->setsockopt()和proto_ops->getsockopt()。

sys_socketpair()调用proto_ops->socketpair(),sys_shutdown()调用proto_ops->shutdown()。

4. 简单基于Socket的Server/Client通信

《Linux/UNIX系统编程手册》第56章 SOCKET:介绍的更多相关文章

  1. Linux/Unix系统编程手册 第三章:系统编程概念

    本章介绍系统编程的基础概念和一些后续章节用到的函数及头文件,并说明了可移植性问题. 系统调用是受控的内核入口,通过系统调用,进程可以请求内核以自己的名义去执行某些动作,比如创建子进程,执行I/O操作, ...

  2. 《Linux/UNIX系统编程手册》第63章 IO多路复用、信号驱动IO以及epoll

    关键词:fasync_helper.kill_async.sigsuspend.sigaction.fcntl.F_SETOWN_EX.F_SETSIG.select().poll().poll_wa ...

  3. 《Linux/Unix系统编程手册》读书笔记5

    <Linux/Unix系统编程手册>读书笔记 目录 第8章 本章讲了用户和组,还有记录用户的密码文件/etc/passwd,shadow密码文件/etc/shadow还有组文件/etc/g ...

  4. 《Linux/Unix系统编程手册》读书笔记8 (文件I/O缓冲)

    <Linux/Unix系统编程手册>读书笔记 目录 第13章 这章主要将了关于文件I/O的缓冲. 系统I/O调用(即内核)和C语言标准库I/O函数(即stdio函数)在对磁盘进行操作的时候 ...

  5. 《Linux/Unix系统编程手册》读书笔记7 (/proc文件的简介和运用)

    <Linux/Unix系统编程手册>读书笔记 目录 第11章 这章主要讲了关于Linux和UNIX的系统资源的限制. 关于限制都存在一个最小值,这些最小值为<limits.h> ...

  6. 《Linux/Unix系统编程手册》读书笔记6

    <Linux/Unix系统编程手册>读书笔记 目录 第9章 这章主要讲了一堆关于进程的ID.实际用户(组)ID.有效用户(组)ID.保存设置用户(组)ID.文件系统用户(组)ID.和辅助组 ...

  7. 《Linux/Unix系统编程手册》读书笔记4

    <Linux/Unix系统编程手册>读书笔记 目录 第7章: 内存分配 通过增加堆的大小分配内存,通过提升program break位置的高度来分配内存. 基本学过C语言的都用过mallo ...

  8. 《Linux/Unix系统编程手册》读书笔记3

    <Linux/Unix系统编程手册>读书笔记 目录 第6章 这章讲进程.虚拟内存和环境变量等. 进程是一个可执行程序的实例.一个程序可以创建很多进程. 进程是由内核定义的抽象实体,内核为此 ...

  9. 《Linux/Unix系统编程手册》读书笔记1

    <Linux/Unix系统编程手册>读书笔记 目录 最近这一个月在看<Linux/Unix系统编程手册>,在学习关于Linux的系统编程.之前学习Linux的时候就打算写关于L ...

随机推荐

  1. SpringCloud 脚手架

    疯狂创客圈 Java 高并发[ 亿级流量聊天室实战]实战系列 [博客园总入口 ] 架构师成长+面试必备之 高并发基础书籍 [Netty Zookeeper Redis 高并发实战 ] 前言 Crazy ...

  2. idea自用快捷键(非常实用)

    最近看了下idea的快捷键,不禁感慨idea真是太强大了,这里记录一下我常用的快捷键 Ctrl+F:在当前文件中查找 Ctrl+R:替换字符串 Alt + shift + 鼠标点击   多处同时编辑 ...

  3. SAP IDOC 通过采购订单输出消息生成销售订单

    题记: 在网络上看到一篇类似的公众号文章,叫<通过IDoc逐步指导PO&SO集成>,个人觉得整个配置过程中还是少了一些重点配置,也少了说明整个功能的核心逻辑,那么,趁着这个机会,就 ...

  4. Python 基础语法-str

    字符串常见操作 find:检测str是否包含在 mystr 中,如果是返回开始的索引值,否则返回 -1 mystr.index(str, start=0, end=len(mystr)) count: ...

  5. MySQL优化/面试,看这一篇就够了

    原文链接:http://www.zhenganwen.top/articles/2018/12/25/1565048860202.html 作者:Anwen~链接:https://www.nowcod ...

  6. JSON对象转JAVA对象--com.alibaba.fastjson.JSONObject

    打印结果:

  7. IT兄弟连 HTML5教程 CSS3揭秘 在HTML文档中放置CSS的几种方式

    有很多方法将样式表加入到HTML中,每种方法都有自己的优点和缺点.新的HTML元素和属性已被加入,以允许样式表与HTML文档更简易地组合起来.将样式表加入到HTML中的常用方法有内联样式表.嵌入一张样 ...

  8. IT兄弟连 HTML5教程 HTML5表单 HTML表单中的get和post方法

    指引 表单在网页应用中十分重要,基本上任何一个网站都必须使用到表单元素,所以表单的美观和易于交互对于网站设计就变得十分重要.HTML5对目前Web表单进行了全面提升,使得我们使用表单更加智能.它在保持 ...

  9. 01-linux介绍

    一.Linux简介 Linux内核最初只是由芬兰人林纳斯.托瓦兹在大学时出于爱好写出来的,是一套免费使用和自由传播的类Unix操作系统,是基于POSIX和UNIX的多用户.多任务.支持多线程和多CPU ...

  10. dev c++必须修改的三处默认设置

    此文档记录参加pat考试并且以dev c++[针对5.11版本]软件作为开发工具时,必须修改的三个默认设置. 1.修改默认语言标准 Dev C++ 5.11 版本支持较新的 C 和 C++ 标准,但是 ...