Linux网络编程:4. 完整的读写函数

4.1 写函数write

[code lang="c" light="true"]ssize_t write(int fd,const void *buf,size_t nbytes)[/code]

write 函数将 buf 中的 nbytes 字节内容写入文件描述符 fd。成功时返回写的字节数,失败时返回 -1,并设置 errno 变量。

在网络程序中,当我们向套接字文件描述符写时有俩种可能:
1) write 的返回值大于0,表示写了部分或者是全部的数据;
2) 返回的值小于0,此时出现了错误,我们要根据错误类型来处理。

如果错误为 EINTR 表示在写的时候出现了中断错误。如果为 EPIPE 表示网络连接出现了问题(对方已经关闭了连接)。

为了处理以上的情况,我们自己编写一个写函数:

[code lang="c"]
int my_write(int fd, void *buffer, int length)
{
int bytes_left;
int written_bytes;
char *ptr;</span>

ptr=buffer;
bytes_left=length;
while(bytes_left>0)
{
/* 开始写 */
written_bytes=write(fd, ptr, bytes_left);
if(written_bytes<=0) /* 出错了*/
{
if(errno==EINTR) /* 中断错误 我们继续写 */
written_bytes=0;
else /* 其他错误 没有办法,只好撤退了 */
return(-1);
}
bytes_left-=written_bytes;
ptr+=written_bytes; /* 从剩下的地方继续写 */
}
return(0);
}
[/code]

4.2 读函数read

[code lang="c" light="true"]ssize_t read(int fd,void *buf,size_t nbyte)[/code]

read 函数是负责从 fd 中读取内容。当读成功时,read 返回实际所读的字节数。如果返回的值是0,表示已经读到文件的结束了。小于0表示出现了错误。如果错误为 EINTR 说明读是由中断引起的,如果是 ECONNREST 表示网络连接出了问题。

和上面一样,我们也写一个自己的读函数:

[code lang="c"]
int my_read(int fd, void *buffer, int length)
{
int bytes_left;
int bytes_read;
char *ptr;</span>

bytes_left=length;
while(bytes_left>0)
{
bytes_read=read(fd,ptr,bytes_read);
if(bytes_read<0)
{
if(errno==EINTR)
bytes_read=0;
else
return(-1);
}
else if(bytes_read==0)
break;
bytes_left-=bytes_read;
ptr+=bytes_read;
}
return(length-bytes_left);
}
[/code]

4.3 数据的传递

有了上面的两个函数,我们就可以向客户端或者是服务端传递数据了。比如我们要传递一个结构。可以使用如下方式:

[code lang="c"]
/* 客户端向服务端写 */
struct my_struct my_struct_client;
write(fd,(void *)&my_struct_client,sizeof(struct my_struct);
[/code]

[code lang="c"]
/* 服务端的读 */
char buffer[sizeof(struct my_struct)];
struct *my_struct_server;
read(fd,(void *)buffer,sizeof(struct my_struct));
my_struct_server=(struct my_struct *)buffer;
[/code]

在网络上传递数据时我们一般都是把数据转化为 char 类型的数据传递。接收的时候也是一样的。注意的是我们没有必要在网络上传递指针(因为传递指针是没有任何意义的),我们必须传递指针所指向的内容。

添加新评论