service.c

LINUX下的一个TCP服务器小程序。学习unix网络编程的朋友可以下来参考参考

#include <stdio.h>
#include <stdlib.h>
#include <netdb.h>
#include <string.h>
#include <error.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

#define PORT 8088
#define BACKLOG 5
#define MAXSIZE 1024

int main()
{
	int sin_size;
	int nbytes = 0;
	int total = 0;
	char buf_read[MAXSIZE] = {0};
	char buf_write[MAXSIZE] = {0};
	int sockfd,new_fd;
	struct sockaddr_in srvaddr, cliaddr;
		
	sockfd = socket(AF_INET,SOCK_STREAM,0);
	if(sockfd < 0)
		{
			perror("Creat socket error!\n");
			exit(1);
		}

	bzero(&srvaddr,sizeof(srvaddr));
	srvaddr.sin_family = AF_INET;
	srvaddr.sin_port = htons(PORT);
	srvaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	
	if((bind(sockfd,(struct sockaddr *)&srvaddr,sizeof(struct sockaddr))) == -1)
		{
			perror("Bind error!\n");
			exit(1);
		}
		
	if(listen(sockfd,BACKLOG) < 0)
		
		{
			perror("Listen error!\n");
			exit(1);
		}

	while(1)
	{
		sin_size = sizeof(struct sockaddr_in);
		bzero(&cliaddr,sizeof(cliaddr));
		new_fd = accept(sockfd,(struct sockaddr *)&cliaddr,&sin_size);
		if(new_fd < 0)
		{
			perror("Accept error!\n");
			continue;
		}
		
		printf("Server:got connection from %s \n",inet_ntoa(cliaddr.sin_addr));		
		while(1)
		{
			if((nbytes = read(new_fd,buf_read,MAXSIZE)) > 0)
			{				
				printf("CLIEENT: %s",buf_read);			
			}	
			
			printf("SERVICE: ");			
			if(fgets(buf_write,MAXSIZE,stdin) != NULL)
			{
				write(new_fd,buf_write,MAXSIZE);
			}					
		}
		close(new_fd);
	}
	close(sockfd);
}