poll函数

代码实现
makefile

src=$(wildcard *.c)                                                             
obj=$(patsubst %.c,%.c,$(src))

all:server client

server: server.o wrap.o
        gcc server.o wrap.o -o server -Wall
client: client.o wrap.o
        gcc client.o wrap.o -o client -Wall

%.c:%.c
        gcc -c $< -Wall

.PHONY: clean all 
clean:
        -rm -rf server client $(obj)

出错处理函数头文件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);
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

出错处理函数wrap.c

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <sys/socket.h>

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;
	printf("----------------in wrap before connect() \n");
	n=connect(fd,sa,salen);
	printf("----------------in wrap connect() return %d\n",n);
	if(n<0){
		perr_exit("connect error");
	}

	return 0;
}

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;//usigned int 剩余未读取的字节数
	ssize_t nread;//int 实际读到的字节数
	char *ptr;

	ptr=vptr;
	nleft=n;//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;
}

/*readline --- fgets*/
//传出参数 vptr
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;
}

服务端函数server.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#include <errno.h>
#include <ctype.h>

#include "wrap.h"

#define MAXLINE 80
#define SERV_PORT 8000
#define OPEN_MAX 1024

int main(int argc,char *argv[])
{
	int i,j,maxi,listenfd,connfd,sockfd;
	int nready;	/*接收poll返回值,记录满足监听事件的fd个数*/
	ssize_t n;

	char buf[MAXLINE],str[INET_ADDRSTRLEN];
	socklen_t clilen;
	struct pollfd client[OPEN_MAX];
	struct sockaddr_in cliaddr,servaddr;

	listenfd=Socket(AF_INET,SOCK_STREAM,0);

	int opt=1;
	setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));

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

	Bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr));
	Listen(listenfd,128);

	client[0].fd=listenfd;		/*要监听的第一个文件描述符 存入client[0]*/
	client[0].events=POLLIN;	/*listened监听普通读事件*/

	for(i=1;i<OPEN_MAX;i++)	
		client[i].fd=-1;	/*用-1初始化client[]里剩下元素0也是文件描述符,不能用*/

	maxi=0;				/*client[]数组有效元素中最大元素下标*/

	for(;;){
		nready=poll(client,maxi+1,-1);	/*阻塞监听是否有客户端链接请求*/
		if(client[0].revents & POLLIN){	/*listenfd有读事件就绪*/
			clilen=sizeof(cliaddr);
			connfd=Accept(listenfd,(struct sockaddr *)&cliaddr,&clilen);	/*接收客户端请求Accept不会阻塞*/
			printf("received from %s at PORT %d\n",
					inet_ntop(AF_INET,&cliaddr.sin_addr,str,sizeof(str)),
					ntohs(cliaddr.sin_port));

			for(i=1;i<OPEN_MAX;i++)
				if(client[i].fd<0){
					client[i].fd=connfd;	/*找到client[]中空闲的位置,存放accept返回的connfd*/
					break;
				}

			if(i==OPEN_MAX)		/*达到了最大客户端数*/
				perr_exit("too many clients");

			client[i].events=POLLIN;	/*设置刚刚返回的connfd,监控读事件*/
			if(i>maxi)
				maxi=i;			/*更新client[]中最大元素下标*/
			if(--nready<=0)
				continue;		/*没有更多就绪事件时,继续回到poll阻塞*/
		}
		for(i=1;i<maxi;i++){			/*前面的if没满足,说明没有listenfd满足,检测client[]看是哪个connfd就绪*/
			if((sockfd=client[i].fd)<0)
				continue;
			if(client[i].revents&POLLIN){
				if((n=Read(sockfd,buf,MAXLINE))<0){
					if(errno==ECONNRESET){	/*收到RST标志*/
						printf("client[%d] aborted connection\n",i);
						Close(sockfd);
						client[i].fd=-1;	/*poll中不监控该文件描述符,直接置为-1即可,不用像select中那样移除*/
					}else
						perr_exit("read error");	

				}else if(n==0){				/*说明客户端先关闭链接*/
					printf("client[%d] closed connection\n",i);
					Close(sockfd);
					client[i].fd=-1;
				}else{
					for(j=0;j<n;j++)
						buf[j]=toupper(buf[j]);
					Writen(sockfd,buf,n);
				}
				if(--nready<=0)
					break;
			}
		}
	}
	return 0;
}

在终端输入:

make
./server

再启动一个终端连接服务端

nc 127.0.0.1 8000
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值