您的位置:首页 > 运维架构 > Linux

linux socket的IO多路复用简单例子(一)

2016-06-13 20:01 387 查看
IO多路复用包括select、poll、epoll,是指通过一种机制,监视多个文件描述符,一旦某个描述符就绪(io描述符就绪一般是读就绪或者写就绪),通知应用程序进行相应的读写操作。对比于使用线程或者进程来接收或者发送信息的socket编程,有更高的性能。

直接上代码:

linux下的socket编程:

server端:

/*
socket编程
用户进程-tcp/udp应用层-ip/ICMP/IGMP网络层-链路层
socket是api 处于用户进程与应用层之间,减少

socket()-bind()-listen()-accect()-read()-write()
*/

#include <sys/socket.h>
#include <sys/types.h>
#include <iostream>
#include <netinet/in.h>
#include <string.h>
#include <stdlib.h>
#include <sys/shm.h>
#include <unistd.h>
#include <stdio.h>
using namespace std;

int main()
{
/*
create new socket
函数原型int socket (int __domain, int __type, int __protocol)
__domain: 地址描述符,表示使用哪种ip协议,目前仅支持IPV4
__type:通信类型。常用的有:流模式(SOCK_STREAM)、数据报模式(SOCK_DGRAM)
__protocol:通信协议。常用的有:IPPROTO_TCP(TCP协议)、IPPROTO_UDP(UDP协议)、IPPROTO_STCP(STCP协议)等。
当为0时,随机使用某一种协议。
Returns a file descriptor for the new socket, or -1 for errors
*/
int socket_fd = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);

/*
通过结构体sockaddr_in设置socket连接的属性
struct sockaddr_in
{
short sin_family;Address family.socket编程只能是AF_INET(TCP/IP协议组)
unsigned short sin_port;
struct in_addr sin_arrd;网络字节序
}
*/
struct sockaddr_in socket_attr;
socket_attr.sin_family=AF_INET;
socket_attr.sin_port=htons(5875);
//从主机字节顺序转换成网络字节序
socket_attr.sin_addr.s_addr=htonl(INADDR_ANY);

/*
将套接字与地址相关联
函数原型bind (int __fd, __CONST_SOCKADDR_ARG __addr, socklen_t __len)
__fd:与哪一个套接字绑定
__addr:关联的地址
__len:当内核复制或传递地址给驱动的时候,它依据这个值来确定需要复制多少数据.
返回0表示成功,-1表示失败
*/
if(bind(socket_fd,(struct sockaddr*)&socket_attr,sizeof(socket_attr)) == -1)
{
perror("bind");
cout<<"bind socket error"<<endl;
return 0;
}
/*
监听端口
函数原型int listen (int __fd, int __n)
__fd:监听的套接字
__n:连接的上限。当服务器接收的客户端连接达n后,拒接收后面的连接
Returns 0 on success, -1 for errors.
*/
if(listen(socket_fd,20)==-1)
{
cout<<"listen error "<<endl;
return 0;
}

char buffer[1024]={0};
struct sockaddr_in client_addr;
/*
int accept (int __fd, __SOCKADDR_ARG __addr,socklen_t *__restrict __addr_len)
Await a connection on socket FD.
When a connection arrives, open a new socket to communicate with it,
set *ADDR (which is *ADDR_LEN bytes long) to the address of the connecting
peer and *ADDR_LEN to the address's actual length, and return the
new socket's descriptor, or -1 for errors.
*/
socklen_t length=sizeof(client_addr);
int conn = accept(socket_fd,(struct sockaddr*)&client_addr,&length);
if(conn ==-1)
{
cout<<"accect error"<<endl;
return 0;
}

while(1)
{
memset(buffer,0,sizeof(buffer)/sizeof(char));
/*
ssize_t recv (int __fd, void *__buf, size_t __n, int __flags)
Read N bytes into BUF from socket FD.
Returns the number read or -1 for errors.
*/
if(recv(conn,buffer,sizeof(buffer),0) == -1)
{
cout<<"recv error"<<endl;
//return;
}
if(send(conn,buffer,sizeof(buffer),0)==-1)
{
cout<<"send error"<<endl;
}
cout<<buffer<<endl;
}
close(conn);
close(socket_fd);
return 0;
}


client端:

#include <iostream>
#include <stdio.h>
#include <sys/socket.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
using namespace std;

int main()
{
int socket_fd = socket(AF_INET,SOCK_STREAM,0);

struct sockaddr_in kk;
kk.sin_family=AF_INET;
kk.sin_port=htons(5875);
kk.sin_addr.s_addr = inet_addr("127.0.0.1");

if(connect(socket_fd,(struct sockaddr*)&kk,sizeof(kk)) <0)
{
perror("connect error");
cout<<"connect error"<<endl;
return 0;
}

char readbuf[1024];
char sendbuf[1024];
while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
{
int  ret =send(socket_fd,sendbuf,strlen(sendbuf),0);
if(ret <0)
{
perror("send error");
}
recv(socket_fd,readbuf,sizeof(readbuf),0);
fputs(readbuf,stdout);
memset(sendbuf,0,sizeof(sendbuf));
memset(readbuf,0,sizeof(readbuf));
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: