Linux网络编程——Socket

Linux网络编程——Socket

  • 1 套接字概念
    • 1.1 基本概念
    • 1.2 表示方式及原理
  • 2 预备知识
    • 2.1 网络字节序
  • 3 网络套接字函数
    • 3.1 socket模型创建流程图
    • 3.2 socket函数
    • 3.3 bind函数
    • 3.4 listen函数
    • 3.5 accept函数
    • 3.6 connect函数
  • 4 C/S模型-TCP
    • 4.1 server
    • 4.2 client
  • 5 出错处理封装函数
    • 5.1 wrap.cpp
    • 5.2 wrap.h

1 套接字概念

1.1 基本概念

       所谓套接字(Socket),就是对网络中不同主机上的应用进程之间进行双向通信的端点的抽象。一个套接字就是网络上进程通信的一端,提供了应用层进程利用网络协议交换数据的机制。从所处的地位来讲,套接字上联应用进程,下联网络协议栈,是应用程序通过网络协议进行通信的接口,是应用程序与网络协议根进行交互的接口【baidu】
       Socket本身有“插座”的意思,在Linux环境下,用于表示进程间网络通信的特殊文件类型。本质为内核借助缓冲区形成的伪文件。
       既然是文件,那么理所当然的,我们可以使用文件描述符引用套接字。与管道类似的,Linux系统将其封装成文件的目的是为了统一接口,使得读写套接字和读写文件的操作一致。区别是管道主要应用于本地进程间通信,而套接字多应用于网络进程间数据的传递。

1.2 表示方式及原理

       套接字Socket=(IP地址:端口号),套接字的表示方法是点分十进制的lP地址后面写上端口号,中间用冒号或逗号隔开。每一个传输层连接唯一地被通信两端的两个端点(即两个套接字)所确定。例如:如果IP地址是210.37.145.1,而端口号是23,那么得到套接字就是(210.37.145.1:23)

       在TCP/IP协议中,“IP地址+TCP或UDP端口号”唯一标识网络通讯中的一个进程。“IP地址+端口号”就对应一个Socket。欲建立连接的两个进程各自有一个Socket来标识,那么这两个Socket组成的socket pair就唯一标识一个连接。因此可以用Socket来描述网络连接的一对一关系。

  • IP地址:在网络环境中唯一标识一台主机
  • 端口号:在主机中唯一标识一个进程
  • IP地址+端口号:唯一标识网络通讯中的一个进程

套接字通信原理如下图所示:
Linux网络编程——Socket_第1张图片

       在网络通信中,套接字一定是成对出现的。一端的发送缓冲区对应对端的接收缓冲区。我们使用同一个文件描述符索发送缓冲区和接收缓冲区。

Linux网络编程——Socket_第2张图片

       TCP/IP协议最早在BSD UNIX上实现,为TCP/IP协议设计的应用层编程接口称为socket API。本章的主要内容是socket API,主要介绍TCP协议的函数接口,最后介绍UDP协议和UNIX Domain Socket的函数接口。

网络编程接口Linux网络编程——Socket_第3张图片

2 预备知识

2.1 网络字节序

       网络字节顺序是TCP/IP中规定好的一种数据表示格式,它与具体的CPU类型、操作系统等无关,从而可以保证数据在不同主机之间传输时能够被正确解释。网络字节顺序采用big-endian(大端)排序方式。
Linux网络编程——Socket_第4张图片

Linux网络编程——Socket_第5张图片

Linux网络编程——Socket_第6张图片

#include 

uint32_t htonl(uint32_t hostlong);
uint16_t htons(uint16_t hostshort);
uint32_t ntohl(uint32_t netlong);
uint16_t ntohs(uint16_t netshort);

h表示host,n表示network,l表示32位长整数,s表示16位短整数。

       如果主机是小端字节序,这些函数将参数做相应的大小端转换然后返回,如果主机是大端字节序,这些函数不做转换,将参数原封不动地返回。

IP地址转换函数
早期:

#include 
#include 
#include 
int inet_aton(const char *cp, struct in_addr *inp);
in_addr_t inet_addr(const char *cp);
char *inet_ntoa(struct in_addr in);
  • 只能处理IPv4的ip地址
  • 不可重入函数
  • 注意参数是struct in_addr

现在:

#include 
//点分十进制  转换为  网络字节序
int inet_pton(int af, const char *src, void *dst);
//网络字节序  转换为  点分十进制字符串
const char *inet_ntop(int af, const void *src, char *dst, socklen_t size);
  • 支持IPv4和IPv6
  • 可重入函数
  • 其中inet_pton和inet_ntop不仅可以转换IPv4的in_addr,还可以转换IPv6的in6_addr。因此函数接口是void *addrptr。

sockaddr数据结构
       strcut sockaddr 很多网络编程函数诞生早于IPv4协议,那时候都使用的是sockaddr结构体,为了向前兼容,现在sockaddr退化成了(void *)的作用,传递一个地址给函数,至于这个函数是sockaddr_in还是sockaddr_in6,由地址族确定,然后函数内部再强制类型转化为所需的地址类型。
Linux网络编程——Socket_第7张图片

sockaddr数据结构
struct sockaddr {
     
	sa_family_t sa_family; 		/* address family, AF_xxx */
	char sa_data[14];			/* 14 bytes of protocol address */
};

使用 sudo grep -r "struct sockaddr_in {" /usr 命令或者man 7 ip可查看到struct sockaddr_in结构体的定义。一般其默认的存储位置:/usr/include/linux/in.h 文件中。

struct sockaddr_in {
     
	__kernel_sa_family_t sin_family; 			/* Address family */  	地址结构类型
	__be16 sin_port;					 		/* Port number */		端口号
	struct in_addr sin_addr;					/* Internet address */	IP地址
	/* Pad to size of `struct sockaddr'. */
	unsigned char __pad[__SOCK_SIZE__ - sizeof(short int) -
	sizeof(unsigned short int) - sizeof(struct in_addr)];
};

struct in_addr {
     						/* Internet address. */
	__be32 s_addr;
};

struct sockaddr_in6 {
     
	unsigned short int sin6_family; 		/* AF_INET6 */
	__be16 sin6_port; 					/* Transport layer port # */
	__be32 sin6_flowinfo; 				/* IPv6 flow information */
	struct in6_addr sin6_addr;			/* IPv6 address */
	__u32 sin6_scope_id; 				/* scope id (new in RFC2553) */
};

struct in6_addr {
     
	union {
     
		__u8 u6_addr8[16];
		__be16 u6_addr16[8];
		__be32 u6_addr32[4];
	} in6_u;
	#define s6_addr 		in6_u.u6_addr8
	#define s6_addr16 	in6_u.u6_addr16
	#define s6_addr32	 	in6_u.u6_addr32
};

#define UNIX_PATH_MAX 108
	struct sockaddr_un {
     
	__kernel_sa_family_t sun_family; 	/* AF_UNIX */
	char sun_path[UNIX_PATH_MAX]; 	/* pathname */
};

       Pv4和IPv6的地址格式定义在netinet/in.h中,IPv4地址用sockaddr_in结构体表示,包括16位端口号和32位IP地址,IPv6地址用sockaddr_in6结构体表示,包括16位端口号、128位IP地址和一些控制字段。UNIX Domain Socket的地址格式定义在sys/un.h中,用sock-addr_un结构体表示。各种socket地址结构体的开头都是相同的,前16位表示整个结构体的长度(并不是所有UNIX的实现都有长度字段,如Linux就没有),后16位表示地址类型。IPv4、IPv6和Unix Domain Socket的地址类型分别定义为常数AF_INET、AF_INET6、AF_UNIX。这样,只要取得某种sockaddr结构体的首地址,不需要知道具体是哪种类型的sockaddr结构体,就可以根据地址类型字段确定结构体中的内容。因此,socket API可以接受各种类型的sockaddr结构体指针做参数,例如bind、accept、connect等函数,这些函数的参数应该设计成void *类型以便接受各种类型的指针,但是sock API的实现早于ANSI C标准化,那时还没有void *类型,因此这些函数的参数都用struct sockaddr *类型表示,在传递参数之前要强制类型转换一下,例如:

struct sockaddr_in servaddr;
bind(listen_fd, (struct sockaddr *)&servaddr, sizeof(servaddr));		/* initialize servaddr */

3 网络套接字函数

3.1 socket模型创建流程图

Linux网络编程——Socket_第8张图片
server.c
1. socket() 建立套接字
2. bind() 绑定IP 端口号 (struct sockaddr_in addr 初始化)
3. listen() 指定最大同时发起连接数
4. accept() 阻塞等待客户端发起连接
5. read()
6. 小–大
7. write 给 客户端
8. close();

client.c
1. socket();
2. bind(); 可以依赖“隐式绑定”
3. connect();发起连接
4. write();
5. read();
6. close();

3.2 socket函数

#include  /* See NOTES */
#include 
int socket(int domain, int type, int protocol);
  • domain:
    • AF_INET 这是大多数用来产生socket的协议,使用TCP或UDP来传输,用IPv4的地址
    • AF_INET6 与上面类似,不过是来用IPv6的地址
    • AF_UNIX 本地协议,使用在Unix和Linux系统上,一般都是当客户端和服务器在同一台及其上的时候使用
  • type:
    • SOCK_STREAM 这个协议是按照顺序的、可靠的、数据完整的基于字节流的连接。这是一个使用最多的socket类型,这个socket是使用TCP来进行传输。
    • SOCK_DGRAM 这个协议是无连接的、固定长度的传输调用。该协议是不可靠的,使用UDP来进行它的连接。
    • SOCK_SEQPACKET该协议是双线路的、可靠的连接,发送固定长度的数据包进行传输。必须把这个包完整的接受才能进行读取。
    • SOCK_RAW socket类型提供单一的网络访问,这个socket类型使用ICMP公共协议。(ping、traceroute使用该协议)
    • SOCK_RDM 这个类型是很少使用的,在大部分的操作系统上没有实现,它是提供给数据链路层使用,不保证数据包的顺序
  • protocol:
    • 传0 表示使用默认协议。
  • 返回值:
    • 成功:返回指向新创建的socket的文件描述符,失败:返回-1,设置errno

       socket()打开一个网络通讯端口,如果成功的话,就像open()一样返回一个文件描述符,应用程序可以像读写文件一样用read/write在网络上收发数据,如果socket()调用出错则返回-1。对于IPv4,domain参数指定为AF_INET。对于TCP协议,type参数指定为SOCK_STREAM,表示面向流的传输协议。如果是UDP协议,则type参数指定为SOCK_DGRAM,表示面向数据报的传输协议。protocol参数的介绍从略,指定为0即可。

3.3 bind函数

#include  /* See NOTES */
#include 
int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
  • sockfd:
    • socket文件描述符
  • addr:
    • 构造出IP地址加端口号
  • addrlen:
    • sizeof(addr)长度
  • 返回值:
    • 成功返回0,失败返回-1, 设置errno

       服务器程序所监听的网络地址和端口号通常是固定不变的,客户端程序得知服务器程序的地址和端口号后就可以向服务器发起连接,因此服务器需要调用bind绑定一个固定的网络地址和端口号。

       bind()的作用是将参数sockfd和addr绑定在一起,使sockfd这个用于网络通讯的文件描述符监听addr所描述的地址和端口号。前面讲过,struct sockaddr *是一个通用指针类型,addr参数实际上可以接受多种协议的sockaddr结构体,而它们的长度各不相同,所以需要第三个参数addrlen指定结构体的长度。如:

struct sockaddr_in servaddr;
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(6666);

       首先将整个结构体清零,然后设置地址类型为AF_INET,网络地址为INADDR_ANY,这个宏表示本地的任意IP地址,因为服务器可能有多个网卡,每个网卡也可能绑定多个IP地址,这样设置可以在所有的IP地址上监听,直到与某个客户端建立了连接时才确定下来到底用哪个IP地址,端口号为6666。

3.4 listen函数

指定同时建立连接数

#include  /* See NOTES */
#include 
int listen(int sockfd, int backlog);
  • sockfd:
    • socket文件描述符
  • backlog:
    • 排队建立3次握手队列和刚刚建立3次握手队列的链接数和

查看系统默认backlog

cat /proc/sys/net/ipv4/tcp_max_syn_backlog

       典型的服务器程序可以同时服务于多个客户端,当有客户端发起连接时,服务器调用的accept()返回并接受这个连接,如果有大量的客户端发起连接而服务器来不及处理,尚未accept的客户端就处于连接等待状态,listen()声明sockfd处于监听状态,并且最多允许有backlog个客户端处于连接待状态,如果接收到更多的连接请求就忽略。listen()成功返回0,失败返回-1。

3.5 accept函数

#include  		/* See NOTES */
#include 
int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
  • sockdf:
    • socket文件描述符
  • addr:
    • 传出参数,返回链接客户端地址信息,含IP地址和端口号
  • addrlen:
    • 传入传出参数(值-结果),传入sizeof(addr)大小,函数返回时返回真正接收到地址结构体的大小
  • 返回值:
    • 成功返回一个新的socket文件描述符,用于和客户端通信,失败返回-1,设置errno

       三方握手完成后,服务器调用accept()接受连接,如果服务器调用accept()时还没有客户端的连接请求,就阻塞等待直到有客户端连接上来。addr是一个传出参数,accept()返回时传出客户端的地址和端口号。addrlen参数是一个传入传出参数(value-result argument),传入的是调用者提供的缓冲区addr的长度以避免缓冲区溢出问题,传出的是客户端地址结构体的实际长度(有可能没有占满调用者提供的缓冲区)。如果给addr参数传NULL,表示不关心客户端的地址。
       我们的服务器程序结构是这样的:

while (1) {
     
	cliaddr_len = sizeof(cliaddr);
	connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len);
	n = read(connfd, buf, MAXLINE);
	......
	close(connfd);
}

       整个是一个while死循环,每次循环处理一个客户端连接。由于cliaddr_len是传入传出参数,每次调用accept()之前应该重新赋初值。accept()的参数listenfd是先前的监听文件描述符,而accept()的返回值是另外一个文件描述符connfd,之后与客户端之间就通过这个connfd通讯,最后关闭connfd断开连接,而不关闭listenfd,再次回到循环开头listenfd仍然用作accept的参数。accept()成功返回一个文件描述符,出错返回-1。

3.6 connect函数

#include  					/* See NOTES */
#include 
int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
  • sockdf:
    • socket文件描述符
  • addr:
    • 传入参数,指定服务器端地址信息,含IP地址和端口号
  • addrlen:
    • 传入参数,传入sizeof(addr)大小
  • 返回值:
    • 成功返回0,失败返回-1,设置errno

       客户端需要调用connect()连接服务器,connect和bind的参数形式一致,区别在于bind的参数是自己的地址,而connect的参数是对方的地址。connect()成功返回0,出错返回-1。

4 C/S模型-TCP

下图是基于TCP协议的客户端/服务器程序的一般流程:
Linux网络编程——Socket_第9张图片

       服务器调用socket()、bind()、listen()完成初始化后,调用accept()阻塞等待,处于监听端口的状态,客户端调用socket()初始化后,调用connect()发出SYN段并阻塞等待服务器应答,服务器应答一个SYN-ACK段,客户端收到后从connect()返回,同时应答一个ACK段,服务器收到后从accept()返回。

       数据传输的过程:
       建立连接后,TCP协议提供全双工的通信服务,但是一般的客户端/服务器程序的流程是由客户端主动发起请求,服务器被动处理请求,一问一答的方式。因此,服务器从accept()返回后立刻调用read(),读socket就像读管道一样,如果没有数据到达就阻塞等待,这时客户端调用write()发送请求给服务器,服务器收到后从read()返回,对客户端的请求进行处理,在此期间客户端调用read()阻塞等待服务器的应答,服务器调用write()将处理结果发回给客户端,再次调用read()阻塞等待下一条请求,客户端收到后从read()返回,发送下一条请求,如此循环下去。

       如果客户端没有更多的请求了,就调用close()关闭连接,就像写端关闭的管道一样,服务器的read()返回0,这样服务器就知道客户端关闭了连接,也调用close()关闭连接。注意,任何一方调用close()后,连接的两个传输方向都关闭,不能再发送数据了。如果一方调用shutdown()则连接处于半关闭状态,仍可接收对方发来的数据。

       在学习socket API时要注意应用程序和TCP协议层是如何交互的: 应用程序调用某个socket函数时TCP协议层完成什么动作,比如调用connect()会发出SYN段 应用程序如何知道TCP协议层的状态变化,比如从某个阻塞的socket函数返回就表明TCP协议收到了某些段,再比如read()返回0就表明收到了FIN段

4.1 server

       下面通过最简单的客户端/服务器程序的实例来学习socket API。
       server.cpp的作用是从客户端读字符,然后将每个字符转换为大写并回送给客户端。(没有写出错处理)

#include 
#include 
#include 
#include 
#include //toupper
#include 
#include 
using namespace std;
#define SER_PORT 6666
#define SER_IP "127.0.0.1"
int main(){
     
    int lfd,cfd;
    struct sockaddr_in ser_addr,clie_addr;
    socklen_t clie_addr_len,clie_IP_len;
    char buf[BUFSIZ],clie_IP[BUFSIZ];
    //创建一个socket,指定IPv4协议族 TCP协议
    lfd = socket(AF_INET,SOCK_STREAM,0);//此套接字不参与与客户端的通信
    //初始化一个地址结构 man 7 ip 可查看对应信息
    ser_addr.sin_family = AF_INET;//IPv4
    ser_addr.sin_port = htons(SER_PORT);//绑定端口号;默认为小端法,调用hons函数转换格式
    ser_addr.sin_addr.s_addr = htonl(INADDR_ANY);//监听本地所有IP地址,宏INADDR_ANY可以绑定本地的任意一个有效ID
    //绑定服务器地址结构
    bind(lfd,(struct sockaddr*)&ser_addr,sizeof(ser_addr));
    //设定链接上限,注意此处不阻塞
    listen(lfd,20);//同一时刻允许向服务器发起链接请求的个数
    cout <<"Accteping connections...." << endl;
    clie_addr_len = sizeof(clie_addr);
    cfd = accept(lfd,(struct sockaddr*)&clie_addr,&clie_addr_len);//返回的是与客户端通信的套接字文件描述符

    cout << " client IP: " 
        << inet_ntop(AF_INET,&clie_addr.sin_addr.s_addr,clie_IP,sizeof(clie_IP))
        << ", cline PORT: " 
        << ntohs(clie_addr.sin_port) << endl;

    int n;

    while(1){
     
        n = read(cfd,buf,sizeof(buf));
        for(int i = 0;i < n ;i++){
     
            buf[i] = toupper(buf[i]);
        }

        write(cfd,buf,n);
    }
    close(lfd);
    close(cfd);

    return 0;
}

4.2 client

       client.cpp的作用是从命令行参数中获得一个字符串发给服务器,然后接收服务器返回的字符串并打印。(没有写出错处理)

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

#define SERV_PORT 6666
#define SERV_IP "127.0.0.1"

int main(){
     
    int cfd;
    struct sockaddr_in serv_addr;
    socklen_t serv_addr_len;
    char buf[BUFSIZ];

    cfd = socket(AF_INET,SOCK_STREAM,0);

    //memset(&serv_addr,0,sizeof(serv_addr));//把结构体清空
    bzero(&serv_addr,sizeof(serv_addr));//把结构体清空

    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(SERV_PORT);
    inet_pton(AF_INET,SERV_IP,&serv_addr.sin_addr.s_addr);

    connect(cfd,(struct sockaddr*)&serv_addr,sizeof(serv_addr));

    while(1){
     
        //从键盘获取数据,写到服务器
        fgets(buf,sizeof(buf),stdin);
        write(cfd,buf,strlen(buf));
        //从服务器读数据然后输出到屏幕
        int n = read(cfd,buf,sizeof(buf));
        write(STDOUT_FILENO,buf,n);
    }
    close(cfd);

    return 0;
}

       由于客户端不需要固定的端口号,因此不必调用bind(),客户端的端口号由内核自动分配。注意,客户端不是不允许调用bind(),只是没有必要调用bind()固定一个端口号,服务器也不是必须调用bind(),但如果服务器不调用bind(),内核会自动给服务器分配监听端口,每次启动服务器时端口号都不一样,客户端要连接服务器就会遇到麻烦。
       客户端和服务器启动后可以使用netstat命令查看链接情况:

netstat -apn|grep 6666

5 出错处理封装函数

       上面的例子不仅功能简单,而且简单到几乎没有什么错误处理,我们知道,系统调用不能保证每次都成功,必须进行出错处理,这样一方面可以保证程序逻辑正常,另一方面可以迅速得到故障信息。

       为使错误处理的代码不影响主程序的可读性,我们把与socket相关的一些系统函数加上错误处理代码包装成新的函数,做成一个模块wrap.cpp:

5.1 wrap.cpp

#include 
#include 
#include 
void perr_exit(const char *s)
{
     
	perror(s);
	exit(1);
}
int Accept(int fd, struct sockaddr *sa, socklen_t *salenptr)
{
     
	int n;
	again:
	if ( (n = accept(fd, sa, salenptr)) < 0) {
     
		if ((errno == ECONNABORTED) || (errno == EINTR))
			goto again;
		else
			perr_exit("accept error");
	}
	return n;
}
int Bind(int fd, const struct sockaddr *sa, socklen_t salen)
{
     
	int n;
	if ((n = bind(fd, sa, salen)) < 0)
		perr_exit("bind error");
	return n;
}
int Connect(int fd, const struct sockaddr *sa, socklen_t salen)
{
     
	int n;
	if ((n = connect(fd, sa, salen)) < 0)
		perr_exit("connect error");
	return n;
}
int Listen(int fd, int backlog)
{
     
	int n;
	if ((n = listen(fd, backlog)) < 0)
		perr_exit("listen error");
	return n;
}
int Socket(int family, int type, int protocol)
{
     
	int n;
	if ( (n = socket(family, type, protocol)) < 0)
		perr_exit("socket error");
	return n;
}
ssize_t Read(int fd, void *ptr, size_t nbytes)
{
     
	ssize_t n;
again:
	if ( (n = read(fd, ptr, nbytes)) == -1) {
     
		if (errno == EINTR)
			goto again;
		else
			return -1;
	}
	return n;
}
ssize_t Write(int fd, const void *ptr, size_t nbytes)
{
     
	ssize_t n;
again:
	if ( (n = write(fd, ptr, nbytes)) == -1) {
     
		if (errno == EINTR)
			goto again;
		else
			return -1;
	}
	return n;
}
int Close(int fd)
{
     
	int n;
	if ((n = close(fd)) == -1)
		perr_exit("close error");
	return n;
}
ssize_t Readn(int fd, void *vptr, size_t n)
{
     
	size_t nleft;
	ssize_t nread;
	char *ptr;

	ptr = vptr;
	nleft = n;

	while (nleft > 0) {
     
		if ( (nread = read(fd, ptr, nleft)) < 0) {
     
			if (errno == EINTR)
				nread = 0;
			else
				return -1;
		} else if (nread == 0)
			break;
		nleft -= nread;
		ptr += nread;
	}
	return n - nleft;
}

ssize_t Writen(int fd, const void *vptr, size_t n)
{
     
	size_t nleft;
	ssize_t nwritten;
	const char *ptr;

	ptr = vptr;
	nleft = n;

	while (nleft > 0) {
     
		if ( (nwritten = write(fd, ptr, nleft)) <= 0) {
     
			if (nwritten < 0 && errno == EINTR)
				nwritten = 0;
			else
				return -1;
		}
		nleft -= nwritten;
		ptr += nwritten;
	}
	return n;
}

static ssize_t my_read(int fd, char *ptr)
{
     
	static int read_cnt;
	static char *read_ptr;
	static char read_buf[100];

	if (read_cnt <= 0) {
     
again:
		if ((read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0) {
     
			if (errno == EINTR)
				goto again;
			return -1;	
		} else if (read_cnt == 0)
			return 0;
		read_ptr = read_buf;
	}
	read_cnt--;
	*ptr = *read_ptr++;
	return 1;
}

ssize_t Readline(int fd, void *vptr, size_t maxlen)
{
     
	ssize_t n, rc;
	char c, *ptr;
	ptr = vptr;

	for (n = 1; n < maxlen; n++) {
     
		if ( (rc = my_read(fd, &c)) == 1) {
     
			*ptr++ = c;
			if (c == '\n')
				break;
		} else if (rc == 0) {
     
			*ptr = 0;
			return n - 1;
		} else
			return -1;
	}
	*ptr = 0;
	return n;
}

5.2 wrap.h

#ifndef __WRAP_H_
#define __WRAP_H_
void perr_exit(const char *s);
int Accept(int fd, struct sockaddr *sa, socklen_t *salenptr);
int Bind(int fd, const struct sockaddr *sa, socklen_t salen);
int Connect(int fd, const struct sockaddr *sa, socklen_t salen);
int Listen(int fd, int backlog);
int Socket(int family, int type, int protocol);
ssize_t Read(int fd, void *ptr, size_t nbytes);
ssize_t Write(int fd, const void *ptr, size_t nbytes);
int Close(int fd);
ssize_t Readn(int fd, void *vptr, size_t n);
ssize_t Writen(int fd, const void *vptr, size_t n);
ssize_t my_read(int fd, char *ptr);
ssize_t Readline(int fd, void *vptr, size_t maxlen);
#endif

你可能感兴趣的:(Linux系统编程,linux,网络编程,socket)